一、背景

代码审计，想必搞安全的同学再熟悉不过了，和标题中的IDE 插件有什么关系呢？莫不是标题党？
偶然挖洞闲暇时，发现了陌陌安全SRC的一个公告SAST开源，赶紧去撸一波源码，最终以看不懂为由挂机了。因此我决定从IDE插件开发学起，撸光它的码子。安全行业难能可贵的是开源精神，谢谢大厂。有请今天的主角momo-code-sec-inspector-java。大家可以先看着源码，看不懂了我们在来看看我下面的从0到1学开发插件。

上一篇介绍完IDEA插件开发的环境准备以及第一个插件开发，从本篇内容开始就对审计不同漏洞类型的代码进行分析，如下表，本篇为1001 拼接SQLi

二、 插件代码分析

2.1 知识储备

java_sec_code

提供了java漏洞代码示例，非专业程序猿请勿模仿，这是漏洞代码，漏洞代码，漏洞代码

代码审计官方示例

提供代码审计模板，可以git clone下来以此为模板

PSI理解

PSI是IDEA 插件开发里面一个特殊命名（Program Structure Interface）

PSI负责解析文件并创建语法和语义代码模型，以支持jetbrains的所有产品语言

通俗的来讲，PSI文件就是一棵大树，那这棵大树就是文件，大树上有很多枝丫，这些枝丫就是PSI element，枝丫上的所有小枝丫都是element

PsiViewer 插件安装

这个插件便于我们理解代码中重写方法的选择，也是IDEA开发获取psi element的一个辅助工具PsiViewer

也可以在plugin中搜索安装

官方SDK文档
SDK文档

2.2 项目结构图

2.3 代码审计流程

1、从plugin.xml 读取功能类及相关介绍描述

2、功能类继承IDEA 相关API，例如代码审计需继承

查看功能类

进入BaseSQLi，这个类是检测SQL注入的一些前置条件，可以通过里面的方法来检测sql语句是否满足这些条件

进入MomoBaseLocalInspectionTool后2个父类就是LocalInspectionTool

3、功能类遍历正在浏览/编辑的java文件，利用我们重写的功能类方法匹配来发现有问题代码

4、将问题代码高亮显示，并提供问题详情和修复建议

5、使用ALT + ENTER 自动修复

修复后代码样例

2.4 拼接SQL注入代码审计

下面是漏洞代码

下面是功能类

变量

如果匹配了代码审计规则，就会代码高亮，并显示问题msg

变量

如果有修复方案，ALT + ENTER会显示修复 msg

变量
修复建议提示名称（alt + enter后的名称）和帮助提示

方法是用来访问PSI 文件的一个父类方法，返回PSI element,所有审计代码都在重写这个方法。

就是当前查看/编辑文件对象，也就是 LocalInspection类的实例，通过方法里面的rule来找到有问题的代码段后返回。

PSI 米素中拼接的String(拼接的必须带变量) ，约定米素名为： PsiPolyadicExpression

所以重写的方法为：来寻找拼接SQLi

中的局部变量List exps是MoExpressionUtils类的实例，将拼接语句split 后分为3段，保存进 list。

是exps利用stream()方法将list 米素转换为 string

①判断转换后的expstr是否是sql 语句，如果是sql 格式，②则再判断 exps LIST中的米素type是否为String，如果是则将String s append 进入 StringBuilder sb中，通过for循环最终将sb添加到List sql_segments中。如果segments为空则idea告警为（字符串拼接有多种方式String.format也可以拼接），否则为拼接型SQLi

①判断代码 BaseSQLi.java

②判断代码先判断是否在定义的三个类型 type中

使用PsiViewer来查看 exps的type类型

对判断结果进行告警

三、总结

本篇讲的是拼接型SQLI 审计讲解，陌陌安全代码中还判断了是否是占位符型的SQLi，其实是多余了，因为在重写的 方法后，接收的expression 只有 PsiPolyadicExpression, 占位符SQL注入的类型是PsiMethodCallExpression。如有不对之处还请大家评论区帮忙更正，谢谢！我下一节讲占位符SQLi类型，请关注我，及时看到更新哦。