]> git.sur5r.net Git - i3/i3status/blobdiff - src/print_wireless_info.c
add yajl compat code
[i3/i3status] / src / print_wireless_info.c
index 60eb90644454a408a232a065a76e4593ea149ae2..a75d28411122736f005406abce8775826a12c2ab 100644 (file)
@@ -1,9 +1,28 @@
 // vim:ts=8:expandtab
 #include <stdio.h>
 #include <string.h>
+#include <yajl/yajl_gen.h>
+#include <yajl/yajl_version.h>
 
 #ifdef LINUX
 #include <iwlib.h>
+#else
+#ifndef __FreeBSD__
+#define IW_ESSID_MAX_SIZE 32
+#endif
+#endif
+
+#ifdef __FreeBSD__
+#include <sys/param.h>
+#include <sys/ioctl.h>
+#include <sys/socket.h>
+#include <ifaddrs.h>
+#include <net/if.h>
+#include <net/if_media.h>
+#include <net80211/ieee80211.h>
+#include <net80211/ieee80211_ioctl.h>
+#include <unistd.h>
+#define IW_ESSID_MAX_SIZE IEEE80211_NWID_LEN
 #endif
 
 #include "i3status.h"
@@ -25,6 +44,7 @@ typedef struct {
         int signal_level_max;
         int noise_level;
         int noise_level_max;
+        int bitrate;
 } wireless_info_t;
 
 static int get_wireless_info(const char *interface, wireless_info_t *info) {
@@ -132,39 +152,108 @@ static int get_wireless_info(const char *interface, wireless_info_t *info) {
                 }
         }
 
+        struct iwreq wrq;
+        if (iw_get_ext(skfd, interface, SIOCGIWRATE, &wrq) >= 0)
+                info->bitrate = wrq.u.bitrate.value;
+
         close(skfd);
         return 1;
 #endif
+#ifdef __FreeBSD__
+        int s, len, inwid;
+        uint8_t buf[24 * 1024], *cp;
+        struct ieee80211req na;
+        char network_id[IEEE80211_NWID_LEN + 1];
+
+        if ((s = socket(AF_INET, SOCK_DGRAM, 0)) == -1)
+                return (0);
+
+        memset(&na, 0, sizeof(na));
+        strlcpy(na.i_name, interface, sizeof(na.i_name));
+        na.i_type = IEEE80211_IOC_SSID;
+        na.i_data = &info->essid[0];
+        na.i_len = IEEE80211_NWID_LEN + 1;
+        if ((inwid = ioctl(s, SIOCG80211, (caddr_t)&na)) == -1) {
+                close(s);
+                return (0);
+        }
+        if (inwid == 0) {
+                if (na.i_len <= IEEE80211_NWID_LEN)
+                        len = na.i_len + 1;
+                else
+                        len = IEEE80211_NWID_LEN + 1;
+                info->essid[len -1] = '\0';
+        } else {
+                close(s);
+                return (0);
+        }
+        info->flags |= WIRELESS_INFO_FLAG_HAS_ESSID;
+
+        memset(&na, 0, sizeof(na));
+        strlcpy(na.i_name, interface, sizeof(na.i_name));
+        na.i_type = IEEE80211_IOC_SCAN_RESULTS;
+        na.i_data = buf;
+        na.i_len = sizeof(buf);
+
+        if (ioctl(s, SIOCG80211, (caddr_t)&na) == -1) {
+                printf("fail\n");
+                close(s);
+                return (0);
+        }
+
+        close(s);
+        len = na.i_len;
+        cp = buf;
+        struct ieee80211req_scan_result *sr;
+        uint8_t *vp;
+        sr = (struct ieee80211req_scan_result *)cp;
+        vp = (u_int8_t *)(sr + 1);
+        strlcpy(network_id, (const char *)vp, sr->isr_ssid_len + 1);
+        if (!strcmp(network_id, &info->essid[0])) {
+                info->signal_level = sr->isr_rssi;
+                info->flags |= WIRELESS_INFO_FLAG_HAS_SIGNAL;
+                info->noise_level = sr->isr_noise;
+                info->flags |= WIRELESS_INFO_FLAG_HAS_NOISE;
+                info->quality = sr->isr_intval;
+                info->flags |= WIRELESS_INFO_FLAG_HAS_QUALITY;
+        }
+
+        return 1;
+#endif
+       return 0;
 }
 
