Python tempfile模块安全使用指南：避免临时文件劫持
本文详解Python中tempfile模块的安全用法，重点分析mktemp的不安全性及软链接劫持风险。通过对比不安全示例与正确代码，介绍mkstemp、TemporaryFile和NamedTemporaryFile的正确使用场景及清理机制，确保临时文件操作的系统安全。
前言
临时文件通常用于存储无法完全加载到内存的数据，或作为必须从文件读取的外部程序的输入源。虽然常见的做法是在/tmp目录下生成唯一文件名，但安全地创建临时文件并非易事，需遵循多项严格规则。切勿自行实现该逻辑，应始终依赖标准库函数。同时，务必注意临时文件的及时清理。
临时文件引发的最大安全隐患在于文件名可预测性。若攻击者能预测临时文件名，便可能通过创建同名的软链接来劫持临时文件，从而执行恶意操作。
tempfile模块介绍
在Python中创建临时文件，主要使用tempfile模块。该模块提供了几种常用函数，具体区别如下：
tempfile.mktemp：不安全，严禁使用。 tempfile.mkstemp：随机生成临时文件名，默认保存在/tmp目录，也可指定其他路径（推荐使用）。 tempfile.TemporaryFile：在内存中创建文件，不写入磁盘，关闭句柄后自动删除（推荐使用）。 tempfile.NamedTemporaryFile(delete=True)：当delete参数为True时，行为与TemporaryFile类似，关闭后自动删除；若为False，则文件会保留在磁盘上（推荐使用）。
示例介绍
以下示例分别展示了创建临时文件的安全与不安全方式。
3.1 错误示例：
错误用法1：
import os
import tempfile
# This will most certainly put you at risk
tmp = os.path.join(tempfile.gettempdir(), filename)
if not os.path.exists(tmp):
with open(tmp, "w") file:
file.write("defaults")错误用法2：
import os
import tempfile
open(tempfile.mktemp(), "w")错误用法3：
filename = "{}/{}.tmp".format(tempfile.gettempdir(), os.getpid())
open(filename, "w")3.2 正确示例
正确用法1：
fd, path = tempfile.mkstemp()
try:
with os.fdopen(fd, 'w') as tmp:
# do stuff with temp file
tmp.write('stuff')
finally:
os.remove(path)正确用法2：
# 句柄关闭，文件即删除
with tempfile.TemporaryFile() as tmp:
# Do stuff with tmp
tmp.write('stuff')正确用法3：
tmp = tempfile.NamedTemporaryFile(delete=True)
try:
# do stuff with temp
tmp.write('stuff')
finally:
tmp.close() # 文件关闭即删除以上即为Python临时文件的安全使用方法详解，更多相关技术文章请关注本站其他内容。
免责声明：本文内容仅供参考，不代表本站观点。转载或引用请注明出处。
