Blame tc/tc.c

Packit Service 3880ab
/*
Packit Service 3880ab
 * tc.c		"tc" utility frontend.
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:	Alexey Kuznetsov, <kuznet@ms2.inr.ac.ru>
Packit Service 3880ab
 *
Packit Service 3880ab
 * Fixes:
Packit Service 3880ab
 *
Packit Service 3880ab
 * Petri Mattila <petri@prihateam.fi> 990308: wrong memset's resulted in faults
Packit Service 3880ab
 */
Packit Service 3880ab
Packit Service 3880ab
#include <stdio.h>
Packit Service 3880ab
#include <stdlib.h>
Packit Service 3880ab
#include <unistd.h>
Packit Service 3880ab
#include <fcntl.h>
Packit Service 3880ab
#include <dlfcn.h>
Packit Service 3880ab
#include <sys/socket.h>
Packit Service 3880ab
#include <netinet/in.h>
Packit Service 3880ab
#include <arpa/inet.h>
Packit Service 3880ab
#include <string.h>
Packit Service 3880ab
#include <errno.h>
Packit Service 3880ab
Packit Service 3880ab
#include "version.h"
Packit Service 3880ab
#include "utils.h"
Packit Service 3880ab
#include "tc_util.h"
Packit Service 3880ab
#include "tc_common.h"
Packit Service 3880ab
#include "namespace.h"
Packit Service 3880ab
#include "rt_names.h"
Packit Service 3880ab
Packit Service 3880ab
int show_stats;
Packit Service 3880ab
int show_details;
Packit Service 3880ab
int show_raw;
Packit Service 3880ab
int show_graph;
Packit Service 3880ab
int timestamp;
Packit Service 3880ab
Packit Service 3880ab
int batch_mode;
Packit Service 3880ab
int use_iec;
Packit Service 3880ab
int force;
Packit Service 3880ab
bool use_names;
Packit Service 3880ab
int json;
Packit Service 3880ab
int color;
Packit Service 3880ab
int oneline;
Packit Service 3880ab
Packit Service 3880ab
static char *conf_file;
Packit Service 3880ab
Packit Service 3880ab
struct rtnl_handle rth;
Packit Service 3880ab
Packit Service 3880ab
static void *BODY;	/* cached handle dlopen(NULL) */
Packit Service 3880ab
static struct qdisc_util *qdisc_list;
Packit Service 3880ab
static struct filter_util *filter_list;
Packit Service 3880ab
Packit Service 3880ab
static int print_noqopt(struct qdisc_util *qu, FILE *f,
Packit Service 3880ab
			struct rtattr *opt)
Packit Service 3880ab
{
Packit Service 3880ab
	if (opt && RTA_PAYLOAD(opt))
Packit Service 3880ab
		fprintf(f, "[Unknown qdisc, optlen=%u] ",
Packit Service 3880ab
			(unsigned int) RTA_PAYLOAD(opt));
Packit Service 3880ab
	return 0;
Packit Service 3880ab
}
Packit Service 3880ab
Packit Service 3880ab
static int parse_noqopt(struct qdisc_util *qu, int argc, char **argv,
Packit Service 3880ab
			struct nlmsghdr *n, const char *dev)
Packit Service 3880ab
{
Packit Service 3880ab
	if (argc) {
Packit Service 3880ab
		fprintf(stderr,
Packit Service 3880ab
			"Unknown qdisc \"%s\", hence option \"%s\" is unparsable\n",
Packit Service 3880ab
			qu->id, *argv);
Packit Service 3880ab
		return -1;
Packit Service 3880ab
	}
Packit Service 3880ab
	return 0;
Packit Service 3880ab
}
Packit Service 3880ab
Packit Service 3880ab
static int print_nofopt(struct filter_util *qu, FILE *f, struct rtattr *opt, __u32 fhandle)
Packit Service 3880ab
{
Packit Service 3880ab
	if (opt && RTA_PAYLOAD(opt))
Packit Service 3880ab
		fprintf(f, "fh %08x [Unknown filter, optlen=%u] ",
Packit Service 3880ab
			fhandle, (unsigned int) RTA_PAYLOAD(opt));
Packit Service 3880ab
	else if (fhandle)
Packit Service 3880ab
		fprintf(f, "fh %08x ", fhandle);
Packit Service 3880ab
	return 0;
Packit Service 3880ab
}
Packit Service 3880ab
Packit Service 3880ab
static int parse_nofopt(struct filter_util *qu, char *fhandle,
Packit Service 3880ab
			int argc, char **argv, struct nlmsghdr *n)
