Uncomment worker loop

This commit is contained in:
2024-03-03 01:28:11 -05:00

View File

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