WPeChatGPT

项目简介：WPeChatGPT

WPeChatGPT是一个功能强大的IDA插件，基于OpenAI发布的gpt-3.5-turbo人工智能模型，这与流行的ChatGPT使用的模型相同。此插件的主要目的是帮助分析人员快速分析二进制文件，从而更高效地进行工作。

功能特点

目前WPeChatGPT支持以下功能：

• 分析功能使用环境、预期目的和功能：帮助分析人员了解代码的背景和用途。
• 重命名函数变量：能够根据功能和上下文对变量进行更合适的命名。
• 尝试使用Python3还原函数：特别对较小代码块（如XOR解密函数）有效。
• 查找函数中的漏洞：识别二进制漏洞以防止后续安全缺陷。
• 尝试生成相应EXP：利用Python生成潜在漏洞的利用代码。
• 自动分析二进制文件：使用GPT进行自动化分析，详细信息见Auto-WPeGPT部分。

插件从v1.0版本的Gepetto基础升级到v2.0版本后，使用OpenAI最新的gpt-3.5-turbo模型。值得注意的是，ChatGPT的分析结果仅供参考，分析工作依然需要人类专家完成。

更新历史

安装指南

1. 运行以下命令安装所需软件包：

   pip install -r ./requirements.txt
   

2. 修改脚本WPeChatGPT.py，设置API密钥至变量openai.api_key，并将变量ZH_CN更改为False（默认中文）。
3. 将脚本文件WPeChatGPT.py及文件夹Auto-WPeGPT_WPeace复制到IDA的插件文件夹中，重启IDA即可使用。

注意：需要将IDA环境设置为Python3，并使用最新的OpenAI Python包。

使用方式

支持在IDA中通过右键菜单、菜单栏或快捷键调用。

• 快捷键：

  • 函数分析 = "Ctrl-Alt-G"
  • 重命名函数变量 = "Ctrl-Alt-R"
  • 漏洞查找 = "Ctrl-Alt-E"

• 右键点击伪代码窗口可弹出相关操作菜单。

• 菜单栏路径：编辑 -> WPeChatGPT

实例展示

可以通过查看项目附带的截图和动图，了解分析功能、重命名、漏洞查找等操作效果。

自动分析功能Auto-WPeGPT

该功能旨在自动分析二进制文件，并输出分析结果和函数调用树等信息。详细的使用方法如下：

• 在菜单栏选择：编辑 -> WPeChatGPT -> Auto-WPeGPT
• 分析完成后，结果会输出在相应的文件夹（如"WPe_+IDB名称"）中。

各输出文件的意义：

GPT-Result.txt -> Auto-WPeGPT分析结果
funcTree.txt -> 函数调用树结构
mainFuncTree.txt -> 主函数树结构
effectiveStrings.txt -> 二进制中的可疑字符串

Auto-WPeGPT v0.1版本测试表明，该版本对函数较少的文件分析效果较好；若函数数量多，则可能超出token范围。我们将在下一个版本中努力改善这一问题。

关于OpenAI-API错误报告

自2023年3月2日起，开发者发现API有时会出现错误，经过对相关问题的调查和处理，可以通过以下方式解决问题：

• 科学上网下，若多次尝试连接API均失败，需检查Python的urllib3版本（1.26版本存在代理问题）。
  • 可使用以下命令将urllib3降级修复：

    pip uninstall urllib3
    pip install urllib3==1.25.11
    

• 可设置插件的前向或反向代理：
  • 填入代理地址和端口信息于proxy变量中（前向代理）。
  • 填入反向代理URL至proxy_address变量中（反向代理）。

联系方式

使用插件时如有问题或疑问，请留言或发送电子邮件。

鸣谢

此项目基于Gepetto并受其启发，开户代码及相关信息请参见：https://github.com/JusticeRage/Gepetto 。