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#include "namespace.h"
33
34int show_stats = 0;
35int show_details = 0;
36int show_raw = 0;
37int show_pretty = 0;
38int show_graph = 0;
39int timestamp;
40
41int batch_mode = 0;
42int resolve_hosts = 0;
43int use_iec = 0;
44int force = 0;
45bool use_names = false;
46
47static char *conf_file;
48
49struct rtnl_handle rth;
50
51static void *BODY = NULL;	/* cached handle dlopen(NULL) */
52static struct qdisc_util * qdisc_list;
53static struct filter_util * filter_list;
54
55#ifdef ANDROID
56extern struct qdisc_util cbq_qdisc_util;
57extern struct qdisc_util htb_qdisc_util;
58extern struct qdisc_util ingress_qdisc_util;
59extern struct filter_util u32_filter_util;
60#endif
61
62static int print_noqopt(struct qdisc_util *qu, FILE *f,
63			struct rtattr *opt)
64{
65	if (opt && RTA_PAYLOAD(opt))
66		fprintf(f, "[Unknown qdisc, optlen=%u] ",
67			(unsigned) RTA_PAYLOAD(opt));
68	return 0;
69}
70
71static int parse_noqopt(struct qdisc_util *qu, int argc, char **argv, struct nlmsghdr *n)
72{
73	if (argc) {
74		fprintf(stderr, "Unknown qdisc \"%s\", hence option \"%s\" is unparsable\n", qu->id, *argv);
75		return -1;
76	}
77	return 0;
78}
79
80static int print_nofopt(struct filter_util *qu, FILE *f, struct rtattr *opt, __u32 fhandle)
81{
82	if (opt && RTA_PAYLOAD(opt))
83		fprintf(f, "fh %08x [Unknown filter, optlen=%u] ",
84			fhandle, (unsigned) RTA_PAYLOAD(opt));
85	else if (fhandle)
86		fprintf(f, "fh %08x ", fhandle);
87	return 0;
88}
89
90static int parse_nofopt(struct filter_util *qu, char *fhandle, int argc, char **argv, struct nlmsghdr *n)
91{
92	__u32 handle;
93
94	if (argc) {
95		fprintf(stderr, "Unknown filter \"%s\", hence option \"%s\" is unparsable\n", qu->id, *argv);
96		return -1;
97	}
98	if (fhandle) {
99		struct tcmsg *t = NLMSG_DATA(n);
100		if (get_u32(&handle, fhandle, 16)) {
101			fprintf(stderr, "Unparsable filter ID \"%s\"\n", fhandle);
102			return -1;
103		}
104		t->tcm_handle = handle;
105	}
106	return 0;
107}
108
109struct qdisc_util *get_qdisc_kind(const char *str)
110{
111	void *dlh;
112	char buf[256];
113	struct qdisc_util *q;
114
115#ifdef ANDROID
116	if (!strcmp(str, "cbq"))
117		return &cbq_qdisc_util;
118	else if (!strcmp(str, "htb"))
119		return &htb_qdisc_util;
120	else if (!strcmp(str, "ingress"))
121		return &ingress_qdisc_util;
122	else {
123		fprintf(stderr, "Android does not support qdisc '%s'\n", str);
124		return NULL;
125	}
126#endif
127	for (q = qdisc_list; q; q = q->next)
128		if (strcmp(q->id, str) == 0)
129			return q;
130
131	snprintf(buf, sizeof(buf), "%s/q_%s.so", get_tc_lib(), str);
132	dlh = dlopen(buf, RTLD_LAZY);
133	if (!dlh) {
134		/* look in current binary, only open once */
135		dlh = BODY;
136		if (dlh == NULL) {
137			dlh = BODY = dlopen(NULL, RTLD_LAZY);
138			if (dlh == NULL)
139				goto noexist;
140		}
141	}
142
143	snprintf(buf, sizeof(buf), "%s_qdisc_util", str);
144	q = dlsym(dlh, buf);
145	if (q == NULL)
146		goto noexist;
147
148reg:
149	q->next = qdisc_list;
150	qdisc_list = q;
151	return q;
152
153noexist:
154	q = malloc(sizeof(*q));
155	if (q) {
156
157		memset(q, 0, sizeof(*q));
158		q->id = strcpy(malloc(strlen(str)+1), str);
159		q->parse_qopt = parse_noqopt;
160		q->print_qopt = print_noqopt;
161		goto reg;
162	}
163	return q;
164}
165
166
167struct filter_util *get_filter_kind(const char *str)
168{
169	void *dlh;
170	char buf[256];
171	struct filter_util *q;
172#ifdef ANDROID
173	if (!strcmp(str, "u32"))
174		return &u32_filter_util;
175	else {
176		fprintf(stderr, "Android does not support filter '%s'\n", str);
177		return NULL;
178	}
179#endif
180
181	for (q = filter_list; q; q = q->next)
182		if (strcmp(q->id, str) == 0)
183			return q;
184
185	snprintf(buf, sizeof(buf), "%s/f_%s.so", get_tc_lib(), str);
186	dlh = dlopen(buf, RTLD_LAZY);
187	if (dlh == NULL) {
188		dlh = BODY;
189		if (dlh == NULL) {
190			dlh = BODY = dlopen(NULL, RTLD_LAZY);
191			if (dlh == NULL)
192				goto noexist;
193		}
194	}
195
196	snprintf(buf, sizeof(buf), "%s_filter_util", str);
197	q = dlsym(dlh, buf);
198	if (q == NULL)
199		goto noexist;
200
201reg:
202	q->next = filter_list;
203	filter_list = q;
204	return q;
205noexist:
206	q = malloc(sizeof(*q));
207	if (q) {
208		memset(q, 0, sizeof(*q));
209		strncpy(q->id, str, 15);
210		q->parse_fopt = parse_nofopt;
211		q->print_fopt = print_nofopt;
212		goto reg;
213	}
214	return q;
215}
216
217static void usage(void)
218{
219	fprintf(stderr, "Usage: tc [ OPTIONS ] OBJECT { COMMAND | help }\n"
220#ifdef ANDROID
221			"       tc [-force]\n"
222#else
223			"       tc [-force] -batch filename\n"
224#endif
225	                "where  OBJECT := { qdisc | class | filter | action | monitor | exec }\n"
226	                "       OPTIONS := { -s[tatistics] | -d[etails] | -r[aw] | -p[retty] | -b[atch] [filename] | "
227			"-n[etns] name |\n"
228			"                    -nm | -nam[es] | { -cf | -conf } path }\n");
229}
230
231static int do_cmd(int argc, char **argv)
232{
233	if (matches(*argv, "qdisc") == 0)
234		return do_qdisc(argc-1, argv+1);
235	if (matches(*argv, "class") == 0)
236		return do_class(argc-1, argv+1);
237	if (matches(*argv, "filter") == 0)
238		return do_filter(argc-1, argv+1);
239	if (matches(*argv, "actions") == 0)
240		return do_action(argc-1, argv+1);
241	if (matches(*argv, "monitor") == 0)
242		return do_tcmonitor(argc-1, argv+1);
243	if (matches(*argv, "exec") == 0)
244		return do_exec(argc-1, argv+1);
245	if (matches(*argv, "help") == 0) {
246		usage();
247		return 0;
248	}
249
250	fprintf(stderr, "Object \"%s\" is unknown, try \"tc help\".\n",
251		*argv);
252	return -1;
253}
254
255#ifndef ANDROID
256static int batch(const char *name)
257{
258	char *line = NULL;
259	size_t len = 0;
260	int ret = 0;
261
262	batch_mode = 1;
263	if (name && strcmp(name, "-") != 0) {
264		if (freopen(name, "r", stdin) == NULL) {
265			fprintf(stderr, "Cannot open file \"%s\" for reading: %s\n",
266				name, strerror(errno));
267			return -1;
268		}
269	}
270
271	tc_core_init();
272
273	if (rtnl_open(&rth, 0) < 0) {
274		fprintf(stderr, "Cannot open rtnetlink\n");
275		return -1;
276	}
277
278	cmdlineno = 0;
279	while (getcmdline(&line, &len, stdin) != -1) {
280		char *largv[100];
281		int largc;
282
283		largc = makeargs(line, largv, 100);
284		if (largc == 0)
285			continue;	/* blank line */
286
287		if (do_cmd(largc, largv)) {
288			fprintf(stderr, "Command failed %s:%d\n", name, cmdlineno);
289			ret = 1;
290			if (!force)
291				break;
292		}
293	}
294	if (line)
295		free(line);
296
297	rtnl_close(&rth);
298	return ret;
299}
300#endif
301
302
303int main(int argc, char **argv)
304{
305	int ret;
306#ifndef ANDROID
307	char *batch_file = NULL;
308#endif
309
310	while (argc > 1) {
311		if (argv[1][0] != '-')
312			break;
313		if (matches(argv[1], "-stats") == 0 ||
314			 matches(argv[1], "-statistics") == 0) {
315			++show_stats;
316		} else if (matches(argv[1], "-details") == 0) {
317			++show_details;
318		} else if (matches(argv[1], "-raw") == 0) {
319			++show_raw;
320		} else if (matches(argv[1], "-pretty") == 0) {
321			++show_pretty;
322		} else if (matches(argv[1], "-graph") == 0) {
323			show_graph = 1;
324		} else if (matches(argv[1], "-Version") == 0) {
325			printf("tc utility, iproute2-ss%s\n", SNAPSHOT);
326			return 0;
327		} else if (matches(argv[1], "-iec") == 0) {
328			++use_iec;
329		} else if (matches(argv[1], "-help") == 0) {
330			usage();
331			return 0;
332		} else if (matches(argv[1], "-force") == 0) {
333			++force;
334#ifndef ANDROID
335		} else if (matches(argv[1], "-batch") == 0) {
336			argc--;	argv++;
337			if (argc <= 1)
338				usage();
339			batch_file = argv[1];
340#endif
341		} else if (matches(argv[1], "-netns") == 0) {
342			NEXT_ARG();
343			if (netns_switch(argv[1]))
344				return -1;
345		} else if (matches(argv[1], "-names") == 0 ||
346				matches(argv[1], "-nm") == 0) {
347			use_names = true;
348		} else if (matches(argv[1], "-cf") == 0 ||
349				matches(argv[1], "-conf") == 0) {
350			NEXT_ARG();
351			conf_file = argv[1];
352		} else if (matches(argv[1], "-timestamp") == 0) {
353			timestamp++;
354		} else if (matches(argv[1], "-tshort") == 0) {
355			++timestamp;
356			++timestamp_short;
357		} else {
358			fprintf(stderr, "Option \"%s\" is unknown, try \"tc -help\".\n", argv[1]);
359			return -1;
360		}
361		argc--;	argv++;
362	}
363
364#ifndef ANDROID
365	if (batch_file)
366		return batch(batch_file);
367#endif
368
369	if (argc <= 1) {
370		usage();
371		return 0;
372	}
373
374	tc_core_init();
375	if (rtnl_open(&rth, 0) < 0) {
376		fprintf(stderr, "Cannot open rtnetlink\n");
377		exit(1);
378	}
379
380	if (use_names && cls_names_init(conf_file)) {
381		ret = -1;
382		goto Exit;
383	}
384
385	ret = do_cmd(argc-1, argv+1);
386Exit:
387	rtnl_close(&rth);
388
389	if (use_names)
390		cls_names_uninit();
391
392	return ret;
393}
394