[mirror_iproute2.git] / tc / tc.c

/*
 * tc.c		"tc" utility frontend.
 *
 *		This program is free software; you can redistribute it and/or
 *		modify it under the terms of the GNU General Public License
 *		as published by the Free Software Foundation; either version
 *		2 of the License, or (at your option) any later version.
 *
 * Authors:	Alexey Kuznetsov, <kuznet@ms2.inr.ac.ru>
 *
 * Fixes:
 *
 * Petri Mattila <petri@prihateam.fi> 990308: wrong memset's resulted in faults
 */

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <syslog.h>
#include <fcntl.h>
#include <dlfcn.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <string.h>
#include <errno.h>

#include "SNAPSHOT.h"
#include "utils.h"
#include "tc_util.h"
#include "tc_common.h"

int show_stats = 0;
int show_details = 0;
int show_raw = 0;
int show_pretty = 0;

int resolve_hosts = 0;
int use_iec = 0;
int force = 0;
struct rtnl_handle rth;

static void *BODY = NULL;	/* cached handle dlopen(NULL) */
static struct qdisc_util * qdisc_list;
static struct filter_util * filter_list;

static int print_noqopt(struct qdisc_util *qu, FILE *f,
			struct rtattr *opt)
{
	if (opt && RTA_PAYLOAD(opt))
		fprintf(f, "[Unknown qdisc, optlen=%u] ",
			(unsigned) RTA_PAYLOAD(opt));
	return 0;
}

static int parse_noqopt(struct qdisc_util *qu, int argc, char **argv, struct nlmsghdr *n)
{
	if (argc) {
		fprintf(stderr, "Unknown qdisc \"%s\", hence option \"%s\" is unparsable\n", qu->id, *argv);
		return -1;
	}
	return 0;
}

static int print_nofopt(struct filter_util *qu, FILE *f, struct rtattr *opt, __u32 fhandle)
{
	if (opt && RTA_PAYLOAD(opt))
		fprintf(f, "fh %08x [Unknown filter, optlen=%u] ",
			fhandle, (unsigned) RTA_PAYLOAD(opt));
	else if (fhandle)
		fprintf(f, "fh %08x ", fhandle);
	return 0;
}

static int parse_nofopt(struct filter_util *qu, char *fhandle, int argc, char **argv, struct nlmsghdr *n)
{
	__u32 handle;

	if (argc) {
		fprintf(stderr, "Unknown filter \"%s\", hence option \"%s\" is unparsable\n", qu->id, *argv);
		return -1;
	}
	if (fhandle) {
		struct tcmsg *t = NLMSG_DATA(n);
		if (get_u32(&handle, fhandle, 16)) {
			fprintf(stderr, "Unparsable filter ID \"%s\"\n", fhandle);
			return -1;
		}
		t->tcm_handle = handle;
	}
	return 0;
}

struct qdisc_util *get_qdisc_kind(const char *str)
{
	void *dlh;
	char buf[256];
	struct qdisc_util *q;

	for (q = qdisc_list; q; q = q->next)
		if (strcmp(q->id, str) == 0)
			return q;

	snprintf(buf, sizeof(buf), "%s/q_%s.so", get_tc_lib(), str);
	dlh = dlopen(buf, RTLD_LAZY);
	if (!dlh) {
		/* look in current binary, only open once */
		dlh = BODY;
		if (dlh == NULL) {
			dlh = BODY = dlopen(NULL, RTLD_LAZY);
			if (dlh == NULL)
				goto noexist;
		}
	}

	snprintf(buf, sizeof(buf), "%s_qdisc_util", str);
	q = dlsym(dlh, buf);
	if (q == NULL)
		goto noexist;

reg:
	q->next = qdisc_list;
	qdisc_list = q;
	return q;

noexist:
	q = malloc(sizeof(*q));
	if (q) {

		memset(q, 0, sizeof(*q));
		q->id = strcpy(malloc(strlen(str)+1), str);
		q->parse_qopt = parse_noqopt;
		q->print_qopt = print_noqopt;
		goto reg;
	}
	return q;
}


struct filter_util *get_filter_kind(const char *str)
{
	void *dlh;
	char buf[256];
	struct filter_util *q;

	for (q = filter_list; q; q = q->next)
		if (strcmp(q->id, str) == 0)
			return q;

