]> rtime.felk.cvut.cz Git - lisovros/iproute2_canprio.git/blobdiff - lib/utils.c
utils: add s32 parser
[lisovros/iproute2_canprio.git] / lib / utils.c
index 6763be2a737315e347b04f9658092cb70c743667..d80f79b4fa26c690acc227ec932235f1e9f6521e 100644 (file)
@@ -8,10 +8,6 @@
  *
  * Authors:    Alexey Kuznetsov, <kuznet@ms2.inr.ac.ru>
  *
- *
- * Changes:
- *
- * Rani Assaf <rani@magic.metawire.com> 980929:        resolve addresses
  */
 
 #include <stdio.h>
 #include <netdb.h>
 #include <arpa/inet.h>
 #include <resolv.h>
+#include <asm/types.h>
 #include <linux/pkt_sched.h>
+#include <time.h>
+#include <sys/time.h>
+#include <errno.h>
+
 
 #include "utils.h"
 
-int get_integer(int *val, char *arg, int base)
+int get_integer(int *val, const char *arg, int base)
 {
        long res;
        char *ptr;
@@ -43,7 +44,42 @@ int get_integer(int *val, char *arg, int base)
        return 0;
 }
 
-int get_unsigned(unsigned *val, char *arg, int base)
+int mask2bits(__u32 netmask)
+{
+       unsigned bits = 0;
+       __u32 mask = ntohl(netmask);
+       __u32 host = ~mask;
+
+       /* a valid netmask must be 2^n - 1 */
+       if ((host & (host + 1)) != 0)
+               return -1;
+
+       for (; mask; mask <<= 1)
+               ++bits;
+       return bits;
+}
+
+static int get_netmask(unsigned *val, const char *arg, int base)
+{
+       inet_prefix addr;
+
+       if (!get_unsigned(val, arg, base))
+               return 0;
+
+       /* try coverting dotted quad to CIDR */
+       if (!get_addr_1(&addr, arg, AF_INET) && addr.family == AF_INET) {
+               int b = mask2bits(addr.data[0]);
+               
+               if (b >= 0) {
+                       *val = b;
+                       return 0;
+               }
+       }
+
+       return -1;
+}
+
+int get_unsigned(unsigned *val, const char *arg, int base)
 {
        unsigned long res;
        char *ptr;
@@ -57,7 +93,71 @@ int get_unsigned(unsigned *val, char *arg, int base)
        return 0;
 }
 
-int get_u32(__u32 *val, char *arg, int base)
+/*
+ * get_time_rtt is "translated" from a similar routine "get_time" in
+ * tc_util.c.  We don't use the exact same routine because tc passes
+ * microseconds to the kernel and the callers of get_time_rtt want to
+ * pass milliseconds (standard unit for rtt values since 2.6.27), and
+ * have a different assumption for the units of a "raw" number.
+ */
+int get_time_rtt(unsigned *val, const char *arg, int *raw)
+{
+       double t;
+       unsigned long res;
+       char *p;
+
+       if (strchr(arg,'.') != NULL) {
+               t = strtod(arg,&p);
+               if (t < 0.0)
+                       return -1;
+       }
+       else {
+               res = strtoul(arg, &p, 0);
+               if (res > UINT_MAX)
+                       return -1;
+               t = (double)res;
+       }
+       if (p == arg)
+               return -1;
+       *raw = 1;
+
+       if (*p) {
+               *raw = 0;
+                if (strcasecmp(p, "s") == 0 || strcasecmp(p, "sec")==0 ||
+                    strcasecmp(p, "secs")==0)
+                        t *= 1000;
+                else if (strcasecmp(p, "ms") == 0 || strcasecmp(p, "msec")==0 ||
+                         strcasecmp(p, "msecs") == 0)
+                       t *= 1.0; /* allow suffix, do nothing */
+                else
+                        return -1;
+        }
+
+       /* emulate ceil() without having to bring-in -lm and always be >= 1 */
+
+       *val = t;
+       if (*val < t)
+               *val += 1;
+       
+        return 0;
+
+}
+
+int get_u64(__u64 *val, const char *arg, int base)
+{
+       unsigned long long res;
+       char *ptr;
+
+       if (!arg || !*arg)
+               return -1;
+       res = strtoull(arg, &ptr, base);
+       if (!ptr || ptr == arg || *ptr || res == 0xFFFFFFFFULL)
+               return -1;
+       *val = res;
+       return 0;
+}
+
+int get_u32(__u32 *val, const char *arg, int base)
 {
        unsigned long res;
        char *ptr;
@@ -71,7 +171,7 @@ int get_u32(__u32 *val, char *arg, int base)
        return 0;
 }
 
-int get_u16(__u16 *val, char *arg, int base)
+int get_u16(__u16 *val, const char *arg, int base)
 {
        unsigned long res;
        char *ptr;
@@ -85,7 +185,7 @@ int get_u16(__u16 *val, char *arg, int base)
        return 0;
 }
 
-int get_u8(__u8 *val, char *arg, int base)
+int get_u8(__u8 *val, const char *arg, int base)
 {
        unsigned long res;
        char *ptr;
@@ -99,7 +199,25 @@ int get_u8(__u8 *val, char *arg, int base)
        return 0;
 }
 
-int get_s16(__s16 *val, char *arg, int base)
+int get_s32(__s32 *val, const char *arg, int base)
+{
+       long res;
+       char *ptr;
+
+       errno = 0;
+
+       if (!arg || !*arg)
+               return -1;
+       res = strtol(arg, &ptr, base);
+       if (ptr == arg || *ptr ||
+           ((res ==  LONG_MIN || res == LONG_MAX) && errno == ERANGE) ||
+           res > INT32_MAX || res < INT32_MIN)
+               return -1;
+       *val = res;
+       return 0;
+}
+
+int get_s16(__s16 *val, const char *arg, int base)
 {
        long res;
        char *ptr;
@@ -113,7 +231,7 @@ int get_s16(__s16 *val, char *arg, int base)
        return 0;
 }
 
-int get_s8(__s8 *val, char *arg, int base)
+int get_s8(__s8 *val, const char *arg, int base)
 {
        long res;
        char *ptr;
@@ -127,12 +245,39 @@ int get_s8(__s8 *val, char *arg, int base)
        return 0;
 }
 
