d00d26b0adb401668dc71e3e93d134e8ee922fab
[project/luci.git] / contrib / package / iwinfo / src / iwinfo_nl80211.c
1 /*
2  * iwinfo - Wireless Information Library - NL80211 Backend
3  *
4  *   Copyright (C) 2010 Jo-Philipp Wich <xm@subsignal.org>
5  *
6  * The iwinfo library is free software: you can redistribute it and/or
7  * modify it under the terms of the GNU General Public License version 2
8  * as published by the Free Software Foundation.
9  *
10  * The iwinfo library is distributed in the hope that it will be useful,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
13  * See the GNU General Public License for more details.
14  *
15  * You should have received a copy of the GNU General Public License along
16  * with the iwinfo library. If not, see http://www.gnu.org/licenses/.
17  *
18  * The signal handling code is derived from the official madwifi tools,
19  * wlanconfig.c in particular. The encryption property handling was
20  * inspired by the hostapd madwifi driver.
21  *
22  * Parts of this code are derived from the Linux iw utility.
23  */
24
25 #include "iwinfo_nl80211.h"
26 #include "iwinfo_wext.h"
27
28 #define min(x, y) ((x) < (y)) ? (x) : (y)
29
30 extern struct iwinfo_iso3166_label ISO3166_Names[];
31 static struct nl80211_state *nls = NULL;
32
33 static int nl80211_init(void)
34 {
35         int err, fd;
36
37         if (!nls)
38         {
39                 nls = malloc(sizeof(struct nl80211_state));
40                 if (!nls) {
41                         err = -ENOMEM;
42                         goto err;
43                 }
44
45                 nls->nl_sock = nl_socket_alloc();
46                 if (!nls->nl_sock) {
47                         err = -ENOMEM;
48                         goto err;
49                 }
50
51                 if( genl_connect(nls->nl_sock)) {
52                         err = -ENOLINK;
53                         goto err;
54                 }
55
56                 fd = nl_socket_get_fd(nls->nl_sock);
57                 if (fcntl(fd, F_SETFD, fcntl(fd, F_GETFD) | FD_CLOEXEC) < 0)
58                 {
59                         err = -EINVAL;
60                         goto err;
61                 }
62
63                 if( genl_ctrl_alloc_cache(nls->nl_sock, &nls->nl_cache)) {
64                         err = -ENOMEM;
65                         goto err;
66                 }
67
68                 nls->nl80211 = genl_ctrl_search_by_name(nls->nl_cache, "nl80211");
69                 if (!nls->nl80211)
70                 {
71                         err = -ENOENT;
72                         goto err;
73                 }
74         }
75
76         return 0;
77
78
79 err:
80         nl80211_close();
81         return err;
82 }
83
84 static int nl80211_msg_error(struct sockaddr_nl *nla,
85         struct nlmsgerr *err, void *arg)
86 {
87         int *ret = arg;
88         *ret = err->error;
89         return NL_STOP;
90 }
91
92 static int nl80211_msg_finish(struct nl_msg *msg, void *arg)
93 {
94         int *ret = arg;
95         *ret = 0;
96         return NL_SKIP;
97 }
98
99 static int nl80211_msg_ack(struct nl_msg *msg, void *arg)
100 {
101         int *ret = arg;
102         *ret = 0;
103         return NL_STOP;
104 }
105
106 static int nl80211_msg_response(struct nl_msg *msg, void *arg)
107 {
108         return NL_SKIP;
109 }
110
111 static void nl80211_free(struct nl80211_msg_conveyor *cv)
112 {
113         if (cv)
114         {
115                 if (cv->cb)
116                         nl_cb_put(cv->cb);
117
118                 if (cv->msg)
119                         nlmsg_free(cv->msg);
120
121                 cv->cb  = NULL;
122                 cv->msg = NULL;
123         }
124 }
125
126 static struct nl80211_msg_conveyor * nl80211_msg(const char *ifname, int cmd, int flags)
127 {
128         static struct nl80211_msg_conveyor cv;
129
130         int ifidx = -1, phyidx = -1;
131         struct nl_msg *req = NULL;
132         struct nl_cb *cb = NULL;
133
134         if (nl80211_init() < 0)
135                 goto err;
136
137         if (!strncmp(ifname, "phy", 3))
138                 phyidx = atoi(&ifname[3]);
139         else if (!strncmp(ifname, "radio", 5))
140                 phyidx = atoi(&ifname[5]);
141         else if (!strncmp(ifname, "mon.", 4))
142                 ifidx = if_nametoindex(&ifname[4]);
143         else
144                 ifidx = if_nametoindex(ifname);
145
146         if ((ifidx < 0) && (phyidx < 0))
147                 return NULL;
148
149         req = nlmsg_alloc();
150         if (!req)
151                 goto err;
152
153         cb = nl_cb_alloc(NL_CB_DEFAULT);
154         if (!cb)
155                 goto err;
156
157         genlmsg_put(req, 0, 0, genl_family_get_id(nls->nl80211), 0,
158                 flags, cmd, 0);
159
160         if (ifidx > -1)
161                 NLA_PUT_U32(req, NL80211_ATTR_IFINDEX, ifidx);
162
163         if (phyidx > -1)
164                 NLA_PUT_U32(req, NL80211_ATTR_WIPHY, phyidx);
165
166         cv.msg = req;
167         cv.cb  = cb;
168
169         return &cv;
170
171 err:
172 nla_put_failure:
173         if (cb)
174                 nl_cb_put(cb);
175
176         if (req)
177                 nlmsg_free(req);
178
179         return NULL;
180 }
181
182 static struct nl80211_msg_conveyor * nl80211_send(
183         struct nl80211_msg_conveyor *cv,
184         int (*cb_func)(struct nl_msg *, void *), void *cb_arg
185 ) {
186         static struct nl80211_msg_conveyor rcv;
187         int err = 1;
188
189         if (cb_func)
190                 nl_cb_set(cv->cb, NL_CB_VALID, NL_CB_CUSTOM, cb_func, cb_arg);
191         else
192                 nl_cb_set(cv->cb, NL_CB_VALID, NL_CB_CUSTOM, nl80211_msg_response, &rcv);
193
194         if (nl_send_auto_complete(nls->nl_sock, cv->msg) < 0)
195                 goto err;
196
197         nl_cb_err(cv->cb,               NL_CB_CUSTOM, nl80211_msg_error,  &err);
198         nl_cb_set(cv->cb, NL_CB_FINISH, NL_CB_CUSTOM, nl80211_msg_finish, &err);
199         nl_cb_set(cv->cb, NL_CB_ACK,    NL_CB_CUSTOM, nl80211_msg_ack,    &err);
200
201         while (err > 0)
202                 nl_recvmsgs(nls->nl_sock, cv->cb);
203
204         return &rcv;
205
206 err:
207         nl_cb_put(cv->cb);
208         nlmsg_free(cv->msg);
209
210         return NULL;
211 }
212
213 static struct nlattr ** nl80211_parse(struct nl_msg *msg)
214 {
215         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
216         static struct nlattr *attr[NL80211_ATTR_MAX + 1];
217
218         nla_parse(attr, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
219                 genlmsg_attrlen(gnlh, 0), NULL);
220
221         return attr;
222 }
223
224 static int nl80211_freq2channel(int freq)
225 {
226     if (freq == 2484)
227         return 14;
228
229     if (freq < 2484)
230         return (freq - 2407) / 5;
231
232     return (freq / 5) - 1000;
233 }
234
235 static char * nl80211_getval(const char *ifname, const char *buf, const char *key)
236 {
237         int i, len;
238         char lkey[64] = { 0 };
239         const char *ln = buf;
240         static char lval[256] = { 0 };
241
242         int matched_if = ifname ? 0 : 1;
243
244
245         for( i = 0, len = strlen(buf); i < len; i++ )
246         {
247                 if (!lkey[0] && (buf[i] == ' ' || buf[i] == '\t'))
248                 {
249                         ln++;
250                 }
251                 else if (!lkey[0] && (buf[i] == '='))
252                 {
253                         if ((&buf[i] - ln) > 0)
254                                 memcpy(lkey, ln, min(sizeof(lkey) - 1, &buf[i] - ln));
255                 }
256                 else if (buf[i] == '\n')
257                 {
258                         if (lkey[0])
259                         {
260                                 memcpy(lval, ln + strlen(lkey) + 1,
261                                         min(sizeof(lval) - 1, &buf[i] - ln - strlen(lkey) - 1));
262
263                                 if ((ifname != NULL) &&
264                                     (!strcmp(lkey, "interface") || !strcmp(lkey, "bss")) )
265                                 {
266                                         matched_if = !strcmp(lval, ifname);
267                                 }
268                                 else if (matched_if && !strcmp(lkey, key))
269                                 {
270                                         return lval;
271                                 }
272                         }
273
274                         ln = &buf[i+1];
275                         memset(lkey, 0, sizeof(lkey));
276                         memset(lval, 0, sizeof(lval));
277                 }
278         }
279
280         return NULL;
281 }
282
283 static int nl80211_ifname2phy_cb(struct nl_msg *msg, void *arg)
284 {
285         char *buf = arg;
286         struct nlattr **attr = nl80211_parse(msg);
287
288         if (attr[NL80211_ATTR_WIPHY_NAME])
289                 sprintf(buf, "%s", nla_data(attr[NL80211_ATTR_WIPHY_NAME]));
290         else
291                 buf[0] = 0;
292
293         return NL_SKIP;
294 }
295
296 static char * nl80211_ifname2phy(const char *ifname)
297 {
298         static char phy[32] = { 0 };
299         struct nl80211_msg_conveyor *req;
300
301         req = nl80211_msg(ifname, NL80211_CMD_GET_WIPHY, 0);
302         if (req)
303         {
304                 nl80211_send(req, nl80211_ifname2phy_cb, phy);
305                 nl80211_free(req);
306         }
307
308         return phy[0] ? phy : NULL;
309 }
310
311 static char * nl80211_hostapd_info(const char *ifname)
312 {
313         char *phy;
314         char path[32] = { 0 };
315         static char buf[4096] = { 0 };
316         FILE *conf;
317
318         if ((phy = nl80211_ifname2phy(ifname)) != NULL)
319         {
320                 snprintf(path, sizeof(path), "/var/run/hostapd-%s.conf", phy);
321
322                 if ((conf = fopen(path, "r")) != NULL)
323                 {
324                         fread(buf, sizeof(buf) - 1, 1, conf);
325                         fclose(conf);
326
327                         return buf;
328                 }
329         }
330
331         return NULL;
332 }
333
334 static inline int nl80211_wpactl_recv(int sock, char *buf, int blen)
335 {
336         fd_set rfds;
337         struct timeval tv = { 2, 0 };
338
339         FD_ZERO(&rfds);
340         FD_SET(sock, &rfds);
341
342         memset(buf, 0, blen);
343
344
345         if (select(sock + 1, &rfds, NULL, NULL, &tv) < 0)
346                 return -1;
347
348         if (!FD_ISSET(sock, &rfds))
349                 return -1;
350
351         return recv(sock, buf, blen, 0);
352 }
353
354 static char * nl80211_wpactl_info(const char *ifname, const char *cmd,
355                                                                    const char *event)
356 {
357         int numtry = 0;
358         int sock = -1;
359         char *rv = NULL;
360         size_t remote_length, local_length;
361         static char buffer[10240] = { 0 };
362
363         struct sockaddr_un local = { 0 };
364         struct sockaddr_un remote = { 0 };
365
366
367         sock = socket(PF_UNIX, SOCK_DGRAM, 0);
368         if (sock < 0)
369                 return NULL;
370
371         remote.sun_family = AF_UNIX;
372         remote_length = sizeof(remote.sun_family) + sprintf(remote.sun_path,
373                 "/var/run/wpa_supplicant-%s/%s", ifname, ifname);
374
375         if (fcntl(sock, F_SETFD, fcntl(sock, F_GETFD) | FD_CLOEXEC) < 0)
376                 goto out;
377
378         if (connect(sock, (struct sockaddr *) &remote, remote_length))
379                 goto out;
380
381         local.sun_family = AF_UNIX;
382         local_length = sizeof(local.sun_family) + sprintf(local.sun_path,
383                 "/var/run/iwinfo-%s-%d", ifname, getpid());
384
385         if (bind(sock, (struct sockaddr *) &local, local_length))
386                 goto out;
387
388
389         send(sock, "ATTACH", 6, 0);
390
391         if (nl80211_wpactl_recv(sock, buffer, sizeof(buffer)) <= 0)
392                 goto out;
393
394
395         send(sock, cmd, strlen(cmd), 0);
396
397         while( numtry++ < 5 )
398         {
399                 if (nl80211_wpactl_recv(sock, buffer, sizeof(buffer)) <= 0)
400                 {
401                         if (event)
402                                 continue;
403
404                         break;
405                 }
406
407                 if ((!event && buffer[0] != '<') || strstr(buffer, event))
408                         break;
409         }
410
411         rv = buffer;
412
413 out:
414         close(sock);
415
416         if (local.sun_family)
417                 unlink(local.sun_path);
418
419         return rv;
420 }
421
422 static inline int nl80211_readint(const char *path)
423 {
424         int fd;
425         int rv = -1;
426         char buffer[16];
427
428         if ((fd = open(path, O_RDONLY)) > -1)
429         {
430                 if (read(fd, buffer, sizeof(buffer)) > 0)
431                         rv = atoi(buffer);
432
433                 close(fd);
434         }
435
436         return rv;
437 }
438
439 static char * nl80211_phy2ifname(const char *ifname)
440 {
441         int fd, ifidx = -1, cifidx = -1, phyidx = -1;
442         char buffer[64];
443         static char nif[IFNAMSIZ] = { 0 };
444
445         DIR *d;
446         struct dirent *e;
447
448         if (!ifname)
449                 return NULL;
450         else if (!strncmp(ifname, "phy", 3))
451                 phyidx = atoi(&ifname[3]);
452         else if (!strncmp(ifname, "radio", 5))
453                 phyidx = atoi(&ifname[5]);
454
455         memset(nif, 0, sizeof(nif));
456
457         if (phyidx > -1)
458         {
459                 if ((d = opendir("/sys/class/net")) != NULL)
460                 {
461                         while( (e = readdir(d)) != NULL )
462                         {
463                                 snprintf(buffer, sizeof(buffer),
464                                         "/sys/class/net/%s/phy80211/index", e->d_name);
465
466                                 if (nl80211_readint(buffer) == phyidx)
467                                 {
468                                         snprintf(buffer, sizeof(buffer),
469                                                 "/sys/class/net/%s/ifindex", e->d_name);
470
471                                         if( (cifidx = nl80211_readint(buffer)) >= 0 &&
472                                             ((ifidx < 0) || (cifidx < ifidx)) )
473                                         {
474                                                 ifidx = cifidx;
475                                                 strncpy(nif, e->d_name, sizeof(nif));
476                                         }
477                                 }
478                         }
479
480                         closedir(d);
481                 }
482         }
483
484         return nif[0] ? nif : NULL;
485 }
486
487 static char * nl80211_ifadd(const char *ifname)
488 {
489         int phyidx;
490         char *rv = NULL;
491         static char nif[IFNAMSIZ] = { 0 };
492         struct nl80211_msg_conveyor *req, *res;
493
494         req = nl80211_msg(ifname, NL80211_CMD_NEW_INTERFACE, 0);
495         if (req)
496         {
497                 snprintf(nif, sizeof(nif), "tmp.%s", ifname);
498
499                 NLA_PUT_STRING(req->msg, NL80211_ATTR_IFNAME, nif);
500                 NLA_PUT_U32(req->msg, NL80211_ATTR_IFTYPE, NL80211_IFTYPE_STATION);
501
502                 nl80211_send(req, NULL, NULL);
503
504                 rv = nif;
505
506         nla_put_failure:
507                 nl80211_free(req);
508         }
509
510         return rv;
511 }
512
513 static void nl80211_ifdel(const char *ifname)
514 {
515         struct nl80211_msg_conveyor *req;
516
517         req = nl80211_msg(ifname, NL80211_CMD_DEL_INTERFACE, 0);
518         if (req)
519         {
520                 NLA_PUT_STRING(req->msg, NL80211_ATTR_IFNAME, ifname);
521
522                 nl80211_send(req, NULL, NULL);
523
524         nla_put_failure:
525                 nl80211_free(req);
526         }
527 }
528
529 static void nl80211_hostapd_hup(const char *ifname)
530 {
531         int fd, pid = 0;
532         char buf[32];
533         char *phy = nl80211_ifname2phy(ifname);
534
535         if (phy)
536         {
537                 snprintf(buf, sizeof(buf), "/var/run/wifi-%s.pid", phy);
538                 if ((fd = open(buf, O_RDONLY)) > 0)
539                 {
540                         if (read(fd, buf, sizeof(buf)) > 0)
541                                 pid = atoi(buf);
542
543                         close(fd);
544                 }
545
546                 if (pid > 0)
547                         kill(pid, 1);
548         }
549 }
550
551
552 int nl80211_probe(const char *ifname)
553 {
554         return !!nl80211_ifname2phy(ifname);
555 }
556
557 void nl80211_close(void)
558 {
559         if (nls)
560         {
561                 if (nls->nl_sock)
562                         nl_socket_free(nls->nl_sock);
563
564                 if (nls->nl_cache)
565                         nl_cache_free(nls->nl_cache);
566
567                 free(nls);
568                 nls = NULL;
569         }
570 }
571
572 int nl80211_get_mode(const char *ifname, char *buf)
573 {
574         return wext_get_mode(ifname, buf);
575 }
576
577 int nl80211_get_ssid(const char *ifname, char *buf)
578 {
579         char *ssid;
580
581         if (!wext_get_ssid(ifname, buf))
582         {
583                 return 0;
584         }
585         else if( (ssid = nl80211_hostapd_info(ifname)) &&
586                  (ssid = nl80211_getval(ifname, ssid, "ssid")) )
587         {
588                 memcpy(buf, ssid, strlen(ssid));
589                 return 0;
590         }
591
592         return -1;
593 }
594
595 int nl80211_get_bssid(const char *ifname, char *buf)
596 {
597         char *bssid;
598         unsigned char mac[6];
599
600         if (!wext_get_bssid(ifname, buf))
601         {
602                 return 0;
603         }
604         else if((bssid = nl80211_hostapd_info(ifname)) &&
605                  (bssid = nl80211_getval(ifname, bssid, "bssid")))
606         {
607                 mac[0] = strtol(&bssid[0],  NULL, 16);
608                 mac[1] = strtol(&bssid[3],  NULL, 16);
609                 mac[2] = strtol(&bssid[6],  NULL, 16);
610                 mac[3] = strtol(&bssid[9],  NULL, 16);
611                 mac[4] = strtol(&bssid[12], NULL, 16);
612                 mac[5] = strtol(&bssid[15], NULL, 16);
613
614                 sprintf(buf, "%02X:%02X:%02X:%02X:%02X:%02X",
615                         mac[0], mac[1], mac[2], mac[3], mac[4], mac[5]);
616
617                 return 0;
618         }
619
620         return -1;
621 }
622
623 int nl80211_get_channel(const char *ifname, int *buf)
624 {
625         char *first;
626
627         if (!wext_get_channel(ifname, buf))
628                 return 0;
629
630         else if ((first = nl80211_phy2ifname(nl80211_ifname2phy(ifname))) != NULL)
631                 return wext_get_channel(first, buf);
632
633         return -1;
634 }
635
636 int nl80211_get_frequency(const char *ifname, int *buf)
637 {
638         char *first;
639
640         if (!wext_get_frequency(ifname, buf))
641                 return 0;
642
643         else if ((first = nl80211_phy2ifname(nl80211_ifname2phy(ifname))) != NULL)
644                 return wext_get_frequency(first, buf);
645
646         return -1;
647 }
648
649 int nl80211_get_txpower(const char *ifname, int *buf)
650 {
651         return wext_get_txpower(ifname, buf);
652 }
653
654
655 static int nl80211_fill_signal_cb(struct nl_msg *msg, void *arg)
656 {
657         int8_t dbm;
658         int16_t mbit;
659         struct nl80211_rssi_rate *rr = arg;
660         struct nlattr **attr = nl80211_parse(msg);
661         struct nlattr *sinfo[NL80211_STA_INFO_MAX + 1];
662         struct nlattr *rinfo[NL80211_RATE_INFO_MAX + 1];
663
664         static struct nla_policy stats_policy[NL80211_STA_INFO_MAX + 1] = {
665                 [NL80211_STA_INFO_INACTIVE_TIME] = { .type = NLA_U32    },
666                 [NL80211_STA_INFO_RX_BYTES]      = { .type = NLA_U32    },
667                 [NL80211_STA_INFO_TX_BYTES]      = { .type = NLA_U32    },
668                 [NL80211_STA_INFO_RX_PACKETS]    = { .type = NLA_U32    },
669                 [NL80211_STA_INFO_TX_PACKETS]    = { .type = NLA_U32    },
670                 [NL80211_STA_INFO_SIGNAL]        = { .type = NLA_U8     },
671                 [NL80211_STA_INFO_TX_BITRATE]    = { .type = NLA_NESTED },
672                 [NL80211_STA_INFO_LLID]          = { .type = NLA_U16    },
673                 [NL80211_STA_INFO_PLID]          = { .type = NLA_U16    },
674                 [NL80211_STA_INFO_PLINK_STATE]   = { .type = NLA_U8     },
675         };
676
677         static struct nla_policy rate_policy[NL80211_RATE_INFO_MAX + 1] = {
678                 [NL80211_RATE_INFO_BITRATE]      = { .type = NLA_U16  },
679                 [NL80211_RATE_INFO_MCS]          = { .type = NLA_U8   },
680                 [NL80211_RATE_INFO_40_MHZ_WIDTH] = { .type = NLA_FLAG },
681                 [NL80211_RATE_INFO_SHORT_GI]     = { .type = NLA_FLAG },
682         };
683
684         if (attr[NL80211_ATTR_STA_INFO])
685         {
686                 if( !nla_parse_nested(sinfo, NL80211_STA_INFO_MAX,
687                                 attr[NL80211_ATTR_STA_INFO], stats_policy) )
688                 {
689                         if (sinfo[NL80211_STA_INFO_SIGNAL])
690                         {
691                                 dbm = nla_get_u8(sinfo[NL80211_STA_INFO_SIGNAL]);
692                                 rr->rssi = rr->rssi ? (int8_t)((rr->rssi + dbm) / 2) : dbm;
693                         }
694
695                         if (sinfo[NL80211_STA_INFO_TX_BITRATE])
696                         {
697                                 if( !nla_parse_nested(rinfo, NL80211_RATE_INFO_MAX,
698                                                 sinfo[NL80211_STA_INFO_TX_BITRATE], rate_policy) )
699                                 {
700                                         if (rinfo[NL80211_RATE_INFO_BITRATE])
701                                         {
702                                                 mbit = nla_get_u16(rinfo[NL80211_RATE_INFO_BITRATE]);
703                                                 rr->rate = rr->rate
704                                                         ? (int16_t)((rr->rate + mbit) / 2) : mbit;
705                                         }
706                                 }
707                         }
708                 }
709         }
710
711         return NL_SKIP;
712 }
713
714 static void nl80211_fill_signal(const char *ifname, struct nl80211_rssi_rate *r)
715 {
716         DIR *d;
717         struct dirent *de;
718         struct nl80211_msg_conveyor *req;
719
720         r->rssi = 0;
721         r->rate = 0;
722
723         if ((d = opendir("/sys/class/net")) != NULL)
724         {
725                 while ((de = readdir(d)) != NULL)
726                 {
727                         if (!strncmp(de->d_name, ifname, strlen(ifname)) &&
728                                 (!de->d_name[strlen(ifname)] ||
729                                  !strncmp(&de->d_name[strlen(ifname)], ".sta", 4)))
730                         {
731                                 req = nl80211_msg(de->d_name, NL80211_CMD_GET_STATION,
732                                                                   NLM_F_DUMP);
733
734                                 if (req)
735                                 {
736                                         nl80211_send(req, nl80211_fill_signal_cb, r);
737                                         nl80211_free(req);
738                                 }
739                         }
740                 }
741
742                 closedir(d);
743         }
744 }
745
746 int nl80211_get_bitrate(const char *ifname, int *buf)
747 {
748         struct nl80211_rssi_rate rr;
749
750         if (!wext_get_bitrate(ifname, buf))
751                 return 0;
752
753         nl80211_fill_signal(ifname, &rr);
754
755         if (rr.rate)
756         {
757                 *buf = (rr.rate * 100);
758                 return 0;
759         }
760
761         return -1;
762 }
763
764 int nl80211_get_signal(const char *ifname, int *buf)
765 {
766         struct nl80211_rssi_rate rr;
767
768         if (!wext_get_signal(ifname, buf))
769                 return 0;
770
771         nl80211_fill_signal(ifname, &rr);
772
773         if (rr.rssi)
774         {
775                 *buf = rr.rssi;
776                 return 0;
777         }
778
779         return -1;
780 }
781
782 static int nl80211_get_noise_cb(struct nl_msg *msg, void *arg)
783 {
784         int8_t *noise = arg;
785         struct nlattr **tb = nl80211_parse(msg);
786         struct nlattr *si[NL80211_SURVEY_INFO_MAX + 1];
787
788         static struct nla_policy sp[NL80211_SURVEY_INFO_MAX + 1] = {
789                 [NL80211_SURVEY_INFO_FREQUENCY] = { .type = NLA_U32 },
790                 [NL80211_SURVEY_INFO_NOISE]     = { .type = NLA_U8  },
791         };
792
793         if (!tb[NL80211_ATTR_SURVEY_INFO])
794                 return NL_SKIP;
795
796         if (nla_parse_nested(si, NL80211_SURVEY_INFO_MAX,
797                                                  tb[NL80211_ATTR_SURVEY_INFO], sp))
798                 return NL_SKIP;
799
800         if (!si[NL80211_SURVEY_INFO_NOISE])
801                 return NL_SKIP;
802
803         if (!*noise || si[NL80211_SURVEY_INFO_IN_USE])
804                 *noise = (int8_t)nla_get_u8(si[NL80211_SURVEY_INFO_NOISE]);
805
806         return NL_SKIP;
807 }
808
809
810 int nl80211_get_noise(const char *ifname, int *buf)
811 {
812         int8_t noise;
813         struct nl80211_msg_conveyor *req;
814
815         req = nl80211_msg(ifname, NL80211_CMD_GET_SURVEY, NLM_F_DUMP);
816         if (req)
817         {
818                 noise = 0;
819
820                 nl80211_send(req, nl80211_get_noise_cb, &noise);
821                 nl80211_free(req);
822
823                 if (noise)
824                 {
825                         *buf = noise;
826                         return 0;
827                 }
828         }
829
830         return -1;
831 }
832
833 int nl80211_get_quality(const char *ifname, int *buf)
834 {
835         int signal;
836
837         if (wext_get_quality(ifname, buf))
838         {
839                 *buf = 0;
840
841                 if (!nl80211_get_signal(ifname, &signal))
842                 {
843                         /* A positive signal level is usually just a quality
844                          * value, pass through as-is */
845                         if (signal >= 0)
846                         {
847                                 *buf = signal;
848                         }
849
850                         /* The cfg80211 wext compat layer assumes a signal range
851                          * of -110 dBm to -40 dBm, the quality value is derived
852                          * by adding 110 to the signal level */
853                         else
854                         {
855                                 if (signal < -110)
856                                         signal = -110;
857                                 else if (signal > -40)
858                                         signal = -40;
859
860                                 *buf = (signal + 110);
861                         }
862                 }
863         }
864
865         return 0;
866 }
867
868 int nl80211_get_quality_max(const char *ifname, int *buf)
869 {
870         if (wext_get_quality_max(ifname, buf))
871                 /* The cfg80211 wext compat layer assumes a maximum
872                  * quality of 70 */
873                 *buf = 70;
874
875         return 0;
876 }
877
878 int nl80211_get_encryption(const char *ifname, char *buf)
879 {
880         int i;
881         char k[9];
882         char *val, *res;
883         struct iwinfo_crypto_entry *c = (struct iwinfo_crypto_entry *)buf;
884
885         /* WPA supplicant */
886         if( (res = nl80211_wpactl_info(ifname, "STATUS", NULL)) &&
887             (val = nl80211_getval(NULL, res, "pairwise_cipher")) )
888         {
889                 /* WEP */
890                 if (strstr(val, "WEP"))
891                 {
892                         if (strstr(val, "WEP-40"))
893                                 c->pair_ciphers |= IWINFO_CIPHER_WEP40;
894
895                         else if (strstr(val, "WEP-104"))
896                                 c->pair_ciphers |= IWINFO_CIPHER_WEP104;
897
898                         c->enabled       = 1;
899                         c->group_ciphers = c->pair_ciphers;
900
901                         c->auth_suites |= IWINFO_KMGMT_NONE;
902                         c->auth_algs   |= IWINFO_AUTH_OPEN; /* XXX: assumption */
903                 }
904
905                 /* WPA */
906                 else
907                 {
908                         if (strstr(val, "TKIP"))
909                                 c->pair_ciphers |= IWINFO_CIPHER_TKIP;
910
911                         else if (strstr(val, "CCMP"))
912                                 c->pair_ciphers |= IWINFO_CIPHER_CCMP;
913
914                         else if (strstr(val, "NONE"))
915                                 c->pair_ciphers |= IWINFO_CIPHER_NONE;
916
917                         else if (strstr(val, "WEP-40"))
918                                 c->pair_ciphers |= IWINFO_CIPHER_WEP40;
919
920                         else if (strstr(val, "WEP-104"))
921                                 c->pair_ciphers |= IWINFO_CIPHER_WEP104;
922
923
924                         if ((val = nl80211_getval(NULL, res, "group_cipher")))
925                         {
926                                 if (strstr(val, "TKIP"))
927                                         c->group_ciphers |= IWINFO_CIPHER_TKIP;
928
929                                 else if (strstr(val, "CCMP"))
930                                         c->group_ciphers |= IWINFO_CIPHER_CCMP;
931
932                                 else if (strstr(val, "NONE"))
933                                         c->group_ciphers |= IWINFO_CIPHER_NONE;
934
935                                 else if (strstr(val, "WEP-40"))
936                                         c->group_ciphers |= IWINFO_CIPHER_WEP40;
937
938                                 else if (strstr(val, "WEP-104"))
939                                         c->group_ciphers |= IWINFO_CIPHER_WEP104;
940                         }
941
942
943                         if ((val = nl80211_getval(NULL, res, "key_mgmt")))
944                         {
945                                 if (strstr(val, "WPA2"))
946                                         c->wpa_version = 2;
947
948                                 else if (strstr(val, "WPA"))
949                                         c->wpa_version = 1;
950
951
952                                 if (strstr(val, "PSK"))
953                                         c->auth_suites |= IWINFO_KMGMT_PSK;
954
955                                 else if (strstr(val, "EAP") || strstr(val, "802.1X"))
956                                         c->auth_suites |= IWINFO_KMGMT_8021x;
957
958                                 else if (strstr(val, "NONE"))
959                                         c->auth_suites |= IWINFO_KMGMT_NONE;
960                         }
961
962                         c->enabled = (c->wpa_version && c->auth_suites) ? 1 : 0;
963                 }
964
965                 return 0;
966         }
967
968         /* Hostapd */
969         else if ((res = nl80211_hostapd_info(ifname)))
970         {
971                 if ((val = nl80211_getval(ifname, res, "wpa")) != NULL)
972                         c->wpa_version = atoi(val);
973
974                 val = nl80211_getval(ifname, res, "wpa_key_mgmt");
975
976                 if (!val || strstr(val, "PSK"))
977                         c->auth_suites |= IWINFO_KMGMT_PSK;
978
979                 if (val && strstr(val, "EAP"))
980                         c->auth_suites |= IWINFO_KMGMT_8021x;
981
982                 if (val && strstr(val, "NONE"))
983                         c->auth_suites |= IWINFO_KMGMT_NONE;
984
985                 if ((val = nl80211_getval(ifname, res, "wpa_pairwise")) != NULL)
986                 {
987                         if (strstr(val, "TKIP"))
988                                 c->pair_ciphers |= IWINFO_CIPHER_TKIP;
989
990                         if (strstr(val, "CCMP"))
991                                 c->pair_ciphers |= IWINFO_CIPHER_CCMP;
992
993                         if (strstr(val, "NONE"))
994                                 c->pair_ciphers |= IWINFO_CIPHER_NONE;
995                 }
996
997                 if ((val = nl80211_getval(ifname, res, "auth_algs")) != NULL)
998                 {
999                         switch(atoi(val)) {
1000                                 case 1:
1001                                         c->auth_algs |= IWINFO_AUTH_OPEN;
1002                                         break;
1003
1004                                 case 2:
1005                                         c->auth_algs |= IWINFO_AUTH_SHARED;
1006                                         break;
1007
1008                                 case 3:
1009                                         c->auth_algs |= IWINFO_AUTH_OPEN;
1010                                         c->auth_algs |= IWINFO_AUTH_SHARED;
1011                                         break;
1012
1013                                 default:
1014                                         break;
1015                         }
1016
1017                         for( i = 0; i < 4; i++ )
1018                         {
1019                                 snprintf(k, sizeof(k), "wep_key%d", i);
1020
1021                                 if ((val = nl80211_getval(ifname, res, k)))
1022                                 {
1023                                         if ((strlen(val) == 5) || (strlen(val) == 10))
1024                                                 c->pair_ciphers |= IWINFO_CIPHER_WEP40;
1025
1026                                         else if ((strlen(val) == 13) || (strlen(val) == 26))
1027                                                 c->pair_ciphers |= IWINFO_CIPHER_WEP104;
1028                                 }
1029                         }
1030                 }
1031
1032                 c->group_ciphers = c->pair_ciphers;
1033                 c->enabled = (c->wpa_version || c->pair_ciphers) ? 1 : 0;
1034
1035                 return 0;
1036         }
1037
1038         return -1;
1039 }
1040
1041
1042 static int nl80211_get_assoclist_cb(struct nl_msg *msg, void *arg)
1043 {
1044         struct nl80211_array_buf *arr = arg;
1045         struct iwinfo_assoclist_entry *e = arr->buf;
1046         struct nlattr **attr = nl80211_parse(msg);
1047         struct nlattr *sinfo[NL80211_STA_INFO_MAX + 1];
1048
1049         static struct nla_policy stats_policy[NL80211_STA_INFO_MAX + 1] = {
1050                 [NL80211_STA_INFO_INACTIVE_TIME] = { .type = NLA_U32    },
1051                 [NL80211_STA_INFO_RX_BYTES]      = { .type = NLA_U32    },
1052                 [NL80211_STA_INFO_TX_BYTES]      = { .type = NLA_U32    },
1053                 [NL80211_STA_INFO_RX_PACKETS]    = { .type = NLA_U32    },
1054                 [NL80211_STA_INFO_TX_PACKETS]    = { .type = NLA_U32    },
1055                 [NL80211_STA_INFO_SIGNAL]        = { .type = NLA_U8     },
1056                 [NL80211_STA_INFO_TX_BITRATE]    = { .type = NLA_NESTED },
1057                 [NL80211_STA_INFO_LLID]          = { .type = NLA_U16    },
1058                 [NL80211_STA_INFO_PLID]          = { .type = NLA_U16    },
1059                 [NL80211_STA_INFO_PLINK_STATE]   = { .type = NLA_U8     },
1060         };
1061
1062         /* advance to end of array */
1063         e += arr->count;
1064
1065         if (attr[NL80211_ATTR_MAC])
1066                 memcpy(e->mac, nla_data(attr[NL80211_ATTR_MAC]), 6);
1067
1068         if (attr[NL80211_ATTR_STA_INFO])
1069         {
1070                 if (!nla_parse_nested(sinfo, NL80211_STA_INFO_MAX,
1071                                 attr[NL80211_ATTR_STA_INFO], stats_policy))
1072                 {
1073                         if (sinfo[NL80211_STA_INFO_SIGNAL])
1074                                 e->signal = nla_get_u8(sinfo[NL80211_STA_INFO_SIGNAL]);
1075                 }
1076         }
1077
1078         e->noise = 0; /* filled in by caller */
1079         arr->count++;
1080
1081         return NL_SKIP;
1082 }
1083
1084 int nl80211_get_assoclist(const char *ifname, char *buf, int *len)
1085 {
1086         DIR *d;
1087         int i, noise = 0;
1088         struct dirent *de;
1089         struct nl80211_msg_conveyor *req;
1090         struct nl80211_array_buf arr = { .buf = buf, .count = 0 };
1091         struct iwinfo_assoclist_entry *e;
1092
1093         if ((d = opendir("/sys/class/net")) != NULL)
1094         {
1095                 while ((de = readdir(d)) != NULL)
1096                 {
1097                         if (!strncmp(de->d_name, ifname, strlen(ifname)) &&
1098                                 (!de->d_name[strlen(ifname)] ||
1099                                  !strncmp(&de->d_name[strlen(ifname)], ".sta", 4)))
1100                         {
1101                                 req = nl80211_msg(de->d_name, NL80211_CMD_GET_STATION,
1102                                                                   NLM_F_DUMP);
1103
1104                                 if (req)
1105                                 {
1106                                         nl80211_send(req, nl80211_get_assoclist_cb, &arr);
1107                                         nl80211_free(req);
1108                                 }
1109
1110                                 break;
1111                         }
1112                 }
1113
1114                 closedir(d);
1115
1116                 if (!nl80211_get_noise(ifname, &noise))
1117                         for (i = 0, e = arr.buf; i < arr.count; i++, e++)
1118                                 e->noise = noise;
1119
1120                 *len = (arr.count * sizeof(struct iwinfo_assoclist_entry));
1121                 return 0;
1122         }
1123
1124         return -1;
1125 }
1126
1127 static int nl80211_get_txpwrlist_cb(struct nl_msg *msg, void *arg)
1128 {
1129         int *dbm_max = arg;
1130         int ch_cur, ch_cmp, bands_remain, freqs_remain;
1131
1132         struct nlattr **attr = nl80211_parse(msg);
1133         struct nlattr *bands[NL80211_BAND_ATTR_MAX + 1];
1134         struct nlattr *freqs[NL80211_FREQUENCY_ATTR_MAX + 1];
1135         struct nlattr *band, *freq;
1136
1137         static struct nla_policy freq_policy[NL80211_FREQUENCY_ATTR_MAX + 1] = {
1138                 [NL80211_FREQUENCY_ATTR_FREQ]         = { .type = NLA_U32  },
1139                 [NL80211_FREQUENCY_ATTR_DISABLED]     = { .type = NLA_FLAG },
1140                 [NL80211_FREQUENCY_ATTR_PASSIVE_SCAN] = { .type = NLA_FLAG },
1141                 [NL80211_FREQUENCY_ATTR_NO_IBSS]      = { .type = NLA_FLAG },
1142                 [NL80211_FREQUENCY_ATTR_RADAR]        = { .type = NLA_FLAG },
1143                 [NL80211_FREQUENCY_ATTR_MAX_TX_POWER] = { .type = NLA_U32  },
1144         };
1145
1146         ch_cur = *dbm_max; /* value int* is initialized with channel by caller */
1147         *dbm_max = -1;
1148
1149         nla_for_each_nested(band, attr[NL80211_ATTR_WIPHY_BANDS], bands_remain)
1150         {
1151                 nla_parse(bands, NL80211_BAND_ATTR_MAX, nla_data(band),
1152                           nla_len(band), NULL);
1153
1154                 nla_for_each_nested(freq,
1155                         bands[NL80211_BAND_ATTR_FREQS], freqs_remain)
1156                 {
1157                         nla_parse(freqs, NL80211_FREQUENCY_ATTR_MAX,
1158                                 nla_data(freq), nla_len(freq), freq_policy);
1159
1160                         ch_cmp = nl80211_freq2channel(
1161                                 nla_get_u32(freqs[NL80211_FREQUENCY_ATTR_FREQ]));
1162
1163                         if( (!ch_cur || (ch_cmp == ch_cur)) &&
1164                                 freqs[NL80211_FREQUENCY_ATTR_MAX_TX_POWER] )
1165                         {
1166                                 *dbm_max = (int)(0.01 * nla_get_u32(
1167                                         freqs[NL80211_FREQUENCY_ATTR_MAX_TX_POWER]));
1168
1169                                 break;
1170                         }
1171                 }
1172         }
1173
1174         return NL_SKIP;
1175 }
1176
1177 int nl80211_get_txpwrlist(const char *ifname, char *buf, int *len)
1178 {
1179         int ch_cur;
1180         int dbm_max = -1, dbm_cur, dbm_cnt;
1181         struct nl80211_msg_conveyor *req;
1182         struct iwinfo_txpwrlist_entry entry;
1183
1184         if (nl80211_get_channel(ifname, &ch_cur))
1185                 ch_cur = 0;
1186
1187         req = nl80211_msg(ifname, NL80211_CMD_GET_WIPHY, 0);
1188         if (req)
1189         {
1190                 /* initialize the value pointer with channel for callback */
1191                 dbm_max = ch_cur;
1192
1193                 nl80211_send(req, nl80211_get_txpwrlist_cb, &dbm_max);
1194                 nl80211_free(req);
1195         }
1196
1197         if (dbm_max > -1)
1198         {
1199                 for (dbm_cur = 0, dbm_cnt = 0;
1200                      dbm_cur < dbm_max;
1201                      dbm_cur++, dbm_cnt++)
1202                 {
1203                         entry.dbm = dbm_cur;
1204                         entry.mw  = iwinfo_dbm2mw(dbm_cur);
1205
1206                         memcpy(&buf[dbm_cnt * sizeof(entry)], &entry, sizeof(entry));
1207                 }
1208
1209                 entry.dbm = dbm_max;
1210                 entry.mw  = iwinfo_dbm2mw(dbm_max);
1211
1212                 memcpy(&buf[dbm_cnt * sizeof(entry)], &entry, sizeof(entry));
1213                 dbm_cnt++;
1214
1215                 *len = dbm_cnt * sizeof(entry);
1216                 return 0;
1217         }
1218
1219         return -1;
1220 }
1221
1222 static void nl80211_get_scancrypto(const char *spec,
1223         struct iwinfo_crypto_entry *c)
1224 {
1225         if (strstr(spec, "WPA") || strstr(spec, "WEP"))
1226         {
1227                 c->enabled = 1;
1228
1229                 if (strstr(spec, "WPA2-") && strstr(spec, "WPA-"))
1230                         c->wpa_version = 3;
1231
1232                 else if (strstr(spec, "WPA2"))
1233                         c->wpa_version = 2;
1234
1235                 else if (strstr(spec, "WPA"))
1236                         c->wpa_version = 1;
1237
1238                 else if (strstr(spec, "WEP"))
1239                         c->auth_algs = IWINFO_AUTH_OPEN | IWINFO_AUTH_SHARED;
1240
1241
1242                 if (strstr(spec, "PSK"))
1243                         c->auth_suites |= IWINFO_KMGMT_PSK;
1244
1245                 if (strstr(spec, "802.1X") || strstr(spec, "EAP"))
1246                         c->auth_suites |= IWINFO_KMGMT_8021x;
1247
1248                 if (strstr(spec, "WPA-NONE"))
1249                         c->auth_suites |= IWINFO_KMGMT_NONE;
1250
1251
1252                 if (strstr(spec, "TKIP"))
1253                         c->pair_ciphers |= IWINFO_CIPHER_TKIP;
1254
1255                 if (strstr(spec, "CCMP"))
1256                         c->pair_ciphers |= IWINFO_CIPHER_CCMP;
1257
1258                 if (strstr(spec, "WEP-40"))
1259                         c->pair_ciphers |= IWINFO_CIPHER_WEP40;
1260
1261                 if (strstr(spec, "WEP-104"))
1262                         c->pair_ciphers |= IWINFO_CIPHER_WEP104;
1263
1264                 c->group_ciphers = c->pair_ciphers;
1265         }
1266         else
1267         {
1268                 c->enabled = 0;
1269         }
1270 }
1271
1272 int nl80211_get_scanlist(const char *ifname, char *buf, int *len)
1273 {
1274         int freq, rssi, qmax, count;
1275         char *res;
1276         char ssid[128] = { 0 };
1277         char bssid[18] = { 0 };
1278         char cipher[256] = { 0 };
1279
1280         /* Got a radioX pseudo interface, find some interface on it or create one */
1281         if (!strncmp(ifname, "radio", 5))
1282         {
1283                 /* Reuse existing interface */
1284                 if ((res = nl80211_phy2ifname(ifname)) != NULL)
1285                 {
1286                         return nl80211_get_scanlist(res, buf, len);
1287                 }
1288
1289                 /* Need to spawn a temporary iface for scanning */
1290                 else if ((res = nl80211_ifadd(ifname)) != NULL)
1291                 {
1292                         count = nl80211_get_scanlist(res, buf, len);
1293                         nl80211_ifdel(res);
1294                         return count;
1295                 }
1296         }
1297
1298         struct iwinfo_scanlist_entry *e = (struct iwinfo_scanlist_entry *)buf;
1299
1300         /* WPA supplicant */
1301         if ((res = nl80211_wpactl_info(ifname, "SCAN", "CTRL-EVENT-SCAN-RESULTS")))
1302         {
1303                 if ((res = nl80211_wpactl_info(ifname, "SCAN_RESULTS", NULL)))
1304                 {
1305                         nl80211_get_quality_max(ifname, &qmax);
1306
1307                         /* skip header line */
1308                         while( *res++ != '\n' );
1309
1310                         count = 0;
1311
1312                         while( sscanf(res, "%17s %d %d %255s%*[ \t]%127[^\n]\n",
1313                                       bssid, &freq, &rssi, cipher, ssid) > 0 )
1314                         {
1315                                 /* BSSID */
1316                                 e->mac[0] = strtol(&bssid[0],  NULL, 16);
1317                                 e->mac[1] = strtol(&bssid[3],  NULL, 16);
1318                                 e->mac[2] = strtol(&bssid[6],  NULL, 16);
1319                                 e->mac[3] = strtol(&bssid[9],  NULL, 16);
1320                                 e->mac[4] = strtol(&bssid[12], NULL, 16);
1321                                 e->mac[5] = strtol(&bssid[15], NULL, 16);
1322
1323                                 /* SSID */
1324                                 memcpy(e->ssid, ssid,
1325                                         min(strlen(ssid), sizeof(e->ssid) - 1));
1326
1327                                 /* Mode (assume master) */
1328                                 sprintf((char *)e->mode, "Master");
1329
1330                                 /* Channel */
1331                                 e->channel = nl80211_freq2channel(freq);
1332
1333                                 /* Signal */
1334                                 e->signal = rssi;
1335
1336                                 /* Quality */
1337                                 if (rssi < 0)
1338                                 {
1339                                         /* The cfg80211 wext compat layer assumes a signal range
1340                                          * of -110 dBm to -40 dBm, the quality value is derived
1341                                          * by adding 110 to the signal level */
1342                                         if (rssi < -110)
1343                                                 rssi = -110;
1344                                         else if (rssi > -40)
1345                                                 rssi = -40;
1346
1347                                         e->quality = (rssi + 110);
1348                                 }
1349                                 else
1350                                 {
1351                                         e->quality = rssi;
1352                                 }
1353
1354                                 /* Max. Quality */
1355                                 e->quality_max = qmax;
1356
1357                                 /* Crypto */
1358                                 nl80211_get_scancrypto(cipher, &e->crypto);
1359
1360                                 /* advance to next line */
1361                                 while( *res && *res++ != '\n' );
1362
1363                                 count++;
1364                                 e++;
1365
1366                                 memset(ssid, 0, sizeof(ssid));
1367                                 memset(bssid, 0, sizeof(bssid));
1368                                 memset(cipher, 0, sizeof(cipher));
1369                         }
1370
1371                         *len = count * sizeof(struct iwinfo_scanlist_entry);
1372                         return 0;
1373                 }
1374         }
1375
1376         /* AP scan */
1377         else
1378         {
1379                 /* Got a temp interface, don't create yet another one */
1380                 if (!strncmp(ifname, "tmp.", 4))
1381                 {
1382                         if (!iwinfo_ifup(ifname))
1383                                 return -1;
1384
1385                         wext_get_scanlist(ifname, buf, len);
1386                         iwinfo_ifdown(ifname);
1387                         return 0;
1388                 }
1389
1390                 /* Spawn a new scan interface */
1391                 else
1392                 {
1393                         if (!(res = nl80211_ifadd(ifname)))
1394                                 goto out;
1395
1396                         if (!iwinfo_ifmac(res))
1397                                 goto out;
1398
1399                         /* if we can take the new interface up, the driver supports an
1400                          * additional interface and there's no need to tear down the ap */
1401                         if (iwinfo_ifup(res))
1402                         {
1403                                 wext_get_scanlist(res, buf, len);
1404                                 iwinfo_ifdown(res);
1405                         }
1406
1407                         /* driver cannot create secondary interface, take down ap
1408                          * during scan */
1409                         else if (iwinfo_ifdown(ifname) && iwinfo_ifup(res))
1410                         {
1411                                 wext_get_scanlist(res, buf, len);
1412                                 iwinfo_ifdown(res);
1413                                 iwinfo_ifup(ifname);
1414                                 nl80211_hostapd_hup(ifname);
1415                         }
1416
1417                 out:
1418                         nl80211_ifdel(res);
1419                         return 0;
1420                 }
1421         }
1422
1423         return -1;
1424 }
1425
1426 static int nl80211_get_freqlist_cb(struct nl_msg *msg, void *arg)
1427 {
1428         int bands_remain, freqs_remain;
1429
1430         struct nl80211_array_buf *arr = arg;
1431         struct iwinfo_freqlist_entry *e = arr->buf;
1432
1433         struct nlattr **attr = nl80211_parse(msg);
1434         struct nlattr *bands[NL80211_BAND_ATTR_MAX + 1];
1435         struct nlattr *freqs[NL80211_FREQUENCY_ATTR_MAX + 1];
1436         struct nlattr *band, *freq;
1437
1438         static struct nla_policy freq_policy[NL80211_FREQUENCY_ATTR_MAX + 1] = {
1439                 [NL80211_FREQUENCY_ATTR_FREQ]         = { .type = NLA_U32  },
1440                 [NL80211_FREQUENCY_ATTR_DISABLED]     = { .type = NLA_FLAG },
1441                 [NL80211_FREQUENCY_ATTR_PASSIVE_SCAN] = { .type = NLA_FLAG },
1442                 [NL80211_FREQUENCY_ATTR_NO_IBSS]      = { .type = NLA_FLAG },
1443                 [NL80211_FREQUENCY_ATTR_RADAR]        = { .type = NLA_FLAG },
1444                 [NL80211_FREQUENCY_ATTR_MAX_TX_POWER] = { .type = NLA_U32  },
1445         };
1446
1447         nla_for_each_nested(band, attr[NL80211_ATTR_WIPHY_BANDS], bands_remain)
1448         {
1449                 nla_parse(bands, NL80211_BAND_ATTR_MAX, nla_data(band),
1450                         nla_len(band), NULL);
1451
1452                 nla_for_each_nested(freq,
1453                         bands[NL80211_BAND_ATTR_FREQS], freqs_remain)
1454                 {
1455                         nla_parse(freqs, NL80211_FREQUENCY_ATTR_MAX,
1456                                 nla_data(freq), nla_len(freq), NULL);
1457
1458                         if( !freqs[NL80211_FREQUENCY_ATTR_FREQ] ||
1459                                 freqs[NL80211_FREQUENCY_ATTR_DISABLED] )
1460                                 continue;
1461
1462                         e->mhz = nla_get_u32(freqs[NL80211_FREQUENCY_ATTR_FREQ]);
1463                         e->channel = nl80211_freq2channel(e->mhz);
1464
1465                         e->restricted = (
1466                                 freqs[NL80211_FREQUENCY_ATTR_PASSIVE_SCAN] ||
1467                                 freqs[NL80211_FREQUENCY_ATTR_NO_IBSS]      ||
1468                                 freqs[NL80211_FREQUENCY_ATTR_RADAR]
1469                         ) ? 1 : 0;
1470
1471                         e++;
1472                         arr->count++;
1473                 }
1474         }
1475
1476         return NL_SKIP;
1477 }
1478
1479 int nl80211_get_freqlist(const char *ifname, char *buf, int *len)
1480 {
1481         struct nl80211_msg_conveyor *req;
1482         struct nl80211_array_buf arr = { .buf = buf, .count = 0 };
1483
1484         req = nl80211_msg(ifname, NL80211_CMD_GET_WIPHY, 0);
1485         if (req)
1486         {
1487                 nl80211_send(req, nl80211_get_freqlist_cb, &arr);
1488                 nl80211_free(req);
1489         }
1490
1491         if (arr.count > 0)
1492         {
1493                 *len = arr.count * sizeof(struct iwinfo_freqlist_entry);
1494                 return 0;
1495         }
1496
1497         return -1;
1498 }
1499
1500 static int nl80211_get_country_cb(struct nl_msg *msg, void *arg)
1501 {
1502         char *buf = arg;
1503         struct nlattr **attr = nl80211_parse(msg);
1504
1505         if (attr[NL80211_ATTR_REG_ALPHA2])
1506                 memcpy(buf, nla_data(attr[NL80211_ATTR_REG_ALPHA2]), 2);
1507         else
1508                 buf[0] = 0;
1509
1510         return NL_SKIP;
1511 }
1512
1513 int nl80211_get_country(const char *ifname, char *buf)
1514 {
1515         int rv = -1;
1516         struct nl80211_msg_conveyor *req;
1517
1518         req = nl80211_msg(ifname, NL80211_CMD_GET_REG, 0);
1519         if (req)
1520         {
1521                 nl80211_send(req, nl80211_get_country_cb, buf);
1522                 nl80211_free(req);
1523
1524                 if (buf[0])
1525                         rv = 0;
1526         }
1527
1528         return rv;
1529 }
1530
1531 int nl80211_get_countrylist(const char *ifname, char *buf, int *len)
1532 {
1533         int i, count;
1534         struct iwinfo_iso3166_label *l;
1535         struct iwinfo_country_entry *e = (struct iwinfo_country_entry *)buf;
1536
1537         for( l = ISO3166_Names, count = 0; l->iso3166; l++, e++, count++ )
1538         {
1539                 e->iso3166 = l->iso3166;
1540                 e->ccode[0] = (l->iso3166 / 256);
1541                 e->ccode[1] = (l->iso3166 % 256);
1542         }
1543
1544         *len = (count * sizeof(struct iwinfo_country_entry));
1545         return 0;
1546 }
1547
1548 static int nl80211_get_hwmodelist_cb(struct nl_msg *msg, void *arg)
1549 {
1550         int *modes = arg;
1551         int bands_remain, freqs_remain;
1552         uint16_t caps = 0;
1553         struct nlattr **attr = nl80211_parse(msg);
1554         struct nlattr *bands[NL80211_BAND_ATTR_MAX + 1];
1555         struct nlattr *freqs[NL80211_FREQUENCY_ATTR_MAX + 1];
1556         struct nlattr *band, *freq;
1557
1558         *modes = 0;
1559
1560         if (attr[NL80211_ATTR_WIPHY_BANDS])
1561         {
1562                 nla_for_each_nested(band, attr[NL80211_ATTR_WIPHY_BANDS], bands_remain)
1563                 {
1564                         nla_parse(bands, NL80211_BAND_ATTR_MAX, nla_data(band),
1565                                 nla_len(band), NULL);
1566
1567                         if (bands[NL80211_BAND_ATTR_HT_CAPA])
1568                                 caps = nla_get_u16(bands[NL80211_BAND_ATTR_HT_CAPA]);
1569
1570                         /* Treat any nonzero capability as 11n */
1571                         if (caps > 0)
1572                                 *modes |= IWINFO_80211_N;
1573
1574                         nla_for_each_nested(freq,
1575                                 bands[NL80211_BAND_ATTR_FREQS], freqs_remain)
1576                         {
1577                                 nla_parse(freqs, NL80211_FREQUENCY_ATTR_MAX,
1578                                         nla_data(freq), nla_len(freq), NULL);
1579
1580                                 if (!freqs[NL80211_FREQUENCY_ATTR_FREQ])
1581                                         continue;
1582
1583                                 if (nla_get_u32(freqs[NL80211_FREQUENCY_ATTR_FREQ]) < 2485)
1584                                 {
1585                                         *modes |= IWINFO_80211_B;
1586                                         *modes |= IWINFO_80211_G;
1587                                 }
1588                                 else
1589                                 {
1590                                         *modes |= IWINFO_80211_A;
1591                                 }
1592                         }
1593                 }
1594         }
1595
1596         return NL_SKIP;
1597 }
1598
1599 int nl80211_get_hwmodelist(const char *ifname, int *buf)
1600 {
1601         struct nl80211_msg_conveyor *req;
1602
1603         req = nl80211_msg(ifname, NL80211_CMD_GET_WIPHY, 0);
1604         if (req)
1605         {
1606                 nl80211_send(req, nl80211_get_hwmodelist_cb, buf);
1607                 nl80211_free(req);
1608         }
1609
1610         return *buf ? 0 : -1;
1611 }
1612
1613 int nl80211_get_mbssid_support(const char *ifname, int *buf)
1614 {
1615         /* Test whether we can create another interface */
1616         char *nif = nl80211_ifadd(ifname);
1617
1618         if (nif)
1619         {
1620                 *buf = (iwinfo_ifmac(nif) && iwinfo_ifup(nif));
1621
1622                 iwinfo_ifdown(nif);
1623                 nl80211_ifdel(nif);
1624
1625                 return 0;
1626         }
1627
1628         return -1;
1629 }