	snprintf(buf, sizeof(buf), "%s/f_%s.so", get_tc_lib(), str);
	dlh = dlopen(buf, RTLD_LAZY);
	if (dlh == NULL) {
		dlh = BODY;
		if (dlh == NULL) {
			dlh = BODY = dlopen(NULL, RTLD_LAZY);
			if (dlh == NULL)
				goto noexist;
		}
	}

	snprintf(buf, sizeof(buf), "%s_filter_util", str);
	q = dlsym(dlh, buf);
	if (q == NULL)
		goto noexist;

reg:
	q->next = filter_list;
	filter_list = q;
	return q;
noexist:
	q = malloc(sizeof(*q));
	if (q) {
		memset(q, 0, sizeof(*q));
		strncpy(q->id, str, 15);
		q->parse_fopt = parse_nofopt;
		q->print_fopt = print_nofopt;
		goto reg;
	}
	return q;
}

static void usage(void)
{
	fprintf(stderr, "Usage: tc [ OPTIONS ] OBJECT { COMMAND | help }\n"
			"       tc [-force] -batch file\n"
	                "where  OBJECT := { qdisc | class | filter | action | monitor }\n"
	                "       OPTIONS := { -s[tatistics] | -d[etails] | -r[aw] | -p[retty] | -b[atch] [file] }\n");
}

static int do_cmd(int argc, char **argv)
{
	if (matches(*argv, "qdisc") == 0)
		return do_qdisc(argc-1, argv+1);

	if (matches(*argv, "class") == 0)
		return do_class(argc-1, argv+1);

	if (matches(*argv, "filter") == 0)
		return do_filter(argc-1, argv+1);

	if (matches(*argv, "actions") == 0)
		return do_action(argc-1, argv+1);

	if (matches(*argv, "monitor") == 0)
		return do_tcmonitor(argc-1, argv+1);

	if (matches(*argv, "help") == 0) {
		usage();
		return 0;
	}

	fprintf(stderr, "Object \"%s\" is unknown, try \"tc help\".\n",
		*argv);
	return -1;
}

static int batch(const char *name)
{
	char *line = NULL;
	size_t len = 0;
	int ret = 0;

	if (name && strcmp(name, "-") != 0) {
		if (freopen(name, "r", stdin) == NULL) {
			fprintf(stderr, "Cannot open file \"%s\" for reading: %s\n",
				name, strerror(errno));
			return -1;
		}
	}

	tc_core_init();

	if (rtnl_open(&rth, 0) < 0) {
		fprintf(stderr, "Cannot open rtnetlink\n");
		return -1;
	}

	cmdlineno = 0;
	while (getcmdline(&line, &len, stdin) != -1) {
		char *largv[100];
		int largc;

		largc = makeargs(line, largv, 100);
		if (largc == 0)
			continue;	/* blank line */

		if (do_cmd(largc, largv)) {
			fprintf(stderr, "Command failed %s:%d\n", name, cmdlineno);
			ret = 1;
			if (!force)
				break;
		}
	}
	if (line)
		free(line);

	rtnl_close(&rth);
	return ret;
}


int main(int argc, char **argv)
{
	int ret;
	int do_batching = 0;
	char *batchfile = NULL;

	while (argc > 1) {
		if (argv[1][0] != '-')
			break;
		if (matches(argv[1], "-stats") == 0 ||
			 matches(argv[1], "-statistics") == 0) {
			++show_stats;
		} else if (matches(argv[1], "-details") == 0) {
			++show_details;
		} else if (matches(argv[1], "-raw") == 0) {
			++show_raw;
		} else if (matches(argv[1], "-pretty") == 0) {
			++show_pretty;
		} else if (matches(argv[1], "-Version") == 0) {
			printf("tc utility, iproute2-ss%s\n", SNAPSHOT);
			return 0;
		} else if (matches(argv[1], "-iec") == 0) {
			++use_iec;
		} else if (matches(argv[1], "-help") == 0) {
			usage();
			return 0;
		} else if (matches(argv[1], "-force") == 0) {
			++force;
		} else 	if (matches(argv[1], "-batch") == 0) {
			do_batching = 1;
			if (argc > 2)
				batchfile = argv[2];
			argc--;	argv++;
		} else {
			fprintf(stderr, "Option \"%s\" is unknown, try \"tc -help\".\n", argv[1]);
			return -1;
		}
		argc--;	argv++;
	}

	if (do_batching)
		return batch(batchfile);

	if (argc <= 1) {
		usage();
		return 0;
	}

	tc_core_init();
	if (rtnl_open(&rth, 0) < 0) {
		fprintf(stderr, "Cannot open rtnetlink\n");
		exit(1);
	}

	ret = do_cmd(argc-1, argv+1);
	rtnl_close(&rth);

	return ret;
}
Commit	Line	Data
aba5acdf SH	1	/*
	2	* tc.c "tc" utility frontend.
	3	*
	4	* This program is free software; you can redistribute it and/or
	5	* modify it under the terms of the GNU General Public License
	6	* as published by the Free Software Foundation; either version
	7	* 2 of the License, or (at your option) any later version.
	8	*
	9	* Authors: Alexey Kuznetsov, <kuznet@ms2.inr.ac.ru>
	10	*
	11	* Fixes:
	12	*
	13	* Petri Mattila <petri@prihateam.fi> 990308: wrong memset's resulted in faults
	14	*/
	15
	16	#include <stdio.h>
	17	#include <stdlib.h>
	18	#include <unistd.h>
	19	#include <syslog.h>
	20	#include <fcntl.h>
	21	#include <dlfcn.h>
	22	#include <sys/socket.h>
	23	#include <netinet/in.h>
	24	#include <arpa/inet.h>
	25	#include <string.h>
	26	#include <errno.h>
	27
	28	#include "SNAPSHOT.h"
	29	#include "utils.h"
	30	#include "tc_util.h"
	31	#include "tc_common.h"
	32
	33	int show_stats = 0;
	34	int show_details = 0;
	35	int show_raw = 0;
