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

酷浩小哥_3396

酷浩小哥_3396

2026-05-12

593人浏览

原创

不能直接信任大模型生成的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; --",仍会因驱动版本或配置问题执行多语句——沙盒必须检测分号和多语句迹象

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

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

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

相关专题

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

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

2023.10.12

4183

8

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

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

2023.10.27

891

4

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

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

2024.02.23

1089

5

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

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

2024.03.06

6081

10

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

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

2024.03.06

2963

4

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

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

2024.04.07

6060

11

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

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

2024.04.29

8141

6

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

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

2024.04.29

1110

5

sql中删除一列的命令是什么
sql中删除一列的命令是什么

在sql中,使用alter table语句可以删除一列,语法为:alter table table_name drop column column_name。想了解更多sql的相关内容,可以阅读本专题下面的文章。

2024.04.29

972

5

热门下载

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

精品课程

更多
热门推荐
/
最新课程
phpStudy极速入门视频教程
phpStudy极速入门视频教程

共6课时 | 54.6万人学习

独孤九贱(4)_PHP视频教程
独孤九贱(4)_PHP视频教程

共89课时 | 133.4万人学习