michael@0: #!/usr/bin/python
michael@0: # -*- Mode: python; c-basic-offset: 4; indent-tabs-mode: nil; tab-width: 40 -*-
michael@0: # vim: set filetype=python:
michael@0: #
michael@0: # This Source Code Form is subject to the terms of the Mozilla Public
michael@0: # License, v. 2.0. If a copy of the MPL was not distributed with this
michael@0: # file, You can obtain one at http://mozilla.org/MPL/2.0/.
michael@0: 
michael@0: import tempfile, os, sys
michael@0: import random
michael@0: import pexpect
michael@0: import subprocess
michael@0: import shutil
michael@0: 
michael@0: libpath = os.path.abspath('../psm_common_py')
michael@0: 
michael@0: sys.path.append(libpath)
michael@0: 
michael@0: import CertUtils
michael@0: 
michael@0: srcdir = os.getcwd()
michael@0: db = tempfile.mkdtemp()
michael@0: 
michael@0: CA_basic_constraints = "basicConstraints = critical, CA:TRUE\n"
michael@0: EE_basic_constraints = "basicConstraints = CA:FALSE\n"
michael@0: 
michael@0: CA_full_ku = ("keyUsage = digitalSignature, nonRepudiation, keyEncipherment, " +
michael@0:               "dataEncipherment, keyAgreement, keyCertSign, cRLSign\n")
michael@0: 
michael@0: CA_eku = ("extendedKeyUsage = critical, serverAuth, clientAuth, " +
michael@0:           "emailProtection, codeSigning\n")
michael@0: 
michael@0: authority_key_ident = "authorityKeyIdentifier = keyid, issuer\n"
michael@0: subject_key_ident = "subjectKeyIdentifier = hash\n"
michael@0: 
michael@0: 
michael@0: 
michael@0: def generate_certs():
michael@0:     key_type = 'rsa'
michael@0:     ca_ext = CA_basic_constraints + CA_full_ku + subject_key_ident + CA_eku;
michael@0:     ee_ext_text = (EE_basic_constraints + authority_key_ident)
michael@0:     [ca_key, ca_cert] = CertUtils.generate_cert_generic(db,
michael@0:                                                         srcdir,
michael@0:                                                         1,
michael@0:                                                         key_type,
michael@0:                                                         'ca',
michael@0:                                                         ca_ext)
michael@0: 
michael@0:     [int_key, int_cert] = CertUtils.generate_cert_generic(db,
michael@0:                                     srcdir,
michael@0:                                     103,
michael@0:                                     key_type,
michael@0:                                     'int',
michael@0:                                     ca_ext,
michael@0:                                     ca_key,
michael@0:                                     ca_cert)
michael@0: 
michael@0:     #now the ee
michael@0:     CertUtils.generate_cert_generic(db,
michael@0:                                     srcdir,
michael@0:                                     100,
michael@0:                                     key_type,
michael@0:                                     'ee',
michael@0:                                     ee_ext_text,
michael@0:                                     int_key,
michael@0:                                     int_cert)
michael@0: 
michael@0: 
michael@0: 
michael@0: 
michael@0: 
michael@0: generate_certs()