本文探讨在动态导入场景下，如何防止目录遍历与非法模块加载，重点分析 importlib.import_module() 的安全性边界，并提供基于存在性校验的可靠防护方案。
本文探讨在动态导入场景下，如何防止目录遍历与非法模块加载，重点分析 `importlib.import_module()` 的安全性边界，并提供基于存在性校验的可靠防护方案。
直接拼接用户输入构建模块路径（如 f"trusted_directory.{submodule_name}"）看似可控，实则存在严重安全隐患。尽管 Python 的模块导入语法确实仅将 . 视为层级分隔符（PEP 328 及后续规范均未赋予 /, \, .., *, ~ 等字符在模块名中的特殊语义），但模块名本身不等于文件系统路径——importlib.import_module() 依赖 sys.path 和已注册的 importer（如 PathFinder）解析模块。只要攻击者能控制 submodule_name，就可能触发非预期行为：
- submodule_name = "os"：若 trusted_directory 位于 sys.path 前端，而标准库 os 恰好未被提前导入，importlib.import_module("trusted_directory.os") 会失败；但若用户输入为 "__main__" 或 "builtins"，且环境异常（如自定义 importer），风险陡增；
- submodule_name = "subpkg.submodule"：split(".")[0] 截断逻辑失效，导致 f"trusted_directory.subpkg.submodule" 被导入——这虽仍在 trusted_directory 下，但已脱离“仅允许一级子模块”的原始约束；
- 最危险的是 submodule_name = "..malicious"：虽然 .. 在模块名中不被 Python 解析为路径向上跳转（模块名是命名空间标识符，非文件路径），但若 trusted_directory 是一个普通包（含 __init__.py），其子包结构可能被恶意构造（如 trusted_directory.__path__ 被篡改），或通过 zipimport / 自定义 MetaPathFinder 引入外部源。
因此，仅靠字符串截断和点号分割无法保证安全。真正可靠的防护必须落实到文件系统层面的存在性验证，即：确保待导入的 .py 文件真实存在于预期目录中，且无符号链接绕过、大小写混淆等隐患。
推荐采用以下加固方案：
import importlib
import os
import pathlib
def safe_import_from_trusted(user_input: str, trusted_root: str = "trusted_directory") -> object:
# 1. 严格清洗：只保留字母、数字、下划线、点号，禁止路径字符
if not all(c.isalnum() or c in "._" for c in user_input):
raise ValueError("Invalid characters in module name")
# 2. 分割并限制层级（例如仅允许一级子模块）
parts = user_input.split(".")
if len(parts) > 1:
raise ValueError("Only top-level modules allowed")
submodule_name = parts[0]
# 3. 构建绝对路径并验证存在性（关键！）
target_path = pathlib.Path(trusted_root) / f"{submodule_name}.py"
# 防御符号链接：resolve() 强制解析真实路径，并检查是否仍在 trusted_root 内
try:
resolved_path = target_path.resolve(strict=True)
if not str(resolved_path).startswith(str(pathlib.Path(trusted_root).resolve())):
raise ValueError("Path traversal attempt detected")
except FileNotFoundError:
raise ImportError(f"Module '{submodule_name}' not found in {trusted_root}")
# 4. 动态导入
full_module_name = f"{trusted_root}.{submodule_name}"
return importlib.import_module(full_module_name)
# 使用示例
try:
user_input = "report_generator" # 安全输入
mod = safe_import_from_trusted(user_input)
mod.main()
except (ValueError, ImportError, AttributeError) as e:
print(f"Import failed: {e}")
关键注意事项：
- ✅ 必须使用 pathlib.Path.resolve(strict=True)：检测并拒绝符号链接指向外部目录；
- ✅ 必须校验 resolved_path 的父目录是否严格等于 trusted_root 的绝对路径：防止 ../../etc/passwd 类绕过（尽管模块名不解析 ..，但文件系统操作需独立防护）；
- ❌ 避免仅用 glob（如答案中所示）：glob 不处理符号链接、权限、大小写敏感等问题，且 root_dir 参数在旧版 Python 中有兼容性风险；
- ⚠️ 永远不要信任 __import__ 或 eval(compile(...))：它们绕过所有模块加载器约束，直接执行任意代码。
总结：模块导入的安全性不取决于语法规范，而取决于运行时对文件系统访问的精确控制。以存在性校验为基石，辅以路径规范化与白名单式字符过滤，才能构建真正可信的动态导入机制。
Python免费学习笔记（深入）：立即使用
在学习笔记中，你将探索 Python 的核心概念和高级技巧！
