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