python - 在 extractall 之前检查 tar 存档

Question

在 python 文档中，建议不要在未事先检查的情况下提取 tar 存档。使用 tarfile python 模块确保存档安全的最佳方法是什么？我是否应该遍历所有文件名并检查它们是否包含绝对路径名？

像下面这样的东西就足够了吗？

import sys
import tarfile
with tarfile.open('sample.tar', 'r') as tarf:
    for n in tarf.names():
        if n[0] == '/' or n[0:2] == '..':
            print 'sample.tar contains unsafe filenames'
            sys.exit(1)
    tarf.extractall()

编辑

此脚本与 2.7 之前的版本不兼容。对照 with and tarfile .

我现在遍历成员:

target_dir = "/target/"
with closing(tarfile.open('sample.tar', mode='r:gz')) as tarf:
    for m in tarf:
        pathn = os.path.abspath(os.path.join(target_dir, m.name))
        if not pathn.startswith(target_dir):
            print 'The tar file contains unsafe filenames. Aborting.'
            sys.exit(1)
        tarf.extract(m, path=tdir)

Answer 1

几乎，尽管仍然有可能有像 foo/../../ 这样的路径。

更好的方法是使用 os.path.join 和 os.path.abspath，它们一起可以正确处理前导 / 和 ..路径中的任意位置:

target_dir = "/target/" # trailing slash is important
with tarfile.open(…) as tarf:
    for n in tarf.names:
        if not os.path.abspath(os.path.join(target_dir, n)).startswith(target_dir):
            print "unsafe filenames!"
            sys.exit(1)
    tarf.extractall(path=target_dir)