| #!/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 |