Pentest Muse 是一款專為網絡安全研究人員和滲透測試人員設計和開發的人工智能 AI 助手,該工具可以幫助滲透測試人員進行頭腦風暴、編寫 Payload、分析代碼或執行網絡偵查任務。除此之外,Pentest Muse 甚至還能夠執行命令行代碼並以迭代方式解決複雜的問題和任務。
工具安裝#
由於該工具基於 Python 3.12 開發,因此我們首先需要在本地設備上安裝並配置好 Python 3.12 + 環境。
接下來,廣大研究人員可以直接使用下列命令將該項目源碼克隆至本地:
git clone https://github.com/pentestmuse-ai/PentestMuse
然後切換到項目目錄中,使用 pip 工具和項目提供的 requirements.txt 安裝該工具所需的其他依賴組件:
cd PentestMuse
pip install -r requirements.txt
除此之外,我們也可以將項目代碼克隆至本地後,將 Pentest Muse 以 Python 包的形式安裝:
pip install.
工具運行#
聊天模式(默認)
在聊天模式中,我們可以直接與 Pentest Muse 聊天,並請它幫我們頭腦風暴、編寫 Payload 或對代碼進行安全分析。參考命令如下:
python run_app.py
或
pmuse
代理模式(Beta)
我們也可以讓 Pentest Muse 使用代理模式來執行操作。在代理模式下,Pentest Muse 能夠幫助我們完成一系列簡單的任務,例如「幫我們在 url 為 xxx 的目標上執行 SQL 注入測試」。下列命令可以直接以代理模式啟動 Pentest Muse:
python run_app.py agent
或
pmuse agent
語言模型選擇
API 管理
在 www.pentestmuse.ai/signup 上註冊後,我們就可以使用 Pentest Muse 來管理 API 了,創建一個賬號,打開 Pentest Muse 的命令行接口,程序將會提示我們進行登錄。
OpenAI API 密鑰
除此之外,我們還可以選擇使用自己的 OpenAI API 密鑰。我們可以直接在啟動 Pentest Muse 腳本時,添加下列命令行參數選項即可:
--openai-api-key=[your openai api key]
項目地址
Pentest Muse: