fix: Parsing detected & undetected urls

pull/322/head
chrisr3d 2019-07-15 23:44:25 +02:00
parent d9b03a7aa5
commit a61d09db8b
No known key found for this signature in database
GPG Key ID: 6BBED1B63A6D639F
1 changed files with 7 additions and 6 deletions

View File

@ -27,8 +27,9 @@ class VirusTotalParser():
results = {key: event[key] for key in ('Attribute', 'Object') if (key in event and event[key])}
return {'results': results}
def parse_detected_urls(self, detected_urls):
for url in detected_urls:
def parse_urls(self, query_result):
for feature in ('detected_urls', 'undetected_urls'):
for url in query_result[feature]:
value = url['url'] if isinstance(url, dict) else url
self.misp_event.add_attribute('url', value)
@ -79,7 +80,7 @@ class DomainQuery(VirusTotalParser):
whois_object.add_attribute('text', type='text', value=query_result[whois])
self.misp_event.add_object(**whois_object)
self.parse_resolutions(query_result['resolutions'], query_result['subdomains'])
self.parse_detected_urls(query_result['detected_urls'])
self.parse_urls(query_result)
for domain in query_result['domain_siblings']:
self.misp_event.add_attribute('domain', domain)
@ -121,7 +122,7 @@ class IpQuery(VirusTotalParser):
attribute_type, relation = value
asn_object.add_attribute(relation, type=attribute_type, value=query_result[key])
self.misp_event.add_object(**asn_object)
self.parse_detected_urls(query_result['detected_urls'])
self.parse_urls(query_result)
if query_result.get('resolutions'):
self.parse_resolutions(query_result['resolutions'])