1/*
2 * tc_util.c		Misc TC utility functions.
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 */
12
13#include <stdio.h>
14#include <stdlib.h>
15#include <unistd.h>
16#include <syslog.h>
17#include <fcntl.h>
18#include <sys/socket.h>
19#include <sys/param.h>
20#include <netinet/in.h>
21#include <arpa/inet.h>
22#include <string.h>
23#include <math.h>
24#include <errno.h>
25
26#include "utils.h"
27#include "names.h"
28#include "tc_util.h"
29#include "tc_common.h"
30
31#ifndef LIBDIR
32#define LIBDIR "/usr/lib"
33#endif
34
35static struct db_names *cls_names = NULL;
36
37#define NAMES_DB "/etc/iproute2/tc_cls"
38
39int cls_names_init(char *path)
40{
41	int ret;
42
43	cls_names = db_names_alloc();
44	if (!cls_names)
45		return -1;
46
47	ret = db_names_load(cls_names, path ?: NAMES_DB);
48	if (ret == -ENOENT && path) {
49		fprintf(stderr, "Can't open class names file: %s\n", path);
50		return -1;
51	}
52	if (ret) {
53		db_names_free(cls_names);
54		cls_names = NULL;
55	}
56
57	return 0;
58}
59
60void cls_names_uninit(void)
61{
62	db_names_free(cls_names);
63}
64
65const char *get_tc_lib(void)
66{
67	const char *lib_dir;
68
69	lib_dir = getenv("TC_LIB_DIR");
70	if (!lib_dir)
71		lib_dir = LIBDIR "/tc/";
72
73	return lib_dir;
74}
75
76int get_qdisc_handle(__u32 *h, const char *str)
77{
78	__u32 maj;
79	char *p;
80
81	maj = TC_H_UNSPEC;
82	if (strcmp(str, "none") == 0)
83		goto ok;
84	maj = strtoul(str, &p, 16);
85	if (p == str)
86		return -1;
87	maj <<= 16;
88	if (*p != ':' && *p!=0)
89		return -1;
90ok:
91	*h = maj;
92	return 0;
93}
94
95int get_tc_classid(__u32 *h, const char *str)
96{
97	__u32 maj, min;
98	char *p;
99
100	maj = TC_H_ROOT;
101	if (strcmp(str, "root") == 0)
102		goto ok;
103	maj = TC_H_UNSPEC;
104	if (strcmp(str, "none") == 0)
105		goto ok;
106	maj = strtoul(str, &p, 16);
107	if (p == str) {
108		maj = 0;
109		if (*p != ':')
110			return -1;
111	}
112	if (*p == ':') {
113		if (maj >= (1<<16))
114			return -1;
115		maj <<= 16;
116		str = p+1;
117		min = strtoul(str, &p, 16);
118		if (*p != 0)
119			return -1;
120		if (min >= (1<<16))
121			return -1;
122		maj |= min;
123	} else if (*p != 0)
124		return -1;
125
126ok:
127	*h = maj;
128	return 0;
129}
130
131int print_tc_classid(char *buf, int blen, __u32 h)
132{
133	SPRINT_BUF(handle) = {};
134	int hlen = SPRINT_BSIZE - 1;
135
136	if (h == TC_H_ROOT)
137		sprintf(handle, "root");
138	else if (h == TC_H_UNSPEC)
139		snprintf(handle, hlen, "none");
140	else if (TC_H_MAJ(h) == 0)
141		snprintf(handle, hlen, ":%x", TC_H_MIN(h));
142	else if (TC_H_MIN(h) == 0)
143		snprintf(handle, hlen, "%x:", TC_H_MAJ(h) >> 16);
144	else
145		snprintf(handle, hlen, "%x:%x", TC_H_MAJ(h) >> 16, TC_H_MIN(h));
146
147	if (use_names) {
148		char clname[IDNAME_MAX] = {};
149
150		if (id_to_name(cls_names, h, clname))
151			snprintf(buf, blen, "%s#%s", clname, handle);
152		else
153			snprintf(buf, blen, "%s", handle);
154	} else {
155		snprintf(buf, blen, "%s", handle);
156	}
157
158	return 0;
159}
160
161char *sprint_tc_classid(__u32 h, char *buf)
162{
163	if (print_tc_classid(buf, SPRINT_BSIZE-1, h))
164		strcpy(buf, "???");
165	return buf;
166}
167
168/* See http://physics.nist.gov/cuu/Units/binary.html */
169static const struct rate_suffix {
170	const char *name;
171	double scale;
172} suffixes[] = {
173	{ "bit",	1. },
174	{ "Kibit",	1024. },
175	{ "kbit",	1000. },
176	{ "mibit",	1024.*1024. },
177	{ "mbit",	1000000. },
178	{ "gibit",	1024.*1024.*1024. },
179	{ "gbit",	1000000000. },
180	{ "tibit",	1024.*1024.*1024.*1024. },
181	{ "tbit",	1000000000000. },
182	{ "Bps",	8. },
183	{ "KiBps",	8.*1024. },
184	{ "KBps",	8000. },
185	{ "MiBps",	8.*1024*1024. },
186	{ "MBps",	8000000. },
187	{ "GiBps",	8.*1024.*1024.*1024. },
188	{ "GBps",	8000000000. },
189	{ "TiBps",	8.*1024.*1024.*1024.*1024. },
190	{ "TBps",	8000000000000. },
191	{ NULL }
192};
193
194
195int get_rate(unsigned *rate, const char *str)
196{
197	char *p;
198	double bps = strtod(str, &p);
199	const struct rate_suffix *s;
200
201	if (p == str)
202		return -1;
203
204	for (s = suffixes; s->name; ++s) {
205		if (strcasecmp(s->name, p) == 0) {
206			bps *= s->scale;
207			p += strlen(p);
208			break;
209		}
210	}
211
212	if (*p)
213		return -1; /* unknown suffix */
214
215	bps /= 8; /* -> bytes per second */
216	*rate = bps;
217	/* detect if an overflow happened */
218	if (*rate != floor(bps))
219		return -1;
220	return 0;
221}
222
223int get_rate64(__u64 *rate, const char *str)
224{
225	char *p;
226	double bps = strtod(str, &p);
227	const struct rate_suffix *s;
228
229	if (p == str)
230		return -1;
231
232	for (s = suffixes; s->name; ++s) {
233		if (strcasecmp(s->name, p) == 0) {
234			bps *= s->scale;
235			p += strlen(p);
236			break;
237		}
238	}
239
240	if (*p)
241		return -1; /* unknown suffix */
242
243	bps /= 8; /* -> bytes per second */
244	*rate = bps;
245	return 0;
246}
247
248void print_rate(char *buf, int len, __u64 rate)
249{
250	extern int use_iec;
251	unsigned long kilo = use_iec ? 1024 : 1000;
252	const char *str = use_iec ? "i" : "";
253	static char *units[5] = {"", "K", "M", "G", "T"};
254	int i;
255
256	rate <<= 3; /* bytes/sec -> bits/sec */
257
258	for (i = 0; i < ARRAY_SIZE(units) - 1; i++)  {
259		if (rate < kilo)
260			break;
261		if (((rate % kilo) != 0) && rate < 1000*kilo)
262			break;
263		rate /= kilo;
264	}
265
266	snprintf(buf, len, "%.0f%s%sbit", (double)rate, units[i], str);
267}
268
269char * sprint_rate(__u64 rate, char *buf)
270{
271	print_rate(buf, SPRINT_BSIZE-1, rate);
272	return buf;
273}
274
275int get_time(unsigned *time, const char *str)
276{
277	double t;
278	char *p;
279
280	t = strtod(str, &p);
281	if (p == str)
282		return -1;
283
284	if (*p) {
285		if (strcasecmp(p, "s") == 0 || strcasecmp(p, "sec")==0 ||
286		    strcasecmp(p, "secs")==0)
287			t *= TIME_UNITS_PER_SEC;
288		else if (strcasecmp(p, "ms") == 0 || strcasecmp(p, "msec")==0 ||
289			 strcasecmp(p, "msecs") == 0)
290			t *= TIME_UNITS_PER_SEC/1000;
291		else if (strcasecmp(p, "us") == 0 || strcasecmp(p, "usec")==0 ||
292			 strcasecmp(p, "usecs") == 0)
293			t *= TIME_UNITS_PER_SEC/1000000;
294		else
295			return -1;
296	}
297
298	*time = t;
299	return 0;
300}
301
302
303void print_time(char *buf, int len, __u32 time)
304{
305	double tmp = time;
306
307	if (tmp >= TIME_UNITS_PER_SEC)
308		snprintf(buf, len, "%.1fs", tmp/TIME_UNITS_PER_SEC);
309	else if (tmp >= TIME_UNITS_PER_SEC/1000)
310		snprintf(buf, len, "%.1fms", tmp/(TIME_UNITS_PER_SEC/1000));
311	else
312		snprintf(buf, len, "%uus", time);
313}
314
315char * sprint_time(__u32 time, char *buf)
316{
317	print_time(buf, SPRINT_BSIZE-1, time);
318	return buf;
319}
320
321char * sprint_ticks(__u32 ticks, char *buf)
322{
323	return sprint_time(tc_core_tick2time(ticks), buf);
324}
325
326int get_size(unsigned *size, const char *str)
327{
328	double sz;
329	char *p;
330
331	sz = strtod(str, &p);
332	if (p == str)
333		return -1;
334
335	if (*p) {
336		if (strcasecmp(p, "kb") == 0 || strcasecmp(p, "k")==0)
337			sz *= 1024;
338		else if (strcasecmp(p, "gb") == 0 || strcasecmp(p, "g")==0)
339			sz *= 1024*1024*1024;
340		else if (strcasecmp(p, "gbit") == 0)
341			sz *= 1024*1024*1024/8;
342		else if (strcasecmp(p, "mb") == 0 || strcasecmp(p, "m")==0)
343			sz *= 1024*1024;
344		else if (strcasecmp(p, "mbit") == 0)
345			sz *= 1024*1024/8;
346		else if (strcasecmp(p, "kbit") == 0)
347			sz *= 1024/8;
348		else if (strcasecmp(p, "b") != 0)
349			return -1;
350	}
351
352	*size = sz;
353	return 0;
354}
355
356int get_size_and_cell(unsigned *size, int *cell_log, char *str)
357{
358	char * slash = strchr(str, '/');
359
360	if (slash)
361		*slash = 0;
362
363	if (get_size(size, str))
364		return -1;
365
366	if (slash) {
367		int cell;
368		int i;
369
370		if (get_integer(&cell, slash+1, 0))
371			return -1;
372		*slash = '/';
373
374		for (i=0; i<32; i++) {
375			if ((1<<i) == cell) {
376				*cell_log = i;
377				return 0;
378			}
379		}
380		return -1;
381	}
382	return 0;
383}
384
385void print_size(char *buf, int len, __u32 sz)
386{
387	double tmp = sz;
388
389	if (sz >= 1024*1024 && fabs(1024*1024*rint(tmp/(1024*1024)) - sz) < 1024)
390		snprintf(buf, len, "%gMb", rint(tmp/(1024*1024)));
391	else if (sz >= 1024 && fabs(1024*rint(tmp/1024) - sz) < 16)
392		snprintf(buf, len, "%gKb", rint(tmp/1024));
393	else
394		snprintf(buf, len, "%ub", sz);
395}
396
397char * sprint_size(__u32 size, char *buf)
398{
399	print_size(buf, SPRINT_BSIZE-1, size);
400	return buf;
401}
402
403void print_qdisc_handle(char *buf, int len, __u32 h)
404{
405	snprintf(buf, len, "%x:", TC_H_MAJ(h)>>16);
406}
407
408char * sprint_qdisc_handle(__u32 h, char *buf)
409{
410	print_qdisc_handle(buf, SPRINT_BSIZE-1, h);
411	return buf;
412}
413
414char * action_n2a(int action, char *buf, int len)
415{
416	switch (action) {
417	case -1:
418		return "continue";
419		break;
420	case TC_ACT_OK:
421		return "pass";
422		break;
423	case TC_ACT_SHOT:
424		return "drop";
425		break;
426	case TC_ACT_RECLASSIFY:
427		return "reclassify";
428	case TC_ACT_PIPE:
429		return "pipe";
430	case TC_ACT_STOLEN:
431		return "stolen";
432	default:
433		snprintf(buf, len, "%d", action);
434		return buf;
435	}
436}
437
438int action_a2n(char *arg, int *result)
439{
440	int res;
441
442	if (matches(arg, "continue") == 0)
443		res = -1;
444	else if (matches(arg, "drop") == 0)
445		res = TC_ACT_SHOT;
446	else if (matches(arg, "shot") == 0)
447		res = TC_ACT_SHOT;
448	else if (matches(arg, "pass") == 0)
449		res = TC_ACT_OK;
450	else if (strcmp(arg, "ok") == 0)
451		res = TC_ACT_OK;
452	else if (matches(arg, "reclassify") == 0)
453		res = TC_ACT_RECLASSIFY;
454	else {
455		char dummy;
456		if (sscanf(arg, "%d%c", &res, &dummy) != 1)
457			return -1;
458	}
459	*result = res;
460	return 0;
461}
462
463int get_linklayer(unsigned *val, const char *arg)
464{
465	int res;
466
467	if (matches(arg, "ethernet") == 0)
468		res = LINKLAYER_ETHERNET;
469	else if (matches(arg, "atm") == 0)
470		res = LINKLAYER_ATM;
471	else if (matches(arg, "adsl") == 0)
472		res = LINKLAYER_ATM;
473	else
474		return -1; /* Indicate error */
475
476	*val = res;
477	return 0;
478}
479
480void print_linklayer(char *buf, int len, unsigned linklayer)
481{
482	switch (linklayer) {
483	case LINKLAYER_UNSPEC:
484		snprintf(buf, len, "%s", "unspec");
485		return;
486	case LINKLAYER_ETHERNET:
487		snprintf(buf, len, "%s", "ethernet");
488		return;
489	case LINKLAYER_ATM:
490		snprintf(buf, len, "%s", "atm");
491		return;
492	default:
493		snprintf(buf, len, "%s", "unknown");
494		return;
495	}
496}
497
498char *sprint_linklayer(unsigned linklayer, char *buf)
499{
500	print_linklayer(buf, SPRINT_BSIZE-1, linklayer);
501	return buf;
502}
503
504void print_tm(FILE * f, const struct tcf_t *tm)
505{
506	int hz = get_user_hz();
507	if (tm->install != 0)
508		fprintf(f, " installed %u sec", (unsigned)(tm->install/hz));
509	if (tm->lastuse != 0)
510		fprintf(f, " used %u sec", (unsigned)(tm->lastuse/hz));
511	if (tm->expires != 0)
512		fprintf(f, " expires %u sec", (unsigned)(tm->expires/hz));
513}
514
515void print_tcstats2_attr(FILE *fp, struct rtattr *rta, char *prefix, struct rtattr **xstats)
516{
517	SPRINT_BUF(b1);
518	struct rtattr *tbs[TCA_STATS_MAX + 1];
519
520	parse_rtattr_nested(tbs, TCA_STATS_MAX, rta);
521
522	if (tbs[TCA_STATS_BASIC]) {
523		struct gnet_stats_basic bs = {0};
524		memcpy(&bs, RTA_DATA(tbs[TCA_STATS_BASIC]), MIN(RTA_PAYLOAD(tbs[TCA_STATS_BASIC]), sizeof(bs)));
525		fprintf(fp, "%sSent %llu bytes %u pkt",
526			prefix, (unsigned long long) bs.bytes, bs.packets);
527	}
528
529	if (tbs[TCA_STATS_QUEUE]) {
530		struct gnet_stats_queue q = {0};
531		memcpy(&q, RTA_DATA(tbs[TCA_STATS_QUEUE]), MIN(RTA_PAYLOAD(tbs[TCA_STATS_QUEUE]), sizeof(q)));
532		fprintf(fp, " (dropped %u, overlimits %u requeues %u) ",
533			q.drops, q.overlimits, q.requeues);
534	}
535
536	if (tbs[TCA_STATS_RATE_EST64]) {
537		struct gnet_stats_rate_est64 re = {0};
538
539		memcpy(&re, RTA_DATA(tbs[TCA_STATS_RATE_EST64]),
540		       MIN(RTA_PAYLOAD(tbs[TCA_STATS_RATE_EST64]),
541			   sizeof(re)));
542		fprintf(fp, "\n%srate %s %llupps ",
543			prefix, sprint_rate(re.bps, b1), re.pps);
544	} else if (tbs[TCA_STATS_RATE_EST]) {
545		struct gnet_stats_rate_est re = {0};
546
547		memcpy(&re, RTA_DATA(tbs[TCA_STATS_RATE_EST]),
548		       MIN(RTA_PAYLOAD(tbs[TCA_STATS_RATE_EST]), sizeof(re)));
549		fprintf(fp, "\n%srate %s %upps ",
550			prefix, sprint_rate(re.bps, b1), re.pps);
551	}
552
553	if (tbs[TCA_STATS_QUEUE]) {
554		struct gnet_stats_queue q = {0};
555		memcpy(&q, RTA_DATA(tbs[TCA_STATS_QUEUE]), MIN(RTA_PAYLOAD(tbs[TCA_STATS_QUEUE]), sizeof(q)));
556		if (!tbs[TCA_STATS_RATE_EST])
557			fprintf(fp, "\n%s", prefix);
558		fprintf(fp, "backlog %s %up requeues %u ",
559			sprint_size(q.backlog, b1), q.qlen, q.requeues);
560	}
561
562	if (xstats)
563		*xstats = tbs[TCA_STATS_APP] ? : NULL;
564}
565
566void print_tcstats_attr(FILE *fp, struct rtattr *tb[], char *prefix, struct rtattr **xstats)
567{
568	SPRINT_BUF(b1);
569
570	if (tb[TCA_STATS2]) {
571		print_tcstats2_attr(fp, tb[TCA_STATS2], prefix, xstats);
572		if (xstats && NULL == *xstats)
573			goto compat_xstats;
574		return;
575	}
576	/* backward compatibility */
577	if (tb[TCA_STATS]) {
578		struct tc_stats st;
579
580		/* handle case where kernel returns more/less than we know about */
581		memset(&st, 0, sizeof(st));
582		memcpy(&st, RTA_DATA(tb[TCA_STATS]), MIN(RTA_PAYLOAD(tb[TCA_STATS]), sizeof(st)));
583
584		fprintf(fp, "%sSent %llu bytes %u pkts (dropped %u, overlimits %u) ",
585			prefix, (unsigned long long)st.bytes, st.packets, st.drops,
586			st.overlimits);
587
588		if (st.bps || st.pps || st.qlen || st.backlog) {
589			fprintf(fp, "\n%s", prefix);
590			if (st.bps || st.pps) {
591				fprintf(fp, "rate ");
592				if (st.bps)
593					fprintf(fp, "%s ", sprint_rate(st.bps, b1));
594				if (st.pps)
595					fprintf(fp, "%upps ", st.pps);
596			}
597			if (st.qlen || st.backlog) {
598				fprintf(fp, "backlog ");
599				if (st.backlog)
600					fprintf(fp, "%s ", sprint_size(st.backlog, b1));
601				if (st.qlen)
602					fprintf(fp, "%up ", st.qlen);
603			}
604		}
605	}
606
607compat_xstats:
608	if (tb[TCA_XSTATS] && xstats)
609		*xstats = tb[TCA_XSTATS];
610}
611