Merge pull request #8 from cybergoof/passwords-argument

Added an argument for username and password
This commit is contained in:
Austin Taylor
2017-11-28 22:18:50 -08:00
committed by GitHub
2 changed files with 19 additions and 4 deletions

View File

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

View File

@ -16,13 +16,14 @@ import logging
class vulnWhisperer(object): 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.verbose = verbose
self.nessus_connect = False self.nessus_connect = False
self.develop = True self.develop = True
self.purge = purge self.purge = purge
if config is not None: if config is not None:
try: try:
self.config = vwConfig(config_in=config) self.config = vwConfig(config_in=config)
@ -31,8 +32,18 @@ class vulnWhisperer(object):
if self.nessus_enabled: if self.nessus_enabled:
self.nessus_hostname = self.config.get('nessus', 'hostname') self.nessus_hostname = self.config.get('nessus', 'hostname')
self.nessus_port = self.config.get('nessus', 'port') 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_writepath = self.config.get('nessus', 'write_path')
self.nessus_dbpath = self.config.get('nessus', 'db_path') self.nessus_dbpath = self.config.get('nessus', 'db_path')
self.nessus_trash = self.config.getbool('nessus', 'trash') self.nessus_trash = self.config.getbool('nessus', 'trash')