]>
Commit | Line | Data |
---|---|---|
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(station); | |
15 | ||
16 | enum plink_state { | |
17 | LISTEN, | |
18 | OPN_SNT, | |
19 | OPN_RCVD, | |
20 | CNF_RCVD, | |
21 | ESTAB, | |
22 | HOLDING, | |
23 | BLOCKED | |
24 | }; | |
25 | ||
26 | enum plink_actions { | |
27 | PLINK_ACTION_UNDEFINED, | |
28 | PLINK_ACTION_OPEN, | |
29 | PLINK_ACTION_BLOCK, | |
30 | }; | |
31 | ||
32 | ||
33 | static int print_sta_handler(struct nl_msg *msg, void *arg) | |
34 | { | |
35 | struct nlattr *tb[NL80211_ATTR_MAX + 1]; | |
36 | struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg)); | |
37 | struct nlattr *sinfo[NL80211_STA_INFO_MAX + 1]; | |
38 | struct nlattr *rinfo[NL80211_RATE_INFO_MAX + 1]; | |
39 | char mac_addr[20], state_name[10], dev[20]; | |
40 | struct nl80211_sta_flag_update *sta_flags; | |
41 | static struct nla_policy stats_policy[NL80211_STA_INFO_MAX + 1] = { | |
42 | [NL80211_STA_INFO_INACTIVE_TIME] = { .type = NLA_U32 }, | |
43 | [NL80211_STA_INFO_RX_BYTES] = { .type = NLA_U32 }, | |
44 | [NL80211_STA_INFO_TX_BYTES] = { .type = NLA_U32 }, | |
45 | [NL80211_STA_INFO_RX_PACKETS] = { .type = NLA_U32 }, | |
46 | [NL80211_STA_INFO_TX_PACKETS] = { .type = NLA_U32 }, | |
47 | [NL80211_STA_INFO_SIGNAL] = { .type = NLA_U8 }, | |
48 | [NL80211_STA_INFO_T_OFFSET] = { .type = NLA_U64 }, | |
49 | [NL80211_STA_INFO_TX_BITRATE] = { .type = NLA_NESTED }, | |
50 | [NL80211_STA_INFO_LLID] = { .type = NLA_U16 }, | |
51 | [NL80211_STA_INFO_PLID] = { .type = NLA_U16 }, | |
52 | [NL80211_STA_INFO_PLINK_STATE] = { .type = NLA_U8 }, | |
53 | [NL80211_STA_INFO_TX_RETRIES] = { .type = NLA_U32 }, | |
54 | [NL80211_STA_INFO_TX_FAILED] = { .type = NLA_U32 }, | |
55 | [NL80211_STA_INFO_STA_FLAGS] = | |
56 | { .minlen = sizeof(struct nl80211_sta_flag_update) }, | |
57 | }; | |
58 | ||
59 | static struct nla_policy rate_policy[NL80211_RATE_INFO_MAX + 1] = { | |
60 | [NL80211_RATE_INFO_BITRATE] = { .type = NLA_U16 }, | |
61 | [NL80211_RATE_INFO_BITRATE32] = { .type = NLA_U32 }, | |
62 | [NL80211_RATE_INFO_MCS] = { .type = NLA_U8 }, | |
63 | [NL80211_RATE_INFO_40_MHZ_WIDTH] = { .type = NLA_FLAG }, | |
64 | [NL80211_RATE_INFO_SHORT_GI] = { .type = NLA_FLAG }, | |
65 | }; | |
66 | ||
67 | nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0), | |
68 | genlmsg_attrlen(gnlh, 0), NULL); | |
69 | ||
70 | /* | |
71 | * TODO: validate the interface and mac address! | |
72 | * Otherwise, there's a race condition as soon as | |
73 | * the kernel starts sending station notifications. | |
74 | */ | |
75 | ||
76 | if (!tb[NL80211_ATTR_STA_INFO]) { | |
77 | fprintf(stderr, "sta stats missing!\n"); | |
78 | return NL_SKIP; | |
79 | } | |
80 | if (nla_parse_nested(sinfo, NL80211_STA_INFO_MAX, | |
81 | tb[NL80211_ATTR_STA_INFO], | |
82 | stats_policy)) { | |
83 | fprintf(stderr, "failed to parse nested attributes!\n"); | |
84 | return NL_SKIP; | |
85 | } | |
86 | ||
87 | mac_addr_n2a(mac_addr, nla_data(tb[NL80211_ATTR_MAC])); | |
88 | if_indextoname(nla_get_u32(tb[NL80211_ATTR_IFINDEX]), dev); | |
89 | printf("Station %s (on %s)", mac_addr, dev); | |
90 | ||
91 | if (sinfo[NL80211_STA_INFO_INACTIVE_TIME]) | |
92 | printf("\n\tinactive time:\t%u ms", | |
93 | nla_get_u32(sinfo[NL80211_STA_INFO_INACTIVE_TIME])); | |
94 | if (sinfo[NL80211_STA_INFO_RX_BYTES]) | |
95 | printf("\n\trx bytes:\t%u", | |
96 | nla_get_u32(sinfo[NL80211_STA_INFO_RX_BYTES])); | |
97 | if (sinfo[NL80211_STA_INFO_RX_PACKETS]) | |
98 | printf("\n\trx packets:\t%u", | |
99 | nla_get_u32(sinfo[NL80211_STA_INFO_RX_PACKETS])); | |
100 | if (sinfo[NL80211_STA_INFO_TX_BYTES]) | |
101 | printf("\n\ttx bytes:\t%u", | |
102 | nla_get_u32(sinfo[NL80211_STA_INFO_TX_BYTES])); | |
103 | if (sinfo[NL80211_STA_INFO_TX_PACKETS]) | |
104 | printf("\n\ttx packets:\t%u", | |
105 | nla_get_u32(sinfo[NL80211_STA_INFO_TX_PACKETS])); | |
106 | if (sinfo[NL80211_STA_INFO_TX_RETRIES]) | |
107 | printf("\n\ttx retries:\t%u", | |
108 | nla_get_u32(sinfo[NL80211_STA_INFO_TX_RETRIES])); | |
109 | if (sinfo[NL80211_STA_INFO_TX_FAILED]) | |
110 | printf("\n\ttx failed:\t%u", | |
111 | nla_get_u32(sinfo[NL80211_STA_INFO_TX_FAILED])); | |
112 | if (sinfo[NL80211_STA_INFO_SIGNAL]) | |
113 | printf("\n\tsignal: \t%d dBm", | |
114 | (int8_t)nla_get_u8(sinfo[NL80211_STA_INFO_SIGNAL])); | |
115 | if (sinfo[NL80211_STA_INFO_SIGNAL_AVG]) | |
116 | printf("\n\tsignal avg:\t%d dBm", | |
117 | (int8_t)nla_get_u8(sinfo[NL80211_STA_INFO_SIGNAL_AVG])); | |
118 | if (sinfo[NL80211_STA_INFO_T_OFFSET]) | |
119 | printf("\n\tToffset:\t%lld us", | |
120 | (unsigned long long)nla_get_u64(sinfo[NL80211_STA_INFO_T_OFFSET])); | |
121 | ||
122 | if (sinfo[NL80211_STA_INFO_TX_BITRATE]) { | |
123 | if (nla_parse_nested(rinfo, NL80211_RATE_INFO_MAX, | |
124 | sinfo[NL80211_STA_INFO_TX_BITRATE], rate_policy)) { | |
125 | fprintf(stderr, "failed to parse nested rate attributes!\n"); | |
126 | } else { | |
127 | int rate = 0; | |
128 | printf("\n\ttx bitrate:\t"); | |
129 | if (rinfo[NL80211_RATE_INFO_BITRATE32]) | |
130 | rate = nla_get_u32(rinfo[NL80211_RATE_INFO_BITRATE32]); | |
131 | else if (rinfo[NL80211_RATE_INFO_BITRATE]) | |
132 | rate = nla_get_u16(rinfo[NL80211_RATE_INFO_BITRATE]); | |
133 | if (rate > 0) | |
134 | printf("%d.%d MBit/s", rate / 10, rate % 10); | |
135 | ||
136 | if (rinfo[NL80211_RATE_INFO_MCS]) | |
137 | printf(" MCS %d", nla_get_u8(rinfo[NL80211_RATE_INFO_MCS])); | |
138 | if (rinfo[NL80211_RATE_INFO_40_MHZ_WIDTH]) | |
139 | printf(" 40Mhz"); | |
140 | if (rinfo[NL80211_RATE_INFO_SHORT_GI]) | |
141 | printf(" short GI"); | |
142 | } | |
143 | } | |
144 | ||
145 | if (sinfo[NL80211_STA_INFO_LLID]) | |
146 | printf("\n\tmesh llid:\t%d", | |
147 | nla_get_u16(sinfo[NL80211_STA_INFO_LLID])); | |
148 | if (sinfo[NL80211_STA_INFO_PLID]) | |
149 | printf("\n\tmesh plid:\t%d", | |
150 | nla_get_u16(sinfo[NL80211_STA_INFO_PLID])); | |
151 | if (sinfo[NL80211_STA_INFO_PLINK_STATE]) { | |
152 | switch (nla_get_u8(sinfo[NL80211_STA_INFO_PLINK_STATE])) { | |
153 | case LISTEN: | |
154 | strcpy(state_name, "LISTEN"); | |
155 | break; | |
156 | case OPN_SNT: | |
157 | strcpy(state_name, "OPN_SNT"); | |
158 | break; | |
159 | case OPN_RCVD: | |
160 | strcpy(state_name, "OPN_RCVD"); | |
161 | break; | |
162 | case CNF_RCVD: | |
163 | strcpy(state_name, "CNF_RCVD"); | |
164 | break; | |
165 | case ESTAB: | |
166 | strcpy(state_name, "ESTAB"); | |
167 | break; | |
168 | case HOLDING: | |
169 | strcpy(state_name, "HOLDING"); | |
170 | break; | |
171 | case BLOCKED: | |
172 | strcpy(state_name, "BLOCKED"); | |
173 | break; | |
174 | default: | |
175 | strcpy(state_name, "UNKNOWN"); | |
176 | break; | |
177 | } | |
178 | printf("\n\tmesh plink:\t%s", state_name); | |
179 | } | |
180 | ||
181 | if (sinfo[NL80211_STA_INFO_STA_FLAGS]) { | |
182 | sta_flags = (struct nl80211_sta_flag_update *) | |
183 | nla_data(sinfo[NL80211_STA_INFO_STA_FLAGS]); | |
184 | ||
185 | if (sta_flags->mask & BIT(NL80211_STA_FLAG_AUTHORIZED)) { | |
186 | printf("\n\tauthorized:\t"); | |
187 | if (sta_flags->set & BIT(NL80211_STA_FLAG_AUTHORIZED)) | |
188 | printf("yes"); | |
189 | else | |
190 | printf("no"); | |
191 | } | |
192 | ||
193 | if (sta_flags->mask & BIT(NL80211_STA_FLAG_AUTHENTICATED)) { | |
194 | printf("\n\tauthenticated:\t"); | |
195 | if (sta_flags->set & BIT(NL80211_STA_FLAG_AUTHENTICATED)) | |
196 | printf("yes"); | |
197 | else | |
198 | printf("no"); | |
199 | } | |
200 | ||
201 | if (sta_flags->mask & BIT(NL80211_STA_FLAG_SHORT_PREAMBLE)) { | |
202 | printf("\n\tpreamble:\t"); | |
203 | if (sta_flags->set & BIT(NL80211_STA_FLAG_SHORT_PREAMBLE)) | |
204 | printf("short"); | |
205 | else | |
206 | printf("long"); | |
207 | } | |
208 | ||
209 | if (sta_flags->mask & BIT(NL80211_STA_FLAG_WME)) { | |
210 | printf("\n\tWMM/WME:\t"); | |
211 | if (sta_flags->set & BIT(NL80211_STA_FLAG_WME)) | |
212 | printf("yes"); | |
213 | else | |
214 | printf("no"); | |
215 | } | |
216 | ||
217 | if (sta_flags->mask & BIT(NL80211_STA_FLAG_MFP)) { | |
218 | printf("\n\tMFP:\t\t"); | |
219 | if (sta_flags->set & BIT(NL80211_STA_FLAG_MFP)) | |
220 | printf("yes"); | |
221 | else | |
222 | printf("no"); | |
223 | } | |
224 | ||
225 | if (sta_flags->mask & BIT(NL80211_STA_FLAG_TDLS_PEER)) { | |
226 | printf("\n\tTDLS peer:\t\t"); | |
227 | if (sta_flags->set & BIT(NL80211_STA_FLAG_TDLS_PEER)) | |
228 | printf("yes"); | |
229 | else | |
230 | printf("no"); | |
231 | } | |
232 | } | |
233 | ||
234 | printf("\n"); | |
235 | return NL_SKIP; | |
236 | } | |
237 | ||
238 | static int handle_station_get(struct nl80211_state *state, | |
239 | struct nl_cb *cb, | |
240 | struct nl_msg *msg, | |
241 | int argc, char **argv, | |
242 | enum id_input id) | |
243 | { | |
244 | unsigned char mac_addr[ETH_ALEN]; | |
245 | ||
246 | if (argc < 1) | |
247 | return 1; | |
248 | ||
249 | if (mac_addr_a2n(mac_addr, argv[0])) { | |
250 | fprintf(stderr, "invalid mac address\n"); | |
251 | return 2; | |
252 | } | |
253 | ||
254 | argc--; | |
255 | argv++; | |
256 | ||
257 | if (argc) | |
258 | return 1; | |
259 | ||
260 | NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, mac_addr); | |
261 | ||
262 | nl_cb_set(cb, NL_CB_VALID, NL_CB_CUSTOM, print_sta_handler, NULL); | |
263 | ||
264 | return 0; | |
265 | nla_put_failure: | |
266 | return -ENOBUFS; | |
267 | } | |
268 | COMMAND(station, get, "<MAC address>", | |
269 | NL80211_CMD_GET_STATION, 0, CIB_NETDEV, handle_station_get, | |
270 | "Get information for a specific station."); | |
271 | COMMAND(station, del, "<MAC address>", | |
272 | NL80211_CMD_DEL_STATION, 0, CIB_NETDEV, handle_station_get, | |
273 | "Remove the given station entry (use with caution!)"); | |
274 | ||
275 | static const struct cmd *station_set_plink; | |
276 | static const struct cmd *station_set_vlan; | |
277 | ||
278 | static const struct cmd *select_station_cmd(int argc, char **argv) | |
279 | { | |
280 | if (argc < 2) | |
281 | return NULL; | |
282 | if (strcmp(argv[1], "plink_action") == 0) | |
283 | return station_set_plink; | |
284 | if (strcmp(argv[1], "vlan") == 0) | |
285 | return station_set_vlan; | |
286 | return NULL; | |
287 | } | |
288 | ||
289 | static int handle_station_set_plink(struct nl80211_state *state, | |
290 | struct nl_cb *cb, | |
291 | struct nl_msg *msg, | |
292 | int argc, char **argv, | |
293 | enum id_input id) | |
294 | { | |
295 | unsigned char plink_action; | |
296 | unsigned char mac_addr[ETH_ALEN]; | |
297 | ||
298 | if (argc < 3) | |
299 | return 1; | |
300 | ||
301 | if (mac_addr_a2n(mac_addr, argv[0])) { | |
302 | fprintf(stderr, "invalid mac address\n"); | |
303 | return 2; | |
304 | } | |
305 | argc--; | |
306 | argv++; | |
307 | ||
308 | if (strcmp("plink_action", argv[0]) != 0) | |
309 | return 1; | |
310 | argc--; | |
311 | argv++; | |
312 | ||
313 | if (strcmp("open", argv[0]) == 0) | |
314 | plink_action = PLINK_ACTION_OPEN; | |
315 | else if (strcmp("block", argv[0]) == 0) | |
316 | plink_action = PLINK_ACTION_BLOCK; | |
317 | else { | |
318 | fprintf(stderr, "plink action not supported\n"); | |
319 | return 2; | |
320 | } | |
321 | argc--; | |
322 | argv++; | |
323 | ||
324 | if (argc) | |
325 | return 1; | |
326 | ||
327 | NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, mac_addr); | |
328 | NLA_PUT_U8(msg, NL80211_ATTR_STA_PLINK_ACTION, plink_action); | |
329 | ||
330 | return 0; | |
331 | nla_put_failure: | |
332 | return -ENOBUFS; | |
333 | } | |
334 | COMMAND_ALIAS(station, set, "<MAC address> plink_action <open|block>", | |
335 | NL80211_CMD_SET_STATION, 0, CIB_NETDEV, handle_station_set_plink, | |
336 | "Set mesh peer link action for this station (peer).", | |
337 | select_station_cmd, station_set_plink); | |
338 | ||
339 | static int handle_station_set_vlan(struct nl80211_state *state, | |
340 | struct nl_cb *cb, | |
341 | struct nl_msg *msg, | |
342 | int argc, char **argv, | |
343 | enum id_input id) | |
344 | { | |
345 | unsigned char mac_addr[ETH_ALEN]; | |
346 | unsigned long sta_vlan = 0; | |
347 | char *err = NULL; | |
348 | ||
349 | if (argc < 3) | |
350 | return 1; | |
351 | ||
352 | if (mac_addr_a2n(mac_addr, argv[0])) { | |
353 | fprintf(stderr, "invalid mac address\n"); | |
354 | return 2; | |
355 | } | |
356 | argc--; | |
357 | argv++; | |
358 | ||
359 | if (strcmp("vlan", argv[0]) != 0) | |
360 | return 1; | |
361 | argc--; | |
362 | argv++; | |
363 | ||
364 | sta_vlan = strtoul(argv[0], &err, 0); | |
365 | if (err && *err) { | |
366 | fprintf(stderr, "invalid vlan id\n"); | |
367 | return 2; | |
368 | } | |
369 | argc--; | |
370 | argv++; | |
371 | ||
372 | if (argc) | |
373 | return 1; | |
374 | ||
375 | NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, mac_addr); | |
376 | NLA_PUT_U32(msg, NL80211_ATTR_STA_VLAN, sta_vlan); | |
377 | ||
378 | return 0; | |
379 | nla_put_failure: | |
380 | return -ENOBUFS; | |
381 | } | |
382 | COMMAND_ALIAS(station, set, "<MAC address> vlan <ifindex>", | |
383 | NL80211_CMD_SET_STATION, 0, CIB_NETDEV, handle_station_set_vlan, | |
384 | "Set an AP VLAN for this station.", | |
385 | select_station_cmd, station_set_vlan); | |
386 | ||
387 | ||
388 | static int handle_station_dump(struct nl80211_state *state, | |
389 | struct nl_cb *cb, | |
390 | struct nl_msg *msg, | |
391 | int argc, char **argv, | |
392 | enum id_input id) | |
393 | { | |
394 | nl_cb_set(cb, NL_CB_VALID, NL_CB_CUSTOM, print_sta_handler, NULL); | |
395 | return 0; | |
396 | } | |
397 | COMMAND(station, dump, NULL, | |
398 | NL80211_CMD_GET_STATION, NLM_F_DUMP, CIB_NETDEV, handle_station_dump, | |
399 | "List all stations known, e.g. the AP on managed interfaces"); |