summaryrefslogtreecommitdiff
path: root/testsuite/tests/ghc-regress/lib/IOExts/hGetBuf003.hs
blob: 6eefdf90e82a8ceb8f061e2c3b4163f47e7d002b (plain)
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
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