JAVA语言搭建白盒静态代码、黑盒网站插件式自动化安全审计平台

public interface IWhiteboxService { /** * 验证该文件是否符合规则需求 * * @param inputFile * 被扫描的文件信息 * @return 验证结果 */ ValidateResult validate(InputFile inputFile); /** * 如果validate验证通过，则开始扫描 * * @param inputFile * 被扫描的文件信息 * @return 扫描结果 */ List<ScanResult> scan(InputFile inputFile); } 

 public void scan(Whitetask task) { //todo:循环插件扫描。或后期根据任务和插件关闭表扫描 if (null == task || StringUtil.isNull(task.getCodepath())) { //todo:记录日志，任务表有问题 return; } /* * 1、读取whiteplugin表所有插件并缓存插件对应的代码，不用每次反射 * 2、循环task.getCodepath()路径下所有文件，默认java文件，根据getFileext()的文件类型和getExcluderule()排除的文件而定 * 根据循环后的文件存储到LIST集合里，循环时排除文件是为了减少LIST的内容 * 3、根据循环LIST结果再循环所有文件，调用插件的代码进行扫描 * 4、扫描结果插入whitescanresult表 * 5、循环完毕后，修改whitetask表的当前任务扫描完毕 */ List<Whiteplugin> allPlugins = whitepluginService.getAll(); if (null == allPlugins || allPlugins.size() < 1) { //todo:记录日志，没有可用的插件 return; } List<WhitePluginData> allPluginDatas = null; if (null != allPlugins) { allPluginDatas = WhitePluginGenerator.generate(allPlugins); } if (null == allPluginDatas || allPluginDatas.size() < 1) { //todo:记录日志，有插件，转换失败 return; } List<FileData> fileDatas = FileGenerator.generate(task.getCodepath(), task.getFileext(), task.getExcluderule()); doScan(fileDatas, allPluginDatas, task); } 

 public void doScan(List<FileData> fileDatas, List<WhitePluginData> allPluginDatas, Whitetask task) { if (null == fileDatas || null == allPluginDatas || fileDatas.size() < 1 || allPluginDatas.size() < 1) { return; } for (int i = 0; i < fileDatas.size(); i++) { //循环每个文件调用每个插件 FileData tempFileData = fileDatas.get(i); for (int j = 0; j < allPluginDatas.size(); j++) { //循环每个插件 IWhiteboxService whiteboxService = allPluginDatas.get(j).getWhiteboxService(); InputFile inputFile = new InputFile(); inputFile.setContents(tempFileData.getContent()); inputFile.setFileExt(FileUtil.getEXT(tempFileData.getFileName())); inputFile.setFileName(tempFileData.getFileName()); //todo:这里传递的是绝对路径，不是包路径 inputFile.setPackagePath(tempFileData.getPath()); List<InputFileContent> inputFileContents = getFileContents(tempFileData.getContent()); inputFile.setFileContents(inputFileContents); ValidateResult validateResult = whiteboxService.validate(inputFile); if (null != validateResult && validateResult.isIssuccess()) { List<ScanResult> scanResults = whiteboxService.scan(inputFile); if (null != scanResults && scanResults.size() > 0) { //插入数据库 for (int z = 0; z < scanResults.size(); z++) { Whitescanresult whitescanresult = getScanResult(inputFile, scanResults.get(z), allPluginDatas.get(j), task); whiteScanResultService.save(whitescanresult); } } } } } } 

DROP TABLE IF EXISTS `vulfeel`.`whiteplugin`; CREATE TABLE `vulfeel`.`whiteplugin` ( `ID` int(11) NOT NULL AUTO_INCREMENT, `name` varchar(200) DEFAULT '' COMMENT '插件名称', `type` tinyint(4) DEFAULT NULL COMMENT '后期扩展分类', `importlevel` tinyint(4) DEFAULT NULL COMMENT '插件重要性，1轻微，2一般，3重要', `classpath` varchar(2000) DEFAULT NULL COMMENT '插件包路径，比如com.xx.vulfeel.plugin.a', `filename` varchar(200) DEFAULT NULL COMMENT 'class名称或jar名称，如果插件是jar包则是jar包名称，比如a.jar；否则是class名称，比如a.class', `functions` varchar(2000) DEFAULT NULL COMMENT '插件功能描述', `version` varchar(200) DEFAULT NULL COMMENT '插件版本', `refurl` varchar(1000) DEFAULT NULL COMMENT '插件原理参考URL等内容', `userid` int(11) DEFAULT NULL COMMENT '插件提供者用户ID', `createtime` datetime DEFAULT NULL, `remark` varchar(2000) DEFAULT NULL COMMENT '插件其他备注', PRIMARY KEY (`ID`) ) ENGINE=InnoDB DEFAULT CHARSET=utf8 COMMENT='白盒扫描插件表'; DROP TABLE IF EXISTS `vulfeel`.`whitescanresult`; CREATE TABLE `vulfeel`.`whitescanresult` ( `id` int(11) NOT NULL AUTO_INCREMENT, `taskid` int(11) DEFAULT NULL COMMENT '关联任务表ID', `pluginid` int(11) DEFAULT NULL COMMENT '关联问题插件ID', `impactlevel` tinyint(4) DEFAULT NULL COMMENT '影响程度，1轻微，2一般，3重要，主要和插件表的importlevel关联', `classpath` varchar(2000) DEFAULT NULL COMMENT '插件包路径，比如com.xx.vulfeel.plugin.a', `filename` varchar(200) DEFAULT NULL COMMENT 'class名称或jar名称，如果插件是jar包则是jar包名称，比如a.jar；否则是class名称，比如a.class', `startline` int(11) DEFAULT NULL COMMENT '问题开始代码行', `stopline` int(11) DEFAULT NULL COMMENT '问题结束代码行', `createtime` datetime DEFAULT NULL, `remark` varchar(2000) DEFAULT NULL COMMENT '其他备注', PRIMARY KEY (`id`) ) ENGINE=InnoDB DEFAULT CHARSET=utf8 COMMENT='白盒扫描结果表'; DROP TABLE IF EXISTS `vulfeel`.`whitetask`; CREATE TABLE `vulfeel`.`whitetask` ( `id` int(11) NOT NULL AUTO_INCREMENT, `name` varchar(200) DEFAULT NULL COMMENT '任务名称', `codepath` varchar(1000) DEFAULT NULL COMMENT '代码路径，不依赖SVN还是GIT等，只需要传入代码地址即可', `fileext` varchar(200) DEFAULT NULL COMMENT '扫描的文件格式，默认java文件，逗号分隔，比如:java,js', `excluderule` varchar(1000) DEFAULT NULL COMMENT '不扫描的路径规则，::两个冒号分割正则，比如：.*test::js$::css$', `status` tinyint(4) DEFAULT NULL COMMENT '任务状态，1初始任务，2正在执行，3任务正常结束，4任务异常结束', `userid` int(11) DEFAULT NULL COMMENT '关联创建任务的用户ID', `createtime` datetime DEFAULT NULL, `lastmodifytime` datetime DEFAULT NULL, `remark` varchar(2000) DEFAULT NULL COMMENT '其他备注', PRIMARY KEY (`id`) ) ENGINE=InnoDB DEFAULT CHARSET=utf8 COMMENT='白盒扫描任务'; 

public class SqlInjectPlugin implements IWhiteboxService { @Override public ValidateResult validate(InputFile inputFile) { // TODO Auto-generated method stub ValidateResult result = new ValidateResult(); result.setIssuccess(false); if (null == inputFile || null == inputFile.getFileContents()) { return result; } if (inputFile.getPackagePath().contains("dao")) { result.setIssuccess(true); } return result; } @Override public List<ScanResult> scan(InputFile inputFile) { // TODO Auto-generated method stub List<ScanResult> result = new ArrayList<ScanResult>(); ScanResult scanResult = new ScanResult(); if (null == inputFile || null == inputFile.getFileContents()) { return result; } Pattern pattern = Pattern.compile("("[^"]+"|\w+)\s*(\+|\+=)\s*\w+"); if (inputFile.getPackagePath().contains("dao")) { //只扫描包路径有dao的文件 //循环inputFile的每一行，如果有拼接sql就说明存在注入 for (int i = 0; i < inputFile.getFileContents().size(); i++) { InputFileContent inputFileContent = inputFile.getFileContents().get(i); Matcher matcher = pattern.matcher(inputFileContent.getContent()); if (matcher.find()) { scanResult.setStartline(inputFileContent.getLine()); scanResult.setStopline(inputFileContent.getLine()); scanResult.setRemark("存在sql拼接！"); result.add(scanResult); } } } return result; } }