[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 <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"
#include "namespace.h"

int show_stats;
int show_details;
int show_raw;
int show_graph;
int timestamp;

int batch_mode;
int use_iec;
int force;
bool use_names;
int json;
int color;
int oneline;
int numeric;

static char *conf_file;

struct rtnl_handle rth;

static void *BODY;	/* 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 int) RTA_PAYLOAD(opt));
	return 0;
}

static int parse_noqopt(struct qdisc_util *qu, int argc, char **argv,
			struct nlmsghdr *n, const char *dev)
{
	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 int) 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 = calloc(1, sizeof(*q));
	if (q) {
		q->id = strdup(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 = calloc(1, sizeof(*q));
	if (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 filename\n"
		"where  OBJECT := { qdisc | class | filter | chain |\n"
		"		    action | monitor | exec }\n"
		"       OPTIONS := { -V[ersion] | -s[tatistics] | -d[etails] | -r[aw] |\n"
		"		    -o[neline] | -j[son] | -p[retty] | -c[olor]\n"
		"		    -b[atch] [filename] | -n[etns] name | -N[umeric] |\n"
		"		     -nm | -nam[es] | { -cf | -conf } path }\n");
}

static int do_cmd(int argc, char **argv, void *buf, size_t buflen)
{
	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, buf, buflen);
	if (matches(*argv, "chain") == 0)
		return do_chain(argc-1, argv+1, buf, buflen);
	if (matches(*argv, "actions") == 0)
		return do_action(argc-1, argv+1, buf, buflen);
	if (matches(*argv, "monitor") == 0)
		return do_tcmonitor(argc-1, argv+1);
	if (matches(*argv, "exec") == 0)
		return do_exec(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;
}

#define TC_MAX_SUBC	10
static bool batchsize_enabled(int argc, char *argv[])
{
	struct {
		char *c;
		char *subc[TC_MAX_SUBC];
	} table[] = {
		{ "filter", { "add", "delete", "change", "replace", NULL} },
		{ "actions", { "add", "change", "replace", NULL} },
		{ NULL },
	}, *iter;
	char *s;
	int i;

	if (argc < 2)
		return false;

	for (iter = table; iter->c; iter++) {
		if (matches(argv[0], iter->c))
			continue;
		for (i = 0; i < TC_MAX_SUBC; i++) {
			s = iter->subc[i];
			if (s && matches(argv[1], s) == 0)
				return true;
		}
	}

	return false;
}

struct batch_buf {
	struct batch_buf	*next;
	char			buf[16420];	/* sizeof (struct nlmsghdr) +
						   max(sizeof (struct tcmsg) +
						   sizeof (struct tcamsg)) +
						   MAX_MSG */
};

static struct batch_buf *get_batch_buf(struct batch_buf **pool,
				       struct batch_buf **head,
				       struct batch_buf **tail)
{
	struct batch_buf *buf;

	if (*pool == NULL)
		buf = calloc(1, sizeof(struct batch_buf));
	else {
		buf = *pool;
		*pool = (*pool)->next;
		memset(buf, 0, sizeof(struct batch_buf));
	}

	if (*head == NULL)
		*head = *tail = buf;
	else {
		(*tail)->next = buf;
		(*tail) = buf;
	}

	return buf;
}

static void put_batch_bufs(struct batch_buf **pool,
			   struct batch_buf **head,
			   struct batch_buf **tail)
{
	if (*head == NULL || *tail == NULL)
		return;

	if (*pool == NULL)
		*pool = *head;
	else {
		(*tail)->next = *pool;
		*pool = *head;
	}
	*head = NULL;
	*tail = NULL;
}

static void free_batch_bufs(struct batch_buf **pool)
{
	struct batch_buf *buf;

	for (buf = *pool; buf != NULL; buf = *pool) {
		*pool = buf->next;
		free(buf);
	}
	*pool = NULL;
}

