Skip to content

Commit 43b9b85

Browse files
Fix code scanning alert no. 27591: Arbitrary file write during tarfile extraction
Co-authored-by: Copilot Autofix powered by AI <62310815+github-advanced-security[bot]@users.noreply.github.com>
1 parent b86a1c9 commit 43b9b85

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

onnxruntime/python/tools/tensorrt/perf/setup_scripts/setup_onnx_zoo.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,9 @@ def extract_and_get_files(file_name):
1818
model_folder = file_name.replace(".tar.gz", "") + "/"
1919
create_model_folder(model_folder)
2020
with tarfile.open(file_name) as model_tar:
21+
for member in model_tar.getmembers():
22+
if os.path.isabs(member.name) or ".." in member.name:
23+
raise ValueError(f"Illegal tar archive entry: {member.name}")
2124
model_tar.extractall(model_folder)
2225
file_list = model_tar.getnames()
2326
file_list.sort()

0 commit comments

Comments
 (0)