package/wificonf/wificonf.c

   1 /*
   2  * Wireless Network Adapter configuration utility
   3  *
   4  * Copyright (C) 2005 Felix Fietkau <nbd@openwrt.org>
   5  *
   6  * This program is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU General Public License
   8  * as published by the Free Software Foundation; either version 2
   9  * of the License, or (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  */
  16
  17 #include <stdio.h>
  18 #include <unistd.h>
  19 #include <iwlib.h>
  20 #include <bcmnvram.h>
  21 #include <shutils.h>
  22 #include <wlioctl.h>
  23 #include <signal.h>
  24
  25 #define ADD_VIF_RETRIES 5
  26 // #define DEBUG
  27
  28 /*------------------------------------------------------------------*/
  29 /*
  30  * Macro to handle errors when setting WE
  31  * Print a nice error message and exit...
  32  * We define them as macro so that "return" do the right thing.
  33  * The "do {...} while(0)" is a standard trick
  34  */
  35 #define ERR_SET_EXT(rname, request) \
  36         fprintf(stderr, "Error for wireless request \"%s\" (%X) :\n", \
  37                 rname, request)
  38
  39 #define ABORT_ARG_NUM(rname, request) \
  40         do { \
  41                 ERR_SET_EXT(rname, request); \
  42                 fprintf(stderr, "    too few arguments.\n"); \
  43         } while(0)
  44
  45 #define ABORT_ARG_TYPE(rname, request, arg) \
  46         do { \
  47                 ERR_SET_EXT(rname, request); \
  48                 fprintf(stderr, "    invalid argument \"%s\".\n", arg); \
  49         } while(0)
  50
  51 #define ABORT_ARG_SIZE(rname, request, max) \
  52         do { \
  53                 ERR_SET_EXT(rname, request); \
  54                 fprintf(stderr, "    argument too big (max %d)\n", max); \
  55         } while(0)
  56
  57 /*------------------------------------------------------------------*/
  58 /*
  59  * Wrapper to push some Wireless Parameter in the driver
  60  * Use standard wrapper and add pretty error message if fail...
  61  */
  62 #define IW_SET_EXT_ERR(skfd, ifname, request, wrq, rname) \
  63         do { \
  64         if(iw_set_ext(skfd, ifname, request, wrq) < 0) { \
  65                 ERR_SET_EXT(rname, request); \
  66                 fprintf(stderr, "    SET failed on device %-1.16s ; %s.\n", \
  67                         ifname, strerror(errno)); \
  68         } } while(0)
  69
  70 /*------------------------------------------------------------------*/
  71 /*
  72  * Wrapper to extract some Wireless Parameter out of the driver
  73  * Use standard wrapper and add pretty error message if fail...
  74  */
  75 #define IW_GET_EXT_ERR(skfd, ifname, request, wrq, rname) \
  76         do { \
  77         if(iw_get_ext(skfd, ifname, request, wrq) < 0) { \
  78                 ERR_SET_EXT(rname, request); \
  79                 fprintf(stderr, "    GET failed on device %-1.16s ; %s.\n", \
  80                         ifname, strerror(errno)); \
  81         } } while(0)
  82
  83 static void set_wext_ssid(int skfd, char *ifname);
  84
  85 static char *prefix;
  86 static char buffer[128];
  87 static int wpa_enc = 0;
  88
  89 static char *wl_var(char *name)
  90 {
  91         sprintf(buffer, "%s_%s", prefix, name);
  92         return buffer;
  93 }
  94
  95 static char *vif_var(int vif, char *name)
  96 {
  97         if (vif == 0)
  98                 return wl_var(name);
  99
 100         sprintf(buffer, "%s.%d_%s", prefix, vif, name);
 101         return buffer;
 102 }
 103
 104 static int nvram_enabled(char *name)
 105 {
 106         return (nvram_match(name, "1") || nvram_match(name, "on") || nvram_match(name, "enabled") || nvram_match(name, "true") || nvram_match(name, "yes") ? 1 : 0);
 107 }
 108
 109 static int nvram_disabled(char *name)
 110 {
 111         return (nvram_match(name, "0") || nvram_match(name, "off") || nvram_match(name, "disabled") || nvram_match(name, "false") || nvram_match(name, "no") ? 1 : 0);
 112 }
 113
 114
 115 /* Quarter dBm units to mW
 116  * Table starts at QDBM_OFFSET, so the first entry is mW for qdBm=153
 117  * Table is offset so the last entry is largest mW value that fits in
 118  * a uint16.
 119  */
 120
 121 #define QDBM_OFFSET 153
 122 #define QDBM_TABLE_LEN 40
 123
 124 /* Smallest mW value that will round up to the first table entry, QDBM_OFFSET.
 125  * Value is ( mW(QDBM_OFFSET - 1) + mW(QDBM_OFFSET) ) / 2
 126  */
 127 #define QDBM_TABLE_LOW_BOUND 6493
 128
 129 /* Largest mW value that will round down to the last table entry,
 130  * QDBM_OFFSET + QDBM_TABLE_LEN-1.
 131  * Value is ( mW(QDBM_OFFSET + QDBM_TABLE_LEN - 1) + mW(QDBM_OFFSET + QDBM_TABLE_LEN) ) / 2.
 132  */
 133 #define QDBM_TABLE_HIGH_BOUND 64938
 134
 135 static const uint16 nqdBm_to_mW_map[QDBM_TABLE_LEN] = {
 136 /* qdBm:        +0              +1              +2              +3              +4              +5              +6              +7      */
 137 /* 153: */      6683,   7079,   7499,   7943,   8414,   8913,   9441,   10000,
 138 /* 161: */      10593,  11220,  11885,  12589,  13335,  14125,  14962,  15849,
 139 /* 169: */      16788,  17783,  18836,  19953,  21135,  22387,  23714,  25119,
 140 /* 177: */      26607,  28184,  29854,  31623,  33497,  35481,  37584,  39811,
 141 /* 185: */      42170,  44668,  47315,  50119,  53088,  56234,  59566,  63096
 142 };
 143
 144 unsigned char mw_to_qdbm(uint16 mw)
 145 {
 146         char qdbm;
 147         int offset;
 148         uint mw_uint = mw;
 149         uint boundary;
 150
 151         /* handle boundary case */
 152         if (mw_uint <= 1)
 153                 return 0;
 154
 155         offset = QDBM_OFFSET;
 156
 157         /* move mw into the range of the table */
 158         while (mw_uint < QDBM_TABLE_LOW_BOUND) {
 159                 mw_uint *= 10;
 160                 offset -= 40;
 161         }
 162
 163         for (qdbm = 0; qdbm < QDBM_TABLE_LEN-1; qdbm++) {
 164                 boundary = nqdBm_to_mW_map[qdbm] + (nqdBm_to_mW_map[qdbm+1] - nqdBm_to_mW_map[qdbm])/2;
 165                 if (mw_uint < boundary) break;
 166         }
 167
 168         qdbm += (unsigned char)offset;
 169
 170         return(qdbm);
 171 }
 172
 173 static int bcom_ioctl(int skfd, char *ifname, int cmd, void *buf, int len)
 174 {
 175         struct ifreq ifr;
 176         wl_ioctl_t ioc;
 177         int ret;
 178
 179         ioc.cmd = cmd;
 180         ioc.buf = buf;
 181         ioc.len = len;
 182
 183         ifr.ifr_data = (caddr_t) &ioc;
 184         strncpy(ifr.ifr_name, ifname, IFNAMSIZ);
 185
 186         ret = ioctl(skfd, SIOCDEVPRIVATE, &ifr);
 187
 188 #ifdef DEBUG
 189         if (ret < 0)
 190                 fprintf(stderr, "IOCTL %d failed: %d\n", cmd, ret);
 191 #endif
 192
 193         return ret;
 194 }
 195
 196 static int bcom_set_var(int skfd, char *ifname, char *var, void *val, int len)
 197 {
 198         char buf[8192];
 199         int ret;
 200
 201         if (strlen(var) + 1 > sizeof(buf) || len > sizeof(buf))
 202                 return -1;
 203
 204         bzero(buf, sizeof(buf));
 205         strcpy(buf, var);
 206         memcpy(&buf[strlen(var) + 1], val, len);
 207
 208         ret = bcom_ioctl(skfd, ifname, WLC_SET_VAR, buf, sizeof(buf));
 209
 210 #ifdef DEBUG
 211         if (ret < 0)
 212                 fprintf(stderr, "SET_VAR %s failed: %d\n", var, ret);
 213 #endif
 214
 215         return ret;
 216 }
 217
 218 static int bcom_get_var(int skfd, char *ifname, char *var, void *buf, int len)
 219 {
 220         int ret;
 221
 222         if (strlen(var) + 1 > sizeof(buf) || len > sizeof(buf))
 223                 return -1;
 224
 225         bzero(buf, sizeof(buf));
 226         strcpy(buf, var);
 227
 228         ret = bcom_ioctl(skfd, ifname, WLC_GET_VAR, buf, sizeof(buf));
 229
 230 #ifdef DEBUG
 231         if (ret < 0)
 232                 fprintf(stderr, "GET_VAR %s failed: %d\n", var, ret);
 233 #endif
 234
 235         return ret;
 236 }
 237
 238 static int bcom_set_bss_var(int skfd, char *ifname, int bss, char *var, void *val, int len)
 239 {
 240         char buf[8192];
 241         int i = 0, ret;
 242
 243         bzero(buf, sizeof(buf));
 244         if (strlen(var) + len + 8 > sizeof(buf) || len > sizeof(buf))
 245                 return -1;
 246
 247         // "bsscfg:<name>\x00" <bss> <data>
 248         i = sprintf(buf, "bsscfg:%s", var);
 249         buf[i++] = 0;
 250
 251         memcpy(buf + i, &bss, sizeof(bss));
 252         i += sizeof(bss);
 253
 254         memcpy(buf + i, val, len);
 255         i += len;
 256
 257         ret = bcom_ioctl(skfd, ifname, WLC_SET_VAR, buf, i);
 258
 259 #ifdef DEBUG
 260         if (ret < 0)
 261                 fprintf(stderr, "SET_BSS_VAR %s failed: %d\n", var, ret);
 262 #endif
 263
 264         return ret;
 265 }
 266
 267 static int bcom_set_int(int skfd, char *ifname, char *var, int val)
 268 {
 269         return bcom_set_var(skfd, ifname, var, &val, sizeof(val));
 270 }
 271
 272 static int bcom_set_bss_int(int skfd, char *ifname, int bss, char *var, int val)
 273 {
 274         return bcom_set_bss_var(skfd, ifname, bss, var, &val, sizeof(val));
 275 }
 276
 277 static int is_new_bcom(int skfd, char *ifname)
 278 {
 279         char buf[8192];
 280
 281         bzero(buf, 8192);
 282         bcom_ioctl(skfd, ifname, WLC_DUMP, buf, 8192);
 283
 284         if (strstr(buf, "3.130"))
 285                 return 1;
 286
 287         return 0;
 288 }
 289
 290 static int bcom_get_wsec(int vif)
 291 {
 292         int val;
 293
 294         if (nvram_match(vif_var(vif, "crypto"), "tkip"))
 295                 val = TKIP_ENABLED;
 296         else if (nvram_match(vif_var(vif, "crypto"), "aes"))
 297                 val = AES_ENABLED;
 298         else if (nvram_match(vif_var(vif, "crypto"), "tkip+aes") || nvram_match(vif_var(vif, "crypto"), "aes+tkip"))
 299                 val = TKIP_ENABLED | AES_ENABLED;
 300         else
 301                 val = 0;
 302
 303         return val;
 304 }
 305
 306 static int bcom_get_wauth(int vif)
 307 {
 308         char *v, *next, var[80];
 309         int res = 0;
 310
 311         if (!(v = nvram_get(vif_var(vif, "akm"))))
 312                 v = nvram_safe_get(vif_var(vif, "auth_mode"));
 313
 314         foreach(var, v, next) {
 315                 if (strcmp(var, "psk") == 0)
 316                         res |= WPA_AUTH_PSK;
 317                 else if (strcmp(var, "psk2") == 0)
 318                         res |= WPA2_AUTH_PSK;
 319                 else if (strcmp(var, "wpa") == 0)
 320                         res |= WPA_AUTH_UNSPECIFIED;
 321                 else if (strcmp(var, "wpa2") == 0)
 322                         res |= WPA2_AUTH_UNSPECIFIED;
 323         }
 324
 325         return res;
 326 }
 327
 328 static void stop_bcom(int skfd, char *ifname)
 329 {
 330         int val = 0;
 331         wlc_ssid_t ssid;
 332
 333         if (bcom_ioctl(skfd, ifname, WLC_GET_MAGIC, &val, sizeof(val)) < 0)
 334                 return;
 335
 336         ssid.SSID_len = 0;
 337         ssid.SSID[0] = 0;
 338         bcom_ioctl(skfd, ifname, WLC_SET_SSID, &ssid, sizeof(ssid));
 339         bcom_ioctl(skfd, ifname, WLC_DOWN, NULL, 0);
 340
 341 }
 342
 343 static void start_bcom(int skfd, char *ifname)
 344 {
 345         int val = 0;
 346
 347         if (bcom_ioctl(skfd, ifname, WLC_GET_MAGIC, &val, sizeof(val)) < 0)
 348                 return;
 349
 350         bcom_ioctl(skfd, ifname, WLC_UP, &val, sizeof(val));
 351 }
 352
 353 static int setup_bcom_wds(int skfd, char *ifname)
 354 {
 355         char buf[8192];
 356         char wbuf[80];
 357         char *v;
 358         int wds_enabled = 0;
 359
 360         if (v = nvram_get(wl_var("wds"))) {
 361                 struct maclist *wdslist = (struct maclist *) buf;
 362                 struct ether_addr *addr = wdslist->ea;
 363                 char *next;
 364
 365                 memset(buf, 0, 8192);
 366                 foreach(wbuf, v, next) {
 367                         if (ether_atoe(wbuf, addr->ether_addr_octet)) {
 368                                 wdslist->count++;
 369                                 addr++;
 370                                 wds_enabled = 1;
 371                         }
 372                 }
 373                 bcom_ioctl(skfd, ifname, WLC_SET_WDSLIST, buf, sizeof(buf));
 374         }
 375         return wds_enabled;
 376 }
 377
 378 static void set_wext_mode(skfd, ifname)
 379 {
 380         struct iwreq wrq;
 381         int ap = 0, infra = 0, wet = 0;
 382
 383         /* Set operation mode */
 384         ap = !nvram_match(wl_var("mode"), "sta") && !nvram_match(wl_var("mode"), "wet");
 385         infra = !nvram_disabled(wl_var("infra"));
 386         wet = !ap && nvram_match(wl_var("mode"), "wet");
 387
 388         wrq.u.mode = (!infra ? IW_MODE_ADHOC : (ap ? IW_MODE_MASTER : (wet ? IW_MODE_REPEAT : IW_MODE_INFRA)));
 389         IW_SET_EXT_ERR(skfd, ifname, SIOCSIWMODE, &wrq, "Set Mode");
 390 }
 391
 392
 393 void start_watchdog(int skfd, char *ifname)
 394 {
 395         FILE *f;
 396         char *v, *next;
 397         unsigned char buf[8192], buf2[8192], wbuf[80], *p, *tmp;
 398         int wds = 0, i, j, restart_wds;
 399         wlc_ssid_t ssid;
 400
 401         if (fork())
 402                 return;
 403
 404         system("kill $(cat /var/run/wifi.pid) 2>&- >&-");
 405         f = fopen("/var/run/wifi.pid", "w");
 406         fprintf(f, "%d\n", getpid());
 407         fclose(f);
 408
 409         v = nvram_safe_get(wl_var("wds"));
 410         memset(buf2, 0, 8192);
 411         p = buf2;
 412         foreach(wbuf, v, next) {
 413                 if (ether_atoe(wbuf, p)) {
 414                         p += 6;
 415                         wds++;
 416                 }
 417         }
 418         v = nvram_safe_get(wl_var("ssid"));
 419         ssid.SSID_len = strlen(v);
 420         strncpy(ssid.SSID, v, 32);
 421
 422         for (;;) {
 423                 sleep(5);
 424
 425                 /* client mode */
 426                 bcom_ioctl(skfd, ifname, WLC_GET_AP, &i, sizeof(i));
 427                 if (!i) {
 428                         i = 0;
 429                         if (bcom_ioctl(skfd, ifname, WLC_GET_BSSID, buf, 6) < 0)
 430                                 i = 1;
 431                         memcpy(buf + 6, "\x00\x00\x00\x00\x00\x00", 6);
 432                         if (memcmp(buf, buf + 6, 6) == 0)
 433                                 i = 1;
 434
 435                         memset(buf, 0, 8192);
 436                         strcpy(buf, "sta_info");
 437                         bcom_ioctl(skfd, ifname, WLC_GET_BSSID, buf + strlen(buf) + 1, 6);
 438                         if (bcom_ioctl(skfd, ifname, WLC_GET_VAR, buf, 8192) < 0) {
 439                                 i = 1;
 440                         } else {
 441                                 sta_info_t *sta = (sta_info_t *) (buf + 4);
 442                                 if ((sta->flags & 0x18) != 0x18)
 443                                         i = 1;
 444                                 if (sta->idle > 20)
 445                                         i = 1;
 446                         }
 447
 448                         if (i)
 449                                 bcom_ioctl(skfd, ifname, WLC_SET_SSID, &ssid, sizeof(ssid));
 450                 }
 451
 452
 453                 /* wds */
 454                 p = buf2;
 455                 restart_wds = 0;
 456                 for (i = 0; i < wds; i++) {
 457                         memset(buf, 0, 8192);
 458                         strcpy(buf, "sta_info");
 459                         memcpy(buf + strlen(buf) + 1, p, 6);
 460                         if (bcom_ioctl(skfd, ifname, WLC_GET_VAR, buf, 8192) < 0) {
 461                         } else {
 462                                 sta_info_t *sta = (sta_info_t *) (buf + 4);
 463                                 if (!(sta->flags & 0x40)) {
 464                                 } else {
 465                                         if (sta->idle > 120)
 466                                                 restart_wds = 1;
 467                                 }
 468                         }
 469                         p += 6;
 470                 }
 471                 if (restart_wds)
 472                         setup_bcom_wds(skfd, ifname);
 473         }
 474 }
 475
 476 static void setup_bcom_vif_sec(int skfd, char *ifname, int vif)
 477 {
 478         int val, wep, wsec, i;
 479         char *v;
 480
 481         wsec = bcom_get_wsec(vif);
 482         if (wsec)
 483                 val = bcom_get_wauth(vif);
 484         else
 485                 val = 0;
 486
 487         bcom_set_bss_int(skfd, ifname, vif, "wpa_auth", val);
 488
 489         if (val) {
 490                 if (WPA_AUTH_PSK | WPA2_AUTH_PSK) {
 491                         v = nvram_safe_get(wl_var("wpa_psk"));
 492                         if ((strlen(v) >= 8) && (strlen(v) < 63) && nvram_match(wl_var("mode"), "wet") && (vif == 0)) {
 493                                 /* Enable in-driver WPA supplicant */
 494                                 wsec_pmk_t pmk;
 495
 496                                 pmk.key_len = (unsigned short) strlen(v);
 497                                 pmk.flags = WSEC_PASSPHRASE;
 498                                 strcpy(pmk.key, v);
 499                                 bcom_ioctl(skfd, ifname, WLC_SET_WSEC_PMK, &pmk, sizeof(pmk));
 500                                 bcom_set_int(skfd, ifname, "sup_wpa", 1);
 501                         } else {
 502                                 bcom_set_int(skfd, ifname, "sup_wpa", 0);
 503                         }
 504                 }
 505                 bcom_set_bss_int(skfd, ifname, vif, "eap_restrict", 1);
 506                 bcom_set_bss_int(skfd, ifname, vif, "wsec", wsec);
 507                 bcom_set_bss_int(skfd, ifname, vif, "wsec_restrict", 1);
 508         } else {
 509                 bcom_set_bss_int(skfd, ifname, vif, "eap_restrict", 0);
 510                 if (wep = nvram_enabled(vif_var(vif, "wep"))) {
 511                         wep = atoi(nvram_safe_get(vif_var(vif, "key")));
 512                         if ((wep >= 1) && (wep <= 4)) {
 513                                 for (i = 1; i < 4; i++) {
 514                                         wl_wsec_key_t k;
 515                                         char name[5] = "key0";
 516                                         unsigned char *kdata = k.data;
 517                                         unsigned char *kstr;
 518
 519                                         bzero(&k, sizeof(k));
 520                                         name[3] += i;
 521                                         kstr = nvram_safe_get(vif_var(vif, name));
 522                                         k.len = strlen(kstr);
 523                                         if ((k.len == 10) || (k.len == 26)) {
 524                                                 k.index = i - 1;
 525 #ifdef DEBUG
 526                                                 fprintf(stderr, "Adding WEP key %d to VIF %d: ", i, vif);
 527 #endif
 528                                                 k.len = 0;
 529                                                 while (*kstr != 0) {
 530                                                         strncpy(name, kstr, 2);
 531                                                         name[2] = 0;
 532                                                         *kdata = (unsigned char) strtoul(name, NULL, 16);
 533 #ifdef DEBUG
 534                                                         fprintf(stderr, "%02x", *kdata);
 535 #endif
 536                                                         kstr += 2;
 537                                                         kdata++;
 538                                                         k.len++;
 539                                                 }
 540 #ifdef DEBUG
 541                                                 fprintf(stderr, "\n");
 542 #endif
 543                                         } else {
 544                                                 k.len = 0;
 545                                         }
 546                                         if ((k.len > 0) && (i == wep))
 547                                                 k.flags = WL_PRIMARY_KEY;
 548
 549                                         bcom_set_bss_var(skfd, ifname, vif, "wsec_key", &k, sizeof(k));
 550                                 }
 551                                 wep = 1;
 552                                 bcom_set_bss_int(skfd, ifname, vif, "wsec", WEP_ENABLED);
 553                                 bcom_set_bss_int(skfd, ifname, vif, "wsec_restrict", 1);
 554                                 bcom_set_bss_int(skfd, ifname, vif, "auth", nvram_enabled(vif_var(vif, "auth")));
 555                         } else {
 556                                 wep = 0;
 557                         }
 558                 }
 559         }
 560
 561         if (!wep && !val) {
 562                 bcom_set_bss_int(skfd, ifname, vif, "wsec", 0);
 563                 bcom_set_bss_int(skfd, ifname, vif, "wsec_restrict", 0);
 564         }
 565
 566 //      bcom_set_bss_int(skfd, ifname, vif, "auth", atoi(nvram_safe_get(vif_var(vif, "auth"))));
 567 }
 568
 569 static void setup_bcom_vif(int skfd, char *ifname, int vif)
 570 {
 571         int val, wep, wsec, i;
 572         char *s, *v;
 573         wlc_ssid_t ssid;
 574
 575         s = nvram_safe_get(vif_var(vif, "ssid"));
 576         strncpy(ssid.SSID, s, sizeof(ssid.SSID));
 577         ssid.SSID_len = strlen(ssid.SSID);
 578         ssid.SSID_len = ((ssid.SSID_len > sizeof(ssid.SSID)) ? sizeof(ssid.SSID) : ssid.SSID_len);
 579         bcom_set_bss_var(skfd, ifname, vif, "ssid", &ssid, sizeof(ssid));
 580
 581         val = nvram_enabled(vif_var(vif, "closed"));
 582         bcom_set_bss_int(skfd, ifname, vif, "closednet", val);
 583
 584         val = nvram_enabled(wl_var("ap_isolate"));
 585         bcom_set_bss_int(skfd, ifname, vif, "ap_isolate", val);
 586
 587 }
 588
 589 static void start_bcom_vif(int skfd, char *ifname, int vif)
 590 {
 591         int cfg[2];
 592         int i;
 593
 594         cfg[0] = vif;
 595         cfg[1] = 1;
 596         for (i = 0; i < ADD_VIF_RETRIES; i++) {
 597                 if (bcom_set_var(skfd, ifname, "bss" , cfg, sizeof(cfg)) == 0)
 598                         break;
 599                 usleep(1000 * 1000);
 600         }
 601 }
 602
 603 static void setup_bcom_common(int skfd, char *ifname)
 604 {
 605         int val = 0, ap;
 606         char buf[8192], wbuf[80], *v;
 607
 608         nvram_set(wl_var("ifname"), ifname);
 609
 610         /* Set Country */
 611         strncpy(buf, nvram_safe_get(wl_var("country_code")), 4);
 612         buf[3] = 0;
 613         bcom_ioctl(skfd, ifname, WLC_SET_COUNTRY, buf, 4);
 614
 615         if (v = nvram_get(wl_var("txpwr"))) {
 616                 val = atoi(v);
 617                 val = mw_to_qdbm(val);
 618                 bcom_set_int(skfd, ifname, "qtxpower", val);
 619         }
 620
 621         /* Set other options */
 622         val = nvram_enabled(wl_var("lazywds"));
 623         bcom_ioctl(skfd, ifname, WLC_SET_LAZYWDS, &val, sizeof(val));
 624
 625         if (v = nvram_get(wl_var("dtim"))) {
 626                 val = atoi(v);
 627                 bcom_ioctl(skfd, ifname, WLC_SET_DTIMPRD, &val, sizeof(val));
 628         }
 629         if (v = nvram_get(wl_var("bcn"))) {
 630                 val = atoi(v);
 631                 bcom_ioctl(skfd, ifname, WLC_SET_BCNPRD, &val, sizeof(val));
 632         }
 633         if (v = nvram_get(wl_var("antdiv"))) {
 634                 val = atoi(v);
 635                 bcom_ioctl(skfd, ifname, WLC_SET_ANTDIV, &val, sizeof(val));
 636         }
 637         if (v = nvram_get(wl_var("txant"))) {
 638                 val = atoi(v);
 639                 bcom_ioctl(skfd, ifname, WLC_SET_TXANT, &val, sizeof(val));
 640         }
 641         if (v = nvram_get(wl_var("maxassoc"))) {
 642                 val = atoi(v);
 643                 bcom_set_int(skfd, ifname, "maxassoc", val);
 644         }
 645
 646         val = nvram_enabled(wl_var("frameburst"));
 647         bcom_ioctl(skfd, ifname, WLC_SET_FAKEFRAG, &val, sizeof(val));
 648
 649         ap = !nvram_match(wl_var("mode"), "sta") && !nvram_match(wl_var("mode"), "wet");
 650
 651         if (ap)
 652                 val = setup_bcom_wds(skfd, ifname);
 653
 654         if ((!ap || val) && is_new_bcom(skfd, ifname))
 655                 start_watchdog(skfd, ifname);
 656
 657         /* Set up afterburner, disabled it if WDS is enabled */
 658         if (val || nvram_enabled(wl_var("lazywds"))) {
 659                 val = ABO_OFF;
 660         } else {
 661                 val = ABO_AUTO;
 662                 if (nvram_enabled(wl_var("afterburner")))
 663                         val = ABO_ON;
 664                 if (nvram_disabled(wl_var("afterburner")))
 665                         val = ABO_OFF;
 666         }
 667
 668         bcom_set_var(skfd, ifname, "afterburner_override", &val, sizeof(val));
 669
 670         /* Set up MAC list */
 671         if (nvram_match(wl_var("macmode"), "allow"))
 672                 val = WLC_MACMODE_ALLOW;
 673         else if (nvram_match(wl_var("macmode"), "deny"))
 674                 val = WLC_MACMODE_DENY;
 675         else
 676                 val = WLC_MACMODE_DISABLED;
 677
 678         if ((val != WLC_MACMODE_DISABLED) && (v = nvram_get(wl_var("maclist")))) {
 679                 struct maclist *mac_list;
 680                 struct ether_addr *addr;
 681                 char *next;
 682
 683                 memset(buf, 0, 8192);
 684                 mac_list = (struct maclist *) buf;
 685                 addr = mac_list->ea;
 686
 687                 foreach(wbuf, v, next) {
 688                         if (ether_atoe(wbuf, addr->ether_addr_octet)) {
 689                                 mac_list->count++;
 690                                 addr++;
 691                         }
 692                 }
 693                 bcom_ioctl(skfd, ifname, WLC_SET_MACLIST, buf, sizeof(buf));
 694         } else {
 695                 val = WLC_MACMODE_DISABLED;
 696         }
 697         bcom_ioctl(skfd, ifname, WLC_SET_MACMODE, &val, sizeof(val));
 698
 699         /* Set up G mode */
 700         bcom_ioctl(skfd, ifname, WLC_GET_PHYTYPE, &val, sizeof(val));
 701         if (val == 2) {
 702                 int override = WLC_G_PROTECTION_OFF;
 703                 int control = WLC_G_PROTECTION_CTL_OFF;
 704
 705                 if (v = nvram_get(wl_var("gmode")))
 706                         val = atoi(v);
 707                 else
 708                         val = 1;
 709
 710                 if (val > 5)
 711                         val = 1;
 712
 713                 bcom_ioctl(skfd, ifname, WLC_SET_GMODE, &val, sizeof(val));
 714
 715                 if (nvram_match(wl_var("gmode_protection"), "auto")) {
 716                         override = WLC_G_PROTECTION_AUTO;
 717                         control = WLC_G_PROTECTION_CTL_OVERLAP;
 718                 }
 719                 if (nvram_enabled(wl_var("gmode_protection"))) {
 720                         override = WLC_G_PROTECTION_ON;
 721                         control = WLC_G_PROTECTION_CTL_OVERLAP;
 722                 }
 723                 bcom_ioctl(skfd, ifname, WLC_SET_GMODE_PROTECTION_CONTROL, &override, sizeof(control));
 724                 bcom_ioctl(skfd, ifname, WLC_SET_GMODE_PROTECTION_OVERRIDE, &override, sizeof(override));
 725
 726                 if (val = 0) {
 727                         if (nvram_match(wl_var("plcphdr"), "long"))
 728                                 val = WLC_PLCP_AUTO;
 729                         else
 730                                 val = WLC_PLCP_SHORT;
 731
 732                         bcom_ioctl(skfd, ifname, WLC_SET_PLCPHDR, &val, sizeof(val));
 733                 }
 734         }
 735 }
 736
 737 static void setup_bcom_new(int skfd, char *ifname)
 738 {
 739         int val = 0, i;
 740         int iface[16], ifaces = 1;
 741         int ap, apsta, sta, wet;
 742         char *v;
 743
 744         if (bcom_ioctl(skfd, ifname, WLC_GET_MAGIC, &val, sizeof(val)) < 0)
 745                 return;
 746
 747         /* Clear all VIFs */
 748         iface[0] = 0;
 749         for (i = 0; i < 16; i++) {
 750                 int cfg[2]; /* index, enabled */
 751
 752                 cfg[0] = i;
 753                 cfg[1] = 0;
 754
 755                 bcom_set_var(skfd, ifname, "bss", cfg, sizeof(cfg));
 756
 757                 if ((i > 0) && nvram_enabled(vif_var(i, "enabled")) && (i == 0 || nvram_get(vif_var(i, "ssid")))) {
 758                         iface[ifaces] = i;
 759                         ifaces++;
 760                 }
 761         }
 762
 763         set_wext_mode(skfd, ifname);
 764
 765         ap = nvram_match(wl_var("mode"), "ap") || nvram_match(wl_var("mode"), "apsta");
 766         apsta = nvram_match(wl_var("mode"), "apsta");
 767         sta = nvram_match(wl_var("mode"), "sta");
 768
 769         bcom_set_int(skfd, ifname, "apsta", apsta);
 770         bcom_set_int(skfd, ifname, "mssid", (ifaces > 1));
 771
 772         for (i = 0; i < (sta ? 0 : ifaces); i++) {
 773 #ifdef DEBUG
 774                 fprintf(stderr, "setup_bcom_vif(%d) start\n", iface[i]);
 775 #endif
 776                 setup_bcom_vif(skfd, ifname, iface[i]);
 777 #ifdef DEBUG
 778                 fprintf(stderr, "setup_bcom_vif(%d) end\n", iface[i]);
 779 #endif
 780         }
 781
 782
 783         if (v = nvram_get(wl_var("rts"))) {
 784                 val = atoi(v);
 785                 bcom_set_int(skfd, ifname, "rtsthresh", val);
 786         }
 787         if (v = nvram_get(wl_var("frag"))) {
 788                 val = atoi(v);
 789                 bcom_set_int(skfd, ifname, "fragthresh", val);
 790         }
 791
 792         val = (nvram_disabled(wl_var("radio")) ? (1 | (1 << 16)) : 0);
 793         bcom_ioctl(skfd, ifname, WLC_SET_RADIO, &val, sizeof(val));
 794
 795         setup_bcom_common(skfd, ifname);
 796         start_bcom(skfd, ifname);
 797
 798         val = atoi(nvram_safe_get(wl_var("channel")));
 799         if (val > 0)
 800                 bcom_ioctl(skfd, ifname, WLC_SET_CHANNEL, &val, sizeof(val));
 801
 802         val = (ap ? 15 : 0);
 803         bcom_ioctl(skfd, ifname, WLC_SET_CS_SCAN_TIMER, &val, sizeof(val));
 804
 805         for (i = 0; i < (sta ? 0 : ifaces); i++) {
 806                 setup_bcom_vif_sec(skfd, ifname, iface[i]);
 807         }
 808
 809         for (i = 0; i < (sta ? 0 : ifaces); i++) {
 810                 start_bcom_vif(skfd, ifname, iface[i]);
 811         }
 812 }
 813
 814 static void setup_bcom_old(int skfd, char *ifname)
 815 {
 816         int val = 0, i;
 817         char buf[8192];
 818         char wbuf[80];
 819         char *v;
 820
 821         if (bcom_ioctl(skfd, ifname, WLC_GET_MAGIC, &val, sizeof(val)) < 0)
 822                 return;
 823
 824         setup_bcom_common(skfd, ifname);
 825
 826         if (v = nvram_get(wl_var("frag"))) {
 827                 val = atoi(v);
 828                 bcom_ioctl(skfd, ifname, WLC_SET_FRAG, &val, sizeof(val));
 829         }
 830         if (v = nvram_get(wl_var("rts"))) {
 831                 val = atoi(v);
 832                 bcom_ioctl(skfd, ifname, WLC_SET_RTS, &val, sizeof(val));
 833         }
 834
 835         val = nvram_enabled(wl_var("closed"));
 836         bcom_ioctl(skfd, ifname, WLC_SET_CLOSED, &val, sizeof(val));
 837
 838         val = nvram_enabled(wl_var("ap_isolate"));
 839         bcom_set_int(skfd, ifname, "ap_isolate", val);
 840
 841         start_bcom(skfd, ifname);
 842         set_wext_ssid(skfd, ifname);
 843
 844         val = bcom_get_wauth(0);
 845         bcom_ioctl(skfd, ifname, WLC_SET_WPA_AUTH, &val, sizeof(val));
 846
 847         if (val & (WPA_AUTH_PSK | WPA2_AUTH_PSK)) {
 848                 v = nvram_safe_get(wl_var("wpa_psk"));
 849                 if ((strlen(v) >= 8) && (strlen(v) < 63) && nvram_match(wl_var("mode"), "wet")) {
 850                         /* Enable in-driver WPA supplicant */
 851                         wsec_pmk_t pmk;
 852
 853                         pmk.key_len = (unsigned short) strlen(v);
 854                         pmk.flags = WSEC_PASSPHRASE;
 855                         strcpy(pmk.key, v);
 856                         bcom_ioctl(skfd, ifname, WLC_SET_WSEC_PMK, &pmk, sizeof(pmk));
 857                         bcom_set_int(skfd, ifname, "sup_wpa", 1);
 858                 }
 859         }
 860         if (val) {
 861                 val = 1;
 862                 bcom_ioctl(skfd, ifname, WLC_SET_EAP_RESTRICT, &val, sizeof(val));
 863                 val = bcom_get_wsec(0);
 864                 bcom_ioctl(skfd, ifname, WLC_SET_WSEC, &val, sizeof(val));
 865         } else {
 866                 val = 0;
 867                 bcom_ioctl(skfd, ifname, WLC_SET_WSEC, &val, sizeof(val));
 868                 bcom_ioctl(skfd, ifname, WLC_SET_EAP_RESTRICT, &val, sizeof(val));
 869                 bcom_set_int(skfd, ifname, "sup_wpa", 0);
 870         }
 871
 872         if (v = nvram_get(wl_var("auth"))) {
 873                 val = atoi(v);
 874                 bcom_ioctl(skfd, ifname, WLC_SET_AUTH, &val, sizeof(val));
 875         }
 876 }
 877
 878
 879 static void set_wext_ssid(int skfd, char *ifname)
 880 {
 881         char *buffer;
 882         char essid[IW_ESSID_MAX_SIZE + 1];
 883         struct iwreq wrq;
 884
 885         buffer = nvram_get(wl_var("ssid"));
 886
 887         if (!buffer || (strlen(buffer) > IW_ESSID_MAX_SIZE))
 888                 buffer = "OpenWrt";
 889
 890         wrq.u.essid.flags = 1;
 891         strcpy(essid, buffer);
 892         wrq.u.essid.pointer = (caddr_t) essid;
 893         wrq.u.essid.length = strlen(essid) + 1;
 894         IW_SET_EXT_ERR(skfd, ifname, SIOCSIWESSID, &wrq, "Set ESSID");
 895 }
 896
 897 static void setup_wext_wep(int skfd, char *ifname)
 898 {
 899         int i, keylen;
 900         struct iwreq wrq;
 901         char keystr[5];
 902         char *keyval;
 903         unsigned char key[IW_ENCODING_TOKEN_MAX];
 904
 905         memset(&wrq, 0, sizeof(wrq));
 906         strcpy(keystr, "key1");
 907         for (i = 1; i <= 4; i++) {
 908                 if (keyval = nvram_get(wl_var(keystr))) {
 909                         keylen = iw_in_key(keyval, key);
 910
 911                         if (keylen > 0) {
 912                                 wrq.u.data.length = keylen;
 913                                 wrq.u.data.pointer = (caddr_t) key;
 914                                 wrq.u.data.flags = i;
 915                                 IW_SET_EXT_ERR(skfd, ifname, SIOCSIWENCODE, &wrq, "Set Encode");
 916                         }
 917                 }
 918                 keystr[3]++;
 919         }
 920
 921         memset(&wrq, 0, sizeof(wrq));
 922         i = atoi(nvram_safe_get(wl_var("key")));
 923         if (i > 0 && i < 4) {
 924                 wrq.u.data.flags = i | IW_ENCODE_RESTRICTED;
 925                 IW_SET_EXT_ERR(skfd, ifname, SIOCSIWENCODE, &wrq, "Set Encode");
 926         }
 927 }
 928
 929 static void setup_wext(int skfd, char *ifname)
 930 {
 931         char *buffer;
 932         struct iwreq wrq;
 933
 934         /* Set channel */
 935         int channel = atoi(nvram_safe_get(wl_var("channel")));
 936
 937         wrq.u.freq.m = -1;
 938         wrq.u.freq.e = 0;
 939         wrq.u.freq.flags = 0;
 940
 941         if (channel > 0) {
 942                 wrq.u.freq.flags = IW_FREQ_FIXED;
 943                 wrq.u.freq.m = channel;
 944                 IW_SET_EXT_ERR(skfd, ifname, SIOCSIWFREQ, &wrq, "Set Frequency");
 945         }
 946
 947         /* Disable radio if wlX_radio is set and not enabled */
 948         wrq.u.txpower.disabled = nvram_disabled(wl_var("radio"));
 949
 950         wrq.u.txpower.value = -1;
 951         wrq.u.txpower.fixed = 1;
 952         wrq.u.txpower.flags = IW_TXPOW_DBM;
 953         IW_SET_EXT_ERR(skfd, ifname, SIOCSIWTXPOW, &wrq, "Set Tx Power");
 954
 955         /* Set up WEP */
 956         if (nvram_enabled(wl_var("wep")) && !wpa_enc)
 957                 setup_wext_wep(skfd, ifname);
 958
 959         /* Set ESSID */
 960         set_wext_ssid(skfd, ifname);
 961
 962 }
 963
 964 static int setup_interfaces(int skfd, char *ifname, char *args[], int count)
 965 {
 966         struct iwreq wrq;
 967         int rc;
 968
 969         /* Avoid "Unused parameter" warning */
 970         args = args; count = count;
 971
 972         if(iw_get_ext(skfd, ifname, SIOCGIWNAME, &wrq) < 0)
 973                 return 0;
 974
 975         if (strncmp(ifname, "ath", 3) == 0) {
 976                 set_wext_mode(skfd, ifname);
 977                 setup_wext(skfd, ifname);
 978         } else {
 979                 if (is_new_bcom(skfd, ifname)) {
 980 #ifdef DEBUG
 981                         fprintf(stderr, "New Broadcom driver detected.\n");
 982 #endif
 983                         stop_bcom(skfd, ifname);
 984 #ifdef DEBUG
 985                         fprintf(stderr, "Setup start.\n");
 986 #endif
 987                         setup_bcom_new(skfd, ifname);
 988 #ifdef DEBUG
 989                         fprintf(stderr, "Setup done.\n");
 990 #endif
 991                 } else {
 992 #ifdef DEBUG
 993                         fprintf(stderr, "Old Broadcom driver detected.\n");
 994 #endif
 995                         stop_bcom(skfd, ifname);
 996                         set_wext_mode(skfd, ifname);
 997                         setup_bcom_old(skfd, ifname);
 998                         setup_wext(skfd, ifname);
 999                 }
1000         }
1001
1002         prefix[2]++;
1003 }
1004
1005 int main(int argc, char **argv)
1006 {
1007         int skfd;
1008         if((skfd = iw_sockets_open()) < 0) {
1009                 perror("socket");
1010                 exit(-1);
1011         }
1012
1013         prefix = strdup("wl0");
1014         iw_enum_devices(skfd, &setup_interfaces, NULL, 0);
1015
1016         return 0;
1017 }