123 lines
3.4 KiB
C
123 lines
3.4 KiB
C
#include <czmq.h>
|
|
|
|
#include "libguestfs-inspect.h"
|
|
|
|
char *guestfs_inspect_endpoint() {
|
|
const char *guestfs_inspect_endpoint = getenv("GUESTFS_INSPECT_ENDPOINT");
|
|
const char *socket_file_name = "/guestfs-inspect.sock";
|
|
char *res = NULL;
|
|
if (guestfs_inspect_endpoint) {
|
|
res = malloc(strlen(guestfs_inspect_endpoint) + 1);
|
|
strcpy(res, guestfs_inspect_endpoint);
|
|
} else {
|
|
const char *xdg_runtime_dir = getenv("XDG_RUNTIME_DIR");
|
|
if (xdg_runtime_dir) {
|
|
res = malloc(strlen(xdg_runtime_dir) + strlen(socket_file_name) + strlen("ipc://") + 1);
|
|
strcpy(res, "ipc://");
|
|
strcat(res, xdg_runtime_dir);
|
|
strcat(res, socket_file_name);
|
|
} else {
|
|
const char *tmp_dir = "/tmp";
|
|
res = malloc(strlen(tmp_dir) + strlen(socket_file_name) + strlen("ipc://") + 1);
|
|
strcpy(res, "ipc://");
|
|
strcat(res, tmp_dir);
|
|
strcat(res, socket_file_name);
|
|
}
|
|
}
|
|
return res;
|
|
}
|
|
|
|
zmsg_t *guestfs_inspect_command_to_zmsg(const struct guestfs_inpsect_command *command) {
|
|
zmsg_t *res = zmsg_new();
|
|
|
|
zmsg_addstr(res, command->name);
|
|
zmsg_addmem(res, &(command->command), sizeof(command->command));
|
|
|
|
switch (command->command) {
|
|
case GUESTFS_COMMAND_LS:
|
|
zmsg_addmem(res, &(command->args.ls.paths_length), sizeof(command->args.ls.paths_length));
|
|
|
|
for (size_t i = 0; i < (command->args.ls.paths_length); i++) {
|
|
zmsg_addstr(res, command->args.ls.paths[i]);
|
|
}
|
|
break;
|
|
case GUESTFS_COMMAND_CAT:
|
|
zmsg_addmem(res, &(command->args.cat.paths_length), sizeof(command->args.cat.paths_length));
|
|
|
|
for (size_t i = 0; i < (command->args.cat.paths_length); i++) {
|
|
zmsg_addstr(res, command->args.cat.paths[i]);
|
|
}
|
|
break;
|
|
}
|
|
|
|
return res;
|
|
}
|
|
|
|
struct guestfs_inpsect_command *guestfs_inspect_zmsg_to_command(zmsg_t *external_msg) {
|
|
// This way we do not modify the argument we are given.
|
|
zmsg_t *msg = zmsg_dup(external_msg);
|
|
struct guestfs_inpsect_command *res = malloc(sizeof(struct guestfs_inpsect_command));
|
|
|
|
if (!(zmsg_size(msg) > 2)) {
|
|
free(res);
|
|
zmsg_destroy(&msg);
|
|
return NULL;
|
|
}
|
|
|
|
res->name = zmsg_popstr(msg);
|
|
zframe_t *next = zmsg_pop(msg);
|
|
res->command = *(zframe_data(next));
|
|
zframe_destroy(&next);
|
|
|
|
switch (res->command) {
|
|
case GUESTFS_COMMAND_LS:
|
|
next = zmsg_pop(msg);
|
|
res->args.ls.paths_length = *(zframe_data(next));
|
|
free(next);
|
|
|
|
res->args.ls.paths = calloc(res->args.ls.paths_length, sizeof(char *));
|
|
|
|
for (size_t i = 0; i < res->args.ls.paths_length; i++) {
|
|
res->args.ls.paths[i] = zmsg_popstr(msg);
|
|
}
|
|
break;
|
|
case GUESTFS_COMMAND_CAT:
|
|
next = zmsg_pop(msg);
|
|
res->args.cat.paths_length = *(zframe_data(next));
|
|
free(next);
|
|
|
|
res->args.cat.paths = calloc(res->args.cat.paths_length, sizeof(char *));
|
|
|
|
for (size_t i = 0; i < res->args.cat.paths_length; i++) {
|
|
res->args.cat.paths[i] = zmsg_popstr(msg);
|
|
}
|
|
break;
|
|
default:
|
|
printf("Command not found\n");
|
|
}
|
|
zmsg_destroy(&msg);
|
|
|
|
return res;
|
|
}
|
|
|
|
void guestfs_inspect_command_destroy(struct guestfs_inpsect_command **c) {
|
|
free((*c)->name);
|
|
switch ((*c)->command) {
|
|
case GUESTFS_COMMAND_LS:
|
|
for (size_t i = 0; i < (*c)->args.ls.paths_length; i++) {
|
|
free((*c)->args.ls.paths[i]);
|
|
}
|
|
free((*c)->args.ls.paths);
|
|
break;
|
|
case GUESTFS_COMMAND_CAT:
|
|
for (size_t i = 0; i < (*c)->args.cat.paths_length; i++) {
|
|
free((*c)->args.cat.paths[i]);
|
|
}
|
|
free((*c)->args.cat.paths);
|
|
break;
|
|
}
|
|
|
|
free(*c);
|
|
*c = NULL;
|
|
}
|