nslookup: return exitcode 1 on resolution errors
[oweals/busybox.git] / networking / tc.c
index 6a5a8504f12fad2399d484965aec826ec8f6b4a4..3e9808328d8b4ab4071020b8ccc9f74df9110f85 100644 (file)
@@ -1,37 +1,86 @@
 /* vi: set sw=4 ts=4: */
 /*
- * tc.c                "tc" utility frontend.
+ * Licensed under GPLv2 or later, see file LICENSE in this source tree.
  *
- * Licensed under the GPL v2 or later, see the file LICENSE in this tarball.
- *
- * Authors:    Alexey Kuznetsov, <kuznet@ms2.inr.ac.ru>
+ * Authors: Alexey Kuznetsov, <kuznet@ms2.inr.ac.ru>
  *
  * Bernhard Reutner-Fischer adjusted for busybox
  */
+//config:config TC
+//config:      bool "tc (8.3 kb)"
+//config:      default y
+//config:      help
+//config:      Show / manipulate traffic control settings
+//config:
+//config:config FEATURE_TC_INGRESS
+//config:      bool "Enable ingress"
+//config:      default y
+//config:      depends on TC
+
+//applet:IF_TC(APPLET(tc, BB_DIR_SBIN, BB_SUID_DROP))
+
+//kbuild:lib-$(CONFIG_TC) += tc.o
+
+//usage:#define tc_trivial_usage
+/* //usage: "[OPTIONS] OBJECT CMD [dev STRING]" */
+//usage:       "OBJECT CMD [dev STRING]"
+//usage:#define tc_full_usage "\n\n"
+//usage:       "OBJECT: qdisc|class|filter\n"
+//usage:       "CMD: add|del|change|replace|show\n"
+//usage:       "\n"
+//usage:       "qdisc [handle QHANDLE] [root|"IF_FEATURE_TC_INGRESS("ingress|")"parent CLASSID]\n"
+/* //usage: "[estimator INTERVAL TIME_CONSTANT]\n" */
+//usage:       "       [[QDISC_KIND] [help|OPTIONS]]\n"
+//usage:       "       QDISC_KIND := [p|b]fifo|tbf|prio|cbq|red|etc.\n"
+//usage:       "qdisc show [dev STRING]"IF_FEATURE_TC_INGRESS(" [ingress]")"\n"
+//usage:       "class [classid CLASSID] [root|parent CLASSID]\n"
+//usage:       "       [[QDISC_KIND] [help|OPTIONS] ]\n"
+//usage:       "class show [ dev STRING ] [root|parent CLASSID]\n"
+//usage:       "filter [pref PRIO] [protocol PROTO]\n"
+/* //usage: "\t[estimator INTERVAL TIME_CONSTANT]\n" */
+//usage:       "       [root|classid CLASSID] [handle FILTERID]\n"
+//usage:       "       [[FILTER_TYPE] [help|OPTIONS]]\n"
+//usage:       "filter show [dev STRING] [root|parent CLASSID]"
 
 #include "libbb.h"
+#include "common_bufsiz.h"
 
 #include "libiproute/utils.h"
 #include "libiproute/ip_common.h"
 #include "libiproute/rt_names.h"
 #include <linux/pkt_sched.h> /* for the TC_H_* macros */
 
+/* This is the deprecated multiqueue interface */
+#ifndef TCA_PRIO_MAX
+enum
+{
+       TCA_PRIO_UNSPEC,
+       TCA_PRIO_MQ,
+       __TCA_PRIO_MAX
+};
+#define TCA_PRIO_MAX    (__TCA_PRIO_MAX - 1)
+#endif
+
 #define parse_rtattr_nested(tb, max, rta) \
        (parse_rtattr((tb), (max), RTA_DATA(rta), RTA_PAYLOAD(rta)))
 
 /* nullifies tb on error */
 #define __parse_rtattr_nested_compat(tb, max, rta, len) \
-       ({if ((RTA_PAYLOAD(rta) >= len) && \
-                (RTA_PAYLOAD(rta) >= RTA_ALIGN(len) + sizeof(struct rtattr))) { \
-                       rta = RTA_DATA(rta) + RTA_ALIGN(len); \
-                       parse_rtattr_nested(tb, max, rta); \
-         } else \
-                       memset(tb, 0, sizeof(struct rtattr *) * (max + 1)); \
-       })
+({ \
+       if ((RTA_PAYLOAD(rta) >= len) \
+        && (RTA_PAYLOAD(rta) >= RTA_ALIGN(len) + sizeof(struct rtattr)) \
+       ) { \
+               rta = RTA_DATA(rta) + RTA_ALIGN(len); \
+               parse_rtattr_nested(tb, max, rta); \
+       } else \
+               memset(tb, 0, sizeof(struct rtattr *) * (max + 1)); \
+})
 
 #define parse_rtattr_nested_compat(tb, max, rta, data, len) \
