Vasanko/sanic
2
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Merge pull request #756 from qwesda/master

Browse Source 
fixes #755 fragmented headers
This commit is contained in:
Raphael Deem 2017-06-30 18:24:51 -07:00 committed by GitHub
commit 1e75265eed
1 changed files with 17 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
sanic/server.py
View File

@ -99,6 +99,7 @@ class HttpProtocol(asyncio.Protocol):
self._request_handler_task = None self._request_handler_task = None
self._request_stream_task = None self._request_stream_task = None
self._keep_alive = keep_alive self._keep_alive = keep_alive
self._header_fragment = b''
self.state = state if state else {} self.state = state if state else {}
if 'requests_count' not in self.state: if 'requests_count' not in self.state:
self.state['requests_count'] = 0 self.state['requests_count'] = 0
@ -173,14 +174,25 @@ class HttpProtocol(asyncio.Protocol):
self.write_error(exception) self.write_error(exception)
def on_url(self, url): def on_url(self, url):
if not self.url:
self.url = url self.url = url
else:
self.url += url
def on_header(self, name, value): def on_header(self, name, value):
if name == b'Content-Length' and int(value) > self.request_max_size: self._header_fragment += name
if value is not None:
if self._header_fragment == b'Content-Length' \
and int(value) > self.request_max_size:
exception = PayloadTooLarge('Payload Too Large') exception = PayloadTooLarge('Payload Too Large')
self.write_error(exception) self.write_error(exception)
self.headers.append((name.decode().casefold(), value.decode())) self.headers.append(
(self._header_fragment.decode().casefold(),
value.decode()))
self._header_fragment = b''
def on_headers_complete(self): def on_headers_complete(self):
self.request = self.request_class( self.request = self.request_class(