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

refactor qualys cvss extraction

Browse Source
This commit is contained in:
pemontto
2019-04-24 09:20:27 +10:00
parent 24cf2ca623
commit d41011a5ed
1 changed files with 5 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
vulnwhisp/frameworks/qualys_vuln.py
View File

@ -11,7 +11,6 @@ import pandas as pd
import qualysapi import qualysapi
class qualysWhisperAPI(object): class qualysWhisperAPI(object):
SCANS = 'api/2.0/fo/scan' SCANS = 'api/2.0/fo/scan'
@ -165,29 +164,11 @@ class qualysVulnScan:
# Contruct the CVSS vector # Contruct the CVSS vector
self.logger.info('Extracting CVSS components') self.logger.info('Extracting CVSS components')
df['cvss_vector'] = ( df['cvss_vector'] = df['cvss_base'].str.extract('\((.*)\)', expand=False)
df.loc[df['cvss_base'].str.contains(' \('), 'cvss_base'] df['cvss_base'] = df['cvss_base'].str.extract('^([^ ]+)', expand=False)
.str.split() df['cvss_temporal_vector'] = df['cvss_temporal'].str.extract('\((.*)\)', expand=False)
.apply(lambda x: x[1]) df['cvss_temporal'] = df['cvss_temporal'].str.extract('^([^ ]+)', expand=False)
.str.strip('()')
)
df['cvss_base'] = (
df.loc[df['cvss_base'].str.contains(' \('), 'cvss_base']
.str.split()
.apply(lambda x: x[0])
)
df['cvss_temporal_vector'] = (
df.loc[df['cvss_temporal'].str.contains(' \('), 'cvss_temporal']
.str.split()
.apply(lambda x: x[1])
.str.strip('()')
)
df['cvss_temporal'] = (
df.loc[df['cvss_temporal'].str.contains(' \('), 'cvss_temporal']
.str.split()
.apply(lambda x: x[0])
)
# Convert Qualys severity to standardised risk number # Convert Qualys severity to standardised risk number
df['risk_number'] = df['severity'].astype(int)-1 df['risk_number'] = df['severity'].astype(int)-1
@ -195,4 +176,4 @@ class qualysVulnScan:
df.fillna('', inplace=True) df.fillna('', inplace=True)
return df return df