From a8ffcb9f7259da2580fcdc979a7179ac27566160 Mon Sep 17 00:00:00 2001
From: DomySh <me@domysh.com>
Date: Sat, 18 Jun 2022 23:48:10 +0200
Subject: [PATCH] General bug fixes

---
 backend/proxy/__init__.py | 22 +++++++++++++---------
 backend/proxy/proxy.cpp   |  4 +++-
 backend/utils.py          |  6 ++++--
 3 files changed, 20 insertions(+), 12 deletions(-)

diff --git a/backend/proxy/__init__.py b/backend/proxy/__init__.py
index 94f4c8c..9c8b2c7 100755
--- a/backend/proxy/__init__.py
+++ b/backend/proxy/__init__.py
@@ -1,7 +1,7 @@
 from signal import SIGUSR1
 from secrets import token_urlsafe
 import subprocess, re, os
-
+from threading import Lock
 
 #c++ -o proxy proxy.cpp
 
@@ -28,6 +28,8 @@ class Filter:
 
 class Proxy:
     def __init__(self, internal_port, public_port, callback_blocked_update=None, filters=None, public_host="0.0.0.0", internal_host="127.0.0.1"):
+        self.filter_map = {}
+        self.filter_map_lock = Lock()
         self.public_host = public_host
         self.public_port = public_port
         self.internal_host = internal_host
@@ -43,8 +45,8 @@ class Proxy:
     
     def start(self, in_pause=False):
         if self.process is None:
-            filter_map = self.compile_filters()
-            filters_codes = list(filter_map.keys()) if not in_pause else []
+            self.filter_map = self.compile_filters()
+            filters_codes = list(self.filter_map.keys()) if not in_pause else []
             proxy_binary_path = os.path.join(os.path.dirname(os.path.abspath(__file__)),"./proxy")
             try:
                 self.__write_config(filters_codes)
@@ -56,8 +58,9 @@ class Proxy:
                 for stdout_line in iter(self.process.stdout.readline, ""):
                     if stdout_line.startswith("BLOCKED"):
                         regex_id = stdout_line.split()[1]
-                        filter_map[regex_id].blocked+=1
-                        if self.callback_blocked_update: self.callback_blocked_update(filter_map[regex_id])
+                        with self.filter_map_lock:
+                            self.filter_map[regex_id].blocked+=1
+                            if self.callback_blocked_update: self.callback_blocked_update(self.filter_map[regex_id])
                 self.process.stdout.close()
                 return self.process.wait()
             finally:
@@ -91,10 +94,11 @@ class Proxy:
 
     def reload(self):
         if self.isactive():
-            filter_map = self.compile_filters()
-            filters_codes = list(filter_map.keys())
-            self.__write_config(filters_codes)
-            self.trigger_reload_config()
+            with self.filter_map_lock:
+                self.filter_map = self.compile_filters()
+                filters_codes = list(self.filter_map.keys())
+                self.__write_config(filters_codes)
+                self.trigger_reload_config()
 
     def isactive(self):
         return True if self.process else False
diff --git a/backend/proxy/proxy.cpp b/backend/proxy/proxy.cpp
index cee8b6f..a58bf3d 100644
--- a/backend/proxy/proxy.cpp
+++ b/backend/proxy/proxy.cpp
@@ -407,12 +407,14 @@ int main(int argc, char* argv[])
    const std::string local_host      = argv[1];
    const std::string forward_host    = argv[3];
    
+   config_file = argv[5];
+
    update_regex();
    signal(SIGUSR1, signal_handler);
    signal(SIGTERM, signal_handler);
    signal(SIGSEGV, signal_handler);
 
-   config_file = argv[5];
+   
    
 
    boost::asio::io_service ios;
diff --git a/backend/utils.py b/backend/utils.py
index def1ecd..bece124 100755
--- a/backend/utils.py
+++ b/backend/utils.py
@@ -1,5 +1,6 @@
 from asyncore import file_dispatcher
 from imp import reload
+from pstats import Stats
 from proxy import Filter, Proxy
 import random, string, os, threading, sqlite3, time, atexit, socket
 from kthread import KThread
@@ -261,8 +262,9 @@ class ProxyManager:
                 else:
                     self.__update_status_db(id, previous_status)
             
-            if restart_required: proxy.restart()
-            elif reload_required: proxy.reload()
+            if previous_status != STATUS.STOP:
+                if restart_required: proxy.restart(in_pause=(previous_status == STATUS.PAUSE))
+                elif reload_required and previous_status != STATUS.PAUSE: proxy.reload()
 
             signal.wait()
             signal.clear()