2025-11-08 17:01:38

使用 Wake Printer 脚本实现 Solidity 智能合约审计自动化

摘要

前提条件和设置在本教程中，我们将通过一个示例项目（workshop 仓库）演示如何利用 Wake Printer 脚本简化 Solidity 审计流程： wake up Bash 复制运行以下命令以启动 Wake 打印机： wake print Bash 复制按名称运行特定的打印机：教程 1：创建你的第一个打印机 – 列出合约我们从一个简单的例子开始：创建一个列出项目中所有合约的打印机。这个示例将帮助你掌握复杂分析中的核心概念。创建打印机结构运行以下命令来搭建你的第

前提条件和设置

在本教程中，我们将通过一个示例项目（workshop 仓库）演示如何利用 Wake Printer 脚本简化 Solidity 审计流程：

wake up

Bash

复制

运行以下命令以启动 Wake 打印机：

wake print

Bash

复制

按名称运行特定的打印机：

教程 1：创建你的第一个打印机 – 列出合约

我们从一个简单的例子开始：创建一个列出项目中所有合约的打印机。这个示例将帮助你掌握复杂分析中的核心概念。

创建打印机结构

运行以下命令来搭建你的第一个打印机：

from __future__ import annotationsimport networkx as nximport rich_click as clickimport wake.ir as irimport wake.ir.types as typesfrom rich importprintfrom wake.cli import SolidityNamefrom wake.printers import Printer, printerclass ListContractsPrinter(Printer):def print(self) -> None:pass @printer.command(name="list-contracts")def cli(self) -> None:pass

Python

复制

以下是模板中每个部分的作用：

print()：显示分析结果的主要执行方法。
cli()：用于自定义参数的命令行界面处理程序。

实现访问者模式

Wake 使用访问者模式遍历合约的抽象语法树 (AST)。这种模式允许自动导航代码结构，并对特定元素（如合约或函数定义）做出反应。

为了列出合约，我们将覆盖 visit_contract_definition 方法，该方法会为代码库中的每个合约调用。

将此方法添加到你的 ListContractsPrinter 类：

wake print list-contracts

Bash

复制

此命令运行你的打印机并打印项目中找到的所有合约名称。

改进输出

基本实现会显示所有合约，包括接口和继承的合约。让我们改进它，仅显示可部署的合约：

def visit_contract_definition(self, node: ir.ContractDefinition) -> None:iflen(node.child_contracts) != 0:returnif node.kind != ir.enums.ContractKind.CONTRACT:returnprint(node.name)

Python

复制

ContractDefinition 类包含可用于过滤结果的属性。有关完整参考，请参见：https://ackee.xyz/wake/docs/latest/api-reference/ir/declarations/contract-definition/

完整实现

这是最终版本，具有适当的关注点分离——在遍历期间收集数据并在 print() 方法中显示它。

教程 2：分析合约函数

了解哪些函数可从外部调用对于安全性至关重要：公共 'withdraw' 或 'transfer' 函数通常定义了合约的攻击面。让我们创建一个打印机，通过列出所有公共和外部函数来绘制攻击面。

设置函数打印机

创建一个新的打印机：

class ListFunctionsPrinter(Printer): contracts: list[ir.ContractDefinition] = []def visit_contract_definition(self, node: ir.ContractDefinition) -> None: self.contracts.append(node)

Python

复制

处理继承层次结构

在 print() 方法中，我们遍历从基合约到派生合约的继承层次结构，显示每个级别的可调用函数：

def get_callable_final_functions(self, contract: ir.ContractDefinition) -> list[ir.FunctionDefinition]:return [\ func for func in contract.functions\iflen(func.child_functions) == 0# 是最终实现\and func.visibility in [ir.enums.Visibility.PUBLIC, ir.enums.Visibility.EXTERNAL]\ ]

Python

复制

运行函数打印机

执行打印机以查看继承层次结构和可调用函数：

Contract: ContextContract: OwnableFunctions: owner renounceOwnership transferOwnershipContract: SingleTokenVaultFunctions: constructor deposit withdraw emergencyWithdraw balanceOf setDepositLimits--------------------Contract: EIP712ExampleFunctions: constructor DOMAIN_SEPARATOR castVoteBySignature getVoteCounts--------------------Contract: ContextContract: IERC20Contract: IERC20MetadataContract: IERC20ErrorsContract: ERC20Functions: name symbol decimals totalSupply balanceOf transfer allowance approve transferFromContract: IERC20PermitContract: IERC5267Contract: EIP712Functions: eip712DomainContract: NoncesContract: ERC20PermitFunctions: permit nonces DOMAIN_SEPARATORContract: PermitTokenFunctions: constructor--------------------Contract: TokenFunctions: constructor mintTokens transfer transferWithBytes getBalance--------------------Contract: ContextContract: IERC20Contract: IERC20MetadataContract: IERC20ErrorsContract: ERC20Functions: name symbol decimals totalSupply balanceOf transfer allowance approve transferFromContract: MockERC20Functions: constructor--------------------

Bash

复制

输出为你提供每个合约的继承和可调用入口点的快速可视化地图。