module Data.X509.EC
(
unserializePoint
, ecPubKeyCurve
, ecPubKeyCurveName
, ecPrivKeyCurve
, ecPrivKeyCurveName
, lookupCurveNameByOID
) where
import Data.ASN1.OID
import Data.List (find)
import Data.X509.OID
import Data.X509.PublicKey
import Data.X509.PrivateKey
import qualified Crypto.PubKey.ECC.Prim as ECC
import qualified Crypto.PubKey.ECC.Types as ECC
import Crypto.Number.Serialize (os2ip)
import qualified Data.ByteString as B
unserializePoint :: ECC.Curve -> SerializedPoint -> Maybe ECC.Point
unserializePoint :: Curve -> SerializedPoint -> Maybe Point
unserializePoint Curve
curve (SerializedPoint ByteString
bs) =
case ByteString -> Maybe (Word8, ByteString)
B.uncons ByteString
bs of
Maybe (Word8, ByteString)
Nothing -> forall a. Maybe a
Nothing
Just (Word8
ptFormat, ByteString
input) ->
case Word8
ptFormat of
Word8
4 -> if ByteString -> Int
B.length ByteString
input forall a. Eq a => a -> a -> Bool
/= Int
2 forall a. Num a => a -> a -> a
* Int
bytes
then forall a. Maybe a
Nothing
else
let (ByteString
x, ByteString
y) = Int -> ByteString -> (ByteString, ByteString)
B.splitAt Int
bytes ByteString
input
p :: Point
p = Integer -> Integer -> Point
ECC.Point (forall ba. ByteArrayAccess ba => ba -> Integer
os2ip ByteString
x) (forall ba. ByteArrayAccess ba => ba -> Integer
os2ip ByteString
y)
in if Curve -> Point -> Bool
ECC.isPointValid Curve
curve Point
p
then forall a. a -> Maybe a
Just Point
p
else forall a. Maybe a
Nothing
Word8
_ -> forall a. Maybe a
Nothing
where bits :: Int
bits = Curve -> Int
ECC.curveSizeBits Curve
curve
bytes :: Int
bytes = (Int
bits forall a. Num a => a -> a -> a
+ Int
7) forall a. Integral a => a -> a -> a
`div` Int
8
ecPubKeyCurve :: PubKeyEC -> Maybe ECC.Curve
ecPubKeyCurve :: PubKeyEC -> Maybe Curve
ecPubKeyCurve (PubKeyEC_Named CurveName
name SerializedPoint
_) = forall a. a -> Maybe a
Just forall a b. (a -> b) -> a -> b
$ CurveName -> Curve
ECC.getCurveByName CurveName
name
ecPubKeyCurve pub :: PubKeyEC
pub@PubKeyEC_Prime{} =
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap Point -> Curve
buildCurve forall a b. (a -> b) -> a -> b
$
Curve -> SerializedPoint -> Maybe Point
unserializePoint (Point -> Curve
buildCurve forall a. HasCallStack => a
undefined) (PubKeyEC -> SerializedPoint
pubkeyEC_generator PubKeyEC
pub)
where
prime :: Integer
prime = PubKeyEC -> Integer
pubkeyEC_prime PubKeyEC
pub
buildCurve :: Point -> Curve
buildCurve Point
g =
let cc :: CurveCommon
cc = ECC.CurveCommon
{ ecc_a :: Integer
ECC.ecc_a = PubKeyEC -> Integer
pubkeyEC_a PubKeyEC
pub
, ecc_b :: Integer
ECC.ecc_b = PubKeyEC -> Integer
pubkeyEC_b PubKeyEC
pub
, ecc_g :: Point
ECC.ecc_g = Point
g
, ecc_n :: Integer
ECC.ecc_n = PubKeyEC -> Integer
pubkeyEC_order PubKeyEC
pub
, ecc_h :: Integer
ECC.ecc_h = PubKeyEC -> Integer
pubkeyEC_cofactor PubKeyEC
pub
}
in CurvePrime -> Curve
ECC.CurveFP (Integer -> CurveCommon -> CurvePrime
ECC.CurvePrime Integer
prime CurveCommon
cc)
ecPubKeyCurveName :: PubKeyEC -> Maybe ECC.CurveName
ecPubKeyCurveName :: PubKeyEC -> Maybe CurveName
ecPubKeyCurveName (PubKeyEC_Named CurveName
name SerializedPoint
_) = forall a. a -> Maybe a
Just CurveName
name
ecPubKeyCurveName pub :: PubKeyEC
pub@PubKeyEC_Prime{} =
forall (t :: * -> *) a. Foldable t => (a -> Bool) -> t a -> Maybe a
find CurveName -> Bool
matchPrimeCurve forall a b. (a -> b) -> a -> b
$ forall a. Enum a => a -> [a]
enumFrom forall a b. (a -> b) -> a -> b
$ forall a. Enum a => Int -> a
toEnum Int
0
where
matchPrimeCurve :: CurveName -> Bool
matchPrimeCurve CurveName
c =
case CurveName -> Curve
ECC.getCurveByName CurveName
c of
ECC.CurveFP (ECC.CurvePrime Integer
p CurveCommon
cc) ->
CurveCommon -> Integer
ECC.ecc_a CurveCommon
cc forall a. Eq a => a -> a -> Bool
== PubKeyEC -> Integer
pubkeyEC_a PubKeyEC
pub Bool -> Bool -> Bool
&&
CurveCommon -> Integer
ECC.ecc_b CurveCommon
cc forall a. Eq a => a -> a -> Bool
== PubKeyEC -> Integer
pubkeyEC_b PubKeyEC
pub Bool -> Bool -> Bool
&&
CurveCommon -> Integer
ECC.ecc_n CurveCommon
cc forall a. Eq a => a -> a -> Bool
== PubKeyEC -> Integer
pubkeyEC_order PubKeyEC
pub Bool -> Bool -> Bool
&&
Integer
p forall a. Eq a => a -> a -> Bool
== PubKeyEC -> Integer
pubkeyEC_prime PubKeyEC
pub
Curve
_ -> Bool
False
ecPrivKeyCurve :: PrivKeyEC -> Maybe ECC.Curve
ecPrivKeyCurve :: PrivKeyEC -> Maybe Curve
ecPrivKeyCurve (PrivKeyEC_Named CurveName
name Integer
_) = forall a. a -> Maybe a
Just forall a b. (a -> b) -> a -> b
$ CurveName -> Curve
ECC.getCurveByName CurveName
name
ecPrivKeyCurve priv :: PrivKeyEC
priv@PrivKeyEC_Prime{} =
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap Point -> Curve
buildCurve forall a b. (a -> b) -> a -> b
$
Curve -> SerializedPoint -> Maybe Point
unserializePoint (Point -> Curve
buildCurve forall a. HasCallStack => a
undefined) (PrivKeyEC -> SerializedPoint
privkeyEC_generator PrivKeyEC
priv)
where
prime :: Integer
prime = PrivKeyEC -> Integer
privkeyEC_prime PrivKeyEC
priv
buildCurve :: Point -> Curve
buildCurve Point
g =
let cc :: CurveCommon
cc = ECC.CurveCommon
{ ecc_a :: Integer
ECC.ecc_a = PrivKeyEC -> Integer
privkeyEC_a PrivKeyEC
priv
, ecc_b :: Integer
ECC.ecc_b = PrivKeyEC -> Integer
privkeyEC_b PrivKeyEC
priv
, ecc_g :: Point
ECC.ecc_g = Point
g
, ecc_n :: Integer
ECC.ecc_n = PrivKeyEC -> Integer
privkeyEC_order PrivKeyEC
priv
, ecc_h :: Integer
ECC.ecc_h = PrivKeyEC -> Integer
privkeyEC_cofactor PrivKeyEC
priv
}
in CurvePrime -> Curve
ECC.CurveFP (Integer -> CurveCommon -> CurvePrime
ECC.CurvePrime Integer
prime CurveCommon
cc)
ecPrivKeyCurveName :: PrivKeyEC -> Maybe ECC.CurveName
ecPrivKeyCurveName :: PrivKeyEC -> Maybe CurveName
ecPrivKeyCurveName (PrivKeyEC_Named CurveName
name Integer
_) = forall a. a -> Maybe a
Just CurveName
name
ecPrivKeyCurveName priv :: PrivKeyEC
priv@PrivKeyEC_Prime{} =
forall (t :: * -> *) a. Foldable t => (a -> Bool) -> t a -> Maybe a
find CurveName -> Bool
matchPrimeCurve forall a b. (a -> b) -> a -> b
$ forall a. Enum a => a -> [a]
enumFrom forall a b. (a -> b) -> a -> b
$ forall a. Enum a => Int -> a
toEnum Int
0
where
matchPrimeCurve :: CurveName -> Bool
matchPrimeCurve CurveName
c =
case CurveName -> Curve
ECC.getCurveByName CurveName
c of
ECC.CurveFP (ECC.CurvePrime Integer
p CurveCommon
cc) ->
CurveCommon -> Integer
ECC.ecc_a CurveCommon
cc forall a. Eq a => a -> a -> Bool
== PrivKeyEC -> Integer
privkeyEC_a PrivKeyEC
priv Bool -> Bool -> Bool
&&
CurveCommon -> Integer
ECC.ecc_b CurveCommon
cc forall a. Eq a => a -> a -> Bool
== PrivKeyEC -> Integer
privkeyEC_b PrivKeyEC
priv Bool -> Bool -> Bool
&&
CurveCommon -> Integer
ECC.ecc_n CurveCommon
cc forall a. Eq a => a -> a -> Bool
== PrivKeyEC -> Integer
privkeyEC_order PrivKeyEC
priv Bool -> Bool -> Bool
&&
Integer
p forall a. Eq a => a -> a -> Bool
== PrivKeyEC -> Integer
privkeyEC_prime PrivKeyEC
priv
Curve
_ -> Bool
False
lookupCurveNameByOID :: OID -> Maybe ECC.CurveName
lookupCurveNameByOID :: OID -> Maybe CurveName
lookupCurveNameByOID = forall a. OIDTable a -> OID -> Maybe a
lookupByOID OIDTable CurveName
curvesOIDTable