wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-04-20 07:22:34 +08:00
Code Issues Actions 10 Packages Projects Releases Wiki Activity

fix: block Windows drive-qualified paths in path traversal validation

Browse Source 
PurePosixPath doesn't understand Windows drive letters, so paths like
C:/Windows/secret.png or C:secret.png would pass validation but escape
the output directory when joined via os.path.join on Windows.

Add PureWindowsPath check to detect drive-qualified paths on all
platforms. Added tests for Windows drive absolute, relative, and
backslash path variants.
This commit is contained in:
Varun Chawla 2026-03-24 21:39:24 -07:00
parent d29f0228a8
commit 2657155dfc
No known key found for this signature in database
2 changed files with 24 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
server.py
View File

@ -1,7 +1,7 @@
import os import os
import sys import sys
import asyncio import asyncio
from pathlib import PurePosixPath from pathlib import PurePosixPath, PureWindowsPath
import traceback import traceback
import time import time
@ -464,7 +464,8 @@ class PromptServer():
# Normalize backslashes and use standard library to parse path components # Normalize backslashes and use standard library to parse path components
normalized = filename.replace('\\', '/') normalized = filename.replace('\\', '/')
path = PurePosixPath(normalized) path = PurePosixPath(normalized)
if path.is_absolute() or '..' in path.parts: win_path = PureWindowsPath(normalized)
if path.is_absolute() or win_path.is_absolute() or win_path.drive or '..' in path.parts:
return web.Response(status=400) return web.Response(status=400)
if output_dir is None: if output_dir is None:

23
tests-unit/server_test/test_view_endpoint.py
View File

@ -5,7 +5,7 @@ from aiohttp import web
from aiohttp.test_utils import AioHTTPTestCase, unittest_run_loop from aiohttp.test_utils import AioHTTPTestCase, unittest_run_loop
import os import os
import tempfile import tempfile
from pathlib import Path, PurePosixPath from pathlib import Path, PurePosixPath, PureWindowsPath
class TestViewEndpointSecurity(AioHTTPTestCase): class TestViewEndpointSecurity(AioHTTPTestCase):
@ -40,7 +40,8 @@ class TestViewEndpointSecurity(AioHTTPTestCase):
# Normalize backslashes and use standard library to parse path components # Normalize backslashes and use standard library to parse path components
normalized = filename.replace('\\', '/') normalized = filename.replace('\\', '/')
path = PurePosixPath(normalized) path = PurePosixPath(normalized)
if path.is_absolute() or '..' in path.parts: win_path = PureWindowsPath(normalized)
if path.is_absolute() or win_path.is_absolute() or win_path.drive or '..' in path.parts:
return web.Response(status=400) return web.Response(status=400)
# For testing, just check if file exists in test directory # For testing, just check if file exists in test directory
@ -146,3 +147,21 @@ class TestViewEndpointSecurity(AioHTTPTestCase):
"""Test that Windows absolute paths with backslash are blocked (e.g., C:\\)""" """Test that Windows absolute paths with backslash are blocked (e.g., C:\\)"""
resp = await self.client.request("GET", "/view?filename=%5Cetc%5Cpasswd") resp = await self.client.request("GET", "/view?filename=%5Cetc%5Cpasswd")
assert resp.status == 400, "Should block backslash absolute paths" assert resp.status == 400, "Should block backslash absolute paths"
@unittest_run_loop
async def test_blocks_windows_drive_absolute_path(self):
"""Test that Windows drive-qualified absolute paths (C:/...) are blocked"""
resp = await self.client.request("GET", "/view?filename=C:/Windows/secret.png")
assert resp.status == 400, "Should block Windows drive-qualified absolute paths"
@unittest_run_loop
async def test_blocks_windows_drive_relative_path(self):
"""Test that Windows drive-qualified relative paths (C:secret.png) are blocked"""
resp = await self.client.request("GET", "/view?filename=C:secret.png")
assert resp.status == 400, "Should block Windows drive-qualified relative paths"
@unittest_run_loop
async def test_blocks_windows_drive_backslash_path(self):
"""Test that Windows drive paths with backslashes are blocked"""
resp = await self.client.request("GET", "/view?filename=C:%5CWindows%5Csecret.png")
assert resp.status == 400, "Should block Windows drive backslash paths"