Added option to fragment all domains

Author: GVCoder09

README.md

@@ -105,8 +105,8 @@ Please report any problems and malfunctions to us on [the Issues page](https://g
 ## Supported arguments / Поддерживаемые аргументы командной строки
 ```
 usage: nodpi [-h] [--host HOST] [--port PORT] [--blacklist BLACKLIST] 
-             [--log_access LOG_ACCESS] [--log_error LOG_ERROR] [-q] [-v] 
-             [--install | --uninstall]
+             [--log_access LOG_ACCESS] [--log_error LOG_ERROR] [--no_blacklist] 
+             [-q] [-v] [--install | --uninstall]
 
 options:
   -h, --help            show this help message and exit
@@ -118,6 +118,7 @@ options:
                         Path to the access control log
   --log_error LOG_ERROR
                         Path to log file for errors
+  --no_blacklist        Use fragmentation for all domains
   -q, --quiet           Remove UI output
   -v, --verbose         Show more info (only for devs)
   --install             Add proxy to Windows autostart (only for EXE)

src/main.py

@@ -30,13 +30,14 @@ def __init__(self, src_ip, dst_domain, method):
 
 class ProxyServer:
 
-    def __init__(self, host, port, blacklist, log_access, log_err, quiet, verbose):
+    def __init__(self, host, port, blacklist, log_access, log_err, no_blacklist, quiet, verbose):
 
         self.host = host
         self.port = port
         self.blacklist = blacklist
         self.log_access_file = log_access
         self.log_err_file = log_err
+        self.no_blacklist = no_blacklist
         self.quiet = quiet
         self.verbose = verbose
 
@@ -387,7 +388,7 @@ async def fragment_data(self, reader, writer):
                 self.print(f"\033[93m[NON-CRITICAL]:\033[97m {e}\033[0m")
             return
 
-        if all(site not in data for site in self.blocked):
+        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()
@@ -447,6 +448,9 @@ def parse_args():
         parser.add_argument(
             "--log_error", required=False, help="Path to log file for errors"
         )
+        parser.add_argument(
+            "--no_blacklist", action="store_true", help="Use fragmentation for all domains"
+        )
         parser.add_argument(
             "-q", "--quiet", action="store_true", help="Remove UI output"
         )
@@ -537,6 +541,7 @@ async def run(cls):
             args.blacklist,
             args.log_access,
             args.log_error,
+            args.no_blacklist,
             args.quiet,
             args.verbose,
         )