Cyberdefense/VulnWhisperer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Put in a check to make sure that the config file exists. FIXES austin-taylor/VulnWhisperer#4

Browse Source
This commit is contained in:
Shaun McCullough
2017-12-08 00:25:15 -05:00
parent f8905e8c4b
commit fcd938b75a
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
bin/vuln_whisperer
View File

@ -6,16 +6,22 @@
from vulnwhisp.vulnwhisp import vulnWhisperer from vulnwhisp.vulnwhisp import vulnWhisperer
from vulnwhisp.utils.cli import bcolors from vulnwhisp.utils.cli import bcolors
import os
import argparse import argparse
import sys import sys
def isFileValid(parser, arg):
if not os.path.exists(arg):
parser.error("The file %s does not exist!" % arg)
else:
return arg
def main(): def main():
parser = argparse.ArgumentParser(description=""" VulnWhisperer is designed to create actionable data from\ parser = argparse.ArgumentParser(description=""" VulnWhisperer is designed to create actionable data from\
your vulnerability scans through aggregation of historical scans.""") your vulnerability scans through aggregation of historical scans.""")
parser.add_argument('-c', '--config', dest='config', required=False, default='frameworks.ini', parser.add_argument('-c', '--config', dest='config', required=False, default='frameworks.ini',
help='Path of config file') help='Path of config file', type=lambda x: isFileValid(parser, x))
parser.add_argument('-v', '--verbose', dest='verbose', action='store_true', default=True, parser.add_argument('-v', '--verbose', dest='verbose', action='store_true', default=True,
help='Prints status out to screen (defaults to True)') help='Prints status out to screen (defaults to True)')
parser.add_argument('-u', '--username', dest='username', required=False, default=None, help='The NESSUS username') parser.add_argument('-u', '--username', dest='username', required=False, default=None, help='The NESSUS username')