]> git.ipfire.org Git - thirdparty/iw.git/blame - vendor.c
iw: don't use NULL pointer in nla_nest_end()
[thirdparty/iw.git] / vendor.c
CommitLineData
894d2e2b
JD
1#include <errno.h>
2#include <string.h>
3
4#include <netlink/genl/genl.h>
5#include <netlink/genl/family.h>
6#include <netlink/genl/ctrl.h>
7#include <netlink/msg.h>
8#include <netlink/attr.h>
9
10#include "nl80211.h"
11#include "iw.h"
12
13SECTION(vendor);
14
15static int read_file(FILE *file, char *buf, size_t size)
16{
0ee571d5
JB
17 size_t count = 0;
18 int data;
894d2e2b
JD
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
0ee571d5 30static int read_hex(unsigned int argc, char **argv, char *buf, size_t size)
894d2e2b 31{
0ee571d5
JB
32 unsigned int i, data;
33 int res;
894d2e2b
JD
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
34b23014 48static int handle_vendor(struct nl80211_state *state,
894d2e2b
JD
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)
6cd36342 59 return 1;
894d2e2b
JD
60
61 res = sscanf(argv[0], "0x%x", &oui);
6cd36342
OO
62 if (res != 1) {
63 printf("Vendor command must start with 0x\n");
64 return 2;
65 }
894d2e2b
JD
66
67 res = sscanf(argv[1], "0x%x", &subcmd);
6cd36342
OO
68 if (res != 1) {
69 printf("Sub command must start with 0x\n");
70 return 2;
71 }
894d2e2b
JD
72
73 if (!strcmp(argv[2], "-"))
74 file = stdin;
75 else
76 file = fopen(argv[2], "r");
77
78 NLA_PUT_U32(msg, NL80211_ATTR_VENDOR_ID, oui);
79 NLA_PUT_U32(msg, NL80211_ATTR_VENDOR_SUBCMD, subcmd);
80
81 if (file) {
82 count = read_file(file, buf, sizeof(buf));
83 fclose(file);
84 } else
85 count = read_hex(argc - 2, &argv[2], buf, sizeof(buf));
86
87 if (count < 0)
88 return -EINVAL;
89
90 if (count > 0)
91 NLA_PUT(msg, NL80211_ATTR_VENDOR_DATA, count, buf);
92
93 return 0;
94
95nla_put_failure:
96 return -ENOBUFS;
97}
98
99COMMAND(vendor, send, "<oui> <subcmd> <filename|-|hex data>", NL80211_CMD_VENDOR, 0, CIB_NETDEV, handle_vendor, "");