Certificate path builder for new certificate verification library

net/data/verify_certificate_chain_unittest/common.py

 #!/usr/bin/python
 # Copyright (c) 2015 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 """Set of helpers to generate signed X.509v3 certificates.
 
 This works by shelling out calls to the 'openssl req' and 'openssl ca'
 commands, and passing the appropriate command line flags and configuration file
 (.cnf).
@@ skipping 54 matching lines @@
   if path_id == 0:
     return name
 
   # Otherwise append the count to make it unique.
   return '%s_%d' % (name, path_id)
 
 
 class Certificate(object):
   """Helper for building an X.509 certificate."""
 
-  def __init__(self, name, cert_type, issuer):
+  def __init__(self, name, cert_type, issuer, key_from=None):
     # The name will be used for the subject's CN, and also as a component of
     # the temporary filenames to help with debugging.
     self.name = name
     self.path_id = GetUniquePathId(name)
 
+    # If specified, use they key from the given Certificate object instead of
+    # generating a new one.
+    self.key_from = key_from
+
     # The issuer is also a Certificate object. Passing |None| means it is a
     # self-signed certificate.
     self.issuer = issuer
     if issuer is None:
       self.issuer = self
 
     # The config contains all the OpenSSL options that will be passed via a
     # .cnf file. Set up defaults.
     self.config = openssl_conf.Config()
     self.init_config()
@@ skipping 69 matching lines @@
     return self.config.get_section('req_ext')
 
 
   def get_path(self, suffix):
     """Forms a path to an output file for this certificate, containing the
     indicated suffix. The certificate's name will be used as its basis."""
     return os.path.join(g_out_dir, '%s%s' % (self.path_id, suffix))
 
 
   def get_key_path(self):
+    if self.key_from is not None:
+      return self.key_from.get_key_path()
     return self.get_path('.key')
 
 
   def get_cert_path(self):
     return self.get_path('.pem')
 
 
   def get_serial_path(self):
     return self.get_path('.serial')
 
@@ skipping 164 matching lines @@
     section = self.config.get_section('crl_ext')
     section.set_property('authorityKeyIdentifier', 'keyid:always')
     section.set_property('authorityInfoAccess', '@issuer_info')
 
 
 def data_to_pem(block_header, block_data):
   return '-----BEGIN %s-----\n%s\n-----END %s-----\n' % (block_header,
           base64.b64encode(block_data), block_header)
 
 
-def write_test_file(description, chain, trusted_certs, utc_time, verify_result):
+def write_test_file(description, chain, trusted_certs, utc_time, verify_result,
+                    out_pem=None):
   """Writes a test file that contains all the inputs necessary to run a
   verification on a certificate chain"""
 
   # Prepend the script name that generated the file to the description.
   test_data = '[Created by: %s]\n\n%s\n' % (sys.argv[0], description)
 
   # Write the certificate chain to the output file.
   for cert in chain:
     test_data += '\n' + cert.get_cert_pem()
 
   # Write the trust store.
   for cert in trusted_certs:
     cert_data = cert.get_cert_pem()
     # Use a different block type in the .pem file.
     cert_data = cert_data.replace('CERTIFICATE', 'TRUSTED_CERTIFICATE')
     test_data += '\n' + cert_data
 
   test_data += '\n' + data_to_pem('TIME', utc_time)
 
   verify_result_string = 'SUCCESS' if verify_result else 'FAIL'
   test_data += '\n' + data_to_pem('VERIFY_RESULT', verify_result_string)
 
-  write_string_to_file(test_data, g_out_pem)
+  write_string_to_file(test_data, out_pem if out_pem else g_out_pem)
 
 
 def write_string_to_file(data, path):
   with open(path, 'w') as f:
     f.write(data)
 
 
 def init(invoking_script_path):
   """Creates an output directory to contain all the temporary files that may be
   created, as well as determining the path for the final output. These paths
@@ skipping 18 matching lines @@
   shutil.rmtree(g_out_dir, True)
   os.makedirs(g_out_dir)
 
   g_out_pem = os.path.join('%s.pem' % (out_name))
 
 
 def create_self_signed_root_certificate(name):
   return Certificate(name, TYPE_CA, None)
 
 
-def create_intermediary_certificate(name, issuer):
-  return Certificate(name, TYPE_CA, issuer)
+def create_intermediary_certificate(name, issuer, **kw_args):
+  return Certificate(name, TYPE_CA, issuer, **kw_args)
 
 
-def create_end_entity_certificate(name, issuer):
-  return Certificate(name, TYPE_END_ENTITY, issuer)
+def create_end_entity_certificate(name, issuer, **kw_args):
+  return Certificate(name, TYPE_END_ENTITY, issuer, **kw_args)
 
 init(sys.argv[0])