Skip to content

Commit

Permalink
fix: [vulnerability_lookup] Avoiding KeyError exceptions on some fields
Browse files Browse the repository at this point in the history
  • Loading branch information
chrisr3d committed Jul 2, 2024
1 parent 42fb1bc commit 7d7dc1e
Showing 1 changed file with 4 additions and 5 deletions.
9 changes: 4 additions & 5 deletions misp_modules/modules/expansion/vulnerability_lookup.py
Original file line number Diff line number Diff line change
Expand Up @@ -152,7 +152,7 @@ def _parse_csaf_description(self, lookup_result: dict) -> str:
for field, relation in self.mapping.csaf_mapping().items():
misp_object.add_attribute(relation, tracking[field])
misp_object.add_attribute('summary', description['title'])
for reference in description['references']:
for reference in description.get('references', []):
misp_object.add_attribute('references', reference['url'])
misp_object.add_attribute('credit', description['publisher']['name'])
misp_object.add_reference(self.misp_attribute.uuid, 'describes')
Expand Down Expand Up @@ -183,7 +183,7 @@ def _parse_csaf_description(self, lookup_result: dict) -> str:

def _parse_cve_description(self, lookup_result: dict) -> str:
misp_object = MISPObject('vulnerability')
cveMetaData = lookup_result['cveMetaData']
cveMetaData = lookup_result['cveMetadata']
for field, relation in self.mapping.cve_mapping().items():
misp_object.add_attribute(relation, cveMetaData[field])
for reference in lookup_result['containers']['cna']['references']:
Expand Down Expand Up @@ -267,9 +267,8 @@ def _parse_standard_description(self, lookup_result: dict) -> str:
misp_object = MISPObject('vulnerability')
for field, relation in self.mapping.standard_mapping().items():
misp_object.add_attribute(relation, lookup_result[field])
misp_object.add_attribute(
'cvss-string', lookup_result['severity']['score']
)
for cvss in lookup_result.get('severity', []):
misp_object.add_attribute('cvss-string', cvss['score'])
for reference in lookup_result['references']:
misp_object.add_attribute('references', reference['url'])
misp_object.add_reference(self.misp_attribute.uuid, 'related-to')
Expand Down

0 comments on commit 7d7dc1e

Please sign in to comment.