source: branches/whiterussian/openwrt/package/wificonf/wificonf.c @ 2471

Last change on this file since 2471 was 2471, checked in by nbd, 10 years ago

enable driver-internal supplicant only in wet mode and set psk2 mode if configured

  • Property svn:eol-style set to native
  • Property svn:keywords set to Author Date Id Revision
File size: 14.3 KB
Line 
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
80static void set_wext_ssid(int skfd, char *ifname);
81
82static char *prefix;
83static char buffer[128];
84static int wpa_enc = 0;
85
86static char *wl_var(char *name)
87{
88        strcpy(buffer, prefix);
89        strcat(buffer, name);
90}
91
92static 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
97static 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
103static 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
121static 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
138static 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
143static 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
158static 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
169static 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
194void 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
245static 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
440static 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
458static 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
490static 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
504static 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
540static 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
564int 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}
Note: See TracBrowser for help on using the repository browser.