summaryrefslogtreecommitdiffstats
path: root/utils/prometheus-node-exporter-lua/files/usr/lib/lua/prometheus-collectors/nat_traffic.lua
blob: 66d8342a6c08ce64e4a367bf29175c2115218020 (plain)
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
local function scrape()
  -- documentation about nf_conntrack:
  -- https://www.frozentux.net/iptables-tutorial/chunkyhtml/x1309.html

  -- two dimensional table to sum bytes for the pair (src/dest)
  local nat = {}
  -- default constructor to initialize unknown pairs
  setmetatable(nat, {
    __index = function (t, addr)
      t[addr] = {}
      setmetatable(t[addr], {
        __index = function () return 0 end
      })
      return t[addr]
    end
  })

  for e in io.lines("/proc/net/nf_conntrack") do
    -- output(string.format("%s\n",e  ))
    local fields = space_split(e)
    local src, dest, bytes;
    bytes = 0;
    for _, field in ipairs(fields) do
      if src == nil and string.match(field, '^src') then
        src = string.match(field,"src=([^ ]+)");
      elseif dest == nil and string.match(field, '^dst') then
        dest = string.match(field,"dst=([^ ]+)");
      elseif string.match(field, '^bytes') then
        local b = string.match(field, "bytes=([^ ]+)");
        bytes = bytes + b;
        -- output(string.format("\t%d %s",ii,field  ));
      end

    end
    -- local src, dest, bytes = string.match(natstat[i], "src=([^ ]+) dst=([^ ]+) .- bytes=([^ ]+)");
    -- local src, dest, bytes = string.match(natstat[i], "src=([^ ]+) dst=([^ ]+) sport=[^ ]+ dport=[^ ]+ packets=[^ ]+ bytes=([^ ]+)")

    -- output(string.format("src=|%s| dest=|%s| bytes=|%s|", src, dest, bytes  ))
    nat[src][dest] = nat[src][dest] + bytes
  end

  nat_metric =  metric("node_nat_traffic", "gauge" )
  for src, values in next, nat do
    for dest, bytes in next, values do
      local labels = { src = src, dest = dest }
      nat_metric(labels, bytes )
    end
  end
end

return { scrape = scrape }