From 8f9932e56bd359881f62c658b8d003fe83b18bea Mon Sep 17 00:00:00 2001 From: pemontto Date: Thu, 11 Apr 2019 21:17:24 +1000 Subject: [PATCH] minor updates --- vulnwhisp/frameworks/nessus.py | 3 +++ vulnwhisp/frameworks/qualys_vuln.py | 7 +++++-- 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/vulnwhisp/frameworks/nessus.py b/vulnwhisp/frameworks/nessus.py index 127e658..9899de6 100755 --- a/vulnwhisp/frameworks/nessus.py +++ b/vulnwhisp/frameworks/nessus.py @@ -223,6 +223,9 @@ class NessusAPI(object): dataframe['cve'] = dataframe['cve'].str.upper() dataframe['protocol'] = dataframe['protocol'].str.lower() + # Copy asset to IP + dataframe['ip'] = dataframe['asset'] + # Map risk to a SEVERITY MAPPING value self.logger.debug('Mapping risk to severity number') dataframe['risk_number'] = dataframe['risk'].str.lower() diff --git a/vulnwhisp/frameworks/qualys_vuln.py b/vulnwhisp/frameworks/qualys_vuln.py index dace770..3eac54f 100644 --- a/vulnwhisp/frameworks/qualys_vuln.py +++ b/vulnwhisp/frameworks/qualys_vuln.py @@ -85,7 +85,8 @@ class qualysVulnScan: 'cvss3_base': 'cvss3', 'ip_status': 'state', 'os': 'operating_system', - 'qid': 'plugin_id' + 'qid': 'plugin_id', + 'title': 'plugin_name' } def __init__( @@ -198,7 +199,9 @@ class qualysVulnScan: .str.rstrip("/nan") .fillna("") ) - + + dataframe.drop('cvss_temporal_vector', axis=1, inplace=True) + # Convert Qualys severity to standardised risk number dataframe['risk_number'] = dataframe['severity'].astype(int)-1