codegeex 可自动识别并修复 sql 注入等安全漏洞,需先安装插件并启用 securitymode;支持右键修复、终端报错反向定位、侧边栏主动提交三种加固方式,均依据 owasp 等标准生成修复方案。
☞☞☞AI 智能聊天, 问答助手, AI 智能搜索, 多模态理解力帮你轻松跨越从0到1的创作门槛☜☜☜

当你在 VSCode 中写出存在 SQL 注入、硬编码密钥、未校验用户输入等安全漏洞的代码时,CodeGeeX 能直接定位风险点并生成符合 OWASP 标准的加固方案,无需切换工具或手动查阅规范。
安装并激活 CodeGeeX 安全增强模块
这一步是后续所有自动修复功能的前提,缺省状态下 CodeGeeX 不启用深度安全分析能力。
打开 VSCode → 按 Ctrl+Shift+X 进入扩展市场 → 搜索 CodeGeeX → 确认发布者为 Zhipu AI → 点击安装 → 安装完成后点击“重新加载”。
重启后,在设置中(Ctrl+,)搜索 codegeex.securityMode → 勾选启用 → 此时插件才会调用安全微调版 CodeGeeX2 模型进行 CWE 漏洞识别。
【必须完成此步,否则右键“修复”仅做语法纠错,不处理安全问题】
对高危代码行一键触发安全修复
适用于已标红波浪线的明显漏洞,如拼接 SQL 字符串、明文写密码、未过滤的 request.getParameter() 等。
将光标停在含漏洞的代码行任意位置 → 鼠标右键 → 选择“使用 CodeGeeX 修复” → 等待 3~5 秒 → 侧边栏弹出带安全说明的修复建议。
修复建议中会明确标注漏洞类型(如 CWE-89)、风险等级(高危/中危)及依据标准(OWASP ASVS 4.0.3),点击“应用修复”即覆盖原代码。
通过终端报错反向定位并修复安全漏洞
当程序运行时报出 java.lang.SecurityException 或 org.springframework.dao.DataIntegrityViolationException 类异常时,往往意味着底层存在被忽略的安全缺陷。
方法一:在终端中完整选中从 Traceback 开始到最后一行的文本 → 右键 → 选择“使用 CodeGeeX 解释” → 模型自动提取异常上下文与调用栈 → 返回结构化报告,指出漏洞根因(例如“用户输入未经白名单过滤直接传入 File 构造函数”)→ 报告末尾附带可直接复制的修复补丁。
方法二:若报错中含文件路径与行号(如 UserService.java:47),可在 VSCode 中按 Ctrl+P 快速跳转 → 将光标置于该行 → 按 Alt+Enter 唤出灯泡菜单 → 选择“使用 CodeGeeX 修复” → 优先采用参数化查询、输入校验、最小权限原则等方案替换原始逻辑。
在侧边栏会话中主动提交漏洞代码请求加固
适用于尚未报错但存在隐蔽风险的代码,比如自定义加密函数缺少盐值、JWT 签名算法未强制指定 HS256、日志中打印敏感字段等。
第一步:点击左侧活动栏 CodeGeeX 图标,打开侧边栏聊天界面。
第二步:粘贴待加固代码段,并附加自然语言指令,例如:“这段 Java 代码用于生成 AES 密钥,请检查是否存在密钥管理缺陷,并按 NIST SP 800-131A Rev.2 要求重写。”
第三步:发送后等待模型返回分析结论,重点查看其是否指出 secrets.randomBytes() 被误用为密钥生成器(应改用 KeyGenerator)、是否遗漏 IV 随机性验证、是否未启用 GCM 模式防篡改。
第四步:确认修复方案后,点击“插入到编辑器”按钮,代码将自动添加到当前文件光标位置。











