ip/ipaddrlabel.c

   1 /*
   2  * ipaddrlabel.c        "ip addrlabel"
   3  *
   4  * Copyright (C)2007 USAGI/WIDE Project
   5  *
   6  * This program is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; either version 2 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with this program; if not, write to the Free Software
  18  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19  *
  20  *
  21  * Based on iprule.c.
  22  *
  23  * Authors:     YOSHIFUJI Hideaki <yoshfuji@linux-ipv6.org>
  24  *
  25  */
  26
  27 #include <stdio.h>
  28 #include <stdlib.h>
  29 #include <unistd.h>
  30 #include <syslog.h>
  31 #include <fcntl.h>
  32 #include <sys/socket.h>
  33 #include <netinet/in.h>
  34 #include <netinet/ip.h>
  35 #include <arpa/inet.h>
  36 #include <string.h>
  37 #include <linux/types.h>
  38 #include <linux/if_addrlabel.h>
  39
  40 #include "rt_names.h"
  41 #include "utils.h"
  42 #include "ip_common.h"
  43
  44 #define IFAL_RTA(r)     ((struct rtattr*)(((char*)(r)) + NLMSG_ALIGN(sizeof(struct ifaddrlblmsg))))
  45 #define IFAL_PAYLOAD(n) NLMSG_PAYLOAD(n,sizeof(struct ifaddrlblmsg))
  46
  47 extern struct rtnl_handle rth;
  48
  49 static void usage(void) __attribute__((noreturn));
  50
  51 static void usage(void)
  52 {
  53         fprintf(stderr, "Usage: ip addrlabel [ list | add | del | flush ] prefix PREFIX [ dev DEV ] [ label LABEL ]\n");
  54         exit(-1);
  55 }
  56
  57 int print_addrlabel(const struct sockaddr_nl *who, struct nlmsghdr *n, void *arg)
  58 {
  59         FILE *fp = (FILE*)arg;
  60         struct ifaddrlblmsg *ifal = NLMSG_DATA(n);
  61         int len = n->nlmsg_len;
  62         int host_len = -1;
  63         struct rtattr *tb[IFAL_MAX+1];
  64         char abuf[256];
  65
  66         if (n->nlmsg_type != RTM_NEWADDRLABEL && n->nlmsg_type != RTM_DELADDRLABEL)
  67                 return 0;
  68
  69         len -= NLMSG_LENGTH(sizeof(*ifal));
  70         if (len < 0)
  71                 return -1;
  72
  73         parse_rtattr(tb, IFAL_MAX, IFAL_RTA(ifal), len);
  74
  75         if (ifal->ifal_family == AF_INET)
  76                 host_len = 32;
  77         else if (ifal->ifal_family == AF_INET6)
  78                 host_len = 128;
  79
  80         if (n->nlmsg_type == RTM_DELADDRLABEL)
  81                 fprintf(fp, "Deleted ");
  82
  83         if (tb[IFAL_ADDRESS]) {
  84                 fprintf(fp, "prefix %s/%u ",
  85                         format_host(ifal->ifal_family,
  86                                     RTA_PAYLOAD(tb[IFAL_ADDRESS]),
  87                                     RTA_DATA(tb[IFAL_ADDRESS]),
  88                                     abuf, sizeof(abuf)),
  89                         ifal->ifal_prefixlen);
  90         }
  91
  92         if (ifal->ifal_index)
  93                 fprintf(fp, "dev %s ", ll_index_to_name(ifal->ifal_index));
  94
  95         if (tb[IFAL_LABEL] && RTA_PAYLOAD(tb[IFAL_LABEL]) == sizeof(int32_t)) {
  96                 int32_t label;
  97                 memcpy(&label, RTA_DATA(tb[IFAL_LABEL]), sizeof(label));
  98                 fprintf(fp, "label %d ", label);
  99         }
 100
 101         fprintf(fp, "\n");
 102         fflush(fp);
 103         return 0;
 104 }
 105
 106 static int ipaddrlabel_list(int argc, char **argv)
 107 {
 108         int af = preferred_family;
 109
 110         if (af == AF_UNSPEC)
 111                 af = AF_INET6;
 112
 113         if (argc > 0) {
 114                 fprintf(stderr, "\"ip addrlabel show\" does not take any arguments.\n");
 115                 return -1;
 116         }
 117
 118         if (rtnl_wilddump_request(&rth, af, RTM_GETADDRLABEL) < 0) {
 119                 perror("Cannot send dump request");
 120                 return 1;
 121         }
 122
 123         if (rtnl_dump_filter(&rth, print_addrlabel, stdout, NULL, NULL) < 0) {
 124                 fprintf(stderr, "Dump terminated\n");
 125                 return 1;
 126         }
 127
 128         return 0;
 129 }
 130
 131
 132 static int ipaddrlabel_modify(int cmd, int argc, char **argv)
 133 {
 134         struct {
 135                 struct nlmsghdr         n;
 136                 struct ifaddrlblmsg     ifal;
 137                 char                    buf[1024];
 138         } req;
 139
 140         inet_prefix prefix;
 141         uint32_t label = 0xffffffffUL;
 142         char *p = NULL;
 143         char *l = NULL;
 144
 145         memset(&req, 0, sizeof(req));
 146         memset(&prefix, 0, sizeof(prefix));
 147
 148         req.n.nlmsg_type = cmd;
 149         req.n.nlmsg_len = NLMSG_LENGTH(sizeof(struct ifaddrlblmsg));
 150         req.n.nlmsg_flags = NLM_F_REQUEST;
 151         req.ifal.ifal_family = preferred_family;
 152         req.ifal.ifal_prefixlen = 0;
 153         req.ifal.ifal_index = 0;
 154
 155         if (cmd == RTM_NEWADDRLABEL) {
 156                 req.n.nlmsg_flags |= NLM_F_CREATE|NLM_F_EXCL;
 157         }
 158
 159         while (argc > 0) {
 160                 if (strcmp(*argv, "prefix") == 0) {
 161                         NEXT_ARG();
 162                         p = *argv;
 163                         get_prefix(&prefix, *argv, preferred_family);
 164                 } else if (strcmp(*argv, "dev") == 0) {
 165                         NEXT_ARG();
 166                         if ((req.ifal.ifal_index = ll_name_to_index(*argv)) == 0)
 167                                 invarg("dev is invalid\n", *argv);
 168                 } else if (strcmp(*argv, "label") == 0) {
 169                         NEXT_ARG();
 170                         l = *argv;
 171                         if (get_u32(&label, *argv, 0) || label == 0xffffffffUL)
 172                                 invarg("label is invalid\n", *argv);
 173                 }
 174                 argc--;
 175                 argv++;
 176         }
 177         if (p == NULL) {
 178                 fprintf(stderr, "Not enough information: \"prefix\" argument is required.\n");
 179                 return -1;
 180         }
 181         if (l == NULL) {
 182                 fprintf(stderr, "Not enough information: \"label\" argument is required.\n");
 183                 return -1;
 184         }
 185         addattr32(&req.n, sizeof(req), IFAL_LABEL, label);
 186         addattr_l(&req.n, sizeof(req), IFAL_ADDRESS, &prefix.data, prefix.bytelen);
 187         req.ifal.ifal_prefixlen = prefix.bitlen;
 188
 189         if (req.ifal.ifal_family == AF_UNSPEC)
 190                 req.ifal.ifal_family = AF_INET6;
 191
 192         if (rtnl_talk(&rth, &req.n, 0, 0, NULL, NULL, NULL) < 0)
 193                 return 2;
 194
 195         return 0;
 196 }
 197
 198
 199 static int flush_addrlabel(const struct sockaddr_nl *who, struct nlmsghdr *n, void *arg)
 200 {
 201         struct rtnl_handle rth2;
 202         struct rtmsg *r = NLMSG_DATA(n);
 203         int len = n->nlmsg_len;
 204         struct rtattr * tb[IFAL_MAX+1];
 205
 206         len -= NLMSG_LENGTH(sizeof(*r));
 207         if (len < 0)
 208                 return -1;
 209
 210         parse_rtattr(tb, IFAL_MAX, RTM_RTA(r), len);
 211
 212         if (tb[IFAL_ADDRESS]) {
 213                 n->nlmsg_type = RTM_DELADDRLABEL;
 214                 n->nlmsg_flags = NLM_F_REQUEST;
 215
 216                 if (rtnl_open(&rth2, 0) < 0)
 217                         return -1;
 218
 219                 if (rtnl_talk(&rth2, n, 0, 0, NULL, NULL, NULL) < 0)
 220                         return -2;
 221
 222                 rtnl_close(&rth2);
 223         }
 224
 225         return 0;
 226 }
 227
 228 static int ipaddrlabel_flush(int argc, char **argv)
 229 {
 230         int af = preferred_family;
 231
 232         if (af == AF_UNSPEC)
 233                 af = AF_INET6;
 234
 235         if (argc > 0) {
 236                 fprintf(stderr, "\"ip addrlabel flush\" does not allow extra arguments\n");
 237                 return -1;
 238         }
 239
 240         if (rtnl_wilddump_request(&rth, af, RTM_GETADDRLABEL) < 0) {
 241                 perror("Cannot send dump request");
 242                 return 1;
 243         }
 244
 245         if (rtnl_dump_filter(&rth, flush_addrlabel, NULL, NULL, NULL) < 0) {
 246                 fprintf(stderr, "Flush terminated\n");
 247                 return 1;
 248         }
 249
 250         return 0;
 251 }
 252
 253 int do_ipaddrlabel(int argc, char **argv)
 254 {
 255         ll_init_map(&rth);
 256
 257         if (argc < 1) {
 258                 return ipaddrlabel_list(0, NULL);
 259         } else if (matches(argv[0], "list") == 0 ||
 260                    matches(argv[0], "show") == 0) {
 261                 return ipaddrlabel_list(argc-1, argv+1);
 262         } else if (matches(argv[0], "add") == 0) {
 263                 return ipaddrlabel_modify(RTM_NEWADDRLABEL, argc-1, argv+1);
 264         } else if (matches(argv[0], "delete") == 0) {
 265                 return ipaddrlabel_modify(RTM_DELADDRLABEL, argc-1, argv+1);
 266         } else if (matches(argv[0], "flush") == 0) {
 267                 return ipaddrlabel_flush(argc-1, argv+1);
 268         } else if (matches(argv[0], "help") == 0)
 269                 usage();
 270
 271         fprintf(stderr, "Command \"%s\" is unknown, try \"ip addrlabel help\".\n", *argv);
 272         exit(-1);
 273 }
 274