44dcfe82 SH	36	int show_pretty = 0;
44dcfe82 SH	37
aba5acdf	38	int resolve_hosts = 0;
3ea2bf45	39	int use_iec = 0;
08856f02	40	int force = 0;
7901660a	41	struct rtnl_handle rth;
aba5acdf	42
f453a0d4	43	static void BODY = NULL; / cached handle dlopen(NULL) */
aba5acdf SH	44	static struct qdisc_util * qdisc_list;
	45	static struct filter_util * filter_list;
	46
ae665a52	47	static int print_noqopt(struct qdisc_util qu, FILE f,
1798c9d5	48	struct rtattr *opt)
aba5acdf SH	49	{
aba5acdf SH	50	if (opt && RTA_PAYLOAD(opt))
ae665a52	51	fprintf(f, "[Unknown qdisc, optlen=%u] ",
bb6a21a4	52	(unsigned) RTA_PAYLOAD(opt));
aba5acdf SH	53	return 0;
	54	}
	55
	56	static int parse_noqopt(struct qdisc_util qu, int argc, char argv, struct nlmsghdr n)
	57	{
	58	if (argc) {
	59	fprintf(stderr, "Unknown qdisc \"%s\", hence option \"%s\" is unparsable\n", qu->id, *argv);
	60	return -1;
	61	}
	62	return 0;
	63	}
	64
	65	static int print_nofopt(struct filter_util qu, FILE f, struct rtattr *opt, __u32 fhandle)
	66	{
	67	if (opt && RTA_PAYLOAD(opt))
ae665a52	68	fprintf(f, "fh %08x [Unknown filter, optlen=%u] ",
bb6a21a4	69	fhandle, (unsigned) RTA_PAYLOAD(opt));
aba5acdf SH	70	else if (fhandle)
	71	fprintf(f, "fh %08x ", fhandle);
	72	return 0;
	73	}
	74
	75	static int parse_nofopt(struct filter_util qu, char fhandle, int argc, char *argv, struct nlmsghdr n)
	76	{
	77	__u32 handle;
	78
	79	if (argc) {
	80	fprintf(stderr, "Unknown filter \"%s\", hence option \"%s\" is unparsable\n", qu->id, *argv);
	81	return -1;
	82	}
	83	if (fhandle) {
	84	struct tcmsg *t = NLMSG_DATA(n);
	85	if (get_u32(&handle, fhandle, 16)) {
	86	fprintf(stderr, "Unparsable filter ID \"%s\"\n", fhandle);
	87	return -1;
	88	}
	89	t->tcm_handle = handle;
	90	}
	91	return 0;
	92	}
	93
