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

抽象语法树（Abstract Syntax Tree，简称 AST）是源代码的树状表现，展示了代码的语法结构。在 Python 中，ast 模块可以资助你剖析和操纵 Python 代码的 AST。
1. 使用 ast 模块剖析 Python 代码

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

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

复制代码

输出示例：

1. Module(
2.     body=[
3.         FunctionDef(
4.             name='add',
5.             args=arguments(
6.                 posonlyargs=[],
7.                 args=[
8.                     arg(arg='a'),
9.                     arg(arg='b')],
10.                 kwonlyargs=[],
11.                 kw_defaults=[],
12.                 defaults=[]),
13.             body=[
14.                 Return(
15.                     value=BinOp(
16.                         left=Name(id='a', ctx=Load()),
17.                         op=Add(),
18.                         right=Name(id='b', ctx=Load())))],
19.             decorator_list=[],
20.             type_params=[]),
21.         Assign(
22.             targets=[
23.                 Name(id='result', ctx=Store())],
24.             value=Call(
25.                 func=Name(id='add', ctx=Load()),
26.                 args=[
27.                     Constant(value=5),
28.                     Constant(value=3)],
29.                 keywords=[])),
30.         Expr(
31.             value=Call(
32.                 func=Name(id='print', ctx=Load()),
33.                 args=[
34.                     Name(id='result', ctx=Load())],
35.                 keywords=[]))],
36.     type_ignores=[])

复制代码

2. 遍历 AST

你可以编写一个访问者（Visitor），遍历 AST 的各个节点，并对特定范例的节点举行处置处罚。比方，下面的代码将遍历 AST 并打印出全部函数界说的名称：

1. import ast
3. class FunctionNameLister(ast.NodeVisitor):
4.     def visit_FunctionDef(self, node):
5.         print(f"Function name: {node.name}")
6.         # 继续遍历函数内部的节点
7.         self.generic_visit(node)
9. # 示例代码
10. code = """
11. def add(a, b):
12.     return a + b
14. def subtract(a, b):
15.     return a - b
16. """
18. # 解析代码为 AST
19. tree = ast.parse(code)
21. # 创建访问者并遍历 AST
22. lister = FunctionNameLister()
23. lister.visit(tree)

复制代码

输出：

1. Function name: add
2. Function name: subtract

复制代码