diff options
Diffstat (limited to 'testsuite/tests/lib/IOExts/hGetBuf003.hs')
-rw-r--r-- | testsuite/tests/lib/IOExts/hGetBuf003.hs | 26 |
1 files changed, 26 insertions, 0 deletions
diff --git a/testsuite/tests/lib/IOExts/hGetBuf003.hs b/testsuite/tests/lib/IOExts/hGetBuf003.hs new file mode 100644 index 0000000000..6eefdf90e8 --- /dev/null +++ b/testsuite/tests/lib/IOExts/hGetBuf003.hs @@ -0,0 +1,26 @@ +import System.IO +import Foreign +import Foreign.C +import Control.Monad + +main = do test True; test False + +test blocking = do + h <- openBinaryFile "hGetBuf003.hs" ReadMode + + let sz = 42 + loop = do + -- mix ordinary char buffering with hGetBuf + eof <- hIsEOF h + when (not eof) $ hGetChar h >>= putChar + b <- allocaBytes sz $ \ptr -> do + r <- (if blocking then hGetBuf else hGetBufNonBlocking) h ptr sz + if (r == 0) + then return True + else do s <- peekCStringLen (ptr,r) + putStr s + return False + if b then return () else loop -- tail call + + loop + |