module Data.ByteString.Pack
( Packer
, Result(..)
, pack
, putWord8
, putWord16
, putWord32
, putStorable
, putByteString
, fillList
, fillUpWith
, putByteStringBase32
, guessEncodedLength
, skip
, skipStorable
) where
import Data.ByteString.Internal (ByteString(..))
import qualified Data.ByteString as B
import qualified Data.ByteString.Internal as B
import Data.Word
import Foreign.ForeignPtr (withForeignPtr)
import Foreign.Ptr
import Foreign.Storable
import System.IO.Unsafe (unsafePerformIO)
import Data.ByteString.Pack.Internal
import Data.ByteString.Pack.Base32
pack :: Packer a -> Int -> Either String ByteString
pack p len =
unsafePerformIO $ do
fptr <- B.mallocByteString len
val <- withForeignPtr fptr $ \ptr ->
runPacker_ p (Cache ptr len)
return $ case val of
PackerMore _ (Cache _ r) -> Right (PS fptr 0 (len r))
PackerFail err -> Left err
fillUpWithWord8' :: Word8 -> Packer ()
fillUpWithWord8' w = Packer $ \(Cache ptr size) -> do
_ <- B.memset ptr w (fromIntegral size)
return $ PackerMore () (Cache (ptr `plusPtr` size) (0))
putStorable :: Storable storable => storable -> Packer ()
putStorable s = actionPacker (sizeOf s) (\ptr -> poke (castPtr ptr) s)
putByteString :: ByteString -> Packer ()
putByteString bs
| neededLength == 0 = return ()
| otherwise = actionPacker neededLength (actionPackerByteString bs)
where
neededLength :: Int
neededLength = B.length bs
actionPackerByteString :: ByteString -> Ptr Word8 -> IO ()
actionPackerByteString (PS fptr off _) ptr =
withForeignPtr fptr $ \srcptr ->
B.memcpy ptr (srcptr `plusPtr` off) neededLength
skip :: Int -> Packer ()
skip n = actionPacker n (\_ -> return ())
skipStorable :: Storable storable => storable -> Packer ()
skipStorable = skip . sizeOf
fillUpWith :: Storable storable => storable -> Packer ()
fillUpWith s = fillList $ repeat s
fillList :: Storable storable => [storable] -> Packer ()
fillList [] = return ()
fillList (x:xs) = putStorable x >> fillList xs
putWord8 :: Word8 -> Packer ()
putWord8 = putStorable
putWord16 :: Word16 -> Packer ()
putWord16 = putStorable
putWord32 :: Word32 -> Packer ()
putWord32 = putStorable