static int batch(const char *name)
{
	struct batch_buf *head = NULL, *tail = NULL, *buf_pool = NULL;
	char *largv[100], *largv_next[100];
	char *line, *line_next = NULL;
	bool bs_enabled = false;
	bool lastline = false;
	int largc, largc_next;
	bool bs_enabled_saved;
	bool bs_enabled_next;
	int batchsize = 0;
	size_t len = 0;
	int ret = 0;
	int err;
	bool send;

	batch_mode = 1;
	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;
	if (getcmdline(&line, &len, stdin) == -1)
		goto Exit;
	largc = makeargs(line, largv, 100);
	bs_enabled = batchsize_enabled(largc, largv);
	do {
		if (getcmdline(&line_next, &len, stdin) == -1)
			lastline = true;

		largc_next = makeargs(line_next, largv_next, 100);
		bs_enabled_next = batchsize_enabled(largc_next, largv_next);
		if (bs_enabled) {
			struct batch_buf *buf;

			buf = get_batch_buf(&buf_pool, &head, &tail);
			if (!buf) {
				fprintf(stderr,
					"failed to allocate batch_buf\n");
				return -1;
			}
			++batchsize;
		}

		/*
		 * In batch mode, if we haven't accumulated enough commands
		 * and this is not the last command and this command & next
		 * command both support the batchsize feature, don't send the
		 * message immediately.
		 */
		if (!lastline && bs_enabled && bs_enabled_next
		    && batchsize != MSG_IOV_MAX)
			send = false;
		else
			send = true;

		line = line_next;
		line_next = NULL;
		len = 0;
		bs_enabled_saved = bs_enabled;
		bs_enabled = bs_enabled_next;

		if (largc == 0) {
			largc = largc_next;
			memcpy(largv, largv_next, largc * sizeof(char *));
			continue;	/* blank line */
		}

		err = do_cmd(largc, largv, tail == NULL ? NULL : tail->buf,
			     tail == NULL ? 0 : sizeof(tail->buf));
		fflush(stdout);
		if (err != 0) {
			fprintf(stderr, "Command failed %s:%d\n", name,
				cmdlineno - 1);
			ret = 1;
			if (!force)
				break;
		}
		largc = largc_next;
		memcpy(largv, largv_next, largc * sizeof(char *));

		if (send && bs_enabled_saved) {
			struct iovec *iov, *iovs;
			struct batch_buf *buf;
			struct nlmsghdr *n;

			iov = iovs = malloc(batchsize * sizeof(struct iovec));
			for (buf = head; buf != NULL; buf = buf->next, ++iov) {
				n = (struct nlmsghdr *)&buf->buf;
				iov->iov_base = n;
				iov->iov_len = n->nlmsg_len;
			}

			err = rtnl_talk_iov(&rth, iovs, batchsize, NULL);
			put_batch_bufs(&buf_pool, &head, &tail);
			free(iovs);
			if (err < 0) {
				fprintf(stderr, "Command failed %s:%d\n", name,
					cmdlineno - (batchsize + err) - 1);
				ret = 1;
				if (!force)
					break;
			}
			batchsize = 0;
		}
	} while (!lastline);

	free_batch_bufs(&buf_pool);
Exit:
	free(line);
	rtnl_close(&rth);

	return ret;
}


int main(int argc, char **argv)
{
	int ret;
	char *batch_file = 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) {
			++pretty;
		} else if (matches(argv[1], "-graph") == 0) {
			show_graph = 1;
		} 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) {
			argc--;	argv++;
			if (argc <= 1)
				usage();
			batch_file = argv[1];
		} else if (matches(argv[1], "-netns") == 0) {
			NEXT_ARG();
			if (netns_switch(argv[1]))
				return -1;
		} else if (matches(argv[1], "-Numeric") == 0) {
			++numeric;
		} else if (matches(argv[1], "-names") == 0 ||
				matches(argv[1], "-nm") == 0) {
			use_names = true;
		} else if (matches(argv[1], "-cf") == 0 ||
				matches(argv[1], "-conf") == 0) {
			NEXT_ARG();
			conf_file = argv[1];
		} else if (matches_color(argv[1], &color)) {
		} else if (matches(argv[1], "-timestamp") == 0) {
			timestamp++;
		} else if (matches(argv[1], "-tshort") == 0) {
			++timestamp;
			++timestamp_short;
		} else if (matches(argv[1], "-json") == 0) {
			++json;
		} else if (matches(argv[1], "-oneline") == 0) {
			++oneline;
		} else {
			fprintf(stderr,
				"Option \"%s\" is unknown, try \"tc -help\".\n",
				argv[1]);
			return -1;
		}
		argc--;	argv++;
	}

	_SL_ = oneline ? "\\" : "\n";

	check_enable_color(color, json);

	if (batch_file)
		return batch(batch_file);

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

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

	if (use_names && cls_names_init(conf_file)) {
		ret = -1;
		goto Exit;
	}

	ret = do_cmd(argc-1, argv+1, NULL, 0);
