Skip to content

Commit d771038

Browse files
bluesentinelsecMichael Long
andauthored
Extend vulnerability severity providers (#98)
* Add severity providers: GHSA, GitLab * Add severity providers: GHSA, GitLab * Add REDHAT_CVE and UBUNTU_CVE providers * rename GHSA to GITHUB --------- Co-authored-by: Michael Long <mlongii@amazon.com>
1 parent 0ef860b commit d771038

File tree

1 file changed

+24
-2
lines changed

1 file changed

+24
-2
lines changed

entrypoint/entrypoint/pkg_vuln.py

Lines changed: 24 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,10 +15,32 @@
1515
class CvssSourceProvider:
1616
NVD = "NVD"
1717
MITRE = "MITRE"
18+
GITHUB = "GITHUB"
19+
GITLAB = "GITLAB"
20+
REDHAT_CVE = "REDHAT_CVE"
21+
UBUNTU_CVE = "UBUNTU_CVE"
1822
AMAZON_INSPECTOR = "AMAZON_INSPECTOR"
19-
2023
DEFAULT_PROVIDER = NVD
2124

25+
def get_rating_providers():
26+
"""
27+
get_rating_providers returns a list of vulnerability
28+
severity providers. The action uses this information
29+
to determine which vuln severity to render when
30+
multiple severity values are present from different
31+
vendors. See the function definition to view the
32+
order in which severity providers are preferred.
33+
"""
34+
35+
# NVD is most preferred, followed by everything
36+
# else in the order listed.
37+
providers = [CvssSourceProvider.NVD,
38+
CvssSourceProvider.MITRE,
39+
CvssSourceProvider.GITHUB,
40+
CvssSourceProvider.GITLAB,
41+
CvssSourceProvider.AMAZON_INSPECTOR
42+
]
43+
return providers
2244

2345
class CvssSeverity:
2446
UNTRIAGED = "untriaged"
@@ -255,7 +277,7 @@ def get_cwes(v) -> str:
255277

256278

257279
def get_cvss_rating(ratings, vulnerability) -> CvssRating:
258-
rating_provider_priority = [CvssSourceProvider.NVD, CvssSourceProvider.MITRE, CvssSourceProvider.AMAZON_INSPECTOR]
280+
rating_provider_priority = get_rating_providers()
259281
for provider in rating_provider_priority:
260282
for rating in ratings:
261283
if rating["source"]["name"] != provider:

0 commit comments

Comments
 (0)