tests/testx509.py

   1 #!/usr/bin/env python
   2 # -*- encoding: utf-8 -*-
   3
   4 from ctypescrypto.x509 import X509,X509Store,utc
   5 from ctypescrypto.oid import Oid
   6 from tempfile import NamedTemporaryFile
   7 import datetime
   8 import unittest
   9
  10
  11
  12 class TestCertInfo(unittest.TestCase):
  13         ca_cert="""-----BEGIN CERTIFICATE-----
  14 MIIEDzCCAvegAwIBAgIJAN9Ejmna3JJ7MA0GCSqGSIb3DQEBBQUAMIGdMQswCQYD
  15 VQQGEwJSVTEVMBMGA1UECAwM0JzQvtGB0LrQstCwMTAwLgYDVQQKDCfQo9C00L7R
  16 gdGC0L7QstC10YDRj9GO0YnQuNC5INGG0LXQvdGC0YAxIjAgBgNVBAMMGdCS0LjQ
  17 utGC0L7RgCDQktCw0LPQvdC10YAxITAfBgkqhkiG9w0BCQEWEnZpdHVzQHdhZ25l
  18 ci5wcC5ydTAeFw0xNDEwMjYxNDQ2MzJaFw0xNzEwMjUxNDQ2MzJaMIGdMQswCQYD
  19 VQQGEwJSVTEVMBMGA1UECAwM0JzQvtGB0LrQstCwMTAwLgYDVQQKDCfQo9C00L7R
  20 gdGC0L7QstC10YDRj9GO0YnQuNC5INGG0LXQvdGC0YAxIjAgBgNVBAMMGdCS0LjQ
  21 utGC0L7RgCDQktCw0LPQvdC10YAxITAfBgkqhkiG9w0BCQEWEnZpdHVzQHdhZ25l
  22 ci5wcC5ydTCCASIwDQYJKoZIhvcNAQEBBQADggEPADCCAQoCggEBAJn+nL8CDaM0
  23 KNafGYdEDuFuCKHFxCcbaT7ecGbwjPKtnqZLOnYpa2iLFY+n7zAYw1MRkFkaR8b+
  24 +AeFPWS9T60ygeWysT9XTS77Fdl69Tmr8HChRk0BuLB3vFCy34vBHSG6Frdm8PtS
  25 PLEleldiwUCHLS+EckrnJylQK13X3HofSbIGzKK53MsuQOtp2dJw3b7MILM/+XBm
  26 RgZlEbTEPYMOH8CE3mu9/LqXfIRQM7+nmzcNZI3RAwxEVHOSHEbgFZaksTj8rMIa
  27 SrJwknmxHntI3P5PSNNbs0SO3TW8ePDIIpbVcjNsMX4qGX8b+8quZuzciKOto8S0
  28 0A6eOBd8Vi0CAwEAAaNQME4wHQYDVR0OBBYEFKzcbd6+N1TKfBjvmyTvw8+DnzAZ
  29 MB8GA1UdIwQYMBaAFKzcbd6+N1TKfBjvmyTvw8+DnzAZMAwGA1UdEwQFMAMBAf8w
  30 DQYJKoZIhvcNAQEFBQADggEBAAa1PpkpL842hh8jLXIpA/nK8aVDDcu5p3pA72/b
  31 noFnZuKcuaSUOz1rrLqxDK2JB3lmChQaVx3pZwqJgA0h0XBScar+8wM2TfeyW+oU
  32 Gr5tOAxoHVRpgn6oCoJkKo0HS2/NA12T/gYsXhXJXn4tuvDjaUzY+K+hhAWh64oL
  33 /c61eKfCZKp50t9Eoua0xHII2Mveb27Ps46j/CZ1r0ts7sGieOqjQo3GZOOikG6F
  34 vFY/2KV16/FdBovTFWMyKrzlYHm0Wgt28IWqhocq/golLfvkz3VAkLQvOF2i6hNc
  35 4feBv69SRTsTCFN9PtJCtxPX/K9LZKeccBKgGjrHQpAF+JU=
  36 -----END CERTIFICATE-----
  37 """
  38         cert1="""-----BEGIN CERTIFICATE-----
  39 MIIEDzCCAvegAwIBAgIJAN9Ejmna3JJ8MA0GCSqGSIb3DQEBBQUAMIGdMQswCQYD
  40 VQQGEwJSVTEVMBMGA1UECAwM0JzQvtGB0LrQstCwMTAwLgYDVQQKDCfQo9C00L7R
  41 gdGC0L7QstC10YDRj9GO0YnQuNC5INGG0LXQvdGC0YAxIjAgBgNVBAMMGdCS0LjQ
  42 utGC0L7RgCDQktCw0LPQvdC10YAxITAfBgkqhkiG9w0BCQEWEnZpdHVzQHdhZ25l
  43 ci5wcC5ydTAeFw0xNDEwMjYxOTA3MTdaFw0yNDEwMjMxOTA3MTdaMIGBMQswCQYD
  44 VQQGEwJSVTEVMBMGA1UECAwM0JzQvtGB0LrQstCwMRUwEwYDVQQHDAzQnNC+0YHQ
  45 utCy0LAxIDAeBgNVBAoMF9Cn0LDRgdGC0L3QvtC1INC70LjRhtC+MSIwIAYDVQQD
  46 DBnQktC40LrRgtC+0YAg0JLQsNCz0L3QtdGAMIIBIjANBgkqhkiG9w0BAQEFAAOC
  47 AQ8AMIIBCgKCAQEArQSfrrxNROyzNEz60G2EHBP+E4BL0b1QytGAZZiQp2XIhhQe
  48 b7mx+c4mpwgvD7/IdAcK+YVGx78nfY723T3wG48U7HzFNbLvNDycxyXecXbvCmRs
  49 xPy8TxkwPf6TIT3UcixtwMMqZFqlAtSTDmOOWSaUuftL/+yFk729xDoYkOZhFwUS
  50 UM5SbEZ0JpufWFjDi3Qwj3ZOTXliHC3e4C7187Me0Nne59dttyKpq1YAThn4Srar
  51 vZYU6Ykk/LUae0FCvfeiKLShWY05XnPVmvPiiFTXJP8/Au8kfezlA4b+eS81zWq2
  52 BFvNlBQsgf04S88oew0CuBBgtjUIIw7XZkS03QIDAQABo2wwajAJBgNVHRMEAjAA
  53 MB0GA1UdDgQWBBRflZBerCFYheRQne/sWL3zY7GiAzAfBgNVHSMEGDAWgBSs3G3e
  54 vjdUynwY75sk78PPg58wGTAdBgNVHREEFjAUgRJ2aXR1c0B3YWduZXIucHAucnUw
  55 DQYJKoZIhvcNAQEFBQADggEBAGx1z0ylq90hP3x/2DmfVUYBA46CiGnV4NSiaOWE
  56 Y18jCuG3W8FcI7JP4uEEjKyz3XbuhTFW2GsZ2L3FGgpA5eXBikgCn5kRpOHgb45r
  57 SxE8u3TwVlYlaF+7RHPYLqmgb25d/O/28McemMmTGecPC9edbtDqLv03aJ0t4gXn
  58 BD+xTJOP74Yhu5IPIV92J6pSBpIoy+qiyOA1iRpOWzrVHVR504vAaFxlfZs3VJhP
  59 uo291iEXyooazJdbWwZwcwk7WrNNKhqktPTg0X1ZHNnGwOAGPzwNJFGPeFj71r0t
  60 aFWU5EMRKaZK75keXq/RdaOAenl+nKF6xA2XHDhGgdndFfY=
  61 -----END CERTIFICATE-----
  62 """
  63         pubkey1="""-----BEGIN PUBLIC KEY-----
  64 MIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8AMIIBCgKCAQEArQSfrrxNROyzNEz60G2E
  65 HBP+E4BL0b1QytGAZZiQp2XIhhQeb7mx+c4mpwgvD7/IdAcK+YVGx78nfY723T3w
  66 G48U7HzFNbLvNDycxyXecXbvCmRsxPy8TxkwPf6TIT3UcixtwMMqZFqlAtSTDmOO
  67 WSaUuftL/+yFk729xDoYkOZhFwUSUM5SbEZ0JpufWFjDi3Qwj3ZOTXliHC3e4C71
  68 87Me0Nne59dttyKpq1YAThn4SrarvZYU6Ykk/LUae0FCvfeiKLShWY05XnPVmvPi
  69 iFTXJP8/Au8kfezlA4b+eS81zWq2BFvNlBQsgf04S88oew0CuBBgtjUIIw7XZkS0
  70 3QIDAQAB
  71 -----END PUBLIC KEY-----
  72 """
  73         digicert_cert="""digicert.crt
  74 -----BEGIN CERTIFICATE-----
  75 MIIG5jCCBc6gAwIBAgIQAze5KDR8YKauxa2xIX84YDANBgkqhkiG9w0BAQUFADBs
  76 MQswCQYDVQQGEwJVUzEVMBMGA1UEChMMRGlnaUNlcnQgSW5jMRkwFwYDVQQLExB3
  77 d3cuZGlnaWNlcnQuY29tMSswKQYDVQQDEyJEaWdpQ2VydCBIaWdoIEFzc3VyYW5j
  78 ZSBFViBSb290IENBMB4XDTA3MTEwOTEyMDAwMFoXDTIxMTExMDAwMDAwMFowaTEL
  79 MAkGA1UEBhMCVVMxFTATBgNVBAoTDERpZ2lDZXJ0IEluYzEZMBcGA1UECxMQd3d3
  80 LmRpZ2ljZXJ0LmNvbTEoMCYGA1UEAxMfRGlnaUNlcnQgSGlnaCBBc3N1cmFuY2Ug
  81 RVYgQ0EtMTCCASIwDQYJKoZIhvcNAQEBBQADggEPADCCAQoCggEBAPOWYth1bhn/
  82 PzR8SU8xfg0ETpmB4rOFVZEwscCvcLssqOcYqj9495BoUoYBiJfiOwZlkKq9ZXbC
  83 7L4QWzd4g2B1Rca9dKq2n6Q6AVAXxDlpufFP74LByvNK28yeUE9NQKM6kOeGZrzw
  84 PnYoTNF1gJ5qNRQ1A57bDIzCKK1Qss72kaPDpQpYSfZ1RGy6+c7pqzoC4E3zrOJ6
  85 4GAiBTyC01Li85xH+DvYskuTVkq/cKs+6WjIHY9YHSpNXic9rQpZL1oRIEDZaARo
  86 LfTAhAsKG3jf7RpY3PtBWm1r8u0c7lwytlzs16YDMqbo3rcoJ1mIgP97rYlY1R4U
  87 pPKwcNSgPqcCAwEAAaOCA4UwggOBMA4GA1UdDwEB/wQEAwIBhjA7BgNVHSUENDAy
  88 BggrBgEFBQcDAQYIKwYBBQUHAwIGCCsGAQUFBwMDBggrBgEFBQcDBAYIKwYBBQUH
  89 AwgwggHEBgNVHSAEggG7MIIBtzCCAbMGCWCGSAGG/WwCATCCAaQwOgYIKwYBBQUH
  90 AgEWLmh0dHA6Ly93d3cuZGlnaWNlcnQuY29tL3NzbC1jcHMtcmVwb3NpdG9yeS5o
  91 dG0wggFkBggrBgEFBQcCAjCCAVYeggFSAEEAbgB5ACAAdQBzAGUAIABvAGYAIAB0
  92 AGgAaQBzACAAQwBlAHIAdABpAGYAaQBjAGEAdABlACAAYwBvAG4AcwB0AGkAdAB1
  93 AHQAZQBzACAAYQBjAGMAZQBwAHQAYQBuAGMAZQAgAG8AZgAgAHQAaABlACAARABp
  94 AGcAaQBDAGUAcgB0ACAARQBWACAAQwBQAFMAIABhAG4AZAAgAHQAaABlACAAUgBl
  95 AGwAeQBpAG4AZwAgAFAAYQByAHQAeQAgAEEAZwByAGUAZQBtAGUAbgB0ACAAdwBo
  96 AGkAYwBoACAAbABpAG0AaQB0ACAAbABpAGEAYgBpAGwAaQB0AHkAIABhAG4AZAAg
  97 AGEAcgBlACAAaQBuAGMAbwByAHAAbwByAGEAdABlAGQAIABoAGUAcgBlAGkAbgAg
  98 AGIAeQAgAHIAZQBmAGUAcgBlAG4AYwBlAC4wEgYDVR0TAQH/BAgwBgEB/wIBADCB
  99 gwYIKwYBBQUHAQEEdzB1MCQGCCsGAQUFBzABhhhodHRwOi8vb2NzcC5kaWdpY2Vy
 100 dC5jb20wTQYIKwYBBQUHMAKGQWh0dHA6Ly93d3cuZGlnaWNlcnQuY29tL0NBQ2Vy
 101 dHMvRGlnaUNlcnRIaWdoQXNzdXJhbmNlRVZSb290Q0EuY3J0MIGPBgNVHR8EgYcw
 102 gYQwQKA+oDyGOmh0dHA6Ly9jcmwzLmRpZ2ljZXJ0LmNvbS9EaWdpQ2VydEhpZ2hB
 103 c3N1cmFuY2VFVlJvb3RDQS5jcmwwQKA+oDyGOmh0dHA6Ly9jcmw0LmRpZ2ljZXJ0
 104 LmNvbS9EaWdpQ2VydEhpZ2hBc3N1cmFuY2VFVlJvb3RDQS5jcmwwHQYDVR0OBBYE
 105 FExYyyXwQU9S9CjIgUObpqig5pLlMB8GA1UdIwQYMBaAFLE+w2kD+L9HAdSYJhoI
 106 Au9jZCvDMA0GCSqGSIb3DQEBBQUAA4IBAQBMeheHKF0XvLIyc7/NLvVYMR3wsXFU
 107 nNabZ5PbLwM+Fm8eA8lThKNWYB54lBuiqG+jpItSkdfdXJW777UWSemlQk808kf/
 108 roF/E1S3IMRwFcuBCoHLdFfcnN8kpCkMGPAc5K4HM+zxST5Vz25PDVR708noFUjU
 109 xbvcNRx3RQdIRYW9135TuMAW2ZXNi419yWBP0aKb49Aw1rRzNubS+QOy46T15bg+
 110 BEkAui6mSnKDcp33C4ypieez12Qf1uNgywPE3IjpnSUBAHHLA7QpYCWP+UbRe3Gu
 111 zVMSW4SOwg/H7ZMZ2cn6j1g0djIvruFQFGHUqFijyDATI+/GJYw2jxyA
 112 -----END CERTIFICATE-----
 113 """
 114         def test_readpubkey(self):
 115                 c=X509(self.cert1)
 116                 p=c.pubkey
 117                 self.assertEqual(p.exportpub(),self.pubkey1)
 118         def test_subject(self):
 119                 c=X509(self.cert1)
 120                 self.assertEqual(unicode(c.subject),u'C=RU,ST=Москва,L=Москва,O=Частное лицо,CN=Виктор Вагнер')
 121         def test_issuer(self):
 122                 c=X509(self.cert1)
 123                 self.assertEqual(unicode(c.issuer),u'C=RU,ST=Москва,O=Удостоверяющий центр,CN=Виктор Вагнер,emailAddress=vitus@wagner.pp.ru')
 124         def test_subjectfields(self):
 125                 c=X509(self.cert1)
 126                 self.assertEqual(c.subject[Oid("C")],"RU")
 127                 self.assertEqual(c.subject[Oid("L")],u'\u041c\u043e\u0441\u043a\u0432\u0430')
 128         def test_notBefore(self):
 129                 c=X509(self.cert1)
 130                 self.assertEqual(c.startDate,datetime.datetime(2014,10,26,19,07,17,0,utc))
 131         def test_notAfter(self):
 132                 c=X509(self.cert1)
 133                 self.assertEqual(c.endDate,datetime.datetime(2024,10,23,19,7,17,0,utc))
 134         def test_namecomp(self):
 135                 c=X509(self.cert1)
 136                 ca=X509(self.ca_cert)
 137                 self.assertEqual(c.issuer,ca.subject)
 138                 self.assertNotEqual(c.subject,c.issuer)
 139                 self.assertEqual(ca.issuer,ca.subject)
 140         def test_serial(self):
 141                 c=X509(self.cert1)
 142                 self.assertEqual(c.serial,0xDF448E69DADC927CL)
 143         def test_version(self):
 144                 c=X509(self.cert1)
 145                 self.assertEqual(c.version,3)
 146         def test_ca_cert(self):
 147                 ca=X509(self.ca_cert)
 148                 self.assertTrue(ca.check_ca())
 149                 notca=X509(self.cert1)
 150                 self.assertFalse(notca.check_ca())
 151         def test_extension_count(self):
 152                 cert=X509(self.cert1)
 153                 self.assertTrue(len(cert.extensions),4)
 154                 ca_cert=X509(self.ca_cert)
 155                 self.assertEqual(len(ca_cert.extensions),3)
 156         def test_extension_outofrange(self):
 157                 cert=X509(self.cert1)
 158                 with self.assertRaises(IndexError):
 159                         cert.extensions[4]
 160                 with self.assertRaises(IndexError):
 161                         cert.extensions[-1]
 162         def test_extension_oid(self):
 163                 cert=X509(self.cert1)
 164                 ext=cert.extensions[0]
 165                 ext_id=ext.oid
 166                 self.assertTrue(isinstance(ext_id,Oid))
 167                 self.assertEqual(ext_id,Oid('basicConstraints'))
 168         def text_extension_text(self):
 169                 cert=X509(self.cert1)
 170                 ext=cert.extensions[0]
 171                 self.assertEqual(str(ext),'CA:FALSE')
 172         def test_extenson_find(self):
 173                 cert=X509(self.cert1)
 174                 exts=cert.extensions.find(Oid('subjectAltName'))
 175                 self.assertEqual(len(exts),1)
 176                 self.assertEqual(exts[0].oid,Oid('subjectAltName'))
 177         def test_extenson_critical(self):
 178                 cert=X509(self.digicert_cert)
 179                 crit_exts=cert.extensions.find_critical()
 180                 self.assertEqual(len(crit_exts),2)
 181                 other_exts=cert.extensions.find_critical(False)
 182                 self.assertEqual(len(crit_exts)+len(other_exts),len(cert.extensions))
 183                 self.assertEqual(crit_exts[0].critical,True)
 184                 self.assertEqual(other_exts[0].critical,False)
 185         def test_verify_by_key(self):
 186                 ca=X509(self.ca_cert)
 187                 pubkey=ca.pubkey
 188                 self.assertTrue(ca.verify(key=pubkey))
 189                 c=X509(self.cert1)
 190                 pk2=c.pubkey
 191                 self.assertFalse(c.verify(key=pk2))
 192                 self.assertTrue(c.verify(key=pubkey))
 193         def test_default_filestore(self):
 194                 store=X509Store(default=True)
 195                 c1=X509(self.cert1)
 196                 # Cert signed by our CA shouldn't be successfully verified
 197                 # by default CA store
 198                 self.assertFalse(c1.verify(store))
 199                 # but cert, downloaded from some commercial CA - should.
 200                 c2=X509(self.digicert_cert)
 201                 self.assertTrue(c2.verify(store))
 202         def test_verify_by_filestore(self):
 203                 trusted=NamedTemporaryFile()
 204                 trusted.write(self.ca_cert)
 205                 trusted.flush()
 206                 goodcert=X509(self.cert1)
 207                 badcert=X509(self.cert1[0:-30]+"GG"+self.cert1[-28:])
 208                 gitcert=X509(self.digicert_cert)
 209                 store=X509Store(file=trusted.name)
 210                 # We should successfuly verify certificate signed by our CA cert
 211                 self.assertTrue(goodcert.verify(store))
 212                 # We should reject corrupted certificate
 213                 self.assertFalse(badcert.verify(store))
 214                 # And if we specify explicitely certificate file, certificate,
 215                 # signed by some commercial CA should be rejected too
 216                 self.assertFalse(gitcert.verify(store))
 217                 trusted.close()
 218         def test_verify_by_dirstore(self):
 219                 pass
 220 if __name__ == '__main__':
 221         unittest.main()