Files
guestfs-inspect/guestfs-inspectd.c
2024-03-02 22:24:11 -05:00

107 lines
3.1 KiB
C

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <czmq.h>
#include <zactor.h>
#include <guestfs.h>
static int compare_key_len(const void *p1, const void *p2) {
const char *key1 = *(char* const*) p1;
const char *key2 = *(char* const*) p2;
return strlen(key1) - strlen(key2);
}
static int count_mountpoints(char *const *argv) {
size_t c;
for (c = 0; argv[c]; c++) {}
return c;
}
static void worker_task(char *disk_path, char *file_path) {
guestfs_h *g;
char **roots, **mountpoints;
char *root;
size_t i, j;
char *file_content;
size_t file_size;
g = guestfs_create();
if (g == NULL) {
exit(EXIT_FAILURE);
}
if (guestfs_add_drive_opts(g, disk_path,
GUESTFS_ADD_DRIVE_OPTS_READONLY, 1,
-1) == -1) {
exit(EXIT_FAILURE);
}
if (guestfs_launch(g) == -1) {
exit(EXIT_FAILURE);
}
roots = guestfs_inspect_os(g);
if (roots == NULL) {
exit(EXIT_FAILURE);
}
for (j = 0; roots[j] != NULL; j++) {
root = roots[j];
mountpoints = guestfs_inspect_get_mountpoints(g, root);
if (mountpoints == NULL) {
exit(EXIT_FAILURE);
}
qsort(mountpoints, count_mountpoints(mountpoints) / 2, 2 * sizeof (char*), compare_key_len);
for (i = 0; mountpoints[i] != NULL; i += 2) {
guestfs_mount_ro(g, mountpoints[i+1], mountpoints[i]);
free(mountpoints[i]);
free(mountpoints[i+1]);
}
free(mountpoints);
if (guestfs_is_file_opts(g, file_path, GUESTFS_IS_FILE_OPTS_FOLLOWSYMLINKS, 1, -1) > 0) {
printf("--- %s ---\n", file_path);
file_content = guestfs_read_file(g, file_path, &file_size);
if (file_content == NULL) {
exit(EXIT_FAILURE);
}
printf("%s\n", file_content);
free(file_content);
}
}
return;
}
char *endpoint(void);
int main(int argc, char **argv) {
if (argc < 2) {
printf("Usage: %s <disk-path> ...\n", argv[0]);
return EXIT_FAILURE;
}
const int worker_count = argc - 1;
// One worker per disk image.
zactor_t *workers[worker_count];
for (int i = 0; i < worker_count; i++) {
char *path = strtok(argv[i+1], ":");
char *name = strtok(NULL, ":");
workers[i] = zactor_new(worker_task, (void *) &((struct { char* path; char *name; }) {.path = path, .name = name}));
}
return EXIT_SUCCESS;
}
char *endpoint() {
// TODO: This should be based on GUESTFS_INSPECT_ENDPOINT, or XDG_RUNTIME_DIR if the former is not set, and default to a sensible path if neither are set.
const char* ep = "ipc:///tmp/guestfs-inspect.sock";
char *res = malloc(strlen(ep) + 1);
strcpy(res, ep);
return res;
}