Added an argument for username and password, which takes precendece over nessus. Fixed #5

This commit is contained in:
Shaun McCullough
2017-11-27 10:02:53 -05:00
parent fedbb18bb2
commit 35b7093762
2 changed files with 19 additions and 4 deletions

View File

@ -18,12 +18,16 @@ def main():
help='Path of config file')
parser.add_argument('-v', '--verbose', dest='verbose', action='store_true', default=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('-p', '--password', dest='password', required=False, default=None, help='The NESSUS password')
args = parser.parse_args()
try:
vw = vulnWhisperer(config=args.config,
verbose=args.verbose)
verbose=args.verbose,
username=args.username,
password=args.password)
vw.whisper_nessus()
sys.exit(1)

View File

@ -16,13 +16,14 @@ import logging
class vulnWhisperer(object):
def __init__(self, config=None, db_name='report_tracker.db', purge=False, verbose=None, debug=False):
def __init__(self, config=None, db_name='report_tracker.db', purge=False, verbose=None, debug=False, username=None, password=None):
self.verbose = verbose
self.nessus_connect = False
self.develop = True
self.purge = purge
if config is not None:
try:
self.config = vwConfig(config_in=config)
@ -31,8 +32,18 @@ class vulnWhisperer(object):
if self.nessus_enabled:
self.nessus_hostname = self.config.get('nessus', 'hostname')
self.nessus_port = self.config.get('nessus', 'port')
self.nessus_username = self.config.get('nessus', 'username')
self.nessus_password = self.config.get('nessus', 'password')
if password:
self.nessus_password = password
else:
self.nessus_password = self.config.get('nessus', 'password')
if username:
self.nessus_username = username
else:
self.nessus_username = self.config.get('nessus', 'username')
self.nessus_writepath = self.config.get('nessus', 'write_path')
self.nessus_dbpath = self.config.get('nessus', 'db_path')
self.nessus_trash = self.config.getbool('nessus', 'trash')