📦 deps(thirdparty): update snapshots
This commit is contained in:
@@ -1,6 +1,7 @@
|
||||
import importlib.util
|
||||
import sys
|
||||
import tempfile
|
||||
import types
|
||||
import unittest
|
||||
import stat
|
||||
import zipfile
|
||||
@@ -8,6 +9,17 @@ from pathlib import Path
|
||||
|
||||
|
||||
REPO_ROOT = Path(__file__).resolve().parents[3]
|
||||
TOOLS_TESTS_DIR = REPO_ROOT / "tools" / "scripts" / "tests"
|
||||
if str(TOOLS_TESTS_DIR) not in sys.path:
|
||||
sys.path.insert(0, str(TOOLS_TESTS_DIR))
|
||||
|
||||
from symlink_test_utils import symlink_or_skip
|
||||
|
||||
defusedxml = types.ModuleType("defusedxml")
|
||||
defusedxml_minidom = types.ModuleType("defusedxml.minidom")
|
||||
defusedxml.minidom = defusedxml_minidom
|
||||
sys.modules.setdefault("defusedxml", defusedxml)
|
||||
sys.modules.setdefault("defusedxml.minidom", defusedxml_minidom)
|
||||
|
||||
|
||||
def load_module(relative_path: str, module_name: str):
|
||||
@@ -67,6 +79,44 @@ class OfficeUnpackSecurityTests(unittest.TestCase):
|
||||
|
||||
self.assertFalse((temp_path / "escape.txt").exists())
|
||||
|
||||
def test_extract_archive_safely_blocks_high_compression_ratio(self):
|
||||
for relative_path, module_name in [
|
||||
("skills/docx-official/ooxml/scripts/unpack.py", "docx_unpack_ratio"),
|
||||
("skills/pptx-official/ooxml/scripts/unpack.py", "pptx_unpack_ratio"),
|
||||
]:
|
||||
module = load_module(relative_path, module_name)
|
||||
module.MAX_COMPRESSION_RATIO = 10
|
||||
|
||||
with self.subTest(module=relative_path):
|
||||
with tempfile.TemporaryDirectory() as temp_dir:
|
||||
temp_path = Path(temp_dir)
|
||||
archive_path = temp_path / "payload.zip"
|
||||
|
||||
with zipfile.ZipFile(archive_path, "w", zipfile.ZIP_DEFLATED) as archive:
|
||||
archive.writestr("word/document.xml", "A" * 100_000)
|
||||
|
||||
with self.assertRaises(ValueError):
|
||||
module.extract_archive_safely(archive_path, temp_path / "output")
|
||||
|
||||
def test_pack_document_blocks_input_symlinks(self):
|
||||
for relative_path, module_name in [
|
||||
("skills/docx-official/ooxml/scripts/pack.py", "docx_pack"),
|
||||
("skills/pptx-official/ooxml/scripts/pack.py", "pptx_pack"),
|
||||
]:
|
||||
module = load_module(relative_path, module_name)
|
||||
|
||||
with self.subTest(module=relative_path):
|
||||
with tempfile.TemporaryDirectory() as temp_dir:
|
||||
temp_path = Path(temp_dir)
|
||||
input_dir = temp_path / "input"
|
||||
outside = temp_path / "outside.txt"
|
||||
input_dir.mkdir()
|
||||
outside.write_text("secret", encoding="utf-8")
|
||||
symlink_or_skip(self, outside, input_dir / "leak.txt")
|
||||
|
||||
with self.assertRaises(ValueError):
|
||||
module.validate_input_tree(input_dir)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
|
||||
Reference in New Issue
Block a user