#!/usr/bin/env python # -*- coding: utf-8 -*- from saml2.cert import OpenSSLWrapper __author__ = 'haho0032' cert_info_ca = { "cn": "test", "country_code": "zz", "state": "zz", "city": "zzzz", "organization": "Zzzzz", "organization_unit": "Zzzzz" } osw = OpenSSLWrapper() ca_cert, ca_key = osw.create_certificate(cert_info_ca, request=False, write_to_file=True, cert_dir="./")
def test_validate_cert_chains(self): cert_info_ca = { "cn": "qwerty", "country_code": "qw", "state": "qwerty", "city": "qwerty", "organization": "qwerty", "organization_unit": "qwerty" } cert_intermediate_1_info = { "cn": "intermediate_1", "country_code": "as", "state": "asdfgh", "city": "asdfgh", "organization": "asdfgh", "organization_unit": "asdfg" } cert_intermediate_2_info = { "cn": "intermediate_2", "country_code": "as", "state": "asdfgh", "city": "asdfgh", "organization": "asdfgh", "organization_unit": "asdfg" } cert_client_cert_info = { "cn": "intermediate_1", "country_code": "as", "state": "asdfgh", "city": "asdfgh", "organization": "asdfgh", "organization_unit": "asdfg" } osw = OpenSSLWrapper() ca_cert_str, ca_key_str = osw.create_certificate(cert_info_ca, request=False) req_cert_str, intermediate_1_key_str = osw.create_certificate( cert_intermediate_1_info, request=True) intermediate_cert_1_str = osw.create_cert_signed_certificate( ca_cert_str, ca_key_str, req_cert_str) req_cert_str, intermediate_2_key_str = osw.create_certificate( cert_intermediate_2_info, request=True) intermediate_cert_2_str = osw.create_cert_signed_certificate( intermediate_cert_1_str, intermediate_1_key_str, req_cert_str) req_cert_str, client_key_str = osw.create_certificate( cert_client_cert_info, request=True) client_cert_str = osw.create_cert_signed_certificate( intermediate_cert_2_str, intermediate_2_key_str, req_cert_str) cert_chain = [ intermediate_cert_2_str, intermediate_cert_1_str, ca_cert_str ] valid, mess = osw.verify_chain(cert_chain, client_cert_str) self.assertTrue(valid)
def test_validate_expire(self): cert_info_ca = { "cn": "qwerty", "country_code": "qw", "state": "qwerty", "city": "qwerty", "organization": "qwerty", "organization_unit": "qwerty" } cert_info = { "cn": "intermediate_1", "country_code": "as", "state": "asdfgh", "city": "asdfgh", "organization": "asdfgh", "organization_unit": "asdfg" } osw = OpenSSLWrapper() ca_cert_str, ca_key_str = osw.create_certificate(cert_info_ca, request=False) req_cert_str, req_key_str = osw.create_certificate(cert_info, request=True) cert_str = osw.create_cert_signed_certificate(ca_cert_str, ca_key_str, req_cert_str) valid, mess = osw.verify(ca_cert_str, cert_str) ca_cert_str, ca_key_str = osw.create_certificate(cert_info_ca, request=False, valid_from=1000, valid_to=100000) req_cert_str, req_key_str = osw.create_certificate(cert_info, request=True) cert_str = osw.create_cert_signed_certificate(ca_cert_str, ca_key_str, req_cert_str) valid, mess = osw.verify(ca_cert_str, cert_str) self.assertFalse(valid) ca_cert_str, ca_key_str = osw.create_certificate(cert_info_ca, request=False) req_cert_str, req_key_str = osw.create_certificate(cert_info, request=True) cert_str = osw.create_cert_signed_certificate(ca_cert_str, ca_key_str, req_cert_str, valid_from=1000, valid_to=100000) valid, mess = osw.verify(ca_cert_str, cert_str) self.assertFalse(valid) ca_cert_str, ca_key_str = osw.create_certificate(cert_info_ca, request=False, valid_from=0, valid_to=1) req_cert_str, req_key_str = osw.create_certificate(cert_info, request=True) cert_str = osw.create_cert_signed_certificate(ca_cert_str, ca_key_str, req_cert_str) time.sleep(2) valid, mess = osw.verify(ca_cert_str, cert_str) self.assertFalse(valid) ca_cert_str, ca_key_str = osw.create_certificate(cert_info_ca, request=False) req_cert_str, req_key_str = osw.create_certificate(cert_info, request=True) cert_str = osw.create_cert_signed_certificate(ca_cert_str, ca_key_str, req_cert_str, valid_from=0, valid_to=1) time.sleep(2) valid, mess = osw.verify(ca_cert_str, cert_str) self.assertFalse(valid)
def test_validate_with_root_cert(self): cert_info_ca = { "cn": "qwerty", "country_code": "qw", "state": "qwerty", "city": "qwerty", "organization": "qwerty", "organization_unit": "qwerty" } cert_info = { "cn": "asdfgh", "country_code": "as", "state": "asdfgh", "city": "asdfgh", "organization": "asdfgh", "organization_unit": "asdfg" } osw = OpenSSLWrapper() ca_cert, ca_key = osw.create_certificate( cert_info_ca, request=False, write_to_file=True, cert_dir=os.path.dirname(os.path.abspath(__file__)) + "/pki") req_cert_str, req_key_str = osw.create_certificate(cert_info, request=True) ca_cert_str = osw.read_str_from_file(ca_cert) ca_key_str = osw.read_str_from_file(ca_key) cert_str = osw.create_cert_signed_certificate(ca_cert_str, ca_key_str, req_cert_str) valid, mess = osw.verify(ca_cert_str, cert_str) self.assertTrue(valid) false_ca_cert, false_ca_key = osw.create_certificate( cert_info_ca, request=False, write_to_file=False) false_req_cert_str_1, false_req_key_str_1 = osw.create_certificate( cert_info_ca, request=True) false_cert_str_1 = osw.create_cert_signed_certificate( false_ca_cert, false_ca_key, false_req_cert_str_1) false_req_cert_str_2, false_req_key_str_2 = osw.create_certificate( cert_info, request=True) false_cert_str_2 = osw.create_cert_signed_certificate( false_ca_cert, false_ca_key, false_req_cert_str_2) valid, mess = osw.verify(false_ca_cert, cert_str) self.assertFalse(valid) valid, mess = osw.verify(false_ca_cert, false_cert_str_1) self.assertFalse(valid) valid, mess = osw.verify(ca_cert_str, false_cert_str_2) self.assertFalse(valid) if 'z' in cert_str: false_cert_str = cert_str.replace('z', 'x') valid, mess = osw.verify(ca_cert_str, false_cert_str) self.assertFalse(valid) remove(ca_cert) remove(ca_key)
def generateCert(cert_info=None, gen_jwks_cert=True): cert_info_ca = { "cn": "localhost", "country_code": "se", "state": "ac", "city": "Test", "organization": "Test org", "organization_unit": "Testers" } if cert_info is not None: cert_info_ca = cert_info osw = OpenSSLWrapper() sn = 1 try: sn = osw.read_str_from_file("sn.txt") if len(sn) > 0: sn = int(sn) sn += 1 else: sn = 1 except: pass ca_cert1, ca_key1 = osw.create_certificate(cert_info_ca, request=False, write_to_file=True, cert_dir="./httpsCert", sn=sn) sn += 1 ca_cert2, ca_key2 = osw.create_certificate(cert_info_ca, request=False, write_to_file=True, cert_dir="./idp_cert", sn=sn) sn += 1 ca_cert3, ca_key3 = osw.create_certificate(cert_info_ca, request=False, write_to_file=True, cert_dir="./sp_cert", sn=sn) sn += 1 ca_cert4 = None if gen_jwks_cert: ca_cert4, ca_key4 = osw.create_certificate(cert_info_ca, request=False, write_to_file=True, cert_dir="./opKeys", sn=sn, key_length=2048) sn += 1 osw.write_str_to_file("sn.txt", str(sn)) return sn, ca_cert4
def verify_encrypt_cert(cert_str): osw = OpenSSLWrapper() ca_cert_str = osw.read_str_from_file( full_path("root_cert/localhost.ca.crt")) valid, mess = osw.verify(ca_cert_str, cert_str) return valid