{-# LANGUAGE GeneralizedNewtypeDeriving #-}
module Crypto.PubKey.DH
( Params(..)
, PublicNumber
, PrivateNumber
, SharedKey
, generateParams
, generatePrivate
, calculatePublic
, generatePublic
, getShared
) where
import Crypto.Number.ModArithmetic (expSafe)
import Crypto.Number.Prime (generateSafePrime)
import Crypto.Number.Generate (generateMax)
import Crypto.Types.PubKey.DH
import Crypto.Random
import Control.Arrow (first)
generateParams :: CPRG g => g -> Int -> Integer -> (Params, g)
generateParams :: g -> Int -> Integer -> (Params, g)
generateParams rng :: g
rng bits :: Int
bits generator :: Integer
generator =
(Integer -> Params) -> (Integer, g) -> (Params, g)
forall (a :: * -> * -> *) b c d.
Arrow a =>
a b c -> a (b, d) (c, d)
first (\p :: Integer
p -> Integer -> Integer -> Params
Params Integer
p Integer
generator) ((Integer, g) -> (Params, g)) -> (Integer, g) -> (Params, g)
forall a b. (a -> b) -> a -> b
$ g -> Int -> (Integer, g)
forall g. CPRG g => g -> Int -> (Integer, g)
generateSafePrime g
rng Int
bits
generatePrivate :: CPRG g => g -> Params -> (PrivateNumber, g)
generatePrivate :: g -> Params -> (PrivateNumber, g)
generatePrivate rng :: g
rng (Params p :: Integer
p _) = (Integer -> PrivateNumber) -> (Integer, g) -> (PrivateNumber, g)
forall (a :: * -> * -> *) b c d.
Arrow a =>
a b c -> a (b, d) (c, d)
first Integer -> PrivateNumber
PrivateNumber ((Integer, g) -> (PrivateNumber, g))
-> (Integer, g) -> (PrivateNumber, g)
forall a b. (a -> b) -> a -> b
$ g -> Integer -> (Integer, g)
forall g. CPRG g => g -> Integer -> (Integer, g)
generateMax g
rng Integer
p
calculatePublic :: Params -> PrivateNumber -> PublicNumber
calculatePublic :: Params -> PrivateNumber -> PublicNumber
calculatePublic (Params p :: Integer
p g :: Integer
g) (PrivateNumber x :: Integer
x) = Integer -> PublicNumber
PublicNumber (Integer -> PublicNumber) -> Integer -> PublicNumber
forall a b. (a -> b) -> a -> b
$ Integer -> Integer -> Integer -> Integer
expSafe Integer
g Integer
x Integer
p
generatePublic :: Params -> PrivateNumber -> PublicNumber
generatePublic :: Params -> PrivateNumber -> PublicNumber
generatePublic = Params -> PrivateNumber -> PublicNumber
calculatePublic
getShared :: Params -> PrivateNumber -> PublicNumber -> SharedKey
getShared :: Params -> PrivateNumber -> PublicNumber -> SharedKey
getShared (Params p :: Integer
p _) (PrivateNumber x :: Integer
x) (PublicNumber y :: Integer
y) = Integer -> SharedKey
SharedKey (Integer -> SharedKey) -> Integer -> SharedKey
forall a b. (a -> b) -> a -> b
$ Integer -> Integer -> Integer -> Integer
expSafe Integer
y Integer
x Integer
p