Packit Service 3880ab
{
Packit Service 3880ab
	__u32 handle;
Packit Service 3880ab
Packit Service 3880ab
	if (argc) {
Packit Service 3880ab
		fprintf(stderr,
Packit Service 3880ab
			"Unknown filter \"%s\", hence option \"%s\" is unparsable\n",
Packit Service 3880ab
			qu->id, *argv);
Packit Service 3880ab
		return -1;
Packit Service 3880ab
	}
Packit Service 3880ab
	if (fhandle) {
Packit Service 3880ab
		struct tcmsg *t = NLMSG_DATA(n);
Packit Service 3880ab
Packit Service 3880ab
		if (get_u32(&handle, fhandle, 16)) {
Packit Service 3880ab
			fprintf(stderr, "Unparsable filter ID \"%s\"\n", fhandle);
Packit Service 3880ab
			return -1;
Packit Service 3880ab
		}
Packit Service 3880ab
		t->tcm_handle = handle;
Packit Service 3880ab
	}
Packit Service 3880ab
	return 0;
Packit Service 3880ab
}
Packit Service 3880ab
Packit Service 3880ab
struct qdisc_util *get_qdisc_kind(const char *str)
Packit Service 3880ab
{
Packit Service 3880ab
	void *dlh;
Packit Service 3880ab
	char buf[256];
Packit Service 3880ab
	struct qdisc_util *q;
Packit Service 3880ab
Packit Service 3880ab
	for (q = qdisc_list; q; q = q->next)
Packit Service 3880ab
		if (strcmp(q->id, str) == 0)
Packit Service 3880ab
			return q;
Packit Service 3880ab
Packit Service 3880ab
	snprintf(buf, sizeof(buf), "%s/q_%s.so", get_tc_lib(), str);
Packit Service 3880ab
	dlh = dlopen(buf, RTLD_LAZY);
Packit Service 3880ab
	if (!dlh) {
Packit Service 3880ab
		/* look in current binary, only open once */
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_qdisc_util", str);
Packit Service 3880ab
	q = dlsym(dlh, buf);
Packit Service 3880ab
	if (q == NULL)
Packit Service 3880ab
		goto noexist;
Packit Service 3880ab
Packit Service 3880ab
reg:
Packit Service 3880ab
	q->next = qdisc_list;
Packit Service 3880ab
	qdisc_list = q;
Packit Service 3880ab
	return q;
Packit Service 3880ab
Packit Service 3880ab
noexist:
Packit Service 3880ab
	q = calloc(1, sizeof(*q));
Packit Service 3880ab
	if (q) {
Packit Service 3880ab
		q->id = strdup(str);
Packit Service 3880ab
		q->parse_qopt = parse_noqopt;
Packit Service 3880ab
		q->print_qopt = print_noqopt;
Packit Service 3880ab
		goto reg;
Packit Service 3880ab
	}
Packit Service 3880ab
	return q;
Packit Service 3880ab
}
Packit Service 3880ab
Packit Service 3880ab
Packit Service 3880ab
struct filter_util *get_filter_kind(const char *str)
Packit Service 3880ab
{
Packit Service 3880ab
	void *dlh;
Packit Service 3880ab
	char buf[256];
Packit Service 3880ab
	struct filter_util *q;
Packit Service 3880ab
Packit Service 3880ab
	for (q = filter_list; q; q = q->next)
Packit Service 3880ab
		if (strcmp(q->id, str) == 0)
Packit Service 3880ab
			return q;
Packit Service 3880ab
Packit Service 3880ab
	snprintf(buf, sizeof(buf), "%s/f_%s.so", get_tc_lib(), str);
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_filter_util", str);
Packit Service 3880ab
	q = dlsym(dlh, buf);
Packit Service 3880ab
	if (q == NULL)
Packit Service 3880ab
		goto noexist;
Packit Service 3880ab
Packit Service 3880ab
reg:
Packit Service 3880ab
	q->next = filter_list;
Packit Service 3880ab
	filter_list = q;
Packit Service 3880ab
	return q;
Packit Service 3880ab
noexist:
Packit Service 3880ab
	q = calloc(1, sizeof(*q));
Packit Service 3880ab
	if (q) {
Packit Service 3880ab
		strncpy(q->id, str, 15);
Packit Service 3880ab
		q->parse_fopt = parse_nofopt;
Packit Service 3880ab
		q->print_fopt = print_nofopt;
Packit Service 3880ab
		goto reg;
Packit Service 3880ab
	}
Packit Service 3880ab
	return q;
Packit Service 3880ab
}
Packit Service 3880ab
Packit Service 3880ab
static void usage(void)
Packit Service 3880ab
{
Packit Service 3880ab
	fprintf(stderr,
Packit Service 3880ab
		"Usage:	tc [ OPTIONS ] OBJECT { COMMAND | help }\n"
Packit Service 3880ab
		"	tc [-force] -batch filename\n"
Packit Service 3880ab
		"where  OBJECT := { qdisc | class | filter | chain |\n"
Packit Service 3880ab
		"		    action | monitor | exec }\n"
Packit Service 3880ab
		"       OPTIONS := { -V[ersion] | -s[tatistics] | -d[etails] | -r[aw] |\n"
Packit Service 3880ab
		"		    -o[neline] | -j[son] | -p[retty] | -c[olor]\n"
Packit Service 3880ab
		"		    -b[atch] [filename] | -n[etns] name | -N[umeric] |\n"
Packit Service 3880ab
		"		     -nm | -nam[es] | { -cf | -conf } path }\n");
Packit Service 3880ab
}
Packit Service 3880ab
Packit Service 3880ab
static int do_cmd(int argc, char **argv)
Packit Service 3880ab
{
Packit Service 3880ab
	if (matches(*argv, "qdisc") == 0)
Packit Service 3880ab
		return do_qdisc(argc-1, argv+1);
Packit Service 3880ab
	if (matches(*argv, "class") == 0)
Packit Service 3880ab
		return do_class(argc-1, argv+1);
Packit Service 3880ab
	if (matches(*argv, "filter") == 0)
Packit Service 3880ab
		return do_filter(argc-1, argv+1);
Packit Service 3880ab
	if (matches(*argv, "chain") == 0)
Packit Service 3880ab
		return do_chain(argc-1, argv+1);
Packit Service 3880ab
	if (matches(*argv, "actions") == 0)
Packit Service 3880ab
		return do_action(argc-1, argv+1);
Packit Service 3880ab
	if (matches(*argv, "monitor") == 0)
Packit Service 3880ab
		return do_tcmonitor(argc-1, argv+1);
Packit Service 3880ab
	if (matches(*argv, "exec") == 0)
Packit Service 3880ab
		return do_exec(argc-1, argv+1);
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
	fprintf(stderr, "Object \"%s\" is unknown, try \"tc help\".\n",
Packit Service 3880ab
		*argv);
Packit Service 3880ab
	return -1;
Packit Service 3880ab
}
Packit Service 3880ab
Packit Service 3880ab
static int batch(const char *name)
Packit Service 3880ab
{
Packit Service 3880ab
	char *line = NULL;
Packit Service 3880ab
	size_t len = 0;
Packit Service 3880ab
	int ret = 0;
Packit Service 3880ab
Packit Service 3880ab
	batch_mode = 1;
Packit Service 3880ab
	if (name && strcmp(name, "-") != 0) {
Packit Service 3880ab
		if (freopen(name, "r", stdin) == NULL) {
Packit Service 3880ab
			fprintf(stderr,
Packit Service 3880ab
				"Cannot open file \"%s\" for reading: %s\n",
Packit Service 3880ab
				name, strerror(errno));
Packit Service 3880ab
			return -1;
Packit Service 3880ab
		}
Packit Service 3880ab
	}
Packit Service 3880ab
Packit Service 3880ab
	tc_core_init();
Packit Service 3880ab
Packit Service 3880ab
	if (rtnl_open(&rth, 0) < 0) {
Packit Service 3880ab
		fprintf(stderr, "Cannot open rtnetlink\n");
Packit Service 3880ab
		return -1;
Packit Service 3880ab
	}
Packit Service 3880ab
Packit Service 3880ab
	cmdlineno = 0;
Packit Service 3880ab
	while (getcmdline(&line, &len, stdin) != -1) {
Packit Service 3880ab
		char *largv[100];
Packit Service 3880ab
		int largc;
Packit Service 3880ab
Packit Service 3880ab
		largc = makeargs(line, largv, 100);
Packit Service 3880ab
		if (largc == 0)
Packit Service 3880ab
			continue;	/* blank line */
Packit Service 3880ab
Packit Service 3880ab
		if (do_cmd(largc, largv)) {
Packit Service 3880ab
			fprintf(stderr, "Command failed %s:%d\n",
Packit Service 3880ab
				name, cmdlineno);
Packit Service 3880ab
			ret = 1;
Packit Service 3880ab
			if (!force)
Packit Service 3880ab
				break;
Packit Service 3880ab
		}
Packit Service 3880ab
		fflush(stdout);
Packit Service 3880ab
	}
Packit Service 3880ab
Packit Service 3880ab
	free(line);
Packit Service 3880ab
	rtnl_close(&rth);
Packit Service 3880ab
	return ret;
Packit Service 3880ab
}
Packit Service 3880ab
Packit Service 3880ab
Packit Service 3880ab
int main(int argc, char **argv)
Packit Service 3880ab
{
Packit Service 3880ab
	int ret;
Packit Service 3880ab
	char *batch_file = NULL;
Packit Service 3880ab
Packit Service 3880ab
	while (argc > 1) {
Packit Service 3880ab
		if (argv[1][0] != '-')
Packit Service 3880ab
			break;
Packit Service 3880ab
		if (matches(argv[1], "-stats") == 0 ||
Packit Service 3880ab
			 matches(argv[1], "-statistics") == 0) {
Packit Service 3880ab
			++show_stats;
Packit Service 3880ab
		} else if (matches(argv[1], "-details") == 0) {
Packit Service 3880ab
			++show_details;
Packit Service 3880ab
		} else if (matches(argv[1], "-raw") == 0) {
Packit Service 3880ab
			++show_raw;
Packit Service 3880ab
		} else if (matches(argv[1], "-pretty") == 0) {
Packit Service 3880ab
			++pretty;
Packit Service 3880ab
		} else if (matches(argv[1], "-graph") == 0) {
Packit Service 3880ab
			show_graph = 1;
Packit Service 3880ab
		} else if (matches(argv[1], "-Version") == 0) {
Packit Service 3880ab
			printf("tc utility, iproute2-%s\n", version);
Packit Service 3880ab
			return 0;
Packit Service 3880ab
		} else if (matches(argv[1], "-iec") == 0) {
Packit Service 3880ab
			++use_iec;
Packit Service 3880ab
		} else if (matches(argv[1], "-help") == 0) {
Packit Service 3880ab
			usage();
Packit Service 3880ab
			return 0;
Packit Service 3880ab
		} else if (matches(argv[1], "-force") == 0) {
Packit Service 3880ab
			++force;
Packit Service 3880ab
		} else if (matches(argv[1], "-batch") == 0) {
Packit Service 3880ab
			argc--;	argv++;
Packit Service 3880ab
			if (argc <= 1)
Packit Service 3880ab
				usage();
Packit Service 3880ab
			batch_file = argv[1];
Packit Service 3880ab
		} else if (matches(argv[1], "-netns") == 0) {
Packit Service 3880ab
			NEXT_ARG();
Packit Service 3880ab
			if (netns_switch(argv[1]))
Packit Service 3880ab
				return -1;
Packit Service 3880ab
		} else if (matches(argv[1], "-Numeric") == 0) {
Packit Service 3880ab
			++numeric;
Packit Service 3880ab
		} else if (matches(argv[1], "-names") == 0 ||
Packit Service 3880ab
				matches(argv[1], "-nm") == 0) {
Packit Service 3880ab
			use_names = true;
Packit Service 3880ab
		} else if (matches(argv[1], "-cf") == 0 ||
Packit Service 3880ab
				matches(argv[1], "-conf") == 0) {
Packit Service 3880ab
			NEXT_ARG();
Packit Service 3880ab
			conf_file = argv[1];
Packit Service 3880ab
		} else if (matches_color(argv[1], &color)) {
Packit Service 3880ab
		} else if (matches(argv[1], "-timestamp") == 0) {
Packit Service 3880ab
			timestamp++;
Packit Service 3880ab
		} else if (matches(argv[1], "-tshort") == 0) {
Packit Service 3880ab
			++timestamp;
Packit Service 3880ab
			++timestamp_short;
Packit Service 3880ab
		} else if (matches(argv[1], "-json") == 0) {
Packit Service 3880ab
			++json;
Packit Service 3880ab
		} else if (matches(argv[1], "-oneline") == 0) {
Packit Service 3880ab
			++oneline;
Packit Service 3880ab
		} else {
Packit Service 3880ab
			fprintf(stderr,
Packit Service 3880ab
				"Option \"%s\" is unknown, try \"tc -help\".\n",
Packit Service 3880ab
				argv[1]);
Packit Service 3880ab
			return -1;
Packit Service 3880ab
		}
Packit Service 3880ab
		argc--;	argv++;
Packit Service 3880ab
	}
Packit Service 3880ab
Packit Service 3880ab
	_SL_ = oneline ? "\\" : "\n";
Packit Service 3880ab
Packit Service 3880ab
	check_enable_color(color, json);
Packit Service 3880ab
Packit Service 3880ab
	if (batch_file)
Packit Service 3880ab
		return batch(batch_file);
Packit Service 3880ab
Packit Service 3880ab
	if (argc <= 1) {
Packit Service 3880ab
		usage();
Packit Service 3880ab
		return 0;
Packit Service 3880ab
	}
Packit Service 3880ab
Packit Service 3880ab
	tc_core_init();
Packit Service 3880ab
	if (rtnl_open(&rth, 0) < 0) {
Packit Service 3880ab
		fprintf(stderr, "Cannot open rtnetlink\n");
Packit Service 3880ab
		exit(1);
Packit Service 3880ab
	}
Packit Service 3880ab
Packit Service 3880ab
	if (use_names && cls_names_init(conf_file)) {
Packit Service 3880ab
		ret = -1;
Packit Service 3880ab
		goto Exit;
Packit Service 3880ab
	}
Packit Service 3880ab
Packit Service 3880ab
	ret = do_cmd(argc-1, argv+1);
Packit Service 3880ab
Exit:
Packit Service 3880ab
	rtnl_close(&rth);
Packit Service 3880ab
Packit Service 3880ab
	if (use_names)
Packit Service 3880ab
		cls_names_uninit();
Packit Service 3880ab
Packit Service 3880ab
	return ret;
Packit Service 3880ab
}