CIRCL
/
lookyloo
mirror of https://github.com/CIRCL/lookyloo
2
0
Fork 0
Code Issues Releases Wiki Activity

chg: Avoid exception on broken captures

pull/497/head
Raphaël Vinot 2022-08-04 21:11:58 +02:00
parent 1ddd38fdcd
commit 4d088f459f
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
bin/async_capture.py
View File

@ -89,9 +89,11 @@ class AsyncCapture(AbstractManager):
with open(tmp_f.name, "wb") as f: with open(tmp_f.name, "wb") as f:
f.write(to_capture[b'document']) f.write(to_capture[b'document'])
url = f'file://{tmp_f.name}' url = f'file://{tmp_f.name}'
else: elif to_capture.get(b'url'):
url = to_capture[b'url'].decode() url = to_capture[b'url'].decode()
self.thirdparty_submit(url) self.thirdparty_submit(url)
else:
self.logger.warning(f'Invalid capture {to_capture}.')
self.logger.info(f'Capturing {url} - {uuid}') self.logger.info(f'Capturing {url} - {uuid}')
success, error_message = await self._capture( success, error_message = await self._capture(