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

fix: Avoid exception in case the DNS files are broken

pull/268/head
Raphaël Vinot 2021-09-29 15:05:31 +02:00
parent 1a0fe677ab
commit 9215522563
1 changed files with 12 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
lookyloo/capturecache.py
View File

@ -321,13 +321,21 @@ class CapturesIndex(Mapping):
ips_path = ct.root_hartree.har.path.parent / 'ips.json' ips_path = ct.root_hartree.har.path.parent / 'ips.json'
host_cnames: Dict[str, Optional[str]] = {} host_cnames: Dict[str, Optional[str]] = {}
if cnames_path.exists(): if cnames_path.exists():
try:
with cnames_path.open() as f: with cnames_path.open() as f:
host_cnames = json.load(f) host_cnames = json.load(f)
except json.decoder.JSONDecodeError:
# The json is broken, delete and re-trigger the requests
host_cnames = {}
host_ips: Dict[str, List[str]] = {} host_ips: Dict[str, List[str]] = {}
if ips_path.exists(): if ips_path.exists():
try:
with ips_path.open() as f: with ips_path.open() as f:
host_ips = json.load(f) host_ips = json.load(f)
except json.decoder.JSONDecodeError:
# The json is broken, delete and re-trigger the requests
host_ips = {}
for node in ct.root_hartree.hostname_tree.traverse(): for node in ct.root_hartree.hostname_tree.traverse():
if node.name not in host_cnames or node.name not in host_ips: if node.name not in host_cnames or node.name not in host_ips: