lib/utils.c

   1 /*
   2  * utils.c
   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 <math.h>
  16 #include <unistd.h>
  17 #include <fcntl.h>
  18 #include <limits.h>
  19 #include <sys/socket.h>
  20 #include <netinet/in.h>
  21 #include <string.h>
  22 #include <ctype.h>
  23 #include <netdb.h>
  24 #include <arpa/inet.h>
  25 #include <asm/types.h>
  26 #include <linux/pkt_sched.h>
  27 #include <linux/param.h>
  28 #include <linux/if_arp.h>
  29 #include <linux/mpls.h>
  30 #include <time.h>
  31 #include <sys/time.h>
  32 #include <errno.h>
  33
  34 #include "rt_names.h"
  35 #include "utils.h"
  36 #include "ll_map.h"
  37 #include "namespace.h"
  38
  39 int resolve_hosts;
  40 int timestamp_short;
  41 int pretty;
  42
  43 int read_prop(const char *dev, char *prop, long *value)
  44 {
  45         char fname[128], buf[80], *endp, *nl;
  46         FILE *fp;
  47         long result;
  48         int ret;
  49
  50         ret = snprintf(fname, sizeof(fname), "/sys/class/net/%s/%s",
  51                         dev, prop);
  52
  53         if (ret <= 0 || ret >= sizeof(fname)) {
  54                 fprintf(stderr, "could not build pathname for property\n");
  55                 return -1;
  56         }
  57
  58         fp = fopen(fname, "r");
  59         if (fp == NULL) {
  60                 fprintf(stderr, "fopen %s: %s\n", fname, strerror(errno));
  61                 return -1;
  62         }
  63
  64         if (!fgets(buf, sizeof(buf), fp)) {
  65                 fprintf(stderr, "property \"%s\" in file %s is currently unknown\n", prop, fname);
  66                 fclose(fp);
  67                 goto out;
  68         }
  69
  70         nl = strchr(buf, '\n');
  71         if (nl)
  72                 *nl = '\0';
  73
  74         fclose(fp);
  75         result = strtol(buf, &endp, 0);
  76
  77         if (*endp || buf == endp) {
  78                 fprintf(stderr, "value \"%s\" in file %s is not a number\n",
  79                         buf, fname);
  80                 goto out;
  81         }
  82
  83         if ((result == LONG_MAX || result == LONG_MIN) && errno == ERANGE) {
  84                 fprintf(stderr, "strtol %s: %s", fname, strerror(errno));
  85                 goto out;
  86         }
  87
  88         *value = result;
  89         return 0;
  90 out:
  91         fprintf(stderr, "Failed to parse %s\n", fname);
  92         return -1;
  93 }
  94
  95 /* Parse a percent e.g: '30%'
  96  * return: 0 = ok, -1 = error, 1 = out of range
  97  */
  98 int parse_percent(double *val, const char *str)
  99 {
 100         char *p;
 101
 102         *val = strtod(str, &p) / 100.;
 103         if (*val == HUGE_VALF || *val == HUGE_VALL)
 104                 return 1;
 105         if (*val == 0.0 || (*p && strcmp(p, "%")))
 106                 return -1;
 107
 108         return 0;
 109 }
 110
 111 int get_hex(char c)
 112 {
 113         if (c >= 'A' && c <= 'F')
 114                 return c - 'A' + 10;
 115         if (c >= 'a' && c <= 'f')
 116                 return c - 'a' + 10;
 117         if (c >= '0' && c <= '9')
 118                 return c - '0';
 119
 120         return -1;
 121 }
 122
 123 int get_integer(int *val, const char *arg, int base)
 124 {
 125         long res;
 126         char *ptr;
 127
 128         if (!arg || !*arg)
 129                 return -1;
 130
 131         res = strtol(arg, &ptr, base);
 132
 133         /* If there were no digits at all, strtol()  stores
 134          * the original value of nptr in *endptr (and returns 0).
 135          * In particular, if *nptr is not '\0' but **endptr is '\0' on return,
 136          * the entire string is valid.
 137          */
 138         if (!ptr || ptr == arg || *ptr)
 139                 return -1;
 140
 141         /* If an underflow occurs, strtol() returns LONG_MIN.
 142          * If an overflow occurs,  strtol() returns LONG_MAX.
 143          * In both cases, errno is set to ERANGE.
 144          */
 145         if ((res == LONG_MAX || res == LONG_MIN) && errno == ERANGE)
 146                 return -1;
 147
 148         /* Outside range of int */
 149         if (res < INT_MIN || res > INT_MAX)
 150                 return -1;
 151
 152         *val = res;
 153         return 0;
 154 }
 155
 156 int mask2bits(__u32 netmask)
 157 {
 158         unsigned int bits = 0;
 159         __u32 mask = ntohl(netmask);
 160         __u32 host = ~mask;
 161
 162         /* a valid netmask must be 2^n - 1 */
 163         if ((host & (host + 1)) != 0)
 164                 return -1;
 165
 166         for (; mask; mask <<= 1)
 167                 ++bits;
 168         return bits;
 169 }
 170
 171 static int get_netmask(unsigned int *val, const char *arg, int base)
 172 {
 173         inet_prefix addr;
 174
 175         if (!get_unsigned(val, arg, base))
 176                 return 0;
 177
 178         /* try coverting dotted quad to CIDR */
 179         if (!get_addr_1(&addr, arg, AF_INET) && addr.family == AF_INET) {
 180                 int b = mask2bits(addr.data[0]);
 181
 182                 if (b >= 0) {
 183                         *val = b;
 184                         return 0;
 185                 }
 186         }
 187
 188         return -1;
 189 }
 190
 191 int get_unsigned(unsigned int *val, const char *arg, int base)
 192 {
 193         unsigned long res;
 194         char *ptr;
 195
 196         if (!arg || !*arg)
 197                 return -1;
 198
 199         res = strtoul(arg, &ptr, base);
 200
 201         /* empty string or trailing non-digits */
 202         if (!ptr || ptr == arg || *ptr)
 203                 return -1;
 204
 205         /* overflow */
 206         if (res == ULONG_MAX && errno == ERANGE)
 207                 return -1;
 208
 209         /* out side range of unsigned */
 210         if (res > UINT_MAX)
 211                 return -1;
 212
 213         *val = res;
 214         return 0;
 215 }
 216
 217 /*
 218  * get_time_rtt is "translated" from a similar routine "get_time" in
 219  * tc_util.c.  We don't use the exact same routine because tc passes
 220  * microseconds to the kernel and the callers of get_time_rtt want to
 221  * pass milliseconds (standard unit for rtt values since 2.6.27), and
 222  * have a different assumption for the units of a "raw" number.
 223  */
 224 int get_time_rtt(unsigned int *val, const char *arg, int *raw)
 225 {
 226         double t;
 227         unsigned long res;
 228         char *p;
 229
 230         if (strchr(arg, '.') != NULL) {
 231                 t = strtod(arg, &p);
 232                 if (t < 0.0)
 233                         return -1;
 234
 235                 /* no digits? */
 236                 if (!p || p == arg)
 237                         return -1;
 238
 239                 /* over/underflow */
 240                 if ((t == HUGE_VALF || t == HUGE_VALL) && errno == ERANGE)
 241                         return -1;
 242         } else {
 243                 res = strtoul(arg, &p, 0);
 244
 245                 /* empty string? */
 246                 if (!p || p == arg)
 247                         return -1;
 248
 249                 /* overflow */
 250                 if (res == ULONG_MAX && errno == ERANGE)
 251                         return -1;
 252
 253                 t = (double)res;
 254         }
 255
 256         if (p == arg)
 257                 return -1;
 258         *raw = 1;
 259
 260         if (*p) {
 261                 *raw = 0;
 262                 if (strcasecmp(p, "s") == 0 ||
 263                     strcasecmp(p, "sec") == 0 ||
 264                     strcasecmp(p, "secs") == 0)
 265                         t *= 1000;
 266                 else if (strcasecmp(p, "ms") == 0 ||
 267                          strcasecmp(p, "msec") == 0 ||
 268                          strcasecmp(p, "msecs") == 0)
 269                         t *= 1.0; /* allow suffix, do nothing */
 270                 else
 271                         return -1;
 272         }
 273
 274         /* emulate ceil() without having to bring-in -lm and always be >= 1 */
 275         *val = t;
 276         if (*val < t)
 277                 *val += 1;
 278
 279         return 0;
 280
 281 }
 282
 283 int get_u64(__u64 *val, const char *arg, int base)
 284 {
 285         unsigned long long res;
 286         char *ptr;
 287
 288         if (!arg || !*arg)
 289                 return -1;
 290
 291         res = strtoull(arg, &ptr, base);
 292
 293         /* empty string or trailing non-digits */
 294         if (!ptr || ptr == arg || *ptr)
 295                 return -1;
 296
 297         /* overflow */
 298         if (res == ULLONG_MAX && errno == ERANGE)
 299                 return -1;
 300
 301         /* in case ULL is 128 bits */
 302         if (res > 0xFFFFFFFFFFFFFFFFULL)
 303                 return -1;
 304
 305         *val = res;
 306         return 0;
 307 }
 308
 309 int get_u32(__u32 *val, const char *arg, int base)
 310 {
 311         unsigned long res;
 312         char *ptr;
 313
 314         if (!arg || !*arg)
 315                 return -1;
 316         res = strtoul(arg, &ptr, base);
 317
 318         /* empty string or trailing non-digits */
 319         if (!ptr || ptr == arg || *ptr)
 320                 return -1;
 321
 322         /* overflow */
 323         if (res == ULONG_MAX && errno == ERANGE)
 324                 return -1;
 325
 326         /* in case UL > 32 bits */
 327         if (res > 0xFFFFFFFFUL)
 328                 return -1;
 329
 330         *val = res;
 331         return 0;
 332 }
 333
 334 int get_u16(__u16 *val, const char *arg, int base)
 335 {
 336         unsigned long res;
 337         char *ptr;
 338
 339         if (!arg || !*arg)
 340                 return -1;
 341         res = strtoul(arg, &ptr, base);
 342
 343         /* empty string or trailing non-digits */
 344         if (!ptr || ptr == arg || *ptr)
 345                 return -1;
 346
 347         /* overflow */
 348         if (res == ULONG_MAX && errno == ERANGE)
 349                 return -1;
 350
 351         if (res > 0xFFFFUL)
 352                 return -1;
 353
 354         *val = res;
 355         return 0;
 356 }
 357
 358 int get_u8(__u8 *val, const char *arg, int base)
 359 {
 360         unsigned long res;
 361         char *ptr;
 362
 363         if (!arg || !*arg)
 364                 return -1;
 365
 366         res = strtoul(arg, &ptr, base);
 367         /* empty string or trailing non-digits */
 368         if (!ptr || ptr == arg || *ptr)
 369                 return -1;
 370
 371         /* overflow */
 372         if (res == ULONG_MAX && errno == ERANGE)
 373                 return -1;
 374
 375         if (res > 0xFFUL)
 376                 return -1;
 377
 378         *val = res;
 379         return 0;
 380 }
 381
 382 int get_s32(__s32 *val, const char *arg, int base)
 383 {
 384         long res;
 385         char *ptr;
 386
 387         errno = 0;
 388
 389         if (!arg || !*arg)
 390                 return -1;
 391         res = strtol(arg, &ptr, base);
 392         if (!ptr || ptr == arg || *ptr)
 393                 return -1;
 394         if ((res == LONG_MIN || res == LONG_MAX) && errno == ERANGE)
 395                 return -1;
 396         if (res > INT32_MAX || res < INT32_MIN)
 397                 return -1;
 398
 399         *val = res;
 400         return 0;
 401 }
 402
 403 int get_s16(__s16 *val, const char *arg, int base)
 404 {
 405         long res;
 406         char *ptr;
 407
 408         if (!arg || !*arg)
 409                 return -1;
 410         res = strtol(arg, &ptr, base);
 411         if (!ptr || ptr == arg || *ptr)
 412                 return -1;
 413         if ((res == LONG_MIN || res == LONG_MAX) && errno == ERANGE)
 414                 return -1;
 415         if (res > 0x7FFF || res < -0x8000)
 416                 return -1;
 417
 418         *val = res;
 419         return 0;
 420 }
 421
 422 int get_s8(__s8 *val, const char *arg, int base)
 423 {
 424         long res;
 425         char *ptr;
 426
 427         if (!arg || !*arg)
 428                 return -1;
 429         res = strtol(arg, &ptr, base);
 430         if (!ptr || ptr == arg || *ptr)
 431                 return -1;
 432         if ((res == LONG_MIN || res == LONG_MAX) && errno == ERANGE)
 433                 return -1;
 434         if (res > 0x7F || res < -0x80)
 435                 return -1;
 436         *val = res;
 437         return 0;
 438 }
 439
 440 int get_be64(__be64 *val, const char *arg, int base)
 441 {
 442         __u64 v;
 443         int ret = get_u64(&v, arg, base);
 444
 445         if (!ret)
 446                 *val = htonll(v);
 447
 448         return ret;
 449 }
 450
 451 int get_be32(__be32 *val, const char *arg, int base)
 452 {
 453         __u32 v;
 454         int ret = get_u32(&v, arg, base);
 455
 456         if (!ret)
 457                 *val = htonl(v);
 458
 459         return ret;
 460 }
 461
 462 int get_be16(__be16 *val, const char *arg, int base)
 463 {
 464         __u16 v;
 465         int ret = get_u16(&v, arg, base);
 466
 467         if (!ret)
 468                 *val = htons(v);
 469
 470         return ret;
 471 }
 472
 473 /* This uses a non-standard parsing (ie not inet_aton, or inet_pton)
 474  * because of legacy choice to parse 10.8 as 10.8.0.0 not 10.0.0.8
 475  */
 476 static int get_addr_ipv4(__u8 *ap, const char *cp)
 477 {
 478         int i;
 479
 480         for (i = 0; i < 4; i++) {
 481                 unsigned long n;
 482                 char *endp;
 483
 484                 n = strtoul(cp, &endp, 0);
 485                 if (n > 255)
 486                         return -1;      /* bogus network value */
 487
 488                 if (endp == cp) /* no digits */
 489                         return -1;
 490
 491                 ap[i] = n;
 492
 493                 if (*endp == '\0')
 494                         break;
 495
 496                 if (i == 3 || *endp != '.')
 497                         return -1;      /* extra characters */
 498                 cp = endp + 1;
 499         }
 500
 501         return 1;
 502 }
 503
 504 int get_addr64(__u64 *ap, const char *cp)
 505 {
 506         int i;
 507
 508         union {
 509                 __u16 v16[4];
 510                 __u64 v64;
 511         } val;
 512
 513         for (i = 0; i < 4; i++) {
 514                 unsigned long n;
 515                 char *endp;
 516
 517                 n = strtoul(cp, &endp, 16);
 518                 if (n > 0xffff)
 519                         return -1;      /* bogus network value */
 520
 521                 if (endp == cp) /* no digits */
 522                         return -1;
 523
 524                 val.v16[i] = htons(n);
 525
 526                 if (*endp == '\0')
 527                         break;
 528
 529                 if (i == 3 || *endp != ':')
 530                         return -1;      /* extra characters */
 531                 cp = endp + 1;
 532         }
 533
 534         *ap = val.v64;
 535
 536         return 1;
 537 }
 538
 539 static void set_address_type(inet_prefix *addr)
 540 {
 541         switch (addr->family) {
 542         case AF_INET:
 543                 if (!addr->data[0])
 544                         addr->flags |= ADDRTYPE_INET_UNSPEC;
 545                 else if (IN_MULTICAST(ntohl(addr->data[0])))
 546                         addr->flags |= ADDRTYPE_INET_MULTI;
 547                 else
 548                         addr->flags |= ADDRTYPE_INET;
 549                 break;
 550         case AF_INET6:
 551                 if (IN6_IS_ADDR_UNSPECIFIED(addr->data))
 552                         addr->flags |= ADDRTYPE_INET_UNSPEC;
 553                 else if (IN6_IS_ADDR_MULTICAST(addr->data))
 554                         addr->flags |= ADDRTYPE_INET_MULTI;
 555                 else
 556                         addr->flags |= ADDRTYPE_INET;
 557                 break;
 558         }
 559 }
 560
 561 static int __get_addr_1(inet_prefix *addr, const char *name, int family)
 562 {
 563         memset(addr, 0, sizeof(*addr));
 564
 565         if (strcmp(name, "default") == 0 ||
 566             strcmp(name, "all") == 0 ||
 567             strcmp(name, "any") == 0) {
 568                 if ((family == AF_DECnet) || (family == AF_MPLS))
 569                         return -1;
 570                 addr->family = (family != AF_UNSPEC) ? family : AF_INET;
 571                 addr->bytelen = af_byte_len(addr->family);
 572                 addr->bitlen = -2;
 573                 return 0;
 574         }
 575
 576         if (family == AF_PACKET) {
 577                 int len;
 578
 579                 len = ll_addr_a2n((char *) &addr->data, sizeof(addr->data),
 580                                   name);
 581                 if (len < 0)
 582                         return -1;
 583
 584                 addr->family = AF_PACKET;
 585                 addr->bytelen = len;
 586                 addr->bitlen = len * 8;
 587                 return 0;
 588         }
 589
 590         if (strchr(name, ':')) {
 591                 addr->family = AF_INET6;
 592                 if (family != AF_UNSPEC && family != AF_INET6)
 593                         return -1;
 594                 if (inet_pton(AF_INET6, name, addr->data) <= 0)
 595                         return -1;
 596                 addr->bytelen = 16;
 597                 addr->bitlen = -1;
 598                 return 0;
 599         }
 600
 601         if (family == AF_DECnet) {
 602                 struct dn_naddr dna;
 603
 604                 addr->family = AF_DECnet;
 605                 if (dnet_pton(AF_DECnet, name, &dna) <= 0)
 606                         return -1;
 607                 memcpy(addr->data, dna.a_addr, 2);
 608                 addr->bytelen = 2;
 609                 addr->bitlen = -1;
 610                 return 0;
 611         }
 612
 613         if (family == AF_MPLS) {
 614                 unsigned int maxlabels;
 615                 int i;
 616
 617                 addr->family = AF_MPLS;
 618                 if (mpls_pton(AF_MPLS, name, addr->data,
 619                               sizeof(addr->data)) <= 0)
 620                         return -1;
 621                 addr->bytelen = 4;
 622                 addr->bitlen = 20;
 623                 /* How many bytes do I need? */
 624                 maxlabels = sizeof(addr->data) / sizeof(struct mpls_label);
 625                 for (i = 0; i < maxlabels; i++) {
 626                         if (ntohl(addr->data[i]) & MPLS_LS_S_MASK) {
 627                                 addr->bytelen = (i + 1)*4;
 628                                 break;
 629                         }
 630                 }
 631                 return 0;
 632         }
 633
 634         addr->family = AF_INET;
 635         if (family != AF_UNSPEC && family != AF_INET)
 636                 return -1;
 637
 638         if (get_addr_ipv4((__u8 *)addr->data, name) <= 0)
 639                 return -1;
 640
 641         addr->bytelen = 4;
 642         addr->bitlen = -1;
 643         return 0;
 644 }
 645
 646 int get_addr_1(inet_prefix *addr, const char *name, int family)
 647 {
 648         int ret;
 649
 650         ret = __get_addr_1(addr, name, family);
 651         if (ret)
 652                 return ret;
 653
 654         set_address_type(addr);
 655         return 0;
 656 }
 657
 658 int af_bit_len(int af)
 659 {
 660         switch (af) {
 661         case AF_INET6:
 662                 return 128;
 663         case AF_INET:
 664                 return 32;
 665         case AF_DECnet:
 666                 return 16;
 667         case AF_IPX:
 668                 return 80;
 669         case AF_MPLS:
 670                 return 20;
 671         }
 672
 673         return 0;
 674 }
 675
 676 int af_byte_len(int af)
 677 {
 678         return af_bit_len(af) / 8;
 679 }
 680
 681 int get_prefix_1(inet_prefix *dst, char *arg, int family)
 682 {
 683         char *slash;
 684         int err, bitlen, flags;
 685
 686         slash = strchr(arg, '/');
 687         if (slash)
 688                 *slash = 0;
 689
 690         err = get_addr_1(dst, arg, family);
 691
 692         if (slash)
 693                 *slash = '/';
 694
 695         if (err)
 696                 return err;
 697
 698         bitlen = af_bit_len(dst->family);
 699
 700         flags = PREFIXLEN_SPECIFIED;
 701         if (slash) {
 702                 unsigned int plen;
 703
 704                 if (dst->bitlen == -2)
 705                         return -1;
 706                 if (get_netmask(&plen, slash + 1, 0))
 707                         return -1;
 708                 if (plen > bitlen)
 709                         return -1;
 710
 711                 bitlen = plen;
 712         } else {
 713                 if (dst->bitlen == -2)
 714                         bitlen = 0;
 715                 else
 716                         flags = 0;
 717         }
 718
 719         dst->flags |= flags;
 720         dst->bitlen = bitlen;
 721
 722         return 0;
 723 }
 724
 725 static const char *family_name_verbose(int family)
 726 {
 727         if (family == AF_UNSPEC)
 728                 return "any valid";
 729         return family_name(family);
 730 }
 731
 732 int get_addr(inet_prefix *dst, const char *arg, int family)
 733 {
 734         if (get_addr_1(dst, arg, family)) {
 735                 fprintf(stderr,
 736                         "Error: %s address is expected rather than \"%s\".\n",
 737                         family_name_verbose(family), arg);
 738                 exit(1);
 739         }
 740         return 0;
 741 }
 742
 743 int get_addr_rta(inet_prefix *dst, const struct rtattr *rta, int family)
 744 {
 745         const int len = RTA_PAYLOAD(rta);
 746         const void *data = RTA_DATA(rta);
 747
 748         switch (len) {
 749         case 4:
 750                 dst->family = AF_INET;
 751                 dst->bytelen = 4;
 752                 memcpy(dst->data, data, 4);
 753                 break;
 754         case 16:
 755                 dst->family = AF_INET6;
 756                 dst->bytelen = 16;
 757                 memcpy(dst->data, data, 16);
 758                 break;
 759         case 2:
 760                 dst->family = AF_DECnet;
 761                 dst->bytelen = 2;
 762                 memcpy(dst->data, data, 2);
 763                 break;
 764         case 10:
 765                 dst->family = AF_IPX;
 766                 dst->bytelen = 10;
 767                 memcpy(dst->data, data, 10);
 768                 break;
 769         default:
 770                 return -1;
 771         }
 772
 773         if (family != AF_UNSPEC && family != dst->family)
 774                 return -2;
 775
 776         dst->bitlen = -1;
 777         dst->flags = 0;
 778
 779         set_address_type(dst);
 780         return 0;
 781 }
 782
 783 int get_prefix(inet_prefix *dst, char *arg, int family)
 784 {
 785         if (family == AF_PACKET) {
 786                 fprintf(stderr,
 787                         "Error: \"%s\" may be inet prefix, but it is not allowed in this context.\n",
 788                         arg);
 789                 exit(1);
 790         }
 791
 792         if (get_prefix_1(dst, arg, family)) {
 793                 fprintf(stderr,
 794                         "Error: %s prefix is expected rather than \"%s\".\n",
 795                         family_name_verbose(family), arg);
 796                 exit(1);
 797         }
 798         return 0;
 799 }
 800
 801 __u32 get_addr32(const char *name)
 802 {
 803         inet_prefix addr;
 804
 805         if (get_addr_1(&addr, name, AF_INET)) {
 806                 fprintf(stderr,
 807                         "Error: an IP address is expected rather than \"%s\"\n",
 808                         name);
 809                 exit(1);
 810         }
 811         return addr.data[0];
 812 }
 813
 814 void incomplete_command(void)
 815 {
 816         fprintf(stderr, "Command line is not complete. Try option \"help\"\n");
 817         exit(-1);
 818 }
 819
 820 void missarg(const char *key)
 821 {
 822         fprintf(stderr, "Error: argument \"%s\" is required\n", key);
 823         exit(-1);
 824 }
 825
 826 void invarg(const char *msg, const char *arg)
 827 {
 828         fprintf(stderr, "Error: argument \"%s\" is wrong: %s\n", arg, msg);
 829         exit(-1);
 830 }
 831
 832 void duparg(const char *key, const char *arg)
 833 {
 834         fprintf(stderr,
 835                 "Error: duplicate \"%s\": \"%s\" is the second value.\n",
 836                 key, arg);
 837         exit(-1);
 838 }
 839
 840 void duparg2(const char *key, const char *arg)
 841 {
 842         fprintf(stderr,
 843                 "Error: either \"%s\" is duplicate, or \"%s\" is a garbage.\n",
 844                 key, arg);
 845         exit(-1);
 846 }
 847
 848 int nodev(const char *dev)
 849 {
 850         fprintf(stderr, "Cannot find device \"%s\"\n", dev);
 851         return -1;
 852 }
 853
 854 int check_ifname(const char *name)
 855 {
 856         /* These checks mimic kernel checks in dev_valid_name */
 857         if (*name == '\0')
 858                 return -1;
 859         if (strlen(name) >= IFNAMSIZ)
 860                 return -1;
 861
 862         while (*name) {
 863                 if (*name == '/' || isspace(*name))
 864                         return -1;
 865                 ++name;
 866         }
 867         return 0;
 868 }
 869
 870 /* buf is assumed to be IFNAMSIZ */
 871 int get_ifname(char *buf, const char *name)
 872 {
 873         int ret;
 874
 875         ret = check_ifname(name);
 876         if (ret == 0)
 877                 strncpy(buf, name, IFNAMSIZ);
 878
 879         return ret;
 880 }
 881
 882 const char *get_ifname_rta(int ifindex, const struct rtattr *rta)
 883 {
 884         const char *name;
 885
 886         if (rta) {
 887                 name = rta_getattr_str(rta);
 888         } else {
 889                 fprintf(stderr,
 890                         "BUG: device with ifindex %d has nil ifname\n",
 891                         ifindex);
 892                 name = ll_idx_n2a(ifindex);
 893         }
 894
 895         if (check_ifname(name))
 896                 return NULL;
 897
 898         return name;
 899 }
 900
 901 int matches(const char *cmd, const char *pattern)
 902 {
 903         int len = strlen(cmd);
 904
 905         if (len > strlen(pattern))
 906                 return -1;
 907         return memcmp(pattern, cmd, len);
 908 }
 909
 910 int inet_addr_match(const inet_prefix *a, const inet_prefix *b, int bits)
 911 {
 912         const __u32 *a1 = a->data;
 913         const __u32 *a2 = b->data;
 914         int words = bits >> 0x05;
 915
 916         bits &= 0x1f;
 917
 918         if (words)
 919                 if (memcmp(a1, a2, words << 2))
 920                         return -1;
 921
 922         if (bits) {
 923                 __u32 w1, w2;
 924                 __u32 mask;
 925
 926                 w1 = a1[words];
 927                 w2 = a2[words];
 928
 929                 mask = htonl((0xffffffff) << (0x20 - bits));
 930
 931                 if ((w1 ^ w2) & mask)
 932                         return 1;
 933         }
 934
 935         return 0;
 936 }
 937
 938 int inet_addr_match_rta(const inet_prefix *m, const struct rtattr *rta)
 939 {
 940         inet_prefix dst;
 941
 942         if (!rta || m->family == AF_UNSPEC || m->bitlen <= 0)
 943                 return 0;
 944
 945         if (get_addr_rta(&dst, rta, m->family))
 946                 return -1;
 947
 948         return inet_addr_match(&dst, m, m->bitlen);
 949 }
 950
 951 int __iproute2_hz_internal;
 952
 953 int __get_hz(void)
 954 {
 955         char name[1024];
 956         int hz = 0;
 957         FILE *fp;
 958
 959         if (getenv("HZ"))
 960                 return atoi(getenv("HZ")) ? : HZ;
 961
 962         if (getenv("PROC_NET_PSCHED"))
 963                 snprintf(name, sizeof(name)-1,
 964                          "%s", getenv("PROC_NET_PSCHED"));
 965         else if (getenv("PROC_ROOT"))
 966                 snprintf(name, sizeof(name)-1,
 967                          "%s/net/psched", getenv("PROC_ROOT"));
 968         else
 969                 strcpy(name, "/proc/net/psched");
 970
 971         fp = fopen(name, "r");
 972
 973         if (fp) {
 974                 unsigned int nom, denom;
 975
 976                 if (fscanf(fp, "%*08x%*08x%08x%08x", &nom, &denom) == 2)
 977                         if (nom == 1000000)
 978                                 hz = denom;
 979                 fclose(fp);
 980         }
 981         if (hz)
 982                 return hz;
 983         return HZ;
 984 }
 985
 986 int __iproute2_user_hz_internal;
 987
 988 int __get_user_hz(void)
 989 {
 990         return sysconf(_SC_CLK_TCK);
 991 }
 992
 993 const char *rt_addr_n2a_r(int af, int len,
 994                           const void *addr, char *buf, int buflen)
 995 {
 996         switch (af) {
 997         case AF_INET:
 998         case AF_INET6:
 999                 return inet_ntop(af, addr, buf, buflen);
1000         case AF_MPLS:
1001                 return mpls_ntop(af, addr, buf, buflen);
1002         case AF_IPX:
1003                 return ipx_ntop(af, addr, buf, buflen);
1004         case AF_DECnet:
1005         {
1006                 struct dn_naddr dna = { 2, { 0, 0, } };
1007
1008                 memcpy(dna.a_addr, addr, 2);
1009                 return dnet_ntop(af, &dna, buf, buflen);
1010         }
1011         case AF_PACKET:
1012                 return ll_addr_n2a(addr, len, ARPHRD_VOID, buf, buflen);
1013         default:
1014                 return "???";
1015         }
1016 }
1017
1018 const char *rt_addr_n2a(int af, int len, const void *addr)
1019 {
1020         static char buf[256];
1021
1022         return rt_addr_n2a_r(af, len, addr, buf, 256);
1023 }
1024
1025 int read_family(const char *name)
1026 {
1027         int family = AF_UNSPEC;
1028
1029         if (strcmp(name, "inet") == 0)
1030                 family = AF_INET;
1031         else if (strcmp(name, "inet6") == 0)
1032                 family = AF_INET6;
1033         else if (strcmp(name, "dnet") == 0)
1034                 family = AF_DECnet;
1035         else if (strcmp(name, "link") == 0)
1036                 family = AF_PACKET;
1037         else if (strcmp(name, "ipx") == 0)
1038                 family = AF_IPX;
1039         else if (strcmp(name, "mpls") == 0)
1040                 family = AF_MPLS;
1041         else if (strcmp(name, "bridge") == 0)
1042                 family = AF_BRIDGE;
1043         return family;
1044 }
1045
1046 const char *family_name(int family)
1047 {
1048         if (family == AF_INET)
1049                 return "inet";
1050         if (family == AF_INET6)
1051                 return "inet6";
1052         if (family == AF_DECnet)
1053                 return "dnet";
1054         if (family == AF_PACKET)
1055                 return "link";
1056         if (family == AF_IPX)
1057                 return "ipx";
1058         if (family == AF_MPLS)
1059                 return "mpls";
1060         if (family == AF_BRIDGE)
1061                 return "bridge";
1062         return "???";
1063 }
1064
1065 #ifdef RESOLVE_HOSTNAMES
1066 struct namerec {
1067         struct namerec *next;
1068         const char *name;
1069         inet_prefix addr;
1070 };
1071
1072 #define NHASH 257
1073 static struct namerec *nht[NHASH];
1074
1075 static const char *resolve_address(const void *addr, int len, int af)
1076 {
1077         struct namerec *n;
1078         struct hostent *h_ent;
1079         unsigned int hash;
1080         static int notfirst;
1081
1082
1083         if (af == AF_INET6 && ((__u32 *)addr)[0] == 0 &&
1084             ((__u32 *)addr)[1] == 0 && ((__u32 *)addr)[2] == htonl(0xffff)) {
1085                 af = AF_INET;
1086                 addr += 12;
1087                 len = 4;
1088         }
1089
1090         hash = *(__u32 *)(addr + len - 4) % NHASH;
1091
1092         for (n = nht[hash]; n; n = n->next) {
1093                 if (n->addr.family == af &&
1094                     n->addr.bytelen == len &&
1095                     memcmp(n->addr.data, addr, len) == 0)
1096                         return n->name;
1097         }
1098         n = malloc(sizeof(*n));
1099         if (n == NULL)
1100                 return NULL;
1101         n->addr.family = af;
1102         n->addr.bytelen = len;
1103         n->name = NULL;
1104         memcpy(n->addr.data, addr, len);
1105         n->next = nht[hash];
1106         nht[hash] = n;
1107         if (++notfirst == 1)
1108                 sethostent(1);
1109         fflush(stdout);
1110
1111         h_ent = gethostbyaddr(addr, len, af);
1112         if (h_ent != NULL)
1113                 n->name = strdup(h_ent->h_name);
1114
1115         /* Even if we fail, "negative" entry is remembered. */
1116         return n->name;
1117 }
1118 #endif
1119
1120 const char *format_host_r(int af, int len, const void *addr,
1121                         char *buf, int buflen)
1122 {
1123 #ifdef RESOLVE_HOSTNAMES
1124         if (resolve_hosts) {
1125                 const char *n;
1126
1127                 len = len <= 0 ? af_byte_len(af) : len;
1128
1129                 if (len > 0 &&
1130                     (n = resolve_address(addr, len, af)) != NULL)
1131                         return n;
1132         }
1133 #endif
1134         return rt_addr_n2a_r(af, len, addr, buf, buflen);
1135 }
1136
1137 const char *format_host(int af, int len, const void *addr)
1138 {
1139         static char buf[256];
1140
1141         return format_host_r(af, len, addr, buf, 256);
1142 }
1143
1144
1145 char *hexstring_n2a(const __u8 *str, int len, char *buf, int blen)
1146 {
1147         char *ptr = buf;
1148         int i;
1149
1150         for (i = 0; i < len; i++) {
1151                 if (blen < 3)
1152                         break;
1153                 sprintf(ptr, "%02x", str[i]);
1154                 ptr += 2;
1155                 blen -= 2;
1156         }
1157         return buf;
1158 }
1159
1160 __u8 *hexstring_a2n(const char *str, __u8 *buf, int blen, unsigned int *len)
1161 {
1162         unsigned int cnt = 0;
1163         char *endptr;
1164
1165         if (strlen(str) % 2)
1166                 return NULL;
1167         while (cnt < blen && strlen(str) > 1) {
1168                 unsigned int tmp;
1169                 char tmpstr[3];
1170
1171                 strncpy(tmpstr, str, 2);
1172                 tmpstr[2] = '\0';
1173                 errno = 0;
1174                 tmp = strtoul(tmpstr, &endptr, 16);
1175                 if (errno != 0 || tmp > 0xFF || *endptr != '\0')
1176                         return NULL;
1177                 buf[cnt++] = tmp;
1178                 str += 2;
1179         }
1180
1181         if (len)
1182                 *len = cnt;
1183
1184         return buf;
1185 }
1186
1187 int hex2mem(const char *buf, uint8_t *mem, int count)
1188 {
1189         int i, j;
1190         int c;
1191
1192         for (i = 0, j = 0; i < count; i++, j += 2) {
1193                 c = get_hex(buf[j]);
1194                 if (c < 0)
1195                         return -1;
1196
1197                 mem[i] = c << 4;
1198
1199                 c = get_hex(buf[j + 1]);
1200                 if (c < 0)
1201                         return -1;
1202
1203                 mem[i] |= c;
1204         }
1205
1206         return 0;
1207 }
1208
1209 int addr64_n2a(__u64 addr, char *buff, size_t len)
1210 {
1211         __u16 *words = (__u16 *)&addr;
1212         __u16 v;
1213         int i, ret;
1214         size_t written = 0;
1215         char *sep = ":";
1216
1217         for (i = 0; i < 4; i++) {
1218                 v = ntohs(words[i]);
1219
1220                 if (i == 3)
1221                         sep = "";
1222
1223                 ret = snprintf(&buff[written], len - written, "%x%s", v, sep);
1224                 if (ret < 0)
1225                         return ret;
1226
1227                 written += ret;
1228         }
1229
1230         return written;
1231 }
1232
1233 /* Print buffer and escape bytes that are !isprint or among 'escape' */
1234 void print_escape_buf(const __u8 *buf, size_t len, const char *escape)
1235 {
1236         size_t i;
1237
1238         for (i = 0; i < len; ++i) {
1239                 if (isprint(buf[i]) && buf[i] != '\\' &&
1240                     !strchr(escape, buf[i]))
1241                         printf("%c", buf[i]);
1242                 else
1243                         printf("\\%03o", buf[i]);
1244         }
1245 }
1246
1247 int print_timestamp(FILE *fp)
1248 {
1249         struct timeval tv;
1250         struct tm *tm;
1251
1252         gettimeofday(&tv, NULL);
1253         tm = localtime(&tv.tv_sec);
1254
1255         if (timestamp_short) {
1256                 char tshort[40];
1257
1258                 strftime(tshort, sizeof(tshort), "%Y-%m-%dT%H:%M:%S", tm);
1259                 fprintf(fp, "[%s.%06ld] ", tshort, tv.tv_usec);
1260         } else {
1261                 char *tstr = asctime(tm);
1262
1263                 tstr[strlen(tstr)-1] = 0;
1264                 fprintf(fp, "Timestamp: %s %ld usec\n",
1265                         tstr, tv.tv_usec);
1266         }
1267
1268         return 0;
1269 }
1270
1271 unsigned int print_name_and_link(const char *fmt,
1272                                  const char *name, struct rtattr *tb[])
1273 {
1274         const char *link = NULL;
1275         unsigned int m_flag = 0;
1276         SPRINT_BUF(b1);
1277
1278         if (tb[IFLA_LINK]) {
1279                 int iflink = rta_getattr_u32(tb[IFLA_LINK]);
1280
1281                 if (iflink) {
1282                         if (tb[IFLA_LINK_NETNSID]) {
1283                                 if (is_json_context()) {
1284                                         print_int(PRINT_JSON,
1285                                                   "link_index", NULL, iflink);
1286                                 } else {
1287                                         link = ll_idx_n2a(iflink);
1288                                 }
1289                         } else {
1290                                 link = ll_index_to_name(iflink);
1291
1292                                 if (is_json_context()) {
1293                                         print_string(PRINT_JSON,
1294                                                      "link", NULL, link);
1295                                         link = NULL;
1296                                 }
1297
1298                                 m_flag = ll_index_to_flags(iflink);
1299                                 m_flag = !(m_flag & IFF_UP);
1300                         }
1301                 } else {
1302                         if (is_json_context())
1303                                 print_null(PRINT_JSON, "link", NULL, NULL);
1304                         else
1305                                 link = "NONE";
1306                 }
1307
1308                 if (link) {
1309                         snprintf(b1, sizeof(b1), "%s@%s", name, link);
1310                         name = b1;
1311                 }
1312         }
1313
1314         print_color_string(PRINT_ANY, COLOR_IFNAME, "ifname", fmt, name);
1315
1316         return m_flag;
1317 }
1318
1319 int cmdlineno;
1320
1321 /* Like glibc getline but handle continuation lines and comments */
1322 ssize_t getcmdline(char **linep, size_t *lenp, FILE *in)
1323 {
1324         ssize_t cc;
1325         char *cp;
1326
1327         cc = getline(linep, lenp, in);
1328         if (cc < 0)
1329                 return cc;      /* eof or error */
1330         ++cmdlineno;
1331
1332         cp = strchr(*linep, '#');
1333         if (cp)
1334                 *cp = '\0';
1335
1336         while ((cp = strstr(*linep, "\\\n")) != NULL) {
1337                 char *line1 = NULL;
1338                 size_t len1 = 0;
1339                 ssize_t cc1;
1340
1341                 cc1 = getline(&line1, &len1, in);
1342                 if (cc1 < 0) {
1343                         fprintf(stderr, "Missing continuation line\n");
1344                         return cc1;
1345                 }
1346
1347                 ++cmdlineno;
1348                 *cp = 0;
1349
1350                 cp = strchr(line1, '#');
1351                 if (cp)
1352                         *cp = '\0';
1353
1354                 *lenp = strlen(*linep) + strlen(line1) + 1;
1355                 *linep = realloc(*linep, *lenp);
1356                 if (!*linep) {
1357                         fprintf(stderr, "Out of memory\n");
1358                         *lenp = 0;
1359                         return -1;
1360                 }
1361                 cc += cc1 - 2;
1362                 strcat(*linep, line1);
1363                 free(line1);
1364         }
1365         return cc;
1366 }
1367
1368 /* split command line into argument vector */
1369 int makeargs(char *line, char *argv[], int maxargs)
1370 {
1371         static const char ws[] = " \t\r\n";
1372         char *cp = line;
1373         int argc = 0;
1374
1375         while (*cp) {
1376                 /* skip leading whitespace */
1377                 cp += strspn(cp, ws);
1378
1379                 if (*cp == '\0')
1380                         break;
1381
1382                 if (argc >= (maxargs - 1)) {
1383                         fprintf(stderr, "Too many arguments to command\n");
1384                         exit(1);
1385                 }
1386
1387                 /* word begins with quote */
1388                 if (*cp == '\'' || *cp == '"') {
1389                         char quote = *cp++;
1390
1391                         argv[argc++] = cp;
1392                         /* find ending quote */
1393                         cp = strchr(cp, quote);
1394                         if (cp == NULL) {
1395                                 fprintf(stderr, "Unterminated quoted string\n");
1396                                 exit(1);
1397                         }
1398                 } else {
1399                         argv[argc++] = cp;
1400
1401                         /* find end of word */
1402                         cp += strcspn(cp, ws);
1403                         if (*cp == '\0')
1404                                 break;
1405                 }
1406
1407                 /* seperate words */
1408                 *cp++ = 0;
1409         }
1410         argv[argc] = NULL;
1411
1412         return argc;
1413 }
1414
1415 void print_nlmsg_timestamp(FILE *fp, const struct nlmsghdr *n)
1416 {
1417         char *tstr;
1418         time_t secs = ((__u32 *)NLMSG_DATA(n))[0];
1419         long usecs = ((__u32 *)NLMSG_DATA(n))[1];
1420
1421         tstr = asctime(localtime(&secs));
1422         tstr[strlen(tstr)-1] = 0;
1423         fprintf(fp, "Timestamp: %s %lu us\n", tstr, usecs);
1424 }
1425
1426 static int on_netns(char *nsname, void *arg)
1427 {
1428         struct netns_func *f = arg;
1429
1430         if (netns_switch(nsname))
1431                 return -1;
1432
1433         return f->func(nsname, f->arg);
1434 }
1435
1436 static int on_netns_label(char *nsname, void *arg)
1437 {
1438         printf("\nnetns: %s\n", nsname);
1439         return on_netns(nsname, arg);
1440 }
1441
1442 int do_each_netns(int (*func)(char *nsname, void *arg), void *arg,
1443                 bool show_label)
1444 {
1445         struct netns_func nsf = { .func = func, .arg = arg };
1446
1447         if (show_label)
1448                 return netns_foreach(on_netns_label, &nsf);
1449
1450         return netns_foreach(on_netns, &nsf);
1451 }
1452
1453 char *int_to_str(int val, char *buf)
1454 {
1455         sprintf(buf, "%d", val);
1456         return buf;
1457 }
1458
1459 int get_guid(__u64 *guid, const char *arg)
1460 {
1461         unsigned long int tmp;
1462         char *endptr;
1463         int i;
1464
1465 #define GUID_STR_LEN 23
1466         /* Verify strict format: format string must be
1467          * xx:xx:xx:xx:xx:xx:xx:xx where xx can be an arbitrary
1468          * hex digit
1469          */
1470
1471         if (strlen(arg) != GUID_STR_LEN)
1472                 return -1;
1473
1474         /* make sure columns are in place */
1475         for (i = 0; i < 7; i++)
1476                 if (arg[2 + i * 3] != ':')
1477                         return -1;
1478
1479         *guid = 0;
1480         for (i = 0; i < 8; i++) {
1481                 tmp = strtoul(arg + i * 3, &endptr, 16);
1482                 if (endptr != arg + i * 3 + 2)
1483                         return -1;
1484
1485                 if (tmp > 255)
1486                         return -1;
1487
1488                  *guid |= tmp << (56 - 8 * i);
1489         }
1490
1491         return 0;
1492 }
1493
1494 /* This is a necessary workaround for multicast route dumps */
1495 int get_real_family(int rtm_type, int rtm_family)
1496 {
1497         if (rtm_type != RTN_MULTICAST)
1498                 return rtm_family;
1499
1500         if (rtm_family == RTNL_FAMILY_IPMR)
1501                 return AF_INET;
1502
1503         if (rtm_family == RTNL_FAMILY_IP6MR)
1504                 return AF_INET6;
1505
1506         return rtm_family;
1507 }
1508
1509 /* Based on copy_rtnl_link_stats() from kernel at net/core/rtnetlink.c */
1510 static void copy_rtnl_link_stats64(struct rtnl_link_stats64 *stats64,
1511                                    const struct rtnl_link_stats *stats)
1512 {
1513         __u64 *a = (__u64 *)stats64;
1514         const __u32 *b = (const __u32 *)stats;
1515         const __u32 *e = b + sizeof(*stats) / sizeof(*b);
1516
1517         while (b < e)
1518                 *a++ = *b++;
1519 }
1520
1521 int get_rtnl_link_stats_rta(struct rtnl_link_stats64 *stats64,
1522                             struct rtattr *tb[])
1523 {
1524         struct rtnl_link_stats stats;
1525         void *s;
1526         struct rtattr *rta;
1527         int size, len;
1528
1529         if (tb[IFLA_STATS64]) {
1530                 rta = tb[IFLA_STATS64];
1531                 size = sizeof(struct rtnl_link_stats64);
1532                 s = stats64;
1533         } else if (tb[IFLA_STATS]) {
1534                 rta = tb[IFLA_STATS];
1535                 size = sizeof(struct rtnl_link_stats);
1536                 s = &stats;
1537         } else {
1538                 return -1;
1539         }
1540
1541         len = RTA_PAYLOAD(rta);
1542         if (len < size)
1543                 memset(s + len, 0, size - len);
1544         else
1545                 len = size;
1546
1547         memcpy(s, RTA_DATA(rta), len);
1548
1549         if (s != stats64)
1550                 copy_rtnl_link_stats64(stats64, s);
1551         return size;
1552 }
1553
1554 #ifdef NEED_STRLCPY
1555 size_t strlcpy(char *dst, const char *src, size_t size)
1556 {
1557         size_t srclen = strlen(src);
1558
1559         if (size) {
1560                 size_t minlen = min(srclen, size - 1);
1561
1562                 memcpy(dst, src, minlen);
1563                 dst[minlen] = '\0';
1564         }
1565         return srclen;
1566 }
1567
1568 size_t strlcat(char *dst, const char *src, size_t size)
1569 {
1570         size_t dlen = strlen(dst);
1571
1572         if (dlen >= size)
1573                 return dlen + strlen(src);
1574
1575         return dlen + strlcpy(dst + dlen, src, size - dlen);
1576 }
1577 #endif