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
read_file(FILE * file,char * buf,size_t size)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
read_hex(int argc,char ** argv,char * buf,size_t size)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
handle_vendor(struct nl80211_state * state,struct nl_cb * cb,struct nl_msg * msg,int argc,char ** argv,enum id_input id)48 static int handle_vendor(struct nl80211_state *state, struct nl_cb *cb,
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, "");
96