fix: [hosts] fix number of hosts extracted

CIRCL · Jan 9, 2024 · bdaa4c5 · bdaa4c5
1 parent 5094b2d
commit bdaa4c5
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/bin/modules/DomClassifier.py b/bin/modules/DomClassifier.py
@@ -22,7 +22,6 @@
 # Import Project packages
 ##################################
 from modules.abstract_module import AbstractModule
-from lib.objects.Items import Item
 from lib.ConfigLoader import ConfigLoader
 from lib import d4
 

diff --git a/bin/modules/Hosts.py b/bin/modules/Hosts.py
@@ -55,7 +55,7 @@ def compute(self, message):
  # if mimetype.split('/')[0] == "text":
 
  content = item.get_content()
- hosts = self.regex_findall(self.host_regex, item.get_id(), content)
+ hosts = self.regex_findall(self.host_regex, item.get_id(), content, r_set=True)
  if hosts:
  print(f'{len(hosts)} host {item.get_id()}')
  for host in hosts: