Read packet file with larger chunk size.

1 files changed, 23 insertions, 0 deletions

diff --git a/lib/ByteStringUtil.hs b/lib/ByteStringUtil.hs
new file mode 100644
index 0000000..c6b509d
--- /dev/null
+++ b/lib/ByteStringUtil.hs
@@ -0,0 +1,23 @@
+module ByteStringUtil where
+
+import Data.ByteString.Lazy.Internal
+import Data.ByteString.Lazy
+import qualified Data.ByteString as S
+import System.IO
+import System.IO.Unsafe
+
+oneMeg :: Int
+oneMeg = 1048576
+
+hGetContentsN :: Int -> Handle -> IO ByteString
+hGetContentsN kk h = lazyRead
+  where
+    k = kk - chunkOverhead
+    lazyRead = unsafeInterleaveIO loop
+
+    loop = do
+        c <- S.hGetSome h k -- only blocks if there is no data available
+        if S.null c
+          then hClose h >> return Empty
+          else do cs <- lazyRead
+                  return (Chunk c cs)