Xinqi Bao's Git

netspeeds: added error condition for openbsd
[slstatus.git] / components / netspeeds.c
index ef8bf93..c846a2c 100644 (file)
                return fmt_scaled((txbytes - oldtxbytes) / interval * 1000);
        }
 #elif defined(__OpenBSD__)
-       /* unimplemented */
+       #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;
+               static uint64_t oldrxbytes;
+               uint64_t rxbytes = 0;
+               const char *rxs;
+               extern const unsigned int interval;
+               char if_ok = 0;
+
+               if (getifaddrs(&ifal) == -1) {
+                       warn("getifaddrs failed");
+                       return NULL;
+               }
+               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;
+               }
+
+               rxs = oldrxbytes ? fmt_scaled((rxbytes - oldrxbytes) /
+                                             interval * 1000) : NULL;
+               return (oldrxbytes = rxbytes, rxs);
+       }
+
+       const char *
+       netspeed_tx(const char *interface)
+       {
+               struct ifaddrs *ifal, *ifa;
+               struct if_data *ifd;
+               static uint64_t oldtxbytes;
+               uint64_t txbytes = 0;
+               const char *txs;
+               extern const unsigned int interval;
+               char if_ok = 0;
+
+               if (getifaddrs(&ifal) == -1) {
+                       warn("getifaddrs failed");
+                       return NULL;
+               }
+               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;
+               }
+
+               txs = oldtxbytes ? fmt_scaled((txbytes - oldtxbytes) /
+                                             interval * 1000) : NULL;
+               return (oldtxbytes = txbytes, txs);
+       }
 #endif