X-Git-Url: https://git.xinqibao.xyz/slstatus.git/blobdiff_plain/f3c23f08bb2b7210decdf695b374fb8ee14e1761..9ac721c23fb640de2a6d1f84c84a79b2ccc26691:/components/wifi.c diff --git a/components/wifi.c b/components/wifi.c index 78acb3f..92c252e 100644 --- a/components/wifi.c +++ b/components/wifi.c @@ -1,44 +1,48 @@ /* See LICENSE file for copyright and license details. */ -#include #include #include #include -#include #include +#include #include #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; - snprintf(path, sizeof(path), "%s%s%s", "/sys/class/net/", iface, - "/operstate"); + if (esnprintf(path, sizeof(path), "/sys/class/net/%s/operstate", + interface) < 0) { + return NULL; + } if (!(fp = fopen(path, "r"))) { - fprintf(stderr, "fopen '%s': %s\n", path, - strerror(errno)); + warn("fopen '%s':", path); return NULL; } p = fgets(status, 5, fp); fclose(fp); - if(!p || strcmp(status, "up\n") != 0) { + if (!p || strcmp(status, "up\n") != 0) { return NULL; } if (!(fp = fopen("/proc/net/wireless", "r"))) { - fprintf(stderr, "fopen '/proc/net/wireless': %s\n", - strerror(errno)); + warn("fopen '/proc/net/wireless':"); return NULL; } @@ -51,36 +55,39 @@ 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 = socket(AF_INET, SOCK_DGRAM, 0); + int sockfd; struct iwreq wreq; memset(&wreq, 0, sizeof(struct iwreq)); wreq.u.essid.length = IW_ESSID_MAX_SIZE+1; - snprintf(wreq.ifr_name, sizeof(wreq.ifr_name), "%s", iface); + if (esnprintf(wreq.ifr_name, sizeof(wreq.ifr_name), "%s", + interface) < 0) { + return NULL; + } - if (sockfd < 0) { - fprintf(stderr, "socket 'AF_INET': %s\n", - strerror(errno)); + if ((sockfd = socket(AF_INET, SOCK_DGRAM, 0)) < 0) { + warn("socket 'AF_INET':"); return NULL; } wreq.u.essid.pointer = id; if (ioctl(sockfd,SIOCGIWESSID, &wreq) < 0) { - fprintf(stderr, "ioctl 'SIOCGIWESSID': %s\n", strerror(errno)); + warn("ioctl 'SIOCGIWESSID':"); close(sockfd); return NULL; } @@ -97,67 +104,169 @@ #include #include #include + #include /* before for NBBY */ #include #include #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; - memset(&bssid, 0, sizeof(bssid); + uint8_t zero_bssid[IEEE80211_ADDR_LEN]; + + memset(&bssid, 0, sizeof(bssid)); memset(nr, 0, sizeof(struct ieee80211_nodereq)); - if ((sockfd = socket(AF_INET, SOCK_DGRAM, 0)) == -1) { - fprintf(stderr, "socket 'AF_INET': %s\n", - strerror(errno)); + if ((sockfd = socket(AF_INET, SOCK_DGRAM, 0)) < 0) { + warn("socket 'AF_INET':"); return 0; } - strlcpy(bssid.i_name, iface, sizeof(bssid.i_name)); - if ((ioctl(sockfd, SIOCG80211BSSID, &bssid)) == -1) { - fprintf(stderr, "ioctl 'SIOCG80211BSSID': %s\n", - strerror(errno)); + strlcpy(bssid.i_name, interface, sizeof(bssid.i_name)); + if ((ioctl(sockfd, SIOCG80211BSSID, &bssid)) < 0) { + warn("ioctl 'SIOCG80211BSSID':"); close(sockfd); return 0; } - strlcpy(nr->nr_ifname, iface, sizeof(nr->nr_ifname)); - memmove(&nr->nr_macaddr, bssid.i_bssid, sizeof(nr->nr_macaddr)); - if ((ioctl(sockfd, SIOCG80211NODE, nr)) == -1 && nr->nr_rssi) { - fprintf(stderr, "ioctl 'SIOCG80211NODE': %s\n", - strerror(errno)); + memset(&zero_bssid, 0, sizeof(zero_bssid)); + if (memcmp(bssid.i_bssid, zero_bssid, + IEEE80211_ADDR_LEN) == 0) { + close(sockfd); + return 0; + } + 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':"); close(sockfd); return 0; } - return close(sockfd), 1; + return close(sockfd), 1; } 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 (nr.nr_max_rssi) + 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))); + } else { + q = RSSI_TO_PERC(nr.nr_rssi); + } return bprintf("%d", q); } + return NULL; } 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