1 files changed, 60 insertions, 0 deletions

diff --git a/whosocket.hs b/whosocket.hs
new file mode 100644
index 00000000..f84e3178
--- /dev/null
+++ b/whosocket.hs
@@ -0,0 +1,60 @@
+{-# LANGUAGE OverloadedStrings #-}
+{-# LANGUAGE TupleSections     #-}
+module Main where
+
+import LocalPeerCred
+import ControlMaybe
+import UTmp
+import ByteStringOperators
+
+import System.Directory
+import Data.Char
+import System.Posix.Types
+import System.Posix.Files
+import qualified Data.ByteString.Lazy.Char8 as L 
+    ( unpack
+    , pack
+    , take
+    , putStrLn
+    )
+import Data.List (groupBy)
+import Data.Maybe (listToMaybe,mapMaybe,catMaybes)
+
+import Network.Socket
+import System.Environment
+import Control.Arrow (first)
+import System.Endian
+
+usage = do
+    putStrLn $ "whosocket numeric-address port"
+
+main = do
+    args <- getArgs
+    case (args??0,args??1) of
+        (Just addr_str,Just port_str) -> whosocket addr_str port_str
+        _                             -> usage
+
+whosocket :: HostName -> ServiceName -> IO ()
+whosocket addr_str port_str = do
+    info <- getAddrInfo (Just $ defaultHints { addrFlags = [ AI_NUMERICHOST ] }) 
+                        (Just addr_str) 
+                        (Just port_str)
+    let addr = head $ map addrAddress info
+    r <- getLocalPeerCred' addr
+    putStrLn $ "r{"++show addr++"} = " ++ show r
+
+    us <- UTmp.users
+    let filterTTYs (_,tty,pid) = 
+            if L.take 3 tty == "tty"
+              then Just (tty,pid)
+              else Nothing
+        tty_pids = mapMaybe filterTTYs us
+
+    tty <- maybe (return Nothing)
+                 (fmap fst . uncurry (identifyTTY tty_pids))
+                 r
+    putStrLn $ "uid = " ++ show (fmap fst r)
+    L.putStrLn $ "tty = " <++?> tty
+
+    return ()
+