Blame tc/tc_exec.c

Packit Service 3880ab
/*
Packit Service 3880ab
 * tc_exec.c	"tc exec".
Packit Service 3880ab
 *
Packit Service 3880ab
 *		This program is free software; you can redistribute it and/or
Packit Service 3880ab
 *		modify it under the terms of the GNU General Public License
Packit Service 3880ab
 *		as published by the Free Software Foundation; either version
Packit Service 3880ab
 *		2 of the License, or (at your option) any later version.
Packit Service 3880ab
 *
Packit Service 3880ab
 * Authors:	Daniel Borkmann <daniel@iogearbox.net>
Packit Service 3880ab
 */
Packit Service 3880ab
Packit Service 3880ab
#include <stdio.h>
Packit Service 3880ab
#include <stdlib.h>
Packit Service 3880ab
#include <dlfcn.h>
Packit Service 3880ab
Packit Service 3880ab
#include "utils.h"
Packit Service 3880ab
Packit Service 3880ab
#include "tc_util.h"
Packit Service 3880ab
#include "tc_common.h"
Packit Service 3880ab
Packit Service 3880ab
static struct exec_util *exec_list;
Packit Service 3880ab
static void *BODY;
Packit Service 3880ab
Packit Service 3880ab
static void usage(void)
Packit Service 3880ab
{
Packit Service 3880ab
	fprintf(stderr,
Packit Service 3880ab
		"Usage: tc exec [ EXEC_TYPE ] [ help | OPTIONS ]\n"
Packit Service 3880ab
		"Where:\n"
Packit Service 3880ab
		"EXEC_TYPE := { bpf | etc. }\n"
Packit Service 3880ab
		"OPTIONS := ... try tc exec <desired EXEC_KIND> help\n");
Packit Service 3880ab
}
Packit Service 3880ab
Packit Service 3880ab
static int parse_noeopt(struct exec_util *eu, int argc, char **argv)
Packit Service 3880ab
{
Packit Service 3880ab
	if (argc) {
Packit Service 3880ab
		fprintf(stderr, "Unknown exec \"%s\", hence option \"%s\" is unparsable\n",
Packit Service 3880ab
			eu->id, *argv);
Packit Service 3880ab
		return -1;
Packit Service 3880ab
	}
Packit Service 3880ab
Packit Service 3880ab
	return 0;
Packit Service 3880ab
}
Packit Service 3880ab
Packit Service 3880ab
static struct exec_util *get_exec_kind(const char *name)
Packit Service 3880ab
{
Packit Service 3880ab
	struct exec_util *eu;
Packit Service 3880ab
	char buf[256];
Packit Service 3880ab
	void *dlh;
Packit Service 3880ab
Packit Service 3880ab
	for (eu = exec_list; eu; eu = eu->next)
Packit Service 3880ab
		if (strcmp(eu->id, name) == 0)
Packit Service 3880ab
			return eu;
Packit Service 3880ab
Packit Service 3880ab
	snprintf(buf, sizeof(buf), "%s/e_%s.so", get_tc_lib(), name);
Packit Service 3880ab
	dlh = dlopen(buf, RTLD_LAZY);
Packit Service 3880ab
	if (dlh == NULL) {
Packit Service 3880ab
		dlh = BODY;
Packit Service 3880ab
		if (dlh == NULL) {
Packit Service 3880ab
			dlh = BODY = dlopen(NULL, RTLD_LAZY);
Packit Service 3880ab
			if (dlh == NULL)
Packit Service 3880ab
				goto noexist;
Packit Service 3880ab
		}
Packit Service 3880ab
	}
Packit Service 3880ab
Packit Service 3880ab
	snprintf(buf, sizeof(buf), "%s_exec_util", name);
Packit Service 3880ab
	eu = dlsym(dlh, buf);
Packit Service 3880ab
	if (eu == NULL)
Packit Service 3880ab
		goto noexist;
Packit Service 3880ab
reg:
Packit Service 3880ab
	eu->next = exec_list;
Packit Service 3880ab
	exec_list = eu;
Packit Service 3880ab
Packit Service 3880ab
	return eu;
Packit Service 3880ab
noexist:
Packit Service 3880ab
	eu = calloc(1, sizeof(*eu));
Packit Service 3880ab
	if (eu) {
Packit Service 3880ab
		strncpy(eu->id, name, sizeof(eu->id) - 1);
Packit Service 3880ab
		eu->parse_eopt = parse_noeopt;
Packit Service 3880ab
		goto reg;
Packit Service 3880ab
	}
Packit Service 3880ab
Packit Service 3880ab
	return eu;
Packit Service 3880ab
}
Packit Service 3880ab
Packit Service 3880ab
int do_exec(int argc, char **argv)
Packit Service 3880ab
{
Packit Service 3880ab
	struct exec_util *eu;
Packit Service 3880ab
	char kind[FILTER_NAMESZ] = {};
Packit Service 3880ab
Packit Service 3880ab
	if (argc < 1) {
Packit Service 3880ab
		fprintf(stderr, "No command given, try \"tc exec help\".\n");
Packit Service 3880ab
		return -1;
Packit Service 3880ab
	}
Packit Service 3880ab
Packit Service 3880ab
	if (matches(*argv, "help") == 0) {
Packit Service 3880ab
		usage();
Packit Service 3880ab
		return 0;
Packit Service 3880ab
	}
Packit Service 3880ab
Packit Service 3880ab
	strncpy(kind, *argv, sizeof(kind) - 1);
Packit Service 3880ab
Packit Service 3880ab
	eu = get_exec_kind(kind);
Packit Service 3880ab
Packit Service 3880ab
	argc--;
Packit Service 3880ab
	argv++;
Packit Service 3880ab
Packit Service 3880ab
	return eu->parse_eopt(eu, argc, argv);
Packit Service 3880ab
}