bridge/mdb.c

   1 /*
   2  * Get mdb table with netlink
   3  */
   4
   5 #include <stdio.h>
   6 #include <stdlib.h>
   7 #include <unistd.h>
   8 #include <fcntl.h>
   9 #include <sys/socket.h>
  10 #include <net/if.h>
  11 #include <netinet/in.h>
  12 #include <linux/if_bridge.h>
  13 #include <linux/if_ether.h>
  14 #include <string.h>
  15 #include <arpa/inet.h>
  16
  17 #include "libnetlink.h"
  18 #include "br_common.h"
  19 #include "rt_names.h"
  20 #include "utils.h"
  21
  22 #ifndef MDBA_RTA
  23 #define MDBA_RTA(r) \
  24         ((struct rtattr*)(((char*)(r)) + NLMSG_ALIGN(sizeof(struct br_port_msg))))
  25 #endif
  26
  27 int filter_index;
  28
  29 static void usage(void)
  30 {
  31         fprintf(stderr, "Usage: bridge mdb { add | del } dev DEV port PORT grp GROUP\n");
  32         fprintf(stderr, "       bridge mdb {show} [ dev DEV ]\n");
  33         exit(-1);
  34 }
  35
  36 static void br_print_router_ports(FILE *f, struct rtattr *attr)
  37 {
  38         uint32_t *port_ifindex;
  39         struct rtattr *i;
  40         int rem;
  41
  42         rem = RTA_PAYLOAD(attr);
  43         for (i = RTA_DATA(attr); RTA_OK(i, rem); i = RTA_NEXT(i, rem)) {
  44                 port_ifindex = RTA_DATA(i);
  45                 fprintf(f, "%s ", ll_index_to_name(*port_ifindex));
  46         }
  47
  48         fprintf(f, "\n");
  49 }
  50
  51 static void print_mdb_entry(FILE *f, int ifindex, struct br_mdb_entry *e)
  52 {
  53         SPRINT_BUF(abuf);
  54
  55         if (e->addr.proto == htons(ETH_P_IP))
  56                 fprintf(f, "bridge %s port %s group %s\n", ll_index_to_name(ifindex),
  57                         ll_index_to_name(e->ifindex),
  58                         inet_ntop(AF_INET, &e->addr.u.ip4, abuf, sizeof(abuf)));
  59         else
  60                 fprintf(f, "bridge %s port %s group %s\n", ll_index_to_name(ifindex),
  61                         ll_index_to_name(e->ifindex),
  62                         inet_ntop(AF_INET6, &e->addr.u.ip6, abuf, sizeof(abuf)));
  63 }
  64
  65 static void br_print_mdb_entry(FILE *f, int ifindex, struct rtattr *attr)
  66 {
  67         struct rtattr *i;
  68         int rem;
  69         struct br_mdb_entry *e;
  70
  71         rem = RTA_PAYLOAD(attr);
  72         for (i = RTA_DATA(attr); RTA_OK(i, rem); i = RTA_NEXT(i, rem)) {
  73                 e = RTA_DATA(i);
  74                 print_mdb_entry(f, ifindex, e);
  75         }
  76 }
  77
  78 int print_mdb(const struct sockaddr_nl *who, struct nlmsghdr *n, void *arg)
  79 {
  80         FILE *fp = arg;
  81         struct br_port_msg *r = NLMSG_DATA(n);
  82         int len = n->nlmsg_len;
  83         struct rtattr * tb[MDBA_MAX+1];
  84
  85         if (n->nlmsg_type != RTM_GETMDB) {
  86                 fprintf(stderr, "Not RTM_GETMDB: %08x %08x %08x\n",
  87                         n->nlmsg_len, n->nlmsg_type, n->nlmsg_flags);
  88
  89                 return 0;
  90         }
  91
  92         len -= NLMSG_LENGTH(sizeof(*r));
  93         if (len < 0) {
  94                 fprintf(stderr, "BUG: wrong nlmsg len %d\n", len);
  95                 return -1;
  96         }
  97
  98         if (filter_index && filter_index != r->ifindex)
  99                 return 0;
 100
 101         parse_rtattr(tb, MDBA_MAX, MDBA_RTA(r), n->nlmsg_len - NLMSG_LENGTH(sizeof(*r)));
 102
 103         if (tb[MDBA_MDB]) {
 104                 struct rtattr *i;
 105                 int rem = RTA_PAYLOAD(tb[MDBA_MDB]);
 106
 107                 for (i = RTA_DATA(tb[MDBA_MDB]); RTA_OK(i, rem); i = RTA_NEXT(i, rem))
 108                         br_print_mdb_entry(fp, r->ifindex, i);
 109         }
 110
 111         if (tb[MDBA_ROUTER]) {
 112                 if (show_details) {
 113                         fprintf(fp, "router ports on %s: ", ll_index_to_name(r->ifindex));
 114                         br_print_router_ports(fp, tb[MDBA_ROUTER]);
 115                 }
 116         }
 117
 118         return 0;
 119 }
 120
 121 static int mdb_show(int argc, char **argv)
 122 {
 123         char *filter_dev = NULL;
 124
 125         while (argc > 0) {
 126                 if (strcmp(*argv, "dev") == 0) {
 127                         NEXT_ARG();
 128                         if (filter_dev)
 129                                 duparg("dev", *argv);
 130                         filter_dev = *argv;
 131                 }
 132                 argc--; argv++;
 133         }
 134
 135         if (filter_dev) {
 136                 filter_index = if_nametoindex(filter_dev);
 137                 if (filter_index == 0) {
 138                         fprintf(stderr, "Cannot find device \"%s\"\n",
 139                                 filter_dev);
 140                         return -1;
 141                 }
 142         }
 143
 144         if (rtnl_wilddump_request(&rth, PF_BRIDGE, RTM_GETMDB) < 0) {
 145                 perror("Cannot send dump request");
 146                 exit(1);
 147         }
 148
 149         if (rtnl_dump_filter(&rth, print_mdb, stdout) < 0) {
 150                 fprintf(stderr, "Dump terminated\n");
 151                 exit(1);
 152         }
 153
 154         return 0;
 155 }
 156
 157 static int mdb_modify(int cmd, int flags, int argc, char **argv)
 158 {
 159         struct {
 160                 struct nlmsghdr         n;
 161                 struct br_port_msg      bpm;
 162                 char                    buf[1024];
 163         } req;
 164         struct br_mdb_entry entry;
 165         char *d = NULL, *p = NULL, *grp = NULL;
 166
 167         memset(&req, 0, sizeof(req));
 168         memset(&entry, 0, sizeof(entry));
 169
 170         req.n.nlmsg_len = NLMSG_LENGTH(sizeof(struct br_port_msg));
 171         req.n.nlmsg_flags = NLM_F_REQUEST|flags;
 172         req.n.nlmsg_type = cmd;
 173         req.bpm.family = PF_BRIDGE;
 174
 175         while (argc > 0) {
 176                 if (strcmp(*argv, "dev") == 0) {
 177                         NEXT_ARG();
 178                         d = *argv;
 179                 } else if (strcmp(*argv, "grp") == 0) {
 180                         NEXT_ARG();
 181                         grp = *argv;
 182                 } else {
 183                         if (strcmp(*argv, "port") == 0) {
 184                                 NEXT_ARG();
 185                                 p = *argv;
 186                         }
 187                         if (matches(*argv, "help") == 0)
 188                                 usage();
 189                 }
 190                 argc--; argv++;
 191         }
 192
 193         if (d == NULL || grp == NULL || p == NULL) {
 194                 fprintf(stderr, "Device, group address and port name are required arguments.\n");
 195                 exit(-1);
 196         }
 197
 198         req.bpm.ifindex = ll_name_to_index(d);
 199         if (req.bpm.ifindex == 0) {
 200                 fprintf(stderr, "Cannot find device \"%s\"\n", d);
 201                 return -1;
 202         }
 203
 204         entry.ifindex = ll_name_to_index(p);
 205         if (entry.ifindex == 0) {
 206                 fprintf(stderr, "Cannot find device \"%s\"\n", p);
 207                 return -1;
 208         }
 209
 210         if (!inet_pton(AF_INET, grp, &entry.addr.u.ip4)) {
 211                 if (!inet_pton(AF_INET6, grp, &entry.addr.u.ip6)) {
 212                         fprintf(stderr, "Invalid address \"%s\"\n", grp);
 213                         return -1;
 214                 } else
 215                         entry.addr.proto = htons(ETH_P_IPV6);
 216         } else
 217                 entry.addr.proto = htons(ETH_P_IP);
 218
 219         addattr_l(&req.n, sizeof(req), MDBA_SET_ENTRY, &entry, sizeof(entry));
 220
 221         if (rtnl_talk(&rth, &req.n, 0, 0, NULL) < 0)
 222                 exit(2);
 223
 224         return 0;
 225 }
 226
 227 int do_mdb(int argc, char **argv)
 228 {
 229         ll_init_map(&rth);
 230
 231         if (argc > 0) {
 232                 if (matches(*argv, "add") == 0)
 233                         return mdb_modify(RTM_NEWMDB, NLM_F_CREATE|NLM_F_EXCL, argc-1, argv+1);
 234                 if (matches(*argv, "delete") == 0)
 235                         return mdb_modify(RTM_DELMDB, 0, argc-1, argv+1);
 236
 237                 if (matches(*argv, "show") == 0 ||
 238                     matches(*argv, "lst") == 0 ||
 239                     matches(*argv, "list") == 0)
 240                         return mdb_show(argc-1, argv+1);
 241                 if (matches(*argv, "help") == 0)
 242                         usage();
 243         } else
 244                 return mdb_show(0, NULL);
 245
 246         fprintf(stderr, "Command \"%s\" is unknown, try \"bridge mdb help\".\n", *argv);
 247         exit(-1);
 248 }