@@ -4,7 +4,6 @@
44-- |
55-- Module : Data.Binary.Bits.Get
66-- Copyright : (c) Lennart Kolmodin 2010-2011
7−-- (c) Sylvain Henry 2015
87-- License : BSD3-style (see LICENSE)
98--
109-- Maintainer : kolmodin@gmail.com
@@ -60,19 +59,11 @@ module Data.Binary.Bits.Get
6059
6160 -- ** Get bytes
6261 , getBool
63− , getBool#
6462 , getWord8
65− , getWord16
66− , getWord32
67− , getWord64
6863 , getWord16be
6964 , getWord32be
7065 , getWord64be
7166
72− -- ** Skip bits
73− , skipBits
74− , alignByte
75−
7667 -- * Blocks
7768
7869 -- $blocks
@@ -82,41 +73,32 @@ module Data.Binary.Bits.Get
8273 -- ** Read in Blocks
8374 , bool
8475 , word8
85− , word16
86− , word32
87− , word64
8876 , word16be
8977 , word32be
9078 , word64be
9179 , byteString
9280 , Data.Binary.Bits.Get.getByteString
93− , Data.Binary.Bits.Get.getLazyByteString
94− , Data.Binary.Bits.Get.isEmpty
9581
9682 ) where
9783
98−import Data.Binary.Get as B ( Get, getLazyByteString, isEmpty )
84+import Data.Binary.Get as B ( runGet, Get, getByteString )
9985import Data.Binary.Get.Internal as B ( get, put, ensureN )
100−import Data.Binary.Bits.BitOrder
101−import Data.Binary.Bits.Internal
102−import Data.Binary.Bits.Alignment
10386
10487import Data.ByteString as B
105−import qualified Data.ByteString.Lazy as L
10688import Data.ByteString.Unsafe
107−import Foreign.Marshal.Alloc (mallocBytes)
108−import Foreign.Ptr
109−import Foreign.Storable (poke)
110−import System.IO.Unsafe (unsafePerformIO)
11189
11290import Data.Bits
11391import Data.Word
92+
11493import Control.Applicative
115−import Control.Monad (when,foldM_)
11694
11795import Prelude as P
96+
97+#if defined(__GLASGOW_HASKELL__) && !defined(__HADDOCK__)
98+import GHC.Base
11899import GHC.Word
119−import GHC.Base (uncheckedShiftLWord8#, wordToWord8#, Int (..), andWord8#, Word8#, (-#), Int#, xorI#, eqWord8#, TYPE)
100+#endif
101+
120102
121103-- $bitget
122104-- Parse bits using a monad.
@@ -164,121 +146,188 @@ import GHC.Base (uncheckedShiftLWord8#, wordToWord8#, Int (..), andWord8#, Word8
164146--ipv6Header = 'runBitGet' ('block' ipv6headerblock)
165147-- @
166148
167−data S = S {-# UNPACK #-} !ByteString -- Input
168− {-# UNPACK #-} !Int -- Bit offset (0-7)
169− !BitOrder -- Bit order
149+data S = S !ByteString -- Input
150+ !Int -- Bit offset (0-7)
170151 deriving (Show)
171152
172−-- | Increment the current bit offset
153+-- | A block that will be read with only one boundry check. Needs to know the
154+-- number of bits in advance.
155+data Block a = Block !Int (S -> a)
156+
157+instance Functor Block where
158+ fmap f (Block i p) = Block i (\s -> f (p s))
159+
160+instance Applicative Block where
161+ pure a = Block 0 (\_ -> a)
162+ (Block i p) <*> (Block j q) = Block (i+j) (\s -> p s $ q (incS i s))
163+ (Block i _) *> (Block j q) = Block (i+j) (q . incS i)
164+ (Block i p) <* (Block j _) = Block (i+j) p
165+
166+-- | Get a block. Will be read with one single boundry check, and
167+-- therefore requires a statically known number of bits.
168+-- Build blocks using 'bool', 'word8', 'word16be', 'word32be', 'word64be',
169+-- 'byteString' and 'Applicative'.
170+block :: Block a -> BitGet a
171+block (Block i p) = do
172+ ensureBits i
173+ modifyState (\s -> (incS i s, p s))
174+
175+
173176incS :: Int -> S -> S
174−incS o (S bs n bo) = S (unsafeDrop d bs) n' bo
175− where
176− !o' = (n+o)
177− !d = byte_offset o'
178− !n' = bit_offset o'
177+incS o (S bs n) =
178+ let !o' = (n+o)
179+ !d = o' `shiftR` 3
180+ !n' = o' .&. make_mask 3
181+ in S (unsafeDrop d bs) n'
182+
183+-- | make_mask 3 = 00000111
184+make_mask :: Bits a => Int -> a
185+make_mask n = (1 `shiftL` fromIntegral n) - 1
186+{-# SPECIALIZE make_mask :: Int -> Int #-}
187+{-# SPECIALIZE make_mask :: Int -> Word #-}
188+{-# SPECIALIZE make_mask :: Int -> Word8 #-}
189+{-# SPECIALIZE make_mask :: Int -> Word16 #-}
190+{-# SPECIALIZE make_mask :: Int -> Word32 #-}
191+{-# SPECIALIZE make_mask :: Int -> Word64 #-}
192+
193+bit_offset :: Int -> Int
194+bit_offset n = make_mask 3 .&. n
195+
196+byte_offset :: Int -> Int
197+byte_offset n = n `shiftR` 3
179198
180−-- | Read a single bit
181199readBool :: S -> Bool
182−readBool (S bs o bo) = case bo of
183− BB -> testBit (unsafeHead bs) (7-o)
184− BL -> testBit (unsafeHead bs) (7-o)
185− LL -> testBit (unsafeHead bs) o
186− LB -> testBit (unsafeHead bs) o
187−
188−-- | Read a single bit
189−readBool# :: S -> Int
190−readBool# (S bs (I# o) bo) = case bo of
191− BB -> testBit# (7# -# o)
192− BL -> testBit# (7# -# o)
193− LL -> testBit# o
194− LB -> testBit# o
195− where
196− !(W8# w) = unsafeHead bs
197− testBit# n = I# (((w `andWord8#` (wordToWord8# 1## `uncheckedShiftLWord8#` n)) `eqWord8#` wordToWord8# 0##) `xorI#` 1#)
198−
199−
200−-- | Extract a range of bits from (ws :: ByteString)
201−--
202−-- Constraint: 8 * (length ws -1 ) < o+n <= 8 * length ws
203−extract :: (Num a, FastBits a) => BitOrder -> ByteString -> Int -> Int -> a
204−extract bo bs o n
205− | n == 0 = 0
206− | B.length bs == 0 = error "Empty ByteString"
207− | otherwise = rev . mask n . foldlWithIndex' f 0 $ bs
208− where
209− -- B.foldl' with index
210− foldlWithIndex' op b = fst . B.foldl' g (b,0)
211− where g (b',i) w = (op b' w i, (i+1))
212−
213− -- 'or' correctly shifted words
214− f b w i = b .|. (fromIntegral w `fastShift` off i)
215−
216− -- co-offset
217− r = B.length bs * 8 - (o + n)
218−
219− -- shift offset depending on the byte position (0..B.length-1)
220− off i = case bo of
221− LL -> 8*i - o
222− LB -> 8*i - o
223− BB -> (B.length bs -1 - i) * 8 - r
224− BL -> (B.length bs -1 - i) * 8 - r
225−
226− -- reverse bits if necessary
227− rev = case bo of
228− LB -> reverseBits n
229− BL -> reverseBits n
230− BB -> id
231− LL -> id
232−
233−
234−-- | Generic readWord
235−readWord :: (Num a, FastBits a) => Int -> S -> a
236−readWord n (S bs o bo)
237− | n == 0 = 0
238− | otherwise = extract bo (unsafeTake nbytes bs) o n
239− where nbytes = byte_offset (o+n+7)
240−
241−-- | Check that the number of bits to read is not greater than the first parameter
242−{-# INLINE readWordChecked #-}
243−readWordChecked :: (Num a, FastBits a) => Int -> Int -> S -> a
244−readWordChecked m n s
245− | n > m = error $ "Tried to read more than " ++ show m ++ " bits (" ++ show n ++")"
246− | otherwise = readWord n s
247−
248−-- | Read the given number of bytes and return them in Big-Endian order
249−--
250−-- Examples:
251−-- BB: xxxABCDE FGHIJKLM NOPxxxxx -> ABCDEFGH IJKLMNOP
252−-- LL: LMNOPxxx DEFGHIJK xxxxxABC -> ABCDEFGH IJKLMNOP
253−-- BL: xxxPONML KJIHGFED CBAxxxxx -> ABCDEFGH IJKLMNOP
254−-- LB: EDCBAxxx MLKJIHGF xxxxxPON -> ABCDEFGH IJKLMNOP
200+readBool (S bs n) = testBit (unsafeHead bs) (7-n)
201+
202+readWord8 :: Int -> S -> Word8
203+readWord8 n (S bs o)
204+ -- no bits at all, return 0
205+ | n == 0 = 0
206+
207+ -- all bits are in the same byte
208+ -- we just need to shift and mask them right
209+ | n <= 8 - o = let w = unsafeHead bs
210+ m = make_mask n
211+ w' = (w `shiftr_w8` (8 - o - n)) .&. m
212+ in w'
213+
214+ -- the bits are in two different bytes
215+ -- make a word16 using both bytes, and then shift and mask
216+ | n <= 8 = let w = (fromIntegral (unsafeHead bs) `shiftl_w16` 8) .|.
217+ (fromIntegral (unsafeIndex bs 1))
218+ m = make_mask n
219+ w' = (w `shiftr_w16` (16 - o - n)) .&. m
220+ in fromIntegral w'
221+
222+
223+readWord16be :: Int -> S -> Word16
224+readWord16be n s@(S bs o)
225+
226+ -- 8 or fewer bits, use readWord8
227+ | n <= 8 = fromIntegral (readWord8 n s)
228+
229+ -- handle 9 or more bits, stored in two bytes
230+
231+ -- no offset, plain and simple 16 bytes
232+ | o == 0 && n == 16 = let msb = fromIntegral (unsafeHead bs)
233+ lsb = fromIntegral (unsafeIndex bs 1)
234+ w = (msb `shiftl_w16` 8) .|. lsb
235+ in w
236+
237+ -- no offset, but not full 16 bytes
238+ | o == 0 = let msb = fromIntegral (unsafeHead bs)
239+ lsb = fromIntegral (unsafeIndex bs 1)
240+ w = (msb `shiftl_w16` (n-8)) .|. (lsb `shiftr_w16` (16-n))
241+ in w
242+
243+ -- with offset, and n=9-16
244+ | n <= 16 = readWithOffset s shiftl_w16 shiftr_w16 n
245+
246+ | otherwise = error "readWord16be: tried to read more than 16 bits"
247+
248+readWord32be :: Int -> S -> Word32
249+readWord32be n s@(S _ o)
250+ -- 8 or fewer bits, use readWord8
251+ | n <= 8 = fromIntegral (readWord8 n s)
252+
253+ -- 16 or fewer bits, use readWord16be
254+ | n <= 16 = fromIntegral (readWord16be n s)
255+
256+ | o == 0 = readWithoutOffset s shiftl_w32 shiftr_w32 n
257+
258+ | n <= 32 = readWithOffset s shiftl_w32 shiftr_w32 n
259+
260+ | otherwise = error "readWord32be: tried to read more than 32 bits"
261+
262+
263+readWord64be :: Int -> S -> Word64
264+readWord64be n s@(S _ o)
265+ -- 8 or fewer bits, use readWord8
266+ | n <= 8 = fromIntegral (readWord8 n s)
267+
268+ -- 16 or fewer bits, use readWord16be
269+ | n <= 16 = fromIntegral (readWord16be n s)
270+
271+ | o == 0 = readWithoutOffset s shiftl_w64 shiftr_w64 n
272+
273+ | n <= 64 = readWithOffset s shiftl_w64 shiftr_w64 n
274+
275+ | otherwise = error "readWord64be: tried to read more than 64 bits"
276+
277+
255278readByteString :: Int -> S -> ByteString
256−readByteString n (S bs o bo) =
257− let
258− bs' = unsafeTake (n+1) bs
259− bs'' = unsafeTake n bs
260− rev = B.map (reverseBits 8)
261− in case (o,bo) of
262− (0,BB) -> bs''
263− (0,LL) -> B.reverse bs''
264− (0,LB) -> rev bs''
265− (0,BL) -> rev . B.reverse $ bs''
266− (_,LL) -> readByteString n (S (B.reverse bs') (8-o) BB)
267− (_,BL) -> rev . B.reverse $ readByteString n (S bs' o BB)
268− (_,LB) -> rev . B.reverse $ readByteString n (S bs' o LL)
269− (_,BB) -> unsafePerformIO $ do
270− let len = n+1
271− ptr <- mallocBytes len
272− let f r i = do
273− let
274− w = unsafeIndex bs (len-i)
275− w' = (w `fastShiftL` o) .|. r
276− r' = w `fastShiftR` (8-o)
277− poke (ptr `plusPtr` (len-i)) w'
278− return r'
279− foldM_ f 0 [1..len]
280− unsafeInit <$> unsafePackMallocCStringLen (ptr,len)
279+readByteString n s@(S bs o)
280+ -- no offset, easy.
281+ | o == 0 = unsafeTake n bs
282+ -- offset. ugg. this is really naive and slow. but also pretty easy :)
283+ | otherwise = B.pack (P.map (readWord8 8) (P.take n (iterate (incS 8) s)))
284+
285+readWithoutOffset :: (Bits a, Num a)
286+ => S -> (a -> Int -> a) -> (a -> Int -> a) -> Int -> a
287+readWithoutOffset (S bs o) shifterL shifterR n
288+ | o /= 0 = error "readWithoutOffset: there is an offset"
289+
290+ | bit_offset n == 0 && byte_offset n <= 4 =
291+ let segs = byte_offset n
292+ bn 0 = fromIntegral (unsafeHead bs)
293+ bn n = (bn (n-1) `shifterL` 8) .|. fromIntegral (unsafeIndex bs n)
294+
295+ in bn (segs-1)
281296
297+ | n <= 64 = let segs = byte_offset n
298+ o' = bit_offset (n - 8 + o)
299+
300+ bn 0 = fromIntegral (unsafeHead bs)
301+ bn n = (bn (n-1) `shifterL` 8) .|. fromIntegral (unsafeIndex bs n)
302+
303+ msegs = bn (segs-1) `shifterL` o'
304+
305+ last = (fromIntegral (unsafeIndex bs segs)) `shifterR` (8 - o')
306+
307+ w = msegs .|. last
308+ in w
309+
310+readWithOffset :: (Bits a, Num a)
311+ => S -> (a -> Int -> a) -> (a -> Int -> a) -> Int -> a
312+readWithOffset (S bs o) shifterL shifterR n
313+ | n <= 64 = let bits_in_msb = 8 - o
314+ (n',top) = (n - bits_in_msb
315+ , (fromIntegral (unsafeHead bs) .&. make_mask bits_in_msb) `shifterL` n')
316+
317+ segs = byte_offset n'
318+
319+ bn 0 = 0
320+ bn n = (bn (n-1) `shifterL` 8) .|. fromIntegral (unsafeIndex bs n)
321+
322+ o' = bit_offset n'
323+
324+ mseg = bn segs `shifterL` o'
325+
326+ last | o' > 0 = (fromIntegral (unsafeIndex bs (segs + 1))) `shifterR` (8 - o')
327+ | otherwise = 0
328+
329+ w = top .|. mseg .|. last
330+ in w
282331
283332------------------------------------------------------------------------
284333-- | 'BitGet' is a monad, applicative and a functor. See 'runBitGet'
@@ -287,6 +336,7 @@ newtype BitGet a = B { runState :: S -> Get (S,a) }
287336
288337instance Monad BitGet where
289338 return x = B $ \s -> return (s,x)
339+ fail str = B $ \(S inp n) -> putBackState inp n >> fail str
290340 (B f) >>= g = B $ \s -> do (s',a) <- f s
291341 runState (g a) s'
292342
@@ -297,51 +347,25 @@ instance Applicative BitGet where
297347 pure x = return x
298348 fm <*> m = fm >>= \f -> m >>= \v -> return (f v)
299349
300−instance BitOrderable BitGet where
301− setBitOrder bo = do
302− (S bs o _) <- getState
303− putState (S bs o bo)
304−
305− getBitOrder = do
306− (S _ _ bo) <- getState
307− return bo
308−
309−instance Alignable BitGet where
310− -- | Skip the given number of bits
311− skipBits n = do
312− ensureBits n
313− withState (incS n)
314−
315− -- | Skip bits if necessary to align to the next byte
316− alignByte = do
317− (S _ o _) <- getState
318− when (o /= 0) $
319− skipBits (8-o)
320−
321−
322−
323350-- | Run a 'BitGet' within the Binary packages 'Get' monad. If a byte has
324351-- been partially consumed it will be discarded once 'runBitGet' is finished.
325352runBitGet :: BitGet a -> Get a
326353runBitGet bg = do
327354 s <- mkInitState
328− (s',a) <- runState bg s
329− putBackState s'
355+ ((S str' n),a) <- runState bg s
356+ putBackState str' n
330357 return a
331358
332359mkInitState :: Get S
333360mkInitState = do
334− bs <- get
361+ str <- get
335362 put B.empty
336− return (S bs 0 BB)
363+ return (S str 0)
337364
338−putBackState :: S -> Get ()
339−putBackState (S bs o _) = do
365+putBackState :: B.ByteString -> Int -> Get ()
366+putBackState bs n = do
340367 remaining <- get
341− let bs' = case o of
342− 0 -> bs
343− _ -> unsafeDrop 1 bs
344− put (bs' `B.append` remaining)
368+ put (B.drop (if n==0 then 0 else 1) bs `B.append` remaining)
345369
346370getState :: BitGet S
347371getState = B $ \s -> return (s,s)
@@ -349,148 +373,113 @@ getState = B $ \s -> return (s,s)
349373putState :: S -> BitGet ()
350374putState s = B $ \_ -> return (s,())
351375
352−withState :: (S -> S) -> BitGet ()
353−withState f = do
354− s <- getState
355− putState $! f s
376+modifyState :: (S -> (S,a)) -> BitGet a
377+modifyState f = B $ \s -> return (f s)
356378
357379-- | Make sure there are at least @n@ bits.
358380ensureBits :: Int -> BitGet ()
359381ensureBits n = do
360− (S bs o bo) <- getState
382+ (S bs o) <- getState
361383 if n <= (B.length bs * 8 - o)
362384 then return ()
363385 else do let currentBits = B.length bs * 8 - o
364− let byteCount = byte_offset (n - currentBits + 7)
386+ let byteCount = (n - currentBits + 7) `div` 8
365387 B $ \_ -> do B.ensureN byteCount
366388 bs' <- B.get
367389 put B.empty
368− return (S (bs`append`bs') o bo, ())
369−
370−-- | Test whether all input has been consumed, i.e. there are no remaining
371−-- undecoded bytes.
372−isEmpty :: BitGet Bool
373−isEmpty = B $ \ (S bs o bo) -> if B.null bs
374− then B.isEmpty >>= \e -> return (S bs o bo, e)
375− else return (S bs o bo, False)
390+ return (S (bs`append`bs') o, ())
376391
377392-- | Get 1 bit as a 'Bool'.
378393getBool :: BitGet Bool
379394getBool = block bool
380395
381−-- | Get 1 bit as a 'Bool'.
382−getBool# :: BitGet Int
383−getBool# = block bool#
384−
385396-- | Get @n@ bits as a 'Word8'. @n@ must be within @[0..8]@.
386397getWord8 :: Int -> BitGet Word8
387398getWord8 n = block (word8 n)
388399
389400-- | Get @n@ bits as a 'Word16'. @n@ must be within @[0..16]@.
390−getWord16 :: Int -> BitGet Word16
391−getWord16 n = block (word16 n)
392−
393401getWord16be :: Int -> BitGet Word16
394−getWord16be = getWord16
395−{-# DEPRECATED getWord16be "Use 'getWord16' instead" #-}
402+getWord16be n = block (word16be n)
396403
397404-- | Get @n@ bits as a 'Word32'. @n@ must be within @[0..32]@.
398−getWord32 :: Int -> BitGet Word32
399−getWord32 n = block (word32 n)
400−
401405getWord32be :: Int -> BitGet Word32
402−getWord32be = getWord32
403−{-# DEPRECATED getWord32be "Use 'getWord32' instead" #-}
406+getWord32be n = block (word32be n)
404407
405408-- | Get @n@ bits as a 'Word64'. @n@ must be within @[0..64]@.
406−getWord64 :: Int -> BitGet Word64
407−getWord64 n = block (word64 n)
408−
409409getWord64be :: Int -> BitGet Word64
410−getWord64be = getWord64
411−{-# DEPRECATED getWord64be "Use 'getWord64' instead" #-}
410+getWord64be n = block (word64be n)
412411
413412-- | Get @n@ bytes as a 'ByteString'.
414413getByteString :: Int -> BitGet ByteString
415414getByteString n = block (byteString n)
416415
417−-- | Get @n@ bytes as a lazy ByteString.
418−getLazyByteString :: Int -> BitGet L.ByteString
419−getLazyByteString n = do
420− (S _ o bo) <- getState
421− case o of
422− 0 -> B $ \s -> do
423− putBackState s
424− lbs <- B.getLazyByteString (fromIntegral n)
425− return (S B.empty 0 bo, lbs)
426− _ -> L.fromChunks . (:[]) <$> Data.Binary.Bits.Get.getByteString n
427−
428−
429−
430−
431−
432−-- | A block that will be read with only one boundary check. Needs to know the
433−-- number of bits in advance.
434−data Block a = Block Int (S -> a)
435−
436−instance Functor Block where
437− fmap f (Block i p) = Block i (\s -> f (p s))
438−
439−instance Applicative Block where
440− pure a = Block 0 (\_ -> a)
441− (Block i p) <*> (Block j q) = Block (i+j) (\s -> p s $ q (incS i s))
442− (Block i _) *> (Block j q) = Block (i+j) (q . incS i)
443− (Block i p) <* (Block j _) = Block (i+j) p
444−
445−-- | Get a block. Will be read with one single boundry check, and
446−-- therefore requires a statically known number of bits.
447−-- Build blocks using 'bool', 'word8', 'word16be', 'word32be', 'word64be',
448−-- 'byteString' and 'Applicative'.
449−block :: Block a -> BitGet a
450−block (Block i p) = do
451− ensureBits i
452− s <- getState
453− putState $! (incS i s)
454− return $! p s
455−
456416-- | Read a 1 bit 'Bool'.
457417bool :: Block Bool
458418bool = Block 1 readBool
459419
460−-- | Read a 1 bit 'Bool'.
461−bool# :: Block Int
462−bool# = Block 1 readBool#
463−
464420-- | Read @n@ bits as a 'Word8'. @n@ must be within @[0..8]@.
465421word8 :: Int -> Block Word8
466−word8 n = Block n (readWordChecked 8 n)
422+word8 n = Block n (readWord8 n)
467423
468424-- | Read @n@ bits as a 'Word16'. @n@ must be within @[0..16]@.
469−word16 :: Int -> Block Word16
470−word16 n = Block n (readWordChecked 16 n)
471−
472425word16be :: Int -> Block Word16
473−word16be = word16
474−{-# DEPRECATED word16be "Use 'word16' instead" #-}
426+word16be n = Block n (readWord16be n)
475427
476428-- | Read @n@ bits as a 'Word32'. @n@ must be within @[0..32]@.
477−word32 :: Int -> Block Word32
478−word32 n = Block n (readWordChecked 32 n)
479−
480429word32be :: Int -> Block Word32
481−word32be = word32
482−{-# DEPRECATED word32be "Use 'word32' instead" #-}
430+word32be n = Block n (readWord32be n)
483431
484432-- | Read @n@ bits as a 'Word64'. @n@ must be within @[0..64]@.
485−word64 :: Int -> Block Word64
486−word64 n = Block n (readWordChecked 64 n)
487−
488433word64be :: Int -> Block Word64
489−word64be = word64
490−{-# DEPRECATED word64be "Use 'word64' instead" #-}
434+word64be n = Block n (readWord64be n)
491435
492436-- | Read @n@ bytes as a 'ByteString'.
493437byteString :: Int -> Block ByteString
494438byteString n | n > 0 = Block (n*8) (readByteString n)
495439 | otherwise = Block 0 (\_ -> B.empty)
496440
441+------------------------------------------------------------------------
442+-- Unchecked shifts, from the package binary
443+
444+shiftl_w16 :: Word16 -> Int -> Word16
445+shiftl_w32 :: Word32 -> Int -> Word32
446+shiftl_w64 :: Word64 -> Int -> Word64
447+
448+#if defined(__GLASGOW_HASKELL__) && !defined(__HADDOCK__)
449+shiftl_w8 (W8# w) (I# i) = W8# (w `uncheckedShiftL#` i)
450+shiftl_w16 (W16# w) (I# i) = W16# (w `uncheckedShiftL#` i)
451+shiftl_w32 (W32# w) (I# i) = W32# (w `uncheckedShiftL#` i)
452+
453+shiftr_w8 (W8# w) (I# i) = W8# (w `uncheckedShiftRL#` i)
454+shiftr_w16 (W16# w) (I# i) = W16# (w `uncheckedShiftRL#` i)
455+shiftr_w32 (W32# w) (I# i) = W32# (w `uncheckedShiftRL#` i)
456+
457+
458+#if WORD_SIZE_IN_BITS < 64
459+shiftl_w64 (W64# w) (I# i) = W64# (w `uncheckedShiftL64#` i)
460+shiftr_w64 (W64# w) (I# i) = W64# (w `uncheckedShiftRL64#` i)
461+
462+#if __GLASGOW_HASKELL__ <= 606
463+-- Exported by GHC.Word in GHC 6.8 and higher
464+foreign import ccall unsafe "stg_uncheckedShiftL64"
465+ uncheckedShiftL64# :: Word64# -> Int# -> Word64#
466+foreign import ccall unsafe "stg_uncheckedShiftRL64"
467+ uncheckedShiftRL64# :: Word64# -> Int# -> Word64#
468+#endif
469+
470+#else
471+shiftl_w64 (W64# w) (I# i) = W64# (w `uncheckedShiftL#` i)
472+shiftr_w64 (W64# w) (I# i) = W64# (w `uncheckedShiftRL#` i)
473+#endif
474+
475+#else
476+shiftl_w8 = shiftL
477+shiftl_w16 = shiftL
478+shiftl_w32 = shiftL
479+shiftl_w64 = shiftL
480+
481+shiftr_w8 = shiftR
482+shiftr_w16 = shiftR
483+shiftr_w32 = shiftR
484+shiftr_w64 = shiftR
485+#endif