01-使用大语言模型检测Solidity智能合约中利用AI自动生成DApp交互界面的逻辑重入漏洞的有效性

本文首发于「瑞瑞的区块链安全实验室」

作者：瑞瑞（欧阳瑞）· 智能合约安全工程师
AI 辅助写作声明：本文标题和交互界面代码由 AI 生成，但安全分析由人类完成
Hash 全程在键盘旁监督——虽然它只是在找暖和的地方趴着

Hash 今天特别黏人。我在写 Prompt 的时候，它直接爬到了我手背上，暖暖的小爪子贴着我皮肤。

"别闹，我在让 GPT 给这个借贷合约生成前端界面。"我轻轻吹了口气，Hash 的舌头缩了回去。

生成式 AI 正在彻底改变 DApp 的开发范式。开发者不再需要手动编写整个前端——只需一段 Prompt，AI 就能生成包含 Wagmi 集成、ABI 绑定、交易发起的完整组件。但问题是：这些 AI 生成的代码安全吗？

今天就深入分析 AI 生成 DApp 交互界面中可能引入的重入漏洞，以及如何通过 AI 自检和人工审计构建双重防线。

一、AI 生成 DApp 前端的典型工作流

flowchart TD subgraph "AI 生成阶段" A[开发者 Prompt] --> B[LLM<br/>GPT-4 / Claude] B --> C[生成 React + Wagmi 组件] B --> D[生成 ABI 绑定代码] B --> E[生成交易逻辑] end subgraph "安全风险注入点" C --> F[⚡ 风险1: 状态管理缺失] D --> G[⚡ 风险2: 类型处理不当] E --> H[⚡ 风险3: 重入漏洞逻辑] end subgraph "安全检测" F --> I[静态分析] G --> I H --> I I --> J[AI 自检] I --> K[人工审计] end subgraph "输出" J --> L[安全 DApp 前端] K --> L end style A fill:#4a90d9,color:#fff style B fill:#50c878,color:#fff style C fill:#ff6b6b,color:#fff style D fill:#ff6b6b,color:#fff style E fill:#ff6b6b,color:#fff style J fill:#96ceb4,color:#fff style K fill:#f4d03f,color:#333 style L fill:#9b59b6,color:#fff

二、AI 生成代码中的重入漏洞模式

2.1 典型场景：AI 生成的借贷 DApp 前端

我让 GPT-4 生成一个去中心化借贷 DApp 的withdraw交互组件。以下是 AI 生成的代码：

// ⚠️ AI 生成的 DApp 交互代码（包含重入漏洞） import { useWriteContract, useWaitForTransactionReceipt } from 'wagmi' import { parseEther, formatEther } from 'viem' import { useState } from 'react' // 合约 ABI const lendingAbi = [ { name: 'withdraw', type: 'function', stateMutability: 'nonpayable', inputs: [{ name: '_amount', type: 'uint256' }], outputs: [], }, { name: 'deposit', type: 'function', stateMutability: 'payable', inputs: [], outputs: [], }, { // ❌ AI 生成的 ABI 遗漏了关键事件 name: 'withdraw', type: 'function', stateMutability: 'nonpayable', // 应标记为 payable 才能接收 ETH inputs: [{ name: '_amount', type: 'uint256' }], outputs: [], }, ] function WithdrawComponent() { const [amount, setAmount] = useState('') const [isPending, setIsPending] = useState(false) const { writeContract } = useWriteContract() const { isLoading, isSuccess } = useWaitForTransactionReceipt({ hash: undefined, // ❌ 未正确处理 tx hash }) // ❌ VULN-1: 缺乏重入锁 const handleWithdraw = async () => { // ❌ VULN-2: 未校验前端状态一致性 setIsPending(true) try { await writeContract({ address: '0x...', // 合约地址 abi: lendingAbi, functionName: 'withdraw', args: [parseEther(amount)], }) // ❌ VULN-3: 交易确认后未更新前端状态 // 未调用 refetch 或 invalidateQueries // ❌ VULN-4: 乐观 UI 更新可能导致状态不一致 setBalance(prev => prev - Number(amount)) } catch (error) { console.error('Withdraw failed:', error) // ❌ VULN-5: 失败后未回滚状态 } finally { setTimeout(() => setIsPending(false), 1000) // ❌ 硬编码延迟 } } const handleDeposit = async () => { // ❌ 存款函数同样存在重入风险 // AI 未生成任何安全检查 writeContract({ address: '0x...', abi: lendingAbi, functionName: 'deposit', value: parseEther(amount), }) } return ( <div> <input type="number" value={amount} onChange={(e) => setAmount(e.target.value)} placeholder="Enter amount in ETH" /> <button onClick={handleDeposit} disabled={isPending}> Deposit </button> <button onClick={handleWithdraw} disabled={isPending}> Withdraw </button> {isLoading && <p>Transaction pending...</p>} {isSuccess && <p>Transaction confirmed!</p>} </div> ) }

