PentestGPT 下載
PentestGPT 介紹
PentestGPT(Penetration Testing GPT)是一個基于大語言模型(LLM)的智能滲透測試助手。它結合了 ChatGPT(或其他 GPT 模型)與滲透測試工具,幫助安全研究人員自動化安全測試、漏洞分析和滲透測試任務。
介紹文檔視頻:https://www.usenix.org/conference/usenixsecurity24/presentation/deng
PentestGPT安裝
安裝演示視頻:https://youtu.be/tGC5z14dE24
源碼地址:https://github.com/GreyDGL/PentestGPT/
新建一個目錄并創建 python 虛擬環境,進入到虛擬環境后依次執行下面命令,
pip3 install git+https://github.com/GreyDGL/PentestGPT
git clone https://github.com/GreyDGL/PentestGPT
cd PentestGPT
pip install -e .
然后接下來就是設置 apikey 了,這里只有國內 deepseek 的 apikey ,所以需要修改部分源代碼,定位到 PentestGPT\pentestgpt\test_connection.py
文件,把 model 換為 deepseek-chat
,
然后跟進到 ChatGPTAPI
,
修改這里的 apikey,然后這里的 base_url 可以直接改為: https://api.deepseek.com/v1
,也可以簡單朔源一下到 ChatGPTConfig
類修改,
這時候在執行 pentestgpt-connection
,下圖這樣基本就行了
但這只是連接測試,接著配置使用也和上面一樣就是直接把 apikey 變量替換為你的 apikey,然后 url 全改為 https://api.deepseek.com/v1
,model 改為 deepseek-caht,最后執行 PentestGPT
,效果如下