blob: 75f4a5ae32bc14e075a080625964d7725b240670 [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
Alex Chanb00ede22017-01-30 07:24:40 +00008from warnings import simplefilter
Jean-Paul Calderone0b88b6a2009-07-05 12:44:41 -04009
Alex Gaynor4b9c96a2014-08-14 09:51:48 -070010import base64
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
Alex Gaynor791212d2015-09-05 15:46:08 -040014import pytest
15
Alex Gaynor9939ba12017-06-25 16:28:24 -040016from cryptography import x509
Paul Kehrer72d968b2016-07-29 15:31:04 +080017from cryptography.hazmat.backends.openssl.backend import backend
18from cryptography.hazmat.primitives import serialization
19from cryptography.hazmat.primitives.asymmetric import rsa
20
Alex Gaynore466bc92017-07-06 23:43:47 -040021import flaky
22
Alex Gaynor01f90a12019-02-07 09:14:48 -050023from OpenSSL.crypto import TYPE_RSA, TYPE_DSA, Error, PKey
24from OpenSSL.crypto import X509, X509Name
Alex Gaynor31287502015-09-05 16:11:27 -040025from OpenSSL.crypto import (
Dan Sully44e767a2016-06-04 18:05:27 -070026 X509Store,
27 X509StoreFlags,
Dan Sully44e767a2016-06-04 18:05:27 -070028 X509StoreContext,
Alex Gaynor03737182020-07-23 20:40:46 -040029 X509StoreContextError,
Alex Gaynor31287502015-09-05 16:11:27 -040030)
Alex Gaynor01f90a12019-02-07 09:14:48 -050031from OpenSSL.crypto import X509Req
32from OpenSSL.crypto import X509Extension
Rick Dean5b7b6372009-04-01 11:34:06 -050033from OpenSSL.crypto import load_certificate, load_privatekey
Cory Benfield6492f7c2015-10-27 16:57:58 +090034from OpenSSL.crypto import load_publickey, dump_publickey
Jean-Paul Calderonef17e4212009-04-01 14:21:40 -040035from OpenSSL.crypto import FILETYPE_PEM, FILETYPE_ASN1, FILETYPE_TEXT
Jean-Paul Calderone71919862009-04-01 13:01:19 -040036from OpenSSL.crypto import dump_certificate, load_certificate_request
37from OpenSSL.crypto import dump_certificate_request, dump_privatekey
Alex Gaynor01f90a12019-02-07 09:14:48 -050038from OpenSSL.crypto import PKCS7, load_pkcs7_data
39from OpenSSL.crypto import PKCS12, load_pkcs12
Dominic Chenf05b2122015-10-13 16:32:35 +000040from OpenSSL.crypto import CRL, Revoked, dump_crl, load_crl
Alex Gaynor01f90a12019-02-07 09:14:48 -050041from OpenSSL.crypto import NetscapeSPKI
Jean-Paul Calderonec09fd582014-04-18 22:00:10 -040042from OpenSSL.crypto import (
Alex Gaynor03737182020-07-23 20:40:46 -040043 sign,
44 verify,
45 get_elliptic_curve,
46 get_elliptic_curves,
47)
Hynek Schlawackf0e66852015-10-16 20:18:38 +020048
Alex Chanb00ede22017-01-30 07:24:40 +000049from .util import EqualityTestsMixin, is_consistent_type, WARNING_TYPE_EXPECTED
Jean-Paul Calderoneadd7bf02010-08-22 17:38:30 -040050
Alex Gaynoraceb3e22015-09-05 12:00:22 -040051
Jean-Paul Calderone9da338d2011-05-04 11:40:54 -040052def normalize_privatekey_pem(pem):
53 return dump_privatekey(FILETYPE_PEM, load_privatekey(FILETYPE_PEM, pem))
54
Jean-Paul Calderoneadd7bf02010-08-22 17:38:30 -040055
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -050056GOOD_CIPHER = "blowfish"
57BAD_CIPHER = "zippers"
58
Anthony Alba2ce737f2015-12-04 11:04:56 +080059GOOD_DIGEST = "SHA1"
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -050060BAD_DIGEST = "monkeys"
61
Paul Kehrera40898b2017-06-11 16:30:58 -100062old_root_cert_pem = b"""-----BEGIN CERTIFICATE-----
Rick Dean94e46fd2009-07-18 14:51:24 -050063MIIC7TCCAlagAwIBAgIIPQzE4MbeufQwDQYJKoZIhvcNAQEFBQAwWDELMAkGA1UE
64BhMCVVMxCzAJBgNVBAgTAklMMRAwDgYDVQQHEwdDaGljYWdvMRAwDgYDVQQKEwdU
65ZXN0aW5nMRgwFgYDVQQDEw9UZXN0aW5nIFJvb3QgQ0EwIhgPMjAwOTAzMjUxMjM2
66NThaGA8yMDE3MDYxMTEyMzY1OFowWDELMAkGA1UEBhMCVVMxCzAJBgNVBAgTAklM
67MRAwDgYDVQQHEwdDaGljYWdvMRAwDgYDVQQKEwdUZXN0aW5nMRgwFgYDVQQDEw9U
68ZXN0aW5nIFJvb3QgQ0EwgZ8wDQYJKoZIhvcNAQEBBQADgY0AMIGJAoGBAPmaQumL
69urpE527uSEHdL1pqcDRmWzu+98Y6YHzT/J7KWEamyMCNZ6fRW1JCR782UQ8a07fy
702xXsKy4WdKaxyG8CcatwmXvpvRQ44dSANMihHELpANTdyVp6DCysED6wkQFurHlF
711dshEaJw8b/ypDhmbVIo6Ci1xvCJqivbLFnbAgMBAAGjgbswgbgwHQYDVR0OBBYE
72FINVdy1eIfFJDAkk51QJEo3IfgSuMIGIBgNVHSMEgYAwfoAUg1V3LV4h8UkMCSTn
73VAkSjch+BK6hXKRaMFgxCzAJBgNVBAYTAlVTMQswCQYDVQQIEwJJTDEQMA4GA1UE
74BxMHQ2hpY2FnbzEQMA4GA1UEChMHVGVzdGluZzEYMBYGA1UEAxMPVGVzdGluZyBS
75b290IENBggg9DMTgxt659DAMBgNVHRMEBTADAQH/MA0GCSqGSIb3DQEBBQUAA4GB
76AGGCDazMJGoWNBpc03u6+smc95dEead2KlZXBATOdFT1VesY3+nUOqZhEhTGlDMi
77hkgaZnzoIq/Uamidegk4hirsCT/R+6vsKAAxNTcBjUeZjlykCJWy5ojShGftXIKY
78w/njVbKMXrvc83qmTdGl3TAM0fxQIpqgcglFLveEBgzn
79-----END CERTIFICATE-----
Alex Gaynore7f51982016-09-11 11:48:14 -040080"""
Rick Dean94e46fd2009-07-18 14:51:24 -050081
Paul Kehrera40898b2017-06-11 16:30:58 -100082root_cert_pem = b"""-----BEGIN CERTIFICATE-----
83MIIC6TCCAlKgAwIBAgIIPQzE4MbeufQwDQYJKoZIhvcNAQEFBQAwWDELMAkGA1UE
84BhMCVVMxCzAJBgNVBAgTAklMMRAwDgYDVQQHEwdDaGljYWdvMRAwDgYDVQQKEwdU
85ZXN0aW5nMRgwFgYDVQQDEw9UZXN0aW5nIFJvb3QgQ0EwHhcNMTcwNjExMjIzMjU5
86WhcNMzcwNjA2MjIzMjU5WjBYMQswCQYDVQQGEwJVUzELMAkGA1UECBMCSUwxEDAO
87BgNVBAcTB0NoaWNhZ28xEDAOBgNVBAoTB1Rlc3RpbmcxGDAWBgNVBAMTD1Rlc3Rp
88bmcgUm9vdCBDQTCBnzANBgkqhkiG9w0BAQEFAAOBjQAwgYkCgYEA+ZpC6Yu6ukTn
89bu5IQd0vWmpwNGZbO773xjpgfNP8nspYRqbIwI1np9FbUkJHvzZRDxrTt/LbFewr
90LhZ0prHIbwJxq3CZe+m9FDjh1IA0yKEcQukA1N3JWnoMLKwQPrCRAW6seUXV2yER
91onDxv/KkOGZtUijoKLXG8ImqK9ssWdsCAwEAAaOBuzCBuDAdBgNVHQ4EFgQUg1V3
92LV4h8UkMCSTnVAkSjch+BK4wgYgGA1UdIwSBgDB+gBSDVXctXiHxSQwJJOdUCRKN
93yH4ErqFcpFowWDELMAkGA1UEBhMCVVMxCzAJBgNVBAgTAklMMRAwDgYDVQQHEwdD
94aGljYWdvMRAwDgYDVQQKEwdUZXN0aW5nMRgwFgYDVQQDEw9UZXN0aW5nIFJvb3Qg
95Q0GCCD0MxODG3rn0MAwGA1UdEwQFMAMBAf8wDQYJKoZIhvcNAQEFBQADgYEANFYQ
96R+T70VcZ+SnvURnwviFgCXeedBzCr21meo+DNHbkp2gudB9W8Xrned/wtUBVymy9
97gjB5jINfU7Lci0H57Evsw96UJJVfhXdUMHpqt1RGCoEd9FWnrDyrSy0NysnBT2bH
98lEqxh3aFEUx9IOQ4sgnx1/NOFXBpkRtivl6O0Ec=
99-----END CERTIFICATE-----
100"""
101
Alex Gaynore7f51982016-09-11 11:48:14 -0400102root_key_pem = b"""-----BEGIN RSA PRIVATE KEY-----
Rick Dean94e46fd2009-07-18 14:51:24 -0500103MIICXQIBAAKBgQD5mkLpi7q6ROdu7khB3S9aanA0Zls7vvfGOmB80/yeylhGpsjA
104jWen0VtSQke/NlEPGtO38tsV7CsuFnSmschvAnGrcJl76b0UOOHUgDTIoRxC6QDU
1053claegwsrBA+sJEBbqx5RdXbIRGicPG/8qQ4Zm1SKOgotcbwiaor2yxZ2wIDAQAB
106AoGBAPCgMpmLxzwDaUmcFbTJUvlLW1hoxNNYSu2jIZm1k/hRAcE60JYwvBkgz3UB
107yMEh0AtLxYe0bFk6EHah11tMUPgscbCq73snJ++8koUw+csk22G65hOs51bVb7Aa
1086JBe67oLzdtvgCUFAA2qfrKzWRZzAdhUirQUZgySZk+Xq1pBAkEA/kZG0A6roTSM
109BVnx7LnPfsycKUsTumorpXiylZJjTi9XtmzxhrYN6wgZlDOOwOLgSQhszGpxVoMD
110u3gByT1b2QJBAPtL3mSKdvwRu/+40zaZLwvSJRxaj0mcE4BJOS6Oqs/hS1xRlrNk
111PpQ7WJ4yM6ZOLnXzm2mKyxm50Mv64109FtMCQQDOqS2KkjHaLowTGVxwC0DijMfr
112I9Lf8sSQk32J5VWCySWf5gGTfEnpmUa41gKTMJIbqZZLucNuDcOtzUaeWZlZAkA8
113ttXigLnCqR486JDPTi9ZscoZkZ+w7y6e/hH8t6d5Vjt48JVyfjPIaJY+km58LcN3
1146AWSeGAdtRFHVzR7oHjVAkB4hutvxiOeiIVQNBhM6RSI9aBPMI21DoX2JRoxvNW2
115cbvAhow217X9V0dVerEOKxnNYspXRrh36h7k4mQA+sDq
116-----END RSA PRIVATE KEY-----
Alex Gaynore7f51982016-09-11 11:48:14 -0400117"""
Rick Dean94e46fd2009-07-18 14:51:24 -0500118
Alex Gaynore7f51982016-09-11 11:48:14 -0400119intermediate_cert_pem = b"""-----BEGIN CERTIFICATE-----
Stephen Holsapple0d9815f2014-08-27 19:36:53 -0700120MIICVzCCAcCgAwIBAgIRAMPzhm6//0Y/g2pmnHR2C4cwDQYJKoZIhvcNAQENBQAw
121WDELMAkGA1UEBhMCVVMxCzAJBgNVBAgTAklMMRAwDgYDVQQHEwdDaGljYWdvMRAw
122DgYDVQQKEwdUZXN0aW5nMRgwFgYDVQQDEw9UZXN0aW5nIFJvb3QgQ0EwHhcNMTQw
123ODI4MDIwNDA4WhcNMjQwODI1MDIwNDA4WjBmMRUwEwYDVQQDEwxpbnRlcm1lZGlh
124dGUxDDAKBgNVBAoTA29yZzERMA8GA1UECxMIb3JnLXVuaXQxCzAJBgNVBAYTAlVT
125MQswCQYDVQQIEwJDQTESMBAGA1UEBxMJU2FuIERpZWdvMIGfMA0GCSqGSIb3DQEB
126AQUAA4GNADCBiQKBgQDYcEQw5lfbEQRjr5Yy4yxAHGV0b9Al+Lmu7wLHMkZ/ZMmK
127FGIbljbviiD1Nz97Oh2cpB91YwOXOTN2vXHq26S+A5xe8z/QJbBsyghMur88CjdT
12821H2qwMa+r5dCQwEhuGIiZ3KbzB/n4DTMYI5zy4IYPv0pjxShZn4aZTCCK2IUwID
129AQABoxMwETAPBgNVHRMBAf8EBTADAQH/MA0GCSqGSIb3DQEBDQUAA4GBAPIWSkLX
130QRMApOjjyC+tMxumT5e2pMqChHmxobQK4NMdrf2VCx+cRT6EmY8sK3/Xl/X8UBQ+
1319n5zXb1ZwhW/sTWgUvmOceJ4/XVs9FkdWOOn1J0XBch9ZIiFe/s5ASIgG7fUdcUF
1329mAWS6FK2ca3xIh5kIupCXOFa0dPvlw/YUFT
133-----END CERTIFICATE-----
Alex Gaynore7f51982016-09-11 11:48:14 -0400134"""
Stephen Holsapple0d9815f2014-08-27 19:36:53 -0700135
Alex Gaynore7f51982016-09-11 11:48:14 -0400136intermediate_key_pem = b"""-----BEGIN RSA PRIVATE KEY-----
Stephen Holsapple0d9815f2014-08-27 19:36:53 -0700137MIICWwIBAAKBgQDYcEQw5lfbEQRjr5Yy4yxAHGV0b9Al+Lmu7wLHMkZ/ZMmKFGIb
138ljbviiD1Nz97Oh2cpB91YwOXOTN2vXHq26S+A5xe8z/QJbBsyghMur88CjdT21H2
139qwMa+r5dCQwEhuGIiZ3KbzB/n4DTMYI5zy4IYPv0pjxShZn4aZTCCK2IUwIDAQAB
140AoGAfSZVV80pSeOKHTYfbGdNY/jHdU9eFUa/33YWriXU+77EhpIItJjkRRgivIfo
141rhFJpBSGmDLblaqepm8emsXMeH4+2QzOYIf0QGGP6E6scjTt1PLqdqKfVJ1a2REN
142147cujNcmFJb/5VQHHMpaPTgttEjlzuww4+BCDPsVRABWrkCQQD3loH36nLoQTtf
143+kQq0T6Bs9/UWkTAGo0ND81ALj0F8Ie1oeZg6RNT96RxZ3aVuFTESTv6/TbjWywO
144wdzlmV1vAkEA38rTJ6PTwaJlw5OttdDzAXGPB9tDmzh9oSi7cHwQQXizYd8MBYx4
145sjHUKD3dCQnb1dxJFhd3BT5HsnkRMbVZXQJAbXduH17ZTzcIOXc9jHDXYiFVZV5D
14652vV0WCbLzVCZc3jMrtSUKa8lPN5EWrdU3UchWybyG0MR5mX8S5lrF4SoQJAIyUD
147DBKaSqpqONCUUx1BTFS9FYrFjzbL4+c1qHCTTPTblt8kUCrDOZjBrKAqeiTmNSum
148/qUot9YUBF8m6BuGsQJATHHmdFy/fG1VLkyBp49CAa8tN3Z5r/CgTznI4DfMTf4C
149NbRHn2UmYlwQBa+L5lg9phewNe8aEwpPyPLoV85U8Q==
150-----END RSA PRIVATE KEY-----
Alex Gaynore7f51982016-09-11 11:48:14 -0400151"""
Stephen Holsapple0d9815f2014-08-27 19:36:53 -0700152
Alex Gaynore7f51982016-09-11 11:48:14 -0400153server_cert_pem = b"""-----BEGIN CERTIFICATE-----
Paul Kehrera40898b2017-06-11 16:30:58 -1000154MIICJDCCAY2gAwIBAgIJAJn/HpR21r/8MA0GCSqGSIb3DQEBBQUAMFgxCzAJBgNV
155BAYTAlVTMQswCQYDVQQIDAJJTDEQMA4GA1UEBwwHQ2hpY2FnbzEQMA4GA1UECgwH
156VGVzdGluZzEYMBYGA1UEAwwPVGVzdGluZyBSb290IENBMB4XDTE3MDYxMjAwMTA1
157N1oXDTM3MDYwNzAwMTA1N1owGDEWMBQGA1UEAwwNbG92ZWx5IHNlcnZlcjCBnzAN
158BgkqhkiG9w0BAQEFAAOBjQAwgYkCgYEAvqb4brndXS2kEL84qXbZXE6LYK+UrhNi
15970sdIM/24NVN7tXkPlOXqrMWhFHHml+aeSpPkH5b1vKnY1TcULmEubnNICtvjmZ5
160SGMQn+J+RmBs1SMd0EgY/0wBBQdlrlYp2QYgm8YC+zxTNSqWvhMFZAgHbj6Un5SS
161T8JGBqytjB0CAwEAAaM2MDQwHQYDVR0OBBYEFINVdy1eIfFJDAkk51QJEo3IfgSu
162MBMGA1UdJQQMMAoGCCsGAQUFBwMBMA0GCSqGSIb3DQEBBQUAA4GBAGki1K6WgHHJ
163qC6aY2EowjaWOXLO6jUZIhGk7BA7vMRfNug429AOZ4m5F6OQhzmJmlw67Jyu2FeI
164h0VtBuQoHPtjqZXF59oX6hMMmGLMs9pV0UA3fJs5MYA4/V5ZcQy0Ie0QoJNejLzE
1656V1Qz1rRTYLUyEcpI7ZCmBg2KQQI8YZI
Rick Dean94e46fd2009-07-18 14:51:24 -0500166-----END CERTIFICATE-----
Alex Gaynore7f51982016-09-11 11:48:14 -0400167"""
Rick Dean94e46fd2009-07-18 14:51:24 -0500168
Alex Gaynor03737182020-07-23 20:40:46 -0400169server_key_pem = normalize_privatekey_pem(
170 b"""-----BEGIN RSA PRIVATE KEY-----
Rick Dean94e46fd2009-07-18 14:51:24 -0500171MIICWwIBAAKBgQC+pvhuud1dLaQQvzipdtlcTotgr5SuE2LvSx0gz/bg1U3u1eQ+
172U5eqsxaEUceaX5p5Kk+QflvW8qdjVNxQuYS5uc0gK2+OZnlIYxCf4n5GYGzVIx3Q
173SBj/TAEFB2WuVinZBiCbxgL7PFM1Kpa+EwVkCAduPpSflJJPwkYGrK2MHQIDAQAB
174AoGAbwuZ0AR6JveahBaczjfnSpiFHf+mve2UxoQdpyr6ROJ4zg/PLW5K/KXrC48G
175j6f3tXMrfKHcpEoZrQWUfYBRCUsGD5DCazEhD8zlxEHahIsqpwA0WWssJA2VOLEN
176j6DuV2pCFbw67rfTBkTSo32ahfXxEKev5KswZk0JIzH3ooECQQDgzS9AI89h0gs8
177Dt+1m11Rzqo3vZML7ZIyGApUzVan+a7hbc33nbGRkAXjHaUBJO31it/H6dTO+uwX
178msWwNG5ZAkEA2RyFKs5xR5USTFaKLWCgpH/ydV96KPOpBND7TKQx62snDenFNNbn
179FwwOhpahld+vqhYk+pfuWWUpQciE+Bu7ZQJASjfT4sQv4qbbKK/scePicnDdx9th
1804e1EeB9xwb+tXXXUo/6Bor/AcUNwfiQ6Zt9PZOK9sR3lMZSsP7rMi7kzuQJABie6
1811sXXjFH7nNJvRG4S39cIxq8YRYTy68II/dlB2QzGpKxV/POCxbJ/zu0CU79tuYK7
182NaeNCFfH3aeTrX0LyQJAMBWjWmeKM2G2sCExheeQK0ROnaBC8itCECD4Jsve4nqf
183r50+LF74iLXFwqysVCebPKMOpDWp/qQ1BbJQIPs7/A==
184-----END RSA PRIVATE KEY-----
Alex Gaynor03737182020-07-23 20:40:46 -0400185"""
186)
Rick Dean94e46fd2009-07-18 14:51:24 -0500187
Alex Gaynore7f51982016-09-11 11:48:14 -0400188intermediate_server_cert_pem = b"""-----BEGIN CERTIFICATE-----
Stephen Holsapple0d9815f2014-08-27 19:36:53 -0700189MIICWDCCAcGgAwIBAgIRAPQFY9jfskSihdiNSNdt6GswDQYJKoZIhvcNAQENBQAw
190ZjEVMBMGA1UEAxMMaW50ZXJtZWRpYXRlMQwwCgYDVQQKEwNvcmcxETAPBgNVBAsT
191CG9yZy11bml0MQswCQYDVQQGEwJVUzELMAkGA1UECBMCQ0ExEjAQBgNVBAcTCVNh
192biBEaWVnbzAeFw0xNDA4MjgwMjEwNDhaFw0yNDA4MjUwMjEwNDhaMG4xHTAbBgNV
193BAMTFGludGVybWVkaWF0ZS1zZXJ2aWNlMQwwCgYDVQQKEwNvcmcxETAPBgNVBAsT
194CG9yZy11bml0MQswCQYDVQQGEwJVUzELMAkGA1UECBMCQ0ExEjAQBgNVBAcTCVNh
195biBEaWVnbzCBnzANBgkqhkiG9w0BAQEFAAOBjQAwgYkCgYEAqpJZygd+w1faLOr1
196iOAmbBhx5SZWcTCZ/ZjHQTJM7GuPT624QkqsixFghRKdDROwpwnAP7gMRukLqiy4
197+kRuGT5OfyGggL95i2xqA+zehjj08lSTlvGHpePJgCyTavIy5+Ljsj4DKnKyuhxm
198biXTRrH83NDgixVkObTEmh/OVK0CAwEAATANBgkqhkiG9w0BAQ0FAAOBgQBa0Npw
199UkzjaYEo1OUE1sTI6Mm4riTIHMak4/nswKh9hYup//WVOlr/RBSBtZ7Q/BwbjobN
2003bfAtV7eSAqBsfxYXyof7G1ALANQERkq3+oyLP1iVt08W1WOUlIMPhdCF/QuCwy6
201x9MJLhUCGLJPM+O2rAPWVD9wCmvq10ALsiH3yA==
202-----END CERTIFICATE-----
Alex Gaynore7f51982016-09-11 11:48:14 -0400203"""
Stephen Holsapple0d9815f2014-08-27 19:36:53 -0700204
Alex Gaynore7f51982016-09-11 11:48:14 -0400205intermediate_server_key_pem = b"""-----BEGIN RSA PRIVATE KEY-----
Stephen Holsapple0d9815f2014-08-27 19:36:53 -0700206MIICXAIBAAKBgQCqklnKB37DV9os6vWI4CZsGHHlJlZxMJn9mMdBMkzsa49PrbhC
207SqyLEWCFEp0NE7CnCcA/uAxG6QuqLLj6RG4ZPk5/IaCAv3mLbGoD7N6GOPTyVJOW
2088Yel48mALJNq8jLn4uOyPgMqcrK6HGZuJdNGsfzc0OCLFWQ5tMSaH85UrQIDAQAB
209AoGAIQ594j5zna3/9WaPsTgnmhlesVctt4AAx/n827DA4ayyuHFlXUuVhtoWR5Pk
2105ezj9mtYW8DyeCegABnsu2vZni/CdvU6uiS1Hv6qM1GyYDm9KWgovIP9rQCDSGaz
211d57IWVGxx7ODFkm3gN5nxnSBOFVHytuW1J7FBRnEsehRroECQQDXHFOv82JuXDcz
212z3+4c74IEURdOHcbycxlppmK9kFqm5lsUdydnnGW+mvwDk0APOB7Wg7vyFyr393e
213dpmBDCzNAkEAyv6tVbTKUYhSjW+QhabJo896/EqQEYUmtMXxk4cQnKeR/Ao84Rkf
214EqD5IykMUfUI0jJU4DGX+gWZ10a7kNbHYQJAVFCuHNFxS4Cpwo0aqtnzKoZaHY/8
215X9ABZfafSHCtw3Op92M+7ikkrOELXdS9KdKyyqbKJAKNEHF3LbOfB44WIQJAA2N4
2169UNNVUsXRbElEnYUS529CdUczo4QdVgQjkvk5RiPAUwSdBd9Q0xYnFOlFwEmIowg
217ipWJWe0aAlP18ZcEQQJBAL+5lekZ/GUdQoZ4HAsN5a9syrzavJ9VvU1KOOPorPZK
218nMRZbbQgP+aSB7yl6K0gaLaZ8XaK0pjxNBh6ASqg9f4=
219-----END RSA PRIVATE KEY-----
Alex Gaynore7f51982016-09-11 11:48:14 -0400220"""
Stephen Holsapple0d9815f2014-08-27 19:36:53 -0700221
Alex Gaynore7f51982016-09-11 11:48:14 -0400222client_cert_pem = b"""-----BEGIN CERTIFICATE-----
Paul Kehrera40898b2017-06-11 16:30:58 -1000223MIICIjCCAYugAwIBAgIJAKxpFI5lODkjMA0GCSqGSIb3DQEBBQUAMFgxCzAJBgNV
224BAYTAlVTMQswCQYDVQQIDAJJTDEQMA4GA1UEBwwHQ2hpY2FnbzEQMA4GA1UECgwH
225VGVzdGluZzEYMBYGA1UEAwwPVGVzdGluZyBSb290IENBMB4XDTE3MDYxMjAwMDQx
226M1oXDTM3MDYwNzAwMDQxM1owFjEUMBIGA1UEAwwLdWdseSBjbGllbnQwgZ8wDQYJ
227KoZIhvcNAQEBBQADgY0AMIGJAoGBAMBmH9JG02bme0xPipvpjMSlOugyWrauf4at
228EdGJn7GQLD8IY2Fu0+Kvv9DFpSPboFKZCsfDVsYoRs+xaJbtt1dJ6ymX7EqKS7gb
2298q+eeZZ14keqyJd5Rm2q6swQtw9ADD3E8cS6GqpQm+8SgxOycISoYz7sO1ugJFqN
230jId+W4BFAgMBAAGjNjA0MB0GA1UdDgQWBBSDVXctXiHxSQwJJOdUCRKNyH4ErjAT
231BgNVHSUEDDAKBggrBgEFBQcDAjANBgkqhkiG9w0BAQUFAAOBgQAMqcHyweaCOZNN
232dWQQOsBKQlL5wqVVZwucHPWqobjxpULKy9gS2ha2zbgkXcB/BnBOSwe0Fm+MJV0T
233NbnTghcGJNpEH7VKn4xSLvIGZmnZZWgxeIB16z4GhpkK2fShBJ+6GKZjsgjT0lSH
234JRgjHbWutZfZvbSHXr9n7PIphG1Ojg==
Rick Dean94e46fd2009-07-18 14:51:24 -0500235-----END CERTIFICATE-----
Alex Gaynore7f51982016-09-11 11:48:14 -0400236"""
Rick Dean94e46fd2009-07-18 14:51:24 -0500237
Alex Gaynor03737182020-07-23 20:40:46 -0400238client_key_pem = normalize_privatekey_pem(
239 b"""-----BEGIN RSA PRIVATE KEY-----
Rick Dean94e46fd2009-07-18 14:51:24 -0500240MIICXgIBAAKBgQDAZh/SRtNm5ntMT4qb6YzEpTroMlq2rn+GrRHRiZ+xkCw/CGNh
241btPir7/QxaUj26BSmQrHw1bGKEbPsWiW7bdXSespl+xKiku4G/KvnnmWdeJHqsiX
242eUZtqurMELcPQAw9xPHEuhqqUJvvEoMTsnCEqGM+7DtboCRajYyHfluARQIDAQAB
243AoGATkZ+NceY5Glqyl4mD06SdcKfV65814vg2EL7V9t8+/mi9rYL8KztSXGlQWPX
244zuHgtRoMl78yQ4ZJYOBVo+nsx8KZNRCEBlE19bamSbQLCeQMenWnpeYyQUZ908gF
245h6L9qsFVJepgA9RDgAjyDoS5CaWCdCCPCH2lDkdcqC54SVUCQQDseuduc4wi8h4t
246V8AahUn9fn9gYfhoNuM0gdguTA0nPLVWz4hy1yJiWYQe0H7NLNNTmCKiLQaJpAbb
247TC6vE8C7AkEA0Ee8CMJUc20BnGEmxwgWcVuqFWaKCo8jTH1X38FlATUsyR3krjW2
248dL3yDD9NwHxsYP7nTKp/U8MV7U9IBn4y/wJBAJl7H0/BcLeRmuJk7IqJ7b635iYB
249D/9beFUw3MUXmQXZUfyYz39xf6CDZsu1GEdEC5haykeln3Of4M9d/4Kj+FcCQQCY
250si6xwT7GzMDkk/ko684AV3KPc/h6G0yGtFIrMg7J3uExpR/VdH2KgwMkZXisSMvw
251JJEQjOMCVsEJlRk54WWjAkEAzoZNH6UhDdBK5F38rVt/y4SEHgbSfJHIAmPS32Kq
252f6GGcfNpip0Uk7q7udTKuX7Q/buZi/C4YW7u3VKAquv9NA==
253-----END RSA PRIVATE KEY-----
Alex Gaynor03737182020-07-23 20:40:46 -0400254"""
255)
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -0400256
Alex Gaynore7f51982016-09-11 11:48:14 -0400257cleartextCertificatePEM = b"""-----BEGIN CERTIFICATE-----
Paul Kehrera40898b2017-06-11 16:30:58 -1000258MIIC6TCCAlKgAwIBAgIIPQzE4MbeufQwDQYJKoZIhvcNAQEFBQAwWDELMAkGA1UE
Jean-Paul Calderone20131f52009-04-01 12:05:45 -0400259BhMCVVMxCzAJBgNVBAgTAklMMRAwDgYDVQQHEwdDaGljYWdvMRAwDgYDVQQKEwdU
Paul Kehrera40898b2017-06-11 16:30:58 -1000260ZXN0aW5nMRgwFgYDVQQDEw9UZXN0aW5nIFJvb3QgQ0EwHhcNMTcwNjExMjIzMjU5
261WhcNMzcwNjA2MjIzMjU5WjBYMQswCQYDVQQGEwJVUzELMAkGA1UECBMCSUwxEDAO
262BgNVBAcTB0NoaWNhZ28xEDAOBgNVBAoTB1Rlc3RpbmcxGDAWBgNVBAMTD1Rlc3Rp
263bmcgUm9vdCBDQTCBnzANBgkqhkiG9w0BAQEFAAOBjQAwgYkCgYEA+ZpC6Yu6ukTn
264bu5IQd0vWmpwNGZbO773xjpgfNP8nspYRqbIwI1np9FbUkJHvzZRDxrTt/LbFewr
265LhZ0prHIbwJxq3CZe+m9FDjh1IA0yKEcQukA1N3JWnoMLKwQPrCRAW6seUXV2yER
266onDxv/KkOGZtUijoKLXG8ImqK9ssWdsCAwEAAaOBuzCBuDAdBgNVHQ4EFgQUg1V3
267LV4h8UkMCSTnVAkSjch+BK4wgYgGA1UdIwSBgDB+gBSDVXctXiHxSQwJJOdUCRKN
268yH4ErqFcpFowWDELMAkGA1UEBhMCVVMxCzAJBgNVBAgTAklMMRAwDgYDVQQHEwdD
269aGljYWdvMRAwDgYDVQQKEwdUZXN0aW5nMRgwFgYDVQQDEw9UZXN0aW5nIFJvb3Qg
270Q0GCCD0MxODG3rn0MAwGA1UdEwQFMAMBAf8wDQYJKoZIhvcNAQEFBQADgYEANFYQ
271R+T70VcZ+SnvURnwviFgCXeedBzCr21meo+DNHbkp2gudB9W8Xrned/wtUBVymy9
272gjB5jINfU7Lci0H57Evsw96UJJVfhXdUMHpqt1RGCoEd9FWnrDyrSy0NysnBT2bH
273lEqxh3aFEUx9IOQ4sgnx1/NOFXBpkRtivl6O0Ec=
Jean-Paul Calderone20131f52009-04-01 12:05:45 -0400274-----END CERTIFICATE-----
Alex Gaynore7f51982016-09-11 11:48:14 -0400275"""
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -0400276
Alex Gaynor03737182020-07-23 20:40:46 -0400277cleartextPrivateKeyPEM = normalize_privatekey_pem(
278 b"""\
Jean-Paul Calderoned50d2042011-05-04 17:00:49 -0400279-----BEGIN RSA PRIVATE KEY-----
Jean-Paul Calderone20131f52009-04-01 12:05:45 -0400280MIICXQIBAAKBgQD5mkLpi7q6ROdu7khB3S9aanA0Zls7vvfGOmB80/yeylhGpsjA
281jWen0VtSQke/NlEPGtO38tsV7CsuFnSmschvAnGrcJl76b0UOOHUgDTIoRxC6QDU
2823claegwsrBA+sJEBbqx5RdXbIRGicPG/8qQ4Zm1SKOgotcbwiaor2yxZ2wIDAQAB
283AoGBAPCgMpmLxzwDaUmcFbTJUvlLW1hoxNNYSu2jIZm1k/hRAcE60JYwvBkgz3UB
284yMEh0AtLxYe0bFk6EHah11tMUPgscbCq73snJ++8koUw+csk22G65hOs51bVb7Aa
2856JBe67oLzdtvgCUFAA2qfrKzWRZzAdhUirQUZgySZk+Xq1pBAkEA/kZG0A6roTSM
286BVnx7LnPfsycKUsTumorpXiylZJjTi9XtmzxhrYN6wgZlDOOwOLgSQhszGpxVoMD
287u3gByT1b2QJBAPtL3mSKdvwRu/+40zaZLwvSJRxaj0mcE4BJOS6Oqs/hS1xRlrNk
288PpQ7WJ4yM6ZOLnXzm2mKyxm50Mv64109FtMCQQDOqS2KkjHaLowTGVxwC0DijMfr
289I9Lf8sSQk32J5VWCySWf5gGTfEnpmUa41gKTMJIbqZZLucNuDcOtzUaeWZlZAkA8
290ttXigLnCqR486JDPTi9ZscoZkZ+w7y6e/hH8t6d5Vjt48JVyfjPIaJY+km58LcN3
2916AWSeGAdtRFHVzR7oHjVAkB4hutvxiOeiIVQNBhM6RSI9aBPMI21DoX2JRoxvNW2
292cbvAhow217X9V0dVerEOKxnNYspXRrh36h7k4mQA+sDq
293-----END RSA PRIVATE KEY-----
Alex Gaynor03737182020-07-23 20:40:46 -0400294"""
295)
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -0400296
Alex Gaynore7f51982016-09-11 11:48:14 -0400297cleartextCertificateRequestPEM = b"""-----BEGIN CERTIFICATE REQUEST-----
Jean-Paul Calderoneadd7bf02010-08-22 17:38:30 -0400298MIIBnjCCAQcCAQAwXjELMAkGA1UEBhMCVVMxCzAJBgNVBAgTAklMMRAwDgYDVQQH
299EwdDaGljYWdvMRcwFQYDVQQKEw5NeSBDb21wYW55IEx0ZDEXMBUGA1UEAxMORnJl
300ZGVyaWNrIERlYW4wgZ8wDQYJKoZIhvcNAQEBBQADgY0AMIGJAoGBANp6Y17WzKSw
301BsUWkXdqg6tnXy8H8hA1msCMWpc+/2KJ4mbv5NyD6UD+/SqagQqulPbF/DFea9nA
302E0zhmHJELcM8gUTIlXv/cgDWnmK4xj8YkjVUiCdqKRAKeuzLG1pGmwwF5lGeJpXN
303xQn5ecR0UYSOWj6TTGXB9VyUMQzCClcBAgMBAAGgADANBgkqhkiG9w0BAQUFAAOB
304gQAAJGuF/R/GGbeC7FbFW+aJgr9ee0Xbl6nlhu7pTe67k+iiKT2dsl2ti68MVTnu
305Vrb3HUNqOkiwsJf6kCtq5oPn3QVYzTa76Dt2y3Rtzv6boRSlmlfrgS92GNma8JfR
306oICQk3nAudi6zl1Dix3BCv1pUp5KMtGn3MeDEi6QFGy2rA==
307-----END CERTIFICATE REQUEST-----
Alex Gaynore7f51982016-09-11 11:48:14 -0400308"""
Rick Dean5b7b6372009-04-01 11:34:06 -0500309
Alex Gaynore7f51982016-09-11 11:48:14 -0400310encryptedPrivateKeyPEM = b"""-----BEGIN RSA PRIVATE KEY-----
Jean-Paul Calderone20131f52009-04-01 12:05:45 -0400311Proc-Type: 4,ENCRYPTED
312DEK-Info: DES-EDE3-CBC,9573604A18579E9E
Jean-Paul Calderone5ef86512008-04-26 19:06:28 -0400313
Jean-Paul Calderone20131f52009-04-01 12:05:45 -0400314SHOho56WxDkT0ht10UTeKc0F5u8cqIa01kzFAmETw0MAs8ezYtK15NPdCXUm3X/2
315a17G7LSF5bkxOgZ7vpXyMzun/owrj7CzvLxyncyEFZWvtvzaAhPhvTJtTIB3kf8B
3168+qRcpTGK7NgXEgYBW5bj1y4qZkD4zCL9o9NQzsKI3Ie8i0239jsDOWR38AxjXBH
317mGwAQ4Z6ZN5dnmM4fhMIWsmFf19sNyAML4gHenQCHhmXbjXeVq47aC2ProInJbrm
318+00TcisbAQ40V9aehVbcDKtS4ZbMVDwncAjpXpcncC54G76N6j7F7wL7L/FuXa3A
319fvSVy9n2VfF/pJ3kYSflLHH2G/DFxjF7dl0GxhKPxJjp3IJi9VtuvmN9R2jZWLQF
320tfC8dXgy/P9CfFQhlinqBTEwgH0oZ/d4k4NVFDSdEMaSdmBAjlHpc+Vfdty3HVnV
321rKXj//wslsFNm9kIwJGIgKUa/n2jsOiydrsk1mgH7SmNCb3YHgZhbbnq0qLat/HC
322gHDt3FHpNQ31QzzL3yrenFB2L9osIsnRsDTPFNi4RX4SpDgNroxOQmyzCCV6H+d4
323o1mcnNiZSdxLZxVKccq0AfRpHqpPAFnJcQHP6xyT9MZp6fBa0XkxDnt9kNU8H3Qw
3247SJWZ69VXjBUzMlQViLuaWMgTnL+ZVyFZf9hTF7U/ef4HMLMAVNdiaGG+G+AjCV/
325MbzjS007Oe4qqBnCWaFPSnJX6uLApeTbqAxAeyCql56ULW5x6vDMNC3dwjvS/CEh
32611n8RkgFIQA0AhuKSIg3CbuartRsJnWOLwgLTzsrKYL4yRog1RJrtw==
327-----END RSA PRIVATE KEY-----
Alex Gaynore7f51982016-09-11 11:48:14 -0400328"""
Jean-Paul Calderone55ec1712009-05-13 14:14:30 -0400329
Alex Gaynore7f51982016-09-11 11:48:14 -0400330encryptedPrivateKeyPEMPassphrase = b"foobar"
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -0400331
Cory Benfield6492f7c2015-10-27 16:57:58 +0900332
Alex Gaynore7f51982016-09-11 11:48:14 -0400333cleartextPublicKeyPEM = b"""-----BEGIN PUBLIC KEY-----
Cory Benfield6492f7c2015-10-27 16:57:58 +0900334MIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8AMIIBCgKCAQEAxszlc+b71LvlLS0ypt/l
335gT/JzSVJtnEqw9WUNGeiChywX2mmQLHEt7KP0JikqUFZOtPclNY823Q4pErMTSWC
33690qlUxI47vNJbXGRfmO2q6Zfw6SE+E9iUb74xezbOJLjBuUIkQzEKEFV+8taiRV+
337ceg1v01yCT2+OjhQW3cxG42zxyRFmqesbQAUWgS3uhPrUQqYQUEiTmVhh4FBUKZ5
338XIneGUpX1S7mXRxTLH6YzRoGFqRoc9A0BBNcoXHTWnxV215k4TeHMFYE5RG0KYAS
3398Xk5iKICEXwnZreIt3jyygqoOKsKZMK/Zl2VhMGhJR6HXRpQCyASzEG7bgtROLhL
340ywIDAQAB
341-----END PUBLIC KEY-----
Alex Gaynore7f51982016-09-11 11:48:14 -0400342"""
Cory Benfield6492f7c2015-10-27 16:57:58 +0900343
Jean-Paul Calderonedc138fa2009-06-27 14:32:07 -0400344# Some PKCS#7 stuff. Generated with the openssl command line:
345#
346# openssl crl2pkcs7 -inform pem -outform pem -certfile s.pem -nocrl
347#
348# with a certificate and key (but the key should be irrelevant) in s.pem
Alex Gaynore7f51982016-09-11 11:48:14 -0400349pkcs7Data = b"""\
Jean-Paul Calderonedc138fa2009-06-27 14:32:07 -0400350-----BEGIN PKCS7-----
351MIIDNwYJKoZIhvcNAQcCoIIDKDCCAyQCAQExADALBgkqhkiG9w0BBwGgggMKMIID
352BjCCAm+gAwIBAgIBATANBgkqhkiG9w0BAQQFADB7MQswCQYDVQQGEwJTRzERMA8G
353A1UEChMITTJDcnlwdG8xFDASBgNVBAsTC00yQ3J5cHRvIENBMSQwIgYDVQQDExtN
354MkNyeXB0byBDZXJ0aWZpY2F0ZSBNYXN0ZXIxHTAbBgkqhkiG9w0BCQEWDm5ncHNA
355cG9zdDEuY29tMB4XDTAwMDkxMDA5NTEzMFoXDTAyMDkxMDA5NTEzMFowUzELMAkG
356A1UEBhMCU0cxETAPBgNVBAoTCE0yQ3J5cHRvMRIwEAYDVQQDEwlsb2NhbGhvc3Qx
357HTAbBgkqhkiG9w0BCQEWDm5ncHNAcG9zdDEuY29tMFwwDQYJKoZIhvcNAQEBBQAD
358SwAwSAJBAKy+e3dulvXzV7zoTZWc5TzgApr8DmeQHTYC8ydfzH7EECe4R1Xh5kwI
359zOuuFfn178FBiS84gngaNcrFi0Z5fAkCAwEAAaOCAQQwggEAMAkGA1UdEwQCMAAw
360LAYJYIZIAYb4QgENBB8WHU9wZW5TU0wgR2VuZXJhdGVkIENlcnRpZmljYXRlMB0G
361A1UdDgQWBBTPhIKSvnsmYsBVNWjj0m3M2z0qVTCBpQYDVR0jBIGdMIGagBT7hyNp
36265w6kxXlxb8pUU/+7Sg4AaF/pH0wezELMAkGA1UEBhMCU0cxETAPBgNVBAoTCE0y
363Q3J5cHRvMRQwEgYDVQQLEwtNMkNyeXB0byBDQTEkMCIGA1UEAxMbTTJDcnlwdG8g
364Q2VydGlmaWNhdGUgTWFzdGVyMR0wGwYJKoZIhvcNAQkBFg5uZ3BzQHBvc3QxLmNv
365bYIBADANBgkqhkiG9w0BAQQFAAOBgQA7/CqT6PoHycTdhEStWNZde7M/2Yc6BoJu
366VwnW8YxGO8Sn6UJ4FeffZNcYZddSDKosw8LtPOeWoK3JINjAk5jiPQ2cww++7QGG
367/g5NDjxFZNDJP1dGiLAxPW6JXwov4v0FmdzfLOZ01jDcgQQZqEpYlgpuI5JEWUQ9
368Ho4EzbYCOaEAMQA=
369-----END PKCS7-----
Alex Gaynore7f51982016-09-11 11:48:14 -0400370"""
Jean-Paul Calderonedc138fa2009-06-27 14:32:07 -0400371
Alex Gaynor03737182020-07-23 20:40:46 -0400372pkcs7DataASN1 = base64.b64decode(
373 b"""
Alex Gaynor4b9c96a2014-08-14 09:51:48 -0700374MIIDNwYJKoZIhvcNAQcCoIIDKDCCAyQCAQExADALBgkqhkiG9w0BBwGgggMKMIID
375BjCCAm+gAwIBAgIBATANBgkqhkiG9w0BAQQFADB7MQswCQYDVQQGEwJTRzERMA8G
376A1UEChMITTJDcnlwdG8xFDASBgNVBAsTC00yQ3J5cHRvIENBMSQwIgYDVQQDExtN
377MkNyeXB0byBDZXJ0aWZpY2F0ZSBNYXN0ZXIxHTAbBgkqhkiG9w0BCQEWDm5ncHNA
378cG9zdDEuY29tMB4XDTAwMDkxMDA5NTEzMFoXDTAyMDkxMDA5NTEzMFowUzELMAkG
379A1UEBhMCU0cxETAPBgNVBAoTCE0yQ3J5cHRvMRIwEAYDVQQDEwlsb2NhbGhvc3Qx
380HTAbBgkqhkiG9w0BCQEWDm5ncHNAcG9zdDEuY29tMFwwDQYJKoZIhvcNAQEBBQAD
381SwAwSAJBAKy+e3dulvXzV7zoTZWc5TzgApr8DmeQHTYC8ydfzH7EECe4R1Xh5kwI
382zOuuFfn178FBiS84gngaNcrFi0Z5fAkCAwEAAaOCAQQwggEAMAkGA1UdEwQCMAAw
383LAYJYIZIAYb4QgENBB8WHU9wZW5TU0wgR2VuZXJhdGVkIENlcnRpZmljYXRlMB0G
384A1UdDgQWBBTPhIKSvnsmYsBVNWjj0m3M2z0qVTCBpQYDVR0jBIGdMIGagBT7hyNp
38565w6kxXlxb8pUU/+7Sg4AaF/pH0wezELMAkGA1UEBhMCU0cxETAPBgNVBAoTCE0y
386Q3J5cHRvMRQwEgYDVQQLEwtNMkNyeXB0byBDQTEkMCIGA1UEAxMbTTJDcnlwdG8g
387Q2VydGlmaWNhdGUgTWFzdGVyMR0wGwYJKoZIhvcNAQkBFg5uZ3BzQHBvc3QxLmNv
388bYIBADANBgkqhkiG9w0BAQQFAAOBgQA7/CqT6PoHycTdhEStWNZde7M/2Yc6BoJu
389VwnW8YxGO8Sn6UJ4FeffZNcYZddSDKosw8LtPOeWoK3JINjAk5jiPQ2cww++7QGG
390/g5NDjxFZNDJP1dGiLAxPW6JXwov4v0FmdzfLOZ01jDcgQQZqEpYlgpuI5JEWUQ9
391Ho4EzbYCOaEAMQA=
Alex Gaynor03737182020-07-23 20:40:46 -0400392"""
393)
Alex Gaynor4b9c96a2014-08-14 09:51:48 -0700394
Alex Gaynore7f51982016-09-11 11:48:14 -0400395crlData = b"""\
Jean-Paul Calderone3eb5cc72010-01-30 15:24:40 -0500396-----BEGIN X509 CRL-----
397MIIBWzCBxTANBgkqhkiG9w0BAQQFADBYMQswCQYDVQQGEwJVUzELMAkGA1UECBMC
398SUwxEDAOBgNVBAcTB0NoaWNhZ28xEDAOBgNVBAoTB1Rlc3RpbmcxGDAWBgNVBAMT
399D1Rlc3RpbmcgUm9vdCBDQRcNMDkwNzI2MDQzNDU2WhcNMTIwOTI3MDI0MTUyWjA8
400MBUCAgOrGA8yMDA5MDcyNTIzMzQ1NlowIwICAQAYDzIwMDkwNzI1MjMzNDU2WjAM
401MAoGA1UdFQQDCgEEMA0GCSqGSIb3DQEBBAUAA4GBAEBt7xTs2htdD3d4ErrcGAw1
4024dKcVnIWTutoI7xxen26Wwvh8VCsT7i/UeP+rBl9rC/kfjWjzQk3/zleaarGTpBT
4030yp4HXRFFoRhhSE/hP+eteaPXRgrsNRLHe9ZDd69wmh7J1wMDb0m81RG7kqcbsid
404vrzEeLDRiiPl92dyyWmu
405-----END X509 CRL-----
Alex Gaynore7f51982016-09-11 11:48:14 -0400406"""
Jean-Paul Calderonee890db32010-08-22 16:55:15 -0400407
Alex Gaynore7f51982016-09-11 11:48:14 -0400408crlDataUnsupportedExtension = b"""\
Paul Kehrer5e3dd4c2016-03-11 09:58:28 -0400409-----BEGIN X509 CRL-----
410MIIGRzCCBS8CAQIwDQYJKoZIhvcNAQELBQAwJzELMAkGA1UEBhMCVVMxGDAWBgNV
411BAMMD2NyeXB0b2dyYXBoeS5pbxgPMjAxNTAxMDEwMDAwMDBaGA8yMDE2MDEwMTAw
412MDAwMFowggTOMBQCAQAYDzIwMTUwMTAxMDAwMDAwWjByAgEBGA8yMDE1MDEwMTAw
413MDAwMFowXDAYBgNVHRgEERgPMjAxNTAxMDEwMDAwMDBaMDQGA1UdHQQtMCukKTAn
414MQswCQYDVQQGEwJVUzEYMBYGA1UEAwwPY3J5cHRvZ3JhcGh5LmlvMAoGA1UdFQQD
415CgEAMHICAQIYDzIwMTUwMTAxMDAwMDAwWjBcMBgGA1UdGAQRGA8yMDE1MDEwMTAw
416MDAwMFowNAYDVR0dBC0wK6QpMCcxCzAJBgNVBAYTAlVTMRgwFgYDVQQDDA9jcnlw
417dG9ncmFwaHkuaW8wCgYDVR0VBAMKAQEwcgIBAxgPMjAxNTAxMDEwMDAwMDBaMFww
418GAYDVR0YBBEYDzIwMTUwMTAxMDAwMDAwWjA0BgNVHR0ELTArpCkwJzELMAkGA1UE
419BhMCVVMxGDAWBgNVBAMMD2NyeXB0b2dyYXBoeS5pbzAKBgNVHRUEAwoBAjByAgEE
420GA8yMDE1MDEwMTAwMDAwMFowXDAYBgNVHRgEERgPMjAxNTAxMDEwMDAwMDBaMDQG
421A1UdHQQtMCukKTAnMQswCQYDVQQGEwJVUzEYMBYGA1UEAwwPY3J5cHRvZ3JhcGh5
422LmlvMAoGA1UdFQQDCgEDMHICAQUYDzIwMTUwMTAxMDAwMDAwWjBcMBgGA1UdGAQR
423GA8yMDE1MDEwMTAwMDAwMFowNAYDVR0dBC0wK6QpMCcxCzAJBgNVBAYTAlVTMRgw
424FgYDVQQDDA9jcnlwdG9ncmFwaHkuaW8wCgYDVR0VBAMKAQQwcgIBBhgPMjAxNTAx
425MDEwMDAwMDBaMFwwGAYDVR0YBBEYDzIwMTUwMTAxMDAwMDAwWjA0BgNVHR0ELTAr
426pCkwJzELMAkGA1UEBhMCVVMxGDAWBgNVBAMMD2NyeXB0b2dyYXBoeS5pbzAKBgNV
427HRUEAwoBBTByAgEHGA8yMDE1MDEwMTAwMDAwMFowXDAYBgNVHRgEERgPMjAxNTAx
428MDEwMDAwMDBaMDQGA1UdHQQtMCukKTAnMQswCQYDVQQGEwJVUzEYMBYGA1UEAwwP
429Y3J5cHRvZ3JhcGh5LmlvMAoGA1UdFQQDCgEGMHICAQgYDzIwMTUwMTAxMDAwMDAw
430WjBcMBgGA1UdGAQRGA8yMDE1MDEwMTAwMDAwMFowNAYDVR0dBC0wK6QpMCcxCzAJ
431BgNVBAYTAlVTMRgwFgYDVQQDDA9jcnlwdG9ncmFwaHkuaW8wCgYDVR0VBAMKAQgw
432cgIBCRgPMjAxNTAxMDEwMDAwMDBaMFwwGAYDVR0YBBEYDzIwMTUwMTAxMDAwMDAw
433WjA0BgNVHR0ELTArpCkwJzELMAkGA1UEBhMCVVMxGDAWBgNVBAMMD2NyeXB0b2dy
434YXBoeS5pbzAKBgNVHRUEAwoBCTByAgEKGA8yMDE1MDEwMTAwMDAwMFowXDAYBgNV
435HRgEERgPMjAxNTAxMDEwMDAwMDBaMDQGA1UdHQQtMCukKTAnMQswCQYDVQQGEwJV
436UzEYMBYGA1UEAwwPY3J5cHRvZ3JhcGh5LmlvMAoGA1UdFQQDCgEKMC4CAQsYDzIw
437MTUwMTAxMDAwMDAwWjAYMAoGA1UdFQQDCgEBMAoGAyoDBAQDCgEAMA0GCSqGSIb3
438DQEBCwUAA4IBAQBTaloHlPaCZzYee8LxkWej5meiqxQVNWFoVdjesroa+f1FRrH+
439drRU60Nq97KCKf7f9GNN/J3ZIlQmYhmuDqh12f+XLpotoj1ZRfBz2hjFCkJlv+2c
440oWWGNHgA70ndFoVtcmX088SYpX8E3ARATivS4q2h9WlwV6rO93mhg3HGIe3JpcK4
4417BcW6Poi/ut/zsDOkVbI00SqaujRpdmdCTht82MH3ztjyDkI9KYaD/YEweKSrWOz
442SdEILd164bfBeLuplVI+xpmTEMVNpXBlSXl7+xIw9Vk7p7Q1Pa3k/SvhOldYCm6y
443C1xAg/AAq6w78yzYt18j5Mj0s6eeHi1YpHKw
444-----END X509 CRL-----
Alex Gaynore7f51982016-09-11 11:48:14 -0400445"""
Paul Kehrer5e3dd4c2016-03-11 09:58:28 -0400446
Jean-Paul Calderone55ec1712009-05-13 14:14:30 -0400447
448# A broken RSA private key which can be used to test the error path through
449# PKey.check.
Alex Gaynore7f51982016-09-11 11:48:14 -0400450inconsistentPrivateKeyPEM = b"""-----BEGIN RSA PRIVATE KEY-----
Jean-Paul Calderone55ec1712009-05-13 14:14:30 -0400451MIIBPAIBAAJBAKy+e3dulvXzV7zoTZWc5TzgApr8DmeQHTYC8ydfzH7EECe4R1Xh
4525kwIzOuuFfn178FBiS84gngaNcrFi0Z5fAkCAwEaAQJBAIqm/bz4NA1H++Vx5Ewx
453OcKp3w19QSaZAwlGRtsUxrP7436QjnREM3Bm8ygU11BjkPVmtrKm6AayQfCHqJoT
454zIECIQDW0BoMoL0HOYM/mrTLhaykYAVqgIeJsPjvkEhTFXWBuQIhAM3deFAvWNu4
455nklUQ37XsCT2c9tmNt1LAT+slG2JOTTRAiAuXDtC/m3NYVwyHfFm+zKHRzHkClk2
456HjubeEgjpj32AQIhAJqMGTaZVOwevTXvvHwNeH+vRWsAYU/gbx+OQB+7VOcBAiEA
457oolb6NMg/R3enNPvS1O4UU1H8wpaF77L4yiSWlE0p4w=
458-----END RSA PRIVATE KEY-----
Alex Gaynore7f51982016-09-11 11:48:14 -0400459"""
Jean-Paul Calderone55ec1712009-05-13 14:14:30 -0400460
Jean-Paul Calderoneff83cdd2013-08-12 18:05:51 -0400461# certificate with NULL bytes in subjectAltName and common name
462
Alex Gaynore7f51982016-09-11 11:48:14 -0400463nulbyteSubjectAltNamePEM = b"""-----BEGIN CERTIFICATE-----
Jean-Paul Calderoneff83cdd2013-08-12 18:05:51 -0400464MIIE2DCCA8CgAwIBAgIBADANBgkqhkiG9w0BAQUFADCBxTELMAkGA1UEBhMCVVMx
465DzANBgNVBAgMBk9yZWdvbjESMBAGA1UEBwwJQmVhdmVydG9uMSMwIQYDVQQKDBpQ
466eXRob24gU29mdHdhcmUgRm91bmRhdGlvbjEgMB4GA1UECwwXUHl0aG9uIENvcmUg
467RGV2ZWxvcG1lbnQxJDAiBgNVBAMMG251bGwucHl0aG9uLm9yZwBleGFtcGxlLm9y
468ZzEkMCIGCSqGSIb3DQEJARYVcHl0aG9uLWRldkBweXRob24ub3JnMB4XDTEzMDgw
469NzEzMTE1MloXDTEzMDgwNzEzMTI1MlowgcUxCzAJBgNVBAYTAlVTMQ8wDQYDVQQI
470DAZPcmVnb24xEjAQBgNVBAcMCUJlYXZlcnRvbjEjMCEGA1UECgwaUHl0aG9uIFNv
471ZnR3YXJlIEZvdW5kYXRpb24xIDAeBgNVBAsMF1B5dGhvbiBDb3JlIERldmVsb3Bt
472ZW50MSQwIgYDVQQDDBtudWxsLnB5dGhvbi5vcmcAZXhhbXBsZS5vcmcxJDAiBgkq
473hkiG9w0BCQEWFXB5dGhvbi1kZXZAcHl0aG9uLm9yZzCCASIwDQYJKoZIhvcNAQEB
474BQADggEPADCCAQoCggEBALXq7cn7Rn1vO3aA3TrzA5QLp6bb7B3f/yN0CJ2XFj+j
475pHs+Gw6WWSUDpybiiKnPec33BFawq3kyblnBMjBU61ioy5HwQqVkJ8vUVjGIUq3P
476vX/wBmQfzCe4o4uM89gpHyUL9UYGG8oCRa17dgqcv7u5rg0Wq2B1rgY+nHwx3JIv
477KRrgSwyRkGzpN8WQ1yrXlxWjgI9de0mPVDDUlywcWze1q2kwaEPTM3hLAmD1PESA
478oY/n8A/RXoeeRs9i/Pm/DGUS8ZPINXk/yOzsR/XvvkTVroIeLZqfmFpnZeF0cHzL
47908LODkVJJ9zjLdT7SA4vnne4FEbAxDbKAq5qkYzaL4UCAwEAAaOB0DCBzTAMBgNV
480HRMBAf8EAjAAMB0GA1UdDgQWBBSIWlXAUv9hzVKjNQ/qWpwkOCL3XDALBgNVHQ8E
481BAMCBeAwgZAGA1UdEQSBiDCBhYIeYWx0bnVsbC5weXRob24ub3JnAGV4YW1wbGUu
482Y29tgSBudWxsQHB5dGhvbi5vcmcAdXNlckBleGFtcGxlLm9yZ4YpaHR0cDovL251
483bGwucHl0aG9uLm9yZwBodHRwOi8vZXhhbXBsZS5vcmeHBMAAAgGHECABDbgAAAAA
484AAAAAAAAAAEwDQYJKoZIhvcNAQEFBQADggEBAKxPRe99SaghcI6IWT7UNkJw9aO9
485i9eo0Fj2MUqxpKbdb9noRDy2CnHWf7EIYZ1gznXPdwzSN4YCjV5d+Q9xtBaowT0j
486HPERs1ZuytCNNJTmhyqZ8q6uzMLoht4IqH/FBfpvgaeC5tBTnTT0rD5A/olXeimk
487kX4LxlEx5RAvpGB2zZVRGr6LobD9rVK91xuHYNIxxxfEGE8tCCWjp0+3ksri9SXx
488VHWBnbM9YaL32u3hxm8sYB/Yb8WSBavJCWJJqRStVRHM1koZlJmXNx2BX4vPo6iW
489RFEIPQsFZRLrtnCAiEhyT8bC2s/Njlu6ly9gtJZWSV46Q3ZjBL4q9sHKqZQ=
Alex Gaynore7f51982016-09-11 11:48:14 -0400490-----END CERTIFICATE-----"""
Jean-Paul Calderoneff83cdd2013-08-12 18:05:51 -0400491
Alex Gaynore7f51982016-09-11 11:48:14 -0400492large_key_pem = b"""-----BEGIN RSA PRIVATE KEY-----
Colleen Murphye09399b2016-03-01 17:40:49 -0800493MIIJYgIBAAKCAg4AtRua8eIeevRfsj+fkcHr1vmse7Kgb+oX1ssJAvCb1R7JQMnH
494hNDjDP6b3vEkZuPUzlDHymP+cNkXvvi4wJ4miVbO3+SeU4Sh+jmsHeHzGIXat9xW
4959PFtuPM5FQq8zvkY8aDeRYmYwN9JKu4/neMBCBqostYlTEWg+bSytO/qWnyHTHKh
496g0GfaDdqUQPsGQw+J0MgaYIjQOCVASHAPlzbDQLCtuOb587rwTLkZA2GwoHB/LyJ
497BwT0HHgBaiObE12Vs6wi2en0Uu11CiwEuK1KIBcZ2XbE6eApaZa6VH9ysEmUxPt7
498TqyZ4E2oMIYaLPNRxuvozdwTlj1svI1k1FrkaXGc5MTjbgigPMKjIb0T7b/4GNzt
499DhP1LvAeUMnrEi3hJJrcJPXHPqS8/RiytR9xQQW6Sdh4LaA3f9MQm3WSevWage3G
500P8YcCLssOVKsArDjuA52NF5LmYuAeUzXprm4ITDi2oO+0iFBpFW6VPEK4A9vO0Yk
501M/6Wt6tG8zyWhaSH1zFUTwfQ9Yvjyt5w1lrUaAJuoTpwbMVZaDJaEhjOaXU0dyPQ
502jOsePDOQcU6dkeTWsQ3LsHPEEug/X6819TLG5mb3V7bvV9nPFBfTJSCEG794kr90
503XgZfIN71FrdByxLerlbuJI21pPs/nZi9SXi9jAWeiS45/azUxMsyYgJArui+gjq7
504sV1pWiBm6/orAgMBAAECggINQp5L6Yu+oIXBqcSjgq8tfF9M5hd30pLuf/EheHZf
505LA7uAqn2fVGFI2OInIJhXIOT5OxsAXO0xXfltzawZxIFpOFMqajj4F7aYjvSpw9V
506J4EdSiJ/zgv8y1qUdbwEZbHVThRZjoSlrtSzilonBoHZAE0mHtqMz7iRFSk1zz6t
507GunRrvo/lROPentf3TsvHquVNUYI5yaapyO1S7xJhecMIIYSb8nbsHI54FBDGNas
5086mFmpPwI/47/6HTwOEWupnn3NicsjrHzUInOUpaMig4cRR+aP5bjqg/ty8xI8AoN
509evEmCytiWTc+Rvbp1ieN+1jpjN18PjUk80/W7qioHUDt4ieLic8uxWH2VD9SCEnX
510Mpi9tA/FqoZ+2A/3m1OfrY6jiZVE2g+asi9lCK7QVWL39eK82H4rPvtp0/dyo1/i
511ZZz68TXg+m8IgEZcp88hngbkuoTTzpGE73QuPKhGA1uMIimDdqPPB5WP76q+03Oi
512IRR5DfZnqPERed49by0enJ7tKa/gFPZizOV8ALKr0Dp+vfAkxGDLPLBLd2A3//tw
513xg0Q/wltihHSBujv4nYlDXdc5oYyMYZ+Lhc/VuOghHfBq3tgEQ1ECM/ofqXEIdy7
514nVcpZn3Eeq8Jl5CrqxE1ee3NxlzsJHn99yGQpr7mOhW/psJF3XNz80Meg3L4m1T8
515sMBK0GbaassuJhdzb5whAoIBBw48sx1b1WR4XxQc5O/HjHva+l16i2pjUnOUTcDF
516RWmSbIhBm2QQ2rVhO8+fak0tkl6ZnMWW4i0U/X5LOEBbC7+IS8bO3j3Revi+Vw5x
517j96LMlIe9XEub5i/saEWgiz7maCvfzLFU08e1OpT4qPDpP293V400ubA6R7WQTCv
518pBkskGwHeu0l/TuKkVqBFFUTu7KEbps8Gjg7MkJaFriAOv1zis/umK8pVS3ZAM6e
5198w5jfpRccn8Xzta2fRwTB5kCmfxdDsY0oYGxPLRAbW72bORoLGuyyPp/ojeGwoik
520JX9RttErc6FjyZtks370Pa8UL5QskyhMbDhrZW2jFD+RXYM1BrvmZRjbAoIBBwy4
521iFJpuDfytJfz1MWtaL5DqEL/kmiZYAXl6hifNhGu5GAipVIIGsDqEYW4i+VC15aa
5227kOCwz/I5zsB3vSDW96IRs4wXtqEZSibc2W/bqfVi+xcvPPl1ZhQ2EAwa4D/x035
523kyf20ffWOU+1yf2cnijzqs3IzlveUm+meLw5s3Rc+iG7DPWWeCoe1hVwANI1euNc
524pqKwKY905yFyjOje2OgiEU2kS4YME4zGeBys8yo7E42hNnN2EPK6xkkUqzdudLLQ
5258OUlKRTc8AbIf3XG1rpA4VUpTv3hhxGGwCRy6If8zgZQsNYchgNztRGk72Gcb8Dm
526vFSEN3ZtwxU64G3YZzntdcr2WPzxAoIBBw30g6Fgdb/gmVnOpL0//T0ePNDKIMPs
527jVJLaRduhoZgB1Bb9qPUPX0SzRzLZtg1tkZSDjBDoHmOHJfhxUaXt+FLCPPbrE4t
528+nq9n/nBaMM779w9ClqhqLOyGrwKoxjSmhi+TVEHyIxCbXMvPHVHfX9WzxjbcGrN
529ZvRaEVZWo+QlIX8yqdSwqxLk1WtAIRzvlcj7NKum8xBxPed6BNFep/PtgIAmoLT5
530L8wb7EWb2iUdc2KbZ4OaY51lDScqpATgXu3WjXfM+Q52G0mX6Wyd0cjlL711Zrjb
531yLbiueZT94lgIHHRRKtKc8CEqcjkQV5OzABS3P/gQSfgZXBdLKjOpTnKDUq7IBeH
532AoIBBweAOEIAPLQg1QRUrr3xRrYKRwlakgZDii9wJt1l5AgBTICzbTA1vzDJ1JM5
533AqSpCV6w9JWyYVcXK+HLdKBRZLaPPNEQDJ5lOxD6uMziWGl2rg8tj+1xNMWfxiPz
534aTCjoe4EoBUMoTq2gwzRcM2usEQNikXVhnj9Wzaivsaeb4bJ3GRPW5DkrO6JSEtT
535w+gvyMqQM2Hy5k7E7BT46sXVwaj/jZxuqGnebRixXtnp0WixdRIqYWUr1UqLf6hQ
536G7WP2BgoxCMaCmNW8+HMD/xuxucEotoIhZ+GgJKBFoNnjl3BX+qxYdSe9RbL/5Tr
5374It6Jxtj8uETJXEbv9Cg6v1agWPS9YY8RLTBAoIBBwrU2AsAUts6h1LgGLKK3UWZ
538oLH5E+4o+7HqSGRcRodVeN9NBXIYdHHOLeEG6YNGJiJ3bFP5ZQEu9iDsyoFVKJ9O
539Mw/y6dKZuxOCZ+X8FopSROg3yWfdOpAm6cnQZp3WqLNX4n/Q6WvKojfyEiPphjwT
5400ymrUJELXLWJmjUyPoAk6HgC0Gs28ZnEXbyhx7CSbZNFyCU/PNUDZwto3GisIPD3
541le7YjqHugezmjMGlA0sDw5aCXjfbl74vowRFYMO6e3ItApfSRgNV86CDoX74WI/5
542AYU/QVM4wGt8XGT2KwDFJaxYGKsGDMWmXY04dS+WPuetCbouWUusyFwRb9SzFave
543vYeU7Ab/
Alex Gaynore7f51982016-09-11 11:48:14 -0400544-----END RSA PRIVATE KEY-----"""
Colleen Murphye09399b2016-03-01 17:40:49 -0800545
Paul Kehrer72d968b2016-07-29 15:31:04 +0800546ec_private_key_pem = b"""-----BEGIN PRIVATE KEY-----
547MIGHAgEAMBMGByqGSM49AgEGCCqGSM49AwEHBG0wawIBAQQgYirTZSx+5O8Y6tlG
548cka6W6btJiocdrdolfcukSoTEk+hRANCAAQkvPNu7Pa1GcsWU4v7ptNfqCJVq8Cx
549zo0MUVPQgwJ3aJtNM1QMOQUayCrRwfklg+D/rFSUwEUqtZh7fJDiFqz3
550-----END PRIVATE KEY-----
551"""
552
Paul Kehrer59d26252017-07-20 10:45:54 +0200553ec_root_key_pem = b"""-----BEGIN EC PRIVATE KEY-----
554MIGlAgEBBDEAz/HOBFPYLB0jLWeTpJn4Yc4m/C4mdWymVHBjOmnwiPHKT326iYN/
555ZhmSs+RM94RsoAcGBSuBBAAioWQDYgAEwE5vDdla/nLpWAPAQ0yFGqwLuw4BcN2r
556U+sKab5EAEHzLeceRa8ffncYdCXNoVsBcdob1y66CFZMEWLetPTmGapyWkBAs6/L
5578kUlkU9OsE+7IVo4QQJkgV5gM+Dim1XE
558-----END EC PRIVATE KEY-----
559"""
560
561ec_root_cert_pem = b"""-----BEGIN CERTIFICATE-----
562MIICLTCCAbKgAwIBAgIMWW/hwTl6ufz6/WkCMAoGCCqGSM49BAMDMFgxGDAWBgNV
563BAMTD1Rlc3RpbmcgUm9vdCBDQTEQMA4GA1UEChMHVGVzdGluZzEQMA4GA1UEBxMH
564Q2hpY2FnbzELMAkGA1UECBMCSUwxCzAJBgNVBAYTAlVTMCAXDTE3MDcxOTIyNDgz
565M1oYDzk5OTkxMjMxMjM1OTU5WjBYMRgwFgYDVQQDEw9UZXN0aW5nIFJvb3QgQ0Ex
566EDAOBgNVBAoTB1Rlc3RpbmcxEDAOBgNVBAcTB0NoaWNhZ28xCzAJBgNVBAgTAklM
567MQswCQYDVQQGEwJVUzB2MBAGByqGSM49AgEGBSuBBAAiA2IABMBObw3ZWv5y6VgD
568wENMhRqsC7sOAXDdq1PrCmm+RABB8y3nHkWvH353GHQlzaFbAXHaG9cuughWTBFi
5693rT05hmqclpAQLOvy/JFJZFPTrBPuyFaOEECZIFeYDPg4ptVxKNDMEEwDwYDVR0T
570AQH/BAUwAwEB/zAPBgNVHQ8BAf8EBQMDBwQAMB0GA1UdDgQWBBSoTrF0H2m8RDzB
571MnY2KReEPfz7ZjAKBggqhkjOPQQDAwNpADBmAjEA3+G1oVCxGjYX4iUN93QYcNHe
572e3fJQJwX9+KsHRut6qNZDUbvRbtO1YIAwB4UJZjwAjEAtXCPURS5A4McZHnSwgTi
573Td8GMrwKz0557OxxtKN6uVVy4ACFMqEw0zN/KJI1vxc9
574-----END CERTIFICATE-----"""
575
Mrmaxmeier8cd3b172020-03-11 22:03:59 +0100576rsa_p_not_prime_pem = """
577-----BEGIN RSA PRIVATE KEY-----
578MBsCAQACAS0CAQcCAQACAQ8CAQMCAQACAQACAQA=
579-----END RSA PRIVATE KEY-----
580"""
581
Jean-Paul Calderone55ec1712009-05-13 14:14:30 -0400582
Alex Chanc6077062016-11-18 13:53:39 +0000583@pytest.fixture
584def x509_data():
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400585 """
Alex Chanc6077062016-11-18 13:53:39 +0000586 Create a new private key and start a certificate request (for a test
587 to finish in one way or another).
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400588 """
Alex Chanc6077062016-11-18 13:53:39 +0000589 # Basic setup stuff to generate a certificate
590 pkey = PKey()
Alex Gaynor6e9f9762018-05-12 07:44:37 -0400591 pkey.generate_key(TYPE_RSA, 512)
Alex Chanc6077062016-11-18 13:53:39 +0000592 req = X509Req()
593 req.set_pubkey(pkey)
594 # Authority good you have.
595 req.get_subject().commonName = "Yoda root CA"
596 x509 = X509()
597 subject = x509.get_subject()
598 subject.commonName = req.get_subject().commonName
599 x509.set_issuer(subject)
600 x509.set_pubkey(pkey)
601 now = datetime.now()
602 expire = datetime.now() + timedelta(days=100)
603 x509.set_notBefore(now.strftime("%Y%m%d%H%M%SZ").encode())
604 x509.set_notAfter(expire.strftime("%Y%m%d%H%M%SZ").encode())
605 yield pkey, x509
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400606
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400607
Alex Chanc6077062016-11-18 13:53:39 +0000608class TestX509Ext(object):
609 """
610 Tests for `OpenSSL.crypto.X509Extension`.
611 """
Jean-Paul Calderoneef9a3dc2013-03-02 16:33:32 -0800612
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -0400613 def test_str(self):
614 """
Alex Chanc6077062016-11-18 13:53:39 +0000615 The string representation of `X509Extension` instances as
616 returned by `str` includes stuff.
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -0400617 """
618 # This isn't necessarily the best string representation. Perhaps it
619 # will be changed/improved in the future.
Alex Chanc6077062016-11-18 13:53:39 +0000620 assert (
Alex Gaynor03737182020-07-23 20:40:46 -0400621 str(X509Extension(b"basicConstraints", True, b"CA:false"))
622 == "CA:FALSE"
Alex Chanc6077062016-11-18 13:53:39 +0000623 )
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -0400624
Jean-Paul Calderone68649052009-07-17 21:14:27 -0400625 def test_type(self):
626 """
Alex Gaynor01f90a12019-02-07 09:14:48 -0500627 `X509Extension` can be used to create instances of that type.
Jean-Paul Calderone68649052009-07-17 21:14:27 -0400628 """
Alex Chanc6077062016-11-18 13:53:39 +0000629 assert is_consistent_type(
Jean-Paul Calderone40dd0992010-08-22 17:52:07 -0400630 X509Extension,
Alex Gaynor03737182020-07-23 20:40:46 -0400631 "X509Extension",
632 b"basicConstraints",
633 True,
634 b"CA:true",
635 )
Jean-Paul Calderone68649052009-07-17 21:14:27 -0400636
Jean-Paul Calderonee7db4b42008-12-31 13:39:24 -0500637 def test_construction(self):
638 """
Alex Chanc6077062016-11-18 13:53:39 +0000639 `X509Extension` accepts an extension type name, a critical flag,
Alex Gaynor01f90a12019-02-07 09:14:48 -0500640 and an extension value and returns an `X509Extension` instance.
Jean-Paul Calderonee7db4b42008-12-31 13:39:24 -0500641 """
Alex Gaynor03737182020-07-23 20:40:46 -0400642 basic = X509Extension(b"basicConstraints", True, b"CA:true")
Alex Gaynor01f90a12019-02-07 09:14:48 -0500643 assert isinstance(basic, X509Extension)
Jean-Paul Calderonee7db4b42008-12-31 13:39:24 -0500644
Alex Gaynor03737182020-07-23 20:40:46 -0400645 comment = X509Extension(b"nsComment", False, b"pyOpenSSL unit test")
Alex Gaynor01f90a12019-02-07 09:14:48 -0500646 assert isinstance(comment, X509Extension)
Jean-Paul Calderonee7db4b42008-12-31 13:39:24 -0500647
Alex Gaynor03737182020-07-23 20:40:46 -0400648 @pytest.mark.parametrize(
649 "type_name, critical, value",
650 [
651 (b"thisIsMadeUp", False, b"hi"),
652 (b"basicConstraints", False, b"blah blah"),
653 # Exercise a weird one (an extension which uses the r2i method).
654 # This exercises the codepath that requires a non-NULL ctx to be
655 # passed to X509V3_EXT_nconf. It can't work now because we provide
656 # no configuration database. It might be made to work in the
657 # future.
658 (
659 b"proxyCertInfo",
660 True,
661 b"language:id-ppl-anyLanguage,pathlen:1,policy:text:AB",
662 ),
663 ],
664 )
Alex Chanc6077062016-11-18 13:53:39 +0000665 def test_invalid_extension(self, type_name, critical, value):
Jean-Paul Calderonee7db4b42008-12-31 13:39:24 -0500666 """
Alex Chanc6077062016-11-18 13:53:39 +0000667 `X509Extension` raises something if it is passed a bad
668 extension name or value.
669 """
670 with pytest.raises(Error):
671 X509Extension(type_name, critical, value)
672
Alex Gaynor03737182020-07-23 20:40:46 -0400673 @pytest.mark.parametrize("critical_flag", [True, False])
Alex Chanc6077062016-11-18 13:53:39 +0000674 def test_get_critical(self, critical_flag):
675 """
676 `X509ExtensionType.get_critical` returns the value of the
Jean-Paul Calderonee7db4b42008-12-31 13:39:24 -0500677 extension's critical flag.
678 """
Alex Gaynor03737182020-07-23 20:40:46 -0400679 ext = X509Extension(b"basicConstraints", critical_flag, b"CA:true")
Alex Chanc6077062016-11-18 13:53:39 +0000680 assert ext.get_critical() == critical_flag
Jean-Paul Calderonee7db4b42008-12-31 13:39:24 -0500681
Alex Gaynor03737182020-07-23 20:40:46 -0400682 @pytest.mark.parametrize(
683 "short_name, value",
684 [(b"basicConstraints", b"CA:true"), (b"nsComment", b"foo bar")],
685 )
Alex Chanc6077062016-11-18 13:53:39 +0000686 def test_get_short_name(self, short_name, value):
Jean-Paul Calderonef8c5fab2008-12-31 15:53:48 -0500687 """
Alex Chanc6077062016-11-18 13:53:39 +0000688 `X509ExtensionType.get_short_name` returns a string giving the
Alex Gaynor31287502015-09-05 16:11:27 -0400689 short type name of the extension.
Jean-Paul Calderonef8c5fab2008-12-31 15:53:48 -0500690 """
Alex Chanc6077062016-11-18 13:53:39 +0000691 ext = X509Extension(short_name, True, value)
692 assert ext.get_short_name() == short_name
Jean-Paul Calderonef8c5fab2008-12-31 15:53:48 -0500693
Jean-Paul Calderone5a9e4612011-04-01 18:27:45 -0400694 def test_get_data(self):
695 """
Alex Chanc6077062016-11-18 13:53:39 +0000696 `X509Extension.get_data` returns a string giving the data of
Alex Gaynor31287502015-09-05 16:11:27 -0400697 the extension.
Jean-Paul Calderone5a9e4612011-04-01 18:27:45 -0400698 """
Alex Gaynor03737182020-07-23 20:40:46 -0400699 ext = X509Extension(b"basicConstraints", True, b"CA:true")
Jean-Paul Calderone5a9e4612011-04-01 18:27:45 -0400700 # Expect to get back the DER encoded form of CA:true.
Alex Gaynor03737182020-07-23 20:40:46 -0400701 assert ext.get_data() == b"0\x03\x01\x01\xff"
Jean-Paul Calderone5a9e4612011-04-01 18:27:45 -0400702
Alex Chanc6077062016-11-18 13:53:39 +0000703 def test_unused_subject(self, x509_data):
Jean-Paul Calderone5a9e4612011-04-01 18:27:45 -0400704 """
Alex Chanc6077062016-11-18 13:53:39 +0000705 The `subject` parameter to `X509Extension` may be provided for an
706 extension which does not use it and is ignored in this case.
Jean-Paul Calderone5a9e4612011-04-01 18:27:45 -0400707 """
Alex Chanc6077062016-11-18 13:53:39 +0000708 pkey, x509 = x509_data
Jean-Paul Calderone40dd0992010-08-22 17:52:07 -0400709 ext1 = X509Extension(
Alex Gaynor03737182020-07-23 20:40:46 -0400710 b"basicConstraints", False, b"CA:TRUE", subject=x509
711 )
Alex Chanc6077062016-11-18 13:53:39 +0000712 x509.add_extensions([ext1])
Alex Gaynor03737182020-07-23 20:40:46 -0400713 x509.sign(pkey, "sha1")
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400714 # This is a little lame. Can we think of a better way?
Alex Chanc6077062016-11-18 13:53:39 +0000715 text = dump_certificate(FILETYPE_TEXT, x509)
Alex Gaynor03737182020-07-23 20:40:46 -0400716 assert b"X509v3 Basic Constraints:" in text
717 assert b"CA:TRUE" in text
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400718
Alex Chanc6077062016-11-18 13:53:39 +0000719 def test_subject(self, x509_data):
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400720 """
Alex Chanc6077062016-11-18 13:53:39 +0000721 If an extension requires a subject, the `subject` parameter to
722 `X509Extension` provides its value.
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400723 """
Alex Chanc6077062016-11-18 13:53:39 +0000724 pkey, x509 = x509_data
Jean-Paul Calderone40dd0992010-08-22 17:52:07 -0400725 ext3 = X509Extension(
Alex Gaynor03737182020-07-23 20:40:46 -0400726 b"subjectKeyIdentifier", False, b"hash", subject=x509
727 )
Alex Chanc6077062016-11-18 13:53:39 +0000728 x509.add_extensions([ext3])
Alex Gaynor03737182020-07-23 20:40:46 -0400729 x509.sign(pkey, "sha1")
Alex Chanc6077062016-11-18 13:53:39 +0000730 text = dump_certificate(FILETYPE_TEXT, x509)
Alex Gaynor03737182020-07-23 20:40:46 -0400731 assert b"X509v3 Subject Key Identifier:" in text
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400732
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400733 def test_missing_subject(self):
734 """
Alex Chanc6077062016-11-18 13:53:39 +0000735 If an extension requires a subject and the `subject` parameter
Alex Gaynor31287502015-09-05 16:11:27 -0400736 is given no value, something happens.
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400737 """
Alex Chanc6077062016-11-18 13:53:39 +0000738 with pytest.raises(Error):
Alex Gaynor03737182020-07-23 20:40:46 -0400739 X509Extension(b"subjectKeyIdentifier", False, b"hash")
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400740
Alex Gaynor03737182020-07-23 20:40:46 -0400741 @pytest.mark.parametrize("bad_obj", [True, object(), "hello", []])
Alex Chanc6077062016-11-18 13:53:39 +0000742 def test_invalid_subject(self, bad_obj):
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400743 """
Alex Chanc6077062016-11-18 13:53:39 +0000744 If the `subject` parameter is given a value which is not an
745 `X509` instance, `TypeError` is raised.
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400746 """
Alex Chanc6077062016-11-18 13:53:39 +0000747 with pytest.raises(TypeError):
748 X509Extension(
Alex Gaynor03737182020-07-23 20:40:46 -0400749 "basicConstraints", False, "CA:TRUE", subject=bad_obj
750 )
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400751
Alex Chanc6077062016-11-18 13:53:39 +0000752 def test_unused_issuer(self, x509_data):
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400753 """
Alex Chanc6077062016-11-18 13:53:39 +0000754 The `issuer` parameter to `X509Extension` may be provided for an
755 extension which does not use it and is ignored in this case.
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400756 """
Alex Chanc6077062016-11-18 13:53:39 +0000757 pkey, x509 = x509_data
Jean-Paul Calderone40dd0992010-08-22 17:52:07 -0400758 ext1 = X509Extension(
Alex Gaynor03737182020-07-23 20:40:46 -0400759 b"basicConstraints", False, b"CA:TRUE", issuer=x509
760 )
Alex Chanc6077062016-11-18 13:53:39 +0000761 x509.add_extensions([ext1])
Alex Gaynor03737182020-07-23 20:40:46 -0400762 x509.sign(pkey, "sha1")
Alex Chanc6077062016-11-18 13:53:39 +0000763 text = dump_certificate(FILETYPE_TEXT, x509)
Alex Gaynor03737182020-07-23 20:40:46 -0400764 assert b"X509v3 Basic Constraints:" in text
765 assert b"CA:TRUE" in text
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400766
Alex Chanc6077062016-11-18 13:53:39 +0000767 def test_issuer(self, x509_data):
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400768 """
Alex Chanc6077062016-11-18 13:53:39 +0000769 If an extension requires an issuer, the `issuer` parameter to
770 `X509Extension` provides its value.
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400771 """
Alex Chanc6077062016-11-18 13:53:39 +0000772 pkey, x509 = x509_data
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400773 ext2 = X509Extension(
Alex Gaynor03737182020-07-23 20:40:46 -0400774 b"authorityKeyIdentifier", False, b"issuer:always", issuer=x509
775 )
Alex Chanc6077062016-11-18 13:53:39 +0000776 x509.add_extensions([ext2])
Alex Gaynor03737182020-07-23 20:40:46 -0400777 x509.sign(pkey, "sha1")
Alex Chanc6077062016-11-18 13:53:39 +0000778 text = dump_certificate(FILETYPE_TEXT, x509)
Alex Gaynor03737182020-07-23 20:40:46 -0400779 assert b"X509v3 Authority Key Identifier:" in text
780 assert b"DirName:/CN=Yoda root CA" in text
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400781
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400782 def test_missing_issuer(self):
783 """
Alex Chanc6077062016-11-18 13:53:39 +0000784 If an extension requires an issue and the `issuer` parameter is
785 given no value, something happens.
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400786 """
Alex Chanc6077062016-11-18 13:53:39 +0000787 with pytest.raises(Error):
788 X509Extension(
Alex Gaynor03737182020-07-23 20:40:46 -0400789 b"authorityKeyIdentifier", False, b"keyid:always,issuer:always"
790 )
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400791
Alex Gaynor03737182020-07-23 20:40:46 -0400792 @pytest.mark.parametrize("bad_obj", [True, object(), "hello", []])
Alex Chanc6077062016-11-18 13:53:39 +0000793 def test_invalid_issuer(self, bad_obj):
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400794 """
Alex Chanc6077062016-11-18 13:53:39 +0000795 If the `issuer` parameter is given a value which is not an
796 `X509` instance, `TypeError` is raised.
Jean-Paul Calderonef0179c72009-07-17 15:48:22 -0400797 """
Alex Chanc6077062016-11-18 13:53:39 +0000798 with pytest.raises(TypeError):
799 X509Extension(
Alex Gaynor03737182020-07-23 20:40:46 -0400800 "basicConstraints",
801 False,
802 "keyid:always,issuer:always",
803 issuer=bad_obj,
804 )
Rick Dean47262da2009-07-08 16:17:17 -0500805
806
Paul Kehrer72d968b2016-07-29 15:31:04 +0800807class TestPKey(object):
808 """
Hynek Schlawack3bcf3152017-02-18 08:25:34 +0100809 Tests for `OpenSSL.crypto.PKey`.
Paul Kehrer72d968b2016-07-29 15:31:04 +0800810 """
811
812 def test_convert_from_cryptography_private_key(self):
813 """
814 PKey.from_cryptography_key creates a proper private PKey.
815 """
816 key = serialization.load_pem_private_key(
817 intermediate_key_pem, None, backend
818 )
819 pkey = PKey.from_cryptography_key(key)
820
821 assert isinstance(pkey, PKey)
822 assert pkey.bits() == key.key_size
823 assert pkey._only_public is False
824 assert pkey._initialized is True
825
826 def test_convert_from_cryptography_public_key(self):
827 """
828 PKey.from_cryptography_key creates a proper public PKey.
829 """
830 key = serialization.load_pem_public_key(cleartextPublicKeyPEM, backend)
831 pkey = PKey.from_cryptography_key(key)
832
833 assert isinstance(pkey, PKey)
834 assert pkey.bits() == key.key_size
835 assert pkey._only_public is True
836 assert pkey._initialized is True
837
838 def test_convert_from_cryptography_unsupported_type(self):
839 """
840 PKey.from_cryptography_key raises TypeError with an unsupported type.
841 """
842 key = serialization.load_pem_private_key(
843 ec_private_key_pem, None, backend
844 )
845 with pytest.raises(TypeError):
846 PKey.from_cryptography_key(key)
847
848 def test_convert_public_pkey_to_cryptography_key(self):
849 """
850 PKey.to_cryptography_key creates a proper cryptography public key.
851 """
852 pkey = load_publickey(FILETYPE_PEM, cleartextPublicKeyPEM)
853 key = pkey.to_cryptography_key()
854
855 assert isinstance(key, rsa.RSAPublicKey)
856 assert pkey.bits() == key.key_size
857
858 def test_convert_private_pkey_to_cryptography_key(self):
859 """
860 PKey.to_cryptography_key creates a proper cryptography private key.
861 """
862 pkey = load_privatekey(FILETYPE_PEM, cleartextPrivateKeyPEM)
863 key = pkey.to_cryptography_key()
864
865 assert isinstance(key, rsa.RSAPrivateKey)
866 assert pkey.bits() == key.key_size
867
Jean-Paul Calderone68649052009-07-17 21:14:27 -0400868 def test_type(self):
869 """
Alex Gaynor01f90a12019-02-07 09:14:48 -0500870 `PKey` can be used to create instances of that type.
Jean-Paul Calderone68649052009-07-17 21:14:27 -0400871 """
Alex Gaynor03737182020-07-23 20:40:46 -0400872 assert is_consistent_type(PKey, "PKey")
Jean-Paul Calderone68649052009-07-17 21:14:27 -0400873
Jean-Paul Calderoned8782ad2008-03-04 23:39:59 -0500874 def test_construction(self):
875 """
Alex Chan9e2a9932017-01-25 14:29:19 +0000876 `PKey` takes no arguments and returns a new `PKey` instance.
Jean-Paul Calderoned8782ad2008-03-04 23:39:59 -0500877 """
Jean-Paul Calderoned8782ad2008-03-04 23:39:59 -0500878 key = PKey()
Alex Chan9e2a9932017-01-25 14:29:19 +0000879 assert isinstance(key, PKey)
Jean-Paul Calderoned8782ad2008-03-04 23:39:59 -0500880
Jean-Paul Calderoned8782ad2008-03-04 23:39:59 -0500881 def test_pregeneration(self):
882 """
Alex Chan9e2a9932017-01-25 14:29:19 +0000883 `PKey.bits` and `PKey.type` return `0` before the key is generated.
884 `PKey.check` raises `TypeError` before the key is generated.
Jean-Paul Calderoned8782ad2008-03-04 23:39:59 -0500885 """
886 key = PKey()
Alex Chan9e2a9932017-01-25 14:29:19 +0000887 assert key.type() == 0
888 assert key.bits() == 0
889 with pytest.raises(TypeError):
890 key.check()
Jean-Paul Calderoned8782ad2008-03-04 23:39:59 -0500891
Alex Chan9e2a9932017-01-25 14:29:19 +0000892 def test_failed_generation(self):
Jean-Paul Calderoned8782ad2008-03-04 23:39:59 -0500893 """
Alex Chan9e2a9932017-01-25 14:29:19 +0000894 `PKey.generate_key` takes two arguments, the first giving the key type
895 as one of `TYPE_RSA` or `TYPE_DSA` and the second giving the number of
896 bits to generate. If an invalid type is specified or generation fails,
897 `Error` is raised. If an invalid number of bits is specified,
898 `ValueError` or `Error` is raised.
Jean-Paul Calderoned8782ad2008-03-04 23:39:59 -0500899 """
900 key = PKey()
Alex Chan9e2a9932017-01-25 14:29:19 +0000901 with pytest.raises(TypeError):
902 key.generate_key("foo", "bar")
903 with pytest.raises(Error):
904 key.generate_key(-1, 0)
Jean-Paul Calderoneab82db72008-03-06 00:09:31 -0500905
Alex Chan9e2a9932017-01-25 14:29:19 +0000906 with pytest.raises(ValueError):
907 key.generate_key(TYPE_RSA, -1)
908 with pytest.raises(ValueError):
909 key.generate_key(TYPE_RSA, 0)
Jean-Paul Calderoned71fe982008-03-06 00:31:50 -0500910
Alex Gaynor5bb2bd12016-07-03 10:48:32 -0400911 with pytest.raises(TypeError):
912 key.generate_key(TYPE_RSA, object())
913
Jean-Paul Calderoned71fe982008-03-06 00:31:50 -0500914 # XXX RSA generation for small values of bits is fairly buggy in a wide
915 # range of OpenSSL versions. I need to figure out what the safe lower
916 # bound for a reasonable number of OpenSSL versions is and explicitly
917 # check for that in the wrapper. The failure behavior is typically an
918 # infinite loop inside OpenSSL.
919
Alex Chan9e2a9932017-01-25 14:29:19 +0000920 # with pytest.raises(Error):
921 # key.generate_key(TYPE_RSA, 2)
Jean-Paul Calderoned8782ad2008-03-04 23:39:59 -0500922
923 # XXX DSA generation seems happy with any number of bits. The DSS
924 # says bits must be between 512 and 1024 inclusive. OpenSSL's DSA
925 # generator doesn't seem to care about the upper limit at all. For
Jean-Paul Calderoneeff3cd92008-03-05 22:35:26 -0500926 # the lower limit, it uses 512 if anything smaller is specified.
Jean-Paul Calderoned8782ad2008-03-04 23:39:59 -0500927 # So, it doesn't seem possible to make generate_key fail for
928 # TYPE_DSA with a bits argument which is at least an int.
929
Alex Chan9e2a9932017-01-25 14:29:19 +0000930 # with pytest.raises(Error):
931 # key.generate_key(TYPE_DSA, -7)
Jean-Paul Calderoned8782ad2008-03-04 23:39:59 -0500932
Alex Chan9e2a9932017-01-25 14:29:19 +0000933 def test_rsa_generation(self):
Jean-Paul Calderoned8782ad2008-03-04 23:39:59 -0500934 """
Alex Chan9e2a9932017-01-25 14:29:19 +0000935 `PKey.generate_key` generates an RSA key when passed `TYPE_RSA` as a
936 type and a reasonable number of bits.
Jean-Paul Calderoned8782ad2008-03-04 23:39:59 -0500937 """
Alex Gaynor6e9f9762018-05-12 07:44:37 -0400938 bits = 512
Jean-Paul Calderoned8782ad2008-03-04 23:39:59 -0500939 key = PKey()
940 key.generate_key(TYPE_RSA, bits)
Alex Chan9e2a9932017-01-25 14:29:19 +0000941 assert key.type() == TYPE_RSA
942 assert key.bits() == bits
943 assert key.check()
Jean-Paul Calderoned8782ad2008-03-04 23:39:59 -0500944
Alex Chan9e2a9932017-01-25 14:29:19 +0000945 def test_dsa_generation(self):
Jean-Paul Calderoned8782ad2008-03-04 23:39:59 -0500946 """
Alex Chan9e2a9932017-01-25 14:29:19 +0000947 `PKey.generate_key` generates a DSA key when passed `TYPE_DSA` as a
948 type and a reasonable number of bits.
Jean-Paul Calderoned8782ad2008-03-04 23:39:59 -0500949 """
950 # 512 is a magic number. The DSS (Digital Signature Standard)
951 # allows a minimum of 512 bits for DSA. DSA_generate_parameters
952 # will silently promote any value below 512 to 512.
953 bits = 512
954 key = PKey()
955 key.generate_key(TYPE_DSA, bits)
Alex Chan9e2a9932017-01-25 14:29:19 +0000956 assert key.type() == TYPE_DSA
957 assert key.bits() == bits
958 with pytest.raises(TypeError):
959 key.check()
Jean-Paul Calderoned8782ad2008-03-04 23:39:59 -0500960
Jean-Paul Calderoned8782ad2008-03-04 23:39:59 -0500961 def test_regeneration(self):
962 """
Alex Chan9e2a9932017-01-25 14:29:19 +0000963 `PKey.generate_key` can be called multiple times on the same key to
964 generate new keys.
Jean-Paul Calderoned8782ad2008-03-04 23:39:59 -0500965 """
966 key = PKey()
967 for type, bits in [(TYPE_RSA, 512), (TYPE_DSA, 576)]:
Alex Gaynor7f636492015-09-04 13:26:52 -0400968 key.generate_key(type, bits)
Alex Chan9e2a9932017-01-25 14:29:19 +0000969 assert key.type() == type
970 assert key.bits() == bits
Jean-Paul Calderoneeff3cd92008-03-05 22:35:26 -0500971
Alex Chan9e2a9932017-01-25 14:29:19 +0000972 def test_inconsistent_key(self):
Jean-Paul Calderone55ec1712009-05-13 14:14:30 -0400973 """
Alex Chan9e2a9932017-01-25 14:29:19 +0000974 `PKey.check` returns `Error` if the key is not consistent.
Jean-Paul Calderone55ec1712009-05-13 14:14:30 -0400975 """
976 key = load_privatekey(FILETYPE_PEM, inconsistentPrivateKeyPEM)
Alex Chan9e2a9932017-01-25 14:29:19 +0000977 with pytest.raises(Error):
978 key.check()
Jean-Paul Calderonee81020e2011-06-12 21:48:57 -0400979
Jean-Paul Calderone02d01972011-10-31 10:39:29 -0400980 def test_check_public_key(self):
981 """
Alex Chan9e2a9932017-01-25 14:29:19 +0000982 `PKey.check` raises `TypeError` if only the public part of the key
983 is available.
Jean-Paul Calderone02d01972011-10-31 10:39:29 -0400984 """
985 # A trick to get a public-only key
986 key = PKey()
987 key.generate_key(TYPE_RSA, 512)
988 cert = X509()
989 cert.set_pubkey(key)
990 pub = cert.get_pubkey()
Alex Chan9e2a9932017-01-25 14:29:19 +0000991 with pytest.raises(TypeError):
992 pub.check()
Jean-Paul Calderone02d01972011-10-31 10:39:29 -0400993
Mrmaxmeier8cd3b172020-03-11 22:03:59 +0100994 def test_check_pr_897(self):
995 """
996 `PKey.check` raises `OpenSSL.crypto.Error` if provided with broken key
997 """
998 pkey = load_privatekey(FILETYPE_PEM, rsa_p_not_prime_pem)
999 with pytest.raises(Error):
1000 pkey.check()
1001
Jean-Paul Calderone02d01972011-10-31 10:39:29 -04001002
Alex Chan9e2a9932017-01-25 14:29:19 +00001003def x509_name(**attrs):
Jean-Paul Calderoneeff3cd92008-03-05 22:35:26 -05001004 """
Alex Chan9e2a9932017-01-25 14:29:19 +00001005 Return a new X509Name with the given attributes.
Jean-Paul Calderoneeff3cd92008-03-05 22:35:26 -05001006 """
Alex Chan9e2a9932017-01-25 14:29:19 +00001007 # XXX There's no other way to get a new X509Name yet.
1008 name = X509().get_subject()
1009 attrs = list(attrs.items())
Alex Gaynoraceb3e22015-09-05 12:00:22 -04001010
Alex Chan9e2a9932017-01-25 14:29:19 +00001011 # Make the order stable - order matters!
1012 def key(attr):
1013 return attr[1]
Alex Gaynor03737182020-07-23 20:40:46 -04001014
Alex Chan9e2a9932017-01-25 14:29:19 +00001015 attrs.sort(key=key)
1016 for k, v in attrs:
1017 setattr(name, k, v)
1018 return name
Alex Gaynor85b49702015-09-05 16:30:59 -04001019
Alex Chan9e2a9932017-01-25 14:29:19 +00001020
1021class TestX509Name(object):
1022 """
1023 Unit tests for `OpenSSL.crypto.X509Name`.
1024 """
Jean-Paul Calderonee098dc72008-03-06 18:36:19 -05001025
Rick Deane15b1472009-07-09 15:53:42 -05001026 def test_type(self):
1027 """
Alex Gaynor01f90a12019-02-07 09:14:48 -05001028 The type of X509Name objects is `X509Name`.
Rick Deane15b1472009-07-09 15:53:42 -05001029 """
Alex Chan9e2a9932017-01-25 14:29:19 +00001030 name = x509_name()
Alex Gaynor01f90a12019-02-07 09:14:48 -05001031 assert isinstance(name, X509Name)
Rick Deane15b1472009-07-09 15:53:42 -05001032
Alex Chan9e2a9932017-01-25 14:29:19 +00001033 def test_only_string_attributes(self):
Jean-Paul Calderone9ce9afb2011-04-22 18:16:22 -04001034 """
Alex Chan9e2a9932017-01-25 14:29:19 +00001035 Attempting to set a non-`str` attribute name on an `X509Name` instance
1036 causes `TypeError` to be raised.
Jean-Paul Calderone9ce9afb2011-04-22 18:16:22 -04001037 """
Alex Chan9e2a9932017-01-25 14:29:19 +00001038 name = x509_name()
Jean-Paul Calderone9ce9afb2011-04-22 18:16:22 -04001039 # Beyond these cases, you may also think that unicode should be
Alex Gaynor31287502015-09-05 16:11:27 -04001040 # rejected. Sorry, you're wrong. unicode is automatically converted
1041 # to str outside of the control of X509Name, so there's no way to
1042 # reject it.
Jean-Paul Calderoneff363be2013-03-03 10:21:23 -08001043
Alex Gaynor31287502015-09-05 16:11:27 -04001044 # Also, this used to test str subclasses, but that test is less
1045 # relevant now that the implementation is in Python instead of C. Also
1046 # PyPy automatically converts str subclasses to str when they are
1047 # passed to setattr, so we can't test it on PyPy. Apparently CPython
1048 # does this sometimes as well.
Alex Chan9e2a9932017-01-25 14:29:19 +00001049 with pytest.raises(TypeError):
1050 setattr(name, None, "hello")
1051 with pytest.raises(TypeError):
1052 setattr(name, 30, "hello")
Jean-Paul Calderone9ce9afb2011-04-22 18:16:22 -04001053
Alex Chan9e2a9932017-01-25 14:29:19 +00001054 def test_set_invalid_attribute(self):
Jean-Paul Calderone9ce9afb2011-04-22 18:16:22 -04001055 """
Alex Chan9e2a9932017-01-25 14:29:19 +00001056 Attempting to set any attribute name on an `X509Name` instance for
1057 which no corresponding NID is defined causes `AttributeError` to be
1058 raised.
Jean-Paul Calderone9ce9afb2011-04-22 18:16:22 -04001059 """
Alex Chan9e2a9932017-01-25 14:29:19 +00001060 name = x509_name()
1061 with pytest.raises(AttributeError):
1062 setattr(name, "no such thing", None)
Jean-Paul Calderone9ce9afb2011-04-22 18:16:22 -04001063
Jean-Paul Calderoneeff3cd92008-03-05 22:35:26 -05001064 def test_attributes(self):
1065 """
Alex Chan9e2a9932017-01-25 14:29:19 +00001066 `X509Name` instances have attributes for each standard (?)
1067 X509Name field.
Jean-Paul Calderoneeff3cd92008-03-05 22:35:26 -05001068 """
Alex Chan9e2a9932017-01-25 14:29:19 +00001069 name = x509_name()
Jean-Paul Calderoneeff3cd92008-03-05 22:35:26 -05001070 name.commonName = "foo"
Alex Gaynor37726112016-07-04 09:51:32 -04001071 assert name.commonName == "foo"
1072 assert name.CN == "foo"
1073
Jean-Paul Calderoneeff3cd92008-03-05 22:35:26 -05001074 name.CN = "baz"
Alex Gaynor37726112016-07-04 09:51:32 -04001075 assert name.commonName == "baz"
1076 assert name.CN == "baz"
1077
Jean-Paul Calderoneeff3cd92008-03-05 22:35:26 -05001078 name.commonName = "bar"
Alex Gaynor37726112016-07-04 09:51:32 -04001079 assert name.commonName == "bar"
1080 assert name.CN == "bar"
1081
Jean-Paul Calderoneeff3cd92008-03-05 22:35:26 -05001082 name.CN = "quux"
Alex Gaynor37726112016-07-04 09:51:32 -04001083 assert name.commonName == "quux"
1084 assert name.CN == "quux"
1085
1086 assert name.OU is None
Jean-Paul Calderoneeff3cd92008-03-05 22:35:26 -05001087
Alex Gaynor7778e792016-07-03 23:38:48 -04001088 with pytest.raises(AttributeError):
1089 name.foobar
1090
Jean-Paul Calderoneeff3cd92008-03-05 22:35:26 -05001091 def test_copy(self):
1092 """
Alex Chan9e2a9932017-01-25 14:29:19 +00001093 `X509Name` creates a new `X509Name` instance with all the same
1094 attributes as an existing `X509Name` instance when called with one.
Jean-Paul Calderoneeff3cd92008-03-05 22:35:26 -05001095 """
Alex Chan9e2a9932017-01-25 14:29:19 +00001096 name = x509_name(commonName="foo", emailAddress="bar@example.com")
Jean-Paul Calderoneeff3cd92008-03-05 22:35:26 -05001097
1098 copy = X509Name(name)
Alex Chan9e2a9932017-01-25 14:29:19 +00001099 assert copy.commonName == "foo"
1100 assert copy.emailAddress == "bar@example.com"
Jean-Paul Calderonee098dc72008-03-06 18:36:19 -05001101
1102 # Mutate the copy and ensure the original is unmodified.
Jean-Paul Calderoneeff3cd92008-03-05 22:35:26 -05001103 copy.commonName = "baz"
Alex Chan9e2a9932017-01-25 14:29:19 +00001104 assert name.commonName == "foo"
Jean-Paul Calderonee098dc72008-03-06 18:36:19 -05001105
1106 # Mutate the original and ensure the copy is unmodified.
Jean-Paul Calderoneeff3cd92008-03-05 22:35:26 -05001107 name.emailAddress = "quux@example.com"
Alex Chan9e2a9932017-01-25 14:29:19 +00001108 assert copy.emailAddress == "bar@example.com"
Jean-Paul Calderoneeff3cd92008-03-05 22:35:26 -05001109
Jean-Paul Calderonee098dc72008-03-06 18:36:19 -05001110 def test_repr(self):
1111 """
Alex Chan9e2a9932017-01-25 14:29:19 +00001112 `repr` passed an `X509Name` instance should return a string containing
1113 a description of the type and the NIDs which have been set on it.
Jean-Paul Calderonee098dc72008-03-06 18:36:19 -05001114 """
Alex Chan9e2a9932017-01-25 14:29:19 +00001115 name = x509_name(commonName="foo", emailAddress="bar")
1116 assert repr(name) == "<X509Name object '/emailAddress=bar/CN=foo'>"
Jean-Paul Calderonee098dc72008-03-06 18:36:19 -05001117
Jean-Paul Calderonee098dc72008-03-06 18:36:19 -05001118 def test_comparison(self):
1119 """
Alex Chan9e2a9932017-01-25 14:29:19 +00001120 `X509Name` instances should compare based on their NIDs.
Jean-Paul Calderonee098dc72008-03-06 18:36:19 -05001121 """
Alex Gaynor03737182020-07-23 20:40:46 -04001122
Alex Chan9e2a9932017-01-25 14:29:19 +00001123 def _equality(a, b, assert_true, assert_false):
1124 assert_true(a == b)
1125 assert_false(a != b)
1126 assert_true(b == a)
1127 assert_false(b != a)
Jean-Paul Calderonee098dc72008-03-06 18:36:19 -05001128
Alex Chan9e2a9932017-01-25 14:29:19 +00001129 def assert_true(x):
1130 assert x
1131
1132 def assert_false(x):
1133 assert not x
1134
1135 def assert_equal(a, b):
1136 _equality(a, b, assert_true, assert_false)
Jean-Paul Calderonee098dc72008-03-06 18:36:19 -05001137
1138 # Instances compare equal to themselves.
Alex Chan9e2a9932017-01-25 14:29:19 +00001139 name = x509_name()
1140 assert_equal(name, name)
Jean-Paul Calderonee098dc72008-03-06 18:36:19 -05001141
1142 # Empty instances should compare equal to each other.
Alex Chan9e2a9932017-01-25 14:29:19 +00001143 assert_equal(x509_name(), x509_name())
Jean-Paul Calderonee098dc72008-03-06 18:36:19 -05001144
1145 # Instances with equal NIDs should compare equal to each other.
Alex Gaynor03737182020-07-23 20:40:46 -04001146 assert_equal(x509_name(commonName="foo"), x509_name(commonName="foo"))
Jean-Paul Calderonee098dc72008-03-06 18:36:19 -05001147
1148 # Instance with equal NIDs set using different aliases should compare
1149 # equal to each other.
Alex Gaynor03737182020-07-23 20:40:46 -04001150 assert_equal(x509_name(commonName="foo"), x509_name(CN="foo"))
Jean-Paul Calderonee098dc72008-03-06 18:36:19 -05001151
1152 # Instances with more than one NID with the same values should compare
1153 # equal to each other.
Alex Gaynor03737182020-07-23 20:40:46 -04001154 assert_equal(
1155 x509_name(CN="foo", organizationalUnitName="bar"),
1156 x509_name(commonName="foo", OU="bar"),
1157 )
Jean-Paul Calderonee098dc72008-03-06 18:36:19 -05001158
Alex Chan9e2a9932017-01-25 14:29:19 +00001159 def assert_not_equal(a, b):
1160 _equality(a, b, assert_false, assert_true)
Jean-Paul Calderonee098dc72008-03-06 18:36:19 -05001161
1162 # Instances with different values for the same NID should not compare
1163 # equal to each other.
Alex Gaynor03737182020-07-23 20:40:46 -04001164 assert_not_equal(x509_name(CN="foo"), x509_name(CN="bar"))
Jean-Paul Calderonee098dc72008-03-06 18:36:19 -05001165
1166 # Instances with different NIDs should not compare equal to each other.
Alex Gaynor03737182020-07-23 20:40:46 -04001167 assert_not_equal(x509_name(CN="foo"), x509_name(OU="foo"))
Jean-Paul Calderonee098dc72008-03-06 18:36:19 -05001168
Alex Chan9e2a9932017-01-25 14:29:19 +00001169 assert_not_equal(x509_name(), object())
Alex Gaynor7778e792016-07-03 23:38:48 -04001170
Alex Chan9e2a9932017-01-25 14:29:19 +00001171 def _inequality(a, b, assert_true, assert_false):
1172 assert_true(a < b)
1173 assert_true(a <= b)
1174 assert_true(b > a)
1175 assert_true(b >= a)
1176 assert_false(a > b)
1177 assert_false(a >= b)
1178 assert_false(b < a)
1179 assert_false(b <= a)
Jean-Paul Calderonee098dc72008-03-06 18:36:19 -05001180
Alex Chan9e2a9932017-01-25 14:29:19 +00001181 def assert_less_than(a, b):
1182 _inequality(a, b, assert_true, assert_false)
Jean-Paul Calderonee098dc72008-03-06 18:36:19 -05001183
1184 # An X509Name with a NID with a value which sorts less than the value
1185 # of the same NID on another X509Name compares less than the other
1186 # X509Name.
Alex Gaynor03737182020-07-23 20:40:46 -04001187 assert_less_than(x509_name(CN="abc"), x509_name(CN="def"))
Jean-Paul Calderonee098dc72008-03-06 18:36:19 -05001188
Alex Chan9e2a9932017-01-25 14:29:19 +00001189 def assert_greater_than(a, b):
1190 _inequality(a, b, assert_false, assert_true)
Jean-Paul Calderonee098dc72008-03-06 18:36:19 -05001191
1192 # An X509Name with a NID with a value which sorts greater than the
1193 # value of the same NID on another X509Name compares greater than the
1194 # other X509Name.
Alex Gaynor03737182020-07-23 20:40:46 -04001195 assert_greater_than(x509_name(CN="def"), x509_name(CN="abc"))
Jean-Paul Calderone2aa2b332008-03-06 21:43:14 -05001196
Jean-Paul Calderone110cd092008-03-24 17:27:42 -04001197 def test_hash(self):
1198 """
Alex Chan9e2a9932017-01-25 14:29:19 +00001199 `X509Name.hash` returns an integer hash based on the value of the name.
Jean-Paul Calderone110cd092008-03-24 17:27:42 -04001200 """
Alex Chan9e2a9932017-01-25 14:29:19 +00001201 a = x509_name(CN="foo")
1202 b = x509_name(CN="foo")
1203 assert a.hash() == b.hash()
Jean-Paul Calderone110cd092008-03-24 17:27:42 -04001204 a.CN = "bar"
Alex Chan9e2a9932017-01-25 14:29:19 +00001205 assert a.hash() != b.hash()
Jean-Paul Calderone110cd092008-03-24 17:27:42 -04001206
Jean-Paul Calderonee957a002008-03-25 15:16:51 -04001207 def test_der(self):
1208 """
Alex Chan9e2a9932017-01-25 14:29:19 +00001209 `X509Name.der` returns the DER encoded form of the name.
Jean-Paul Calderonee957a002008-03-25 15:16:51 -04001210 """
Alex Chan9e2a9932017-01-25 14:29:19 +00001211 a = x509_name(CN="foo", C="US")
Alex Gaynor03737182020-07-23 20:40:46 -04001212 assert (
1213 a.der() == b"0\x1b1\x0b0\t\x06\x03U\x04\x06\x13\x02US"
1214 b"1\x0c0\n\x06\x03U\x04\x03\x0c\x03foo"
1215 )
Jean-Paul Calderonee957a002008-03-25 15:16:51 -04001216
Jean-Paul Calderonec54cc182008-03-26 21:11:07 -04001217 def test_get_components(self):
1218 """
Alex Chan9e2a9932017-01-25 14:29:19 +00001219 `X509Name.get_components` returns a `list` of two-tuples of `str`
Jean-Paul Calderonec54cc182008-03-26 21:11:07 -04001220 giving the NIDs and associated values which make up the name.
1221 """
Alex Chan9e2a9932017-01-25 14:29:19 +00001222 a = x509_name()
1223 assert a.get_components() == []
Jean-Paul Calderonec54cc182008-03-26 21:11:07 -04001224 a.CN = "foo"
Alex Chan9e2a9932017-01-25 14:29:19 +00001225 assert a.get_components() == [(b"CN", b"foo")]
Jean-Paul Calderonec54cc182008-03-26 21:11:07 -04001226 a.organizationalUnitName = "bar"
Alex Chan9e2a9932017-01-25 14:29:19 +00001227 assert a.get_components() == [(b"CN", b"foo"), (b"OU", b"bar")]
Jean-Paul Calderonec54cc182008-03-26 21:11:07 -04001228
Jean-Paul Calderone4bf75c62013-08-23 15:39:53 -04001229 def test_load_nul_byte_attribute(self):
1230 """
Alex Chan9e2a9932017-01-25 14:29:19 +00001231 An `X509Name` from an `X509` instance loaded from a file can have a
Jean-Paul Calderone4bf75c62013-08-23 15:39:53 -04001232 NUL byte in the value of one of its attributes.
1233 """
1234 cert = load_certificate(FILETYPE_PEM, nulbyteSubjectAltNamePEM)
1235 subject = cert.get_subject()
Alex Chan9e2a9932017-01-25 14:29:19 +00001236 assert "null.python.org\x00example.org" == subject.commonName
Jean-Paul Calderone4bf75c62013-08-23 15:39:53 -04001237
Romuald Brunet4183beb2019-01-21 19:38:33 +01001238 def test_load_nul_byte_components(self):
1239 """
1240 An `X509Name` from an `X509` instance loaded from a file can have a
1241 NUL byte in the value of its components
1242 """
1243 cert = load_certificate(FILETYPE_PEM, nulbyteSubjectAltNamePEM)
1244 subject = cert.get_subject()
1245 components = subject.get_components()
Alex Gaynor03737182020-07-23 20:40:46 -04001246 ccn = [value for name, value in components if name == b"CN"]
1247 assert ccn[0] == b"null.python.org\x00example.org"
Romuald Brunet4183beb2019-01-21 19:38:33 +01001248
Alex Chan9e2a9932017-01-25 14:29:19 +00001249 def test_set_attribute_failure(self):
Jean-Paul Calderone5300d6a2013-12-29 16:36:50 -05001250 """
1251 If the value of an attribute cannot be set for some reason then
Alex Chan9e2a9932017-01-25 14:29:19 +00001252 `Error` is raised.
Jean-Paul Calderone5300d6a2013-12-29 16:36:50 -05001253 """
Alex Chan9e2a9932017-01-25 14:29:19 +00001254 name = x509_name()
Jean-Paul Calderone5300d6a2013-12-29 16:36:50 -05001255 # This value is too long
Alex Chan9e2a9932017-01-25 14:29:19 +00001256 with pytest.raises(Error):
1257 setattr(name, "O", b"x" * 512)
Jean-Paul Calderone5300d6a2013-12-29 16:36:50 -05001258
1259
Jean-Paul Calderoneac0d95f2008-03-10 00:00:42 -04001260class _PKeyInteractionTestsMixin:
1261 """
1262 Tests which involve another thing and a PKey.
1263 """
Alex Gaynoraceb3e22015-09-05 12:00:22 -04001264
Jean-Paul Calderoneac0d95f2008-03-10 00:00:42 -04001265 def signable(self):
1266 """
Alex Chanfb078d82017-04-20 11:16:15 +01001267 Return something with a `set_pubkey`, `set_pubkey`, and `sign` method.
Jean-Paul Calderoneac0d95f2008-03-10 00:00:42 -04001268 """
1269 raise NotImplementedError()
1270
Alex Chanb00ede22017-01-30 07:24:40 +00001271 def test_sign_with_ungenerated(self):
Jean-Paul Calderoneac0d95f2008-03-10 00:00:42 -04001272 """
Alex Chanb00ede22017-01-30 07:24:40 +00001273 `X509Req.sign` raises `ValueError` when passed a `PKey` with no parts.
Jean-Paul Calderoneac0d95f2008-03-10 00:00:42 -04001274 """
1275 request = self.signable()
1276 key = PKey()
Alex Chanb00ede22017-01-30 07:24:40 +00001277 with pytest.raises(ValueError):
1278 request.sign(key, GOOD_DIGEST)
Jean-Paul Calderoneac0d95f2008-03-10 00:00:42 -04001279
Alex Chanb00ede22017-01-30 07:24:40 +00001280 def test_sign_with_public_key(self):
Jean-Paul Calderoneac0d95f2008-03-10 00:00:42 -04001281 """
Alex Chanb00ede22017-01-30 07:24:40 +00001282 `X509Req.sign` raises `ValueError` when passed a `PKey` with no private
1283 part as the signing key.
Jean-Paul Calderoneac0d95f2008-03-10 00:00:42 -04001284 """
1285 request = self.signable()
1286 key = PKey()
1287 key.generate_key(TYPE_RSA, 512)
1288 request.set_pubkey(key)
1289 pub = request.get_pubkey()
Alex Chanb00ede22017-01-30 07:24:40 +00001290 with pytest.raises(ValueError):
1291 request.sign(pub, GOOD_DIGEST)
Jean-Paul Calderoneac0d95f2008-03-10 00:00:42 -04001292
Alex Chanb00ede22017-01-30 07:24:40 +00001293 def test_sign_with_unknown_digest(self):
Jean-Paul Calderonecc05a912010-08-03 18:24:19 -04001294 """
Alex Chanb00ede22017-01-30 07:24:40 +00001295 `X509Req.sign` raises `ValueError` when passed a digest name which is
1296 not known.
Jean-Paul Calderonecc05a912010-08-03 18:24:19 -04001297 """
1298 request = self.signable()
1299 key = PKey()
1300 key.generate_key(TYPE_RSA, 512)
Alex Chanb00ede22017-01-30 07:24:40 +00001301 with pytest.raises(ValueError):
1302 request.sign(key, BAD_DIGEST)
Jean-Paul Calderonecc05a912010-08-03 18:24:19 -04001303
Jean-Paul Calderoneb9725592010-08-03 18:17:22 -04001304 def test_sign(self):
1305 """
Alex Chanb00ede22017-01-30 07:24:40 +00001306 `X509Req.sign` succeeds when passed a private key object and a
1307 valid digest function. `X509Req.verify` can be used to check
Alex Gaynor31287502015-09-05 16:11:27 -04001308 the signature.
Jean-Paul Calderoneb9725592010-08-03 18:17:22 -04001309 """
1310 request = self.signable()
1311 key = PKey()
1312 key.generate_key(TYPE_RSA, 512)
1313 request.set_pubkey(key)
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05001314 request.sign(key, GOOD_DIGEST)
Jean-Paul Calderoneb9725592010-08-03 18:17:22 -04001315 # If the type has a verify method, cover that too.
Alex Gaynor03737182020-07-23 20:40:46 -04001316 if getattr(request, "verify", None) is not None:
Jean-Paul Calderoneb9725592010-08-03 18:17:22 -04001317 pub = request.get_pubkey()
Alex Chanb00ede22017-01-30 07:24:40 +00001318 assert request.verify(pub)
Jean-Paul Calderoneb9725592010-08-03 18:17:22 -04001319 # Make another key that won't verify.
1320 key = PKey()
1321 key.generate_key(TYPE_RSA, 512)
Alex Chanb00ede22017-01-30 07:24:40 +00001322 with pytest.raises(Error):
1323 request.verify(key)
Jean-Paul Calderoneb9725592010-08-03 18:17:22 -04001324
1325
Alex Chanb00ede22017-01-30 07:24:40 +00001326class TestX509Req(_PKeyInteractionTestsMixin):
Jean-Paul Calderone2aa2b332008-03-06 21:43:14 -05001327 """
Alex Chanb00ede22017-01-30 07:24:40 +00001328 Tests for `OpenSSL.crypto.X509Req`.
Jean-Paul Calderone2aa2b332008-03-06 21:43:14 -05001329 """
Alex Gaynoraceb3e22015-09-05 12:00:22 -04001330
Jean-Paul Calderoneac0d95f2008-03-10 00:00:42 -04001331 def signable(self):
1332 """
Alex Chanb00ede22017-01-30 07:24:40 +00001333 Create and return a new `X509Req`.
Jean-Paul Calderoneac0d95f2008-03-10 00:00:42 -04001334 """
1335 return X509Req()
1336
Jean-Paul Calderone68649052009-07-17 21:14:27 -04001337 def test_type(self):
1338 """
Alex Gaynor01f90a12019-02-07 09:14:48 -05001339 `X509Req` can be used to create instances of that type.
Jean-Paul Calderone68649052009-07-17 21:14:27 -04001340 """
Alex Gaynor03737182020-07-23 20:40:46 -04001341 assert is_consistent_type(X509Req, "X509Req")
Jean-Paul Calderoneac0d95f2008-03-10 00:00:42 -04001342
Jean-Paul Calderone2aa2b332008-03-06 21:43:14 -05001343 def test_construction(self):
1344 """
Alex Gaynor01f90a12019-02-07 09:14:48 -05001345 `X509Req` takes no arguments and returns an `X509Req` instance.
Jean-Paul Calderone2aa2b332008-03-06 21:43:14 -05001346 """
1347 request = X509Req()
Alex Gaynor01f90a12019-02-07 09:14:48 -05001348 assert isinstance(request, X509Req)
Jean-Paul Calderone2aa2b332008-03-06 21:43:14 -05001349
Jean-Paul Calderone8dd19b82008-12-28 20:41:16 -05001350 def test_version(self):
1351 """
Alex Chanb00ede22017-01-30 07:24:40 +00001352 `X509Req.set_version` sets the X.509 version of the certificate
1353 request. `X509Req.get_version` returns the X.509 version of the
1354 certificate request. The initial value of the version is 0.
Jean-Paul Calderone8dd19b82008-12-28 20:41:16 -05001355 """
1356 request = X509Req()
Alex Chanb00ede22017-01-30 07:24:40 +00001357 assert request.get_version() == 0
Jean-Paul Calderone8dd19b82008-12-28 20:41:16 -05001358 request.set_version(1)
Alex Chanb00ede22017-01-30 07:24:40 +00001359 assert request.get_version() == 1
Jean-Paul Calderone8dd19b82008-12-28 20:41:16 -05001360 request.set_version(3)
Alex Chanb00ede22017-01-30 07:24:40 +00001361 assert request.get_version() == 3
Jean-Paul Calderone8dd19b82008-12-28 20:41:16 -05001362
Jean-Paul Calderone54e49e92010-07-30 11:04:46 -04001363 def test_version_wrong_args(self):
Jean-Paul Calderoneaa6c0692010-09-08 22:43:09 -04001364 """
Alex Chanb00ede22017-01-30 07:24:40 +00001365 `X509Req.set_version` raises `TypeError` if called with a non-`int`
1366 argument.
Jean-Paul Calderoneaa6c0692010-09-08 22:43:09 -04001367 """
Jean-Paul Calderone54e49e92010-07-30 11:04:46 -04001368 request = X509Req()
Alex Chanb00ede22017-01-30 07:24:40 +00001369 with pytest.raises(TypeError):
1370 request.set_version("foo")
Jean-Paul Calderone54e49e92010-07-30 11:04:46 -04001371
Jean-Paul Calderone2aa2b332008-03-06 21:43:14 -05001372 def test_get_subject(self):
1373 """
Alex Chanb00ede22017-01-30 07:24:40 +00001374 `X509Req.get_subject` returns an `X509Name` for the subject of the
1375 request and which is valid even after the request object is
1376 otherwise dead.
Jean-Paul Calderone2aa2b332008-03-06 21:43:14 -05001377 """
1378 request = X509Req()
1379 subject = request.get_subject()
Alex Gaynor01f90a12019-02-07 09:14:48 -05001380 assert isinstance(subject, X509Name)
Jean-Paul Calderone2aa2b332008-03-06 21:43:14 -05001381 subject.commonName = "foo"
Alex Chanb00ede22017-01-30 07:24:40 +00001382 assert request.get_subject().commonName == "foo"
Jean-Paul Calderone2aa2b332008-03-06 21:43:14 -05001383 del request
1384 subject.commonName = "bar"
Alex Chanb00ede22017-01-30 07:24:40 +00001385 assert subject.commonName == "bar"
Jean-Paul Calderone54e49e92010-07-30 11:04:46 -04001386
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -04001387 def test_add_extensions(self):
1388 """
Alex Chanb00ede22017-01-30 07:24:40 +00001389 `X509Req.add_extensions` accepts a `list` of `X509Extension` instances
1390 and adds them to the X509 request.
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -04001391 """
1392 request = X509Req()
Alex Gaynor03737182020-07-23 20:40:46 -04001393 request.add_extensions(
1394 [X509Extension(b"basicConstraints", True, b"CA:false")]
1395 )
Stephen Holsappleca545b72014-01-28 21:43:25 -08001396 exts = request.get_extensions()
Alex Chanb00ede22017-01-30 07:24:40 +00001397 assert len(exts) == 1
Alex Gaynor03737182020-07-23 20:40:46 -04001398 assert exts[0].get_short_name() == b"basicConstraints"
Alex Chanb00ede22017-01-30 07:24:40 +00001399 assert exts[0].get_critical() == 1
Alex Gaynor03737182020-07-23 20:40:46 -04001400 assert exts[0].get_data() == b"0\x00"
Stephen Holsapple7fbdf642014-03-01 20:05:47 -08001401
Stephen Holsapple7fbdf642014-03-01 20:05:47 -08001402 def test_get_extensions(self):
1403 """
Alex Chanb00ede22017-01-30 07:24:40 +00001404 `X509Req.get_extensions` returns a `list` of extensions added to this
1405 X509 request.
Stephen Holsapple7fbdf642014-03-01 20:05:47 -08001406 """
1407 request = X509Req()
1408 exts = request.get_extensions()
Alex Chanb00ede22017-01-30 07:24:40 +00001409 assert exts == []
Alex Gaynor03737182020-07-23 20:40:46 -04001410 request.add_extensions(
1411 [
1412 X509Extension(b"basicConstraints", True, b"CA:true"),
1413 X509Extension(b"keyUsage", False, b"digitalSignature"),
1414 ]
1415 )
Stephen Holsapple7fbdf642014-03-01 20:05:47 -08001416 exts = request.get_extensions()
Alex Chanb00ede22017-01-30 07:24:40 +00001417 assert len(exts) == 2
Alex Gaynor03737182020-07-23 20:40:46 -04001418 assert exts[0].get_short_name() == b"basicConstraints"
Alex Chanb00ede22017-01-30 07:24:40 +00001419 assert exts[0].get_critical() == 1
Alex Gaynor03737182020-07-23 20:40:46 -04001420 assert exts[0].get_data() == b"0\x03\x01\x01\xff"
1421 assert exts[1].get_short_name() == b"keyUsage"
Alex Chanb00ede22017-01-30 07:24:40 +00001422 assert exts[1].get_critical() == 0
Alex Gaynor03737182020-07-23 20:40:46 -04001423 assert exts[1].get_data() == b"\x03\x02\x07\x80"
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -04001424
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -04001425 def test_add_extensions_wrong_args(self):
1426 """
Alex Chanb00ede22017-01-30 07:24:40 +00001427 `X509Req.add_extensions` raises `TypeError` if called with a
1428 non-`list`. Or it raises `ValueError` if called with a `list`
1429 containing objects other than `X509Extension` instances.
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -04001430 """
1431 request = X509Req()
Alex Chanb00ede22017-01-30 07:24:40 +00001432 with pytest.raises(TypeError):
1433 request.add_extensions(object())
1434 with pytest.raises(ValueError):
1435 request.add_extensions([object()])
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -04001436
Jean-Paul Calderone5565f0f2013-03-06 11:10:20 -08001437 def test_verify_wrong_args(self):
1438 """
Alex Chanb00ede22017-01-30 07:24:40 +00001439 `X509Req.verify` raises `TypeError` if passed anything other than a
1440 `PKey` instance as its single argument.
Jean-Paul Calderone5565f0f2013-03-06 11:10:20 -08001441 """
1442 request = X509Req()
Alex Chanb00ede22017-01-30 07:24:40 +00001443 with pytest.raises(TypeError):
1444 request.verify(object())
Jean-Paul Calderone5565f0f2013-03-06 11:10:20 -08001445
Jean-Paul Calderone5565f0f2013-03-06 11:10:20 -08001446 def test_verify_uninitialized_key(self):
1447 """
Alex Chanb00ede22017-01-30 07:24:40 +00001448 `X509Req.verify` raises `OpenSSL.crypto.Error` if called with a
1449 `OpenSSL.crypto.PKey` which contains no key data.
Jean-Paul Calderone5565f0f2013-03-06 11:10:20 -08001450 """
1451 request = X509Req()
1452 pkey = PKey()
Alex Chanb00ede22017-01-30 07:24:40 +00001453 with pytest.raises(Error):
1454 request.verify(pkey)
Jean-Paul Calderone5565f0f2013-03-06 11:10:20 -08001455
Jean-Paul Calderone5565f0f2013-03-06 11:10:20 -08001456 def test_verify_wrong_key(self):
1457 """
Alex Chanb00ede22017-01-30 07:24:40 +00001458 `X509Req.verify` raises `OpenSSL.crypto.Error` if called with a
1459 `OpenSSL.crypto.PKey` which does not represent the public part of the
Alex Gaynor31287502015-09-05 16:11:27 -04001460 key which signed the request.
Jean-Paul Calderone5565f0f2013-03-06 11:10:20 -08001461 """
1462 request = X509Req()
1463 pkey = load_privatekey(FILETYPE_PEM, cleartextPrivateKeyPEM)
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05001464 request.sign(pkey, GOOD_DIGEST)
Alex Chanb00ede22017-01-30 07:24:40 +00001465 another_pkey = load_privatekey(FILETYPE_PEM, client_key_pem)
1466 with pytest.raises(Error):
1467 request.verify(another_pkey)
1468
1469 def test_verify_success(self):
1470 """
1471 `X509Req.verify` returns `True` if called with a `OpenSSL.crypto.PKey`
1472 which represents the public part of the key which signed the request.
1473 """
1474 request = X509Req()
1475 pkey = load_privatekey(FILETYPE_PEM, cleartextPrivateKeyPEM)
1476 request.sign(pkey, GOOD_DIGEST)
1477 assert request.verify(pkey)
Jean-Paul Calderone5565f0f2013-03-06 11:10:20 -08001478
Paul Kehrer41c10242017-06-29 18:24:17 -05001479 def test_convert_from_cryptography(self):
1480 crypto_req = x509.load_pem_x509_csr(
1481 cleartextCertificateRequestPEM, backend
1482 )
1483 req = X509Req.from_cryptography(crypto_req)
1484 assert isinstance(req, X509Req)
1485
1486 def test_convert_from_cryptography_unsupported_type(self):
1487 with pytest.raises(TypeError):
1488 X509Req.from_cryptography(object())
1489
1490 def test_convert_to_cryptography_key(self):
1491 req = load_certificate_request(
1492 FILETYPE_PEM, cleartextCertificateRequestPEM
1493 )
1494 crypto_req = req.to_cryptography()
1495 assert isinstance(crypto_req, x509.CertificateSigningRequest)
1496
Jean-Paul Calderone5565f0f2013-03-06 11:10:20 -08001497
Alex Chanb00ede22017-01-30 07:24:40 +00001498class TestX509(_PKeyInteractionTestsMixin):
Jean-Paul Calderone78381d22008-03-06 23:35:22 -05001499 """
Alex Chanb00ede22017-01-30 07:24:40 +00001500 Tests for `OpenSSL.crypto.X509`.
Jean-Paul Calderone78381d22008-03-06 23:35:22 -05001501 """
Alex Gaynor03737182020-07-23 20:40:46 -04001502
Jean-Paul Calderone5ef86512008-04-26 19:06:28 -04001503 pemData = cleartextCertificatePEM + cleartextPrivateKeyPEM
Jean-Paul Calderone8114b452008-03-25 15:27:59 -04001504
Roland Hedberg7e4930e2008-04-22 22:58:50 +02001505 extpem = """
1506-----BEGIN CERTIFICATE-----
1507MIIC3jCCAkegAwIBAgIJAJHFjlcCgnQzMA0GCSqGSIb3DQEBBQUAMEcxCzAJBgNV
1508BAYTAlNFMRUwEwYDVQQIEwxXZXN0ZXJib3R0b20xEjAQBgNVBAoTCUNhdGFsb2dp
1509eDENMAsGA1UEAxMEUm9vdDAeFw0wODA0MjIxNDQ1MzhaFw0wOTA0MjIxNDQ1Mzha
1510MFQxCzAJBgNVBAYTAlNFMQswCQYDVQQIEwJXQjEUMBIGA1UEChMLT3Blbk1ldGFk
1511aXIxIjAgBgNVBAMTGW5vZGUxLm9tMi5vcGVubWV0YWRpci5vcmcwgZ8wDQYJKoZI
1512hvcNAQEBBQADgY0AMIGJAoGBAPIcQMrwbk2nESF/0JKibj9i1x95XYAOwP+LarwT
1513Op4EQbdlI9SY+uqYqlERhF19w7CS+S6oyqx0DRZSk4Y9dZ9j9/xgm2u/f136YS1u
1514zgYFPvfUs6PqYLPSM8Bw+SjJ+7+2+TN+Tkiof9WP1cMjodQwOmdsiRbR0/J7+b1B
1515hec1AgMBAAGjgcQwgcEwCQYDVR0TBAIwADAsBglghkgBhvhCAQ0EHxYdT3BlblNT
1516TCBHZW5lcmF0ZWQgQ2VydGlmaWNhdGUwHQYDVR0OBBYEFIdHsBcMVVMbAO7j6NCj
151703HgLnHaMB8GA1UdIwQYMBaAFL2h9Bf9Mre4vTdOiHTGAt7BRY/8MEYGA1UdEQQ/
1518MD2CDSouZXhhbXBsZS5vcmeCESoub20yLmV4bWFwbGUuY29thwSC7wgKgRNvbTJA
1519b3Blbm1ldGFkaXIub3JnMA0GCSqGSIb3DQEBBQUAA4GBALd7WdXkp2KvZ7/PuWZA
1520MPlIxyjS+Ly11+BNE0xGQRp9Wz+2lABtpgNqssvU156+HkKd02rGheb2tj7MX9hG
1521uZzbwDAZzJPjzDQDD7d3cWsrVcfIdqVU7epHqIadnOF+X0ghJ39pAm6VVadnSXCt
1522WpOdIpB8KksUTCzV591Nr1wd
1523-----END CERTIFICATE-----
1524 """
Alex Gaynoraceb3e22015-09-05 12:00:22 -04001525
Jean-Paul Calderoneac0d95f2008-03-10 00:00:42 -04001526 def signable(self):
1527 """
Alex Chanb00ede22017-01-30 07:24:40 +00001528 Create and return a new `X509`.
Jean-Paul Calderoneac0d95f2008-03-10 00:00:42 -04001529 """
1530 return X509()
1531
Jean-Paul Calderone68649052009-07-17 21:14:27 -04001532 def test_type(self):
1533 """
Alex Gaynor01f90a12019-02-07 09:14:48 -05001534 `X509` can be used to create instances of that type.
Jean-Paul Calderone68649052009-07-17 21:14:27 -04001535 """
Alex Gaynor03737182020-07-23 20:40:46 -04001536 assert is_consistent_type(X509, "X509")
Jean-Paul Calderone68649052009-07-17 21:14:27 -04001537
Jean-Paul Calderone78381d22008-03-06 23:35:22 -05001538 def test_construction(self):
1539 """
Alex Gaynor01f90a12019-02-07 09:14:48 -05001540 `X509` takes no arguments and returns an instance of `X509`.
Jean-Paul Calderone78381d22008-03-06 23:35:22 -05001541 """
1542 certificate = X509()
Alex Gaynor01f90a12019-02-07 09:14:48 -05001543 assert isinstance(certificate, X509)
Alex Gaynor03737182020-07-23 20:40:46 -04001544 assert type(certificate).__name__ == "X509"
Alex Chanb00ede22017-01-30 07:24:40 +00001545 assert type(certificate) == X509
Jean-Paul Calderone3544eb42010-07-30 22:09:47 -04001546
Jean-Paul Calderone3544eb42010-07-30 22:09:47 -04001547 def test_set_version_wrong_args(self):
1548 """
Alex Chanb00ede22017-01-30 07:24:40 +00001549 `X509.set_version` raises `TypeError` if invoked with an argument
1550 not of type `int`.
Jean-Paul Calderone3544eb42010-07-30 22:09:47 -04001551 """
1552 cert = X509()
Alex Chanb00ede22017-01-30 07:24:40 +00001553 with pytest.raises(TypeError):
1554 cert.set_version(None)
Jean-Paul Calderone3544eb42010-07-30 22:09:47 -04001555
Jean-Paul Calderone3544eb42010-07-30 22:09:47 -04001556 def test_version(self):
1557 """
Alex Chanb00ede22017-01-30 07:24:40 +00001558 `X509.set_version` sets the certificate version number.
1559 `X509.get_version` retrieves it.
Jean-Paul Calderone3544eb42010-07-30 22:09:47 -04001560 """
1561 cert = X509()
1562 cert.set_version(1234)
Alex Chanb00ede22017-01-30 07:24:40 +00001563 assert cert.get_version() == 1234
Jean-Paul Calderone3544eb42010-07-30 22:09:47 -04001564
Jean-Paul Calderone78381d22008-03-06 23:35:22 -05001565 def test_serial_number(self):
1566 """
Alex Chanb00ede22017-01-30 07:24:40 +00001567 The serial number of an `X509` can be retrieved and
1568 modified with `X509.get_serial_number` and
1569 `X509.set_serial_number`.
Jean-Paul Calderone78381d22008-03-06 23:35:22 -05001570 """
1571 certificate = X509()
Alex Chanb00ede22017-01-30 07:24:40 +00001572 with pytest.raises(TypeError):
1573 certificate.set_serial_number("1")
1574 assert certificate.get_serial_number() == 0
Jean-Paul Calderone78381d22008-03-06 23:35:22 -05001575 certificate.set_serial_number(1)
Alex Chanb00ede22017-01-30 07:24:40 +00001576 assert certificate.get_serial_number() == 1
Jean-Paul Calderone78381d22008-03-06 23:35:22 -05001577 certificate.set_serial_number(2 ** 32 + 1)
Alex Chanb00ede22017-01-30 07:24:40 +00001578 assert certificate.get_serial_number() == 2 ** 32 + 1
Jean-Paul Calderone78381d22008-03-06 23:35:22 -05001579 certificate.set_serial_number(2 ** 64 + 1)
Alex Chanb00ede22017-01-30 07:24:40 +00001580 assert certificate.get_serial_number() == 2 ** 64 + 1
Jean-Paul Calderone525ef802008-03-09 20:39:42 -04001581 certificate.set_serial_number(2 ** 128 + 1)
Alex Chanb00ede22017-01-30 07:24:40 +00001582 assert certificate.get_serial_number() == 2 ** 128 + 1
Jean-Paul Calderone525ef802008-03-09 20:39:42 -04001583
Jean-Paul Calderone525ef802008-03-09 20:39:42 -04001584 def _setBoundTest(self, which):
1585 """
Alex Chanb00ede22017-01-30 07:24:40 +00001586 `X509.set_notBefore` takes a string in the format of an
Alex Gaynor31287502015-09-05 16:11:27 -04001587 ASN1 GENERALIZEDTIME and sets the beginning of the certificate's
1588 validity period to it.
Jean-Paul Calderone525ef802008-03-09 20:39:42 -04001589 """
1590 certificate = X509()
Alex Gaynor03737182020-07-23 20:40:46 -04001591 set = getattr(certificate, "set_not" + which)
1592 get = getattr(certificate, "get_not" + which)
Jean-Paul Calderone525ef802008-03-09 20:39:42 -04001593
Jean-Paul Calderonee0615b52008-03-09 21:44:46 -04001594 # Starts with no value.
Alex Chanb00ede22017-01-30 07:24:40 +00001595 assert get() is None
Jean-Paul Calderonee0615b52008-03-09 21:44:46 -04001596
Jean-Paul Calderone525ef802008-03-09 20:39:42 -04001597 # GMT (Or is it UTC?) -exarkun
Alex Gaynore7f51982016-09-11 11:48:14 -04001598 when = b"20040203040506Z"
Jean-Paul Calderone525ef802008-03-09 20:39:42 -04001599 set(when)
Alex Chanb00ede22017-01-30 07:24:40 +00001600 assert get() == when
Jean-Paul Calderone525ef802008-03-09 20:39:42 -04001601
1602 # A plus two hours and thirty minutes offset
Alex Gaynore7f51982016-09-11 11:48:14 -04001603 when = b"20040203040506+0530"
Jean-Paul Calderone525ef802008-03-09 20:39:42 -04001604 set(when)
Alex Chanb00ede22017-01-30 07:24:40 +00001605 assert get() == when
Jean-Paul Calderone525ef802008-03-09 20:39:42 -04001606
1607 # A minus one hour fifteen minutes offset
Alex Gaynore7f51982016-09-11 11:48:14 -04001608 when = b"20040203040506-0115"
Jean-Paul Calderone525ef802008-03-09 20:39:42 -04001609 set(when)
Alex Chanb00ede22017-01-30 07:24:40 +00001610 assert get() == when
Jean-Paul Calderone525ef802008-03-09 20:39:42 -04001611
1612 # An invalid string results in a ValueError
Alex Chanb00ede22017-01-30 07:24:40 +00001613 with pytest.raises(ValueError):
1614 set(b"foo bar")
Jean-Paul Calderone525ef802008-03-09 20:39:42 -04001615
Jean-Paul Calderone31ca2002010-01-30 15:14:43 -05001616 # The wrong number of arguments results in a TypeError.
Alex Chanb00ede22017-01-30 07:24:40 +00001617 with pytest.raises(TypeError):
1618 set()
Alex Gaynor85b49702015-09-05 16:30:59 -04001619 with pytest.raises(TypeError):
1620 set(b"20040203040506Z", b"20040203040506Z")
Alex Chanb00ede22017-01-30 07:24:40 +00001621 with pytest.raises(TypeError):
1622 get(b"foo bar")
Jean-Paul Calderonee890db32010-08-22 16:55:15 -04001623
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -04001624 # XXX ASN1_TIME (not GENERALIZEDTIME)
Jean-Paul Calderone525ef802008-03-09 20:39:42 -04001625
1626 def test_set_notBefore(self):
1627 """
Alex Chanb00ede22017-01-30 07:24:40 +00001628 `X509.set_notBefore` takes a string in the format of an
Alex Gaynor31287502015-09-05 16:11:27 -04001629 ASN1 GENERALIZEDTIME and sets the beginning of the certificate's
1630 validity period to it.
Jean-Paul Calderone525ef802008-03-09 20:39:42 -04001631 """
1632 self._setBoundTest("Before")
1633
Jean-Paul Calderone525ef802008-03-09 20:39:42 -04001634 def test_set_notAfter(self):
1635 """
Alex Chanb00ede22017-01-30 07:24:40 +00001636 `X509.set_notAfter` takes a string in the format of an ASN1
Jean-Paul Calderone525ef802008-03-09 20:39:42 -04001637 GENERALIZEDTIME and sets the end of the certificate's validity period
1638 to it.
1639 """
1640 self._setBoundTest("After")
Jean-Paul Calderone76576d52008-03-24 16:04:46 -04001641
Jean-Paul Calderone38a646d2008-03-25 15:16:18 -04001642 def test_get_notBefore(self):
1643 """
Alex Chanb00ede22017-01-30 07:24:40 +00001644 `X509.get_notBefore` returns a string in the format of an
Alex Gaynor31287502015-09-05 16:11:27 -04001645 ASN1 GENERALIZEDTIME even for certificates which store it as UTCTIME
Jean-Paul Calderone38a646d2008-03-25 15:16:18 -04001646 internally.
1647 """
Paul Kehrera40898b2017-06-11 16:30:58 -10001648 cert = load_certificate(FILETYPE_PEM, old_root_cert_pem)
Alex Chanb00ede22017-01-30 07:24:40 +00001649 assert cert.get_notBefore() == b"20090325123658Z"
Jean-Paul Calderone38a646d2008-03-25 15:16:18 -04001650
Rick Dean38a05c82009-07-18 01:41:30 -05001651 def test_get_notAfter(self):
1652 """
Alex Chanb00ede22017-01-30 07:24:40 +00001653 `X509.get_notAfter` returns a string in the format of an
Alex Gaynor31287502015-09-05 16:11:27 -04001654 ASN1 GENERALIZEDTIME even for certificates which store it as UTCTIME
Rick Dean38a05c82009-07-18 01:41:30 -05001655 internally.
1656 """
Paul Kehrera40898b2017-06-11 16:30:58 -10001657 cert = load_certificate(FILETYPE_PEM, old_root_cert_pem)
Alex Chanb00ede22017-01-30 07:24:40 +00001658 assert cert.get_notAfter() == b"20170611123658Z"
Rick Dean38a05c82009-07-18 01:41:30 -05001659
Jean-Paul Calderoneccf9d482010-07-30 22:36:42 -04001660 def test_gmtime_adj_notBefore_wrong_args(self):
1661 """
Alex Chanb00ede22017-01-30 07:24:40 +00001662 `X509.gmtime_adj_notBefore` raises `TypeError` if called with a
1663 non-`int` argument.
Jean-Paul Calderoneccf9d482010-07-30 22:36:42 -04001664 """
1665 cert = X509()
Alex Chanb00ede22017-01-30 07:24:40 +00001666 with pytest.raises(TypeError):
1667 cert.gmtime_adj_notBefore(None)
Jean-Paul Calderoneccf9d482010-07-30 22:36:42 -04001668
Alex Gaynor7f5610c2017-07-07 00:09:34 -04001669 @flaky.flaky
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001670 def test_gmtime_adj_notBefore(self):
1671 """
Alex Chanb00ede22017-01-30 07:24:40 +00001672 `X509.gmtime_adj_notBefore` changes the not-before timestamp to be the
1673 current time plus the number of seconds passed in.
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001674 """
1675 cert = load_certificate(FILETYPE_PEM, self.pemData)
Alex Gaynor03737182020-07-23 20:40:46 -04001676 not_before_min = datetime.utcnow().replace(microsecond=0) + timedelta(
1677 seconds=100
Alex Gaynor85b49702015-09-05 16:30:59 -04001678 )
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001679 cert.gmtime_adj_notBefore(100)
Alex Gaynor85b49702015-09-05 16:30:59 -04001680 not_before = datetime.strptime(
1681 cert.get_notBefore().decode(), "%Y%m%d%H%M%SZ"
1682 )
Alex Gaynor7f5610c2017-07-07 00:09:34 -04001683 not_before_max = datetime.utcnow() + timedelta(seconds=100)
Alex Chanb00ede22017-01-30 07:24:40 +00001684 assert not_before_min <= not_before <= not_before_max
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001685
Jean-Paul Calderoneccf9d482010-07-30 22:36:42 -04001686 def test_gmtime_adj_notAfter_wrong_args(self):
1687 """
Alex Chanb00ede22017-01-30 07:24:40 +00001688 `X509.gmtime_adj_notAfter` raises `TypeError` if called with a
1689 non-`int` argument.
Jean-Paul Calderoneccf9d482010-07-30 22:36:42 -04001690 """
1691 cert = X509()
Alex Chanb00ede22017-01-30 07:24:40 +00001692 with pytest.raises(TypeError):
1693 cert.gmtime_adj_notAfter(None)
Jean-Paul Calderoneccf9d482010-07-30 22:36:42 -04001694
Alex Gaynor642de6f2017-07-24 00:57:38 -04001695 @flaky.flaky
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001696 def test_gmtime_adj_notAfter(self):
1697 """
Alex Chanb00ede22017-01-30 07:24:40 +00001698 `X509.gmtime_adj_notAfter` changes the not-after timestamp
Alex Gaynor31287502015-09-05 16:11:27 -04001699 to be the current time plus the number of seconds passed in.
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001700 """
1701 cert = load_certificate(FILETYPE_PEM, self.pemData)
Alex Gaynor03737182020-07-23 20:40:46 -04001702 not_after_min = datetime.utcnow().replace(microsecond=0) + timedelta(
1703 seconds=100
Alex Gaynor85b49702015-09-05 16:30:59 -04001704 )
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001705 cert.gmtime_adj_notAfter(100)
Alex Gaynor85b49702015-09-05 16:30:59 -04001706 not_after = datetime.strptime(
1707 cert.get_notAfter().decode(), "%Y%m%d%H%M%SZ"
1708 )
Maximilian Hilsbed25c92015-07-25 12:58:07 +02001709 not_after_max = datetime.utcnow() + timedelta(seconds=100)
Alex Chanb00ede22017-01-30 07:24:40 +00001710 assert not_after_min <= not_after <= not_after_max
Jean-Paul Calderoneccf9d482010-07-30 22:36:42 -04001711
Jean-Paul Calderoneccf9d482010-07-30 22:36:42 -04001712 def test_has_expired(self):
1713 """
Alex Chanb00ede22017-01-30 07:24:40 +00001714 `X509.has_expired` returns `True` if the certificate's not-after time
1715 is in the past.
Jean-Paul Calderoneccf9d482010-07-30 22:36:42 -04001716 """
1717 cert = X509()
1718 cert.gmtime_adj_notAfter(-1)
Alex Chanb00ede22017-01-30 07:24:40 +00001719 assert cert.has_expired()
Jean-Paul Calderoneccf9d482010-07-30 22:36:42 -04001720
Jean-Paul Calderoneccf9d482010-07-30 22:36:42 -04001721 def test_has_not_expired(self):
1722 """
Alex Chanb00ede22017-01-30 07:24:40 +00001723 `X509.has_expired` returns `False` if the certificate's not-after time
1724 is in the future.
Jean-Paul Calderoneccf9d482010-07-30 22:36:42 -04001725 """
1726 cert = X509()
1727 cert.gmtime_adj_notAfter(2)
Alex Chanb00ede22017-01-30 07:24:40 +00001728 assert not cert.has_expired()
Jean-Paul Calderoneccf9d482010-07-30 22:36:42 -04001729
Jeff Tangfc18f7b2015-04-15 17:42:33 -04001730 def test_root_has_not_expired(self):
1731 """
Alex Chanb00ede22017-01-30 07:24:40 +00001732 `X509.has_expired` returns `False` if the certificate's not-after time
1733 is in the future.
Jeff Tangfc18f7b2015-04-15 17:42:33 -04001734 """
1735 cert = load_certificate(FILETYPE_PEM, root_cert_pem)
Alex Chanb00ede22017-01-30 07:24:40 +00001736 assert not cert.has_expired()
Jeff Tangfc18f7b2015-04-15 17:42:33 -04001737
Rick Dean38a05c82009-07-18 01:41:30 -05001738 def test_digest(self):
1739 """
Alex Chanb00ede22017-01-30 07:24:40 +00001740 `X509.digest` returns a string giving ":"-separated hex-encoded
Alex Gaynor31287502015-09-05 16:11:27 -04001741 words of the digest of the certificate.
Rick Dean38a05c82009-07-18 01:41:30 -05001742 """
Paul Kehrera40898b2017-06-11 16:30:58 -10001743 cert = load_certificate(FILETYPE_PEM, old_root_cert_pem)
Alex Chanb00ede22017-01-30 07:24:40 +00001744 assert (
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05001745 # This is MD5 instead of GOOD_DIGEST because the digest algorithm
1746 # actually matters to the assertion (ie, another arbitrary, good
1747 # digest will not product the same digest).
Jeff Tangfc18f7b2015-04-15 17:42:33 -04001748 # Digest verified with the command:
1749 # openssl x509 -in root_cert.pem -noout -fingerprint -md5
Alex Gaynor03737182020-07-23 20:40:46 -04001750 cert.digest("MD5")
1751 == b"19:B3:05:26:2B:F8:F2:FF:0B:8F:21:07:A8:28:B8:75"
1752 )
Rick Dean38a05c82009-07-18 01:41:30 -05001753
Jean-Paul Calderone83a593d2011-04-01 17:45:07 -04001754 def _extcert(self, pkey, extensions):
1755 cert = X509()
David Benjamin6b799472020-06-24 17:14:16 -04001756 # Certificates with extensions must be X.509v3, which is encoded with a
1757 # version of two.
1758 cert.set_version(2)
Jean-Paul Calderone83a593d2011-04-01 17:45:07 -04001759 cert.set_pubkey(pkey)
1760 cert.get_subject().commonName = "Unit Tests"
1761 cert.get_issuer().commonName = "Unit Tests"
Alex Gaynore7f51982016-09-11 11:48:14 -04001762 when = datetime.now().strftime("%Y%m%d%H%M%SZ").encode("ascii")
Jean-Paul Calderone83a593d2011-04-01 17:45:07 -04001763 cert.set_notBefore(when)
1764 cert.set_notAfter(when)
1765
1766 cert.add_extensions(extensions)
Alex Gaynor03737182020-07-23 20:40:46 -04001767 cert.sign(pkey, "sha1")
Jean-Paul Calderone83a593d2011-04-01 17:45:07 -04001768 return load_certificate(
Alex Gaynor03737182020-07-23 20:40:46 -04001769 FILETYPE_PEM, dump_certificate(FILETYPE_PEM, cert)
1770 )
Jean-Paul Calderone83a593d2011-04-01 17:45:07 -04001771
Roland Hedberg7e4930e2008-04-22 22:58:50 +02001772 def test_extension_count(self):
1773 """
Alex Chanb00ede22017-01-30 07:24:40 +00001774 `X509.get_extension_count` returns the number of extensions
Alex Gaynor31287502015-09-05 16:11:27 -04001775 that are present in the certificate.
Roland Hedberg7e4930e2008-04-22 22:58:50 +02001776 """
Jean-Paul Calderonef7b3ee62011-04-01 17:36:24 -04001777 pkey = load_privatekey(FILETYPE_PEM, client_key_pem)
Alex Gaynor03737182020-07-23 20:40:46 -04001778 ca = X509Extension(b"basicConstraints", True, b"CA:FALSE")
1779 key = X509Extension(b"keyUsage", True, b"digitalSignature")
Jean-Paul Calderonef7b3ee62011-04-01 17:36:24 -04001780 subjectAltName = X509Extension(
Alex Gaynor03737182020-07-23 20:40:46 -04001781 b"subjectAltName", True, b"DNS:example.com"
1782 )
Jean-Paul Calderonef7b3ee62011-04-01 17:36:24 -04001783
1784 # Try a certificate with no extensions at all.
Jean-Paul Calderone83a593d2011-04-01 17:45:07 -04001785 c = self._extcert(pkey, [])
Alex Chanb00ede22017-01-30 07:24:40 +00001786 assert c.get_extension_count() == 0
Jean-Paul Calderonef7b3ee62011-04-01 17:36:24 -04001787
1788 # And a certificate with one
Jean-Paul Calderone83a593d2011-04-01 17:45:07 -04001789 c = self._extcert(pkey, [ca])
Alex Chanb00ede22017-01-30 07:24:40 +00001790 assert c.get_extension_count() == 1
Jean-Paul Calderonef7b3ee62011-04-01 17:36:24 -04001791
1792 # And a certificate with several
Jean-Paul Calderone83a593d2011-04-01 17:45:07 -04001793 c = self._extcert(pkey, [ca, key, subjectAltName])
Alex Chanb00ede22017-01-30 07:24:40 +00001794 assert c.get_extension_count() == 3
Roland Hedberg7e4930e2008-04-22 22:58:50 +02001795
Jean-Paul Calderone83a593d2011-04-01 17:45:07 -04001796 def test_get_extension(self):
1797 """
Alex Chanb00ede22017-01-30 07:24:40 +00001798 `X509.get_extension` takes an integer and returns an
1799 `X509Extension` corresponding to the extension at that index.
Jean-Paul Calderone83a593d2011-04-01 17:45:07 -04001800 """
1801 pkey = load_privatekey(FILETYPE_PEM, client_key_pem)
Alex Gaynor03737182020-07-23 20:40:46 -04001802 ca = X509Extension(b"basicConstraints", True, b"CA:FALSE")
1803 key = X509Extension(b"keyUsage", True, b"digitalSignature")
Jean-Paul Calderone83a593d2011-04-01 17:45:07 -04001804 subjectAltName = X509Extension(
Alex Gaynor03737182020-07-23 20:40:46 -04001805 b"subjectAltName", False, b"DNS:example.com"
1806 )
Jean-Paul Calderone83a593d2011-04-01 17:45:07 -04001807
1808 cert = self._extcert(pkey, [ca, key, subjectAltName])
1809
1810 ext = cert.get_extension(0)
Alex Chanb00ede22017-01-30 07:24:40 +00001811 assert isinstance(ext, X509Extension)
1812 assert ext.get_critical()
Alex Gaynor03737182020-07-23 20:40:46 -04001813 assert ext.get_short_name() == b"basicConstraints"
Jean-Paul Calderone83a593d2011-04-01 17:45:07 -04001814
1815 ext = cert.get_extension(1)
Alex Chanb00ede22017-01-30 07:24:40 +00001816 assert isinstance(ext, X509Extension)
1817 assert ext.get_critical()
Alex Gaynor03737182020-07-23 20:40:46 -04001818 assert ext.get_short_name() == b"keyUsage"
Jean-Paul Calderone83a593d2011-04-01 17:45:07 -04001819
1820 ext = cert.get_extension(2)
Alex Chanb00ede22017-01-30 07:24:40 +00001821 assert isinstance(ext, X509Extension)
1822 assert not ext.get_critical()
Alex Gaynor03737182020-07-23 20:40:46 -04001823 assert ext.get_short_name() == b"subjectAltName"
Jean-Paul Calderone83a593d2011-04-01 17:45:07 -04001824
Alex Chanb00ede22017-01-30 07:24:40 +00001825 with pytest.raises(IndexError):
1826 cert.get_extension(-1)
1827 with pytest.raises(IndexError):
1828 cert.get_extension(4)
1829 with pytest.raises(TypeError):
1830 cert.get_extension("hello")
Jean-Paul Calderone83a593d2011-04-01 17:45:07 -04001831
Jean-Paul Calderone4bf75c62013-08-23 15:39:53 -04001832 def test_nullbyte_subjectAltName(self):
Jean-Paul Calderoneff83cdd2013-08-12 18:05:51 -04001833 """
Jean-Paul Calderone9af07b02013-08-23 16:07:31 -04001834 The fields of a `subjectAltName` extension on an X509 may contain NUL
Jean-Paul Calderone4bf75c62013-08-23 15:39:53 -04001835 bytes and this value is reflected in the string representation of the
1836 extension object.
Jean-Paul Calderoneff83cdd2013-08-12 18:05:51 -04001837 """
Jean-Paul Calderone4bf75c62013-08-23 15:39:53 -04001838 cert = load_certificate(FILETYPE_PEM, nulbyteSubjectAltNamePEM)
Jean-Paul Calderoneff83cdd2013-08-12 18:05:51 -04001839
1840 ext = cert.get_extension(3)
Alex Gaynor03737182020-07-23 20:40:46 -04001841 assert ext.get_short_name() == b"subjectAltName"
Alex Chanb00ede22017-01-30 07:24:40 +00001842 assert (
Alex Gaynore7f51982016-09-11 11:48:14 -04001843 b"DNS:altnull.python.org\x00example.com, "
1844 b"email:null@python.org\x00user@example.org, "
1845 b"URI:http://null.python.org\x00http://example.org, "
Alex Gaynor03737182020-07-23 20:40:46 -04001846 b"IP Address:192.0.2.1, IP Address:2001:DB8:0:0:0:0:0:1\n"
1847 == str(ext).encode("ascii")
1848 )
Jean-Paul Calderone4bf75c62013-08-23 15:39:53 -04001849
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -04001850 def test_invalid_digest_algorithm(self):
1851 """
Alex Chanb00ede22017-01-30 07:24:40 +00001852 `X509.digest` raises `ValueError` if called with an unrecognized hash
1853 algorithm.
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -04001854 """
1855 cert = X509()
Alex Chanb00ede22017-01-30 07:24:40 +00001856 with pytest.raises(ValueError):
1857 cert.digest(BAD_DIGEST)
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001858
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001859 def test_get_subject(self):
1860 """
Alex Chanb00ede22017-01-30 07:24:40 +00001861 `X509.get_subject` returns an `X509Name` instance.
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001862 """
1863 cert = load_certificate(FILETYPE_PEM, self.pemData)
1864 subj = cert.get_subject()
Alex Chanb00ede22017-01-30 07:24:40 +00001865 assert isinstance(subj, X509Name)
Alex Gaynor03737182020-07-23 20:40:46 -04001866 assert subj.get_components() == [
1867 (b"C", b"US"),
1868 (b"ST", b"IL"),
1869 (b"L", b"Chicago"),
1870 (b"O", b"Testing"),
1871 (b"CN", b"Testing Root CA"),
1872 ]
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001873
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001874 def test_set_subject_wrong_args(self):
1875 """
Alex Chanb00ede22017-01-30 07:24:40 +00001876 `X509.set_subject` raises a `TypeError` if called with an argument not
1877 of type `X509Name`.
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001878 """
1879 cert = X509()
Alex Gaynor85b49702015-09-05 16:30:59 -04001880 with pytest.raises(TypeError):
Alex Chanb00ede22017-01-30 07:24:40 +00001881 cert.set_subject(None)
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001882
1883 def test_set_subject(self):
1884 """
Alex Chanb00ede22017-01-30 07:24:40 +00001885 `X509.set_subject` changes the subject of the certificate to the one
1886 passed in.
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001887 """
1888 cert = X509()
1889 name = cert.get_subject()
Alex Gaynor03737182020-07-23 20:40:46 -04001890 name.C = "AU"
1891 name.OU = "Unit Tests"
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001892 cert.set_subject(name)
Alex Gaynor03737182020-07-23 20:40:46 -04001893 assert cert.get_subject().get_components() == [
1894 (b"C", b"AU"),
1895 (b"OU", b"Unit Tests"),
1896 ]
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001897
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001898 def test_get_issuer(self):
1899 """
Alex Chanb00ede22017-01-30 07:24:40 +00001900 `X509.get_issuer` returns an `X509Name` instance.
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001901 """
1902 cert = load_certificate(FILETYPE_PEM, self.pemData)
1903 subj = cert.get_issuer()
Alex Chanb00ede22017-01-30 07:24:40 +00001904 assert isinstance(subj, X509Name)
Jean-Paul Calderone30a4cb32010-08-11 23:54:12 -04001905 comp = subj.get_components()
Alex Gaynor03737182020-07-23 20:40:46 -04001906 assert comp == [
1907 (b"C", b"US"),
1908 (b"ST", b"IL"),
1909 (b"L", b"Chicago"),
1910 (b"O", b"Testing"),
1911 (b"CN", b"Testing Root CA"),
1912 ]
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001913
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001914 def test_set_issuer_wrong_args(self):
1915 """
Alex Chanb00ede22017-01-30 07:24:40 +00001916 `X509.set_issuer` raises a `TypeError` if called with an argument not
1917 of type `X509Name`.
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001918 """
1919 cert = X509()
Alex Chanb00ede22017-01-30 07:24:40 +00001920 with pytest.raises(TypeError):
1921 cert.set_issuer(None)
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001922
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001923 def test_set_issuer(self):
1924 """
Alex Chanb00ede22017-01-30 07:24:40 +00001925 `X509.set_issuer` changes the issuer of the certificate to the
Alex Gaynor31287502015-09-05 16:11:27 -04001926 one passed in.
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001927 """
1928 cert = X509()
1929 name = cert.get_issuer()
Alex Gaynor03737182020-07-23 20:40:46 -04001930 name.C = "AU"
1931 name.OU = "Unit Tests"
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001932 cert.set_issuer(name)
Alex Gaynor03737182020-07-23 20:40:46 -04001933 assert cert.get_issuer().get_components() == [
1934 (b"C", b"AU"),
1935 (b"OU", b"Unit Tests"),
1936 ]
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001937
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001938 def test_get_pubkey_uninitialized(self):
1939 """
Alex Chanb00ede22017-01-30 07:24:40 +00001940 When called on a certificate with no public key, `X509.get_pubkey`
1941 raises `OpenSSL.crypto.Error`.
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001942 """
1943 cert = X509()
Alex Chanb00ede22017-01-30 07:24:40 +00001944 with pytest.raises(Error):
1945 cert.get_pubkey()
Jean-Paul Calderone4f237b22010-07-30 22:29:39 -04001946
Alex Gaynor7778e792016-07-03 23:38:48 -04001947 def test_set_pubkey_wrong_type(self):
1948 """
Alex Chanb00ede22017-01-30 07:24:40 +00001949 `X509.set_pubkey` raises `TypeError` when given an object of the
1950 wrong type.
Alex Gaynor7778e792016-07-03 23:38:48 -04001951 """
1952 cert = X509()
1953 with pytest.raises(TypeError):
1954 cert.set_pubkey(object())
1955
Jean-Paul Calderoneccf9d482010-07-30 22:36:42 -04001956 def test_subject_name_hash(self):
1957 """
Alex Chanb00ede22017-01-30 07:24:40 +00001958 `X509.subject_name_hash` returns the hash of the certificate's
Alex Gaynor31287502015-09-05 16:11:27 -04001959 subject name.
Jean-Paul Calderoneccf9d482010-07-30 22:36:42 -04001960 """
1961 cert = load_certificate(FILETYPE_PEM, self.pemData)
Alex Gaynor4cb05202019-02-02 11:06:41 -05001962 # SHA1
1963 assert cert.subject_name_hash() == 3278919224
Jean-Paul Calderoneccf9d482010-07-30 22:36:42 -04001964
Jean-Paul Calderone2755fa52011-05-18 19:42:10 -04001965 def test_get_signature_algorithm(self):
1966 """
Alex Chanb00ede22017-01-30 07:24:40 +00001967 `X509.get_signature_algorithm` returns a string which means
Jean-Paul Calderone2755fa52011-05-18 19:42:10 -04001968 the algorithm used to sign the certificate.
1969 """
1970 cert = load_certificate(FILETYPE_PEM, self.pemData)
Alex Chanb00ede22017-01-30 07:24:40 +00001971 assert b"sha1WithRSAEncryption" == cert.get_signature_algorithm()
Jean-Paul Calderone2755fa52011-05-18 19:42:10 -04001972
Jean-Paul Calderone2755fa52011-05-18 19:42:10 -04001973 def test_get_undefined_signature_algorithm(self):
Jean-Paul Calderone5d8e4052011-05-19 17:51:43 -04001974 """
Alex Chanb00ede22017-01-30 07:24:40 +00001975 `X509.get_signature_algorithm` raises `ValueError` if the signature
1976 algorithm is undefined or unknown.
Jean-Paul Calderone5d8e4052011-05-19 17:51:43 -04001977 """
1978 # This certificate has been modified to indicate a bogus OID in the
1979 # signature algorithm field so that OpenSSL does not recognize it.
Alex Gaynore7f51982016-09-11 11:48:14 -04001980 certPEM = b"""\
Jean-Paul Calderone2755fa52011-05-18 19:42:10 -04001981-----BEGIN CERTIFICATE-----
1982MIIC/zCCAmigAwIBAgIBATAGBgJ8BQUAMHsxCzAJBgNVBAYTAlNHMREwDwYDVQQK
1983EwhNMkNyeXB0bzEUMBIGA1UECxMLTTJDcnlwdG8gQ0ExJDAiBgNVBAMTG00yQ3J5
1984cHRvIENlcnRpZmljYXRlIE1hc3RlcjEdMBsGCSqGSIb3DQEJARYObmdwc0Bwb3N0
1985MS5jb20wHhcNMDAwOTEwMDk1MTMwWhcNMDIwOTEwMDk1MTMwWjBTMQswCQYDVQQG
1986EwJTRzERMA8GA1UEChMITTJDcnlwdG8xEjAQBgNVBAMTCWxvY2FsaG9zdDEdMBsG
1987CSqGSIb3DQEJARYObmdwc0Bwb3N0MS5jb20wXDANBgkqhkiG9w0BAQEFAANLADBI
1988AkEArL57d26W9fNXvOhNlZzlPOACmvwOZ5AdNgLzJ1/MfsQQJ7hHVeHmTAjM664V
1989+fXvwUGJLziCeBo1ysWLRnl8CQIDAQABo4IBBDCCAQAwCQYDVR0TBAIwADAsBglg
1990hkgBhvhCAQ0EHxYdT3BlblNTTCBHZW5lcmF0ZWQgQ2VydGlmaWNhdGUwHQYDVR0O
1991BBYEFM+EgpK+eyZiwFU1aOPSbczbPSpVMIGlBgNVHSMEgZ0wgZqAFPuHI2nrnDqT
1992FeXFvylRT/7tKDgBoX+kfTB7MQswCQYDVQQGEwJTRzERMA8GA1UEChMITTJDcnlw
1993dG8xFDASBgNVBAsTC00yQ3J5cHRvIENBMSQwIgYDVQQDExtNMkNyeXB0byBDZXJ0
1994aWZpY2F0ZSBNYXN0ZXIxHTAbBgkqhkiG9w0BCQEWDm5ncHNAcG9zdDEuY29tggEA
1995MA0GCSqGSIb3DQEBBAUAA4GBADv8KpPo+gfJxN2ERK1Y1l17sz/ZhzoGgm5XCdbx
1996jEY7xKfpQngV599k1xhl11IMqizDwu0855agrckg2MCTmOI9DZzDD77tAYb+Dk0O
1997PEVk0Mk/V0aIsDE9bolfCi/i/QWZ3N8s5nTWMNyBBBmoSliWCm4jkkRZRD0ejgTN
1998tgI5
1999-----END CERTIFICATE-----
Alex Gaynore7f51982016-09-11 11:48:14 -04002000"""
Jean-Paul Calderone2755fa52011-05-18 19:42:10 -04002001 cert = load_certificate(FILETYPE_PEM, certPEM)
Alex Chanb00ede22017-01-30 07:24:40 +00002002 with pytest.raises(ValueError):
2003 cert.get_signature_algorithm()
Jean-Paul Calderoneccf9d482010-07-30 22:36:42 -04002004
Alex Gaynor37726112016-07-04 09:51:32 -04002005 def test_sign_bad_pubkey_type(self):
2006 """
Alex Chanb00ede22017-01-30 07:24:40 +00002007 `X509.sign` raises `TypeError` when called with the wrong type.
Alex Gaynor37726112016-07-04 09:51:32 -04002008 """
2009 cert = X509()
2010 with pytest.raises(TypeError):
2011 cert.sign(object(), b"sha256")
2012
Alex Gaynor9939ba12017-06-25 16:28:24 -04002013 def test_convert_from_cryptography(self):
2014 crypto_cert = x509.load_pem_x509_certificate(
2015 intermediate_cert_pem, backend
2016 )
2017 cert = X509.from_cryptography(crypto_cert)
2018
2019 assert isinstance(cert, X509)
2020 assert cert.get_version() == crypto_cert.version.value
2021
2022 def test_convert_from_cryptography_unsupported_type(self):
2023 with pytest.raises(TypeError):
2024 X509.from_cryptography(object())
2025
2026 def test_convert_to_cryptography_key(self):
2027 cert = load_certificate(FILETYPE_PEM, intermediate_cert_pem)
2028 crypto_cert = cert.to_cryptography()
2029
2030 assert isinstance(crypto_cert, x509.Certificate)
2031 assert crypto_cert.version.value == cert.get_version()
2032
Jean-Paul Calderoneccf9d482010-07-30 22:36:42 -04002033
Alex Chan9e2a9932017-01-25 14:29:19 +00002034class TestX509Store(object):
Jean-Paul Calderonea63714c2013-03-05 17:02:26 -08002035 """
Alex Chan9e2a9932017-01-25 14:29:19 +00002036 Test for `OpenSSL.crypto.X509Store`.
Jean-Paul Calderonea63714c2013-03-05 17:02:26 -08002037 """
Alex Gaynoraceb3e22015-09-05 12:00:22 -04002038
Jean-Paul Calderonea63714c2013-03-05 17:02:26 -08002039 def test_type(self):
2040 """
Alex Chan9e2a9932017-01-25 14:29:19 +00002041 `X509Store` is a type object.
Jean-Paul Calderonea63714c2013-03-05 17:02:26 -08002042 """
Alex Gaynor03737182020-07-23 20:40:46 -04002043 assert is_consistent_type(X509Store, "X509Store")
Jean-Paul Calderonea63714c2013-03-05 17:02:26 -08002044
Jean-Paul Calderonea63714c2013-03-05 17:02:26 -08002045 def test_add_cert(self):
Jean-Paul Calderonee6f32b82013-03-06 10:27:57 -08002046 """
Alex Chan9e2a9932017-01-25 14:29:19 +00002047 `X509Store.add_cert` adds a `X509` instance to the certificate store.
Jean-Paul Calderonee6f32b82013-03-06 10:27:57 -08002048 """
2049 cert = load_certificate(FILETYPE_PEM, cleartextCertificatePEM)
Jean-Paul Calderonea63714c2013-03-05 17:02:26 -08002050 store = X509Store()
Jean-Paul Calderonee6f32b82013-03-06 10:27:57 -08002051 store.add_cert(cert)
2052
Alex Gaynor03737182020-07-23 20:40:46 -04002053 @pytest.mark.parametrize("cert", [None, 1.0, "cert", object()])
Alex Chanfb078d82017-04-20 11:16:15 +01002054 def test_add_cert_wrong_args(self, cert):
2055 """
2056 `X509Store.add_cert` raises `TypeError` if passed a non-X509 object
2057 as its first argument.
2058 """
2059 store = X509Store()
2060 with pytest.raises(TypeError):
2061 store.add_cert(cert)
2062
Paul Kehrer0e6c5532018-08-23 10:52:15 -05002063 def test_add_cert_accepts_duplicate(self):
Jean-Paul Calderonee6f32b82013-03-06 10:27:57 -08002064 """
Paul Kehrer0e6c5532018-08-23 10:52:15 -05002065 `X509Store.add_cert` doesn't raise `OpenSSL.crypto.Error` if an attempt
2066 is made to add the same certificate to the store more than once.
Jean-Paul Calderonee6f32b82013-03-06 10:27:57 -08002067 """
2068 cert = load_certificate(FILETYPE_PEM, cleartextCertificatePEM)
2069 store = X509Store()
2070 store.add_cert(cert)
Paul Kehrer0e6c5532018-08-23 10:52:15 -05002071 store.add_cert(cert)
Jean-Paul Calderonea63714c2013-03-05 17:02:26 -08002072
2073
Alex Chanb00ede22017-01-30 07:24:40 +00002074class TestPKCS12(object):
Rick Dean623ee362009-07-17 12:22:16 -05002075 """
Alex Chanb00ede22017-01-30 07:24:40 +00002076 Test for `OpenSSL.crypto.PKCS12` and `OpenSSL.crypto.load_pkcs12`.
Rick Dean623ee362009-07-17 12:22:16 -05002077 """
Alex Gaynor03737182020-07-23 20:40:46 -04002078
Rick Dean623ee362009-07-17 12:22:16 -05002079 pemData = cleartextCertificatePEM + cleartextPrivateKeyPEM
2080
Jean-Paul Calderonec3a41f72009-07-25 12:36:02 -04002081 def test_type(self):
2082 """
Alex Gaynor01f90a12019-02-07 09:14:48 -05002083 `PKCS12` is a type object.
Jean-Paul Calderonec3a41f72009-07-25 12:36:02 -04002084 """
Alex Gaynor03737182020-07-23 20:40:46 -04002085 assert is_consistent_type(PKCS12, "PKCS12")
Jean-Paul Calderonec3a41f72009-07-25 12:36:02 -04002086
Rick Deanf94096c2009-07-18 14:23:06 -05002087 def test_empty_construction(self):
Rick Dean38a05c82009-07-18 01:41:30 -05002088 """
Alex Chanb00ede22017-01-30 07:24:40 +00002089 `PKCS12` returns a new instance of `PKCS12` with no certificate,
2090 private key, CA certificates, or friendly name.
Rick Dean38a05c82009-07-18 01:41:30 -05002091 """
Jean-Paul Calderonea202edb2009-07-25 12:22:12 -04002092 p12 = PKCS12()
Alex Chanb00ede22017-01-30 07:24:40 +00002093 assert None is p12.get_certificate()
2094 assert None is p12.get_privatekey()
2095 assert None is p12.get_ca_certificates()
2096 assert None is p12.get_friendlyname()
Rick Dean623ee362009-07-17 12:22:16 -05002097
2098 def test_type_errors(self):
Rick Dean38a05c82009-07-18 01:41:30 -05002099 """
Alex Chanb00ede22017-01-30 07:24:40 +00002100 The `PKCS12` setter functions (`set_certificate`, `set_privatekey`,
2101 `set_ca_certificates`, and `set_friendlyname`) raise `TypeError`
2102 when passed objects of types other than those expected.
Rick Dean38a05c82009-07-18 01:41:30 -05002103 """
Jean-Paul Calderonea202edb2009-07-25 12:22:12 -04002104 p12 = PKCS12()
Alex Chanb00ede22017-01-30 07:24:40 +00002105 for bad_arg in [3, PKey(), X509]:
2106 with pytest.raises(TypeError):
2107 p12.set_certificate(bad_arg)
Alex Gaynor03737182020-07-23 20:40:46 -04002108 for bad_arg in [3, "legbone", X509()]:
Alex Chanb00ede22017-01-30 07:24:40 +00002109 with pytest.raises(TypeError):
2110 p12.set_privatekey(bad_arg)
2111 for bad_arg in [3, X509(), (3, 4), (PKey(),)]:
2112 with pytest.raises(TypeError):
2113 p12.set_ca_certificates(bad_arg)
Alex Gaynor03737182020-07-23 20:40:46 -04002114 for bad_arg in [6, ("foo", "bar")]:
Alex Chanb00ede22017-01-30 07:24:40 +00002115 with pytest.raises(TypeError):
2116 p12.set_friendlyname(bad_arg)
Rick Dean623ee362009-07-17 12:22:16 -05002117
2118 def test_key_only(self):
2119 """
Alex Chanb00ede22017-01-30 07:24:40 +00002120 A `PKCS12` with only a private key can be exported using
2121 `PKCS12.export` and loaded again using `load_pkcs12`.
Rick Dean623ee362009-07-17 12:22:16 -05002122 """
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05002123 passwd = b"blah"
Rick Dean623ee362009-07-17 12:22:16 -05002124 p12 = PKCS12()
Jean-Paul Calderonea202edb2009-07-25 12:22:12 -04002125 pkey = load_privatekey(FILETYPE_PEM, cleartextPrivateKeyPEM)
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04002126 p12.set_privatekey(pkey)
Alex Chanb00ede22017-01-30 07:24:40 +00002127 assert None is p12.get_certificate()
2128 assert pkey == p12.get_privatekey()
Rick Dean321a0512009-08-13 17:21:29 -05002129 try:
2130 dumped_p12 = p12.export(passphrase=passwd, iter=2, maciter=3)
2131 except Error:
2132 # Some versions of OpenSSL will throw an exception
2133 # for this nearly useless PKCS12 we tried to generate:
2134 # [('PKCS12 routines', 'PKCS12_create', 'invalid null argument')]
2135 return
Rick Dean623ee362009-07-17 12:22:16 -05002136 p12 = load_pkcs12(dumped_p12, passwd)
Alex Chanb00ede22017-01-30 07:24:40 +00002137 assert None is p12.get_ca_certificates()
2138 assert None is p12.get_certificate()
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04002139
2140 # OpenSSL fails to bring the key back to us. So sad. Perhaps in the
2141 # future this will be improved.
Alex Chanb00ede22017-01-30 07:24:40 +00002142 assert isinstance(p12.get_privatekey(), (PKey, type(None)))
Rick Dean623ee362009-07-17 12:22:16 -05002143
2144 def test_cert_only(self):
2145 """
Alex Chanb00ede22017-01-30 07:24:40 +00002146 A `PKCS12` with only a certificate can be exported using
2147 `PKCS12.export` and loaded again using `load_pkcs12`.
Rick Dean623ee362009-07-17 12:22:16 -05002148 """
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05002149 passwd = b"blah"
Rick Dean623ee362009-07-17 12:22:16 -05002150 p12 = PKCS12()
Jean-Paul Calderonea202edb2009-07-25 12:22:12 -04002151 cert = load_certificate(FILETYPE_PEM, cleartextCertificatePEM)
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04002152 p12.set_certificate(cert)
Alex Chanb00ede22017-01-30 07:24:40 +00002153 assert cert == p12.get_certificate()
2154 assert None is p12.get_privatekey()
Rick Dean321a0512009-08-13 17:21:29 -05002155 try:
2156 dumped_p12 = p12.export(passphrase=passwd, iter=2, maciter=3)
2157 except Error:
2158 # Some versions of OpenSSL will throw an exception
2159 # for this nearly useless PKCS12 we tried to generate:
2160 # [('PKCS12 routines', 'PKCS12_create', 'invalid null argument')]
2161 return
Rick Dean623ee362009-07-17 12:22:16 -05002162 p12 = load_pkcs12(dumped_p12, passwd)
Alex Chanb00ede22017-01-30 07:24:40 +00002163 assert None is p12.get_privatekey()
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04002164
2165 # OpenSSL fails to bring the cert back to us. Groany mcgroan.
Alex Chanb00ede22017-01-30 07:24:40 +00002166 assert isinstance(p12.get_certificate(), (X509, type(None)))
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04002167
2168 # Oh ho. It puts the certificate into the ca certificates list, in
2169 # fact. Totally bogus, I would think. Nevertheless, let's exploit
2170 # that to check to see if it reconstructed the certificate we expected
2171 # it to. At some point, hopefully this will change so that
2172 # p12.get_certificate() is actually what returns the loaded
2173 # certificate.
Alex Gaynor03737182020-07-23 20:40:46 -04002174 assert cleartextCertificatePEM == dump_certificate(
2175 FILETYPE_PEM, p12.get_ca_certificates()[0]
2176 )
Rick Dean623ee362009-07-17 12:22:16 -05002177
Alex Gaynor03737182020-07-23 20:40:46 -04002178 def gen_pkcs12(
2179 self, cert_pem=None, key_pem=None, ca_pem=None, friendly_name=None
2180 ):
Rick Dean623ee362009-07-17 12:22:16 -05002181 """
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04002182 Generate a PKCS12 object with components from PEM. Verify that the set
2183 functions return None.
Rick Dean623ee362009-07-17 12:22:16 -05002184 """
Rick Deanf94096c2009-07-18 14:23:06 -05002185 p12 = PKCS12()
2186 if cert_pem:
2187 ret = p12.set_certificate(load_certificate(FILETYPE_PEM, cert_pem))
Alex Chanb00ede22017-01-30 07:24:40 +00002188 assert ret is None
Rick Deanf94096c2009-07-18 14:23:06 -05002189 if key_pem:
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04002190 ret = p12.set_privatekey(load_privatekey(FILETYPE_PEM, key_pem))
Alex Chanb00ede22017-01-30 07:24:40 +00002191 assert ret is None
Rick Deanf94096c2009-07-18 14:23:06 -05002192 if ca_pem:
Alex Gaynor85b49702015-09-05 16:30:59 -04002193 ret = p12.set_ca_certificates(
2194 (load_certificate(FILETYPE_PEM, ca_pem),)
2195 )
Alex Chanb00ede22017-01-30 07:24:40 +00002196 assert ret is None
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04002197 if friendly_name:
2198 ret = p12.set_friendlyname(friendly_name)
Alex Chanb00ede22017-01-30 07:24:40 +00002199 assert ret is None
Rick Deanf94096c2009-07-18 14:23:06 -05002200 return p12
2201
Alex Gaynor03737182020-07-23 20:40:46 -04002202 def check_recovery(
2203 self, p12_str, key=None, cert=None, ca=None, passwd=b"", extra=()
2204 ):
Rick Deanf94096c2009-07-18 14:23:06 -05002205 """
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04002206 Use openssl program to confirm three components are recoverable from a
2207 PKCS12 string.
Rick Deanf94096c2009-07-18 14:23:06 -05002208 """
2209 if key:
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04002210 recovered_key = _runopenssl(
Alex Gaynor03737182020-07-23 20:40:46 -04002211 p12_str,
2212 b"pkcs12",
2213 b"-nocerts",
2214 b"-nodes",
2215 b"-passin",
2216 b"pass:" + passwd,
2217 *extra
2218 )
2219 assert recovered_key[-len(key) :] == key
Rick Deanf94096c2009-07-18 14:23:06 -05002220 if cert:
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04002221 recovered_cert = _runopenssl(
Alex Gaynor03737182020-07-23 20:40:46 -04002222 p12_str,
2223 b"pkcs12",
2224 b"-clcerts",
2225 b"-nodes",
2226 b"-passin",
2227 b"pass:" + passwd,
2228 b"-nokeys",
2229 *extra
2230 )
2231 assert recovered_cert[-len(cert) :] == cert
Rick Deanf94096c2009-07-18 14:23:06 -05002232 if ca:
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04002233 recovered_cert = _runopenssl(
Alex Gaynor03737182020-07-23 20:40:46 -04002234 p12_str,
2235 b"pkcs12",
2236 b"-cacerts",
2237 b"-nodes",
2238 b"-passin",
2239 b"pass:" + passwd,
2240 b"-nokeys",
2241 *extra
2242 )
2243 assert recovered_cert[-len(ca) :] == ca
Rick Deanf94096c2009-07-18 14:23:06 -05002244
Stephen Holsapple38482622014-04-05 20:29:34 -07002245 def verify_pkcs12_container(self, p12):
2246 """
2247 Verify that the PKCS#12 container contains the correct client
2248 certificate and private key.
Jean-Paul Calderonef0ff13b2014-05-05 12:48:33 -04002249
2250 :param p12: The PKCS12 instance to verify.
Alex Chanb00ede22017-01-30 07:24:40 +00002251 :type p12: `PKCS12`
Stephen Holsapple38482622014-04-05 20:29:34 -07002252 """
2253 cert_pem = dump_certificate(FILETYPE_PEM, p12.get_certificate())
2254 key_pem = dump_privatekey(FILETYPE_PEM, p12.get_privatekey())
Alex Gaynor03737182020-07-23 20:40:46 -04002255 assert (client_cert_pem, client_key_pem, None) == (
2256 cert_pem,
2257 key_pem,
2258 p12.get_ca_certificates(),
2259 )
Stephen Holsapple38482622014-04-05 20:29:34 -07002260
Rick Deanf94096c2009-07-18 14:23:06 -05002261 def test_load_pkcs12(self):
2262 """
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04002263 A PKCS12 string generated using the openssl command line can be loaded
Alex Chanb00ede22017-01-30 07:24:40 +00002264 with `load_pkcs12` and its components extracted and examined.
Rick Deanf94096c2009-07-18 14:23:06 -05002265 """
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05002266 passwd = b"whatever"
Rick Dean623ee362009-07-17 12:22:16 -05002267 pem = client_key_pem + client_cert_pem
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04002268 p12_str = _runopenssl(
Alex Gaynor85b49702015-09-05 16:30:59 -04002269 pem,
2270 b"pkcs12",
2271 b"-export",
2272 b"-clcerts",
2273 b"-passout",
Alex Gaynor03737182020-07-23 20:40:46 -04002274 b"pass:" + passwd,
Alex Gaynor85b49702015-09-05 16:30:59 -04002275 )
Stephen Holsapple38482622014-04-05 20:29:34 -07002276 p12 = load_pkcs12(p12_str, passphrase=passwd)
2277 self.verify_pkcs12_container(p12)
2278
Abraham Martinc5484ba2015-03-25 15:33:05 +00002279 def test_load_pkcs12_text_passphrase(self):
2280 """
2281 A PKCS12 string generated using the openssl command line can be loaded
Alex Chanb00ede22017-01-30 07:24:40 +00002282 with `load_pkcs12` and its components extracted and examined.
Abraham Martinc5484ba2015-03-25 15:33:05 +00002283 Using text as passphrase instead of bytes. DeprecationWarning expected.
2284 """
2285 pem = client_key_pem + client_cert_pem
2286 passwd = b"whatever"
Alex Gaynor03737182020-07-23 20:40:46 -04002287 p12_str = _runopenssl(
2288 pem,
2289 b"pkcs12",
2290 b"-export",
2291 b"-clcerts",
2292 b"-passout",
2293 b"pass:" + passwd,
2294 )
Alex Chanb00ede22017-01-30 07:24:40 +00002295 with pytest.warns(DeprecationWarning) as w:
Abraham Martinc5484ba2015-03-25 15:33:05 +00002296 simplefilter("always")
Jean-Paul Calderone13a0e652015-03-29 07:58:51 -04002297 p12 = load_pkcs12(p12_str, passphrase=b"whatever".decode("ascii"))
Alex Gaynor03737182020-07-23 20:40:46 -04002298 msg = "{0} for passphrase is no longer accepted, use bytes".format(
2299 WARNING_TYPE_EXPECTED
2300 )
2301 assert msg == str(w[-1].message)
Jean-Paul Calderone6462b072015-03-29 07:03:11 -04002302
Abraham Martinc5484ba2015-03-25 15:33:05 +00002303 self.verify_pkcs12_container(p12)
2304
Stephen Holsapple38482622014-04-05 20:29:34 -07002305 def test_load_pkcs12_no_passphrase(self):
2306 """
Jean-Paul Calderonef0ff13b2014-05-05 12:48:33 -04002307 A PKCS12 string generated using openssl command line can be loaded with
Alex Chanb00ede22017-01-30 07:24:40 +00002308 `load_pkcs12` without a passphrase and its components extracted
Jean-Paul Calderonef0ff13b2014-05-05 12:48:33 -04002309 and examined.
Stephen Holsapple38482622014-04-05 20:29:34 -07002310 """
2311 pem = client_key_pem + client_cert_pem
2312 p12_str = _runopenssl(
Alex Gaynor03737182020-07-23 20:40:46 -04002313 pem, b"pkcs12", b"-export", b"-clcerts", b"-passout", b"pass:"
2314 )
Stephen Holsapple38482622014-04-05 20:29:34 -07002315 p12 = load_pkcs12(p12_str)
2316 self.verify_pkcs12_container(p12)
2317
Stephen Holsapple38482622014-04-05 20:29:34 -07002318 def _dump_and_load(self, dump_passphrase, load_passphrase):
2319 """
2320 A helper method to dump and load a PKCS12 object.
2321 """
2322 p12 = self.gen_pkcs12(client_cert_pem, client_key_pem)
2323 dumped_p12 = p12.export(passphrase=dump_passphrase, iter=2, maciter=3)
2324 return load_pkcs12(dumped_p12, passphrase=load_passphrase)
2325
Stephen Holsapple38482622014-04-05 20:29:34 -07002326 def test_load_pkcs12_null_passphrase_load_empty(self):
2327 """
2328 A PKCS12 string can be dumped with a null passphrase, loaded with an
Alex Chanb00ede22017-01-30 07:24:40 +00002329 empty passphrase with `load_pkcs12`, and its components
Stephen Holsapple38482622014-04-05 20:29:34 -07002330 extracted and examined.
2331 """
Jean-Paul Calderonef0ff13b2014-05-05 12:48:33 -04002332 self.verify_pkcs12_container(
Alex Gaynor03737182020-07-23 20:40:46 -04002333 self._dump_and_load(dump_passphrase=None, load_passphrase=b"")
2334 )
Stephen Holsapple38482622014-04-05 20:29:34 -07002335
Stephen Holsapple38482622014-04-05 20:29:34 -07002336 def test_load_pkcs12_null_passphrase_load_null(self):
2337 """
2338 A PKCS12 string can be dumped with a null passphrase, loaded with a
Alex Chanb00ede22017-01-30 07:24:40 +00002339 null passphrase with `load_pkcs12`, and its components
Stephen Holsapple38482622014-04-05 20:29:34 -07002340 extracted and examined.
2341 """
Jean-Paul Calderonef0ff13b2014-05-05 12:48:33 -04002342 self.verify_pkcs12_container(
Alex Gaynor03737182020-07-23 20:40:46 -04002343 self._dump_and_load(dump_passphrase=None, load_passphrase=None)
2344 )
Stephen Holsapple38482622014-04-05 20:29:34 -07002345
Stephen Holsapple38482622014-04-05 20:29:34 -07002346 def test_load_pkcs12_empty_passphrase_load_empty(self):
2347 """
2348 A PKCS12 string can be dumped with an empty passphrase, loaded with an
Alex Chanb00ede22017-01-30 07:24:40 +00002349 empty passphrase with `load_pkcs12`, and its components
Stephen Holsapple38482622014-04-05 20:29:34 -07002350 extracted and examined.
2351 """
Jean-Paul Calderonef0ff13b2014-05-05 12:48:33 -04002352 self.verify_pkcs12_container(
Alex Gaynor03737182020-07-23 20:40:46 -04002353 self._dump_and_load(dump_passphrase=b"", load_passphrase=b"")
2354 )
Stephen Holsapple38482622014-04-05 20:29:34 -07002355
Stephen Holsapple38482622014-04-05 20:29:34 -07002356 def test_load_pkcs12_empty_passphrase_load_null(self):
2357 """
2358 A PKCS12 string can be dumped with an empty passphrase, loaded with a
Alex Chanb00ede22017-01-30 07:24:40 +00002359 null passphrase with `load_pkcs12`, and its components
Stephen Holsapple38482622014-04-05 20:29:34 -07002360 extracted and examined.
2361 """
Jean-Paul Calderonef0ff13b2014-05-05 12:48:33 -04002362 self.verify_pkcs12_container(
Alex Gaynor03737182020-07-23 20:40:46 -04002363 self._dump_and_load(dump_passphrase=b"", load_passphrase=None)
2364 )
Rick Deanf94096c2009-07-18 14:23:06 -05002365
Rick Deanee568302009-07-24 09:56:29 -05002366 def test_load_pkcs12_garbage(self):
2367 """
Alex Chanb00ede22017-01-30 07:24:40 +00002368 `load_pkcs12` raises `OpenSSL.crypto.Error` when passed
Alex Gaynor85b49702015-09-05 16:30:59 -04002369 a string which is not a PKCS12 dump.
Rick Deanee568302009-07-24 09:56:29 -05002370 """
Alex Gaynor03737182020-07-23 20:40:46 -04002371 passwd = "whatever"
Alex Chanb00ede22017-01-30 07:24:40 +00002372 with pytest.raises(Error) as err:
Alex Gaynor03737182020-07-23 20:40:46 -04002373 load_pkcs12(b"fruit loops", passwd)
2374 assert err.value.args[0][0][0] == "asn1 encoding routines"
Alex Chanb00ede22017-01-30 07:24:40 +00002375 assert len(err.value.args[0][0]) == 3
Rick Deanee568302009-07-24 09:56:29 -05002376
Rick Deanf94096c2009-07-18 14:23:06 -05002377 def test_replace(self):
2378 """
Alex Chanb00ede22017-01-30 07:24:40 +00002379 `PKCS12.set_certificate` replaces the certificate in a PKCS12
2380 cluster. `PKCS12.set_privatekey` replaces the private key.
2381 `PKCS12.set_ca_certificates` replaces the CA certificates.
Rick Deanf94096c2009-07-18 14:23:06 -05002382 """
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04002383 p12 = self.gen_pkcs12(client_cert_pem, client_key_pem, root_cert_pem)
2384 p12.set_certificate(load_certificate(FILETYPE_PEM, server_cert_pem))
2385 p12.set_privatekey(load_privatekey(FILETYPE_PEM, server_key_pem))
Jean-Paul Calderonea202edb2009-07-25 12:22:12 -04002386 root_cert = load_certificate(FILETYPE_PEM, root_cert_pem)
Rick Deanf94096c2009-07-18 14:23:06 -05002387 client_cert = load_certificate(FILETYPE_PEM, client_cert_pem)
Alex Gaynoraceb3e22015-09-05 12:00:22 -04002388 p12.set_ca_certificates([root_cert]) # not a tuple
Alex Chanb00ede22017-01-30 07:24:40 +00002389 assert 1 == len(p12.get_ca_certificates())
2390 assert root_cert == p12.get_ca_certificates()[0]
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04002391 p12.set_ca_certificates([client_cert, root_cert])
Alex Chanb00ede22017-01-30 07:24:40 +00002392 assert 2 == len(p12.get_ca_certificates())
2393 assert client_cert == p12.get_ca_certificates()[0]
2394 assert root_cert == p12.get_ca_certificates()[1]
Rick Deanf94096c2009-07-18 14:23:06 -05002395
Rick Deanf94096c2009-07-18 14:23:06 -05002396 def test_friendly_name(self):
2397 """
Jean-Paul Calderone64efa2c2011-09-11 10:00:09 -04002398 The *friendlyName* of a PKCS12 can be set and retrieved via
Alex Chanb00ede22017-01-30 07:24:40 +00002399 `PKCS12.get_friendlyname` and `PKCS12_set_friendlyname`, and a
2400 `PKCS12` with a friendly name set can be dumped with `PKCS12.export`.
Rick Deanf94096c2009-07-18 14:23:06 -05002401 """
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05002402 passwd = b'Dogmeat[]{}!@#$%^&*()~`?/.,<>-_+=";:'
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04002403 p12 = self.gen_pkcs12(server_cert_pem, server_key_pem, root_cert_pem)
Alex Gaynor03737182020-07-23 20:40:46 -04002404 for friendly_name in [b"Serverlicious", None, b"###"]:
Rick Dean42d69e12009-07-20 11:36:08 -05002405 p12.set_friendlyname(friendly_name)
Alex Chanb00ede22017-01-30 07:24:40 +00002406 assert p12.get_friendlyname() == friendly_name
Jean-Paul Calderonea202edb2009-07-25 12:22:12 -04002407 dumped_p12 = p12.export(passphrase=passwd, iter=2, maciter=3)
Rick Dean42d69e12009-07-20 11:36:08 -05002408 reloaded_p12 = load_pkcs12(dumped_p12, passwd)
Alex Chanb00ede22017-01-30 07:24:40 +00002409 assert p12.get_friendlyname() == reloaded_p12.get_friendlyname()
Jean-Paul Calderonea202edb2009-07-25 12:22:12 -04002410 # We would use the openssl program to confirm the friendly
2411 # name, but it is not possible. The pkcs12 command
2412 # does not store the friendly name in the cert's
Rick Dean42d69e12009-07-20 11:36:08 -05002413 # alias, which we could then extract.
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04002414 self.check_recovery(
Alex Gaynor03737182020-07-23 20:40:46 -04002415 dumped_p12,
2416 key=server_key_pem,
2417 cert=server_cert_pem,
2418 ca=root_cert_pem,
2419 passwd=passwd,
2420 )
Rick Deanf94096c2009-07-18 14:23:06 -05002421
Rick Deanf94096c2009-07-18 14:23:06 -05002422 def test_various_empty_passphrases(self):
2423 """
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04002424 Test that missing, None, and '' passphrases are identical for PKCS12
2425 export.
Rick Deanf94096c2009-07-18 14:23:06 -05002426 """
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04002427 p12 = self.gen_pkcs12(client_cert_pem, client_key_pem, root_cert_pem)
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05002428 passwd = b""
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04002429 dumped_p12_empty = p12.export(iter=2, maciter=0, passphrase=passwd)
2430 dumped_p12_none = p12.export(iter=3, maciter=2, passphrase=None)
2431 dumped_p12_nopw = p12.export(iter=9, maciter=4)
2432 for dumped_p12 in [dumped_p12_empty, dumped_p12_none, dumped_p12_nopw]:
2433 self.check_recovery(
Alex Gaynor03737182020-07-23 20:40:46 -04002434 dumped_p12,
2435 key=client_key_pem,
2436 cert=client_cert_pem,
2437 ca=root_cert_pem,
2438 passwd=passwd,
2439 )
Rick Deanf94096c2009-07-18 14:23:06 -05002440
Rick Deanf94096c2009-07-18 14:23:06 -05002441 def test_removing_ca_cert(self):
2442 """
Alex Chanb00ede22017-01-30 07:24:40 +00002443 Passing `None` to `PKCS12.set_ca_certificates` removes all CA
2444 certificates.
Rick Deanf94096c2009-07-18 14:23:06 -05002445 """
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04002446 p12 = self.gen_pkcs12(server_cert_pem, server_key_pem, root_cert_pem)
2447 p12.set_ca_certificates(None)
Alex Chanb00ede22017-01-30 07:24:40 +00002448 assert None is p12.get_ca_certificates()
Rick Deanf94096c2009-07-18 14:23:06 -05002449
Rick Deanf94096c2009-07-18 14:23:06 -05002450 def test_export_without_mac(self):
2451 """
Alex Chanb00ede22017-01-30 07:24:40 +00002452 Exporting a PKCS12 with a `maciter` of `-1` excludes the MAC entirely.
Rick Deanf94096c2009-07-18 14:23:06 -05002453 """
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05002454 passwd = b"Lake Michigan"
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04002455 p12 = self.gen_pkcs12(server_cert_pem, server_key_pem, root_cert_pem)
Rick Dean623ee362009-07-17 12:22:16 -05002456 dumped_p12 = p12.export(maciter=-1, passphrase=passwd, iter=2)
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04002457 self.check_recovery(
Alex Gaynor03737182020-07-23 20:40:46 -04002458 dumped_p12,
2459 key=server_key_pem,
2460 cert=server_cert_pem,
2461 passwd=passwd,
2462 extra=(b"-nomacver",),
2463 )
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04002464
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04002465 def test_load_without_mac(self):
2466 """
2467 Loading a PKCS12 without a MAC does something other than crash.
2468 """
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05002469 passwd = b"Lake Michigan"
Jean-Paul Calderone7426ed82009-07-25 21:19:23 -04002470 p12 = self.gen_pkcs12(server_cert_pem, server_key_pem, root_cert_pem)
2471 dumped_p12 = p12.export(maciter=-1, passphrase=passwd, iter=2)
Rick Dean321a0512009-08-13 17:21:29 -05002472 try:
2473 recovered_p12 = load_pkcs12(dumped_p12, passwd)
2474 # The person who generated this PCKS12 should be flogged,
2475 # or better yet we should have a means to determine
2476 # whether a PCKS12 had a MAC that was verified.
2477 # Anyway, libopenssl chooses to allow it, so the
2478 # pyopenssl binding does as well.
Alex Chanb00ede22017-01-30 07:24:40 +00002479 assert isinstance(recovered_p12, PKCS12)
Rick Dean321a0512009-08-13 17:21:29 -05002480 except Error:
2481 # Failing here with an exception is preferred as some openssl
2482 # versions do.
2483 pass
Rick Dean623ee362009-07-17 12:22:16 -05002484
Rick Dean25bcc1f2009-07-20 11:53:13 -05002485 def test_zero_len_list_for_ca(self):
2486 """
Jean-Paul Calderoneda1ffa72009-07-25 21:24:34 -04002487 A PKCS12 with an empty CA certificates list can be exported.
Rick Dean25bcc1f2009-07-20 11:53:13 -05002488 """
Alex Gaynor03737182020-07-23 20:40:46 -04002489 passwd = b"Hobie 18"
Jean-Paul Calderoneda1ffa72009-07-25 21:24:34 -04002490 p12 = self.gen_pkcs12(server_cert_pem, server_key_pem)
Alex Gaynor85b49702015-09-05 16:30:59 -04002491 p12.set_ca_certificates([])
Alex Chanb00ede22017-01-30 07:24:40 +00002492 assert () == p12.get_ca_certificates()
Alex Gaynor85b49702015-09-05 16:30:59 -04002493 dumped_p12 = p12.export(passphrase=passwd, iter=3)
2494 self.check_recovery(
Alex Gaynor03737182020-07-23 20:40:46 -04002495 dumped_p12, key=server_key_pem, cert=server_cert_pem, passwd=passwd
2496 )
Rick Dean25bcc1f2009-07-20 11:53:13 -05002497
Rick Deanf94096c2009-07-18 14:23:06 -05002498 def test_export_without_args(self):
Jean-Paul Calderone38a646d2008-03-25 15:16:18 -04002499 """
Alex Chanb00ede22017-01-30 07:24:40 +00002500 All the arguments to `PKCS12.export` are optional.
Jean-Paul Calderone38a646d2008-03-25 15:16:18 -04002501 """
Jean-Paul Calderoneda1ffa72009-07-25 21:24:34 -04002502 p12 = self.gen_pkcs12(server_cert_pem, server_key_pem, root_cert_pem)
Rick Deanf94096c2009-07-18 14:23:06 -05002503 dumped_p12 = p12.export() # no args
Jean-Paul Calderoneda1ffa72009-07-25 21:24:34 -04002504 self.check_recovery(
Alex Gaynor03737182020-07-23 20:40:46 -04002505 dumped_p12, key=server_key_pem, cert=server_cert_pem, passwd=b""
2506 )
Rick Deanf94096c2009-07-18 14:23:06 -05002507
Abraham Martinc5484ba2015-03-25 15:33:05 +00002508 def test_export_without_bytes(self):
2509 """
Alex Chanb00ede22017-01-30 07:24:40 +00002510 Test `PKCS12.export` with text not bytes as passphrase
Abraham Martinc5484ba2015-03-25 15:33:05 +00002511 """
2512 p12 = self.gen_pkcs12(server_cert_pem, server_key_pem, root_cert_pem)
2513
Alex Chanb00ede22017-01-30 07:24:40 +00002514 with pytest.warns(DeprecationWarning) as w:
Abraham Martinc5484ba2015-03-25 15:33:05 +00002515 simplefilter("always")
Jean-Paul Calderone13a0e652015-03-29 07:58:51 -04002516 dumped_p12 = p12.export(passphrase=b"randomtext".decode("ascii"))
Alex Gaynor03737182020-07-23 20:40:46 -04002517 msg = "{0} for passphrase is no longer accepted, use bytes".format(
2518 WARNING_TYPE_EXPECTED
2519 )
2520 assert msg == str(w[-1].message)
Abraham Martinc5484ba2015-03-25 15:33:05 +00002521 self.check_recovery(
Alex Gaynor791212d2015-09-05 15:46:08 -04002522 dumped_p12,
2523 key=server_key_pem,
2524 cert=server_cert_pem,
Alex Gaynor03737182020-07-23 20:40:46 -04002525 passwd=b"randomtext",
Alex Gaynor791212d2015-09-05 15:46:08 -04002526 )
Abraham Martinc5484ba2015-03-25 15:33:05 +00002527
Rick Deanf94096c2009-07-18 14:23:06 -05002528 def test_key_cert_mismatch(self):
2529 """
Alex Chanb00ede22017-01-30 07:24:40 +00002530 `PKCS12.export` raises an exception when a key and certificate
Jean-Paul Calderoneda1ffa72009-07-25 21:24:34 -04002531 mismatch.
Rick Deanf94096c2009-07-18 14:23:06 -05002532 """
Jean-Paul Calderoneda1ffa72009-07-25 21:24:34 -04002533 p12 = self.gen_pkcs12(server_cert_pem, client_key_pem, root_cert_pem)
Alex Chanb00ede22017-01-30 07:24:40 +00002534 with pytest.raises(Error):
2535 p12.export()
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002536
2537
Rick Dean4c9ad612009-07-17 15:05:22 -05002538def _runopenssl(pem, *args):
2539 """
2540 Run the command line openssl tool with the given arguments and write
Rick Dean55d1ce62009-08-13 17:40:24 -05002541 the given PEM to its stdin. Not safe for quotes.
Rick Dean4c9ad612009-07-17 15:05:22 -05002542 """
Alex Gaynor6cbc69a2017-07-25 09:07:04 -04002543 proc = Popen([b"openssl"] + list(args), stdin=PIPE, stdout=PIPE)
Jean-Paul Calderone62ca8da2010-08-11 19:58:08 -04002544 proc.stdin.write(pem)
2545 proc.stdin.close()
Jean-Paul Calderonee82e3252013-03-03 10:14:10 -08002546 output = proc.stdout.read()
2547 proc.stdout.close()
2548 proc.wait()
2549 return output
Rick Dean4c9ad612009-07-17 15:05:22 -05002550
2551
Hynek Schlawack40d448f2016-06-03 16:15:14 -07002552class TestLoadPublicKey(object):
Paul Kehrer32fc4e62016-06-03 15:21:44 -07002553 """
Hynek Schlawack40d448f2016-06-03 16:15:14 -07002554 Tests for :func:`load_publickey`.
Paul Kehrer32fc4e62016-06-03 15:21:44 -07002555 """
Alex Gaynor03737182020-07-23 20:40:46 -04002556
Hynek Schlawack40d448f2016-06-03 16:15:14 -07002557 def test_loading_works(self):
Paul Kehrer32fc4e62016-06-03 15:21:44 -07002558 """
Hynek Schlawack40d448f2016-06-03 16:15:14 -07002559 load_publickey loads public keys and sets correct attributes.
Paul Kehrer32fc4e62016-06-03 15:21:44 -07002560 """
2561 key = load_publickey(FILETYPE_PEM, cleartextPublicKeyPEM)
Hynek Schlawack40d448f2016-06-03 16:15:14 -07002562
2563 assert True is key._only_public
2564 assert 2048 == key.bits()
2565 assert TYPE_RSA == key.type()
2566
2567 def test_invalid_type(self):
2568 """
2569 load_publickey doesn't support FILETYPE_TEXT.
2570 """
2571 with pytest.raises(ValueError):
2572 load_publickey(FILETYPE_TEXT, cleartextPublicKeyPEM)
2573
2574 def test_invalid_key_format(self):
2575 """
2576 load_publickey explodes on incorrect keys.
2577 """
2578 with pytest.raises(Error):
2579 load_publickey(FILETYPE_ASN1, cleartextPublicKeyPEM)
2580
2581 def test_tolerates_unicode_strings(self):
2582 """
2583 load_publickey works with text strings, not just bytes.
2584 """
Alex Gaynor03737182020-07-23 20:40:46 -04002585 serialized = cleartextPublicKeyPEM.decode("ascii")
Hynek Schlawack40d448f2016-06-03 16:15:14 -07002586 key = load_publickey(FILETYPE_PEM, serialized)
2587 dumped_pem = dump_publickey(FILETYPE_PEM, key)
2588
2589 assert dumped_pem == cleartextPublicKeyPEM
Paul Kehrer32fc4e62016-06-03 15:21:44 -07002590
2591
Alex Chanb00ede22017-01-30 07:24:40 +00002592class TestFunction(object):
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002593 """
Alex Chanb00ede22017-01-30 07:24:40 +00002594 Tests for free-functions in the `OpenSSL.crypto` module.
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002595 """
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -04002596
2597 def test_load_privatekey_invalid_format(self):
2598 """
Alex Chanb00ede22017-01-30 07:24:40 +00002599 `load_privatekey` raises `ValueError` if passed an unknown filetype.
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -04002600 """
Alex Chanb00ede22017-01-30 07:24:40 +00002601 with pytest.raises(ValueError):
2602 load_privatekey(100, root_key_pem)
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -04002603
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -04002604 def test_load_privatekey_invalid_passphrase_type(self):
2605 """
Alex Chanb00ede22017-01-30 07:24:40 +00002606 `load_privatekey` raises `TypeError` if passed a passphrase that is
2607 neither a `str` nor a callable.
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -04002608 """
Alex Chanb00ede22017-01-30 07:24:40 +00002609 with pytest.raises(TypeError):
2610 load_privatekey(
Alex Gaynor03737182020-07-23 20:40:46 -04002611 FILETYPE_PEM, encryptedPrivateKeyPEMPassphrase, object()
2612 )
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -04002613
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002614 def test_load_privatekey_wrongPassphrase(self):
2615 """
Alex Chanb00ede22017-01-30 07:24:40 +00002616 `load_privatekey` raises `OpenSSL.crypto.Error` when it is passed an
2617 encrypted PEM and an incorrect passphrase.
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002618 """
Alex Chanb00ede22017-01-30 07:24:40 +00002619 with pytest.raises(Error) as err:
2620 load_privatekey(FILETYPE_PEM, encryptedPrivateKeyPEM, b"quack")
2621 assert err.value.args[0] != []
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002622
Ziga Seilnacht376cf972009-12-22 16:04:10 +01002623 def test_load_privatekey_passphraseWrongType(self):
2624 """
Alex Chanb00ede22017-01-30 07:24:40 +00002625 `load_privatekey` raises `ValueError` when it is passeda passphrase
2626 with a private key encoded in a format, that doesn't support
2627 encryption.
Ziga Seilnacht376cf972009-12-22 16:04:10 +01002628 """
2629 key = load_privatekey(FILETYPE_PEM, cleartextPrivateKeyPEM)
2630 blob = dump_privatekey(FILETYPE_ASN1, key)
Alex Chanb00ede22017-01-30 07:24:40 +00002631 with pytest.raises(ValueError):
2632 load_privatekey(FILETYPE_ASN1, blob, "secret")
Ziga Seilnacht376cf972009-12-22 16:04:10 +01002633
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002634 def test_load_privatekey_passphrase(self):
2635 """
Alex Chanb00ede22017-01-30 07:24:40 +00002636 `load_privatekey` can create a `PKey` object from an encrypted PEM
2637 string if given the passphrase.
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002638 """
2639 key = load_privatekey(
Alex Gaynor03737182020-07-23 20:40:46 -04002640 FILETYPE_PEM,
2641 encryptedPrivateKeyPEM,
2642 encryptedPrivateKeyPEMPassphrase,
2643 )
Alex Gaynor01f90a12019-02-07 09:14:48 -05002644 assert isinstance(key, PKey)
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002645
Ziga Seilnacht6b90a402009-12-22 14:33:47 +01002646 def test_load_privatekey_passphrase_exception(self):
2647 """
Alex Gaynor791212d2015-09-05 15:46:08 -04002648 If the passphrase callback raises an exception, that exception is
Alex Chanb00ede22017-01-30 07:24:40 +00002649 raised by `load_privatekey`.
Ziga Seilnacht6b90a402009-12-22 14:33:47 +01002650 """
Alex Gaynor03737182020-07-23 20:40:46 -04002651
Ziga Seilnacht6b90a402009-12-22 14:33:47 +01002652 def cb(ignored):
2653 raise ArithmeticError
2654
Alex Gaynor791212d2015-09-05 15:46:08 -04002655 with pytest.raises(ArithmeticError):
2656 load_privatekey(FILETYPE_PEM, encryptedPrivateKeyPEM, cb)
Ziga Seilnacht6b90a402009-12-22 14:33:47 +01002657
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002658 def test_load_privatekey_wrongPassphraseCallback(self):
2659 """
Alex Chanb00ede22017-01-30 07:24:40 +00002660 `load_privatekey` raises `OpenSSL.crypto.Error` when it
Jean-Paul Calderoned440a082011-09-14 11:02:05 -04002661 is passed an encrypted PEM and a passphrase callback which returns an
2662 incorrect passphrase.
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002663 """
2664 called = []
Alex Gaynoraceb3e22015-09-05 12:00:22 -04002665
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002666 def cb(*a):
2667 called.append(None)
Alex Gaynore7f51982016-09-11 11:48:14 -04002668 return b"quack"
Alex Gaynor03737182020-07-23 20:40:46 -04002669
Alex Chanb00ede22017-01-30 07:24:40 +00002670 with pytest.raises(Error) as err:
2671 load_privatekey(FILETYPE_PEM, encryptedPrivateKeyPEM, cb)
2672 assert called
2673 assert err.value.args[0] != []
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002674
2675 def test_load_privatekey_passphraseCallback(self):
2676 """
Alex Chanb00ede22017-01-30 07:24:40 +00002677 `load_privatekey` can create a `PKey` object from an encrypted PEM
2678 string if given a passphrase callback which returns the correct
2679 password.
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002680 """
2681 called = []
Alex Gaynoraceb3e22015-09-05 12:00:22 -04002682
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002683 def cb(writing):
2684 called.append(writing)
2685 return encryptedPrivateKeyPEMPassphrase
Alex Gaynor03737182020-07-23 20:40:46 -04002686
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002687 key = load_privatekey(FILETYPE_PEM, encryptedPrivateKeyPEM, cb)
Alex Gaynor01f90a12019-02-07 09:14:48 -05002688 assert isinstance(key, PKey)
Alex Chanb00ede22017-01-30 07:24:40 +00002689 assert called == [False]
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002690
Jean-Paul Calderone105cb952011-09-14 10:16:46 -04002691 def test_load_privatekey_passphrase_wrong_return_type(self):
2692 """
Alex Chanb00ede22017-01-30 07:24:40 +00002693 `load_privatekey` raises `ValueError` if the passphrase callback
2694 returns something other than a byte string.
Jean-Paul Calderone105cb952011-09-14 10:16:46 -04002695 """
Alex Chanb00ede22017-01-30 07:24:40 +00002696 with pytest.raises(ValueError):
2697 load_privatekey(
Alex Gaynor03737182020-07-23 20:40:46 -04002698 FILETYPE_PEM, encryptedPrivateKeyPEM, lambda *args: 3
2699 )
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -04002700
Alex Chanfb078d82017-04-20 11:16:15 +01002701 def test_dump_privatekey_wrong_args(self):
2702 """
2703 `dump_privatekey` raises `TypeError` if called with a `cipher`
2704 argument but no `passphrase` argument.
2705 """
2706 key = PKey()
2707 key.generate_key(TYPE_RSA, 512)
2708 with pytest.raises(TypeError):
2709 dump_privatekey(FILETYPE_PEM, key, cipher=GOOD_CIPHER)
2710
Paul Kehrercded9932017-06-29 18:43:42 -05002711 def test_dump_privatekey_not_rsa_key(self):
2712 """
2713 `dump_privatekey` raises `TypeError` if called with a key that is
2714 not RSA.
2715 """
2716 key = PKey()
2717 key.generate_key(TYPE_DSA, 512)
2718 with pytest.raises(TypeError):
2719 dump_privatekey(FILETYPE_TEXT, key)
2720
2721 def test_dump_privatekey_invalid_pkey(self):
2722 with pytest.raises(TypeError):
2723 dump_privatekey(FILETYPE_TEXT, object())
2724
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -04002725 def test_dump_privatekey_unknown_cipher(self):
2726 """
Alex Chanb00ede22017-01-30 07:24:40 +00002727 `dump_privatekey` raises `ValueError` if called with an unrecognized
2728 cipher name.
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -04002729 """
2730 key = PKey()
2731 key.generate_key(TYPE_RSA, 512)
Alex Chanb00ede22017-01-30 07:24:40 +00002732 with pytest.raises(ValueError):
2733 dump_privatekey(FILETYPE_PEM, key, BAD_CIPHER, "passphrase")
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -04002734
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -04002735 def test_dump_privatekey_invalid_passphrase_type(self):
2736 """
Alex Chanb00ede22017-01-30 07:24:40 +00002737 `dump_privatekey` raises `TypeError` if called with a passphrase which
2738 is neither a `str` nor a callable.
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -04002739 """
2740 key = PKey()
2741 key.generate_key(TYPE_RSA, 512)
Alex Chanb00ede22017-01-30 07:24:40 +00002742 with pytest.raises(TypeError):
2743 dump_privatekey(FILETYPE_PEM, key, GOOD_CIPHER, object())
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -04002744
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -04002745 def test_dump_privatekey_invalid_filetype(self):
2746 """
Alex Chanb00ede22017-01-30 07:24:40 +00002747 `dump_privatekey` raises `ValueError` if called with an unrecognized
2748 filetype.
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -04002749 """
2750 key = PKey()
2751 key.generate_key(TYPE_RSA, 512)
Alex Chanb00ede22017-01-30 07:24:40 +00002752 with pytest.raises(ValueError):
2753 dump_privatekey(100, key)
Jean-Paul Calderonefe1b9bd2010-08-03 18:00:21 -04002754
Alex Chanb00ede22017-01-30 07:24:40 +00002755 def test_load_privatekey_passphrase_callback_length(self):
Ziga Seilnacht781295a2009-12-22 14:58:01 +01002756 """
Alex Chanb00ede22017-01-30 07:24:40 +00002757 `crypto.load_privatekey` should raise an error when the passphrase
2758 provided by the callback is too long, not silently truncate it.
Ziga Seilnacht781295a2009-12-22 14:58:01 +01002759 """
Alex Gaynor03737182020-07-23 20:40:46 -04002760
Ziga Seilnacht781295a2009-12-22 14:58:01 +01002761 def cb(ignored):
2762 return "a" * 1025
2763
Alex Gaynor791212d2015-09-05 15:46:08 -04002764 with pytest.raises(ValueError):
2765 load_privatekey(FILETYPE_PEM, encryptedPrivateKeyPEM, cb)
Ziga Seilnacht781295a2009-12-22 14:58:01 +01002766
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002767 def test_dump_privatekey_passphrase(self):
2768 """
Alex Chanb00ede22017-01-30 07:24:40 +00002769 `dump_privatekey` writes an encrypted PEM when given a passphrase.
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002770 """
Alex Gaynore7f51982016-09-11 11:48:14 -04002771 passphrase = b"foo"
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002772 key = load_privatekey(FILETYPE_PEM, cleartextPrivateKeyPEM)
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05002773 pem = dump_privatekey(FILETYPE_PEM, key, GOOD_CIPHER, passphrase)
Alex Gaynor12576002019-11-18 00:18:50 -05002774 assert isinstance(pem, bytes)
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002775 loadedKey = load_privatekey(FILETYPE_PEM, pem, passphrase)
Alex Gaynor01f90a12019-02-07 09:14:48 -05002776 assert isinstance(loadedKey, PKey)
Alex Chanb00ede22017-01-30 07:24:40 +00002777 assert loadedKey.type() == key.type()
2778 assert loadedKey.bits() == key.bits()
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002779
Alex Chanb00ede22017-01-30 07:24:40 +00002780 def test_dump_privatekey_passphrase_wrong_type(self):
Ziga Seilnacht376cf972009-12-22 16:04:10 +01002781 """
Alex Chanb00ede22017-01-30 07:24:40 +00002782 `dump_privatekey` raises `ValueError` when it is passed a passphrase
2783 with a private key encoded in a format, that doesn't support
2784 encryption.
Ziga Seilnacht376cf972009-12-22 16:04:10 +01002785 """
2786 key = load_privatekey(FILETYPE_PEM, cleartextPrivateKeyPEM)
Alex Gaynor791212d2015-09-05 15:46:08 -04002787 with pytest.raises(ValueError):
2788 dump_privatekey(FILETYPE_ASN1, key, GOOD_CIPHER, "secret")
Ziga Seilnacht376cf972009-12-22 16:04:10 +01002789
Rick Dean5b7b6372009-04-01 11:34:06 -05002790 def test_dump_certificate(self):
2791 """
Alex Chanb00ede22017-01-30 07:24:40 +00002792 `dump_certificate` writes PEM, DER, and text.
Rick Dean5b7b6372009-04-01 11:34:06 -05002793 """
Jean-Paul Calderonef17e4212009-04-01 14:21:40 -04002794 pemData = cleartextCertificatePEM + cleartextPrivateKeyPEM
Rick Dean5b7b6372009-04-01 11:34:06 -05002795 cert = load_certificate(FILETYPE_PEM, pemData)
2796 dumped_pem = dump_certificate(FILETYPE_PEM, cert)
Alex Chanb00ede22017-01-30 07:24:40 +00002797 assert dumped_pem == cleartextCertificatePEM
Rick Dean5b7b6372009-04-01 11:34:06 -05002798 dumped_der = dump_certificate(FILETYPE_ASN1, cert)
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05002799 good_der = _runopenssl(dumped_pem, b"x509", b"-outform", b"DER")
Alex Chanb00ede22017-01-30 07:24:40 +00002800 assert dumped_der == good_der
Rick Dean5b7b6372009-04-01 11:34:06 -05002801 cert2 = load_certificate(FILETYPE_ASN1, dumped_der)
2802 dumped_pem2 = dump_certificate(FILETYPE_PEM, cert2)
Alex Chanb00ede22017-01-30 07:24:40 +00002803 assert dumped_pem2 == cleartextCertificatePEM
Rick Dean5b7b6372009-04-01 11:34:06 -05002804 dumped_text = dump_certificate(FILETYPE_TEXT, cert)
Alex Gaynor316aa2c2016-09-10 14:42:53 -04002805 good_text = _runopenssl(
Alex Gaynor03737182020-07-23 20:40:46 -04002806 dumped_pem, b"x509", b"-noout", b"-text", b"-nameopt", b""
2807 )
Alex Chanb00ede22017-01-30 07:24:40 +00002808 assert dumped_text == good_text
Rick Dean5b7b6372009-04-01 11:34:06 -05002809
Alex Gaynor37726112016-07-04 09:51:32 -04002810 def test_dump_certificate_bad_type(self):
2811 """
Alex Chanb00ede22017-01-30 07:24:40 +00002812 `dump_certificate` raises a `ValueError` if it's called with
Alex Gaynor37726112016-07-04 09:51:32 -04002813 a bad type.
2814 """
2815 cert = load_certificate(FILETYPE_PEM, cleartextCertificatePEM)
2816 with pytest.raises(ValueError):
2817 dump_certificate(object(), cert)
2818
Jean-Paul Calderonee82e3252013-03-03 10:14:10 -08002819 def test_dump_privatekey_pem(self):
Rick Dean5b7b6372009-04-01 11:34:06 -05002820 """
Alex Chanb00ede22017-01-30 07:24:40 +00002821 `dump_privatekey` writes a PEM
Rick Dean5b7b6372009-04-01 11:34:06 -05002822 """
2823 key = load_privatekey(FILETYPE_PEM, cleartextPrivateKeyPEM)
Alex Chanb00ede22017-01-30 07:24:40 +00002824 assert key.check()
Rick Dean5b7b6372009-04-01 11:34:06 -05002825 dumped_pem = dump_privatekey(FILETYPE_PEM, key)
Alex Chanb00ede22017-01-30 07:24:40 +00002826 assert dumped_pem == cleartextPrivateKeyPEM
Jean-Paul Calderonee82e3252013-03-03 10:14:10 -08002827
Jean-Paul Calderonee82e3252013-03-03 10:14:10 -08002828 def test_dump_privatekey_asn1(self):
2829 """
Alex Chanb00ede22017-01-30 07:24:40 +00002830 `dump_privatekey` writes a DER
Jean-Paul Calderonee82e3252013-03-03 10:14:10 -08002831 """
2832 key = load_privatekey(FILETYPE_PEM, cleartextPrivateKeyPEM)
2833 dumped_pem = dump_privatekey(FILETYPE_PEM, key)
2834
Rick Dean5b7b6372009-04-01 11:34:06 -05002835 dumped_der = dump_privatekey(FILETYPE_ASN1, key)
Jean-Paul Calderonef17e4212009-04-01 14:21:40 -04002836 # XXX This OpenSSL call writes "writing RSA key" to standard out. Sad.
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05002837 good_der = _runopenssl(dumped_pem, b"rsa", b"-outform", b"DER")
Alex Chanb00ede22017-01-30 07:24:40 +00002838 assert dumped_der == good_der
Rick Dean5b7b6372009-04-01 11:34:06 -05002839 key2 = load_privatekey(FILETYPE_ASN1, dumped_der)
2840 dumped_pem2 = dump_privatekey(FILETYPE_PEM, key2)
Alex Chanb00ede22017-01-30 07:24:40 +00002841 assert dumped_pem2 == cleartextPrivateKeyPEM
Jean-Paul Calderonee82e3252013-03-03 10:14:10 -08002842
Jean-Paul Calderonee82e3252013-03-03 10:14:10 -08002843 def test_dump_privatekey_text(self):
2844 """
Alex Chanb00ede22017-01-30 07:24:40 +00002845 `dump_privatekey` writes a text
Jean-Paul Calderonee82e3252013-03-03 10:14:10 -08002846 """
2847 key = load_privatekey(FILETYPE_PEM, cleartextPrivateKeyPEM)
2848 dumped_pem = dump_privatekey(FILETYPE_PEM, key)
2849
Rick Dean5b7b6372009-04-01 11:34:06 -05002850 dumped_text = dump_privatekey(FILETYPE_TEXT, key)
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05002851 good_text = _runopenssl(dumped_pem, b"rsa", b"-noout", b"-text")
Alex Chanb00ede22017-01-30 07:24:40 +00002852 assert dumped_text == good_text
Rick Dean5b7b6372009-04-01 11:34:06 -05002853
Cory Benfield6492f7c2015-10-27 16:57:58 +09002854 def test_dump_publickey_pem(self):
2855 """
Cory Benfield11c10192015-10-27 17:23:03 +09002856 dump_publickey writes a PEM.
Cory Benfield6492f7c2015-10-27 16:57:58 +09002857 """
2858 key = load_publickey(FILETYPE_PEM, cleartextPublicKeyPEM)
2859 dumped_pem = dump_publickey(FILETYPE_PEM, key)
Cory Benfieldd86f1d82015-10-27 17:25:17 +09002860 assert dumped_pem == cleartextPublicKeyPEM
Cory Benfield6492f7c2015-10-27 16:57:58 +09002861
2862 def test_dump_publickey_asn1(self):
2863 """
Cory Benfield11c10192015-10-27 17:23:03 +09002864 dump_publickey writes a DER.
Cory Benfield6492f7c2015-10-27 16:57:58 +09002865 """
2866 key = load_publickey(FILETYPE_PEM, cleartextPublicKeyPEM)
2867 dumped_der = dump_publickey(FILETYPE_ASN1, key)
2868 key2 = load_publickey(FILETYPE_ASN1, dumped_der)
2869 dumped_pem2 = dump_publickey(FILETYPE_PEM, key2)
Cory Benfieldd86f1d82015-10-27 17:25:17 +09002870 assert dumped_pem2 == cleartextPublicKeyPEM
Cory Benfield6492f7c2015-10-27 16:57:58 +09002871
Cory Benfielde02c7d82015-10-27 17:34:49 +09002872 def test_dump_publickey_invalid_type(self):
2873 """
2874 dump_publickey doesn't support FILETYPE_TEXT.
2875 """
2876 key = load_publickey(FILETYPE_PEM, cleartextPublicKeyPEM)
2877
2878 with pytest.raises(ValueError):
2879 dump_publickey(FILETYPE_TEXT, key)
2880
Rick Dean5b7b6372009-04-01 11:34:06 -05002881 def test_dump_certificate_request(self):
2882 """
Alex Chanb00ede22017-01-30 07:24:40 +00002883 `dump_certificate_request` writes a PEM, DER, and text.
Rick Dean5b7b6372009-04-01 11:34:06 -05002884 """
Alex Gaynor31287502015-09-05 16:11:27 -04002885 req = load_certificate_request(
Alex Gaynor03737182020-07-23 20:40:46 -04002886 FILETYPE_PEM, cleartextCertificateRequestPEM
2887 )
Rick Dean5b7b6372009-04-01 11:34:06 -05002888 dumped_pem = dump_certificate_request(FILETYPE_PEM, req)
Alex Chanb00ede22017-01-30 07:24:40 +00002889 assert dumped_pem == cleartextCertificateRequestPEM
Rick Dean5b7b6372009-04-01 11:34:06 -05002890 dumped_der = dump_certificate_request(FILETYPE_ASN1, req)
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05002891 good_der = _runopenssl(dumped_pem, b"req", b"-outform", b"DER")
Alex Chanb00ede22017-01-30 07:24:40 +00002892 assert dumped_der == good_der
Rick Dean5b7b6372009-04-01 11:34:06 -05002893 req2 = load_certificate_request(FILETYPE_ASN1, dumped_der)
2894 dumped_pem2 = dump_certificate_request(FILETYPE_PEM, req2)
Alex Chanb00ede22017-01-30 07:24:40 +00002895 assert dumped_pem2 == cleartextCertificateRequestPEM
Rick Dean5b7b6372009-04-01 11:34:06 -05002896 dumped_text = dump_certificate_request(FILETYPE_TEXT, req)
Alex Gaynor316aa2c2016-09-10 14:42:53 -04002897 good_text = _runopenssl(
Alex Gaynor03737182020-07-23 20:40:46 -04002898 dumped_pem, b"req", b"-noout", b"-text", b"-nameopt", b""
2899 )
Alex Chanb00ede22017-01-30 07:24:40 +00002900 assert dumped_text == good_text
2901 with pytest.raises(ValueError):
2902 dump_certificate_request(100, req)
Rick Dean5b7b6372009-04-01 11:34:06 -05002903
Alex Chanb00ede22017-01-30 07:24:40 +00002904 def test_dump_privatekey_passphrase_callback(self):
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002905 """
Alex Chanb00ede22017-01-30 07:24:40 +00002906 `dump_privatekey` writes an encrypted PEM when given a callback
Alex Gaynor791212d2015-09-05 15:46:08 -04002907 which returns the correct passphrase.
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002908 """
Alex Gaynore7f51982016-09-11 11:48:14 -04002909 passphrase = b"foo"
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002910 called = []
Alex Gaynoraceb3e22015-09-05 12:00:22 -04002911
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002912 def cb(writing):
2913 called.append(writing)
2914 return passphrase
Alex Gaynor03737182020-07-23 20:40:46 -04002915
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002916 key = load_privatekey(FILETYPE_PEM, cleartextPrivateKeyPEM)
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05002917 pem = dump_privatekey(FILETYPE_PEM, key, GOOD_CIPHER, cb)
Alex Gaynor12576002019-11-18 00:18:50 -05002918 assert isinstance(pem, bytes)
Alex Chanb00ede22017-01-30 07:24:40 +00002919 assert called == [True]
Jean-Paul Calderone828c9cb2008-04-26 18:06:54 -04002920 loadedKey = load_privatekey(FILETYPE_PEM, pem, passphrase)
Alex Gaynor01f90a12019-02-07 09:14:48 -05002921 assert isinstance(loadedKey, PKey)
Alex Chanb00ede22017-01-30 07:24:40 +00002922 assert loadedKey.type() == key.type()
2923 assert loadedKey.bits() == key.bits()
Rick Dean5b7b6372009-04-01 11:34:06 -05002924
Ziga Seilnacht6b90a402009-12-22 14:33:47 +01002925 def test_dump_privatekey_passphrase_exception(self):
2926 """
Alex Chanb00ede22017-01-30 07:24:40 +00002927 `dump_privatekey` should not overwrite the exception raised
Ziga Seilnacht6b90a402009-12-22 14:33:47 +01002928 by the passphrase callback.
2929 """
Alex Gaynor03737182020-07-23 20:40:46 -04002930
Ziga Seilnacht6b90a402009-12-22 14:33:47 +01002931 def cb(ignored):
2932 raise ArithmeticError
2933
2934 key = load_privatekey(FILETYPE_PEM, cleartextPrivateKeyPEM)
Alex Gaynor85b49702015-09-05 16:30:59 -04002935 with pytest.raises(ArithmeticError):
2936 dump_privatekey(FILETYPE_PEM, key, GOOD_CIPHER, cb)
Ziga Seilnacht6b90a402009-12-22 14:33:47 +01002937
Ziga Seilnacht781295a2009-12-22 14:58:01 +01002938 def test_dump_privatekey_passphraseCallbackLength(self):
2939 """
Alex Chanb00ede22017-01-30 07:24:40 +00002940 `crypto.dump_privatekey` should raise an error when the passphrase
2941 provided by the callback is too long, not silently truncate it.
Ziga Seilnacht781295a2009-12-22 14:58:01 +01002942 """
Alex Gaynor03737182020-07-23 20:40:46 -04002943
Ziga Seilnacht781295a2009-12-22 14:58:01 +01002944 def cb(ignored):
2945 return "a" * 1025
2946
2947 key = load_privatekey(FILETYPE_PEM, cleartextPrivateKeyPEM)
Alex Gaynor85b49702015-09-05 16:30:59 -04002948 with pytest.raises(ValueError):
2949 dump_privatekey(FILETYPE_PEM, key, GOOD_CIPHER, cb)
Ziga Seilnacht781295a2009-12-22 14:58:01 +01002950
Alex Gaynor4b9c96a2014-08-14 09:51:48 -07002951 def test_load_pkcs7_data_pem(self):
Jean-Paul Calderonedc138fa2009-06-27 14:32:07 -04002952 """
Alex Chanb00ede22017-01-30 07:24:40 +00002953 `load_pkcs7_data` accepts a PKCS#7 string and returns an instance of
2954 `PKCS`.
Jean-Paul Calderonedc138fa2009-06-27 14:32:07 -04002955 """
2956 pkcs7 = load_pkcs7_data(FILETYPE_PEM, pkcs7Data)
Alex Chanb00ede22017-01-30 07:24:40 +00002957 assert isinstance(pkcs7, PKCS7)
Jean-Paul Calderonedc138fa2009-06-27 14:32:07 -04002958
Alex Gaynor4b9c96a2014-08-14 09:51:48 -07002959 def test_load_pkcs7_data_asn1(self):
Alex Gaynor9875a912014-08-14 13:35:05 -07002960 """
Alex Chanb00ede22017-01-30 07:24:40 +00002961 `load_pkcs7_data` accepts a bytes containing ASN1 data representing
2962 PKCS#7 and returns an instance of `PKCS7`.
Alex Gaynor9875a912014-08-14 13:35:05 -07002963 """
Alex Gaynor4b9c96a2014-08-14 09:51:48 -07002964 pkcs7 = load_pkcs7_data(FILETYPE_ASN1, pkcs7DataASN1)
Alex Chanb00ede22017-01-30 07:24:40 +00002965 assert isinstance(pkcs7, PKCS7)
Alex Gaynor4b9c96a2014-08-14 09:51:48 -07002966
Jean-Paul Calderonee82e3252013-03-03 10:14:10 -08002967 def test_load_pkcs7_data_invalid(self):
2968 """
Alex Chanb00ede22017-01-30 07:24:40 +00002969 If the data passed to `load_pkcs7_data` is invalid, `Error` is raised.
Jean-Paul Calderonee82e3252013-03-03 10:14:10 -08002970 """
Alex Chanb00ede22017-01-30 07:24:40 +00002971 with pytest.raises(Error):
2972 load_pkcs7_data(FILETYPE_PEM, b"foo")
Jean-Paul Calderonee82e3252013-03-03 10:14:10 -08002973
Alex Gaynor09a386e2016-07-03 09:32:44 -04002974 def test_load_pkcs7_type_invalid(self):
2975 """
Alex Chanb00ede22017-01-30 07:24:40 +00002976 If the type passed to `load_pkcs7_data`, `ValueError` is raised.
Alex Gaynor09a386e2016-07-03 09:32:44 -04002977 """
2978 with pytest.raises(ValueError):
2979 load_pkcs7_data(object(), b"foo")
2980
Jean-Paul Calderonee82e3252013-03-03 10:14:10 -08002981
Alex Chan9e2a9932017-01-25 14:29:19 +00002982class TestLoadCertificate(object):
Jean-Paul Calderone4a68b402013-12-29 16:54:58 -05002983 """
Alex Chan9e2a9932017-01-25 14:29:19 +00002984 Tests for `load_certificate_request`.
Jean-Paul Calderone4a68b402013-12-29 16:54:58 -05002985 """
Alex Gaynoraceb3e22015-09-05 12:00:22 -04002986
Alex Chan9e2a9932017-01-25 14:29:19 +00002987 def test_bad_file_type(self):
Jean-Paul Calderone4a68b402013-12-29 16:54:58 -05002988 """
Alex Chan9e2a9932017-01-25 14:29:19 +00002989 If the file type passed to `load_certificate_request` is neither
2990 `FILETYPE_PEM` nor `FILETYPE_ASN1` then `ValueError` is raised.
Jean-Paul Calderone4a68b402013-12-29 16:54:58 -05002991 """
Alex Gaynor7778e792016-07-03 23:38:48 -04002992 with pytest.raises(ValueError):
2993 load_certificate_request(object(), b"")
2994 with pytest.raises(ValueError):
2995 load_certificate(object(), b"")
Jean-Paul Calderone4a68b402013-12-29 16:54:58 -05002996
Alex Gaynor37726112016-07-04 09:51:32 -04002997 def test_bad_certificate(self):
2998 """
Alex Chan9e2a9932017-01-25 14:29:19 +00002999 If the bytes passed to `load_certificate` are not a valid certificate,
3000 an exception is raised.
Alex Gaynor37726112016-07-04 09:51:32 -04003001 """
3002 with pytest.raises(Error):
3003 load_certificate(FILETYPE_ASN1, b"lol")
3004
Jean-Paul Calderone4a68b402013-12-29 16:54:58 -05003005
Alex Chanb00ede22017-01-30 07:24:40 +00003006class TestPKCS7(object):
Jean-Paul Calderone68649052009-07-17 21:14:27 -04003007 """
Alex Chanb00ede22017-01-30 07:24:40 +00003008 Tests for `PKCS7`.
Jean-Paul Calderone68649052009-07-17 21:14:27 -04003009 """
Alex Gaynoraceb3e22015-09-05 12:00:22 -04003010
Jean-Paul Calderone07c93742010-07-30 10:53:41 -04003011 def test_type_is_signed(self):
Jean-Paul Calderoneaa6c0692010-09-08 22:43:09 -04003012 """
Alex Chanb00ede22017-01-30 07:24:40 +00003013 `PKCS7.type_is_signed` returns `True` if the PKCS7 object is of
3014 the type *signed*.
Jean-Paul Calderoneaa6c0692010-09-08 22:43:09 -04003015 """
Jean-Paul Calderone07c93742010-07-30 10:53:41 -04003016 pkcs7 = load_pkcs7_data(FILETYPE_PEM, pkcs7Data)
Alex Chanb00ede22017-01-30 07:24:40 +00003017 assert pkcs7.type_is_signed()
Jean-Paul Calderone07c93742010-07-30 10:53:41 -04003018
Jean-Paul Calderone07c93742010-07-30 10:53:41 -04003019 def test_type_is_enveloped(self):
Jean-Paul Calderoneaa6c0692010-09-08 22:43:09 -04003020 """
Alex Chanb00ede22017-01-30 07:24:40 +00003021 `PKCS7.type_is_enveloped` returns `False` if the PKCS7 object is not
3022 of the type *enveloped*.
Jean-Paul Calderoneaa6c0692010-09-08 22:43:09 -04003023 """
Jean-Paul Calderone07c93742010-07-30 10:53:41 -04003024 pkcs7 = load_pkcs7_data(FILETYPE_PEM, pkcs7Data)
Alex Chanb00ede22017-01-30 07:24:40 +00003025 assert not pkcs7.type_is_enveloped()
Jean-Paul Calderone07c93742010-07-30 10:53:41 -04003026
Alex Chanb00ede22017-01-30 07:24:40 +00003027 def test_type_is_signed_and_enveloped(self):
Jean-Paul Calderoneaa6c0692010-09-08 22:43:09 -04003028 """
Alex Chanb00ede22017-01-30 07:24:40 +00003029 `PKCS7.type_is_signedAndEnveloped` returns `False`
Alex Gaynor791212d2015-09-05 15:46:08 -04003030 if the PKCS7 object is not of the type *signed and enveloped*.
Jean-Paul Calderoneaa6c0692010-09-08 22:43:09 -04003031 """
Jean-Paul Calderone07c93742010-07-30 10:53:41 -04003032 pkcs7 = load_pkcs7_data(FILETYPE_PEM, pkcs7Data)
Alex Chanb00ede22017-01-30 07:24:40 +00003033 assert not pkcs7.type_is_signedAndEnveloped()
Jean-Paul Calderone07c93742010-07-30 10:53:41 -04003034
Jean-Paul Calderoneb4754b92010-07-30 11:00:08 -04003035 def test_type_is_data(self):
Jean-Paul Calderoneaa6c0692010-09-08 22:43:09 -04003036 """
Alex Chanb00ede22017-01-30 07:24:40 +00003037 `PKCS7.type_is_data` returns `False` if the PKCS7 object is not of
3038 the type data.
Jean-Paul Calderoneaa6c0692010-09-08 22:43:09 -04003039 """
Jean-Paul Calderoneb4754b92010-07-30 11:00:08 -04003040 pkcs7 = load_pkcs7_data(FILETYPE_PEM, pkcs7Data)
Alex Chanb00ede22017-01-30 07:24:40 +00003041 assert not pkcs7.type_is_data()
Jean-Paul Calderone97b28ca2010-07-30 10:56:07 -04003042
Jean-Paul Calderone4cbe05e2010-07-30 10:55:30 -04003043 def test_get_type_name(self):
Jean-Paul Calderoneaa6c0692010-09-08 22:43:09 -04003044 """
Alex Chanb00ede22017-01-30 07:24:40 +00003045 `PKCS7.get_type_name` returns a `str` giving the
Alex Gaynor791212d2015-09-05 15:46:08 -04003046 type name.
Jean-Paul Calderoneaa6c0692010-09-08 22:43:09 -04003047 """
Jean-Paul Calderone4cbe05e2010-07-30 10:55:30 -04003048 pkcs7 = load_pkcs7_data(FILETYPE_PEM, pkcs7Data)
Alex Gaynor03737182020-07-23 20:40:46 -04003049 assert pkcs7.get_type_name() == b"pkcs7-signedData"
Jean-Paul Calderone4cbe05e2010-07-30 10:55:30 -04003050
Jean-Paul Calderone4cbe05e2010-07-30 10:55:30 -04003051 def test_attribute(self):
Jean-Paul Calderoneaa6c0692010-09-08 22:43:09 -04003052 """
Alex Gaynor791212d2015-09-05 15:46:08 -04003053 If an attribute other than one of the methods tested here is accessed
Alex Chanb00ede22017-01-30 07:24:40 +00003054 on an instance of `PKCS7`, `AttributeError` is raised.
Jean-Paul Calderoneaa6c0692010-09-08 22:43:09 -04003055 """
Jean-Paul Calderone4cbe05e2010-07-30 10:55:30 -04003056 pkcs7 = load_pkcs7_data(FILETYPE_PEM, pkcs7Data)
Alex Chanb00ede22017-01-30 07:24:40 +00003057 with pytest.raises(AttributeError):
3058 pkcs7.foo
Jean-Paul Calderone4cbe05e2010-07-30 10:55:30 -04003059
3060
Alex Chanb00ede22017-01-30 07:24:40 +00003061class TestNetscapeSPKI(_PKeyInteractionTestsMixin):
Jean-Paul Calderonedc138fa2009-06-27 14:32:07 -04003062 """
Alex Chanb00ede22017-01-30 07:24:40 +00003063 Tests for `OpenSSL.crypto.NetscapeSPKI`.
Jean-Paul Calderonedc138fa2009-06-27 14:32:07 -04003064 """
Alex Gaynoraceb3e22015-09-05 12:00:22 -04003065
Jean-Paul Calderoneb9725592010-08-03 18:17:22 -04003066 def signable(self):
3067 """
Alex Chanb00ede22017-01-30 07:24:40 +00003068 Return a new `NetscapeSPKI` for use with signing tests.
Jean-Paul Calderoneb9725592010-08-03 18:17:22 -04003069 """
3070 return NetscapeSPKI()
3071
Jean-Paul Calderone68649052009-07-17 21:14:27 -04003072 def test_type(self):
3073 """
Alex Gaynor01f90a12019-02-07 09:14:48 -05003074 `NetscapeSPKI` can be used to create instances of that type.
Jean-Paul Calderone68649052009-07-17 21:14:27 -04003075 """
Alex Gaynor03737182020-07-23 20:40:46 -04003076 assert is_consistent_type(NetscapeSPKI, "NetscapeSPKI")
Jean-Paul Calderone68649052009-07-17 21:14:27 -04003077
Jean-Paul Calderonedc138fa2009-06-27 14:32:07 -04003078 def test_construction(self):
3079 """
Alex Gaynor01f90a12019-02-07 09:14:48 -05003080 `NetscapeSPKI` returns an instance of `NetscapeSPKI`.
Jean-Paul Calderonedc138fa2009-06-27 14:32:07 -04003081 """
3082 nspki = NetscapeSPKI()
Alex Gaynor01f90a12019-02-07 09:14:48 -05003083 assert isinstance(nspki, NetscapeSPKI)
Jean-Paul Calderonedc138fa2009-06-27 14:32:07 -04003084
Jean-Paul Calderone969efaa2010-08-03 18:19:19 -04003085 def test_invalid_attribute(self):
3086 """
Alex Chanb00ede22017-01-30 07:24:40 +00003087 Accessing a non-existent attribute of a `NetscapeSPKI` instance
3088 causes an `AttributeError` to be raised.
Jean-Paul Calderone969efaa2010-08-03 18:19:19 -04003089 """
3090 nspki = NetscapeSPKI()
Alex Chanb00ede22017-01-30 07:24:40 +00003091 with pytest.raises(AttributeError):
3092 nspki.foo
Jean-Paul Calderone969efaa2010-08-03 18:19:19 -04003093
Jean-Paul Calderone06ada9f2010-08-03 18:26:52 -04003094 def test_b64_encode(self):
3095 """
Alex Chanb00ede22017-01-30 07:24:40 +00003096 `NetscapeSPKI.b64_encode` encodes the certificate to a base64 blob.
Jean-Paul Calderone06ada9f2010-08-03 18:26:52 -04003097 """
3098 nspki = NetscapeSPKI()
3099 blob = nspki.b64_encode()
Alex Gaynor12576002019-11-18 00:18:50 -05003100 assert isinstance(blob, bytes)
Jean-Paul Calderone06ada9f2010-08-03 18:26:52 -04003101
3102
Paul Kehrer2c605ba2016-03-11 11:17:26 -04003103class TestRevoked(object):
3104 """
Alex Chandeec9342016-12-19 22:00:38 +00003105 Tests for `OpenSSL.crypto.Revoked`.
Paul Kehrer2c605ba2016-03-11 11:17:26 -04003106 """
Alex Gaynor03737182020-07-23 20:40:46 -04003107
Paul Kehrer2c605ba2016-03-11 11:17:26 -04003108 def test_ignores_unsupported_revoked_cert_extension_get_reason(self):
3109 """
3110 The get_reason method on the Revoked class checks to see if the
3111 extension is NID_crl_reason and should skip it otherwise. This test
3112 loads a CRL with extensions it should ignore.
3113 """
3114 crl = load_crl(FILETYPE_PEM, crlDataUnsupportedExtension)
3115 revoked = crl.get_revoked()
3116 reason = revoked[1].get_reason()
Alex Gaynor03737182020-07-23 20:40:46 -04003117 assert reason == b"Unspecified"
Paul Kehrer2c605ba2016-03-11 11:17:26 -04003118
3119 def test_ignores_unsupported_revoked_cert_extension_set_new_reason(self):
3120 crl = load_crl(FILETYPE_PEM, crlDataUnsupportedExtension)
3121 revoked = crl.get_revoked()
3122 revoked[1].set_reason(None)
3123 reason = revoked[1].get_reason()
3124 assert reason is None
3125
Rick Dean536ba022009-07-24 23:57:27 -05003126 def test_construction(self):
3127 """
Alex Chandeec9342016-12-19 22:00:38 +00003128 Confirm we can create `OpenSSL.crypto.Revoked`. Check that it is
3129 empty.
Rick Dean536ba022009-07-24 23:57:27 -05003130 """
3131 revoked = Revoked()
Alex Chandeec9342016-12-19 22:00:38 +00003132 assert isinstance(revoked, Revoked)
3133 assert type(revoked) == Revoked
Alex Gaynor03737182020-07-23 20:40:46 -04003134 assert revoked.get_serial() == b"00"
Alex Chandeec9342016-12-19 22:00:38 +00003135 assert revoked.get_rev_date() is None
3136 assert revoked.get_reason() is None
Jean-Paul Calderone4f74bfe2010-01-30 14:32:49 -05003137
Rick Dean536ba022009-07-24 23:57:27 -05003138 def test_serial(self):
3139 """
Jean-Paul Calderoneb98eae02010-01-30 13:18:04 -05003140 Confirm we can set and get serial numbers from
Alex Chandeec9342016-12-19 22:00:38 +00003141 `OpenSSL.crypto.Revoked`. Confirm errors are handled with grace.
Rick Dean536ba022009-07-24 23:57:27 -05003142 """
3143 revoked = Revoked()
Alex Gaynor03737182020-07-23 20:40:46 -04003144 ret = revoked.set_serial(b"10b")
Alex Chandeec9342016-12-19 22:00:38 +00003145 assert ret is None
Rick Dean536ba022009-07-24 23:57:27 -05003146 ser = revoked.get_serial()
Alex Gaynor03737182020-07-23 20:40:46 -04003147 assert ser == b"010B"
Rick Dean536ba022009-07-24 23:57:27 -05003148
Alex Gaynor03737182020-07-23 20:40:46 -04003149 revoked.set_serial(b"31ppp") # a type error would be nice
Rick Dean536ba022009-07-24 23:57:27 -05003150 ser = revoked.get_serial()
Alex Gaynor03737182020-07-23 20:40:46 -04003151 assert ser == b"31"
Rick Dean536ba022009-07-24 23:57:27 -05003152
Alex Chandeec9342016-12-19 22:00:38 +00003153 with pytest.raises(ValueError):
Alex Gaynor03737182020-07-23 20:40:46 -04003154 revoked.set_serial(b"pqrst")
Alex Chandeec9342016-12-19 22:00:38 +00003155 with pytest.raises(TypeError):
3156 revoked.set_serial(100)
Rick Dean536ba022009-07-24 23:57:27 -05003157
Rick Dean536ba022009-07-24 23:57:27 -05003158 def test_date(self):
3159 """
Jean-Paul Calderoneb98eae02010-01-30 13:18:04 -05003160 Confirm we can set and get revocation dates from
Alex Chandeec9342016-12-19 22:00:38 +00003161 `OpenSSL.crypto.Revoked`. Confirm errors are handled with grace.
Rick Dean536ba022009-07-24 23:57:27 -05003162 """
3163 revoked = Revoked()
3164 date = revoked.get_rev_date()
Alex Chandeec9342016-12-19 22:00:38 +00003165 assert date is None
Rick Dean536ba022009-07-24 23:57:27 -05003166
Alex Gaynore7f51982016-09-11 11:48:14 -04003167 now = datetime.now().strftime("%Y%m%d%H%M%SZ").encode("ascii")
Rick Dean536ba022009-07-24 23:57:27 -05003168 ret = revoked.set_rev_date(now)
Alex Chandeec9342016-12-19 22:00:38 +00003169 assert ret is None
Rick Dean536ba022009-07-24 23:57:27 -05003170 date = revoked.get_rev_date()
Alex Chandeec9342016-12-19 22:00:38 +00003171 assert date == now
Rick Dean536ba022009-07-24 23:57:27 -05003172
Rick Dean6385faf2009-07-26 00:07:47 -05003173 def test_reason(self):
3174 """
Jean-Paul Calderoneb98eae02010-01-30 13:18:04 -05003175 Confirm we can set and get revocation reasons from
Alex Chandeec9342016-12-19 22:00:38 +00003176 `OpenSSL.crypto.Revoked`. The "get" need to work as "set".
3177 Likewise, each reason of all_reasons() must work.
Rick Dean6385faf2009-07-26 00:07:47 -05003178 """
3179 revoked = Revoked()
3180 for r in revoked.all_reasons():
Jean-Paul Calderoneeacad4a2010-08-22 17:12:55 -04003181 for x in range(2):
Rick Dean6385faf2009-07-26 00:07:47 -05003182 ret = revoked.set_reason(r)
Alex Chandeec9342016-12-19 22:00:38 +00003183 assert ret is None
Rick Dean6385faf2009-07-26 00:07:47 -05003184 reason = revoked.get_reason()
Alex Gaynor03737182020-07-23 20:40:46 -04003185 assert reason.lower().replace(b" ", b"") == r.lower().replace(
3186 b" ", b""
3187 )
Alex Gaynoraceb3e22015-09-05 12:00:22 -04003188 r = reason # again with the resp of get
Rick Dean6385faf2009-07-26 00:07:47 -05003189
3190 revoked.set_reason(None)
Alex Chandeec9342016-12-19 22:00:38 +00003191 assert revoked.get_reason() is None
Rick Dean6385faf2009-07-26 00:07:47 -05003192
Alex Gaynor03737182020-07-23 20:40:46 -04003193 @pytest.mark.parametrize("reason", [object(), 1.0, u"foo"])
Alex Chanfb078d82017-04-20 11:16:15 +01003194 def test_set_reason_wrong_args(self, reason):
3195 """
3196 `Revoked.set_reason` raises `TypeError` if called with an argument
3197 which is neither `None` nor a byte string.
3198 """
3199 revoked = Revoked()
3200 with pytest.raises(TypeError):
3201 revoked.set_reason(reason)
3202
Alex Chandeec9342016-12-19 22:00:38 +00003203 def test_set_reason_invalid_reason(self):
Rick Dean6385faf2009-07-26 00:07:47 -05003204 """
Alex Chandeec9342016-12-19 22:00:38 +00003205 Calling `OpenSSL.crypto.Revoked.set_reason` with an argument which
3206 isn't a valid reason results in `ValueError` being raised.
Rick Dean6385faf2009-07-26 00:07:47 -05003207 """
3208 revoked = Revoked()
Alex Chandeec9342016-12-19 22:00:38 +00003209 with pytest.raises(ValueError):
Alex Gaynor03737182020-07-23 20:40:46 -04003210 revoked.set_reason(b"blue")
Jean-Paul Calderone4f74bfe2010-01-30 14:32:49 -05003211
3212
Alex Chan7be83a52017-01-24 15:19:29 +00003213class TestCRL(object):
Rick Dean536ba022009-07-24 23:57:27 -05003214 """
Alex Chan7be83a52017-01-24 15:19:29 +00003215 Tests for `OpenSSL.crypto.CRL`.
Rick Dean536ba022009-07-24 23:57:27 -05003216 """
Alex Gaynor03737182020-07-23 20:40:46 -04003217
Rick Dean536ba022009-07-24 23:57:27 -05003218 cert = load_certificate(FILETYPE_PEM, cleartextCertificatePEM)
3219 pkey = load_privatekey(FILETYPE_PEM, cleartextPrivateKeyPEM)
3220
Dan Sully44e767a2016-06-04 18:05:27 -07003221 root_cert = load_certificate(FILETYPE_PEM, root_cert_pem)
3222 root_key = load_privatekey(FILETYPE_PEM, root_key_pem)
3223 intermediate_cert = load_certificate(FILETYPE_PEM, intermediate_cert_pem)
3224 intermediate_key = load_privatekey(FILETYPE_PEM, intermediate_key_pem)
3225 intermediate_server_cert = load_certificate(
Alex Gaynor03737182020-07-23 20:40:46 -04003226 FILETYPE_PEM, intermediate_server_cert_pem
3227 )
Dan Sully44e767a2016-06-04 18:05:27 -07003228 intermediate_server_key = load_privatekey(
Alex Gaynor03737182020-07-23 20:40:46 -04003229 FILETYPE_PEM, intermediate_server_key_pem
3230 )
Dan Sully44e767a2016-06-04 18:05:27 -07003231
Rick Dean536ba022009-07-24 23:57:27 -05003232 def test_construction(self):
3233 """
Alex Chan7be83a52017-01-24 15:19:29 +00003234 Confirm we can create `OpenSSL.crypto.CRL`. Check
Rick Dean536ba022009-07-24 23:57:27 -05003235 that it is empty
3236 """
3237 crl = CRL()
Alex Chan7be83a52017-01-24 15:19:29 +00003238 assert isinstance(crl, CRL)
3239 assert crl.get_revoked() is None
Jean-Paul Calderone2efd03e2010-01-30 13:59:55 -05003240
Jean-Paul Calderone60432792015-04-13 12:26:07 -04003241 def _get_crl(self):
Rick Dean536ba022009-07-24 23:57:27 -05003242 """
Jean-Paul Calderone60432792015-04-13 12:26:07 -04003243 Get a new ``CRL`` with a revocation.
Rick Dean536ba022009-07-24 23:57:27 -05003244 """
3245 crl = CRL()
3246 revoked = Revoked()
Alex Gaynore7f51982016-09-11 11:48:14 -04003247 now = datetime.now().strftime("%Y%m%d%H%M%SZ").encode("ascii")
Rick Dean536ba022009-07-24 23:57:27 -05003248 revoked.set_rev_date(now)
Alex Gaynor03737182020-07-23 20:40:46 -04003249 revoked.set_serial(b"3ab")
3250 revoked.set_reason(b"sUpErSeDEd")
Rick Dean536ba022009-07-24 23:57:27 -05003251 crl.add_revoked(revoked)
Jean-Paul Calderone60432792015-04-13 12:26:07 -04003252 return crl
Rick Dean536ba022009-07-24 23:57:27 -05003253
Jean-Paul Calderone60432792015-04-13 12:26:07 -04003254 def test_export_pem(self):
3255 """
3256 If not passed a format, ``CRL.export`` returns a "PEM" format string
3257 representing a serial number, a revoked reason, and certificate issuer
3258 information.
3259 """
Rick Dean536ba022009-07-24 23:57:27 -05003260 # PEM format
Paul Kehrer7d5a3bf2019-01-21 12:24:02 -06003261 dumped_crl = self._get_crl().export(
Alex Gaynor173e4ba2017-06-30 08:01:12 -07003262 self.cert, self.pkey, days=20, digest=b"sha256"
3263 )
Paul Kehrer7d5a3bf2019-01-21 12:24:02 -06003264 crl = x509.load_pem_x509_crl(dumped_crl, backend)
3265 revoked = crl.get_revoked_certificate_by_serial_number(0x03AB)
3266 assert revoked is not None
Alex Gaynor03737182020-07-23 20:40:46 -04003267 assert crl.issuer == x509.Name(
3268 [
3269 x509.NameAttribute(x509.NameOID.COUNTRY_NAME, u"US"),
3270 x509.NameAttribute(x509.NameOID.STATE_OR_PROVINCE_NAME, u"IL"),
3271 x509.NameAttribute(x509.NameOID.LOCALITY_NAME, u"Chicago"),
3272 x509.NameAttribute(x509.NameOID.ORGANIZATION_NAME, u"Testing"),
3273 x509.NameAttribute(
3274 x509.NameOID.COMMON_NAME, u"Testing Root CA"
3275 ),
3276 ]
3277 )
Jean-Paul Calderone60432792015-04-13 12:26:07 -04003278
Jean-Paul Calderone60432792015-04-13 12:26:07 -04003279 def test_export_der(self):
3280 """
3281 If passed ``FILETYPE_ASN1`` for the format, ``CRL.export`` returns a
3282 "DER" format string representing a serial number, a revoked reason, and
3283 certificate issuer information.
3284 """
3285 crl = self._get_crl()
Rick Dean536ba022009-07-24 23:57:27 -05003286
3287 # DER format
Paul Kehrer7d5a3bf2019-01-21 12:24:02 -06003288 dumped_crl = self._get_crl().export(
Alex Gaynor173e4ba2017-06-30 08:01:12 -07003289 self.cert, self.pkey, FILETYPE_ASN1, digest=b"md5"
3290 )
Paul Kehrer7d5a3bf2019-01-21 12:24:02 -06003291 crl = x509.load_der_x509_crl(dumped_crl, backend)
3292 revoked = crl.get_revoked_certificate_by_serial_number(0x03AB)
3293 assert revoked is not None
Alex Gaynor03737182020-07-23 20:40:46 -04003294 assert crl.issuer == x509.Name(
3295 [
3296 x509.NameAttribute(x509.NameOID.COUNTRY_NAME, u"US"),
3297 x509.NameAttribute(x509.NameOID.STATE_OR_PROVINCE_NAME, u"IL"),
3298 x509.NameAttribute(x509.NameOID.LOCALITY_NAME, u"Chicago"),
3299 x509.NameAttribute(x509.NameOID.ORGANIZATION_NAME, u"Testing"),
3300 x509.NameAttribute(
3301 x509.NameOID.COMMON_NAME, u"Testing Root CA"
3302 ),
3303 ]
3304 )
Jean-Paul Calderone60432792015-04-13 12:26:07 -04003305
Alex Gaynore466bc92017-07-06 23:43:47 -04003306 # Flaky because we compare the output of running commands which sometimes
3307 # varies by 1 second
3308 @flaky.flaky
Jean-Paul Calderone60432792015-04-13 12:26:07 -04003309 def test_export_text(self):
3310 """
3311 If passed ``FILETYPE_TEXT`` for the format, ``CRL.export`` returns a
3312 text format string like the one produced by the openssl command line
3313 tool.
3314 """
3315 crl = self._get_crl()
3316
Alex Gaynor173e4ba2017-06-30 08:01:12 -07003317 dumped_crl = crl.export(
3318 self.cert, self.pkey, FILETYPE_ASN1, digest=b"md5"
3319 )
Jean-Paul Calderone60432792015-04-13 12:26:07 -04003320 text = _runopenssl(
Alex Gaynor03737182020-07-23 20:40:46 -04003321 dumped_crl,
3322 b"crl",
3323 b"-noout",
3324 b"-text",
3325 b"-inform",
3326 b"DER",
3327 b"-nameopt",
3328 b"",
Jean-Paul Calderone60432792015-04-13 12:26:07 -04003329 )
Rick Dean536ba022009-07-24 23:57:27 -05003330
3331 # text format
Alex Gaynor173e4ba2017-06-30 08:01:12 -07003332 dumped_text = crl.export(
3333 self.cert, self.pkey, type=FILETYPE_TEXT, digest=b"md5"
3334 )
Alex Chan7be83a52017-01-24 15:19:29 +00003335 assert text == dumped_text
Rick Dean536ba022009-07-24 23:57:27 -05003336
Jean-Paul Calderone60432792015-04-13 12:26:07 -04003337 def test_export_custom_digest(self):
3338 """
3339 If passed the name of a digest function, ``CRL.export`` uses a
3340 signature algorithm based on that digest function.
3341 """
3342 crl = self._get_crl()
Jean-Paul Calderonecce22d02015-04-13 13:56:09 -04003343 dumped_crl = crl.export(self.cert, self.pkey, digest=b"sha1")
Bulat Gaifullin1966c972014-09-22 09:47:20 +04003344 text = _runopenssl(dumped_crl, b"crl", b"-noout", b"-text")
Alex Gaynor03737182020-07-23 20:40:46 -04003345 text.index(b"Signature Algorithm: sha1")
Bulat Gaifullin1966c972014-09-22 09:47:20 +04003346
Jean-Paul Calderonecce22d02015-04-13 13:56:09 -04003347 def test_export_md5_digest(self):
3348 """
3349 If passed md5 as the digest function, ``CRL.export`` uses md5 and does
3350 not emit a deprecation warning.
3351 """
3352 crl = self._get_crl()
Alex Chan7be83a52017-01-24 15:19:29 +00003353 with pytest.warns(None) as catcher:
Jean-Paul Calderonecce22d02015-04-13 13:56:09 -04003354 simplefilter("always")
Alex Chan7be83a52017-01-24 15:19:29 +00003355 assert 0 == len(catcher)
Jean-Paul Calderonecce22d02015-04-13 13:56:09 -04003356 dumped_crl = crl.export(self.cert, self.pkey, digest=b"md5")
3357 text = _runopenssl(dumped_crl, b"crl", b"-noout", b"-text")
Alex Gaynor03737182020-07-23 20:40:46 -04003358 text.index(b"Signature Algorithm: md5")
Jean-Paul Calderonecce22d02015-04-13 13:56:09 -04003359
Jean-Paul Calderone60432792015-04-13 12:26:07 -04003360 def test_export_default_digest(self):
3361 """
Alex Gaynor173e4ba2017-06-30 08:01:12 -07003362 If not passed the name of a digest function, ``CRL.export`` raises a
3363 ``TypeError``.
Jean-Paul Calderone60432792015-04-13 12:26:07 -04003364 """
3365 crl = self._get_crl()
Alex Gaynor173e4ba2017-06-30 08:01:12 -07003366 with pytest.raises(TypeError):
3367 crl.export(self.cert, self.pkey)
Bulat Gaifullin1966c972014-09-22 09:47:20 +04003368
Jean-Paul Calderonec7293bc2011-09-13 15:24:38 -04003369 def test_export_invalid(self):
3370 """
Alex Chan7be83a52017-01-24 15:19:29 +00003371 If `CRL.export` is used with an uninitialized `X509` instance,
3372 `OpenSSL.crypto.Error` is raised.
Jean-Paul Calderonec7293bc2011-09-13 15:24:38 -04003373 """
3374 crl = CRL()
Alex Chan7be83a52017-01-24 15:19:29 +00003375 with pytest.raises(Error):
Alex Gaynor173e4ba2017-06-30 08:01:12 -07003376 crl.export(X509(), PKey(), digest=b"sha256")
Jean-Paul Calderonec7293bc2011-09-13 15:24:38 -04003377
Jean-Paul Calderone56515342010-01-30 13:49:38 -05003378 def test_add_revoked_keyword(self):
3379 """
Alex Chan7be83a52017-01-24 15:19:29 +00003380 `OpenSSL.CRL.add_revoked` accepts its single argument as the
Jean-Paul Calderone64efa2c2011-09-11 10:00:09 -04003381 ``revoked`` keyword argument.
Jean-Paul Calderone56515342010-01-30 13:49:38 -05003382 """
3383 crl = CRL()
3384 revoked = Revoked()
Paul Kehrerb11bffc2016-03-10 18:30:29 -04003385 revoked.set_serial(b"01")
Paul Kehrer2fe23b02016-03-09 22:02:15 -04003386 revoked.set_rev_date(b"20160310020145Z")
Jean-Paul Calderone56515342010-01-30 13:49:38 -05003387 crl.add_revoked(revoked=revoked)
Alex Chan7be83a52017-01-24 15:19:29 +00003388 assert isinstance(crl.get_revoked()[0], Revoked)
Jean-Paul Calderone56515342010-01-30 13:49:38 -05003389
Jean-Paul Calderone883ca4b2010-01-30 13:55:13 -05003390 def test_export_wrong_args(self):
3391 """
Alex Chan7be83a52017-01-24 15:19:29 +00003392 Calling `OpenSSL.CRL.export` with arguments other than the certificate,
Jean-Paul Calderonef1515862010-01-30 13:57:03 -05003393 private key, integer file type, and integer number of days it
Alex Chan7be83a52017-01-24 15:19:29 +00003394 expects, results in a `TypeError` being raised.
Jean-Paul Calderone883ca4b2010-01-30 13:55:13 -05003395 """
3396 crl = CRL()
Alex Gaynor85b49702015-09-05 16:30:59 -04003397 with pytest.raises(TypeError):
3398 crl.export(None, self.pkey, FILETYPE_PEM, 10)
3399 with pytest.raises(TypeError):
3400 crl.export(self.cert, None, FILETYPE_PEM, 10)
3401 with pytest.raises(TypeError):
3402 crl.export(self.cert, self.pkey, None, 10)
Alex Chan7be83a52017-01-24 15:19:29 +00003403 with pytest.raises(TypeError):
3404 crl.export(self.cert, FILETYPE_PEM, None)
Jean-Paul Calderonef1515862010-01-30 13:57:03 -05003405
Jean-Paul Calderoneea198422010-01-30 13:58:23 -05003406 def test_export_unknown_filetype(self):
3407 """
Alex Chan7be83a52017-01-24 15:19:29 +00003408 Calling `OpenSSL.CRL.export` with a file type other than
3409 `FILETYPE_PEM`, `FILETYPE_ASN1`, or
3410 `FILETYPE_TEXT` results in a `ValueError` being raised.
Jean-Paul Calderoneea198422010-01-30 13:58:23 -05003411 """
3412 crl = CRL()
Alex Gaynor85b49702015-09-05 16:30:59 -04003413 with pytest.raises(ValueError):
Alex Gaynor173e4ba2017-06-30 08:01:12 -07003414 crl.export(self.cert, self.pkey, 100, 10, digest=b"sha256")
Jean-Paul Calderoneea198422010-01-30 13:58:23 -05003415
Bulat Gaifullin925f7862014-09-22 10:10:44 +04003416 def test_export_unknown_digest(self):
Bulat Gaifullin5f9eea42014-09-23 19:35:15 +04003417 """
Alex Chan7be83a52017-01-24 15:19:29 +00003418 Calling `OpenSSL.CRL.export` with an unsupported digest results
3419 in a `ValueError` being raised.
Bulat Gaifullin5f9eea42014-09-23 19:35:15 +04003420 """
Bulat Gaifullin925f7862014-09-22 10:10:44 +04003421 crl = CRL()
Alex Chan7be83a52017-01-24 15:19:29 +00003422 with pytest.raises(ValueError):
3423 crl.export(
Alex Gaynor03737182020-07-23 20:40:46 -04003424 self.cert, self.pkey, FILETYPE_PEM, 10, b"strange-digest"
3425 )
Bulat Gaifullin925f7862014-09-22 10:10:44 +04003426
Rick Dean536ba022009-07-24 23:57:27 -05003427 def test_get_revoked(self):
3428 """
Alex Chan7be83a52017-01-24 15:19:29 +00003429 Use python to create a simple CRL with two revocations. Get back the
3430 `Revoked` using `OpenSSL.CRL.get_revoked` and verify them.
Rick Dean536ba022009-07-24 23:57:27 -05003431 """
3432 crl = CRL()
3433
3434 revoked = Revoked()
Alex Gaynore7f51982016-09-11 11:48:14 -04003435 now = datetime.now().strftime("%Y%m%d%H%M%SZ").encode("ascii")
Rick Dean536ba022009-07-24 23:57:27 -05003436 revoked.set_rev_date(now)
Alex Gaynor03737182020-07-23 20:40:46 -04003437 revoked.set_serial(b"3ab")
Rick Dean536ba022009-07-24 23:57:27 -05003438 crl.add_revoked(revoked)
Alex Gaynor03737182020-07-23 20:40:46 -04003439 revoked.set_serial(b"100")
3440 revoked.set_reason(b"sUpErSeDEd")
Rick Dean536ba022009-07-24 23:57:27 -05003441 crl.add_revoked(revoked)
3442
3443 revs = crl.get_revoked()
Alex Chan7be83a52017-01-24 15:19:29 +00003444 assert len(revs) == 2
3445 assert type(revs[0]) == Revoked
3446 assert type(revs[1]) == Revoked
Alex Gaynor03737182020-07-23 20:40:46 -04003447 assert revs[0].get_serial() == b"03AB"
3448 assert revs[1].get_serial() == b"0100"
Alex Chan7be83a52017-01-24 15:19:29 +00003449 assert revs[0].get_rev_date() == now
3450 assert revs[1].get_rev_date() == now
Jean-Paul Calderoneecef6fa2010-01-30 13:47:18 -05003451
Rick Dean536ba022009-07-24 23:57:27 -05003452 def test_load_crl(self):
3453 """
Alex Chan7be83a52017-01-24 15:19:29 +00003454 Load a known CRL and inspect its revocations. Both EM and DER formats
3455 are loaded.
Rick Dean536ba022009-07-24 23:57:27 -05003456 """
Jean-Paul Calderone3eb5cc72010-01-30 15:24:40 -05003457 crl = load_crl(FILETYPE_PEM, crlData)
Rick Dean536ba022009-07-24 23:57:27 -05003458 revs = crl.get_revoked()
Alex Chan7be83a52017-01-24 15:19:29 +00003459 assert len(revs) == 2
Alex Gaynor03737182020-07-23 20:40:46 -04003460 assert revs[0].get_serial() == b"03AB"
Alex Chan7be83a52017-01-24 15:19:29 +00003461 assert revs[0].get_reason() is None
Alex Gaynor03737182020-07-23 20:40:46 -04003462 assert revs[1].get_serial() == b"0100"
3463 assert revs[1].get_reason() == b"Superseded"
Rick Dean536ba022009-07-24 23:57:27 -05003464
Jean-Paul Calderone4f0467a2014-01-11 11:58:41 -05003465 der = _runopenssl(crlData, b"crl", b"-outform", b"DER")
Jean-Paul Calderoneb98eae02010-01-30 13:18:04 -05003466 crl = load_crl(FILETYPE_ASN1, der)
Rick Dean536ba022009-07-24 23:57:27 -05003467 revs = crl.get_revoked()
Alex Chan7be83a52017-01-24 15:19:29 +00003468 assert len(revs) == 2
Alex Gaynor03737182020-07-23 20:40:46 -04003469 assert revs[0].get_serial() == b"03AB"
Alex Chan7be83a52017-01-24 15:19:29 +00003470 assert revs[0].get_reason() is None
Alex Gaynor03737182020-07-23 20:40:46 -04003471 assert revs[1].get_serial() == b"0100"
3472 assert revs[1].get_reason() == b"Superseded"
Jean-Paul Calderone3eb5cc72010-01-30 15:24:40 -05003473
Jean-Paul Calderone3eb5cc72010-01-30 15:24:40 -05003474 def test_load_crl_bad_filetype(self):
3475 """
Alex Chan7be83a52017-01-24 15:19:29 +00003476 Calling `OpenSSL.crypto.load_crl` with an unknown file type raises a
3477 `ValueError`.
Jean-Paul Calderone3eb5cc72010-01-30 15:24:40 -05003478 """
Alex Chan7be83a52017-01-24 15:19:29 +00003479 with pytest.raises(ValueError):
3480 load_crl(100, crlData)
Jean-Paul Calderone3eb5cc72010-01-30 15:24:40 -05003481
Jean-Paul Calderone3eb5cc72010-01-30 15:24:40 -05003482 def test_load_crl_bad_data(self):
3483 """
Alex Chan7be83a52017-01-24 15:19:29 +00003484 Calling `OpenSSL.crypto.load_crl` with file data which can't be loaded
3485 raises a `OpenSSL.crypto.Error`.
Jean-Paul Calderone3eb5cc72010-01-30 15:24:40 -05003486 """
Alex Chan7be83a52017-01-24 15:19:29 +00003487 with pytest.raises(Error):
3488 load_crl(FILETYPE_PEM, b"hello, world")
Jean-Paul Calderone3eb5cc72010-01-30 15:24:40 -05003489
Dan Sully44e767a2016-06-04 18:05:27 -07003490 def test_get_issuer(self):
3491 """
Alex Chan7be83a52017-01-24 15:19:29 +00003492 Load a known CRL and assert its issuer's common name is what we expect
3493 from the encoded crlData string.
Dan Sully44e767a2016-06-04 18:05:27 -07003494 """
3495 crl = load_crl(FILETYPE_PEM, crlData)
Alex Chan7be83a52017-01-24 15:19:29 +00003496 assert isinstance(crl.get_issuer(), X509Name)
Alex Gaynor03737182020-07-23 20:40:46 -04003497 assert crl.get_issuer().CN == "Testing Root CA"
Dan Sully44e767a2016-06-04 18:05:27 -07003498
Dominic Chenf05b2122015-10-13 16:32:35 +00003499 def test_dump_crl(self):
3500 """
3501 The dumped CRL matches the original input.
3502 """
3503 crl = load_crl(FILETYPE_PEM, crlData)
3504 buf = dump_crl(FILETYPE_PEM, crl)
3505 assert buf == crlData
3506
Dan Sully44e767a2016-06-04 18:05:27 -07003507 def _make_test_crl(self, issuer_cert, issuer_key, certs=()):
3508 """
3509 Create a CRL.
3510
3511 :param list[X509] certs: A list of certificates to revoke.
3512 :rtype: CRL
3513 """
3514 crl = CRL()
3515 for cert in certs:
3516 revoked = Revoked()
3517 # FIXME: This string splicing is an unfortunate implementation
3518 # detail that has been reported in
3519 # https://github.com/pyca/pyopenssl/issues/258
Alex Gaynor03737182020-07-23 20:40:46 -04003520 serial = hex(cert.get_serial_number())[2:].encode("utf-8")
Dan Sully44e767a2016-06-04 18:05:27 -07003521 revoked.set_serial(serial)
Alex Gaynor03737182020-07-23 20:40:46 -04003522 revoked.set_reason(b"unspecified")
3523 revoked.set_rev_date(b"20140601000000Z")
Dan Sully44e767a2016-06-04 18:05:27 -07003524 crl.add_revoked(revoked)
3525 crl.set_version(1)
Alex Gaynor03737182020-07-23 20:40:46 -04003526 crl.set_lastUpdate(b"20140601000000Z")
3527 crl.set_nextUpdate(b"20180601000000Z")
3528 crl.sign(issuer_cert, issuer_key, digest=b"sha512")
Dan Sully44e767a2016-06-04 18:05:27 -07003529 return crl
3530
3531 def test_verify_with_revoked(self):
3532 """
Alex Chan7be83a52017-01-24 15:19:29 +00003533 `verify_certificate` raises error when an intermediate certificate is
3534 revoked.
Dan Sully44e767a2016-06-04 18:05:27 -07003535 """
3536 store = X509Store()
3537 store.add_cert(self.root_cert)
3538 store.add_cert(self.intermediate_cert)
3539 root_crl = self._make_test_crl(
Alex Gaynor03737182020-07-23 20:40:46 -04003540 self.root_cert, self.root_key, certs=[self.intermediate_cert]
3541 )
Dan Sully44e767a2016-06-04 18:05:27 -07003542 intermediate_crl = self._make_test_crl(
Alex Gaynor03737182020-07-23 20:40:46 -04003543 self.intermediate_cert, self.intermediate_key, certs=[]
3544 )
Dan Sully44e767a2016-06-04 18:05:27 -07003545 store.add_crl(root_crl)
3546 store.add_crl(intermediate_crl)
3547 store.set_flags(
Alex Gaynor03737182020-07-23 20:40:46 -04003548 X509StoreFlags.CRL_CHECK | X509StoreFlags.CRL_CHECK_ALL
3549 )
Dan Sully44e767a2016-06-04 18:05:27 -07003550 store_ctx = X509StoreContext(store, self.intermediate_server_cert)
Alex Chan7be83a52017-01-24 15:19:29 +00003551 with pytest.raises(X509StoreContextError) as err:
3552 store_ctx.verify_certificate()
Alex Gaynor03737182020-07-23 20:40:46 -04003553 assert err.value.args[0][2] == "certificate revoked"
Dan Sully44e767a2016-06-04 18:05:27 -07003554
3555 def test_verify_with_missing_crl(self):
3556 """
Alex Chan7be83a52017-01-24 15:19:29 +00003557 `verify_certificate` raises error when an intermediate certificate's
3558 CRL is missing.
Dan Sully44e767a2016-06-04 18:05:27 -07003559 """
3560 store = X509Store()
3561 store.add_cert(self.root_cert)
3562 store.add_cert(self.intermediate_cert)
3563 root_crl = self._make_test_crl(
Alex Gaynor03737182020-07-23 20:40:46 -04003564 self.root_cert, self.root_key, certs=[self.intermediate_cert]
3565 )
Dan Sully44e767a2016-06-04 18:05:27 -07003566 store.add_crl(root_crl)
3567 store.set_flags(
Alex Gaynor03737182020-07-23 20:40:46 -04003568 X509StoreFlags.CRL_CHECK | X509StoreFlags.CRL_CHECK_ALL
3569 )
Dan Sully44e767a2016-06-04 18:05:27 -07003570 store_ctx = X509StoreContext(store, self.intermediate_server_cert)
Alex Chan7be83a52017-01-24 15:19:29 +00003571 with pytest.raises(X509StoreContextError) as err:
3572 store_ctx.verify_certificate()
Alex Gaynor03737182020-07-23 20:40:46 -04003573 assert err.value.args[0][2] == "unable to get certificate CRL"
3574 assert err.value.certificate.get_subject().CN == "intermediate-service"
Dan Sully44e767a2016-06-04 18:05:27 -07003575
Paul Kehrer41c10242017-06-29 18:24:17 -05003576 def test_convert_from_cryptography(self):
3577 crypto_crl = x509.load_pem_x509_crl(crlData, backend)
3578 crl = CRL.from_cryptography(crypto_crl)
3579 assert isinstance(crl, CRL)
3580
3581 def test_convert_from_cryptography_unsupported_type(self):
3582 with pytest.raises(TypeError):
3583 CRL.from_cryptography(object())
3584
3585 def test_convert_to_cryptography_key(self):
3586 crl = load_crl(FILETYPE_PEM, crlData)
3587 crypto_crl = crl.to_cryptography()
3588 assert isinstance(crypto_crl, x509.CertificateRevocationList)
3589
Jean-Paul Calderonedc138fa2009-06-27 14:32:07 -04003590
Alex Chan7be83a52017-01-24 15:19:29 +00003591class TestX509StoreContext(object):
Stephen Holsapple0d9815f2014-08-27 19:36:53 -07003592 """
Alex Chan7be83a52017-01-24 15:19:29 +00003593 Tests for `OpenSSL.crypto.X509StoreContext`.
Stephen Holsapple0d9815f2014-08-27 19:36:53 -07003594 """
Alex Gaynor03737182020-07-23 20:40:46 -04003595
Stephen Holsapple0d9815f2014-08-27 19:36:53 -07003596 root_cert = load_certificate(FILETYPE_PEM, root_cert_pem)
3597 intermediate_cert = load_certificate(FILETYPE_PEM, intermediate_cert_pem)
Alex Gaynor31287502015-09-05 16:11:27 -04003598 intermediate_server_cert = load_certificate(
Alex Gaynor03737182020-07-23 20:40:46 -04003599 FILETYPE_PEM, intermediate_server_cert_pem
3600 )
Stephen Holsapple0d9815f2014-08-27 19:36:53 -07003601
3602 def test_valid(self):
3603 """
Alex Chan7be83a52017-01-24 15:19:29 +00003604 `verify_certificate` returns ``None`` when called with a certificate
3605 and valid chain.
Stephen Holsapple0d9815f2014-08-27 19:36:53 -07003606 """
3607 store = X509Store()
3608 store.add_cert(self.root_cert)
3609 store.add_cert(self.intermediate_cert)
3610 store_ctx = X509StoreContext(store, self.intermediate_server_cert)
Alex Chan7be83a52017-01-24 15:19:29 +00003611 assert store_ctx.verify_certificate() is None
Stephen Holsapple0d9815f2014-08-27 19:36:53 -07003612
3613 def test_reuse(self):
3614 """
Alex Chan7be83a52017-01-24 15:19:29 +00003615 `verify_certificate` can be called multiple times with the same
Jean-Paul Calderone06e01b92015-01-18 15:43:13 -05003616 ``X509StoreContext`` instance to produce the same result.
Stephen Holsapple0d9815f2014-08-27 19:36:53 -07003617 """
3618 store = X509Store()
3619 store.add_cert(self.root_cert)
3620 store.add_cert(self.intermediate_cert)
3621 store_ctx = X509StoreContext(store, self.intermediate_server_cert)
Alex Chan7be83a52017-01-24 15:19:29 +00003622 assert store_ctx.verify_certificate() is None
3623 assert store_ctx.verify_certificate() is None
Stephen Holsapple0d9815f2014-08-27 19:36:53 -07003624
3625 def test_trusted_self_signed(self):
3626 """
Alex Chan7be83a52017-01-24 15:19:29 +00003627 `verify_certificate` returns ``None`` when called with a self-signed
3628 certificate and itself in the chain.
Stephen Holsapple0d9815f2014-08-27 19:36:53 -07003629 """
3630 store = X509Store()
3631 store.add_cert(self.root_cert)
3632 store_ctx = X509StoreContext(store, self.root_cert)
Alex Chan7be83a52017-01-24 15:19:29 +00003633 assert store_ctx.verify_certificate() is None
Stephen Holsapple0d9815f2014-08-27 19:36:53 -07003634
3635 def test_untrusted_self_signed(self):
3636 """
Alex Chan7be83a52017-01-24 15:19:29 +00003637 `verify_certificate` raises error when a self-signed certificate is
3638 verified without itself in the chain.
Stephen Holsapple0d9815f2014-08-27 19:36:53 -07003639 """
3640 store = X509Store()
3641 store_ctx = X509StoreContext(store, self.root_cert)
Alex Gaynor85b49702015-09-05 16:30:59 -04003642 with pytest.raises(X509StoreContextError) as exc:
3643 store_ctx.verify_certificate()
3644
Alex Gaynor03737182020-07-23 20:40:46 -04003645 assert exc.value.args[0][2] == "self signed certificate"
3646 assert exc.value.certificate.get_subject().CN == "Testing Root CA"
Jean-Paul Calderone517816e2015-01-18 15:39:26 -05003647
Stephen Holsapple0d9815f2014-08-27 19:36:53 -07003648 def test_invalid_chain_no_root(self):
3649 """
Alex Chan7be83a52017-01-24 15:19:29 +00003650 `verify_certificate` raises error when a root certificate is missing
3651 from the chain.
Stephen Holsapple0d9815f2014-08-27 19:36:53 -07003652 """
3653 store = X509Store()
3654 store.add_cert(self.intermediate_cert)
3655 store_ctx = X509StoreContext(store, self.intermediate_server_cert)
Alex Gaynor85b49702015-09-05 16:30:59 -04003656
3657 with pytest.raises(X509StoreContextError) as exc:
3658 store_ctx.verify_certificate()
3659
Alex Gaynor03737182020-07-23 20:40:46 -04003660 assert exc.value.args[0][2] == "unable to get issuer certificate"
3661 assert exc.value.certificate.get_subject().CN == "intermediate"
Jean-Paul Calderone517816e2015-01-18 15:39:26 -05003662
Stephen Holsapple0d9815f2014-08-27 19:36:53 -07003663 def test_invalid_chain_no_intermediate(self):
3664 """
Alex Chan7be83a52017-01-24 15:19:29 +00003665 `verify_certificate` raises error when an intermediate certificate is
3666 missing from the chain.
Stephen Holsapple0d9815f2014-08-27 19:36:53 -07003667 """
3668 store = X509Store()
3669 store.add_cert(self.root_cert)
3670 store_ctx = X509StoreContext(store, self.intermediate_server_cert)
Jean-Paul Calderone517816e2015-01-18 15:39:26 -05003671
Alex Gaynor85b49702015-09-05 16:30:59 -04003672 with pytest.raises(X509StoreContextError) as exc:
3673 store_ctx.verify_certificate()
3674
Alex Gaynor03737182020-07-23 20:40:46 -04003675 assert exc.value.args[0][2] == "unable to get local issuer certificate"
3676 assert exc.value.certificate.get_subject().CN == "intermediate-service"
Stephen Holsapple0d9815f2014-08-27 19:36:53 -07003677
Stephen Holsapple46a09252015-02-12 14:45:43 -08003678 def test_modification_pre_verify(self):
3679 """
Alex Chan7be83a52017-01-24 15:19:29 +00003680 `verify_certificate` can use a store context modified after
Stephen Holsapple46a09252015-02-12 14:45:43 -08003681 instantiation.
3682 """
3683 store_bad = X509Store()
3684 store_bad.add_cert(self.intermediate_cert)
3685 store_good = X509Store()
3686 store_good.add_cert(self.root_cert)
3687 store_good.add_cert(self.intermediate_cert)
3688 store_ctx = X509StoreContext(store_bad, self.intermediate_server_cert)
Alex Gaynor85b49702015-09-05 16:30:59 -04003689
3690 with pytest.raises(X509StoreContextError) as exc:
3691 store_ctx.verify_certificate()
3692
Alex Gaynor03737182020-07-23 20:40:46 -04003693 assert exc.value.args[0][2] == "unable to get issuer certificate"
3694 assert exc.value.certificate.get_subject().CN == "intermediate"
Alex Gaynor85b49702015-09-05 16:30:59 -04003695
Stephen Holsapple46a09252015-02-12 14:45:43 -08003696 store_ctx.set_store(store_good)
Alex Chan7be83a52017-01-24 15:19:29 +00003697 assert store_ctx.verify_certificate() is None
Stephen Holsapple46a09252015-02-12 14:45:43 -08003698
Thomas Sileoe15e60a2016-11-22 18:13:30 +01003699 def test_verify_with_time(self):
3700 """
3701 `verify_certificate` raises error when the verification time is
3702 set at notAfter.
3703 """
3704 store = X509Store()
3705 store.add_cert(self.root_cert)
3706 store.add_cert(self.intermediate_cert)
3707
3708 expire_time = self.intermediate_server_cert.get_notAfter()
3709 expire_datetime = datetime.strptime(
Alex Gaynor03737182020-07-23 20:40:46 -04003710 expire_time.decode("utf-8"), "%Y%m%d%H%M%SZ"
Thomas Sileoe15e60a2016-11-22 18:13:30 +01003711 )
3712 store.set_time(expire_datetime)
3713
3714 store_ctx = X509StoreContext(store, self.intermediate_server_cert)
3715 with pytest.raises(X509StoreContextError) as exc:
3716 store_ctx.verify_certificate()
3717
Alex Gaynor03737182020-07-23 20:40:46 -04003718 assert exc.value.args[0][2] == "certificate has expired"
Thomas Sileoe15e60a2016-11-22 18:13:30 +01003719
Stephen Holsapple46a09252015-02-12 14:45:43 -08003720
Alex Chan7be83a52017-01-24 15:19:29 +00003721class TestSignVerify(object):
James Yonan7c2e5d32010-02-27 05:45:50 -07003722 """
Alex Chan7be83a52017-01-24 15:19:29 +00003723 Tests for `OpenSSL.crypto.sign` and `OpenSSL.crypto.verify`.
James Yonan7c2e5d32010-02-27 05:45:50 -07003724 """
Alex Gaynoraceb3e22015-09-05 12:00:22 -04003725
James Yonan7c2e5d32010-02-27 05:45:50 -07003726 def test_sign_verify(self):
Jean-Paul Calderonef3cb9d82010-06-22 10:29:33 -04003727 """
Alex Chan7be83a52017-01-24 15:19:29 +00003728 `sign` generates a cryptographic signature which `verify` can check.
Jean-Paul Calderonef3cb9d82010-06-22 10:29:33 -04003729 """
Alex Gaynore7f51982016-09-11 11:48:14 -04003730 content = (
3731 b"It was a bright cold day in April, and the clocks were striking "
3732 b"thirteen. Winston Smith, his chin nuzzled into his breast in an "
3733 b"effort to escape the vile wind, slipped quickly through the "
3734 b"glass doors of Victory Mansions, though not quickly enough to "
Alex Gaynor03737182020-07-23 20:40:46 -04003735 b"prevent a swirl of gritty dust from entering along with him."
3736 )
Jean-Paul Calderoneb98ce212010-06-22 09:46:27 -04003737
3738 # sign the content with this private key
Jean-Paul Calderonef3cb9d82010-06-22 10:29:33 -04003739 priv_key = load_privatekey(FILETYPE_PEM, root_key_pem)
Jean-Paul Calderoneb98ce212010-06-22 09:46:27 -04003740 # verify the content with this cert
3741 good_cert = load_certificate(FILETYPE_PEM, root_cert_pem)
3742 # certificate unrelated to priv_key, used to trigger an error
3743 bad_cert = load_certificate(FILETYPE_PEM, server_cert_pem)
James Yonan7c2e5d32010-02-27 05:45:50 -07003744
Alex Gaynor03737182020-07-23 20:40:46 -04003745 for digest in ["md5", "sha1"]:
James Yonan7c2e5d32010-02-27 05:45:50 -07003746 sig = sign(priv_key, content, digest)
3747
Alex Gaynoraceb3e22015-09-05 12:00:22 -04003748 # Verify the signature of content, will throw an exception if
3749 # error.
James Yonan7c2e5d32010-02-27 05:45:50 -07003750 verify(good_cert, sig, content, digest)
3751
3752 # This should fail because the certificate doesn't match the
3753 # private key that was used to sign the content.
Alex Chan7be83a52017-01-24 15:19:29 +00003754 with pytest.raises(Error):
3755 verify(bad_cert, sig, content, digest)
James Yonan7c2e5d32010-02-27 05:45:50 -07003756
3757 # This should fail because we've "tainted" the content after
3758 # signing it.
Alex Chan7be83a52017-01-24 15:19:29 +00003759 with pytest.raises(Error):
3760 verify(good_cert, sig, content + b"tainted", digest)
James Yonan7c2e5d32010-02-27 05:45:50 -07003761
3762 # test that unknown digest types fail
Alex Chan7be83a52017-01-24 15:19:29 +00003763 with pytest.raises(ValueError):
3764 sign(priv_key, content, "strange-digest")
3765 with pytest.raises(ValueError):
3766 verify(good_cert, sig, content, "strange-digest")
James Yonan7c2e5d32010-02-27 05:45:50 -07003767
Abraham Martinc5484ba2015-03-25 15:33:05 +00003768 def test_sign_verify_with_text(self):
3769 """
Alex Chan7be83a52017-01-24 15:19:29 +00003770 `sign` generates a cryptographic signature which
3771 `verify` can check. Deprecation warnings raised because using
Alex Gaynor791212d2015-09-05 15:46:08 -04003772 text instead of bytes as content
Abraham Martinc5484ba2015-03-25 15:33:05 +00003773 """
Jean-Paul Calderone6462b072015-03-29 07:03:11 -04003774 content = (
Jean-Paul Calderone362c1f52015-03-29 08:01:39 -04003775 b"It was a bright cold day in April, and the clocks were striking "
3776 b"thirteen. Winston Smith, his chin nuzzled into his breast in an "
3777 b"effort to escape the vile wind, slipped quickly through the "
3778 b"glass doors of Victory Mansions, though not quickly enough to "
3779 b"prevent a swirl of gritty dust from entering along with him."
Jean-Paul Calderone13a0e652015-03-29 07:58:51 -04003780 ).decode("ascii")
Abraham Martinc5484ba2015-03-25 15:33:05 +00003781
3782 priv_key = load_privatekey(FILETYPE_PEM, root_key_pem)
3783 cert = load_certificate(FILETYPE_PEM, root_cert_pem)
Alex Gaynor03737182020-07-23 20:40:46 -04003784 for digest in ["md5", "sha1"]:
Alex Chan7be83a52017-01-24 15:19:29 +00003785 with pytest.warns(DeprecationWarning) as w:
Abraham Martinc5484ba2015-03-25 15:33:05 +00003786 simplefilter("always")
3787 sig = sign(priv_key, content, digest)
Alex Gaynor03737182020-07-23 20:40:46 -04003788 assert "{0} for data is no longer accepted, use bytes".format(
3789 WARNING_TYPE_EXPECTED
3790 ) == str(w[-1].message)
Jean-Paul Calderone6462b072015-03-29 07:03:11 -04003791
Alex Chan7be83a52017-01-24 15:19:29 +00003792 with pytest.warns(DeprecationWarning) as w:
Abraham Martinc5484ba2015-03-25 15:33:05 +00003793 simplefilter("always")
3794 verify(cert, sig, content, digest)
Alex Gaynor03737182020-07-23 20:40:46 -04003795 assert "{0} for data is no longer accepted, use bytes".format(
3796 WARNING_TYPE_EXPECTED
3797 ) == str(w[-1].message)
Abraham Martinc5484ba2015-03-25 15:33:05 +00003798
Paul Kehrer59d26252017-07-20 10:45:54 +02003799 def test_sign_verify_ecdsa(self):
3800 """
3801 `sign` generates a cryptographic signature which `verify` can check.
3802 ECDSA Signatures in the X9.62 format may have variable length,
3803 different from the length of the private key.
3804 """
3805 content = (
3806 b"It was a bright cold day in April, and the clocks were striking "
3807 b"thirteen. Winston Smith, his chin nuzzled into his breast in an "
3808 b"effort to escape the vile wind, slipped quickly through the "
3809 b"glass doors of Victory Mansions, though not quickly enough to "
3810 b"prevent a swirl of gritty dust from entering along with him."
3811 ).decode("ascii")
3812 priv_key = load_privatekey(FILETYPE_PEM, ec_root_key_pem)
3813 cert = load_certificate(FILETYPE_PEM, ec_root_cert_pem)
3814 sig = sign(priv_key, content, "sha1")
3815 verify(cert, sig, content, "sha1")
3816
Jean-Paul Calderone9828f662010-12-08 22:57:26 -05003817 def test_sign_nulls(self):
3818 """
Alex Chan7be83a52017-01-24 15:19:29 +00003819 `sign` produces a signature for a string with embedded nulls.
Jean-Paul Calderone9828f662010-12-08 22:57:26 -05003820 """
Alex Gaynore7f51982016-09-11 11:48:14 -04003821 content = b"Watch out! \0 Did you see it?"
Jean-Paul Calderone9828f662010-12-08 22:57:26 -05003822 priv_key = load_privatekey(FILETYPE_PEM, root_key_pem)
3823 good_cert = load_certificate(FILETYPE_PEM, root_cert_pem)
3824 sig = sign(priv_key, content, "sha1")
3825 verify(good_cert, sig, content, "sha1")
3826
Colleen Murphye09399b2016-03-01 17:40:49 -08003827 def test_sign_with_large_key(self):
3828 """
Alex Chan7be83a52017-01-24 15:19:29 +00003829 `sign` produces a signature for a string when using a long key.
Colleen Murphye09399b2016-03-01 17:40:49 -08003830 """
Alex Gaynore7f51982016-09-11 11:48:14 -04003831 content = (
3832 b"It was a bright cold day in April, and the clocks were striking "
3833 b"thirteen. Winston Smith, his chin nuzzled into his breast in an "
3834 b"effort to escape the vile wind, slipped quickly through the "
3835 b"glass doors of Victory Mansions, though not quickly enough to "
Alex Gaynor03737182020-07-23 20:40:46 -04003836 b"prevent a swirl of gritty dust from entering along with him."
3837 )
Colleen Murphye09399b2016-03-01 17:40:49 -08003838
3839 priv_key = load_privatekey(FILETYPE_PEM, large_key_pem)
3840 sign(priv_key, content, "sha1")
3841
Jean-Paul Calderone9828f662010-12-08 22:57:26 -05003842
Alex Chan63ef9bc2016-12-19 12:02:06 +00003843class TestEllipticCurve(object):
Jean-Paul Calderonec09fd582014-04-18 22:00:10 -04003844 """
Alex Chan63ef9bc2016-12-19 12:02:06 +00003845 Tests for `_EllipticCurve`, `get_elliptic_curve`, and
3846 `get_elliptic_curves`.
Jean-Paul Calderonec09fd582014-04-18 22:00:10 -04003847 """
Alex Gaynoraceb3e22015-09-05 12:00:22 -04003848
Jean-Paul Calderonec09fd582014-04-18 22:00:10 -04003849 def test_set(self):
3850 """
Alex Chan63ef9bc2016-12-19 12:02:06 +00003851 `get_elliptic_curves` returns a `set`.
Jean-Paul Calderonec09fd582014-04-18 22:00:10 -04003852 """
Alex Chan63ef9bc2016-12-19 12:02:06 +00003853 assert isinstance(get_elliptic_curves(), set)
Jean-Paul Calderonec09fd582014-04-18 22:00:10 -04003854
Jean-Paul Calderonec09fd582014-04-18 22:00:10 -04003855 def test_a_curve(self):
3856 """
Alex Chan63ef9bc2016-12-19 12:02:06 +00003857 `get_elliptic_curve` can be used to retrieve a particular supported
3858 curve.
Jean-Paul Calderonec09fd582014-04-18 22:00:10 -04003859 """
3860 curves = get_elliptic_curves()
Alex Chan63ef9bc2016-12-19 12:02:06 +00003861 curve = next(iter(curves))
3862 assert curve.name == get_elliptic_curve(curve.name).name
Jean-Paul Calderonec09fd582014-04-18 22:00:10 -04003863
Jean-Paul Calderonec09fd582014-04-18 22:00:10 -04003864 def test_not_a_curve(self):
3865 """
Alex Chan63ef9bc2016-12-19 12:02:06 +00003866 `get_elliptic_curve` raises `ValueError` if called with a name which
3867 does not identify a supported curve.
Jean-Paul Calderonec09fd582014-04-18 22:00:10 -04003868 """
Alex Chan63ef9bc2016-12-19 12:02:06 +00003869 with pytest.raises(ValueError):
3870 get_elliptic_curve(u"this curve was just invented")
Jean-Paul Calderonec09fd582014-04-18 22:00:10 -04003871
Jean-Paul Calderonec09fd582014-04-18 22:00:10 -04003872 def test_repr(self):
3873 """
3874 The string representation of a curve object includes simply states the
3875 object is a curve and what its name is.
3876 """
3877 curves = get_elliptic_curves()
Alex Chan63ef9bc2016-12-19 12:02:06 +00003878 curve = next(iter(curves))
3879 assert "<Curve %r>" % (curve.name,) == repr(curve)
Jean-Paul Calderonec09fd582014-04-18 22:00:10 -04003880
Jean-Paul Calderonec09fd582014-04-18 22:00:10 -04003881 def test_to_EC_KEY(self):
3882 """
3883 The curve object can export a version of itself as an EC_KEY* via the
Alex Chan63ef9bc2016-12-19 12:02:06 +00003884 private `_EllipticCurve._to_EC_KEY`.
Jean-Paul Calderonec09fd582014-04-18 22:00:10 -04003885 """
3886 curves = get_elliptic_curves()
Alex Chan63ef9bc2016-12-19 12:02:06 +00003887 curve = next(iter(curves))
3888 # It's not easy to assert anything about this object. However, see
3889 # leakcheck/crypto.py for a test that demonstrates it at least does
3890 # not leak memory.
3891 curve._to_EC_KEY()
Jean-Paul Calderonec09fd582014-04-18 22:00:10 -04003892
3893
Jean-Paul Calderone1be77082014-04-30 18:17:41 -04003894class EllipticCurveFactory(object):
3895 """
3896 A helper to get the names of two curves.
3897 """
Alex Gaynoraceb3e22015-09-05 12:00:22 -04003898
Jean-Paul Calderone1be77082014-04-30 18:17:41 -04003899 def __init__(self):
3900 curves = iter(get_elliptic_curves())
Alex Chan63ef9bc2016-12-19 12:02:06 +00003901 self.curve_name = next(curves).name
3902 self.another_curve_name = next(curves).name
Jean-Paul Calderone1be77082014-04-30 18:17:41 -04003903
3904
Alex Chan63ef9bc2016-12-19 12:02:06 +00003905class TestEllipticCurveEquality(EqualityTestsMixin):
Jean-Paul Calderone1be77082014-04-30 18:17:41 -04003906 """
Paul Kehrer7d5a3bf2019-01-21 12:24:02 -06003907 Tests `_EllipticCurve`'s implementation of ``==`` and ``!=``.
Jean-Paul Calderone1be77082014-04-30 18:17:41 -04003908 """
Alex Gaynor03737182020-07-23 20:40:46 -04003909
Jean-Paul Calderone1be77082014-04-30 18:17:41 -04003910 curve_factory = EllipticCurveFactory()
3911
3912 if curve_factory.curve_name is None:
3913 skip = "There are no curves available there can be no curve objects."
3914
Jean-Paul Calderone1be77082014-04-30 18:17:41 -04003915 def anInstance(self):
3916 """
3917 Get the curve object for an arbitrary curve supported by the system.
3918 """
3919 return get_elliptic_curve(self.curve_factory.curve_name)
3920
Jean-Paul Calderone1be77082014-04-30 18:17:41 -04003921 def anotherInstance(self):
3922 """
3923 Get the curve object for an arbitrary curve supported by the system -
3924 but not the one returned by C{anInstance}.
3925 """
3926 return get_elliptic_curve(self.curve_factory.another_curve_name)
3927
3928
Alex Chan63ef9bc2016-12-19 12:02:06 +00003929class TestEllipticCurveHash(object):
Jean-Paul Calderone22c31242014-05-01 07:49:47 -04003930 """
Alex Chan63ef9bc2016-12-19 12:02:06 +00003931 Tests for `_EllipticCurve`'s implementation of hashing (thus use as
3932 an item in a `dict` or `set`).
Jean-Paul Calderone22c31242014-05-01 07:49:47 -04003933 """
Alex Gaynor03737182020-07-23 20:40:46 -04003934
Jean-Paul Calderone22c31242014-05-01 07:49:47 -04003935 curve_factory = EllipticCurveFactory()
3936
3937 if curve_factory.curve_name is None:
3938 skip = "There are no curves available there can be no curve objects."
3939
Jean-Paul Calderone22c31242014-05-01 07:49:47 -04003940 def test_contains(self):
3941 """
Alex Chan63ef9bc2016-12-19 12:02:06 +00003942 The ``in`` operator reports that a `set` containing a curve does
3943 contain that curve.
Jean-Paul Calderone22c31242014-05-01 07:49:47 -04003944 """
3945 curve = get_elliptic_curve(self.curve_factory.curve_name)
3946 curves = set([curve])
Alex Chan63ef9bc2016-12-19 12:02:06 +00003947 assert curve in curves
Jean-Paul Calderone22c31242014-05-01 07:49:47 -04003948
Jean-Paul Calderone22c31242014-05-01 07:49:47 -04003949 def test_does_not_contain(self):
3950 """
Alex Chan63ef9bc2016-12-19 12:02:06 +00003951 The ``in`` operator reports that a `set` not containing a curve
3952 does not contain that curve.
Jean-Paul Calderone22c31242014-05-01 07:49:47 -04003953 """
3954 curve = get_elliptic_curve(self.curve_factory.curve_name)
Alex Gaynor03737182020-07-23 20:40:46 -04003955 curves = set(
3956 [get_elliptic_curve(self.curve_factory.another_curve_name)]
3957 )
Alex Chan63ef9bc2016-12-19 12:02:06 +00003958 assert curve not in curves