blob: ad466877fe045be36bfebdcfb30026f2245deae9 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
|
#!/usr/bin/python
from mod_pywebsocket import common, stream
from mod_pywebsocket.handshake import AbortedByUserException, hybi
def web_socket_do_extra_handshake(request):
# Send simple response header. This test implements the handshake manually,
# so that we can send the header in the same packet as the close frame.
msg = (b'HTTP/1.1 101 Switching Protocols:\x0D\x0A'
b'Connection: Upgrade\x0D\x0A'
b'Upgrade: WebSocket\x0D\x0A'
b'Set-Cookie: ws_test=test\x0D\x0A'
b'Sec-WebSocket-Origin: %s\x0D\x0A'
b'Sec-WebSocket-Accept: %s\x0D\x0A\x0D\x0A') % (request.ws_origin.encode(
'UTF-8'), hybi.compute_accept_from_unicode(request.headers_in.get(common.SEC_WEBSOCKET_KEY_HEADER)))
# Create a clean close frame.
close_body = stream.create_closing_handshake_body(1001, 'PASS')
close_frame = stream.create_close_frame(close_body)
# Concatenate the header and the close frame and write them to the socket.
request.connection.write(msg + close_frame)
# Wait for the responding close frame from the user agent. It's not possible
# to use the stream methods at this point because the stream hasn't been
# established from pywebsocket's point of view. Instead just read the
# correct number of bytes.
# Warning: reading the wrong number of bytes here will make the test
# flaky.
MASK_LENGTH = 4
request.connection.read(len(close_frame) + MASK_LENGTH)
# Close the socket without pywebsocket sending its own handshake response.
raise AbortedByUserException('Abort the connection')
def web_socket_transfer_data(request):
pass
|
