refactor(storage): 统一数据库文件存储到sqlite_dbs目录

- 创建sqlite_dbs目录以统一存放所有数据库文件
- 修改IPC数据库初始化路径，固定存储到sqlite_dbs目录下ipc.db
- 修改Header数据库初始化，默认路径指向sqlite_dbs目录的headers.db
- 修改Scan Preset数据库初始化，默认路径使用sqlite_dbs目录的scan_presets.db
- 调整get_scan_preset_db函数以支持传入数据库路径参数
- 修改.idea项目配置文件，修正Python SDK路径和测试源目录设置
- 在任务监控中添加调试日志，完善任务信息输出
- Task类增加__str__方法，便于任务信息打印调试

Author: lanshuizhiyue

.idea/misc.xml

@@ -132,25 +132,29 @@ def main(username, password):
         global_username = username
         global_password = password
 
-        _, Database.filepath = tempfile.mkstemp(prefix=MKSTEMP_PREFIX.IPC, text=False)
-        os.close(_)
+        # 创建 sqlite_dbs 目录（所有数据库文件统一存放）
+        sqlite_dbs_dir = os.path.join(current_dir, "sqlite_dbs")
+        os.makedirs(sqlite_dbs_dir, exist_ok=True)
+        logger.info(f"[*] SQLite database directory: {sqlite_dbs_dir}")
 
-        # Initialize IPC database
-        # pdb.set_trace()
+        # Initialize IPC database (固定存储到 sqlite_dbs 目录)
+        Database.filepath = os.path.join(sqlite_dbs_dir, f"{MKSTEMP_PREFIX.IPC}ipc.db")
         DataStore.current_db = Database()
         logger.info(f"id(DataStore.current_db): {id(DataStore.current_db)}")
         DataStore.current_db.connect()
         DataStore.current_db.init()
         logger.info("[*] IPC database initialized")
 
-        # Initialize Header database
-        DataStore.header_db = HeaderDatabase()
+        # Initialize Header database (存储到 sqlite_dbs 目录)
+        header_db_path = os.path.join(sqlite_dbs_dir, "headers.db")
+        DataStore.header_db = HeaderDatabase(database_path=header_db_path)
         DataStore.header_db.connect()
         DataStore.header_db.init()
         logger.info("[*] Header database initialized")
 
-        # Initialize Scan Preset database
-        DataStore.scan_preset_db = get_scan_preset_db()
+        # Initialize Scan Preset database (存储到 sqlite_dbs 目录)
+        scan_preset_db_path = os.path.join(sqlite_dbs_dir, "scan_presets.db")
+        DataStore.scan_preset_db = get_scan_preset_db(database_path=scan_preset_db_path)
         logger.info("[*] Scan preset database initialized")
 
         scheduler = BackgroundScheduler()

.idea/sqlmapWebUI.iml

@@ -12,13 +12,13 @@ class HeaderDatabase(Database):
     """独立的请求头管理数据库"""
 
     def __init__(self, database_path=None):
-        # 如果没有指定数据库路径，则使用程序所在目录下的headers.db文件
+        # 如果没有指定数据库路径，则使用 sqlite_dbs 目录下的 headers.db 文件
         if database_path is None:
-            # 获取当前脚本所在目录
             current_dir = os.path.dirname(os.path.abspath(__file__))
-            # 确保是src目录的上一级目录
             project_dir = os.path.dirname(current_dir)
-            database_path = os.path.join(project_dir, "headers.db")
+            sqlite_dbs_dir = os.path.join(project_dir, "sqlite_dbs")
+            os.makedirs(sqlite_dbs_dir, exist_ok=True)
+            database_path = os.path.join(sqlite_dbs_dir, "headers.db")
 
         super().__init__(database_path)
         self.database_path = database_path

.idea/workspace.xml

@@ -37,11 +37,13 @@ def __init__(self, database_path=None):
         if self._initialized:
             return
 
-        # 如果没有指定数据库路径，使用默认路径
+        # 如果没有指定数据库路径，使用 sqlite_dbs 目录下的默认路径
         if database_path is None:
             current_dir = os.path.dirname(os.path.abspath(__file__))
             project_dir = os.path.dirname(current_dir)
-            database_path = os.path.join(project_dir, "scan_presets.db")
+            sqlite_dbs_dir = os.path.join(project_dir, "sqlite_dbs")
+            os.makedirs(sqlite_dbs_dir, exist_ok=True)
+            database_path = os.path.join(sqlite_dbs_dir, "scan_presets.db")
 
         super().__init__(database_path)
         self.database_path = database_path
@@ -605,10 +607,10 @@ def _row_to_preset(self, row) -> Optional[ScanPreset]:
 _scan_preset_db: Optional[ScanPresetDatabase] = None
 
 
-def get_scan_preset_db() -> ScanPresetDatabase:
+def get_scan_preset_db(database_path=None) -> ScanPresetDatabase:
     """获取扫描配置预设数据库实例"""
     global _scan_preset_db
     if _scan_preset_db is None:
-        _scan_preset_db = ScanPresetDatabase()
+        _scan_preset_db = ScanPresetDatabase(database_path=database_path)
         _scan_preset_db.init()
     return _scan_preset_db

src/backEnd/main.py

@@ -76,6 +76,10 @@ def __init__(self, taskid, remote_addr, scanUrl, host, method, headers, body):
 
         logger.debug(f"[{self.taskid}] Task initialization completed")
 
+    
+    def __str__(self):
+        return f"Task(taskid={self.taskid}, status={self.status}, create_datetime={self.create_datetime}, start_datetime={self.start_datetime}, scanUrl={self.scanUrl}, host={self.host}, method={self.method}, headers={self.headers}, body={self.body}, remote_addr={self.remote_addr}, process={self.process}, output_directory={self.output_directory}, options={self.options}, _original_options={self._original_options}, _user_set_options={self._user_set_options}, _header_rules_applied={self._header_rules_applied}, _body_field_rules_applied={self._body_field_rules_applied}, _request_file_path={self._request_file_path})"
+
     def initialize_options(self, taskid):
         datatype = {"boolean": False, "string": None, "integer": None, "float": None}
         self.options = AttribDict()

src/backEnd/model/HeaderDatabase.py

@@ -34,7 +34,7 @@ def get_max_tasks_count():
 
 
 def monitor(max_tasks_count=None):
-    # logger.info("monitor...")
+    logger.debug("monitor triggered...")
     # logger.debug(f"monitor -> id(DataStore.tasks): {id(DataStore.tasks)}")
     # logger.debug(f"monitor -> id(DataStore.current_db): {id(DataStore.current_db)}")
     local_max_tasks_count = 0
@@ -56,7 +56,6 @@ def monitor(max_tasks_count=None):
     with DataStore.tasks_lock:
         runnable_list = []
         running_task_cnt = 0
-
         for taskid in DataStore.tasks:
             task = DataStore.tasks[taskid]
             task_orin_status = task.status
@@ -87,7 +86,8 @@ def monitor(max_tasks_count=None):
                         continue
                 else:
                     running_task_cnt += 1
-                    # logger.debug(f"monitor -> task_id: {task.options.taskid} task.start_datetime: {task.start_datetime}")
+                    logger.debug(f"monitor -> task_id: {task.options.taskid} task.start_datetime: {task.start_datetime}")
+                    logger.debug(f"monitor -> task_info: {task}")
                     task.start_datetime = datetime.now()
                     task.engine_start()
                     task.status = TaskStatus.Running