blob: 51da99b3226c17534469dc63c87d9ca19e3e5b14 [file] [log] [blame]
Jean-Paul Calderone8671c852011-03-02 19:26:20 -05001# Copyright (c) Jean-Paul Calderone
2# See LICENSE file for details.
Jean-Paul Calderone8b63d452008-03-21 18:31:12 -04003
Jean-Paul Calderoned8782ad2008-03-04 23:39:59 -05004"""
Jonathan Ballet648875f2011-07-16 14:14:58 +09005Unit tests for :py:mod:`OpenSSL.crypto`.
Jean-Paul Calderoned8782ad2008-03-04 23:39:59 -05006"""
7
Jean-Paul Calderone0b88b6a2009-07-05 12:44:41 -04008from unittest import main
9
Jean-Paul Calderone7b874db2009-08-27 12:32:47 -040010import os, re
Jean-Paul Calderone62ca8da2010-08-11 19:58:08 -040011from subprocess import PIPE, Popen
Rick Dean47262da2009-07-08 16:17:17 -050012from datetime import datetime, timedelta
Jean-Paul Calderoned8782ad2008-03-04 23:39:59 -050013
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -050014from six import binary_type
15
Jean-Paul Calderoned8782ad2008-03-04 23:39:59 -050016from OpenSSL.crypto import TYPE_RSA, TYPE_DSA, Error, PKey, PKeyType
Jean-Paul Calderone78381d22008-03-06 23:35:22 -050017from OpenSSL.crypto import X509, X509Type, X509Name, X509NameType
Jean-Paul Calderonea63714c2013-03-05 17:02:26 -080018from OpenSSL.crypto import X509Store, X509StoreType, X509Req, X509ReqType
Jean-Paul Calderonee7db4b42008-12-31 13:39:24 -050019from OpenSSL.crypto import X509Extension, X509ExtensionType
Rick Dean5b7b6372009-04-01 11:34:06 -050020from OpenSSL.crypto import load_certificate, load_privatekey
Jean-Paul Calderonef17e4212009-04-01 14:21:40 -040021from OpenSSL.crypto import FILETYPE_PEM, FILETYPE_ASN1, FILETYPE_TEXT
Jean-Paul Calderone71919862009-04-01 13:01:19 -040022from OpenSSL.crypto import dump_certificate, load_certificate_request
23from OpenSSL.crypto import dump_certificate_request, dump_privatekey
Jean-Paul Calderonedc138fa2009-06-27 14:32:07 -040024from OpenSSL.crypto import PKCS7Type, load_pkcs7_data
Jean-Paul Calderone9178ee62010-01-25 17:55:30 -050025from OpenSSL.crypto import PKCS12, PKCS12Type, load_pkcs12
Rick Dean536ba022009-07-24 23:57:27 -050026from OpenSSL.crypto import CRL, Revoked, load_crl
Jean-Paul Calderonedc138fa2009-06-27 14:32:07 -040027from OpenSSL.crypto import NetscapeSPKI, NetscapeSPKIType
Jean-Paul Calderonef3cb9d82010-06-22 10:29:33 -040028from OpenSSL.crypto import sign, verify
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -050029from OpenSSL.test.util import TestCase, b
30from OpenSSL._util import native
Jean-Paul Calderoneadd7bf02010-08-22 17:38:30 -040031
Jean-Paul Calderone9da338d2011-05-04 11:40:54 -040032def normalize_certificate_pem(pem):
33 return dump_certificate(FILETYPE_PEM, load_certificate(FILETYPE_PEM, pem))
34
35
36def normalize_privatekey_pem(pem):
37 return dump_privatekey(FILETYPE_PEM, load_privatekey(FILETYPE_PEM, pem))
38
Jean-Paul Calderoneadd7bf02010-08-22 17:38:30 -040039
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -050040GOOD_CIPHER = "blowfish"
41BAD_CIPHER = "zippers"
42
43GOOD_DIGEST = "MD5"
44BAD_DIGEST = "monkeys"
45
Jean-Paul Calderoneadd7bf02010-08-22 17:38:30 -040046root_cert_pem = b("""-----BEGIN CERTIFICATE-----
Rick Dean94e46fd2009-07-18 14:51:24 -050047MIIC7TCCAlagAwIBAgIIPQzE4MbeufQwDQYJKoZIhvcNAQEFBQAwWDELMAkGA1UE
48BhMCVVMxCzAJBgNVBAgTAklMMRAwDgYDVQQHEwdDaGljYWdvMRAwDgYDVQQKEwdU
49ZXN0aW5nMRgwFgYDVQQDEw9UZXN0aW5nIFJvb3QgQ0EwIhgPMjAwOTAzMjUxMjM2
50NThaGA8yMDE3MDYxMTEyMzY1OFowWDELMAkGA1UEBhMCVVMxCzAJBgNVBAgTAklM
51MRAwDgYDVQQHEwdDaGljYWdvMRAwDgYDVQQKEwdUZXN0aW5nMRgwFgYDVQQDEw9U
52ZXN0aW5nIFJvb3QgQ0EwgZ8wDQYJKoZIhvcNAQEBBQADgY0AMIGJAoGBAPmaQumL
53urpE527uSEHdL1pqcDRmWzu+98Y6YHzT/J7KWEamyMCNZ6fRW1JCR782UQ8a07fy
542xXsKy4WdKaxyG8CcatwmXvpvRQ44dSANMihHELpANTdyVp6DCysED6wkQFurHlF
551dshEaJw8b/ypDhmbVIo6Ci1xvCJqivbLFnbAgMBAAGjgbswgbgwHQYDVR0OBBYE
56FINVdy1eIfFJDAkk51QJEo3IfgSuMIGIBgNVHSMEgYAwfoAUg1V3LV4h8UkMCSTn
57VAkSjch+BK6hXKRaMFgxCzAJBgNVBAYTAlVTMQswCQYDVQQIEwJJTDEQMA4GA1UE
58BxMHQ2hpY2FnbzEQMA4GA1UEChMHVGVzdGluZzEYMBYGA1UEAxMPVGVzdGluZyBS
59b290IENBggg9DMTgxt659DAMBgNVHRMEBTADAQH/MA0GCSqGSIb3DQEBBQUAA4GB
60AGGCDazMJGoWNBpc03u6+smc95dEead2KlZXBATOdFT1VesY3+nUOqZhEhTGlDMi
61hkgaZnzoIq/Uamidegk4hirsCT/R+6vsKAAxNTcBjUeZjlykCJWy5ojShGftXIKY
62w/njVbKMXrvc83qmTdGl3TAM0fxQIpqgcglFLveEBgzn
63-----END CERTIFICATE-----
Jean-Paul Calderoneadd7bf02010-08-22 17:38:30 -040064""")
Rick Dean94e46fd2009-07-18 14:51:24 -050065
Jean-Paul Calderoneadd7bf02010-08-22 17:38:30 -040066root_key_pem = b("""-----BEGIN RSA PRIVATE KEY-----
Rick Dean94e46fd2009-07-18 14:51:24 -050067MIICXQIBAAKBgQD5mkLpi7q6ROdu7khB3S9aanA0Zls7vvfGOmB80/yeylhGpsjA
68jWen0VtSQke/NlEPGtO38tsV7CsuFnSmschvAnGrcJl76b0UOOHUgDTIoRxC6QDU
693claegwsrBA+sJEBbqx5RdXbIRGicPG/8qQ4Zm1SKOgotcbwiaor2yxZ2wIDAQAB
70AoGBAPCgMpmLxzwDaUmcFbTJUvlLW1hoxNNYSu2jIZm1k/hRAcE60JYwvBkgz3UB
71yMEh0AtLxYe0bFk6EHah11tMUPgscbCq73snJ++8koUw+csk22G65hOs51bVb7Aa
726JBe67oLzdtvgCUFAA2qfrKzWRZzAdhUirQUZgySZk+Xq1pBAkEA/kZG0A6roTSM
73BVnx7LnPfsycKUsTumorpXiylZJjTi9XtmzxhrYN6wgZlDOOwOLgSQhszGpxVoMD
74u3gByT1b2QJBAPtL3mSKdvwRu/+40zaZLwvSJRxaj0mcE4BJOS6Oqs/hS1xRlrNk
75PpQ7WJ4yM6ZOLnXzm2mKyxm50Mv64109FtMCQQDOqS2KkjHaLowTGVxwC0DijMfr
76I9Lf8sSQk32J5VWCySWf5gGTfEnpmUa41gKTMJIbqZZLucNuDcOtzUaeWZlZAkA8
77ttXigLnCqR486JDPTi9ZscoZkZ+w7y6e/hH8t6d5Vjt48JVyfjPIaJY+km58LcN3
786AWSeGAdtRFHVzR7oHjVAkB4hutvxiOeiIVQNBhM6RSI9aBPMI21DoX2JRoxvNW2
79cbvAhow217X9V0dVerEOKxnNYspXRrh36h7k4mQA+sDq
80-----END RSA PRIVATE KEY-----
Jean-Paul Calderoneadd7bf02010-08-22 17:38:30 -040081""")
Rick Dean94e46fd2009-07-18 14:51:24 -050082
Jean-Paul Calderoneadd7bf02010-08-22 17:38:30 -040083server_cert_pem = b("""-----BEGIN CERTIFICATE-----
Rick Dean94e46fd2009-07-18 14:51:24 -050084MIICKDCCAZGgAwIBAgIJAJn/HpR21r/8MA0GCSqGSIb3DQEBBQUAMFgxCzAJBgNV
85BAYTAlVTMQswCQYDVQQIEwJJTDEQMA4GA1UEBxMHQ2hpY2FnbzEQMA4GA1UEChMH
86VGVzdGluZzEYMBYGA1UEAxMPVGVzdGluZyBSb290IENBMCIYDzIwMDkwMzI1MTIz
87NzUzWhgPMjAxNzA2MTExMjM3NTNaMBgxFjAUBgNVBAMTDWxvdmVseSBzZXJ2ZXIw
88gZ8wDQYJKoZIhvcNAQEBBQADgY0AMIGJAoGBAL6m+G653V0tpBC/OKl22VxOi2Cv
89lK4TYu9LHSDP9uDVTe7V5D5Tl6qzFoRRx5pfmnkqT5B+W9byp2NU3FC5hLm5zSAr
90b45meUhjEJ/ifkZgbNUjHdBIGP9MAQUHZa5WKdkGIJvGAvs8UzUqlr4TBWQIB24+
91lJ+Ukk/CRgasrYwdAgMBAAGjNjA0MB0GA1UdDgQWBBS4kC7Ij0W1TZXZqXQFAM2e
92gKEG2DATBgNVHSUEDDAKBggrBgEFBQcDATANBgkqhkiG9w0BAQUFAAOBgQBh30Li
93dJ+NlxIOx5343WqIBka3UbsOb2kxWrbkVCrvRapCMLCASO4FqiKWM+L0VDBprqIp
942mgpFQ6FHpoIENGvJhdEKpptQ5i7KaGhnDNTfdy3x1+h852G99f1iyj0RmbuFcM8
95uzujnS8YXWvM7DM1Ilozk4MzPug8jzFp5uhKCQ==
96-----END CERTIFICATE-----
Jean-Paul Calderoneadd7bf02010-08-22 17:38:30 -040097""")
Rick Dean94e46fd2009-07-18 14:51:24 -050098
Jean-Paul Calderone9da338d2011-05-04 11:40:54 -040099server_key_pem = normalize_privatekey_pem(b("""-----BEGIN RSA PRIVATE KEY-----
Rick Dean94e46fd2009-07-18 14:51:24 -0500100MIICWwIBAAKBgQC+pvhuud1dLaQQvzipdtlcTotgr5SuE2LvSx0gz/bg1U3u1eQ+
101U5eqsxaEUceaX5p5Kk+QflvW8qdjVNxQuYS5uc0gK2+OZnlIYxCf4n5GYGzVIx3Q
102SBj/TAEFB2WuVinZBiCbxgL7PFM1Kpa+EwVkCAduPpSflJJPwkYGrK2MHQIDAQAB
103AoGAbwuZ0AR6JveahBaczjfnSpiFHf+mve2UxoQdpyr6ROJ4zg/PLW5K/KXrC48G
104j6f3tXMrfKHcpEoZrQWUfYBRCUsGD5DCazEhD8zlxEHahIsqpwA0WWssJA2VOLEN
105j6DuV2pCFbw67rfTBkTSo32ahfXxEKev5KswZk0JIzH3ooECQQDgzS9AI89h0gs8
106Dt+1m11Rzqo3vZML7ZIyGApUzVan+a7hbc33nbGRkAXjHaUBJO31it/H6dTO+uwX
107msWwNG5ZAkEA2RyFKs5xR5USTFaKLWCgpH/ydV96KPOpBND7TKQx62snDenFNNbn
108FwwOhpahld+vqhYk+pfuWWUpQciE+Bu7ZQJASjfT4sQv4qbbKK/scePicnDdx9th
1094e1EeB9xwb+tXXXUo/6Bor/AcUNwfiQ6Zt9PZOK9sR3lMZSsP7rMi7kzuQJABie6
1101sXXjFH7nNJvRG4S39cIxq8YRYTy68II/dlB2QzGpKxV/POCxbJ/zu0CU79tuYK7
111NaeNCFfH3aeTrX0LyQJAMBWjWmeKM2G2sCExheeQK0ROnaBC8itCECD4Jsve4nqf
112r50+LF74iLXFwqysVCebPKMOpDWp/qQ1BbJQIPs7/A==
113-----END RSA PRIVATE KEY-----
Jean-Paul Calderone9da338d2011-05-04 11:40:54 -0400114"""))
Rick Dean94e46fd2009-07-18 14:51:24 -0500115
Jean-Paul Calderoneadd7bf02010-08-22 17:38:30 -0400116client_cert_pem = b("""-----BEGIN CERTIFICATE-----
Rick Dean94e46fd2009-07-18 14:51:24 -0500117MIICJjCCAY+gAwIBAgIJAKxpFI5lODkjMA0GCSqGSIb3DQEBBQUAMFgxCzAJBgNV
118BAYTAlVTMQswCQYDVQQIEwJJTDEQMA4GA1UEBxMHQ2hpY2FnbzEQMA4GA1UEChMH
119VGVzdGluZzEYMBYGA1UEAxMPVGVzdGluZyBSb290IENBMCIYDzIwMDkwMzI1MTIz
120ODA1WhgPMjAxNzA2MTExMjM4MDVaMBYxFDASBgNVBAMTC3VnbHkgY2xpZW50MIGf
121MA0GCSqGSIb3DQEBAQUAA4GNADCBiQKBgQDAZh/SRtNm5ntMT4qb6YzEpTroMlq2
122rn+GrRHRiZ+xkCw/CGNhbtPir7/QxaUj26BSmQrHw1bGKEbPsWiW7bdXSespl+xK
123iku4G/KvnnmWdeJHqsiXeUZtqurMELcPQAw9xPHEuhqqUJvvEoMTsnCEqGM+7Dtb
124oCRajYyHfluARQIDAQABozYwNDAdBgNVHQ4EFgQUNQB+qkaOaEVecf1J3TTUtAff
1250fAwEwYDVR0lBAwwCgYIKwYBBQUHAwIwDQYJKoZIhvcNAQEFBQADgYEAyv/Jh7gM
126Q3OHvmsFEEvRI+hsW8y66zK4K5de239Y44iZrFYkt7Q5nBPMEWDj4F2hLYWL/qtI
1279Zdr0U4UDCU9SmmGYh4o7R4TZ5pGFvBYvjhHbkSFYFQXZxKUi+WUxplP6I0wr2KJ
128PSTJCjJOn3xo2NTKRgV1gaoTf2EhL+RG8TQ=
129-----END CERTIFICATE-----
Jean-Paul Calderoneadd7bf02010-08-22 17:38:30 -0400130""")
Rick Dean94e46fd2009-07-18 14:51:24 -0500131
Jean-Paul Calderone22cbe502011-05-04 17:01:43 -0400132client_key_pem = normalize_privatekey_pem(b("""-----BEGIN RSA PRIVATE KEY-----
Rick Dean94e46fd2009-07-18 14:51:24 -0500133MIICXgIBAAKBgQDAZh/SRtNm5ntMT4qb6YzEpTroMlq2rn+GrRHRiZ+xkCw/CGNh
134btPir7/QxaUj26BSmQrHw1bGKEbPsWiW7bdXSespl+xKiku4G/KvnnmWdeJHqsiX
135eUZtqurMELcPQAw9xPHEuhqqUJvvEoMTsnCEqGM+7DtboCRajYyHfluARQIDAQAB
136AoGATkZ+NceY5Glqyl4mD06SdcKfV65814vg2EL7V9t8+/mi9rYL8KztSXGlQWPX
137zuHgtRoMl78yQ4ZJYOBVo+nsx8KZNRCEBlE19bamSbQLCeQMenWnpeYyQUZ908gF
138h6L9qsFVJepgA9RDgAjyDoS5CaWCdCCPCH2lDkdcqC54SVUCQQDseuduc4wi8h4t
139V8AahUn9fn9gYfhoNuM0gdguTA0nPLVWz4hy1yJiWYQe0H7NLNNTmCKiLQaJpAbb
140TC6vE8C7AkEA0Ee8CMJUc20BnGEmxwgWcVuqFWaKCo8jTH1X38FlATUsyR3krjW2
141dL3yDD9NwHxsYP7nTKp/U8MV7U9IBn4y/wJBAJl7H0/BcLeRmuJk7IqJ7b635iYB
142D/9beFUw3MUXmQXZUfyYz39xf6CDZsu1GEdEC5haykeln3Of4M9d/4Kj+FcCQQCY
143si6xwT7GzMDkk/ko684AV3KPc/h6G0yGtFIrMg7J3uExpR/VdH2KgwMkZXisSMvw
144JJEQjOMCVsEJlRk54WWjAkEAzoZNH6UhDdBK5F38rVt/y4SEHgbSfJHIAmPS32Kq
145f6GGcfNpip0Uk7q7udTKuX7Q/buZi/C4YW7u3VKAquv9NA==
146-----END RSA PRIVATE KEY-----
Jean-Paul Calderone22cbe502011-05-04 17:01:43 -0400147"""))
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -0400148
Jean-Paul Calderoneadd7bf02010-08-22 17:38:30 -0400149cleartextCertificatePEM = b("""-----BEGIN CERTIFICATE-----
Jean-Paul Calderone20131f52009-04-01 12:05:45 -0400150MIIC7TCCAlagAwIBAgIIPQzE4MbeufQwDQYJKoZIhvcNAQEFBQAwWDELMAkGA1UE
151BhMCVVMxCzAJBgNVBAgTAklMMRAwDgYDVQQHEwdDaGljYWdvMRAwDgYDVQQKEwdU
152ZXN0aW5nMRgwFgYDVQQDEw9UZXN0aW5nIFJvb3QgQ0EwIhgPMjAwOTAzMjUxMjM2
153NThaGA8yMDE3MDYxMTEyMzY1OFowWDELMAkGA1UEBhMCVVMxCzAJBgNVBAgTAklM
154MRAwDgYDVQQHEwdDaGljYWdvMRAwDgYDVQQKEwdUZXN0aW5nMRgwFgYDVQQDEw9U
155ZXN0aW5nIFJvb3QgQ0EwgZ8wDQYJKoZIhvcNAQEBBQADgY0AMIGJAoGBAPmaQumL
156urpE527uSEHdL1pqcDRmWzu+98Y6YHzT/J7KWEamyMCNZ6fRW1JCR782UQ8a07fy
1572xXsKy4WdKaxyG8CcatwmXvpvRQ44dSANMihHELpANTdyVp6DCysED6wkQFurHlF
1581dshEaJw8b/ypDhmbVIo6Ci1xvCJqivbLFnbAgMBAAGjgbswgbgwHQYDVR0OBBYE
159FINVdy1eIfFJDAkk51QJEo3IfgSuMIGIBgNVHSMEgYAwfoAUg1V3LV4h8UkMCSTn
160VAkSjch+BK6hXKRaMFgxCzAJBgNVBAYTAlVTMQswCQYDVQQIEwJJTDEQMA4GA1UE
161BxMHQ2hpY2FnbzEQMA4GA1UEChMHVGVzdGluZzEYMBYGA1UEAxMPVGVzdGluZyBS
162b290IENBggg9DMTgxt659DAMBgNVHRMEBTADAQH/MA0GCSqGSIb3DQEBBQUAA4GB
163AGGCDazMJGoWNBpc03u6+smc95dEead2KlZXBATOdFT1VesY3+nUOqZhEhTGlDMi
164hkgaZnzoIq/Uamidegk4hirsCT/R+6vsKAAxNTcBjUeZjlykCJWy5ojShGftXIKY
165w/njVbKMXrvc83qmTdGl3TAM0fxQIpqgcglFLveEBgzn
166-----END CERTIFICATE-----
Jean-Paul Calderoneadd7bf02010-08-22 17:38:30 -0400167""")
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -0400168
Jean-Paul Calderoned50d2042011-05-04 17:00:49 -0400169cleartextPrivateKeyPEM = normalize_privatekey_pem(b("""\
170-----BEGIN RSA PRIVATE KEY-----
Jean-Paul Calderone20131f52009-04-01 12:05:45 -0400171MIICXQIBAAKBgQD5mkLpi7q6ROdu7khB3S9aanA0Zls7vvfGOmB80/yeylhGpsjA
172jWen0VtSQke/NlEPGtO38tsV7CsuFnSmschvAnGrcJl76b0UOOHUgDTIoRxC6QDU
1733claegwsrBA+sJEBbqx5RdXbIRGicPG/8qQ4Zm1SKOgotcbwiaor2yxZ2wIDAQAB
174AoGBAPCgMpmLxzwDaUmcFbTJUvlLW1hoxNNYSu2jIZm1k/hRAcE60JYwvBkgz3UB
175yMEh0AtLxYe0bFk6EHah11tMUPgscbCq73snJ++8koUw+csk22G65hOs51bVb7Aa
1766JBe67oLzdtvgCUFAA2qfrKzWRZzAdhUirQUZgySZk+Xq1pBAkEA/kZG0A6roTSM
177BVnx7LnPfsycKUsTumorpXiylZJjTi9XtmzxhrYN6wgZlDOOwOLgSQhszGpxVoMD
178u3gByT1b2QJBAPtL3mSKdvwRu/+40zaZLwvSJRxaj0mcE4BJOS6Oqs/hS1xRlrNk
179PpQ7WJ4yM6ZOLnXzm2mKyxm50Mv64109FtMCQQDOqS2KkjHaLowTGVxwC0DijMfr
180I9Lf8sSQk32J5VWCySWf5gGTfEnpmUa41gKTMJIbqZZLucNuDcOtzUaeWZlZAkA8
181ttXigLnCqR486JDPTi9ZscoZkZ+w7y6e/hH8t6d5Vjt48JVyfjPIaJY+km58LcN3
1826AWSeGAdtRFHVzR7oHjVAkB4hutvxiOeiIVQNBhM6RSI9aBPMI21DoX2JRoxvNW2
183cbvAhow217X9V0dVerEOKxnNYspXRrh36h7k4mQA+sDq
184-----END RSA PRIVATE KEY-----
Jean-Paul Calderoned50d2042011-05-04 17:00:49 -0400185"""))
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -0400186
Jean-Paul Calderoneadd7bf02010-08-22 17:38:30 -0400187cleartextCertificateRequestPEM = b("""-----BEGIN CERTIFICATE REQUEST-----
188MIIBnjCCAQcCAQAwXjELMAkGA1UEBhMCVVMxCzAJBgNVBAgTAklMMRAwDgYDVQQH
189EwdDaGljYWdvMRcwFQYDVQQKEw5NeSBDb21wYW55IEx0ZDEXMBUGA1UEAxMORnJl
190ZGVyaWNrIERlYW4wgZ8wDQYJKoZIhvcNAQEBBQADgY0AMIGJAoGBANp6Y17WzKSw
191BsUWkXdqg6tnXy8H8hA1msCMWpc+/2KJ4mbv5NyD6UD+/SqagQqulPbF/DFea9nA
192E0zhmHJELcM8gUTIlXv/cgDWnmK4xj8YkjVUiCdqKRAKeuzLG1pGmwwF5lGeJpXN
193xQn5ecR0UYSOWj6TTGXB9VyUMQzCClcBAgMBAAGgADANBgkqhkiG9w0BAQUFAAOB
194gQAAJGuF/R/GGbeC7FbFW+aJgr9ee0Xbl6nlhu7pTe67k+iiKT2dsl2ti68MVTnu
195Vrb3HUNqOkiwsJf6kCtq5oPn3QVYzTa76Dt2y3Rtzv6boRSlmlfrgS92GNma8JfR
196oICQk3nAudi6zl1Dix3BCv1pUp5KMtGn3MeDEi6QFGy2rA==
197-----END CERTIFICATE REQUEST-----
198""")
Rick Dean5b7b6372009-04-01 11:34:06 -0500199
Jean-Paul Calderoneadd7bf02010-08-22 17:38:30 -0400200encryptedPrivateKeyPEM = b("""-----BEGIN RSA PRIVATE KEY-----
Jean-Paul Calderone20131f52009-04-01 12:05:45 -0400201Proc-Type: 4,ENCRYPTED
202DEK-Info: DES-EDE3-CBC,9573604A18579E9E
Jean-Paul Calderone5ef86512008-04-26 19:06:28 -0400203
Jean-Paul Calderone20131f52009-04-01 12:05:45 -0400204SHOho56WxDkT0ht10UTeKc0F5u8cqIa01kzFAmETw0MAs8ezYtK15NPdCXUm3X/2
205a17G7LSF5bkxOgZ7vpXyMzun/owrj7CzvLxyncyEFZWvtvzaAhPhvTJtTIB3kf8B
2068+qRcpTGK7NgXEgYBW5bj1y4qZkD4zCL9o9NQzsKI3Ie8i0239jsDOWR38AxjXBH
207mGwAQ4Z6ZN5dnmM4fhMIWsmFf19sNyAML4gHenQCHhmXbjXeVq47aC2ProInJbrm
208+00TcisbAQ40V9aehVbcDKtS4ZbMVDwncAjpXpcncC54G76N6j7F7wL7L/FuXa3A
209fvSVy9n2VfF/pJ3kYSflLHH2G/DFxjF7dl0GxhKPxJjp3IJi9VtuvmN9R2jZWLQF
210tfC8dXgy/P9CfFQhlinqBTEwgH0oZ/d4k4NVFDSdEMaSdmBAjlHpc+Vfdty3HVnV
211rKXj//wslsFNm9kIwJGIgKUa/n2jsOiydrsk1mgH7SmNCb3YHgZhbbnq0qLat/HC
212gHDt3FHpNQ31QzzL3yrenFB2L9osIsnRsDTPFNi4RX4SpDgNroxOQmyzCCV6H+d4
213o1mcnNiZSdxLZxVKccq0AfRpHqpPAFnJcQHP6xyT9MZp6fBa0XkxDnt9kNU8H3Qw
2147SJWZ69VXjBUzMlQViLuaWMgTnL+ZVyFZf9hTF7U/ef4HMLMAVNdiaGG+G+AjCV/
215MbzjS007Oe4qqBnCWaFPSnJX6uLApeTbqAxAeyCql56ULW5x6vDMNC3dwjvS/CEh
21611n8RkgFIQA0AhuKSIg3CbuartRsJnWOLwgLTzsrKYL4yRog1RJrtw==
217-----END RSA PRIVATE KEY-----
Jean-Paul Calderoneadd7bf02010-08-22 17:38:30 -0400218""")
Jean-Paul Calderone55ec1712009-05-13 14:14:30 -0400219
Jean-Paul Calderoneadd7bf02010-08-22 17:38:30 -0400220encryptedPrivateKeyPEMPassphrase = b("foobar")
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -0400221
Jean-Paul Calderonedc138fa2009-06-27 14:32:07 -0400222# Some PKCS#7 stuff. Generated with the openssl command line:
223#
224# openssl crl2pkcs7 -inform pem -outform pem -certfile s.pem -nocrl
225#
226# with a certificate and key (but the key should be irrelevant) in s.pem
Jean-Paul Calderoneadd7bf02010-08-22 17:38:30 -0400227pkcs7Data = b("""\
Jean-Paul Calderonedc138fa2009-06-27 14:32:07 -0400228-----BEGIN PKCS7-----
229MIIDNwYJKoZIhvcNAQcCoIIDKDCCAyQCAQExADALBgkqhkiG9w0BBwGgggMKMIID
230BjCCAm+gAwIBAgIBATANBgkqhkiG9w0BAQQFADB7MQswCQYDVQQGEwJTRzERMA8G
231A1UEChMITTJDcnlwdG8xFDASBgNVBAsTC00yQ3J5cHRvIENBMSQwIgYDVQQDExtN
232MkNyeXB0byBDZXJ0aWZpY2F0ZSBNYXN0ZXIxHTAbBgkqhkiG9w0BCQEWDm5ncHNA
233cG9zdDEuY29tMB4XDTAwMDkxMDA5NTEzMFoXDTAyMDkxMDA5NTEzMFowUzELMAkG
234A1UEBhMCU0cxETAPBgNVBAoTCE0yQ3J5cHRvMRIwEAYDVQQDEwlsb2NhbGhvc3Qx
235HTAbBgkqhkiG9w0BCQEWDm5ncHNAcG9zdDEuY29tMFwwDQYJKoZIhvcNAQEBBQAD
236SwAwSAJBAKy+e3dulvXzV7zoTZWc5TzgApr8DmeQHTYC8ydfzH7EECe4R1Xh5kwI
237zOuuFfn178FBiS84gngaNcrFi0Z5fAkCAwEAAaOCAQQwggEAMAkGA1UdEwQCMAAw
238LAYJYIZIAYb4QgENBB8WHU9wZW5TU0wgR2VuZXJhdGVkIENlcnRpZmljYXRlMB0G
239A1UdDgQWBBTPhIKSvnsmYsBVNWjj0m3M2z0qVTCBpQYDVR0jBIGdMIGagBT7hyNp
24065w6kxXlxb8pUU/+7Sg4AaF/pH0wezELMAkGA1UEBhMCU0cxETAPBgNVBAoTCE0y
241Q3J5cHRvMRQwEgYDVQQLEwtNMkNyeXB0byBDQTEkMCIGA1UEAxMbTTJDcnlwdG8g
242Q2VydGlmaWNhdGUgTWFzdGVyMR0wGwYJKoZIhvcNAQkBFg5uZ3BzQHBvc3QxLmNv
243bYIBADANBgkqhkiG9w0BAQQFAAOBgQA7/CqT6PoHycTdhEStWNZde7M/2Yc6BoJu
244VwnW8YxGO8Sn6UJ4FeffZNcYZddSDKosw8LtPOeWoK3JINjAk5jiPQ2cww++7QGG
245/g5NDjxFZNDJP1dGiLAxPW6JXwov4v0FmdzfLOZ01jDcgQQZqEpYlgpuI5JEWUQ9
246Ho4EzbYCOaEAMQA=
247-----END PKCS7-----
Jean-Paul Calderoneadd7bf02010-08-22 17:38:30 -0400248""")
Jean-Paul Calderonedc138fa2009-06-27 14:32:07 -0400249
Jean-Paul Calderoneadd7bf02010-08-22 17:38:30 -0400250crlData = b("""\
Jean-Paul Calderone3eb5cc72010-01-30 15:24:40 -0500251-----BEGIN X509 CRL-----
252MIIBWzCBxTANBgkqhkiG9w0BAQQFADBYMQswCQYDVQQGEwJVUzELMAkGA1UECBMC
253SUwxEDAOBgNVBAcTB0NoaWNhZ28xEDAOBgNVBAoTB1Rlc3RpbmcxGDAWBgNVBAMT
254D1Rlc3RpbmcgUm9vdCBDQRcNMDkwNzI2MDQzNDU2WhcNMTIwOTI3MDI0MTUyWjA8
255MBUCAgOrGA8yMDA5MDcyNTIzMzQ1NlowIwICAQAYDzIwMDkwNzI1MjMzNDU2WjAM
256MAoGA1UdFQQDCgEEMA0GCSqGSIb3DQEBBAUAA4GBAEBt7xTs2htdD3d4ErrcGAw1
2574dKcVnIWTutoI7xxen26Wwvh8VCsT7i/UeP+rBl9rC/kfjWjzQk3/zleaarGTpBT
2580yp4HXRFFoRhhSE/hP+eteaPXRgrsNRLHe9ZDd69wmh7J1wMDb0m81RG7kqcbsid
259vrzEeLDRiiPl92dyyWmu
260-----END X509 CRL-----
Jean-Paul Calderoneadd7bf02010-08-22 17:38:30 -0400261""")
Jean-Paul Calderonee890db32010-08-22 16:55:15 -0400262
Jean-Paul Calderone55ec1712009-05-13 14:14:30 -0400263
264# A broken RSA private key which can be used to test the error path through
265# PKey.check.
266inconsistentPrivateKeyPEM = b("""-----BEGIN RSA PRIVATE KEY-----
267MIIBPAIBAAJBAKy+e3dulvXzV7zoTZWc5TzgApr8DmeQHTYC8ydfzH7EECe4R1Xh
2685kwIzOuuFfn178FBiS84gngaNcrFi0Z5fAkCAwEaAQJBAIqm/bz4NA1H++Vx5Ewx
269OcKp3w19QSaZAwlGRtsUxrP7436QjnREM3Bm8ygU11BjkPVmtrKm6AayQfCHqJoT
270zIECIQDW0BoMoL0HOYM/mrTLhaykYAVqgIeJsPjvkEhTFXWBuQIhAM3deFAvWNu4
271nklUQ37XsCT2c9tmNt1LAT+slG2JOTTRAiAuXDtC/m3NYVwyHfFm+zKHRzHkClk2
272HjubeEgjpj32AQIhAJqMGTaZVOwevTXvvHwNeH+vRWsAYU/gbx+OQB+7VOcBAiEA
273oolb6NMg/R3enNPvS1O4UU1H8wpaF77L4yiSWlE0p4w=
274-----END RSA PRIVATE KEY-----
275""")
276
Jean-Paul Calderoneff83cdd2013-08-12 18:05:51 -0400277# certificate with NULL bytes in subjectAltName and common name
278
Jean-Paul Calderone4bf75c62013-08-23 15:39:53 -0400279nulbyteSubjectAltNamePEM = b("""-----BEGIN CERTIFICATE-----
Jean-Paul Calderoneff83cdd2013-08-12 18:05:51 -0400280MIIE2DCCA8CgAwIBAgIBADANBgkqhkiG9w0BAQUFADCBxTELMAkGA1UEBhMCVVMx
281DzANBgNVBAgMBk9yZWdvbjESMBAGA1UEBwwJQmVhdmVydG9uMSMwIQYDVQQKDBpQ
282eXRob24gU29mdHdhcmUgRm91bmRhdGlvbjEgMB4GA1UECwwXUHl0aG9uIENvcmUg
283RGV2ZWxvcG1lbnQxJDAiBgNVBAMMG251bGwucHl0aG9uLm9yZwBleGFtcGxlLm9y
284ZzEkMCIGCSqGSIb3DQEJARYVcHl0aG9uLWRldkBweXRob24ub3JnMB4XDTEzMDgw
285NzEzMTE1MloXDTEzMDgwNzEzMTI1MlowgcUxCzAJBgNVBAYTAlVTMQ8wDQYDVQQI
286DAZPcmVnb24xEjAQBgNVBAcMCUJlYXZlcnRvbjEjMCEGA1UECgwaUHl0aG9uIFNv
287ZnR3YXJlIEZvdW5kYXRpb24xIDAeBgNVBAsMF1B5dGhvbiBDb3JlIERldmVsb3Bt
288ZW50MSQwIgYDVQQDDBtudWxsLnB5dGhvbi5vcmcAZXhhbXBsZS5vcmcxJDAiBgkq
289hkiG9w0BCQEWFXB5dGhvbi1kZXZAcHl0aG9uLm9yZzCCASIwDQYJKoZIhvcNAQEB
290BQADggEPADCCAQoCggEBALXq7cn7Rn1vO3aA3TrzA5QLp6bb7B3f/yN0CJ2XFj+j
291pHs+Gw6WWSUDpybiiKnPec33BFawq3kyblnBMjBU61ioy5HwQqVkJ8vUVjGIUq3P
292vX/wBmQfzCe4o4uM89gpHyUL9UYGG8oCRa17dgqcv7u5rg0Wq2B1rgY+nHwx3JIv
293KRrgSwyRkGzpN8WQ1yrXlxWjgI9de0mPVDDUlywcWze1q2kwaEPTM3hLAmD1PESA
294oY/n8A/RXoeeRs9i/Pm/DGUS8ZPINXk/yOzsR/XvvkTVroIeLZqfmFpnZeF0cHzL
29508LODkVJJ9zjLdT7SA4vnne4FEbAxDbKAq5qkYzaL4UCAwEAAaOB0DCBzTAMBgNV
296HRMBAf8EAjAAMB0GA1UdDgQWBBSIWlXAUv9hzVKjNQ/qWpwkOCL3XDALBgNVHQ8E
297BAMCBeAwgZAGA1UdEQSBiDCBhYIeYWx0bnVsbC5weXRob24ub3JnAGV4YW1wbGUu
298Y29tgSBudWxsQHB5dGhvbi5vcmcAdXNlckBleGFtcGxlLm9yZ4YpaHR0cDovL251
299bGwucHl0aG9uLm9yZwBodHRwOi8vZXhhbXBsZS5vcmeHBMAAAgGHECABDbgAAAAA
300AAAAAAAAAAEwDQYJKoZIhvcNAQEFBQADggEBAKxPRe99SaghcI6IWT7UNkJw9aO9
301i9eo0Fj2MUqxpKbdb9noRDy2CnHWf7EIYZ1gznXPdwzSN4YCjV5d+Q9xtBaowT0j
302HPERs1ZuytCNNJTmhyqZ8q6uzMLoht4IqH/FBfpvgaeC5tBTnTT0rD5A/olXeimk
303kX4LxlEx5RAvpGB2zZVRGr6LobD9rVK91xuHYNIxxxfEGE8tCCWjp0+3ksri9SXx
304VHWBnbM9YaL32u3hxm8sYB/Yb8WSBavJCWJJqRStVRHM1koZlJmXNx2BX4vPo6iW
305RFEIPQsFZRLrtnCAiEhyT8bC2s/Njlu6ly9gtJZWSV46Q3ZjBL4q9sHKqZQ=
306-----END CERTIFICATE-----""")
307
Jean-Paul Calderone55ec1712009-05-13 14:14:30 -0400308
Jean-Paul Calderone18808652009-07-05 12:54:05 -0400309class X509ExtTests(TestCase):
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400310 """
Jonathan Ballet648875f2011-07-16 14:14:58 +0900311 Tests for :py:class:`OpenSSL.crypto.X509Extension`.
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400312 """
313
314 def setUp(self):
315 """
316 Create a new private key and start a certificate request (for a test
317 method to finish in one way or another).
318 """
Jean-Paul Calderoneef9a3dc2013-03-02 16:33:32 -0800319 super(X509ExtTests, self).setUp()
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400320 # Basic setup stuff to generate a certificate
321 self.pkey = PKey()
322 self.pkey.generate_key(TYPE_RSA, 384)
323 self.req = X509Req()
324 self.req.set_pubkey(self.pkey)
325 # Authority good you have.
326 self.req.get_subject().commonName = "Yoda root CA"
327 self.x509 = X509()
328 self.subject = self.x509.get_subject()
329 self.subject.commonName = self.req.get_subject().commonName
330 self.x509.set_issuer(self.subject)
331 self.x509.set_pubkey(self.pkey)
Jean-Paul Calderone40dd0992010-08-22 17:52:07 -0400332 now = b(datetime.now().strftime("%Y%m%d%H%M%SZ"))
333 expire = b((datetime.now() + timedelta(days=100)).strftime("%Y%m%d%H%M%SZ"))
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400334 self.x509.set_notBefore(now)
335 self.x509.set_notAfter(expire)
336
337
Jean-Paul Calderoneef9a3dc2013-03-02 16:33:32 -0800338 def tearDown(self):
339 """
340 Forget all of the pyOpenSSL objects so they can be garbage collected,
341 their memory released, and not interfere with the leak detection code.
342 """
343 self.pkey = self.req = self.x509 = self.subject = None
344 super(X509ExtTests, self).tearDown()
345
346
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -0400347 def test_str(self):
348 """
Jonathan Ballet648875f2011-07-16 14:14:58 +0900349 The string representation of :py:class:`X509Extension` instances as returned by
350 :py:data:`str` includes stuff.
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -0400351 """
352 # This isn't necessarily the best string representation. Perhaps it
353 # will be changed/improved in the future.
354 self.assertEquals(
Jean-Paul Calderone40dd0992010-08-22 17:52:07 -0400355 str(X509Extension(b('basicConstraints'), True, b('CA:false'))),
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -0400356 'CA:FALSE')
357
358
Jean-Paul Calderone68649052009-07-17 21:14:27 -0400359 def test_type(self):
360 """
Jonathan Ballet648875f2011-07-16 14:14:58 +0900361 :py:class:`X509Extension` and :py:class:`X509ExtensionType` refer to the same type object
Jean-Paul Calderone68649052009-07-17 21:14:27 -0400362 and can be used to create instances of that type.
363 """
364 self.assertIdentical(X509Extension, X509ExtensionType)
365 self.assertConsistentType(
Jean-Paul Calderone40dd0992010-08-22 17:52:07 -0400366 X509Extension,
367 'X509Extension', b('basicConstraints'), True, b('CA:true'))
Jean-Paul Calderone68649052009-07-17 21:14:27 -0400368
369
Jean-Paul Calderonee7db4b42008-12-31 13:39:24 -0500370 def test_construction(self):
371 """
Jonathan Ballet648875f2011-07-16 14:14:58 +0900372 :py:class:`X509Extension` accepts an extension type name, a critical flag,
373 and an extension value and returns an :py:class:`X509ExtensionType` instance.
Jean-Paul Calderonee7db4b42008-12-31 13:39:24 -0500374 """
Jean-Paul Calderone40dd0992010-08-22 17:52:07 -0400375 basic = X509Extension(b('basicConstraints'), True, b('CA:true'))
Jean-Paul Calderonee7db4b42008-12-31 13:39:24 -0500376 self.assertTrue(
377 isinstance(basic, X509ExtensionType),
378 "%r is of type %r, should be %r" % (
379 basic, type(basic), X509ExtensionType))
380
Jean-Paul Calderone40dd0992010-08-22 17:52:07 -0400381 comment = X509Extension(
382 b('nsComment'), False, b('pyOpenSSL unit test'))
Jean-Paul Calderonee7db4b42008-12-31 13:39:24 -0500383 self.assertTrue(
384 isinstance(comment, X509ExtensionType),
385 "%r is of type %r, should be %r" % (
386 comment, type(comment), X509ExtensionType))
387
388
Jean-Paul Calderone391585f2008-12-31 14:36:31 -0500389 def test_invalid_extension(self):
390 """
Jonathan Ballet648875f2011-07-16 14:14:58 +0900391 :py:class:`X509Extension` raises something if it is passed a bad extension
Jean-Paul Calderone391585f2008-12-31 14:36:31 -0500392 name or value.
393 """
394 self.assertRaises(
Jean-Paul Calderone40dd0992010-08-22 17:52:07 -0400395 Error, X509Extension, b('thisIsMadeUp'), False, b('hi'))
Jean-Paul Calderone391585f2008-12-31 14:36:31 -0500396 self.assertRaises(
Jean-Paul Calderone40dd0992010-08-22 17:52:07 -0400397 Error, X509Extension, b('basicConstraints'), False, b('blah blah'))
Jean-Paul Calderone391585f2008-12-31 14:36:31 -0500398
Jean-Paul Calderone2ee1e7c2008-12-31 14:58:38 -0500399 # Exercise a weird one (an extension which uses the r2i method). This
400 # exercises the codepath that requires a non-NULL ctx to be passed to
401 # X509V3_EXT_nconf. It can't work now because we provide no
402 # configuration database. It might be made to work in the future.
403 self.assertRaises(
Jean-Paul Calderone40dd0992010-08-22 17:52:07 -0400404 Error, X509Extension, b('proxyCertInfo'), True,
405 b('language:id-ppl-anyLanguage,pathlen:1,policy:text:AB'))
Jean-Paul Calderone2ee1e7c2008-12-31 14:58:38 -0500406
Jean-Paul Calderone391585f2008-12-31 14:36:31 -0500407
Jean-Paul Calderonee7db4b42008-12-31 13:39:24 -0500408 def test_get_critical(self):
409 """
Jonathan Ballet648875f2011-07-16 14:14:58 +0900410 :py:meth:`X509ExtensionType.get_critical` returns the value of the
Jean-Paul Calderonee7db4b42008-12-31 13:39:24 -0500411 extension's critical flag.
412 """
Jean-Paul Calderone40dd0992010-08-22 17:52:07 -0400413 ext = X509Extension(b('basicConstraints'), True, b('CA:true'))
Jean-Paul Calderonee7db4b42008-12-31 13:39:24 -0500414 self.assertTrue(ext.get_critical())
Jean-Paul Calderone40dd0992010-08-22 17:52:07 -0400415 ext = X509Extension(b('basicConstraints'), False, b('CA:true'))
Jean-Paul Calderonee7db4b42008-12-31 13:39:24 -0500416 self.assertFalse(ext.get_critical())
417
Jean-Paul Calderone7535dab2008-03-06 18:53:11 -0500418
Jean-Paul Calderonef8c5fab2008-12-31 15:53:48 -0500419 def test_get_short_name(self):
420 """
Jonathan Ballet648875f2011-07-16 14:14:58 +0900421 :py:meth:`X509ExtensionType.get_short_name` returns a string giving the short
Jean-Paul Calderonef8c5fab2008-12-31 15:53:48 -0500422 type name of the extension.
423 """
Jean-Paul Calderone40dd0992010-08-22 17:52:07 -0400424 ext = X509Extension(b('basicConstraints'), True, b('CA:true'))
425 self.assertEqual(ext.get_short_name(), b('basicConstraints'))
426 ext = X509Extension(b('nsComment'), True, b('foo bar'))
427 self.assertEqual(ext.get_short_name(), b('nsComment'))
Jean-Paul Calderonef8c5fab2008-12-31 15:53:48 -0500428
429
Jean-Paul Calderone5a9e4612011-04-01 18:27:45 -0400430 def test_get_data(self):
431 """
Jonathan Ballet648875f2011-07-16 14:14:58 +0900432 :py:meth:`X509Extension.get_data` returns a string giving the data of the
Jean-Paul Calderone5a9e4612011-04-01 18:27:45 -0400433 extension.
434 """
435 ext = X509Extension(b('basicConstraints'), True, b('CA:true'))
436 # Expect to get back the DER encoded form of CA:true.
437 self.assertEqual(ext.get_data(), b('0\x03\x01\x01\xff'))
438
439
440 def test_get_data_wrong_args(self):
441 """
Jonathan Ballet648875f2011-07-16 14:14:58 +0900442 :py:meth:`X509Extension.get_data` raises :py:exc:`TypeError` if passed any arguments.
Jean-Paul Calderone5a9e4612011-04-01 18:27:45 -0400443 """
444 ext = X509Extension(b('basicConstraints'), True, b('CA:true'))
445 self.assertRaises(TypeError, ext.get_data, None)
446 self.assertRaises(TypeError, ext.get_data, "foo")
447 self.assertRaises(TypeError, ext.get_data, 7)
448
449
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400450 def test_unused_subject(self):
Rick Dean47262da2009-07-08 16:17:17 -0500451 """
Jonathan Ballet648875f2011-07-16 14:14:58 +0900452 The :py:data:`subject` parameter to :py:class:`X509Extension` may be provided for an
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400453 extension which does not use it and is ignored in this case.
Rick Dean47262da2009-07-08 16:17:17 -0500454 """
Jean-Paul Calderone40dd0992010-08-22 17:52:07 -0400455 ext1 = X509Extension(
456 b('basicConstraints'), False, b('CA:TRUE'), subject=self.x509)
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400457 self.x509.add_extensions([ext1])
458 self.x509.sign(self.pkey, 'sha1')
459 # This is a little lame. Can we think of a better way?
460 text = dump_certificate(FILETYPE_TEXT, self.x509)
Jean-Paul Calderone40dd0992010-08-22 17:52:07 -0400461 self.assertTrue(b('X509v3 Basic Constraints:') in text)
462 self.assertTrue(b('CA:TRUE') in text)
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400463
464
465 def test_subject(self):
466 """
Jonathan Ballet648875f2011-07-16 14:14:58 +0900467 If an extension requires a subject, the :py:data:`subject` parameter to
468 :py:class:`X509Extension` provides its value.
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400469 """
Jean-Paul Calderone40dd0992010-08-22 17:52:07 -0400470 ext3 = X509Extension(
471 b('subjectKeyIdentifier'), False, b('hash'), subject=self.x509)
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400472 self.x509.add_extensions([ext3])
473 self.x509.sign(self.pkey, 'sha1')
474 text = dump_certificate(FILETYPE_TEXT, self.x509)
Jean-Paul Calderone40dd0992010-08-22 17:52:07 -0400475 self.assertTrue(b('X509v3 Subject Key Identifier:') in text)
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400476
477
478 def test_missing_subject(self):
479 """
Jonathan Ballet648875f2011-07-16 14:14:58 +0900480 If an extension requires a subject and the :py:data:`subject` parameter is
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400481 given no value, something happens.
482 """
483 self.assertRaises(
Jean-Paul Calderone40dd0992010-08-22 17:52:07 -0400484 Error, X509Extension, b('subjectKeyIdentifier'), False, b('hash'))
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400485
486
487 def test_invalid_subject(self):
488 """
Jonathan Ballet648875f2011-07-16 14:14:58 +0900489 If the :py:data:`subject` parameter is given a value which is not an
490 :py:class:`X509` instance, :py:exc:`TypeError` is raised.
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400491 """
492 for badObj in [True, object(), "hello", [], self]:
493 self.assertRaises(
494 TypeError,
495 X509Extension,
496 'basicConstraints', False, 'CA:TRUE', subject=badObj)
497
498
499 def test_unused_issuer(self):
500 """
Jonathan Ballet648875f2011-07-16 14:14:58 +0900501 The :py:data:`issuer` parameter to :py:class:`X509Extension` may be provided for an
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400502 extension which does not use it and is ignored in this case.
503 """
Jean-Paul Calderone40dd0992010-08-22 17:52:07 -0400504 ext1 = X509Extension(
505 b('basicConstraints'), False, b('CA:TRUE'), issuer=self.x509)
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400506 self.x509.add_extensions([ext1])
507 self.x509.sign(self.pkey, 'sha1')
508 text = dump_certificate(FILETYPE_TEXT, self.x509)
Jean-Paul Calderone40dd0992010-08-22 17:52:07 -0400509 self.assertTrue(b('X509v3 Basic Constraints:') in text)
510 self.assertTrue(b('CA:TRUE') in text)
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400511
512
513 def test_issuer(self):
514 """
Jonathan Ballet648875f2011-07-16 14:14:58 +0900515 If an extension requires a issuer, the :py:data:`issuer` parameter to
516 :py:class:`X509Extension` provides its value.
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400517 """
518 ext2 = X509Extension(
Jean-Paul Calderone40dd0992010-08-22 17:52:07 -0400519 b('authorityKeyIdentifier'), False, b('issuer:always'),
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400520 issuer=self.x509)
521 self.x509.add_extensions([ext2])
522 self.x509.sign(self.pkey, 'sha1')
523 text = dump_certificate(FILETYPE_TEXT, self.x509)
Jean-Paul Calderone40dd0992010-08-22 17:52:07 -0400524 self.assertTrue(b('X509v3 Authority Key Identifier:') in text)
525 self.assertTrue(b('DirName:/CN=Yoda root CA') in text)
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400526
527
528 def test_missing_issuer(self):
529 """
Jonathan Ballet648875f2011-07-16 14:14:58 +0900530 If an extension requires an issue and the :py:data:`issuer` parameter is given
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400531 no value, something happens.
532 """
533 self.assertRaises(
534 Error,
535 X509Extension,
Jean-Paul Calderone40dd0992010-08-22 17:52:07 -0400536 b('authorityKeyIdentifier'), False,
537 b('keyid:always,issuer:always'))
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400538
539
540 def test_invalid_issuer(self):
541 """
Jonathan Ballet648875f2011-07-16 14:14:58 +0900542 If the :py:data:`issuer` parameter is given a value which is not an
543 :py:class:`X509` instance, :py:exc:`TypeError` is raised.
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400544 """
545 for badObj in [True, object(), "hello", [], self]:
546 self.assertRaises(
547 TypeError,
548 X509Extension,
549 'authorityKeyIdentifier', False, 'keyid:always,issuer:always',
550 issuer=badObj)
Rick Dean47262da2009-07-08 16:17:17 -0500551
552
Jean-Paul Calderone391585f2008-12-31 14:36:31 -0500553
Jean-Paul Calderone18808652009-07-05 12:54:05 -0400554class PKeyTests(TestCase):
Jean-Paul Calderoneac930e12008-03-06 18:50:51 -0500555 """
Jonathan Ballet648875f2011-07-16 14:14:58 +0900556 Unit tests for :py:class:`OpenSSL.crypto.PKey`.
Jean-Paul Calderoneac930e12008-03-06 18:50:51 -0500557 """
Jean-Paul Calderone68649052009-07-17 21:14:27 -0400558 def test_type(self):
559 """
Jonathan Ballet648875f2011-07-16 14:14:58 +0900560 :py:class:`PKey` and :py:class:`PKeyType` refer to the same type object
561 and can be used to create instances of that type.
Jean-Paul Calderone68649052009-07-17 21:14:27 -0400562 """
563 self.assertIdentical(PKey, PKeyType)
564 self.assertConsistentType(PKey, 'PKey')
565
566
Jean-Paul Calderoned8782ad2008-03-04 23:39:59 -0500567 def test_construction(self):
568 """
Jonathan Ballet648875f2011-07-16 14:14:58 +0900569 :py:class:`PKey` takes no arguments and returns a new :py:class:`PKey` instance.
Jean-Paul Calderoned8782ad2008-03-04 23:39:59 -0500570 """
571 self.assertRaises(TypeError, PKey, None)
572 key = PKey()
573 self.assertTrue(
574 isinstance(key, PKeyType),
575 "%r is of type %r, should be %r" % (key, type(key), PKeyType))
576
577
578 def test_pregeneration(self):
579 """
Jonathan Ballet648875f2011-07-16 14:14:58 +0900580 :py:attr:`PKeyType.bits` and :py:attr:`PKeyType.type` return :py:data:`0` before the key is
581 generated. :py:attr:`PKeyType.check` raises :py:exc:`TypeError` before the key is
Jean-Paul Calderoned8782ad2008-03-04 23:39:59 -0500582 generated.
583 """
584 key = PKey()
585 self.assertEqual(key.type(), 0)
586 self.assertEqual(key.bits(), 0)
Jean-Paul Calderone55ec1712009-05-13 14:14:30 -0400587 self.assertRaises(TypeError, key.check)
Jean-Paul Calderoned8782ad2008-03-04 23:39:59 -0500588
589
590 def test_failedGeneration(self):
591 """
Jonathan Ballet648875f2011-07-16 14:14:58 +0900592 :py:meth:`PKeyType.generate_key` takes two arguments, the first giving the key
593 type as one of :py:data:`TYPE_RSA` or :py:data:`TYPE_DSA` and the second giving the
Jean-Paul Calderoneab82db72008-03-06 00:09:31 -0500594 number of bits to generate. If an invalid type is specified or
Jonathan Ballet648875f2011-07-16 14:14:58 +0900595 generation fails, :py:exc:`Error` is raised. If an invalid number of bits is
596 specified, :py:exc:`ValueError` or :py:exc:`Error` is raised.
Jean-Paul Calderoned8782ad2008-03-04 23:39:59 -0500597 """
598 key = PKey()
599 self.assertRaises(TypeError, key.generate_key)
600 self.assertRaises(TypeError, key.generate_key, 1, 2, 3)
601 self.assertRaises(TypeError, key.generate_key, "foo", "bar")
602 self.assertRaises(Error, key.generate_key, -1, 0)
Jean-Paul Calderoneab82db72008-03-06 00:09:31 -0500603
Jean-Paul Calderoneab82db72008-03-06 00:09:31 -0500604 self.assertRaises(ValueError, key.generate_key, TYPE_RSA, -1)
605 self.assertRaises(ValueError, key.generate_key, TYPE_RSA, 0)
Jean-Paul Calderoned71fe982008-03-06 00:31:50 -0500606
607 # XXX RSA generation for small values of bits is fairly buggy in a wide
608 # range of OpenSSL versions. I need to figure out what the safe lower
609 # bound for a reasonable number of OpenSSL versions is and explicitly
610 # check for that in the wrapper. The failure behavior is typically an
611 # infinite loop inside OpenSSL.
612
613 # self.assertRaises(Error, key.generate_key, TYPE_RSA, 2)
Jean-Paul Calderoned8782ad2008-03-04 23:39:59 -0500614
615 # XXX DSA generation seems happy with any number of bits. The DSS
616 # says bits must be between 512 and 1024 inclusive. OpenSSL's DSA
617 # generator doesn't seem to care about the upper limit at all. For
Jean-Paul Calderoneeff3cd92008-03-05 22:35:26 -0500618 # the lower limit, it uses 512 if anything smaller is specified.
Jean-Paul Calderoned8782ad2008-03-04 23:39:59 -0500619 # So, it doesn't seem possible to make generate_key fail for
620 # TYPE_DSA with a bits argument which is at least an int.
621
622 # self.assertRaises(Error, key.generate_key, TYPE_DSA, -7)
623
624
625 def test_rsaGeneration(self):
626 """
Jonathan Ballet648875f2011-07-16 14:14:58 +0900627 :py:meth:`PKeyType.generate_key` generates an RSA key when passed
628 :py:data:`TYPE_RSA` as a type and a reasonable number of bits.
Jean-Paul Calderoned8782ad2008-03-04 23:39:59 -0500629 """
630 bits = 128
631 key = PKey()
632 key.generate_key(TYPE_RSA, bits)
633 self.assertEqual(key.type(), TYPE_RSA)
634 self.assertEqual(key.bits(), bits)
Jean-Paul Calderone8e6ce972009-05-13 12:32:49 -0400635 self.assertTrue(key.check())
Jean-Paul Calderoned8782ad2008-03-04 23:39:59 -0500636
637
638 def test_dsaGeneration(self):
639 """
Jonathan Ballet648875f2011-07-16 14:14:58 +0900640 :py:meth:`PKeyType.generate_key` generates a DSA key when passed
641 :py:data:`TYPE_DSA` as a type and a reasonable number of bits.
Jean-Paul Calderoned8782ad2008-03-04 23:39:59 -0500642 """
643 # 512 is a magic number. The DSS (Digital Signature Standard)
644 # allows a minimum of 512 bits for DSA. DSA_generate_parameters
645 # will silently promote any value below 512 to 512.
646 bits = 512
647 key = PKey()
648 key.generate_key(TYPE_DSA, bits)
Jean-Paul Calderonef6745b32013-03-01 15:08:46 -0800649 # self.assertEqual(key.type(), TYPE_DSA)
650 # self.assertEqual(key.bits(), bits)
651 # self.assertRaises(TypeError, key.check)
Jean-Paul Calderoned8782ad2008-03-04 23:39:59 -0500652
653
654 def test_regeneration(self):
655 """
Jonathan Ballet648875f2011-07-16 14:14:58 +0900656 :py:meth:`PKeyType.generate_key` can be called multiple times on the same
Jean-Paul Calderoned8782ad2008-03-04 23:39:59 -0500657 key to generate new keys.
658 """
659 key = PKey()
660 for type, bits in [(TYPE_RSA, 512), (TYPE_DSA, 576)]:
661 key.generate_key(type, bits)
662 self.assertEqual(key.type(), type)
663 self.assertEqual(key.bits(), bits)
Jean-Paul Calderoneeff3cd92008-03-05 22:35:26 -0500664
665
Jean-Paul Calderone55ec1712009-05-13 14:14:30 -0400666 def test_inconsistentKey(self):
667 """
Jonathan Ballet648875f2011-07-16 14:14:58 +0900668 :py:`PKeyType.check` returns :py:exc:`Error` if the key is not consistent.
Jean-Paul Calderone55ec1712009-05-13 14:14:30 -0400669 """
670 key = load_privatekey(FILETYPE_PEM, inconsistentPrivateKeyPEM)
Jean-Paul Calderoned338e4e2009-05-13 15:45:07 -0400671 self.assertRaises(Error, key.check)
Jean-Paul Calderone55ec1712009-05-13 14:14:30 -0400672
Jean-Paul Calderoneeff3cd92008-03-05 22:35:26 -0500673
Jean-Paul Calderonee81020e2011-06-12 21:48:57 -0400674 def test_check_wrong_args(self):
675 """
Jonathan Ballet648875f2011-07-16 14:14:58 +0900676 :py:meth:`PKeyType.check` raises :py:exc:`TypeError` if called with any arguments.
Jean-Paul Calderonee81020e2011-06-12 21:48:57 -0400677 """
678 self.assertRaises(TypeError, PKey().check, None)
679 self.assertRaises(TypeError, PKey().check, object())
680 self.assertRaises(TypeError, PKey().check, 1)
681
682
Jean-Paul Calderone02d01972011-10-31 10:39:29 -0400683 def test_check_public_key(self):
684 """
685 :py:meth:`PKeyType.check` raises :py:exc:`TypeError` if only the public
686 part of the key is available.
687 """
688 # A trick to get a public-only key
689 key = PKey()
690 key.generate_key(TYPE_RSA, 512)
691 cert = X509()
692 cert.set_pubkey(key)
693 pub = cert.get_pubkey()
694 self.assertRaises(TypeError, pub.check)
695
696
Jean-Paul Calderonee81020e2011-06-12 21:48:57 -0400697
Jean-Paul Calderone18808652009-07-05 12:54:05 -0400698class X509NameTests(TestCase):
Jean-Paul Calderoneeff3cd92008-03-05 22:35:26 -0500699 """
Jonathan Ballet648875f2011-07-16 14:14:58 +0900700 Unit tests for :py:class:`OpenSSL.crypto.X509Name`.
Jean-Paul Calderoneeff3cd92008-03-05 22:35:26 -0500701 """
Jean-Paul Calderonee098dc72008-03-06 18:36:19 -0500702 def _x509name(self, **attrs):
703 # XXX There's no other way to get a new X509Name yet.
704 name = X509().get_subject()
Jean-Paul Calderone40dd0992010-08-22 17:52:07 -0400705 attrs = list(attrs.items())
Jean-Paul Calderonee098dc72008-03-06 18:36:19 -0500706 # Make the order stable - order matters!
Jean-Paul Calderone40dd0992010-08-22 17:52:07 -0400707 def key(attr):
708 return attr[1]
709 attrs.sort(key=key)
Jean-Paul Calderonee098dc72008-03-06 18:36:19 -0500710 for k, v in attrs:
711 setattr(name, k, v)
712 return name
713
714
Rick Deane15b1472009-07-09 15:53:42 -0500715 def test_type(self):
716 """
Jonathan Ballet648875f2011-07-16 14:14:58 +0900717 The type of X509Name objects is :py:class:`X509NameType`.
Rick Deane15b1472009-07-09 15:53:42 -0500718 """
Jean-Paul Calderone68649052009-07-17 21:14:27 -0400719 self.assertIdentical(X509Name, X509NameType)
720 self.assertEqual(X509NameType.__name__, 'X509Name')
721 self.assertTrue(isinstance(X509NameType, type))
722
Rick Deane15b1472009-07-09 15:53:42 -0500723 name = self._x509name()
724 self.assertTrue(
725 isinstance(name, X509NameType),
726 "%r is of type %r, should be %r" % (
727 name, type(name), X509NameType))
Rick Deane15b1472009-07-09 15:53:42 -0500728
729
Jean-Paul Calderone9ce9afb2011-04-22 18:16:22 -0400730 def test_onlyStringAttributes(self):
731 """
Jonathan Ballet648875f2011-07-16 14:14:58 +0900732 Attempting to set a non-:py:data:`str` attribute name on an :py:class:`X509NameType`
733 instance causes :py:exc:`TypeError` to be raised.
Jean-Paul Calderone9ce9afb2011-04-22 18:16:22 -0400734 """
735 name = self._x509name()
736 # Beyond these cases, you may also think that unicode should be
737 # rejected. Sorry, you're wrong. unicode is automatically converted to
738 # str outside of the control of X509Name, so there's no way to reject
739 # it.
Jean-Paul Calderoneff363be2013-03-03 10:21:23 -0800740
741 # Also, this used to test str subclasses, but that test is less relevant
742 # now that the implementation is in Python instead of C. Also PyPy
743 # automatically converts str subclasses to str when they are passed to
744 # setattr, so we can't test it on PyPy. Apparently CPython does this
745 # sometimes as well.
Jean-Paul Calderone9ce9afb2011-04-22 18:16:22 -0400746 self.assertRaises(TypeError, setattr, name, None, "hello")
747 self.assertRaises(TypeError, setattr, name, 30, "hello")
Jean-Paul Calderone9ce9afb2011-04-22 18:16:22 -0400748
Jean-Paul Calderone5300d6a2013-12-29 16:36:50 -0500749
Jean-Paul Calderone9ce9afb2011-04-22 18:16:22 -0400750 def test_setInvalidAttribute(self):
751 """
Jonathan Ballet648875f2011-07-16 14:14:58 +0900752 Attempting to set any attribute name on an :py:class:`X509NameType` instance for
753 which no corresponding NID is defined causes :py:exc:`AttributeError` to be
Jean-Paul Calderone9ce9afb2011-04-22 18:16:22 -0400754 raised.
755 """
756 name = self._x509name()
757 self.assertRaises(AttributeError, setattr, name, "no such thing", None)
758
759
Jean-Paul Calderoneeff3cd92008-03-05 22:35:26 -0500760 def test_attributes(self):
761 """
Jonathan Ballet648875f2011-07-16 14:14:58 +0900762 :py:class:`X509NameType` instances have attributes for each standard (?)
Jean-Paul Calderoneeff3cd92008-03-05 22:35:26 -0500763 X509Name field.
764 """
Jean-Paul Calderonee098dc72008-03-06 18:36:19 -0500765 name = self._x509name()
Jean-Paul Calderoneeff3cd92008-03-05 22:35:26 -0500766 name.commonName = "foo"
767 self.assertEqual(name.commonName, "foo")
768 self.assertEqual(name.CN, "foo")
769 name.CN = "baz"
770 self.assertEqual(name.commonName, "baz")
771 self.assertEqual(name.CN, "baz")
772 name.commonName = "bar"
773 self.assertEqual(name.commonName, "bar")
774 self.assertEqual(name.CN, "bar")
775 name.CN = "quux"
776 self.assertEqual(name.commonName, "quux")
777 self.assertEqual(name.CN, "quux")
778
779
Jean-Paul Calderoneeff3cd92008-03-05 22:35:26 -0500780 def test_copy(self):
781 """
Jonathan Ballet648875f2011-07-16 14:14:58 +0900782 :py:class:`X509Name` creates a new :py:class:`X509NameType` instance with all the same
783 attributes as an existing :py:class:`X509NameType` instance when called with
Jean-Paul Calderoneeff3cd92008-03-05 22:35:26 -0500784 one.
785 """
Jean-Paul Calderonee098dc72008-03-06 18:36:19 -0500786 name = self._x509name(commonName="foo", emailAddress="bar@example.com")
Jean-Paul Calderoneeff3cd92008-03-05 22:35:26 -0500787
788 copy = X509Name(name)
789 self.assertEqual(copy.commonName, "foo")
790 self.assertEqual(copy.emailAddress, "bar@example.com")
Jean-Paul Calderonee098dc72008-03-06 18:36:19 -0500791
792 # Mutate the copy and ensure the original is unmodified.
Jean-Paul Calderoneeff3cd92008-03-05 22:35:26 -0500793 copy.commonName = "baz"
794 self.assertEqual(name.commonName, "foo")
Jean-Paul Calderonee098dc72008-03-06 18:36:19 -0500795
796 # Mutate the original and ensure the copy is unmodified.
Jean-Paul Calderoneeff3cd92008-03-05 22:35:26 -0500797 name.emailAddress = "quux@example.com"
798 self.assertEqual(copy.emailAddress, "bar@example.com")
799
Jean-Paul Calderonee098dc72008-03-06 18:36:19 -0500800
801 def test_repr(self):
802 """
Jonathan Ballet648875f2011-07-16 14:14:58 +0900803 :py:func:`repr` passed an :py:class:`X509NameType` instance should return a string
Jean-Paul Calderonee098dc72008-03-06 18:36:19 -0500804 containing a description of the type and the NIDs which have been set
805 on it.
806 """
807 name = self._x509name(commonName="foo", emailAddress="bar")
808 self.assertEqual(
809 repr(name),
810 "<X509Name object '/emailAddress=bar/CN=foo'>")
811
812
813 def test_comparison(self):
814 """
Jonathan Ballet648875f2011-07-16 14:14:58 +0900815 :py:class:`X509NameType` instances should compare based on their NIDs.
Jean-Paul Calderonee098dc72008-03-06 18:36:19 -0500816 """
817 def _equality(a, b, assertTrue, assertFalse):
818 assertTrue(a == b, "(%r == %r) --> False" % (a, b))
819 assertFalse(a != b)
820 assertTrue(b == a)
821 assertFalse(b != a)
822
823 def assertEqual(a, b):
824 _equality(a, b, self.assertTrue, self.assertFalse)
825
826 # Instances compare equal to themselves.
827 name = self._x509name()
828 assertEqual(name, name)
829
830 # Empty instances should compare equal to each other.
831 assertEqual(self._x509name(), self._x509name())
832
833 # Instances with equal NIDs should compare equal to each other.
834 assertEqual(self._x509name(commonName="foo"),
835 self._x509name(commonName="foo"))
836
837 # Instance with equal NIDs set using different aliases should compare
838 # equal to each other.
839 assertEqual(self._x509name(commonName="foo"),
840 self._x509name(CN="foo"))
841
842 # Instances with more than one NID with the same values should compare
843 # equal to each other.
844 assertEqual(self._x509name(CN="foo", organizationalUnitName="bar"),
845 self._x509name(commonName="foo", OU="bar"))
846
847 def assertNotEqual(a, b):
848 _equality(a, b, self.assertFalse, self.assertTrue)
849
850 # Instances with different values for the same NID should not compare
851 # equal to each other.
852 assertNotEqual(self._x509name(CN="foo"),
853 self._x509name(CN="bar"))
854
855 # Instances with different NIDs should not compare equal to each other.
856 assertNotEqual(self._x509name(CN="foo"),
857 self._x509name(OU="foo"))
858
859 def _inequality(a, b, assertTrue, assertFalse):
860 assertTrue(a < b)
861 assertTrue(a <= b)
862 assertTrue(b > a)
863 assertTrue(b >= a)
864 assertFalse(a > b)
865 assertFalse(a >= b)
866 assertFalse(b < a)
867 assertFalse(b <= a)
868
869 def assertLessThan(a, b):
870 _inequality(a, b, self.assertTrue, self.assertFalse)
871
872 # An X509Name with a NID with a value which sorts less than the value
873 # of the same NID on another X509Name compares less than the other
874 # X509Name.
875 assertLessThan(self._x509name(CN="abc"),
876 self._x509name(CN="def"))
877
878 def assertGreaterThan(a, b):
879 _inequality(a, b, self.assertFalse, self.assertTrue)
880
881 # An X509Name with a NID with a value which sorts greater than the
882 # value of the same NID on another X509Name compares greater than the
883 # other X509Name.
884 assertGreaterThan(self._x509name(CN="def"),
885 self._x509name(CN="abc"))
Jean-Paul Calderone2aa2b332008-03-06 21:43:14 -0500886
887
Jean-Paul Calderone110cd092008-03-24 17:27:42 -0400888 def test_hash(self):
889 """
Jonathan Ballet648875f2011-07-16 14:14:58 +0900890 :py:meth:`X509Name.hash` returns an integer hash based on the value of the
Jean-Paul Calderone110cd092008-03-24 17:27:42 -0400891 name.
892 """
893 a = self._x509name(CN="foo")
894 b = self._x509name(CN="foo")
895 self.assertEqual(a.hash(), b.hash())
896 a.CN = "bar"
897 self.assertNotEqual(a.hash(), b.hash())
898
899
Jean-Paul Calderonee957a002008-03-25 15:16:51 -0400900 def test_der(self):
901 """
Jonathan Ballet648875f2011-07-16 14:14:58 +0900902 :py:meth:`X509Name.der` returns the DER encoded form of the name.
Jean-Paul Calderonee957a002008-03-25 15:16:51 -0400903 """
904 a = self._x509name(CN="foo", C="US")
905 self.assertEqual(
906 a.der(),
Jean-Paul Calderone2ac721b2010-08-22 19:20:00 -0400907 b('0\x1b1\x0b0\t\x06\x03U\x04\x06\x13\x02US'
908 '1\x0c0\n\x06\x03U\x04\x03\x13\x03foo'))
Jean-Paul Calderonee957a002008-03-25 15:16:51 -0400909
910
Jean-Paul Calderonec54cc182008-03-26 21:11:07 -0400911 def test_get_components(self):
912 """
Jonathan Ballet648875f2011-07-16 14:14:58 +0900913 :py:meth:`X509Name.get_components` returns a :py:data:`list` of
914 two-tuples of :py:data:`str`
Jean-Paul Calderonec54cc182008-03-26 21:11:07 -0400915 giving the NIDs and associated values which make up the name.
916 """
917 a = self._x509name()
918 self.assertEqual(a.get_components(), [])
919 a.CN = "foo"
Jean-Paul Calderone2ac721b2010-08-22 19:20:00 -0400920 self.assertEqual(a.get_components(), [(b("CN"), b("foo"))])
Jean-Paul Calderonec54cc182008-03-26 21:11:07 -0400921 a.organizationalUnitName = "bar"
922 self.assertEqual(
923 a.get_components(),
Jean-Paul Calderone2ac721b2010-08-22 19:20:00 -0400924 [(b("CN"), b("foo")), (b("OU"), b("bar"))])
Jean-Paul Calderonec54cc182008-03-26 21:11:07 -0400925
Jean-Paul Calderone110cd092008-03-24 17:27:42 -0400926
Jean-Paul Calderone4bf75c62013-08-23 15:39:53 -0400927 def test_load_nul_byte_attribute(self):
928 """
Jean-Paul Calderone9af07b02013-08-23 16:07:31 -0400929 An :py:class:`OpenSSL.crypto.X509Name` from an
930 :py:class:`OpenSSL.crypto.X509` instance loaded from a file can have a
Jean-Paul Calderone4bf75c62013-08-23 15:39:53 -0400931 NUL byte in the value of one of its attributes.
932 """
933 cert = load_certificate(FILETYPE_PEM, nulbyteSubjectAltNamePEM)
934 subject = cert.get_subject()
935 self.assertEqual(
Jean-Paul Calderone06754fc2013-08-23 15:47:47 -0400936 "null.python.org\x00example.org", subject.commonName)
Jean-Paul Calderone4bf75c62013-08-23 15:39:53 -0400937
938
Jean-Paul Calderone5300d6a2013-12-29 16:36:50 -0500939 def test_setAttributeFailure(self):
940 """
941 If the value of an attribute cannot be set for some reason then
942 :py:class:`OpenSSL.crypto.Error` is raised.
943 """
944 name = self._x509name()
945 # This value is too long
946 self.assertRaises(Error, setattr, name, "O", b"x" * 512)
947
948
Jean-Paul Calderone4bf75c62013-08-23 15:39:53 -0400949
Jean-Paul Calderoneac0d95f2008-03-10 00:00:42 -0400950class _PKeyInteractionTestsMixin:
951 """
952 Tests which involve another thing and a PKey.
953 """
954 def signable(self):
955 """
Jonathan Ballet648875f2011-07-16 14:14:58 +0900956 Return something with a :py:meth:`set_pubkey`, :py:meth:`set_pubkey`,
957 and :py:meth:`sign` method.
Jean-Paul Calderoneac0d95f2008-03-10 00:00:42 -0400958 """
959 raise NotImplementedError()
960
961
962 def test_signWithUngenerated(self):
963 """
Jonathan Ballet648875f2011-07-16 14:14:58 +0900964 :py:meth:`X509Req.sign` raises :py:exc:`ValueError` when pass a
965 :py:class:`PKey` with no parts.
Jean-Paul Calderoneac0d95f2008-03-10 00:00:42 -0400966 """
967 request = self.signable()
968 key = PKey()
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -0500969 self.assertRaises(ValueError, request.sign, key, GOOD_DIGEST)
Jean-Paul Calderoneac0d95f2008-03-10 00:00:42 -0400970
971
972 def test_signWithPublicKey(self):
973 """
Jonathan Ballet648875f2011-07-16 14:14:58 +0900974 :py:meth:`X509Req.sign` raises :py:exc:`ValueError` when pass a
975 :py:class:`PKey` with no private part as the signing key.
Jean-Paul Calderoneac0d95f2008-03-10 00:00:42 -0400976 """
977 request = self.signable()
978 key = PKey()
979 key.generate_key(TYPE_RSA, 512)
980 request.set_pubkey(key)
981 pub = request.get_pubkey()
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -0500982 self.assertRaises(ValueError, request.sign, pub, GOOD_DIGEST)
Jean-Paul Calderoneac0d95f2008-03-10 00:00:42 -0400983
984
Jean-Paul Calderonecc05a912010-08-03 18:24:19 -0400985 def test_signWithUnknownDigest(self):
986 """
Jonathan Ballet648875f2011-07-16 14:14:58 +0900987 :py:meth:`X509Req.sign` raises :py:exc:`ValueError` when passed a digest name which is
Jean-Paul Calderonecc05a912010-08-03 18:24:19 -0400988 not known.
989 """
990 request = self.signable()
991 key = PKey()
992 key.generate_key(TYPE_RSA, 512)
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -0500993 self.assertRaises(ValueError, request.sign, key, BAD_DIGEST)
Jean-Paul Calderonecc05a912010-08-03 18:24:19 -0400994
995
Jean-Paul Calderoneb9725592010-08-03 18:17:22 -0400996 def test_sign(self):
997 """
Jonathan Ballet648875f2011-07-16 14:14:58 +0900998 :py:meth:`X509Req.sign` succeeds when passed a private key object and a valid
999 digest function. :py:meth:`X509Req.verify` can be used to check the signature.
Jean-Paul Calderoneb9725592010-08-03 18:17:22 -04001000 """
1001 request = self.signable()
1002 key = PKey()
1003 key.generate_key(TYPE_RSA, 512)
1004 request.set_pubkey(key)
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05001005 request.sign(key, GOOD_DIGEST)
Jean-Paul Calderoneb9725592010-08-03 18:17:22 -04001006 # If the type has a verify method, cover that too.
1007 if getattr(request, 'verify', None) is not None:
1008 pub = request.get_pubkey()
1009 self.assertTrue(request.verify(pub))
1010 # Make another key that won't verify.
1011 key = PKey()
1012 key.generate_key(TYPE_RSA, 512)
1013 self.assertRaises(Error, request.verify, key)
1014
1015
1016
Jean-Paul Calderoneac0d95f2008-03-10 00:00:42 -04001017
Jean-Paul Calderone18808652009-07-05 12:54:05 -04001018class X509ReqTests(TestCase, _PKeyInteractionTestsMixin):
Jean-Paul Calderone2aa2b332008-03-06 21:43:14 -05001019 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001020 Tests for :py:class:`OpenSSL.crypto.X509Req`.
Jean-Paul Calderone2aa2b332008-03-06 21:43:14 -05001021 """
Jean-Paul Calderoneac0d95f2008-03-10 00:00:42 -04001022 def signable(self):
1023 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001024 Create and return a new :py:class:`X509Req`.
Jean-Paul Calderoneac0d95f2008-03-10 00:00:42 -04001025 """
1026 return X509Req()
1027
1028
Jean-Paul Calderone68649052009-07-17 21:14:27 -04001029 def test_type(self):
1030 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001031 :py:obj:`X509Req` and :py:obj:`X509ReqType` refer to the same type object and can be
Jean-Paul Calderone68649052009-07-17 21:14:27 -04001032 used to create instances of that type.
1033 """
1034 self.assertIdentical(X509Req, X509ReqType)
1035 self.assertConsistentType(X509Req, 'X509Req')
Jean-Paul Calderone2aa2b332008-03-06 21:43:14 -05001036
1037
1038 def test_construction(self):
1039 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001040 :py:obj:`X509Req` takes no arguments and returns an :py:obj:`X509ReqType` instance.
Jean-Paul Calderone2aa2b332008-03-06 21:43:14 -05001041 """
1042 request = X509Req()
1043 self.assertTrue(
1044 isinstance(request, X509ReqType),
1045 "%r is of type %r, should be %r" % (request, type(request), X509ReqType))
1046
1047
Jean-Paul Calderone8dd19b82008-12-28 20:41:16 -05001048 def test_version(self):
1049 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001050 :py:obj:`X509ReqType.set_version` sets the X.509 version of the certificate
1051 request. :py:obj:`X509ReqType.get_version` returns the X.509 version of
Jean-Paul Calderone8dd19b82008-12-28 20:41:16 -05001052 the certificate request. The initial value of the version is 0.
1053 """
1054 request = X509Req()
1055 self.assertEqual(request.get_version(), 0)
1056 request.set_version(1)
1057 self.assertEqual(request.get_version(), 1)
1058 request.set_version(3)
1059 self.assertEqual(request.get_version(), 3)
1060
1061
Jean-Paul Calderone54e49e92010-07-30 11:04:46 -04001062 def test_version_wrong_args(self):
Jean-Paul Calderoneaa6c0692010-09-08 22:43:09 -04001063 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001064 :py:obj:`X509ReqType.set_version` raises :py:obj:`TypeError` if called with the wrong
1065 number of arguments or with a non-:py:obj:`int` argument.
1066 :py:obj:`X509ReqType.get_version` raises :py:obj:`TypeError` if called with any
Jean-Paul Calderoneaa6c0692010-09-08 22:43:09 -04001067 arguments.
1068 """
Jean-Paul Calderone54e49e92010-07-30 11:04:46 -04001069 request = X509Req()
1070 self.assertRaises(TypeError, request.set_version)
1071 self.assertRaises(TypeError, request.set_version, "foo")
1072 self.assertRaises(TypeError, request.set_version, 1, 2)
1073 self.assertRaises(TypeError, request.get_version, None)
1074
1075
Jean-Paul Calderone2aa2b332008-03-06 21:43:14 -05001076 def test_get_subject(self):
1077 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001078 :py:obj:`X509ReqType.get_subject` returns an :py:obj:`X509Name` for the subject of
Jean-Paul Calderone2aa2b332008-03-06 21:43:14 -05001079 the request and which is valid even after the request object is
1080 otherwise dead.
1081 """
1082 request = X509Req()
1083 subject = request.get_subject()
1084 self.assertTrue(
1085 isinstance(subject, X509NameType),
1086 "%r is of type %r, should be %r" % (subject, type(subject), X509NameType))
1087 subject.commonName = "foo"
1088 self.assertEqual(request.get_subject().commonName, "foo")
1089 del request
1090 subject.commonName = "bar"
1091 self.assertEqual(subject.commonName, "bar")
Jean-Paul Calderone78381d22008-03-06 23:35:22 -05001092
1093
Jean-Paul Calderone54e49e92010-07-30 11:04:46 -04001094 def test_get_subject_wrong_args(self):
Jean-Paul Calderoneaa6c0692010-09-08 22:43:09 -04001095 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001096 :py:obj:`X509ReqType.get_subject` raises :py:obj:`TypeError` if called with any
Jean-Paul Calderoneaa6c0692010-09-08 22:43:09 -04001097 arguments.
1098 """
Jean-Paul Calderone54e49e92010-07-30 11:04:46 -04001099 request = X509Req()
1100 self.assertRaises(TypeError, request.get_subject, None)
1101
1102
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -04001103 def test_add_extensions(self):
1104 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001105 :py:obj:`X509Req.add_extensions` accepts a :py:obj:`list` of :py:obj:`X509Extension`
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -04001106 instances and adds them to the X509 request.
1107 """
1108 request = X509Req()
1109 request.add_extensions([
Jean-Paul Calderonee16f9f32010-08-22 19:20:37 -04001110 X509Extension(b('basicConstraints'), True, b('CA:false'))])
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -04001111 # XXX Add get_extensions so the rest of this unit test can be written.
1112
1113
1114 def test_add_extensions_wrong_args(self):
1115 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001116 :py:obj:`X509Req.add_extensions` raises :py:obj:`TypeError` if called with the wrong
1117 number of arguments or with a non-:py:obj:`list`. Or it raises :py:obj:`ValueError`
1118 if called with a :py:obj:`list` containing objects other than :py:obj:`X509Extension`
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -04001119 instances.
1120 """
1121 request = X509Req()
1122 self.assertRaises(TypeError, request.add_extensions)
1123 self.assertRaises(TypeError, request.add_extensions, object())
1124 self.assertRaises(ValueError, request.add_extensions, [object()])
1125 self.assertRaises(TypeError, request.add_extensions, [], None)
1126
1127
Jean-Paul Calderone5565f0f2013-03-06 11:10:20 -08001128 def test_verify_wrong_args(self):
1129 """
1130 :py:obj:`X509Req.verify` raises :py:obj:`TypeError` if called with zero
1131 arguments or more than one argument or if passed anything other than a
1132 :py:obj:`PKey` instance as its single argument.
1133 """
1134 request = X509Req()
1135 self.assertRaises(TypeError, request.verify)
1136 self.assertRaises(TypeError, request.verify, object())
1137 self.assertRaises(TypeError, request.verify, PKey(), object())
1138
1139
1140 def test_verify_uninitialized_key(self):
1141 """
1142 :py:obj:`X509Req.verify` raises :py:obj:`OpenSSL.crypto.Error` if called
1143 with a :py:obj:`OpenSSL.crypto.PKey` which contains no key data.
1144 """
1145 request = X509Req()
1146 pkey = PKey()
1147 self.assertRaises(Error, request.verify, pkey)
1148
1149
1150 def test_verify_wrong_key(self):
1151 """
1152 :py:obj:`X509Req.verify` raises :py:obj:`OpenSSL.crypto.Error` if called
1153 with a :py:obj:`OpenSSL.crypto.PKey` which does not represent the public
1154 part of the key which signed the request.
1155 """
1156 request = X509Req()
1157 pkey = load_privatekey(FILETYPE_PEM, cleartextPrivateKeyPEM)
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05001158 request.sign(pkey, GOOD_DIGEST)
Jean-Paul Calderone5565f0f2013-03-06 11:10:20 -08001159 another_pkey = load_privatekey(FILETYPE_PEM, client_key_pem)
1160 self.assertRaises(Error, request.verify, another_pkey)
1161
1162
1163 def test_verify_success(self):
1164 """
1165 :py:obj:`X509Req.verify` returns :py:obj:`True` if called with a
1166 :py:obj:`OpenSSL.crypto.PKey` which represents the public part ofthe key
1167 which signed the request.
1168 """
1169 request = X509Req()
1170 pkey = load_privatekey(FILETYPE_PEM, cleartextPrivateKeyPEM)
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05001171 request.sign(pkey, GOOD_DIGEST)
Jean-Paul Calderone5565f0f2013-03-06 11:10:20 -08001172 self.assertEqual(True, request.verify(pkey))
1173
1174
Jean-Paul Calderone78381d22008-03-06 23:35:22 -05001175
Jean-Paul Calderone18808652009-07-05 12:54:05 -04001176class X509Tests(TestCase, _PKeyInteractionTestsMixin):
Jean-Paul Calderone78381d22008-03-06 23:35:22 -05001177 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001178 Tests for :py:obj:`OpenSSL.crypto.X509`.
Jean-Paul Calderone78381d22008-03-06 23:35:22 -05001179 """
Jean-Paul Calderone5ef86512008-04-26 19:06:28 -04001180 pemData = cleartextCertificatePEM + cleartextPrivateKeyPEM
Jean-Paul Calderone8114b452008-03-25 15:27:59 -04001181
Roland Hedberg7e4930e2008-04-22 22:58:50 +02001182 extpem = """
1183-----BEGIN CERTIFICATE-----
1184MIIC3jCCAkegAwIBAgIJAJHFjlcCgnQzMA0GCSqGSIb3DQEBBQUAMEcxCzAJBgNV
1185BAYTAlNFMRUwEwYDVQQIEwxXZXN0ZXJib3R0b20xEjAQBgNVBAoTCUNhdGFsb2dp
1186eDENMAsGA1UEAxMEUm9vdDAeFw0wODA0MjIxNDQ1MzhaFw0wOTA0MjIxNDQ1Mzha
1187MFQxCzAJBgNVBAYTAlNFMQswCQYDVQQIEwJXQjEUMBIGA1UEChMLT3Blbk1ldGFk
1188aXIxIjAgBgNVBAMTGW5vZGUxLm9tMi5vcGVubWV0YWRpci5vcmcwgZ8wDQYJKoZI
1189hvcNAQEBBQADgY0AMIGJAoGBAPIcQMrwbk2nESF/0JKibj9i1x95XYAOwP+LarwT
1190Op4EQbdlI9SY+uqYqlERhF19w7CS+S6oyqx0DRZSk4Y9dZ9j9/xgm2u/f136YS1u
1191zgYFPvfUs6PqYLPSM8Bw+SjJ+7+2+TN+Tkiof9WP1cMjodQwOmdsiRbR0/J7+b1B
1192hec1AgMBAAGjgcQwgcEwCQYDVR0TBAIwADAsBglghkgBhvhCAQ0EHxYdT3BlblNT
1193TCBHZW5lcmF0ZWQgQ2VydGlmaWNhdGUwHQYDVR0OBBYEFIdHsBcMVVMbAO7j6NCj
119403HgLnHaMB8GA1UdIwQYMBaAFL2h9Bf9Mre4vTdOiHTGAt7BRY/8MEYGA1UdEQQ/
1195MD2CDSouZXhhbXBsZS5vcmeCESoub20yLmV4bWFwbGUuY29thwSC7wgKgRNvbTJA
1196b3Blbm1ldGFkaXIub3JnMA0GCSqGSIb3DQEBBQUAA4GBALd7WdXkp2KvZ7/PuWZA
1197MPlIxyjS+Ly11+BNE0xGQRp9Wz+2lABtpgNqssvU156+HkKd02rGheb2tj7MX9hG
1198uZzbwDAZzJPjzDQDD7d3cWsrVcfIdqVU7epHqIadnOF+X0ghJ39pAm6VVadnSXCt
1199WpOdIpB8KksUTCzV591Nr1wd
1200-----END CERTIFICATE-----
1201 """
Jean-Paul Calderoneac0d95f2008-03-10 00:00:42 -04001202 def signable(self):
1203 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001204 Create and return a new :py:obj:`X509`.
Jean-Paul Calderoneac0d95f2008-03-10 00:00:42 -04001205 """
1206 return X509()
1207
1208
Jean-Paul Calderone68649052009-07-17 21:14:27 -04001209 def test_type(self):
1210 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001211 :py:obj:`X509` and :py:obj:`X509Type` refer to the same type object and can be used
Jean-Paul Calderone68649052009-07-17 21:14:27 -04001212 to create instances of that type.
1213 """
1214 self.assertIdentical(X509, X509Type)
1215 self.assertConsistentType(X509, 'X509')
1216
1217
Jean-Paul Calderone78381d22008-03-06 23:35:22 -05001218 def test_construction(self):
1219 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001220 :py:obj:`X509` takes no arguments and returns an instance of :py:obj:`X509Type`.
Jean-Paul Calderone78381d22008-03-06 23:35:22 -05001221 """
1222 certificate = X509()
1223 self.assertTrue(
1224 isinstance(certificate, X509Type),
1225 "%r is of type %r, should be %r" % (certificate,
1226 type(certificate),
1227 X509Type))
Rick Deane15b1472009-07-09 15:53:42 -05001228 self.assertEqual(type(X509Type).__name__, 'type')
1229 self.assertEqual(type(certificate).__name__, 'X509')
1230 self.assertEqual(type(certificate), X509Type)
Rick Dean04113e72009-07-16 12:06:35 -05001231 self.assertEqual(type(certificate), X509)
Jean-Paul Calderone78381d22008-03-06 23:35:22 -05001232
1233
Jean-Paul Calderone3544eb42010-07-30 22:09:47 -04001234 def test_get_version_wrong_args(self):
1235 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001236 :py:obj:`X509.get_version` raises :py:obj:`TypeError` if invoked with any arguments.
Jean-Paul Calderone3544eb42010-07-30 22:09:47 -04001237 """
1238 cert = X509()
1239 self.assertRaises(TypeError, cert.get_version, None)
1240
1241
1242 def test_set_version_wrong_args(self):
1243 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001244 :py:obj:`X509.set_version` raises :py:obj:`TypeError` if invoked with the wrong number
1245 of arguments or an argument not of type :py:obj:`int`.
Jean-Paul Calderone3544eb42010-07-30 22:09:47 -04001246 """
1247 cert = X509()
1248 self.assertRaises(TypeError, cert.set_version)
1249 self.assertRaises(TypeError, cert.set_version, None)
1250 self.assertRaises(TypeError, cert.set_version, 1, None)
1251
1252
1253 def test_version(self):
1254 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001255 :py:obj:`X509.set_version` sets the certificate version number.
1256 :py:obj:`X509.get_version` retrieves it.
Jean-Paul Calderone3544eb42010-07-30 22:09:47 -04001257 """
1258 cert = X509()
1259 cert.set_version(1234)
1260 self.assertEquals(cert.get_version(), 1234)
1261
1262
1263 def test_get_serial_number_wrong_args(self):
1264 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001265 :py:obj:`X509.get_serial_number` raises :py:obj:`TypeError` if invoked with any
Jean-Paul Calderone3544eb42010-07-30 22:09:47 -04001266 arguments.
1267 """
1268 cert = X509()
1269 self.assertRaises(TypeError, cert.get_serial_number, None)
1270
1271
Jean-Paul Calderone78381d22008-03-06 23:35:22 -05001272 def test_serial_number(self):
1273 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001274 The serial number of an :py:obj:`X509Type` can be retrieved and modified with
1275 :py:obj:`X509Type.get_serial_number` and :py:obj:`X509Type.set_serial_number`.
Jean-Paul Calderone78381d22008-03-06 23:35:22 -05001276 """
1277 certificate = X509()
1278 self.assertRaises(TypeError, certificate.set_serial_number)
1279 self.assertRaises(TypeError, certificate.set_serial_number, 1, 2)
1280 self.assertRaises(TypeError, certificate.set_serial_number, "1")
1281 self.assertRaises(TypeError, certificate.set_serial_number, 5.5)
1282 self.assertEqual(certificate.get_serial_number(), 0)
1283 certificate.set_serial_number(1)
1284 self.assertEqual(certificate.get_serial_number(), 1)
1285 certificate.set_serial_number(2 ** 32 + 1)
1286 self.assertEqual(certificate.get_serial_number(), 2 ** 32 + 1)
1287 certificate.set_serial_number(2 ** 64 + 1)
1288 self.assertEqual(certificate.get_serial_number(), 2 ** 64 + 1)
Jean-Paul Calderone525ef802008-03-09 20:39:42 -04001289 certificate.set_serial_number(2 ** 128 + 1)
1290 self.assertEqual(certificate.get_serial_number(), 2 ** 128 + 1)
1291
1292
1293 def _setBoundTest(self, which):
1294 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001295 :py:obj:`X509Type.set_notBefore` takes a string in the format of an ASN1
Jean-Paul Calderone525ef802008-03-09 20:39:42 -04001296 GENERALIZEDTIME and sets the beginning of the certificate's validity
1297 period to it.
1298 """
1299 certificate = X509()
1300 set = getattr(certificate, 'set_not' + which)
1301 get = getattr(certificate, 'get_not' + which)
1302
Jean-Paul Calderonee0615b52008-03-09 21:44:46 -04001303 # Starts with no value.
1304 self.assertEqual(get(), None)
1305
Jean-Paul Calderone525ef802008-03-09 20:39:42 -04001306 # GMT (Or is it UTC?) -exarkun
Jean-Paul Calderonee890db32010-08-22 16:55:15 -04001307 when = b("20040203040506Z")
Jean-Paul Calderone525ef802008-03-09 20:39:42 -04001308 set(when)
1309 self.assertEqual(get(), when)
1310
1311 # A plus two hours and thirty minutes offset
Jean-Paul Calderonee890db32010-08-22 16:55:15 -04001312 when = b("20040203040506+0530")
Jean-Paul Calderone525ef802008-03-09 20:39:42 -04001313 set(when)
1314 self.assertEqual(get(), when)
1315
1316 # A minus one hour fifteen minutes offset
Jean-Paul Calderonee890db32010-08-22 16:55:15 -04001317 when = b("20040203040506-0115")
Jean-Paul Calderone525ef802008-03-09 20:39:42 -04001318 set(when)
1319 self.assertEqual(get(), when)
1320
1321 # An invalid string results in a ValueError
Jean-Paul Calderonee890db32010-08-22 16:55:15 -04001322 self.assertRaises(ValueError, set, b("foo bar"))
Jean-Paul Calderone525ef802008-03-09 20:39:42 -04001323
Jean-Paul Calderone31ca2002010-01-30 15:14:43 -05001324 # The wrong number of arguments results in a TypeError.
1325 self.assertRaises(TypeError, set)
Jean-Paul Calderonee890db32010-08-22 16:55:15 -04001326 self.assertRaises(TypeError, set, b("20040203040506Z"), b("20040203040506Z"))
1327 self.assertRaises(TypeError, get, b("foo bar"))
1328
Jean-Paul Calderone31ca2002010-01-30 15:14:43 -05001329
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -04001330 # XXX ASN1_TIME (not GENERALIZEDTIME)
Jean-Paul Calderone525ef802008-03-09 20:39:42 -04001331
1332 def test_set_notBefore(self):
1333 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001334 :py:obj:`X509Type.set_notBefore` takes a string in the format of an ASN1
Jean-Paul Calderone525ef802008-03-09 20:39:42 -04001335 GENERALIZEDTIME and sets the beginning of the certificate's validity
1336 period to it.
1337 """
1338 self._setBoundTest("Before")
1339
1340
1341 def test_set_notAfter(self):
1342 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001343 :py:obj:`X509Type.set_notAfter` takes a string in the format of an ASN1
Jean-Paul Calderone525ef802008-03-09 20:39:42 -04001344 GENERALIZEDTIME and sets the end of the certificate's validity period
1345 to it.
1346 """
1347 self._setBoundTest("After")
Jean-Paul Calderone76576d52008-03-24 16:04:46 -04001348
1349
Jean-Paul Calderone38a646d2008-03-25 15:16:18 -04001350 def test_get_notBefore(self):
1351 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001352 :py:obj:`X509Type.get_notBefore` returns a string in the format of an ASN1
Jean-Paul Calderone38a646d2008-03-25 15:16:18 -04001353 GENERALIZEDTIME even for certificates which store it as UTCTIME
1354 internally.
1355 """
Jean-Paul Calderone8114b452008-03-25 15:27:59 -04001356 cert = load_certificate(FILETYPE_PEM, self.pemData)
Jean-Paul Calderonee890db32010-08-22 16:55:15 -04001357 self.assertEqual(cert.get_notBefore(), b("20090325123658Z"))
Jean-Paul Calderone38a646d2008-03-25 15:16:18 -04001358
Rick Dean38a05c82009-07-18 01:41:30 -05001359
1360 def test_get_notAfter(self):
1361 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001362 :py:obj:`X509Type.get_notAfter` returns a string in the format of an ASN1
Rick Dean38a05c82009-07-18 01:41:30 -05001363 GENERALIZEDTIME even for certificates which store it as UTCTIME
1364 internally.
1365 """
1366 cert = load_certificate(FILETYPE_PEM, self.pemData)
Jean-Paul Calderonee890db32010-08-22 16:55:15 -04001367 self.assertEqual(cert.get_notAfter(), b("20170611123658Z"))
Rick Dean38a05c82009-07-18 01:41:30 -05001368
1369
Jean-Paul Calderoneccf9d482010-07-30 22:36:42 -04001370 def test_gmtime_adj_notBefore_wrong_args(self):
1371 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001372 :py:obj:`X509Type.gmtime_adj_notBefore` raises :py:obj:`TypeError` if called with the
1373 wrong number of arguments or a non-:py:obj:`int` argument.
Jean-Paul Calderoneccf9d482010-07-30 22:36:42 -04001374 """
1375 cert = X509()
1376 self.assertRaises(TypeError, cert.gmtime_adj_notBefore)
1377 self.assertRaises(TypeError, cert.gmtime_adj_notBefore, None)
1378 self.assertRaises(TypeError, cert.gmtime_adj_notBefore, 123, None)
1379
1380
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001381 def test_gmtime_adj_notBefore(self):
1382 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001383 :py:obj:`X509Type.gmtime_adj_notBefore` changes the not-before timestamp to be
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001384 the current time plus the number of seconds passed in.
1385 """
1386 cert = load_certificate(FILETYPE_PEM, self.pemData)
1387 now = datetime.utcnow() + timedelta(seconds=100)
1388 cert.gmtime_adj_notBefore(100)
Jean-Paul Calderonee890db32010-08-22 16:55:15 -04001389 self.assertEqual(cert.get_notBefore(), b(now.strftime("%Y%m%d%H%M%SZ")))
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001390
1391
Jean-Paul Calderoneccf9d482010-07-30 22:36:42 -04001392 def test_gmtime_adj_notAfter_wrong_args(self):
1393 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001394 :py:obj:`X509Type.gmtime_adj_notAfter` raises :py:obj:`TypeError` if called with the
1395 wrong number of arguments or a non-:py:obj:`int` argument.
Jean-Paul Calderoneccf9d482010-07-30 22:36:42 -04001396 """
1397 cert = X509()
1398 self.assertRaises(TypeError, cert.gmtime_adj_notAfter)
1399 self.assertRaises(TypeError, cert.gmtime_adj_notAfter, None)
1400 self.assertRaises(TypeError, cert.gmtime_adj_notAfter, 123, None)
1401
1402
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001403 def test_gmtime_adj_notAfter(self):
1404 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001405 :py:obj:`X509Type.gmtime_adj_notAfter` changes the not-after timestamp to be
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001406 the current time plus the number of seconds passed in.
1407 """
1408 cert = load_certificate(FILETYPE_PEM, self.pemData)
1409 now = datetime.utcnow() + timedelta(seconds=100)
1410 cert.gmtime_adj_notAfter(100)
Jean-Paul Calderonee890db32010-08-22 16:55:15 -04001411 self.assertEqual(cert.get_notAfter(), b(now.strftime("%Y%m%d%H%M%SZ")))
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001412
1413
Jean-Paul Calderoneccf9d482010-07-30 22:36:42 -04001414 def test_has_expired_wrong_args(self):
1415 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001416 :py:obj:`X509Type.has_expired` raises :py:obj:`TypeError` if called with any
Jean-Paul Calderoneccf9d482010-07-30 22:36:42 -04001417 arguments.
1418 """
1419 cert = X509()
1420 self.assertRaises(TypeError, cert.has_expired, None)
1421
1422
1423 def test_has_expired(self):
1424 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001425 :py:obj:`X509Type.has_expired` returns :py:obj:`True` if the certificate's not-after
Jean-Paul Calderoneccf9d482010-07-30 22:36:42 -04001426 time is in the past.
1427 """
1428 cert = X509()
1429 cert.gmtime_adj_notAfter(-1)
1430 self.assertTrue(cert.has_expired())
1431
1432
1433 def test_has_not_expired(self):
1434 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001435 :py:obj:`X509Type.has_expired` returns :py:obj:`False` if the certificate's not-after
Jean-Paul Calderoneccf9d482010-07-30 22:36:42 -04001436 time is in the future.
1437 """
1438 cert = X509()
1439 cert.gmtime_adj_notAfter(2)
1440 self.assertFalse(cert.has_expired())
1441
1442
Rick Dean38a05c82009-07-18 01:41:30 -05001443 def test_digest(self):
1444 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001445 :py:obj:`X509.digest` returns a string giving ":"-separated hex-encoded words
Rick Dean38a05c82009-07-18 01:41:30 -05001446 of the digest of the certificate.
1447 """
1448 cert = X509()
1449 self.assertEqual(
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05001450 # This is MD5 instead of GOOD_DIGEST because the digest algorithm
1451 # actually matters to the assertion (ie, another arbitrary, good
1452 # digest will not product the same digest).
1453 cert.digest("MD5"),
Jean-Paul Calderonee890db32010-08-22 16:55:15 -04001454 b("A8:EB:07:F8:53:25:0A:F2:56:05:C5:A5:C4:C4:C7:15"))
Rick Dean38a05c82009-07-18 01:41:30 -05001455
1456
Jean-Paul Calderone83a593d2011-04-01 17:45:07 -04001457 def _extcert(self, pkey, extensions):
1458 cert = X509()
1459 cert.set_pubkey(pkey)
1460 cert.get_subject().commonName = "Unit Tests"
1461 cert.get_issuer().commonName = "Unit Tests"
1462 when = b(datetime.now().strftime("%Y%m%d%H%M%SZ"))
1463 cert.set_notBefore(when)
1464 cert.set_notAfter(when)
1465
1466 cert.add_extensions(extensions)
1467 return load_certificate(
1468 FILETYPE_PEM, dump_certificate(FILETYPE_PEM, cert))
1469
1470
Roland Hedberg7e4930e2008-04-22 22:58:50 +02001471 def test_extension_count(self):
1472 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001473 :py:obj:`X509.get_extension_count` returns the number of extensions that are
Jean-Paul Calderonef7b3ee62011-04-01 17:36:24 -04001474 present in the certificate.
Roland Hedberg7e4930e2008-04-22 22:58:50 +02001475 """
Jean-Paul Calderonef7b3ee62011-04-01 17:36:24 -04001476 pkey = load_privatekey(FILETYPE_PEM, client_key_pem)
Jean-Paul Calderone90abbc02011-04-06 18:20:22 -04001477 ca = X509Extension(b('basicConstraints'), True, b('CA:FALSE'))
1478 key = X509Extension(b('keyUsage'), True, b('digitalSignature'))
Jean-Paul Calderonef7b3ee62011-04-01 17:36:24 -04001479 subjectAltName = X509Extension(
Jean-Paul Calderone90abbc02011-04-06 18:20:22 -04001480 b('subjectAltName'), True, b('DNS:example.com'))
Jean-Paul Calderonef7b3ee62011-04-01 17:36:24 -04001481
1482 # Try a certificate with no extensions at all.
Jean-Paul Calderone83a593d2011-04-01 17:45:07 -04001483 c = self._extcert(pkey, [])
Jean-Paul Calderonef7b3ee62011-04-01 17:36:24 -04001484 self.assertEqual(c.get_extension_count(), 0)
1485
1486 # And a certificate with one
Jean-Paul Calderone83a593d2011-04-01 17:45:07 -04001487 c = self._extcert(pkey, [ca])
Jean-Paul Calderonef7b3ee62011-04-01 17:36:24 -04001488 self.assertEqual(c.get_extension_count(), 1)
1489
1490 # And a certificate with several
Jean-Paul Calderone83a593d2011-04-01 17:45:07 -04001491 c = self._extcert(pkey, [ca, key, subjectAltName])
Jean-Paul Calderonef7b3ee62011-04-01 17:36:24 -04001492 self.assertEqual(c.get_extension_count(), 3)
Roland Hedberg7e4930e2008-04-22 22:58:50 +02001493
1494
Jean-Paul Calderone83a593d2011-04-01 17:45:07 -04001495 def test_get_extension(self):
1496 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001497 :py:obj:`X509.get_extension` takes an integer and returns an :py:obj:`X509Extension`
Jean-Paul Calderone83a593d2011-04-01 17:45:07 -04001498 corresponding to the extension at that index.
1499 """
1500 pkey = load_privatekey(FILETYPE_PEM, client_key_pem)
Jean-Paul Calderone90abbc02011-04-06 18:20:22 -04001501 ca = X509Extension(b('basicConstraints'), True, b('CA:FALSE'))
1502 key = X509Extension(b('keyUsage'), True, b('digitalSignature'))
Jean-Paul Calderone83a593d2011-04-01 17:45:07 -04001503 subjectAltName = X509Extension(
Jean-Paul Calderone90abbc02011-04-06 18:20:22 -04001504 b('subjectAltName'), False, b('DNS:example.com'))
Jean-Paul Calderone83a593d2011-04-01 17:45:07 -04001505
1506 cert = self._extcert(pkey, [ca, key, subjectAltName])
1507
1508 ext = cert.get_extension(0)
1509 self.assertTrue(isinstance(ext, X509Extension))
1510 self.assertTrue(ext.get_critical())
Jean-Paul Calderone90abbc02011-04-06 18:20:22 -04001511 self.assertEqual(ext.get_short_name(), b('basicConstraints'))
Jean-Paul Calderone83a593d2011-04-01 17:45:07 -04001512
1513 ext = cert.get_extension(1)
1514 self.assertTrue(isinstance(ext, X509Extension))
1515 self.assertTrue(ext.get_critical())
Jean-Paul Calderone90abbc02011-04-06 18:20:22 -04001516 self.assertEqual(ext.get_short_name(), b('keyUsage'))
Jean-Paul Calderone83a593d2011-04-01 17:45:07 -04001517
1518 ext = cert.get_extension(2)
1519 self.assertTrue(isinstance(ext, X509Extension))
1520 self.assertFalse(ext.get_critical())
Jean-Paul Calderone90abbc02011-04-06 18:20:22 -04001521 self.assertEqual(ext.get_short_name(), b('subjectAltName'))
Jean-Paul Calderone83a593d2011-04-01 17:45:07 -04001522
1523 self.assertRaises(IndexError, cert.get_extension, -1)
1524 self.assertRaises(IndexError, cert.get_extension, 4)
1525 self.assertRaises(TypeError, cert.get_extension, "hello")
1526
1527
Jean-Paul Calderone4bf75c62013-08-23 15:39:53 -04001528 def test_nullbyte_subjectAltName(self):
Jean-Paul Calderoneff83cdd2013-08-12 18:05:51 -04001529 """
Jean-Paul Calderone9af07b02013-08-23 16:07:31 -04001530 The fields of a `subjectAltName` extension on an X509 may contain NUL
Jean-Paul Calderone4bf75c62013-08-23 15:39:53 -04001531 bytes and this value is reflected in the string representation of the
1532 extension object.
Jean-Paul Calderoneff83cdd2013-08-12 18:05:51 -04001533 """
Jean-Paul Calderone4bf75c62013-08-23 15:39:53 -04001534 cert = load_certificate(FILETYPE_PEM, nulbyteSubjectAltNamePEM)
Jean-Paul Calderoneff83cdd2013-08-12 18:05:51 -04001535
1536 ext = cert.get_extension(3)
1537 self.assertEqual(ext.get_short_name(), b('subjectAltName'))
Jean-Paul Calderone4bf75c62013-08-23 15:39:53 -04001538 self.assertEqual(
1539 b("DNS:altnull.python.org\x00example.com, "
1540 "email:null@python.org\x00user@example.org, "
1541 "URI:http://null.python.org\x00http://example.org, "
1542 "IP Address:192.0.2.1, IP Address:2001:DB8:0:0:0:0:0:1\n"),
1543 b(str(ext)))
1544
Rick Dean38a05c82009-07-18 01:41:30 -05001545
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -04001546 def test_invalid_digest_algorithm(self):
1547 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001548 :py:obj:`X509.digest` raises :py:obj:`ValueError` if called with an unrecognized hash
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -04001549 algorithm.
1550 """
1551 cert = X509()
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05001552 self.assertRaises(ValueError, cert.digest, BAD_DIGEST)
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -04001553
1554
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001555 def test_get_subject_wrong_args(self):
1556 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001557 :py:obj:`X509.get_subject` raises :py:obj:`TypeError` if called with any arguments.
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001558 """
1559 cert = X509()
1560 self.assertRaises(TypeError, cert.get_subject, None)
1561
1562
1563 def test_get_subject(self):
1564 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001565 :py:obj:`X509.get_subject` returns an :py:obj:`X509Name` instance.
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001566 """
1567 cert = load_certificate(FILETYPE_PEM, self.pemData)
1568 subj = cert.get_subject()
1569 self.assertTrue(isinstance(subj, X509Name))
1570 self.assertEquals(
1571 subj.get_components(),
Jean-Paul Calderonedc3275f2010-08-22 17:04:09 -04001572 [(b('C'), b('US')), (b('ST'), b('IL')), (b('L'), b('Chicago')),
1573 (b('O'), b('Testing')), (b('CN'), b('Testing Root CA'))])
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001574
1575
1576 def test_set_subject_wrong_args(self):
1577 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001578 :py:obj:`X509.set_subject` raises a :py:obj:`TypeError` if called with the wrong
1579 number of arguments or an argument not of type :py:obj:`X509Name`.
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001580 """
1581 cert = X509()
1582 self.assertRaises(TypeError, cert.set_subject)
1583 self.assertRaises(TypeError, cert.set_subject, None)
1584 self.assertRaises(TypeError, cert.set_subject, cert.get_subject(), None)
1585
1586
1587 def test_set_subject(self):
1588 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001589 :py:obj:`X509.set_subject` changes the subject of the certificate to the one
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001590 passed in.
1591 """
1592 cert = X509()
1593 name = cert.get_subject()
1594 name.C = 'AU'
1595 name.O = 'Unit Tests'
1596 cert.set_subject(name)
1597 self.assertEquals(
1598 cert.get_subject().get_components(),
Jean-Paul Calderonedc3275f2010-08-22 17:04:09 -04001599 [(b('C'), b('AU')), (b('O'), b('Unit Tests'))])
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001600
1601
1602 def test_get_issuer_wrong_args(self):
1603 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001604 :py:obj:`X509.get_issuer` raises :py:obj:`TypeError` if called with any arguments.
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001605 """
1606 cert = X509()
1607 self.assertRaises(TypeError, cert.get_issuer, None)
1608
1609
1610 def test_get_issuer(self):
1611 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001612 :py:obj:`X509.get_issuer` returns an :py:obj:`X509Name` instance.
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001613 """
1614 cert = load_certificate(FILETYPE_PEM, self.pemData)
1615 subj = cert.get_issuer()
1616 self.assertTrue(isinstance(subj, X509Name))
Jean-Paul Calderone30a4cb32010-08-11 23:54:12 -04001617 comp = subj.get_components()
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001618 self.assertEquals(
Jean-Paul Calderone30a4cb32010-08-11 23:54:12 -04001619 comp,
Jean-Paul Calderonedc3275f2010-08-22 17:04:09 -04001620 [(b('C'), b('US')), (b('ST'), b('IL')), (b('L'), b('Chicago')),
1621 (b('O'), b('Testing')), (b('CN'), b('Testing Root CA'))])
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001622
1623
1624 def test_set_issuer_wrong_args(self):
1625 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001626 :py:obj:`X509.set_issuer` raises a :py:obj:`TypeError` if called with the wrong
1627 number of arguments or an argument not of type :py:obj:`X509Name`.
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001628 """
1629 cert = X509()
1630 self.assertRaises(TypeError, cert.set_issuer)
1631 self.assertRaises(TypeError, cert.set_issuer, None)
1632 self.assertRaises(TypeError, cert.set_issuer, cert.get_issuer(), None)
1633
1634
1635 def test_set_issuer(self):
1636 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001637 :py:obj:`X509.set_issuer` changes the issuer of the certificate to the one
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001638 passed in.
1639 """
1640 cert = X509()
1641 name = cert.get_issuer()
1642 name.C = 'AU'
1643 name.O = 'Unit Tests'
1644 cert.set_issuer(name)
1645 self.assertEquals(
1646 cert.get_issuer().get_components(),
Jean-Paul Calderonedc3275f2010-08-22 17:04:09 -04001647 [(b('C'), b('AU')), (b('O'), b('Unit Tests'))])
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001648
1649
1650 def test_get_pubkey_uninitialized(self):
1651 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001652 When called on a certificate with no public key, :py:obj:`X509.get_pubkey`
1653 raises :py:obj:`OpenSSL.crypto.Error`.
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001654 """
1655 cert = X509()
1656 self.assertRaises(Error, cert.get_pubkey)
1657
1658
Jean-Paul Calderoneccf9d482010-07-30 22:36:42 -04001659 def test_subject_name_hash_wrong_args(self):
1660 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001661 :py:obj:`X509.subject_name_hash` raises :py:obj:`TypeError` if called with any
Jean-Paul Calderoneccf9d482010-07-30 22:36:42 -04001662 arguments.
1663 """
1664 cert = X509()
1665 self.assertRaises(TypeError, cert.subject_name_hash, None)
1666
1667
1668 def test_subject_name_hash(self):
1669 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001670 :py:obj:`X509.subject_name_hash` returns the hash of the certificate's subject
Jean-Paul Calderoneccf9d482010-07-30 22:36:42 -04001671 name.
1672 """
1673 cert = load_certificate(FILETYPE_PEM, self.pemData)
Jean-Paul Calderone060a57e2011-05-04 18:02:49 -04001674 self.assertIn(
Jean-Paul Calderone2755fa52011-05-18 19:42:10 -04001675 cert.subject_name_hash(),
Jean-Paul Calderone060a57e2011-05-04 18:02:49 -04001676 [3350047874, # OpenSSL 0.9.8, MD5
1677 3278919224, # OpenSSL 1.0.0, SHA1
1678 ])
Jean-Paul Calderoneccf9d482010-07-30 22:36:42 -04001679
1680
Jean-Paul Calderone2755fa52011-05-18 19:42:10 -04001681 def test_get_signature_algorithm(self):
1682 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001683 :py:obj:`X509Type.get_signature_algorithm` returns a string which means
Jean-Paul Calderone2755fa52011-05-18 19:42:10 -04001684 the algorithm used to sign the certificate.
1685 """
1686 cert = load_certificate(FILETYPE_PEM, self.pemData)
Jean-Paul Calderone5d8e4052011-05-19 17:51:43 -04001687 self.assertEqual(
1688 b("sha1WithRSAEncryption"), cert.get_signature_algorithm())
Jean-Paul Calderone2755fa52011-05-18 19:42:10 -04001689
1690
1691 def test_get_undefined_signature_algorithm(self):
Jean-Paul Calderone5d8e4052011-05-19 17:51:43 -04001692 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001693 :py:obj:`X509Type.get_signature_algorithm` raises :py:obj:`ValueError` if the
Jean-Paul Calderone5d8e4052011-05-19 17:51:43 -04001694 signature algorithm is undefined or unknown.
1695 """
1696 # This certificate has been modified to indicate a bogus OID in the
1697 # signature algorithm field so that OpenSSL does not recognize it.
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05001698 certPEM = b("""\
Jean-Paul Calderone2755fa52011-05-18 19:42:10 -04001699-----BEGIN CERTIFICATE-----
1700MIIC/zCCAmigAwIBAgIBATAGBgJ8BQUAMHsxCzAJBgNVBAYTAlNHMREwDwYDVQQK
1701EwhNMkNyeXB0bzEUMBIGA1UECxMLTTJDcnlwdG8gQ0ExJDAiBgNVBAMTG00yQ3J5
1702cHRvIENlcnRpZmljYXRlIE1hc3RlcjEdMBsGCSqGSIb3DQEJARYObmdwc0Bwb3N0
1703MS5jb20wHhcNMDAwOTEwMDk1MTMwWhcNMDIwOTEwMDk1MTMwWjBTMQswCQYDVQQG
1704EwJTRzERMA8GA1UEChMITTJDcnlwdG8xEjAQBgNVBAMTCWxvY2FsaG9zdDEdMBsG
1705CSqGSIb3DQEJARYObmdwc0Bwb3N0MS5jb20wXDANBgkqhkiG9w0BAQEFAANLADBI
1706AkEArL57d26W9fNXvOhNlZzlPOACmvwOZ5AdNgLzJ1/MfsQQJ7hHVeHmTAjM664V
1707+fXvwUGJLziCeBo1ysWLRnl8CQIDAQABo4IBBDCCAQAwCQYDVR0TBAIwADAsBglg
1708hkgBhvhCAQ0EHxYdT3BlblNTTCBHZW5lcmF0ZWQgQ2VydGlmaWNhdGUwHQYDVR0O
1709BBYEFM+EgpK+eyZiwFU1aOPSbczbPSpVMIGlBgNVHSMEgZ0wgZqAFPuHI2nrnDqT
1710FeXFvylRT/7tKDgBoX+kfTB7MQswCQYDVQQGEwJTRzERMA8GA1UEChMITTJDcnlw
1711dG8xFDASBgNVBAsTC00yQ3J5cHRvIENBMSQwIgYDVQQDExtNMkNyeXB0byBDZXJ0
1712aWZpY2F0ZSBNYXN0ZXIxHTAbBgkqhkiG9w0BCQEWDm5ncHNAcG9zdDEuY29tggEA
1713MA0GCSqGSIb3DQEBBAUAA4GBADv8KpPo+gfJxN2ERK1Y1l17sz/ZhzoGgm5XCdbx
1714jEY7xKfpQngV599k1xhl11IMqizDwu0855agrckg2MCTmOI9DZzDD77tAYb+Dk0O
1715PEVk0Mk/V0aIsDE9bolfCi/i/QWZ3N8s5nTWMNyBBBmoSliWCm4jkkRZRD0ejgTN
1716tgI5
1717-----END CERTIFICATE-----
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05001718""")
Jean-Paul Calderone2755fa52011-05-18 19:42:10 -04001719 cert = load_certificate(FILETYPE_PEM, certPEM)
1720 self.assertRaises(ValueError, cert.get_signature_algorithm)
Jean-Paul Calderoneccf9d482010-07-30 22:36:42 -04001721
1722
Rick Dean38a05c82009-07-18 01:41:30 -05001723
Jean-Paul Calderonea63714c2013-03-05 17:02:26 -08001724class X509StoreTests(TestCase):
1725 """
1726 Test for :py:obj:`OpenSSL.crypto.X509Store`.
1727 """
1728 def test_type(self):
1729 """
1730 :py:obj:`X509StoreType` is a type object.
1731 """
1732 self.assertIdentical(X509Store, X509StoreType)
1733 self.assertConsistentType(X509Store, 'X509Store')
1734
1735
1736 def test_add_cert_wrong_args(self):
1737 store = X509Store()
1738 self.assertRaises(TypeError, store.add_cert)
1739 self.assertRaises(TypeError, store.add_cert, object())
1740 self.assertRaises(TypeError, store.add_cert, X509(), object())
1741
1742
1743 def test_add_cert(self):
Jean-Paul Calderonee6f32b82013-03-06 10:27:57 -08001744 """
1745 :py:obj:`X509Store.add_cert` adds a :py:obj:`X509` instance to the
1746 certificate store.
1747 """
1748 cert = load_certificate(FILETYPE_PEM, cleartextCertificatePEM)
Jean-Paul Calderonea63714c2013-03-05 17:02:26 -08001749 store = X509Store()
Jean-Paul Calderonee6f32b82013-03-06 10:27:57 -08001750 store.add_cert(cert)
1751
1752
1753 def test_add_cert_rejects_duplicate(self):
1754 """
1755 :py:obj:`X509Store.add_cert` raises :py:obj:`OpenSSL.crypto.Error` if an
1756 attempt is made to add the same certificate to the store more than once.
1757 """
1758 cert = load_certificate(FILETYPE_PEM, cleartextCertificatePEM)
1759 store = X509Store()
1760 store.add_cert(cert)
1761 self.assertRaises(Error, store.add_cert, cert)
Jean-Paul Calderonea63714c2013-03-05 17:02:26 -08001762
1763
1764
Rick Dean623ee362009-07-17 12:22:16 -05001765class PKCS12Tests(TestCase):
1766 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001767 Test for :py:obj:`OpenSSL.crypto.PKCS12` and :py:obj:`OpenSSL.crypto.load_pkcs12`.
Rick Dean623ee362009-07-17 12:22:16 -05001768 """
1769 pemData = cleartextCertificatePEM + cleartextPrivateKeyPEM
1770
Jean-Paul Calderonec3a41f72009-07-25 12:36:02 -04001771 def test_type(self):
1772 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001773 :py:obj:`PKCS12Type` is a type object.
Jean-Paul Calderonec3a41f72009-07-25 12:36:02 -04001774 """
1775 self.assertIdentical(PKCS12, PKCS12Type)
1776 self.assertConsistentType(PKCS12, 'PKCS12')
1777
1778
Rick Deanf94096c2009-07-18 14:23:06 -05001779 def test_empty_construction(self):
Rick Dean38a05c82009-07-18 01:41:30 -05001780 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001781 :py:obj:`PKCS12` returns a new instance of :py:obj:`PKCS12` with no certificate,
Jean-Paul Calderonedc857b52009-07-25 12:25:07 -04001782 private key, CA certificates, or friendly name.
Rick Dean38a05c82009-07-18 01:41:30 -05001783 """
Jean-Paul Calderonea202edb2009-07-25 12:22:12 -04001784 p12 = PKCS12()
Rick Dean623ee362009-07-17 12:22:16 -05001785 self.assertEqual(None, p12.get_certificate())
1786 self.assertEqual(None, p12.get_privatekey())
1787 self.assertEqual(None, p12.get_ca_certificates())
Rick Dean42d69e12009-07-20 11:36:08 -05001788 self.assertEqual(None, p12.get_friendlyname())
Rick Dean623ee362009-07-17 12:22:16 -05001789
Rick Dean38a05c82009-07-18 01:41:30 -05001790
Rick Dean623ee362009-07-17 12:22:16 -05001791 def test_type_errors(self):
Rick Dean38a05c82009-07-18 01:41:30 -05001792 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001793 The :py:obj:`PKCS12` setter functions (:py:obj:`set_certificate`, :py:obj:`set_privatekey`,
1794 :py:obj:`set_ca_certificates`, and :py:obj:`set_friendlyname`) raise :py:obj:`TypeError`
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04001795 when passed objects of types other than those expected.
Rick Dean38a05c82009-07-18 01:41:30 -05001796 """
Jean-Paul Calderonea202edb2009-07-25 12:22:12 -04001797 p12 = PKCS12()
Rick Dean623ee362009-07-17 12:22:16 -05001798 self.assertRaises(TypeError, p12.set_certificate, 3)
Rick Deanf94096c2009-07-18 14:23:06 -05001799 self.assertRaises(TypeError, p12.set_certificate, PKey())
1800 self.assertRaises(TypeError, p12.set_certificate, X509)
Rick Dean623ee362009-07-17 12:22:16 -05001801 self.assertRaises(TypeError, p12.set_privatekey, 3)
Rick Deanf94096c2009-07-18 14:23:06 -05001802 self.assertRaises(TypeError, p12.set_privatekey, 'legbone')
1803 self.assertRaises(TypeError, p12.set_privatekey, X509())
Rick Dean623ee362009-07-17 12:22:16 -05001804 self.assertRaises(TypeError, p12.set_ca_certificates, 3)
1805 self.assertRaises(TypeError, p12.set_ca_certificates, X509())
1806 self.assertRaises(TypeError, p12.set_ca_certificates, (3, 4))
Rick Deanf94096c2009-07-18 14:23:06 -05001807 self.assertRaises(TypeError, p12.set_ca_certificates, ( PKey(), ))
Rick Dean42d69e12009-07-20 11:36:08 -05001808 self.assertRaises(TypeError, p12.set_friendlyname, 6)
1809 self.assertRaises(TypeError, p12.set_friendlyname, ('foo', 'bar'))
Rick Dean623ee362009-07-17 12:22:16 -05001810
Rick Dean38a05c82009-07-18 01:41:30 -05001811
Rick Dean623ee362009-07-17 12:22:16 -05001812 def test_key_only(self):
1813 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001814 A :py:obj:`PKCS12` with only a private key can be exported using
1815 :py:obj:`PKCS12.export` and loaded again using :py:obj:`load_pkcs12`.
Rick Dean623ee362009-07-17 12:22:16 -05001816 """
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05001817 passwd = b"blah"
Rick Dean623ee362009-07-17 12:22:16 -05001818 p12 = PKCS12()
Jean-Paul Calderonea202edb2009-07-25 12:22:12 -04001819 pkey = load_privatekey(FILETYPE_PEM, cleartextPrivateKeyPEM)
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04001820 p12.set_privatekey(pkey)
Rick Dean623ee362009-07-17 12:22:16 -05001821 self.assertEqual(None, p12.get_certificate())
1822 self.assertEqual(pkey, p12.get_privatekey())
Rick Dean321a0512009-08-13 17:21:29 -05001823 try:
1824 dumped_p12 = p12.export(passphrase=passwd, iter=2, maciter=3)
1825 except Error:
1826 # Some versions of OpenSSL will throw an exception
1827 # for this nearly useless PKCS12 we tried to generate:
1828 # [('PKCS12 routines', 'PKCS12_create', 'invalid null argument')]
1829 return
Rick Dean623ee362009-07-17 12:22:16 -05001830 p12 = load_pkcs12(dumped_p12, passwd)
1831 self.assertEqual(None, p12.get_ca_certificates())
1832 self.assertEqual(None, p12.get_certificate())
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04001833
1834 # OpenSSL fails to bring the key back to us. So sad. Perhaps in the
1835 # future this will be improved.
1836 self.assertTrue(isinstance(p12.get_privatekey(), (PKey, type(None))))
Rick Dean623ee362009-07-17 12:22:16 -05001837
Rick Dean38a05c82009-07-18 01:41:30 -05001838
Rick Dean623ee362009-07-17 12:22:16 -05001839 def test_cert_only(self):
1840 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001841 A :py:obj:`PKCS12` with only a certificate can be exported using
1842 :py:obj:`PKCS12.export` and loaded again using :py:obj:`load_pkcs12`.
Rick Dean623ee362009-07-17 12:22:16 -05001843 """
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05001844 passwd = b"blah"
Rick Dean623ee362009-07-17 12:22:16 -05001845 p12 = PKCS12()
Jean-Paul Calderonea202edb2009-07-25 12:22:12 -04001846 cert = load_certificate(FILETYPE_PEM, cleartextCertificatePEM)
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04001847 p12.set_certificate(cert)
Rick Dean623ee362009-07-17 12:22:16 -05001848 self.assertEqual(cert, p12.get_certificate())
1849 self.assertEqual(None, p12.get_privatekey())
Rick Dean321a0512009-08-13 17:21:29 -05001850 try:
1851 dumped_p12 = p12.export(passphrase=passwd, iter=2, maciter=3)
1852 except Error:
1853 # Some versions of OpenSSL will throw an exception
1854 # for this nearly useless PKCS12 we tried to generate:
1855 # [('PKCS12 routines', 'PKCS12_create', 'invalid null argument')]
1856 return
Rick Dean623ee362009-07-17 12:22:16 -05001857 p12 = load_pkcs12(dumped_p12, passwd)
1858 self.assertEqual(None, p12.get_privatekey())
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04001859
1860 # OpenSSL fails to bring the cert back to us. Groany mcgroan.
1861 self.assertTrue(isinstance(p12.get_certificate(), (X509, type(None))))
1862
1863 # Oh ho. It puts the certificate into the ca certificates list, in
1864 # fact. Totally bogus, I would think. Nevertheless, let's exploit
1865 # that to check to see if it reconstructed the certificate we expected
1866 # it to. At some point, hopefully this will change so that
1867 # p12.get_certificate() is actually what returns the loaded
1868 # certificate.
1869 self.assertEqual(
1870 cleartextCertificatePEM,
1871 dump_certificate(FILETYPE_PEM, p12.get_ca_certificates()[0]))
Rick Dean623ee362009-07-17 12:22:16 -05001872
1873
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04001874 def gen_pkcs12(self, cert_pem=None, key_pem=None, ca_pem=None, friendly_name=None):
Rick Dean623ee362009-07-17 12:22:16 -05001875 """
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04001876 Generate a PKCS12 object with components from PEM. Verify that the set
1877 functions return None.
Rick Dean623ee362009-07-17 12:22:16 -05001878 """
Rick Deanf94096c2009-07-18 14:23:06 -05001879 p12 = PKCS12()
1880 if cert_pem:
1881 ret = p12.set_certificate(load_certificate(FILETYPE_PEM, cert_pem))
1882 self.assertEqual(ret, None)
1883 if key_pem:
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04001884 ret = p12.set_privatekey(load_privatekey(FILETYPE_PEM, key_pem))
Rick Deanf94096c2009-07-18 14:23:06 -05001885 self.assertEqual(ret, None)
1886 if ca_pem:
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04001887 ret = p12.set_ca_certificates((load_certificate(FILETYPE_PEM, ca_pem),))
Rick Deanf94096c2009-07-18 14:23:06 -05001888 self.assertEqual(ret, None)
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04001889 if friendly_name:
1890 ret = p12.set_friendlyname(friendly_name)
Rick Dean42d69e12009-07-20 11:36:08 -05001891 self.assertEqual(ret, None)
Rick Deanf94096c2009-07-18 14:23:06 -05001892 return p12
1893
1894
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05001895 def check_recovery(self, p12_str, key=None, cert=None, ca=None, passwd=b"",
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04001896 extra=()):
Rick Deanf94096c2009-07-18 14:23:06 -05001897 """
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04001898 Use openssl program to confirm three components are recoverable from a
1899 PKCS12 string.
Rick Deanf94096c2009-07-18 14:23:06 -05001900 """
1901 if key:
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04001902 recovered_key = _runopenssl(
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05001903 p12_str, b"pkcs12", b"-nocerts", b"-nodes", b"-passin",
1904 b"pass:" + passwd, *extra)
Rick Deanf94096c2009-07-18 14:23:06 -05001905 self.assertEqual(recovered_key[-len(key):], key)
1906 if cert:
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04001907 recovered_cert = _runopenssl(
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05001908 p12_str, b"pkcs12", b"-clcerts", b"-nodes", b"-passin",
1909 b"pass:" + passwd, b"-nokeys", *extra)
Rick Deanf94096c2009-07-18 14:23:06 -05001910 self.assertEqual(recovered_cert[-len(cert):], cert)
1911 if ca:
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04001912 recovered_cert = _runopenssl(
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05001913 p12_str, b"pkcs12", b"-cacerts", b"-nodes", b"-passin",
1914 b"pass:" + passwd, b"-nokeys", *extra)
Rick Deanf94096c2009-07-18 14:23:06 -05001915 self.assertEqual(recovered_cert[-len(ca):], ca)
1916
1917
1918 def test_load_pkcs12(self):
1919 """
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04001920 A PKCS12 string generated using the openssl command line can be loaded
Jonathan Ballet648875f2011-07-16 14:14:58 +09001921 with :py:obj:`load_pkcs12` and its components extracted and examined.
Rick Deanf94096c2009-07-18 14:23:06 -05001922 """
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05001923 passwd = b"whatever"
Rick Dean623ee362009-07-17 12:22:16 -05001924 pem = client_key_pem + client_cert_pem
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04001925 p12_str = _runopenssl(
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05001926 pem, b"pkcs12", b"-export", b"-clcerts", b"-passout", b"pass:" + passwd)
Rick Dean623ee362009-07-17 12:22:16 -05001927 p12 = load_pkcs12(p12_str, passwd)
Jean-Paul Calderonea202edb2009-07-25 12:22:12 -04001928 # verify
Rick Deanf94096c2009-07-18 14:23:06 -05001929 self.assertTrue(isinstance(p12, PKCS12))
Rick Dean623ee362009-07-17 12:22:16 -05001930 cert_pem = dump_certificate(FILETYPE_PEM, p12.get_certificate())
1931 self.assertEqual(cert_pem, client_cert_pem)
1932 key_pem = dump_privatekey(FILETYPE_PEM, p12.get_privatekey())
1933 self.assertEqual(key_pem, client_key_pem)
1934 self.assertEqual(None, p12.get_ca_certificates())
Rick Deanf94096c2009-07-18 14:23:06 -05001935
Rick Deanee568302009-07-24 09:56:29 -05001936
1937 def test_load_pkcs12_garbage(self):
1938 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001939 :py:obj:`load_pkcs12` raises :py:obj:`OpenSSL.crypto.Error` when passed a string
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04001940 which is not a PKCS12 dump.
Rick Deanee568302009-07-24 09:56:29 -05001941 """
1942 passwd = 'whatever'
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05001943 e = self.assertRaises(Error, load_pkcs12, b'fruit loops', passwd)
1944 self.assertEqual( e.args[0][0][0], b'asn1 encoding routines')
Jean-Paul Calderoneadd7bf02010-08-22 17:38:30 -04001945 self.assertEqual( len(e.args[0][0]), 3)
Rick Deanee568302009-07-24 09:56:29 -05001946
1947
Rick Deanf94096c2009-07-18 14:23:06 -05001948 def test_replace(self):
1949 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09001950 :py:obj:`PKCS12.set_certificate` replaces the certificate in a PKCS12 cluster.
1951 :py:obj:`PKCS12.set_privatekey` replaces the private key.
1952 :py:obj:`PKCS12.set_ca_certificates` replaces the CA certificates.
Rick Deanf94096c2009-07-18 14:23:06 -05001953 """
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04001954 p12 = self.gen_pkcs12(client_cert_pem, client_key_pem, root_cert_pem)
1955 p12.set_certificate(load_certificate(FILETYPE_PEM, server_cert_pem))
1956 p12.set_privatekey(load_privatekey(FILETYPE_PEM, server_key_pem))
Jean-Paul Calderonea202edb2009-07-25 12:22:12 -04001957 root_cert = load_certificate(FILETYPE_PEM, root_cert_pem)
Rick Deanf94096c2009-07-18 14:23:06 -05001958 client_cert = load_certificate(FILETYPE_PEM, client_cert_pem)
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04001959 p12.set_ca_certificates([root_cert]) # not a tuple
Rick Dean623ee362009-07-17 12:22:16 -05001960 self.assertEqual(1, len(p12.get_ca_certificates()))
1961 self.assertEqual(root_cert, p12.get_ca_certificates()[0])
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04001962 p12.set_ca_certificates([client_cert, root_cert])
Rick Deanf94096c2009-07-18 14:23:06 -05001963 self.assertEqual(2, len(p12.get_ca_certificates()))
1964 self.assertEqual(client_cert, p12.get_ca_certificates()[0])
1965 self.assertEqual(root_cert, p12.get_ca_certificates()[1])
1966
1967
1968 def test_friendly_name(self):
1969 """
Jean-Paul Calderone64efa2c2011-09-11 10:00:09 -04001970 The *friendlyName* of a PKCS12 can be set and retrieved via
Jonathan Ballet648875f2011-07-16 14:14:58 +09001971 :py:obj:`PKCS12.get_friendlyname` and :py:obj:`PKCS12_set_friendlyname`, and a
1972 :py:obj:`PKCS12` with a friendly name set can be dumped with :py:obj:`PKCS12.export`.
Rick Deanf94096c2009-07-18 14:23:06 -05001973 """
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05001974 passwd = b'Dogmeat[]{}!@#$%^&*()~`?/.,<>-_+=";:'
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04001975 p12 = self.gen_pkcs12(server_cert_pem, server_key_pem, root_cert_pem)
Jean-Paul Calderoneadd7bf02010-08-22 17:38:30 -04001976 for friendly_name in [b('Serverlicious'), None, b('###')]:
Rick Dean42d69e12009-07-20 11:36:08 -05001977 p12.set_friendlyname(friendly_name)
1978 self.assertEqual(p12.get_friendlyname(), friendly_name)
Jean-Paul Calderonea202edb2009-07-25 12:22:12 -04001979 dumped_p12 = p12.export(passphrase=passwd, iter=2, maciter=3)
Rick Dean42d69e12009-07-20 11:36:08 -05001980 reloaded_p12 = load_pkcs12(dumped_p12, passwd)
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04001981 self.assertEqual(
Jean-Paul Calderone9da338d2011-05-04 11:40:54 -04001982 p12.get_friendlyname(), reloaded_p12.get_friendlyname())
Jean-Paul Calderonea202edb2009-07-25 12:22:12 -04001983 # We would use the openssl program to confirm the friendly
1984 # name, but it is not possible. The pkcs12 command
1985 # does not store the friendly name in the cert's
Rick Dean42d69e12009-07-20 11:36:08 -05001986 # alias, which we could then extract.
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04001987 self.check_recovery(
1988 dumped_p12, key=server_key_pem, cert=server_cert_pem,
1989 ca=root_cert_pem, passwd=passwd)
Rick Deanf94096c2009-07-18 14:23:06 -05001990
1991
1992 def test_various_empty_passphrases(self):
1993 """
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04001994 Test that missing, None, and '' passphrases are identical for PKCS12
1995 export.
Rick Deanf94096c2009-07-18 14:23:06 -05001996 """
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04001997 p12 = self.gen_pkcs12(client_cert_pem, client_key_pem, root_cert_pem)
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05001998 passwd = b""
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04001999 dumped_p12_empty = p12.export(iter=2, maciter=0, passphrase=passwd)
2000 dumped_p12_none = p12.export(iter=3, maciter=2, passphrase=None)
2001 dumped_p12_nopw = p12.export(iter=9, maciter=4)
2002 for dumped_p12 in [dumped_p12_empty, dumped_p12_none, dumped_p12_nopw]:
2003 self.check_recovery(
2004 dumped_p12, key=client_key_pem, cert=client_cert_pem,
2005 ca=root_cert_pem, passwd=passwd)
Rick Deanf94096c2009-07-18 14:23:06 -05002006
2007
2008 def test_removing_ca_cert(self):
2009 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002010 Passing :py:obj:`None` to :py:obj:`PKCS12.set_ca_certificates` removes all CA
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04002011 certificates.
Rick Deanf94096c2009-07-18 14:23:06 -05002012 """
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04002013 p12 = self.gen_pkcs12(server_cert_pem, server_key_pem, root_cert_pem)
2014 p12.set_ca_certificates(None)
Rick Dean623ee362009-07-17 12:22:16 -05002015 self.assertEqual(None, p12.get_ca_certificates())
Rick Deanf94096c2009-07-18 14:23:06 -05002016
2017
2018 def test_export_without_mac(self):
2019 """
Jean-Paul Calderone64efa2c2011-09-11 10:00:09 -04002020 Exporting a PKCS12 with a :py:obj:`maciter` of ``-1`` excludes the MAC
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04002021 entirely.
Rick Deanf94096c2009-07-18 14:23:06 -05002022 """
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05002023 passwd = b"Lake Michigan"
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04002024 p12 = self.gen_pkcs12(server_cert_pem, server_key_pem, root_cert_pem)
Rick Dean623ee362009-07-17 12:22:16 -05002025 dumped_p12 = p12.export(maciter=-1, passphrase=passwd, iter=2)
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04002026 self.check_recovery(
2027 dumped_p12, key=server_key_pem, cert=server_cert_pem,
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05002028 passwd=passwd, extra=(b"-nomacver",))
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04002029
2030
2031 def test_load_without_mac(self):
2032 """
2033 Loading a PKCS12 without a MAC does something other than crash.
2034 """
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05002035 passwd = b"Lake Michigan"
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04002036 p12 = self.gen_pkcs12(server_cert_pem, server_key_pem, root_cert_pem)
2037 dumped_p12 = p12.export(maciter=-1, passphrase=passwd, iter=2)
Rick Dean321a0512009-08-13 17:21:29 -05002038 try:
2039 recovered_p12 = load_pkcs12(dumped_p12, passwd)
2040 # The person who generated this PCKS12 should be flogged,
2041 # or better yet we should have a means to determine
2042 # whether a PCKS12 had a MAC that was verified.
2043 # Anyway, libopenssl chooses to allow it, so the
2044 # pyopenssl binding does as well.
2045 self.assertTrue(isinstance(recovered_p12, PKCS12))
2046 except Error:
2047 # Failing here with an exception is preferred as some openssl
2048 # versions do.
2049 pass
Rick Dean623ee362009-07-17 12:22:16 -05002050
Jean-Paul Calderone38a646d2008-03-25 15:16:18 -04002051
Rick Dean25bcc1f2009-07-20 11:53:13 -05002052 def test_zero_len_list_for_ca(self):
2053 """
Jean-Paul Calderoneda1ffa72009-07-25 21:24:34 -04002054 A PKCS12 with an empty CA certificates list can be exported.
Rick Dean25bcc1f2009-07-20 11:53:13 -05002055 """
2056 passwd = 'Hobie 18'
Jean-Paul Calderoneda1ffa72009-07-25 21:24:34 -04002057 p12 = self.gen_pkcs12(server_cert_pem, server_key_pem)
Jean-Paul Calderonef6745b32013-03-01 15:08:46 -08002058 # p12.set_ca_certificates([])
2059 # self.assertEqual((), p12.get_ca_certificates())
2060 # dumped_p12 = p12.export(passphrase=passwd, iter=3)
2061 # self.check_recovery(
2062 # dumped_p12, key=server_key_pem, cert=server_cert_pem,
2063 # passwd=passwd)
Rick Dean25bcc1f2009-07-20 11:53:13 -05002064
2065
Rick Deanf94096c2009-07-18 14:23:06 -05002066 def test_export_without_args(self):
Jean-Paul Calderone38a646d2008-03-25 15:16:18 -04002067 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002068 All the arguments to :py:obj:`PKCS12.export` are optional.
Jean-Paul Calderone38a646d2008-03-25 15:16:18 -04002069 """
Jean-Paul Calderoneda1ffa72009-07-25 21:24:34 -04002070 p12 = self.gen_pkcs12(server_cert_pem, server_key_pem, root_cert_pem)
Rick Deanf94096c2009-07-18 14:23:06 -05002071 dumped_p12 = p12.export() # no args
Jean-Paul Calderoneda1ffa72009-07-25 21:24:34 -04002072 self.check_recovery(
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05002073 dumped_p12, key=server_key_pem, cert=server_cert_pem, passwd=b"")
Rick Deanf94096c2009-07-18 14:23:06 -05002074
2075
2076 def test_key_cert_mismatch(self):
2077 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002078 :py:obj:`PKCS12.export` raises an exception when a key and certificate
Jean-Paul Calderoneda1ffa72009-07-25 21:24:34 -04002079 mismatch.
Rick Deanf94096c2009-07-18 14:23:06 -05002080 """
Jean-Paul Calderoneda1ffa72009-07-25 21:24:34 -04002081 p12 = self.gen_pkcs12(server_cert_pem, client_key_pem, root_cert_pem)
2082 self.assertRaises(Error, p12.export)
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002083
2084
2085
Jean-Paul Calderone7b874db2009-08-27 12:32:47 -04002086# These quoting functions taken directly from Twisted's twisted.python.win32.
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05002087_cmdLineQuoteRe = re.compile(br'(\\*)"')
2088_cmdLineQuoteRe2 = re.compile(br'(\\+)\Z')
Jean-Paul Calderone7b874db2009-08-27 12:32:47 -04002089def cmdLineQuote(s):
2090 """
2091 Internal method for quoting a single command-line argument.
2092
Jean-Paul Calderone64efa2c2011-09-11 10:00:09 -04002093 See http://www.perlmonks.org/?node_id=764004
2094
Jonathan Ballet648875f2011-07-16 14:14:58 +09002095 :type: :py:obj:`str`
Jonathan Ballet78b92a22011-07-16 08:07:26 +09002096 :param s: A single unquoted string to quote for something that is expecting
Jean-Paul Calderone7b874db2009-08-27 12:32:47 -04002097 cmd.exe-style quoting
2098
Jonathan Ballet648875f2011-07-16 14:14:58 +09002099 :rtype: :py:obj:`str`
Jonathan Ballet78b92a22011-07-16 08:07:26 +09002100 :return: A cmd.exe-style quoted string
Jean-Paul Calderone7b874db2009-08-27 12:32:47 -04002101 """
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05002102 s = _cmdLineQuoteRe2.sub(br"\1\1", _cmdLineQuoteRe.sub(br'\1\1\\"', s))
2103 return b'"' + s + b'"'
Jean-Paul Calderone7b874db2009-08-27 12:32:47 -04002104
2105
2106
2107def quoteArguments(arguments):
2108 """
2109 Quote an iterable of command-line arguments for passing to CreateProcess or
Jonathan Ballet648875f2011-07-16 14:14:58 +09002110 a similar API. This allows the list passed to :py:obj:`reactor.spawnProcess` to
2111 match the child process's :py:obj:`sys.argv` properly.
Jean-Paul Calderone7b874db2009-08-27 12:32:47 -04002112
Jonathan Ballet648875f2011-07-16 14:14:58 +09002113 :type arguments: :py:obj:`iterable` of :py:obj:`str`
Jonathan Ballet78b92a22011-07-16 08:07:26 +09002114 :param arguments: An iterable of unquoted arguments to quote
Jean-Paul Calderone7b874db2009-08-27 12:32:47 -04002115
Jonathan Ballet648875f2011-07-16 14:14:58 +09002116 :rtype: :py:obj:`str`
2117 :return: A space-delimited string containing quoted versions of :py:obj:`arguments`
Jean-Paul Calderone7b874db2009-08-27 12:32:47 -04002118 """
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05002119 return b' '.join(map(cmdLineQuote, arguments))
Jean-Paul Calderone7b874db2009-08-27 12:32:47 -04002120
2121
Jean-Paul Calderoneb98eae02010-01-30 13:18:04 -05002122
Rick Dean4c9ad612009-07-17 15:05:22 -05002123def _runopenssl(pem, *args):
2124 """
2125 Run the command line openssl tool with the given arguments and write
Rick Dean55d1ce62009-08-13 17:40:24 -05002126 the given PEM to its stdin. Not safe for quotes.
Rick Dean4c9ad612009-07-17 15:05:22 -05002127 """
Jean-Paul Calderone038de952009-08-21 12:16:06 -04002128 if os.name == 'posix':
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05002129 command = b"openssl " + b" ".join([
2130 (b"'" + arg.replace(b"'", b"'\\''") + b"'")
2131 for arg in args])
Rick Dean55d1ce62009-08-13 17:40:24 -05002132 else:
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05002133 command = b"openssl " + quoteArguments(args)
2134 proc = Popen(native(command), shell=True, stdin=PIPE, stdout=PIPE)
Jean-Paul Calderone62ca8da2010-08-11 19:58:08 -04002135 proc.stdin.write(pem)
2136 proc.stdin.close()
Jean-Paul Calderonee82e3252013-03-03 10:14:10 -08002137 output = proc.stdout.read()
2138 proc.stdout.close()
2139 proc.wait()
2140 return output
Rick Dean4c9ad612009-07-17 15:05:22 -05002141
2142
2143
Jean-Paul Calderone18808652009-07-05 12:54:05 -04002144class FunctionTests(TestCase):
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002145 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002146 Tests for free-functions in the :py:obj:`OpenSSL.crypto` module.
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002147 """
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -04002148
2149 def test_load_privatekey_invalid_format(self):
2150 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002151 :py:obj:`load_privatekey` raises :py:obj:`ValueError` if passed an unknown filetype.
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -04002152 """
2153 self.assertRaises(ValueError, load_privatekey, 100, root_key_pem)
2154
2155
2156 def test_load_privatekey_invalid_passphrase_type(self):
2157 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002158 :py:obj:`load_privatekey` raises :py:obj:`TypeError` if passed a passphrase that is
Jean-Paul Calderone64efa2c2011-09-11 10:00:09 -04002159 neither a :py:obj:`str` nor a callable.
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -04002160 """
2161 self.assertRaises(
2162 TypeError,
2163 load_privatekey,
2164 FILETYPE_PEM, encryptedPrivateKeyPEMPassphrase, object())
2165
2166
2167 def test_load_privatekey_wrong_args(self):
2168 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002169 :py:obj:`load_privatekey` raises :py:obj:`TypeError` if called with the wrong number
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -04002170 of arguments.
2171 """
2172 self.assertRaises(TypeError, load_privatekey)
2173
2174
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002175 def test_load_privatekey_wrongPassphrase(self):
2176 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002177 :py:obj:`load_privatekey` raises :py:obj:`OpenSSL.crypto.Error` when it is passed an
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002178 encrypted PEM and an incorrect passphrase.
2179 """
2180 self.assertRaises(
2181 Error,
Jean-Paul Calderone5be230f2010-08-22 19:27:58 -04002182 load_privatekey, FILETYPE_PEM, encryptedPrivateKeyPEM, b("quack"))
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002183
2184
Ziga Seilnacht376cf972009-12-22 16:04:10 +01002185 def test_load_privatekey_passphraseWrongType(self):
2186 """
Jean-Paul Calderone5d9d7f12011-09-14 09:48:58 -04002187 :py:obj:`load_privatekey` raises :py:obj:`ValueError` when it is passed a passphrase
Ziga Seilnacht376cf972009-12-22 16:04:10 +01002188 with a private key encoded in a format, that doesn't support
2189 encryption.
2190 """
2191 key = load_privatekey(FILETYPE_PEM, cleartextPrivateKeyPEM)
2192 blob = dump_privatekey(FILETYPE_ASN1, key)
2193 self.assertRaises(ValueError,
2194 load_privatekey, FILETYPE_ASN1, blob, "secret")
2195
2196
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002197 def test_load_privatekey_passphrase(self):
2198 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002199 :py:obj:`load_privatekey` can create a :py:obj:`PKey` object from an encrypted PEM
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002200 string if given the passphrase.
2201 """
2202 key = load_privatekey(
2203 FILETYPE_PEM, encryptedPrivateKeyPEM,
2204 encryptedPrivateKeyPEMPassphrase)
2205 self.assertTrue(isinstance(key, PKeyType))
2206
2207
Ziga Seilnacht6b90a402009-12-22 14:33:47 +01002208 def test_load_privatekey_passphrase_exception(self):
2209 """
2210 If the passphrase callback raises an exception, that exception is raised
2211 by :py:obj:`load_privatekey`.
2212 """
2213 def cb(ignored):
2214 raise ArithmeticError
2215
2216 self.assertRaises(ArithmeticError,
2217 load_privatekey, FILETYPE_PEM, encryptedPrivateKeyPEM, cb)
2218
2219
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002220 def test_load_privatekey_wrongPassphraseCallback(self):
2221 """
Jean-Paul Calderoned440a082011-09-14 11:02:05 -04002222 :py:obj:`load_privatekey` raises :py:obj:`OpenSSL.crypto.Error` when it
2223 is passed an encrypted PEM and a passphrase callback which returns an
2224 incorrect passphrase.
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002225 """
2226 called = []
2227 def cb(*a):
2228 called.append(None)
Jean-Paul Calderoned440a082011-09-14 11:02:05 -04002229 return b("quack")
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002230 self.assertRaises(
2231 Error,
2232 load_privatekey, FILETYPE_PEM, encryptedPrivateKeyPEM, cb)
2233 self.assertTrue(called)
2234
Jean-Paul Calderoneaa6c0692010-09-08 22:43:09 -04002235
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002236 def test_load_privatekey_passphraseCallback(self):
2237 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002238 :py:obj:`load_privatekey` can create a :py:obj:`PKey` object from an encrypted PEM
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002239 string if given a passphrase callback which returns the correct
2240 password.
2241 """
2242 called = []
2243 def cb(writing):
2244 called.append(writing)
2245 return encryptedPrivateKeyPEMPassphrase
2246 key = load_privatekey(FILETYPE_PEM, encryptedPrivateKeyPEM, cb)
2247 self.assertTrue(isinstance(key, PKeyType))
2248 self.assertEqual(called, [False])
2249
2250
Jean-Paul Calderone105cb952011-09-14 10:16:46 -04002251 def test_load_privatekey_passphrase_wrong_return_type(self):
2252 """
2253 :py:obj:`load_privatekey` raises :py:obj:`ValueError` if the passphrase
2254 callback returns something other than a byte string.
2255 """
2256 self.assertRaises(
2257 ValueError,
2258 load_privatekey,
2259 FILETYPE_PEM, encryptedPrivateKeyPEM, lambda *args: 3)
2260
2261
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -04002262 def test_dump_privatekey_wrong_args(self):
2263 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002264 :py:obj:`dump_privatekey` raises :py:obj:`TypeError` if called with the wrong number
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -04002265 of arguments.
2266 """
2267 self.assertRaises(TypeError, dump_privatekey)
Jean-Paul Calderone1eeccd82011-09-14 10:18:52 -04002268 # If cipher name is given, password is required.
2269 self.assertRaises(
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05002270 TypeError, dump_privatekey, FILETYPE_PEM, PKey(), GOOD_CIPHER)
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -04002271
2272
2273 def test_dump_privatekey_unknown_cipher(self):
2274 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002275 :py:obj:`dump_privatekey` raises :py:obj:`ValueError` if called with an unrecognized
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -04002276 cipher name.
2277 """
2278 key = PKey()
2279 key.generate_key(TYPE_RSA, 512)
2280 self.assertRaises(
2281 ValueError, dump_privatekey,
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05002282 FILETYPE_PEM, key, BAD_CIPHER, "passphrase")
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -04002283
2284
2285 def test_dump_privatekey_invalid_passphrase_type(self):
2286 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002287 :py:obj:`dump_privatekey` raises :py:obj:`TypeError` if called with a passphrase which
2288 is neither a :py:obj:`str` nor a callable.
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -04002289 """
2290 key = PKey()
2291 key.generate_key(TYPE_RSA, 512)
2292 self.assertRaises(
2293 TypeError,
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05002294 dump_privatekey, FILETYPE_PEM, key, GOOD_CIPHER, object())
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -04002295
2296
2297 def test_dump_privatekey_invalid_filetype(self):
2298 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002299 :py:obj:`dump_privatekey` raises :py:obj:`ValueError` if called with an unrecognized
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -04002300 filetype.
2301 """
2302 key = PKey()
2303 key.generate_key(TYPE_RSA, 512)
2304 self.assertRaises(ValueError, dump_privatekey, 100, key)
2305
2306
Ziga Seilnacht781295a2009-12-22 14:58:01 +01002307 def test_load_privatekey_passphraseCallbackLength(self):
2308 """
Jean-Paul Calderone5d9d7f12011-09-14 09:48:58 -04002309 :py:obj:`crypto.load_privatekey` should raise an error when the passphrase
Ziga Seilnacht781295a2009-12-22 14:58:01 +01002310 provided by the callback is too long, not silently truncate it.
2311 """
2312 def cb(ignored):
2313 return "a" * 1025
2314
2315 self.assertRaises(ValueError,
2316 load_privatekey, FILETYPE_PEM, encryptedPrivateKeyPEM, cb)
2317
2318
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002319 def test_dump_privatekey_passphrase(self):
2320 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002321 :py:obj:`dump_privatekey` writes an encrypted PEM when given a passphrase.
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002322 """
Jean-Paul Calderone5be230f2010-08-22 19:27:58 -04002323 passphrase = b("foo")
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002324 key = load_privatekey(FILETYPE_PEM, cleartextPrivateKeyPEM)
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05002325 pem = dump_privatekey(FILETYPE_PEM, key, GOOD_CIPHER, passphrase)
2326 self.assertTrue(isinstance(pem, binary_type))
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002327 loadedKey = load_privatekey(FILETYPE_PEM, pem, passphrase)
2328 self.assertTrue(isinstance(loadedKey, PKeyType))
2329 self.assertEqual(loadedKey.type(), key.type())
2330 self.assertEqual(loadedKey.bits(), key.bits())
2331
Jean-Paul Calderonef17e4212009-04-01 14:21:40 -04002332
Ziga Seilnacht376cf972009-12-22 16:04:10 +01002333 def test_dump_privatekey_passphraseWrongType(self):
2334 """
Jean-Paul Calderone5d9d7f12011-09-14 09:48:58 -04002335 :py:obj:`dump_privatekey` raises :py:obj:`ValueError` when it is passed a passphrase
Ziga Seilnacht376cf972009-12-22 16:04:10 +01002336 with a private key encoded in a format, that doesn't support
2337 encryption.
2338 """
2339 key = load_privatekey(FILETYPE_PEM, cleartextPrivateKeyPEM)
2340 self.assertRaises(ValueError,
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05002341 dump_privatekey, FILETYPE_ASN1, key, GOOD_CIPHER, "secret")
Ziga Seilnacht376cf972009-12-22 16:04:10 +01002342
2343
Rick Dean5b7b6372009-04-01 11:34:06 -05002344 def test_dump_certificate(self):
2345 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002346 :py:obj:`dump_certificate` writes PEM, DER, and text.
Rick Dean5b7b6372009-04-01 11:34:06 -05002347 """
Jean-Paul Calderonef17e4212009-04-01 14:21:40 -04002348 pemData = cleartextCertificatePEM + cleartextPrivateKeyPEM
Rick Dean5b7b6372009-04-01 11:34:06 -05002349 cert = load_certificate(FILETYPE_PEM, pemData)
2350 dumped_pem = dump_certificate(FILETYPE_PEM, cert)
2351 self.assertEqual(dumped_pem, cleartextCertificatePEM)
2352 dumped_der = dump_certificate(FILETYPE_ASN1, cert)
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05002353 good_der = _runopenssl(dumped_pem, b"x509", b"-outform", b"DER")
Rick Dean5b7b6372009-04-01 11:34:06 -05002354 self.assertEqual(dumped_der, good_der)
2355 cert2 = load_certificate(FILETYPE_ASN1, dumped_der)
2356 dumped_pem2 = dump_certificate(FILETYPE_PEM, cert2)
2357 self.assertEqual(dumped_pem2, cleartextCertificatePEM)
2358 dumped_text = dump_certificate(FILETYPE_TEXT, cert)
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05002359 good_text = _runopenssl(dumped_pem, b"x509", b"-noout", b"-text")
Rick Dean5b7b6372009-04-01 11:34:06 -05002360 self.assertEqual(dumped_text, good_text)
2361
2362
Jean-Paul Calderonee82e3252013-03-03 10:14:10 -08002363 def test_dump_privatekey_pem(self):
Rick Dean5b7b6372009-04-01 11:34:06 -05002364 """
Jean-Paul Calderonee82e3252013-03-03 10:14:10 -08002365 :py:obj:`dump_privatekey` writes a PEM
Rick Dean5b7b6372009-04-01 11:34:06 -05002366 """
2367 key = load_privatekey(FILETYPE_PEM, cleartextPrivateKeyPEM)
Jean-Paul Calderone8e6ce972009-05-13 12:32:49 -04002368 self.assertTrue(key.check())
Rick Dean5b7b6372009-04-01 11:34:06 -05002369 dumped_pem = dump_privatekey(FILETYPE_PEM, key)
2370 self.assertEqual(dumped_pem, cleartextPrivateKeyPEM)
Jean-Paul Calderonee82e3252013-03-03 10:14:10 -08002371
2372
2373 def test_dump_privatekey_asn1(self):
2374 """
2375 :py:obj:`dump_privatekey` writes a DER
2376 """
2377 key = load_privatekey(FILETYPE_PEM, cleartextPrivateKeyPEM)
2378 dumped_pem = dump_privatekey(FILETYPE_PEM, key)
2379
Rick Dean5b7b6372009-04-01 11:34:06 -05002380 dumped_der = dump_privatekey(FILETYPE_ASN1, key)
Jean-Paul Calderonef17e4212009-04-01 14:21:40 -04002381 # XXX This OpenSSL call writes "writing RSA key" to standard out. Sad.
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05002382 good_der = _runopenssl(dumped_pem, b"rsa", b"-outform", b"DER")
Rick Dean5b7b6372009-04-01 11:34:06 -05002383 self.assertEqual(dumped_der, good_der)
2384 key2 = load_privatekey(FILETYPE_ASN1, dumped_der)
2385 dumped_pem2 = dump_privatekey(FILETYPE_PEM, key2)
2386 self.assertEqual(dumped_pem2, cleartextPrivateKeyPEM)
Jean-Paul Calderonee82e3252013-03-03 10:14:10 -08002387
2388
2389 def test_dump_privatekey_text(self):
2390 """
2391 :py:obj:`dump_privatekey` writes a text
2392 """
2393 key = load_privatekey(FILETYPE_PEM, cleartextPrivateKeyPEM)
2394 dumped_pem = dump_privatekey(FILETYPE_PEM, key)
2395
Rick Dean5b7b6372009-04-01 11:34:06 -05002396 dumped_text = dump_privatekey(FILETYPE_TEXT, key)
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05002397 good_text = _runopenssl(dumped_pem, b"rsa", b"-noout", b"-text")
Rick Dean5b7b6372009-04-01 11:34:06 -05002398 self.assertEqual(dumped_text, good_text)
2399
Jean-Paul Calderonef17e4212009-04-01 14:21:40 -04002400
Rick Dean5b7b6372009-04-01 11:34:06 -05002401 def test_dump_certificate_request(self):
2402 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002403 :py:obj:`dump_certificate_request` writes a PEM, DER, and text.
Rick Dean5b7b6372009-04-01 11:34:06 -05002404 """
2405 req = load_certificate_request(FILETYPE_PEM, cleartextCertificateRequestPEM)
2406 dumped_pem = dump_certificate_request(FILETYPE_PEM, req)
2407 self.assertEqual(dumped_pem, cleartextCertificateRequestPEM)
2408 dumped_der = dump_certificate_request(FILETYPE_ASN1, req)
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05002409 good_der = _runopenssl(dumped_pem, b"req", b"-outform", b"DER")
Rick Dean5b7b6372009-04-01 11:34:06 -05002410 self.assertEqual(dumped_der, good_der)
2411 req2 = load_certificate_request(FILETYPE_ASN1, dumped_der)
2412 dumped_pem2 = dump_certificate_request(FILETYPE_PEM, req2)
2413 self.assertEqual(dumped_pem2, cleartextCertificateRequestPEM)
2414 dumped_text = dump_certificate_request(FILETYPE_TEXT, req)
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05002415 good_text = _runopenssl(dumped_pem, b"req", b"-noout", b"-text")
Rick Dean5b7b6372009-04-01 11:34:06 -05002416 self.assertEqual(dumped_text, good_text)
Jean-Paul Calderoneaf43cdf2010-08-03 18:40:52 -04002417 self.assertRaises(ValueError, dump_certificate_request, 100, req)
Rick Dean5b7b6372009-04-01 11:34:06 -05002418
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002419
2420 def test_dump_privatekey_passphraseCallback(self):
2421 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002422 :py:obj:`dump_privatekey` writes an encrypted PEM when given a callback which
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002423 returns the correct passphrase.
2424 """
Jean-Paul Calderone5be230f2010-08-22 19:27:58 -04002425 passphrase = b("foo")
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002426 called = []
2427 def cb(writing):
2428 called.append(writing)
2429 return passphrase
2430 key = load_privatekey(FILETYPE_PEM, cleartextPrivateKeyPEM)
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05002431 pem = dump_privatekey(FILETYPE_PEM, key, GOOD_CIPHER, cb)
2432 self.assertTrue(isinstance(pem, binary_type))
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002433 self.assertEqual(called, [True])
2434 loadedKey = load_privatekey(FILETYPE_PEM, pem, passphrase)
2435 self.assertTrue(isinstance(loadedKey, PKeyType))
2436 self.assertEqual(loadedKey.type(), key.type())
2437 self.assertEqual(loadedKey.bits(), key.bits())
Rick Dean5b7b6372009-04-01 11:34:06 -05002438
2439
Ziga Seilnacht6b90a402009-12-22 14:33:47 +01002440 def test_dump_privatekey_passphrase_exception(self):
2441 """
Jean-Paul Calderone5d9d7f12011-09-14 09:48:58 -04002442 :py:obj:`dump_privatekey` should not overwrite the exception raised
Ziga Seilnacht6b90a402009-12-22 14:33:47 +01002443 by the passphrase callback.
2444 """
2445 def cb(ignored):
2446 raise ArithmeticError
2447
2448 key = load_privatekey(FILETYPE_PEM, cleartextPrivateKeyPEM)
2449 self.assertRaises(ArithmeticError,
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05002450 dump_privatekey, FILETYPE_PEM, key, GOOD_CIPHER, cb)
Ziga Seilnacht6b90a402009-12-22 14:33:47 +01002451
2452
Ziga Seilnacht781295a2009-12-22 14:58:01 +01002453 def test_dump_privatekey_passphraseCallbackLength(self):
2454 """
Jean-Paul Calderone5d9d7f12011-09-14 09:48:58 -04002455 :py:obj:`crypto.dump_privatekey` should raise an error when the passphrase
Ziga Seilnacht781295a2009-12-22 14:58:01 +01002456 provided by the callback is too long, not silently truncate it.
2457 """
2458 def cb(ignored):
2459 return "a" * 1025
2460
2461 key = load_privatekey(FILETYPE_PEM, cleartextPrivateKeyPEM)
2462 self.assertRaises(ValueError,
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05002463 dump_privatekey, FILETYPE_PEM, key, GOOD_CIPHER, cb)
Ziga Seilnacht781295a2009-12-22 14:58:01 +01002464
2465
Jean-Paul Calderonedc138fa2009-06-27 14:32:07 -04002466 def test_load_pkcs7_data(self):
2467 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002468 :py:obj:`load_pkcs7_data` accepts a PKCS#7 string and returns an instance of
2469 :py:obj:`PKCS7Type`.
Jean-Paul Calderonedc138fa2009-06-27 14:32:07 -04002470 """
2471 pkcs7 = load_pkcs7_data(FILETYPE_PEM, pkcs7Data)
2472 self.assertTrue(isinstance(pkcs7, PKCS7Type))
2473
2474
Jean-Paul Calderonee82e3252013-03-03 10:14:10 -08002475 def test_load_pkcs7_data_invalid(self):
2476 """
2477 If the data passed to :py:obj:`load_pkcs7_data` is invalid,
2478 :py:obj:`Error` is raised.
2479 """
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05002480 self.assertRaises(Error, load_pkcs7_data, FILETYPE_PEM, b"foo")
Jean-Paul Calderonee82e3252013-03-03 10:14:10 -08002481
2482
Jean-Paul Calderonedc138fa2009-06-27 14:32:07 -04002483
Jean-Paul Calderone4a68b402013-12-29 16:54:58 -05002484class LoadCertificateTests(TestCase):
2485 """
2486 Tests for :py:obj:`load_certificate_request`.
2487 """
2488 def test_badFileType(self):
2489 """
2490 If the file type passed to :py:obj:`load_certificate_request` is
2491 neither :py:obj:`FILETYPE_PEM` nor :py:obj:`FILETYPE_ASN1` then
2492 :py:class:`ValueError` is raised.
2493 """
2494 self.assertRaises(ValueError, load_certificate_request, object(), b"")
2495
2496
2497
Jean-Paul Calderone68649052009-07-17 21:14:27 -04002498class PKCS7Tests(TestCase):
2499 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002500 Tests for :py:obj:`PKCS7Type`.
Jean-Paul Calderone68649052009-07-17 21:14:27 -04002501 """
2502 def test_type(self):
2503 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002504 :py:obj:`PKCS7Type` is a type object.
Jean-Paul Calderone68649052009-07-17 21:14:27 -04002505 """
2506 self.assertTrue(isinstance(PKCS7Type, type))
2507 self.assertEqual(PKCS7Type.__name__, 'PKCS7')
2508
2509 # XXX This doesn't currently work.
2510 # self.assertIdentical(PKCS7, PKCS7Type)
2511
2512
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -04002513 # XXX Opposite results for all these following methods
2514
Jean-Paul Calderone07c93742010-07-30 10:53:41 -04002515 def test_type_is_signed_wrong_args(self):
Jean-Paul Calderoneaa6c0692010-09-08 22:43:09 -04002516 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002517 :py:obj:`PKCS7Type.type_is_signed` raises :py:obj:`TypeError` if called with any
Jean-Paul Calderoneaa6c0692010-09-08 22:43:09 -04002518 arguments.
2519 """
Jean-Paul Calderone07c93742010-07-30 10:53:41 -04002520 pkcs7 = load_pkcs7_data(FILETYPE_PEM, pkcs7Data)
2521 self.assertRaises(TypeError, pkcs7.type_is_signed, None)
2522
2523
2524 def test_type_is_signed(self):
Jean-Paul Calderoneaa6c0692010-09-08 22:43:09 -04002525 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002526 :py:obj:`PKCS7Type.type_is_signed` returns :py:obj:`True` if the PKCS7 object is of
Jean-Paul Calderone64efa2c2011-09-11 10:00:09 -04002527 the type *signed*.
Jean-Paul Calderoneaa6c0692010-09-08 22:43:09 -04002528 """
Jean-Paul Calderone07c93742010-07-30 10:53:41 -04002529 pkcs7 = load_pkcs7_data(FILETYPE_PEM, pkcs7Data)
2530 self.assertTrue(pkcs7.type_is_signed())
2531
2532
2533 def test_type_is_enveloped_wrong_args(self):
Jean-Paul Calderoneaa6c0692010-09-08 22:43:09 -04002534 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002535 :py:obj:`PKCS7Type.type_is_enveloped` raises :py:obj:`TypeError` if called with any
Jean-Paul Calderoneaa6c0692010-09-08 22:43:09 -04002536 arguments.
2537 """
Jean-Paul Calderone07c93742010-07-30 10:53:41 -04002538 pkcs7 = load_pkcs7_data(FILETYPE_PEM, pkcs7Data)
2539 self.assertRaises(TypeError, pkcs7.type_is_enveloped, None)
2540
2541
2542 def test_type_is_enveloped(self):
Jean-Paul Calderoneaa6c0692010-09-08 22:43:09 -04002543 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002544 :py:obj:`PKCS7Type.type_is_enveloped` returns :py:obj:`False` if the PKCS7 object is
Jean-Paul Calderone64efa2c2011-09-11 10:00:09 -04002545 not of the type *enveloped*.
Jean-Paul Calderoneaa6c0692010-09-08 22:43:09 -04002546 """
Jean-Paul Calderone07c93742010-07-30 10:53:41 -04002547 pkcs7 = load_pkcs7_data(FILETYPE_PEM, pkcs7Data)
2548 self.assertFalse(pkcs7.type_is_enveloped())
2549
2550
2551 def test_type_is_signedAndEnveloped_wrong_args(self):
Jean-Paul Calderoneaa6c0692010-09-08 22:43:09 -04002552 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002553 :py:obj:`PKCS7Type.type_is_signedAndEnveloped` raises :py:obj:`TypeError` if called
Jean-Paul Calderoneaa6c0692010-09-08 22:43:09 -04002554 with any arguments.
2555 """
Jean-Paul Calderone07c93742010-07-30 10:53:41 -04002556 pkcs7 = load_pkcs7_data(FILETYPE_PEM, pkcs7Data)
2557 self.assertRaises(TypeError, pkcs7.type_is_signedAndEnveloped, None)
2558
2559
2560 def test_type_is_signedAndEnveloped(self):
Jean-Paul Calderoneaa6c0692010-09-08 22:43:09 -04002561 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002562 :py:obj:`PKCS7Type.type_is_signedAndEnveloped` returns :py:obj:`False` if the PKCS7
Jean-Paul Calderone64efa2c2011-09-11 10:00:09 -04002563 object is not of the type *signed and enveloped*.
Jean-Paul Calderoneaa6c0692010-09-08 22:43:09 -04002564 """
Jean-Paul Calderone07c93742010-07-30 10:53:41 -04002565 pkcs7 = load_pkcs7_data(FILETYPE_PEM, pkcs7Data)
2566 self.assertFalse(pkcs7.type_is_signedAndEnveloped())
2567
2568
Jean-Paul Calderoneb4754b92010-07-30 11:00:08 -04002569 def test_type_is_data(self):
Jean-Paul Calderoneaa6c0692010-09-08 22:43:09 -04002570 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002571 :py:obj:`PKCS7Type.type_is_data` returns :py:obj:`False` if the PKCS7 object is not of
Jean-Paul Calderoneaa6c0692010-09-08 22:43:09 -04002572 the type data.
2573 """
Jean-Paul Calderoneb4754b92010-07-30 11:00:08 -04002574 pkcs7 = load_pkcs7_data(FILETYPE_PEM, pkcs7Data)
2575 self.assertFalse(pkcs7.type_is_data())
2576
2577
2578 def test_type_is_data_wrong_args(self):
Jean-Paul Calderoneaa6c0692010-09-08 22:43:09 -04002579 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002580 :py:obj:`PKCS7Type.type_is_data` raises :py:obj:`TypeError` if called with any
Jean-Paul Calderoneaa6c0692010-09-08 22:43:09 -04002581 arguments.
2582 """
Jean-Paul Calderoneb4754b92010-07-30 11:00:08 -04002583 pkcs7 = load_pkcs7_data(FILETYPE_PEM, pkcs7Data)
2584 self.assertRaises(TypeError, pkcs7.type_is_data, None)
2585
2586
Jean-Paul Calderone97b28ca2010-07-30 10:56:07 -04002587 def test_get_type_name_wrong_args(self):
Jean-Paul Calderoneaa6c0692010-09-08 22:43:09 -04002588 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002589 :py:obj:`PKCS7Type.get_type_name` raises :py:obj:`TypeError` if called with any
Jean-Paul Calderoneaa6c0692010-09-08 22:43:09 -04002590 arguments.
2591 """
Jean-Paul Calderone97b28ca2010-07-30 10:56:07 -04002592 pkcs7 = load_pkcs7_data(FILETYPE_PEM, pkcs7Data)
2593 self.assertRaises(TypeError, pkcs7.get_type_name, None)
2594
2595
Jean-Paul Calderone4cbe05e2010-07-30 10:55:30 -04002596 def test_get_type_name(self):
Jean-Paul Calderoneaa6c0692010-09-08 22:43:09 -04002597 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002598 :py:obj:`PKCS7Type.get_type_name` returns a :py:obj:`str` giving the type name.
Jean-Paul Calderoneaa6c0692010-09-08 22:43:09 -04002599 """
Jean-Paul Calderone4cbe05e2010-07-30 10:55:30 -04002600 pkcs7 = load_pkcs7_data(FILETYPE_PEM, pkcs7Data)
Jean-Paul Calderone07640f12010-08-22 17:14:43 -04002601 self.assertEquals(pkcs7.get_type_name(), b('pkcs7-signedData'))
Jean-Paul Calderone4cbe05e2010-07-30 10:55:30 -04002602
2603
2604 def test_attribute(self):
Jean-Paul Calderoneaa6c0692010-09-08 22:43:09 -04002605 """
2606 If an attribute other than one of the methods tested here is accessed on
Jonathan Ballet648875f2011-07-16 14:14:58 +09002607 an instance of :py:obj:`PKCS7Type`, :py:obj:`AttributeError` is raised.
Jean-Paul Calderoneaa6c0692010-09-08 22:43:09 -04002608 """
Jean-Paul Calderone4cbe05e2010-07-30 10:55:30 -04002609 pkcs7 = load_pkcs7_data(FILETYPE_PEM, pkcs7Data)
2610 self.assertRaises(AttributeError, getattr, pkcs7, "foo")
2611
2612
Jean-Paul Calderone68649052009-07-17 21:14:27 -04002613
Jean-Paul Calderoneb9725592010-08-03 18:17:22 -04002614class NetscapeSPKITests(TestCase, _PKeyInteractionTestsMixin):
Jean-Paul Calderonedc138fa2009-06-27 14:32:07 -04002615 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002616 Tests for :py:obj:`OpenSSL.crypto.NetscapeSPKI`.
Jean-Paul Calderonedc138fa2009-06-27 14:32:07 -04002617 """
Jean-Paul Calderoneb9725592010-08-03 18:17:22 -04002618 def signable(self):
2619 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002620 Return a new :py:obj:`NetscapeSPKI` for use with signing tests.
Jean-Paul Calderoneb9725592010-08-03 18:17:22 -04002621 """
2622 return NetscapeSPKI()
2623
2624
Jean-Paul Calderone68649052009-07-17 21:14:27 -04002625 def test_type(self):
2626 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002627 :py:obj:`NetscapeSPKI` and :py:obj:`NetscapeSPKIType` refer to the same type object
Jean-Paul Calderone68649052009-07-17 21:14:27 -04002628 and can be used to create instances of that type.
2629 """
2630 self.assertIdentical(NetscapeSPKI, NetscapeSPKIType)
2631 self.assertConsistentType(NetscapeSPKI, 'NetscapeSPKI')
2632
2633
Jean-Paul Calderonedc138fa2009-06-27 14:32:07 -04002634 def test_construction(self):
2635 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002636 :py:obj:`NetscapeSPKI` returns an instance of :py:obj:`NetscapeSPKIType`.
Jean-Paul Calderonedc138fa2009-06-27 14:32:07 -04002637 """
2638 nspki = NetscapeSPKI()
2639 self.assertTrue(isinstance(nspki, NetscapeSPKIType))
2640
Jean-Paul Calderoneb9725592010-08-03 18:17:22 -04002641
Jean-Paul Calderone969efaa2010-08-03 18:19:19 -04002642 def test_invalid_attribute(self):
2643 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002644 Accessing a non-existent attribute of a :py:obj:`NetscapeSPKI` instance causes
2645 an :py:obj:`AttributeError` to be raised.
Jean-Paul Calderone969efaa2010-08-03 18:19:19 -04002646 """
2647 nspki = NetscapeSPKI()
2648 self.assertRaises(AttributeError, lambda: nspki.foo)
2649
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -04002650
Jean-Paul Calderone06ada9f2010-08-03 18:26:52 -04002651 def test_b64_encode(self):
2652 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002653 :py:obj:`NetscapeSPKI.b64_encode` encodes the certificate to a base64 blob.
Jean-Paul Calderone06ada9f2010-08-03 18:26:52 -04002654 """
2655 nspki = NetscapeSPKI()
2656 blob = nspki.b64_encode()
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05002657 self.assertTrue(isinstance(blob, binary_type))
Jean-Paul Calderone06ada9f2010-08-03 18:26:52 -04002658
2659
Jean-Paul Calderonedc138fa2009-06-27 14:32:07 -04002660
Rick Dean536ba022009-07-24 23:57:27 -05002661class RevokedTests(TestCase):
2662 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002663 Tests for :py:obj:`OpenSSL.crypto.Revoked`
Rick Dean536ba022009-07-24 23:57:27 -05002664 """
2665 def test_construction(self):
2666 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002667 Confirm we can create :py:obj:`OpenSSL.crypto.Revoked`. Check
Rick Dean536ba022009-07-24 23:57:27 -05002668 that it is empty.
2669 """
2670 revoked = Revoked()
Jean-Paul Calderoneeacad4a2010-08-22 17:12:55 -04002671 self.assertTrue(isinstance(revoked, Revoked))
2672 self.assertEquals(type(revoked), Revoked)
2673 self.assertEquals(revoked.get_serial(), b('00'))
2674 self.assertEquals(revoked.get_rev_date(), None)
2675 self.assertEquals(revoked.get_reason(), None)
Rick Dean536ba022009-07-24 23:57:27 -05002676
2677
Jean-Paul Calderone4f74bfe2010-01-30 14:32:49 -05002678 def test_construction_wrong_args(self):
2679 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002680 Calling :py:obj:`OpenSSL.crypto.Revoked` with any arguments results
2681 in a :py:obj:`TypeError` being raised.
Jean-Paul Calderone4f74bfe2010-01-30 14:32:49 -05002682 """
2683 self.assertRaises(TypeError, Revoked, None)
2684 self.assertRaises(TypeError, Revoked, 1)
2685 self.assertRaises(TypeError, Revoked, "foo")
2686
2687
Rick Dean536ba022009-07-24 23:57:27 -05002688 def test_serial(self):
2689 """
Jean-Paul Calderoneb98eae02010-01-30 13:18:04 -05002690 Confirm we can set and get serial numbers from
Jonathan Ballet648875f2011-07-16 14:14:58 +09002691 :py:obj:`OpenSSL.crypto.Revoked`. Confirm errors are handled
Rick Dean536ba022009-07-24 23:57:27 -05002692 with grace.
2693 """
2694 revoked = Revoked()
Jean-Paul Calderoneeacad4a2010-08-22 17:12:55 -04002695 ret = revoked.set_serial(b('10b'))
2696 self.assertEquals(ret, None)
Rick Dean536ba022009-07-24 23:57:27 -05002697 ser = revoked.get_serial()
Jean-Paul Calderoneeacad4a2010-08-22 17:12:55 -04002698 self.assertEquals(ser, b('010B'))
Rick Dean536ba022009-07-24 23:57:27 -05002699
Jean-Paul Calderoneeacad4a2010-08-22 17:12:55 -04002700 revoked.set_serial(b('31ppp')) # a type error would be nice
Rick Dean536ba022009-07-24 23:57:27 -05002701 ser = revoked.get_serial()
Jean-Paul Calderoneeacad4a2010-08-22 17:12:55 -04002702 self.assertEquals(ser, b('31'))
Rick Dean536ba022009-07-24 23:57:27 -05002703
Jean-Paul Calderoneeacad4a2010-08-22 17:12:55 -04002704 self.assertRaises(ValueError, revoked.set_serial, b('pqrst'))
Rick Dean536ba022009-07-24 23:57:27 -05002705 self.assertRaises(TypeError, revoked.set_serial, 100)
Jean-Paul Calderone4f74bfe2010-01-30 14:32:49 -05002706 self.assertRaises(TypeError, revoked.get_serial, 1)
2707 self.assertRaises(TypeError, revoked.get_serial, None)
2708 self.assertRaises(TypeError, revoked.get_serial, "")
Rick Dean536ba022009-07-24 23:57:27 -05002709
2710
2711 def test_date(self):
2712 """
Jean-Paul Calderoneb98eae02010-01-30 13:18:04 -05002713 Confirm we can set and get revocation dates from
Jonathan Ballet648875f2011-07-16 14:14:58 +09002714 :py:obj:`OpenSSL.crypto.Revoked`. Confirm errors are handled
Rick Dean536ba022009-07-24 23:57:27 -05002715 with grace.
2716 """
2717 revoked = Revoked()
2718 date = revoked.get_rev_date()
Jean-Paul Calderoneeacad4a2010-08-22 17:12:55 -04002719 self.assertEquals(date, None)
Rick Dean536ba022009-07-24 23:57:27 -05002720
Jean-Paul Calderoneeacad4a2010-08-22 17:12:55 -04002721 now = b(datetime.now().strftime("%Y%m%d%H%M%SZ"))
Rick Dean536ba022009-07-24 23:57:27 -05002722 ret = revoked.set_rev_date(now)
Jean-Paul Calderoneeacad4a2010-08-22 17:12:55 -04002723 self.assertEqual(ret, None)
Rick Dean536ba022009-07-24 23:57:27 -05002724 date = revoked.get_rev_date()
Jean-Paul Calderoneeacad4a2010-08-22 17:12:55 -04002725 self.assertEqual(date, now)
Rick Dean536ba022009-07-24 23:57:27 -05002726
2727
Rick Dean6385faf2009-07-26 00:07:47 -05002728 def test_reason(self):
2729 """
Jean-Paul Calderoneb98eae02010-01-30 13:18:04 -05002730 Confirm we can set and get revocation reasons from
Jonathan Ballet648875f2011-07-16 14:14:58 +09002731 :py:obj:`OpenSSL.crypto.Revoked`. The "get" need to work
Rick Dean6385faf2009-07-26 00:07:47 -05002732 as "set". Likewise, each reason of all_reasons() must work.
2733 """
2734 revoked = Revoked()
2735 for r in revoked.all_reasons():
Jean-Paul Calderoneeacad4a2010-08-22 17:12:55 -04002736 for x in range(2):
Rick Dean6385faf2009-07-26 00:07:47 -05002737 ret = revoked.set_reason(r)
Jean-Paul Calderoneeacad4a2010-08-22 17:12:55 -04002738 self.assertEquals(ret, None)
Rick Dean6385faf2009-07-26 00:07:47 -05002739 reason = revoked.get_reason()
Jean-Paul Calderoneeacad4a2010-08-22 17:12:55 -04002740 self.assertEquals(
2741 reason.lower().replace(b(' '), b('')),
2742 r.lower().replace(b(' '), b('')))
Rick Dean6385faf2009-07-26 00:07:47 -05002743 r = reason # again with the resp of get
2744
2745 revoked.set_reason(None)
2746 self.assertEqual(revoked.get_reason(), None)
2747
2748
Jean-Paul Calderone4f74bfe2010-01-30 14:32:49 -05002749 def test_set_reason_wrong_arguments(self):
Rick Dean6385faf2009-07-26 00:07:47 -05002750 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002751 Calling :py:obj:`OpenSSL.crypto.Revoked.set_reason` with other than
Jean-Paul Calderone4f74bfe2010-01-30 14:32:49 -05002752 one argument, or an argument which isn't a valid reason,
Jonathan Ballet648875f2011-07-16 14:14:58 +09002753 results in :py:obj:`TypeError` or :py:obj:`ValueError` being raised.
Rick Dean6385faf2009-07-26 00:07:47 -05002754 """
2755 revoked = Revoked()
2756 self.assertRaises(TypeError, revoked.set_reason, 100)
Jean-Paul Calderone281b62b2010-08-28 14:22:29 -04002757 self.assertRaises(ValueError, revoked.set_reason, b('blue'))
Rick Dean6385faf2009-07-26 00:07:47 -05002758
Rick Dean536ba022009-07-24 23:57:27 -05002759
Jean-Paul Calderone4f74bfe2010-01-30 14:32:49 -05002760 def test_get_reason_wrong_arguments(self):
2761 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002762 Calling :py:obj:`OpenSSL.crypto.Revoked.get_reason` with any
2763 arguments results in :py:obj:`TypeError` being raised.
Jean-Paul Calderone4f74bfe2010-01-30 14:32:49 -05002764 """
2765 revoked = Revoked()
2766 self.assertRaises(TypeError, revoked.get_reason, None)
2767 self.assertRaises(TypeError, revoked.get_reason, 1)
2768 self.assertRaises(TypeError, revoked.get_reason, "foo")
2769
2770
2771
Rick Dean536ba022009-07-24 23:57:27 -05002772class CRLTests(TestCase):
2773 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002774 Tests for :py:obj:`OpenSSL.crypto.CRL`
Rick Dean536ba022009-07-24 23:57:27 -05002775 """
2776 cert = load_certificate(FILETYPE_PEM, cleartextCertificatePEM)
2777 pkey = load_privatekey(FILETYPE_PEM, cleartextPrivateKeyPEM)
2778
2779 def test_construction(self):
2780 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002781 Confirm we can create :py:obj:`OpenSSL.crypto.CRL`. Check
Rick Dean536ba022009-07-24 23:57:27 -05002782 that it is empty
2783 """
2784 crl = CRL()
2785 self.assertTrue( isinstance(crl, CRL) )
2786 self.assertEqual(crl.get_revoked(), None)
2787
2788
Jean-Paul Calderone2efd03e2010-01-30 13:59:55 -05002789 def test_construction_wrong_args(self):
2790 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002791 Calling :py:obj:`OpenSSL.crypto.CRL` with any number of arguments
2792 results in a :py:obj:`TypeError` being raised.
Jean-Paul Calderone2efd03e2010-01-30 13:59:55 -05002793 """
2794 self.assertRaises(TypeError, CRL, 1)
2795 self.assertRaises(TypeError, CRL, "")
2796 self.assertRaises(TypeError, CRL, None)
2797
2798
Rick Dean536ba022009-07-24 23:57:27 -05002799 def test_export(self):
2800 """
2801 Use python to create a simple CRL with a revocation, and export
2802 the CRL in formats of PEM, DER and text. Those outputs are verified
2803 with the openssl program.
2804 """
2805 crl = CRL()
2806 revoked = Revoked()
Jean-Paul Calderoneb894fe12010-08-22 19:30:19 -04002807 now = b(datetime.now().strftime("%Y%m%d%H%M%SZ"))
Rick Dean536ba022009-07-24 23:57:27 -05002808 revoked.set_rev_date(now)
Jean-Paul Calderoneb894fe12010-08-22 19:30:19 -04002809 revoked.set_serial(b('3ab'))
2810 revoked.set_reason(b('sUpErSeDEd'))
Rick Dean536ba022009-07-24 23:57:27 -05002811 crl.add_revoked(revoked)
2812
2813 # PEM format
2814 dumped_crl = crl.export(self.cert, self.pkey, days=20)
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05002815 text = _runopenssl(dumped_crl, b"crl", b"-noout", b"-text")
Jean-Paul Calderoneb894fe12010-08-22 19:30:19 -04002816 text.index(b('Serial Number: 03AB'))
2817 text.index(b('Superseded'))
2818 text.index(b('Issuer: /C=US/ST=IL/L=Chicago/O=Testing/CN=Testing Root CA'))
Rick Dean536ba022009-07-24 23:57:27 -05002819
2820 # DER format
2821 dumped_crl = crl.export(self.cert, self.pkey, FILETYPE_ASN1)
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05002822 text = _runopenssl(dumped_crl, b"crl", b"-noout", b"-text", b"-inform", b"DER")
Jean-Paul Calderoneb894fe12010-08-22 19:30:19 -04002823 text.index(b('Serial Number: 03AB'))
2824 text.index(b('Superseded'))
2825 text.index(b('Issuer: /C=US/ST=IL/L=Chicago/O=Testing/CN=Testing Root CA'))
Rick Dean536ba022009-07-24 23:57:27 -05002826
2827 # text format
2828 dumped_text = crl.export(self.cert, self.pkey, type=FILETYPE_TEXT)
2829 self.assertEqual(text, dumped_text)
2830
2831
Jean-Paul Calderonec7293bc2011-09-13 15:24:38 -04002832 def test_export_invalid(self):
2833 """
2834 If :py:obj:`CRL.export` is used with an uninitialized :py:obj:`X509`
Jean-Paul Calderoneb5871072012-03-09 14:58:00 -08002835 instance, :py:obj:`OpenSSL.crypto.Error` is raised.
Jean-Paul Calderonec7293bc2011-09-13 15:24:38 -04002836 """
2837 crl = CRL()
2838 self.assertRaises(Error, crl.export, X509(), PKey())
2839
2840
Jean-Paul Calderone56515342010-01-30 13:49:38 -05002841 def test_add_revoked_keyword(self):
2842 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002843 :py:obj:`OpenSSL.CRL.add_revoked` accepts its single argument as the
Jean-Paul Calderone64efa2c2011-09-11 10:00:09 -04002844 ``revoked`` keyword argument.
Jean-Paul Calderone56515342010-01-30 13:49:38 -05002845 """
2846 crl = CRL()
2847 revoked = Revoked()
2848 crl.add_revoked(revoked=revoked)
2849 self.assertTrue(isinstance(crl.get_revoked()[0], Revoked))
2850
Rick Dean6385faf2009-07-26 00:07:47 -05002851
Jean-Paul Calderone883ca4b2010-01-30 13:55:13 -05002852 def test_export_wrong_args(self):
2853 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002854 Calling :py:obj:`OpenSSL.CRL.export` with fewer than two or more than
Jean-Paul Calderonef1515862010-01-30 13:57:03 -05002855 four arguments, or with arguments other than the certificate,
2856 private key, integer file type, and integer number of days it
Jonathan Ballet648875f2011-07-16 14:14:58 +09002857 expects, results in a :py:obj:`TypeError` being raised.
Jean-Paul Calderone883ca4b2010-01-30 13:55:13 -05002858 """
2859 crl = CRL()
2860 self.assertRaises(TypeError, crl.export)
2861 self.assertRaises(TypeError, crl.export, self.cert)
2862 self.assertRaises(TypeError, crl.export, self.cert, self.pkey, FILETYPE_PEM, 10, "foo")
2863
Jean-Paul Calderonef1515862010-01-30 13:57:03 -05002864 self.assertRaises(TypeError, crl.export, None, self.pkey, FILETYPE_PEM, 10)
2865 self.assertRaises(TypeError, crl.export, self.cert, None, FILETYPE_PEM, 10)
2866 self.assertRaises(TypeError, crl.export, self.cert, self.pkey, None, 10)
2867 self.assertRaises(TypeError, crl.export, self.cert, FILETYPE_PEM, None)
2868
Jean-Paul Calderone883ca4b2010-01-30 13:55:13 -05002869
Jean-Paul Calderoneea198422010-01-30 13:58:23 -05002870 def test_export_unknown_filetype(self):
2871 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002872 Calling :py:obj:`OpenSSL.CRL.export` with a file type other than
2873 :py:obj:`FILETYPE_PEM`, :py:obj:`FILETYPE_ASN1`, or :py:obj:`FILETYPE_TEXT` results
2874 in a :py:obj:`ValueError` being raised.
Jean-Paul Calderoneea198422010-01-30 13:58:23 -05002875 """
2876 crl = CRL()
2877 self.assertRaises(ValueError, crl.export, self.cert, self.pkey, 100, 10)
2878
2879
Rick Dean536ba022009-07-24 23:57:27 -05002880 def test_get_revoked(self):
2881 """
Jean-Paul Calderoneb98eae02010-01-30 13:18:04 -05002882 Use python to create a simple CRL with two revocations.
Jonathan Ballet648875f2011-07-16 14:14:58 +09002883 Get back the :py:obj:`Revoked` using :py:obj:`OpenSSL.CRL.get_revoked` and
Rick Dean536ba022009-07-24 23:57:27 -05002884 verify them.
2885 """
2886 crl = CRL()
2887
2888 revoked = Revoked()
Jean-Paul Calderoneb894fe12010-08-22 19:30:19 -04002889 now = b(datetime.now().strftime("%Y%m%d%H%M%SZ"))
Rick Dean536ba022009-07-24 23:57:27 -05002890 revoked.set_rev_date(now)
Jean-Paul Calderoneb894fe12010-08-22 19:30:19 -04002891 revoked.set_serial(b('3ab'))
Rick Dean536ba022009-07-24 23:57:27 -05002892 crl.add_revoked(revoked)
Jean-Paul Calderoneb894fe12010-08-22 19:30:19 -04002893 revoked.set_serial(b('100'))
2894 revoked.set_reason(b('sUpErSeDEd'))
Rick Dean536ba022009-07-24 23:57:27 -05002895 crl.add_revoked(revoked)
2896
2897 revs = crl.get_revoked()
2898 self.assertEqual(len(revs), 2)
2899 self.assertEqual(type(revs[0]), Revoked)
2900 self.assertEqual(type(revs[1]), Revoked)
Jean-Paul Calderoneb894fe12010-08-22 19:30:19 -04002901 self.assertEqual(revs[0].get_serial(), b('03AB'))
2902 self.assertEqual(revs[1].get_serial(), b('0100'))
Rick Dean536ba022009-07-24 23:57:27 -05002903 self.assertEqual(revs[0].get_rev_date(), now)
2904 self.assertEqual(revs[1].get_rev_date(), now)
2905
2906
Jean-Paul Calderone46bdce42010-01-30 13:44:16 -05002907 def test_get_revoked_wrong_args(self):
2908 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002909 Calling :py:obj:`OpenSSL.CRL.get_revoked` with any arguments results
2910 in a :py:obj:`TypeError` being raised.
Jean-Paul Calderone46bdce42010-01-30 13:44:16 -05002911 """
2912 crl = CRL()
2913 self.assertRaises(TypeError, crl.get_revoked, None)
2914 self.assertRaises(TypeError, crl.get_revoked, 1)
2915 self.assertRaises(TypeError, crl.get_revoked, "")
2916 self.assertRaises(TypeError, crl.get_revoked, "", 1, None)
2917
2918
Jean-Paul Calderoneecef6fa2010-01-30 13:47:18 -05002919 def test_add_revoked_wrong_args(self):
2920 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002921 Calling :py:obj:`OpenSSL.CRL.add_revoked` with other than one
2922 argument results in a :py:obj:`TypeError` being raised.
Jean-Paul Calderoneecef6fa2010-01-30 13:47:18 -05002923 """
2924 crl = CRL()
2925 self.assertRaises(TypeError, crl.add_revoked)
2926 self.assertRaises(TypeError, crl.add_revoked, 1, 2)
2927 self.assertRaises(TypeError, crl.add_revoked, "foo", "bar")
2928
2929
Rick Dean536ba022009-07-24 23:57:27 -05002930 def test_load_crl(self):
2931 """
2932 Load a known CRL and inspect its revocations. Both
2933 PEM and DER formats are loaded.
2934 """
Jean-Paul Calderone3eb5cc72010-01-30 15:24:40 -05002935 crl = load_crl(FILETYPE_PEM, crlData)
Rick Dean536ba022009-07-24 23:57:27 -05002936 revs = crl.get_revoked()
2937 self.assertEqual(len(revs), 2)
Jean-Paul Calderoneb894fe12010-08-22 19:30:19 -04002938 self.assertEqual(revs[0].get_serial(), b('03AB'))
Rick Dean6385faf2009-07-26 00:07:47 -05002939 self.assertEqual(revs[0].get_reason(), None)
Jean-Paul Calderoneb894fe12010-08-22 19:30:19 -04002940 self.assertEqual(revs[1].get_serial(), b('0100'))
2941 self.assertEqual(revs[1].get_reason(), b('Superseded'))
Rick Dean536ba022009-07-24 23:57:27 -05002942
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05002943 der = _runopenssl(crlData, b"crl", b"-outform", b"DER")
Jean-Paul Calderoneb98eae02010-01-30 13:18:04 -05002944 crl = load_crl(FILETYPE_ASN1, der)
Rick Dean536ba022009-07-24 23:57:27 -05002945 revs = crl.get_revoked()
2946 self.assertEqual(len(revs), 2)
Jean-Paul Calderoneb894fe12010-08-22 19:30:19 -04002947 self.assertEqual(revs[0].get_serial(), b('03AB'))
Rick Dean6385faf2009-07-26 00:07:47 -05002948 self.assertEqual(revs[0].get_reason(), None)
Jean-Paul Calderoneb894fe12010-08-22 19:30:19 -04002949 self.assertEqual(revs[1].get_serial(), b('0100'))
2950 self.assertEqual(revs[1].get_reason(), b('Superseded'))
Rick Dean536ba022009-07-24 23:57:27 -05002951
2952
Jean-Paul Calderone3eb5cc72010-01-30 15:24:40 -05002953 def test_load_crl_wrong_args(self):
2954 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002955 Calling :py:obj:`OpenSSL.crypto.load_crl` with other than two
2956 arguments results in a :py:obj:`TypeError` being raised.
Jean-Paul Calderone3eb5cc72010-01-30 15:24:40 -05002957 """
2958 self.assertRaises(TypeError, load_crl)
2959 self.assertRaises(TypeError, load_crl, FILETYPE_PEM)
2960 self.assertRaises(TypeError, load_crl, FILETYPE_PEM, crlData, None)
2961
2962
2963 def test_load_crl_bad_filetype(self):
2964 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002965 Calling :py:obj:`OpenSSL.crypto.load_crl` with an unknown file type
2966 raises a :py:obj:`ValueError`.
Jean-Paul Calderone3eb5cc72010-01-30 15:24:40 -05002967 """
2968 self.assertRaises(ValueError, load_crl, 100, crlData)
2969
2970
2971 def test_load_crl_bad_data(self):
2972 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002973 Calling :py:obj:`OpenSSL.crypto.load_crl` with file data which can't
2974 be loaded raises a :py:obj:`OpenSSL.crypto.Error`.
Jean-Paul Calderone3eb5cc72010-01-30 15:24:40 -05002975 """
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05002976 self.assertRaises(Error, load_crl, FILETYPE_PEM, b"hello, world")
Jean-Paul Calderone3eb5cc72010-01-30 15:24:40 -05002977
Jean-Paul Calderonedc138fa2009-06-27 14:32:07 -04002978
Jean-Paul Calderone5300d6a2013-12-29 16:36:50 -05002979
James Yonan7c2e5d32010-02-27 05:45:50 -07002980class SignVerifyTests(TestCase):
2981 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002982 Tests for :py:obj:`OpenSSL.crypto.sign` and :py:obj:`OpenSSL.crypto.verify`.
James Yonan7c2e5d32010-02-27 05:45:50 -07002983 """
2984 def test_sign_verify(self):
Jean-Paul Calderonef3cb9d82010-06-22 10:29:33 -04002985 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09002986 :py:obj:`sign` generates a cryptographic signature which :py:obj:`verify` can check.
Jean-Paul Calderonef3cb9d82010-06-22 10:29:33 -04002987 """
Jean-Paul Calderoneea305c52010-08-28 14:41:07 -04002988 content = b(
Jean-Paul Calderoneb98ce212010-06-22 09:46:27 -04002989 "It was a bright cold day in April, and the clocks were striking "
2990 "thirteen. Winston Smith, his chin nuzzled into his breast in an "
2991 "effort to escape the vile wind, slipped quickly through the "
2992 "glass doors of Victory Mansions, though not quickly enough to "
2993 "prevent a swirl of gritty dust from entering along with him.")
2994
2995 # sign the content with this private key
Jean-Paul Calderonef3cb9d82010-06-22 10:29:33 -04002996 priv_key = load_privatekey(FILETYPE_PEM, root_key_pem)
Jean-Paul Calderoneb98ce212010-06-22 09:46:27 -04002997 # verify the content with this cert
2998 good_cert = load_certificate(FILETYPE_PEM, root_cert_pem)
2999 # certificate unrelated to priv_key, used to trigger an error
3000 bad_cert = load_certificate(FILETYPE_PEM, server_cert_pem)
James Yonan7c2e5d32010-02-27 05:45:50 -07003001
Jean-Paul Calderoned08feb02010-10-12 21:53:24 -04003002 for digest in ['md5', 'sha1']:
James Yonan7c2e5d32010-02-27 05:45:50 -07003003 sig = sign(priv_key, content, digest)
3004
3005 # Verify the signature of content, will throw an exception if error.
3006 verify(good_cert, sig, content, digest)
3007
3008 # This should fail because the certificate doesn't match the
3009 # private key that was used to sign the content.
3010 self.assertRaises(Error, verify, bad_cert, sig, content, digest)
3011
3012 # This should fail because we've "tainted" the content after
3013 # signing it.
Jean-Paul Calderoneea305c52010-08-28 14:41:07 -04003014 self.assertRaises(
3015 Error, verify,
3016 good_cert, sig, content + b("tainted"), digest)
James Yonan7c2e5d32010-02-27 05:45:50 -07003017
3018 # test that unknown digest types fail
Jean-Paul Calderoneea305c52010-08-28 14:41:07 -04003019 self.assertRaises(
Jean-Paul Calderone9538f142010-10-13 22:13:40 -04003020 ValueError, sign, priv_key, content, "strange-digest")
Jean-Paul Calderoneea305c52010-08-28 14:41:07 -04003021 self.assertRaises(
Jean-Paul Calderone9538f142010-10-13 22:13:40 -04003022 ValueError, verify, good_cert, sig, content, "strange-digest")
James Yonan7c2e5d32010-02-27 05:45:50 -07003023
Jean-Paul Calderoneb98ce212010-06-22 09:46:27 -04003024
Jean-Paul Calderone9828f662010-12-08 22:57:26 -05003025 def test_sign_nulls(self):
3026 """
Jonathan Ballet648875f2011-07-16 14:14:58 +09003027 :py:obj:`sign` produces a signature for a string with embedded nulls.
Jean-Paul Calderone9828f662010-12-08 22:57:26 -05003028 """
3029 content = b("Watch out! \0 Did you see it?")
3030 priv_key = load_privatekey(FILETYPE_PEM, root_key_pem)
3031 good_cert = load_certificate(FILETYPE_PEM, root_cert_pem)
3032 sig = sign(priv_key, content, "sha1")
3033 verify(good_cert, sig, content, "sha1")
3034
3035
Rick Dean5b7b6372009-04-01 11:34:06 -05003036if __name__ == '__main__':
3037 main()