-int get_addr_1(inet_prefix *addr, char *name, int family)
+/* This uses a non-standard parsing (ie not inet_aton, or inet_pton)
+ * because of legacy choice to parse 10.8 as 10.8.0.0 not 10.0.0.8
+ */
+static int get_addr_ipv4(__u8 *ap, const char *cp)
 {
-       char *cp;
-       unsigned char *ap = (unsigned char*)addr->data;
        int i;
 
+       for (i = 0; i < 4; i++) {
+               unsigned long n;
+               char *endp;
+               
+               n = strtoul(cp, &endp, 0);
+               if (n > 255)
+                       return -1;      /* bogus network value */
+
+               if (endp == cp) /* no digits */
+                       return -1;
+
+               ap[i] = n;
+
+               if (*endp == '\0')
+                       break;
+
+               if (i == 3 || *endp != '.')
+                       return -1;      /* extra characters */
+               cp = endp + 1;
+       }
+
+       return 1;
+}
+
+int get_addr_1(inet_prefix *addr, const char *name, int family)
+{
        memset(addr, 0, sizeof(*addr));
 
        if (strcmp(name, "default") == 0 ||
@@ -171,17 +316,12 @@ int get_addr_1(inet_prefix *addr, char *name, int family)
        addr->family = AF_INET;
        if (family != AF_UNSPEC && family != AF_INET)
                return -1;
+
+       if (get_addr_ipv4((__u8 *)addr->data, name) <= 0)
+               return -1;
+
        addr->bytelen = 4;
        addr->bitlen = -1;
-       for (cp=name, i=0; *cp; cp++) {
-               if (*cp <= '9' && *cp >= '0') {
-                       ap[i] = 10*ap[i] + (*cp-'0');
-                       continue;
-               }
-               if (*cp == '.' && ++i <= 3)
-                       continue;
-               return -1;
-       }
        return 0;
 }
 
@@ -207,6 +347,7 @@ int get_prefix_1(inet_prefix *dst, char *arg, int family)
        slash = strchr(arg, '/');
        if (slash)
                *slash = 0;
+
        err = get_addr_1(dst, arg, family);
        if (err == 0) {
                switch(dst->family) {
@@ -221,10 +362,12 @@ int get_prefix_1(inet_prefix *dst, char *arg, int family)
                                dst->bitlen = 32;
                }
                if (slash) {
-                       if (get_integer(&plen, slash+1, 0) || plen > dst->bitlen) {
+                       if (get_netmask(&plen, slash+1, 0)
+                                       || plen > dst->bitlen) {
                                err = -1;
                                goto done;
                        }
+                       dst->flags |= PREFIXLEN_SPECIFIED;
                        dst->bitlen = plen;
                }
        }
@@ -234,7 +377,7 @@ done:
        return err;
 }
 
-int get_addr(inet_prefix *dst, char *arg, int family)
+int get_addr(inet_prefix *dst, const char *arg, int family)
 {
        if (family == AF_PACKET) {
                fprintf(stderr, "Error: \"%s\" may be inet address, but it is not allowed in this context.\n", arg);
@@ -260,7 +403,7 @@ int get_prefix(inet_prefix *dst, char *arg, int family)
        return 0;
 }
 
-__u32 get_addr32(char *name)
+__u32 get_addr32(const char *name)
 {
        inet_prefix addr;
        if (get_addr_1(&addr, name, AF_INET)) {
@@ -270,31 +413,37 @@ __u32 get_addr32(char *name)
        return addr.data[0];
 }
 
-void incomplete_command()
+void incomplete_command(void)
 {
        fprintf(stderr, "Command line is not complete. Try option \"help\"\n");
        exit(-1);
 }
 
-void invarg(char *msg, char *arg)
+void missarg(const char *key)
+{
+       fprintf(stderr, "Error: argument \"%s\" is required\n", key);
+       exit(-1);
+}
+
+void invarg(const char *msg, const char *arg)
 {
        fprintf(stderr, "Error: argument \"%s\" is wrong: %s\n", arg, msg);
        exit(-1);
 }
 
-void duparg(char *key, char *arg)
+void duparg(const char *key, const char *arg)
 {
        fprintf(stderr, "Error: duplicate \"%s\": \"%s\" is the second value.\n", key, arg);
        exit(-1);
 }
 
-void duparg2(char *key, char *arg)
+void duparg2(const char *key, const char *arg)
 {
        fprintf(stderr, "Error: either \"%s\" is duplicate, or \"%s\" is a garbage.\n", key, arg);
        exit(-1);
 }
 
-int matches(char *cmd, char *pattern)
+int matches(const char *cmd, const char *pattern)
 {
        int len = strlen(cmd);
        if (len > strlen(pattern))
@@ -302,10 +451,10 @@ int matches(char *cmd, char *pattern)
        return memcmp(pattern, cmd, len);
 }
 
-int inet_addr_match(inet_prefix *a, inet_prefix *b, int bits)
+int inet_addr_match(const inet_prefix *a, const inet_prefix *b, int bits)
 {
-       __u32 *a1 = a->data;
-       __u32 *a2 = b->data;
+       const __u32 *a1 = a->data;
+       const __u32 *a2 = b->data;
        int words = bits >> 0x05;
 
        bits &= 0x1f;
@@ -343,7 +492,7 @@ int __get_hz(void)
 
        if (getenv("PROC_NET_PSCHED")) {
                snprintf(name, sizeof(name)-1, "%s", getenv("PROC_NET_PSCHED"));
-       } else if (getenv("PROC_ROOT")) { 
+       } else if (getenv("PROC_ROOT")) {
                snprintf(name, sizeof(name)-1, "%s/net/psched", getenv("PROC_ROOT"));
        } else {
                strcpy(name, "/proc/net/psched");
@@ -362,7 +511,14 @@ int __get_hz(void)
        return HZ;
 }
 
-const char *rt_addr_n2a(int af, int len, void *addr, char *buf, int buflen)
+int __iproute2_user_hz_internal;
+
+int __get_user_hz(void)
+{
+       return sysconf(_SC_CLK_TCK);
+}
+
+const char *rt_addr_n2a(int af, int len, const void *addr, char *buf, int buflen)
 {
        switch (af) {
        case AF_INET:
@@ -385,13 +541,14 @@ const char *rt_addr_n2a(int af, int len, void *addr, char *buf, int buflen)
 struct namerec
 {
        struct namerec *next;
+       const char *name;
        inet_prefix addr;
-       char        *name;
 };
 
-static struct namerec *nht[256];
+#define NHASH 257
+static struct namerec *nht[NHASH];
 
-char *resolve_address(char *addr, int len, int af)
+static const char *resolve_address(const void *addr, int len, int af)
 {
        struct namerec *n;
        struct hostent *h_ent;
@@ -406,7 +563,7 @@ char *resolve_address(char *addr, int len, int af)
                len = 4;
        }
 
-       hash = addr[len-1] ^ addr[len-2] ^ addr[len-3] ^ addr[len-4];
+       hash = *(__u32 *)(addr + len - 4) % NHASH;
 
        for (n = nht[hash]; n; n = n->next) {
                if (n->addr.family == af &&
@@ -435,11 +592,13 @@ char *resolve_address(char *addr, int len, int af)
 #endif
 
 
-const char *format_host(int af, int len, void *addr, char *buf, int buflen)
+const char *format_host(int af, int len, const void *addr,
+                       char *buf, int buflen)
 {
 #ifdef RESOLVE_HOSTNAMES
        if (resolve_hosts) {
-               char *n;
+               const char *n;
+
                if (len <= 0) {
                        switch (af) {
                        case AF_INET:
@@ -470,9 +629,9 @@ const char *format_host(int af, int len, void *addr, char *buf, int buflen)
 }
 
 
-__u8* hexstring_n2a(const __u8 *str, int len, __u8 *buf, int blen)
+char *hexstring_n2a(const __u8 *str, int len, char *buf, int blen)
 {
-       __u8 *ptr = buf;
+       char *ptr = buf;
        int i;
 
        for (i=0; i<len; i++) {
@@ -489,7 +648,7 @@ __u8* hexstring_n2a(const __u8 *str, int len, __u8 *buf, int blen)
        return buf;
 }
 
-__u8* hexstring_a2n(const __u8 *str, __u8 *buf, int blen)
+__u8* hexstring_a2n(const char *str, __u8 *buf, int blen)
 {
        int cnt = 0;
 
@@ -526,3 +685,83 @@ __u8* hexstring_a2n(const __u8 *str, __u8 *buf, int blen)
                memset(buf+cnt, 0, blen-cnt);
        return buf;
 }
+
+int print_timestamp(FILE *fp)
+{
+       struct timeval tv;
+       char *tstr;
+
+       memset(&tv, 0, sizeof(tv));
+       gettimeofday(&tv, NULL);
+
+       tstr = asctime(localtime(&tv.tv_sec));
+       tstr[strlen(tstr)-1] = 0;
+       fprintf(fp, "Timestamp: %s %lu usec\n", tstr, tv.tv_usec);
+       return 0;
+}
+
+int cmdlineno;
+
+/* Like glibc getline but handle continuation lines and comments */
+ssize_t getcmdline(char **linep, size_t *lenp, FILE *in)
+{
+       ssize_t cc;
+       char *cp;
+
+       if ((cc = getline(linep, lenp, in)) < 0)
+               return cc;      /* eof or error */
+       ++cmdlineno;
+
+       cp = strchr(*linep, '#');
+       if (cp)
+               *cp = '\0';
+
+       while ((cp = strstr(*linep, "\\\n")) != NULL) {
+               char *line1 = NULL;
+               size_t len1 = 0;
+               ssize_t cc1;
+
+               if ((cc1 = getline(&line1, &len1, in)) < 0) {
+                       fprintf(stderr, "Missing continuation line\n");
+                       return cc1;
+               }
+
+               ++cmdlineno;
+               *cp = 0;
+
+               cp = strchr(line1, '#');
+               if (cp)
+                       *cp = '\0';
+
+               *lenp = strlen(*linep) + strlen(line1) + 1;
+               *linep = realloc(*linep, *lenp);
+               if (!*linep) {
+                       fprintf(stderr, "Out of memory\n");
+                       *lenp = 0;
+                       return -1;
+               }
+               cc += cc1 - 2;
+               strcat(*linep, line1);
+               free(line1);
+       }
+       return cc;
+}
+
+/* split command line into argument vector */
+int makeargs(char *line, char *argv[], int maxargs)
+{
+       static const char ws[] = " \t\r\n";
+       char *cp;
+       int argc = 0;
+
+       for (cp = strtok(line, ws); cp; cp = strtok(NULL, ws)) {
+               if (argc >= (maxargs - 1)) {
+                       fprintf(stderr, "Too many arguments to command\n");
+                       exit(1);
+               }
+               argv[argc++] = cp;
+       }
+       argv[argc] = NULL;
+
+       return argc;
+}