summaryrefslogtreecommitdiffhomepage
path: root/src/Plugins/Monitors/Net.hs
diff options
context:
space:
mode:
Diffstat (limited to 'src/Plugins/Monitors/Net.hs')
-rw-r--r--src/Plugins/Monitors/Net.hs34
1 files changed, 23 insertions, 11 deletions
diff --git a/src/Plugins/Monitors/Net.hs b/src/Plugins/Monitors/Net.hs
index 10ca9cc..8234670 100644
--- a/src/Plugins/Monitors/Net.hs
+++ b/src/Plugins/Monitors/Net.hs
@@ -22,7 +22,9 @@ import Plugins.Monitors.Common
import Data.IORef (IORef, newIORef, readIORef, writeIORef)
import Data.Time.Clock (UTCTime, getCurrentTime, diffUTCTime)
-import Control.Monad (forM)
+import Control.Monad (forM, filterM)
+import System.Directory (getDirectoryContents, doesFileExist)
+import System.FilePath ((</>))
import qualified Data.ByteString.Lazy.Char8 as B
@@ -32,10 +34,9 @@ data NetDev = NA
type NetDevRef = IORef (NetDev, UTCTime)
-{- The more information available, the better.
- - Note that names don't matter. Therefore, if only the names differ,
- - a compare evaluates to EQ while (==) evaluates to False.
- -}
+-- The more information available, the better.
+-- Note that names don't matter. Therefore, if only the names differ,
+-- a compare evaluates to EQ while (==) evaluates to False.
instance Ord NetDev where
compare NA NA = EQ
compare NA _ = LT
@@ -54,9 +55,18 @@ netConfig = mkMConfig
"<dev>: <rx>KB|<tx>KB" -- template
["dev", "rx", "tx", "rxbar", "txbar"] -- available replacements
+operstateDir :: String -> FilePath
+operstateDir d = "/sys/class/net" </> d </> "operstate"
+
+existingDevs :: IO [String]
+existingDevs = getDirectoryContents "/sys/class/net" >>= filterM isDev
+ where isDev d | d `elem` excludes = return False
+ | otherwise = doesFileExist (operstateDir d)
+ excludes = [".", "..", "lo"]
+
isUp :: String -> IO Bool
isUp d = do
- operstate <- B.readFile $ "/sys/class/net/" ++ d ++ "/operstate"
+ operstate <- B.readFile (operstateDir d)
return $ "up" == (B.unpack . head . B.lines) operstate
readNetDev :: [String] -> IO NetDev
@@ -139,13 +149,15 @@ startNet i a r cb = do
_ <- parseNet nref i
runM a netConfig (runNet nref i) r cb
-startDynNet :: [String] -> [String] -> Int -> (String -> IO ()) -> IO ()
-startDynNet is a r cb = do
- refs <- forM is $ \i -> do
+startDynNet :: [String] -> Int -> (String -> IO ()) -> IO ()
+startDynNet a r cb = do
+ devs <- existingDevs
+ refs <- forM devs $ \d -> do
t <- getCurrentTime
nref <- newIORef (NA, t)
- _ <- parseNet nref i
- return (nref, i)
+ _ <- parseNet nref d
+ return (nref, d)
runM a netConfig (runNets refs) r cb
-- TODO: Prelude.head: empty list
+-- TODO: remember last active interface.