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

fix: Avoid exception if rendered page is missing on broken tree

pull/212/head
Raphaël Vinot 2021-06-03 19:07:02 -07:00
parent 89f19d4b3e
commit 88eb85df46
1 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
lookyloo/modules.py
View File

@ -21,7 +21,7 @@ from pysanejs import SaneJS
from pyeupi import PyEUPI from pyeupi import PyEUPI
from pymisp import PyMISP, MISPEvent, MISPAttribute from pymisp import PyMISP, MISPEvent, MISPAttribute
from har2tree import CrawledTree, HostNode, URLNode from har2tree import CrawledTree, HostNode, URLNode, Har2TreeError
class MISP(): class MISP():
@ -189,10 +189,14 @@ class UniversalWhois():
if auto_trigger and not self.allow_auto_trigger: if auto_trigger and not self.allow_auto_trigger:
return None return None
hostnode = crawled_tree.root_hartree.get_host_node_by_uuid(crawled_tree.root_hartree.rendered_node.hostnode_uuid) try:
self.query_whois_hostnode(hostnode) hostnode = crawled_tree.root_hartree.get_host_node_by_uuid(crawled_tree.root_hartree.rendered_node.hostnode_uuid)
for n in hostnode.get_ancestors(): except Har2TreeError as e:
self.query_whois_hostnode(n) self.logger.warning(e)
else:
self.query_whois_hostnode(hostnode)
for n in hostnode.get_ancestors():
self.query_whois_hostnode(n)
def whois(self, query: str) -> str: def whois(self, query: str) -> str:
if not self.available: if not self.available: