slstatus/components/netspeeds.c
2021-08-07 01:31:57 -04:00

139 lines
2.9 KiB
C

/* See LICENSE file for copyright and license details. */
#include <stdio.h>
#include <limits.h>
#include "../util.h"
#if defined(__linux__)
#include <stdint.h>
const char *
netspeed_rx(const char *interface)
{
uintmax_t oldrxbytes;
static uintmax_t rxbytes;
extern const unsigned int interval;
char path[PATH_MAX];
oldrxbytes = rxbytes;
if (esnprintf(path, sizeof(path),
"/sys/class/net/%s/statistics/rx_bytes",
interface) < 0) {
return NULL;
}
if (pscanf(path, "%ju", &rxbytes) != 1) {
return NULL;
}
if (oldrxbytes == 0) {
return NULL;
}
return fmt_human((rxbytes - oldrxbytes) * 1000 / interval,
1024);
}
const char *
netspeed_tx(const char *interface)
{
uintmax_t oldtxbytes;
static uintmax_t txbytes;
extern const unsigned int interval;
char path[PATH_MAX];
oldtxbytes = txbytes;
if (esnprintf(path, sizeof(path),
"/sys/class/net/%s/statistics/tx_bytes",
interface) < 0) {
return NULL;
}
if (pscanf(path, "%ju", &txbytes) != 1) {
return NULL;
}
if (oldtxbytes == 0) {
return NULL;
}
return fmt_human((txbytes - oldtxbytes) * 1000 / interval,
1024);
}
#elif defined(__OpenBSD__) | defined(__FreeBSD__)
#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;
uintmax_t oldrxbytes;
static uintmax_t rxbytes;
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;
}
if (oldrxbytes == 0) {
return NULL;
}
return fmt_human((rxbytes - oldrxbytes) * 1000 / interval,
1024);
}
const char *
netspeed_tx(const char *interface)
{
struct ifaddrs *ifal, *ifa;
struct if_data *ifd;
uintmax_t oldtxbytes;
static uintmax_t txbytes;
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;
}
if (oldtxbytes == 0) {
return NULL;
}
return fmt_human((txbytes - oldtxbytes) * 1000 / interval,
1024);
}
#endif