]> git.enpas.org Git - openwrt.git/blob - package/wificonf/wificonf.c
fix wificonf segfault when maclist is enabled
[openwrt.git] / 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
24 /*------------------------------------------------------------------*/
25 /*
26  * Macro to handle errors when setting WE
27  * Print a nice error message and exit...
28  * We define them as macro so that "return" do the right thing.
29  * The "do {...} while(0)" is a standard trick
30  */
31 #define ERR_SET_EXT(rname, request) \
32         fprintf(stderr, "Error for wireless request \"%s\" (%X) :\n", \
33                 rname, request)
34
35 #define ABORT_ARG_NUM(rname, request) \
36         do { \
37                 ERR_SET_EXT(rname, request); \
38                 fprintf(stderr, "    too few arguments.\n"); \
39                 return; \
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                 return; \
47         } while(0)
48
49 #define ABORT_ARG_SIZE(rname, request, max) \
50         do { \
51                 ERR_SET_EXT(rname, request); \
52                 fprintf(stderr, "    argument too big (max %d)\n", max); \
53                 return; \
54         } while(0)
55
56 /*------------------------------------------------------------------*/
57 /*
58  * Wrapper to push some Wireless Parameter in the driver
59  * Use standard wrapper and add pretty error message if fail...
60  */
61 #define IW_SET_EXT_ERR(skfd, ifname, request, wrq, rname) \
62         do { \
63         if(iw_set_ext(skfd, ifname, request, wrq) < 0) { \
64                 ERR_SET_EXT(rname, request); \
65                 fprintf(stderr, "    SET failed on device %-1.16s ; %s.\n", \
66                         ifname, strerror(errno)); \
67                 return; \
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                 return; \
82         } } while(0)
83
84 char *prefix;
85 char buffer[128];
86
87 char *wl_var(char *name)
88 {
89         strcpy(buffer, prefix);
90         strcat(buffer, name);
91 }
92
93 int nvram_enabled(char *name)
94 {
95         return (nvram_match(name, "1") || nvram_match(name, "on") || nvram_match(name, "enabled") || nvram_match(name, "true") || nvram_match(name, "yes") ? 1 : 0);
96 }
97
98 int nvram_disabled(char *name)
99 {
100         return (nvram_match(name, "0") || nvram_match(name, "off") || nvram_match(name, "disabled") || nvram_match(name, "false") || nvram_match(name, "no") ? 1 : 0);
101 }
102
103
104 int bcom_ioctl(int skfd, char *ifname, int cmd, void *buf, int len)
105 {
106         struct ifreq ifr;
107         wl_ioctl_t ioc;
108         int ret;
109         
110         ioc.cmd = cmd;
111         ioc.buf = buf;
112         ioc.len = len;
113         
114         ifr.ifr_data = (caddr_t) &ioc;
115         strncpy(ifr.ifr_name, ifname, IFNAMSIZ);
116
117         ret = ioctl(skfd, SIOCDEVPRIVATE, &ifr);
118         if (ret < 0)
119                 fprintf(stderr, "bcom_ioctl [cmd=%d, buf=%08x, len=%d] failed: %d\n", cmd, buf, len, ret);
120
121         return ret;
122 }
123
124 int bcom_set_val(int skfd, char *ifname, char *var, void *val, int len)
125 {
126         char buf[8192];
127         int ret;
128         
129         if (strlen(var) + 1 > sizeof(buf) || len > sizeof(buf))
130                 return -1;
131
132         strcpy(buf, var);
133         
134         if ((ret = bcom_ioctl(skfd, ifname, WLC_GET_VAR, buf, sizeof(buf))))
135                 return ret;
136
137         memcpy(val, buf, len);
138         return 0;       
139 }
140
141 int bcom_set_int(int skfd, char *ifname, char *var, int val)
142 {
143         return bcom_set_val(skfd, ifname, var, &val, sizeof(val));
144 }
145
146 void setup_bcom(int skfd, char *ifname)
147 {
148         int val = 0;
149         char *v;
150         
151         if (bcom_ioctl(skfd, ifname, WLC_GET_MAGIC, &val, sizeof(val)) < 0)
152                 return;
153
154         bcom_ioctl(skfd, ifname, WLC_DOWN, NULL, 0);
155         
156         /* Set up WPA */
157         if (nvram_match(wl_var("crypto"), "tkip"))
158                 val = TKIP_ENABLED;
159         else if (nvram_match(wl_var("crypto"), "aes"))
160                 val = AES_ENABLED;
161         else if (nvram_match(wl_var("crypto"), "tkip+aes"))
162                 val = TKIP_ENABLED | AES_ENABLED;
163         else
164                 val = 0;
165         bcom_ioctl(skfd, ifname, WLC_SET_WSEC, &val, sizeof(val));
166
167         if (val && nvram_get(wl_var("wpa_psk"))) {
168                 val = 1;
169                 bcom_ioctl(skfd, ifname, WLC_SET_EAP_RESTRICT, &val, sizeof(val));
170         }
171
172         /* Set up afterburner */
173         val = ABO_AUTO;
174         if (nvram_enabled(wl_var("afterburner")))
175                 val = ABO_ON;
176         if (nvram_disabled(wl_var("afterburner")))
177                 val = ABO_OFF;
178         bcom_set_val(skfd, ifname, "afterburner_override", &val, sizeof(val));
179         
180         /* Set other options */
181         val = atoi(nvram_safe_get(wl_var("lazywds")));
182         bcom_ioctl(skfd, ifname, WLC_SET_LAZYWDS, &val, sizeof(val));
183         val = atoi(nvram_safe_get(wl_var("frag")));
184         bcom_ioctl(skfd, ifname, WLC_SET_FRAG, &val, sizeof(val));
185         val = atoi(nvram_safe_get(wl_var("dtim")));
186         bcom_ioctl(skfd, ifname, WLC_SET_DTIMPRD, &val, sizeof(val));
187         val = atoi(nvram_safe_get(wl_var("bcn")));
188         bcom_ioctl(skfd, ifname, WLC_SET_BCNPRD, &val, sizeof(val));
189         val = atoi(nvram_safe_get(wl_var("rts")));
190         bcom_ioctl(skfd, ifname, WLC_SET_RTS, &val, sizeof(val));
191         val = atoi(nvram_safe_get(wl_var("antdiv")));
192         bcom_ioctl(skfd, ifname, WLC_SET_ANTDIV, &val, sizeof(val));
193         val = atoi(nvram_safe_get(wl_var("txant")));
194         bcom_ioctl(skfd, ifname, WLC_SET_TXANT, &val, sizeof(val));
195
196         val = nvram_enabled(wl_var("ap_isolate"));
197         bcom_set_int(skfd, ifname, "ap_isolate", val);
198
199         val = nvram_enabled(wl_var("frameburst"));
200         bcom_ioctl(skfd, ifname, WLC_SET_FAKEFRAG, &val, sizeof(val));
201
202         /* Set up MAC list */
203         if (nvram_match(wl_var("macmode"), "allow"))
204                 val = WLC_MACMODE_ALLOW;
205         else if (nvram_match(wl_var("macmode"), "deny"))
206                 val = WLC_MACMODE_DENY;
207         else
208                 val = WLC_MACMODE_DISABLED;
209
210         if ((val != WLC_MACMODE_DISABLED) && (v = nvram_get(wl_var("maclist")))) {
211                 char buf[8192];
212                 struct maclist *mac_list;
213                 struct ether_addr *addr;
214                 char *next;
215                 
216                 memset(buf, 0, 8192);
217                 mac_list = (struct maclist *) buf;
218                 addr = mac_list->ea;
219                 
220                 v = malloc(80);
221                 foreach(v, nvram_safe_get(wl_var("maclist")), next) {
222                         if (ether_atoe(v, addr->ether_addr_octet)) {
223                                 mac_list->count++;
224                                 addr++;
225                         }
226                 }
227                 free(v);
228                 bcom_ioctl(skfd, ifname, WLC_SET_MACLIST, buf, sizeof(buf));
229         } else {
230                 val = WLC_MACMODE_DISABLED;
231         }
232         bcom_ioctl(skfd, ifname, WLC_SET_MACMODE, &val, sizeof(val));
233         
234         /* Set up G mode */
235         bcom_ioctl(skfd, ifname, WLC_GET_PHYTYPE, &val, sizeof(val));
236         if (val == 2) {
237                 int override = WLC_G_PROTECTION_OFF;
238                 int control = WLC_G_PROTECTION_CTL_OFF;
239                 
240                 val = atoi(nvram_safe_get(wl_var("gmode")));
241                 if (val > 5)
242                         val = 1;
243                 bcom_ioctl(skfd, ifname, WLC_SET_GMODE, &val, sizeof(val));
244                 
245                 if (nvram_match(wl_var("gmode_protection"), "auto")) {
246                         override = WLC_G_PROTECTION_AUTO;
247                         control = WLC_G_PROTECTION_CTL_OVERLAP;
248                 }
249                 if (nvram_enabled(wl_var("gmode_protection"))) {
250                         override = WLC_G_PROTECTION_ON;
251                         control = WLC_G_PROTECTION_CTL_OVERLAP;
252                 }
253                 bcom_ioctl(skfd, ifname, WLC_SET_GMODE_PROTECTION_CONTROL, &override, sizeof(control));
254                 bcom_ioctl(skfd, ifname, WLC_SET_GMODE_PROTECTION_OVERRIDE, &override, sizeof(override));
255         }
256 }
257
258 void set_wext_ssid(int skfd, char *ifname)
259 {
260         char *buffer;
261         struct iwreq wrq;
262
263         if (buffer = nvram_get(wl_var("ssid"))) {
264                 if (strlen(buffer) > IW_ESSID_MAX_SIZE) {
265                         ABORT_ARG_SIZE("Set ESSID", SIOCSIWESSID, IW_ESSID_MAX_SIZE);
266                 } else {
267                         char essid[IW_ESSID_MAX_SIZE + 1];
268
269                         wrq.u.essid.flags = 1;
270                         strcpy(essid, buffer);
271                         wrq.u.essid.pointer = (caddr_t) essid;
272                         wrq.u.essid.length = strlen(essid) + 1;
273                         IW_SET_EXT_ERR(skfd, ifname, SIOCSIWESSID, &wrq, "Set ESSID");
274                 }
275         }
276 }
277
278 void start_bcom(int skfd, char *ifname)
279 {
280         int val = 0;
281         
282         if (bcom_ioctl(skfd, ifname, WLC_GET_MAGIC, &val, sizeof(val)) < 0)
283                 return;
284
285         bcom_ioctl(skfd, ifname, WLC_UP, &val, sizeof(val));
286         
287         /* Need to re-set SSID after WLC_UP */
288         set_wext_ssid(skfd, ifname);
289 }
290
291 void setup_wext_wep(int skfd, char *ifname)
292 {
293         int i, keylen;
294         struct iwreq wrq;
295         char keystr[5];
296         char *keyval;
297         unsigned char key[IW_ENCODING_TOKEN_MAX];
298
299         strcpy(keystr, "key1");
300         for (i = 1; i <= 4; i++) {
301                 if (keyval = nvram_get(wl_var(keystr))) {
302                         keylen = iw_in_key(keyval, key);
303                         
304                         if (keylen > 0) {
305                                 wrq.u.data.length = keylen;
306                                 wrq.u.data.pointer = (caddr_t) key;
307                                 wrq.u.data.flags = i;
308                                 IW_SET_EXT_ERR(skfd, ifname, SIOCSIWENCODE, &wrq, "Set Encode");
309                         }
310                 }
311                 keystr[3]++;
312         }
313         
314         
315         i = atoi(nvram_safe_get(wl_var("key")));
316         if (i > 0 && i < 4) {
317                 wrq.u.data.flags = i | IW_ENCODE_RESTRICTED;
318                 IW_SET_EXT_ERR(skfd, ifname, SIOCSIWENCODE, &wrq, "Set Encode");
319         }
320 }
321
322 void setup_wext(int skfd, char *ifname)
323 {
324         char *buffer;
325         struct iwreq wrq;
326
327         /* Set ESSID */
328         set_wext_ssid(skfd, ifname);
329         
330         /* Set channel */
331         int channel = atoi(nvram_safe_get(wl_var("channel")));
332         
333         wrq.u.freq.m = -1;
334         wrq.u.freq.e = 0;
335         wrq.u.freq.flags = 0;
336
337         if (channel > 0) {
338                 wrq.u.freq.flags = IW_FREQ_FIXED;
339                 wrq.u.freq.m = channel;
340         }
341         IW_SET_EXT_ERR(skfd, ifname, SIOCSIWFREQ, &wrq, "Set Frequency");
342         
343         /* Set operation mode */
344         int ap = 0, infra = 0, wet = 0;
345         
346         ap = (nvram_match(wl_var("mode"), "ap") || nvram_match(wl_var("mode"), "wds"));
347         infra = nvram_enabled(wl_var("infra"));
348         wet = nvram_enabled(wl_var("wet"));
349
350         wrq.u.mode = (!infra ? IW_MODE_ADHOC : (ap ? IW_MODE_MASTER : (wet ? IW_MODE_REPEAT : IW_MODE_INFRA)));
351         IW_SET_EXT_ERR(skfd, ifname, SIOCSIWMODE, &wrq, "Set Mode");
352
353         /* Disable radio if wlX_radio is set and not enabled */
354         if (nvram_disabled(wl_var("radio")))
355                 wrq.u.txpower.disabled = 1;
356         else 
357                 wrq.u.txpower.disabled = 0;
358
359         wrq.u.txpower.value = -1;
360         wrq.u.txpower.fixed = 1;
361         wrq.u.txpower.flags = IW_TXPOW_MWATT;
362         IW_SET_EXT_ERR(skfd, ifname, SIOCSIWTXPOW, &wrq, "Set Tx Power");
363
364         /* Set up WEP */
365         if (nvram_enabled(wl_var("wep")))
366                 setup_wext_wep(skfd, ifname);
367
368 }
369
370
371 static int setup_interfaces(int skfd, char *ifname, char *args[], int count)
372 {
373         struct iwreq wrq;
374         int rc;
375         
376         /* Avoid "Unused parameter" warning */
377         args = args; count = count;
378         
379         if(iw_get_ext(skfd, ifname, SIOCGIWNAME, &wrq) < 0)
380                 return 0;
381
382         setup_bcom(skfd, ifname);
383         setup_wext(skfd, ifname);
384         start_bcom(skfd, ifname);
385         prefix[2]++;
386 }
387
388 int main(int argc, char **argv)
389 {
390         int skfd;
391         if((skfd = iw_sockets_open()) < 0) {
392                 perror("socket");
393                 exit(-1);
394         }
395
396         prefix = strdup("wl0_");
397         iw_enum_devices(skfd, &setup_interfaces, NULL, 0);
398         
399         return 0;
400 }