Pass the proper type when uploading files. (#11927)

The Content-Length header should be treated as an int, not
a string. This shouldn't have any user-facing change.
This commit is contained in:
Patrick Cloke 2022-02-07 10:06:52 -05:00 committed by GitHub
parent e03dde259b
commit 314ca4c86d
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 10 additions and 4 deletions

1
changelog.d/11927.misc Normal file
View file

@ -0,0 +1 @@
Use the proper type for the Content-Length header in the `UploadResource`.

View file

@ -49,10 +49,14 @@ class UploadResource(DirectServeJsonResource):
async def _async_render_POST(self, request: SynapseRequest) -> None: async def _async_render_POST(self, request: SynapseRequest) -> None:
requester = await self.auth.get_user_by_req(request) requester = await self.auth.get_user_by_req(request)
content_length = request.getHeader("Content-Length") raw_content_length = request.getHeader("Content-Length")
if content_length is None: if raw_content_length is None:
raise SynapseError(msg="Request must specify a Content-Length", code=400) raise SynapseError(msg="Request must specify a Content-Length", code=400)
if int(content_length) > self.max_upload_size: try:
content_length = int(raw_content_length)
except ValueError:
raise SynapseError(msg="Content-Length value is invalid", code=400)
if content_length > self.max_upload_size:
raise SynapseError( raise SynapseError(
msg="Upload request body is too large", msg="Upload request body is too large",
code=413, code=413,
@ -66,7 +70,8 @@ class UploadResource(DirectServeJsonResource):
upload_name: Optional[str] = upload_name_bytes.decode("utf8") upload_name: Optional[str] = upload_name_bytes.decode("utf8")
except UnicodeDecodeError: except UnicodeDecodeError:
raise SynapseError( raise SynapseError(
msg="Invalid UTF-8 filename parameter: %r" % (upload_name), code=400 msg="Invalid UTF-8 filename parameter: %r" % (upload_name_bytes,),
code=400,
) )
# If the name is falsey (e.g. an empty byte string) ensure it is None. # If the name is falsey (e.g. an empty byte string) ensure it is None.