-void print_wireless_info(const char *interface, const char *format_up, const char *format_down) {
+void print_wireless_info(yajl_gen json_gen, char *buffer, const char *interface, const char *format_up, const char *format_down) {
         const char *walk;
+        char *outwalk = buffer;
         wireless_info_t info;
+
+        INSTANCE(interface);
+
         if (get_wireless_info(interface, &info)) {
                 walk = format_up;
                 if (info.flags & WIRELESS_INFO_FLAG_HAS_QUALITY)
-                        printf("%s", info.quality < info.quality_average ? color("color_degraded") : color("color_good"));
+                        START_COLOR((info.quality < info.quality_average ? "color_degraded" : "color_good"));
         }
         else {
                 walk = format_down;
-                printf("%s", color("color_bad"));
+                START_COLOR("color_bad");
         }
 
         for (; *walk != '\0'; walk++) {
                 if (*walk != '%') {
-                        putchar(*walk);
+                        *(outwalk++) = *walk;
                         continue;
                 }
 
                 if (BEGINS_WITH(walk+1, "quality")) {
                         if (info.flags & WIRELESS_INFO_FLAG_HAS_QUALITY) {
                                 if (info.quality_max)
-                                        printf("%03d%%", PERCENT_VALUE(info.quality, info.quality_max));
+                                        outwalk += sprintf(outwalk, "%03d%%", PERCENT_VALUE(info.quality, info.quality_max));
                                 else
-                                        printf("%d", info.quality);
-                        }
-                        else {
-                                printf("no value");
+                                        outwalk += sprintf(outwalk, "%d", info.quality);
+                        } else {
+                                *(outwalk++) = '?';
                         }
                         walk += strlen("quality");
                 }
@@ -172,12 +261,11 @@ void print_wireless_info(const char *interface, const char *format_up, const cha
                 if (BEGINS_WITH(walk+1, "signal")) {
                         if (info.flags & WIRELESS_INFO_FLAG_HAS_SIGNAL) {
                                 if (info.signal_level_max)
-                                        printf("%03d%%", PERCENT_VALUE(info.signal_level, info.signal_level_max));
+                                        outwalk += sprintf(outwalk, "%03d%%", PERCENT_VALUE(info.signal_level, info.signal_level_max));
                                 else
-                                        printf("%d dBm", info.signal_level);
-                        }
-                        else {
-                                printf("no value");
+                                        outwalk += sprintf(outwalk, "%d dBm", info.signal_level);
+                        } else {
+                                *(outwalk++) = '?';
                         }
                         walk += strlen("signal");
                 }
@@ -185,32 +273,41 @@ void print_wireless_info(const char *interface, const char *format_up, const cha
                 if (BEGINS_WITH(walk+1, "noise")) {
                         if (info.flags & WIRELESS_INFO_FLAG_HAS_NOISE) {
                                 if (info.noise_level_max)
-                                        printf("%03d%%", PERCENT_VALUE(info.noise_level, info.noise_level_max));
+                                        outwalk += sprintf(outwalk, "%03d%%", PERCENT_VALUE(info.noise_level, info.noise_level_max));
                                 else
-                                        printf("%d dBm", info.noise_level);
-                        }
-                        else {
-                                printf("no value");
+                                        outwalk += sprintf(outwalk, "%d dBm", info.noise_level);
+                        } else {
+                                *(outwalk++) = '?';
                         }
                         walk += strlen("noise");
                 }
 
                 if (BEGINS_WITH(walk+1, "essid")) {
                         if (info.flags & WIRELESS_INFO_FLAG_HAS_ESSID)
-                                printf("%s", info.essid);
+                                outwalk += sprintf(outwalk, "%s", info.essid);
                         else
-                                printf("no value");
+                                *(outwalk++) = '?';
                         walk += strlen("essid");
                 }
 
                 if (BEGINS_WITH(walk+1, "ip")) {
                         const char *ip_address = get_ip_addr(interface);
-                        if (ip_address != NULL)
-                                (void)printf("%s", get_ip_addr(interface));
-                        else (void)printf("no IP");
+                        outwalk += sprintf(outwalk, "%s", (ip_address ? ip_address : "no IP"));
                         walk += strlen("ip");
                 }
+
+#ifdef LINUX
+                if (BEGINS_WITH(walk+1, "bitrate")) {
+                        char br_buffer[128];
+
+                        iw_print_bitrate(br_buffer, sizeof(br_buffer), info.bitrate);
+
+                        outwalk += sprintf(outwalk, "%s", br_buffer);
+                        walk += strlen("bitrate");
+                }
+#endif
         }
 
-        (void)printf("%s", endcolor());
+        END_COLOR;
+        OUTPUT_FULL_TEXT(buffer);
 }