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

View File

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