mpath.c

   1 #include <net/if.h>
   2 #include <errno.h>
   3 #include <string.h>
   4
   5 #include <netlink/genl/genl.h>
   6 #include <netlink/genl/family.h>
   7 #include <netlink/genl/ctrl.h>
   8 #include <netlink/msg.h>
   9 #include <netlink/attr.h>
  10
  11 #include "nl80211.h"
  12 #include "iw.h"
  13
  14 enum plink_state {
  15         LISTEN,
  16         OPN_SNT,
  17         OPN_RCVD,
  18         CNF_RCVD,
  19         ESTAB,
  20         HOLDING,
  21         BLOCKED
  22 };
  23
  24 enum plink_actions {
  25         PLINK_ACTION_UNDEFINED,
  26         PLINK_ACTION_OPEN,
  27         PLINK_ACTION_BLOCK,
  28 };
  29
  30
  31 static int print_mpath_handler(struct nl_msg *msg, void *arg)
  32 {
  33         struct nlattr *tb[NL80211_ATTR_MAX + 1];
  34         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
  35         struct nlattr *pinfo[NL80211_MPATH_INFO_MAX + 1];
  36         char dst[20], next_hop[20], dev[20];
  37         static struct nla_policy mpath_policy[NL80211_MPATH_INFO_MAX + 1] = {
  38                 [NL80211_MPATH_INFO_FRAME_QLEN] = { .type = NLA_U32 },
  39                 [NL80211_MPATH_INFO_DSN] = { .type = NLA_U32 },
  40                 [NL80211_MPATH_INFO_METRIC] = { .type = NLA_U32 },
  41                 [NL80211_MPATH_INFO_EXPTIME] = { .type = NLA_U32 },
  42                 [NL80211_MPATH_INFO_DISCOVERY_TIMEOUT] = { .type = NLA_U32 },
  43                 [NL80211_MPATH_INFO_DISCOVERY_RETRIES] = { .type = NLA_U8 },
  44                 [NL80211_MPATH_INFO_FLAGS] = { .type = NLA_U8 },
  45         };
  46
  47         nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
  48                   genlmsg_attrlen(gnlh, 0), NULL);
  49
  50         /*
  51          * TODO: validate the interface and mac address!
  52          * Otherwise, there's a race condition as soon as
  53          * the kernel starts sending mpath notifications.
  54          */
  55
  56         if (!tb[NL80211_ATTR_MPATH_INFO]) {
  57                 fprintf(stderr, "mpath info missing!");
  58                 return NL_SKIP;
  59         }
  60         if (nla_parse_nested(pinfo, NL80211_MPATH_INFO_MAX,
  61                              tb[NL80211_ATTR_MPATH_INFO],
  62                              mpath_policy)) {
  63                 fprintf(stderr, "failed to parse nested attributes!");
  64                 return NL_SKIP;
  65         }
  66
  67         mac_addr_n2a(dst, nla_data(tb[NL80211_ATTR_MAC]));
  68         mac_addr_n2a(next_hop, nla_data(tb[NL80211_ATTR_MPATH_NEXT_HOP]));
  69         if_indextoname(nla_get_u32(tb[NL80211_ATTR_IFINDEX]), dev);
  70         printf("%s %s %s", dst, next_hop, dev);
  71         if (pinfo[NL80211_MPATH_INFO_DSN])
  72                 printf("\t%u",
  73                         nla_get_u32(pinfo[NL80211_MPATH_INFO_DSN]));
  74         if (pinfo[NL80211_MPATH_INFO_METRIC])
  75                 printf("\t%u",
  76                         nla_get_u32(pinfo[NL80211_MPATH_INFO_METRIC]));
  77         if (pinfo[NL80211_MPATH_INFO_FRAME_QLEN])
  78                 printf("\t%u",
  79                         nla_get_u32(pinfo[NL80211_MPATH_INFO_FRAME_QLEN]));
  80         if (pinfo[NL80211_MPATH_INFO_EXPTIME])
  81                 printf("\t%u",
  82                         nla_get_u32(pinfo[NL80211_MPATH_INFO_EXPTIME]));
  83         if (pinfo[NL80211_MPATH_INFO_DISCOVERY_TIMEOUT])
  84                 printf("\t%u",
  85                 nla_get_u32(pinfo[NL80211_MPATH_INFO_DISCOVERY_TIMEOUT]));
  86         if (pinfo[NL80211_MPATH_INFO_DISCOVERY_RETRIES])
  87                 printf("\t%u",
  88                 nla_get_u8(pinfo[NL80211_MPATH_INFO_DISCOVERY_RETRIES]));
  89         if (pinfo[NL80211_MPATH_INFO_FLAGS])
  90                 printf("\t0x%x",
  91                         nla_get_u8(pinfo[NL80211_MPATH_INFO_FLAGS]));
  92
  93         printf("\n");
  94         return NL_SKIP;
  95 }
  96
  97 static int handle_mpath_get(struct nl80211_state *state,
  98                             struct nl_cb *cb,
  99                             struct nl_msg *msg,
 100                             int argc, char **argv)
 101 {
 102         unsigned char dst[ETH_ALEN];
 103
 104         if (argc < 1)
 105                 return 1;
 106
 107         if (mac_addr_a2n(dst, argv[0])) {
 108                 fprintf(stderr, "invalid mac address\n");
 109                 return 2;
 110         }
 111         argc--;
 112         argv++;
 113
 114         if (argc)
 115                 return 1;
 116
 117         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, dst);
 118
 119         nl_cb_set(cb, NL_CB_VALID, NL_CB_CUSTOM, print_mpath_handler, NULL);
 120
 121         return 0;
 122  nla_put_failure:
 123         return -ENOBUFS;
 124 }
 125 COMMAND(mpath, get, "<MAC address>",
 126         NL80211_CMD_GET_MPATH, 0, CIB_NETDEV, handle_mpath_get);
 127 COMMAND(mpath, del, "<MAC address>",
 128         NL80211_CMD_DEL_MPATH, 0, CIB_NETDEV, handle_mpath_get);
 129
 130 static int handle_mpath_set(struct nl80211_state *state,
 131                             struct nl_cb *cb,
 132                             struct nl_msg *msg,
 133                             int argc, char **argv)
 134 {
 135         unsigned char dst[ETH_ALEN];
 136         unsigned char next_hop[ETH_ALEN];
 137
 138         if (argc < 3)
 139                 return 1;
 140
 141         if (mac_addr_a2n(dst, argv[0])) {
 142                 fprintf(stderr, "invalid destination mac address\n");
 143                 return 2;
 144         }
 145         argc--;
 146         argv++;
 147
 148         if (strcmp("next_hop", argv[0]) != 0)
 149                 return 1;
 150         argc--;
 151         argv++;
 152
 153         if (mac_addr_a2n(next_hop, argv[0])) {
 154                 fprintf(stderr, "invalid next hop mac address\n");
 155                 return 2;
 156         }
 157         argc--;
 158         argv++;
 159
 160         if (argc)
 161                 return 1;
 162
 163         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, dst);
 164         NLA_PUT(msg, NL80211_ATTR_MPATH_NEXT_HOP, ETH_ALEN, next_hop);
 165
 166         nl_cb_set(cb, NL_CB_VALID, NL_CB_CUSTOM, print_mpath_handler, NULL);
 167         return 0;
 168  nla_put_failure:
 169         return -ENOBUFS;
 170 }
 171 COMMAND(mpath, new, "<destination MAC address> next_hop <next hop MAC address>",
 172         NL80211_CMD_NEW_MPATH, 0, CIB_NETDEV, handle_mpath_set);
 173 COMMAND(mpath, set, "<destination MAC address> next_hop <next hop MAC address>",
 174         NL80211_CMD_SET_MPATH, 0, CIB_NETDEV, handle_mpath_set);
 175
 176 static int handle_mpath_dump(struct nl80211_state *state,
 177                              struct nl_cb *cb,
 178                              struct nl_msg *msg,
 179                              int argc, char **argv)
 180 {
 181         nl_cb_set(cb, NL_CB_VALID, NL_CB_CUSTOM, print_mpath_handler, NULL);
 182         return 0;
 183 }
 184 COMMAND(mpath, dump, NULL,
 185         NL80211_CMD_GET_MPATH, NLM_F_DUMP, CIB_NETDEV, handle_mpath_dump);