optimized Data.Binary.Get.getWord8

1 files changed, 9 insertions, 1 deletions

diff --git a/src/Data/Binary/Get.hs b/src/Data/Binary/Get.hs
index d92567e45..bdc07d985 100644
--- a/src/Data/Binary/Get.hs
+++ b/src/Data/Binary/Get.hs
@@ -416,7 +416,15 @@ getPtr n = do
 
 -- | Read a Word8 from the monad state
 getWord8 :: Get Word8
-getWord8 = getPtr (sizeOf (undefined :: Word8))
+getWord8 = do
+    S s ss bytes <- get
+    case B.uncons s of
+      Just (w,rest) -> do put $! S rest ss (bytes + 1)
+                          return $! w
+      Nothing       -> case L.uncons ss of
+                         Just (w,rest) -> do put $! mkState rest (bytes + 1)
+                                             return $! w
+                         Nothing       -> fail "too few bytes"
 {-# INLINE getWord8 #-}
 
 -- | Read a Word16 in big endian format