openwrt/package/wificonf/wificonf.c

   1 /*
   2  * Wireless Network Adapter configuration utility
   3  *
   4  * Copyright (C) 2005 Felix Fietkau <nbd@vd-s.ath.cx>
   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 /*------------------------------------------------------------------*/
  26 /*
  27  * Macro to handle errors when setting WE
  28  * Print a nice error message and exit...
  29  * We define them as macro so that "return" do the right thing.
  30  * The "do {...} while(0)" is a standard trick
  31  */
  32 #define ERR_SET_EXT(rname, request) \
  33         fprintf(stderr, "Error for wireless request \"%s\" (%X) :\n", \
  34                 rname, request)
  35
  36 #define ABORT_ARG_NUM(rname, request) \
  37         do { \
  38                 ERR_SET_EXT(rname, request); \
  39                 fprintf(stderr, "    too few arguments.\n"); \
  40         } while(0)
  41
  42 #define ABORT_ARG_TYPE(rname, request, arg) \
  43         do { \
  44                 ERR_SET_EXT(rname, request); \
  45                 fprintf(stderr, "    invalid argument \"%s\".\n", arg); \
  46         } while(0)
  47
  48 #define ABORT_ARG_SIZE(rname, request, max) \
  49         do { \
  50                 ERR_SET_EXT(rname, request); \
  51                 fprintf(stderr, "    argument too big (max %d)\n", max); \
  52         } while(0)
  53
  54 /*------------------------------------------------------------------*/
  55 /*
  56  * Wrapper to push some Wireless Parameter in the driver
  57  * Use standard wrapper and add pretty error message if fail...
  58  */
  59 #define IW_SET_EXT_ERR(skfd, ifname, request, wrq, rname) \
  60         do { \
  61         if(iw_set_ext(skfd, ifname, request, wrq) < 0) { \
  62                 ERR_SET_EXT(rname, request); \
  63                 fprintf(stderr, "    SET failed on device %-1.16s ; %s.\n", \
  64                         ifname, strerror(errno)); \
  65         } } while(0)
  66
  67 /*------------------------------------------------------------------*/
  68 /*
  69  * Wrapper to extract some Wireless Parameter out of the driver
  70  * Use standard wrapper and add pretty error message if fail...
  71  */
  72 #define IW_GET_EXT_ERR(skfd, ifname, request, wrq, rname) \
  73         do { \
  74         if(iw_get_ext(skfd, ifname, request, wrq) < 0) { \
  75                 ERR_SET_EXT(rname, request); \
  76                 fprintf(stderr, "    GET failed on device %-1.16s ; %s.\n", \
  77                         ifname, strerror(errno)); \
  78         } } while(0)
  79
  80 static void set_wext_ssid(int skfd, char *ifname);
  81
  82 static char *prefix;
  83 static char buffer[128];
  84 static int wpa_enc = 0;
  85
  86 static char *wl_var(char *name)
  87 {
  88         strcpy(buffer, prefix);
  89         strcat(buffer, name);
  90 }
  91
  92 static int nvram_enabled(char *name)
  93 {
  94         return (nvram_match(name, "1") || nvram_match(name, "on") || nvram_match(name, "enabled") || nvram_match(name, "true") || nvram_match(name, "yes") ? 1 : 0);
  95 }
  96
  97 static int nvram_disabled(char *name)
  98 {
  99         return (nvram_match(name, "0") || nvram_match(name, "off") || nvram_match(name, "disabled") || nvram_match(name, "false") || nvram_match(name, "no") ? 1 : 0);
 100 }
 101
 102
 103 static int bcom_ioctl(int skfd, char *ifname, int cmd, void *buf, int len)
 104 {
 105         struct ifreq ifr;
 106         wl_ioctl_t ioc;
 107         int ret;
 108
 109         ioc.cmd = cmd;
 110         ioc.buf = buf;
 111         ioc.len = len;
 112
 113         ifr.ifr_data = (caddr_t) &ioc;
 114         strncpy(ifr.ifr_name, ifname, IFNAMSIZ);
 115
 116         ret = ioctl(skfd, SIOCDEVPRIVATE, &ifr);
 117
 118         return ret;
 119 }
 120
 121 static int bcom_set_val(int skfd, char *ifname, char *var, void *val, int len)
 122 {
 123         char buf[8192];
 124         int ret;
 125
 126         if (strlen(var) + 1 > sizeof(buf) || len > sizeof(buf))
 127                 return -1;
 128
 129         strcpy(buf, var);
 130         memcpy(&buf[strlen(var) + 1], val, len);
 131
 132         if ((ret = bcom_ioctl(skfd, ifname, WLC_SET_VAR, buf, sizeof(buf))))
 133                 return ret;
 134
 135         return 0;
 136 }
 137
 138 static int bcom_set_int(int skfd, char *ifname, char *var, int val)
 139 {
 140         return bcom_set_val(skfd, ifname, var, &val, sizeof(val));
 141 }
 142
 143 static void stop_bcom(int skfd, char *ifname)
 144 {
 145         int val = 0;
 146         wlc_ssid_t ssid;
 147
 148         if (bcom_ioctl(skfd, ifname, WLC_GET_MAGIC, &val, sizeof(val)) < 0)
 149                 return;
 150
 151         ssid.SSID_len = 0;
 152         ssid.SSID[0] = 0;
 153         bcom_ioctl(skfd, ifname, WLC_SET_SSID, &ssid, sizeof(ssid));
 154         bcom_ioctl(skfd, ifname, WLC_DOWN, NULL, 0);
 155
 156 }
 157
 158 static void start_bcom(int skfd, char *ifname)
 159 {
 160         int val = 0;
 161
 162         if (bcom_ioctl(skfd, ifname, WLC_GET_MAGIC, &val, sizeof(val)) < 0)
 163                 return;
 164
 165         bcom_ioctl(skfd, ifname, WLC_UP, &val, sizeof(val));
 166         set_wext_ssid(skfd, ifname);
 167 }
 168
 169 static int setup_bcom_wds(int skfd, char *ifname)
 170 {
 171         char buf[8192];
 172         char wbuf[80];
 173         char *v;
 174         int wds_enabled = 0;
 175
 176         if (v = nvram_get(wl_var("wds"))) {
 177                 struct maclist *wdslist = (struct maclist *) buf;
 178                 struct ether_addr *addr = wdslist->ea;
 179                 char *next;
 180
 181                 memset(buf, 0, 8192);
 182                 foreach(wbuf, v, next) {
 183                         if (ether_atoe(wbuf, addr->ether_addr_octet)) {
 184                                 wdslist->count++;
 185                                 addr++;
 186                                 wds_enabled = 1;
 187                         }
 188                 }
 189                 bcom_ioctl(skfd, ifname, WLC_SET_WDSLIST, buf, sizeof(buf));
 190         }
 191         return wds_enabled;
 192 }
 193
 194 void start_watchdog(int skfd, char *ifname)
 195 {
 196         FILE *f;
 197         unsigned char buf[8192], buf2[8192], wbuf[80], *v, *p, *next, *tmp;
 198         int wds = 0, i, restart_wds;
 199
 200         if (fork())
 201                 return;
 202
 203         system("kill $(cat /var/run/wifi.pid) 2>&- >&-");
 204         f = fopen("/var/run/wifi.pid", "w");
 205         fprintf(f, "%d\n", getpid());
 206         fclose(f);
 207
 208         v = nvram_safe_get(wl_var("wds"));
 209         memset(buf2, 0, 8192);
 210         p = buf2;
 211         foreach(wbuf, v, next) {
 212                 if (ether_atoe(wbuf, p)) {
 213                         p += 6;
 214                         wds++;
 215                 }
 216         }
 217         v = nvram_safe_get(wl_var("ssid"));
 218
 219         for (;;) {
 220                 sleep(5);
 221                 if (bcom_ioctl(skfd, ifname, WLC_GET_BSSID, buf, 6) < 0)
 222                         bcom_ioctl(skfd, ifname, WLC_SET_SSID, v, strlen(v));
 223                 p = buf2;
 224                 restart_wds = 0;
 225                 for (i = 0; i < wds; i++) {
 226                         memset(buf, 0, 8192);
 227                         strcpy(buf, "sta_info");
 228                         memcpy(buf + strlen(buf) + 1, p, 6);
 229                         if (bcom_ioctl(skfd, ifname, WLC_GET_VAR, buf, 8192) < 0) {
 230                         } else {
 231                                 sta_info_t *sta = (sta_info_t *) (buf + 4);
 232                                 if (!(sta->flags & 0x40)) {
 233                                 } else {
 234                                         if (sta->idle > 120)
 235                                                 restart_wds = 1;
 236                                 }
 237                         }
 238                         p += 6;
 239                 }
 240                 if (restart_wds)
 241                         setup_bcom_wds(skfd, ifname);
 242         }
 243 }
 244
 245 static void setup_bcom(int skfd, char *ifname)
 246 {
 247         int val = 0, ap;
 248         char buf[8192];
 249         char wbuf[80];
 250         char *v;
 251         int wds_enabled = 0;
 252
 253         if (bcom_ioctl(skfd, ifname, WLC_GET_MAGIC, &val, sizeof(val)) < 0)
 254                 return;
 255
 256         nvram_set(wl_var("ifname"), ifname);
 257
 258         stop_bcom(skfd, ifname);
 259
 260         /* Set Country */
 261         strncpy(buf, nvram_safe_get(wl_var("country_code")), 4);
 262         buf[3] = 0;
 263         bcom_ioctl(skfd, ifname, WLC_SET_COUNTRY, buf, 4);
 264
 265         /* Set other options */
 266         val = nvram_enabled(wl_var("lazywds"));
 267         wds_enabled = val;
 268         bcom_ioctl(skfd, ifname, WLC_SET_LAZYWDS, &val, sizeof(val));
 269
 270         if (v = nvram_get(wl_var("frag"))) {
 271                 val = atoi(v);
 272                 bcom_ioctl(skfd, ifname, WLC_SET_FRAG, &val, sizeof(val));
 273         }
 274         if (v = nvram_get(wl_var("dtim"))) {
 275                 val = atoi(v);
 276                 bcom_ioctl(skfd, ifname, WLC_SET_DTIMPRD, &val, sizeof(val));
 277         }
 278         if (v = nvram_get(wl_var("bcn"))) {
 279                 val = atoi(v);
 280                 bcom_ioctl(skfd, ifname, WLC_SET_BCNPRD, &val, sizeof(val));
 281         }
 282         if (v = nvram_get(wl_var("rts"))) {
 283                 val = atoi(v);
 284                 bcom_ioctl(skfd, ifname, WLC_SET_RTS, &val, sizeof(val));
 285         }
 286         if (v = nvram_get(wl_var("antdiv"))) {
 287                 val = atoi(v);
 288                 bcom_ioctl(skfd, ifname, WLC_SET_ANTDIV, &val, sizeof(val));
 289         }
 290         if (v = nvram_get(wl_var("txant"))) {
 291                 val = atoi(v);
 292                 bcom_ioctl(skfd, ifname, WLC_SET_TXANT, &val, sizeof(val));
 293         }
 294
 295         val = nvram_enabled(wl_var("closed"));
 296         bcom_ioctl(skfd, ifname, WLC_SET_CLOSED, &val, sizeof(val));
 297
 298         val = nvram_enabled(wl_var("ap_isolate"));
 299         bcom_set_int(skfd, ifname, "ap_isolate", val);
 300
 301         val = nvram_enabled(wl_var("frameburst"));
 302         bcom_ioctl(skfd, ifname, WLC_SET_FAKEFRAG, &val, sizeof(val));
 303
 304         /* Set up MAC list */
 305         if (nvram_match(wl_var("macmode"), "allow"))
 306                 val = WLC_MACMODE_ALLOW;
 307         else if (nvram_match(wl_var("macmode"), "deny"))
 308                 val = WLC_MACMODE_DENY;
 309         else
 310                 val = WLC_MACMODE_DISABLED;
 311
 312         if ((val != WLC_MACMODE_DISABLED) && (v = nvram_get(wl_var("maclist")))) {
 313                 struct maclist *mac_list;
 314                 struct ether_addr *addr;
 315                 char *next;
 316
 317                 memset(buf, 0, 8192);
 318                 mac_list = (struct maclist *) buf;
 319                 addr = mac_list->ea;
 320
 321                 foreach(wbuf, v, next) {
 322                         if (ether_atoe(wbuf, addr->ether_addr_octet)) {
 323                                 mac_list->count++;
 324                                 addr++;
 325                         }
 326                 }
 327                 bcom_ioctl(skfd, ifname, WLC_SET_MACLIST, buf, sizeof(buf));
 328         } else {
 329                 val = WLC_MACMODE_DISABLED;
 330         }
 331         bcom_ioctl(skfd, ifname, WLC_SET_MACMODE, &val, sizeof(val));
 332
 333         if (ap = !nvram_match(wl_var("mode"), "sta") && !nvram_match(wl_var("mode"), "wet"))
 334                 wds_enabled = setup_bcom_wds(skfd, ifname);
 335
 336         if (!ap || wds_enabled)
 337                 start_watchdog(skfd, ifname);
 338
 339         /* Set up afterburner, disabled it if WDS is enabled */
 340         if (wds_enabled) {
 341                 val = ABO_OFF;
 342         } else {
 343                 val = ABO_AUTO;
 344                 if (nvram_enabled(wl_var("afterburner")))
 345                         val = ABO_ON;
 346                 if (nvram_disabled(wl_var("afterburner")))
 347                         val = ABO_OFF;
 348         }
 349
 350         bcom_set_val(skfd, ifname, "afterburner_override", &val, sizeof(val));
 351
 352         /* Set up G mode */
 353         bcom_ioctl(skfd, ifname, WLC_GET_PHYTYPE, &val, sizeof(val));
 354         if (val == 2) {
 355                 int override = WLC_G_PROTECTION_OFF;
 356                 int control = WLC_G_PROTECTION_CTL_OFF;
 357
 358                 if (v = nvram_get(wl_var("gmode")))
 359                         val = atoi(v);
 360                 else
 361                         val = 1;
 362
 363                 if (val > 5)
 364                         val = 1;
 365
 366                 bcom_ioctl(skfd, ifname, WLC_SET_GMODE, &val, sizeof(val));
 367
 368                 if (nvram_match(wl_var("gmode_protection"), "auto")) {
 369                         override = WLC_G_PROTECTION_AUTO;
 370                         control = WLC_G_PROTECTION_CTL_OVERLAP;
 371                 }
 372                 if (nvram_enabled(wl_var("gmode_protection"))) {
 373                         override = WLC_G_PROTECTION_ON;
 374                         control = WLC_G_PROTECTION_CTL_OVERLAP;
 375                 }
 376                 bcom_ioctl(skfd, ifname, WLC_SET_GMODE_PROTECTION_CONTROL, &override, sizeof(control));
 377                 bcom_ioctl(skfd, ifname, WLC_SET_GMODE_PROTECTION_OVERRIDE, &override, sizeof(override));
 378
 379                 if (val = 0) {
 380                         if (nvram_match(wl_var("plcphdr"), "long"))
 381                                 val = WLC_PLCP_AUTO;
 382                         else
 383                                 val = WLC_PLCP_SHORT;
 384
 385                         bcom_ioctl(skfd, ifname, WLC_SET_PLCPHDR, &val, sizeof(val));
 386                 }
 387         }
 388
 389         start_bcom(skfd, ifname);
 390
 391         if (!(v = nvram_get(wl_var("akm"))))
 392                 v = nvram_safe_get(wl_var("auth_mode"));
 393
 394         if (strstr(v, "wpa") || strstr(v, "psk")) {
 395                 wpa_enc = 1;
 396
 397                 /* Set up WPA */
 398                 if (nvram_match(wl_var("crypto"), "tkip"))
 399                         val = TKIP_ENABLED;
 400                 else if (nvram_match(wl_var("crypto"), "aes"))
 401                         val = AES_ENABLED;
 402                 else if (nvram_match(wl_var("crypto"), "tkip+aes") || nvram_match(wl_var("crypto"), "aes+tkip"))
 403                         val = TKIP_ENABLED | AES_ENABLED;
 404                 else
 405                         val = 0;
 406                 bcom_ioctl(skfd, ifname, WLC_SET_WSEC, &val, sizeof(val));
 407
 408                 if (val && strstr(v, "psk")) {
 409                         val = (strstr(v, "psk2") ? 0x84 : 0x4);
 410                         v = nvram_safe_get(wl_var("wpa_psk"));
 411                         if ((strlen(v) >= 8) && (strlen(v) < 63)) {
 412
 413                                 bcom_ioctl(skfd, ifname, WLC_SET_WPA_AUTH, &val, sizeof(val));
 414
 415                                 if (!nvram_match(wl_var("mode"), "wet")) {
 416                                         /* Enable in-driver WPA supplicant */
 417                                         wsec_pmk_t pmk;
 418
 419                                         pmk.key_len = (unsigned short) strlen(v);
 420                                         pmk.flags = WSEC_PASSPHRASE;
 421                                         strcpy(pmk.key, v);
 422                                         bcom_ioctl(skfd, ifname, WLC_SET_WSEC_PMK, &pmk, sizeof(pmk));
 423                                         bcom_set_int(skfd, ifname, "sup_wpa", 1);
 424                                 }
 425                         }
 426                 } else  {
 427                         val = 1;
 428                         bcom_ioctl(skfd, ifname, WLC_SET_EAP_RESTRICT, &val, sizeof(val));
 429                 }
 430         } else {
 431                 val = 0;
 432
 433                 bcom_ioctl(skfd, ifname, WLC_SET_WSEC, &val, sizeof(val));
 434                 bcom_ioctl(skfd, ifname, WLC_SET_WPA_AUTH, &val, sizeof(val));
 435                 bcom_ioctl(skfd, ifname, WLC_SET_EAP_RESTRICT, &val, sizeof(val));
 436                 bcom_set_int(skfd, ifname, "sup_wpa", 0);
 437         }
 438 }
 439
 440 static void set_wext_ssid(int skfd, char *ifname)
 441 {
 442         char *buffer;
 443         char essid[IW_ESSID_MAX_SIZE + 1];
 444         struct iwreq wrq;
 445
 446         buffer = nvram_get(wl_var("ssid"));
 447
 448         if (!buffer || (strlen(buffer) > IW_ESSID_MAX_SIZE))
 449                 buffer = "OpenWrt";
 450
 451         wrq.u.essid.flags = 1;
 452         strcpy(essid, buffer);
 453         wrq.u.essid.pointer = (caddr_t) essid;
 454         wrq.u.essid.length = strlen(essid) + 1;
 455         IW_SET_EXT_ERR(skfd, ifname, SIOCSIWESSID, &wrq, "Set ESSID");
 456 }
 457
 458 static void setup_wext_wep(int skfd, char *ifname)
 459 {
 460         int i, keylen;
 461         struct iwreq wrq;
 462         char keystr[5];
 463         char *keyval;
 464         unsigned char key[IW_ENCODING_TOKEN_MAX];
 465
 466         memset(&wrq, 0, sizeof(wrq));
 467         strcpy(keystr, "key1");
 468         for (i = 1; i <= 4; i++) {
 469                 if (keyval = nvram_get(wl_var(keystr))) {
 470                         keylen = iw_in_key(keyval, key);
 471
 472                         if (keylen > 0) {
 473                                 wrq.u.data.length = keylen;
 474                                 wrq.u.data.pointer = (caddr_t) key;
 475                                 wrq.u.data.flags = i;
 476                                 IW_SET_EXT_ERR(skfd, ifname, SIOCSIWENCODE, &wrq, "Set Encode");
 477                         }
 478                 }
 479                 keystr[3]++;
 480         }
 481
 482         memset(&wrq, 0, sizeof(wrq));
 483         i = atoi(nvram_safe_get(wl_var("key")));
 484         if (i > 0 && i < 4) {
 485                 wrq.u.data.flags = i | IW_ENCODE_RESTRICTED;
 486                 IW_SET_EXT_ERR(skfd, ifname, SIOCSIWENCODE, &wrq, "Set Encode");
 487         }
 488 }
 489
 490 static void set_wext_mode(skfd, ifname)
 491 {
 492         struct iwreq wrq;
 493         int ap = 0, infra = 0, wet = 0;
 494
 495         /* Set operation mode */
 496         ap = !nvram_match(wl_var("mode"), "sta") && !nvram_match(wl_var("mode"), "wet");
 497         infra = !nvram_disabled(wl_var("infra"));
 498         wet = !ap && nvram_match(wl_var("mode"), "wet");
 499
 500         wrq.u.mode = (!infra ? IW_MODE_ADHOC : (ap ? IW_MODE_MASTER : (wet ? IW_MODE_REPEAT : IW_MODE_INFRA)));
 501         IW_SET_EXT_ERR(skfd, ifname, SIOCSIWMODE, &wrq, "Set Mode");
 502 }
 503
 504 static void setup_wext(int skfd, char *ifname)
 505 {
 506         char *buffer;
 507         struct iwreq wrq;
 508
 509         /* Set channel */
 510         int channel = atoi(nvram_safe_get(wl_var("channel")));
 511
 512         wrq.u.freq.m = -1;
 513         wrq.u.freq.e = 0;
 514         wrq.u.freq.flags = 0;
 515
 516         if (channel > 0) {
 517                 wrq.u.freq.flags = IW_FREQ_FIXED;
 518                 wrq.u.freq.m = channel;
 519                 IW_SET_EXT_ERR(skfd, ifname, SIOCSIWFREQ, &wrq, "Set Frequency");
 520         }
 521
 522
 523         /* Disable radio if wlX_radio is set and not enabled */
 524         wrq.u.txpower.disabled = nvram_disabled(wl_var("radio"));
 525
 526         wrq.u.txpower.value = -1;
 527         wrq.u.txpower.fixed = 1;
 528         wrq.u.txpower.flags = IW_TXPOW_DBM;
 529         IW_SET_EXT_ERR(skfd, ifname, SIOCSIWTXPOW, &wrq, "Set Tx Power");
 530
 531         /* Set up WEP */
 532         if (nvram_enabled(wl_var("wep")) && !wpa_enc)
 533                 setup_wext_wep(skfd, ifname);
 534
 535         /* Set ESSID */
 536         set_wext_ssid(skfd, ifname);
 537
 538 }
 539
 540 static int setup_interfaces(int skfd, char *ifname, char *args[], int count)
 541 {
 542         struct iwreq wrq;
 543         int rc;
 544
 545         /* Avoid "Unused parameter" warning */
 546         args = args; count = count;
 547
 548         if(iw_get_ext(skfd, ifname, SIOCGIWNAME, &wrq) < 0)
 549                 return 0;
 550
 551         if (strncmp(ifname, "ath", 3) == 0) {
 552                 set_wext_mode(skfd, ifname);
 553                 setup_wext(skfd, ifname);
 554         } else {
 555                 stop_bcom(skfd, ifname);
 556                 set_wext_mode(skfd, ifname);
 557                 setup_bcom(skfd, ifname);
 558                 setup_wext(skfd, ifname);
 559         }
 560
 561         prefix[2]++;
 562 }
 563
 564 int main(int argc, char **argv)
 565 {
 566         int skfd;
 567         if((skfd = iw_sockets_open()) < 0) {
 568                 perror("socket");
 569                 exit(-1);
 570         }
 571
 572         prefix = strdup("wl0_");
 573         iw_enum_devices(skfd, &setup_interfaces, NULL, 0);
 574
 575         return 0;
 576 }