基于不可信输入安全导入指定目录Python模块的可行性问询

目录遍历风险：不存在突破可能 #

你的代码中submodule_name = user_input.split(".")[0]的处理，结合Python的导入机制，不会出现目录遍历的情况：

• 如果用户输入类似"..os"，分割后得到".."，但Python的模块名不允许包含..（模块名必须符合标识符规则：仅字母、数字、下划线，不能以数字开头），尝试导入trusted_directory..会直接抛出ImportError。
• 若输入包含斜杠、反斜杠等路径字符（比如"/etc/passwd"），分割后的子模块名是无效标识符，同样会触发导入错误，无法加载可信目录外的模块。

导入中的特殊字符：仅点号有层级分隔意义 #

除了PEP 328定义的点号（用于分隔模块层级），Python的模块命名严格遵循标识符规则，其他特殊字符（如斜杠、减号、@、空格等）都不允许出现在合法模块名中。只要模块名包含这些字符，导入操作会直接失败，不存在利用特殊字符绕过限制的可能。

当前方法的安全性评估 #

在你完全信任trusted_directory内所有代码的前提下，当前方法基本安全，但存在几个可优化的细节：

• 若用户输入为空字符串，split(".")[0]会得到空值，导入trusted_directory.会触发错误，需要额外处理空输入的情况。
• 若submodule_name与Python内置模块同名（如"os"），只会尝试导入trusted_directory.os，只要你的可信目录中没有该模块，就会报错，不会加载内置模块。

不过当前的输入过滤逻辑较简单，建议补充更严格的验证步骤，避免不必要的错误和潜在风险。

推荐的输入清理与验证方法 #

1. 验证模块名合法性
   使用正则表达式检查子模块名是否符合Python标识符规则，过滤非法输入：

   import re
   submodule_name = user_input.split(".")[0]
   if not re.fullmatch(r'^[a-zA-Z_][a-zA-Z0-9_]*$', submodule_name):
       raise ValueError("无效的模块名称")
   

2. 检查模块是否存在于可信目录
   在导入前，先确认模块文件确实存在于trusted_directory中，避免导入失败：

   import os
   trusted_dir = os.path.abspath("trusted_directory")
   # 检查是否是单文件模块或包
   module_file = os.path.join(trusted_dir, f"{submodule_name}.py")
   package_dir = os.path.join(trusted_dir, submodule_name)
   if not (os.path.isfile(module_file) or (os.path.isdir(package_dir) and os.path.isfile(os.path.join(package_dir, "__init__.py")))):
       raise ValueError("可信目录中不存在该模块")
   

3. 使用白名单机制（最安全）
   预先定义允许导入的模块列表，仅允许加载白名单内的模块，彻底杜绝非法输入的风险：

   ALLOWED_MODULES = ["user_module1", "user_module2", "payment_processor"]
   submodule_name = user_input.split(".")[0]
   if submodule_name not in ALLOWED_MODULES:
       raise ValueError("该模块不允许被加载")
   

安全导入不可信输入的进阶方案 #

如果你需要更严谨的控制，可以绕过模块搜索路径，直接从文件加载模块，确保仅加载可信目录内的文件：

import importlib.util
import os

trusted_dir = os.path.abspath("trusted_directory")
submodule_name = user_input.split(".")[0]
module_path = os.path.join(trusted_dir, f"{submodule_name}.py")

# 验证文件存在且在可信目录内（防止符号链接绕过）
if not os.path.isfile(module_path):
    raise ValueError("模块不存在")
if not os.path.abspath(module_path).startswith(trusted_dir + os.sep):
    raise ValueError("非法模块路径")

# 直接从文件加载模块
spec = importlib.util.spec_from_file_location(f"trusted_directory.{submodule_name}", module_path)
imported_module = importlib.util.module_from_spec(spec)
spec.loader.exec_module(imported_module)
imported_module.main()

这种方式完全不受Python模块搜索路径的影响，从根源上避免了加载外部模块的可能。

内容的提问来源于stack exchange，提问作者Pierre couy