-       ({data = RTA_PAYLOAD(rta) >= len ? RTA_DATA(rta) : NULL; \
-       __parse_rtattr_nested_compat(tb, max, rta, len); })
+({ \
+       data = RTA_PAYLOAD(rta) >= len ? RTA_DATA(rta) : NULL; \
+       __parse_rtattr_nested_compat(tb, max, rta, len); \
+})
 
 #define show_details (0) /* not implemented. Does anyone need it? */
 #define use_iec (0) /* not currently documented in the upstream manpage */
 
 struct globals {
        int filter_ifindex;
-       __u32 filter_qdisc;
-       __u32 filter_parent;
-       __u32 filter_prio;
-       __u32 filter_proto;
+       uint32_t filter_qdisc;
+       uint32_t filter_parent;
+       uint32_t filter_prio;
+       uint32_t filter_proto;
 } FIX_ALIASING;
-#define G (*(struct globals*)&bb_common_bufsiz1)
+#define G (*(struct globals*)bb_common_bufsiz1)
 #define filter_ifindex (G.filter_ifindex)
 #define filter_qdisc (G.filter_qdisc)
 #define filter_parent (G.filter_parent)
 #define filter_prio (G.filter_prio)
 #define filter_proto (G.filter_proto)
-
-void BUG_tc_globals_too_big(void);
 #define INIT_G() do { \
-       if (sizeof(G) > COMMON_BUFSIZE) \
-               BUG_tc_globals_too_big(); \
+       setup_common_bufsiz(); \
+       BUILD_BUG_ON(sizeof(G) > COMMON_BUFSIZE); \
 } while (0)
 
 /* Allocates a buffer containing the name of a class id.
@@ -77,12 +124,12 @@ static char* print_tc_classid(uint32_t cid)
 }
 
 /* Get a qdisc handle.  Return 0 on success, !0 otherwise.  */
