Gepetto:使用chatGPT来对函数功能进行分析并重命名变量的IDA插件

最近OpenAI的chatGPT很火,chatGPT是一个大型的语言模型,能够生成人类语言的文本,主要用于对话式的问答和聊天,以及模拟人类的对话行为

有关chatGPT的介绍就不多赘述了,相关内容很多,这是它们的官网OpenAI API

而Geptto是一个IDA插件,它使用了chatGPT的API,通过调用API,能够在IDA中让chatGPT识别函数,对函数实现的功能进行描述,并且根据函数的功能将变量重命名为易于理解的形式(而不是IDA中的v1 v2...)

Geptto开源,地址如下JusticeRage/Gepetto: IDA plugin which queries OpenAI's davinci-003 language model to speed up reverse-engineering (github.com)

Gepetto安装

以下是在windows中安装Gepetto插件,首先进入它们的仓库下载源码,源码内容如下图

 

然后,将gepetto.py复制,保存在IDA/plugins文件夹下。

此外,你需要注册一个chatGPT的账号,并获取API key,注册chatGPT需要有梯子,具体方法见此处注册ChatGPT全攻略 (qq.com)

现在假设你已经注册好了chatGPT的账号,点击右上角的个人头像查看API key:

选择创建一个API key(这里我已经创建过了)

复制API key,然后打开保存在plugins里的gepetto.py文件,在此处添加你的API key,作为变量openai.api_key的值

 

Gepetto的使用

现在,使用你的IDA随意打开一个二进制文件,选择一个函数按F5完成反汇编,然后在反汇编窗口右键,会有一个Gepetto的选项卡

Explain function可以解释函数功能,Rename variables则用于重命名变量,也可以使用快捷键调用。根据作者的测试,重命名变量通常在解释函数功能之后使用效果更好。

以下是两种方式使用前后的对比图

 

posted @ 2022-12-06 11:03  Uiharu  阅读(2385)  评论(0编辑  收藏  举报