4094db72	94	struct qdisc_util get_qdisc_kind(const char str)
aba5acdf SH	95	{
	96	void *dlh;
	97	char buf[256];
	98	struct qdisc_util *q;
	99
	100	for (q = qdisc_list; q; q = q->next)
	101	if (strcmp(q->id, str) == 0)
	102	return q;
	103
aa27f88c	104	snprintf(buf, sizeof(buf), "%s/q_%s.so", get_tc_lib(), str);
aba5acdf	105	dlh = dlopen(buf, RTLD_LAZY);
b7a45150 SH	106	if (!dlh) {
b7a45150 SH	107	/* look in current binary, only open once */
aba5acdf SH	108	dlh = BODY;
	109	if (dlh == NULL) {
	110	dlh = BODY = dlopen(NULL, RTLD_LAZY);
	111	if (dlh == NULL)
	112	goto noexist;
	113	}
	114	}
	115
95812b56	116	snprintf(buf, sizeof(buf), "%s_qdisc_util", str);
aba5acdf SH	117	q = dlsym(dlh, buf);
	118	if (q == NULL)
	119	goto noexist;
	120
	121	reg:
	122	q->next = qdisc_list;
	123	qdisc_list = q;
	124	return q;
	125
	126	noexist:
	127	q = malloc(sizeof(*q));
	128	if (q) {
1798c9d5	129
aba5acdf	130	memset(q, 0, sizeof(*q));
1798c9d5	131	q->id = strcpy(malloc(strlen(str)+1), str);
aba5acdf SH	132	q->parse_qopt = parse_noqopt;
	133	q->print_qopt = print_noqopt;
	134	goto reg;
	135	}
	136	return q;
	137	}
	138
	139
4094db72	140	struct filter_util get_filter_kind(const char str)
aba5acdf SH	141	{
	142	void *dlh;
	143	char buf[256];
	144	struct filter_util *q;
	145
	146	for (q = filter_list; q; q = q->next)
	147	if (strcmp(q->id, str) == 0)
	148	return q;
	149
aa27f88c	150	snprintf(buf, sizeof(buf), "%s/f_%s.so", get_tc_lib(), str);
aba5acdf SH	151	dlh = dlopen(buf, RTLD_LAZY);
	152	if (dlh == NULL) {
	153	dlh = BODY;
	154	if (dlh == NULL) {
	155	dlh = BODY = dlopen(NULL, RTLD_LAZY);
	156	if (dlh == NULL)
	157	goto noexist;
	158	}
	159	}
	160
95812b56	161	snprintf(buf, sizeof(buf), "%s_filter_util", str);
aba5acdf SH	162	q = dlsym(dlh, buf);
	163	if (q == NULL)
	164	goto noexist;
	165
	166	reg:
	167	q->next = filter_list;
	168	filter_list = q;
	169	return q;
aba5acdf SH	170	noexist:
	171	q = malloc(sizeof(*q));
	172	if (q) {
	173	memset(q, 0, sizeof(*q));
	174	strncpy(q->id, str, 15);
	175	q->parse_fopt = parse_nofopt;
	176	q->print_fopt = print_nofopt;
	177	goto reg;
	178	}
	179	return q;
	180	}
	181