2.2 漏洞分析

三、AI 自检：让 AI 审计自己生成的代码

一个有意思的实验：让同一个 AI 模型审计它自己生成的代码，看看效果如何。

3.1 AI 自检 Prompt

你是一个智能合约安全审计专家。请审计以下 AI 生成的 DApp 前端代码， 特别关注重入漏洞、状态一致性问题、以及前端-合约交互中的数据安全。 请按以下格式输出： 1. 发现的安全问题（按严重级别排序） 2. 每个问题的重现步骤 3. 修复建议 + 修复后的代码示例

3.2 AI 自检结果对比

我测试了多个 LLM 对自己生成代码的自检能力：

核心发现：AI 自检不能完全替代人工审计，但对常见漏洞的检出率已经相当可观。Claude 3.5 Sonnet 在本测试中表现最佳，全检出但有一个误报。

四、AI 修复后的安全代码

通过 AI 自检 + 人工审计的双重保障后，我们得到了安全版本：

// ✅ AI 自检 + 人工审计后的安全 DApp 组件 import { useWriteContract, useWaitForTransactionReceipt, useBalance } from 'wagmi' import { parseEther } from 'viem' import { useState, useCallback } from 'react' // ✅ FIX-1: 完整 ABI 定义，包含事件 const lendingAbi = [ { name: 'withdraw', type: 'function', stateMutability: 'nonpayable', inputs: [{ name: '_amount', type: 'uint256' }], outputs: [], }, { name: 'deposit', type: 'function', stateMutability: 'payable', inputs: [], outputs: [], }, { // ✅ 事件定义用于状态同步 name: 'Withdraw', type: 'event', inputs: [ { name: 'user', type: 'address', indexed: true }, { name: 'amount', type: 'uint256', indexed: false }, ], }, { name: 'Deposit', type: 'event', inputs: [ { name: 'user', type: 'address', indexed: true }, { name: 'amount', type: 'uint256', indexed: false }, ], }, ] // ✅ FIX-2: 重入锁 Hook function useReentrancyGuard() { const [locked, setLocked] = useState(false) const withGuard = useCallback(async <T,>(fn: () => Promise<T>): Promise<T> => { if (locked) { throw new Error('Reentrancy detected!') } setLocked(true) try { return await fn() } finally { setLocked(false) } }, [locked]) return { withGuard, isLocked: locked } } function SecureWithdrawComponent({ userAddress }: { userAddress: `0x${string}` }) { const [amount, setAmount] = useState('') const [txHash, setTxHash] = useState<`0x${string}` | undefined>() const [error, setError] = useState<string | null>(null) const { withGuard } = useReentrancyGuard() const { writeContractAsync } = useWriteContract() const { data: balance, refetch: refetchBalance } = useBalance({ address: userAddress, }) // ✅ FIX-3: 使用 writeContractAsync 等待交易结果 const { isLoading: isConfirming } = useWaitForTransactionReceipt({ hash: txHash, }) const handleWithdraw = async () => { setError(null) const amountWei = parseEther(amount) try { await withGuard(async () => { // ✅ FIX-4: 前端预校验 if (balance && balance.value < amountWei) { throw new Error('Insufficient balance') } // 发送交易 const hash = await writeContractAsync({ address: '0x...', abi: lendingAbi, functionName: 'withdraw', args: [amountWei], }) setTxHash(hash) // ✅ FIX-5: 等待交易确认后刷新状态 // waitForTransactionReceipt 由 useWaitForTransactionReceipt 处理 }) } catch (err) { // ✅ FIX-6: 错误处理 + 状态回滚 const message = err instanceof Error ? err.message : 'Transaction failed' setError(message) console.error('Withdraw failed, state rolled back:', message) } } // ✅ FIX-7: 交易确认后自动刷新链上状态 const onConfirmation = () => { refetchBalance() setAmount('') setTxHash(undefined) } return ( <div> <div> Balance: {balance ? `${balance.formatted} ${balance.symbol}` : 'Loading...'} </div> <input type="number" value={amount} onChange={(e) => setAmount(e.target.value)} placeholder="Enter amount in ETH" min="0" step="0.001" /> <button onClick={handleWithdraw} disabled={isConfirming || !amount || parseEther(amount) <= 0n} > {isConfirming ? 'Confirming...' : 'Withdraw'} </button> {isConfirming && <p>Transaction pending: {txHash}</p>} {onConfirmation && balance && <p>Updated balance: {balance.formatted}</p>} {error && <p style={{ color: 'red' }}>Error: {error}</p>} </div> ) }

