fixed: robust validation when model downloading #2

This commit is contained in:
Dr.Lt.Data
2025-03-12 21:24:31 +09:00
parent bbb54d4a08
commit c3eed981c0
3 changed files with 6 additions and 2 deletions

View File

@@ -279,6 +279,10 @@ def get_model_dir(data, show_log=False) -> str | None:
else:
models_base = folder_paths.models_dir
# NOTE: Validate to prevent path traversal.
if any(char in data['filename'] for char in {'/', '\\', ':'}):
return None
def resolve_custom_node(save_path):
save_path = save_path[13:] # remove 'custom_nodes/'