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

Only set IP if it's an IP

Browse Source
This commit is contained in:
pemontto
2019-05-08 17:51:05 +01:00
parent 488bd09dad
commit f600d0b88c
1 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
vulnwhisp/frameworks/nessus.py
View File

@ -206,10 +206,6 @@ class NessusAPI(object):
self.logger.debug('Dropping redundant tenable fields') self.logger.debug('Dropping redundant tenable fields')
df.drop('CVSS', axis=1, inplace=True, errors='ignore') df.drop('CVSS', axis=1, inplace=True, errors='ignore')
if self.profile == 'nessus':
# Set IP from Host field
df['ip'] = df['Host']
# Lowercase and map fields from COLUMN_MAPPING # Lowercase and map fields from COLUMN_MAPPING
df.columns = [x.lower() for x in df.columns] df.columns = [x.lower() for x in df.columns]
df.rename(columns=self.COLUMN_MAPPING, inplace=True) df.rename(columns=self.COLUMN_MAPPING, inplace=True)
@ -222,6 +218,14 @@ class NessusAPI(object):
df.fillna('', inplace=True) df.fillna('', inplace=True)
if self.profile == 'nessus':
# Set IP from asset field
df.loc[
df["asset"].str.match("^\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}$"), "ip"
] = df.loc[
df["asset"].str.match("^\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}$"), "asset"
]
# upper/lowercase fields # upper/lowercase fields
self.logger.debug('Changing case of fields') self.logger.debug('Changing case of fields')
df['cve'] = df['cve'].str.upper() df['cve'] = df['cve'].str.upper()