yes i try i
This commit is contained in:
@@ -1,6 +1,11 @@
|
|||||||
FROM debian
|
FROM debian as builder
|
||||||
|
|
||||||
RUN apt update && apt upgrade -y
|
RUN apt update && apt upgrade -y
|
||||||
RUN apt install -y libczmq-dev libguestfs-dev gcc valgrind gdb make pkg-config
|
RUN apt install -y libczmq-dev libguestfs-dev gcc valgrind gdb make pkg-config
|
||||||
|
|
||||||
WORKDIR /opt
|
COPY . /opt
|
||||||
|
WORKDIR /opt
|
||||||
|
|
||||||
|
FROM builder as compile
|
||||||
|
|
||||||
|
RUN make
|
||||||
|
|||||||
1
Makefile
1
Makefile
@@ -1,4 +1,5 @@
|
|||||||
build: daemon client
|
build: daemon client
|
||||||
|
debug: daemon-debug client-debug
|
||||||
|
|
||||||
daemon:
|
daemon:
|
||||||
gcc -ggdb3 -Wall guestfs-inspectd.c libguestfs-inspect.c -o guestfs-inspectd `pkg-config libguestfs libczmq --cflags --libs`
|
gcc -ggdb3 -Wall guestfs-inspectd.c libguestfs-inspect.c -o guestfs-inspectd `pkg-config libguestfs libczmq --cflags --libs`
|
||||||
|
|||||||
@@ -94,21 +94,20 @@ static void worker_task(zsock_t *pipe, char *disk_path) {
|
|||||||
|
|
||||||
g = init_guestfs(disk_path);
|
g = init_guestfs(disk_path);
|
||||||
|
|
||||||
|
zsock_t *worker = zsock_new_dealer("inproc://workers");
|
||||||
|
|
||||||
// ZeroMQ Opens here
|
// ZeroMQ Opens here
|
||||||
zsock_signal(pipe, 0);
|
zsock_signal(pipe, 0);
|
||||||
|
|
||||||
while (true) {
|
while (true) {
|
||||||
zmsg_t *msg = zmsg_recv(pipe);
|
zmsg_t *msg = zmsg_recv(pipe);
|
||||||
if (!msg) {
|
if (!msg) {
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
// Process message
|
|
||||||
|
|
||||||
// do something here
|
|
||||||
printf("Received a message in the worker\n");
|
printf("Received a message in the worker\n");
|
||||||
|
zframe_t *client_id = zmsg_pop(msg);
|
||||||
|
|
||||||
printf("Size: %zu\tContent size: %zu\n", zmsg_size(msg), zmsg_content_size(msg));
|
|
||||||
struct guestfs_inpsect_command *command = guestfs_inspect_zmsg_to_command(msg);
|
struct guestfs_inpsect_command *command = guestfs_inspect_zmsg_to_command(msg);
|
||||||
printf("Size OF:: %zu\n", (sizeof(struct guestfs_inpsect_command)));
|
|
||||||
zmsg_t *reply = zmsg_new();
|
zmsg_t *reply = zmsg_new();
|
||||||
if (!reply) {
|
if (!reply) {
|
||||||
printf("wuddahec\n");
|
printf("wuddahec\n");
|
||||||
@@ -116,75 +115,139 @@ static void worker_task(zsock_t *pipe, char *disk_path) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
switch (command->command) {
|
switch (command->command) {
|
||||||
case GUESTFS_COMMAND_LS:
|
case GUESTFS_COMMAND_LS: printf("Enter ls command\n");
|
||||||
zmsg_pushstr(reply, "From worker! -- not implemented");
|
zmsg_pushstr(reply, "From worker! -- not implemented");
|
||||||
break;
|
break;
|
||||||
case GUESTFS_COMMAND_CAT:
|
case GUESTFS_COMMAND_CAT: printf("Entering cat command\n");
|
||||||
char *res;
|
char *res = NULL;
|
||||||
size_t s;
|
size_t s = 0;
|
||||||
puts("catting file...");
|
cat_file(g, command->args.cat.paths[0], &res, &s);
|
||||||
cat_file(g, command->args.cat.paths[0], &res, &s);
|
zmsg_addmem(reply, res, s);
|
||||||
puts("Done catting file contents...");
|
free(res);
|
||||||
zmsg_addmem(reply, res, s);
|
break;
|
||||||
free(res);
|
default:
|
||||||
break;
|
zmsg_pushstr(reply, "Unknown command");
|
||||||
|
break;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Sending reply
|
// Sending reply
|
||||||
zmsg_send(&reply, pipe);
|
zmsg_pushstr(reply, "");
|
||||||
|
zmsg_prepend(reply, &client_id);
|
||||||
|
zmsg_send(&reply, worker);
|
||||||
|
|
||||||
guestfs_inspect_command_destroy(&command);
|
guestfs_inspect_command_destroy(&command);
|
||||||
|
zframe_destroy(&client_id);
|
||||||
zmsg_destroy(&msg);
|
zmsg_destroy(&msg);
|
||||||
zmsg_destroy(&reply);
|
zmsg_destroy(&reply);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
zsock_destroy(&worker);
|
||||||
guestfs_close(g);
|
guestfs_close(g);
|
||||||
exit(EXIT_SUCCESS);
|
exit(EXIT_SUCCESS);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
typedef struct {
|
||||||
|
char *name;
|
||||||
|
zactor_t *worker;
|
||||||
|
} zactor_worker_map;
|
||||||
|
|
||||||
|
void mapping_zactor_worker(char *name, zactor_t *worker, zactor_worker_map **map, size_t mapper_size) {
|
||||||
|
|
||||||
|
map[mapper_size - 1] = malloc(sizeof(zactor_worker_map));
|
||||||
|
(*map[mapper_size - 1]).name = strdup(name);
|
||||||
|
(*map[mapper_size - 1]).worker = worker;
|
||||||
|
printf("Registering inside mapper %s :: %p\n", name, worker);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
zactor_t * zactor_worker_lookup(char *name, zactor_worker_map **map, size_t mapper_size) {
|
||||||
|
size_t i = 0;
|
||||||
|
printf("inside lookup::%zu\n", mapper_size);
|
||||||
|
for (i = 0; i < mapper_size; i++) {
|
||||||
|
if (strcmp((*map[i]).name, name) == 0) {
|
||||||
|
printf("found at %zu\n", i);
|
||||||
|
printf("found %p\n", map[i]->worker);
|
||||||
|
return map[i]->worker;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
int main(int argc, char **argv) {
|
int main(int argc, char **argv) {
|
||||||
if (argc < 2) {
|
if (argc < 2) {
|
||||||
printf("Usage: %s <disk-path>:<name> ...\n", argv[0]);
|
printf("Usage: %s <disk-path>:<name> ...\n", argv[0]);
|
||||||
return EXIT_FAILURE;
|
return EXIT_FAILURE;
|
||||||
}
|
}
|
||||||
|
|
||||||
char *path = strtok(argv[1], ":");
|
// Zactor setup
|
||||||
char *name = strtok(NULL, ":");
|
size_t worker_map_size = 0;
|
||||||
printf("name: '%s'\n", name);
|
zactor_worker_map **worker_map = malloc(sizeof(zactor_worker_map *) * worker_map_size);
|
||||||
zactor_t *worker = zactor_new(worker_task, path);
|
|
||||||
|
|
||||||
|
for (int i = 0; i < argc - 1; i++) {
|
||||||
|
worker_map_size++;
|
||||||
|
worker_map = realloc(worker_map, sizeof(zactor_worker_map *) * worker_map_size);
|
||||||
|
char *path = strtok(argv[i+1], ":");
|
||||||
|
char *name = strtok(NULL, ":");
|
||||||
|
printf("--- Registering\n");
|
||||||
|
printf("name: '%s'\n", name);
|
||||||
|
printf("path: '%s'\n", path);
|
||||||
|
mapping_zactor_worker(name, zactor_new((void *)worker_task, path), worker_map, worker_map_size);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Setup daemon ipc endpoint
|
||||||
char *ep = guestfs_inspect_endpoint();
|
char *ep = guestfs_inspect_endpoint();
|
||||||
printf("ep: %s\n", ep);
|
printf("ep: %s\n", ep);
|
||||||
|
|
||||||
|
// Setup ZMQ routers
|
||||||
zsock_t *frontend = zsock_new(ZMQ_ROUTER);
|
zsock_t *frontend = zsock_new(ZMQ_ROUTER);
|
||||||
zsock_bind(frontend, ep);
|
zsock_bind(frontend, ep);
|
||||||
free(ep);
|
free(ep);
|
||||||
|
|
||||||
|
zsock_t *backend = zsock_new(ZMQ_ROUTER);
|
||||||
|
zsock_bind(backend, "inproc://workers");
|
||||||
|
|
||||||
|
for (int i = 0; i < worker_map_size; i++) {
|
||||||
|
printf("Found:: %s :: %p\n", worker_map[i]->name, worker_map[i]->worker);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Connecting frontend to backend
|
||||||
|
zpoller_t *poller = zpoller_new(frontend, backend, NULL);
|
||||||
|
|
||||||
while (true) {
|
while (true) {
|
||||||
zmsg_t *msg = zmsg_recv(frontend);
|
void *sock = zpoller_wait(poller, -1);
|
||||||
if (!msg) {
|
if (!sock) {
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
|
|
||||||
// reply id
|
if (sock == frontend) {
|
||||||
char *id = zmsg_popstr(msg);
|
zmsg_t *msg = zmsg_recv(frontend);
|
||||||
// null frame
|
// reply id
|
||||||
zmsg_pop(msg);
|
zframe_t *identity = zmsg_pop(msg);
|
||||||
struct guestfs_inpsect_command *c = guestfs_inspect_zmsg_to_command(msg);
|
// Null frame
|
||||||
|
zmsg_pop(msg);
|
||||||
|
struct guestfs_inpsect_command *c = guestfs_inspect_zmsg_to_command(msg);
|
||||||
|
|
||||||
if (STREQ(name, c->name)) {
|
zactor_t *zactor_tmp = zactor_worker_lookup(c->name, worker_map, worker_map_size);
|
||||||
// send request to worker
|
if (zactor_tmp != NULL) {
|
||||||
printf("name equal\n");
|
// send request to worker
|
||||||
zactor_send(worker, &msg);
|
zmsg_prepend(msg, &identity);
|
||||||
|
zactor_send(zactor_tmp, &msg);
|
||||||
} else {
|
} else {
|
||||||
printf("name not equal\n");
|
// add failure responce
|
||||||
|
}
|
||||||
|
|
||||||
|
zactor_tmp = NULL;
|
||||||
|
guestfs_inspect_command_destroy(&c);
|
||||||
|
zmsg_destroy(&msg);
|
||||||
|
zframe_destroy(&identity);
|
||||||
|
} else if (sock == backend) {
|
||||||
|
zmsg_t *msg = zmsg_recv(backend);
|
||||||
|
zmsg_pop(msg); // Removing backend id
|
||||||
|
zmsg_send(&msg, frontend);
|
||||||
|
zmsg_destroy(&msg);
|
||||||
}
|
}
|
||||||
|
|
||||||
/* zmsg_t *reply = zmsg_recv(frontend); */
|
|
||||||
}
|
}
|
||||||
|
//zactor_destroy(&worker);
|
||||||
zactor_destroy(&worker);
|
|
||||||
return EXIT_SUCCESS;
|
return EXIT_SUCCESS;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user