xref: /aosp_15_r20/external/cronet/net/data/websocket/trailing-whitespace_wsh.py (revision 6777b5387eb2ff775bb5750e3f5d96f37fb7352b)
1# Copyright 2015 The Chromium Authors
2# Use of this source code is governed by a BSD-style license that can be
3# found in the LICENSE file.
4#
5# The purpose of this test is to verify that the WebSocket handshake correctly
6# ignores trailing whitespace on response headers.
7# It is used by test case WebSocketEndToEndTest.TrailingWhitespace.
8
9from mod_pywebsocket import handshake
10from mod_pywebsocket.handshake.hybi import compute_accept_from_unicode
11
12
13def web_socket_do_extra_handshake(request):
14  accept = compute_accept_from_unicode(request.headers_in['Sec-WebSocket-Key'])
15  message = (b'HTTP/1.1 101 Switching Protocols\r\n'
16             b'Upgrade: websocket\r\n'
17             b'Connection: Upgrade\r\n'
18             b'Sec-WebSocket-Accept: %s\r\n'
19             b'Sec-WebSocket-Protocol: sip \r\n'
20             b'\r\n' % accept)
21  request.connection.write(message)
22  # Prevent pywebsocket from sending its own handshake message.
23  raise handshake.AbortedByUserException('Close the connection')
24
25
26def web_socket_transfer_data(request):
27  pass
28