Skip to content

Commit 3f1763d

Browse files
committed
fix: [CE Detector] avoid retagging
1 parent 3fc2112 commit 3f1763d

File tree

1 file changed

+10
-5
lines changed

1 file changed

+10
-5
lines changed

bin/modules/CEDetector.py

Lines changed: 10 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -60,10 +60,10 @@ def load_world_file(self, path):
6060
words.add(word[0])
6161
return words
6262

63-
def compute(self, message): # TODO LIMIT TO DARKWEB ???
63+
def compute(self, message):
6464
to_tag = False
6565
content = self.obj.get_content().lower()
66-
# print(content)
66+
domain_id = self.get_message()
6767

6868
is_csam = False
6969
is_child_word = False
@@ -91,10 +91,15 @@ def compute(self, message): # TODO LIMIT TO DARKWEB ???
9191
print(f'CSAM DETECTED {content}')
9292
# print()
9393
self.add_message_to_queue(message=self.ce_tag, queue='Tags')
94-
# Domains
95-
for dom in self.obj.get_correlation('domain').get('domain', []):
96-
domain = Domain(dom[1:])
94+
# Domain
95+
if domain_id:
96+
domain = Domain(domain_id)
9797
self.add_message_to_queue(obj=domain, message=self.ce_tag, queue='Tags')
98+
# Domains
99+
else:
100+
for dom in self.obj.get_correlation('domain').get('domain', []):
101+
domain = Domain(dom[1:])
102+
self.add_message_to_queue(obj=domain, message=self.ce_tag, queue='Tags')
98103

99104
return to_tag
100105

0 commit comments

Comments
 (0)