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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
|
-----------------------------------------------------------------------------
-- |
-- Module : Plugins.Monitors.Net
-- Copyright : (c) 2011, 2012 Jose Antonio Ortega Ruiz
-- (c) 2007-2010 Andrea Rossato
-- License : BSD-style (see LICENSE)
--
-- Maintainer : Jose A. Ortega Ruiz <jao@gnu.org>
-- Stability : unstable
-- Portability : unportable
--
-- A net device monitor for Xmobar
--
-----------------------------------------------------------------------------
module Plugins.Monitors.Net (startNet) where
import Plugins.Monitors.Common
import Data.IORef (IORef, newIORef, readIORef, writeIORef)
import Data.Time.Clock (UTCTime, getCurrentTime, diffUTCTime)
import qualified Data.ByteString.Lazy.Char8 as B
data NetDev = NA
| NI String
| ND String Float Float deriving (Eq,Show,Read)
type NetDevRef = IORef (NetDev, UTCTime)
netConfig :: IO MConfig
netConfig = mkMConfig
"<dev>: <rx>KB|<tx>KB" -- template
["dev", "rx", "tx", "rxbar", "txbar"] -- available replacements
isUp :: String -> IO Bool
isUp d = do
operstate <- B.readFile $ "/sys/class/net/" ++ d ++ "/operstate"
return $ "up" == (B.unpack . head . B.lines) operstate
readNetDev :: [String] -> IO NetDev
readNetDev (d:x:y:_) = do
up <- isUp d
return (if up then ND d (r x) (r y) else NI d)
where r s | s == "" = 0
| otherwise = read s / 1024
readNetDev _ = return NA
netParser :: B.ByteString -> IO [NetDev]
netParser = mapM (readNetDev . splitDevLine) . readDevLines
where readDevLines = drop 2 . B.lines
splitDevLine = selectCols . wordsBy (`elem` " :") . B.unpack
selectCols cols = map (cols!!) [0,1,9]
wordsBy f s = case dropWhile f s of
[] -> []
s' -> w : wordsBy f s'' where (w, s'') = break f s'
findNetDev :: String -> IO NetDev
findNetDev dev = do
nds <- B.readFile "/proc/net/dev" >>= netParser
case filter isDev nds of
x:_ -> return x
_ -> return NA
where isDev (ND d _ _) = d == dev
isDev (NI d) = d == dev
isDev NA = False
formatNet :: Float -> Monitor (String, String)
formatNet d = do
s <- getConfigValue useSuffix
dd <- getConfigValue decDigits
let str = if s then (++"Kb/s") . showDigits dd else showDigits dd
b <- showLogBar 0.9 d
x <- showWithColors str d
return (x, b)
printNet :: NetDev -> Monitor String
printNet nd =
case nd of
ND d r t -> do
(rx, rb) <- formatNet r
(tx, tb) <- formatNet t
parseTemplate [d,rx,tx,rb,tb]
NI _ -> return ""
NA -> return "N/A"
parseNet :: NetDevRef -> String -> IO NetDev
parseNet nref nd = do
(n0, t0) <- readIORef nref
n1 <- findNetDev nd
t1 <- getCurrentTime
writeIORef nref (n1, t1)
let scx = realToFrac (diffUTCTime t1 t0)
scx' = if scx > 0 then scx else 1
rate da db = takeDigits 2 $ (db - da) / scx'
diffRate (ND d ra ta) (ND _ rb tb) = ND d (rate ra rb) (rate ta tb)
diffRate (NI d) _ = NI d
diffRate _ (NI d) = NI d
diffRate _ _ = NA
return $ diffRate n0 n1
runNet :: NetDevRef -> String -> [String] -> Monitor String
runNet nref i _ = io (parseNet nref i) >>= printNet
startNet :: String -> [String] -> Int -> (String -> IO ()) -> IO ()
startNet i a r cb = do
t0 <- getCurrentTime
nref <- newIORef (NA, t0)
_ <- parseNet nref i
runM a netConfig (runNet nref i) r cb
|
