diff --git a/guestfs-inspectd.c b/guestfs-inspectd.c index 916b11b..6ce5d04 100644 --- a/guestfs-inspectd.c +++ b/guestfs-inspectd.c @@ -26,8 +26,17 @@ static int count_mountpoints(char *const *argv) { return c; } -static void *worker_task(zsock_t *pipe, char *disk_path) { - guestfs_h *g; +static void cat_file(guestfs_h *g, char *file_path, char **file_content, size_t *file_size) { + if (guestfs_is_file_opts(g, file_path, GUESTFS_IS_FILE_OPTS_FOLLOWSYMLINKS, 1, -1) > 0) { + (*file_content) = guestfs_read_file(g, file_path, file_size); + if ((*file_content) == NULL) { + exit(EXIT_FAILURE); + } + } + return; +} + +static void init_guestfs(guestfs_h **g, char *disk_path) { char **roots, **mountpoints; char *root; size_t i, j; @@ -36,23 +45,23 @@ static void *worker_task(zsock_t *pipe, char *disk_path) { //size_t file_size; // Create a connection handle - g = guestfs_create(); + (*g) = guestfs_create(); if (g == NULL) { exit(EXIT_FAILURE); } // Adding disk_path to connection handle - if (guestfs_add_drive_opts(g, disk_path, GUESTFS_ADD_DRIVE_OPTS_READONLY, 1, -1) == -1) { + if (guestfs_add_drive_opts((*g), disk_path, GUESTFS_ADD_DRIVE_OPTS_READONLY, 1, -1) == -1) { exit(EXIT_FAILURE); } // Launching connection handle - if (guestfs_launch(g) == -1) { + if (guestfs_launch((*g)) == -1) { exit(EXIT_FAILURE); } // Pulling rootfs information - roots = guestfs_inspect_os(g); + roots = guestfs_inspect_os((*g)); if (roots == NULL) { exit(EXIT_FAILURE); } @@ -60,20 +69,27 @@ static void *worker_task(zsock_t *pipe, char *disk_path) { // Looping through roots to mount mountpoints for (j = 0; roots[j] != NULL; j++) { root = roots[j]; - mountpoints = guestfs_inspect_get_mountpoints(g, root); + mountpoints = guestfs_inspect_get_mountpoints((*g), root); if (mountpoints == NULL) { exit(EXIT_FAILURE); } // Sorting mountpoints to be in {'${device_path}', '${mount_path}'} format 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]); + guestfs_mount_ro((*g), mountpoints[i+1], mountpoints[i]); free(mountpoints[i]); free(mountpoints[i+1]); } free(mountpoints); free(root); } + free(roots); +} + +static void *worker_task(zsock_t *pipe, char *disk_path) { + guestfs_h *g = NULL; + + init_guestfs(&g, disk_path); // ZeroMQ Opens here zsock_signal(pipe, 0); @@ -90,16 +106,7 @@ static void *worker_task(zsock_t *pipe, char *disk_path) { zmsg_send(&msg, pipe); zmsg_destroy(&msg); } - /*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); - }*/ - free(roots); + guestfs_close(g); exit(EXIT_SUCCESS); }