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