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="/Users/haho0032/Develop/openSSL/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 generate_cert(self):
sn = uuid.uuid4().urn
cert_info = {
"cn": "localhost",
"country_code": "se",
"state": "ac",
"city": "Umea",
"organization": "ITS",
"organization_unit": "DIRG"
}
osw = OpenSSLWrapper()
ca_cert_str = osw.read_str_from_file(self.base_dir + "/root_cert/localhost.ca.crt")
ca_key_str = osw.read_str_from_file(self.base_dir + "/root_cert/localhost.ca.key")
req_cert_str, req_key_str = osw.create_certificate(cert_info, request=True, sn=sn, key_length=2048)
cert_str = osw.create_cert_signed_certificate(ca_cert_str, ca_key_str, req_cert_str)
return cert_str, req_key_str
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 generate_cert():
sn = uuid.uuid4().urn
cert_info = {
"cn": "localhost",
"country_code": "se",
"state": "ac",
"city": "Umea",
"organization": "ITS",
"organization_unit": "DIRG"
}
osw = OpenSSLWrapper()
ca_cert_str = osw.read_str_from_file(
full_path("root_cert/localhost.ca.crt"))
ca_key_str = osw.read_str_from_file(
full_path("root_cert/localhost.ca.key"))
req_cert_str, req_key_str = osw.create_certificate(cert_info,
request=True,
sn=sn,
key_length=2048)
cert_str = osw.create_cert_signed_certificate(ca_cert_str, ca_key_str,
req_cert_str)
return cert_str, req_key_str
def generatePublicKey(cert_file):
osw = OpenSSLWrapper()
cert_str = osw.read_str_from_file(cert_file)
cert = crypto.load_certificate(crypto.FILETYPE_PEM, cert_str)
pub_key = cert.get_pubkey()
src = crypto.dump_privatekey(crypto.FILETYPE_ASN1, pub_key)
pub_der = DerSequence()
pub_der.decode(src)
pub_key_rsa = RSA.construct((long(pub_der._seq[1]), long(pub_der._seq[2])))
pemSeq = DerSequence()
pemSeq[:] = [pub_key_rsa.key.n, pub_key_rsa.key.e]
s = pub_key_str = b64encode(pemSeq.encode())
pem_src = '-----BEGIN RSA PUBLIC KEY-----\n'
while True:
pem_src += s[:64] + '\n'
s = s[64:]
if s == '':
break
pem_src += '-----END RSA PUBLIC KEY-----'
jwks = {
"keys": [{
"use": "enc",
"e": "AQAB",
"kty": "RSA",
"n": pub_key_str
}, {
"use": "sig",
"e": "AQAB",
"kty": "RSA",
"n": pub_key_str
}]
}
jwks_str = json.dumps(jwks)
osw.write_str_to_file("./static/jwks.json", jwks_str)
def generateCert(cert_info=None):
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, ca_key4 = osw.create_certificate(cert_info_ca, request=False, write_to_file=True,
cert_dir="./opKeys", sn=sn, key_length=2048)
cert_str = osw.read_str_from_file(ca_cert4)
cert = crypto.load_certificate(crypto.FILETYPE_PEM, cert_str)
pub_key = cert.get_pubkey()
src = crypto.dump_privatekey(crypto.FILETYPE_ASN1, pub_key)
pub_der = DerSequence()
pub_der.decode(src)
pub_key_rsa = RSA.construct((long(pub_der._seq[1]), long(pub_der._seq[2])))
pemSeq = DerSequence()
pemSeq[:] = [pub_key_rsa.key.n, pub_key_rsa.key.e]
s = pub_key_str = b64encode(pemSeq.encode())
pem_src = '-----BEGIN RSA PUBLIC KEY-----\n'
while True:
pem_src += s[:64] + '\n'
s = s[64:]
if s == '':
break
pem_src += '-----END RSA PUBLIC KEY-----'
jwks = {"keys": [{"use": "enc", "e": "AQAB", "kty": "RSA", "n": pub_key_str},
{"use": "sig", "e": "AQAB", "kty": "RSA", "n": pub_key_str}]}
jwks_str = json.dumps(jwks)
osw.write_str_to_file("./static/jwks.json", jwks_str)
sn += 1
osw.write_str_to_file("sn.txt", str(sn))
return sn
def verify_encrypt_cert(cert_str):
osw = OpenSSLWrapper()
ca_cert_str = osw.read_str_from_file("root_cert/localhost.ca.crt")
valid, mess = osw.verify(ca_cert_str, cert_str)
return 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 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
