69b2487650
Long, wrapped, multi-line if statements suck to read. This fixes readability in the worst places by packing format strings for paths into defines.
138 lines
2.9 KiB
C
138 lines
2.9 KiB
C
/* See LICENSE file for copyright and license details. */
|
|
#include <stdio.h>
|
|
#include <limits.h>
|
|
|
|
#include "../util.h"
|
|
#include "../slstatus.h"
|
|
|
|
#if defined(__linux__)
|
|
#include <stdint.h>
|
|
|
|
#define NET_RX_BYTES "/sys/class/net/%s/statistics/rx_bytes"
|
|
#define NET_TX_BYTES "/sys/class/net/%s/statistics/tx_bytes"
|
|
|
|
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), NET_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), NET_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
|