hs-tls/Tests/Certificate.hs

60 lines
1.7 KiB
Haskell

module Tests.Certificate
( arbitraryX509
, arbitraryX509WithPublicKey
) where
import Test.QuickCheck
import qualified Data.Certificate.X509 as X509
import qualified Data.Certificate.X509.Cert as Cert
import Control.Monad
import Data.Time.Calendar (fromGregorian)
import Data.Time.Clock (secondsToDiffTime)
import qualified Data.ByteString as B
import Tests.PubKey
readableChar :: Gen Char
readableChar = elements (['a'..'z'] ++ ['A'..'Z'] ++ ['0'..'9'])
arbitraryDN = return []
arbitraryTime = do
year <- choose (1951, 2050)
month <- choose (1, 12)
day <- choose (1, 30)
hour <- choose (0, 23)
minute <- choose (0, 59)
second <- choose (0, 59)
z <- arbitrary
return (fromGregorian year month day
, secondsToDiffTime (hour * 3600 + minute * 60 + second)
, z)
arbitraryX509Cert pubKey = do
version <- choose (1,3)
serial <- choose (0,2^24)
issuerdn <- arbitraryDN
subjectdn <- arbitraryDN
time1 <- arbitraryTime
time2 <- arbitraryTime
let sigalg = X509.SignatureALG X509.HashMD5 X509.PubKeyALG_RSA
return $ Cert.Certificate
{ X509.certVersion = version
, X509.certSerial = serial
, X509.certSignatureAlg = sigalg
, X509.certIssuerDN = issuerdn
, X509.certSubjectDN = subjectdn
, X509.certValidity = (time1, time2)
, X509.certPubKey = pubKey
, X509.certExtensions = Nothing
}
arbitraryX509WithPublicKey pubKey = do
cert <- arbitraryX509Cert (X509.PubKeyRSA pubKey)
sig <- resize 40 $ listOf1 arbitrary
let sigalg = X509.SignatureALG X509.HashMD5 X509.PubKeyALG_RSA
return (X509.X509 cert Nothing Nothing sigalg sig)
arbitraryX509 = do
let pubKey = fst $ getGlobalRSAPair
arbitraryX509WithPublicKey pubKey