c2f3a0f9	182	static void usage(void)
aba5acdf SH	183	{
aba5acdf SH	184	fprintf(stderr, "Usage: tc [ OPTIONS ] OBJECT { COMMAND \| help }\n"
4073448c	185	" tc [-force] -batch file\n"
5bec3484	186	"where OBJECT := { qdisc \| class \| filter \| action \| monitor }\n"
44dcfe82	187	" OPTIONS := { -s[tatistics] \| -d[etails] \| -r[aw] \| -p[retty] \| -b[atch] [file] }\n");
c2f3a0f9	188	}
	189
	190	static int do_cmd(int argc, char **argv)
	191	{
	192	if (matches(*argv, "qdisc") == 0)
	193	return do_qdisc(argc-1, argv+1);
	194
	195	if (matches(*argv, "class") == 0)
	196	return do_class(argc-1, argv+1);
	197
	198	if (matches(*argv, "filter") == 0)
	199	return do_filter(argc-1, argv+1);
	200
	201	if (matches(*argv, "actions") == 0)
	202	return do_action(argc-1, argv+1);
	203
5bec3484 JHS	204	if (matches(*argv, "monitor") == 0)
	205	return do_tcmonitor(argc-1, argv+1);
	206
c2f3a0f9	207	if (matches(*argv, "help") == 0) {
	208	usage();
	209	return 0;
	210	}
ae665a52 SH	211
ae665a52 SH	212	fprintf(stderr, "Object \"%s\" is unknown, try \"tc help\".\n",
c2f3a0f9	213	*argv);
044ebf35	214	return -1;
aba5acdf SH	215	}
aba5acdf SH	216
c2f3a0f9	217	static int batch(const char *name)
aba5acdf	218	{
c2f3a0f9	219	char *line = NULL;
c2f3a0f9	220	size_t len = 0;
08856f02	221	int ret = 0;
c2f3a0f9	222
dd3e9085	223	if (name && strcmp(name, "-") != 0) {
c2f3a0f9	224	if (freopen(name, "r", stdin) == NULL) {
84d66882	225	fprintf(stderr, "Cannot open file \"%s\" for reading: %s\n",
c2f3a0f9	226	name, strerror(errno));
044ebf35	227	return -1;
aba5acdf	228	}
c2f3a0f9	229	}
	230
	231	tc_core_init();
	232
	233	if (rtnl_open(&rth, 0) < 0) {
	234	fprintf(stderr, "Cannot open rtnetlink\n");
	235	return -1;
	236	}
	237
351efcde	238	cmdlineno = 0;
08856f02 SH	239	while (getcmdline(&line, &len, stdin) != -1) {
	240	char *largv[100];
	241	int largc;
aba5acdf	242
c2f3a0f9	243	largc = makeargs(line, largv, 100);
08856f02 SH	244	if (largc == 0)
08856f02 SH	245	continue; /* blank line */
aba5acdf	246
08856f02	247	if (do_cmd(largc, largv)) {
351efcde	248	fprintf(stderr, "Command failed %s:%d\n", name, cmdlineno);
08856f02 SH	249	ret = 1;
	250	if (!force)
	251	break;
aba5acdf	252	}
c2f3a0f9	253	}
8ed63ab1 SH	254	if (line)
8ed63ab1 SH	255	free(line);
7901660a	256
c2f3a0f9	257	rtnl_close(&rth);
	258	return ret;
	259	}
7901660a	260
c2f3a0f9	261
	262	int main(int argc, char **argv)
	263	{
	264	int ret;
dd3e9085	265	int do_batching = 0;
08856f02	266	char *batchfile = NULL;
aba5acdf SH	267
	268	while (argc > 1) {
	269	if (argv[1][0] != '-')
	270	break;
	271	if (matches(argv[1], "-stats") == 0 \|\|
3ea2bf45	272	matches(argv[1], "-statistics") == 0) {
aba5acdf SH	273	++show_stats;
	274	} else if (matches(argv[1], "-details") == 0) {
	275	++show_details;
	276	} else if (matches(argv[1], "-raw") == 0) {
	277	++show_raw;
44dcfe82 SH	278	} else if (matches(argv[1], "-pretty") == 0) {
44dcfe82 SH	279	++show_pretty;
aba5acdf SH	280	} else if (matches(argv[1], "-Version") == 0) {
aba5acdf SH	281	printf("tc utility, iproute2-ss%s\n", SNAPSHOT);
044ebf35	282	return 0;
3ea2bf45 SH	283	} else if (matches(argv[1], "-iec") == 0) {
3ea2bf45 SH	284	++use_iec;
aba5acdf SH	285	} else if (matches(argv[1], "-help") == 0) {
aba5acdf SH	286	usage();
c2f3a0f9	287	return 0;
08856f02 SH	288	} else if (matches(argv[1], "-force") == 0) {
08856f02 SH	289	++force;
c2f3a0f9	290	} else if (matches(argv[1], "-batch") == 0) {
dd3e9085 SH	291	do_batching = 1;
	292	if (argc > 2)
	293	batchfile = argv[2];
08856f02	294	argc--; argv++;
aba5acdf SH	295	} else {
aba5acdf SH	296	fprintf(stderr, "Option \"%s\" is unknown, try \"tc -help\".\n", argv[1]);
044ebf35	297	return -1;
aba5acdf SH	298	}
	299	argc--; argv++;
	300	}
	301
dd3e9085	302	if (do_batching)
08856f02 SH	303	return batch(batchfile);
08856f02 SH	304
c2f3a0f9	305	if (argc <= 1) {
	306	usage();
	307	return 0;
	308	}
	309
aba5acdf	310	tc_core_init();
7901660a SH	311	if (rtnl_open(&rth, 0) < 0) {
	312	fprintf(stderr, "Cannot open rtnetlink\n");
	313	exit(1);
	314	}
aba5acdf	315
c2f3a0f9	316	ret = do_cmd(argc-1, argv+1);
7901660a	317	rtnl_close(&rth);
c2f3a0f9	318
c2f3a0f9	319	return ret;
aba5acdf	320	}