vendor.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 SECTION(vendor);
  15
  16 static int read_file(FILE *file, char *buf, size_t size)
  17 {
  18         int data, count = 0;
  19
  20         while ((data = fgetc(file)) != EOF) {
  21                 if (count >= size)
  22                         return -EINVAL;
  23                 buf[count] = data;
  24                 count++;
  25         }
  26
  27         return count;
  28 }
  29
  30 static int read_hex(int argc, char **argv, char *buf, size_t size)
  31 {
  32         int i, res;
  33         unsigned int data;
  34
  35         if (argc > size)
  36                 return -EINVAL;
  37
  38         for (i = 0; i < argc; i++) {
  39                 res = sscanf(argv[i], "0x%x", &data);
  40                 if (res != 1 || data > 0xff)
  41                         return -EINVAL;
  42                 buf[i] = data;
  43         }
  44
  45         return argc;
  46 }
  47
  48 static int handle_vendor(struct nl80211_state *state,
  49                          struct nl_msg *msg, int argc, char **argv,
  50                          enum id_input id)
  51 {
  52         unsigned int oui;
  53         unsigned int subcmd;
  54         char buf[2048] = {};
  55         int res, count = 0;
  56         FILE *file = NULL;
  57
  58         if (argc < 3)
  59                 return -EINVAL;
  60
  61         res = sscanf(argv[0], "0x%x", &oui);
  62         if (res != 1)
  63                 return -EINVAL;
  64
  65         res = sscanf(argv[1], "0x%x", &subcmd);
  66         if (res != 1)
  67                 return -EINVAL;
  68
  69         if (!strcmp(argv[2], "-"))
  70                 file = stdin;
  71         else
  72                 file = fopen(argv[2], "r");
  73
  74         NLA_PUT_U32(msg, NL80211_ATTR_VENDOR_ID, oui);
  75         NLA_PUT_U32(msg, NL80211_ATTR_VENDOR_SUBCMD, subcmd);
  76
  77         if (file) {
  78                 count = read_file(file, buf, sizeof(buf));
  79                 fclose(file);
  80         } else
  81                 count = read_hex(argc - 2, &argv[2], buf, sizeof(buf));
  82
  83         if (count < 0)
  84                 return -EINVAL;
  85
  86         if (count > 0)
  87                 NLA_PUT(msg, NL80211_ATTR_VENDOR_DATA, count, buf);
  88
  89         return 0;
  90
  91 nla_put_failure:
  92         return -ENOBUFS;
  93 }
  94
  95 COMMAND(vendor, send, "<oui> <subcmd> <filename|-|hex data>", NL80211_CMD_VENDOR, 0, CIB_NETDEV, handle_vendor, "");