CTF-web:查看python代码抽象语法树

抽象语法树（Abstract Syntax Tree，简称 AST）是源代码的树状表示，展示了代码的语法结构。在 Python 中，ast 模块可以帮助你解析和操作 Python 代码的 AST。

1. 使用 ast 模块解析 Python 代码

我们可以使用 ast.parse 将一段 Python 代码解析为 AST，然后使用 ast.dump 查看其结构。

import ast # 示例代码 code = """ def add(a, b): return a + b result = add(5, 3) print(result) """ # 解析代码为 AST tree = ast.parse(code) # 打印 AST 的结构 print(ast.dump(tree, indent=4))

输出示例：

Module( body=[ FunctionDef( name='add', args=arguments( posonlyargs=[], args=[ arg(arg='a'), arg(arg='b')], kwonlyargs=[], kw_defaults=[], defaults=[]), body=[ Return( value=BinOp( left=Name(id='a', ctx=Load()), op=Add(), right=Name(id='b', ctx=Load())))], decorator_list=[], type_params=[]), Assign( targets=[ Name(id='result', ctx=Store())], value=Call( func=Name(id='add', ctx=Load()), args=[ Constant(value=5), Constant(value=3)], keywords=[])), Expr( value=Call( func=Name(id='print', ctx=Load()), args=[ Name(id='result', ctx=Load())], keywords=[]))], type_ignores=[]) 2. 遍历 AST

你可以编写一个访问者（Visitor），遍历 AST 的各个节点，并对特定类型的节点进行处理。例如，下面的代码将遍历 AST 并打印出所有函数定义的名称：

import ast class FunctionNameLister(ast.NodeVisitor): def visit_FunctionDef(self, node): print(f"Function name: {node.name}") # 继续遍历函数内部的节点 self.generic_visit(node) # 示例代码 code = """ def add(a, b): return a + b def subtract(a, b): return a - b """ # 解析代码为 AST tree = ast.parse(code) # 创建访问者并遍历 AST lister = FunctionNameLister() lister.visit(tree)

输出：

Function name: add Function name: subtract