django/daphne
1
1
Fork 0
mirror of https://github.com/django/daphne.git synced 2025-04-17 23:32:12 +03:00
Code Issues Packages Projects Releases Wiki Activity

Use code when rejecting connection

Browse Source 
Fixed #486
This commit is contained in:
Dave Johansen 2023-10-05 18:53:01 -06:00 committed by GitHub
parent 2d4dcbf149
commit 7234cb1638
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
daphne/ws_protocol.py
View File

@ -185,7 +185,7 @@ class WebSocketProtocol(WebSocketServerProtocol):
self.serverAccept(message.get("subprotocol", None))
elif message["type"] == "websocket.close":
if self.state == self.STATE_CONNECTING:
self.serverReject()
self.serverReject(code=message.get("code", 403))
else:
self.serverClose(code=message.get("code", None))
elif message["type"] == "websocket.send":
@ -222,12 +222,12 @@ class WebSocketProtocol(WebSocketServerProtocol):
del self.handshake_deferred
logger.debug("WebSocket %s accepted by application", self.client_addr)
def serverReject(self):
def serverReject(self, code):
"""
Called when we get a message saying to reject the connection.
"""
self.handshake_deferred.errback(
ConnectionDeny(code=403, reason="Access denied")
ConnectionDeny(code=code, reason="Connection closed")
)
del self.handshake_deferred
self.server.protocol_disconnected(self)
@ -284,7 +284,7 @@ class WebSocketProtocol(WebSocketServerProtocol):
# If we're still connecting, deny the connection
if self.state == self.STATE_CONNECTING:
if self.duration() > self.server.websocket_connect_timeout:
self.serverReject()
self.serverReject(408)
elif self.state == self.STATE_OPEN:
if (time.time() - self.last_ping) > self.server.ping_interval:
self._sendAutoPing()