summaryrefslogtreecommitdiffhomepage
path: root/Network/DNS/StateBinary.hs
diff options
context:
space:
mode:
Diffstat (limited to 'Network/DNS/StateBinary.hs')
-rw-r--r--Network/DNS/StateBinary.hs8
1 files changed, 5 insertions, 3 deletions
diff --git a/Network/DNS/StateBinary.hs b/Network/DNS/StateBinary.hs
index 5099104..bd54a75 100644
--- a/Network/DNS/StateBinary.hs
+++ b/Network/DNS/StateBinary.hs
@@ -3,6 +3,8 @@ module Network.DNS.StateBinary where
import Blaze.ByteString.Builder
import Control.Monad.State
import Data.Binary.Get
+import Data.ByteString (ByteString)
+import qualified Data.ByteString.Char8 as BS
import qualified Data.ByteString.Lazy.Char8 as L
import Data.Char
import Data.Int
@@ -85,10 +87,10 @@ getPosition = fromIntegral <$> lift bytesRead
getNBytes :: Int -> SGet [Int]
getNBytes len = toInts <$> getNByteString len
where
- toInts = map ord . L.unpack
+ toInts = map ord . BS.unpack
-getNByteString :: Int -> SGet L.ByteString
-getNByteString len = lift . getLazyByteString . fromIntegral $ len
+getNByteString :: Int -> SGet ByteString
+getNByteString = lift . getByteString . fromIntegral
----------------------------------------------------------------