libiwinfo: fix detection of wep encryption and open networks in nl80211 backend
[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         if (phyidx > -1)
456         {
457                 if ((d = opendir("/sys/class/net")) != NULL)
458                 {
459                         while( (e = readdir(d)) != NULL )
460                         {
461                                 snprintf(buffer, sizeof(buffer),
462                                         "/sys/class/net/%s/phy80211/index", e->d_name);
463
464                                 if (nl80211_readint(buffer) == phyidx)
465                                 {
466                                         snprintf(buffer, sizeof(buffer),
467                                                 "/sys/class/net/%s/ifindex", e->d_name);
468
469                                         if( (cifidx = nl80211_readint(buffer)) >= 0 &&
470                                             ((ifidx < 0) || (cifidx < ifidx)) )
471                                         {
472                                                 ifidx = cifidx;
473                                                 strncpy(nif, e->d_name, sizeof(nif));
474                                         }
475                                 }
476                         }
477
478                         closedir(d);
479                 }
480         }
481
482         return nif[0] ? nif : NULL;
483 }
484
485 static char * nl80211_ifadd(const char *ifname)
486 {
487         int phyidx;
488         char *rv = NULL;
489         static char nif[IFNAMSIZ] = { 0 };
490         struct nl80211_msg_conveyor *req, *res;
491
492         req = nl80211_msg(ifname, NL80211_CMD_NEW_INTERFACE, 0);
493         if (req)
494         {
495                 snprintf(nif, sizeof(nif), "tmp.%s", ifname);
496
497                 NLA_PUT_STRING(req->msg, NL80211_ATTR_IFNAME, nif);
498                 NLA_PUT_U32(req->msg, NL80211_ATTR_IFTYPE, NL80211_IFTYPE_STATION);
499
500                 nl80211_send(req, NULL, NULL);
501
502                 rv = nif;
503
504         nla_put_failure:
505                 nl80211_free(req);
506         }
507
508         return rv;
509 }
510
511 static void nl80211_ifdel(const char *ifname)
512 {
513         struct nl80211_msg_conveyor *req;
514
515         req = nl80211_msg(ifname, NL80211_CMD_DEL_INTERFACE, 0);
516         if (req)
517         {
518                 NLA_PUT_STRING(req->msg, NL80211_ATTR_IFNAME, ifname);
519
520                 nl80211_send(req, NULL, NULL);
521
522         nla_put_failure:
523                 nl80211_free(req);
524         }
525 }
526
527 static void nl80211_hostapd_hup(const char *ifname)
528 {
529         int fd, pid = 0;
530         char buf[32];
531         char *phy = nl80211_ifname2phy(ifname);
532
533         if (phy)
534         {
535                 snprintf(buf, sizeof(buf), "/var/run/wifi-%s.pid", phy);
536                 if ((fd = open(buf, O_RDONLY)) > 0)
537                 {
538                         if (read(fd, buf, sizeof(buf)) > 0)
539                                 pid = atoi(buf);
540
541                         close(fd);
542                 }
543
544                 if (pid > 0)
545                         kill(pid, 1);
546         }
547 }
548
549
550 int nl80211_probe(const char *ifname)
551 {
552         return !!nl80211_ifname2phy(ifname);
553 }
554
555 void nl80211_close(void)
556 {
557         if (nls)
558         {
559                 if (nls->nl_sock)
560                         nl_socket_free(nls->nl_sock);
561
562                 if (nls->nl_cache)
563                         nl_cache_free(nls->nl_cache);
564
565                 free(nls);
566                 nls = NULL;
567         }
568 }
569
570 int nl80211_get_mode(const char *ifname, char *buf)
571 {
572         return wext_get_mode(ifname, buf);
573 }
574
575 int nl80211_get_ssid(const char *ifname, char *buf)
576 {
577         char *ssid;
578
579         if (!wext_get_ssid(ifname, buf))
580         {
581                 return 0;
582         }
583         else if( (ssid = nl80211_hostapd_info(ifname)) &&
584                  (ssid = nl80211_getval(ifname, ssid, "ssid")) )
585         {
586                 memcpy(buf, ssid, strlen(ssid));
587                 return 0;
588         }
589
590         return -1;
591 }
592
593 int nl80211_get_bssid(const char *ifname, char *buf)
594 {
595         char *bssid;
596         unsigned char mac[6];
597
598         if (!wext_get_bssid(ifname, buf))
599         {
600                 return 0;
601         }
602         else if((bssid = nl80211_hostapd_info(ifname)) &&
603                  (bssid = nl80211_getval(ifname, bssid, "bssid")))
604         {
605                 mac[0] = strtol(&bssid[0],  NULL, 16);
606                 mac[1] = strtol(&bssid[3],  NULL, 16);
607                 mac[2] = strtol(&bssid[6],  NULL, 16);
608                 mac[3] = strtol(&bssid[9],  NULL, 16);
609                 mac[4] = strtol(&bssid[12], NULL, 16);
610                 mac[5] = strtol(&bssid[15], NULL, 16);
611
612                 sprintf(buf, "%02X:%02X:%02X:%02X:%02X:%02X",
613                         mac[0], mac[1], mac[2], mac[3], mac[4], mac[5]);
614
615                 return 0;
616         }
617
618         return -1;
619 }
620
621 int nl80211_get_channel(const char *ifname, int *buf)
622 {
623         char *first;
624
625         if (!wext_get_channel(ifname, buf))
626                 return 0;
627
628         else if ((first = nl80211_phy2ifname(nl80211_ifname2phy(ifname))) != NULL)
629                 return wext_get_channel(first, buf);
630
631         return -1;
632 }
633
634 int nl80211_get_frequency(const char *ifname, int *buf)
635 {
636         char *first;
637
638         if (!wext_get_frequency(ifname, buf))
639                 return 0;
640
641         else if ((first = nl80211_phy2ifname(nl80211_ifname2phy(ifname))) != NULL)
642                 return wext_get_frequency(first, buf);
643
644         return -1;
645 }
646
647 int nl80211_get_txpower(const char *ifname, int *buf)
648 {
649         return wext_get_txpower(ifname, buf);
650 }
651
652
653 static int nl80211_fill_signal_cb(struct nl_msg *msg, void *arg)
654 {
655         int8_t dbm;
656         int16_t mbit;
657         struct nl80211_rssi_rate *rr = arg;
658         struct nlattr **attr = nl80211_parse(msg);
659         struct nlattr *sinfo[NL80211_STA_INFO_MAX + 1];
660         struct nlattr *rinfo[NL80211_RATE_INFO_MAX + 1];
661
662         static struct nla_policy stats_policy[NL80211_STA_INFO_MAX + 1] = {
663                 [NL80211_STA_INFO_INACTIVE_TIME] = { .type = NLA_U32    },
664                 [NL80211_STA_INFO_RX_BYTES]      = { .type = NLA_U32    },
665                 [NL80211_STA_INFO_TX_BYTES]      = { .type = NLA_U32    },
666                 [NL80211_STA_INFO_RX_PACKETS]    = { .type = NLA_U32    },
667                 [NL80211_STA_INFO_TX_PACKETS]    = { .type = NLA_U32    },
668                 [NL80211_STA_INFO_SIGNAL]        = { .type = NLA_U8     },
669                 [NL80211_STA_INFO_TX_BITRATE]    = { .type = NLA_NESTED },
670                 [NL80211_STA_INFO_LLID]          = { .type = NLA_U16    },
671                 [NL80211_STA_INFO_PLID]          = { .type = NLA_U16    },
672                 [NL80211_STA_INFO_PLINK_STATE]   = { .type = NLA_U8     },
673         };
674
675         static struct nla_policy rate_policy[NL80211_RATE_INFO_MAX + 1] = {
676                 [NL80211_RATE_INFO_BITRATE]      = { .type = NLA_U16  },
677                 [NL80211_RATE_INFO_MCS]          = { .type = NLA_U8   },
678                 [NL80211_RATE_INFO_40_MHZ_WIDTH] = { .type = NLA_FLAG },
679                 [NL80211_RATE_INFO_SHORT_GI]     = { .type = NLA_FLAG },
680         };
681
682         if (attr[NL80211_ATTR_STA_INFO])
683         {
684                 if( !nla_parse_nested(sinfo, NL80211_STA_INFO_MAX,
685                                 attr[NL80211_ATTR_STA_INFO], stats_policy) )
686                 {
687                         if (sinfo[NL80211_STA_INFO_SIGNAL])
688                         {
689                                 dbm = nla_get_u8(sinfo[NL80211_STA_INFO_SIGNAL]);
690                                 rr->rssi = rr->rssi ? (int8_t)((rr->rssi + dbm) / 2) : dbm;
691                         }
692
693                         if (sinfo[NL80211_STA_INFO_TX_BITRATE])
694                         {
695                                 if( !nla_parse_nested(rinfo, NL80211_RATE_INFO_MAX,
696                                                 sinfo[NL80211_STA_INFO_TX_BITRATE], rate_policy) )
697                                 {
698                                         if (rinfo[NL80211_RATE_INFO_BITRATE])
699                                         {
700                                                 mbit = nla_get_u16(rinfo[NL80211_RATE_INFO_BITRATE]);
701                                                 rr->rate = rr->rate
702                                                         ? (int16_t)((rr->rate + mbit) / 2) : mbit;
703                                         }
704                                 }
705                         }
706                 }
707         }
708
709         return NL_SKIP;
710 }
711
712 static void nl80211_fill_signal(const char *ifname, struct nl80211_rssi_rate *r)
713 {
714         DIR *d;
715         struct dirent *de;
716         struct nl80211_msg_conveyor *req;
717
718         r->rssi = 0;
719         r->rate = 0;
720
721         if ((d = opendir("/sys/class/net")) != NULL)
722         {
723                 while ((de = readdir(d)) != NULL)
724                 {
725                         if (!strncmp(de->d_name, ifname, strlen(ifname)) &&
726                                 (!de->d_name[strlen(ifname)] ||
727                                  !strncmp(&de->d_name[strlen(ifname)], ".sta", 4)))
728                         {
729                                 req = nl80211_msg(de->d_name, NL80211_CMD_GET_STATION,
730                                                                   NLM_F_DUMP);
731
732                                 if (req)
733                                 {
734                                         nl80211_send(req, nl80211_fill_signal_cb, r);
735                                         nl80211_free(req);
736                                 }
737                         }
738                 }
739
740                 closedir(d);
741         }
742 }
743
744 int nl80211_get_bitrate(const char *ifname, int *buf)
745 {
746         struct nl80211_rssi_rate rr;
747
748         if (!wext_get_bitrate(ifname, buf))
749                 return 0;
750
751         nl80211_fill_signal(ifname, &rr);
752
753         if (rr.rate)
754         {
755                 *buf = (rr.rate * 100);
756                 return 0;
757         }
758
759         return -1;
760 }
761
762 int nl80211_get_signal(const char *ifname, int *buf)
763 {
764         struct nl80211_rssi_rate rr;
765
766         if (!wext_get_signal(ifname, buf))
767                 return 0;
768
769         nl80211_fill_signal(ifname, &rr);
770
771         if (rr.rssi)
772         {
773                 *buf = rr.rssi;
774                 return 0;
775         }
776
777         return -1;
778 }
779
780 static int nl80211_get_noise_cb(struct nl_msg *msg, void *arg)
781 {
782         int8_t *noise = arg;
783         struct nlattr **tb = nl80211_parse(msg);
784         struct nlattr *si[NL80211_SURVEY_INFO_MAX + 1];
785
786         static struct nla_policy sp[NL80211_SURVEY_INFO_MAX + 1] = {
787                 [NL80211_SURVEY_INFO_FREQUENCY] = { .type = NLA_U32 },
788                 [NL80211_SURVEY_INFO_NOISE]     = { .type = NLA_U8  },
789         };
790
791         if (!tb[NL80211_ATTR_SURVEY_INFO])
792                 return NL_SKIP;
793
794         if (nla_parse_nested(si, NL80211_SURVEY_INFO_MAX,
795                                                  tb[NL80211_ATTR_SURVEY_INFO], sp))
796                 return NL_SKIP;
797
798         if (!si[NL80211_SURVEY_INFO_NOISE])
799                 return NL_SKIP;
800
801         if (!*noise || si[NL80211_SURVEY_INFO_IN_USE])
802                 *noise = (int8_t)nla_get_u8(si[NL80211_SURVEY_INFO_NOISE]);
803
804         return NL_SKIP;
805 }
806
807
808 int nl80211_get_noise(const char *ifname, int *buf)
809 {
810         int8_t noise;
811         struct nl80211_msg_conveyor *req;
812
813         req = nl80211_msg(ifname, NL80211_CMD_GET_SURVEY, NLM_F_DUMP);
814         if (req)
815         {
816                 noise = 0;
817
818                 nl80211_send(req, nl80211_get_noise_cb, &noise);
819                 nl80211_free(req);
820
821                 if (noise)
822                 {
823                         *buf = noise;
824                         return 0;
825                 }
826         }
827
828         return -1;
829 }
830
831 int nl80211_get_quality(const char *ifname, int *buf)
832 {
833         int signal;
834
835         if (wext_get_quality(ifname, buf))
836         {
837                 *buf = 0;
838
839                 if (!nl80211_get_signal(ifname, &signal))
840                 {
841                         /* A positive signal level is usually just a quality
842                          * value, pass through as-is */
843                         if (signal >= 0)
844                         {
845                                 *buf = signal;
846                         }
847
848                         /* The cfg80211 wext compat layer assumes a signal range
849                          * of -110 dBm to -40 dBm, the quality value is derived
850                          * by adding 110 to the signal level */
851                         else
852                         {
853                                 if (signal < -110)
854                                         signal = -110;
855                                 else if (signal > -40)
856                                         signal = -40;
857
858                                 *buf = (signal + 110);
859                         }
860                 }
861         }
862
863         return 0;
864 }
865
866 int nl80211_get_quality_max(const char *ifname, int *buf)
867 {
868         if (wext_get_quality_max(ifname, buf))
869                 /* The cfg80211 wext compat layer assumes a maximum
870                  * quality of 70 */
871                 *buf = 70;
872
873         return 0;
874 }
875
876 int nl80211_get_encryption(const char *ifname, char *buf)
877 {
878         int i;
879         char k[9];
880         char *val, *res;
881         struct iwinfo_crypto_entry *c = (struct iwinfo_crypto_entry *)buf;
882
883         /* WPA supplicant */
884         if( (res = nl80211_wpactl_info(ifname, "STATUS", NULL)) &&
885             (val = nl80211_getval(NULL, res, "pairwise_cipher")) )
886         {
887                 /* WEP */
888                 if (strstr(val, "WEP"))
889                 {
890                         if (strstr(val, "WEP-40"))
891                                 c->pair_ciphers |= IWINFO_CIPHER_WEP40;
892
893                         else if (strstr(val, "WEP-104"))
894                                 c->pair_ciphers |= IWINFO_CIPHER_WEP104;
895
896                         c->enabled       = 1;
897                         c->group_ciphers = c->pair_ciphers;
898
899                         c->auth_suites |= IWINFO_KMGMT_NONE;
900                         c->auth_algs   |= IWINFO_AUTH_OPEN; /* XXX: assumption */
901                 }
902
903                 /* WPA */
904                 else
905                 {
906                         if (strstr(val, "TKIP"))
907                                 c->pair_ciphers |= IWINFO_CIPHER_TKIP;
908
909                         else if (strstr(val, "CCMP"))
910                                 c->pair_ciphers |= IWINFO_CIPHER_CCMP;
911
912                         else if (strstr(val, "NONE"))
913                                 c->pair_ciphers |= IWINFO_CIPHER_NONE;
914
915                         else if (strstr(val, "WEP-40"))
916                                 c->pair_ciphers |= IWINFO_CIPHER_WEP40;
917
918                         else if (strstr(val, "WEP-104"))
919                                 c->pair_ciphers |= IWINFO_CIPHER_WEP104;
920
921
922                         if ((val = nl80211_getval(NULL, res, "group_cipher")))
923                         {
924                                 if (strstr(val, "TKIP"))
925                                         c->group_ciphers |= IWINFO_CIPHER_TKIP;
926
927                                 else if (strstr(val, "CCMP"))
928                                         c->group_ciphers |= IWINFO_CIPHER_CCMP;
929
930                                 else if (strstr(val, "NONE"))
931                                         c->group_ciphers |= IWINFO_CIPHER_NONE;
932
933                                 else if (strstr(val, "WEP-40"))
934                                         c->group_ciphers |= IWINFO_CIPHER_WEP40;
935
936                                 else if (strstr(val, "WEP-104"))
937                                         c->group_ciphers |= IWINFO_CIPHER_WEP104;
938                         }
939
940
941                         if ((val = nl80211_getval(NULL, res, "key_mgmt")))
942                         {
943                                 if (strstr(val, "WPA2"))
944                                         c->wpa_version = 2;
945
946                                 else if (strstr(val, "WPA"))
947                                         c->wpa_version = 1;
948
949
950                                 if (strstr(val, "PSK"))
951                                         c->auth_suites |= IWINFO_KMGMT_PSK;
952
953                                 else if (strstr(val, "EAP") || strstr(val, "802.1X"))
954                                         c->auth_suites |= IWINFO_KMGMT_8021x;
955
956                                 else if (strstr(val, "NONE"))
957                                         c->auth_suites |= IWINFO_KMGMT_NONE;
958                         }
959
960                         c->enabled = (c->wpa_version && c->auth_suites) ? 1 : 0;
961                 }
962
963                 return 0;
964         }
965
966         /* Hostapd */
967         else if ((res = nl80211_hostapd_info(ifname)))
968         {
969                 if ((val = nl80211_getval(ifname, res, "wpa")) != NULL)
970                         c->wpa_version = atoi(val);
971
972                 val = nl80211_getval(ifname, res, "wpa_key_mgmt");
973
974                 if (!val || strstr(val, "PSK"))
975                         c->auth_suites |= IWINFO_KMGMT_PSK;
976
977                 if (val && strstr(val, "EAP"))
978                         c->auth_suites |= IWINFO_KMGMT_8021x;
979
980                 if (val && strstr(val, "NONE"))
981                         c->auth_suites |= IWINFO_KMGMT_NONE;
982
983                 if ((val = nl80211_getval(ifname, res, "wpa_pairwise")) != NULL)
984                 {
985                         if (strstr(val, "TKIP"))
986                                 c->pair_ciphers |= IWINFO_CIPHER_TKIP;
987
988                         if (strstr(val, "CCMP"))
989                                 c->pair_ciphers |= IWINFO_CIPHER_CCMP;
990
991                         if (strstr(val, "NONE"))
992                                 c->pair_ciphers |= IWINFO_CIPHER_NONE;
993                 }
994
995                 if ((val = nl80211_getval(ifname, res, "auth_algs")) != NULL)
996                 {
997                         switch(atoi(val)) {
998                                 case 1:
999                                         c->auth_algs |= IWINFO_AUTH_OPEN;
1000                                         break;
1001
1002                                 case 2:
1003                                         c->auth_algs |= IWINFO_AUTH_SHARED;
1004                                         break;
1005
1006                                 case 3:
1007                                         c->auth_algs |= IWINFO_AUTH_OPEN;
1008                                         c->auth_algs |= IWINFO_AUTH_SHARED;
1009                                         break;
1010
1011                                 default:
1012                                         break;
1013                         }
1014
1015                         for( i = 0; i < 4; i++ )
1016                         {
1017                                 snprintf(k, sizeof(k), "wep_key%d", i);
1018
1019                                 if ((val = nl80211_getval(ifname, res, k)))
1020                                 {
1021                                         if ((strlen(val) == 5) || (strlen(val) == 10))
1022                                                 c->pair_ciphers |= IWINFO_CIPHER_WEP40;
1023
1024                                         else if ((strlen(val) == 13) || (strlen(val) == 26))
1025                                                 c->pair_ciphers |= IWINFO_CIPHER_WEP104;
1026                                 }
1027                         }
1028                 }
1029
1030                 c->group_ciphers = c->pair_ciphers;
1031                 c->enabled = (c->wpa_version || c->pair_ciphers) ? 1 : 0;
1032
1033                 return 0;
1034         }
1035
1036         return -1;
1037 }
1038
1039
1040 static int nl80211_get_assoclist_cb(struct nl_msg *msg, void *arg)
1041 {
1042         struct nl80211_array_buf *arr = arg;
1043         struct iwinfo_assoclist_entry *e = arr->buf;
1044         struct nlattr **attr = nl80211_parse(msg);
1045         struct nlattr *sinfo[NL80211_STA_INFO_MAX + 1];
1046
1047         static struct nla_policy stats_policy[NL80211_STA_INFO_MAX + 1] = {
1048                 [NL80211_STA_INFO_INACTIVE_TIME] = { .type = NLA_U32    },
1049                 [NL80211_STA_INFO_RX_BYTES]      = { .type = NLA_U32    },
1050                 [NL80211_STA_INFO_TX_BYTES]      = { .type = NLA_U32    },
1051                 [NL80211_STA_INFO_RX_PACKETS]    = { .type = NLA_U32    },
1052                 [NL80211_STA_INFO_TX_PACKETS]    = { .type = NLA_U32    },
1053                 [NL80211_STA_INFO_SIGNAL]        = { .type = NLA_U8     },
1054                 [NL80211_STA_INFO_TX_BITRATE]    = { .type = NLA_NESTED },
1055                 [NL80211_STA_INFO_LLID]          = { .type = NLA_U16    },
1056                 [NL80211_STA_INFO_PLID]          = { .type = NLA_U16    },
1057                 [NL80211_STA_INFO_PLINK_STATE]   = { .type = NLA_U8     },
1058         };
1059
1060         /* advance to end of array */
1061         e += arr->count;
1062
1063         if (attr[NL80211_ATTR_MAC])
1064                 memcpy(e->mac, nla_data(attr[NL80211_ATTR_MAC]), 6);
1065
1066         if (attr[NL80211_ATTR_STA_INFO])
1067         {
1068                 if (!nla_parse_nested(sinfo, NL80211_STA_INFO_MAX,
1069                                 attr[NL80211_ATTR_STA_INFO], stats_policy))
1070                 {
1071                         if (sinfo[NL80211_STA_INFO_SIGNAL])
1072                                 e->signal = nla_get_u8(sinfo[NL80211_STA_INFO_SIGNAL]);
1073                 }
1074         }
1075
1076         e->noise = 0; /* filled in by caller */
1077         arr->count++;
1078
1079         return NL_SKIP;
1080 }
1081
1082 int nl80211_get_assoclist(const char *ifname, char *buf, int *len)
1083 {
1084         DIR *d;
1085         int i, noise = 0;
1086         struct dirent *de;
1087         struct nl80211_msg_conveyor *req;
1088         struct nl80211_array_buf arr = { .buf = buf, .count = 0 };
1089         struct iwinfo_assoclist_entry *e;
1090
1091         if ((d = opendir("/sys/class/net")) != NULL)
1092         {
1093                 while ((de = readdir(d)) != NULL)
1094                 {
1095                         if (!strncmp(de->d_name, ifname, strlen(ifname)) &&
1096                                 (!de->d_name[strlen(ifname)] ||
1097                                  !strncmp(&de->d_name[strlen(ifname)], ".sta", 4)))
1098                         {
1099                                 req = nl80211_msg(de->d_name, NL80211_CMD_GET_STATION,
1100                                                                   NLM_F_DUMP);
1101
1102                                 if (req)
1103                                 {
1104                                         nl80211_send(req, nl80211_get_assoclist_cb, &arr);
1105                                         nl80211_free(req);
1106                                 }
1107
1108                                 break;
1109                         }
1110                 }
1111
1112                 closedir(d);
1113
1114                 if (!nl80211_get_noise(ifname, &noise))
1115                         for (i = 0, e = arr.buf; i < arr.count; i++, e++)
1116                                 e->noise = noise;
1117
1118                 *len = (arr.count * sizeof(struct iwinfo_assoclist_entry));
1119                 return 0;
1120         }
1121
1122         return -1;
1123 }
1124
1125 static int nl80211_get_txpwrlist_cb(struct nl_msg *msg, void *arg)
1126 {
1127         int *dbm_max = arg;
1128         int ch_cur, ch_cmp, bands_remain, freqs_remain;
1129
1130         struct nlattr **attr = nl80211_parse(msg);
1131         struct nlattr *bands[NL80211_BAND_ATTR_MAX + 1];
1132         struct nlattr *freqs[NL80211_FREQUENCY_ATTR_MAX + 1];
1133         struct nlattr *band, *freq;
1134
1135         static struct nla_policy freq_policy[NL80211_FREQUENCY_ATTR_MAX + 1] = {
1136                 [NL80211_FREQUENCY_ATTR_FREQ]         = { .type = NLA_U32  },
1137                 [NL80211_FREQUENCY_ATTR_DISABLED]     = { .type = NLA_FLAG },
1138                 [NL80211_FREQUENCY_ATTR_PASSIVE_SCAN] = { .type = NLA_FLAG },
1139                 [NL80211_FREQUENCY_ATTR_NO_IBSS]      = { .type = NLA_FLAG },
1140                 [NL80211_FREQUENCY_ATTR_RADAR]        = { .type = NLA_FLAG },
1141                 [NL80211_FREQUENCY_ATTR_MAX_TX_POWER] = { .type = NLA_U32  },
1142         };
1143
1144         ch_cur = *dbm_max; /* value int* is initialized with channel by caller */
1145         *dbm_max = -1;
1146
1147         nla_for_each_nested(band, attr[NL80211_ATTR_WIPHY_BANDS], bands_remain)
1148         {
1149                 nla_parse(bands, NL80211_BAND_ATTR_MAX, nla_data(band),
1150                           nla_len(band), NULL);
1151
1152                 nla_for_each_nested(freq,
1153                         bands[NL80211_BAND_ATTR_FREQS], freqs_remain)
1154                 {
1155                         nla_parse(freqs, NL80211_FREQUENCY_ATTR_MAX,
1156                                 nla_data(freq), nla_len(freq), freq_policy);
1157
1158                         ch_cmp = nl80211_freq2channel(
1159                                 nla_get_u32(freqs[NL80211_FREQUENCY_ATTR_FREQ]));
1160
1161                         if( (!ch_cur || (ch_cmp == ch_cur)) &&
1162                                 freqs[NL80211_FREQUENCY_ATTR_MAX_TX_POWER] )
1163                         {
1164                                 *dbm_max = (int)(0.01 * nla_get_u32(
1165                                         freqs[NL80211_FREQUENCY_ATTR_MAX_TX_POWER]));
1166
1167                                 break;
1168                         }
1169                 }
1170         }
1171
1172         return NL_SKIP;
1173 }
1174
1175 int nl80211_get_txpwrlist(const char *ifname, char *buf, int *len)
1176 {
1177         int ch_cur;
1178         int dbm_max = -1, dbm_cur, dbm_cnt;
1179         struct nl80211_msg_conveyor *req;
1180         struct iwinfo_txpwrlist_entry entry;
1181
1182         if (nl80211_get_channel(ifname, &ch_cur))
1183                 ch_cur = 0;
1184
1185         req = nl80211_msg(ifname, NL80211_CMD_GET_WIPHY, 0);
1186         if (req)
1187         {
1188                 /* initialize the value pointer with channel for callback */
1189                 dbm_max = ch_cur;
1190
1191                 nl80211_send(req, nl80211_get_txpwrlist_cb, &dbm_max);
1192                 nl80211_free(req);
1193         }
1194
1195         if (dbm_max > -1)
1196         {
1197                 for (dbm_cur = 0, dbm_cnt = 0;
1198                      dbm_cur < dbm_max;
1199                      dbm_cur++, dbm_cnt++)
1200                 {
1201                         entry.dbm = dbm_cur;
1202                         entry.mw  = iwinfo_dbm2mw(dbm_cur);
1203
1204                         memcpy(&buf[dbm_cnt * sizeof(entry)], &entry, sizeof(entry));
1205                 }
1206
1207                 entry.dbm = dbm_max;
1208                 entry.mw  = iwinfo_dbm2mw(dbm_max);
1209
1210                 memcpy(&buf[dbm_cnt * sizeof(entry)], &entry, sizeof(entry));
1211                 dbm_cnt++;
1212
1213                 *len = dbm_cnt * sizeof(entry);
1214                 return 0;
1215         }
1216
1217         return -1;
1218 }
1219
1220 static void nl80211_get_scancrypto(const char *spec,
1221         struct iwinfo_crypto_entry *c)
1222 {
1223         if (strstr(spec, "WPA") || strstr(spec, "WEP"))
1224         {
1225                 c->enabled = 1;
1226
1227                 if (strstr(spec, "WPA2-") && strstr(spec, "WPA-"))
1228                         c->wpa_version = 3;
1229
1230                 else if (strstr(spec, "WPA2"))
1231                         c->wpa_version = 2;
1232
1233                 else if (strstr(spec, "WPA"))
1234                         c->wpa_version = 1;
1235
1236                 else if (strstr(spec, "WEP"))
1237                         c->auth_algs = IWINFO_AUTH_OPEN | IWINFO_AUTH_SHARED;
1238
1239
1240                 if (strstr(spec, "PSK"))
1241                         c->auth_suites |= IWINFO_KMGMT_PSK;
1242
1243                 if (strstr(spec, "802.1X") || strstr(spec, "EAP"))
1244                         c->auth_suites |= IWINFO_KMGMT_8021x;
1245
1246                 if (strstr(spec, "WPA-NONE"))
1247                         c->auth_suites |= IWINFO_KMGMT_NONE;
1248
1249
1250                 if (strstr(spec, "TKIP"))
1251                         c->pair_ciphers |= IWINFO_CIPHER_TKIP;
1252
1253                 if (strstr(spec, "CCMP"))
1254                         c->pair_ciphers |= IWINFO_CIPHER_CCMP;
1255
1256                 if (strstr(spec, "WEP-40"))
1257                         c->pair_ciphers |= IWINFO_CIPHER_WEP40;
1258
1259                 if (strstr(spec, "WEP-104"))
1260                         c->pair_ciphers |= IWINFO_CIPHER_WEP104;
1261
1262                 c->group_ciphers = c->pair_ciphers;
1263         }
1264         else
1265         {
1266                 c->enabled = 0;
1267         }
1268 }
1269
1270 int nl80211_get_scanlist(const char *ifname, char *buf, int *len)
1271 {
1272         int freq, rssi, qmax, count;
1273         char *res;
1274         char ssid[128] = { 0 };
1275         char bssid[18] = { 0 };
1276         char cipher[256] = { 0 };
1277
1278         /* Got a radioX pseudo interface, find some interface on it or create one */
1279         if (!strncmp(ifname, "radio", 5))
1280         {
1281                 /* Reuse existing interface */
1282                 if ((res = nl80211_phy2ifname(ifname)) != NULL)
1283                 {
1284                         return nl80211_get_scanlist(res, buf, len);
1285                 }
1286
1287                 /* Need to spawn a temporary iface for scanning */
1288                 else if ((res = nl80211_ifadd(ifname)) != NULL)
1289                 {
1290                         count = nl80211_get_scanlist(res, buf, len);
1291                         nl80211_ifdel(res);
1292                         return count;
1293                 }
1294         }
1295
1296         struct iwinfo_scanlist_entry *e = (struct iwinfo_scanlist_entry *)buf;
1297
1298         /* WPA supplicant */
1299         if ((res = nl80211_wpactl_info(ifname, "SCAN", "CTRL-EVENT-SCAN-RESULTS")))
1300         {
1301                 if ((res = nl80211_wpactl_info(ifname, "SCAN_RESULTS", NULL)))
1302                 {
1303                         nl80211_get_quality_max(ifname, &qmax);
1304
1305                         /* skip header line */
1306                         while( *res++ != '\n' );
1307
1308                         count = 0;
1309
1310                         while( sscanf(res, "%17s %d %d %255s%*[ \t]%127[^\n]\n",
1311                                       bssid, &freq, &rssi, cipher, ssid) > 0 )
1312                         {
1313                                 /* BSSID */
1314                                 e->mac[0] = strtol(&bssid[0],  NULL, 16);
1315                                 e->mac[1] = strtol(&bssid[3],  NULL, 16);
1316                                 e->mac[2] = strtol(&bssid[6],  NULL, 16);
1317                                 e->mac[3] = strtol(&bssid[9],  NULL, 16);
1318                                 e->mac[4] = strtol(&bssid[12], NULL, 16);
1319                                 e->mac[5] = strtol(&bssid[15], NULL, 16);
1320
1321                                 /* SSID */
1322                                 memcpy(e->ssid, ssid,
1323                                         min(strlen(ssid), sizeof(e->ssid) - 1));
1324
1325                                 /* Mode (assume master) */
1326                                 sprintf((char *)e->mode, "Master");
1327
1328                                 /* Channel */
1329                                 e->channel = nl80211_freq2channel(freq);
1330
1331                                 /* Signal */
1332                                 e->signal = rssi;
1333
1334                                 /* Quality */
1335                                 if (rssi < 0)
1336                                 {
1337                                         /* The cfg80211 wext compat layer assumes a signal range
1338                                          * of -110 dBm to -40 dBm, the quality value is derived
1339                                          * by adding 110 to the signal level */
1340                                         if (rssi < -110)
1341                                                 rssi = -110;
1342                                         else if (rssi > -40)
1343                                                 rssi = -40;
1344
1345                                         e->quality = (rssi + 110);
1346                                 }
1347                                 else
1348                                 {
1349                                         e->quality = rssi;
1350                                 }
1351
1352                                 /* Max. Quality */
1353                                 e->quality_max = qmax;
1354
1355                                 /* Crypto */
1356                                 nl80211_get_scancrypto(cipher, &e->crypto);
1357
1358                                 /* advance to next line */
1359                                 while( *res && *res++ != '\n' );
1360
1361                                 count++;
1362                                 e++;
1363
1364                                 memset(ssid, 0, sizeof(ssid));
1365                                 memset(bssid, 0, sizeof(bssid));
1366                                 memset(cipher, 0, sizeof(cipher));
1367                         }
1368
1369                         *len = count * sizeof(struct iwinfo_scanlist_entry);
1370                         return 0;
1371                 }
1372         }
1373
1374         /* AP scan */
1375         else
1376         {
1377                 /* Got a temp interface, don't create yet another one */
1378                 if (!strncmp(ifname, "tmp.", 4))
1379                 {
1380                         if (!iwinfo_ifup(ifname))
1381                                 return -1;
1382
1383                         wext_get_scanlist(ifname, buf, len);
1384                         iwinfo_ifdown(ifname);
1385                         return 0;
1386                 }
1387
1388                 /* Spawn a new scan interface */
1389                 else
1390                 {
1391                         if (!(res = nl80211_ifadd(ifname)))
1392                                 goto out;
1393
1394                         if (!iwinfo_ifmac(res))
1395                                 goto out;
1396
1397                         /* if we can take the new interface up, the driver supports an
1398                          * additional interface and there's no need to tear down the ap */
1399                         if (iwinfo_ifup(res))
1400                         {
1401                                 wext_get_scanlist(res, buf, len);
1402                                 iwinfo_ifdown(res);
1403                         }
1404
1405                         /* driver cannot create secondary interface, take down ap
1406                          * during scan */
1407                         else if (iwinfo_ifdown(ifname) && iwinfo_ifup(res))
1408                         {
1409                                 wext_get_scanlist(res, buf, len);
1410                                 iwinfo_ifdown(res);
1411                                 iwinfo_ifup(ifname);
1412                                 nl80211_hostapd_hup(ifname);
1413                         }
1414
1415                 out:
1416                         nl80211_ifdel(res);
1417                         return 0;
1418                 }
1419         }
1420
1421         return -1;
1422 }
1423
1424 static int nl80211_get_freqlist_cb(struct nl_msg *msg, void *arg)
1425 {
1426         int bands_remain, freqs_remain;
1427
1428         struct nl80211_array_buf *arr = arg;
1429         struct iwinfo_freqlist_entry *e = arr->buf;
1430
1431         struct nlattr **attr = nl80211_parse(msg);
1432         struct nlattr *bands[NL80211_BAND_ATTR_MAX + 1];
1433         struct nlattr *freqs[NL80211_FREQUENCY_ATTR_MAX + 1];
1434         struct nlattr *band, *freq;
1435
1436         static struct nla_policy freq_policy[NL80211_FREQUENCY_ATTR_MAX + 1] = {
1437                 [NL80211_FREQUENCY_ATTR_FREQ]         = { .type = NLA_U32  },
1438                 [NL80211_FREQUENCY_ATTR_DISABLED]     = { .type = NLA_FLAG },
1439                 [NL80211_FREQUENCY_ATTR_PASSIVE_SCAN] = { .type = NLA_FLAG },
1440                 [NL80211_FREQUENCY_ATTR_NO_IBSS]      = { .type = NLA_FLAG },
1441                 [NL80211_FREQUENCY_ATTR_RADAR]        = { .type = NLA_FLAG },
1442                 [NL80211_FREQUENCY_ATTR_MAX_TX_POWER] = { .type = NLA_U32  },
1443         };
1444
1445         nla_for_each_nested(band, attr[NL80211_ATTR_WIPHY_BANDS], bands_remain)
1446         {
1447                 nla_parse(bands, NL80211_BAND_ATTR_MAX, nla_data(band),
1448                         nla_len(band), NULL);
1449
1450                 nla_for_each_nested(freq,
1451                         bands[NL80211_BAND_ATTR_FREQS], freqs_remain)
1452                 {
1453                         nla_parse(freqs, NL80211_FREQUENCY_ATTR_MAX,
1454                                 nla_data(freq), nla_len(freq), NULL);
1455
1456                         if( !freqs[NL80211_FREQUENCY_ATTR_FREQ] ||
1457                                 freqs[NL80211_FREQUENCY_ATTR_DISABLED] )
1458                                 continue;
1459
1460                         e->mhz = nla_get_u32(freqs[NL80211_FREQUENCY_ATTR_FREQ]);
1461                         e->channel = nl80211_freq2channel(e->mhz);
1462
1463                         e->restricted = (
1464                                 freqs[NL80211_FREQUENCY_ATTR_PASSIVE_SCAN] ||
1465                                 freqs[NL80211_FREQUENCY_ATTR_NO_IBSS]      ||
1466                                 freqs[NL80211_FREQUENCY_ATTR_RADAR]
1467                         ) ? 1 : 0;
1468
1469                         e++;
1470                         arr->count++;
1471                 }
1472         }
1473
1474         return NL_SKIP;
1475 }
1476
1477 int nl80211_get_freqlist(const char *ifname, char *buf, int *len)
1478 {
1479         struct nl80211_msg_conveyor *req;
1480         struct nl80211_array_buf arr = { .buf = buf, .count = 0 };
1481
1482         req = nl80211_msg(ifname, NL80211_CMD_GET_WIPHY, 0);
1483         if (req)
1484         {
1485                 nl80211_send(req, nl80211_get_freqlist_cb, &arr);
1486                 nl80211_free(req);
1487         }
1488
1489         if (arr.count > 0)
1490         {
1491                 *len = arr.count * sizeof(struct iwinfo_freqlist_entry);
1492                 return 0;
1493         }
1494
1495         return -1;
1496 }
1497
1498 static int nl80211_get_country_cb(struct nl_msg *msg, void *arg)
1499 {
1500         char *buf = arg;
1501         struct nlattr **attr = nl80211_parse(msg);
1502
1503         if (attr[NL80211_ATTR_REG_ALPHA2])
1504                 memcpy(buf, nla_data(attr[NL80211_ATTR_REG_ALPHA2]), 2);
1505         else
1506                 buf[0] = 0;
1507
1508         return NL_SKIP;
1509 }
1510
1511 int nl80211_get_country(const char *ifname, char *buf)
1512 {
1513         int rv = -1;
1514         struct nl80211_msg_conveyor *req;
1515
1516         req = nl80211_msg(ifname, NL80211_CMD_GET_REG, 0);
1517         if (req)
1518         {
1519                 nl80211_send(req, nl80211_get_country_cb, buf);
1520                 nl80211_free(req);
1521
1522                 if (buf[0])
1523                         rv = 0;
1524         }
1525
1526         return rv;
1527 }
1528
1529 int nl80211_get_countrylist(const char *ifname, char *buf, int *len)
1530 {
1531         int i, count;
1532         struct iwinfo_iso3166_label *l;
1533         struct iwinfo_country_entry *e = (struct iwinfo_country_entry *)buf;
1534
1535         for( l = ISO3166_Names, count = 0; l->iso3166; l++, e++, count++ )
1536         {
1537                 e->iso3166 = l->iso3166;
1538                 e->ccode[0] = (l->iso3166 / 256);
1539                 e->ccode[1] = (l->iso3166 % 256);
1540         }
1541
1542         *len = (count * sizeof(struct iwinfo_country_entry));
1543         return 0;
1544 }
1545
1546 static int nl80211_get_hwmodelist_cb(struct nl_msg *msg, void *arg)
1547 {
1548         int *modes = arg;
1549         int bands_remain, freqs_remain;
1550         uint16_t caps = 0;
1551         struct nlattr **attr = nl80211_parse(msg);
1552         struct nlattr *bands[NL80211_BAND_ATTR_MAX + 1];
1553         struct nlattr *freqs[NL80211_FREQUENCY_ATTR_MAX + 1];
1554         struct nlattr *band, *freq;
1555
1556         *modes = 0;
1557
1558         if (attr[NL80211_ATTR_WIPHY_BANDS])
1559         {
1560                 nla_for_each_nested(band, attr[NL80211_ATTR_WIPHY_BANDS], bands_remain)
1561                 {
1562                         nla_parse(bands, NL80211_BAND_ATTR_MAX, nla_data(band),
1563                                 nla_len(band), NULL);
1564
1565                         if (bands[NL80211_BAND_ATTR_HT_CAPA])
1566                                 caps = nla_get_u16(bands[NL80211_BAND_ATTR_HT_CAPA]);
1567
1568                         /* Treat any nonzero capability as 11n */
1569                         if (caps > 0)
1570                                 *modes |= IWINFO_80211_N;
1571
1572                         nla_for_each_nested(freq,
1573                                 bands[NL80211_BAND_ATTR_FREQS], freqs_remain)
1574                         {
1575                                 nla_parse(freqs, NL80211_FREQUENCY_ATTR_MAX,
1576                                         nla_data(freq), nla_len(freq), NULL);
1577
1578                                 if (!freqs[NL80211_FREQUENCY_ATTR_FREQ])
1579                                         continue;
1580
1581                                 if (nla_get_u32(freqs[NL80211_FREQUENCY_ATTR_FREQ]) < 2485)
1582                                 {
1583                                         *modes |= IWINFO_80211_B;
1584                                         *modes |= IWINFO_80211_G;
1585                                 }
1586                                 else
1587                                 {
1588                                         *modes |= IWINFO_80211_A;
1589                                 }
1590                         }
1591                 }
1592         }
1593
1594         return NL_SKIP;
1595 }
1596
1597 int nl80211_get_hwmodelist(const char *ifname, int *buf)
1598 {
1599         struct nl80211_msg_conveyor *req;
1600
1601         req = nl80211_msg(ifname, NL80211_CMD_GET_WIPHY, 0);
1602         if (req)
1603         {
1604                 nl80211_send(req, nl80211_get_hwmodelist_cb, buf);
1605                 nl80211_free(req);
1606         }
1607
1608         return *buf ? 0 : -1;
1609 }
1610
1611 int nl80211_get_mbssid_support(const char *ifname, int *buf)
1612 {
1613         /* Test whether we can create another interface */
1614         char *nif = nl80211_ifadd(ifname);
1615
1616         if (nif)
1617         {
1618                 *buf = (iwinfo_ifmac(nif) && iwinfo_ifup(nif));
1619
1620                 iwinfo_ifdown(nif);
1621                 nl80211_ifdel(nif);
1622
1623                 return 0;
1624         }
1625
1626         return -1;
1627 }