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

Ensure empty fields are output as nulls

Browse Source
This commit is contained in:
pemontto
2019-04-18 11:33:25 +10:00
parent e3907940bc
commit a2e27d816b
1 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
vulnwhisp/vulnwhisp.py
View File

@ -9,6 +9,7 @@ from frameworks.qualys_vuln import qualysVulnScan
from frameworks.openvas import OpenVAS_API from frameworks.openvas import OpenVAS_API
from reporting.jira_api import JiraAPI from reporting.jira_api import JiraAPI
import pandas as pd import pandas as pd
import numpy as np
from lxml import objectify from lxml import objectify
import sys import sys
import os import os
@ -282,7 +283,8 @@ 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'].isnull(), 'cvss_severity'] = 'info' # df.loc[df['cvss'].isnull(), 'cvss_severity'] = 'info'
df.loc[df['cvss'] == 0, 'cvss3_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'
@ -293,13 +295,16 @@ class vulnWhispererBase(object):
# 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'].isnull(), 'cvss3_severity'] = 'info' # df.loc[df['cvss3'].isnull(), 'cvss3_severity'] = 'info'
df.loc[df['cvss3'] == 0, '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'
df.loc[(df['cvss3'] > 9) & (df['cvss3'].notnull()), 'cvss3_severity'] = 'critical' df.loc[(df['cvss3'] > 9) & (df['cvss3'].notnull()), 'cvss3_severity'] = 'critical'
df.fillna('', inplace=True) # Ensure empty strings are output as nulls
df.replace({'': np.nan}, inplace=True)
return df return df