--- embedtools/src/dwds.c 2010/10/27 15:23:49 1.1.2.4 +++ embedtools/src/dwds.c 2010/11/02 02:25:46 1.1.2.13 @@ -5,7 +5,6 @@ sl_config cfg; int Verbose, Kill, nif; char **ifs, szConfig[MAXPATHLEN] = DWDS_CONFIG; -struct dwds_if *wds; extern char compiled[], compiledby[], compilehost[]; @@ -14,7 +13,7 @@ Usage() { printf( "-= dWDS =- WiFi dynamic WDS service managment for VAP\n" "=== %s === %s@%s ===\n\n" - " Syntax: dwds [options] [interface [...]]\n" + " Syntax: dwds [options] [interface [ath0, ...]]\n" "\n" "\t-v\t\tVerbose ...\n" "\t-f\t\tForeground, not demonize process ...\n" @@ -54,8 +53,131 @@ sigHandler(int sig) } static int -RtMsg(struct rt_msghdr *msg, size_t len) +RtMsg(struct dwds_if **wds, struct rt_msghdr *msg, size_t len) { + struct if_announcemsghdr *ifan; + const u_char *v; + struct ether_addr bssid; + char szStr[STRSIZ] = { 0 }, szCmd[MAXPATHLEN] = { 0 }, szIdent[STRSIZ] = { 0 }, + szName[IFNAMSIZ] = { 0 }; + int f, stat; + + assert(wds); + assert(msg); + + if (msg->rtm_version != RTM_VERSION) { + syslog(LOG_ERR, "Error:: routing message version %d not understood!\n", msg->rtm_version); + return -1; + } + + switch (msg->rtm_type) { + case RTM_IFANNOUNCE: + ifan = (struct if_announcemsghdr*) msg; + switch (ifan->ifan_what) { + case IFAN_ARRIVAL: + VERB(1) syslog(LOG_INFO, "RTM_IFANNOUNCE: if# %d, what: arrival\n", + ifan->ifan_index); + break; + case IFAN_DEPARTURE: + VERB(1) syslog(LOG_INFO, "RTM_IFANNOUNCE: if# %d, what: departure\n", + ifan->ifan_index); + wifi_destroyWDS(ifan->ifan_name, wds); + break; + } + break; + case RTM_IEEE80211: +#define V(type) ((struct type *)(&ifan[1])) + ifan = (struct if_announcemsghdr*) msg; + switch (ifan->ifan_what) { + case RTM_IEEE80211_DISASSOC: + v = cfg_GetAttribute(&cfg, CFG("dwds"), CFG("discover_on_join")); + if (!v || !strtol((char*) v, NULL, 0)) + break; + /* fall thru ... */ + case RTM_IEEE80211_LEAVE: + if (!wifi_chkIface(ifan->ifan_name, ifs, nif)) + break; + memcpy(&bssid, V(ieee80211_leave_event)->iev_addr, ETHER_ADDR_LEN); + VERB(1) syslog(LOG_INFO, "BSSID:%s Station leave\n", ether_ntoa(&bssid)); + if (!wifi_leaveWDS(bssid, wds, szName, IFNAMSIZ)) { + cfg_LoadAttribute(&cfg, CFG("dwds"), CFG("name"), + CFG(szIdent), STRSIZ, DWDS_NAME); + /* delete state file ... */ + v = cfg_GetAttribute(&cfg, CFG("dwds"), CFG("state_dir")); + if (v) { + memset(szCmd, 0, STRSIZ); + snprintf(szCmd, MAXPATHLEN, "%s/%s-%s-%s", (char*) v, + szName, ether_ntoa(&bssid), szIdent); + unlink(szCmd); + VERB(2) syslog(LOG_DEBUG, "Debug:: delete session name %s\n", szCmd); + } + + /* Launch script ... */ + cfg_LoadAttribute(&cfg, CFG("dwds"), CFG("disassoc_event"), + CFG(szStr), STRSIZ, NULL); + if (*szStr) { + cfg_LoadAttribute(&cfg, CFG("dwds"), CFG("name"), + CFG(szIdent), STRSIZ, DWDS_NAME); + memset(szCmd, 0, MAXPATHLEN); + snprintf(szCmd, MAXPATHLEN, "%s %s %s %s", szStr, + szName, ether_ntoa(&bssid), szIdent); + VERB(3) syslog(LOG_DEBUG, "Debug:: Command line: %s\n", szCmd); + + if ((stat = system(szCmd))) + syslog(LOG_ERR, "VAP down script %s exited " + "with status %d\n", szStr, stat); + } + } + break; + + case RTM_IEEE80211_JOIN: + case RTM_IEEE80211_REJOIN: + case RTM_IEEE80211_ASSOC: + case RTM_IEEE80211_REASSOC: + v = cfg_GetAttribute(&cfg, CFG("dwds"), CFG("discover_on_join")); + if (!v || !strtol((char*) v, NULL, 0)) + break; + /* fall thru ... */ + case RTM_IEEE80211_WDS: + memcpy(&bssid, V(ieee80211_leave_event)->iev_addr, ETHER_ADDR_LEN); + VERB(1) syslog(LOG_INFO, "BSSID:%s WDS discovery\n", ether_ntoa(&bssid)); + if (!wifi_chkIface(ifan->ifan_name, ifs, nif)) + break; + if (!wifi_createWDS(ifan->ifan_name, bssid, wds)) { + cfg_LoadAttribute(&cfg, CFG("dwds"), CFG("name"), + CFG(szIdent), STRSIZ, DWDS_NAME); + /* create state file ... */ + v = cfg_GetAttribute(&cfg, CFG("dwds"), CFG("state_dir")); + if (v) { + memset(szCmd, 0, MAXPATHLEN); + snprintf(szCmd, MAXPATHLEN, "%s/%s-%s-%s", (char*) v, + (*wds)->if_name, ether_ntoa(&bssid), szIdent); + f = open(szCmd, O_WRONLY | O_CREAT, 0644); + if (f != -1) + close(f); + VERB(2) syslog(LOG_DEBUG, "Debug:: create session name %s\n", szCmd); + } + + /* Launch script ... */ + cfg_LoadAttribute(&cfg, CFG("dwds"), CFG("assoc_event"), + CFG(szStr), STRSIZ, NULL); + if (*szStr) { + memset(szCmd, 0, MAXPATHLEN); + snprintf(szCmd, MAXPATHLEN, "%s %s %s %s", szStr, + (*wds)->if_name, ether_ntoa(&bssid), szIdent); + VERB(3) syslog(LOG_DEBUG, "Debug:: Command line: %s\n", szCmd); + + if ((stat = system(szCmd))) + syslog(LOG_ERR, "VAP up script %s exited " + "with status %d\n", szStr, stat); + } + } + break; + } +#undef V + break; + } + return 0; } @@ -69,6 +191,7 @@ main(int argc, char **argv) int s; struct sigaction sa; size_t len; + struct dwds_if *wds = NULL; while ((ch = getopt(argc, argv, "hvfc:")) != -1) switch (ch) { @@ -101,7 +224,7 @@ main(int argc, char **argv) return 1; } - if (fg) + if (!fg) switch (fork()) { case -1: printf("Error:: when fork() #%d - %s\n", errno, strerror(errno)); @@ -134,13 +257,18 @@ main(int argc, char **argv) goto end; } + if (!(wds = wifi_buildWDS(s, ifs, nif))) { + syslog(LOG_ERR, "Error:: Go to dead ...\n"); + goto end; + } + while (!Kill) { len = read(s, (void*) msg, sizeof msg); if (len == -1) { - syslog(LOG_ERR, "Error:: read() #%d - %s\n", errno, strerror(errno)); + VERB(5) syslog(LOG_ERR, "Error:: read() #%d - %s\n", errno, strerror(errno)); Kill++; } else - RtMsg((struct rt_msghdr*) msg, len); + RtMsg(&wds, (struct rt_msghdr*) msg, len); } close(s);