ip6addrctl.tproj/ip6addrctl.c

   1 /*      $KAME: ip6addrctl.c,v 1.3 2003/12/16 08:14:28 suz Exp $ */
   2
   3 /*
   4  * Copyright (C) 2001 WIDE Project.
   5  * All rights reserved.
   6  *
   7  * Redistribution and use in source and binary forms, with or without
   8  * modification, are permitted provided that the following conditions
   9  * are met:
  10  * 1. Redistributions of source code must retain the above copyright
  11  *    notice, this list of conditions and the following disclaimer.
  12  * 2. Redistributions in binary form must reproduce the above copyright
  13  *    notice, this list of conditions and the following disclaimer in the
  14  *    documentation and/or other materials provided with the distribution.
  15  * 3. Neither the name of the project nor the names of its contributors
  16  *    may be used to endorse or promote products derived from this software
  17  *    without specific prior written permission.
  18  *
  19  * THIS SOFTWARE IS PROVIDED BY THE PROJECT AND CONTRIBUTORS ``AS IS'' AND
  20  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  21  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  22  * ARE DISCLAIMED.  IN NO EVENT SHALL THE PROJECT OR CONTRIBUTORS BE LIABLE
  23  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  24  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  25  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  26  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  27  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  28  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  29  * SUCH DAMAGE.
  30  *
  31  * $FreeBSD$
  32  */
  33
  34 #include <sys/types.h>
  35 #include <sys/socket.h>
  36 #include <sys/queue.h>
  37 #include <sys/param.h>
  38 #include <sys/ioctl.h>
  39 #include <sys/sysctl.h>
  40
  41 #include <net/if.h>
  42 #include <net/if_var.h>
  43
  44 #include <netinet/in.h>
  45 #include <netinet6/in6_var.h>
  46
  47 #include <stdlib.h>
  48 #include <netdb.h>
  49 #include <stdio.h>
  50 #include <unistd.h>
  51 #include <limits.h>
  52 #include <string.h>
  53 #include <err.h>
  54
  55 static char *configfile;
  56
  57 struct policyqueue {
  58         TAILQ_ENTRY(policyqueue) pc_entry;
  59         struct in6_addrpolicy pc_policy;
  60 };
  61 TAILQ_HEAD(policyhead, policyqueue);
  62 struct policyhead policyhead;
  63
  64 static void usage __P((void));
  65 static void get_policy __P((void));
  66 static void dump_policy __P((void));
  67 static int mask2plen __P((struct sockaddr_in6 *));
  68 static int parse_prefix __P((const char *, struct in6_addrpolicy *));
  69 static void make_policy_fromfile __P((char *));
  70 static void plen2mask __P((struct sockaddr_in6 *, int));
  71 static void set_policy __P((void));
  72 static void add_policy __P((char *, char *, char *));
  73 static void delete_policy __P((char *));
  74 static void flush_policy __P(());
  75
  76 int
  77 main(argc, argv)
  78         int argc;
  79         char *argv[];
  80 {
  81         TAILQ_INIT(&policyhead);
  82
  83         if (argc == 1 || strcasecmp(argv[1], "show") == 0) {
  84                 get_policy();
  85                 dump_policy();
  86         } else if (strcasecmp(argv[1], "add") == 0) {
  87                 if (argc < 5)
  88                         usage();
  89                 add_policy(argv[2], argv[3], argv[4]);
  90         } else if (strcasecmp(argv[1], "delete") == 0) {
  91                 if (argc < 3)
  92                         usage();
  93                 delete_policy(argv[2]);
  94         } else if (strcasecmp(argv[1], "flush") == 0) {
  95                 get_policy();
  96                 flush_policy();
  97         } else if (strcasecmp(argv[1], "install") == 0) {
  98                 if (argc < 3)
  99                         usage();
 100                 configfile = argv[2];
 101                 make_policy_fromfile(configfile);
 102                 set_policy();
 103         } else
 104                 usage();
 105
 106         exit(0);
 107 }
 108
 109 static void
 110 get_policy()
 111 {
 112         int mib[] = { CTL_NET, PF_INET6, IPPROTO_IPV6, IPV6CTL_ADDRCTLPOLICY };
 113         size_t l;
 114         char *buf;
 115         struct in6_addrpolicy *pol, *ep;
 116
 117         if (sysctl(mib, sizeof(mib) / sizeof(mib[0]), NULL, &l, NULL, 0) < 0) {
 118                 err(1, "sysctl(IPV6CTL_ADDRCTLPOLICY)");
 119                 /* NOTREACHED */
 120         }
 121         if (l == 0) {
 122                 printf("no source-address-selection policy is installed\n");
 123                 return;
 124         }
 125         if ((buf = malloc(l)) == NULL) {
 126                 errx(1, "malloc failed");
 127                 /* NOTREACHED */
 128         }
 129         if (sysctl(mib, sizeof(mib) / sizeof(mib[0]), buf, &l, NULL, 0) < 0) {
 130                 err(1, "sysctl(IPV6CTL_ADDRCTLPOLICY)");
 131                 /* NOTREACHED */
 132         }
 133
 134         ep = (struct in6_addrpolicy *)(buf + l);
 135         for (pol = (struct in6_addrpolicy *)buf; pol + 1 <= ep; pol++) {
 136                 struct policyqueue *new;
 137
 138                 if ((new = malloc(sizeof(*new))) == NULL)
 139                         errx(1, "malloc failed\n");
 140                 new->pc_policy = *pol;
 141                 TAILQ_INSERT_TAIL(&policyhead, new, pc_entry);
 142         }
 143
 144         free(buf);
 145 }
 146
 147 static void
 148 dump_policy()
 149 {
 150         size_t addrlen;
 151         char addrbuf[NI_MAXHOST];
 152         struct in6_addrpolicy *pol;
 153         struct policyqueue *ent;
 154         int plen, first = 1;
 155
 156         for (ent = TAILQ_FIRST(&policyhead); ent;
 157              ent = TAILQ_NEXT(ent, pc_entry)) {
 158                 pol = &ent->pc_policy;
 159                 if (first) {
 160                         printf("%-30s %5s %5s %8s\n",
 161                                "Prefix", "Prec", "Label", "Use");
 162                         first = 0;
 163                 }
 164
 165                 if ((getnameinfo((struct sockaddr *)&pol->addr,
 166                                  sizeof(pol->addr), addrbuf, sizeof(addrbuf),
 167                                  NULL, 0, NI_NUMERICHOST))) {
 168                         warnx("getnameinfo for prefix address failed");
 169                         continue;
 170                 }
 171                 if ((plen = mask2plen(&pol->addrmask)) < 0) {
 172                         warnx("invalid address mask");
 173                         continue;
 174                 }
 175                 addrlen = strlen(addrbuf);
 176                 if (addrlen + sizeof("/128") < sizeof(addrbuf)) {
 177                         snprintf(&addrbuf[addrlen],
 178                                  sizeof(addrbuf) - addrlen - 1,
 179                                  "/%d", plen);
 180                         printf("%-30s", addrbuf);
 181                 } else          /* XXX */
 182                         printf("%s/%d", addrbuf, plen);
 183                 printf(" %5d %5d %8llu\n", pol->preced, pol->label,
 184                     (unsigned long long)pol->use);
 185         }
 186 }
 187
 188 #define SKIP_WHITE(p, emptyok) \
 189         do { \
 190                 while((*(p) == ' ' || *(p) == '\t')) \
 191                         (p)++; \
 192                 if ((*(p) == '\0' || (*(p) == '\n')) && !(emptyok)) \
 193                         goto bad; \
 194         } while (0);
 195 #define SKIP_WORD(p) \
 196         do { \
 197                 while(*(p) != ' ' && *(p) != '\t') \
 198                         (p)++; \
 199                 if (*(p) == '\0' || *(p) == '\n') \
 200                         goto bad; \
 201         } while (0);
 202
 203 static void
 204 make_policy_fromfile(conf)
 205         char *conf;
 206 {
 207         char line[_POSIX2_LINE_MAX], *cp;
 208         char *addrstr;
 209         FILE *fp;
 210         int count = 0;
 211         struct in6_addrpolicy pol0;
 212         struct policyqueue *new;
 213
 214         if ((fp = fopen(conf, "r")) == NULL)
 215                 err(1, "fopen: %s", conf);
 216
 217         while(fgets(line, sizeof(line), fp)) {
 218                 count++;
 219                 cp = line;
 220
 221                 memset(&pol0, 0, sizeof(pol0));
 222
 223                 /* get prefix */
 224                 SKIP_WHITE(cp, 1);
 225                 if (*cp == '\n') /* empty line */
 226                         continue;
 227                 if (*cp == '#')
 228                         continue;
 229                 addrstr = cp;
 230                 if (parse_prefix((const char *)addrstr, &pol0))
 231                         goto bad;
 232
 233                 /* get precedence value */
 234                 SKIP_WORD(cp);
 235                 SKIP_WHITE(cp, 0);
 236                 pol0.preced = atoi(cp);
 237
 238                 /* get label */
 239                 SKIP_WORD(cp);
 240                 SKIP_WHITE(cp, 0);
 241                 pol0.label = atoi(cp);
 242
 243                 /* parse succeeded.  make a control buffer entry. */
 244                 if ((new = malloc(sizeof(*new))) == NULL)
 245                         errx(1, "malloc failed\n");
 246                 memset(new, 0, sizeof(*new));
 247                 new->pc_policy = pol0;
 248                 TAILQ_INSERT_TAIL(&policyhead, new, pc_entry);
 249         }
 250
 251         fclose(fp);
 252         return;
 253
 254   bad:
 255         errx(1, "parse failed at line %d", count);
 256         /* NOTREACHED */
 257 }
 258
 259 static int
 260 parse_prefix(prefix0, pol)
 261         const char *prefix0;
 262         struct in6_addrpolicy *pol;
 263 {
 264         int e = 0, plen;
 265         char *prefix, *plenstr;
 266         struct addrinfo hints, *res;
 267
 268         if ((prefix = strdup(prefix0)) == NULL)
 269                 errx(1, "strdup failed");
 270
 271         if ((plenstr = strchr(prefix, '/')) == NULL) {
 272                 e = -1;
 273                 goto end;
 274         }
 275         *plenstr = '\0';
 276
 277         memset(&hints, 0, sizeof(hints));
 278         hints.ai_flags = AI_NUMERICHOST;
 279         hints.ai_family = AF_INET6;
 280
 281         if ((e = getaddrinfo(prefix, NULL, &hints, &res)) != 0) {
 282                 warnx("getaddrinfo failed for %s: %s", prefix,
 283                       gai_strerror(e));
 284                 goto end;
 285         }
 286         memcpy(&pol->addr, res->ai_addr, res->ai_addrlen);
 287         freeaddrinfo(res);
 288         plen = atoi(plenstr + 1);
 289         if (plen < 0 || plen > 128) {
 290                 warnx("invalid prefix length: %d", plen);
 291                 e = -1;
 292                 goto end;
 293         }
 294         plen2mask(&pol->addrmask, plen);
 295
 296   end:
 297         free(prefix);
 298         return(e);
 299 }
 300
 301 static void
 302 plen2mask(mask, plen)
 303         struct sockaddr_in6 *mask;
 304         int plen;
 305 {
 306         u_char *cp = (unsigned char *)&mask->sin6_addr;
 307
 308         memset(mask, 0, sizeof(*mask));
 309         mask->sin6_family = AF_INET6; /* just in case */
 310         mask->sin6_len = sizeof(*mask);
 311
 312         for(; plen >= 8; plen -= 8)
 313                 *cp++ = 0xff;
 314         if (plen > 0)
 315                 *cp = (0xff << (8 - plen));
 316 }
 317
 318 static void
 319 set_policy()
 320 {
 321         struct policyqueue *ent;
 322         int s;
 323
 324         if ((s = socket(AF_INET6, SOCK_DGRAM, IPPROTO_UDP)) < 0)
 325                 err(1, "socket(UDP)");
 326
 327         for (ent = TAILQ_FIRST(&policyhead); ent;
 328              ent = TAILQ_NEXT(ent, pc_entry)) {
 329                 if (ioctl(s, SIOCAADDRCTL_POLICY, &ent->pc_policy))
 330                         warn("ioctl(SIOCAADDRCTL_POLICY)");
 331         }
 332
 333         close(s);
 334 }
 335
 336 static int
 337 mask2plen(mask)
 338         struct sockaddr_in6 *mask;
 339 {
 340         int masklen, final = 0;
 341         u_char *p, *lim;
 342
 343         masklen = 0;
 344         lim = (u_char *)(mask + 1);
 345         for (p = (u_char *)(&mask->sin6_addr); p < lim; p++) {
 346                 if (final && *p) {
 347                         goto bad;
 348                 }
 349
 350                 switch (*p & 0xff) {
 351                 case 0xff:
 352                         masklen += 8;
 353                         break;
 354                 case 0xfe:
 355                         masklen += 7;
 356                         final++;
 357                         break;
 358                 case 0xfc:
 359                         masklen += 6;
 360                         final++;
 361                         break;
 362                 case 0xf8:
 363                         masklen += 5;
 364                         final++;
 365                         break;
 366                 case 0xf0:
 367                         masklen += 4;
 368                         final++;
 369                         break;
 370                 case 0xe0:
 371                         masklen += 3;
 372                         final++;
 373                         break;
 374                 case 0xc0:
 375                         masklen += 2;
 376                         final++;
 377                         break;
 378                 case 0x80:
 379                         masklen += 1;
 380                         final++;
 381                         break;
 382                 case 0x00:
 383                         final++;
 384                         break;
 385                 default:
 386                         goto bad;
 387                         break;
 388                 }
 389         }
 390         return(masklen);
 391
 392   bad:
 393         return(-1);
 394 }
 395
 396 static void
 397 add_policy(prefix, prec, label)
 398         char *prefix, *prec, *label;
 399 {
 400         struct in6_addrpolicy p;
 401         int s;
 402
 403         memset(&p, 0, sizeof(p));
 404
 405         if (parse_prefix((const char *)prefix, &p))
 406                 errx(1, "bad prefix: %s", prefix);
 407         p.preced = atoi(prec);
 408         p.label = atoi(label);
 409
 410         if ((s = socket(AF_INET6, SOCK_DGRAM, IPPROTO_UDP)) < 0)
 411                 err(1, "socket(UDP)");
 412         if (ioctl(s, SIOCAADDRCTL_POLICY, &p))
 413                 err(1, "ioctl(SIOCAADDRCTL_POLICY)");
 414
 415         close(s);
 416 }
 417
 418 static void
 419 delete_policy(prefix)
 420         char *prefix;
 421 {
 422         struct in6_addrpolicy p;
 423         int s;
 424
 425         memset(&p, 0, sizeof(p));
 426
 427         if (parse_prefix((const char *)prefix, &p))
 428                 errx(1, "bad prefix: %s", prefix);
 429
 430         if ((s = socket(AF_INET6, SOCK_DGRAM, IPPROTO_UDP)) < 0)
 431                 err(1, "socket(UDP)");
 432         if (ioctl(s, SIOCDADDRCTL_POLICY, &p))
 433                 err(1, "ioctl(SIOCDADDRCTL_POLICY)");
 434
 435         close(s);
 436 }
 437
 438 static void
 439 flush_policy()
 440 {
 441         struct policyqueue *ent;
 442         int s;
 443
 444         if ((s = socket(AF_INET6, SOCK_DGRAM, IPPROTO_UDP)) < 0)
 445                 err(1, "socket(UDP)");
 446
 447         for (ent = TAILQ_FIRST(&policyhead); ent;
 448              ent = TAILQ_NEXT(ent, pc_entry)) {
 449                 if (ioctl(s, SIOCDADDRCTL_POLICY, &ent->pc_policy))
 450                         warn("ioctl(SIOCDADDRCTL_POLICY)");
 451         }
 452
 453         close(s);
 454 }
 455
 456 static void
 457 usage()
 458 {
 459         fprintf(stderr, "usage: ip6addrctl [show]\n");
 460         fprintf(stderr, "       ip6addrctl add "
 461                 "<prefix> <precedence> <label>\n");
 462         fprintf(stderr, "       ip6addrctl delete <prefix>\n");
 463         fprintf(stderr, "       ip6addrctl flush\n");
 464         fprintf(stderr, "       ip6addrctl install <configfile>\n");
 465
 466         exit(1);
 467 }