Exit:
	rtnl_close(&rth);

	if (use_names)
		cls_names_uninit();

	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>
aba5acdf SH	19	#include <fcntl.h>
	20	#include <dlfcn.h>
	21	#include <sys/socket.h>
	22	#include <netinet/in.h>
	23	#include <arpa/inet.h>
	24	#include <string.h>
	25	#include <errno.h>
	26
	27	#include "SNAPSHOT.h"
	28	#include "utils.h"
	29	#include "tc_util.h"
	30	#include "tc_common.h"
67e1d73b	31	#include "namespace.h"
aba5acdf	32
32a121cb SH	33	int show_stats;
	34	int show_details;
	35	int show_raw;
32a121cb	36	int show_graph;
32a6fbe5	37	int timestamp;
44dcfe82	38
32a121cb	39	int batch_mode;
32a121cb SH	40	int use_iec;
	41	int force;
	42	bool use_names;
c91d262f	43	int json;
2d165c08	44	int color;
6e8634eb	45	int oneline;
ca697cee	46	int numeric;
4612d04d VK	47
	48	static char *conf_file;
	49
7901660a	50	struct rtnl_handle rth;
aba5acdf	51
32a121cb SH	52	static void BODY; / cached handle dlopen(NULL) */
	53	static struct qdisc_util *qdisc_list;
	54	static struct filter_util *filter_list;
aba5acdf	55
ae665a52	56	static int print_noqopt(struct qdisc_util qu, FILE f,
1798c9d5	57	struct rtattr *opt)
aba5acdf SH	58	{
aba5acdf SH	59	if (opt && RTA_PAYLOAD(opt))
ae665a52	60	fprintf(f, "[Unknown qdisc, optlen=%u] ",
32a121cb	61	(unsigned int) RTA_PAYLOAD(opt));
aba5acdf SH	62	return 0;
	63	}
	64
fec62c0e SH	65	static int parse_noqopt(struct qdisc_util qu, int argc, char *argv,
fec62c0e SH	66	struct nlmsghdr n, const char dev)
aba5acdf SH	67	{
aba5acdf SH	68	if (argc) {
fec62c0e SH	69	fprintf(stderr,
	70	"Unknown qdisc \"%s\", hence option \"%s\" is unparsable\n",
	71	qu->id, *argv);
aba5acdf SH	72	return -1;
	73	}
	74	return 0;
	75	}
	76
	77	static int print_nofopt(struct filter_util qu, FILE f, struct rtattr *opt, __u32 fhandle)
	78	{
	79	if (opt && RTA_PAYLOAD(opt))
ae665a52	80	fprintf(f, "fh %08x [Unknown filter, optlen=%u] ",
32a121cb	81	fhandle, (unsigned int) RTA_PAYLOAD(opt));
aba5acdf SH	82	else if (fhandle)
	83	fprintf(f, "fh %08x ", fhandle);
	84	return 0;
	85	}
	86
fec62c0e SH	87	static int parse_nofopt(struct filter_util qu, char fhandle,
fec62c0e SH	88	int argc, char *argv, struct nlmsghdr n)
aba5acdf SH	89	{
	90	__u32 handle;
	91
	92	if (argc) {
fec62c0e SH	93	fprintf(stderr,
	94	"Unknown filter \"%s\", hence option \"%s\" is unparsable\n",
	95	qu->id, *argv);
aba5acdf SH	96	return -1;
	97	}
	98	if (fhandle) {
	99	struct tcmsg *t = NLMSG_DATA(n);
32a121cb	100
aba5acdf SH	101	if (get_u32(&handle, fhandle, 16)) {
	102	fprintf(stderr, "Unparsable filter ID \"%s\"\n", fhandle);
	103	return -1;
	104	}
	105	t->tcm_handle = handle;
	106	}
	107	return 0;
	108	}
	109
