mali/ssl-checker
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix CSV output to write into rows instead of columns

Browse Source
This commit is contained in:
rotarydrone 2018-09-18 12:19:09 -04:00
parent 85d461942f
commit 3d66f1ada8
1 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
ssl_checker.py
View File

@ -6,6 +6,7 @@ from argparse import ArgumentParser, SUPPRESS
from datetime import datetime from datetime import datetime
from ssl import PROTOCOL_TLSv1 from ssl import PROTOCOL_TLSv1
from time import sleep from time import sleep
from csv import DictWriter
try: try:
from OpenSSL import SSL, crypto from OpenSSL import SSL, crypto
@ -98,6 +99,8 @@ def get_cert_sans(x509cert):
ext = x509cert.get_extension(i) ext = x509cert.get_extension(i)
if 'subjectAltName' in str(ext.get_short_name()): if 'subjectAltName' in str(ext.get_short_name()):
san = ext.__str__() san = ext.__str__()
# replace commas to not break csv output
san = san.replace(',', ';')
return san return san
def get_cert_info(host, cert): def get_cert_info(host, cert):
@ -160,7 +163,7 @@ def print_status(host, context, analyze=False):
print('\t\tExpired: {}'.format(context[host]['cert_exp'])) print('\t\tExpired: {}'.format(context[host]['cert_exp']))
print('\t\tCertificate SANs: ') print('\t\tCertificate SANs: ')
for san in context[host]['cert_sans'].split(','): print('\t\t\t{}'.format(san)) for san in context[host]['cert_sans'].split(';'): print('\t\t\t{}'.format(san))
@ -216,15 +219,14 @@ def show_result(user_args):
else: else:
print(context) print(context)
def export_csv(context, filename): def export_csv(context, filename):
""" Export all context results to CSV file.""" """ Export all context results to CSV file."""
# prepend dict keys to write column headers
with open(filename, 'w') as csv_file: with open(filename, 'w') as csv_file:
csv_writer = DictWriter(csv_file, context.items()[0][1].keys())
csv_writer.writeheader()
for host in context.keys(): for host in context.keys():
csv_file.write('{}\n'.format(host)) csv_writer.writerow(context[host])
for key, value in context[host].items():
csv_file.write('{},{}\n'.format(key, value))
def filter_hostname(host): def filter_hostname(host):
"""Remove unused characters and split by address and port.""" """Remove unused characters and split by address and port."""