AI 代码的安全检查
AI 生成的代码上线前,过一遍安全清单:密钥硬编码、越权访问、注入——三分钟避免一次事故。
AI 写代码又快又勤,但不会自动替你写安全的代码。它生成的代码里,翻来覆去就三类坑:密钥硬编码、越权访问、注入。上线前过一遍,三分钟就能避免一次事故。
你会遇到的现象:
- API 密钥明晃晃写在代码里,跟着仓库一起上了云
- 接口只查了「登录了没」,没查「能不能看这条数据」
- 用户输入直接拼进了数据库查询字符串
三类高频坑
先知道坑长什么样,才知道要查什么:
- **密钥硬编码。**把 API 密钥、数据库密码、签名密钥直接写进代码或提交到仓库。一旦代码入库上云,密钥等于公开——攻击者拿它做合法调用,账单记在你头上。
- **越权访问。**接口校验了「是否登录」,但没校验「能不能看这条数据」。用户 A 登录后能看用户 B 的订单、改用户 B 的资料。AI 生成的 CRUD 接口尤其容易漏这一层。owasp-bac
- 注入。把用户输入直接拼进 SQL、命令、HTML。本质是把「数据」当「指令」——AI 生成代码时常用字符串拼接,恰好是最危险的那种写法。参见提示注入。owasp-injection
三分钟检查清单
不管代码是 AI 写的还是人写的,上线前用这份清单过一遍:
- 一、扫密钥。用扫描工具把全仓库搜一遍:密钥、密码、Token。发现已提交的,立刻吊销重发——只是删掉远远不够,删掉的那份已经泄了。
- **二、查每个接口的权限。**逐接口问两遍:有没有校验登录?有没有校验「当前用户对这条数据的访问权」?测试方法很简单——用 A 的会话去访问 B 的资源(比如
GET /orders/B-1024),返回 403 才是查了权限,返回 200 说明只查了「登录没」。 - **三、查所有「拼」的地方。**凡是用户输入(或任何外部输入)进入查询、命令、HTML 的地方,都要是参数化写法,而不是字符串拼接。
- **四、让 AI 自己先自查。**生成代码时在提示词里加一句:「写出符合安全规范的版本:密钥不硬编码、每个接口校验权限、用户输入全部参数化。」效果立竿见影,成本为零。
- **五、当流程走。**AI 代码要走和人写代码一样的评审、扫描、测试。恰恰因为代码量上来了,审查更不能省。参见走查清单。
为什么 AI 特别容易犯
三个原因,都跟 AI 的工作方式有关:
- **它学的是「常见写法」。**AI 从海量代码里学「一般怎么写」。而公开代码里,密钥硬编码、不校验权限的示例比比皆是——它学到的「正常」恰好就是不安全的。
- **它只看到局部。**生成一段代码时,它看不到你的权限系统、密钥管理、安全规范。为了先让功能跑通,它默认把密钥写进代码里。
- **它追求「能跑」,不追求「安全」。**安全是隐性质量要求,除非你在提示词里明确要求,它不会主动加。
原理讲得更细的版本在AI 代码的常见漏洞。这里只补一条认知:**别问「AI 为什么老写不安全代码」,问你自己为什么不审查。**工具提速不等于质量提高,安全这条线本来就要人来守。交给 AI 之前,先判断这单活配不配——参见什么能交给 AI。
最后把这份清单存成文件,每次上线前照着走一遍,走完就勾掉归档。三分钟是它每次花的时间,但它挡住的是一次可能花掉一整周的事故。
