Skip to content

Commit f1e9c5a

Browse files
committed
Fix null constraint violations in multiple v1 exploit collection pipelines
Signed-off-by: ziad hany <ziadhany2016@gmail.com>
1 parent b8f3936 commit f1e9c5a

File tree

3 files changed

+6
-3
lines changed

3 files changed

+6
-3
lines changed

vulnerabilities/pipelines/enhance_with_exploitdb.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -88,7 +88,8 @@ def add_vulnerability_exploit(row, logger):
8888
for raw_alias in aliases:
8989
try:
9090
if alias := Alias.objects.get(alias=raw_alias):
91-
vulnerabilities.add(alias.vulnerability)
91+
if alias.vulnerability:
92+
vulnerabilities.add(alias.vulnerability)
9293
except Alias.DoesNotExist:
9394
continue
9495

vulnerabilities/pipelines/enhance_with_kev.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,8 @@ def add_vulnerability_exploit(kev_vul, logger):
7474
vulnerability = None
7575
try:
7676
if alias := Alias.objects.get(alias=cve_id):
77-
vulnerability = alias.vulnerability
77+
if alias.vulnerability:
78+
vulnerability = alias.vulnerability
7879
except Alias.DoesNotExist:
7980
logger(f"No vulnerability found for aliases {cve_id}")
8081
return 0

vulnerabilities/pipelines/enhance_with_metasploit.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,8 @@ def add_vulnerability_exploit(record, logger):
7979
for ref in interesting_references:
8080
try:
8181
if alias := Alias.objects.get(alias=ref):
82-
vulnerabilities.add(alias.vulnerability)
82+
if alias.vulnerability:
83+
vulnerabilities.add(alias.vulnerability)
8384
except Alias.DoesNotExist:
8485
continue
8586

0 commit comments

Comments
 (0)