fix: use strict domain matching & add reg patterns

Author: GVCoder09

src/main.py

@@ -3,6 +3,7 @@
 import argparse
 import asyncio
 import random
+import re
 import logging
 import os
 import sys
@@ -65,6 +66,7 @@ def __init__(self, host, port, blacklist, log_access, log_err, no_blacklist, aut
         self.tasks_lock = asyncio.Lock()
 
         self.blocked = []
+        self.blocked_regex = []
         self.whitelist = []
         self.tasks = []
         self.server = None
@@ -137,7 +139,23 @@ def load_blacklist(self):
             sys.exit(1)
 
         with open(self.blacklist, "r", encoding="utf-8") as f:
-            self.blocked = [line.rstrip().encode() for line in f]
+            for line in f:
+                line = line.strip()
+                if not line or line.startswith('#'):
+                    continue
+
+                if line.startswith('regex:'):
+                    pattern = line[6:].strip()
+                    try:
+                        self.blocked_regex.append(re.compile(pattern.encode()))
+                    except re.error as e:
+                        self.print(
+                            f"\033[91m[ERROR]: Invalid regex pattern {pattern}: {e}\033[0m")
+                        self.logger.error(
+                            "Invalid regex pattern %s: %s", pattern, e)
+                else:
+                    escaped_domain = re.escape(line).encode()
+                    self.blocked.append(escaped_domain)
 
     async def run(self):
         """
@@ -435,10 +453,11 @@ async def fragment_data(self, reader, writer):
             return
 
         if not self.no_blacklist and all(site not in data for site in self.blocked):
-            self.allowed_connections += 1
-            writer.write(head + data)
-            await writer.drain()
-            return
+            if not any(regex.search(data) for regex in self.blocked_regex):
+                self.allowed_connections += 1
+                writer.write(head + data)
+                await writer.drain()
+                return
 
         self.blocked_connections += 1