initalize

sketchybar/bridge/network_load/Makefile

@@ -0,0 +1,5 @@
+bin/network_load: network_load.c network_load.h ../sketchybar.h | bin
+	clang -std=c99 -O3 $< -o $@
+
+bin:
+	mkdir bin

sketchybar/bridge/network_load/network_load.c

@@ -0,0 +1,39 @@
+#include "network_load.h"
+#include "../sketchybar.h"
+#include <unistd.h>
+
+int main(int argc, char **argv) {
+  float update_freq;
+  if (argc < 4 || (sscanf(argv[3], "%f", &update_freq) != 1)) {
+    printf("Usage: %s \"<interface>\" \"<event-name>\" \"<event_freq>\"\n",
+           argv[0]);
+    exit(1);
+  }
+
+  alarm(0);
+  // Setup the event in sketchybar
+  char event_message[512];
+  snprintf(event_message, 512, "--add event '%s'", argv[2]);
+  sketchybar(event_message);
+
+  struct network network;
+  network_init(&network, argv[1]);
+  char trigger_message[512];
+  for (;;) {
+    // Acquire new info
+    network_update(&network);
+
+    // Prepare the event message
+    snprintf(trigger_message, 512,
+             "--trigger '%s' upload='%03d%s' download='%03d%s'", argv[2],
+             network.up, unit_str[network.up_unit], network.down,
+             unit_str[network.down_unit]);
+
+    // Trigger the event
+    sketchybar(trigger_message);
+
+    // Wait
+    usleep(update_freq * 1000000);
+  }
+  return 0;
+}

sketchybar/bridge/network_load/network_load.h

@@ -0,0 +1,93 @@
+#include <math.h>
+#include <net/if.h>
+#include <net/if_mib.h>
+#include <stdio.h>
+#include <string.h>
+#include <sys/select.h>
+#include <sys/sysctl.h>
+
+static char unit_str[3][6] = {
+    {" Bps"},
+    {"KBps"},
+    {"MBps"},
+};
+
+enum unit { UNIT_BPS, UNIT_KBPS, UNIT_MBPS };
+struct network {
+  uint32_t row;
+  struct ifmibdata data;
+  struct timeval tv_nm1, tv_n, tv_delta;
+
+  int up;
+  int down;
+  enum unit up_unit, down_unit;
+};
+
+static inline void ifdata(uint32_t net_row, struct ifmibdata *data) {
+  static size_t size = sizeof(struct ifmibdata);
+  static int32_t data_option[] = {CTL_NET,      PF_LINK, NETLINK_GENERIC,
+                                  IFMIB_IFDATA, 0,       IFDATA_GENERAL};
+  data_option[4] = net_row;
+  sysctl(data_option, 6, data, &size, NULL, 0);
+}
+
+static inline void network_init(struct network *net, char *ifname) {
+  memset(net, 0, sizeof(struct network));
+
+  static int count_option[] = {CTL_NET, PF_LINK, NETLINK_GENERIC, IFMIB_SYSTEM,
+                               IFMIB_IFCOUNT};
+  uint32_t interface_count = 0;
+  size_t size = sizeof(uint32_t);
+  sysctl(count_option, 5, &interface_count, &size, NULL, 0);
+
+  for (int i = 0; i < interface_count; i++) {
+    ifdata(i, &net->data);
+    if (strcmp(net->data.ifmd_name, ifname) == 0) {
+      net->row = i;
+      break;
+    }
+  }
+}
+
+static inline void network_update(struct network *net) {
+  gettimeofday(&net->tv_n, NULL);
+  timersub(&net->tv_n, &net->tv_nm1, &net->tv_delta);
+  net->tv_nm1 = net->tv_n;
+
+  uint64_t ibytes_nm1 = net->data.ifmd_data.ifi_ibytes;
+  uint64_t obytes_nm1 = net->data.ifmd_data.ifi_obytes;
+  ifdata(net->row, &net->data);
+
+  double time_scale = (net->tv_delta.tv_sec + 1e-6 * net->tv_delta.tv_usec);
+  if (time_scale < 1e-6 || time_scale > 1e2)
+    return;
+  double delta_ibytes =
+      (double)(net->data.ifmd_data.ifi_ibytes - ibytes_nm1) / time_scale;
+  double delta_obytes =
+      (double)(net->data.ifmd_data.ifi_obytes - obytes_nm1) / time_scale;
+
+  double exponent_ibytes = log10(delta_ibytes);
+  double exponent_obytes = log10(delta_obytes);
+
+  if (exponent_ibytes < 3) {
+    net->down_unit = UNIT_BPS;
+    net->down = delta_ibytes;
+  } else if (exponent_ibytes < 6) {
+    net->down_unit = UNIT_KBPS;
+    net->down = delta_ibytes / 1000.0;
+  } else if (exponent_ibytes < 9) {
+    net->down_unit = UNIT_MBPS;
+    net->down = delta_ibytes / 1000000.0;
+  }
+
+  if (exponent_obytes < 3) {
+    net->up_unit = UNIT_BPS;
+    net->up = delta_obytes;
+  } else if (exponent_obytes < 6) {
+    net->up_unit = UNIT_KBPS;
+    net->up = delta_obytes / 1000.0;
+  } else if (exponent_obytes < 9) {
+    net->up_unit = UNIT_MBPS;
+    net->up = delta_obytes / 1000000.0;
+  }
+}