1 // vim:ts=4:sw=4:expandtab
4 #include <yajl/yajl_gen.h>
5 #include <yajl/yajl_version.h>
10 #include <netlink/netlink.h>
11 #include <netlink/genl/genl.h>
12 #include <netlink/genl/ctrl.h>
13 #include <linux/nl80211.h>
14 #include <linux/if_ether.h>
15 #define IW_ESSID_MAX_SIZE 32
19 #include <sys/socket.h>
20 #define IW_ESSID_MAX_SIZE 32
24 #include <sys/param.h>
25 #include <sys/ioctl.h>
26 #include <sys/socket.h>
29 #include <net/if_media.h>
30 #include <net80211/ieee80211.h>
31 #include <net80211/ieee80211_ioctl.h>
33 #define IW_ESSID_MAX_SIZE IEEE80211_NWID_LEN
37 #include <sys/param.h>
38 #include <sys/ioctl.h>
39 #include <sys/socket.h>
43 #include <net/if_media.h>
44 #include <netproto/802_11/ieee80211.h>
45 #include <netproto/802_11/ieee80211_ioctl.h>
47 #define IW_ESSID_MAX_SIZE IEEE80211_NWID_LEN
51 #include <sys/ioctl.h>
52 #include <sys/socket.h>
54 #include <sys/select.h>
55 #include <sys/types.h>
56 #include <netinet/in.h>
57 #include <netinet/if_ether.h>
58 #include <net80211/ieee80211.h>
59 #include <net80211/ieee80211_ioctl.h>
60 #define IW_ESSID_MAX_SIZE IEEE80211_NWID_LEN
64 #include <sys/types.h>
65 #include <net80211/ieee80211.h>
66 #define IW_ESSID_MAX_SIZE IEEE80211_NWID_LEN
71 #define WIRELESS_INFO_FLAG_HAS_ESSID (1 << 0)
72 #define WIRELESS_INFO_FLAG_HAS_QUALITY (1 << 1)
73 #define WIRELESS_INFO_FLAG_HAS_SIGNAL (1 << 2)
74 #define WIRELESS_INFO_FLAG_HAS_NOISE (1 << 3)
75 #define WIRELESS_INFO_FLAG_HAS_FREQUENCY (1 << 4)
77 #define PERCENT_VALUE(value, total) ((int)(value * 100 / (float)total + 0.5f))
81 #ifdef IW_ESSID_MAX_SIZE
82 char essid[IW_ESSID_MAX_SIZE + 1];
85 uint8_t bssid[ETH_ALEN];
99 // Like iw_print_bitrate, but without the dependency on libiw.
100 static void print_bitrate(char *buffer, int buflen, int bitrate) {
101 const int kilo = 1e3;
102 const int mega = 1e6;
103 const int giga = 1e9;
105 const double rate = bitrate;
112 } else if (rate >= mega) {
119 snprintf(buffer, buflen, "%g %cb/s", rate / divisor, scale);
122 // Based on NetworkManager/src/platform/wifi/wifi-utils-nl80211.c
123 static uint32_t nl80211_xbm_to_percent(int32_t xbm, int32_t divisor) {
124 #define NOISE_FLOOR_DBM -90
125 #define SIGNAL_MAX_DBM -20
128 if (xbm < NOISE_FLOOR_DBM)
129 xbm = NOISE_FLOOR_DBM;
130 if (xbm > SIGNAL_MAX_DBM)
131 xbm = SIGNAL_MAX_DBM;
133 return 100 - 70 * (((float)SIGNAL_MAX_DBM - (float)xbm) / ((float)SIGNAL_MAX_DBM - (float)NOISE_FLOOR_DBM));
136 // Based on NetworkManager/src/platform/wifi/wifi-utils-nl80211.c
137 static void find_ssid(uint8_t *ies, uint32_t ies_len, uint8_t **ssid, uint32_t *ssid_len) {
138 #define WLAN_EID_SSID 0
142 while (ies_len > 2 && ies[0] != WLAN_EID_SSID) {
143 ies_len -= ies[1] + 2;
148 if (ies_len < (uint32_t)(2 + ies[1]))
155 static int gwi_sta_cb(struct nl_msg *msg, void *data) {
156 wireless_info_t *info = data;
158 struct nlattr *tb[NL80211_ATTR_MAX + 1];
159 struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
160 struct nlattr *sinfo[NL80211_STA_INFO_MAX + 1];
161 struct nlattr *rinfo[NL80211_RATE_INFO_MAX + 1];
162 static struct nla_policy stats_policy[NL80211_STA_INFO_MAX + 1] = {
163 [NL80211_STA_INFO_RX_BITRATE] = {.type = NLA_NESTED},
166 static struct nla_policy rate_policy[NL80211_RATE_INFO_MAX + 1] = {
167 [NL80211_RATE_INFO_BITRATE] = {.type = NLA_U16},
170 if (nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0), genlmsg_attrlen(gnlh, 0), NULL) < 0)
173 if (tb[NL80211_ATTR_STA_INFO] == NULL)
176 if (nla_parse_nested(sinfo, NL80211_STA_INFO_MAX, tb[NL80211_ATTR_STA_INFO], stats_policy))
179 if (sinfo[NL80211_STA_INFO_RX_BITRATE] == NULL)
182 if (nla_parse_nested(rinfo, NL80211_RATE_INFO_MAX, sinfo[NL80211_STA_INFO_RX_BITRATE], rate_policy))
185 if (rinfo[NL80211_RATE_INFO_BITRATE] == NULL)
188 // NL80211_RATE_INFO_BITRATE is specified in units of 100 kbit/s, but iw
189 // used to specify bit/s, so we convert to use the same code path.
190 info->bitrate = (int)nla_get_u16(rinfo[NL80211_RATE_INFO_BITRATE]) * 100 * 1000;
195 static int gwi_scan_cb(struct nl_msg *msg, void *data) {
196 wireless_info_t *info = data;
197 struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
198 struct nlattr *tb[NL80211_ATTR_MAX + 1];
199 struct nlattr *bss[NL80211_BSS_MAX + 1];
200 struct nla_policy bss_policy[NL80211_BSS_MAX + 1] = {
201 [NL80211_BSS_FREQUENCY] = {.type = NLA_U32},
202 [NL80211_BSS_BSSID] = {.type = NLA_UNSPEC},
203 [NL80211_BSS_INFORMATION_ELEMENTS] = {.type = NLA_UNSPEC},
204 [NL80211_BSS_SIGNAL_MBM] = {.type = NLA_U32},
205 [NL80211_BSS_SIGNAL_UNSPEC] = {.type = NLA_U8},
206 [NL80211_BSS_STATUS] = {.type = NLA_U32},
209 if (nla_parse(tb, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0), genlmsg_attrlen(gnlh, 0), NULL) < 0)
212 if (tb[NL80211_ATTR_BSS] == NULL)
215 if (nla_parse_nested(bss, NL80211_BSS_MAX, tb[NL80211_ATTR_BSS], bss_policy))
218 if (bss[NL80211_BSS_STATUS] == NULL)
221 const uint32_t status = nla_get_u32(bss[NL80211_BSS_STATUS]);
223 if (status != NL80211_BSS_STATUS_ASSOCIATED &&
224 status != NL80211_BSS_STATUS_IBSS_JOINED)
227 if (bss[NL80211_BSS_BSSID] == NULL)
230 memcpy(info->bssid, nla_data(bss[NL80211_BSS_BSSID]), ETH_ALEN);
232 if (bss[NL80211_BSS_FREQUENCY]) {
233 info->flags |= WIRELESS_INFO_FLAG_HAS_FREQUENCY;
234 info->frequency = (double)nla_get_u32(bss[NL80211_BSS_FREQUENCY]) * 1e6;
237 if (bss[NL80211_BSS_SIGNAL_UNSPEC]) {
238 info->flags |= WIRELESS_INFO_FLAG_HAS_SIGNAL;
239 info->signal_level = nla_get_u8(bss[NL80211_BSS_SIGNAL_UNSPEC]);
240 info->signal_level_max = 100;
242 info->flags |= WIRELESS_INFO_FLAG_HAS_QUALITY;
243 info->quality = info->signal_level;
244 info->quality_max = 100;
245 info->quality_average = 50;
248 if (bss[NL80211_BSS_SIGNAL_MBM]) {
249 info->flags |= WIRELESS_INFO_FLAG_HAS_SIGNAL;
250 info->signal_level = (int)nla_get_u32(bss[NL80211_BSS_SIGNAL_MBM]) / 100;
252 info->flags |= WIRELESS_INFO_FLAG_HAS_QUALITY;
253 info->quality = nl80211_xbm_to_percent(nla_get_u32(bss[NL80211_BSS_SIGNAL_MBM]), 100);
254 info->quality_max = 100;
255 info->quality_average = 50;
258 if (bss[NL80211_BSS_INFORMATION_ELEMENTS]) {
262 find_ssid(nla_data(bss[NL80211_BSS_INFORMATION_ELEMENTS]),
263 nla_len(bss[NL80211_BSS_INFORMATION_ELEMENTS]),
265 if (ssid && ssid_len) {
266 info->flags |= WIRELESS_INFO_FLAG_HAS_ESSID;
267 snprintf(info->essid, sizeof(info->essid), "%.*s", ssid_len, ssid);
275 static int get_wireless_info(const char *interface, wireless_info_t *info) {
276 memset(info, 0, sizeof(wireless_info_t));
279 struct nl_sock *sk = nl_socket_alloc();
280 if (genl_connect(sk) != 0)
283 if (nl_socket_modify_cb(sk, NL_CB_VALID, NL_CB_CUSTOM, gwi_scan_cb, info) < 0)
286 const int nl80211_id = genl_ctrl_resolve(sk, "nl80211");
290 const unsigned int ifidx = if_nametoindex(interface);
294 struct nl_msg *msg = NULL;
295 if ((msg = nlmsg_alloc()) == NULL)
298 if (!genlmsg_put(msg, NL_AUTO_PORT, NL_AUTO_SEQ, nl80211_id, 0, NLM_F_DUMP, NL80211_CMD_GET_SCAN, 0) ||
299 nla_put_u32(msg, NL80211_ATTR_IFINDEX, ifidx) < 0)
302 if (nl_send_sync(sk, msg) < 0)
303 // nl_send_sync calls nlmsg_free()
307 if (nl_socket_modify_cb(sk, NL_CB_VALID, NL_CB_CUSTOM, gwi_sta_cb, info) < 0)
310 if ((msg = nlmsg_alloc()) == NULL)
313 if (!genlmsg_put(msg, NL_AUTO_PORT, NL_AUTO_SEQ, nl80211_id, 0, NLM_F_DUMP, NL80211_CMD_GET_STATION, 0) || nla_put_u32(msg, NL80211_ATTR_IFINDEX, ifidx) < 0 || nla_put(msg, NL80211_ATTR_MAC, 6, info->bssid) < 0)
316 if (nl_send_sync(sk, msg) < 0)
317 // nl_send_sync calls nlmsg_free()
331 #if defined(__FreeBSD__) || defined(__DragonFly__)
334 struct ieee80211req_sta_req req;
335 uint8_t buf[24 * 1024];
337 struct ieee80211req na;
338 char bssid[IEEE80211_ADDR_LEN];
341 if ((s = socket(AF_INET, SOCK_DGRAM, 0)) == -1)
344 memset(&na, 0, sizeof(na));
345 strlcpy(na.i_name, interface, sizeof(na.i_name));
346 na.i_type = IEEE80211_IOC_SSID;
347 na.i_data = &info->essid[0];
348 na.i_len = IEEE80211_NWID_LEN + 1;
349 if ((inwid = ioctl(s, SIOCG80211, (caddr_t)&na)) == -1) {
354 if (na.i_len <= IEEE80211_NWID_LEN)
357 len = IEEE80211_NWID_LEN + 1;
358 info->essid[len - 1] = '\0';
363 info->flags |= WIRELESS_INFO_FLAG_HAS_ESSID;
365 memset(&na, 0, sizeof(na));
366 strlcpy(na.i_name, interface, sizeof(na.i_name));
367 na.i_type = IEEE80211_IOC_BSSID;
369 na.i_len = sizeof(bssid);
371 if (ioctl(s, SIOCG80211, (caddr_t)&na) == -1) {
376 memcpy(u.req.is_u.macaddr, bssid, sizeof(bssid));
377 memset(&na, 0, sizeof(na));
378 strlcpy(na.i_name, interface, sizeof(na.i_name));
379 na.i_type = IEEE80211_IOC_STA_INFO;
381 na.i_len = sizeof(u);
383 if (ioctl(s, SIOCG80211, (caddr_t)&na) == -1) {
389 if (na.i_len >= sizeof(u.req)) {
391 * Just use the first BSSID returned even if there are
392 * multiple APs sharing the same BSSID.
394 info->signal_level = u.req.info[0].isi_rssi / 2 +
395 u.req.info[0].isi_noise;
396 info->flags |= WIRELESS_INFO_FLAG_HAS_SIGNAL;
397 info->noise_level = u.req.info[0].isi_noise;
398 info->flags |= WIRELESS_INFO_FLAG_HAS_NOISE;
405 struct ieee80211_bssid bssid;
406 struct ieee80211_nwid nwid;
407 struct ieee80211_nodereq nr;
409 struct ether_addr ea;
411 int s, len, ibssid, inwid;
412 u_int8_t zero_bssid[IEEE80211_ADDR_LEN];
414 if ((s = socket(AF_INET, SOCK_DGRAM, 0)) == -1)
417 memset(&ifr, 0, sizeof(ifr));
418 ifr.ifr_data = (caddr_t)&nwid;
419 (void)strlcpy(ifr.ifr_name, interface, sizeof(ifr.ifr_name));
420 inwid = ioctl(s, SIOCG80211NWID, (caddr_t)&ifr);
422 memset(&bssid, 0, sizeof(bssid));
423 strlcpy(bssid.i_name, interface, sizeof(bssid.i_name));
424 ibssid = ioctl(s, SIOCG80211BSSID, &bssid);
426 if (ibssid != 0 || inwid != 0) {
433 if (nwid.i_len <= IEEE80211_NWID_LEN)
434 len = nwid.i_len + 1;
436 len = IEEE80211_NWID_LEN + 1;
438 strncpy(&info->essid[0], (char *)nwid.i_nwid, len);
439 info->essid[IW_ESSID_MAX_SIZE] = '\0';
440 info->flags |= WIRELESS_INFO_FLAG_HAS_ESSID;
443 /* Signal strength */
445 memset(&zero_bssid, 0, sizeof(zero_bssid));
446 if (ibssid == 0 && memcmp(bssid.i_bssid, zero_bssid, IEEE80211_ADDR_LEN) != 0) {
447 memcpy(&ea.ether_addr_octet, bssid.i_bssid, sizeof(ea.ether_addr_octet));
449 bzero(&nr, sizeof(nr));
450 bcopy(bssid.i_bssid, &nr.nr_macaddr, sizeof(nr.nr_macaddr));
451 strlcpy(nr.nr_ifname, interface, sizeof(nr.nr_ifname));
453 if (ioctl(s, SIOCG80211NODE, &nr) == 0 && nr.nr_rssi) {
454 info->signal_level = nr.nr_rssi;
456 info->signal_level_max = nr.nr_max_rssi;
458 info->flags |= WIRELESS_INFO_FLAG_HAS_SIGNAL;
469 /* Table summarizing what is the decision to prefer IPv4 or IPv6
470 * based their values.
472 * | ipv4_address | ipv6_address | Chosen IP | Color |
473 * |--------------|--------------|-----------|-------------------|
474 * | NULL | NULL | None | bad (red) |
475 * | NULL | no IP | IPv6 | degraded (orange) |
476 * | NULL | ::1/128 | IPv6 | ok (green) |
477 * | no IP | NULL | IPv4 | degraded |
478 * | no IP | no IP | IPv4 | degraded |
479 * | no IP | ::1/128 | IPv6 | ok |
480 * | 127.0.0.1 | NULL | IPv4 | ok |
481 * | 127.0.0.1 | no IP | IPv4 | ok |
482 * | 127.0.0.1 | ::1/128 | IPv4 | ok |
484 void print_wireless_info(yajl_gen json_gen, char *buffer, const char *interface, const char *format_up, const char *format_down, const char *format_quality) {
486 char *outwalk = buffer;
487 wireless_info_t info;
491 char *ipv4_address = sstrdup(get_ip_addr(interface, AF_INET));
492 char *ipv6_address = sstrdup(get_ip_addr(interface, AF_INET6));
495 * Removing '%' and following characters from IPv6 since the interface identifier is redundant,
496 * as the output already includes the interface name.
498 if (ipv6_address != NULL) {
499 char *prct_ptr = strstr(ipv6_address, "%");
500 if (prct_ptr != NULL) {
505 bool prefer_ipv4 = true;
506 if (ipv4_address == NULL) {
507 if (ipv6_address == NULL) {
508 START_COLOR("color_bad");
509 outwalk += sprintf(outwalk, "%s", format_down);
514 } else if (BEGINS_WITH(ipv4_address, "no IP") && ipv6_address != NULL && !BEGINS_WITH(ipv6_address, "no IP")) {
518 const char *ip_address = (prefer_ipv4) ? ipv4_address : ipv6_address;
519 if (!get_wireless_info(interface, &info)) {
521 START_COLOR("color_bad");
524 if (info.flags & WIRELESS_INFO_FLAG_HAS_QUALITY)
525 START_COLOR((info.quality < info.quality_average ? "color_degraded" : "color_good"));
527 if (BEGINS_WITH(ip_address, "no IP")) {
528 START_COLOR("color_degraded");
530 START_COLOR("color_good");
535 for (; *walk != '\0'; walk++) {
537 *(outwalk++) = *walk;
539 } else if (BEGINS_WITH(walk + 1, "quality")) {
540 if (info.flags & WIRELESS_INFO_FLAG_HAS_QUALITY) {
541 if (info.quality_max)
542 outwalk += sprintf(outwalk, format_quality, PERCENT_VALUE(info.quality, info.quality_max), pct_mark);
544 outwalk += sprintf(outwalk, "%d", info.quality);
548 walk += strlen("quality");
550 } else if (BEGINS_WITH(walk + 1, "signal")) {
551 if (info.flags & WIRELESS_INFO_FLAG_HAS_SIGNAL) {
552 if (info.signal_level_max)
553 outwalk += sprintf(outwalk, "%3d%s", PERCENT_VALUE(info.signal_level, info.signal_level_max), pct_mark);
555 outwalk += sprintf(outwalk, "%d dBm", info.signal_level);
559 walk += strlen("signal");
561 } else if (BEGINS_WITH(walk + 1, "noise")) {
562 if (info.flags & WIRELESS_INFO_FLAG_HAS_NOISE) {
563 if (info.noise_level_max)
564 outwalk += sprintf(outwalk, "%3d%s", PERCENT_VALUE(info.noise_level, info.noise_level_max), pct_mark);
566 outwalk += sprintf(outwalk, "%d dBm", info.noise_level);
570 walk += strlen("noise");
572 } else if (BEGINS_WITH(walk + 1, "essid")) {
573 #ifdef IW_ESSID_MAX_SIZE
574 if (info.flags & WIRELESS_INFO_FLAG_HAS_ESSID)
575 maybe_escape_markup(info.essid, &outwalk);
579 walk += strlen("essid");
581 } else if (BEGINS_WITH(walk + 1, "frequency")) {
582 if (info.flags & WIRELESS_INFO_FLAG_HAS_FREQUENCY)
583 outwalk += sprintf(outwalk, "%1.1f GHz", info.frequency / 1e9);
586 walk += strlen("frequency");
588 } else if (BEGINS_WITH(walk + 1, "ip")) {
589 outwalk += sprintf(outwalk, "%s", ip_address);
590 walk += strlen("ip");
593 else if (BEGINS_WITH(walk + 1, "bitrate")) {
596 print_bitrate(br_buffer, sizeof(br_buffer), info.bitrate);
598 outwalk += sprintf(outwalk, "%s", br_buffer);
599 walk += strlen("bitrate");
611 OUTPUT_FULL_TEXT(buffer);