static struct snobj *handle_list_workers(struct snobj *q)
{
struct snobj *r;
r = snobj_list();
for (int wid = 0; wid < MAX_WORKERS; wid++) {
struct snobj *worker;
if (!is_worker_active(wid))
continue;
worker = snobj_map();
snobj_map_set(worker, "wid", snobj_int(wid));
snobj_map_set(worker, "running",
snobj_int(is_worker_running(wid)));
snobj_map_set(worker, "core",
snobj_int(workers[wid]->core));
snobj_map_set(worker, "num_tcs",
snobj_int(workers[wid]->s->num_classes));
snobj_list_add(r, worker);
}
return r;
}
static struct snobj *docker_container_pid(char *cid, int *container_pid)
{
char buf[1024];
FILE *fp;
int ret;
int exit_code;
if (!cid)
return snobj_err(EINVAL, "field 'docker' should be " \
"a containder ID or name in string");
ret = snprintf(buf, sizeof(buf),
"docker inspect --format '{{.State.Pid}}' " \
"%s 2>&1", cid);
if (ret >= sizeof(buf))
return snobj_err(EINVAL, "The specified Docker " \
"container ID or name is too long");
fp = popen(buf, "r");
if (!fp)
return snobj_err_details(ESRCH,
snobj_str_fmt("popen() errno=%d (%s)",
errno, strerror(errno)),
"Command 'docker' is not available. " \
"(not installed?)");
ret = fread(buf, 1, sizeof(buf) - 1, fp);
if (ret == 0)
return snobj_err(ENOENT, "Cannot find the PID of " \
"container %s", cid);
buf[ret] = '\0';
ret = pclose(fp);
exit_code = WEXITSTATUS(ret);
if (exit_code != 0 || sscanf(buf, "%d", container_pid) == 0) {
struct snobj *details = snobj_map();
snobj_map_set(details, "exit_code", snobj_int(exit_code));
snobj_map_set(details, "docker_err", snobj_str(buf));
return snobj_err_details(ESRCH, details,
"Cannot find the PID of container %s", cid);
}
return NULL;
}
static struct snobj *handle_list_tcs(struct snobj *q)
{
struct snobj *r;
struct snobj *t;
unsigned int wid_filter = MAX_WORKERS;
struct ns_iter iter;
struct tc *c;
t = snobj_eval(q, "wid");
if (t) {
wid_filter = snobj_uint_get(t);
if (wid_filter >= MAX_WORKERS)
return snobj_err(EINVAL,
"'wid' must be between 0 and %d",
MAX_WORKERS - 1);
if (!is_worker_active(wid_filter))
return snobj_err(EINVAL, "worker:%d does not exist",
wid_filter);
}
r = snobj_list();
ns_init_iterator(&iter, NS_TYPE_TC);
while ((c = (struct tc *)ns_next(&iter)) != NULL) {
int wid;
if (wid_filter < MAX_WORKERS) {
if (workers[wid_filter]->s != c->s)
continue;
wid = wid_filter;
} else {
for (wid = 0; wid < MAX_WORKERS; wid++)
if (is_worker_active(wid) &&
workers[wid]->s == c->s)
break;
}
struct snobj *elem = snobj_map();
snobj_map_set(elem, "name", snobj_str(c->name));
snobj_map_set(elem, "tasks", snobj_int(c->num_tasks));
snobj_map_set(elem, "parent", snobj_str(c->parent->name));
snobj_map_set(elem, "priority", snobj_int(c->priority));
if (wid < MAX_WORKERS)
snobj_map_set(elem, "wid", snobj_uint(wid));
else
snobj_map_set(elem, "wid", snobj_int(-1));
snobj_list_add(r, elem);
}
ns_release_iterator(&iter);
return r;
}
