WebSocket: Copy existing WebSocket tests
[WebKit-https.git] / LayoutTests / http / tests / websocket / tests / hybi / handshake-fail-by-prepended-null_wsh.py
1 # Copyright (C) Research In Motion Limited 2011. All rights reserved.
2 #
3 # Redistribution and use in source and binary forms, with or without modification, are permitted provided that the
4 # following conditions are met:
5 #
6 # 1. Redistributions of source code must retain the above copyright notice, this list of conditions and the following
7 # disclaimer.
8 #
9 # 2. Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the
10 # following disclaimer in the documentation and/or other materials provided with the distribution.
11 #
12 # THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS “AS IS” AND ANY EXPRESS OR
13 # IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
14 # MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL
15 # APPLE INC. OR ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
16 # EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
17 # SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
18 # HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR
19 # TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
20 # SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
21
22 import time
23
24
25 def web_socket_do_extra_handshake(request):
26     # This simulates a broken server that sends a WebSocket frame before the
27     # handshake, and more frames afterwards.  It is important that if this
28     # happens the client does not buffer all the frames as the server continues
29     # to send more data - it should abort after reading a reasonable number of
30     # bytes (set arbitrarily to 1024).
31     frame = '\0Frame-contains-thirty-two-bytes'
32
33     msg = frame
34     msg += 'HTTP/1.1 101 WebSocket Protocol Handshake\r\n'
35     msg += 'Upgrade: WebSocket\r\n'
36     msg += 'Connection: Upgrade\r\n'
37     msg += 'Sec-WebSocket-Location: ' + request.ws_location + '\r\n'
38     msg += 'Sec-WebSocket-Origin: ' + request.ws_origin + '\r\n'
39     msg += '\r\n'
40     msg += request.ws_challenge_md5
41     request.connection.write(msg)
42     # continue writing data until the client disconnects
43     while True:
44         time.sleep(1)
45         numFrames = 1024 / len(frame) # write over 1024 bytes including the above handshake
46         for i in range(0, numFrames):
47             request.connection.write(frame)
48
49
50 def web_socket_transfer_data(request):
51     pass