#include <bcmnvram.h>
#include <shutils.h>
#include <wlioctl.h>
+#include <signal.h>
/*------------------------------------------------------------------*/
/*
do { \
ERR_SET_EXT(rname, request); \
fprintf(stderr, " too few arguments.\n"); \
- return; \
} while(0)
#define ABORT_ARG_TYPE(rname, request, arg) \
do { \
ERR_SET_EXT(rname, request); \
fprintf(stderr, " invalid argument \"%s\".\n", arg); \
- return; \
} while(0)
#define ABORT_ARG_SIZE(rname, request, max) \
do { \
ERR_SET_EXT(rname, request); \
fprintf(stderr, " argument too big (max %d)\n", max); \
- return; \
} while(0)
/*------------------------------------------------------------------*/
ERR_SET_EXT(rname, request); \
fprintf(stderr, " SET failed on device %-1.16s ; %s.\n", \
ifname, strerror(errno)); \
- return; \
} } while(0)
/*------------------------------------------------------------------*/
ERR_SET_EXT(rname, request); \
fprintf(stderr, " GET failed on device %-1.16s ; %s.\n", \
ifname, strerror(errno)); \
- return; \
} } while(0)
-char *prefix;
-char buffer[128];
+static void set_wext_ssid(int skfd, char *ifname);
-char *wl_var(char *name)
+static char *prefix;
+static char buffer[128];
+static int wpa_enc = 0;
+
+static char *wl_var(char *name)
{
strcpy(buffer, prefix);
strcat(buffer, name);
}
-int nvram_enabled(char *name)
+static int nvram_enabled(char *name)
{
return (nvram_match(name, "1") || nvram_match(name, "on") || nvram_match(name, "enabled") || nvram_match(name, "true") || nvram_match(name, "yes") ? 1 : 0);
}
-int nvram_disabled(char *name)
+static int nvram_disabled(char *name)
{
return (nvram_match(name, "0") || nvram_match(name, "off") || nvram_match(name, "disabled") || nvram_match(name, "false") || nvram_match(name, "no") ? 1 : 0);
}
-int bcom_ioctl(int skfd, char *ifname, int cmd, void *buf, int len)
+static int bcom_ioctl(int skfd, char *ifname, int cmd, void *buf, int len)
{
struct ifreq ifr;
wl_ioctl_t ioc;
strncpy(ifr.ifr_name, ifname, IFNAMSIZ);
ret = ioctl(skfd, SIOCDEVPRIVATE, &ifr);
- if (ret < 0)
- fprintf(stderr, "bcom_ioctl [cmd=%d, buf=%08x, len=%d] failed: %d\n", cmd, buf, len, ret);
return ret;
}
-int bcom_set_val(int skfd, char *ifname, char *var, void *val, int len)
+static int bcom_set_val(int skfd, char *ifname, char *var, void *val, int len)
{
char buf[8192];
int ret;
return -1;
strcpy(buf, var);
+ memcpy(&buf[strlen(var) + 1], val, len);
- if ((ret = bcom_ioctl(skfd, ifname, WLC_GET_VAR, buf, sizeof(buf))))
+ if ((ret = bcom_ioctl(skfd, ifname, WLC_SET_VAR, buf, sizeof(buf))))
return ret;
- memcpy(val, buf, len);
return 0;
}
-int bcom_set_int(int skfd, char *ifname, char *var, int val)
+static int bcom_set_int(int skfd, char *ifname, char *var, int val)
{
return bcom_set_val(skfd, ifname, var, &val, sizeof(val));
}
-void setup_bcom(int skfd, char *ifname)
+static void stop_bcom(int skfd, char *ifname)
{
int val = 0;
- char *v;
+ wlc_ssid_t ssid;
if (bcom_ioctl(skfd, ifname, WLC_GET_MAGIC, &val, sizeof(val)) < 0)
return;
-
+
+ ssid.SSID_len = 0;
+ ssid.SSID[0] = 0;
+ bcom_ioctl(skfd, ifname, WLC_SET_SSID, &ssid, sizeof(ssid));
bcom_ioctl(skfd, ifname, WLC_DOWN, NULL, 0);
+
+}
+
+static void start_bcom(int skfd, char *ifname)
+{
+ int val = 0;
- /* Set up WPA */
- if (nvram_match(wl_var("crypto"), "tkip"))
- val = TKIP_ENABLED;
- else if (nvram_match(wl_var("crypto"), "aes"))
- val = AES_ENABLED;
- else if (nvram_match(wl_var("crypto"), "tkip+aes"))
- val = TKIP_ENABLED | AES_ENABLED;
- else
- val = 0;
- bcom_ioctl(skfd, ifname, WLC_SET_WSEC, &val, sizeof(val));
+ if (bcom_ioctl(skfd, ifname, WLC_GET_MAGIC, &val, sizeof(val)) < 0)
+ return;
- if (val && nvram_get(wl_var("wpa_psk"))) {
- val = 1;
- bcom_ioctl(skfd, ifname, WLC_SET_EAP_RESTRICT, &val, sizeof(val));
+ bcom_ioctl(skfd, ifname, WLC_UP, &val, sizeof(val));
+ set_wext_ssid(skfd, ifname);
+}
+
+static int setup_bcom_wds(int skfd, char *ifname)
+{
+ char buf[8192];
+ char wbuf[80];
+ char *v;
+ int wds_enabled = 0;
+
+ if (v = nvram_get(wl_var("wds"))) {
+ struct maclist *wdslist = (struct maclist *) buf;
+ struct ether_addr *addr = wdslist->ea;
+ char *next;
+
+ memset(buf, 0, 8192);
+ foreach(wbuf, v, next) {
+ if (ether_atoe(wbuf, addr->ether_addr_octet)) {
+ wdslist->count++;
+ addr++;
+ wds_enabled = 1;
+ }
+ }
+ bcom_ioctl(skfd, ifname, WLC_SET_WDSLIST, buf, sizeof(buf));
}
+ return wds_enabled;
+}
- /* Set up afterburner */
- val = ABO_AUTO;
- if (nvram_enabled(wl_var("afterburner")))
- val = ABO_ON;
- if (nvram_disabled(wl_var("afterburner")))
- val = ABO_OFF;
- bcom_set_val(skfd, ifname, "afterburner_override", &val, sizeof(val));
+void start_watchdog(int skfd, char *ifname)
+{
+ FILE *f;
+ unsigned char buf[8192], buf2[8192], wbuf[80], *v, *p, *next, *tmp;
+ int wds = 0, i, restart_wds;
+
+ if (fork())
+ return;
+
+ system("kill $(cat /var/run/wifi.pid) 2>&- >&-");
+ f = fopen("/var/run/wifi.pid", "w");
+ fprintf(f, "%d\n", getpid());
+ fclose(f);
- /* Set other options */
- if (v = nvram_get(wl_var("lazywds"))) {
- val = atoi(v);
- bcom_ioctl(skfd, ifname, WLC_SET_LAZYWDS, &val, sizeof(val));
+ v = nvram_safe_get(wl_var("wds"));
+ memset(buf2, 0, 8192);
+ p = buf2;
+ foreach(wbuf, v, next) {
+ if (ether_atoe(wbuf, p)) {
+ p += 6;
+ wds++;
+ }
}
+ v = nvram_safe_get(wl_var("ssid"));
+
+ for (;;) {
+ sleep(5);
+ if (bcom_ioctl(skfd, ifname, WLC_GET_BSSID, buf, 6) < 0)
+ bcom_ioctl(skfd, ifname, WLC_SET_SSID, v, strlen(v));
+ p = buf2;
+ restart_wds = 0;
+ for (i = 0; i < wds; i++) {
+ memset(buf, 0, 8192);
+ strcpy(buf, "sta_info");
+ memcpy(buf + strlen(buf) + 1, p, 6);
+ if (bcom_ioctl(skfd, ifname, WLC_GET_VAR, buf, 8192) < 0) {
+ } else {
+ sta_info_t *sta = (sta_info_t *) (buf + 4);
+ if (!(sta->flags & 0x40)) {
+ } else {
+ if (sta->idle > 120)
+ restart_wds = 1;
+ }
+ }
+ p += 6;
+ }
+ if (restart_wds)
+ setup_bcom_wds(skfd, ifname);
+ }
+}
+
+static void setup_bcom(int skfd, char *ifname)
+{
+ int val = 0, ap;
+ char buf[8192];
+ char wbuf[80];
+ char *v;
+ int wds_enabled = 0;
+
+ if (bcom_ioctl(skfd, ifname, WLC_GET_MAGIC, &val, sizeof(val)) < 0)
+ return;
+
+ nvram_set(wl_var("ifname"), ifname);
+
+ stop_bcom(skfd, ifname);
+
+ /* Set Country */
+ strncpy(buf, nvram_safe_get(wl_var("country_code")), 4);
+ buf[3] = 0;
+ bcom_ioctl(skfd, ifname, WLC_SET_COUNTRY, buf, 4);
+
+ /* Set other options */
+ val = nvram_enabled(wl_var("lazywds"));
+ wds_enabled = val;
+ bcom_ioctl(skfd, ifname, WLC_SET_LAZYWDS, &val, sizeof(val));
+
if (v = nvram_get(wl_var("frag"))) {
val = atoi(v);
bcom_ioctl(skfd, ifname, WLC_SET_FRAG, &val, sizeof(val));
val = WLC_MACMODE_DISABLED;
if ((val != WLC_MACMODE_DISABLED) && (v = nvram_get(wl_var("maclist")))) {
- char buf[8192];
- char wbuf[80];
struct maclist *mac_list;
struct ether_addr *addr;
char *next;
mac_list = (struct maclist *) buf;
addr = mac_list->ea;
- foreach(wbuf, nvram_safe_get(wl_var("maclist")), next) {
+ foreach(wbuf, v, next) {
if (ether_atoe(wbuf, addr->ether_addr_octet)) {
mac_list->count++;
addr++;
val = WLC_MACMODE_DISABLED;
}
bcom_ioctl(skfd, ifname, WLC_SET_MACMODE, &val, sizeof(val));
+
+ if (ap = !nvram_match(wl_var("mode"), "sta") && !nvram_match(wl_var("mode"), "wet"))
+ wds_enabled = setup_bcom_wds(skfd, ifname);
+
+ if (!ap || wds_enabled)
+ start_watchdog(skfd, ifname);
+
+ /* Set up afterburner, disabled it if WDS is enabled */
+ if (wds_enabled) {
+ val = ABO_OFF;
+ } else {
+ val = ABO_AUTO;
+ if (nvram_enabled(wl_var("afterburner")))
+ val = ABO_ON;
+ if (nvram_disabled(wl_var("afterburner")))
+ val = ABO_OFF;
+ }
+
+ bcom_set_val(skfd, ifname, "afterburner_override", &val, sizeof(val));
/* Set up G mode */
bcom_ioctl(skfd, ifname, WLC_GET_PHYTYPE, &val, sizeof(val));
int override = WLC_G_PROTECTION_OFF;
int control = WLC_G_PROTECTION_CTL_OFF;
- val = atoi(nvram_safe_get(wl_var("gmode")));
+ if (v = nvram_get(wl_var("gmode")))
+ val = atoi(v);
+ else
+ val = 1;
+
if (val > 5)
val = 1;
+
bcom_ioctl(skfd, ifname, WLC_SET_GMODE, &val, sizeof(val));
if (nvram_match(wl_var("gmode_protection"), "auto")) {
}
bcom_ioctl(skfd, ifname, WLC_SET_GMODE_PROTECTION_CONTROL, &override, sizeof(control));
bcom_ioctl(skfd, ifname, WLC_SET_GMODE_PROTECTION_OVERRIDE, &override, sizeof(override));
+
+ if (val = 0) {
+ if (nvram_match(wl_var("plcphdr"), "long"))
+ val = WLC_PLCP_AUTO;
+ else
+ val = WLC_PLCP_SHORT;
+
+ bcom_ioctl(skfd, ifname, WLC_SET_PLCPHDR, &val, sizeof(val));
+ }
}
-}
-void set_wext_ssid(int skfd, char *ifname)
-{
- char *buffer;
- struct iwreq wrq;
+ start_bcom(skfd, ifname);
- if (buffer = nvram_get(wl_var("ssid"))) {
- if (strlen(buffer) > IW_ESSID_MAX_SIZE) {
- ABORT_ARG_SIZE("Set ESSID", SIOCSIWESSID, IW_ESSID_MAX_SIZE);
- } else {
- char essid[IW_ESSID_MAX_SIZE + 1];
-
- wrq.u.essid.flags = 1;
- strcpy(essid, buffer);
- wrq.u.essid.pointer = (caddr_t) essid;
- wrq.u.essid.length = strlen(essid) + 1;
- IW_SET_EXT_ERR(skfd, ifname, SIOCSIWESSID, &wrq, "Set ESSID");
+ if (!(v = nvram_get(wl_var("akm"))))
+ v = nvram_safe_get(wl_var("auth_mode"));
+
+ if (strstr(v, "wpa") || strstr(v, "psk")) {
+ wpa_enc = 1;
+
+ /* Set up WPA */
+ if (nvram_match(wl_var("crypto"), "tkip"))
+ val = TKIP_ENABLED;
+ else if (nvram_match(wl_var("crypto"), "aes"))
+ val = AES_ENABLED;
+ else if (nvram_match(wl_var("crypto"), "tkip+aes") || nvram_match(wl_var("crypto"), "aes+tkip"))
+ val = TKIP_ENABLED | AES_ENABLED;
+ else
+ val = 0;
+ bcom_ioctl(skfd, ifname, WLC_SET_WSEC, &val, sizeof(val));
+
+ if (val && strstr(v, "psk")) {
+ val = (strstr(v, "psk2") ? 0x84 : 0x4);
+ v = nvram_safe_get(wl_var("wpa_psk"));
+ if ((strlen(v) >= 8) && (strlen(v) < 63)) {
+
+ bcom_ioctl(skfd, ifname, WLC_SET_WPA_AUTH, &val, sizeof(val));
+
+ if (!nvram_match(wl_var("mode"), "wet")) {
+ /* Enable in-driver WPA supplicant */
+ wsec_pmk_t pmk;
+
+ pmk.key_len = (unsigned short) strlen(v);
+ pmk.flags = WSEC_PASSPHRASE;
+ strcpy(pmk.key, v);
+ bcom_ioctl(skfd, ifname, WLC_SET_WSEC_PMK, &pmk, sizeof(pmk));
+ bcom_set_int(skfd, ifname, "sup_wpa", 1);
+ }
+ }
+ } else {
+ val = 1;
+ bcom_ioctl(skfd, ifname, WLC_SET_EAP_RESTRICT, &val, sizeof(val));
}
+ } else {
+ val = 0;
+
+ bcom_ioctl(skfd, ifname, WLC_SET_WSEC, &val, sizeof(val));
+ bcom_ioctl(skfd, ifname, WLC_SET_WPA_AUTH, &val, sizeof(val));
+ bcom_ioctl(skfd, ifname, WLC_SET_EAP_RESTRICT, &val, sizeof(val));
+ bcom_set_int(skfd, ifname, "sup_wpa", 0);
}
}
-void start_bcom(int skfd, char *ifname)
+static void set_wext_ssid(int skfd, char *ifname)
{
- int val = 0;
-
- if (bcom_ioctl(skfd, ifname, WLC_GET_MAGIC, &val, sizeof(val)) < 0)
- return;
+ char *buffer;
+ char essid[IW_ESSID_MAX_SIZE + 1];
+ struct iwreq wrq;
- bcom_ioctl(skfd, ifname, WLC_UP, &val, sizeof(val));
+ buffer = nvram_get(wl_var("ssid"));
- /* Need to re-set SSID after WLC_UP */
- set_wext_ssid(skfd, ifname);
+ if (!buffer || (strlen(buffer) > IW_ESSID_MAX_SIZE))
+ buffer = "OpenWrt";
+
+ wrq.u.essid.flags = 1;
+ strcpy(essid, buffer);
+ wrq.u.essid.pointer = (caddr_t) essid;
+ wrq.u.essid.length = strlen(essid) + 1;
+ IW_SET_EXT_ERR(skfd, ifname, SIOCSIWESSID, &wrq, "Set ESSID");
}
-void setup_wext_wep(int skfd, char *ifname)
+static void setup_wext_wep(int skfd, char *ifname)
{
int i, keylen;
struct iwreq wrq;
char keystr[5];
char *keyval;
unsigned char key[IW_ENCODING_TOKEN_MAX];
-
+
+ memset(&wrq, 0, sizeof(wrq));
strcpy(keystr, "key1");
for (i = 1; i <= 4; i++) {
if (keyval = nvram_get(wl_var(keystr))) {
keystr[3]++;
}
-
+ memset(&wrq, 0, sizeof(wrq));
i = atoi(nvram_safe_get(wl_var("key")));
if (i > 0 && i < 4) {
wrq.u.data.flags = i | IW_ENCODE_RESTRICTED;
}
}
-void setup_wext(int skfd, char *ifname)
+static void set_wext_mode(skfd, ifname)
+{
+ struct iwreq wrq;
+ int ap = 0, infra = 0, wet = 0;
+
+ /* Set operation mode */
+ ap = !nvram_match(wl_var("mode"), "sta") && !nvram_match(wl_var("mode"), "wet");
+ infra = !nvram_disabled(wl_var("infra"));
+ wet = !ap && nvram_match(wl_var("mode"), "wet");
+
+ wrq.u.mode = (!infra ? IW_MODE_ADHOC : (ap ? IW_MODE_MASTER : (wet ? IW_MODE_REPEAT : IW_MODE_INFRA)));
+ IW_SET_EXT_ERR(skfd, ifname, SIOCSIWMODE, &wrq, "Set Mode");
+}
+
+static void setup_wext(int skfd, char *ifname)
{
char *buffer;
struct iwreq wrq;
- /* Set ESSID */
- set_wext_ssid(skfd, ifname);
-
/* Set channel */
int channel = atoi(nvram_safe_get(wl_var("channel")));
wrq.u.freq.m = channel;
IW_SET_EXT_ERR(skfd, ifname, SIOCSIWFREQ, &wrq, "Set Frequency");
}
-
- /* Set operation mode */
- int ap = 0, infra = 0, wet = 0;
-
- ap = !nvram_match(wl_var("mode"), "sta");
- infra = !nvram_disabled(wl_var("infra"));
- wet = nvram_enabled(wl_var("wet"));
- wrq.u.mode = (!infra ? IW_MODE_ADHOC : (ap ? IW_MODE_MASTER : (wet ? IW_MODE_REPEAT : IW_MODE_INFRA)));
- IW_SET_EXT_ERR(skfd, ifname, SIOCSIWMODE, &wrq, "Set Mode");
/* Disable radio if wlX_radio is set and not enabled */
wrq.u.txpower.disabled = nvram_disabled(wl_var("radio"));
wrq.u.txpower.value = -1;
wrq.u.txpower.fixed = 1;
- wrq.u.txpower.flags = IW_TXPOW_MWATT;
+ wrq.u.txpower.flags = IW_TXPOW_DBM;
IW_SET_EXT_ERR(skfd, ifname, SIOCSIWTXPOW, &wrq, "Set Tx Power");
/* Set up WEP */
- if (nvram_enabled(wl_var("wep")))
+ if (nvram_enabled(wl_var("wep")) && !wpa_enc)
setup_wext_wep(skfd, ifname);
+
+ /* Set ESSID */
+ set_wext_ssid(skfd, ifname);
}
-
static int setup_interfaces(int skfd, char *ifname, char *args[], int count)
{
struct iwreq wrq;
if(iw_get_ext(skfd, ifname, SIOCGIWNAME, &wrq) < 0)
return 0;
- setup_bcom(skfd, ifname);
- setup_wext(skfd, ifname);
- start_bcom(skfd, ifname);
+ if (strncmp(ifname, "ath", 3) == 0) {
+ set_wext_mode(skfd, ifname);
+ setup_wext(skfd, ifname);
+ } else {
+ stop_bcom(skfd, ifname);
+ set_wext_mode(skfd, ifname);
+ setup_bcom(skfd, ifname);
+ setup_wext(skfd, ifname);
+ }
+
prefix[2]++;
}