diff --git a/index.py b/index.py index 606e59e..ed8d49b 100644 --- a/index.py +++ b/index.py @@ -54,8 +54,9 @@ def tlsa_validation(DOMAIN): conn = ssl.create_connection((DOMAIN , 443)) context = ssl.SSLContext(ssl.PROTOCOL_TLSv1_2) sock = context.wrap_socket(conn, server_hostname=DOMAIN) - cert = ssl.DER_cert_to_PEM_cert(sock.getpeercert(True)) - cert = cert.encode('ascii') + dercert = sock.getpeercert(True) + pemcert = ssl.DER_cert_to_PEM_cert(dercert) + cert = pemcert.encode('ascii') except: return(False) @@ -68,7 +69,7 @@ def tlsa_validation(DOMAIN): raise ValueError("selector type %d not recognized" % selector) if mtype == "0": - hexdata2 = hexdump(certdata) + certdata = dercert elif mtype == "1": hexdata2 = compute_hash(hashlib.sha256, certdata) elif mtype == "2":