add erothots

burp/start_burp.py

@@ -2,7 +2,7 @@ import pyautogui
 import time
 import os
 import subprocess
-import glob
+import datetime
 
 BURP_JAR = "/headless/burpsuite_community.jar"
 CONFIG_FILE = "/app/burp/project_options.json"
@@ -19,8 +19,15 @@ time.sleep(5)
 
 print("Starting Burp Suite...")
 burp_process = start_burp()
+end_time = datetime.datetime.now() + datetime.timedelta(days=1)
 button = None
 while True:
+    if datetime.datetime.now() > end_time:
+        print("Burp Suite has been running for 24 hours, restarting...")
+        burp_process.terminate()
+        time.sleep(1)
+        burp_process = start_burp()
+        end_time = datetime.datetime.now() + datetime.timedelta(days=1)
     try:
         button = pyautogui.locateCenterOnScreen("/app/burp/next_button.png", confidence=0.8)
     except:
@@ -73,8 +80,8 @@ while True:
                 print("Clicking on the 'Sorting' button...")
                 pyautogui.click(button)
 
-            time.sleep(60*60*24)
-            burp_process.terminate()
-            print("Starting Burp Suite...")
-            burp_process = start_burp()
+            # time.sleep(60*60*24)
+            # burp_process.terminate()
+            # print("Starting Burp Suite...")
+            # burp_process = start_burp()