widget/net.lua

   1 --[[
   2
   3      Licensed under GNU General Public License v2
   4       * (c) 2013,      Luca CPZ
   5       * (c) 2010-2012, Peter Hofmann
   6
   7 --]]
   8
   9 local helpers = require("lain.helpers")
  10 local naughty = require("naughty")
  11 local wibox   = require("wibox")
  12 local string  = { format = string.format, match = string.match }
  13
  14 -- Network infos
  15 -- lain.widget.net
  16
  17 local function factory(args)
  18     local net        = { widget = wibox.widget.textbox(), devices = {} }
  19     local args       = args or {}
  20     local timeout    = args.timeout or 2
  21     local units      = args.units or 1024 -- KB
  22     local notify     = args.notify or "on"
  23     local wifi_state = args.wifi_state or "off"
  24     local eth_state  = args.eth_state or "off"
  25     local screen     = args.screen or 1
  26     local settings   = args.settings or function() end
  27
  28     -- Compatibility with old API where iface was a string corresponding to 1 interface
  29     net.iface = (args.iface and (type(args.iface) == "string" and {args.iface}) or
  30                 (type(args.iface) == "table" and args.iface)) or {}
  31
  32     function net.get_device()
  33         helpers.line_callback("ip link", function(line)
  34             net.iface[#net.iface + 1] = not string.match(line, "LOOPBACK") and string.match(line, "(%w+): <") or nil
  35         end)
  36     end
  37
  38     if #net.iface == 0 then net.get_device() end
  39
  40     function net.update()
  41         -- These are the totals over all specified interfaces
  42         net_now = {
  43             devices  = {},
  44             -- Bytes since last iteration
  45             sent     = 0,
  46             received = 0
  47         }
  48
  49         for _, dev in ipairs(net.iface) do
  50             local dev_now    = {}
  51             local dev_before = net.devices[dev] or { last_t = 0, last_r = 0 }
  52             local now_t      = tonumber(helpers.first_line(string.format("/sys/class/net/%s/statistics/tx_bytes", dev)) or 0)
  53             local now_r      = tonumber(helpers.first_line(string.format("/sys/class/net/%s/statistics/rx_bytes", dev)) or 0)
  54
  55             dev_now.carrier  = helpers.first_line(string.format("/sys/class/net/%s/carrier", dev)) or "0"
  56             dev_now.state    = helpers.first_line(string.format("/sys/class/net/%s/operstate", dev)) or "down"
  57
  58             dev_now.sent     = (now_t - dev_before.last_t) / timeout / units
  59             dev_now.received = (now_r - dev_before.last_r) / timeout / units
  60
  61             net_now.sent     = net_now.sent + dev_now.sent
  62             net_now.received = net_now.received + dev_now.received
  63
  64             dev_now.sent     = string.format("%.1f", dev_now.sent)
  65             dev_now.received = string.format("%.1f", dev_now.received)
  66
  67             dev_now.last_t   = now_t
  68             dev_now.last_r   = now_r
  69
  70             if wifi_state == "on" and helpers.first_line(string.format("/sys/class/net/%s/uevent", dev)) == "DEVTYPE=wlan" and string.match(dev_now.carrier, "1") then
  71                 dev_now.wifi   = true
  72                 dev_now.signal = tonumber(string.match(helpers.lines_from("/proc/net/wireless")[3], "(%-%d+%.)")) or nil
  73             end
  74
  75             if eth_state == "on" and helpers.first_line(string.format("/sys/class/net/%s/uevent", dev)) ~= "DEVTYPE=wlan" and string.match(dev_now.carrier, "1") then
  76                 dev_now.ethernet = true
  77             end
  78
  79             net.devices[dev] = dev_now
  80
  81             -- Notify only once when connection is lost
  82             if string.match(dev_now.carrier, "0") and notify == "on" and helpers.get_map(dev) then
  83                 naughty.notify {
  84                     title    = dev,
  85                     text     = "No carrier",
  86                     icon     = helpers.icons_dir .. "no_net.png",
  87                     screen   = screen
  88                 }
  89                 helpers.set_map(dev, false)
  90             elseif string.match(dev_now.carrier, "1") then
  91                 helpers.set_map(dev, true)
  92             end
  93
  94             net_now.carrier = dev_now.carrier
  95             net_now.state = dev_now.state
  96             net_now.devices[dev] = dev_now
  97             -- net_now.sent and net_now.received will be
  98             -- the totals across all specified devices
  99         end
 100
 101         net_now.sent = string.format("%.1f", net_now.sent)
 102         net_now.received = string.format("%.1f", net_now.received)
 103
 104         widget = net.widget
 105         settings()
 106     end
 107
 108     helpers.newtimer("network", timeout, net.update)
 109
 110     return net
 111 end
 112
 113 return factory