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

fix cvss info severity

Browse Source
This commit is contained in:
pemontto
2019-04-17 16:31:23 +10:00
parent e8d0c71bfb
commit eea417a0d9
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
vulnwhisp/vulnwhisp.py
View File

@ -282,7 +282,7 @@ class vulnWhispererBase(object):
# Map CVSS to severity name # Map CVSS to severity name
df.loc[df['cvss'] == '', 'cvss'] = None df.loc[df['cvss'] == '', 'cvss'] = None
df['cvss'] = df['cvss'].astype('float') df['cvss'] = df['cvss'].astype('float')
df.loc[df['cvss'] == 0, 'cvss_severity'] = 'info' df.loc[df['cvss'].isnull(), 'cvss_severity'] = 'info'
df.loc[(df['cvss'] > 0) & (df['cvss'] < 3), 'cvss_severity'] = 'low' df.loc[(df['cvss'] > 0) & (df['cvss'] < 3), 'cvss_severity'] = 'low'
df.loc[(df['cvss'] >= 3) & (df['cvss'] < 6), 'cvss_severity'] = 'medium' df.loc[(df['cvss'] >= 3) & (df['cvss'] < 6), 'cvss_severity'] = 'medium'
df.loc[(df['cvss'] >= 6) & (df['cvss'] < 9), 'cvss_severity'] = 'high' df.loc[(df['cvss'] >= 6) & (df['cvss'] < 9), 'cvss_severity'] = 'high'
@ -291,9 +291,9 @@ class vulnWhispererBase(object):
if 'cvss3' in df: if 'cvss3' in df:
self.logger.info('Normalising CVSS3 severity') self.logger.info('Normalising CVSS3 severity')
# Map CVSS to severity name # Map CVSS to severity name
df.loc[df['cvss3'] =='', 'cvss3'] = None df.loc[df['cvss3'] == '', 'cvss3'] = None
df['cvss3'] = df['cvss3'].astype('float') df['cvss3'] = df['cvss3'].astype('float')
df.loc[df['cvss3'] == 0, 'cvss3_severity'] = 'info' df.loc[df['cvss3'].isnull(), 'cvss3_severity'] = 'info'
df.loc[(df['cvss3'] > 0) & (df['cvss3'] < 3), 'cvss3_severity'] = 'low' df.loc[(df['cvss3'] > 0) & (df['cvss3'] < 3), 'cvss3_severity'] = 'low'
df.loc[(df['cvss3'] >= 3) & (df['cvss3'] < 6), 'cvss3_severity'] = 'medium' df.loc[(df['cvss3'] >= 3) & (df['cvss3'] < 6), 'cvss3_severity'] = 'medium'
df.loc[(df['cvss3'] >= 6) & (df['cvss3'] < 9), 'cvss3_severity'] = 'high' df.loc[(df['cvss3'] >= 6) & (df['cvss3'] < 9), 'cvss3_severity'] = 'high'