X-Git-Url: https://git.xinqibao.xyz/slstatus.git/blobdiff_plain/8c8091f4f165974c44d75dd9ccc7cc4dcd1d1c96..a9193a757a510e07e69b18ca64d2e50d467f01c1:/components/wifi.c diff --git a/components/wifi.c b/components/wifi.c index 3751ae3..92c252e 100644 --- a/components/wifi.c +++ b/components/wifi.c @@ -8,23 +8,27 @@ #include "../util.h" +#define RSSI_TO_PERC(rssi) \ + rssi >= -50 ? 100 : \ + (rssi <= -100 ? 0 : \ + (2 * (rssi + 100))) + #if defined(__linux__) #include #include const char * - wifi_perc(const char *iface) + wifi_perc(const char *interface) { - int i, cur; - int total = 70; /* the max of /proc/net/wireless */ + int cur; + size_t i; char *p, *datastart; char path[PATH_MAX]; char status[5]; FILE *fp; - if (esnprintf(path, sizeof(path), - "/sys/class/net/%s/operstate", - iface) < 0) { + if (esnprintf(path, sizeof(path), "/sys/class/net/%s/operstate", + interface) < 0) { return NULL; } if (!(fp = fopen(path, "r"))) { @@ -33,7 +37,7 @@ } p = fgets(status, 5, fp); fclose(fp); - if(!p || strcmp(status, "up\n") != 0) { + if (!p || strcmp(status, "up\n") != 0) { return NULL; } @@ -51,19 +55,20 @@ return NULL; } - if (!(datastart = strstr(buf, iface))) { + if (!(datastart = strstr(buf, interface))) { return NULL; } - datastart = (datastart+(strlen(iface)+1)); + datastart = (datastart+(strlen(interface)+1)); sscanf(datastart + 1, " %*d %d %*d %*d\t\t %*d\t " "%*d\t\t%*d\t\t %*d\t %*d\t\t %*d", &cur); - return bprintf("%d", (int)((float)cur / total * 100)); + /* 70 is the max of /proc/net/wireless */ + return bprintf("%d", (int)((float)cur / 70 * 100)); } const char * - wifi_essid(const char *iface) + wifi_essid(const char *interface) { static char id[IW_ESSID_MAX_SIZE+1]; int sockfd; @@ -71,8 +76,8 @@ memset(&wreq, 0, sizeof(struct iwreq)); wreq.u.essid.length = IW_ESSID_MAX_SIZE+1; - if (esnprintf(wreq.ifr_name, sizeof(wreq.ifr_name), - "%s", iface) < 0) { + if (esnprintf(wreq.ifr_name, sizeof(wreq.ifr_name), "%s", + interface) < 0) { return NULL; } @@ -105,7 +110,7 @@ #include static int - load_ieee80211_nodereq(const char *iface, struct ieee80211_nodereq *nr) + load_ieee80211_nodereq(const char *interface, struct ieee80211_nodereq *nr) { struct ieee80211_bssid bssid; int sockfd; @@ -117,7 +122,7 @@ warn("socket 'AF_INET':"); return 0; } - strlcpy(bssid.i_name, iface, sizeof(bssid.i_name)); + strlcpy(bssid.i_name, interface, sizeof(bssid.i_name)); if ((ioctl(sockfd, SIOCG80211BSSID, &bssid)) < 0) { warn("ioctl 'SIOCG80211BSSID':"); close(sockfd); @@ -129,7 +134,7 @@ close(sockfd); return 0; } - strlcpy(nr->nr_ifname, iface, sizeof(nr->nr_ifname)); + strlcpy(nr->nr_ifname, interface, sizeof(nr->nr_ifname)); memcpy(&nr->nr_macaddr, bssid.i_bssid, sizeof(nr->nr_macaddr)); if ((ioctl(sockfd, SIOCG80211NODE, nr)) < 0 && nr->nr_rssi) { warn("ioctl 'SIOCG80211NODE':"); @@ -141,17 +146,16 @@ } const char * - wifi_perc(const char *iface) + wifi_perc(const char *interface) { struct ieee80211_nodereq nr; int q; - if (load_ieee80211_nodereq(iface, &nr)) { + if (load_ieee80211_nodereq(interface, &nr)) { if (nr.nr_max_rssi) { q = IEEE80211_NODEREQ_RSSI(&nr); } else { - q = nr.nr_rssi >= -50 ? 100 : (nr.nr_rssi <= -100 ? 0 : - (2 * (nr.nr_rssi + 100))); + q = RSSI_TO_PERC(nr.nr_rssi); } return bprintf("%d", q); } @@ -160,14 +164,109 @@ } const char * - wifi_essid(const char *iface) + wifi_essid(const char *interface) { struct ieee80211_nodereq nr; - if (load_ieee80211_nodereq(iface, &nr)) { + if (load_ieee80211_nodereq(interface, &nr)) { return bprintf("%s", nr.nr_nwid); } return NULL; } +#elif defined(__FreeBSD__) + #include + #include + + int + load_ieee80211req(int sock, const char *interface, void *data, int type, size_t *len) + { + char warn_buf[256]; + struct ieee80211req ireq; + memset(&ireq, 0, sizeof(ireq)); + ireq.i_type = type; + ireq.i_data = (caddr_t) data; + ireq.i_len = *len; + + strlcpy(ireq.i_name, interface, sizeof(ireq.i_name)); + if (ioctl(sock, SIOCG80211, &ireq) < 0) { + snprintf(warn_buf, sizeof(warn_buf), + "ioctl: 'SIOCG80211': %d", type); + warn(warn_buf); + return 0; + } + + *len = ireq.i_len; + return 1; + } + + const char * + wifi_perc(const char *interface) + { + union { + struct ieee80211req_sta_req sta; + uint8_t buf[24 * 1024]; + } info; + uint8_t bssid[IEEE80211_ADDR_LEN]; + int rssi_dbm; + int sockfd; + size_t len; + const char *fmt; + + if ((sockfd = socket(AF_INET, SOCK_DGRAM, 0)) < 0) { + warn("socket 'AF_INET':"); + return NULL; + } + + /* Retreive MAC address of interface */ + len = IEEE80211_ADDR_LEN; + fmt = NULL; + if (load_ieee80211req(sockfd, interface, &bssid, IEEE80211_IOC_BSSID, &len)) + { + /* Retrieve info on station with above BSSID */ + memset(&info, 0, sizeof(info)); + memcpy(info.sta.is_u.macaddr, bssid, sizeof(bssid)); + + len = sizeof(info); + if (load_ieee80211req(sockfd, interface, &info, IEEE80211_IOC_STA_INFO, &len)) { + rssi_dbm = info.sta.info[0].isi_noise + + info.sta.info[0].isi_rssi / 2; + + fmt = bprintf("%d", RSSI_TO_PERC(rssi_dbm)); + } + } + + close(sockfd); + return fmt; + } + + const char * + wifi_essid(const char *interface) + { + char ssid[IEEE80211_NWID_LEN + 1]; + size_t len; + int sockfd; + const char *fmt; + + if ((sockfd = socket(AF_INET, SOCK_DGRAM, 0)) < 0) { + warn("socket 'AF_INET':"); + return NULL; + } + + fmt = NULL; + len = sizeof(ssid); + memset(&ssid, 0, len); + if (load_ieee80211req(sockfd, interface, &ssid, IEEE80211_IOC_SSID, &len )) { + if (len < sizeof(ssid)) + len += 1; + else + len = sizeof(ssid); + + ssid[len - 1] = '\0'; + fmt = bprintf("%s", ssid); + } + + close(sockfd); + return fmt; + } #endif