MISP
/
misp-modules
mirror of https://github.com/MISP/misp-modules
2
0
Fork 0
Code Issues Releases Wiki Activity

Merge pull request #640 from Cosive/feature/virustotal_enrichment 

Fix regression in Virustotal modules
pull/618/merge
Alexandre Dulaunoy 2023-10-13 10:19:40 +02:00 committed by GitHub
parent 7a0666b8af 0f5532b2a1
commit a4e84abb38
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
misp_modules/modules/expansion/virustotal.py
View File

@ -153,11 +153,11 @@ class VirusTotalParser:
('contacted_domains', 'communicates-with'), ('contacted_domains', 'communicates-with'),
('contacted_ips', 'communicates-with') ('contacted_ips', 'communicates-with')
]: ]:
files_iterator = self.client.iterator(f'/files/{file_report.id}/{relationship_name}', limit=self.limit) related_files_iterator = self.client.iterator(f'/files/{file_report.id}/{relationship_name}', limit=self.limit)
for file in files_iterator: for related_file in related_files_iterator:
file_object = self.create_misp_object(file) related_file_object = self.create_misp_object(related_file)
file_object.add_reference(file_object.uuid, misp_name) related_file_object.add_reference(file_object.uuid, misp_name)
self.misp_event.add_object(**file_object) self.misp_event.add_object(**related_file_object)
self.misp_event.add_object(**file_object) self.misp_event.add_object(**file_object)
return file_object.uuid return file_object.uuid

10
misp_modules/modules/expansion/virustotal_public.py
View File

@ -138,11 +138,11 @@ class VirusTotalParser:
('contacted_domains', 'communicates-with'), ('contacted_domains', 'communicates-with'),
('contacted_ips', 'communicates-with') ('contacted_ips', 'communicates-with')
]: ]:
files_iterator = self.client.iterator(f'/files/{file_report.id}/{relationship_name}', limit=self.limit) related_files_iterator = self.client.iterator(f'/files/{file_report.id}/{relationship_name}', limit=self.limit)
for file in files_iterator: for related_file in related_files_iterator:
file_object = self.create_misp_object(file) related_file_object = self.create_misp_object(related_file)
file_object.add_reference(file_object.uuid, misp_name) related_file_object.add_reference(file_object.uuid, misp_name)
self.misp_event.add_object(**file_object) self.misp_event.add_object(**related_file_object)
self.misp_event.add_object(**file_object) self.misp_event.add_object(**file_object)
return file_object.uuid return file_object.uuid