commit 82fed08a126c97d67bf6157d34f7017ebdf92e21 Author: brent s Date: Tue Dec 18 04:14:17 2018 -0500 initial commit diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..e16177d --- /dev/null +++ b/.gitignore @@ -0,0 +1,18 @@ +*/ +.*.swp +*.pkg.tar.xz +src/ +pkg/ +*.tar +*.tar.bz2 +*.tar.xz +*.tar.gz +*.tgz +*.txz +*.tbz +*.tbz2 +*.zip +*.run +*.7z +*.rar +*.deb diff --git a/certparser.py b/certparser.py new file mode 100755 index 0000000..7372595 --- /dev/null +++ b/certparser.py @@ -0,0 +1,474 @@ +#!/usr/bin/env python3 + +# stdlib +import argparse +import collections +import copy +import datetime +import hashlib +import importlib +import ipaddress +import json +import os +import pprint +import re +import shutil +import socket +import ssl +from urllib import parse +# PyPi/PIP +# These are handled automagically. +# If you'd rather install them via your distro's package manager (YOU SHOULD), +# then install them first then run this script. +# Otherwise you'll have to use pip to remove them. +thrd_prty = {'OpenSSL': 'pyOpenSSL', + #'pyasn1': 'pyasn1', + #'jinja2': 'Jinja2', + 'validators': 'validators'} + +cols = shutil.get_terminal_size((80, 20)).columns + +for mod in thrd_prty: + try: + globals()[mod] = importlib.import_module(mod) + except ImportError: + import pip + pip.main(['install', '--quiet', '--quiet', '--quiet', + '--user', thrd_prty[mod]]) + globals()[mod] = importlib.import_module(mod) + +class CertParse(object): + def __init__(self, target, port = 443, force = None, cert_type = 'pem', + json_fmt = False, starttls = False, extensions = False, + alt_names = False): + self.target = target + self.port = port + self.force_type = force + self.cert_type = cert_type + self.starttls = starttls + self.json_fmt = json_fmt + self.extensions = extensions + self.alt_names = alt_names + self.cert = None + self.certinfo = None + self.get_type() + + def getCert(self): + if self.cert_type.lower() == 'pem': + self.cert_type = OpenSSL.crypto.FILETYPE_PEM + elif self.cert_type.lower() == 'asn1': + self.cert_type = OpenSSL.crypto.FILETYPE_ASN1 + else: + raise ValueError(('{0} is not a valid cert type; must be either ' + + '"pem" or "asn1"').format(self.cert_type)) + if not self.force_type in ('url', 'domain', 'ip'): + with open(self.target, 'rb') as f: + self.cert = OpenSSL.crypto.load_certificate(self.cert_type, + f.read()) + else: + _cert = ssl.get_server_certificate((self.target, self.port)) + self.cert = OpenSSL.crypto.load_certificate(self.cert_type, + _cert) + return() + + def parseCert(self): + certinfo = collections.OrderedDict() + timefmt = '%Y%m%d%H%M%SZ' + certinfo['Subject'] = self.parse_name(self.cert.get_subject().\ + get_components()) + certinfo['EXPIRED'] = self.cert.has_expired() + certinfo['Issuer'] = self.parse_name(self.cert.get_issuer().\ + get_components()) + certinfo['Issued'] = str(datetime.datetime.strptime( + self.cert.get_notBefore().decode('utf-8'), + timefmt)) + certinfo['Expires'] = str(datetime.datetime.strptime( + self.cert.get_notAfter().decode('utf-8'), + timefmt)) + if self.extensions: + certinfo['Extensions'] = self.parse_ext() + elif self.alt_names: + certinfo['SANs'] = self.parse_ext_san_only() + certinfo['Pubkey'] = self.get_pubkey() + certinfo['Serial'] = int(self.cert.get_serial_number()) + certinfo['Signature Algorithm'] = self.cert.get_signature_algorithm().\ + decode('utf-8') + certinfo['Version'] = self.cert.get_version() + certinfo['Subject Name Hash'] = self.cert.subject_name_hash() + certinfo['Fingerprints'] = self.gen_hashes() + self.certinfo = certinfo + return() + + def print(self, json_fmt = None): + if json_fmt is None: + json_fmt = self.json_fmt + if json_fmt: + output = json.dumps(self.certinfo, indent = 4) + else: + output = self.certinfo + if __name__ == '__main__': + if not json_fmt: + pprint.pprint(output, compact = False, width = cols) + else: + print(output) + return() + return(output) + + def get_pubkey(self): + pubkey = {} + key = self.cert.get_pubkey() + pubkey['Bit Length'] = key.bits() + # I wish there was a more comfortable way of comparing these. + if key.type() == OpenSSL.crypto.TYPE_RSA: + pubkey['Algorithm'] = 'RSA' + elif key.type() == OpenSSL.crypto.TYPE_DSA: + pubkey['Algorithm'] = 'DSA' + return(pubkey) + + def gen_hashes(self): + hashes = {} + # Note: MD2 is *so old* that they aren't even + # *supported in python 3*. + # If we NEED to implement, https://urchin.earth.li/~twic/md2.py + fpt_types = sorted([i.lower() for i in ['md2', 'md5', 'sha1', 'mdc2', + 'ripemd160', 'blake2b512', + 'blake2s256', 'sha224', + 'sha256', 'sha384', 'sha512']]) + supported_types = sorted([i.lower() for i in \ + list(hashlib.algorithms_available)]) + cert_hash_types = [i for i in fpt_types if i in supported_types] + for h in cert_hash_types: + hashes[h.upper()] = self.cert.digest(h).decode('utf-8') + return(hashes) + + def parse_name(self, item): + component_map = {'C': 'Country', + 'countryName': 'Country', + 'ST': 'State/Province', + 'stateOrProvinceName': 'State/Province', + 'L': 'Locality/City/Town/Region', + 'localityName': 'Locality/City/Town/Region', + 'O': 'Organization', + 'organizationName': 'Organization', + 'OU': 'Department/Team/Organization Unit', + 'organizationalUnitName': ('Department/Team/' + + 'Organization Unit'), + 'CN': 'Common name', + 'commonName': 'Common name', + 'emailAddress': 'eMail Address'} + info = {} + for c in item: + item = c[0].decode('utf-8') + value = c[1].decode('utf-8') + if item in component_map.keys(): + info[component_map[item]] = value + else: + info[item] = value + return(info) + + def parse_ext_san_only(self): + SANs = [] + for idx in range(0, self.cert.get_extension_count()): + ext = self.cert.get_extension(idx) + name = ext.get_short_name().decode('utf-8').lower() + x = str(ext).strip() + if name == 'subjectaltname': + val_lst = [i.strip() for i in x.split(',')] + for v in val_lst: + parsed_val = re.sub('^\s*DNS:\s*(.*)', '\g<1>', v) + if parsed_val not in ('\n', ''): + SANs.append(parsed_val.lower()) + return(SANs) + + def parse_ext(self): + exts = {} + for idx in range(0, self.cert.get_extension_count()): + ext = self.cert.get_extension(idx) + keyname = ext.get_short_name().decode('utf-8') + value_str = str(ext).strip() + # These should be split into lists by commas. + if keyname in ('subjectAltName', 'keyUsage', 'extendedKeyUsage', + 'basicConstraints'): + val_lst = [i.strip() for i in value_str.split(',')] + value_str = [] + for v in val_lst: + parsed_val = re.sub('^\s*DNS:\s*(.*)', '\g<1>', v) + if parsed_val not in ('\n', ''): + value_str.append(parsed_val) + # These should be split into lists by lines. + elif keyname in ('certificatePolicies', 'ct_precert_scts', + 'authorityInfoAccess'): + val_lst = [i.strip() for i in value_str.splitlines()] + value_str = [] + for v in val_lst: + value_str.append(v) + exts[keyname] = value_str + # These are split FURTHER into dicts but require unique... massaging. + # authorityInfoAccess + if 'authorityInfoAccess' in exts.keys(): + _tmp = copy.deepcopy(exts['authorityInfoAccess']) + exts['authorityInfoAccess'] = {} + for i in _tmp: + x = [n.strip() for n in i.split('-', 1)] + y = [n.strip() for n in x[1].split(':', 1)] + exts['authorityInfoAccess'][x[0]] = {y[0]: y[1]} + # authorityKeyIdentifier + if 'authorityKeyIdentifier' in exts.keys(): + _tmp = copy.deepcopy(exts['authorityKeyIdentifier']) + exts['authorityKeyIdentifier'] = {_tmp.split(':', 1)[0]: + _tmp.split(':', 1)[1]} + # basicConstraints + if 'basicConstraints' in exts.keys(): + _tmp = copy.deepcopy(exts['basicConstraints']) + exts['basicConstraints'] = {} + for i in _tmp: + x = [n.strip() for n in i.split(':', 1)] + if len(x) >= 1: + if x[1].lower() in ('true', 'false'): + x[1] = (x[1].lower() == 'true') + exts['basicConstraints'][x[0]] = x[1] + else: + exts['basicConstraints'][x[0]] = True + # certificatePolicies + # What a mess. + if 'certificatePolicies' in exts.keys(): + _tmp = copy.deepcopy(exts['certificatePolicies']) + exts['certificatePolicies'] = {} + last_key = None + for i in [n.strip() for n in _tmp]: + l = [y for y in i.split(':', 1) if y not in ('', None)] + if len(l) > 1: + # It MAY be a key:value. + if re.search('^\s+', l[1]): + val = l[1].strip() + if last_key == 'Policy': + if not isinstance(exts['certificatePolicies']\ + [last_key], + list): + exts['certificatePolicies'][last_key] = [ + exts['certificatePolicies'][last_key]] + exts['certificatePolicies'][last_key].append(val) + # I can't seem to get CPS as a separate dict. + # Patches welcome. + # Also, are CPS and User Notice *subitems* of Policy + # items? + elif last_key not in ('User Notice', 'CPS'): + # It's a value. + last_key = l[0].strip() + exts['certificatePolicies'][last_key] = val + else: + k = l[0].strip() + exts['certificatePolicies'][last_key][k] = val + else: + # Standalone key line + last_key = l[0].strip() + exts['certificatePolicies'][last_key] = {} + # ct_precert_scts + # another mess. a much. much, bigger mess. + if 'ct_precert_scts' in exts.keys(): + _tmp = copy.deepcopy(exts['ct_precert_scts']) + exts['ct_precert_scts'] = {} + last_key = None + last_sub_key = None + cnt = 0 + for i in [n.strip() for n in _tmp]: + l = [y for y in i.split(':', 1) if y not in ('', None)] + if len(l) > 1: + # Is it a line continuation (of a hex value)? + if ((re.search('^[0-9A-Z]{2}$', l[0])) and + (re.search('^[0-9A-Z:]*:?$', ':'.join(l)))): + exts['ct_precert_scts'][last_key][cnt]\ + [last_sub_key] += ':'.join(l) + continue + # It MAY be a key:value. + if re.search('^\s+', l[1]) and ( + last_key != + 'Signed Certificate Timestamp'): + # It's a value. + last_key = l[0].strip() + val = l[1].strip() + if val.lower() == 'none': + val = None + exts['ct_precert_scts'][last_key] = val + elif re.search('^\s+', l[1]): + last_sub_key = l[0].strip() + val = l[1].strip() + if val.lower() == 'none': + val = None + if last_sub_key == 'Signature': + val += ' ' + exts['ct_precert_scts'][last_key][cnt]\ + [last_sub_key] = val + else: + # Standalone key line + last_key = l[0].strip() + if last_key == 'Signed Certificate Timestamp': + if last_key not in exts['ct_precert_scts'].keys(): + exts['ct_precert_scts'][last_key] = [{}] + else: + exts['ct_precert_scts'][last_key].append({}) + cnt += 1 + # some laaaast bit of cleanup... + if 'Signed Certificate Timestamp' in exts['ct_precert_scts']: + for i in exts['ct_precert_scts']\ + ['Signed Certificate Timestamp']: + if 'Signature' in i.keys(): + d = i['Signature'].split() + i['Signature'] = {d[0]: d[1]} + return(exts) + + def get_domain_from_url(self, url): + orig_url = url + # Needed in case a URL is passed with no http:// or https://, etc. + url = re.sub('^((ht|f)tps?://)*', + 'https://', + url, + re.IGNORECASE).lower() + if not self.validURL(url): + raise ValueError(('{0} is not a valid URL').format(orig_url)) + domain = parse.urlparse(url).netloc + return(domain) + + def validIP(self, ip): + is_valid = False + try: + ipaddress.ip_address(self.target) + is_valid = True + except ValueError: + pass + return(is_valid) + + def validDomain(self, domain): + is_valid = False + if not isinstance(validators.domain(domain), + validators.utils.ValidationFailure): + is_valid = True + return(is_valid) + + def validURL(self, url): + is_valid = False + if not isinstance(validators.url(url), + validators.utils.ValidationFailure): + is_valid = True + return(is_valid) + + def validPath(self, path): + is_valid = False + if os.path.isfile(path): + is_valid = True + return(is_valid) + + def get_type(self): + if self.force_type: + # Just run the validator and some cleanup. + if self.force_type == 'url': + self.target = self.get_domain_from_url(self.target) + chk = self.validURL(self.target) + if chk: + self.force_type = 'domain' + elif self.force_type == 'ip': + chk = self.validIP(self.target) + elif self.force_type == 'domain': + chk = self.validDomain(self.target) + elif self.force_type == 'file': + self.target = os.path.abspath(os.path.expanduser(self.target)) + chk = self.validPath(self.target) + if not chk: + raise TypeError(('{0} does not appear to be a valid ' + + 'instance of type {1}'.format(self.target, + self.force_type) + )) + if self.force_type in ('url', 'domain', 'ip'): + self.remote = True + else: + self.remote = False + return() + # Is it an IP address? + if self.validIP(self.target): + self.force_type = 'ip' + return() + # Is it a filepath? + fpath = os.path.abspath(os.path.expanduser(self.target)) + if self.validPath(fpath): + self.target = fpath + self.force_type = 'file' + return() + # Is it a domain? + if self.validDomain(self.target): + self.force_type = 'domain' + return() + # Lastly, is it a URL? + if self.validURL(self.target): + domain = self.get_domain_from_url(self.target) + if self.validDomain(domain): + self.target = domain + self.force_type = 'domain' + if not self.force_type: # We couldn't detect it + raise RuntimeError(('Automatic type detection of {0} requested ' + + 'but we could not determine what type of ' + + 'resource it is')) + return() + +def parseArgs(): + args = argparse.ArgumentParser() + args.add_argument('-e', '--extensions', + dest = 'extensions', + action = 'store_true', + help = ('If specified, include ALL extension info ' + + '(this DRASTICALLY increases the output. You ' + + 'have been warned)')) + args.add_argument('-a', '--alt-names', + dest = 'alt_names', + action = 'store_true', + help = ('If specified, ONLY include the SAN (Subject ' + + 'Alt Name) extension. This is highly ' + + 'recommended over -e/--extensions. Ignored if ' + + '-e/--extensions is set (as the SANs are ' + + 'included in that)')) + args.add_argument('-j','--json', + dest = 'json_fmt', + action = 'store_true', + help = ('If specified, return the results in JSON')) + args.add_argument('-f', '--force', + choices = ['url', 'domain', 'ip', 'file'], + default = None, + help = ('If specified, force the TARGET to be parsed ' + + 'as the given type')) + args.add_argument('-p', '--port', + dest = 'port', + type = int, + default = 443, + help = ('Use a port other than 443 (only used for ' + + 'URL/domain/IP address targets)')) + args.add_argument('-t', '--cert-type', + dest = 'cert_type', + default = 'pem', + choices = ['pem', 'asn1'], + help = ('The type of certificate (only used for ' + 'file targets). Note that "DER"-encoded ' + + 'certificates should use "asn1". The default ' + + 'is pem')) +# TODO: I think the starttls process depends on the protocol? If so, this... +# won't be feasible. +# args.add_argument('-s', '--starttls', +# dest = 'starttls', +# action = 'store_true', +# help = ('If specified, initiate STARTTLS on the ' + +# 'target instead of pure SSL/TLS')) + args.add_argument('TARGET', + help = ('The target to gather cert info for. Can be a ' + + 'filepath (to the certificate, not key etc.), ' + + 'a URL/domain, or IP address')) + return(args) + +def main(): + args = vars(parseArgs().parse_args()) + args['target'] = copy.deepcopy(args['TARGET']) + del(args['TARGET']) + p = CertParse(**args) + p.getCert() + p.parseCert() + p.print() + +if __name__ == '__main__': + main()