module Data.Digest.CRC32 (
CRC32, crc32, crc32Update
) where
import Foreign
import Foreign.C.Types
import Foreign.ForeignPtr
import GHC.Ptr
import qualified Data.ByteString as S
import qualified Data.ByteString.Internal as BI
import qualified Data.ByteString.Lazy as L
import qualified Data.ByteString.Lazy.Internal as LI
class CRC32 a where
crc32 :: a -> Word32
crc32 = crc32Update 0
crc32Update :: Word32 -> a -> Word32
instance CRC32 S.ByteString where
crc32Update = crc32_s_update
instance CRC32 L.ByteString where
crc32Update = crc32_l_update
instance CRC32 [Word8] where
crc32Update n = (crc32Update n) . L.pack
crc32_s_update :: Word32 -> S.ByteString -> Word32
crc32_s_update n s = crc32_l_update n (LI.Chunk s LI.Empty)
crc32_l_update :: Word32 -> L.ByteString -> Word32
crc32_l_update n = LI.foldlChunks updateCRC n
where updateCRC crc bs = fromIntegral $ crc32_c (fromIntegral crc) buf (fromIntegral len)
where (ptr, offset, len) = BI.toForeignPtr bs
buf = (unsafeForeignPtrToPtr ptr) `plusPtr` offset
foreign import ccall unsafe "zlib.h crc32"
crc32_c :: CInt -> Ptr Word8 -> CInt -> CInt