4094db72	110	struct qdisc_util get_qdisc_kind(const char str)
aba5acdf SH	111	{
	112	void *dlh;
	113	char buf[256];
	114	struct qdisc_util *q;
	115
	116	for (q = qdisc_list; q; q = q->next)
	117	if (strcmp(q->id, str) == 0)
	118	return q;
	119
aa27f88c	120	snprintf(buf, sizeof(buf), "%s/q_%s.so", get_tc_lib(), str);
aba5acdf	121	dlh = dlopen(buf, RTLD_LAZY);
b7a45150 SH	122	if (!dlh) {
b7a45150 SH	123	/* look in current binary, only open once */
aba5acdf SH	124	dlh = BODY;
	125	if (dlh == NULL) {
	126	dlh = BODY = dlopen(NULL, RTLD_LAZY);
	127	if (dlh == NULL)
	128	goto noexist;
	129	}
	130	}
	131
95812b56	132	snprintf(buf, sizeof(buf), "%s_qdisc_util", str);
aba5acdf SH	133	q = dlsym(dlh, buf);
	134	if (q == NULL)
	135	goto noexist;
	136
	137	reg:
	138	q->next = qdisc_list;
	139	qdisc_list = q;
	140	return q;
	141
	142	noexist:
f89bb021	143	q = calloc(1, sizeof(*q));
aba5acdf	144	if (q) {
f89bb021	145	q->id = strdup(str);
aba5acdf SH	146	q->parse_qopt = parse_noqopt;
	147	q->print_qopt = print_noqopt;
	148	goto reg;
	149	}
	150	return q;
	151	}
	152
	153
4094db72	154	struct filter_util get_filter_kind(const char str)
aba5acdf SH	155	{
	156	void *dlh;
	157	char buf[256];
	158	struct filter_util *q;
	159
	160	for (q = filter_list; q; q = q->next)
	161	if (strcmp(q->id, str) == 0)
	162	return q;
	163
aa27f88c	164	snprintf(buf, sizeof(buf), "%s/f_%s.so", get_tc_lib(), str);
aba5acdf SH	165	dlh = dlopen(buf, RTLD_LAZY);
	166	if (dlh == NULL) {
	167	dlh = BODY;
	168	if (dlh == NULL) {
	169	dlh = BODY = dlopen(NULL, RTLD_LAZY);
	170	if (dlh == NULL)
	171	goto noexist;
	172	}
	173	}
	174
95812b56	175	snprintf(buf, sizeof(buf), "%s_filter_util", str);
aba5acdf SH	176	q = dlsym(dlh, buf);
	177	if (q == NULL)
	178	goto noexist;
	179
	180	reg:
	181	q->next = filter_list;
	182	filter_list = q;
	183	return q;
aba5acdf	184	noexist:
f89bb021	185	q = calloc(1, sizeof(*q));
aba5acdf	186	if (q) {
aba5acdf SH	187	strncpy(q->id, str, 15);
	188	q->parse_fopt = parse_nofopt;
	189	q->print_fopt = print_nofopt;
	190	goto reg;
	191	}
	192	return q;
	193	}
	194
