AIL/ail-framework
7
0
Fork 0
mirror of https://github.com/ail-project/ail-framework.git synced 2024-11-10 08:38:28 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: [DomClassifier] improve perf

Browse source
This commit is contained in:
terrtia 2024-01-09 11:38:54 +01:00
parent 7263a9777c
commit 5094b2dcbb
No known key found for this signature in database
GPG key ID: 1E1B1F50D84613D0
2 changed files with 15 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
bin/modules/DomClassifier.py
View file

@ -73,15 +73,17 @@ class DomClassifier(AbstractModule):
for dns_record in self.c.vdomain: for dns_record in self.c.vdomain:
self.add_message_to_queue(obj=None, message=dns_record) self.add_message_to_queue(obj=None, message=dns_record)
localizeddomains = self.c.include(expression=self.cc_tld) if self.cc_tld:
if localizeddomains: localizeddomains = self.c.include(expression=self.cc_tld)
print(localizeddomains) if localizeddomains:
self.redis_logger.warning(f"DomainC;{item_source};{item_date};{item_basename};Checked {localizeddomains} located in {self.cc_tld};{item.get_id()}") print(localizeddomains)
self.redis_logger.warning(f"DomainC;{item_source};{item_date};{item_basename};Checked {localizeddomains} located in {self.cc_tld};{item.get_id()}")
localizeddomains = self.c.localizedomain(cc=self.cc) if self.cc:
if localizeddomains: localizeddomains = self.c.localizedomain(cc=self.cc)
print(localizeddomains) if localizeddomains:
self.redis_logger.warning(f"DomainC;{item_source};{item_date};{item_basename};Checked {localizeddomains} located in {self.cc};{item.get_id()}") print(localizeddomains)
self.redis_logger.warning(f"DomainC;{item_source};{item_date};{item_basename};Checked {localizeddomains} located in {self.cc};{item.get_id()}")
if r_result: if r_result:
return self.c.vdomain return self.c.vdomain

7
configs/core.cfg.sample
View file

@ -222,10 +222,13 @@ password = ail_trackers
cc_critical = DE cc_critical = DE
[DomClassifier] [DomClassifier]
cc = DE #cc = DE
cc_tld = r'\.de$' #cc_tld = r'\.de$'
cc =
cc_tld =
dns = 8.8.8.8 dns = 8.8.8.8
[Mail] [Mail]
dns = 8.8.8.8 dns = 8.8.8.8