-static int get_qdisc_handle(__u32 *h, const char *str) {
-       __u32 maj;
+static int get_qdisc_handle(uint32_t *h, const char *str) {
+       uint32_t maj;
        char *p;
 
        maj = TC_H_UNSPEC;
-       if (!strcmp(str, "none"))
+       if (strcmp(str, "none") == 0)
                goto ok;
        maj = strtoul(str, &p, 16);
        if (p == str)
@@ -96,15 +143,15 @@ static int get_qdisc_handle(__u32 *h, const char *str) {
 }
 
 /* Get class ID.  Return 0 on success, !0 otherwise.  */
-static int get_tc_classid(__u32 *h, const char *str) {
-       __u32 maj, min;
+static int get_tc_classid(uint32_t *h, const char *str) {
+       uint32_t maj, min;
        char *p;
 
        maj = TC_H_ROOT;
-       if (!strcmp(str, "root"))
+       if (strcmp(str, "root") == 0)
                goto ok;
        maj = TC_H_UNSPEC;
-       if (!strcmp(str, "none"))
+       if (strcmp(str, "none") == 0)
                goto ok;
        maj = strtoul(str, &p, 16);
        if (p == str) {
@@ -134,27 +181,29 @@ static void print_rate(char *buf, int len, uint32_t rate)
        double tmp = (double)rate*8;
 
        if (use_iec) {
-               if (tmp >= 1000.0*1024.0*1024.0)
-                       snprintf(buf, len, "%.0fMibit", tmp/1024.0*1024.0);
-               else if (tmp >= 1000.0*1024)
+               if (tmp >= 1000*1024*1024)
+                       snprintf(buf, len, "%.0fMibit", tmp/(1024*1024));
+               else if (tmp >= 1000*1024)
                        snprintf(buf, len, "%.0fKibit", tmp/1024);
                else
                        snprintf(buf, len, "%.0fbit", tmp);
        } else {
-               if (tmp >= 1000.0*1000000.0)
-                       snprintf(buf, len, "%.0fMbit", tmp/1000000.0);
-               else if (tmp >= 1000.0 * 1000.0)
-                       snprintf(buf, len, "%.0fKbit", tmp/1000.0);
+               if (tmp >= 1000*1000000)
+                       snprintf(buf, len, "%.0fMbit", tmp/1000000);
+               else if (tmp >= 1000*1000)
+                       snprintf(buf, len, "%.0fKbit", tmp/1000);
                else
                        snprintf(buf, len, "%.0fbit",  tmp);
        }
 }
 
+#if 0
 /* This is "pfifo_fast".  */
 static int prio_parse_opt(int argc, char **argv, struct nlmsghdr *n)
 {
        return 0;
 }
+#endif
 static int prio_print_opt(struct rtattr *opt)
 {
        int i;
@@ -177,11 +226,13 @@ static int prio_print_opt(struct rtattr *opt)
        return 0;
 }
 
+#if 0
 /* Class Based Queue */
 static int cbq_parse_opt(int argc, char **argv, struct nlmsghdr *n)
 {
        return 0;
 }
+#endif
 static int cbq_print_opt(struct rtattr *opt)
 {
        struct rtattr *tb[TCA_CBQ_MAX+1];
@@ -274,8 +325,10 @@ static int cbq_print_opt(struct rtattr *opt)
        return 0;
 }
 
-static int print_qdisc(const struct sockaddr_nl *who UNUSED_PARAM,
-                                               struct nlmsghdr *hdr, void *arg UNUSED_PARAM)
+static FAST_FUNC int print_qdisc(
+               const struct sockaddr_nl *who UNUSED_PARAM,
+               struct nlmsghdr *hdr,
+               void *arg UNUSED_PARAM)
 {
        struct tcmsg *msg = NLMSG_DATA(hdr);
        int len = hdr->nlmsg_len;
@@ -330,8 +383,10 @@ static int print_qdisc(const struct sockaddr_nl *who UNUSED_PARAM,
        return 0;
 }
 
-static int print_class(const struct sockaddr_nl *who UNUSED_PARAM,
-                                               struct nlmsghdr *hdr, void *arg UNUSED_PARAM)
+static FAST_FUNC int print_class(
+               const struct sockaddr_nl *who UNUSED_PARAM,
+               struct nlmsghdr *hdr,
+               void *arg UNUSED_PARAM)
 {
        struct tcmsg *msg = NLMSG_DATA(hdr);
        int len = hdr->nlmsg_len;
@@ -374,7 +429,7 @@ static int print_class(const struct sockaddr_nl *who UNUSED_PARAM,
                printf("root ");
        else if (msg->tcm_parent) {
                classid = print_tc_classid(filter_qdisc ?
-                                                                  TC_H_MIN(msg->tcm_parent) : msg->tcm_parent);
+                               TC_H_MIN(msg->tcm_parent) : msg->tcm_parent);
                printf("parent %s ", classid);
                if (ENABLE_FEATURE_CLEAN_UP)
                        free(classid);
@@ -398,12 +453,11 @@ static int print_class(const struct sockaddr_nl *who UNUSED_PARAM,
        return 0;
 }
 
-static int print_filter(const struct sockaddr_nl *who UNUSED_PARAM,
-                                               struct nlmsghdr *hdr, void *arg UNUSED_PARAM)
+static FAST_FUNC int print_filter(
+               const struct sockaddr_nl *who UNUSED_PARAM,
+               struct nlmsghdr *hdr UNUSED_PARAM,
+               void *arg UNUSED_PARAM)
 {
-       struct tcmsg *msg = NLMSG_DATA(hdr);
-       int len = hdr->nlmsg_len;
-       struct rtattr * tb[TCA_MAX+1];
        return 0;
 }
 
@@ -420,6 +474,12 @@ int tc_main(int argc UNUSED_PARAM, char **argv)
                "replace\0"
                "show\0""list\0"
                ;
+       enum {
+               CMD_add = 0, CMD_del, CMD_change,
+               CMD_link,
+               CMD_replace,
+               CMD_show
+       };
        static const char args[] ALIGN1 =
                "dev\0" /* qdisc, class, filter */
                "root\0" /* class, filter */
@@ -429,9 +489,15 @@ int tc_main(int argc UNUSED_PARAM, char **argv)
                "classid\0" /* change: for class use "handle" */
                "preference\0""priority\0""protocol\0" /* filter */
                ;
-       enum { CMD_add = 0, CMD_del, CMD_change, CMD_link, CMD_replace, CMD_show };
-       enum { ARG_dev = 0, ARG_root, ARG_parent, ARG_qdisc,
-                       ARG_handle, ARG_classid, ARG_pref, ARG_prio, ARG_proto};
+       enum {
+               ARG_dev = 0,
+               ARG_root,
+               ARG_parent,
+               ARG_qdisc,
+               ARG_handle,
+               ARG_classid,
+               ARG_pref, ARG_prio, ARG_proto
+       };
        struct rtnl_handle rth;
        struct tcmsg msg;
        int ret, obj, cmd, arg;
@@ -446,19 +512,22 @@ int tc_main(int argc UNUSED_PARAM, char **argv)
 
        obj = index_in_substrings(objects, *argv++);
 
-       if (obj < OBJ_qdisc)
+       if (obj < 0)
                bb_show_usage();
        if (!*argv)
                cmd = CMD_show; /* list is the default */
        else {
                cmd = index_in_substrings(commands, *argv);
                if (cmd < 0)
-                       bb_error_msg_and_die(bb_msg_invalid_arg, *argv, applet_name);
+                       invarg_1_to_2(*argv, argv[-1]);
                argv++;
        }
+
        memset(&msg, 0, sizeof(msg));
-       msg.tcm_family = AF_UNSPEC;
+       if (AF_UNSPEC != 0)
+               msg.tcm_family = AF_UNSPEC;
        ll_init_map(&rth);
+
        while (*argv) {
                arg = index_in_substrings(args, *argv);
                if (arg == ARG_dev) {
@@ -476,7 +545,7 @@ int tc_main(int argc UNUSED_PARAM, char **argv)
                        NEXT_ARG();
                        /* We don't care about duparg2("qdisc handle",*argv) for now */
                        if (get_qdisc_handle(&filter_qdisc, *argv))
-                               invarg(*argv, "qdisc");
+                               invarg_1_to_2(*argv, "qdisc");
                } else
                if (obj != OBJ_qdisc
                 && (arg == ARG_root
@@ -486,7 +555,7 @@ int tc_main(int argc UNUSED_PARAM, char **argv)
                ) {
                        /* nothing */
                } else {
-                       invarg(*argv, "command");
+                       invarg_1_to_2(*argv, "command");
                }
                NEXT_ARG();
                if (arg == ARG_root) {
@@ -495,39 +564,50 @@ int tc_main(int argc UNUSED_PARAM, char **argv)
                        msg.tcm_parent = TC_H_ROOT;
                        if (obj == OBJ_filter)
                                filter_parent = TC_H_ROOT;
-               } else if (arg == ARG_parent) {
-                       __u32 handle;
+               } else
+               if (arg == ARG_parent) {
+                       uint32_t handle;
                        if (msg.tcm_parent)
                                duparg(*argv, "parent");
                        if (get_tc_classid(&handle, *argv))
-                               invarg(*argv, "parent");
+                               invarg_1_to_2(*argv, "parent");
                        msg.tcm_parent = handle;
                        if (obj == OBJ_filter)
                                filter_parent = handle;
-               } else if (arg == ARG_handle) { /* filter::list */
+               } else
+               if (arg == ARG_handle) { /* filter::list */
                        if (msg.tcm_handle)
                                duparg(*argv, "handle");
                        /* reject LONG_MIN || LONG_MAX */
                        /* TODO: for fw
-                          if ((slash = strchr(handle, '/')) != NULL)
-                                  *slash = '\0';
+                       slash = strchr(handle, '/');
+                       if (slash != NULL)
+                               *slash = '\0';
                         */
                        msg.tcm_handle = get_u32(*argv, "handle");
-                       /* if (slash) {if (get_u32(__u32 &mask, slash+1, NULL)) inv mask; addattr32(n, MAX_MSG, TCA_FW_MASK, mask); */
-               } else if (arg == ARG_classid && obj == OBJ_class && cmd == CMD_change){
-               } else if (arg == ARG_pref || arg == ARG_prio) { /* filter::list */
+                       /* if (slash) {if (get_u32(uint32_t &mask, slash+1, NULL)) inv mask; addattr32(n, MAX_MSG, TCA_FW_MASK, mask); */
+               } else
+               if (arg == ARG_classid
+                && obj == OBJ_class
+                && cmd == CMD_change
+               ) {
+                       /* TODO */
+               } else
+               if (arg == ARG_pref || arg == ARG_prio) { /* filter::list */
                        if (filter_prio)
                                duparg(*argv, "priority");
                        filter_prio = get_u32(*argv, "priority");
-               } else if (arg == ARG_proto) { /* filter::list */
+               } else
+               if (arg == ARG_proto) { /* filter::list */
                        uint16_t tmp;
                        if (filter_proto)
                                duparg(*argv, "protocol");
                        if (ll_proto_a2n(&tmp, *argv))
-                               invarg(*argv, "protocol");
+                               invarg_1_to_2(*argv, "protocol");
                        filter_proto = tmp;
                }
        }
+
        if (cmd >= CMD_show) { /* show or list */
                if (obj == OBJ_filter)
                        msg.tcm_info = TC_H_MAKE(filter_prio<<16, filter_proto);