agl@chromium.org | 77a9ad9 | 2012-03-20 15:14:27 +0000 | [diff] [blame] | 1 | # 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 | |
| 5 | import asn1 |
dadrian | 4ccf51c | 2016-07-20 15:36:58 -0700 | [diff] [blame] | 6 | import datetime |
agl@chromium.org | 77a9ad9 | 2012-03-20 15:14:27 +0000 | [diff] [blame] | 7 | import hashlib |
dadrian | 4ccf51c | 2016-07-20 15:36:58 -0700 | [diff] [blame] | 8 | import itertools |
agl@chromium.org | 77a9ad9 | 2012-03-20 15:14:27 +0000 | [diff] [blame] | 9 | import os |
dadrian | 4ccf51c | 2016-07-20 15:36:58 -0700 | [diff] [blame] | 10 | import time |
agl@chromium.org | 77a9ad9 | 2012-03-20 15:14:27 +0000 | [diff] [blame] | 11 | |
dadrian | 4ccf51c | 2016-07-20 15:36:58 -0700 | [diff] [blame] | 12 | GENERALIZED_TIME_FORMAT = "%Y%m%d%H%M%SZ" |
| 13 | |
| 14 | OCSP_STATE_GOOD = 1 |
| 15 | OCSP_STATE_REVOKED = 2 |
| 16 | OCSP_STATE_INVALID_RESPONSE = 3 |
| 17 | OCSP_STATE_UNAUTHORIZED = 4 |
| 18 | OCSP_STATE_UNKNOWN = 5 |
| 19 | OCSP_STATE_TRY_LATER = 6 |
| 20 | OCSP_STATE_INVALID_RESPONSE_DATA = 7 |
| 21 | OCSP_STATE_MISMATCHED_SERIAL = 8 |
| 22 | |
| 23 | OCSP_DATE_VALID = 1 |
| 24 | OCSP_DATE_OLD = 2 |
| 25 | OCSP_DATE_EARLY = 3 |
| 26 | OCSP_DATE_LONG = 4 |
| 27 | |
| 28 | OCSP_PRODUCED_VALID = 1 |
| 29 | OCSP_PRODUCED_BEFORE_CERT = 2 |
| 30 | OCSP_PRODUCED_AFTER_CERT = 3 |
agl@chromium.org | 77a9ad9 | 2012-03-20 15:14:27 +0000 | [diff] [blame] | 31 | |
| 32 | # This file implements very minimal certificate and OCSP generation. It's |
| 33 | # designed to test revocation checking. |
| 34 | |
| 35 | def 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 | |
| 45 | def 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 | |
rsleevi | 85d2177 | 2014-08-28 15:46:51 -0700 | [diff] [blame] | 55 | # PKCS1v15_SHA256_PREFIX is the ASN.1 prefix for a SHA256 signature. |
| 56 | PKCS1v15_SHA256_PREFIX = '3031300d060960864801650304020105000420'.decode('hex') |
agl@chromium.org | 77a9ad9 | 2012-03-20 15:14:27 +0000 | [diff] [blame] | 57 | |
| 58 | class 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): |
rsleevi | 85d2177 | 2014-08-28 15:46:51 -0700 | [diff] [blame] | 71 | digest = hashlib.sha256(message).digest() |
| 72 | prefix = PKCS1v15_SHA256_PREFIX |
agl@chromium.org | 77a9ad9 | 2012-03-20 15:14:27 +0000 | [diff] [blame] | 73 | |
| 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 | |
| 96 | def 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 | |
mattm | 10ede84 | 2016-11-29 11:57:16 -0800 | [diff] [blame] | 131 | # This is the root private key |
| 132 | ROOT_KEY = RSA(0x00c1541fac63d3b969aa231a02cb2e0d9ee7b26724f136c121b2c28bdae5caa87733cc407ad83842ef20ec67d941b448a1ce3557cf5ddebf3c9bde8f36f253ee73e670d1c4c6631d1ddc0e39cbde09b833f66347ea379c3fa891d61a0ca005b38b0b2cad1058e3589c9f30600be81e4ff4ac220972c17b74f92f03d72b496f643543d0b27a5227f1efee13c138888b23cb101877b3b4dc091f0b3bb6fc3c792187b05ab38e97862f8af6156bcbfbb824385132c6741e6c65cfcd5f13142421a210b95185884c4866f3ea644dfb8006133d14e72a4704f3e700cf827ca5ffd2ef74c2ab6a5259ffff40f0f7f607891388f917fc9fc9e65742df1bfa0b322140bb65, |
| 133 | 65537, |
| 134 | 0x00980f2db66ef249e4954074a5fbdf663135363a3071554ac4d19079661bd5b179c890ffaa5fc4a8c8e3116e81104fd7cd049f2a48dd2165332bb9fad511f6f817cb09b3c45cf1fa25d13e9331099c8578c173c74dae9dc3e83784ba0a7216e9e8144af8786221b741c167d033ad47a245e4da04aa710a44aff5cdc480b48adbba3575d1315555690f081f9f69691e801e34c21240bcd3df9573ec5f9aa290c5ed19404fb911ab28b7680e0be086487273db72da6621f24d8c66197a5f1b7687efe1d9e3b6655af2891d4540482e1246ff5f62ce61b8b5dcb2c66ade6bb41e0bf071445fb8544aa0a489780f770a6f1031ee19347641794f4ad17354d579a9d061) |
agl@chromium.org | 77a9ad9 | 2012-03-20 15:14:27 +0000 | [diff] [blame] | 135 | |
| 136 | # Root certificate CN |
mattm | 10ede84 | 2016-11-29 11:57:16 -0800 | [diff] [blame] | 137 | ROOT_CN = "Testing CA" |
agl@chromium.org | 77a9ad9 | 2012-03-20 15:14:27 +0000 | [diff] [blame] | 138 | |
| 139 | # All certificates are issued under this policy OID, in the Google arc: |
| 140 | CERT_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----- |
mattm | 10ede84 | 2016-11-29 11:57:16 -0800 | [diff] [blame] | 144 | # 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.org | 77a9ad9 | 2012-03-20 15:14:27 +0000 | [diff] [blame] | 160 | # -----END CERTIFICATE----- |
| 161 | |
mattm | adace8e | 2016-10-03 14:07:15 -0700 | [diff] [blame] | 162 | # If you update any of the above, you can generate a new root by running this |
| 163 | # file as a script. |
agl@chromium.org | 77a9ad9 | 2012-03-20 15:14:27 +0000 | [diff] [blame] | 164 | |
mattm | 10ede84 | 2016-11-29 11:57:16 -0800 | [diff] [blame] | 165 | INTERMEDIATE_KEY = RSA(0x00c661afcc659f88855a83ade8fb792dc13d0cf388b17bece9149cf0b8556d27b19101d081fb2a842d13a2ac95d8308ddd66783843ecc5806513959eb6b30dd69b2845d97e10d0bbbf653d686dc8828935022cc96f9e030b567157257d3d6526734080bb9727cee0d30f4209d5820e1d662f358fc789c0e9366d84f89adf1beb8d843f74e6f325876ac35d5c11691fcb296967c06edf69450c16bb2314c14599fe90725d5ec90f2db6698afae72bba0cfbf77967c7e8b49f2172f9381827c27ab7f9471c62bd8da4a6c657966ec1385cf41d739449835888f30d64971619dcd380408cd74f25c3be19833a92620c9cf710da67e15ac8cef69bc7e4e5e7f813c1ed, |
| 166 | 65537, |
| 167 | 0x77c5e2edf52d2cafd6c649e9b06aa9455226cfa26805fa337f4e81c7c94bedfb3721715208e2d28aa4a042b2f5a3db03212ad44dae564ffeb6a44efedf7c2b65e21aca056301a3591b36c82600394fbdc16268fc0adaabadb5207871f4ef6d17888a30b84240955cd889768681cf23d0de0fe88f008c8841643e341acd397e2d1104a23242e566088b7617c26ae8b48a85b6c9b7dc64ef1fa5e9b124ff8c1659a82d8225f28a820cc6ca07beff0354364c631a9142309fea1d8b054f6e00e23c54b493a21fcbe89a646b39d1acba5bc2ace9bba0252671d42a15202f3afccc912114d6c20eb3131e74289f2c744c5b39e7d3780fe21402ab1c3ae65854fee401) |
| 168 | |
| 169 | # Intermediate certificate CN |
| 170 | INTERMEDIATE_CN = "Testing Intermediate CA" |
| 171 | |
| 172 | LEAF_KEY = RSA(0x00cd12d317b39cfbb160fb1dc9c9f0dc8fef3604dda4d8c557392ce1d616483713f78216cadbefd1c76ea0f3bbbe410e24b233b1b73583922b09314e249b2cfde1be0995e13f160fb630c10d447750da20ffaa4880006717feaa3e4db602e4f511b5cc312f770f44b037784effec62640f948aa189c3769f03bdd0e22a36ecfa5951f5577de195a4fba33c879b657968b79138fd7ab389a9968522f7389c6052be1ff78bc168d3ea961e132a044eba33ac07ead95367c7b815e91eca924d914fd0d811349b8bf500707ba71a43a2901a545f34e1792e72654f6649fab9716f4ba17379ee8042186bbba9b9bac416a60474cc60686f0e6e4b01259cc3cb5873edf9, |
| 173 | 65537, |
| 174 | 0x009c23e81bd4c30314743dded9646b82d408937db2f0afa7d9988be6cba59d886a287aa13605ad9c7117776efc94885de76cd3554da46e301d9a5b331f4613449edb9ddac36cd0345848d8c46c4bd880acbd5cfee48ee9efe813e16a33da124fd213348c8292494ac84d03ca4aabc5e25fc67ea32e0c6845fc884b01d8988768b8b931c41de49708dbcd5fcb61823f9a1f7507c6f364be4cb5a8cf24af4925997030dd4f67a0c9c6813401cc8b2f5d1971ee0022770239b7042fde8228c33942e9c0a0b18854cb1b5542be928338ab33ac936bbba174e55457007b16f36011dbb8f4258abe64e42b1cfa79803d30170b7ecf3e7c595d42003fff72591e07acd9cd) |
| 175 | |
| 176 | LEAF_KEY_PEM = '''-----BEGIN RSA PRIVATE KEY----- |
| 177 | MIIEpQIBAAKCAQEAzRLTF7Oc+7Fg+x3JyfDcj+82BN2k2MVXOSzh1hZINxP3ghbK |
| 178 | 2+/Rx26g87u+QQ4ksjOxtzWDkisJMU4kmyz94b4JleE/Fg+2MMENRHdQ2iD/qkiA |
| 179 | AGcX/qo+TbYC5PURtcwxL3cPRLA3eE7/7GJkD5SKoYnDdp8DvdDiKjbs+llR9Vd9 |
| 180 | 4ZWk+6M8h5tleWi3kTj9erOJqZaFIvc4nGBSvh/3i8Fo0+qWHhMqBE66M6wH6tlT |
| 181 | Z8e4FekeypJNkU/Q2BE0m4v1AHB7pxpDopAaVF804XkucmVPZkn6uXFvS6Fzee6A |
| 182 | Qhhru6m5usQWpgR0zGBobw5uSwElnMPLWHPt+QIDAQABAoIBAQCcI+gb1MMDFHQ9 |
| 183 | 3tlka4LUCJN9svCvp9mYi+bLpZ2Iaih6oTYFrZxxF3du/JSIXeds01VNpG4wHZpb |
| 184 | Mx9GE0Se253aw2zQNFhI2MRsS9iArL1c/uSO6e/oE+FqM9oST9ITNIyCkklKyE0D |
| 185 | ykqrxeJfxn6jLgxoRfyISwHYmIdouLkxxB3klwjbzV/LYYI/mh91B8bzZL5MtajP |
| 186 | JK9JJZlwMN1PZ6DJxoE0AcyLL10Zce4AIncCObcEL96CKMM5QunAoLGIVMsbVUK+ |
| 187 | koM4qzOsk2u7oXTlVFcAexbzYBHbuPQlir5k5Csc+nmAPTAXC37PPnxZXUIAP/9y |
| 188 | WR4HrNnNAoGBAPmOqTe7ntto6rDEsU1cKOJFKIZ7UVcSByyz8aLrvj1Rb2mkrNJU |
| 189 | SdTqJvtqrvDXgO0HuGtFOzsZrRV9+XRPd2P0mP0uhfRiYGWT8hnILGyI2+7zlC/w |
| 190 | HDtLEefelhtdOVKgUaLQXptSn7aGalUHghZKWjRNT5ah+U85MoI2ZkDbAoGBANJe |
| 191 | KvrBBPSFLj+x2rsMhG+ksK0I6tivapVvSTtDV3ME1DvA/4BIMV/nIZyoH4AYI72c |
| 192 | m/vD66+eCqh75cq5BzbVD63tR+ZRi/VdT1HJcl2IFXynk6eaBw8v7gpQyx6t3iSK |
| 193 | lx/dIdpLt1BQuR4qI6x1wYp7Utn98soEkiFXzgq7AoGBAJTLBYPQXvgNBxlcPSaV |
| 194 | 016Nw4rjTe0vN43kwCbWjjf7LQV9BPnm/Zpv/cwboLDCnQE2gDOdNKKZPYS59pjt |
| 195 | pI65UNpr+bxrR3RpEIlku2/+7br8ChfG/t4vdT6djTxFih8ErYf42t+bFNT8Mbv+ |
| 196 | 3QYzULMsgU6bxo0A2meezbrPAoGBAK/IxmtQXP6iRxosWRUSCZxs5sFAgVVdh1el |
| 197 | bXEa/Xj8IQhpZlbgfHmh3oFULzZPdZYcxm7jsQ7HpipRlZwHbtLPyNFSRFFd9PCr |
| 198 | 7vrttSYY77OBKC3V1G5JY8S07HYPXV/1ewDCPGZ3/I8dVQKyvap/n6FDGeFUhctv |
| 199 | dFhuUZq/AoGAWLXlbcIl1cvOhfFJ5owohJhzh9oW9tlCtjV5/dlix2RaE5CtDZWS |
| 200 | oMm4sQu9HiA8jLDP1MEEMRFPrPXdrZnxnSqVd1DgabSegD1/ZCb1QlWwQWkk5QU+ |
| 201 | wotPOMI33L50kZqUaDP+1XSL0Dyfo/pYpm4tYy/5QmP6WKXCtFUXybI= |
| 202 | -----END RSA PRIVATE KEY----- |
| 203 | ''' |
agl@chromium.org | 77a9ad9 | 2012-03-20 15:14:27 +0000 | [diff] [blame] | 204 | |
| 205 | # Various OIDs |
| 206 | |
| 207 | AIA_OCSP = asn1.OID([1, 3, 6, 1, 5, 5, 7, 48, 1]) |
mattm | 10ede84 | 2016-11-29 11:57:16 -0800 | [diff] [blame] | 208 | AIA_CA_ISSUERS = asn1.OID([1, 3, 6, 1, 5, 5, 7, 48, 2]) |
agl@chromium.org | 77a9ad9 | 2012-03-20 15:14:27 +0000 | [diff] [blame] | 209 | AUTHORITY_INFORMATION_ACCESS = asn1.OID([1, 3, 6, 1, 5, 5, 7, 1, 1]) |
| 210 | BASIC_CONSTRAINTS = asn1.OID([2, 5, 29, 19]) |
| 211 | CERT_POLICIES = asn1.OID([2, 5, 29, 32]) |
| 212 | COMMON_NAME = asn1.OID([2, 5, 4, 3]) |
| 213 | COUNTRY = asn1.OID([2, 5, 4, 6]) |
| 214 | HASH_SHA1 = asn1.OID([1, 3, 14, 3, 2, 26]) |
| 215 | OCSP_TYPE_BASIC = asn1.OID([1, 3, 6, 1, 5, 5, 7, 48, 1, 1]) |
| 216 | ORGANIZATION = asn1.OID([2, 5, 4, 10]) |
| 217 | PUBLIC_KEY_RSA = asn1.OID([1, 2, 840, 113549, 1, 1, 1]) |
rsleevi | 85d2177 | 2014-08-28 15:46:51 -0700 | [diff] [blame] | 218 | SHA256_WITH_RSA_ENCRYPTION = asn1.OID([1, 2, 840, 113549, 1, 1, 11]) |
agl@chromium.org | 77a9ad9 | 2012-03-20 15:14:27 +0000 | [diff] [blame] | 219 | |
| 220 | |
| 221 | def MakeCertificate( |
mattm | 10ede84 | 2016-11-29 11:57:16 -0800 | [diff] [blame] | 222 | issuer_cn, subject_cn, serial, pubkey, privkey, ocsp_url = None, |
| 223 | ca_issuers_url = None, is_ca=False, path_len=None): |
agl@chromium.org | 77a9ad9 | 2012-03-20 15:14:27 +0000 | [diff] [blame] | 224 | '''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 | |
mattm | 10ede84 | 2016-11-29 11:57:16 -0800 | [diff] [blame] | 231 | if is_ca: |
agl@chromium.org | 77a9ad9 | 2012-03-20 15:14:27 +0000 | [diff] [blame] | 232 | # Root certificate. |
| 233 | c = None |
| 234 | o = None |
| 235 | extensions.children.append( |
| 236 | asn1.SEQUENCE([ |
mattm | adace8e | 2016-10-03 14:07:15 -0700 | [diff] [blame] | 237 | BASIC_CONSTRAINTS, |
agl@chromium.org | 77a9ad9 | 2012-03-20 15:14:27 +0000 | [diff] [blame] | 238 | True, |
| 239 | asn1.OCTETSTRING(asn1.ToDER(asn1.SEQUENCE([ |
mattm | 10ede84 | 2016-11-29 11:57:16 -0800 | [diff] [blame] | 240 | True, # IsCA |
| 241 | ] + ([path_len] if path_len is not None else []) # Path len |
| 242 | ))), |
agl@chromium.org | 77a9ad9 | 2012-03-20 15:14:27 +0000 | [diff] [blame] | 243 | ])) |
| 244 | |
mattm | 10ede84 | 2016-11-29 11:57:16 -0800 | [diff] [blame] | 245 | 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.org | 77a9ad9 | 2012-03-20 15:14:27 +0000 | [diff] [blame] | 260 | extensions.children.append( |
| 261 | asn1.SEQUENCE([ |
| 262 | AUTHORITY_INFORMATION_ACCESS, |
mattm | adace8e | 2016-10-03 14:07:15 -0700 | [diff] [blame] | 263 | # There is implicitly a critical=False here. Since false is the default, |
| 264 | # encoding the value would be invalid DER. |
mattm | 10ede84 | 2016-11-29 11:57:16 -0800 | [diff] [blame] | 265 | asn1.OCTETSTRING(asn1.ToDER(asn1.SEQUENCE(aia_entries))), |
| 266 | ])) |
agl@chromium.org | 77a9ad9 | 2012-03-20 15:14:27 +0000 | [diff] [blame] | 267 | |
| 268 | extensions.children.append( |
| 269 | asn1.SEQUENCE([ |
| 270 | CERT_POLICIES, |
mattm | adace8e | 2016-10-03 14:07:15 -0700 | [diff] [blame] | 271 | # There is implicitly a critical=False here. Since false is the default, |
| 272 | # encoding the value would be invalid DER. |
agl@chromium.org | 77a9ad9 | 2012-03-20 15:14:27 +0000 | [diff] [blame] | 273 | 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, |
rsleevi | 85d2177 | 2014-08-28 15:46:51 -0700 | [diff] [blame] | 284 | asn1.SEQUENCE([SHA256_WITH_RSA_ENCRYPTION, None]), # SignatureAlgorithm |
agl@chromium.org | 77a9ad9 | 2012-03-20 15:14:27 +0000 | [diff] [blame] | 285 | 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([ |
rsleevi | 85d2177 | 2014-08-28 15:46:51 -0700 | [diff] [blame] | 304 | SHA256_WITH_RSA_ENCRYPTION, |
agl@chromium.org | 77a9ad9 | 2012-03-20 15:14:27 +0000 | [diff] [blame] | 305 | None, |
| 306 | ]), |
| 307 | asn1.BitString(privkey.Sign(tbsCert)), |
| 308 | ])) |
| 309 | |
dadrian | 4ccf51c | 2016-07-20 15:36:58 -0700 | [diff] [blame] | 310 | def MakeOCSPSingleResponse( |
| 311 | issuer_name_hash, issuer_key_hash, serial, ocsp_state, ocsp_date): |
agl@chromium.org | 77a9ad9 | 2012-03-20 15:14:27 +0000 | [diff] [blame] | 312 | cert_status = None |
agl@chromium.org | b5f388a | 2012-06-29 17:54:06 +0000 | [diff] [blame] | 313 | if ocsp_state == OCSP_STATE_REVOKED: |
agl@chromium.org | 77a9ad9 | 2012-03-20 15:14:27 +0000 | [diff] [blame] | 314 | cert_status = asn1.Explicit(1, asn1.GeneralizedTime("20100101060000Z")) |
agl@chromium.org | b5f388a | 2012-06-29 17:54:06 +0000 | [diff] [blame] | 315 | 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.org | 77a9ad9 | 2012-03-20 15:14:27 +0000 | [diff] [blame] | 318 | cert_status = asn1.Raw(asn1.TagAndLength(0x80 | 0, 0)) |
dadrian | 4ccf51c | 2016-07-20 15:36:58 -0700 | [diff] [blame] | 319 | elif ocsp_state == OCSP_STATE_MISMATCHED_SERIAL: |
| 320 | cert_status = asn1.Raw(asn1.TagAndLength(0x80 | 0, 0)) |
| 321 | serial -= 1 |
agl@chromium.org | b5f388a | 2012-06-29 17:54:06 +0000 | [diff] [blame] | 322 | else: |
| 323 | raise ValueError('Bad OCSP state: ' + str(ocsp_state)) |
agl@chromium.org | 77a9ad9 | 2012-03-20 15:14:27 +0000 | [diff] [blame] | 324 | |
dadrian | 4ccf51c | 2016-07-20 15:36:58 -0700 | [diff] [blame] | 325 | 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: |
dadrian | 9a07cc9 | 2016-07-22 10:44:37 -0700 | [diff] [blame] | 330 | thisUpdate = now - datetime.timedelta(days=1, weeks=1) |
dadrian | 4ccf51c | 2016-07-20 15:36:58 -0700 | [diff] [blame] | 331 | nextUpdate = thisUpdate + datetime.timedelta(weeks=1) |
| 332 | elif ocsp_date == OCSP_DATE_EARLY: |
dadrian | 9a07cc9 | 2016-07-22 10:44:37 -0700 | [diff] [blame] | 333 | thisUpdate = now + datetime.timedelta(days=1) |
dadrian | 4ccf51c | 2016-07-20 15:36:58 -0700 | [diff] [blame] | 334 | nextUpdate = thisUpdate + datetime.timedelta(weeks=1) |
| 335 | elif ocsp_date == OCSP_DATE_LONG: |
| 336 | thisUpdate = now - datetime.timedelta(days=365) |
dadrian | 9a07cc9 | 2016-07-22 10:44:37 -0700 | [diff] [blame] | 337 | nextUpdate = thisUpdate + datetime.timedelta(days=366) |
dadrian | 4ccf51c | 2016-07-20 15:36:58 -0700 | [diff] [blame] | 338 | 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 | |
| 361 | def 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.org | 77a9ad9 | 2012-03-20 15:14:27 +0000 | [diff] [blame] | 388 | basic_resp_data_der = asn1.ToDER(asn1.SEQUENCE([ |
| 389 | asn1.Explicit(2, issuer_key_hash), |
dadrian | 4ccf51c | 2016-07-20 15:36:58 -0700 | [diff] [blame] | 390 | asn1.GeneralizedTime(producedAt.strftime(GENERALIZED_TIME_FORMAT)), |
| 391 | asn1.SEQUENCE(single_responses), |
agl@chromium.org | 77a9ad9 | 2012-03-20 15:14:27 +0000 | [diff] [blame] | 392 | ])) |
| 393 | |
| 394 | basic_resp = asn1.SEQUENCE([ |
| 395 | asn1.Raw(basic_resp_data_der), |
| 396 | asn1.SEQUENCE([ |
rsleevi | 85d2177 | 2014-08-28 15:46:51 -0700 | [diff] [blame] | 397 | SHA256_WITH_RSA_ENCRYPTION, |
agl@chromium.org | 77a9ad9 | 2012-03-20 15:14:27 +0000 | [diff] [blame] | 398 | 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 | |
| 414 | def DERToPEM(der): |
| 415 | pem = '-----BEGIN CERTIFICATE-----\n' |
| 416 | pem += der.encode('base64') |
| 417 | pem += '-----END CERTIFICATE-----\n' |
| 418 | return pem |
| 419 | |
agl@chromium.org | b5f388a | 2012-06-29 17:54:06 +0000 | [diff] [blame] | 420 | # unauthorizedDER is an OCSPResponse with a status of 6: |
| 421 | # SEQUENCE { ENUM(6) } |
| 422 | unauthorizedDER = '30030a0106'.decode('hex') |
agl@chromium.org | 77a9ad9 | 2012-03-20 15:14:27 +0000 | [diff] [blame] | 423 | |
| 424 | def GenerateCertKeyAndOCSP(subject = "127.0.0.1", |
| 425 | ocsp_url = "http://127.0.0.1", |
dadrian | 4ccf51c | 2016-07-20 15:36:58 -0700 | [diff] [blame] | 426 | ocsp_states = None, |
| 427 | ocsp_dates = None, |
| 428 | ocsp_produced = OCSP_PRODUCED_VALID, |
agl@chromium.org | df77814 | 2013-07-31 21:57:28 +0000 | [diff] [blame] | 429 | serial = 0): |
agl@chromium.org | 77a9ad9 | 2012-03-20 15:14:27 +0000 | [diff] [blame] | 430 | '''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 | |
dadrian | 4ccf51c | 2016-07-20 15:36:58 -0700 | [diff] [blame] | 436 | 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.org | df77814 | 2013-07-31 21:57:28 +0000 | [diff] [blame] | 441 | if serial == 0: |
| 442 | serial = RandomNumber(16) |
mattm | 10ede84 | 2016-11-29 11:57:16 -0800 | [diff] [blame] | 443 | cert_der = MakeCertificate(ROOT_CN, bytes(subject), serial, LEAF_KEY, |
| 444 | ROOT_KEY, bytes(ocsp_url)) |
agl@chromium.org | 77a9ad9 | 2012-03-20 15:14:27 +0000 | [diff] [blame] | 445 | cert_pem = DERToPEM(cert_der) |
| 446 | |
| 447 | ocsp_der = None |
| 448 | if ocsp_url is not None: |
dadrian | 4ccf51c | 2016-07-20 15:36:58 -0700 | [diff] [blame] | 449 | if ocsp_states[0] == OCSP_STATE_UNAUTHORIZED: |
agl@chromium.org | b5f388a | 2012-06-29 17:54:06 +0000 | [diff] [blame] | 450 | ocsp_der = unauthorizedDER |
dadrian | 4ccf51c | 2016-07-20 15:36:58 -0700 | [diff] [blame] | 451 | elif ocsp_states[0] == OCSP_STATE_INVALID_RESPONSE: |
agl@chromium.org | b5f388a | 2012-06-29 17:54:06 +0000 | [diff] [blame] | 452 | ocsp_der = '3' |
dadrian | 4ccf51c | 2016-07-20 15:36:58 -0700 | [diff] [blame] | 453 | 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 | ]), |
mattm | 10ede84 | 2016-11-29 11:57:16 -0800 | [diff] [blame] | 466 | asn1.BitString(ROOT_KEY.Sign(invalid_data)), |
dadrian | 4ccf51c | 2016-07-20 15:36:58 -0700 | [diff] [blame] | 467 | ]) |
| 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.org | b5f388a | 2012-06-29 17:54:06 +0000 | [diff] [blame] | 476 | else: |
dadrian | 4ccf51c | 2016-07-20 15:36:58 -0700 | [diff] [blame] | 477 | ocsp_der = MakeOCSPResponse( |
mattm | 10ede84 | 2016-11-29 11:57:16 -0800 | [diff] [blame] | 478 | ROOT_CN, ROOT_KEY, serial, ocsp_states, ocsp_dates, ocsp_produced) |
agl@chromium.org | 77a9ad9 | 2012-03-20 15:14:27 +0000 | [diff] [blame] | 479 | |
mattm | 10ede84 | 2016-11-29 11:57:16 -0800 | [diff] [blame] | 480 | return (cert_pem + LEAF_KEY_PEM, ocsp_der) |
| 481 | |
| 482 | |
| 483 | def 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 |
mattm | adace8e | 2016-10-03 14:07:15 -0700 | [diff] [blame] | 503 | |
| 504 | |
| 505 | if __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 | |
mattm | 10ede84 | 2016-11-29 11:57:16 -0800 | [diff] [blame] | 514 | der_root = MakeCertificate(ROOT_CN, ROOT_CN, 1, ROOT_KEY, ROOT_KEY, |
| 515 | is_ca=True, path_len=1) |
mattm | adace8e | 2016-10-03 14:07:15 -0700 | [diff] [blame] | 516 | 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)) |