讲师中心 微信公众号
AI工具推荐 视频效率加速

如何解决大语言模型生成SQL代码时的注入风险_对生成的SQL进行沙盒校验

酷浩小哥_3396

酷浩小哥_3396

发布时间:2026-05-12 09:18:20

|

593人浏览过

|

来源于php中文网

原创

不能直接信任大模型生成的SQL,因其常将用户输入直接拼接进字符串导致SQL注入;需通过SQL沙盒在执行前静态分析,拦截高危关键词、校验参数化格式、检测注释符与分号等风险特征。

如何解决大语言模型生成sql代码时的注入风险_对生成的sql进行沙盒校验

为什么不能直接信任大模型生成的SQL

大模型生成SQL时,常把用户输入原样拼进字符串,比如输出 "SELECT * FROM users WHERE name = '" + user_input + "'" 这类模板。它不区分上下文——哪怕你提示“请用参数化”,只要输入里混了单引号、注释符或布尔逻辑片段(如 ' OR 1=1 --),生成结果就可能带注入payload。更危险的是,模型可能“自作聪明”补全语句,把 user_input 当成代码片段而非数据处理。

SQL沙盒校验的核心动作:拦截+解析+约束

沙盒不是运行SQL,而是静态分析其结构与风险特征。关键点在于三步过滤:

  • 拒绝含 UNION、EXEC、EXECUTE、DECLARE、CREATE、DROP、ALTER 等非查询关键词的语句
  • 提取所有字符串字面量,检查是否包含 SQL 注释符(--、/*)、分号(;)、布尔运算符(OR、AND)出现在引号内但未被转义
  • 验证占位符形式:只允许 ?、:name、$1 等标准参数标记,禁止出现 "' + input + '" 类拼接模式

示例:模型输出 "SELECT id FROM logs WHERE level = '" + level + "' AND ts > '" + since + "'" 会被直接拒收;而 "SELECT id FROM logs WHERE level = ? AND ts > ?" 才进入下一流程。

在LangChain等框架中嵌入校验层的实际做法

不要依赖LLM自己“说它安全了”。必须在调用插件前加一道中间校验:

  • 用 sqlparse(Python)或 node-sql-parser(JS)做语法树解析,遍历 WHERE 子句中的 Comparison 节点,确认右侧全是参数节点,而非 StringLiteral
  • 对 LangChain 的 SQLDatabaseChain 或自定义工具函数,包装一层 validate_sql_safety(sql: string): boolean,返回 false 时抛出 ValueError("Unsafe SQL detected") 并中断执行
  • 禁用 allow_dml=True 以外的 DML 操作,尤其防止模型生成 INSERT INTO users VALUES (...) 后又自动补上 SELECT * FROM users —— 这类链式操作需显式白名单控制

最容易被忽略的盲区:ORM映射层与日志回显

即使SQL本身通过了沙盒,风险仍可能从别处漏出:

  • ORM 如 SQLAlchemy 的 text() 构造器若传入未校验的字符串,等于绕过所有参数化保护;校验必须覆盖 text("...") 内容本身
  • 错误日志如果把原始SQL连同用户输入一起打出来(如 logger.error(f"Query failed: {raw_sql}")),会把注入payload直接暴露在日志系统中,形成信息泄露通道
  • 某些数据库驱动(如旧版 psycopg2)支持 execute("""SELECT %s""", (user_input,)),但若 user_input 是 "'a'; DROP TABLE users; --",仍会因驱动版本或配置问题执行多语句——沙盒必须检测分号和多语句迹象

真正的防护不在“生成端”,而在“执行前最后一毫秒”。任何跳过校验直连数据库的路径,都是给提示注入留的后门。

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

热门AI工具

更多
Seko
Seko Hot

一款AI视频创作工具,主要用于商汤科技推出的创编一体的AI短视频创作Agent,适合需要提升相关任务效率的用户。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

UpDream
UpDream Hot

一款AI视频创作工具,主要用于哔哩哔哩推出的自研AI视频创作工具,适合需要提升相关任务效率的用户。

Loomy
Loomy Hot

一款AI工具,主要用于科大讯飞发布的桌面级 AI 助理,比 OpenClaw 更易用、更安全!,适合需要提升相关任务效率的用户。

讯飞智作

讯飞智作是一款AI视频创作工具,AI文本配音工具,数字人课程、营销视频制作。

立刻MV
立刻MV Hot

立刻MV是一款AI文本写作工具,AI 音乐视频(MV)创作工具。

音述AI
音述AI Hot

一款AI音频处理工具,主要用于音述AI是一个以“用声音述说故事”为核心的 AI 音乐创作与声音分享社区,适合需要提升相关任务效率的用户。

相关专题

更多
数据分析工具有哪些
数据分析工具有哪些

数据分析工具有Excel、SQL、Python、R、Tableau、Power BI、SAS、SPSS和MATLAB等。详细介绍:1、Excel,具有强大的计算和数据处理功能;2、SQL,可以进行数据查询、过滤、排序、聚合等操作;3、Python,拥有丰富的数据分析库;4、R,拥有丰富的统计分析库和图形库;5、Tableau,提供了直观易用的用户界面等等。

4303

2023.10.12

SQL中distinct的用法
SQL中distinct的用法

SQL中distinct的语法是“SELECT DISTINCT column1, column2,...,FROM table_name;”。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

911

2023.10.27

SQL中months_between使用方法
SQL中months_between使用方法

在SQL中,MONTHS_BETWEEN 是一个常见的函数,用于计算两个日期之间的月份差。想了解更多SQL的相关内容,可以阅读本专题下面的文章。

1109

2024.02.23

SQL出现5120错误解决方法
SQL出现5120错误解决方法

SQL Server错误5120是由于没有足够的权限来访问或操作指定的数据库或文件引起的。想了解更多sql错误的相关内容,可以阅读本专题下面的文章。

6221

2024.03.06

sql procedure语法错误解决方法
sql procedure语法错误解决方法

sql procedure语法错误解决办法:1、仔细检查错误消息;2、检查语法规则;3、检查括号和引号;4、检查变量和参数;5、检查关键字和函数;6、逐步调试;7、参考文档和示例。想了解更多语法错误的相关内容,可以阅读本专题下面的文章。

3063

2024.03.06

oracle数据库运行sql方法
oracle数据库运行sql方法

运行sql步骤包括:打开sql plus工具并连接到数据库。在提示符下输入sql语句。按enter键运行该语句。查看结果,错误消息或退出sql plus。想了解更多oracle数据库的相关内容,可以阅读本专题下面的文章。

6200

2024.04.07

sql中where的含义
sql中where的含义

sql中where子句用于从表中过滤数据,它基于指定条件选择特定的行。想了解更多where的相关内容,可以阅读本专题下面的文章。

8381

2024.04.29

sql中删除表的语句是什么
sql中删除表的语句是什么

sql中用于删除表的语句是drop table。语法为drop table table_name;该语句将永久删除指定表的表和数据。想了解更多sql的相关内容,可以阅读本专题下面的文章。

1130

2024.04.29

OpenJDK JVM参数怎么调优
OpenJDK JVM参数怎么调优

Java应用运行缓慢,如何调整JVM参数?本专题介绍OpenJDK JVM性能调优的基本流程,涵盖性能指标确认、运行数据采集、内存配置和垃圾回收分析,帮助你先定位瓶颈,再通过对照测试验证效果,避免直接套用参数而影响应用稳定性。

0

2026.10.16

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
热门推荐
/
最新课程
关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn