homeio

1import { describe, expect, it } from "vitest";
2import {
3  createNetworkRateTracker,
4  parseDefaultRouteInterface,
5  readInterfaceAddresses,
6  readInterfaceCounters,
7} from "@/lib/server/modules/system/network-activity";
8
9// Real layout of /proc/net/route (tab separated, addresses in little-endian hex).
10const ROUTES = [
11  "Iface\tDestination\tGateway \tFlags\tRefCnt\tUse\tMetric\tMask\t\tMTU\tWindow\tIRTT",
12  "wlan0\t00000000\t0101A8C0\t0003\t0\t0\t600\t00000000\t0\t0\t0",
13  "eth0\t00000000\t0101A8C0\t0003\t0\t0\t100\t00000000\t0\t0\t0",
14  "eth0\t0001A8C0\t00000000\t0001\t0\t0\t100\t00FFFFFF\t0\t0\t0",
15  "docker0\t000011AC\t00000000\t0001\t0\t0\t0\t0000FFFF\t0\t0\t0",
16].join("\n");
17
18describe("network activity", () => {
19  it("finds the interface carrying the default route, lowest metric first", () => {
20    expect(parseDefaultRouteInterface(ROUTES)).toBe("eth0");
21  });
22
23  it("finds no default route on a machine without one", () => {
24    expect(parseDefaultRouteInterface(ROUTES.split("\n").slice(0, 1).join("\n"))).toBeNull();
25    expect(parseDefaultRouteInterface("")).toBeNull();
26  });
27
28  it("refuses interface names that could leave /sys/class/net", async () => {
29    expect(await readInterfaceCounters("../../etc")).toBeNull();
30  });
31
32  it("turns two readings of an interface into rates", () => {
33    const track = createNetworkRateTracker();
34
35    expect(track("eth0", { rxBytes: 0, txBytes: 0 }, 10_000)).toBeNull();
36    expect(track("eth0", { rxBytes: 6_000_000, txBytes: 2_000_000 }, 12_000)).toEqual({
37      rxBytesPerSec: 3_000_000,
38      txBytesPerSec: 1_000_000,
39    });
40  });
41
42  it("starts a new baseline when the interface changes or the counters reset", () => {
43    const track = createNetworkRateTracker();
44    track("eth0", { rxBytes: 5_000, txBytes: 5_000 }, 10_000);
45
46    expect(track("wlan0", { rxBytes: 9_000, txBytes: 9_000 }, 12_000)).toBeNull();
47    expect(track("wlan0", { rxBytes: 100, txBytes: 100 }, 14_000)).toBeNull();
48    expect(track("wlan0", { rxBytes: 2_100, txBytes: 1_100 }, 16_000)).toEqual({
49      rxBytesPerSec: 1_000,
50      txBytesPerSec: 500,
51    });
52  });
53
54  it("gives no rate across a long gap", () => {
55    const track = createNetworkRateTracker(60_000);
56    track("eth0", { rxBytes: 0, txBytes: 0 }, 0);
57
58    expect(track("eth0", { rxBytes: 1_000, txBytes: 1_000 }, 61_000)).toBeNull();
59  });
60
61  it("reads an interface's addresses, preferring a global IPv6 over link-local", () => {
62    const interfaces = {
63      eth0: [
64        { address: "fe80::1", family: "IPv6", internal: false },
65        { address: "192.168.1.27", family: "IPv4", internal: false },
66        { address: "2a01:cb00::27", family: "IPv6", internal: false },
67      ],
68      lo: [{ address: "127.0.0.1", family: "IPv4", internal: true }],
69    } as unknown as Parameters<typeof readInterfaceAddresses>[1];
70
71    expect(readInterfaceAddresses("eth0", interfaces)).toEqual({
72      ipv4: "192.168.1.27",
73      ipv6: "2a01:cb00::27",
74    });
75    expect(readInterfaceAddresses("lo", interfaces)).toEqual({ ipv4: null, ipv6: null });
76    expect(readInterfaceAddresses("wlan9", interfaces)).toEqual({ ipv4: null, ipv6: null });
77  });
78});
79