c2f3a0f9	195	static void usage(void)
aba5acdf	196	{
fec62c0e	197	fprintf(stderr,
8589eb4e MC	198	"Usage: tc [ OPTIONS ] OBJECT { COMMAND \| help }\n"
8589eb4e MC	199	" tc [-force] -batch filename\n"
afcd0699	200	"where OBJECT := { qdisc \| class \| filter \| chain \|\n"
8589eb4e	201	" action \| monitor \| exec }\n"
fec62c0e	202	" OPTIONS := { -V[ersion] \| -s[tatistics] \| -d[etails] \| -r[aw] \|\n"
8589eb4e	203	" -o[neline] \| -j[son] \| -p[retty] \| -c[olor]\n"
ca697cee	204	" -b[atch] [filename] \| -n[etns] name \| -N[umeric] \|\n"
8589eb4e	205	" -nm \| -nam[es] \| { -cf \| -conf } path }\n");
c2f3a0f9	206	}
c2f3a0f9	207
485d0c60	208	static int do_cmd(int argc, char *argv, void buf, size_t buflen)
c2f3a0f9	209	{
	210	if (matches(*argv, "qdisc") == 0)
	211	return do_qdisc(argc-1, argv+1);
c2f3a0f9	212	if (matches(*argv, "class") == 0)
c2f3a0f9	213	return do_class(argc-1, argv+1);
c2f3a0f9	214	if (matches(*argv, "filter") == 0)
485d0c60	215	return do_filter(argc-1, argv+1, buf, buflen);
afcd0699 JP	216	if (matches(*argv, "chain") == 0)
afcd0699 JP	217	return do_chain(argc-1, argv+1, buf, buflen);
c2f3a0f9	218	if (matches(*argv, "actions") == 0)
485d0c60	219	return do_action(argc-1, argv+1, buf, buflen);
5bec3484 JHS	220	if (matches(*argv, "monitor") == 0)
5bec3484 JHS	221	return do_tcmonitor(argc-1, argv+1);
4bd62446 DB	222	if (matches(*argv, "exec") == 0)
4bd62446 DB	223	return do_exec(argc-1, argv+1);
c2f3a0f9	224	if (matches(*argv, "help") == 0) {
	225	usage();
	226	return 0;
	227	}
ae665a52 SH	228
ae665a52 SH	229	fprintf(stderr, "Object \"%s\" is unknown, try \"tc help\".\n",
c2f3a0f9	230	*argv);
044ebf35	231	return -1;
aba5acdf SH	232	}
aba5acdf SH	233
485d0c60 CM	234	#define TC_MAX_SUBC 10
	235	static bool batchsize_enabled(int argc, char *argv[])
	236	{
	237	struct {
	238	char *c;
	239	char *subc[TC_MAX_SUBC];
	240	} table[] = {
fec62c0e SH	241	{ "filter", { "add", "delete", "change", "replace", NULL} },
fec62c0e SH	242	{ "actions", { "add", "change", "replace", NULL} },
485d0c60 CM	243	{ NULL },
	244	}, *iter;
	245	char *s;
	246	int i;
	247
	248	if (argc < 2)
	249	return false;
	250
	251	for (iter = table; iter->c; iter++) {
	252	if (matches(argv[0], iter->c))
	253	continue;
	254	for (i = 0; i < TC_MAX_SUBC; i++) {
	255	s = iter->subc[i];
	256	if (s && matches(argv[1], s) == 0)
	257	return true;
	258	}
	259	}
	260
	261	return false;
	262	}
	263
	264	struct batch_buf {
	265	struct batch_buf *next;
	266	char buf[16420]; /* sizeof (struct nlmsghdr) +
	267	max(sizeof (struct tcmsg) +
	268	sizeof (struct tcamsg)) +
	269	MAX_MSG */
	270	};
	271
	272	static struct batch_buf get_batch_buf(struct batch_buf *pool,
	273	struct batch_buf **head,
	274	struct batch_buf **tail)
	275	{
	276	struct batch_buf *buf;
	277
	278	if (*pool == NULL)
fec62c0e	279	buf = calloc(1, sizeof(struct batch_buf));
485d0c60 CM	280	else {
	281	buf = *pool;
	282	pool = (pool)->next;
fec62c0e	283	memset(buf, 0, sizeof(struct batch_buf));
485d0c60 CM	284	}
	285
	286	if (*head == NULL)
	287	head = tail = buf;
	288	else {
	289	(*tail)->next = buf;
	290	(*tail) = buf;
	291	}
	292
	293	return buf;
	294	}
	295
	296	static void put_batch_bufs(struct batch_buf **pool,
	297	struct batch_buf **head,
	298	struct batch_buf **tail)
	299	{
	300	if (head == NULL \|\| tail == NULL)
	301	return;
	302
	303	if (*pool == NULL)
	304	pool = head;
	305	else {
	306	(tail)->next = pool;
	307	pool = head;
	308	}
	309	*head = NULL;
	310	*tail = NULL;
	311	}
	312
	313	static void free_batch_bufs(struct batch_buf **pool)
	314	{
	315	struct batch_buf *buf;
	316
	317	for (buf = pool; buf != NULL; buf = pool) {
	318	*pool = buf->next;
	319	free(buf);
	320	}
	321	*pool = NULL;
	322	}
	323
