secp256k1

commit 2247e534684d46c339f1a961918f4028ca0d8468
parent 581dd31fde3b38a6f0c360b377d3bff0dc9e0d9e
Author: Jared Tobin <jared@jtobin.io>
Date:   Wed,  9 Oct 2024 16:28:00 +0400

test: wycheproof skeleton

Diffstat:
M
lib/Crypto/Curve/Secp256k1.hs
 | 
20
+++++++++++++++++++-
M
ppad-secp256k1.cabal
 | 
2
++
M
test/Main.hs
 | 
11
+++++++++++
A
test/Wycheproof.hs
 | 
53
+++++++++++++++++++++++++++++++++++++++++++++++++++++

4 files changed, 85 insertions(+), 1 deletion(-)
diff --git a/lib/Crypto/Curve/Secp256k1.hs b/lib/Crypto/Curve/Secp256k1.hs
@@ -6,7 +6,25 @@
 {-# LANGUAGE UnboxedSums #-}
 {-# LANGUAGE ViewPatterns #-}
 
-module Crypto.Curve.Secp256k1 where
+module Crypto.Curve.Secp256k1 (
+    Affine(..)
+  , Projective(..)
+  , affine
+  , projective
+
+  , neg
+  , add
+  , double
+  , mul
+  , mul_safe
+
+  , parse_point
+
+  , ECDSA(..)
+  , sign
+  , sign_unrestricted
+  , verify
+  ) where
 
 import Control.Monad (when)
 import Control.Monad.ST
diff --git a/ppad-secp256k1.cabal b/ppad-secp256k1.cabal
@@ -37,6 +37,8 @@ test-suite secp256k1-tests
   default-language:    Haskell2010
   hs-source-dirs:      test
   main-is:             Main.hs
+  other-modules:
+    Wycheproof
 
   ghc-options:
     -rtsopts -Wall
diff --git a/test/Main.hs b/test/Main.hs
@@ -1,12 +1,23 @@
+{-# LANGUAGE BangPatterns #-}
 {-# LANGUAGE OverloadedStrings #-}
 
 module Main where
 
+import qualified Data.Bits as B
+import qualified Data.Attoparsec.ByteString as A
 import qualified Data.ByteString as BS
+import qualified Data.ByteString.Base16 as B16
 import Crypto.Curve.Secp256k1
 import Test.Tasty
 import Test.Tasty.HUnit
 
+fi :: (Integral a, Num b) => a -> b
+fi = fromIntegral
+{-# INLINE fi #-}
+
+test_sig :: BS.ByteString
+test_sig = "3046022100f80ae4f96cdbc9d853f83d47aae225bf407d51c56b7776cd67d0dc195d99a9dc022100b303e26be1f73465315221f0b331528807a1a9b6eb068ede6eebeaaa49af8a36"
+
 main :: IO ()
 main = defaultMain units
 
diff --git a/test/Wycheproof.hs b/test/Wycheproof.hs
@@ -0,0 +1,53 @@
+{-# LANGUAGE BangPatterns #-}
+{-# LANGUAGE OverloadedStrings #-}
+
+module Wycheproof where
+
+import Crypto.Curve.Secp256k1
+import qualified Data.Bits as B
+import qualified Data.Attoparsec.ByteString as A
+import qualified Data.ByteString as BS
+import qualified Data.ByteString.Base16 as B16
+
+fi :: (Integral a, Num b) => a -> b
+fi = fromIntegral
+{-# INLINE fi #-}
+
+parse_DER :: A.Parser ECDSA
+parse_DER = do
+  _ <- A.word8 0x30
+  _ <- A.takeTill (== 0x02)
+  _ <- A.word8 0x02
+  r_len <- fmap fi A.anyWord8
+  bs_r <- A.take r_len
+  _ <- A.word8 0x02
+  s_len <- fmap fi A.anyWord8
+  bs_s <- A.take s_len
+  let r = fi (parse_integer_be bs_r)
+      s = fi (parse_integer_be bs_s)
+  pure (ECDSA r s)
+
+-- dumb integer parser
+parse_integer_be :: BS.ByteString -> Integer
+parse_integer_be = BS.foldl' alg 0 where
+  alg !acc b = B.shiftL acc 8 + fromIntegral b
+{-# INLINE parse_integer_be #-}
+
+-- XX delete me when done
+
+test_pub :: BS.ByteString
+test_pub = "04782c8ed17e3b2a783b5464f33b09652a71c678e05ec51e84e2bcfc663a3de963af9acb4280b8c7f7c42f4ef9aba6245ec1ec1712fd38a0fa96418d8cd6aa6152"
+
+pub =
+  let Just p = parse_point test_pub
+  in  p
+
+test_msg :: BS.ByteString
+test_msg = mempty
+
+test_sig :: BS.ByteString
+test_sig = "3046022100f80ae4f96cdbc9d853f83d47aae225bf407d51c56b7776cd67d0dc195d99a9dc022100b303e26be1f73465315221f0b331528807a1a9b6eb068ede6eebeaaa49af8a36"
+
+sig =
+  let Right s = A.parseOnly parse_DER (B16.decodeLenient test_sig)
+  in  s