1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
|
#include "netio.h"
#include "common/netif.h"
#include "common/strutil.h"
#include <kstat.h>
static inline void kstatFreeWrap(kstat_ctl_t** pkc) {
assert(pkc);
if (*pkc) {
kstat_close(*pkc);
}
}
const char* ffNetIOGetIoCounters(FFlist* result, FFNetIOOptions* options) {
FF_A_CLEANUP(kstatFreeWrap) kstat_ctl_t* kc = kstat_open();
if (!kc) {
return "kstat_open() failed";
}
const char* defaultRouteIfName = ffNetifGetDefaultRouteV4()->ifName;
for (kstat_t* ks = kc->kc_chain; ks; ks = ks->ks_next) {
if (!ffStrEquals(ks->ks_class, "net") || !ffStrEquals(ks->ks_module, "link")) {
continue;
}
if (options->namePrefix.length && strncmp(ks->ks_name, options->namePrefix.chars, options->namePrefix.length) != 0) {
continue;
}
bool isDefaultRoute = ffStrEquals(ks->ks_name, defaultRouteIfName);
if (options->defaultRouteOnly && !isDefaultRoute) {
continue;
}
if (kstat_read(kc, ks, NULL) < 0) {
continue;
}
FFNetIOResult* counters = FF_LIST_ADD(FFNetIOResult, *result);
kstat_named_t* wbytes = (kstat_named_t*) kstat_data_lookup(ks, "obytes64");
kstat_named_t* rbytes = (kstat_named_t*) kstat_data_lookup(ks, "rbytes64");
kstat_named_t* wpkts = (kstat_named_t*) kstat_data_lookup(ks, "opackets64");
kstat_named_t* rpkts = (kstat_named_t*) kstat_data_lookup(ks, "ipackets64");
kstat_named_t* werrs = (kstat_named_t*) kstat_data_lookup(ks, "oerrors");
kstat_named_t* rerrs = (kstat_named_t*) kstat_data_lookup(ks, "ierrors");
*counters = (FFNetIOResult) {
.name = ffStrbufCreateS(ks->ks_name),
.txBytes = wbytes->value.ui64,
.rxBytes = rbytes->value.ui64,
.txPackets = wpkts->value.ui64,
.rxPackets = rpkts->value.ui64,
.txErrors = werrs->value.ui64,
.rxErrors = rerrs->value.ui64,
.txDrops = 0, // unsupported
.rxDrops = 0,
.defaultRoute = isDefaultRoute,
};
if (options->defaultRouteOnly) {
break;
}
}
return NULL;
}
|
