blob: d91aaaecf1a8350eaaf1904ba3d8529db91cebee [file] [log] [blame]
/*
* 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 resolve_hosts = 0;
void *BODY;
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] ", 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, 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;
}
#if 0
/* Builtin filter types */
static int f_parse_noopt(struct filter_util *qu, char *fhandle, int argc, char **argv, struct nlmsghdr *n)
{
if (argc || fhandle) {
fprintf(stderr, "Filter \"%s\" has no options.\n", qu->id);
return -1;
}
return 0;
}
#endif
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), "q_%s.so", 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_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));
strncpy(q->id, str, 15);
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), "f_%s.so", 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_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) __attribute__((noreturn));
static void usage(void)
{
fprintf(stderr, "Usage: tc [ OPTIONS ] OBJECT { COMMAND | help }\n"
"where OBJECT := { qdisc | class | filter }\n"
" OPTIONS := { -s[tatistics] | -d[etails] | -r[aw] | -b[atch] file }\n");
exit(-1);
}
int main(int argc, char **argv)
{
char *basename;
basename = strrchr(argv[0], '/');
if (basename == NULL)
basename = argv[0];
else
basename++;
/* batch mode */
if (argc > 1 && matches(argv[1], "-batch") == 0) {
FILE *batch;
char line[400];
char *largv[100];
int largc, ret=0;
#define BMAXARG (sizeof(largv)/sizeof(char *)-2)
if (argc != 3) {
fprintf(stderr, "Wrong number of arguments in batch mode\n");
exit(-1);
}
if (matches(argv[2], "-") != 0) {
if ((batch = fopen(argv[2], "r")) == NULL) {
fprintf(stderr, "Cannot open file \"%s\" for reading: %s=n", argv[2], strerror(errno));
exit(-1);
}
} else {
if ((batch = fdopen(0, "r")) == NULL) {
fprintf(stderr, "Cannot open stdin for reading: %s=n", strerror(errno));
exit(-1);
}
}
tc_core_init();
while (fgets(line, sizeof(line)-1, batch)) {
if (line[strlen(line)-1]=='\n') {
line[strlen(line)-1] = '\0';
} else {
fprintf(stderr, "No newline at the end of line, looks like to long (%d chars or more)\n", strlen(line));
exit(-1);
}
largc = 0;
largv[largc]=strtok(line, " ");
while ((largv[++largc]=strtok(NULL, " ")) != NULL) {
if (largc > BMAXARG) {
fprintf(stderr, "Over %d arguments in batch mode, enough!\n", BMAXARG);
exit(-1);
}
}
if (matches(largv[0], "qdisc") == 0) {
ret += do_qdisc(largc-1, largv+1);
} else if (matches(largv[0], "class") == 0) {
ret += do_class(largc-1, largv+1);
} else if (matches(largv[0], "filter") == 0) {
ret += do_filter(largc-1, largv+1);
} else if (matches(largv[0], "help") == 0) {
usage(); /* note that usage() doesn't return */
} else {
fprintf(stderr, "Object \"%s\" is unknown, try \"tc help\".\n", largv[1]);
exit(-1);
}
}
fclose(batch);
exit(0); /* end of batch, that's all */
}
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], "-Version") == 0) {
printf("tc utility, iproute2-ss%s\n", SNAPSHOT);
exit(0);
} else if (matches(argv[1], "-help") == 0) {
usage();
} else {
fprintf(stderr, "Option \"%s\" is unknown, try \"tc -help\".\n", argv[1]);
exit(-1);
}
argc--; argv++;
}
tc_core_init();
if (argc > 1) {
if (matches(argv[1], "qdisc") == 0)
return do_qdisc(argc-2, argv+2);
if (matches(argv[1], "class") == 0)
return do_class(argc-2, argv+2);
if (matches(argv[1], "filter") == 0)
return do_filter(argc-2, argv+2);
if (matches(argv[1], "help") == 0)
usage();
fprintf(stderr, "Object \"%s\" is unknown, try \"tc help\".\n", argv[1]);
exit(-1);
}
usage();
}