haskell/cryptonite
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge last cshakeUpdate with cshakeFinalize

Browse Source
This commit is contained in:
Olivier Chéron 2019-05-01 07:03:45 +02:00
parent 8b235612be
commit ae107a9285
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
Crypto/MAC/KMAC.hs
View File

@ -58,10 +58,13 @@ cshakeUpdates :: (HashSHAKE a, ByteArrayAccess ba)
=> H.Context a -> [ba] -> H.Context a => H.Context a -> [ba] -> H.Context a
cshakeUpdates = H.hashUpdates cshakeUpdates = H.hashUpdates
cshakeFinalize :: forall a . HashSHAKE a => H.Context a -> Digest a cshakeFinalize :: forall a suffix . (HashSHAKE a, ByteArrayAccess suffix)
cshakeFinalize !c = => H.Context a -> suffix -> Digest a
cshakeFinalize !c s =
Digest $ B.allocAndFreeze (hashDigestSize (undefined :: a)) $ \dig -> do Digest $ B.allocAndFreeze (hashDigestSize (undefined :: a)) $ \dig -> do
((!_) :: B.Bytes) <- B.copy c $ \(ctx :: Ptr (H.Context a)) -> ((!_) :: B.Bytes) <- B.copy c $ \(ctx :: Ptr (H.Context a)) -> do
B.withByteArray s $ \d ->
hashInternalUpdate ctx d (fromIntegral $ B.length s)
cshakeInternalFinalize ctx dig cshakeInternalFinalize ctx dig
return () return ()
@ -108,7 +111,7 @@ updates (Context ctx) = Context . cshakeUpdates ctx
-- | Finalize a KMAC context and return the KMAC. -- | Finalize a KMAC context and return the KMAC.
finalize :: forall a . HashSHAKE a => Context a -> KMAC a finalize :: forall a . HashSHAKE a => Context a -> KMAC a
finalize (Context ctx) = KMAC $ cshakeFinalize $ cshakeUpdate ctx suffix finalize (Context ctx) = KMAC $ cshakeFinalize ctx suffix
where where
l = cshakeOutputLength (undefined :: a) l = cshakeOutputLength (undefined :: a)
suffix = builderAllocAndFreeze (rightEncode l) :: B.Bytes suffix = builderAllocAndFreeze (rightEncode l) :: B.Bytes