blob: 3395daf97ac7301bd34bfd1ae57f321197de4a62 [file] [log] [blame]
agl@chromium.org77a9ad92012-03-20 15:14:27 +00001# Copyright (c) 2012 The Chromium Authors. All rights reserved.
2# Use of this source code is governed by a BSD-style license that can be
3# found in the LICENSE file.
4
5import asn1
dadrian4ccf51c2016-07-20 15:36:58 -07006import datetime
agl@chromium.org77a9ad92012-03-20 15:14:27 +00007import hashlib
dadrian4ccf51c2016-07-20 15:36:58 -07008import itertools
agl@chromium.org77a9ad92012-03-20 15:14:27 +00009import os
dadrian4ccf51c2016-07-20 15:36:58 -070010import time
agl@chromium.org77a9ad92012-03-20 15:14:27 +000011
dadrian4ccf51c2016-07-20 15:36:58 -070012GENERALIZED_TIME_FORMAT = "%Y%m%d%H%M%SZ"
13
14OCSP_STATE_GOOD = 1
15OCSP_STATE_REVOKED = 2
16OCSP_STATE_INVALID_RESPONSE = 3
17OCSP_STATE_UNAUTHORIZED = 4
18OCSP_STATE_UNKNOWN = 5
19OCSP_STATE_TRY_LATER = 6
20OCSP_STATE_INVALID_RESPONSE_DATA = 7
21OCSP_STATE_MISMATCHED_SERIAL = 8
22
23OCSP_DATE_VALID = 1
24OCSP_DATE_OLD = 2
25OCSP_DATE_EARLY = 3
26OCSP_DATE_LONG = 4
27
28OCSP_PRODUCED_VALID = 1
29OCSP_PRODUCED_BEFORE_CERT = 2
30OCSP_PRODUCED_AFTER_CERT = 3
agl@chromium.org77a9ad92012-03-20 15:14:27 +000031
32# This file implements very minimal certificate and OCSP generation. It's
33# designed to test revocation checking.
34
35def RandomNumber(length_in_bytes):
36 '''RandomNumber returns a random number of length 8*|length_in_bytes| bits'''
37 rand = os.urandom(length_in_bytes)
38 n = 0
39 for x in rand:
40 n <<= 8
41 n |= ord(x)
42 return n
43
44
45def ModExp(n, e, p):
46 '''ModExp returns n^e mod p'''
47 r = 1
48 while e != 0:
49 if e & 1:
50 r = (r*n) % p
51 e >>= 1
52 n = (n*n) % p
53 return r
54
rsleevi85d21772014-08-28 15:46:51 -070055# PKCS1v15_SHA256_PREFIX is the ASN.1 prefix for a SHA256 signature.
56PKCS1v15_SHA256_PREFIX = '3031300d060960864801650304020105000420'.decode('hex')
agl@chromium.org77a9ad92012-03-20 15:14:27 +000057
58class RSA(object):
59 def __init__(self, modulus, e, d):
60 self.m = modulus
61 self.e = e
62 self.d = d
63
64 self.modlen = 0
65 m = modulus
66 while m != 0:
67 self.modlen += 1
68 m >>= 8
69
70 def Sign(self, message):
rsleevi85d21772014-08-28 15:46:51 -070071 digest = hashlib.sha256(message).digest()
72 prefix = PKCS1v15_SHA256_PREFIX
agl@chromium.org77a9ad92012-03-20 15:14:27 +000073
74 em = ['\xff'] * (self.modlen - 1 - len(prefix) - len(digest))
75 em[0] = '\x00'
76 em[1] = '\x01'
77 em += "\x00" + prefix + digest
78
79 n = 0
80 for x in em:
81 n <<= 8
82 n |= ord(x)
83
84 s = ModExp(n, self.d, self.m)
85 out = []
86 while s != 0:
87 out.append(s & 0xff)
88 s >>= 8
89 out.reverse()
90 return '\x00' * (self.modlen - len(out)) + asn1.ToBytes(out)
91
92 def ToDER(self):
93 return asn1.ToDER(asn1.SEQUENCE([self.m, self.e]))
94
95
96def Name(cn = None, c = None, o = None):
97 names = asn1.SEQUENCE([])
98
99 if cn is not None:
100 names.children.append(
101 asn1.SET([
102 asn1.SEQUENCE([
103 COMMON_NAME, cn,
104 ])
105 ])
106 )
107
108 if c is not None:
109 names.children.append(
110 asn1.SET([
111 asn1.SEQUENCE([
112 COUNTRY, c,
113 ])
114 ])
115 )
116
117 if o is not None:
118 names.children.append(
119 asn1.SET([
120 asn1.SEQUENCE([
121 ORGANIZATION, o,
122 ])
123 ])
124 )
125
126 return names
127
128
129# The private key and root certificate name are hard coded here:
130
mattm10ede842016-11-29 11:57:16 -0800131# This is the root private key
132ROOT_KEY = RSA(0x00c1541fac63d3b969aa231a02cb2e0d9ee7b26724f136c121b2c28bdae5caa87733cc407ad83842ef20ec67d941b448a1ce3557cf5ddebf3c9bde8f36f253ee73e670d1c4c6631d1ddc0e39cbde09b833f66347ea379c3fa891d61a0ca005b38b0b2cad1058e3589c9f30600be81e4ff4ac220972c17b74f92f03d72b496f643543d0b27a5227f1efee13c138888b23cb101877b3b4dc091f0b3bb6fc3c792187b05ab38e97862f8af6156bcbfbb824385132c6741e6c65cfcd5f13142421a210b95185884c4866f3ea644dfb8006133d14e72a4704f3e700cf827ca5ffd2ef74c2ab6a5259ffff40f0f7f607891388f917fc9fc9e65742df1bfa0b322140bb65,
133 65537,
134 0x00980f2db66ef249e4954074a5fbdf663135363a3071554ac4d19079661bd5b179c890ffaa5fc4a8c8e3116e81104fd7cd049f2a48dd2165332bb9fad511f6f817cb09b3c45cf1fa25d13e9331099c8578c173c74dae9dc3e83784ba0a7216e9e8144af8786221b741c167d033ad47a245e4da04aa710a44aff5cdc480b48adbba3575d1315555690f081f9f69691e801e34c21240bcd3df9573ec5f9aa290c5ed19404fb911ab28b7680e0be086487273db72da6621f24d8c66197a5f1b7687efe1d9e3b6655af2891d4540482e1246ff5f62ce61b8b5dcb2c66ade6bb41e0bf071445fb8544aa0a489780f770a6f1031ee19347641794f4ad17354d579a9d061)
agl@chromium.org77a9ad92012-03-20 15:14:27 +0000135
136# Root certificate CN
mattm10ede842016-11-29 11:57:16 -0800137ROOT_CN = "Testing CA"
agl@chromium.org77a9ad92012-03-20 15:14:27 +0000138
139# All certificates are issued under this policy OID, in the Google arc:
140CERT_POLICY_OID = asn1.OID([1, 3, 6, 1, 4, 1, 11129, 2, 4, 1])
141
142# These result in the following root certificate:
143# -----BEGIN CERTIFICATE-----
mattm10ede842016-11-29 11:57:16 -0800144# MIIC1DCCAbygAwIBAgIBATANBgkqhkiG9w0BAQsFADAVMRMwEQYDVQQDEwpUZXN0
145# aW5nIENBMB4XDTEwMDEwMTA2MDAwMFoXDTMyMTIwMTA2MDAwMFowFTETMBEGA1UE
146# AxMKVGVzdGluZyBDQTCCASIwDQYJKoZIhvcNAQEBBQADggEPADCCAQoCggEBAMFU
147# H6xj07lpqiMaAssuDZ7nsmck8TbBIbLCi9rlyqh3M8xAetg4Qu8g7GfZQbRIoc41
148# V89d3r88m96PNvJT7nPmcNHExmMdHdwOOcveCbgz9mNH6jecP6iR1hoMoAWziwss
149# rRBY41icnzBgC+geT/SsIglywXt0+S8D1ytJb2Q1Q9CyelIn8e/uE8E4iIsjyxAY
150# d7O03AkfCzu2/Dx5IYewWrOOl4YvivYVa8v7uCQ4UTLGdB5sZc/NXxMUJCGiELlR
151# hYhMSGbz6mRN+4AGEz0U5ypHBPPnAM+CfKX/0u90wqtqUln//0Dw9/YHiROI+Rf8
152# n8nmV0LfG/oLMiFAu2UCAwEAAaMvMC0wEgYDVR0TAQH/BAgwBgEB/wIBATAXBgNV
153# HSAEEDAOMAwGCisGAQQB1nkCBAEwDQYJKoZIhvcNAQELBQADggEBADNrvoAyqAVm
154# bydPBBfLRqyH4DXt2vuMVmnSdnWnOxYiEezGmNSNiO1k1ZFBwVSsd+JHrT24lax9
155# kvU1yQDW//PBu3ijfZOCaIUleQiGXHMGfV4MjzgYbxpvHOvEUC6IXmYCsIEwcZgK
156# lrwnfJQ3MVU4hOgGTlOTWYPtCwvTsBObNRLdIs+ifMQiWmzPBlM8XeX4e5acDjTb
157# emcN4szU3EcgmCA0LvBIRI4F6NWpaIJl2WnLyMUDyKq4vjpRJOZkNwAC+525duDr
158# JFE4PKR2Lh53nJQIJv6mcTZQkX1mmw0yzqWxcGCoHACma3TgSwOHryvSopL+t26+
159# ZlQvP2ygwqY=
agl@chromium.org77a9ad92012-03-20 15:14:27 +0000160# -----END CERTIFICATE-----
161
mattmadace8e2016-10-03 14:07:15 -0700162# If you update any of the above, you can generate a new root by running this
163# file as a script.
agl@chromium.org77a9ad92012-03-20 15:14:27 +0000164
mattm10ede842016-11-29 11:57:16 -0800165INTERMEDIATE_KEY = RSA(0x00c661afcc659f88855a83ade8fb792dc13d0cf388b17bece9149cf0b8556d27b19101d081fb2a842d13a2ac95d8308ddd66783843ecc5806513959eb6b30dd69b2845d97e10d0bbbf653d686dc8828935022cc96f9e030b567157257d3d6526734080bb9727cee0d30f4209d5820e1d662f358fc789c0e9366d84f89adf1beb8d843f74e6f325876ac35d5c11691fcb296967c06edf69450c16bb2314c14599fe90725d5ec90f2db6698afae72bba0cfbf77967c7e8b49f2172f9381827c27ab7f9471c62bd8da4a6c657966ec1385cf41d739449835888f30d64971619dcd380408cd74f25c3be19833a92620c9cf710da67e15ac8cef69bc7e4e5e7f813c1ed,
166 65537,
167 0x77c5e2edf52d2cafd6c649e9b06aa9455226cfa26805fa337f4e81c7c94bedfb3721715208e2d28aa4a042b2f5a3db03212ad44dae564ffeb6a44efedf7c2b65e21aca056301a3591b36c82600394fbdc16268fc0adaabadb5207871f4ef6d17888a30b84240955cd889768681cf23d0de0fe88f008c8841643e341acd397e2d1104a23242e566088b7617c26ae8b48a85b6c9b7dc64ef1fa5e9b124ff8c1659a82d8225f28a820cc6ca07beff0354364c631a9142309fea1d8b054f6e00e23c54b493a21fcbe89a646b39d1acba5bc2ace9bba0252671d42a15202f3afccc912114d6c20eb3131e74289f2c744c5b39e7d3780fe21402ab1c3ae65854fee401)
168
169# Intermediate certificate CN
170INTERMEDIATE_CN = "Testing Intermediate CA"
171
172LEAF_KEY = RSA(0x00cd12d317b39cfbb160fb1dc9c9f0dc8fef3604dda4d8c557392ce1d616483713f78216cadbefd1c76ea0f3bbbe410e24b233b1b73583922b09314e249b2cfde1be0995e13f160fb630c10d447750da20ffaa4880006717feaa3e4db602e4f511b5cc312f770f44b037784effec62640f948aa189c3769f03bdd0e22a36ecfa5951f5577de195a4fba33c879b657968b79138fd7ab389a9968522f7389c6052be1ff78bc168d3ea961e132a044eba33ac07ead95367c7b815e91eca924d914fd0d811349b8bf500707ba71a43a2901a545f34e1792e72654f6649fab9716f4ba17379ee8042186bbba9b9bac416a60474cc60686f0e6e4b01259cc3cb5873edf9,
173 65537,
174 0x009c23e81bd4c30314743dded9646b82d408937db2f0afa7d9988be6cba59d886a287aa13605ad9c7117776efc94885de76cd3554da46e301d9a5b331f4613449edb9ddac36cd0345848d8c46c4bd880acbd5cfee48ee9efe813e16a33da124fd213348c8292494ac84d03ca4aabc5e25fc67ea32e0c6845fc884b01d8988768b8b931c41de49708dbcd5fcb61823f9a1f7507c6f364be4cb5a8cf24af4925997030dd4f67a0c9c6813401cc8b2f5d1971ee0022770239b7042fde8228c33942e9c0a0b18854cb1b5542be928338ab33ac936bbba174e55457007b16f36011dbb8f4258abe64e42b1cfa79803d30170b7ecf3e7c595d42003fff72591e07acd9cd)
175
176LEAF_KEY_PEM = '''-----BEGIN RSA PRIVATE KEY-----
177MIIEpQIBAAKCAQEAzRLTF7Oc+7Fg+x3JyfDcj+82BN2k2MVXOSzh1hZINxP3ghbK
1782+/Rx26g87u+QQ4ksjOxtzWDkisJMU4kmyz94b4JleE/Fg+2MMENRHdQ2iD/qkiA
179AGcX/qo+TbYC5PURtcwxL3cPRLA3eE7/7GJkD5SKoYnDdp8DvdDiKjbs+llR9Vd9
1804ZWk+6M8h5tleWi3kTj9erOJqZaFIvc4nGBSvh/3i8Fo0+qWHhMqBE66M6wH6tlT
181Z8e4FekeypJNkU/Q2BE0m4v1AHB7pxpDopAaVF804XkucmVPZkn6uXFvS6Fzee6A
182Qhhru6m5usQWpgR0zGBobw5uSwElnMPLWHPt+QIDAQABAoIBAQCcI+gb1MMDFHQ9
1833tlka4LUCJN9svCvp9mYi+bLpZ2Iaih6oTYFrZxxF3du/JSIXeds01VNpG4wHZpb
184Mx9GE0Se253aw2zQNFhI2MRsS9iArL1c/uSO6e/oE+FqM9oST9ITNIyCkklKyE0D
185ykqrxeJfxn6jLgxoRfyISwHYmIdouLkxxB3klwjbzV/LYYI/mh91B8bzZL5MtajP
186JK9JJZlwMN1PZ6DJxoE0AcyLL10Zce4AIncCObcEL96CKMM5QunAoLGIVMsbVUK+
187koM4qzOsk2u7oXTlVFcAexbzYBHbuPQlir5k5Csc+nmAPTAXC37PPnxZXUIAP/9y
188WR4HrNnNAoGBAPmOqTe7ntto6rDEsU1cKOJFKIZ7UVcSByyz8aLrvj1Rb2mkrNJU
189SdTqJvtqrvDXgO0HuGtFOzsZrRV9+XRPd2P0mP0uhfRiYGWT8hnILGyI2+7zlC/w
190HDtLEefelhtdOVKgUaLQXptSn7aGalUHghZKWjRNT5ah+U85MoI2ZkDbAoGBANJe
191KvrBBPSFLj+x2rsMhG+ksK0I6tivapVvSTtDV3ME1DvA/4BIMV/nIZyoH4AYI72c
192m/vD66+eCqh75cq5BzbVD63tR+ZRi/VdT1HJcl2IFXynk6eaBw8v7gpQyx6t3iSK
193lx/dIdpLt1BQuR4qI6x1wYp7Utn98soEkiFXzgq7AoGBAJTLBYPQXvgNBxlcPSaV
194016Nw4rjTe0vN43kwCbWjjf7LQV9BPnm/Zpv/cwboLDCnQE2gDOdNKKZPYS59pjt
195pI65UNpr+bxrR3RpEIlku2/+7br8ChfG/t4vdT6djTxFih8ErYf42t+bFNT8Mbv+
1963QYzULMsgU6bxo0A2meezbrPAoGBAK/IxmtQXP6iRxosWRUSCZxs5sFAgVVdh1el
197bXEa/Xj8IQhpZlbgfHmh3oFULzZPdZYcxm7jsQ7HpipRlZwHbtLPyNFSRFFd9PCr
1987vrttSYY77OBKC3V1G5JY8S07HYPXV/1ewDCPGZ3/I8dVQKyvap/n6FDGeFUhctv
199dFhuUZq/AoGAWLXlbcIl1cvOhfFJ5owohJhzh9oW9tlCtjV5/dlix2RaE5CtDZWS
200oMm4sQu9HiA8jLDP1MEEMRFPrPXdrZnxnSqVd1DgabSegD1/ZCb1QlWwQWkk5QU+
201wotPOMI33L50kZqUaDP+1XSL0Dyfo/pYpm4tYy/5QmP6WKXCtFUXybI=
202-----END RSA PRIVATE KEY-----
203'''
agl@chromium.org77a9ad92012-03-20 15:14:27 +0000204
205# Various OIDs
206
207AIA_OCSP = asn1.OID([1, 3, 6, 1, 5, 5, 7, 48, 1])
mattm10ede842016-11-29 11:57:16 -0800208AIA_CA_ISSUERS = asn1.OID([1, 3, 6, 1, 5, 5, 7, 48, 2])
agl@chromium.org77a9ad92012-03-20 15:14:27 +0000209AUTHORITY_INFORMATION_ACCESS = asn1.OID([1, 3, 6, 1, 5, 5, 7, 1, 1])
210BASIC_CONSTRAINTS = asn1.OID([2, 5, 29, 19])
211CERT_POLICIES = asn1.OID([2, 5, 29, 32])
212COMMON_NAME = asn1.OID([2, 5, 4, 3])
213COUNTRY = asn1.OID([2, 5, 4, 6])
214HASH_SHA1 = asn1.OID([1, 3, 14, 3, 2, 26])
215OCSP_TYPE_BASIC = asn1.OID([1, 3, 6, 1, 5, 5, 7, 48, 1, 1])
216ORGANIZATION = asn1.OID([2, 5, 4, 10])
217PUBLIC_KEY_RSA = asn1.OID([1, 2, 840, 113549, 1, 1, 1])
rsleevi85d21772014-08-28 15:46:51 -0700218SHA256_WITH_RSA_ENCRYPTION = asn1.OID([1, 2, 840, 113549, 1, 1, 11])
agl@chromium.org77a9ad92012-03-20 15:14:27 +0000219
220
221def MakeCertificate(
mattm10ede842016-11-29 11:57:16 -0800222 issuer_cn, subject_cn, serial, pubkey, privkey, ocsp_url = None,
223 ca_issuers_url = None, is_ca=False, path_len=None):
agl@chromium.org77a9ad92012-03-20 15:14:27 +0000224 '''MakeCertificate returns a DER encoded certificate, signed by privkey.'''
225 extensions = asn1.SEQUENCE([])
226
227 # Default subject name fields
228 c = "XX"
229 o = "Testing Org"
230
mattm10ede842016-11-29 11:57:16 -0800231 if is_ca:
agl@chromium.org77a9ad92012-03-20 15:14:27 +0000232 # Root certificate.
233 c = None
234 o = None
235 extensions.children.append(
236 asn1.SEQUENCE([
mattmadace8e2016-10-03 14:07:15 -0700237 BASIC_CONSTRAINTS,
agl@chromium.org77a9ad92012-03-20 15:14:27 +0000238 True,
239 asn1.OCTETSTRING(asn1.ToDER(asn1.SEQUENCE([
mattm10ede842016-11-29 11:57:16 -0800240 True, # IsCA
241 ] + ([path_len] if path_len is not None else []) # Path len
242 ))),
agl@chromium.org77a9ad92012-03-20 15:14:27 +0000243 ]))
244
mattm10ede842016-11-29 11:57:16 -0800245 if ocsp_url is not None or ca_issuers_url is not None:
246 aia_entries = []
247 if ocsp_url is not None:
248 aia_entries.append(
249 asn1.SEQUENCE([
250 AIA_OCSP,
251 asn1.Raw(asn1.TagAndLength(0x86, len(ocsp_url)) + ocsp_url),
252 ]))
253 if ca_issuers_url is not None:
254 aia_entries.append(
255 asn1.SEQUENCE([
256 AIA_CA_ISSUERS,
257 asn1.Raw(asn1.TagAndLength(0x86,
258 len(ca_issuers_url)) + ca_issuers_url),
259 ]))
agl@chromium.org77a9ad92012-03-20 15:14:27 +0000260 extensions.children.append(
261 asn1.SEQUENCE([
262 AUTHORITY_INFORMATION_ACCESS,
mattmadace8e2016-10-03 14:07:15 -0700263 # There is implicitly a critical=False here. Since false is the default,
264 # encoding the value would be invalid DER.
mattm10ede842016-11-29 11:57:16 -0800265 asn1.OCTETSTRING(asn1.ToDER(asn1.SEQUENCE(aia_entries))),
266 ]))
agl@chromium.org77a9ad92012-03-20 15:14:27 +0000267
268 extensions.children.append(
269 asn1.SEQUENCE([
270 CERT_POLICIES,
mattmadace8e2016-10-03 14:07:15 -0700271 # There is implicitly a critical=False here. Since false is the default,
272 # encoding the value would be invalid DER.
agl@chromium.org77a9ad92012-03-20 15:14:27 +0000273 asn1.OCTETSTRING(asn1.ToDER(asn1.SEQUENCE([
274 asn1.SEQUENCE([ # PolicyInformation
275 CERT_POLICY_OID,
276 ]),
277 ]))),
278 ])
279 )
280
281 tbsCert = asn1.ToDER(asn1.SEQUENCE([
282 asn1.Explicit(0, 2), # Version
283 serial,
rsleevi85d21772014-08-28 15:46:51 -0700284 asn1.SEQUENCE([SHA256_WITH_RSA_ENCRYPTION, None]), # SignatureAlgorithm
agl@chromium.org77a9ad92012-03-20 15:14:27 +0000285 Name(cn = issuer_cn), # Issuer
286 asn1.SEQUENCE([ # Validity
287 asn1.UTCTime("100101060000Z"), # NotBefore
288 asn1.UTCTime("321201060000Z"), # NotAfter
289 ]),
290 Name(cn = subject_cn, c = c, o = o), # Subject
291 asn1.SEQUENCE([ # SubjectPublicKeyInfo
292 asn1.SEQUENCE([ # Algorithm
293 PUBLIC_KEY_RSA,
294 None,
295 ]),
296 asn1.BitString(asn1.ToDER(pubkey)),
297 ]),
298 asn1.Explicit(3, extensions),
299 ]))
300
301 return asn1.ToDER(asn1.SEQUENCE([
302 asn1.Raw(tbsCert),
303 asn1.SEQUENCE([
rsleevi85d21772014-08-28 15:46:51 -0700304 SHA256_WITH_RSA_ENCRYPTION,
agl@chromium.org77a9ad92012-03-20 15:14:27 +0000305 None,
306 ]),
307 asn1.BitString(privkey.Sign(tbsCert)),
308 ]))
309
dadrian4ccf51c2016-07-20 15:36:58 -0700310def MakeOCSPSingleResponse(
311 issuer_name_hash, issuer_key_hash, serial, ocsp_state, ocsp_date):
agl@chromium.org77a9ad92012-03-20 15:14:27 +0000312 cert_status = None
agl@chromium.orgb5f388a2012-06-29 17:54:06 +0000313 if ocsp_state == OCSP_STATE_REVOKED:
agl@chromium.org77a9ad92012-03-20 15:14:27 +0000314 cert_status = asn1.Explicit(1, asn1.GeneralizedTime("20100101060000Z"))
agl@chromium.orgb5f388a2012-06-29 17:54:06 +0000315 elif ocsp_state == OCSP_STATE_UNKNOWN:
316 cert_status = asn1.Raw(asn1.TagAndLength(0x80 | 2, 0))
317 elif ocsp_state == OCSP_STATE_GOOD:
agl@chromium.org77a9ad92012-03-20 15:14:27 +0000318 cert_status = asn1.Raw(asn1.TagAndLength(0x80 | 0, 0))
dadrian4ccf51c2016-07-20 15:36:58 -0700319 elif ocsp_state == OCSP_STATE_MISMATCHED_SERIAL:
320 cert_status = asn1.Raw(asn1.TagAndLength(0x80 | 0, 0))
321 serial -= 1
agl@chromium.orgb5f388a2012-06-29 17:54:06 +0000322 else:
323 raise ValueError('Bad OCSP state: ' + str(ocsp_state))
agl@chromium.org77a9ad92012-03-20 15:14:27 +0000324
dadrian4ccf51c2016-07-20 15:36:58 -0700325 now = datetime.datetime.fromtimestamp(time.mktime(time.gmtime()))
326 if ocsp_date == OCSP_DATE_VALID:
327 thisUpdate = now - datetime.timedelta(days=1)
328 nextUpdate = thisUpdate + datetime.timedelta(weeks=1)
329 elif ocsp_date == OCSP_DATE_OLD:
dadrian9a07cc92016-07-22 10:44:37 -0700330 thisUpdate = now - datetime.timedelta(days=1, weeks=1)
dadrian4ccf51c2016-07-20 15:36:58 -0700331 nextUpdate = thisUpdate + datetime.timedelta(weeks=1)
332 elif ocsp_date == OCSP_DATE_EARLY:
dadrian9a07cc92016-07-22 10:44:37 -0700333 thisUpdate = now + datetime.timedelta(days=1)
dadrian4ccf51c2016-07-20 15:36:58 -0700334 nextUpdate = thisUpdate + datetime.timedelta(weeks=1)
335 elif ocsp_date == OCSP_DATE_LONG:
336 thisUpdate = now - datetime.timedelta(days=365)
dadrian9a07cc92016-07-22 10:44:37 -0700337 nextUpdate = thisUpdate + datetime.timedelta(days=366)
dadrian4ccf51c2016-07-20 15:36:58 -0700338 else:
339 raise ValueError('Bad OCSP date: ' + str(ocsp_date))
340
341 return asn1.SEQUENCE([ # SingleResponse
342 asn1.SEQUENCE([ # CertID
343 asn1.SEQUENCE([ # hashAlgorithm
344 HASH_SHA1,
345 None,
346 ]),
347 issuer_name_hash,
348 issuer_key_hash,
349 serial,
350 ]),
351 cert_status,
352 asn1.GeneralizedTime( # thisUpdate
353 thisUpdate.strftime(GENERALIZED_TIME_FORMAT)
354 ),
355 asn1.Explicit( # nextUpdate
356 0,
357 asn1.GeneralizedTime(nextUpdate.strftime(GENERALIZED_TIME_FORMAT))
358 ),
359 ])
360
361def MakeOCSPResponse(
362 issuer_cn, issuer_key, serial, ocsp_states, ocsp_dates, ocsp_produced):
363 # https://tools.ietf.org/html/rfc2560
364 issuer_name_hash = asn1.OCTETSTRING(
365 hashlib.sha1(asn1.ToDER(Name(cn = issuer_cn))).digest())
366
367 issuer_key_hash = asn1.OCTETSTRING(
368 hashlib.sha1(asn1.ToDER(issuer_key)).digest())
369
370 now = datetime.datetime.fromtimestamp(time.mktime(time.gmtime()))
371 if ocsp_produced == OCSP_PRODUCED_VALID:
372 producedAt = now - datetime.timedelta(days=1)
373 elif ocsp_produced == OCSP_PRODUCED_BEFORE_CERT:
374 producedAt = datetime.datetime.strptime(
375 "19100101050000Z", GENERALIZED_TIME_FORMAT)
376 elif ocsp_produced == OCSP_PRODUCED_AFTER_CERT:
377 producedAt = datetime.datetime.strptime(
378 "20321201070000Z", GENERALIZED_TIME_FORMAT)
379 else:
380 raise ValueError('Bad OCSP produced: ' + str(ocsp_produced))
381
382 single_responses = [
383 MakeOCSPSingleResponse(issuer_name_hash, issuer_key_hash, serial,
384 ocsp_state, ocsp_date)
385 for ocsp_state, ocsp_date in itertools.izip(ocsp_states, ocsp_dates)
386 ]
387
agl@chromium.org77a9ad92012-03-20 15:14:27 +0000388 basic_resp_data_der = asn1.ToDER(asn1.SEQUENCE([
389 asn1.Explicit(2, issuer_key_hash),
dadrian4ccf51c2016-07-20 15:36:58 -0700390 asn1.GeneralizedTime(producedAt.strftime(GENERALIZED_TIME_FORMAT)),
391 asn1.SEQUENCE(single_responses),
agl@chromium.org77a9ad92012-03-20 15:14:27 +0000392 ]))
393
394 basic_resp = asn1.SEQUENCE([
395 asn1.Raw(basic_resp_data_der),
396 asn1.SEQUENCE([
rsleevi85d21772014-08-28 15:46:51 -0700397 SHA256_WITH_RSA_ENCRYPTION,
agl@chromium.org77a9ad92012-03-20 15:14:27 +0000398 None,
399 ]),
400 asn1.BitString(issuer_key.Sign(basic_resp_data_der)),
401 ])
402
403 resp = asn1.SEQUENCE([
404 asn1.ENUMERATED(0),
405 asn1.Explicit(0, asn1.SEQUENCE([
406 OCSP_TYPE_BASIC,
407 asn1.OCTETSTRING(asn1.ToDER(basic_resp)),
408 ]))
409 ])
410
411 return asn1.ToDER(resp)
412
413
414def DERToPEM(der):
415 pem = '-----BEGIN CERTIFICATE-----\n'
416 pem += der.encode('base64')
417 pem += '-----END CERTIFICATE-----\n'
418 return pem
419
agl@chromium.orgb5f388a2012-06-29 17:54:06 +0000420# unauthorizedDER is an OCSPResponse with a status of 6:
421# SEQUENCE { ENUM(6) }
422unauthorizedDER = '30030a0106'.decode('hex')
agl@chromium.org77a9ad92012-03-20 15:14:27 +0000423
424def GenerateCertKeyAndOCSP(subject = "127.0.0.1",
425 ocsp_url = "http://127.0.0.1",
dadrian4ccf51c2016-07-20 15:36:58 -0700426 ocsp_states = None,
427 ocsp_dates = None,
428 ocsp_produced = OCSP_PRODUCED_VALID,
agl@chromium.orgdf778142013-07-31 21:57:28 +0000429 serial = 0):
agl@chromium.org77a9ad92012-03-20 15:14:27 +0000430 '''GenerateCertKeyAndOCSP returns a (cert_and_key_pem, ocsp_der) where:
431 * cert_and_key_pem contains a certificate and private key in PEM format
432 with the given subject common name and OCSP URL.
433 * ocsp_der contains a DER encoded OCSP response or None if ocsp_url is
434 None'''
435
dadrian4ccf51c2016-07-20 15:36:58 -0700436 if ocsp_states is None:
437 ocsp_states = [OCSP_STATE_GOOD]
438 if ocsp_dates is None:
439 ocsp_dates = [OCSP_DATE_VALID]
440
agl@chromium.orgdf778142013-07-31 21:57:28 +0000441 if serial == 0:
442 serial = RandomNumber(16)
mattm10ede842016-11-29 11:57:16 -0800443 cert_der = MakeCertificate(ROOT_CN, bytes(subject), serial, LEAF_KEY,
444 ROOT_KEY, bytes(ocsp_url))
agl@chromium.org77a9ad92012-03-20 15:14:27 +0000445 cert_pem = DERToPEM(cert_der)
446
447 ocsp_der = None
448 if ocsp_url is not None:
dadrian4ccf51c2016-07-20 15:36:58 -0700449 if ocsp_states[0] == OCSP_STATE_UNAUTHORIZED:
agl@chromium.orgb5f388a2012-06-29 17:54:06 +0000450 ocsp_der = unauthorizedDER
dadrian4ccf51c2016-07-20 15:36:58 -0700451 elif ocsp_states[0] == OCSP_STATE_INVALID_RESPONSE:
agl@chromium.orgb5f388a2012-06-29 17:54:06 +0000452 ocsp_der = '3'
dadrian4ccf51c2016-07-20 15:36:58 -0700453 elif ocsp_states[0] == OCSP_STATE_TRY_LATER:
454 resp = asn1.SEQUENCE([
455 asn1.ENUMERATED(3),
456 ])
457 ocsp_der = asn1.ToDER(resp)
458 elif ocsp_states[0] == OCSP_STATE_INVALID_RESPONSE_DATA:
459 invalid_data = asn1.ToDER(asn1.OCTETSTRING('not ocsp data'))
460 basic_resp = asn1.SEQUENCE([
461 asn1.Raw(invalid_data),
462 asn1.SEQUENCE([
463 SHA256_WITH_RSA_ENCRYPTION,
464 None,
465 ]),
mattm10ede842016-11-29 11:57:16 -0800466 asn1.BitString(ROOT_KEY.Sign(invalid_data)),
dadrian4ccf51c2016-07-20 15:36:58 -0700467 ])
468 resp = asn1.SEQUENCE([
469 asn1.ENUMERATED(0),
470 asn1.Explicit(0, asn1.SEQUENCE([
471 OCSP_TYPE_BASIC,
472 asn1.OCTETSTRING(asn1.ToDER(basic_resp)),
473 ])),
474 ])
475 ocsp_der = asn1.ToDER(resp)
agl@chromium.orgb5f388a2012-06-29 17:54:06 +0000476 else:
dadrian4ccf51c2016-07-20 15:36:58 -0700477 ocsp_der = MakeOCSPResponse(
mattm10ede842016-11-29 11:57:16 -0800478 ROOT_CN, ROOT_KEY, serial, ocsp_states, ocsp_dates, ocsp_produced)
agl@chromium.org77a9ad92012-03-20 15:14:27 +0000479
mattm10ede842016-11-29 11:57:16 -0800480 return (cert_pem + LEAF_KEY_PEM, ocsp_der)
481
482
483def GenerateCertKeyAndIntermediate(subject, ca_issuers_url, serial=0):
484 '''Returns a (cert_and_key_pem, intermediate_cert_pem) where:
485 * cert_and_key_pem contains a certificate and private key in PEM format
486 with the given subject common name and caIssuers URL.
487 * intermediate_cert_pem contains a PEM encoded certificate that signed
488 cert_and_key_pem and was signed by ocsp-test-root.pem.'''
489 if serial == 0:
490 serial = RandomNumber(16)
491 target_cert_der = MakeCertificate(INTERMEDIATE_CN, bytes(subject), serial,
492 LEAF_KEY, INTERMEDIATE_KEY,
493 ca_issuers_url=bytes(ca_issuers_url))
494 target_cert_pem = DERToPEM(target_cert_der)
495
496 intermediate_serial = RandomNumber(16)
497 intermediate_cert_der = MakeCertificate(ROOT_CN, INTERMEDIATE_CN,
498 intermediate_serial,
499 INTERMEDIATE_KEY, ROOT_KEY,
500 is_ca=True)
501
502 return target_cert_pem + LEAF_KEY_PEM, intermediate_cert_der
mattmadace8e2016-10-03 14:07:15 -0700503
504
505if __name__ == '__main__':
506 def bin_to_array(s):
507 return ' '.join(['0x%02x,'%ord(c) for c in s])
508
509 import sys
510 sys.path.append(os.path.join(os.path.dirname(os.path.abspath(__file__)), '..',
511 '..', 'data', 'ssl', 'scripts'))
512 import crlsetutil
513
mattm10ede842016-11-29 11:57:16 -0800514 der_root = MakeCertificate(ROOT_CN, ROOT_CN, 1, ROOT_KEY, ROOT_KEY,
515 is_ca=True, path_len=1)
mattmadace8e2016-10-03 14:07:15 -0700516 print 'ocsp-test-root.pem:'
517 print DERToPEM(der_root)
518
519 print
520 print 'kOCSPTestCertFingerprint:'
521 print bin_to_array(hashlib.sha1(der_root).digest())
522
523 print
524 print 'kOCSPTestCertSPKI:'
525 print bin_to_array(crlsetutil.der_cert_to_spki_hash(der_root))