fixing SignatureAlgorithmsCert.

haskell-tls · Aug 6, 2018 · 33ba430 · 33ba430
1 parent 03f5fa8
commit 33ba430
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/core/Network/TLS/Extension.hs b/core/Network/TLS/Extension.hs
@@ -357,7 +357,16 @@ instance Extension SignatureAlgorithms where
             len <- getWord16
             SignatureAlgorithms <$> getList (fromIntegral len) (getSignatureHashAlgorithm >>= \sh -> return (2, sh))
 
-type SignatureAlgorithmsCert = SignatureAlgorithms
+newtype SignatureAlgorithmsCert = SignatureAlgorithmsCert [HashAndSignatureAlgorithm] deriving (Show,Eq)
+
+instance Extension SignatureAlgorithmsCert where
+    extensionID _ = extensionID_SignatureAlgorithms
+    extensionEncode (SignatureAlgorithmsCert algs) =
+        runPut $ putWord16 (fromIntegral (length algs * 2)) >> mapM_ putSignatureHashAlgorithm algs
+    extensionDecode _ =
+        runGetMaybe $ do
+            len <- getWord16
+            SignatureAlgorithmsCert <$> getList (fromIntegral len) (getSignatureHashAlgorithm >>= \sh -> return (2, sh))
 
 data SupportedVersions =
     SupportedVersionsClientHello [Version]