iwinfo_nl80211.c

   1 /*
   2  * iwinfo - Wireless Information Library - NL80211 Backend
   3  *
   4  *   Copyright (C) 2010-2013 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 <limits.h>
  26 #include <glob.h>
  27 #include <fnmatch.h>
  28 #include <stdarg.h>
  29
  30 #include "iwinfo_nl80211.h"
  31
  32 #define min(x, y) ((x) < (y)) ? (x) : (y)
  33
  34 #define BIT(x) (1ULL<<(x))
  35
  36 static struct nl80211_state *nls = NULL;
  37
  38 static void nl80211_close(void)
  39 {
  40         if (nls)
  41         {
  42                 if (nls->nlctrl)
  43                         genl_family_put(nls->nlctrl);
  44
  45                 if (nls->nl80211)
  46                         genl_family_put(nls->nl80211);
  47
  48                 if (nls->nl_sock)
  49                         nl_socket_free(nls->nl_sock);
  50
  51                 if (nls->nl_cache)
  52                         nl_cache_free(nls->nl_cache);
  53
  54                 free(nls);
  55                 nls = NULL;
  56         }
  57 }
  58
  59 static int nl80211_init(void)
  60 {
  61         int err, fd;
  62
  63         if (!nls)
  64         {
  65                 nls = malloc(sizeof(struct nl80211_state));
  66                 if (!nls) {
  67                         err = -ENOMEM;
  68                         goto err;
  69                 }
  70
  71                 memset(nls, 0, sizeof(*nls));
  72
  73                 nls->nl_sock = nl_socket_alloc();
  74                 if (!nls->nl_sock) {
  75                         err = -ENOMEM;
  76                         goto err;
  77                 }
  78
  79                 if (genl_connect(nls->nl_sock)) {
  80                         err = -ENOLINK;
  81                         goto err;
  82                 }
  83
  84                 fd = nl_socket_get_fd(nls->nl_sock);
  85                 if (fcntl(fd, F_SETFD, fcntl(fd, F_GETFD) | FD_CLOEXEC) < 0) {
  86                         err = -EINVAL;
  87                         goto err;
  88                 }
  89
  90                 if (genl_ctrl_alloc_cache(nls->nl_sock, &nls->nl_cache)) {
  91                         err = -ENOMEM;
  92                         goto err;
  93                 }
  94
  95                 nls->nl80211 = genl_ctrl_search_by_name(nls->nl_cache, "nl80211");
  96                 if (!nls->nl80211) {
  97                         err = -ENOENT;
  98                         goto err;
  99                 }
 100
 101                 nls->nlctrl = genl_ctrl_search_by_name(nls->nl_cache, "nlctrl");
 102                 if (!nls->nlctrl) {
 103                         err = -ENOENT;
 104                         goto err;
 105                 }
 106         }
 107
 108         return 0;
 109
 110
 111 err:
 112         nl80211_close();
 113         return err;
 114 }
 115
 116 static int nl80211_readint(const char *path)
 117 {
 118         int fd;
 119         int rv = -1;
 120         char buffer[16];
 121
 122         if ((fd = open(path, O_RDONLY)) > -1)
 123         {
 124                 if (read(fd, buffer, sizeof(buffer)) > 0)
 125                         rv = atoi(buffer);
 126
 127                 close(fd);
 128         }
 129
 130         return rv;
 131 }
 132
 133 static int nl80211_readstr(const char *path, char *buffer, int length)
 134 {
 135         int fd;
 136         int rv = -1;
 137
 138         if ((fd = open(path, O_RDONLY)) > -1)
 139         {
 140                 if ((rv = read(fd, buffer, length - 1)) > 0)
 141                 {
 142                         if (buffer[rv - 1] == '\n')
 143                                 rv--;
 144
 145                         buffer[rv] = 0;
 146                 }
 147
 148                 close(fd);
 149         }
 150
 151         return rv;
 152 }
 153
 154
 155 static int nl80211_msg_error(struct sockaddr_nl *nla,
 156         struct nlmsgerr *err, void *arg)
 157 {
 158         int *ret = arg;
 159         *ret = err->error;
 160         return NL_STOP;
 161 }
 162
 163 static int nl80211_msg_finish(struct nl_msg *msg, void *arg)
 164 {
 165         int *ret = arg;
 166         *ret = 0;
 167         return NL_SKIP;
 168 }
 169
 170 static int nl80211_msg_ack(struct nl_msg *msg, void *arg)
 171 {
 172         int *ret = arg;
 173         *ret = 0;
 174         return NL_STOP;
 175 }
 176
 177 static int nl80211_msg_response(struct nl_msg *msg, void *arg)
 178 {
 179         return NL_SKIP;
 180 }
 181
 182 static void nl80211_free(struct nl80211_msg_conveyor *cv)
 183 {
 184         if (cv)
 185         {
 186                 if (cv->cb)
 187                         nl_cb_put(cv->cb);
 188
 189                 if (cv->msg)
 190                         nlmsg_free(cv->msg);
 191
 192                 cv->cb  = NULL;
 193                 cv->msg = NULL;
 194         }
 195 }
 196
 197 static struct nl80211_msg_conveyor * nl80211_new(struct genl_family *family,
 198                                                  int cmd, int flags)
 199 {
 200         static struct nl80211_msg_conveyor cv;
 201
 202         struct nl_msg *req = NULL;
 203         struct nl_cb *cb = NULL;
 204
 205         req = nlmsg_alloc();
 206         if (!req)
 207                 goto err;
 208
 209         cb = nl_cb_alloc(NL_CB_DEFAULT);
 210         if (!cb)
 211                 goto err;
 212
 213         genlmsg_put(req, 0, 0, genl_family_get_id(family), 0, flags, cmd, 0);
 214
 215         cv.msg = req;
 216         cv.cb  = cb;
 217
 218         return &cv;
 219
 220 err:
 221         if (req)
 222                 nlmsg_free(req);
 223
 224         return NULL;
 225 }
 226
 227 static struct nl80211_msg_conveyor * nl80211_ctl(int cmd, int flags)
 228 {
 229         if (nl80211_init() < 0)
 230                 return NULL;
 231
 232         return nl80211_new(nls->nlctrl, cmd, flags);
 233 }
 234
 235 static int nl80211_phy_idx_from_uci_path(struct uci_section *s)
 236 {
 237         const char *opt;
 238         char buf[128];
 239         int idx = -1;
 240         glob_t gl;
 241
 242         opt = uci_lookup_option_string(uci_ctx, s, "path");
 243         if (!opt)
 244                 return -1;
 245
 246         snprintf(buf, sizeof(buf), "/sys/devices/%s/ieee80211/*/index", opt);  /**/
 247         if (glob(buf, 0, NULL, &gl))
 248                 snprintf(buf, sizeof(buf), "/sys/devices/platform/%s/ieee80211/*/index", opt);  /**/
 249
 250         if (glob(buf, 0, NULL, &gl))
 251                 return -1;
 252
 253         if (gl.gl_pathc > 0)
 254                 idx = nl80211_readint(gl.gl_pathv[0]);
 255
 256         globfree(&gl);
 257
 258         return idx;
 259 }
 260
 261 static int nl80211_phy_idx_from_uci_macaddr(struct uci_section *s)
 262 {
 263         const char *opt;
 264         char buf[128];
 265         int i, idx = -1;
 266         glob_t gl;
 267
 268         opt = uci_lookup_option_string(uci_ctx, s, "macaddr");
 269         if (!opt)
 270                 return -1;
 271
 272         snprintf(buf, sizeof(buf), "/sys/class/ieee80211/*");   /**/
 273         if (glob(buf, 0, NULL, &gl))
 274                 return -1;
 275
 276         for (i = 0; i < gl.gl_pathc; i++)
 277         {
 278                 snprintf(buf, sizeof(buf), "%s/macaddress", gl.gl_pathv[i]);
 279                 if (nl80211_readstr(buf, buf, sizeof(buf)) <= 0)
 280                         continue;
 281
 282                 if (fnmatch(opt, buf, FNM_CASEFOLD))
 283                         continue;
 284
 285                 snprintf(buf, sizeof(buf), "%s/index", gl.gl_pathv[i]);
 286                 if ((idx = nl80211_readint(buf)) > -1)
 287                         break;
 288         }
 289
 290         globfree(&gl);
 291
 292         return idx;
 293 }
 294
 295 static int nl80211_phy_idx_from_uci_phy(struct uci_section *s)
 296 {
 297         const char *opt;
 298         char buf[128];
 299
 300         opt = uci_lookup_option_string(uci_ctx, s, "phy");
 301         if (!opt)
 302                 return -1;
 303
 304         snprintf(buf, sizeof(buf), "/sys/class/ieee80211/%s/index", opt);
 305         return nl80211_readint(buf);
 306 }
 307
 308 static int nl80211_phy_idx_from_uci(const char *name)
 309 {
 310         struct uci_section *s;
 311         int idx = -1;
 312
 313         s = iwinfo_uci_get_radio(name, "mac80211");
 314         if (!s)
 315                 goto free;
 316
 317         idx = nl80211_phy_idx_from_uci_path(s);
 318
 319         if (idx < 0)
 320                 idx = nl80211_phy_idx_from_uci_macaddr(s);
 321
 322         if (idx < 0)
 323                 idx = nl80211_phy_idx_from_uci_phy(s);
 324
 325 free:
 326         iwinfo_uci_free();
 327         return idx;
 328 }
 329
 330 static struct nl80211_msg_conveyor * nl80211_msg(const char *ifname,
 331                                                  int cmd, int flags)
 332 {
 333         int ifidx = -1, phyidx = -1;
 334         struct nl80211_msg_conveyor *cv;
 335
 336         if (ifname == NULL)
 337                 return NULL;
 338
 339         if (nl80211_init() < 0)
 340                 return NULL;
 341
 342         if (!strncmp(ifname, "phy", 3))
 343                 phyidx = atoi(&ifname[3]);
 344         else if (!strncmp(ifname, "radio", 5))
 345                 phyidx = nl80211_phy_idx_from_uci(ifname);
 346         else if (!strncmp(ifname, "mon.", 4))
 347                 ifidx = if_nametoindex(&ifname[4]);
 348         else
 349                 ifidx = if_nametoindex(ifname);
 350
 351         /* Valid ifidx must be greater than 0 */
 352         if ((ifidx <= 0) && (phyidx < 0))
 353                 return NULL;
 354
 355         cv = nl80211_new(nls->nl80211, cmd, flags);
 356         if (!cv)
 357                 return NULL;
 358
 359         if (ifidx > -1)
 360                 NLA_PUT_U32(cv->msg, NL80211_ATTR_IFINDEX, ifidx);
 361
 362         if (phyidx > -1)
 363                 NLA_PUT_U32(cv->msg, NL80211_ATTR_WIPHY, phyidx);
 364
 365         return cv;
 366
 367 nla_put_failure:
 368         nl80211_free(cv);
 369         return NULL;
 370 }
 371
 372 static int nl80211_send(struct nl80211_msg_conveyor *cv,
 373                         int (*cb_func)(struct nl_msg *, void *),
 374                         void *cb_arg)
 375 {
 376         static struct nl80211_msg_conveyor rcv;
 377         int err;
 378
 379         if (cb_func)
 380                 nl_cb_set(cv->cb, NL_CB_VALID, NL_CB_CUSTOM, cb_func, cb_arg);
 381         else
 382                 nl_cb_set(cv->cb, NL_CB_VALID, NL_CB_CUSTOM, nl80211_msg_response, &rcv);
 383
 384         err = nl_send_auto_complete(nls->nl_sock, cv->msg);
 385
 386         if (err < 0)
 387                 goto out;
 388
 389         err = 1;
 390
 391         nl_cb_err(cv->cb,               NL_CB_CUSTOM, nl80211_msg_error,  &err);
 392         nl_cb_set(cv->cb, NL_CB_FINISH, NL_CB_CUSTOM, nl80211_msg_finish, &err);
 393         nl_cb_set(cv->cb, NL_CB_ACK,    NL_CB_CUSTOM, nl80211_msg_ack,    &err);
 394
 395         while (err > 0)
 396                 nl_recvmsgs(nls->nl_sock, cv->cb);
 397
 398 out:
 399         nl80211_free(cv);
 400         return err;
 401 }
 402
 403 static int nl80211_request(const char *ifname, int cmd, int flags,
 404                            int (*cb_func)(struct nl_msg *, void *),
 405                            void *cb_arg)
 406 {
 407         struct nl80211_msg_conveyor *cv;
 408
 409         cv = nl80211_msg(ifname, cmd, flags);
 410
 411         if (!cv)
 412                 return -ENOMEM;
 413
 414         return nl80211_send(cv, cb_func, cb_arg);
 415 }
 416
 417 static struct nlattr ** nl80211_parse(struct nl_msg *msg)
 418 {
 419         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
 420         static struct nlattr *attr[NL80211_ATTR_MAX + 1];
 421
 422         nla_parse(attr, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
 423                   genlmsg_attrlen(gnlh, 0), NULL);
 424
 425         return attr;
 426 }
 427
 428 static int nl80211_get_protocol_features_cb(struct nl_msg *msg, void *arg)
 429 {
 430         uint32_t *features = arg;
 431         struct nlattr **attr = nl80211_parse(msg);
 432
 433         if (attr[NL80211_ATTR_PROTOCOL_FEATURES])
 434                 *features = nla_get_u32(attr[NL80211_ATTR_PROTOCOL_FEATURES]);
 435
 436         return NL_SKIP;
 437 }
 438
 439 static int nl80211_get_protocol_features(const char *ifname)
 440 {
 441         struct nl80211_msg_conveyor *req;
 442         uint32_t features = 0;
 443
 444         req = nl80211_msg(ifname, NL80211_CMD_GET_PROTOCOL_FEATURES, 0);
 445         if (req) {
 446                 nl80211_send(req, nl80211_get_protocol_features_cb, &features);
 447                 nl80211_free(req);
 448         }
 449
 450         return features;
 451 }
 452
 453 static int nl80211_subscribe_cb(struct nl_msg *msg, void *arg)
 454 {
 455         struct nl80211_group_conveyor *cv = arg;
 456
 457         struct nlattr **attr = nl80211_parse(msg);
 458         struct nlattr *mgrpinfo[CTRL_ATTR_MCAST_GRP_MAX + 1];
 459         struct nlattr *mgrp;
 460         int mgrpidx;
 461
 462         if (!attr[CTRL_ATTR_MCAST_GROUPS])
 463                 return NL_SKIP;
 464
 465         nla_for_each_nested(mgrp, attr[CTRL_ATTR_MCAST_GROUPS], mgrpidx)
 466         {
 467                 nla_parse(mgrpinfo, CTRL_ATTR_MCAST_GRP_MAX,
 468                           nla_data(mgrp), nla_len(mgrp), NULL);
 469
 470                 if (mgrpinfo[CTRL_ATTR_MCAST_GRP_ID] &&
 471                     mgrpinfo[CTRL_ATTR_MCAST_GRP_NAME] &&
 472                     !strncmp(nla_data(mgrpinfo[CTRL_ATTR_MCAST_GRP_NAME]),
 473                              cv->name, nla_len(mgrpinfo[CTRL_ATTR_MCAST_GRP_NAME])))
 474                 {
 475                         cv->id = nla_get_u32(mgrpinfo[CTRL_ATTR_MCAST_GRP_ID]);
 476                         break;
 477                 }
 478         }
 479
 480         return NL_SKIP;
 481 }
 482
 483 static int nl80211_subscribe(const char *family, const char *group)
 484 {
 485         struct nl80211_group_conveyor cv = { .name = group, .id = -ENOENT };
 486         struct nl80211_msg_conveyor *req;
 487         int err;
 488
 489         req = nl80211_ctl(CTRL_CMD_GETFAMILY, 0);
 490         if (req)
 491         {
 492                 NLA_PUT_STRING(req->msg, CTRL_ATTR_FAMILY_NAME, family);
 493                 err = nl80211_send(req, nl80211_subscribe_cb, &cv);
 494
 495                 if (err)
 496                         return err;
 497
 498                 return nl_socket_add_membership(nls->nl_sock, cv.id);
 499
 500 nla_put_failure:
 501                 nl80211_free(req);
 502         }
 503
 504         return -ENOMEM;
 505 }
 506
 507
 508 static int nl80211_wait_cb(struct nl_msg *msg, void *arg)
 509 {
 510         struct nl80211_event_conveyor *cv = arg;
 511         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
 512
 513         if (cv->wait[gnlh->cmd / 32] & (1 << (gnlh->cmd % 32)))
 514                 cv->recv = gnlh->cmd;
 515
 516         return NL_SKIP;
 517 }
 518
 519 static int nl80211_wait_seq_check(struct nl_msg *msg, void *arg)
 520 {
 521         return NL_OK;
 522 }
 523
 524 static int __nl80211_wait(const char *family, const char *group, ...)
 525 {
 526         struct nl80211_event_conveyor cv = { };
 527         struct nl_cb *cb;
 528         int err = 0;
 529         int cmd;
 530         va_list ap;
 531
 532         if (nl80211_subscribe(family, group))
 533                 return -ENOENT;
 534
 535         cb = nl_cb_alloc(NL_CB_DEFAULT);
 536
 537         if (!cb)
 538                 return -ENOMEM;
 539
 540         nl_cb_err(cb,                  NL_CB_CUSTOM, nl80211_msg_error,      &err);
 541         nl_cb_set(cb, NL_CB_SEQ_CHECK, NL_CB_CUSTOM, nl80211_wait_seq_check, NULL);
 542         nl_cb_set(cb, NL_CB_VALID,     NL_CB_CUSTOM, nl80211_wait_cb,        &cv );
 543
 544         va_start(ap, group);
 545
 546         for (cmd = va_arg(ap, int); cmd != 0; cmd = va_arg(ap, int))
 547                 cv.wait[cmd / 32] |= (1 << (cmd % 32));
 548
 549         va_end(ap);
 550
 551         while (!cv.recv && !err)
 552                 nl_recvmsgs(nls->nl_sock, cb);
 553
 554         nl_cb_put(cb);
 555
 556         return err;
 557 }
 558
 559 #define nl80211_wait(family, group, ...) \
 560         __nl80211_wait(family, group, __VA_ARGS__, 0)
 561
 562
 563 static int nl80211_freq2channel(int freq)
 564 {
 565         if (freq == 2484)
 566                 return 14;
 567         else if (freq < 2484)
 568                 return (freq - 2407) / 5;
 569         else if (freq >= 4910 && freq <= 4980)
 570                 return (freq - 4000) / 5;
 571         else
 572                 return (freq - 5000) / 5;
 573 }
 574
 575 static int nl80211_channel2freq(int channel, const char *band)
 576 {
 577         if (!band || band[0] != 'a')
 578         {
 579                 if (channel == 14)
 580                         return 2484;
 581                 else if (channel < 14)
 582                         return (channel * 5) + 2407;
 583         }
 584         else
 585         {
 586                 if (channel >= 182 && channel <= 196)
 587                         return (channel * 5) + 4000;
 588                 else
 589                         return (channel * 5) + 5000;
 590         }
 591
 592         return 0;
 593 }
 594
 595 static int nl80211_ifname2phy_cb(struct nl_msg *msg, void *arg)
 596 {
 597         char *buf = arg;
 598         struct nlattr **attr = nl80211_parse(msg);
 599
 600         if (attr[NL80211_ATTR_WIPHY_NAME])
 601                 memcpy(buf, nla_data(attr[NL80211_ATTR_WIPHY_NAME]),
 602                        nla_len(attr[NL80211_ATTR_WIPHY_NAME]));
 603         else
 604                 buf[0] = 0;
 605
 606         return NL_SKIP;
 607 }
 608
 609 static char * nl80211_ifname2phy(const char *ifname)
 610 {
 611         static char phy[32] = { 0 };
 612
 613         memset(phy, 0, sizeof(phy));
 614
 615         nl80211_request(ifname, NL80211_CMD_GET_WIPHY, 0,
 616                         nl80211_ifname2phy_cb, phy);
 617
 618         return phy[0] ? phy : NULL;
 619 }
 620
 621 static char * nl80211_phy2ifname(const char *ifname)
 622 {
 623         int ifidx = -1, cifidx = -1, phyidx = -1;
 624         char buffer[64];
 625         static char nif[IFNAMSIZ] = { 0 };
 626
 627         DIR *d;
 628         struct dirent *e;
 629
 630         /* Only accept phy name of the form phy%d or radio%d */
 631         if (!ifname)
 632                 return NULL;
 633         else if (!strncmp(ifname, "phy", 3))
 634                 phyidx = atoi(&ifname[3]);
 635         else if (!strncmp(ifname, "radio", 5))
 636                 phyidx = nl80211_phy_idx_from_uci(ifname);
 637         else
 638                 return NULL;
 639
 640         memset(nif, 0, sizeof(nif));
 641
 642         if (phyidx > -1)
 643         {
 644                 if ((d = opendir("/sys/class/net")) != NULL)
 645                 {
 646                         while ((e = readdir(d)) != NULL)
 647                         {
 648                                 snprintf(buffer, sizeof(buffer),
 649                                          "/sys/class/net/%s/phy80211/index", e->d_name);
 650
 651                                 if (nl80211_readint(buffer) == phyidx)
 652                                 {
 653                                         snprintf(buffer, sizeof(buffer),
 654                                                  "/sys/class/net/%s/ifindex", e->d_name);
 655
 656                                         if ((cifidx = nl80211_readint(buffer)) >= 0 &&
 657                                             ((ifidx < 0) || (cifidx < ifidx)))
 658                                         {
 659                                                 ifidx = cifidx;
 660                                                 strncpy(nif, e->d_name, sizeof(nif) - 1);
 661                                         }
 662                                 }
 663                         }
 664
 665                         closedir(d);
 666                 }
 667         }
 668
 669         return nif[0] ? nif : NULL;
 670 }
 671
 672 static int nl80211_get_mode_cb(struct nl_msg *msg, void *arg)
 673 {
 674         int *mode = arg;
 675         struct nlattr **tb = nl80211_parse(msg);
 676         const int ifmodes[NL80211_IFTYPE_MAX + 1] = {
 677                 IWINFO_OPMODE_UNKNOWN,          /* unspecified */
 678                 IWINFO_OPMODE_ADHOC,            /* IBSS */
 679                 IWINFO_OPMODE_CLIENT,           /* managed */
 680                 IWINFO_OPMODE_MASTER,           /* AP */
 681                 IWINFO_OPMODE_AP_VLAN,          /* AP/VLAN */
 682                 IWINFO_OPMODE_WDS,                      /* WDS */
 683                 IWINFO_OPMODE_MONITOR,          /* monitor */
 684                 IWINFO_OPMODE_MESHPOINT,        /* mesh point */
 685                 IWINFO_OPMODE_P2P_CLIENT,       /* P2P-client */
 686                 IWINFO_OPMODE_P2P_GO,           /* P2P-GO */
 687         };
 688
 689         if (tb[NL80211_ATTR_IFTYPE])
 690                 *mode = ifmodes[nla_get_u32(tb[NL80211_ATTR_IFTYPE])];
 691
 692         return NL_SKIP;
 693 }
 694
 695
 696 static int nl80211_get_mode(const char *ifname, int *buf)
 697 {
 698         char *res;
 699
 700         *buf = IWINFO_OPMODE_UNKNOWN;
 701
 702         res = nl80211_phy2ifname(ifname);
 703
 704         nl80211_request(res ? res : ifname, NL80211_CMD_GET_INTERFACE, 0,
 705                         nl80211_get_mode_cb, buf);
 706
 707         return (*buf == IWINFO_OPMODE_UNKNOWN) ? -1 : 0;
 708 }
 709
 710 static int __nl80211_hostapd_query(const char *ifname, ...)
 711 {
 712         va_list ap, ap_cur;
 713         char *phy, *search, *dest, *key, *val, buf[128];
 714         int len, mode, found = 0, match = 1;
 715         FILE *fp;
 716
 717         if (nl80211_get_mode(ifname, &mode))
 718                 return 0;
 719
 720         if (mode != IWINFO_OPMODE_MASTER && mode != IWINFO_OPMODE_AP_VLAN)
 721                 return 0;
 722
 723         phy = nl80211_ifname2phy(ifname);
 724
 725         if (!phy)
 726                 return 0;
 727
 728         snprintf(buf, sizeof(buf), "/var/run/hostapd-%s.conf", phy);
 729         fp = fopen(buf, "r");
 730
 731         if (!fp)
 732                 return 0;
 733
 734         va_start(ap, ifname);
 735
 736         /* clear all destination buffers */
 737         va_copy(ap_cur, ap);
 738
 739         while ((search = va_arg(ap_cur, char *)) != NULL)
 740         {
 741                 dest = va_arg(ap_cur, char *);
 742                 len  = va_arg(ap_cur, int);
 743
 744                 memset(dest, 0, len);
 745         }
 746
 747         va_end(ap_cur);
 748
 749         /* iterate applicable lines and copy found values into dest buffers */
 750         while (fgets(buf, sizeof(buf), fp))
 751         {
 752                 key = strtok(buf, " =\t\n");
 753                 val = strtok(NULL, "\n");
 754
 755                 if (!key || !val || !*key || *key == '#')
 756                         continue;
 757
 758                 if (!strcmp(key, "interface") || !strcmp(key, "bss"))
 759                         match = !strcmp(ifname, val);
 760
 761                 if (!match)
 762                         continue;
 763
 764                 va_copy(ap_cur, ap);
 765
 766                 while ((search = va_arg(ap_cur, char *)) != NULL)
 767                 {
 768                         dest = va_arg(ap_cur, char *);
 769                         len  = va_arg(ap_cur, int);
 770
 771                         if (!strcmp(search, key))
 772                         {
 773                                 strncpy(dest, val, len - 1);
 774                                 found++;
 775                                 break;
 776                         }
 777                 }
 778
 779                 va_end(ap_cur);
 780         }
 781
 782         fclose(fp);
 783
 784         va_end(ap);
 785
 786         return found;
 787 }
 788
 789 #define nl80211_hostapd_query(ifname, ...) \
 790         __nl80211_hostapd_query(ifname, ##__VA_ARGS__, NULL)
 791
 792
 793 static inline int nl80211_wpactl_recv(int sock, char *buf, int blen)
 794 {
 795         fd_set rfds;
 796         struct timeval tv = { 0, 256000 };
 797
 798         FD_ZERO(&rfds);
 799         FD_SET(sock, &rfds);
 800
 801         memset(buf, 0, blen);
 802
 803         if (select(sock + 1, &rfds, NULL, NULL, &tv) < 0)
 804                 return -1;
 805
 806         if (!FD_ISSET(sock, &rfds))
 807                 return -1;
 808
 809         return recv(sock, buf, blen - 1, 0);
 810 }
 811
 812 static int nl80211_wpactl_connect(const char *ifname, struct sockaddr_un *local)
 813 {
 814         struct sockaddr_un remote = { 0 };
 815         size_t remote_length, local_length;
 816
 817         int sock = socket(PF_UNIX, SOCK_DGRAM, 0);
 818         if (sock < 0)
 819                 return sock;
 820
 821         remote.sun_family = AF_UNIX;
 822         remote_length = sizeof(remote.sun_family) +
 823                 sprintf(remote.sun_path, "/var/run/wpa_supplicant-%s/%s",
 824                         ifname, ifname);
 825
 826         if (fcntl(sock, F_SETFD, fcntl(sock, F_GETFD) | FD_CLOEXEC) < 0)
 827         {
 828                 close(sock);
 829                 return -1;
 830         }
 831
 832         if (connect(sock, (struct sockaddr *)&remote, remote_length))
 833         {
 834                 remote_length = sizeof(remote.sun_family) +
 835                         sprintf(remote.sun_path, "/var/run/wpa_supplicant/%s", ifname);
 836
 837                 if (connect(sock, (struct sockaddr *)&remote, remote_length))
 838                 {
 839                         close(sock);
 840                         return -1;
 841                 }
 842         }
 843
 844         local->sun_family = AF_UNIX;
 845         local_length = sizeof(local->sun_family) +
 846                 sprintf(local->sun_path, "/var/run/iwinfo-%s-%d", ifname, getpid());
 847
 848         if (bind(sock, (struct sockaddr *)local, local_length) < 0)
 849         {
 850                 close(sock);
 851                 return -1;
 852         }
 853
 854         return sock;
 855 }
 856
 857 static int __nl80211_wpactl_query(const char *ifname, ...)
 858 {
 859         va_list ap, ap_cur;
 860         struct sockaddr_un local = { 0 };
 861         int len, mode, found = 0, sock = -1;
 862         char *search, *dest, *key, *val, *line, *pos, buf[512];
 863
 864         if (nl80211_get_mode(ifname, &mode))
 865                 return 0;
 866
 867         if (mode != IWINFO_OPMODE_CLIENT && mode != IWINFO_OPMODE_ADHOC)
 868                 return 0;
 869
 870         sock = nl80211_wpactl_connect(ifname, &local);
 871
 872         if (sock < 0)
 873                 return 0;
 874
 875         va_start(ap, ifname);
 876
 877         /* clear all destination buffers */
 878         va_copy(ap_cur, ap);
 879
 880         while ((search = va_arg(ap_cur, char *)) != NULL)
 881         {
 882                 dest = va_arg(ap_cur, char *);
 883                 len  = va_arg(ap_cur, int);
 884
 885                 memset(dest, 0, len);
 886         }
 887
 888         va_end(ap_cur);
 889
 890         send(sock, "STATUS", 6, 0);
 891
 892         while (true)
 893         {
 894                 if (nl80211_wpactl_recv(sock, buf, sizeof(buf)) <= 0)
 895                         break;
 896
 897                 if (buf[0] == '<')
 898                         continue;
 899
 900                 for (line = strtok_r(buf, "\n", &pos);
 901                          line != NULL;
 902                          line = strtok_r(NULL, "\n", &pos))
 903                 {
 904                         key = strtok(line, "=");
 905                         val = strtok(NULL, "\n");
 906
 907                         if (!key || !val)
 908                                 continue;
 909
 910                         va_copy(ap_cur, ap);
 911
 912                         while ((search = va_arg(ap_cur, char *)) != NULL)
 913                         {
 914                                 dest = va_arg(ap_cur, char *);
 915                                 len  = va_arg(ap_cur, int);
 916
 917                                 if (!strcmp(search, key))
 918                                 {
 919                                         strncpy(dest, val, len - 1);
 920                                         found++;
 921                                         break;
 922                                 }
 923                         }
 924
 925                         va_end(ap_cur);
 926                 }
 927
 928                 break;
 929         }
 930
 931         va_end(ap);
 932
 933         close(sock);
 934         unlink(local.sun_path);
 935
 936         return found;
 937 }
 938
 939 #define nl80211_wpactl_query(ifname, ...) \
 940         __nl80211_wpactl_query(ifname, ##__VA_ARGS__, NULL)
 941
 942
 943 static char * nl80211_ifadd(const char *ifname)
 944 {
 945         char path[PATH_MAX];
 946         static char nif[IFNAMSIZ] = { 0 };
 947         struct nl80211_msg_conveyor *req;
 948         FILE *sysfs;
 949
 950         req = nl80211_msg(ifname, NL80211_CMD_NEW_INTERFACE, 0);
 951         if (req)
 952         {
 953                 snprintf(nif, sizeof(nif), "tmp.%s", ifname);
 954
 955                 NLA_PUT_STRING(req->msg, NL80211_ATTR_IFNAME, nif);
 956                 NLA_PUT_U32(req->msg, NL80211_ATTR_IFTYPE, NL80211_IFTYPE_STATION);
 957
 958                 nl80211_send(req, NULL, NULL);
 959
 960                 snprintf(path, sizeof(path) - 1,
 961                          "/proc/sys/net/ipv6/conf/%s/disable_ipv6", nif);
 962
 963                 if ((sysfs = fopen(path, "w")) != NULL)
 964                 {
 965                         fwrite("0\n", 1, 2, sysfs);
 966                         fclose(sysfs);
 967                 }
 968
 969                 return nif;
 970
 971         nla_put_failure:
 972                 nl80211_free(req);
 973         }
 974
 975         return NULL;
 976 }
 977
 978 static void nl80211_ifdel(const char *ifname)
 979 {
 980         struct nl80211_msg_conveyor *req;
 981         int err;
 982
 983         req = nl80211_msg(ifname, NL80211_CMD_DEL_INTERFACE, 0);
 984         if (req)
 985         {
 986                 NLA_PUT_STRING(req->msg, NL80211_ATTR_IFNAME, ifname);
 987
 988                 nl80211_send(req, NULL, NULL);
 989                 return;
 990
 991         nla_put_failure:
 992                 nl80211_free(req);
 993         }
 994 }
 995
 996 static void nl80211_hostapd_hup(const char *ifname)
 997 {
 998         int fd, pid = 0;
 999         char buf[32];
1000         char *phy = nl80211_ifname2phy(ifname);
1001
1002         if (phy)
1003         {
1004                 snprintf(buf, sizeof(buf), "/var/run/wifi-%s.pid", phy);
1005                 if ((fd = open(buf, O_RDONLY)) >= 0)
1006                 {
1007                         if (read(fd, buf, sizeof(buf)) > 0)
1008                                 pid = atoi(buf);
1009
1010                         close(fd);
1011                 }
1012
1013                 if (pid > 0)
1014                         kill(pid, 1);
1015         }
1016 }
1017
1018
1019 static int nl80211_probe(const char *ifname)
1020 {
1021         return !!nl80211_ifname2phy(ifname);
1022 }
1023
1024 struct nl80211_ssid_bssid {
1025         unsigned char *ssid;
1026         unsigned char bssid[7];
1027 };
1028
1029 static int nl80211_get_ssid_bssid_cb(struct nl_msg *msg, void *arg)
1030 {
1031         int ielen;
1032         unsigned char *ie;
1033         struct nl80211_ssid_bssid *sb = arg;
1034         struct nlattr **tb = nl80211_parse(msg);
1035         struct nlattr *bss[NL80211_BSS_MAX + 1];
1036
1037         static struct nla_policy bss_policy[NL80211_BSS_MAX + 1] = {
1038                 [NL80211_BSS_INFORMATION_ELEMENTS] = { 0 },
1039                 [NL80211_BSS_STATUS]               = { .type = NLA_U32 },
1040         };
1041
1042         if (!tb[NL80211_ATTR_BSS] ||
1043             nla_parse_nested(bss, NL80211_BSS_MAX, tb[NL80211_ATTR_BSS],
1044                              bss_policy) ||
1045             !bss[NL80211_BSS_BSSID] ||
1046             !bss[NL80211_BSS_STATUS] ||
1047             !bss[NL80211_BSS_INFORMATION_ELEMENTS])
1048         {
1049                 return NL_SKIP;
1050         }
1051
1052         switch (nla_get_u32(bss[NL80211_BSS_STATUS]))
1053         {
1054         case NL80211_BSS_STATUS_ASSOCIATED:
1055         case NL80211_BSS_STATUS_AUTHENTICATED:
1056         case NL80211_BSS_STATUS_IBSS_JOINED:
1057
1058                 if (sb->ssid)
1059                 {
1060                         ie = nla_data(bss[NL80211_BSS_INFORMATION_ELEMENTS]);
1061                         ielen = nla_len(bss[NL80211_BSS_INFORMATION_ELEMENTS]);
1062
1063                         while (ielen >= 2 && ielen >= ie[1])
1064                         {
1065                                 if (ie[0] == 0)
1066                                 {
1067                                         memcpy(sb->ssid, ie + 2, min(ie[1], IWINFO_ESSID_MAX_SIZE));
1068                                         return NL_SKIP;
1069                                 }
1070
1071                                 ielen -= ie[1] + 2;
1072                                 ie += ie[1] + 2;
1073                         }
1074                 }
1075                 else
1076                 {
1077                         sb->bssid[0] = 1;
1078                         memcpy(sb->bssid + 1, nla_data(bss[NL80211_BSS_BSSID]), 6);
1079                         return NL_SKIP;
1080                 }
1081
1082         default:
1083                 return NL_SKIP;
1084         }
1085 }
1086
1087 static int nl80211_get_ssid(const char *ifname, char *buf)
1088 {
1089         char *res;
1090         struct nl80211_ssid_bssid sb = { .ssid = (unsigned char *)buf };
1091
1092         /* try to find ssid from scan dump results */
1093         res = nl80211_phy2ifname(ifname);
1094         sb.ssid[0] = 0;
1095
1096         nl80211_request(res ? res : ifname, NL80211_CMD_GET_SCAN, NLM_F_DUMP,
1097                         nl80211_get_ssid_bssid_cb, &sb);
1098
1099         /* failed, try to find from hostapd info */
1100         if (sb.ssid[0] == 0)
1101                 nl80211_hostapd_query(ifname, "ssid", sb.ssid,
1102                                       IWINFO_ESSID_MAX_SIZE + 1);
1103
1104         return (sb.ssid[0] == 0) ? -1 : 0;
1105 }
1106
1107 static int nl80211_get_bssid(const char *ifname, char *buf)
1108 {
1109         char *res, bssid[sizeof("FF:FF:FF:FF:FF:FF\0")];
1110         struct nl80211_ssid_bssid sb = { };
1111
1112         /* try to find bssid from scan dump results */
1113         res = nl80211_phy2ifname(ifname);
1114
1115         nl80211_request(res ? res : ifname, NL80211_CMD_GET_SCAN, NLM_F_DUMP,
1116                         nl80211_get_ssid_bssid_cb, &sb);
1117
1118         /* failed, try to find mac from hostapd info */
1119         if ((sb.bssid[0] == 0) &&
1120             nl80211_hostapd_query(ifname, "bssid", bssid, sizeof(bssid)))
1121         {
1122                 sb.bssid[0] = 1;
1123                 sb.bssid[1] = strtol(&bssid[0],  NULL, 16);
1124                 sb.bssid[2] = strtol(&bssid[3],  NULL, 16);
1125                 sb.bssid[3] = strtol(&bssid[6],  NULL, 16);
1126                 sb.bssid[4] = strtol(&bssid[9],  NULL, 16);
1127                 sb.bssid[5] = strtol(&bssid[12], NULL, 16);
1128                 sb.bssid[6] = strtol(&bssid[15], NULL, 16);
1129         }
1130
1131         if (sb.bssid[0])
1132         {
1133                 sprintf(buf, "%02X:%02X:%02X:%02X:%02X:%02X",
1134                         sb.bssid[1], sb.bssid[2], sb.bssid[3],
1135                         sb.bssid[4], sb.bssid[5], sb.bssid[6]);
1136
1137                 return 0;
1138         }
1139
1140         return -1;
1141 }
1142
1143
1144 static int nl80211_get_frequency_scan_cb(struct nl_msg *msg, void *arg)
1145 {
1146         int *freq = arg;
1147         struct nlattr **attr = nl80211_parse(msg);
1148         struct nlattr *binfo[NL80211_BSS_MAX + 1];
1149
1150         static struct nla_policy bss_policy[NL80211_BSS_MAX + 1] = {
1151                 [NL80211_BSS_FREQUENCY] = { .type = NLA_U32 },
1152                 [NL80211_BSS_STATUS]    = { .type = NLA_U32 },
1153         };
1154
1155         if (attr[NL80211_ATTR_BSS] &&
1156             !nla_parse_nested(binfo, NL80211_BSS_MAX,
1157                               attr[NL80211_ATTR_BSS], bss_policy))
1158         {
1159                 if (binfo[NL80211_BSS_STATUS] && binfo[NL80211_BSS_FREQUENCY])
1160                         *freq = nla_get_u32(binfo[NL80211_BSS_FREQUENCY]);
1161         }
1162
1163         return NL_SKIP;
1164 }
1165
1166 static int nl80211_get_frequency_info_cb(struct nl_msg *msg, void *arg)
1167 {
1168         int *freq = arg;
1169         struct nlattr **tb = nl80211_parse(msg);
1170
1171         if (tb[NL80211_ATTR_WIPHY_FREQ])
1172                 *freq = nla_get_u32(tb[NL80211_ATTR_WIPHY_FREQ]);
1173
1174         return NL_SKIP;
1175 }
1176
1177 static int nl80211_get_frequency(const char *ifname, int *buf)
1178 {
1179         char *res, channel[4], hwmode[2];
1180
1181         /* try to find frequency from interface info */
1182         res = nl80211_phy2ifname(ifname);
1183         *buf = 0;
1184
1185         nl80211_request(res ? res : ifname, NL80211_CMD_GET_INTERFACE, 0,
1186                         nl80211_get_frequency_info_cb, buf);
1187
1188         /* failed, try to find frequency from hostapd info */
1189         if ((*buf == 0) &&
1190             nl80211_hostapd_query(ifname, "hw_mode", hwmode, sizeof(hwmode),
1191                                           "channel", channel, sizeof(channel)) == 2)
1192         {
1193                 *buf = nl80211_channel2freq(atoi(channel), hwmode);
1194         }
1195
1196         /* failed, try to find frequency from scan results */
1197         if (*buf == 0)
1198         {
1199                 res = nl80211_phy2ifname(ifname);
1200
1201                 nl80211_request(res ? res : ifname, NL80211_CMD_GET_SCAN, NLM_F_DUMP,
1202                                 nl80211_get_frequency_scan_cb, buf);
1203         }
1204
1205         return (*buf == 0) ? -1 : 0;
1206 }
1207
1208 static int nl80211_get_channel(const char *ifname, int *buf)
1209 {
1210         if (!nl80211_get_frequency(ifname, buf))
1211         {
1212                 *buf = nl80211_freq2channel(*buf);
1213                 return 0;
1214         }
1215
1216         return -1;
1217 }
1218
1219 static int nl80211_get_txpower_cb(struct nl_msg *msg, void *arg)
1220 {
1221         int *buf = arg;
1222         struct nlattr **tb = nl80211_parse(msg);
1223
1224         if (tb[NL80211_ATTR_WIPHY_TX_POWER_LEVEL])
1225                 *buf = iwinfo_mbm2dbm(nla_get_u32(tb[NL80211_ATTR_WIPHY_TX_POWER_LEVEL]));
1226
1227         return NL_SKIP;
1228 }
1229
1230 static int nl80211_get_txpower(const char *ifname, int *buf)
1231 {
1232         char *res;
1233
1234         res = nl80211_phy2ifname(ifname);
1235         *buf = 0;
1236
1237         if (nl80211_request(res ? res : ifname, NL80211_CMD_GET_INTERFACE, 0,
1238                             nl80211_get_txpower_cb, buf))
1239                 return -1;
1240
1241         return 0;
1242 }
1243
1244
1245 static int nl80211_fill_signal_cb(struct nl_msg *msg, void *arg)
1246 {
1247         int8_t dbm;
1248         int16_t mbit;
1249         struct nl80211_rssi_rate *rr = arg;
1250         struct nlattr **attr = nl80211_parse(msg);
1251         struct nlattr *sinfo[NL80211_STA_INFO_MAX + 1];
1252         struct nlattr *rinfo[NL80211_RATE_INFO_MAX + 1];
1253
1254         static struct nla_policy stats_policy[NL80211_STA_INFO_MAX + 1] = {
1255                 [NL80211_STA_INFO_INACTIVE_TIME] = { .type = NLA_U32    },
1256                 [NL80211_STA_INFO_RX_BYTES]      = { .type = NLA_U32    },
1257                 [NL80211_STA_INFO_TX_BYTES]      = { .type = NLA_U32    },
1258                 [NL80211_STA_INFO_RX_PACKETS]    = { .type = NLA_U32    },
1259                 [NL80211_STA_INFO_TX_PACKETS]    = { .type = NLA_U32    },
1260                 [NL80211_STA_INFO_SIGNAL]        = { .type = NLA_U8     },
1261                 [NL80211_STA_INFO_TX_BITRATE]    = { .type = NLA_NESTED },
1262                 [NL80211_STA_INFO_LLID]          = { .type = NLA_U16    },
1263                 [NL80211_STA_INFO_PLID]          = { .type = NLA_U16    },
1264                 [NL80211_STA_INFO_PLINK_STATE]   = { .type = NLA_U8     },
1265         };
1266
1267         static struct nla_policy rate_policy[NL80211_RATE_INFO_MAX + 1] = {
1268                 [NL80211_RATE_INFO_BITRATE]      = { .type = NLA_U16  },
1269                 [NL80211_RATE_INFO_MCS]          = { .type = NLA_U8   },
1270                 [NL80211_RATE_INFO_40_MHZ_WIDTH] = { .type = NLA_FLAG },
1271                 [NL80211_RATE_INFO_SHORT_GI]     = { .type = NLA_FLAG },
1272         };
1273
1274         if (attr[NL80211_ATTR_STA_INFO])
1275         {
1276                 if (!nla_parse_nested(sinfo, NL80211_STA_INFO_MAX,
1277                                       attr[NL80211_ATTR_STA_INFO], stats_policy))
1278                 {
1279                         if (sinfo[NL80211_STA_INFO_SIGNAL])
1280                         {
1281                                 dbm = nla_get_u8(sinfo[NL80211_STA_INFO_SIGNAL]);
1282                                 rr->rssi = rr->rssi ? (int8_t)((rr->rssi + dbm) / 2) : dbm;
1283                         }
1284
1285                         if (sinfo[NL80211_STA_INFO_TX_BITRATE])
1286                         {
1287                                 if (!nla_parse_nested(rinfo, NL80211_RATE_INFO_MAX,
1288                                                       sinfo[NL80211_STA_INFO_TX_BITRATE],
1289                                                       rate_policy))
1290                                 {
1291                                         if (rinfo[NL80211_RATE_INFO_BITRATE])
1292                                         {
1293                                                 mbit = nla_get_u16(rinfo[NL80211_RATE_INFO_BITRATE]);
1294                                                 rr->rate = rr->rate
1295                                                         ? (int16_t)((rr->rate + mbit) / 2) : mbit;
1296                                         }
1297                                 }
1298                         }
1299                 }
1300         }
1301
1302         return NL_SKIP;
1303 }
1304
1305 static void nl80211_fill_signal(const char *ifname, struct nl80211_rssi_rate *r)
1306 {
1307         DIR *d;
1308         struct dirent *de;
1309
1310         r->rssi = 0;
1311         r->rate = 0;
1312
1313         if ((d = opendir("/sys/class/net")) != NULL)
1314         {
1315                 while ((de = readdir(d)) != NULL)
1316                 {
1317                         if (!strncmp(de->d_name, ifname, strlen(ifname)) &&
1318                             (!de->d_name[strlen(ifname)] ||
1319                              !strncmp(&de->d_name[strlen(ifname)], ".sta", 4)))
1320                         {
1321                                 nl80211_request(de->d_name, NL80211_CMD_GET_STATION,
1322                                                 NLM_F_DUMP, nl80211_fill_signal_cb, r);
1323                         }
1324                 }
1325
1326                 closedir(d);
1327         }
1328 }
1329
1330 static int nl80211_get_bitrate(const char *ifname, int *buf)
1331 {
1332         struct nl80211_rssi_rate rr;
1333
1334         nl80211_fill_signal(ifname, &rr);
1335
1336         if (rr.rate)
1337         {
1338                 *buf = (rr.rate * 100);
1339                 return 0;
1340         }
1341
1342         return -1;
1343 }
1344
1345 static int nl80211_get_signal(const char *ifname, int *buf)
1346 {
1347         struct nl80211_rssi_rate rr;
1348
1349         nl80211_fill_signal(ifname, &rr);
1350
1351         if (rr.rssi)
1352         {
1353                 *buf = rr.rssi;
1354                 return 0;
1355         }
1356
1357         return -1;
1358 }
1359
1360 static int nl80211_get_noise_cb(struct nl_msg *msg, void *arg)
1361 {
1362         int8_t *noise = arg;
1363         struct nlattr **tb = nl80211_parse(msg);
1364         struct nlattr *si[NL80211_SURVEY_INFO_MAX + 1];
1365
1366         static struct nla_policy sp[NL80211_SURVEY_INFO_MAX + 1] = {
1367                 [NL80211_SURVEY_INFO_FREQUENCY] = { .type = NLA_U32 },
1368                 [NL80211_SURVEY_INFO_NOISE]     = { .type = NLA_U8  },
1369         };
1370
1371         if (!tb[NL80211_ATTR_SURVEY_INFO])
1372                 return NL_SKIP;
1373
1374         if (nla_parse_nested(si, NL80211_SURVEY_INFO_MAX,
1375                              tb[NL80211_ATTR_SURVEY_INFO], sp))
1376                 return NL_SKIP;
1377
1378         if (!si[NL80211_SURVEY_INFO_NOISE])
1379                 return NL_SKIP;
1380
1381         if (!*noise || si[NL80211_SURVEY_INFO_IN_USE])
1382                 *noise = (int8_t)nla_get_u8(si[NL80211_SURVEY_INFO_NOISE]);
1383
1384         return NL_SKIP;
1385 }
1386
1387
1388 static int nl80211_get_noise(const char *ifname, int *buf)
1389 {
1390         int8_t noise = 0;
1391
1392         if (nl80211_request(ifname, NL80211_CMD_GET_SURVEY, NLM_F_DUMP,
1393                             nl80211_get_noise_cb, &noise))
1394                 goto out;
1395
1396         *buf = noise;
1397         return 0;
1398
1399 out:
1400         *buf = 0;
1401         return -1;
1402 }
1403
1404 static int nl80211_get_quality(const char *ifname, int *buf)
1405 {
1406         int signal;
1407
1408         if (!nl80211_get_signal(ifname, &signal))
1409         {
1410                 /* A positive signal level is usually just a quality
1411                  * value, pass through as-is */
1412                 if (signal >= 0)
1413                 {
1414                         *buf = signal;
1415                 }
1416
1417                 /* The cfg80211 wext compat layer assumes a signal range
1418                  * of -110 dBm to -40 dBm, the quality value is derived
1419                  * by adding 110 to the signal level */
1420                 else
1421                 {
1422                         if (signal < -110)
1423                                 signal = -110;
1424                         else if (signal > -40)
1425                                 signal = -40;
1426
1427                         *buf = (signal + 110);
1428                 }
1429
1430                 return 0;
1431         }
1432
1433         return -1;
1434 }
1435
1436 static int nl80211_get_quality_max(const char *ifname, int *buf)
1437 {
1438         /* The cfg80211 wext compat layer assumes a maximum
1439          * quality of 70 */
1440         *buf = 70;
1441
1442         return 0;
1443 }
1444
1445 static int nl80211_check_wepkey(const char *key)
1446 {
1447         if (key && *key)
1448         {
1449                 switch (strlen(key))
1450                 {
1451                 case 5:
1452                 case 10:
1453                         return IWINFO_CIPHER_WEP40;
1454
1455                 case 13:
1456                 case 26:
1457                         return IWINFO_CIPHER_WEP104;
1458                 }
1459         }
1460
1461         return 0;
1462 }
1463
1464 static int nl80211_get_encryption(const char *ifname, char *buf)
1465 {
1466         char wpa[2], wpa_key_mgmt[16], wpa_pairwise[16], wpa_groupwise[16];
1467         char auth_algs[2], wep_key0[27], wep_key1[27], wep_key2[27], wep_key3[27];
1468
1469         struct iwinfo_crypto_entry *c = (struct iwinfo_crypto_entry *)buf;
1470
1471         /* WPA supplicant */
1472         if (nl80211_wpactl_query(ifname,
1473                         "pairwise_cipher", wpa_pairwise,  sizeof(wpa_pairwise),
1474                         "group_cipher",    wpa_groupwise, sizeof(wpa_groupwise),
1475                         "key_mgmt",        wpa_key_mgmt,  sizeof(wpa_key_mgmt)))
1476         {
1477                 /* WEP */
1478                 if (!strcmp(wpa_key_mgmt, "NONE"))
1479                 {
1480                         if (strstr(wpa_pairwise, "WEP-40"))
1481                                 c->pair_ciphers |= IWINFO_CIPHER_WEP40;
1482                         else if (strstr(wpa_pairwise, "WEP-104"))
1483                                 c->pair_ciphers |= IWINFO_CIPHER_WEP104;
1484
1485                         if (strstr(wpa_groupwise, "WEP-40"))
1486                                 c->group_ciphers |= IWINFO_CIPHER_WEP40;
1487                         else if (strstr(wpa_groupwise, "WEP-104"))
1488                                 c->group_ciphers |= IWINFO_CIPHER_WEP104;
1489
1490                         c->enabled      = !!(c->pair_ciphers | c->group_ciphers);
1491                         c->auth_suites |= IWINFO_KMGMT_NONE;
1492                         c->auth_algs   |= IWINFO_AUTH_OPEN; /* XXX: assumption */
1493                 }
1494
1495                 /* WPA */
1496                 else if (strstr(wpa_key_mgmt, "WPA"))
1497                 {
1498                         if (strstr(wpa_pairwise, "TKIP"))
1499                                 c->pair_ciphers |= IWINFO_CIPHER_TKIP;
1500                         else if (strstr(wpa_pairwise, "CCMP"))
1501                                 c->pair_ciphers |= IWINFO_CIPHER_CCMP;
1502                         else if (strstr(wpa_pairwise, "NONE"))
1503                                 c->pair_ciphers |= IWINFO_CIPHER_NONE;
1504                         else if (strstr(wpa_pairwise, "WEP-40"))
1505                                 c->pair_ciphers |= IWINFO_CIPHER_WEP40;
1506                         else if (strstr(wpa_pairwise, "WEP-104"))
1507                                 c->pair_ciphers |= IWINFO_CIPHER_WEP104;
1508
1509                         if (strstr(wpa_groupwise, "TKIP"))
1510                                 c->group_ciphers |= IWINFO_CIPHER_TKIP;
1511                         else if (strstr(wpa_groupwise, "CCMP"))
1512                                 c->group_ciphers |= IWINFO_CIPHER_CCMP;
1513                         else if (strstr(wpa_groupwise, "NONE"))
1514                                 c->group_ciphers |= IWINFO_CIPHER_NONE;
1515                         else if (strstr(wpa_groupwise, "WEP-40"))
1516                                 c->group_ciphers |= IWINFO_CIPHER_WEP40;
1517                         else if (strstr(wpa_groupwise, "WEP-104"))
1518                                 c->group_ciphers |= IWINFO_CIPHER_WEP104;
1519
1520                         if (strstr(wpa_key_mgmt, "WPA2"))
1521                                 c->wpa_version = 2;
1522                         else if (strstr(wpa_key_mgmt, "WPA"))
1523                                 c->wpa_version = 1;
1524
1525                         if (strstr(wpa_key_mgmt, "PSK"))
1526                                 c->auth_suites |= IWINFO_KMGMT_PSK;
1527                         else if (strstr(wpa_key_mgmt, "EAP") ||
1528                                  strstr(wpa_key_mgmt, "802.1X"))
1529                                 c->auth_suites |= IWINFO_KMGMT_8021x;
1530                         else if (strstr(wpa_key_mgmt, "NONE"))
1531                                 c->auth_suites |= IWINFO_KMGMT_NONE;
1532
1533                         c->enabled = !!(c->wpa_version && c->auth_suites);
1534                 }
1535
1536                 return 0;
1537         }
1538
1539         /* Hostapd */
1540         else if (nl80211_hostapd_query(ifname,
1541                                 "wpa",          wpa,          sizeof(wpa),
1542                                 "wpa_key_mgmt", wpa_key_mgmt, sizeof(wpa_key_mgmt),
1543                                 "wpa_pairwise", wpa_pairwise, sizeof(wpa_pairwise),
1544                                 "auth_algs",    auth_algs,    sizeof(auth_algs),
1545                                 "wep_key0",     wep_key0,     sizeof(wep_key0),
1546                                 "wep_key1",     wep_key1,     sizeof(wep_key1),
1547                                 "wep_key2",     wep_key2,     sizeof(wep_key2),
1548                                 "wep_key3",     wep_key3,     sizeof(wep_key3)))
1549         {
1550                 c->wpa_version = wpa[0] ? atoi(wpa) : 0;
1551
1552                 if (wpa_key_mgmt[0])
1553                 {
1554                         if (strstr(wpa_key_mgmt, "PSK"))
1555                                 c->auth_suites |= IWINFO_KMGMT_PSK;
1556
1557                         if (strstr(wpa_key_mgmt, "EAP"))
1558                                 c->auth_suites |= IWINFO_KMGMT_8021x;
1559
1560                         if (strstr(wpa_key_mgmt, "NONE"))
1561                                 c->auth_suites |= IWINFO_KMGMT_NONE;
1562                 }
1563                 else
1564                 {
1565                         c->auth_suites |= IWINFO_KMGMT_PSK;
1566                 }
1567
1568                 if (wpa_pairwise[0])
1569                 {
1570                         if (strstr(wpa_pairwise, "TKIP"))
1571                                 c->pair_ciphers |= IWINFO_CIPHER_TKIP;
1572
1573                         if (strstr(wpa_pairwise, "CCMP"))
1574                                 c->pair_ciphers |= IWINFO_CIPHER_CCMP;
1575
1576                         if (strstr(wpa_pairwise, "NONE"))
1577                                 c->pair_ciphers |= IWINFO_CIPHER_NONE;
1578                 }
1579
1580                 if (auth_algs[0])
1581                 {
1582                         switch(atoi(auth_algs))
1583                         {
1584                         case 1:
1585                                 c->auth_algs |= IWINFO_AUTH_OPEN;
1586                                 break;
1587
1588                         case 2:
1589                                 c->auth_algs |= IWINFO_AUTH_SHARED;
1590                                 break;
1591
1592                         case 3:
1593                                 c->auth_algs |= IWINFO_AUTH_OPEN;
1594                                 c->auth_algs |= IWINFO_AUTH_SHARED;
1595                                 break;
1596                         }
1597
1598                         c->pair_ciphers |= nl80211_check_wepkey(wep_key0);
1599                         c->pair_ciphers |= nl80211_check_wepkey(wep_key1);
1600                         c->pair_ciphers |= nl80211_check_wepkey(wep_key2);
1601                         c->pair_ciphers |= nl80211_check_wepkey(wep_key3);
1602                 }
1603
1604                 c->group_ciphers = c->pair_ciphers;
1605                 c->enabled = (c->wpa_version || c->pair_ciphers) ? 1 : 0;
1606
1607                 return 0;
1608         }
1609
1610         return -1;
1611 }
1612
1613 static int nl80211_get_phyname(const char *ifname, char *buf)
1614 {
1615         const char *name;
1616
1617         name = nl80211_ifname2phy(ifname);
1618
1619         if (name)
1620         {
1621                 strcpy(buf, name);
1622                 return 0;
1623         }
1624         else if ((name = nl80211_phy2ifname(ifname)) != NULL)
1625         {
1626                 name = nl80211_ifname2phy(name);
1627
1628                 if (name)
1629                 {
1630                         strcpy(buf, ifname);
1631                         return 0;
1632                 }
1633         }
1634
1635         return -1;
1636 }
1637
1638
1639 static void nl80211_parse_rateinfo(struct nlattr **ri,
1640                                    struct iwinfo_rate_entry *re)
1641 {
1642         if (ri[NL80211_RATE_INFO_BITRATE32])
1643                 re->rate = nla_get_u32(ri[NL80211_RATE_INFO_BITRATE32]) * 100;
1644         else if (ri[NL80211_RATE_INFO_BITRATE])
1645                 re->rate = nla_get_u16(ri[NL80211_RATE_INFO_BITRATE]) * 100;
1646
1647         if (ri[NL80211_RATE_INFO_VHT_MCS])
1648         {
1649                 re->is_vht = 1;
1650                 re->mcs = nla_get_u8(ri[NL80211_RATE_INFO_VHT_MCS]);
1651
1652                 if (ri[NL80211_RATE_INFO_VHT_NSS])
1653                         re->nss = nla_get_u8(ri[NL80211_RATE_INFO_VHT_NSS]);
1654         }
1655         else if (ri[NL80211_RATE_INFO_MCS])
1656         {
1657                 re->is_ht = 1;
1658                 re->mcs = nla_get_u8(ri[NL80211_RATE_INFO_MCS]);
1659         }
1660
1661         if (ri[NL80211_RATE_INFO_5_MHZ_WIDTH])
1662                 re->mhz = 5;
1663         else if (ri[NL80211_RATE_INFO_10_MHZ_WIDTH])
1664                 re->mhz = 10;
1665         else if (ri[NL80211_RATE_INFO_40_MHZ_WIDTH])
1666                 re->mhz = 40;
1667         else if (ri[NL80211_RATE_INFO_80_MHZ_WIDTH])
1668                 re->mhz = 80;
1669         else if (ri[NL80211_RATE_INFO_80P80_MHZ_WIDTH] ||
1670                  ri[NL80211_RATE_INFO_160_MHZ_WIDTH])
1671                 re->mhz = 160;
1672         else
1673                 re->mhz = 20;
1674
1675         if (ri[NL80211_RATE_INFO_SHORT_GI])
1676                 re->is_short_gi = 1;
1677
1678         re->is_40mhz = (re->mhz == 40);
1679 }
1680
1681 static int nl80211_get_assoclist_cb(struct nl_msg *msg, void *arg)
1682 {
1683         struct nl80211_array_buf *arr = arg;
1684         struct iwinfo_assoclist_entry *e = arr->buf;
1685         struct nlattr **attr = nl80211_parse(msg);
1686         struct nlattr *sinfo[NL80211_STA_INFO_MAX + 1];
1687         struct nlattr *rinfo[NL80211_RATE_INFO_MAX + 1];
1688         struct nl80211_sta_flag_update *sta_flags;
1689
1690         static struct nla_policy stats_policy[NL80211_STA_INFO_MAX + 1] = {
1691                 [NL80211_STA_INFO_INACTIVE_TIME] = { .type = NLA_U32    },
1692                 [NL80211_STA_INFO_RX_PACKETS]    = { .type = NLA_U32    },
1693                 [NL80211_STA_INFO_TX_PACKETS]    = { .type = NLA_U32    },
1694                 [NL80211_STA_INFO_RX_BITRATE]    = { .type = NLA_NESTED },
1695                 [NL80211_STA_INFO_TX_BITRATE]    = { .type = NLA_NESTED },
1696                 [NL80211_STA_INFO_SIGNAL]        = { .type = NLA_U8     },
1697                 [NL80211_STA_INFO_RX_BYTES]      = { .type = NLA_U32    },
1698                 [NL80211_STA_INFO_TX_BYTES]      = { .type = NLA_U32    },
1699                 [NL80211_STA_INFO_TX_RETRIES]    = { .type = NLA_U32    },
1700                 [NL80211_STA_INFO_TX_FAILED]     = { .type = NLA_U32    },
1701                 [NL80211_STA_INFO_T_OFFSET]      = { .type = NLA_U64    },
1702                 [NL80211_STA_INFO_STA_FLAGS] =
1703                         { .minlen = sizeof(struct nl80211_sta_flag_update) },
1704         };
1705
1706         static struct nla_policy rate_policy[NL80211_RATE_INFO_MAX + 1] = {
1707                 [NL80211_RATE_INFO_BITRATE]      = { .type = NLA_U16    },
1708                 [NL80211_RATE_INFO_MCS]          = { .type = NLA_U8     },
1709                 [NL80211_RATE_INFO_40_MHZ_WIDTH] = { .type = NLA_FLAG   },
1710                 [NL80211_RATE_INFO_SHORT_GI]     = { .type = NLA_FLAG   },
1711         };
1712
1713         /* advance to end of array */
1714         e += arr->count;
1715         memset(e, 0, sizeof(*e));
1716
1717         if (attr[NL80211_ATTR_MAC])
1718                 memcpy(e->mac, nla_data(attr[NL80211_ATTR_MAC]), 6);
1719
1720         if (attr[NL80211_ATTR_STA_INFO] &&
1721             !nla_parse_nested(sinfo, NL80211_STA_INFO_MAX,
1722                               attr[NL80211_ATTR_STA_INFO], stats_policy))
1723         {
1724                 if (sinfo[NL80211_STA_INFO_SIGNAL])
1725                         e->signal = nla_get_u8(sinfo[NL80211_STA_INFO_SIGNAL]);
1726
1727                 if (sinfo[NL80211_STA_INFO_INACTIVE_TIME])
1728                         e->inactive = nla_get_u32(sinfo[NL80211_STA_INFO_INACTIVE_TIME]);
1729
1730                 if (sinfo[NL80211_STA_INFO_RX_PACKETS])
1731                         e->rx_packets = nla_get_u32(sinfo[NL80211_STA_INFO_RX_PACKETS]);
1732
1733                 if (sinfo[NL80211_STA_INFO_TX_PACKETS])
1734                         e->tx_packets = nla_get_u32(sinfo[NL80211_STA_INFO_TX_PACKETS]);
1735
1736                 if (sinfo[NL80211_STA_INFO_RX_BITRATE] &&
1737                     !nla_parse_nested(rinfo, NL80211_RATE_INFO_MAX,
1738                                       sinfo[NL80211_STA_INFO_RX_BITRATE], rate_policy))
1739                         nl80211_parse_rateinfo(rinfo, &e->rx_rate);
1740
1741                 if (sinfo[NL80211_STA_INFO_TX_BITRATE] &&
1742                     !nla_parse_nested(rinfo, NL80211_RATE_INFO_MAX,
1743                                       sinfo[NL80211_STA_INFO_TX_BITRATE], rate_policy))
1744                         nl80211_parse_rateinfo(rinfo, &e->tx_rate);
1745
1746                 if (sinfo[NL80211_STA_INFO_RX_BYTES])
1747                         e->rx_bytes = nla_get_u32(sinfo[NL80211_STA_INFO_RX_BYTES]);
1748
1749                 if (sinfo[NL80211_STA_INFO_TX_BYTES])
1750                         e->tx_bytes = nla_get_u32(sinfo[NL80211_STA_INFO_TX_BYTES]);
1751
1752                 if (sinfo[NL80211_STA_INFO_TX_RETRIES])
1753                         e->tx_retries = nla_get_u32(sinfo[NL80211_STA_INFO_TX_RETRIES]);
1754
1755                 if (sinfo[NL80211_STA_INFO_TX_FAILED])
1756                         e->tx_failed = nla_get_u32(sinfo[NL80211_STA_INFO_TX_FAILED]);
1757
1758                 if (sinfo[NL80211_STA_INFO_T_OFFSET])
1759                         e->t_offset = nla_get_u64(sinfo[NL80211_STA_INFO_T_OFFSET]);
1760
1761                 /* Station flags */
1762                 if (sinfo[NL80211_STA_INFO_STA_FLAGS])
1763                 {
1764                         sta_flags = (struct nl80211_sta_flag_update *)
1765                                 nla_data(sinfo[NL80211_STA_INFO_STA_FLAGS]);
1766
1767                         if (sta_flags->mask & BIT(NL80211_STA_FLAG_AUTHORIZED) &&
1768                             sta_flags->set & BIT(NL80211_STA_FLAG_AUTHORIZED))
1769                                 e->is_authorized = 1;
1770
1771                         if (sta_flags->mask & BIT(NL80211_STA_FLAG_AUTHENTICATED) &&
1772                             sta_flags->set & BIT(NL80211_STA_FLAG_AUTHENTICATED))
1773                                 e->is_authenticated = 1;
1774
1775                         if (sta_flags->mask & BIT(NL80211_STA_FLAG_SHORT_PREAMBLE) &&
1776                             sta_flags->set & BIT(NL80211_STA_FLAG_SHORT_PREAMBLE))
1777                                 e->is_preamble_short = 1;
1778
1779                         if (sta_flags->mask & BIT(NL80211_STA_FLAG_WME) &&
1780                             sta_flags->set & BIT(NL80211_STA_FLAG_WME))
1781                                 e->is_wme = 1;
1782
1783                         if (sta_flags->mask & BIT(NL80211_STA_FLAG_MFP) &&
1784                             sta_flags->set & BIT(NL80211_STA_FLAG_MFP))
1785                                 e->is_mfp = 1;
1786
1787                         if (sta_flags->mask & BIT(NL80211_STA_FLAG_TDLS_PEER) &&
1788                             sta_flags->set & BIT(NL80211_STA_FLAG_TDLS_PEER))
1789                                 e->is_tdls = 1;
1790                 }
1791         }
1792
1793         e->noise = 0; /* filled in by caller */
1794         arr->count++;
1795
1796         return NL_SKIP;
1797 }
1798
1799 static int nl80211_get_assoclist(const char *ifname, char *buf, int *len)
1800 {
1801         DIR *d;
1802         int i, noise = 0;
1803         struct dirent *de;
1804         struct nl80211_array_buf arr = { .buf = buf, .count = 0 };
1805         struct iwinfo_assoclist_entry *e;
1806
1807         if ((d = opendir("/sys/class/net")) != NULL)
1808         {
1809                 while ((de = readdir(d)) != NULL)
1810                 {
1811                         if (!strncmp(de->d_name, ifname, strlen(ifname)) &&
1812                             (!de->d_name[strlen(ifname)] ||
1813                              !strncmp(&de->d_name[strlen(ifname)], ".sta", 4)))
1814                         {
1815                                 nl80211_request(de->d_name, NL80211_CMD_GET_STATION,
1816                                                 NLM_F_DUMP, nl80211_get_assoclist_cb, &arr);
1817                         }
1818                 }
1819
1820                 closedir(d);
1821
1822                 if (!nl80211_get_noise(ifname, &noise))
1823                         for (i = 0, e = arr.buf; i < arr.count; i++, e++)
1824                                 e->noise = noise;
1825
1826                 *len = (arr.count * sizeof(struct iwinfo_assoclist_entry));
1827                 return 0;
1828         }
1829
1830         return -1;
1831 }
1832
1833 static int nl80211_get_txpwrlist_cb(struct nl_msg *msg, void *arg)
1834 {
1835         int *dbm_max = arg;
1836         int ch_cur, ch_cmp, bands_remain, freqs_remain;
1837
1838         struct nlattr **attr = nl80211_parse(msg);
1839         struct nlattr *bands[NL80211_BAND_ATTR_MAX + 1];
1840         struct nlattr *freqs[NL80211_FREQUENCY_ATTR_MAX + 1];
1841         struct nlattr *band, *freq;
1842
1843         static struct nla_policy freq_policy[NL80211_FREQUENCY_ATTR_MAX + 1] = {
1844                 [NL80211_FREQUENCY_ATTR_FREQ]         = { .type = NLA_U32  },
1845                 [NL80211_FREQUENCY_ATTR_DISABLED]     = { .type = NLA_FLAG },
1846                 [NL80211_FREQUENCY_ATTR_PASSIVE_SCAN] = { .type = NLA_FLAG },
1847                 [NL80211_FREQUENCY_ATTR_NO_IBSS]      = { .type = NLA_FLAG },
1848                 [NL80211_FREQUENCY_ATTR_RADAR]        = { .type = NLA_FLAG },
1849                 [NL80211_FREQUENCY_ATTR_MAX_TX_POWER] = { .type = NLA_U32  },
1850         };
1851
1852         ch_cur = *dbm_max; /* value int* is initialized with channel by caller */
1853         *dbm_max = -1;
1854
1855         nla_for_each_nested(band, attr[NL80211_ATTR_WIPHY_BANDS], bands_remain)
1856         {
1857                 nla_parse(bands, NL80211_BAND_ATTR_MAX, nla_data(band),
1858                           nla_len(band), NULL);
1859
1860                 nla_for_each_nested(freq, bands[NL80211_BAND_ATTR_FREQS], freqs_remain)
1861                 {
1862                         nla_parse(freqs, NL80211_FREQUENCY_ATTR_MAX,
1863                                   nla_data(freq), nla_len(freq), freq_policy);
1864
1865                         ch_cmp = nl80211_freq2channel(nla_get_u32(
1866                                 freqs[NL80211_FREQUENCY_ATTR_FREQ]));
1867
1868                         if ((!ch_cur || (ch_cmp == ch_cur)) &&
1869                             freqs[NL80211_FREQUENCY_ATTR_MAX_TX_POWER])
1870                         {
1871                                 *dbm_max = (int)(0.01 * nla_get_u32(
1872                                         freqs[NL80211_FREQUENCY_ATTR_MAX_TX_POWER]));
1873
1874                                 break;
1875                         }
1876                 }
1877         }
1878
1879         return NL_SKIP;
1880 }
1881
1882 static int nl80211_get_txpwrlist(const char *ifname, char *buf, int *len)
1883 {
1884         int err, ch_cur;
1885         int dbm_max = -1, dbm_cur, dbm_cnt;
1886         struct nl80211_msg_conveyor *req;
1887         struct iwinfo_txpwrlist_entry entry;
1888
1889         if (nl80211_get_channel(ifname, &ch_cur))
1890                 ch_cur = 0;
1891
1892         /* initialize the value pointer with channel for callback */
1893         dbm_max = ch_cur;
1894
1895         err = nl80211_request(ifname, NL80211_CMD_GET_WIPHY, 0,
1896                               nl80211_get_txpwrlist_cb, &dbm_max);
1897
1898         if (!err)
1899         {
1900                 for (dbm_cur = 0, dbm_cnt = 0;
1901                      dbm_cur < dbm_max;
1902                      dbm_cur++, dbm_cnt++)
1903                 {
1904                         entry.dbm = dbm_cur;
1905                         entry.mw  = iwinfo_dbm2mw(dbm_cur);
1906
1907                         memcpy(&buf[dbm_cnt * sizeof(entry)], &entry, sizeof(entry));
1908                 }
1909
1910                 entry.dbm = dbm_max;
1911                 entry.mw  = iwinfo_dbm2mw(dbm_max);
1912
1913                 memcpy(&buf[dbm_cnt * sizeof(entry)], &entry, sizeof(entry));
1914                 dbm_cnt++;
1915
1916                 *len = dbm_cnt * sizeof(entry);
1917                 return 0;
1918         }
1919
1920         return -1;
1921 }
1922
1923 static void nl80211_get_scancrypto(const char *spec,
1924         struct iwinfo_crypto_entry *c)
1925 {
1926         if (strstr(spec, "WPA") || strstr(spec, "WEP"))
1927         {
1928                 c->enabled = 1;
1929
1930                 if (strstr(spec, "WPA2-") && strstr(spec, "WPA-"))
1931                         c->wpa_version = 3;
1932
1933                 else if (strstr(spec, "WPA2"))
1934                         c->wpa_version = 2;
1935
1936                 else if (strstr(spec, "WPA"))
1937                         c->wpa_version = 1;
1938
1939                 else if (strstr(spec, "WEP"))
1940                         c->auth_algs = IWINFO_AUTH_OPEN | IWINFO_AUTH_SHARED;
1941
1942
1943                 if (strstr(spec, "PSK"))
1944                         c->auth_suites |= IWINFO_KMGMT_PSK;
1945
1946                 if (strstr(spec, "802.1X") || strstr(spec, "EAP"))
1947                         c->auth_suites |= IWINFO_KMGMT_8021x;
1948
1949                 if (strstr(spec, "WPA-NONE"))
1950                         c->auth_suites |= IWINFO_KMGMT_NONE;
1951
1952
1953                 if (strstr(spec, "TKIP"))
1954                         c->pair_ciphers |= IWINFO_CIPHER_TKIP;
1955
1956                 if (strstr(spec, "CCMP"))
1957                         c->pair_ciphers |= IWINFO_CIPHER_CCMP;
1958
1959                 if (strstr(spec, "WEP-40"))
1960                         c->pair_ciphers |= IWINFO_CIPHER_WEP40;
1961
1962                 if (strstr(spec, "WEP-104"))
1963                         c->pair_ciphers |= IWINFO_CIPHER_WEP104;
1964
1965                 c->group_ciphers = c->pair_ciphers;
1966         }
1967         else
1968         {
1969                 c->enabled = 0;
1970         }
1971 }
1972
1973
1974 struct nl80211_scanlist {
1975         struct iwinfo_scanlist_entry *e;
1976         int len;
1977 };
1978
1979
1980 static void nl80211_get_scanlist_ie(struct nlattr **bss,
1981                                     struct iwinfo_scanlist_entry *e)
1982 {
1983         int ielen = nla_len(bss[NL80211_BSS_INFORMATION_ELEMENTS]);
1984         unsigned char *ie = nla_data(bss[NL80211_BSS_INFORMATION_ELEMENTS]);
1985         static unsigned char ms_oui[3] = { 0x00, 0x50, 0xf2 };
1986         int len;
1987
1988         while (ielen >= 2 && ielen >= ie[1])
1989         {
1990                 switch (ie[0])
1991                 {
1992                 case 0: /* SSID */
1993                         len = min(ie[1], IWINFO_ESSID_MAX_SIZE);
1994                         memcpy(e->ssid, ie + 2, len);
1995                         e->ssid[len] = 0;
1996                         break;
1997
1998                 case 48: /* RSN */
1999                         iwinfo_parse_rsn(&e->crypto, ie + 2, ie[1],
2000                                          IWINFO_CIPHER_CCMP, IWINFO_KMGMT_8021x);
2001                         break;
2002
2003                 case 221: /* Vendor */
2004                         if (ie[1] >= 4 && !memcmp(ie + 2, ms_oui, 3) && ie[5] == 1)
2005                                 iwinfo_parse_rsn(&e->crypto, ie + 6, ie[1] - 4,
2006                                                  IWINFO_CIPHER_TKIP, IWINFO_KMGMT_PSK);
2007                         break;
2008                 }
2009
2010                 ielen -= ie[1] + 2;
2011                 ie += ie[1] + 2;
2012         }
2013 }
2014
2015 static int nl80211_get_scanlist_cb(struct nl_msg *msg, void *arg)
2016 {
2017         int8_t rssi;
2018         uint16_t caps;
2019
2020         struct nl80211_scanlist *sl = arg;
2021         struct nlattr **tb = nl80211_parse(msg);
2022         struct nlattr *bss[NL80211_BSS_MAX + 1];
2023
2024         static struct nla_policy bss_policy[NL80211_BSS_MAX + 1] = {
2025                 [NL80211_BSS_TSF]                  = { .type = NLA_U64 },
2026                 [NL80211_BSS_FREQUENCY]            = { .type = NLA_U32 },
2027                 [NL80211_BSS_BSSID]                = { 0 },
2028                 [NL80211_BSS_BEACON_INTERVAL]      = { .type = NLA_U16 },
2029                 [NL80211_BSS_CAPABILITY]           = { .type = NLA_U16 },
2030                 [NL80211_BSS_INFORMATION_ELEMENTS] = { 0 },
2031                 [NL80211_BSS_SIGNAL_MBM]           = { .type = NLA_U32 },
2032                 [NL80211_BSS_SIGNAL_UNSPEC]        = { .type = NLA_U8  },
2033                 [NL80211_BSS_STATUS]               = { .type = NLA_U32 },
2034                 [NL80211_BSS_SEEN_MS_AGO]          = { .type = NLA_U32 },
2035                 [NL80211_BSS_BEACON_IES]           = { 0 },
2036         };
2037
2038         if (!tb[NL80211_ATTR_BSS] ||
2039                 nla_parse_nested(bss, NL80211_BSS_MAX, tb[NL80211_ATTR_BSS],
2040                                  bss_policy) ||
2041                 !bss[NL80211_BSS_BSSID])
2042         {
2043                 return NL_SKIP;
2044         }
2045
2046         if (bss[NL80211_BSS_CAPABILITY])
2047                 caps = nla_get_u16(bss[NL80211_BSS_CAPABILITY]);
2048         else
2049                 caps = 0;
2050
2051         memset(sl->e, 0, sizeof(*sl->e));
2052         memcpy(sl->e->mac, nla_data(bss[NL80211_BSS_BSSID]), 6);
2053
2054         if (caps & (1<<1))
2055                 sl->e->mode = IWINFO_OPMODE_ADHOC;
2056         else if (caps & (1<<0))
2057                 sl->e->mode = IWINFO_OPMODE_MASTER;
2058         else
2059                 sl->e->mode = IWINFO_OPMODE_MESHPOINT;
2060
2061         if (caps & (1<<4))
2062                 sl->e->crypto.enabled = 1;
2063
2064         if (bss[NL80211_BSS_FREQUENCY])
2065                 sl->e->channel = nl80211_freq2channel(nla_get_u32(
2066                         bss[NL80211_BSS_FREQUENCY]));
2067
2068         if (bss[NL80211_BSS_INFORMATION_ELEMENTS])
2069                 nl80211_get_scanlist_ie(bss, sl->e);
2070
2071         if (bss[NL80211_BSS_SIGNAL_MBM])
2072         {
2073                 sl->e->signal =
2074                         (uint8_t)((int32_t)nla_get_u32(bss[NL80211_BSS_SIGNAL_MBM]) / 100);
2075
2076                 rssi = sl->e->signal - 0x100;
2077
2078                 if (rssi < -110)
2079                         rssi = -110;
2080                 else if (rssi > -40)
2081                         rssi = -40;
2082
2083                 sl->e->quality = (rssi + 110);
2084                 sl->e->quality_max = 70;
2085         }
2086
2087         if (sl->e->crypto.enabled && !sl->e->crypto.wpa_version)
2088         {
2089                 sl->e->crypto.auth_algs    = IWINFO_AUTH_OPEN | IWINFO_AUTH_SHARED;
2090                 sl->e->crypto.pair_ciphers = IWINFO_CIPHER_WEP40 | IWINFO_CIPHER_WEP104;
2091         }
2092
2093         sl->e++;
2094         sl->len++;
2095
2096         return NL_SKIP;
2097 }
2098
2099 static int nl80211_get_scanlist_nl(const char *ifname, char *buf, int *len)
2100 {
2101         struct nl80211_scanlist sl = { .e = (struct iwinfo_scanlist_entry *)buf };
2102
2103         if (nl80211_request(ifname, NL80211_CMD_TRIGGER_SCAN, 0, NULL, NULL))
2104                 goto out;
2105
2106         if (nl80211_wait("nl80211", "scan",
2107                          NL80211_CMD_NEW_SCAN_RESULTS, NL80211_CMD_SCAN_ABORTED))
2108                 goto out;
2109
2110         if (nl80211_request(ifname, NL80211_CMD_GET_SCAN, NLM_F_DUMP,
2111                             nl80211_get_scanlist_cb, &sl))
2112                 goto out;
2113
2114         *len = sl.len * sizeof(struct iwinfo_scanlist_entry);
2115         return 0;
2116
2117 out:
2118         *len = 0;
2119         return -1;
2120 }
2121
2122 static int wpasupp_ssid_decode(const char *in, char *out, int outlen)
2123 {
2124 #define hex(x) \
2125         (((x) >= 'a') ? ((x) - 'a' + 10) : \
2126                 (((x) >= 'A') ? ((x) - 'A' + 10) : ((x) - '0')))
2127
2128         int len = 0;
2129
2130         while (*in)
2131         {
2132                 if (len + 1 >= outlen)
2133                         break;
2134
2135                 switch (*in)
2136                 {
2137                 case '\\':
2138                         in++;
2139                         switch (*in)
2140                         {
2141                         case 'n':
2142                                 out[len++] = '\n'; in++;
2143                                 break;
2144
2145                         case 'r':
2146                                 out[len++] = '\r'; in++;
2147                                 break;
2148
2149                         case 't':
2150                                 out[len++] = '\t'; in++;
2151                                 break;
2152
2153                         case 'e':
2154                                 out[len++] = '\033'; in++;
2155                                 break;
2156
2157                         case 'x':
2158                                 if (isxdigit(*(in+1)) && isxdigit(*(in+2)))
2159                                         out[len++] = hex(*(in+1)) * 16 + hex(*(in+2));
2160                                 in += 3;
2161                                 break;
2162
2163                         default:
2164                                 out[len++] = *in++;
2165                                 break;
2166                         }
2167                         break;
2168
2169                 default:
2170                         out[len++] = *in++;
2171                         break;
2172                 }
2173         }
2174
2175         if (outlen > len)
2176                 out[len] = '\0';
2177
2178         return len;
2179 }
2180
2181 static int nl80211_get_scanlist_wpactl(const char *ifname, char *buf, int *len)
2182 {
2183         int sock, qmax, rssi, tries, count = -1, ready = 0;
2184         char *pos, *line, *bssid, *freq, *signal, *flags, *ssid, reply[4096];
2185         struct sockaddr_un local = { 0 };
2186         struct iwinfo_scanlist_entry *e = (struct iwinfo_scanlist_entry *)buf;
2187
2188         sock = nl80211_wpactl_connect(ifname, &local);
2189
2190         if (sock < 0)
2191                 return sock;
2192
2193         send(sock, "ATTACH", 6, 0);
2194         send(sock, "SCAN", 4, 0);
2195
2196         /*
2197          * wait for scan results:
2198          *   nl80211_wpactl_recv() will use a timeout of 256ms and we need to scan
2199          *   72 channels at most. We'll also receive two "OK" messages acknowledging
2200          *   the "ATTACH" and "SCAN" commands and the driver might need a bit extra
2201          *   time to process the results, so try 72 + 2 + 1 times.
2202          */
2203         for (tries = 0; tries < 75; tries++)
2204         {
2205                 if (nl80211_wpactl_recv(sock, reply, sizeof(reply)) <= 0)
2206                         continue;
2207
2208                 /* got an event notification */
2209                 if (reply[0] == '<')
2210                 {
2211                         /* scan results are ready */
2212                         if (strstr(reply, "CTRL-EVENT-SCAN-RESULTS"))
2213                         {
2214                                 /* send "SCAN_RESULTS" command */
2215                                 ready = (send(sock, "SCAN_RESULTS", 12, 0) == 12);
2216                                 break;
2217                         }
2218
2219                         /* is another unrelated event, retry */
2220                         tries--;
2221                 }
2222         }
2223
2224         /* receive and parse scan results if the wait above didn't time out */
2225         while (ready && nl80211_wpactl_recv(sock, reply, sizeof(reply)) > 0)
2226         {
2227                 /* received an event notification, receive again */
2228                 if (reply[0] == '<')
2229                         continue;
2230
2231                 nl80211_get_quality_max(ifname, &qmax);
2232
2233                 for (line = strtok_r(reply, "\n", &pos);
2234                      line != NULL;
2235                      line = strtok_r(NULL, "\n", &pos))
2236                 {
2237                         /* skip header line */
2238                         if (count < 0)
2239                         {
2240                                 count++;
2241                                 continue;
2242                         }
2243
2244                         bssid  = strtok(line, "\t");
2245                         freq   = strtok(NULL, "\t");
2246                         signal = strtok(NULL, "\t");
2247                         flags  = strtok(NULL, "\t");
2248                         ssid   = strtok(NULL, "\n");
2249
2250                         if (!bssid || !freq || !signal || !flags || !ssid)
2251                                 continue;
2252
2253                         /* BSSID */
2254                         e->mac[0] = strtol(&bssid[0],  NULL, 16);
2255                         e->mac[1] = strtol(&bssid[3],  NULL, 16);
2256                         e->mac[2] = strtol(&bssid[6],  NULL, 16);
2257                         e->mac[3] = strtol(&bssid[9],  NULL, 16);
2258                         e->mac[4] = strtol(&bssid[12], NULL, 16);
2259                         e->mac[5] = strtol(&bssid[15], NULL, 16);
2260
2261                         /* SSID */
2262                         wpasupp_ssid_decode(ssid, e->ssid, sizeof(e->ssid));
2263
2264                         /* Mode */
2265                         if (strstr(flags, "[MESH]"))
2266                                 e->mode = IWINFO_OPMODE_MESHPOINT;
2267                         else if (strstr(flags, "[IBSS]"))
2268                                 e->mode = IWINFO_OPMODE_ADHOC;
2269                         else
2270                                 e->mode = IWINFO_OPMODE_MASTER;
2271
2272                         /* Channel */
2273                         e->channel = nl80211_freq2channel(atoi(freq));
2274
2275                         /* Signal */
2276                         rssi = atoi(signal);
2277                         e->signal = rssi;
2278
2279                         /* Quality */
2280                         if (rssi < 0)
2281                         {
2282                                 /* The cfg80211 wext compat layer assumes a signal range
2283                                  * of -110 dBm to -40 dBm, the quality value is derived
2284                                  * by adding 110 to the signal level */
2285                                 if (rssi < -110)
2286                                         rssi = -110;
2287                                 else if (rssi > -40)
2288                                         rssi = -40;
2289
2290                                 e->quality = (rssi + 110);
2291                         }
2292                         else
2293                         {
2294                                 e->quality = rssi;
2295                         }
2296
2297                         /* Max. Quality */
2298                         e->quality_max = qmax;
2299
2300                         /* Crypto */
2301                         nl80211_get_scancrypto(flags, &e->crypto);
2302
2303                         count++;
2304                         e++;
2305                 }
2306
2307                 *len = count * sizeof(struct iwinfo_scanlist_entry);
2308                 break;
2309         }
2310
2311         close(sock);
2312         unlink(local.sun_path);
2313
2314         return (count >= 0) ? 0 : -1;
2315 }
2316
2317 static int nl80211_get_scanlist(const char *ifname, char *buf, int *len)
2318 {
2319         char *res;
2320         int rv, mode;
2321
2322         *len = 0;
2323
2324         /* Got a radioX pseudo interface, find some interface on it or create one */
2325         if (!strncmp(ifname, "radio", 5))
2326         {
2327                 /* Reuse existing interface */
2328                 if ((res = nl80211_phy2ifname(ifname)) != NULL)
2329                 {
2330                         return nl80211_get_scanlist(res, buf, len);
2331                 }
2332
2333                 /* Need to spawn a temporary iface for scanning */
2334                 else if ((res = nl80211_ifadd(ifname)) != NULL)
2335                 {
2336                         rv = nl80211_get_scanlist(res, buf, len);
2337                         nl80211_ifdel(res);
2338                         return rv;
2339                 }
2340         }
2341
2342         /* WPA supplicant */
2343         if (!nl80211_get_scanlist_wpactl(ifname, buf, len))
2344         {
2345                 return 0;
2346         }
2347
2348         /* station / ad-hoc / monitor scan */
2349         else if (!nl80211_get_mode(ifname, &mode) &&
2350                  (mode == IWINFO_OPMODE_ADHOC ||
2351                   mode == IWINFO_OPMODE_MASTER ||
2352                   mode == IWINFO_OPMODE_CLIENT ||
2353                   mode == IWINFO_OPMODE_MONITOR) &&
2354                  iwinfo_ifup(ifname))
2355         {
2356                 return nl80211_get_scanlist_nl(ifname, buf, len);
2357         }
2358
2359         /* AP scan */
2360         else
2361         {
2362                 /* Got a temp interface, don't create yet another one */
2363                 if (!strncmp(ifname, "tmp.", 4))
2364                 {
2365                         if (!iwinfo_ifup(ifname))
2366                                 return -1;
2367
2368                         rv = nl80211_get_scanlist_nl(ifname, buf, len);
2369                         iwinfo_ifdown(ifname);
2370                         return rv;
2371                 }
2372
2373                 /* Spawn a new scan interface */
2374                 else
2375                 {
2376                         if (!(res = nl80211_ifadd(ifname)))
2377                                 return -1;
2378
2379                         iwinfo_ifmac(res);
2380
2381                         /* if we can take the new interface up, the driver supports an
2382                          * additional interface and there's no need to tear down the ap */
2383                         if (iwinfo_ifup(res))
2384                         {
2385                                 rv = nl80211_get_scanlist_nl(res, buf, len);
2386                                 iwinfo_ifdown(res);
2387                         }
2388
2389                         /* driver cannot create secondary interface, take down ap
2390                          * during scan */
2391                         else if (iwinfo_ifdown(ifname) && iwinfo_ifup(res))
2392                         {
2393                                 rv = nl80211_get_scanlist_nl(res, buf, len);
2394                                 iwinfo_ifdown(res);
2395                                 iwinfo_ifup(ifname);
2396                                 nl80211_hostapd_hup(ifname);
2397                         }
2398
2399                         nl80211_ifdel(res);
2400                         return rv;
2401                 }
2402         }
2403
2404         return -1;
2405 }
2406
2407 static int nl80211_get_freqlist_cb(struct nl_msg *msg, void *arg)
2408 {
2409         int bands_remain, freqs_remain;
2410
2411         struct nl80211_array_buf *arr = arg;
2412         struct iwinfo_freqlist_entry *e;
2413
2414         struct nlattr **attr = nl80211_parse(msg);
2415         struct nlattr *bands[NL80211_BAND_ATTR_MAX + 1];
2416         struct nlattr *freqs[NL80211_FREQUENCY_ATTR_MAX + 1];
2417         struct nlattr *band, *freq;
2418
2419         e = arr->buf;
2420         e += arr->count;
2421
2422         if (attr[NL80211_ATTR_WIPHY_BANDS]) {
2423                 nla_for_each_nested(band, attr[NL80211_ATTR_WIPHY_BANDS], bands_remain)
2424                 {
2425                         nla_parse(bands, NL80211_BAND_ATTR_MAX,
2426                                   nla_data(band), nla_len(band), NULL);
2427
2428                         if (bands[NL80211_BAND_ATTR_FREQS]) {
2429                                 nla_for_each_nested(freq, bands[NL80211_BAND_ATTR_FREQS], freqs_remain)
2430                                 {
2431                                         nla_parse(freqs, NL80211_FREQUENCY_ATTR_MAX,
2432                                                   nla_data(freq), nla_len(freq), NULL);
2433
2434                                         if (!freqs[NL80211_FREQUENCY_ATTR_FREQ] ||
2435                                             freqs[NL80211_FREQUENCY_ATTR_DISABLED])
2436                                                 continue;
2437
2438                                         e->mhz = nla_get_u32(freqs[NL80211_FREQUENCY_ATTR_FREQ]);
2439                                         e->channel = nl80211_freq2channel(e->mhz);
2440
2441                                         e->restricted = (
2442                                                 freqs[NL80211_FREQUENCY_ATTR_NO_IR] &&
2443                                                 !freqs[NL80211_FREQUENCY_ATTR_RADAR]
2444                                         ) ? 1 : 0;
2445
2446                                         if (freqs[NL80211_FREQUENCY_ATTR_NO_HT40_MINUS])
2447                                                 e->flags |= IWINFO_FREQ_NO_HT40MINUS;
2448                                         if (freqs[NL80211_FREQUENCY_ATTR_NO_HT40_PLUS])
2449                                                 e->flags |= IWINFO_FREQ_NO_HT40PLUS;
2450                                         if (freqs[NL80211_FREQUENCY_ATTR_NO_80MHZ])
2451                                                 e->flags |= IWINFO_FREQ_NO_80MHZ;
2452                                         if (freqs[NL80211_FREQUENCY_ATTR_NO_160MHZ])
2453                                                 e->flags |= IWINFO_FREQ_NO_160MHZ;
2454                                         if (freqs[NL80211_FREQUENCY_ATTR_NO_20MHZ])
2455                                                 e->flags |= IWINFO_FREQ_NO_20MHZ;
2456                                         if (freqs[NL80211_FREQUENCY_ATTR_NO_10MHZ])
2457                                                 e->flags |= IWINFO_FREQ_NO_10MHZ;
2458
2459                                         e++;
2460                                         arr->count++;
2461                                 }
2462                         }
2463                 }
2464         }
2465
2466         return NL_SKIP;
2467 }
2468
2469 static int nl80211_get_freqlist(const char *ifname, char *buf, int *len)
2470 {
2471         struct nl80211_msg_conveyor *cv;
2472         struct nl80211_array_buf arr = { .buf = buf, .count = 0 };
2473         uint32_t features = nl80211_get_protocol_features(ifname);
2474         int flags;
2475
2476         flags = features & NL80211_PROTOCOL_FEATURE_SPLIT_WIPHY_DUMP ? NLM_F_DUMP : 0;
2477         cv = nl80211_msg(ifname, NL80211_CMD_GET_WIPHY, flags);
2478         if (!cv)
2479                 goto out;
2480
2481         NLA_PUT_FLAG(cv->msg, NL80211_ATTR_SPLIT_WIPHY_DUMP);
2482         if (nl80211_send(cv, nl80211_get_freqlist_cb, &arr))
2483                 goto out;
2484
2485         *len = arr.count * sizeof(struct iwinfo_freqlist_entry);
2486         return 0;
2487
2488 nla_put_failure:
2489         nl80211_free(cv);
2490 out:
2491         *len = 0;
2492         return -1;
2493 }
2494
2495 static int nl80211_get_country_cb(struct nl_msg *msg, void *arg)
2496 {
2497         char *buf = arg;
2498         struct nlattr **attr = nl80211_parse(msg);
2499
2500         if (attr[NL80211_ATTR_REG_ALPHA2])
2501                 memcpy(buf, nla_data(attr[NL80211_ATTR_REG_ALPHA2]), 2);
2502         else
2503                 buf[0] = 0;
2504
2505         return NL_SKIP;
2506 }
2507
2508 static int nl80211_get_country(const char *ifname, char *buf)
2509 {
2510         if (nl80211_request(ifname, NL80211_CMD_GET_REG, 0,
2511                             nl80211_get_country_cb, buf))
2512                 return -1;
2513
2514         return 0;
2515 }
2516
2517 static int nl80211_get_countrylist(const char *ifname, char *buf, int *len)
2518 {
2519         int count;
2520         struct iwinfo_country_entry *e = (struct iwinfo_country_entry *)buf;
2521         const struct iwinfo_iso3166_label *l;
2522
2523         for (l = IWINFO_ISO3166_NAMES, count = 0; l->iso3166; l++, e++, count++)
2524         {
2525                 e->iso3166 = l->iso3166;
2526                 e->ccode[0] = (l->iso3166 / 256);
2527                 e->ccode[1] = (l->iso3166 % 256);
2528                 e->ccode[2] = 0;
2529         }
2530
2531         *len = (count * sizeof(struct iwinfo_country_entry));
2532         return 0;
2533 }
2534
2535
2536 struct nl80211_modes
2537 {
2538         bool ok;
2539         uint32_t hw;
2540         uint32_t ht;
2541 };
2542
2543 static int nl80211_get_modelist_cb(struct nl_msg *msg, void *arg)
2544 {
2545         struct nl80211_modes *m = arg;
2546         int bands_remain, freqs_remain;
2547         uint16_t caps = 0;
2548         uint32_t vht_caps = 0;
2549         struct nlattr **attr = nl80211_parse(msg);
2550         struct nlattr *bands[NL80211_BAND_ATTR_MAX + 1];
2551         struct nlattr *freqs[NL80211_FREQUENCY_ATTR_MAX + 1];
2552         struct nlattr *band, *freq;
2553
2554         if (attr[NL80211_ATTR_WIPHY_BANDS])
2555         {
2556                 nla_for_each_nested(band, attr[NL80211_ATTR_WIPHY_BANDS], bands_remain)
2557                 {
2558                         nla_parse(bands, NL80211_BAND_ATTR_MAX,
2559                                   nla_data(band), nla_len(band), NULL);
2560
2561                         if (bands[NL80211_BAND_ATTR_HT_CAPA])
2562                                 caps = nla_get_u16(bands[NL80211_BAND_ATTR_HT_CAPA]);
2563
2564                         /* Treat any nonzero capability as 11n */
2565                         if (caps > 0)
2566                         {
2567                                 m->hw |= IWINFO_80211_N;
2568                                 m->ht |= IWINFO_HTMODE_HT20;
2569
2570                                 if (caps & (1 << 1))
2571                                         m->ht |= IWINFO_HTMODE_HT40;
2572                         }
2573
2574                         nla_for_each_nested(freq, bands[NL80211_BAND_ATTR_FREQS],
2575                                             freqs_remain)
2576                         {
2577                                 nla_parse(freqs, NL80211_FREQUENCY_ATTR_MAX,
2578                                           nla_data(freq), nla_len(freq), NULL);
2579
2580                                 if (!freqs[NL80211_FREQUENCY_ATTR_FREQ])
2581                                         continue;
2582
2583                                 if (nla_get_u32(freqs[NL80211_FREQUENCY_ATTR_FREQ]) < 2485)
2584                                 {
2585                                         m->hw |= IWINFO_80211_B;
2586                                         m->hw |= IWINFO_80211_G;
2587                                 }
2588                                 else if (bands[NL80211_BAND_ATTR_VHT_CAPA])
2589                                 {
2590                                         vht_caps = nla_get_u32(bands[NL80211_BAND_ATTR_VHT_CAPA]);
2591
2592                                         /* Treat any nonzero capability as 11ac */
2593                                         if (vht_caps > 0)
2594                                         {
2595                                                 m->hw |= IWINFO_80211_AC;
2596                                                 m->ht |= IWINFO_HTMODE_VHT20 | IWINFO_HTMODE_VHT40 | IWINFO_HTMODE_VHT80;
2597
2598                                                 switch ((vht_caps >> 2) & 3)
2599                                                 {
2600                                                 case 2:
2601                                                         m->ht |= IWINFO_HTMODE_VHT80_80;
2602                                                         /* fall through */
2603
2604                                                 case 1:
2605                                                         m->ht |= IWINFO_HTMODE_VHT160;
2606                                                 }
2607                                         }
2608                                 }
2609                                 else if (!(m->hw & IWINFO_80211_AC))
2610                                 {
2611                                         m->hw |= IWINFO_80211_A;
2612                                 }
2613                         }
2614                 }
2615
2616                 m->ok = 1;
2617         }
2618
2619         return NL_SKIP;
2620 }
2621
2622 static int nl80211_get_hwmodelist(const char *ifname, int *buf)
2623 {
2624         struct nl80211_modes m = { 0 };
2625
2626         if (nl80211_request(ifname, NL80211_CMD_GET_WIPHY, 0,
2627                             nl80211_get_modelist_cb, &m))
2628                 goto out;
2629
2630         if (!m.ok)
2631                 goto out;
2632
2633         *buf = m.hw;
2634         return 0;
2635
2636 out:
2637         *buf = 0;
2638         return -1;
2639 }
2640
2641 static int nl80211_get_htmodelist(const char *ifname, int *buf)
2642 {
2643         struct nl80211_modes m = { 0 };
2644
2645         if (nl80211_request(ifname, NL80211_CMD_GET_WIPHY, 0,
2646                             nl80211_get_modelist_cb, &m))
2647                 goto out;
2648
2649         if (!m.ok)
2650                 goto out;
2651
2652         *buf = m.ht;
2653         return 0;
2654
2655 out:
2656         *buf = 0;
2657         return -1;
2658 }
2659
2660
2661 static int nl80211_get_ifcomb_cb(struct nl_msg *msg, void *arg)
2662 {
2663         struct nlattr **attr = nl80211_parse(msg);
2664         struct nlattr *comb;
2665         int *ret = arg;
2666         int comb_rem, limit_rem, mode_rem;
2667
2668         *ret = 0;
2669         if (!attr[NL80211_ATTR_INTERFACE_COMBINATIONS])
2670                 return NL_SKIP;
2671
2672         nla_for_each_nested(comb, attr[NL80211_ATTR_INTERFACE_COMBINATIONS], comb_rem)
2673         {
2674                 static struct nla_policy iface_combination_policy[NUM_NL80211_IFACE_COMB] = {
2675                         [NL80211_IFACE_COMB_LIMITS] = { .type = NLA_NESTED },
2676                         [NL80211_IFACE_COMB_MAXNUM] = { .type = NLA_U32 },
2677                 };
2678                 struct nlattr *tb_comb[NUM_NL80211_IFACE_COMB+1];
2679                 static struct nla_policy iface_limit_policy[NUM_NL80211_IFACE_LIMIT] = {
2680                         [NL80211_IFACE_LIMIT_TYPES] = { .type = NLA_NESTED },
2681                         [NL80211_IFACE_LIMIT_MAX] = { .type = NLA_U32 },
2682                 };
2683                 struct nlattr *tb_limit[NUM_NL80211_IFACE_LIMIT+1];
2684                 struct nlattr *limit;
2685
2686                 nla_parse_nested(tb_comb, NUM_NL80211_IFACE_COMB, comb, iface_combination_policy);
2687
2688                 if (!tb_comb[NL80211_IFACE_COMB_LIMITS])
2689                         continue;
2690
2691                 nla_for_each_nested(limit, tb_comb[NL80211_IFACE_COMB_LIMITS], limit_rem)
2692                 {
2693                         struct nlattr *mode;
2694
2695                         nla_parse_nested(tb_limit, NUM_NL80211_IFACE_LIMIT, limit, iface_limit_policy);
2696
2697                         if (!tb_limit[NL80211_IFACE_LIMIT_TYPES] ||
2698                             !tb_limit[NL80211_IFACE_LIMIT_MAX])
2699                                 continue;
2700
2701                         if (nla_get_u32(tb_limit[NL80211_IFACE_LIMIT_MAX]) < 2)
2702                                 continue;
2703
2704                         nla_for_each_nested(mode, tb_limit[NL80211_IFACE_LIMIT_TYPES], mode_rem) {
2705                                 if (nla_type(mode) == NL80211_IFTYPE_AP)
2706                                         *ret = 1;
2707                         }
2708                 }
2709         }
2710
2711         return NL_SKIP;
2712 }
2713
2714 static int nl80211_get_mbssid_support(const char *ifname, int *buf)
2715 {
2716         if (nl80211_request(ifname, NL80211_CMD_GET_WIPHY, 0,
2717                             nl80211_get_ifcomb_cb, buf))
2718                 return -1;
2719
2720         return 0;
2721 }
2722
2723 static int nl80211_get_hardware_id(const char *ifname, char *buf)
2724 {
2725         int rv = -1;
2726         char *res;
2727
2728         /* Got a radioX pseudo interface, find some interface on it or create one */
2729         if (!strncmp(ifname, "radio", 5))
2730         {
2731                 /* Reuse existing interface */
2732                 if ((res = nl80211_phy2ifname(ifname)) != NULL)
2733                 {
2734                         rv = wext_ops.hardware_id(res, buf);
2735                 }
2736
2737                 /* Need to spawn a temporary iface for finding IDs */
2738                 else if ((res = nl80211_ifadd(ifname)) != NULL)
2739                 {
2740                         rv = wext_ops.hardware_id(res, buf);
2741                         nl80211_ifdel(res);
2742                 }
2743         }
2744         else
2745         {
2746                 rv = wext_ops.hardware_id(ifname, buf);
2747         }
2748
2749         /* Failed to obtain hardware IDs, search board config */
2750         if (rv)
2751         {
2752                 rv = iwinfo_hardware_id_from_mtd((struct iwinfo_hardware_id *)buf);
2753         }
2754
2755         return rv;
2756 }
2757
2758 static const struct iwinfo_hardware_entry *
2759 nl80211_get_hardware_entry(const char *ifname)
2760 {
2761         struct iwinfo_hardware_id id;
2762
2763         if (nl80211_get_hardware_id(ifname, (char *)&id))
2764                 return NULL;
2765
2766         return iwinfo_hardware(&id);
2767 }
2768
2769 static int nl80211_get_hardware_name(const char *ifname, char *buf)
2770 {
2771         const struct iwinfo_hardware_entry *hw;
2772
2773         if (!(hw = nl80211_get_hardware_entry(ifname)))
2774                 sprintf(buf, "Generic MAC80211");
2775         else
2776                 sprintf(buf, "%s %s", hw->vendor_name, hw->device_name);
2777
2778         return 0;
2779 }
2780
2781 static int nl80211_get_txpower_offset(const char *ifname, int *buf)
2782 {
2783         const struct iwinfo_hardware_entry *hw;
2784
2785         if (!(hw = nl80211_get_hardware_entry(ifname)))
2786                 return -1;
2787
2788         *buf = hw->txpower_offset;
2789         return 0;
2790 }
2791
2792 static int nl80211_get_frequency_offset(const char *ifname, int *buf)
2793 {
2794         const struct iwinfo_hardware_entry *hw;
2795
2796         if (!(hw = nl80211_get_hardware_entry(ifname)))
2797                 return -1;
2798
2799         *buf = hw->frequency_offset;
2800         return 0;
2801 }
2802
2803 static int nl80211_lookup_phyname(const char *section, char *buf)
2804 {
2805         int idx;
2806
2807         if ((idx = nl80211_phy_idx_from_uci(section)) < 0)
2808                 return -1;
2809
2810         sprintf(buf, "phy%d", idx);
2811         return 0;
2812 }
2813
2814 const struct iwinfo_ops nl80211_ops = {
2815         .name             = "nl80211",
2816         .probe            = nl80211_probe,
2817         .channel          = nl80211_get_channel,
2818         .frequency        = nl80211_get_frequency,
2819         .frequency_offset = nl80211_get_frequency_offset,
2820         .txpower          = nl80211_get_txpower,
2821         .txpower_offset   = nl80211_get_txpower_offset,
2822         .bitrate          = nl80211_get_bitrate,
2823         .signal           = nl80211_get_signal,
2824         .noise            = nl80211_get_noise,
2825         .quality          = nl80211_get_quality,
2826         .quality_max      = nl80211_get_quality_max,
2827         .mbssid_support   = nl80211_get_mbssid_support,
2828         .hwmodelist       = nl80211_get_hwmodelist,
2829         .htmodelist       = nl80211_get_htmodelist,
2830         .mode             = nl80211_get_mode,
2831         .ssid             = nl80211_get_ssid,
2832         .bssid            = nl80211_get_bssid,
2833         .country          = nl80211_get_country,
2834         .hardware_id      = nl80211_get_hardware_id,
2835         .hardware_name    = nl80211_get_hardware_name,
2836         .encryption       = nl80211_get_encryption,
2837         .phyname          = nl80211_get_phyname,
2838         .assoclist        = nl80211_get_assoclist,
2839         .txpwrlist        = nl80211_get_txpwrlist,
2840         .scanlist         = nl80211_get_scanlist,
2841         .freqlist         = nl80211_get_freqlist,
2842         .countrylist      = nl80211_get_countrylist,
2843         .lookup_phy       = nl80211_lookup_phyname,
2844         .close            = nl80211_close
2845 };