prometheus-node-exporter-lua: add hostapd exporter

This exporter exposes information of the connected stations acquired
from hostapd. These contain additional information compared to the
existing station exporter, however they require a full build of hostapd
/ wpad.

Signed-off-by: David Bauer <mail@david-bauer.net>

1 files changed, 110 insertions, 0 deletions

diff --git a/utils/prometheus-node-exporter-lua/files/usr/lib/lua/prometheus-collectors/hostapd_stations.lua b/utils/prometheus-node-exporter-lua/files/usr/lib/lua/prometheus-collectors/hostapd_stations.lua
new file mode 100644
index 000000000..47ae9a5c5
--- /dev/null
+++ b/utils/prometheus-node-exporter-lua/files/usr/lib/lua/prometheus-collectors/hostapd_stations.lua
@@ -0,0 +1,110 @@
+local ubus = require "ubus"
+local bit32 = require "bit32"
+
+local function get_wifi_interfaces()
+  local u = ubus.connect()
+  local status = u:call("network.wireless", "status", {})
+  local interfaces = {}
+
+  for dev, dev_table in pairs(status) do
+    for _, intf in ipairs(dev_table['interfaces']) do
+      table.insert(interfaces, intf['config']['ifname'])
+    end
+  end
+
+  return interfaces
+end
+
+local function scrape()
+  local metric_hostapd_station_vht = metric("hostapd_station_vht", "gauge")
+  local metric_hostapd_station_ht = metric("hostapd_station_ht", "gauge")
+  local metric_hostapd_station_wmm = metric("hostapd_station_wmm", "gauge")
+  local metric_hostapd_station_mfp = metric("hostapd_station_mfp", "gauge")
+
+  local metric_hostapd_station_rx_packets = metric("hostapd_station_rx_packets", "counter")
+  local metric_hostapd_station_rx_bytes = metric("hostapd_station_rx_bytes", "counter")
+  local metric_hostapd_station_tx_packets = metric("hostapd_station_tx_packets", "counter")
+  local metric_hostapd_station_tx_bytes = metric("hostapd_station_tx_bytes", "counter")
+  
+  local metric_hostapd_station_inactive_msec = metric("hostapd_station_inactive_msec", "counter")
+
+  local metric_hostapd_station_signal = metric("hostapd_station_signal", "gauge")
+  local metric_hostapd_station_connected_time = metric("hostapd_station_connected_time", "counter")
+
+  local metric_hostapd_station_wpa = metric("hostapd_station_wpa", "gauge")
+  local metric_hostapd_station_sae_group = metric("hostapd_station_sae_group", "gauge")
+
+  local metric_hostapd_station_vht_capb_su_beamformee = metric("hostapd_station_vht_capb_su_beamformee", "gauge")
+  local metric_hostapd_station_vht_capb_mu_beamformee = metric("hostapd_station_vht_capb_mu_beamformee", "gauge")
+
+  local function evaluate_metrics(ifname, station, vals)
+    local label_station = {
+      ifname = ifname,
+      station = station,
+    }
+
+    for k, v in pairs(vals) do
+      if k == "flags" then
+        if string.match(v, "[VHT]") then
+          metric_hostapd_station_vht(label_station, 1)
+        end
+        if string.match(v, "[HT]") then
+          metric_hostapd_station_ht(label_station, 1)
+        end
+        if string.match(v, "[WMM]") then
+          metric_hostapd_station_wmm(label_station, 1)
+        end
+        if string.match(v, "[MFP]") then
+          metric_hostapd_station_mfp(label_station, 1)
+        end
+      elseif k == "wpa" then
+        metric_hostapd_station_wpa(label_station, v)
+      elseif k == "rx_packets" then
+        metric_hostapd_station_rx_packets(label_station, v)
+      elseif k == "rx_bytes" then
+        metric_hostapd_station_rx_bytes(label_station, v)
+      elseif k == "tx_packets" then
+        metric_hostapd_station_tx_packets(label_station, v)
+      elseif k == "tx_bytes" then
+        metric_hostapd_station_tx_bytes(label_station, v)
+      elseif k == "inactive_msec" then
+        metric_hostapd_station_inactive_msec(label_station, v)
+      elseif k == "signal" then
+        metric_hostapd_station_signal(label_station, v)
+      elseif k == "connected_time" then
+        metric_hostapd_station_connected_time(label_station, v)
+      elseif k == "sae_group" then
+        metric_hostapd_station_sae_group(label_station, v)
+      elseif k == "vht_caps_info" then
+	      local caps = tonumber(string.gsub(v, "0x", ""), 16)
+	      metric_hostapd_station_vht_capb_su_beamformee(label_station, bit32.band(bit32.lshift(1, 12), caps) > 0 and 1 or 0)
+	      metric_hostapd_station_vht_capb_mu_beamformee(label_station, bit32.band(bit32.lshift(1, 20), caps) > 0 and 1 or 0)
+      end
+    end
+  end
+
+  for _, ifname in ipairs(get_wifi_interfaces()) do
+    local handle = io.popen("hostapd_cli -i " .. ifname .." all_sta")
+    local all_sta = handle:read("*a")
+    handle:close()
+
+    local current_station = nil
+    local current_station_values = {}
+
+    for line in all_sta:gmatch("[^\r\n]+") do
+      if string.match(line, "^%x[0123456789aAbBcCdDeE]:%x%x:%x%x:%x%x:%x%x:%x%x$") then
+        if current_station ~= nil then
+          evaluate_metrics(ifname, current_station, current_station_values)
+        end
+        current_station = line
+        current_station_values = {}
+      else
+        local name, value = string.match(line, "(.+)=(.+)")
+        current_station_values[name] = value
+      end
+    end
+    evaluate_metrics(ifname, current_station, current_station_values)
+  end
+end
+
+return { scrape = scrape }