Use pathlib for path resolution (#2506)

This commit is contained in:
Adam Hopkins 2022-07-31 08:49:02 +03:00
parent 7e0b0deb11
commit 0985d130e2
No known key found for this signature in database
GPG Key ID: 9F85EE6C807303FB
3 changed files with 37 additions and 34 deletions

View File

@ -4,6 +4,7 @@ coverage:
default: default:
target: auto target: auto
threshold: 0.75 threshold: 0.75
informational: true
project: project:
default: default:
target: auto target: auto

View File

@ -3,8 +3,8 @@ from contextlib import suppress
from functools import partial, wraps from functools import partial, wraps
from inspect import getsource, signature from inspect import getsource, signature
from mimetypes import guess_type from mimetypes import guess_type
from os import path, sep from os import path
from pathlib import PurePath from pathlib import Path, PurePath
from textwrap import dedent from textwrap import dedent
from time import gmtime, strftime from time import gmtime, strftime
from typing import ( from typing import (
@ -26,12 +26,7 @@ from sanic.base.meta import SanicMeta
from sanic.compat import stat_async from sanic.compat import stat_async
from sanic.constants import DEFAULT_HTTP_CONTENT_TYPE, HTTP_METHODS from sanic.constants import DEFAULT_HTTP_CONTENT_TYPE, HTTP_METHODS
from sanic.errorpages import RESPONSE_MAPPING from sanic.errorpages import RESPONSE_MAPPING
from sanic.exceptions import ( from sanic.exceptions import FileNotFound, HeaderNotFound, RangeNotSatisfiable
BadRequest,
FileNotFound,
HeaderNotFound,
RangeNotSatisfiable,
)
from sanic.handlers import ContentRangeHandler from sanic.handlers import ContentRangeHandler
from sanic.log import error_logger from sanic.log import error_logger
from sanic.models.futures import FutureRoute, FutureStatic from sanic.models.futures import FutureRoute, FutureStatic
@ -806,31 +801,39 @@ class RouteMixin(metaclass=SanicMeta):
__file_uri__=None, __file_uri__=None,
): ):
# Merge served directory and requested file if provided # Merge served directory and requested file if provided
root_path = file_path = path.abspath(unquote(file_or_directory)) file_path_raw = Path(unquote(file_or_directory))
root_path = file_path = file_path_raw.resolve()
not_found = FileNotFound(
"File not found",
path=file_or_directory,
relative_url=__file_uri__,
)
if __file_uri__: if __file_uri__:
# Strip all / that in the beginning of the URL to help prevent # Strip all / that in the beginning of the URL to help prevent
# python from herping a derp and treating the uri as an # python from herping a derp and treating the uri as an
# absolute path # absolute path
unquoted_file_uri = unquote(__file_uri__).lstrip("/") unquoted_file_uri = unquote(__file_uri__).lstrip("/")
file_path_raw = Path(file_or_directory, unquoted_file_uri)
file_path = file_path_raw.resolve()
if (
file_path < root_path and not file_path_raw.is_symlink()
) or file_path_raw.match("../**/*"):
error_logger.exception(
f"File not found: path={file_or_directory}, "
f"relative_url={__file_uri__}"
)
raise not_found
segments = unquoted_file_uri.split("/") try:
if ".." in segments or any(sep in segment for segment in segments): file_path.relative_to(root_path)
raise BadRequest("Invalid URL") except ValueError:
if not file_path_raw.is_symlink():
file_path = path.join(file_or_directory, unquoted_file_uri) error_logger.exception(
file_path = path.abspath(file_path) f"File not found: path={file_or_directory}, "
f"relative_url={__file_uri__}"
if not file_path.startswith(root_path): )
error_logger.exception( raise not_found
f"File not found: path={file_or_directory}, "
f"relative_url={__file_uri__}"
)
raise FileNotFound(
"File not found",
path=file_or_directory,
relative_url=__file_uri__,
)
try: try:
headers = {} headers = {}
# Check if the client has been sent this file before # Check if the client has been sent this file before
@ -898,11 +901,7 @@ class RouteMixin(metaclass=SanicMeta):
except RangeNotSatisfiable: except RangeNotSatisfiable:
raise raise
except FileNotFoundError: except FileNotFoundError:
raise FileNotFound( raise not_found
"File not found",
path=file_or_directory,
relative_url=__file_uri__,
)
except Exception: except Exception:
error_logger.exception( error_logger.exception(
f"Exception in static request handler: " f"Exception in static request handler: "

View File

@ -616,8 +616,11 @@ def test_dotted_dir_ok(
def test_breakout(app: Sanic, static_file_directory: str): def test_breakout(app: Sanic, static_file_directory: str):
app.static("/foo", static_file_directory) app.static("/foo", static_file_directory)
_, response = app.test_client.get("/foo/..%2Ffake/server.py")
assert response.status == 404
_, response = app.test_client.get("/foo/..%2Fstatic/test.file") _, response = app.test_client.get("/foo/..%2Fstatic/test.file")
assert response.status == 400 assert response.status == 404
@pytest.mark.skipif( @pytest.mark.skipif(
@ -629,6 +632,6 @@ def test_double_backslash_prohibited_on_win32(
app.static("/foo", static_file_directory) app.static("/foo", static_file_directory)
_, response = app.test_client.get("/foo/static/..\\static/test.file") _, response = app.test_client.get("/foo/static/..\\static/test.file")
assert response.status == 400 assert response.status == 404
_, response = app.test_client.get("/foo/static\\../static/test.file") _, response = app.test_client.get("/foo/static\\../static/test.file")
assert response.status == 400 assert response.status == 404