### 如何在Python程序中调用Sqlmap 为了在 Python 中集成并调用 Sqlmap 进行 SQL 注入测试，可以采用多种方法。最常见的方式是通过命令行接口 (CLI) 调用 Sqlmap 或者使用其 API 接口。 #### 方法一：使用 subprocess 模块执行 CLI 命令 这种方法适用于希望保持简单操作而不深入修改现有代码的情况： ```python import subprocess def run_sqlmap(url, options): command = ['python', 'sqlmap.py', '-u', url] + options.split() try: result = subprocess.run(command, capture_output=True, text=True) print(result.stdout) if result.stderr: print(f"Error occurred:\n{result.stderr}") except Exception as e: print(e) # Example usage run_sqlmap('http://example.com/vulnerable.php?id=1', '--batch --level=5') ``` 此函数 `run_sqlmap` 将 URL 和附加选项作为参数传递给 Sqlmap 的 CLI 工具[^2]。请注意，在实际应用中应当处理好输入验证以及错误捕捉逻辑以增强稳定性。 #### 方法二：利用 Sqlmap 提供的 RESTful API 对于更复杂的场景或是需要频繁交互的应用来说，推荐使用 Sqlmap 自身提供的 RESTful API 来实现自动化流程。这通常涉及到启动一个独立的服务进程来监听 HTTP 请求，并允许外部应用程序发送 JSON 数据包来进行配置和控制。 下面是一段简单的例子展示了怎样初始化这个服务并与之通信： ```python from requests import post, get API_URL = "http://localhost:8775" def start_task(target_url, parameters=None): task_new_response = post(API_URL + '/task/new').json() if not task_new_response['success']: raise ValueError("Failed to create new task") task_id = task_new_response.get('taskid') data = { 'url': target_url, **(parameters or {}) } configure_response = post( f"{API_URL}/option/{task_id}/set", json=data ).json() if not configure_response['success']: raise ValueError("Configuration failed") scan_start_response = post( f"{API_URL}/scan/{task_id}/start" ).json() return task_id if __name__ == '__main__': tid = start_task('http://testphp.vulnweb.com/artists.php?artist=1', {'risk': 3}) status_resp = get(f'{API_URL}/scan/{tid}/status').json() while status_resp["status"] != "terminated": # Wait until the scan completes... pass results = get(f'{API_URL}/scan/{tid}/data').json() print(results) ``` 上述代码片段首先创建一个新的扫描任务，接着设置目标网站及其他必要参数，最后发起一次完整的扫描请求。完成后可以通过查询状态端点(`/scan/<task-id>/status`)轮询获取当前进度直至完成，之后可以从数据端点提取最终报告[^4]。