Use `int` for flags

[slstatus.git] / components / netspeeds.c

diff --git a/components/netspeeds.c b/components/netspeeds.c
index ef8bf93..b4e6972 100644 (file)
--- a/components/netspeeds.c
+++ b/components/netspeeds.c
@@ -5,49 +5,117 @@
 #include "../util.h"
 
 #if defined(__linux__)
 #include "../util.h"
 
 #if defined(__linux__)

+       #include <stdint.h>
+

        const char *
        netspeed_rx(const char *interface)
        {
        const char *
        netspeed_rx(const char *interface)
        {

-               static int valid;
-               static unsigned long long rxbytes;
-               unsigned long oldrxbytes;
+               uint64_t oldrxbytes;
+               static uint64_t rxbytes = 0;

                extern const unsigned int interval;
                char path[PATH_MAX];
 
                oldrxbytes = rxbytes;
                extern const unsigned int interval;
                char path[PATH_MAX];
 
                oldrxbytes = rxbytes;

-               snprintf(path, sizeof(path), "/sys/class/net/%s/statistics/rx_bytes", interface);
+
+               snprintf(path, sizeof(path),
+                        "/sys/class/net/%s/statistics/rx_bytes", interface);

                if (pscanf(path, "%llu", &rxbytes) != 1) {
                        return NULL;
                }
                if (pscanf(path, "%llu", &rxbytes) != 1) {
                        return NULL;
                }

-               if (!valid) {
-                       valid = 1;
-                       return NULL;
-               }

 
 

-               return fmt_scaled((rxbytes - oldrxbytes) / interval * 1000);
+               return oldrxbytes ? fmt_scaled((rxbytes - oldrxbytes) /
+                                              interval * 1000) : NULL;

        }
 
        const char *
        netspeed_tx(const char *interface)
        {
        }
 
        const char *
        netspeed_tx(const char *interface)
        {

-               static int valid;
-               static unsigned long long txbytes;
-               unsigned long oldtxbytes;
+               uint64_t oldtxbytes;
+               static uint64_t txbytes = 0;

                extern const unsigned int interval;
                char path[PATH_MAX];
 
                oldtxbytes = txbytes;
                extern const unsigned int interval;
                char path[PATH_MAX];
 
                oldtxbytes = txbytes;

-               snprintf(path, sizeof(path), "/sys/class/net/%s/statistics/tx_bytes", interface);
+
+               snprintf(path, sizeof(path),
+                        "/sys/class/net/%s/statistics/tx_bytes", interface);

                if (pscanf(path, "%llu", &txbytes) != 1) {
                        return NULL;
                }
                if (pscanf(path, "%llu", &txbytes) != 1) {
                        return NULL;
                }

-               if (!valid) {
-                       valid = 1;
+
+               return oldtxbytes ? fmt_scaled((txbytes - oldtxbytes) /
+                                              interval * 1000) : NULL;
+       }
+#elif defined(__OpenBSD__)
+       #include <string.h>
+       #include <ifaddrs.h>
+       #include <sys/types.h>
+       #include <sys/socket.h>
+       #include <net/if.h>
+
+       const char *
+       netspeed_rx(const char *interface)
+       {
+               struct ifaddrs *ifal, *ifa;
+               struct if_data *ifd;
+               uint64_t oldrxbytes;
+               static uint64_t rxbytes = 0;
+               extern const unsigned int interval;
+               int if_ok = 0;
+
+               oldrxbytes = rxbytes;
+
+               if (getifaddrs(&ifal) == -1) {
+                       warn("getifaddrs failed");
+                       return NULL;
+               }
+               rxbytes = 0;
+               for (ifa = ifal; ifa; ifa = ifa->ifa_next) {
+                       if (!strcmp(ifa->ifa_name, interface) &&
+                          (ifd = (struct if_data *)ifa->ifa_data)) {
+                               rxbytes += ifd->ifi_ibytes, if_ok = 1;
+                       }
+               }
+               freeifaddrs(ifal);
+               if (!if_ok) {
+                       warn("reading 'if_data' failed");

                        return NULL;
                }
 
                        return NULL;
                }
 

-               return fmt_scaled((txbytes - oldtxbytes) / interval * 1000);
+               return oldrxbytes ? fmt_scaled((rxbytes - oldrxbytes) /
+                                              interval * 1000) : NULL;
+       }
+
+       const char *
+       netspeed_tx(const char *interface)
+       {
+               struct ifaddrs *ifal, *ifa;
+               struct if_data *ifd;
+               uint64_t oldtxbytes;
+               static uint64_t txbytes = 0;
+               extern const unsigned int interval;
+               int if_ok = 0;
+
+               oldtxbytes = txbytes;
+
+               if (getifaddrs(&ifal) == -1) {
+                       warn("getifaddrs failed");
+                       return NULL;
+               }
+               txbytes = 0;
+               for (ifa = ifal; ifa; ifa = ifa->ifa_next) {
+                       if (!strcmp(ifa->ifa_name, interface) &&
+                          (ifd = (struct if_data *)ifa->ifa_data)) {
+                               txbytes += ifd->ifi_obytes, if_ok = 1;
+                       }
+               }
+               freeifaddrs(ifal);
+               if (!if_ok) {
+                       warn("reading 'if_data' failed");
+                       return NULL;
+               }
+
+               return oldtxbytes ? fmt_scaled((txbytes - oldtxbytes) /
+                                              interval * 1000) : NULL;

        }
        }

-#elif defined(__OpenBSD__)
-       /* unimplemented */

 #endif
 #endif