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