`RestrictedPython` 是一个用于创建安全执行环境的库，它通过限制 Python 内置函数、模块和语法来防止潜在危险操作。它可以用于在线编程平台、插件系统等需要运行不受信任代码的场景。 --- ## 示例：使用 `RestrictedPython` 创建沙盒 首先安装 `RestrictedPython`： ```bash pip install RestrictedPython ``` 然后可以使用以下代码创建一个基本的沙盒环境： ```python from RestrictedPython import compile_restricted from RestrictedPython.Guards import safe_builtins # 用户提供的不受信任代码 user_code = """ def calc_sum(a, b): return a + b print(calc_sum(3, 5)) """ # 编译受限代码 byte_code = compile_restricted( source=user_code, filename='<inline code>', mode='exec' ) # 定义全局命名空间 global_namespace = { '__builtins__': safe_builtins, # 使用受限的内置函数 } # 执行受限代码 try: exec(byte_code, global_namespace) except Exception as e: print("Error:", e) ``` ### 输出： ``` 8 ``` --- ## 解释： - **compile_restricted**: 将源码编译为字节码，并在编译阶段进行语法检查，禁止某些不安全结构（如 `import`、`exec`、`eval`）。 - **safe_builtins**: 提供一组安全的内置函数（如 `len`, `range`, `int`），禁用 `open`, `eval`, `exec` 等危险函数。 - **exec(byte_code, global_namespace)**: 在指定的命名空间中执行代码，隔离对全局环境的影响。 --- ## 自定义限制：阻止特定函数或变量访问 你可以通过继承 `ExecutionFilter` 或自定义 `__builtins__` 来进一步限制行为： ```python from RestrictedPython import compile_restricted from RestrictedPython.Guards import safe_builtins, guarded_iter_unpack_sequence # 自定义安全 builtins restricted_builtins = dict(safe_builtins) restricted_builtins['_getiter_'] = iter restricted_builtins['_unpack_sequence_'] = guarded_iter_unpack_sequence # 用户代码尝试导入模块或执行危险操作会被拒绝 user_code = """ import os os.system('echo "This should not run!"') """ try: byte_code = compile_restricted( source=user_code, filename='<inline code>', mode='exec' ) global_namespace = {'__builtins__': restricted_builtins} exec(byte_code, global_namespace) except Exception as e: print("Compilation or execution error:", e) ``` ### 输出： ``` Compilation or execution error: Line 2: SyntaxError: import is not allowed ``` --- ## 更多控制：限制函数调用、属性访问 你还可以使用 `AccessPolicy` 和 `RestrictionDecider` 来控制对象属性访问和方法调用。 例如，限制只允许访问某些类的方法： ```python from RestrictedPython import compile_restricted from RestrictedPython.Guards import safe_builtins, full_write_guard class MySafeClass: def greet(self): return "Hello from safe class!" # 用户代码 user_code = """ obj = MyClass() print(obj.greet()) """ # 自定义命名空间 global_namespace = { '__builtins__': safe_builtins, 'MyClass': MySafeClass, } # 编写写保护钩子 def custom_write_guard(obj, name, value): print(f"Write to {name} on {obj} is restricted") raise AttributeError("Attribute write not allowed") # 设置写保护 global_namespace['__write__'] = custom_write_guard # 编译并执行 try: byte_code = compile_restricted(user_code, '<code>', 'exec') exec(byte_code, global_namespace) except Exception as e: print("Error:", e) ``` --- ## 总结 | 功能 | 实现方式 | |------|----------| | 防止 `import`、`exec` 等语法 | 使用 `compile_restricted` | | 控制内置函数 | 替换 `__builtins__` 为 `safe_builtins` | | 限制属性/方法访问 | 使用 `guarded_iter_unpack_sequence` 和自定义钩子 | | 拦截对象修改 | 使用 `__write__` 钩子 | ---