c2f3a0f9	324	static int batch(const char *name)
aba5acdf	325	{
485d0c60 CM	326	struct batch_buf head = NULL, tail = NULL, *buf_pool = NULL;
	327	char largv[100], largv_next[100];
	328	char line, line_next = NULL;
485d0c60 CM	329	bool bs_enabled = false;
	330	bool lastline = false;
	331	int largc, largc_next;
	332	bool bs_enabled_saved;
6358bbc3	333	bool bs_enabled_next;
485d0c60	334	int batchsize = 0;
c2f3a0f9	335	size_t len = 0;
08856f02	336	int ret = 0;
b1333924	337	int err;
485d0c60	338	bool send;
c2f3a0f9	339
a3aa47a5	340	batch_mode = 1;
dd3e9085	341	if (name && strcmp(name, "-") != 0) {
c2f3a0f9	342	if (freopen(name, "r", stdin) == NULL) {
fec62c0e SH	343	fprintf(stderr,
fec62c0e SH	344	"Cannot open file \"%s\" for reading: %s\n",
c2f3a0f9	345	name, strerror(errno));
044ebf35	346	return -1;
aba5acdf	347	}
c2f3a0f9	348	}
	349
	350	tc_core_init();
	351
	352	if (rtnl_open(&rth, 0) < 0) {
	353	fprintf(stderr, "Cannot open rtnetlink\n");
	354	return -1;
	355	}
	356
351efcde	357	cmdlineno = 0;
485d0c60 CM	358	if (getcmdline(&line, &len, stdin) == -1)
	359	goto Exit;
	360	largc = makeargs(line, largv, 100);
	361	bs_enabled = batchsize_enabled(largc, largv);
485d0c60 CM	362	do {
	363	if (getcmdline(&line_next, &len, stdin) == -1)
	364	lastline = true;
	365
	366	largc_next = makeargs(line_next, largv_next, 100);
	367	bs_enabled_next = batchsize_enabled(largc_next, largv_next);
	368	if (bs_enabled) {
	369	struct batch_buf *buf;
	370
	371	buf = get_batch_buf(&buf_pool, &head, &tail);
	372	if (!buf) {
	373	fprintf(stderr,
	374	"failed to allocate batch_buf\n");
	375	return -1;
	376	}
	377	++batchsize;
	378	}
aba5acdf	379
485d0c60 CM	380	/*
	381	* In batch mode, if we haven't accumulated enough commands
	382	* and this is not the last command and this command & next
	383	* command both support the batchsize feature, don't send the
	384	* message immediately.
	385	*/
	386	if (!lastline && bs_enabled && bs_enabled_next
	387	&& batchsize != MSG_IOV_MAX)
	388	send = false;
	389	else
	390	send = true;
	391
	392	line = line_next;
	393	line_next = NULL;
	394	len = 0;
	395	bs_enabled_saved = bs_enabled;
	396	bs_enabled = bs_enabled_next;
485d0c60 CM	397
	398	if (largc == 0) {
	399	largc = largc_next;
	400	memcpy(largv, largv_next, largc * sizeof(char *));
08856f02	401	continue; /* blank line */
485d0c60	402	}
aba5acdf	403
b1333924	404	err = do_cmd(largc, largv, tail == NULL ? NULL : tail->buf,
fec62c0e	405	tail == NULL ? 0 : sizeof(tail->buf));
d66fdfda	406	fflush(stdout);
b1333924	407	if (err != 0) {
485d0c60 CM	408	fprintf(stderr, "Command failed %s:%d\n", name,
485d0c60 CM	409	cmdlineno - 1);
08856f02 SH	410	ret = 1;
	411	if (!force)
	412	break;
aba5acdf	413	}
485d0c60 CM	414	largc = largc_next;
	415	memcpy(largv, largv_next, largc * sizeof(char *));
	416
	417	if (send && bs_enabled_saved) {
	418	struct iovec iov, iovs;
	419	struct batch_buf *buf;
	420	struct nlmsghdr *n;
	421
fec62c0e	422	iov = iovs = malloc(batchsize * sizeof(struct iovec));
485d0c60 CM	423	for (buf = head; buf != NULL; buf = buf->next, ++iov) {
	424	n = (struct nlmsghdr *)&buf->buf;
	425	iov->iov_base = n;
	426	iov->iov_len = n->nlmsg_len;
	427	}
	428
b1333924 VB	429	err = rtnl_talk_iov(&rth, iovs, batchsize, NULL);
	430	put_batch_bufs(&buf_pool, &head, &tail);
	431	free(iovs);
	432	if (err < 0) {
485d0c60	433	fprintf(stderr, "Command failed %s:%d\n", name,
b1333924 VB	434	cmdlineno - (batchsize + err) - 1);
	435	ret = 1;
	436	if (!force)
	437	break;
485d0c60	438	}
485d0c60	439	batchsize = 0;
485d0c60 CM	440	}
485d0c60 CM	441	} while (!lastline);
7901660a	442
485d0c60 CM	443	free_batch_bufs(&buf_pool);
	444	Exit:
	445	free(line);
c2f3a0f9	446	rtnl_close(&rth);
485d0c60	447
c2f3a0f9	448	return ret;
c2f3a0f9	449	}
7901660a	450
c2f3a0f9	451
	452	int main(int argc, char **argv)
	453	{
	454	int ret;
a3aa47a5	455	char *batch_file = NULL;
aba5acdf SH	456
	457	while (argc > 1) {
	458	if (argv[1][0] != '-')
	459	break;
	460	if (matches(argv[1], "-stats") == 0 \|\|
3ea2bf45	461	matches(argv[1], "-statistics") == 0) {
aba5acdf SH	462	++show_stats;
	463	} else if (matches(argv[1], "-details") == 0) {
	464	++show_details;
	465	} else if (matches(argv[1], "-raw") == 0) {
	466	++show_raw;
44dcfe82	467	} else if (matches(argv[1], "-pretty") == 0) {
54336567	468	++pretty;
d954b34a VK	469	} else if (matches(argv[1], "-graph") == 0) {
d954b34a VK	470	show_graph = 1;
aba5acdf SH	471	} else if (matches(argv[1], "-Version") == 0) {
aba5acdf SH	472	printf("tc utility, iproute2-ss%s\n", SNAPSHOT);
044ebf35	473	return 0;
3ea2bf45 SH	474	} else if (matches(argv[1], "-iec") == 0) {
3ea2bf45 SH	475	++use_iec;
aba5acdf SH	476	} else if (matches(argv[1], "-help") == 0) {
aba5acdf SH	477	usage();
c2f3a0f9	478	return 0;
08856f02 SH	479	} else if (matches(argv[1], "-force") == 0) {
08856f02 SH	480	++force;
4612d04d	481	} else if (matches(argv[1], "-batch") == 0) {
08856f02	482	argc--; argv++;
a3aa47a5 SH	483	if (argc <= 1)
	484	usage();
	485	batch_file = argv[1];
67e1d73b VK	486	} else if (matches(argv[1], "-netns") == 0) {
	487	NEXT_ARG();
	488	if (netns_switch(argv[1]))
	489	return -1;
ca697cee HL	490	} else if (matches(argv[1], "-Numeric") == 0) {
ca697cee HL	491	++numeric;
4612d04d VK	492	} else if (matches(argv[1], "-names") == 0 \|\|
	493	matches(argv[1], "-nm") == 0) {
	494	use_names = true;
	495	} else if (matches(argv[1], "-cf") == 0 \|\|
	496	matches(argv[1], "-conf") == 0) {
	497	NEXT_ARG();
	498	conf_file = argv[1];
ff1ab8ed	499	} else if (matches_color(argv[1], &color)) {
32a6fbe5 ED	500	} else if (matches(argv[1], "-timestamp") == 0) {
	501	timestamp++;
	502	} else if (matches(argv[1], "-tshort") == 0) {
	503	++timestamp;
	504	++timestamp_short;
c91d262f JP	505	} else if (matches(argv[1], "-json") == 0) {
c91d262f JP	506	++json;
6e8634eb RM	507	} else if (matches(argv[1], "-oneline") == 0) {
6e8634eb RM	508	++oneline;
aba5acdf	509	} else {
fec62c0e SH	510	fprintf(stderr,
	511	"Option \"%s\" is unknown, try \"tc -help\".\n",
	512	argv[1]);
044ebf35	513	return -1;
aba5acdf SH	514	}
	515	argc--; argv++;
	516	}
	517
6e8634eb RM	518	_SL_ = oneline ? "\\" : "\n";
6e8634eb RM	519
4d82962c	520	check_enable_color(color, json);
2d165c08	521
a3aa47a5 SH	522	if (batch_file)
a3aa47a5 SH	523	return batch(batch_file);
08856f02	524
c2f3a0f9	525	if (argc <= 1) {
	526	usage();
	527	return 0;
	528	}
	529
aba5acdf	530	tc_core_init();
7901660a SH	531	if (rtnl_open(&rth, 0) < 0) {
	532	fprintf(stderr, "Cannot open rtnetlink\n");
	533	exit(1);
	534	}
aba5acdf	535
4612d04d VK	536	if (use_names && cls_names_init(conf_file)) {
	537	ret = -1;
	538	goto Exit;
	539	}
	540
485d0c60	541	ret = do_cmd(argc-1, argv+1, NULL, 0);
4612d04d	542	Exit:
7901660a	543	rtnl_close(&rth);
c2f3a0f9	544
4612d04d VK	545	if (use_names)
	546	cls_names_uninit();
	547
c2f3a0f9	548	return ret;
aba5acdf	549	}