Skip to content

Latest commit

 

History

History
86 lines (78 loc) · 2.87 KB

README.md

File metadata and controls

86 lines (78 loc) · 2.87 KB
Raw

项目介绍

基于Tai-e插件系统开发的Web应用分析程序，可以直接用来分析Web项目。目前只对Spring项目的动态特性做了处理。

  • URL映射：提取Spring Web应用中的路由信息，生成URL与其对应的入口方法映射关系
  • Bean提取：自动识别和提取Spring应用中的Bean
  • 调用图构造：为每个入口方法生成独立的调用图
    • 处理依赖注入的对象，能够正确处理依赖注入对象的方法调用
    • 尚未处理面向切面编程
  • 漏洞检测：基于已识别的调用流，检测潜在的漏洞点，识别漏洞类型
    • 相关代码在llm分支，在使用大模型进行检测前，请在DefaultLLMConnector.java中填写api key，目前仅实现了GPT与讯飞星火大模型客户端
    • 使用前请加入运行参数：-v
    • 这部分会消耗大量Token
    • 效果待优化...更好的处理方式正在开发中...

项目尚处于开发阶段
针对web多入口生成独立调用图功能目前可用
漏洞检测需要填写自己的api key（GPT或讯飞星火大模型） 或者根据ModelClient.java接口实现任意大模型客户端

使用说明

  1. options.yml
optionsFile: null
printHelp: false
# 第三方依赖路径，支持输入jar包
classPath :
    - WebGoat/BOOT-INF/lib
# 应用程序类路径
appClassPath: [WebGoat/BOOT-INF/classes]
mainClass:
inputClasses: []
javaVersion: 17
prependJVM: true
allowPhantom: true
worldBuilderClass: pascal.taie.frontend.soot.SootWorldBuilder
outputDir: output
preBuildIR: false
worldCacheMode: false
# 分析范围：APP，REACHABLE，ALL
scope: APP
nativeModel: true
planFile: null
analyses:
  ir-dumper: ;
  # 分析路由
  routerAnalysis: ""
  # 提取Bean
  beanAnalysis: ""
  cg: ""
  #  cfg: ""
  #  icfg: ""
  pta: "plugins:[org.example.spring.DICGConstructorPlugin]"
onlyGenPlan: false
keepResult:
  - $KEEP-ALL
  1. 运行参数
-o="configs/options.yml"

或者

--options-file="configs/options.yml"

关于options.yml中选项的详细信息请阅读：Tai-e 配置

输出

  • output/urls.txt: 路由映射关系
  • output/callFlows/*.dot: 不同入口的调用图，以入口方法命名

TODO

  • 处理面向切面编程特性，完善调用图
  • 抽象出入口方法参数类型的对象，完善调用图
  • 优化路由信息提取流程，输出结果
  • 优化Bean输出结果
  • 自动添加污点流 Source 点
  • 更改大模型检测漏洞的方案
  • 更改输出效果，考虑增加前端界面

Reference

参考了

处理了三种依赖注入情况与“this”变量，而非仅处理Field注入