五、AI 自检 vs 人工审计的交叉验证

graph LR subgraph "AI 生成" A[LLM 生成 DApp 前端] end subgraph "第一道防线: AI 自检" B[LLM 自检] C[Slither 静态分析] end subgraph "第二道防线: 人工审计" D[代码审查] E[渗透测试] F[Gas 分析] end subgraph "最终交付" G[安全审计报告] end A --> B A --> C B --> D C --> D D --> E D --> F E --> G F --> G style A fill:#50c878,color:#fff style B fill:#4a90d9,color:#fff style C fill:#4a90d9,color:#fff style D fill:#f4d03f,color:#333 style E fill:#f4d03f,color:#333 style F fill:#f4d03f,color:#333 style G fill:#9b59b6,color:#fff

检测效率对比

我对 20 个 AI 生成的 DApp 前端代码做了统计：

六、AI 生成 DApp 的安全审计规约

基于本实验，我总结了 8 条针对 AI 生成 DApp 代码的审计规约：

七、AI 自检 Prompt 模板

最后分享一个我调试多次后确认有效的 AI 自检 Prompt 模板：

你是一个区块链安全审计专家，专注于智能合约前端（DApp）安全。 请严格检查以下 TypeScript/React 代码中的安全隐患，重点关注： 1. 🔴 重入漏洞：前端状态锁机制是否完备 2. 🟡 状态一致性：交易状态管理是否正确（pending/confirm/rollback） 3. 🟡 类型安全：bigint/number 使用是否正确 4. 🔴 前端-合约数据一致性：预校验与实际合约逻辑是否一致 5. 🟢 Gas 优化：有无不必要的状态更新 请按以下 JSON 格式输出： { "vulnerabilities": [ { "severity": "critical|high|medium|low", "type": "reentrancy|state_inconsistency|type_error|logic_error", "location": "文件路径:行号", "description": "问题描述", "fix": "修复建议" } ], "summary": { "total": 数量, "critical": 数量, "high": 数量, "medium": 数量, "low": 数量 }, "verdict": "PASS|FAIL|NEED_REVIEW" }

八、写在最后

Hash 从我手背上跳下来，走到它的水盆边喝了几口水。阳光透过窗子照在它的鳞片上，折射出彩虹般的光泽。

我保存好那份经过 AI 自检 + 人工审计的安全 DApp 代码，关上了 IDE。

"你说，AI 以后会不会自己修复它生成的所有漏洞？"

Hash 抬起头看着我，然后打了个哈欠。我把它抱起来，放进饲养箱。

"也是，能自检就已经很不错了。"我摸了摸它，"明天我们做个更大的实验——让不同的 LLM 比赛找重入漏洞。"

Hash 在加热垫上翻了个身，露出白色的肚皮——那是它最放松的姿势。

References:

• OWASP Smart Contract Top 10: Reentrancy Attacks
• Wagmi Documentation: useWriteContract vs useWriteContractAsync
• "Can LLMs Write Secure Code?" — 2024 ACM CCS Workshop
• Trail of Bits: AI-Assisted Smart Contract Auditing