Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 5 additions & 4 deletions gunicorn/asgi/message.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,7 @@

from gunicorn.http.errors import (
InvalidHeader, InvalidHeaderName, NoMoreData,
ChunkMissingTerminator, InvalidChunkSize,
InvalidRequestLine, InvalidRequestMethod, InvalidHTTPVersion,
LimitRequestLine, LimitRequestHeaders,
UnsupportedTransferCoding, ObsoleteFolding,
Expand Down Expand Up @@ -177,7 +178,7 @@ async def _read_into(self, buf, stop=False):
data = await self.unreader.read()
if not data:
if stop:
raise StopIteration()
raise StopAsyncIteration
raise NoMoreData(bytes(buf))
buf.extend(data)

Expand Down Expand Up @@ -628,9 +629,9 @@ async def _chunked_body_reader(self):
chunk_size = chunk_size.rstrip(b" \t")

if any(n not in b"0123456789abcdefABCDEF" for n in chunk_size):
raise InvalidHeader("Invalid chunk size")
raise InvalidChunkSize(chunk_size)
if len(chunk_size) == 0:
raise InvalidHeader("Invalid chunk size")
raise InvalidChunkSize(chunk_size)

chunk_size = int(chunk_size, 16)

Expand Down Expand Up @@ -658,7 +659,7 @@ async def _chunked_body_reader(self):
break
crlf += more
if crlf != b"\r\n":
raise InvalidHeader("Missing chunk terminator")
raise ChunkMissingTerminator(crlf)

async def _read_chunk_size_line(self):
"""Read a chunk size line."""
Expand Down
3 changes: 3 additions & 0 deletions tests/requests/invalid/006.py
Original file line number Diff line number Diff line change
@@ -1,2 +1,5 @@
from gunicorn.config import Config
from gunicorn.http.errors import LimitRequestLine
cfg = Config()
cfg.set("limit_request_line", 4094)
request = LimitRequestLine
4 changes: 4 additions & 0 deletions tests/requests/valid/040.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,7 @@
from gunicorn.config import Config
cfg = Config()
cfg.set("header_map", "drop")

request = {
"method": "GET",
"uri": uri("/keep/same/as?invalid/040"),
Expand Down
Loading