Protect against negative argument

haskell-crypto · Feb 24, 2020 · 981b97a · 981b97a
1 parent 2e0a60f
commit 981b97a
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/Crypto/Internal/Builder.hs b/Crypto/Internal/Builder.hs
@@ -26,7 +26,7 @@ import           Data.Memory.PtrMethods (memSet)
 import           Foreign.Ptr (Ptr, plusPtr)
 import           Foreign.Storable (poke)
 
-import           Crypto.Internal.Imports
+import           Crypto.Internal.Imports hiding (empty)
 
 data Builder =  Builder !Int (Ptr Word8 -> IO ())  -- size and initializer
 
@@ -47,4 +47,7 @@ bytes :: ByteArrayAccess ba => ba -> Builder
 bytes bs = Builder (B.length bs) (B.copyByteArrayToPtr bs)
 
 zero :: Int -> Builder
-zero s = Builder s (\p -> memSet p 0 s)
+zero s = if s > 0 then Builder s (\p -> memSet p 0 s) else empty
+
+empty :: Builder
+empty = Builder 0 (const $ return ())