new testcase: cokiki ssh-client

1 files changed, 19 insertions, 0 deletions

diff --git a/lib/ProcessUtils.hs b/lib/ProcessUtils.hs
index 06f6893..a6902be 100644
--- a/lib/ProcessUtils.hs
+++ b/lib/ProcessUtils.hs
@@ -3,6 +3,7 @@ module ProcessUtils
     ( ExitCode(ExitFailure,ExitSuccess)
     , systemEnv
     , readPipe
+    , readProcessWithErrorH
     ) where
 
 import GHC.IO.Exception ( ioException, IOErrorType(..) )
@@ -14,6 +15,8 @@ import Data.Maybe ( isNothing )
 import System.IO.Error ( mkIOError, ioeSetErrorString ) 
 import System.Exit ( ExitCode(..) )
 import System.IO
+import Control.Applicative
+import Control.Exception (bracket)
 
 
 -- | systemEnv
@@ -83,3 +86,19 @@ readPipe ((cmd,args):xs) stdin0 = do
             (Nothing,Just sout,Just serr, ph) <- createProcess p
             readPipe0 xs sout
         readPipe0 [] h = hGetContents h
+
+
+readProcessWithErrorH :: FilePath -> [String] -> String -> Handle -> IO String
+readProcessWithErrorH cmd args stdin erH = do
+    let p = (shell cmd) { std_in = CreatePipe
+                        , std_out = CreatePipe
+                        , std_err = UseHandle erH
+                        , cmdspec = RawCommand cmd args
+                        }
+    bracket (createProcess p) 
+            (\(Just sinh,Just sout,_, ph) ->
+                mapM_ hClose [sinh,sout]         )
+            (\(Just sinh,Just sout,_, ph) -> do
+                hPutStr sinh stdin
+                hClose sinh
+                hGetContents sout                )