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

Add support for file inputs

Browse Source
This commit is contained in:
rotarydrone 2018-09-18 10:44:47 -04:00
parent 30d51db336
commit beb92ca506
1 changed files with 15 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
ssl_checker.py
View File

@ -223,8 +223,16 @@ def get_args():
"""Set argparse options."""
parser = ArgumentParser(prog='ssl_checker.py', add_help=False,
description="""Collects useful information about given host's SSL certificates.""")
parser.add_argument('-H', '--host', dest='hosts', nargs='*', required=True,
group = parser.add_mutually_exclusive_group()
group.add_argument('-H', '--host', dest='hosts', nargs='*', required=False,
help='Hosts as input separated by space')
group.add_argument('-f', '--host-file', dest='host_file', required=False,
help='Hosts as input from file')
parser.add_argument('-s', '--socks', dest='socks',
default=False, metavar='HOST:PORT',
help='Enable SOCKS proxy for connection')
@ -246,6 +254,12 @@ def get_args():
args = parser.parse_args()
# Get hosts from file if provided
if args.host_file:
hosts_file = open(args.host_file, 'r')
args.hosts = hosts_file.readlines()
hosts_file.close()
# Checks hosts list
if isinstance(args.hosts, list):
if len(args.hosts) == 0: