11477Szelenkov@nginx.comimport struct
21130Szelenkov@nginx.comimport time
31477Szelenkov@nginx.com
41635Szelenkov@nginx.comimport pytest
51635Szelenkov@nginx.com
61130Szelenkov@nginx.comfrom unit.applications.lang.node import TestApplicationNode
71130Szelenkov@nginx.comfrom unit.applications.websockets import TestApplicationWebsocket
81730Szelenkov@nginx.comfrom unit.option import option
91130Szelenkov@nginx.com
101150Szelenkov@nginx.com
111130Szelenkov@nginx.comclass TestNodeWebsockets(TestApplicationNode):
121467Szelenkov@nginx.com    prerequisites = {'modules': {'node': 'any'}}
131130Szelenkov@nginx.com
141130Szelenkov@nginx.com    ws = TestApplicationWebsocket()
151130Szelenkov@nginx.com
16*1736Szelenkov@nginx.com    @pytest.fixture(autouse=True)
17*1736Szelenkov@nginx.com    def setup_method_fixture(self, request, skip_alert):
181596Szelenkov@nginx.com        assert 'success' in self.conf(
191596Szelenkov@nginx.com            {'http': {'websocket': {'keepalive_interval': 0}}}, 'settings'
201596Szelenkov@nginx.com        ), 'clear keepalive_interval'
211138Szelenkov@nginx.com
221596Szelenkov@nginx.com        skip_alert(r'socket close\(\d+\) failed')
231130Szelenkov@nginx.com
241130Szelenkov@nginx.com    def close_connection(self, sock):
251596Szelenkov@nginx.com        assert self.recvall(sock, read_timeout=0.1) == b'', 'empty soc'
261130Szelenkov@nginx.com
271130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CLOSE, self.ws.serialize_close())
281130Szelenkov@nginx.com
291130Szelenkov@nginx.com        self.check_close(sock)
301130Szelenkov@nginx.com
311150Szelenkov@nginx.com    def check_close(self, sock, code=1000, no_close=False):
321130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
331130Szelenkov@nginx.com
341596Szelenkov@nginx.com        assert frame['fin'] == True, 'close fin'
351596Szelenkov@nginx.com        assert frame['opcode'] == self.ws.OP_CLOSE, 'close opcode'
361596Szelenkov@nginx.com        assert frame['code'] == code, 'close code'
371130Szelenkov@nginx.com
381130Szelenkov@nginx.com        if not no_close:
391130Szelenkov@nginx.com            sock.close()
401130Szelenkov@nginx.com
411130Szelenkov@nginx.com    def check_frame(self, frame, fin, opcode, payload, decode=True):
421130Szelenkov@nginx.com        if opcode == self.ws.OP_BINARY or not decode:
431130Szelenkov@nginx.com            data = frame['data']
441130Szelenkov@nginx.com        else:
451130Szelenkov@nginx.com            data = frame['data'].decode('utf-8')
461130Szelenkov@nginx.com
471596Szelenkov@nginx.com        assert frame['fin'] == fin, 'fin'
481596Szelenkov@nginx.com        assert frame['opcode'] == opcode, 'opcode'
491596Szelenkov@nginx.com        assert data == payload, 'payload'
501130Szelenkov@nginx.com
511130Szelenkov@nginx.com    def test_node_websockets_handshake(self):
521130Szelenkov@nginx.com        self.load('websockets/mirror')
531130Szelenkov@nginx.com
541130Szelenkov@nginx.com        resp, sock, key = self.ws.upgrade()
551130Szelenkov@nginx.com        sock.close()
561130Szelenkov@nginx.com
571596Szelenkov@nginx.com        assert resp['status'] == 101, 'status'
581596Szelenkov@nginx.com        assert resp['headers']['Upgrade'] == 'websocket', 'upgrade'
591596Szelenkov@nginx.com        assert resp['headers']['Connection'] == 'Upgrade', 'connection'
601596Szelenkov@nginx.com        assert resp['headers']['Sec-WebSocket-Accept'] == self.ws.accept(
611596Szelenkov@nginx.com            key
621596Szelenkov@nginx.com        ), 'key'
631130Szelenkov@nginx.com
641130Szelenkov@nginx.com    def test_node_websockets_mirror(self):
651130Szelenkov@nginx.com        self.load('websockets/mirror')
661130Szelenkov@nginx.com
671130Szelenkov@nginx.com        message = 'blah'
681130Szelenkov@nginx.com
691130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
701130Szelenkov@nginx.com
711130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, message)
721130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
731130Szelenkov@nginx.com
741596Szelenkov@nginx.com        assert message == frame['data'].decode('utf-8'), 'mirror'
751130Szelenkov@nginx.com
761130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, message)
771130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
781130Szelenkov@nginx.com
791596Szelenkov@nginx.com        assert message == frame['data'].decode('utf-8'), 'mirror 2'
801130Szelenkov@nginx.com
811130Szelenkov@nginx.com        sock.close()
821130Szelenkov@nginx.com
831130Szelenkov@nginx.com    def test_node_websockets_no_mask(self):
841130Szelenkov@nginx.com        self.load('websockets/mirror')
851130Szelenkov@nginx.com
861130Szelenkov@nginx.com        message = 'blah'
871130Szelenkov@nginx.com
881130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
891130Szelenkov@nginx.com
901130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, message, mask=False)
911130Szelenkov@nginx.com
921130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
931130Szelenkov@nginx.com
941596Szelenkov@nginx.com        assert frame['opcode'] == self.ws.OP_CLOSE, 'no mask opcode'
951596Szelenkov@nginx.com        assert frame['code'] == 1002, 'no mask close code'
961130Szelenkov@nginx.com
971130Szelenkov@nginx.com        sock.close()
981130Szelenkov@nginx.com
991130Szelenkov@nginx.com    def test_node_websockets_fragmentation(self):
1001130Szelenkov@nginx.com        self.load('websockets/mirror')
1011130Szelenkov@nginx.com
1021130Szelenkov@nginx.com        message = 'blah'
1031130Szelenkov@nginx.com
1041130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
1051130Szelenkov@nginx.com
1061130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, message, fin=False)
1071130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CONT, ' ', fin=False)
1081130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CONT, message)
1091130Szelenkov@nginx.com
1101130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
1111130Szelenkov@nginx.com
1121596Szelenkov@nginx.com        assert message + ' ' + message == frame['data'].decode(
1131596Szelenkov@nginx.com            'utf-8'
1141596Szelenkov@nginx.com        ), 'mirror framing'
1151130Szelenkov@nginx.com
1161130Szelenkov@nginx.com        sock.close()
1171130Szelenkov@nginx.com
1181130Szelenkov@nginx.com    def test_node_websockets_frame_fragmentation_invalid(self):
1191130Szelenkov@nginx.com        self.load('websockets/mirror')
1201130Szelenkov@nginx.com
1211130Szelenkov@nginx.com        message = 'blah'
1221130Szelenkov@nginx.com
1231130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
1241130Szelenkov@nginx.com
1251130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PING, message, fin=False)
1261130Szelenkov@nginx.com
1271130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
1281130Szelenkov@nginx.com
1291130Szelenkov@nginx.com        frame.pop('data')
1301596Szelenkov@nginx.com        assert frame == {
1311596Szelenkov@nginx.com            'fin': True,
1321596Szelenkov@nginx.com            'rsv1': False,
1331596Szelenkov@nginx.com            'rsv2': False,
1341596Szelenkov@nginx.com            'rsv3': False,
1351596Szelenkov@nginx.com            'opcode': self.ws.OP_CLOSE,
1361596Szelenkov@nginx.com            'mask': 0,
1371596Szelenkov@nginx.com            'code': 1002,
1381596Szelenkov@nginx.com            'reason': 'Fragmented control frame',
1391596Szelenkov@nginx.com        }, 'close frame'
1401130Szelenkov@nginx.com
1411130Szelenkov@nginx.com        sock.close()
1421130Szelenkov@nginx.com
1431130Szelenkov@nginx.com    def test_node_websockets_large(self):
1441130Szelenkov@nginx.com        self.load('websockets/mirror_fragmentation')
1451130Szelenkov@nginx.com
1461130Szelenkov@nginx.com        message = '0123456789' * 3000
1471130Szelenkov@nginx.com
1481130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
1491130Szelenkov@nginx.com
1501130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, message)
1511130Szelenkov@nginx.com
1521130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
1531130Szelenkov@nginx.com        data = frame['data'].decode('utf-8')
1541130Szelenkov@nginx.com
1551130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
1561130Szelenkov@nginx.com        data += frame['data'].decode('utf-8')
1571130Szelenkov@nginx.com
1581596Szelenkov@nginx.com        assert message == data, 'large'
1591130Szelenkov@nginx.com
1601130Szelenkov@nginx.com        sock.close()
1611130Szelenkov@nginx.com
1621130Szelenkov@nginx.com    def test_node_websockets_two_clients(self):
1631130Szelenkov@nginx.com        self.load('websockets/mirror')
1641130Szelenkov@nginx.com
1651130Szelenkov@nginx.com        message1 = 'blah1'
1661130Szelenkov@nginx.com        message2 = 'blah2'
1671130Szelenkov@nginx.com
1681130Szelenkov@nginx.com        _, sock1, _ = self.ws.upgrade()
1691130Szelenkov@nginx.com        _, sock2, _ = self.ws.upgrade()
1701130Szelenkov@nginx.com
1711130Szelenkov@nginx.com        self.ws.frame_write(sock1, self.ws.OP_TEXT, message1)
1721130Szelenkov@nginx.com        self.ws.frame_write(sock2, self.ws.OP_TEXT, message2)
1731130Szelenkov@nginx.com
1741130Szelenkov@nginx.com        frame1 = self.ws.frame_read(sock1)
1751130Szelenkov@nginx.com        frame2 = self.ws.frame_read(sock2)
1761130Szelenkov@nginx.com
1771596Szelenkov@nginx.com        assert message1 == frame1['data'].decode('utf-8'), 'client 1'
1781596Szelenkov@nginx.com        assert message2 == frame2['data'].decode('utf-8'), 'client 2'
1791130Szelenkov@nginx.com
1801130Szelenkov@nginx.com        sock1.close()
1811130Szelenkov@nginx.com        sock2.close()
1821130Szelenkov@nginx.com
1831596Szelenkov@nginx.com    @pytest.mark.skip('not yet')
1841150Szelenkov@nginx.com    def test_node_websockets_handshake_upgrade_absent(
1851150Szelenkov@nginx.com        self
1861150Szelenkov@nginx.com    ):  # FAIL https://tools.ietf.org/html/rfc6455#section-4.2.1
1871130Szelenkov@nginx.com        self.load('websockets/mirror')
1881130Szelenkov@nginx.com
1891150Szelenkov@nginx.com        resp = self.get(
1901150Szelenkov@nginx.com            headers={
1911150Szelenkov@nginx.com                'Host': 'localhost',
1921150Szelenkov@nginx.com                'Connection': 'Upgrade',
1931262Szelenkov@nginx.com                'Sec-WebSocket-Key': self.ws.key(),
1941150Szelenkov@nginx.com                'Sec-WebSocket-Protocol': 'chat',
1951150Szelenkov@nginx.com                'Sec-WebSocket-Version': 13,
1961150Szelenkov@nginx.com            },
1971150Szelenkov@nginx.com        )
1981130Szelenkov@nginx.com
1991596Szelenkov@nginx.com        assert resp['status'] == 400, 'upgrade absent'
2001130Szelenkov@nginx.com
2011130Szelenkov@nginx.com    def test_node_websockets_handshake_case_insensitive(self):
2021130Szelenkov@nginx.com        self.load('websockets/mirror')
2031130Szelenkov@nginx.com
2041262Szelenkov@nginx.com        resp, sock, _ = self.ws.upgrade(
2051150Szelenkov@nginx.com            headers={
2061150Szelenkov@nginx.com                'Host': 'localhost',
2071150Szelenkov@nginx.com                'Upgrade': 'WEBSOCKET',
2081150Szelenkov@nginx.com                'Connection': 'UPGRADE',
2091262Szelenkov@nginx.com                'Sec-WebSocket-Key': self.ws.key(),
2101150Szelenkov@nginx.com                'Sec-WebSocket-Protocol': 'chat',
2111150Szelenkov@nginx.com                'Sec-WebSocket-Version': 13,
2121262Szelenkov@nginx.com            }
2131150Szelenkov@nginx.com        )
2141262Szelenkov@nginx.com        sock.close()
2151130Szelenkov@nginx.com
2161596Szelenkov@nginx.com        assert resp['status'] == 101, 'status'
2171130Szelenkov@nginx.com
2181596Szelenkov@nginx.com    @pytest.mark.skip('not yet')
2191150Szelenkov@nginx.com    def test_node_websockets_handshake_connection_absent(self):  # FAIL
2201130Szelenkov@nginx.com        self.load('websockets/mirror')
2211130Szelenkov@nginx.com
2221150Szelenkov@nginx.com        resp = self.get(
2231150Szelenkov@nginx.com            headers={
2241150Szelenkov@nginx.com                'Host': 'localhost',
2251150Szelenkov@nginx.com                'Upgrade': 'websocket',
2261262Szelenkov@nginx.com                'Sec-WebSocket-Key': self.ws.key(),
2271150Szelenkov@nginx.com                'Sec-WebSocket-Protocol': 'chat',
2281150Szelenkov@nginx.com                'Sec-WebSocket-Version': 13,
2291150Szelenkov@nginx.com            },
2301150Szelenkov@nginx.com        )
2311130Szelenkov@nginx.com
2321596Szelenkov@nginx.com        assert resp['status'] == 400, 'status'
2331130Szelenkov@nginx.com
2341130Szelenkov@nginx.com    def test_node_websockets_handshake_version_absent(self):
2351130Szelenkov@nginx.com        self.load('websockets/mirror')
2361130Szelenkov@nginx.com
2371150Szelenkov@nginx.com        resp = self.get(
2381150Szelenkov@nginx.com            headers={
2391150Szelenkov@nginx.com                'Host': 'localhost',
2401150Szelenkov@nginx.com                'Upgrade': 'websocket',
2411150Szelenkov@nginx.com                'Connection': 'Upgrade',
2421262Szelenkov@nginx.com                'Sec-WebSocket-Key': self.ws.key(),
2431150Szelenkov@nginx.com                'Sec-WebSocket-Protocol': 'chat',
2441150Szelenkov@nginx.com            },
2451150Szelenkov@nginx.com        )
2461130Szelenkov@nginx.com
2471596Szelenkov@nginx.com        assert resp['status'] == 426, 'status'
2481130Szelenkov@nginx.com
2491596Szelenkov@nginx.com    @pytest.mark.skip('not yet')
2501130Szelenkov@nginx.com    def test_node_websockets_handshake_key_invalid(self):
2511130Szelenkov@nginx.com        self.load('websockets/mirror')
2521130Szelenkov@nginx.com
2531150Szelenkov@nginx.com        resp = self.get(
2541150Szelenkov@nginx.com            headers={
2551150Szelenkov@nginx.com                'Host': 'localhost',
2561150Szelenkov@nginx.com                'Upgrade': 'websocket',
2571150Szelenkov@nginx.com                'Connection': 'Upgrade',
2581150Szelenkov@nginx.com                'Sec-WebSocket-Key': '!',
2591150Szelenkov@nginx.com                'Sec-WebSocket-Protocol': 'chat',
2601150Szelenkov@nginx.com                'Sec-WebSocket-Version': 13,
2611150Szelenkov@nginx.com            },
2621150Szelenkov@nginx.com        )
2631130Szelenkov@nginx.com
2641596Szelenkov@nginx.com        assert resp['status'] == 400, 'key length'
2651130Szelenkov@nginx.com
2661130Szelenkov@nginx.com        key = self.ws.key()
2671150Szelenkov@nginx.com        resp = self.get(
2681150Szelenkov@nginx.com            headers={
2691150Szelenkov@nginx.com                'Host': 'localhost',
2701150Szelenkov@nginx.com                'Upgrade': 'websocket',
2711150Szelenkov@nginx.com                'Connection': 'Upgrade',
2721150Szelenkov@nginx.com                'Sec-WebSocket-Key': [key, key],
2731150Szelenkov@nginx.com                'Sec-WebSocket-Protocol': 'chat',
2741150Szelenkov@nginx.com                'Sec-WebSocket-Version': 13,
2751150Szelenkov@nginx.com            },
2761150Szelenkov@nginx.com        )
2771130Szelenkov@nginx.com
2781596Szelenkov@nginx.com        assert resp['status'] == 400, 'key double'  # FAIL https://tools.ietf.org/html/rfc6455#section-11.3.1
2791130Szelenkov@nginx.com
2801130Szelenkov@nginx.com    def test_node_websockets_handshake_method_invalid(self):
2811130Szelenkov@nginx.com        self.load('websockets/mirror')
2821130Szelenkov@nginx.com
2831150Szelenkov@nginx.com        resp = self.post(
2841150Szelenkov@nginx.com            headers={
2851150Szelenkov@nginx.com                'Host': 'localhost',
2861150Szelenkov@nginx.com                'Upgrade': 'websocket',
2871150Szelenkov@nginx.com                'Connection': 'Upgrade',
2881262Szelenkov@nginx.com                'Sec-WebSocket-Key': self.ws.key(),
2891150Szelenkov@nginx.com                'Sec-WebSocket-Protocol': 'chat',
2901150Szelenkov@nginx.com                'Sec-WebSocket-Version': 13,
2911150Szelenkov@nginx.com            },
2921150Szelenkov@nginx.com        )
2931130Szelenkov@nginx.com
2941596Szelenkov@nginx.com        assert resp['status'] == 400, 'status'
2951130Szelenkov@nginx.com
2961130Szelenkov@nginx.com    def test_node_websockets_handshake_http_10(self):
2971130Szelenkov@nginx.com        self.load('websockets/mirror')
2981130Szelenkov@nginx.com
2991150Szelenkov@nginx.com        resp = self.get(
3001150Szelenkov@nginx.com            headers={
3011150Szelenkov@nginx.com                'Host': 'localhost',
3021150Szelenkov@nginx.com                'Upgrade': 'websocket',
3031150Szelenkov@nginx.com                'Connection': 'Upgrade',
3041262Szelenkov@nginx.com                'Sec-WebSocket-Key': self.ws.key(),
3051150Szelenkov@nginx.com                'Sec-WebSocket-Protocol': 'chat',
3061150Szelenkov@nginx.com                'Sec-WebSocket-Version': 13,
3071150Szelenkov@nginx.com            },
3081150Szelenkov@nginx.com            http_10=True,
3091150Szelenkov@nginx.com        )
3101130Szelenkov@nginx.com
3111596Szelenkov@nginx.com        assert resp['status'] == 400, 'status'
3121130Szelenkov@nginx.com
3131130Szelenkov@nginx.com    def test_node_websockets_handshake_uri_invalid(self):
3141130Szelenkov@nginx.com        self.load('websockets/mirror')
3151130Szelenkov@nginx.com
3161150Szelenkov@nginx.com        resp = self.get(
3171150Szelenkov@nginx.com            headers={
3181150Szelenkov@nginx.com                'Host': 'localhost',
3191150Szelenkov@nginx.com                'Upgrade': 'websocket',
3201150Szelenkov@nginx.com                'Connection': 'Upgrade',
3211262Szelenkov@nginx.com                'Sec-WebSocket-Key': self.ws.key(),
3221150Szelenkov@nginx.com                'Sec-WebSocket-Protocol': 'chat',
3231150Szelenkov@nginx.com                'Sec-WebSocket-Version': 13,
3241150Szelenkov@nginx.com            },
3251150Szelenkov@nginx.com            url='!',
3261150Szelenkov@nginx.com        )
3271130Szelenkov@nginx.com
3281596Szelenkov@nginx.com        assert resp['status'] == 400, 'status'
3291130Szelenkov@nginx.com
3301130Szelenkov@nginx.com    def test_node_websockets_protocol_absent(self):
3311130Szelenkov@nginx.com        self.load('websockets/mirror')
3321130Szelenkov@nginx.com
3331130Szelenkov@nginx.com        key = self.ws.key()
3341262Szelenkov@nginx.com        resp, sock, _ = self.ws.upgrade(
3351150Szelenkov@nginx.com            headers={
3361150Szelenkov@nginx.com                'Host': 'localhost',
3371150Szelenkov@nginx.com                'Upgrade': 'websocket',
3381150Szelenkov@nginx.com                'Connection': 'Upgrade',
3391150Szelenkov@nginx.com                'Sec-WebSocket-Key': key,
3401150Szelenkov@nginx.com                'Sec-WebSocket-Version': 13,
3411262Szelenkov@nginx.com            }
3421150Szelenkov@nginx.com        )
3431262Szelenkov@nginx.com        sock.close()
3441130Szelenkov@nginx.com
3451596Szelenkov@nginx.com        assert resp['status'] == 101, 'status'
3461596Szelenkov@nginx.com        assert resp['headers']['Upgrade'] == 'websocket', 'upgrade'
3471596Szelenkov@nginx.com        assert resp['headers']['Connection'] == 'Upgrade', 'connection'
3481596Szelenkov@nginx.com        assert resp['headers']['Sec-WebSocket-Accept'] == self.ws.accept(
3491596Szelenkov@nginx.com            key
3501596Szelenkov@nginx.com        ), 'key'
3511130Szelenkov@nginx.com
3521130Szelenkov@nginx.com    # autobahn-testsuite
3531150Szelenkov@nginx.com    #
3541130Szelenkov@nginx.com    # Some following tests fail because of Unit does not support UTF-8
3551130Szelenkov@nginx.com    # validation for websocket frames.  It should be implemented
3561130Szelenkov@nginx.com    # by application, if necessary.
3571130Szelenkov@nginx.com
3581130Szelenkov@nginx.com    def test_node_websockets_1_1_1__1_1_8(self):
3591130Szelenkov@nginx.com        self.load('websockets/mirror')
3601130Szelenkov@nginx.com
3611130Szelenkov@nginx.com        opcode = self.ws.OP_TEXT
3621130Szelenkov@nginx.com
3631130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
3641130Szelenkov@nginx.com
3651130Szelenkov@nginx.com        def check_length(length, chopsize=None):
3661130Szelenkov@nginx.com            payload = '*' * length
3671130Szelenkov@nginx.com
3681130Szelenkov@nginx.com            self.ws.frame_write(sock, opcode, payload, chopsize=chopsize)
3691130Szelenkov@nginx.com
3701130Szelenkov@nginx.com            frame = self.ws.frame_read(sock)
3711130Szelenkov@nginx.com            self.check_frame(frame, True, opcode, payload)
3721130Szelenkov@nginx.com
3731130Szelenkov@nginx.com        check_length(0)                      # 1_1_1
3741130Szelenkov@nginx.com        check_length(125)                    # 1_1_2
3751130Szelenkov@nginx.com        check_length(126)                    # 1_1_3
3761130Szelenkov@nginx.com        check_length(127)                    # 1_1_4
3771130Szelenkov@nginx.com        check_length(128)                    # 1_1_5
3781130Szelenkov@nginx.com        check_length(65535)                  # 1_1_6
3791130Szelenkov@nginx.com        check_length(65536)                  # 1_1_7
3801130Szelenkov@nginx.com        check_length(65536, chopsize = 997)  # 1_1_8
3811130Szelenkov@nginx.com
3821130Szelenkov@nginx.com        self.close_connection(sock)
3831130Szelenkov@nginx.com
3841130Szelenkov@nginx.com    def test_node_websockets_1_2_1__1_2_8(self):
3851130Szelenkov@nginx.com        self.load('websockets/mirror')
3861130Szelenkov@nginx.com
3871130Szelenkov@nginx.com        opcode = self.ws.OP_BINARY
3881130Szelenkov@nginx.com
3891130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
3901130Szelenkov@nginx.com
3911130Szelenkov@nginx.com        def check_length(length, chopsize=None):
3921130Szelenkov@nginx.com            payload = b'\xfe' * length
3931130Szelenkov@nginx.com
3941130Szelenkov@nginx.com            self.ws.frame_write(sock, opcode, payload, chopsize=chopsize)
3951130Szelenkov@nginx.com            frame = self.ws.frame_read(sock)
3961130Szelenkov@nginx.com
3971130Szelenkov@nginx.com            self.check_frame(frame, True, opcode, payload)
3981130Szelenkov@nginx.com
3991130Szelenkov@nginx.com        check_length(0)                      # 1_2_1
4001130Szelenkov@nginx.com        check_length(125)                    # 1_2_2
4011130Szelenkov@nginx.com        check_length(126)                    # 1_2_3
4021130Szelenkov@nginx.com        check_length(127)                    # 1_2_4
4031130Szelenkov@nginx.com        check_length(128)                    # 1_2_5
4041130Szelenkov@nginx.com        check_length(65535)                  # 1_2_6
4051130Szelenkov@nginx.com        check_length(65536)                  # 1_2_7
4061130Szelenkov@nginx.com        check_length(65536, chopsize = 997)  # 1_2_8
4071130Szelenkov@nginx.com
4081130Szelenkov@nginx.com        self.close_connection(sock)
4091130Szelenkov@nginx.com
4101130Szelenkov@nginx.com    def test_node_websockets_2_1__2_6(self):
4111130Szelenkov@nginx.com        self.load('websockets/mirror')
4121130Szelenkov@nginx.com
4131130Szelenkov@nginx.com        op_ping = self.ws.OP_PING
4141130Szelenkov@nginx.com        op_pong = self.ws.OP_PONG
4151130Szelenkov@nginx.com
4161130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
4171130Szelenkov@nginx.com
4181130Szelenkov@nginx.com        def check_ping(payload, chopsize=None, decode=True):
4191130Szelenkov@nginx.com            self.ws.frame_write(sock, op_ping, payload, chopsize=chopsize)
4201130Szelenkov@nginx.com            frame = self.ws.frame_read(sock)
4211130Szelenkov@nginx.com
4221130Szelenkov@nginx.com            self.check_frame(frame, True, op_pong, payload, decode=decode)
4231130Szelenkov@nginx.com
4241130Szelenkov@nginx.com        check_ping('')                                                 # 2_1
4251130Szelenkov@nginx.com        check_ping('Hello, world!')                                    # 2_2
4261130Szelenkov@nginx.com        check_ping(b'\x00\xff\xfe\xfd\xfc\xfb\x00\xff', decode=False)  # 2_3
4271130Szelenkov@nginx.com        check_ping(b'\xfe' * 125, decode=False)                        # 2_4
4281130Szelenkov@nginx.com        check_ping(b'\xfe' * 125, chopsize=1, decode=False)            # 2_6
4291130Szelenkov@nginx.com
4301130Szelenkov@nginx.com        self.close_connection(sock)
4311130Szelenkov@nginx.com
4321130Szelenkov@nginx.com        # 2_5
4331130Szelenkov@nginx.com
4341130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
4351130Szelenkov@nginx.com
4361130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PING, b'\xfe' * 126)
4371130Szelenkov@nginx.com        self.check_close(sock, 1002)
4381130Szelenkov@nginx.com
4391130Szelenkov@nginx.com    def test_node_websockets_2_7__2_9(self):
4401130Szelenkov@nginx.com        self.load('websockets/mirror')
4411130Szelenkov@nginx.com
4421130Szelenkov@nginx.com        # 2_7
4431130Szelenkov@nginx.com
4441130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
4451130Szelenkov@nginx.com
4461130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PONG, '')
4471596Szelenkov@nginx.com        assert self.recvall(sock, read_timeout=0.1) == b'', '2_7'
4481130Szelenkov@nginx.com
4491130Szelenkov@nginx.com        # 2_8
4501130Szelenkov@nginx.com
4511130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PONG, 'unsolicited pong payload')
4521596Szelenkov@nginx.com        assert self.recvall(sock, read_timeout=0.1) == b'', '2_8'
4531130Szelenkov@nginx.com
4541130Szelenkov@nginx.com        # 2_9
4551130Szelenkov@nginx.com
4561130Szelenkov@nginx.com        payload = 'ping payload'
4571130Szelenkov@nginx.com
4581130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PONG, 'unsolicited pong payload')
4591130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PING, payload)
4601130Szelenkov@nginx.com
4611130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
4621130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_PONG, payload)
4631130Szelenkov@nginx.com
4641130Szelenkov@nginx.com        self.close_connection(sock)
4651130Szelenkov@nginx.com
4661130Szelenkov@nginx.com    def test_node_websockets_2_10__2_11(self):
4671130Szelenkov@nginx.com        self.load('websockets/mirror')
4681130Szelenkov@nginx.com
4691130Szelenkov@nginx.com        # 2_10
4701130Szelenkov@nginx.com
4711130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
4721130Szelenkov@nginx.com
4731130Szelenkov@nginx.com        for i in range(0, 10):
4741130Szelenkov@nginx.com            self.ws.frame_write(sock, self.ws.OP_PING, 'payload-%d' % i)
4751130Szelenkov@nginx.com
4761130Szelenkov@nginx.com        for i in range(0, 10):
4771130Szelenkov@nginx.com            frame = self.ws.frame_read(sock)
4781130Szelenkov@nginx.com            self.check_frame(frame, True, self.ws.OP_PONG, 'payload-%d' % i)
4791130Szelenkov@nginx.com
4801130Szelenkov@nginx.com        # 2_11
4811130Szelenkov@nginx.com
4821130Szelenkov@nginx.com        for i in range(0, 10):
4831130Szelenkov@nginx.com            opcode = self.ws.OP_PING
4841130Szelenkov@nginx.com            self.ws.frame_write(sock, opcode, 'payload-%d' % i, chopsize=1)
4851130Szelenkov@nginx.com
4861130Szelenkov@nginx.com        for i in range(0, 10):
4871130Szelenkov@nginx.com            frame = self.ws.frame_read(sock)
4881130Szelenkov@nginx.com            self.check_frame(frame, True, self.ws.OP_PONG, 'payload-%d' % i)
4891130Szelenkov@nginx.com
4901130Szelenkov@nginx.com        self.close_connection(sock)
4911130Szelenkov@nginx.com
4921596Szelenkov@nginx.com    @pytest.mark.skip('not yet')
4931130Szelenkov@nginx.com    def test_node_websockets_3_1__3_7(self):
4941130Szelenkov@nginx.com        self.load('websockets/mirror')
4951130Szelenkov@nginx.com
4961130Szelenkov@nginx.com        payload = 'Hello, world!'
4971130Szelenkov@nginx.com
4981130Szelenkov@nginx.com        # 3_1
4991130Szelenkov@nginx.com
5001130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
5011130Szelenkov@nginx.com
5021130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, payload, rsv1=True)
5031130Szelenkov@nginx.com        self.check_close(sock, 1002)
5041130Szelenkov@nginx.com
5051130Szelenkov@nginx.com        # 3_2
5061130Szelenkov@nginx.com
5071130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
5081130Szelenkov@nginx.com
5091130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, payload)
5101130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, payload, rsv2=True)
5111130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PING, '')
5121130Szelenkov@nginx.com
5131130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
5141130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_TEXT, payload)
5151130Szelenkov@nginx.com
5161150Szelenkov@nginx.com        self.check_close(sock, 1002, no_close=True)
5171130Szelenkov@nginx.com
5181596Szelenkov@nginx.com        assert self.recvall(sock, read_timeout=0.1) == b'', 'empty 3_2'
5191130Szelenkov@nginx.com        sock.close()
5201130Szelenkov@nginx.com
5211130Szelenkov@nginx.com        # 3_3
5221130Szelenkov@nginx.com
5231130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
5241130Szelenkov@nginx.com
5251130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, payload)
5261130Szelenkov@nginx.com
5271130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
5281130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_TEXT, payload)
5291130Szelenkov@nginx.com
5301130Szelenkov@nginx.com        self.ws.frame_write(
5311150Szelenkov@nginx.com            sock, self.ws.OP_TEXT, payload, rsv1=True, rsv2=True
5321130Szelenkov@nginx.com        )
5331130Szelenkov@nginx.com
5341150Szelenkov@nginx.com        self.check_close(sock, 1002, no_close=True)
5351130Szelenkov@nginx.com
5361596Szelenkov@nginx.com        assert self.recvall(sock, read_timeout=0.1) == b'', 'empty 3_3'
5371130Szelenkov@nginx.com        sock.close()
5381130Szelenkov@nginx.com
5391130Szelenkov@nginx.com        # 3_4
5401130Szelenkov@nginx.com
5411130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
5421130Szelenkov@nginx.com
5431130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, payload, chopsize=1)
5441130Szelenkov@nginx.com        self.ws.frame_write(
5451150Szelenkov@nginx.com            sock, self.ws.OP_TEXT, payload, rsv3=True, chopsize=1
5461130Szelenkov@nginx.com        )
5471130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PING, '')
5481130Szelenkov@nginx.com
5491130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
5501130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_TEXT, payload)
5511130Szelenkov@nginx.com
5521150Szelenkov@nginx.com        self.check_close(sock, 1002, no_close=True)
5531130Szelenkov@nginx.com
5541596Szelenkov@nginx.com        assert self.recvall(sock, read_timeout=0.1) == b'', 'empty 3_4'
5551130Szelenkov@nginx.com        sock.close()
5561130Szelenkov@nginx.com
5571130Szelenkov@nginx.com        # 3_5
5581130Szelenkov@nginx.com
5591130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
5601130Szelenkov@nginx.com
5611130Szelenkov@nginx.com        self.ws.frame_write(
5621130Szelenkov@nginx.com            sock,
5631130Szelenkov@nginx.com            self.ws.OP_BINARY,
5641130Szelenkov@nginx.com            b'\x00\xff\xfe\xfd\xfc\xfb\x00\xff',
5651130Szelenkov@nginx.com            rsv1=True,
5661130Szelenkov@nginx.com            rsv3=True,
5671130Szelenkov@nginx.com        )
5681130Szelenkov@nginx.com
5691130Szelenkov@nginx.com        self.check_close(sock, 1002)
5701130Szelenkov@nginx.com
5711130Szelenkov@nginx.com        # 3_6
5721130Szelenkov@nginx.com
5731130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
5741130Szelenkov@nginx.com
5751130Szelenkov@nginx.com        self.ws.frame_write(
5761150Szelenkov@nginx.com            sock, self.ws.OP_PING, payload, rsv2=True, rsv3=True
5771130Szelenkov@nginx.com        )
5781130Szelenkov@nginx.com
5791130Szelenkov@nginx.com        self.check_close(sock, 1002)
5801130Szelenkov@nginx.com
5811130Szelenkov@nginx.com        # 3_7
5821130Szelenkov@nginx.com
5831130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
5841130Szelenkov@nginx.com
5851130Szelenkov@nginx.com        self.ws.frame_write(
5861150Szelenkov@nginx.com            sock, self.ws.OP_CLOSE, payload, rsv1=True, rsv2=True, rsv3=True
5871130Szelenkov@nginx.com        )
5881130Szelenkov@nginx.com
5891130Szelenkov@nginx.com        self.check_close(sock, 1002)
5901130Szelenkov@nginx.com
5911130Szelenkov@nginx.com    def test_node_websockets_4_1_1__4_2_5(self):
5921130Szelenkov@nginx.com        self.load('websockets/mirror')
5931130Szelenkov@nginx.com
5941130Szelenkov@nginx.com        payload = 'Hello, world!'
5951130Szelenkov@nginx.com
5961130Szelenkov@nginx.com        # 4_1_1
5971130Szelenkov@nginx.com
5981130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
5991130Szelenkov@nginx.com
6001130Szelenkov@nginx.com        self.ws.frame_write(sock, 0x03, '')
6011130Szelenkov@nginx.com        self.check_close(sock, 1002)
6021130Szelenkov@nginx.com
6031130Szelenkov@nginx.com        # 4_1_2
6041130Szelenkov@nginx.com
6051130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
6061130Szelenkov@nginx.com
6071130Szelenkov@nginx.com        self.ws.frame_write(sock, 0x04, 'reserved opcode payload')
6081130Szelenkov@nginx.com        self.check_close(sock, 1002)
6091130Szelenkov@nginx.com
6101130Szelenkov@nginx.com        # 4_1_3
6111130Szelenkov@nginx.com
6121130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
6131130Szelenkov@nginx.com
6141130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, payload)
6151130Szelenkov@nginx.com
6161130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
6171130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_TEXT, payload)
6181130Szelenkov@nginx.com
6191130Szelenkov@nginx.com        self.ws.frame_write(sock, 0x05, '')
6201130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PING, '')
6211130Szelenkov@nginx.com
6221130Szelenkov@nginx.com        self.check_close(sock, 1002)
6231130Szelenkov@nginx.com
6241130Szelenkov@nginx.com        # 4_1_4
6251130Szelenkov@nginx.com
6261130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
6271130Szelenkov@nginx.com
6281130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, payload)
6291130Szelenkov@nginx.com
6301130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
6311130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_TEXT, payload)
6321130Szelenkov@nginx.com
6331130Szelenkov@nginx.com        self.ws.frame_write(sock, 0x06, payload)
6341130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PING, '')
6351130Szelenkov@nginx.com
6361130Szelenkov@nginx.com        self.check_close(sock, 1002)
6371130Szelenkov@nginx.com
6381130Szelenkov@nginx.com        # 4_1_5
6391130Szelenkov@nginx.com
6401130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
6411130Szelenkov@nginx.com
6421130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, payload, chopsize=1)
6431130Szelenkov@nginx.com
6441130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
6451130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_TEXT, payload)
6461130Szelenkov@nginx.com
6471130Szelenkov@nginx.com        self.ws.frame_write(sock, 0x07, payload, chopsize=1)
6481130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PING, '')
6491130Szelenkov@nginx.com
6501130Szelenkov@nginx.com        self.check_close(sock, 1002)
6511130Szelenkov@nginx.com
6521130Szelenkov@nginx.com        # 4_2_1
6531130Szelenkov@nginx.com
6541130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
6551130Szelenkov@nginx.com
6561130Szelenkov@nginx.com        self.ws.frame_write(sock, 0x0B, '')
6571130Szelenkov@nginx.com        self.check_close(sock, 1002)
6581130Szelenkov@nginx.com
6591130Szelenkov@nginx.com        # 4_2_2
6601130Szelenkov@nginx.com
6611130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
6621130Szelenkov@nginx.com
6631130Szelenkov@nginx.com        self.ws.frame_write(sock, 0x0C, 'reserved opcode payload')
6641130Szelenkov@nginx.com        self.check_close(sock, 1002)
6651130Szelenkov@nginx.com
6661130Szelenkov@nginx.com        # 4_2_3
6671130Szelenkov@nginx.com
6681130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
6691130Szelenkov@nginx.com
6701130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, payload)
6711130Szelenkov@nginx.com
6721130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
6731130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_TEXT, payload)
6741130Szelenkov@nginx.com
6751130Szelenkov@nginx.com        self.ws.frame_write(sock, 0x0D, '')
6761130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PING, '')
6771130Szelenkov@nginx.com
6781130Szelenkov@nginx.com        self.check_close(sock, 1002)
6791130Szelenkov@nginx.com
6801130Szelenkov@nginx.com        # 4_2_4
6811130Szelenkov@nginx.com
6821130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
6831130Szelenkov@nginx.com
6841130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, payload)
6851130Szelenkov@nginx.com
6861130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
6871130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_TEXT, payload)
6881130Szelenkov@nginx.com
6891130Szelenkov@nginx.com        self.ws.frame_write(sock, 0x0E, payload)
6901130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PING, '')
6911130Szelenkov@nginx.com
6921130Szelenkov@nginx.com        self.check_close(sock, 1002)
6931130Szelenkov@nginx.com
6941130Szelenkov@nginx.com        # 4_2_5
6951130Szelenkov@nginx.com
6961130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
6971130Szelenkov@nginx.com
6981130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, payload, chopsize=1)
6991130Szelenkov@nginx.com
7001130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
7011130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_TEXT, payload)
7021130Szelenkov@nginx.com
7031130Szelenkov@nginx.com        self.ws.frame_write(sock, 0x0F, payload, chopsize=1)
7041130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PING, '')
7051130Szelenkov@nginx.com
7061130Szelenkov@nginx.com        self.check_close(sock, 1002)
7071130Szelenkov@nginx.com
7081130Szelenkov@nginx.com    def test_node_websockets_5_1__5_20(self):
7091130Szelenkov@nginx.com        self.load('websockets/mirror')
7101130Szelenkov@nginx.com
7111130Szelenkov@nginx.com        # 5_1
7121130Szelenkov@nginx.com
7131130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
7141130Szelenkov@nginx.com
7151130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PING, 'fragment1', fin=False)
7161130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CONT, 'fragment2', fin=True)
7171130Szelenkov@nginx.com        self.check_close(sock, 1002)
7181130Szelenkov@nginx.com
7191130Szelenkov@nginx.com        # 5_2
7201130Szelenkov@nginx.com
7211130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
7221130Szelenkov@nginx.com
7231130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PONG, 'fragment1', fin=False)
7241130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CONT, 'fragment2', fin=True)
7251130Szelenkov@nginx.com        self.check_close(sock, 1002)
7261130Szelenkov@nginx.com
7271130Szelenkov@nginx.com        # 5_3
7281130Szelenkov@nginx.com
7291130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
7301130Szelenkov@nginx.com
7311130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, 'fragment1', fin=False)
7321130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CONT, 'fragment2', fin=True)
7331130Szelenkov@nginx.com
7341130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
7351130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_TEXT, 'fragment1fragment2')
7361130Szelenkov@nginx.com
7371130Szelenkov@nginx.com        # 5_4
7381130Szelenkov@nginx.com
7391130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, 'fragment1', fin=False)
7401596Szelenkov@nginx.com        assert self.recvall(sock, read_timeout=0.1) == b'', '5_4'
7411130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CONT, 'fragment2', fin=True)
7421130Szelenkov@nginx.com
7431130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
7441130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_TEXT, 'fragment1fragment2')
7451130Szelenkov@nginx.com
7461130Szelenkov@nginx.com        # 5_5
7471130Szelenkov@nginx.com
7481130Szelenkov@nginx.com        self.ws.frame_write(
7491150Szelenkov@nginx.com            sock, self.ws.OP_TEXT, 'fragment1', fin=False, chopsize=1
7501130Szelenkov@nginx.com        )
7511130Szelenkov@nginx.com        self.ws.frame_write(
7521150Szelenkov@nginx.com            sock, self.ws.OP_CONT, 'fragment2', fin=True, chopsize=1
7531130Szelenkov@nginx.com        )
7541130Szelenkov@nginx.com
7551130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
7561130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_TEXT, 'fragment1fragment2')
7571130Szelenkov@nginx.com
7581130Szelenkov@nginx.com        # 5_6
7591130Szelenkov@nginx.com
7601130Szelenkov@nginx.com        ping_payload = 'ping payload'
7611130Szelenkov@nginx.com
7621130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, 'fragment1', fin=False)
7631130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PING, ping_payload)
7641130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CONT, 'fragment2', fin=True)
7651130Szelenkov@nginx.com
7661130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
7671130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_PONG, ping_payload)
7681130Szelenkov@nginx.com
7691130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
7701130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_TEXT, 'fragment1fragment2')
7711130Szelenkov@nginx.com
7721130Szelenkov@nginx.com        # 5_7
7731130Szelenkov@nginx.com
7741130Szelenkov@nginx.com        ping_payload = 'ping payload'
7751130Szelenkov@nginx.com
7761130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, 'fragment1', fin=False)
7771596Szelenkov@nginx.com        assert self.recvall(sock, read_timeout=0.1) == b'', '5_7'
7781130Szelenkov@nginx.com
7791130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PING, ping_payload)
7801130Szelenkov@nginx.com
7811130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
7821130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_PONG, ping_payload)
7831130Szelenkov@nginx.com
7841130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CONT, 'fragment2', fin=True)
7851130Szelenkov@nginx.com
7861130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
7871130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_TEXT, 'fragment1fragment2')
7881130Szelenkov@nginx.com
7891130Szelenkov@nginx.com        # 5_8
7901130Szelenkov@nginx.com
7911130Szelenkov@nginx.com        ping_payload = 'ping payload'
7921130Szelenkov@nginx.com
7931130Szelenkov@nginx.com        self.ws.frame_write(
7941150Szelenkov@nginx.com            sock, self.ws.OP_TEXT, 'fragment1', fin=False, chopsize=1
7951130Szelenkov@nginx.com        )
7961130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PING, ping_payload, chopsize=1)
7971130Szelenkov@nginx.com        self.ws.frame_write(
7981150Szelenkov@nginx.com            sock, self.ws.OP_CONT, 'fragment2', fin=True, chopsize=1
7991130Szelenkov@nginx.com        )
8001130Szelenkov@nginx.com
8011130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
8021130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_PONG, ping_payload)
8031130Szelenkov@nginx.com
8041130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
8051130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_TEXT, 'fragment1fragment2')
8061130Szelenkov@nginx.com
8071130Szelenkov@nginx.com        # 5_9
8081130Szelenkov@nginx.com
8091130Szelenkov@nginx.com        self.ws.frame_write(
8101150Szelenkov@nginx.com            sock, self.ws.OP_CONT, 'non-continuation payload', fin=True
8111130Szelenkov@nginx.com        )
8121130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, 'Hello, world!', fin=True)
8131130Szelenkov@nginx.com        self.check_close(sock, 1002)
8141130Szelenkov@nginx.com
8151130Szelenkov@nginx.com        # 5_10
8161130Szelenkov@nginx.com
8171130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
8181130Szelenkov@nginx.com
8191130Szelenkov@nginx.com        self.ws.frame_write(
8201150Szelenkov@nginx.com            sock, self.ws.OP_CONT, 'non-continuation payload', fin=True
8211130Szelenkov@nginx.com        )
8221130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, 'Hello, world!', fin=True)
8231130Szelenkov@nginx.com        self.check_close(sock, 1002)
8241130Szelenkov@nginx.com
8251130Szelenkov@nginx.com        # 5_11
8261130Szelenkov@nginx.com
8271130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
8281130Szelenkov@nginx.com
8291130Szelenkov@nginx.com        self.ws.frame_write(
8301130Szelenkov@nginx.com            sock,
8311130Szelenkov@nginx.com            self.ws.OP_CONT,
8321130Szelenkov@nginx.com            'non-continuation payload',
8331130Szelenkov@nginx.com            fin=True,
8341130Szelenkov@nginx.com            chopsize=1,
8351130Szelenkov@nginx.com        )
8361130Szelenkov@nginx.com        self.ws.frame_write(
8371150Szelenkov@nginx.com            sock, self.ws.OP_TEXT, 'Hello, world!', fin=True, chopsize=1
8381130Szelenkov@nginx.com        )
8391130Szelenkov@nginx.com        self.check_close(sock, 1002)
8401130Szelenkov@nginx.com
8411130Szelenkov@nginx.com        # 5_12
8421130Szelenkov@nginx.com
8431130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
8441130Szelenkov@nginx.com
8451130Szelenkov@nginx.com        self.ws.frame_write(
8461150Szelenkov@nginx.com            sock, self.ws.OP_CONT, 'non-continuation payload', fin=False
8471130Szelenkov@nginx.com        )
8481130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, 'Hello, world!', fin=True)
8491130Szelenkov@nginx.com        self.check_close(sock, 1002)
8501130Szelenkov@nginx.com
8511130Szelenkov@nginx.com        # 5_13
8521130Szelenkov@nginx.com
8531130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
8541130Szelenkov@nginx.com
8551130Szelenkov@nginx.com        self.ws.frame_write(
8561150Szelenkov@nginx.com            sock, self.ws.OP_CONT, 'non-continuation payload', fin=False
8571130Szelenkov@nginx.com        )
8581130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, 'Hello, world!', fin=True)
8591130Szelenkov@nginx.com        self.check_close(sock, 1002)
8601130Szelenkov@nginx.com
8611130Szelenkov@nginx.com        # 5_14
8621130Szelenkov@nginx.com
8631130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
8641130Szelenkov@nginx.com
8651130Szelenkov@nginx.com        self.ws.frame_write(
8661130Szelenkov@nginx.com            sock,
8671130Szelenkov@nginx.com            self.ws.OP_CONT,
8681130Szelenkov@nginx.com            'non-continuation payload',
8691130Szelenkov@nginx.com            fin=False,
8701130Szelenkov@nginx.com            chopsize=1,
8711130Szelenkov@nginx.com        )
8721130Szelenkov@nginx.com        self.ws.frame_write(
8731150Szelenkov@nginx.com            sock, self.ws.OP_TEXT, 'Hello, world!', fin=True, chopsize=1
8741130Szelenkov@nginx.com        )
8751130Szelenkov@nginx.com        self.check_close(sock, 1002)
8761130Szelenkov@nginx.com
8771130Szelenkov@nginx.com        # 5_15
8781130Szelenkov@nginx.com
8791130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
8801130Szelenkov@nginx.com
8811130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, 'fragment1', fin=False)
8821130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CONT, 'fragment2', fin=True)
8831130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CONT, 'fragment3', fin=False)
8841130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, 'fragment4', fin=True)
8851130Szelenkov@nginx.com        self.check_close(sock, 1002)
8861130Szelenkov@nginx.com
8871130Szelenkov@nginx.com        # 5_16
8881130Szelenkov@nginx.com
8891130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
8901130Szelenkov@nginx.com
8911130Szelenkov@nginx.com        for i in range(0, 2):
8921130Szelenkov@nginx.com            self.ws.frame_write(sock, self.ws.OP_CONT, 'fragment1', fin=False)
8931130Szelenkov@nginx.com            self.ws.frame_write(sock, self.ws.OP_TEXT, 'fragment2', fin=False)
8941130Szelenkov@nginx.com            self.ws.frame_write(sock, self.ws.OP_CONT, 'fragment3', fin=True)
8951130Szelenkov@nginx.com        self.check_close(sock, 1002)
8961130Szelenkov@nginx.com
8971130Szelenkov@nginx.com        # 5_17
8981130Szelenkov@nginx.com
8991130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
9001130Szelenkov@nginx.com
9011130Szelenkov@nginx.com        for i in range(0, 2):
9021130Szelenkov@nginx.com            self.ws.frame_write(sock, self.ws.OP_CONT, 'fragment1', fin=True)
9031130Szelenkov@nginx.com            self.ws.frame_write(sock, self.ws.OP_TEXT, 'fragment2', fin=False)
9041130Szelenkov@nginx.com            self.ws.frame_write(sock, self.ws.OP_CONT, 'fragment3', fin=True)
9051130Szelenkov@nginx.com        self.check_close(sock, 1002)
9061130Szelenkov@nginx.com
9071130Szelenkov@nginx.com        # 5_18
9081130Szelenkov@nginx.com
9091130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
9101130Szelenkov@nginx.com
9111130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, 'fragment1', fin=False)
9121130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, 'fragment2')
9131130Szelenkov@nginx.com        self.check_close(sock, 1002)
9141130Szelenkov@nginx.com
9151130Szelenkov@nginx.com        # 5_19
9161130Szelenkov@nginx.com
9171130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
9181130Szelenkov@nginx.com
9191130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, 'fragment1', fin=False)
9201130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CONT, 'fragment2', fin=False)
9211130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PING, 'pongme 1!')
9221130Szelenkov@nginx.com
9231130Szelenkov@nginx.com        time.sleep(1)
9241130Szelenkov@nginx.com
9251130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CONT, 'fragment3', fin=False)
9261130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CONT, 'fragment4', fin=False)
9271130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PING, 'pongme 2!')
9281130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CONT, 'fragment5')
9291130Szelenkov@nginx.com
9301130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
9311130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_PONG, 'pongme 1!')
9321130Szelenkov@nginx.com
9331130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
9341130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_PONG, 'pongme 2!')
9351130Szelenkov@nginx.com
9361130Szelenkov@nginx.com        self.check_frame(
9371130Szelenkov@nginx.com            self.ws.frame_read(sock),
9381130Szelenkov@nginx.com            True,
9391130Szelenkov@nginx.com            self.ws.OP_TEXT,
9401130Szelenkov@nginx.com            'fragment1fragment2fragment3fragment4fragment5',
9411130Szelenkov@nginx.com        )
9421130Szelenkov@nginx.com
9431130Szelenkov@nginx.com        # 5_20
9441130Szelenkov@nginx.com
9451130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, 'fragment1', fin=False)
9461130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CONT, 'fragment2', fin=False)
9471130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PING, 'pongme 1!')
9481130Szelenkov@nginx.com
9491130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
9501130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_PONG, 'pongme 1!')
9511130Szelenkov@nginx.com
9521130Szelenkov@nginx.com        time.sleep(1)
9531130Szelenkov@nginx.com
9541130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CONT, 'fragment3', fin=False)
9551130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CONT, 'fragment4', fin=False)
9561130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PING, 'pongme 2!')
9571130Szelenkov@nginx.com
9581130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
9591130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_PONG, 'pongme 2!')
9601130Szelenkov@nginx.com
9611596Szelenkov@nginx.com        assert self.recvall(sock, read_timeout=0.1) == b'', '5_20'
9621130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CONT, 'fragment5')
9631130Szelenkov@nginx.com
9641130Szelenkov@nginx.com        self.check_frame(
9651130Szelenkov@nginx.com            self.ws.frame_read(sock),
9661130Szelenkov@nginx.com            True,
9671130Szelenkov@nginx.com            self.ws.OP_TEXT,
9681130Szelenkov@nginx.com            'fragment1fragment2fragment3fragment4fragment5',
9691130Szelenkov@nginx.com        )
9701130Szelenkov@nginx.com
9711130Szelenkov@nginx.com        self.close_connection(sock)
9721130Szelenkov@nginx.com
9731130Szelenkov@nginx.com    def test_node_websockets_6_1_1__6_4_4(self):
9741130Szelenkov@nginx.com        self.load('websockets/mirror')
9751130Szelenkov@nginx.com
9761130Szelenkov@nginx.com        # 6_1_1
9771130Szelenkov@nginx.com
9781130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
9791130Szelenkov@nginx.com
9801130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, '')
9811130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
9821130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_TEXT, '')
9831130Szelenkov@nginx.com
9841130Szelenkov@nginx.com        # 6_1_2
9851130Szelenkov@nginx.com
9861130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, '', fin=False)
9871130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CONT, '', fin=False)
9881130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CONT, '')
9891130Szelenkov@nginx.com
9901130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
9911130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_TEXT, '')
9921130Szelenkov@nginx.com
9931130Szelenkov@nginx.com        # 6_1_3
9941130Szelenkov@nginx.com
9951130Szelenkov@nginx.com        payload = 'middle frame payload'
9961130Szelenkov@nginx.com
9971130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, '', fin=False)
9981130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CONT, payload, fin=False)
9991130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CONT, '')
10001130Szelenkov@nginx.com
10011130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
10021130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_TEXT, payload)
10031130Szelenkov@nginx.com
10041130Szelenkov@nginx.com        # 6_2_1
10051130Szelenkov@nginx.com
10061130Szelenkov@nginx.com        payload = 'Hello-µ@ßöäüàá-UTF-8!!'
10071130Szelenkov@nginx.com
10081130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, payload)
10091130Szelenkov@nginx.com
10101130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
10111130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_TEXT, payload)
10121130Szelenkov@nginx.com
10131130Szelenkov@nginx.com        # 6_2_2
10141130Szelenkov@nginx.com
10151130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, payload[:12], fin=False)
10161130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CONT, payload[12:])
10171130Szelenkov@nginx.com
10181130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
10191130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_TEXT, payload)
10201130Szelenkov@nginx.com
10211130Szelenkov@nginx.com        # 6_2_3
10221130Szelenkov@nginx.com
10231130Szelenkov@nginx.com        self.ws.message(sock, self.ws.OP_TEXT, payload, fragmention_size=1)
10241130Szelenkov@nginx.com
10251130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
10261130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_TEXT, payload)
10271130Szelenkov@nginx.com
10281130Szelenkov@nginx.com        # 6_2_4
10291130Szelenkov@nginx.com
10301130Szelenkov@nginx.com        payload = '\xce\xba\xe1\xbd\xb9\xcf\x83\xce\xbc\xce\xb5'
10311130Szelenkov@nginx.com
10321130Szelenkov@nginx.com        self.ws.message(sock, self.ws.OP_TEXT, payload, fragmention_size=1)
10331130Szelenkov@nginx.com
10341130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
10351130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_TEXT, payload)
10361130Szelenkov@nginx.com
10371130Szelenkov@nginx.com        self.close_connection(sock)
10381130Szelenkov@nginx.com
10391150Szelenkov@nginx.com#        Unit does not support UTF-8 validation
10401150Szelenkov@nginx.com#
10411130Szelenkov@nginx.com#        # 6_3_1 FAIL
10421130Szelenkov@nginx.com#
10431130Szelenkov@nginx.com#        payload_1 = '\xce\xba\xe1\xbd\xb9\xcf\x83\xce\xbc\xce\xb5'
10441130Szelenkov@nginx.com#        payload_2 = '\xed\xa0\x80'
10451130Szelenkov@nginx.com#        payload_3 = '\x65\x64\x69\x74\x65\x64'
10461130Szelenkov@nginx.com#
10471130Szelenkov@nginx.com#        payload = payload_1 + payload_2 + payload_3
10481130Szelenkov@nginx.com#
10491130Szelenkov@nginx.com#        self.ws.message(sock, self.ws.OP_TEXT, payload)
10501130Szelenkov@nginx.com#        self.check_close(sock, 1007)
10511130Szelenkov@nginx.com#
10521130Szelenkov@nginx.com#        # 6_3_2 FAIL
10531130Szelenkov@nginx.com#
10541130Szelenkov@nginx.com#        _, sock, _ = self.ws.upgrade()
10551130Szelenkov@nginx.com#
10561130Szelenkov@nginx.com#        self.ws.message(sock, self.ws.OP_TEXT, payload, fragmention_size=1)
10571130Szelenkov@nginx.com#        self.check_close(sock, 1007)
10581130Szelenkov@nginx.com#
10591130Szelenkov@nginx.com#        # 6_4_1 ... 6_4_4 FAIL
10601130Szelenkov@nginx.com
10611130Szelenkov@nginx.com    def test_node_websockets_7_1_1__7_5_1(self):
10621130Szelenkov@nginx.com        self.load('websockets/mirror')
10631130Szelenkov@nginx.com
10641130Szelenkov@nginx.com        # 7_1_1
10651130Szelenkov@nginx.com
10661130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
10671130Szelenkov@nginx.com
10681130Szelenkov@nginx.com        payload = "Hello World!"
10691130Szelenkov@nginx.com
10701130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, payload)
10711130Szelenkov@nginx.com
10721130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
10731130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_TEXT, payload)
10741130Szelenkov@nginx.com
10751130Szelenkov@nginx.com        self.close_connection(sock)
10761130Szelenkov@nginx.com
10771130Szelenkov@nginx.com        # 7_1_2
10781130Szelenkov@nginx.com
10791130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
10801130Szelenkov@nginx.com
10811130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CLOSE, self.ws.serialize_close())
10821130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CLOSE, self.ws.serialize_close())
10831130Szelenkov@nginx.com
10841130Szelenkov@nginx.com        self.check_close(sock)
10851130Szelenkov@nginx.com
10861130Szelenkov@nginx.com        # 7_1_3
10871130Szelenkov@nginx.com
10881130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
10891130Szelenkov@nginx.com
10901130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CLOSE, self.ws.serialize_close())
10911150Szelenkov@nginx.com        self.check_close(sock, no_close=True)
10921130Szelenkov@nginx.com
10931130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PING, '')
10941596Szelenkov@nginx.com        assert self.recvall(sock, read_timeout=0.1) == b'', 'empty soc'
10951130Szelenkov@nginx.com
10961130Szelenkov@nginx.com        sock.close()
10971130Szelenkov@nginx.com
10981130Szelenkov@nginx.com        # 7_1_4
10991130Szelenkov@nginx.com
11001130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
11011130Szelenkov@nginx.com
11021130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CLOSE, self.ws.serialize_close())
11031150Szelenkov@nginx.com        self.check_close(sock, no_close=True)
11041130Szelenkov@nginx.com
11051130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, payload)
11061596Szelenkov@nginx.com        assert self.recvall(sock, read_timeout=0.1) == b'', 'empty soc'
11071130Szelenkov@nginx.com
11081130Szelenkov@nginx.com        sock.close()
11091130Szelenkov@nginx.com
11101130Szelenkov@nginx.com        # 7_1_5
11111130Szelenkov@nginx.com
11121130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
11131130Szelenkov@nginx.com
11141130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, 'fragment1', fin=False)
11151130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CLOSE, self.ws.serialize_close())
11161150Szelenkov@nginx.com        self.check_close(sock, no_close=True)
11171130Szelenkov@nginx.com
11181130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CONT, 'fragment2')
11191596Szelenkov@nginx.com        assert self.recvall(sock, read_timeout=0.1) == b'', 'empty soc'
11201130Szelenkov@nginx.com
11211130Szelenkov@nginx.com        sock.close()
11221130Szelenkov@nginx.com
11231130Szelenkov@nginx.com        # 7_1_6
11241130Szelenkov@nginx.com
11251130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
11261130Szelenkov@nginx.com
11271150Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, 'BAsd7&jh23' * 26 * 2 ** 10)
11281130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_TEXT, payload)
11291130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CLOSE, self.ws.serialize_close())
11301130Szelenkov@nginx.com
11311130Szelenkov@nginx.com        self.recvall(sock, read_timeout=1)
11321130Szelenkov@nginx.com
11331130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_PING, '')
11341596Szelenkov@nginx.com        assert self.recvall(sock, read_timeout=0.1) == b'', 'empty soc'
11351130Szelenkov@nginx.com
11361130Szelenkov@nginx.com        sock.close()
11371130Szelenkov@nginx.com
11381262Szelenkov@nginx.com        # 7_3_1
11391130Szelenkov@nginx.com
11401130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
11411130Szelenkov@nginx.com
11421130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CLOSE, '')
11431130Szelenkov@nginx.com        self.check_close(sock)
11441130Szelenkov@nginx.com
11451130Szelenkov@nginx.com        # 7_3_2
11461130Szelenkov@nginx.com
11471130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
11481130Szelenkov@nginx.com
11491130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CLOSE, 'a')
11501130Szelenkov@nginx.com        self.check_close(sock, 1002)
11511130Szelenkov@nginx.com
11521130Szelenkov@nginx.com        # 7_3_3
11531130Szelenkov@nginx.com
11541130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
11551130Szelenkov@nginx.com
11561130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CLOSE, self.ws.serialize_close())
11571130Szelenkov@nginx.com        self.check_close(sock)
11581130Szelenkov@nginx.com
11591130Szelenkov@nginx.com        # 7_3_4
11601130Szelenkov@nginx.com
11611130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
11621130Szelenkov@nginx.com
11631150Szelenkov@nginx.com        payload = self.ws.serialize_close(reason='Hello World!')
11641130Szelenkov@nginx.com
11651130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CLOSE, payload)
11661130Szelenkov@nginx.com        self.check_close(sock)
11671130Szelenkov@nginx.com
11681130Szelenkov@nginx.com        # 7_3_5
11691130Szelenkov@nginx.com
11701130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
11711130Szelenkov@nginx.com
11721150Szelenkov@nginx.com        payload = self.ws.serialize_close(reason='*' * 123)
11731130Szelenkov@nginx.com
11741130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CLOSE, payload)
11751130Szelenkov@nginx.com        self.check_close(sock)
11761130Szelenkov@nginx.com
11771130Szelenkov@nginx.com        # 7_3_6
11781130Szelenkov@nginx.com
11791130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
11801130Szelenkov@nginx.com
11811150Szelenkov@nginx.com        payload = self.ws.serialize_close(reason='*' * 124)
11821130Szelenkov@nginx.com
11831130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CLOSE, payload)
11841130Szelenkov@nginx.com        self.check_close(sock, 1002)
11851130Szelenkov@nginx.com
11861150Szelenkov@nginx.com#        # 7_5_1 FAIL Unit does not support UTF-8 validation
11871150Szelenkov@nginx.com#
11881130Szelenkov@nginx.com#        _, sock, _ = self.ws.upgrade()
11891130Szelenkov@nginx.com#
11901130Szelenkov@nginx.com#        payload = self.ws.serialize_close(reason = '\xce\xba\xe1\xbd\xb9\xcf' \
11911130Szelenkov@nginx.com#            '\x83\xce\xbc\xce\xb5\xed\xa0\x80\x65\x64\x69\x74\x65\x64')
11921130Szelenkov@nginx.com#
11931130Szelenkov@nginx.com#        self.ws.frame_write(sock, self.ws.OP_CLOSE, payload)
11941130Szelenkov@nginx.com#        self.check_close(sock, 1007)
11951130Szelenkov@nginx.com
11961130Szelenkov@nginx.com    def test_node_websockets_7_7_X__7_9_X(self):
11971130Szelenkov@nginx.com        self.load('websockets/mirror')
11981130Szelenkov@nginx.com
11991130Szelenkov@nginx.com        valid_codes = [
12001130Szelenkov@nginx.com            1000,
12011130Szelenkov@nginx.com            1001,
12021130Szelenkov@nginx.com            1002,
12031130Szelenkov@nginx.com            1003,
12041130Szelenkov@nginx.com            1007,
12051130Szelenkov@nginx.com            1008,
12061130Szelenkov@nginx.com            1009,
12071130Szelenkov@nginx.com            1010,
12081130Szelenkov@nginx.com            1011,
12091130Szelenkov@nginx.com            3000,
12101130Szelenkov@nginx.com            3999,
12111130Szelenkov@nginx.com            4000,
12121130Szelenkov@nginx.com            4999,
12131130Szelenkov@nginx.com        ]
12141130Szelenkov@nginx.com
12151130Szelenkov@nginx.com        invalid_codes = [0, 999, 1004, 1005, 1006, 1016, 1100, 2000, 2999]
12161130Szelenkov@nginx.com
12171130Szelenkov@nginx.com        for code in valid_codes:
12181130Szelenkov@nginx.com            _, sock, _ = self.ws.upgrade()
12191130Szelenkov@nginx.com
12201150Szelenkov@nginx.com            payload = self.ws.serialize_close(code=code)
12211130Szelenkov@nginx.com
12221130Szelenkov@nginx.com            self.ws.frame_write(sock, self.ws.OP_CLOSE, payload)
12231130Szelenkov@nginx.com            self.check_close(sock)
12241130Szelenkov@nginx.com
12251130Szelenkov@nginx.com        for code in invalid_codes:
12261130Szelenkov@nginx.com            _, sock, _ = self.ws.upgrade()
12271130Szelenkov@nginx.com
12281150Szelenkov@nginx.com            payload = self.ws.serialize_close(code=code)
12291130Szelenkov@nginx.com
12301130Szelenkov@nginx.com            self.ws.frame_write(sock, self.ws.OP_CLOSE, payload)
12311130Szelenkov@nginx.com            self.check_close(sock, 1002)
12321130Szelenkov@nginx.com
12331130Szelenkov@nginx.com    def test_node_websockets_7_13_1__7_13_2(self):
12341130Szelenkov@nginx.com        self.load('websockets/mirror')
12351130Szelenkov@nginx.com
12361130Szelenkov@nginx.com        # 7_13_1
12371130Szelenkov@nginx.com
12381130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
12391130Szelenkov@nginx.com
12401150Szelenkov@nginx.com        payload = self.ws.serialize_close(code=5000)
12411130Szelenkov@nginx.com
12421130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CLOSE, payload)
12431130Szelenkov@nginx.com        self.check_close(sock, 1002)
12441130Szelenkov@nginx.com
12451130Szelenkov@nginx.com        # 7_13_2
12461130Szelenkov@nginx.com
12471130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
12481130Szelenkov@nginx.com
12491130Szelenkov@nginx.com        payload = struct.pack('!I', 65536) + ''.encode('utf-8')
12501130Szelenkov@nginx.com
12511130Szelenkov@nginx.com        self.ws.frame_write(sock, self.ws.OP_CLOSE, payload)
12521130Szelenkov@nginx.com        self.check_close(sock, 1002)
12531130Szelenkov@nginx.com
12541596Szelenkov@nginx.com    def test_node_websockets_9_1_1__9_6_6(self, is_unsafe):
12551596Szelenkov@nginx.com        if not is_unsafe:
12561596Szelenkov@nginx.com            pytest.skip('unsafe, long run')
12571130Szelenkov@nginx.com
12581130Szelenkov@nginx.com        self.load('websockets/mirror')
12591130Szelenkov@nginx.com
12601596Szelenkov@nginx.com        assert 'success' in self.conf(
12611596Szelenkov@nginx.com            {
12621596Szelenkov@nginx.com                'http': {
12631596Szelenkov@nginx.com                    'websocket': {
12641596Szelenkov@nginx.com                        'max_frame_size': 33554432,
12651596Szelenkov@nginx.com                        'keepalive_interval': 0,
12661130Szelenkov@nginx.com                    }
12671596Szelenkov@nginx.com                }
12681596Szelenkov@nginx.com            },
12691596Szelenkov@nginx.com            'settings',
12701596Szelenkov@nginx.com        ), 'increase max_frame_size and keepalive_interval'
12711130Szelenkov@nginx.com
12721130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
12731130Szelenkov@nginx.com
12741130Szelenkov@nginx.com        op_text = self.ws.OP_TEXT
12751130Szelenkov@nginx.com        op_binary = self.ws.OP_BINARY
12761130Szelenkov@nginx.com
12771130Szelenkov@nginx.com        def check_payload(opcode, length, chopsize=None):
12781130Szelenkov@nginx.com            if opcode == self.ws.OP_TEXT:
12791130Szelenkov@nginx.com                payload = '*' * length
12801130Szelenkov@nginx.com            else:
12811130Szelenkov@nginx.com                payload = b'*' * length
12821130Szelenkov@nginx.com
12831130Szelenkov@nginx.com            self.ws.frame_write(sock, opcode, payload, chopsize=chopsize)
12841130Szelenkov@nginx.com            frame = self.ws.frame_read(sock, read_timeout=5)
12851130Szelenkov@nginx.com            self.check_frame(frame, True, opcode, payload)
12861130Szelenkov@nginx.com
12871130Szelenkov@nginx.com        def check_message(opcode, f_size):
12881130Szelenkov@nginx.com            if opcode == self.ws.OP_TEXT:
12891150Szelenkov@nginx.com                payload = '*' * 4 * 2 ** 20
12901130Szelenkov@nginx.com            else:
12911150Szelenkov@nginx.com                payload = b'*' * 4 * 2 ** 20
12921130Szelenkov@nginx.com
12931130Szelenkov@nginx.com            self.ws.message(sock, opcode, payload, fragmention_size=f_size)
12941130Szelenkov@nginx.com            frame = self.ws.frame_read(sock, read_timeout=5)
12951130Szelenkov@nginx.com            self.check_frame(frame, True, opcode, payload)
12961130Szelenkov@nginx.com
12971150Szelenkov@nginx.com        check_payload(op_text, 64 * 2 ** 10)              # 9_1_1
12981150Szelenkov@nginx.com        check_payload(op_text, 256 * 2 ** 10)             # 9_1_2
12991150Szelenkov@nginx.com        check_payload(op_text, 2 ** 20)                   # 9_1_3
13001150Szelenkov@nginx.com        check_payload(op_text, 4 * 2 ** 20)               # 9_1_4
13011150Szelenkov@nginx.com        check_payload(op_text, 8 * 2 ** 20)               # 9_1_5
13021150Szelenkov@nginx.com        check_payload(op_text, 16 * 2 ** 20)              # 9_1_6
13031130Szelenkov@nginx.com
13041150Szelenkov@nginx.com        check_payload(op_binary, 64 * 2 ** 10)            # 9_2_1
13051150Szelenkov@nginx.com        check_payload(op_binary, 256 * 2 ** 10)           # 9_2_2
13061150Szelenkov@nginx.com        check_payload(op_binary, 2 ** 20)                 # 9_2_3
13071150Szelenkov@nginx.com        check_payload(op_binary, 4 * 2 ** 20)             # 9_2_4
13081150Szelenkov@nginx.com        check_payload(op_binary, 8 * 2 ** 20)             # 9_2_5
13091150Szelenkov@nginx.com        check_payload(op_binary, 16 * 2 ** 20)            # 9_2_6
13101130Szelenkov@nginx.com
13111596Szelenkov@nginx.com        if option.system != 'Darwin' and option.system != 'FreeBSD':
13121150Szelenkov@nginx.com            check_message(op_text, 64)                    # 9_3_1
13131150Szelenkov@nginx.com            check_message(op_text, 256)                   # 9_3_2
13141150Szelenkov@nginx.com            check_message(op_text, 2 ** 10)               # 9_3_3
13151150Szelenkov@nginx.com            check_message(op_text, 4 * 2 ** 10)           # 9_3_4
13161150Szelenkov@nginx.com            check_message(op_text, 16 * 2 ** 10)          # 9_3_5
13171150Szelenkov@nginx.com            check_message(op_text, 64 * 2 ** 10)          # 9_3_6
13181150Szelenkov@nginx.com            check_message(op_text, 256 * 2 ** 10)         # 9_3_7
13191150Szelenkov@nginx.com            check_message(op_text, 2 ** 20)               # 9_3_8
13201150Szelenkov@nginx.com            check_message(op_text, 4 * 2 ** 20)           # 9_3_9
13211130Szelenkov@nginx.com
13221150Szelenkov@nginx.com            check_message(op_binary, 64)                  # 9_4_1
13231150Szelenkov@nginx.com            check_message(op_binary, 256)                 # 9_4_2
13241150Szelenkov@nginx.com            check_message(op_binary, 2 ** 10)             # 9_4_3
13251150Szelenkov@nginx.com            check_message(op_binary, 4 * 2 ** 10)         # 9_4_4
13261150Szelenkov@nginx.com            check_message(op_binary, 16 * 2 ** 10)        # 9_4_5
13271150Szelenkov@nginx.com            check_message(op_binary, 64 * 2 ** 10)        # 9_4_6
13281150Szelenkov@nginx.com            check_message(op_binary, 256 * 2 ** 10)       # 9_4_7
13291150Szelenkov@nginx.com            check_message(op_binary, 2 ** 20)             # 9_4_8
13301150Szelenkov@nginx.com            check_message(op_binary, 4 * 2 ** 20)         # 9_4_9
13311130Szelenkov@nginx.com
13321150Szelenkov@nginx.com        check_payload(op_text, 2 ** 20, chopsize=64)      # 9_5_1
13331150Szelenkov@nginx.com        check_payload(op_text, 2 ** 20, chopsize=128)     # 9_5_2
13341150Szelenkov@nginx.com        check_payload(op_text, 2 ** 20, chopsize=256)     # 9_5_3
13351150Szelenkov@nginx.com        check_payload(op_text, 2 ** 20, chopsize=512)     # 9_5_4
13361150Szelenkov@nginx.com        check_payload(op_text, 2 ** 20, chopsize=1024)    # 9_5_5
13371150Szelenkov@nginx.com        check_payload(op_text, 2 ** 20, chopsize=2048)    # 9_5_6
13381130Szelenkov@nginx.com
13391150Szelenkov@nginx.com        check_payload(op_binary, 2 ** 20, chopsize=64)    # 9_6_1
13401150Szelenkov@nginx.com        check_payload(op_binary, 2 ** 20, chopsize=128)   # 9_6_2
13411150Szelenkov@nginx.com        check_payload(op_binary, 2 ** 20, chopsize=256)   # 9_6_3
13421150Szelenkov@nginx.com        check_payload(op_binary, 2 ** 20, chopsize=512)   # 9_6_4
13431150Szelenkov@nginx.com        check_payload(op_binary, 2 ** 20, chopsize=1024)  # 9_6_5
13441150Szelenkov@nginx.com        check_payload(op_binary, 2 ** 20, chopsize=2048)  # 9_6_6
13451130Szelenkov@nginx.com
13461130Szelenkov@nginx.com        self.close_connection(sock)
13471130Szelenkov@nginx.com
13481130Szelenkov@nginx.com    def test_node_websockets_10_1_1(self):
13491130Szelenkov@nginx.com        self.load('websockets/mirror')
13501130Szelenkov@nginx.com
13511130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
13521130Szelenkov@nginx.com
13531130Szelenkov@nginx.com        payload = '*' * 65536
13541130Szelenkov@nginx.com
13551130Szelenkov@nginx.com        self.ws.message(sock, self.ws.OP_TEXT, payload, fragmention_size=1300)
13561130Szelenkov@nginx.com
13571130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
13581130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_TEXT, payload)
13591130Szelenkov@nginx.com
13601130Szelenkov@nginx.com        self.close_connection(sock)
13611130Szelenkov@nginx.com
13621130Szelenkov@nginx.com    # settings
13631130Szelenkov@nginx.com
13641130Szelenkov@nginx.com    def test_node_websockets_max_frame_size(self):
13651130Szelenkov@nginx.com        self.load('websockets/mirror')
13661130Szelenkov@nginx.com
13671596Szelenkov@nginx.com        assert 'success' in self.conf(
13681596Szelenkov@nginx.com            {'http': {'websocket': {'max_frame_size': 100}}}, 'settings'
13691596Szelenkov@nginx.com        ), 'configure max_frame_size'
13701130Szelenkov@nginx.com
13711130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
13721130Szelenkov@nginx.com
13731130Szelenkov@nginx.com        payload = '*' * 94
13741130Szelenkov@nginx.com        opcode = self.ws.OP_TEXT
13751130Szelenkov@nginx.com
13761130Szelenkov@nginx.com        self.ws.frame_write(sock, opcode, payload)  # frame length is 100
13771130Szelenkov@nginx.com
13781130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
13791130Szelenkov@nginx.com        self.check_frame(frame, True, opcode, payload)
13801130Szelenkov@nginx.com
13811130Szelenkov@nginx.com        payload = '*' * 95
13821130Szelenkov@nginx.com
13831130Szelenkov@nginx.com        self.ws.frame_write(sock, opcode, payload)  # frame length is 101
13841150Szelenkov@nginx.com        self.check_close(sock, 1009)  # 1009 - CLOSE_TOO_LARGE
13851130Szelenkov@nginx.com
13861130Szelenkov@nginx.com    def test_node_websockets_read_timeout(self):
13871130Szelenkov@nginx.com        self.load('websockets/mirror')
13881130Szelenkov@nginx.com
13891596Szelenkov@nginx.com        assert 'success' in self.conf(
13901596Szelenkov@nginx.com            {'http': {'websocket': {'read_timeout': 5}}}, 'settings'
13911596Szelenkov@nginx.com        ), 'configure read_timeout'
13921130Szelenkov@nginx.com
13931130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
13941130Szelenkov@nginx.com
13951130Szelenkov@nginx.com        frame = self.ws.frame_to_send(self.ws.OP_TEXT, 'blah')
13961130Szelenkov@nginx.com        sock.sendall(frame[:2])
13971130Szelenkov@nginx.com
13981130Szelenkov@nginx.com        time.sleep(2)
13991130Szelenkov@nginx.com
14001150Szelenkov@nginx.com        self.check_close(sock, 1001)  # 1001 - CLOSE_GOING_AWAY
14011130Szelenkov@nginx.com
14021130Szelenkov@nginx.com    def test_node_websockets_keepalive_interval(self):
14031130Szelenkov@nginx.com        self.load('websockets/mirror')
14041130Szelenkov@nginx.com
14051596Szelenkov@nginx.com        assert 'success' in self.conf(
14061596Szelenkov@nginx.com            {'http': {'websocket': {'keepalive_interval': 5}}}, 'settings'
14071596Szelenkov@nginx.com        ), 'configure keepalive_interval'
14081130Szelenkov@nginx.com
14091130Szelenkov@nginx.com        _, sock, _ = self.ws.upgrade()
14101130Szelenkov@nginx.com
14111130Szelenkov@nginx.com        frame = self.ws.frame_to_send(self.ws.OP_TEXT, 'blah')
14121130Szelenkov@nginx.com        sock.sendall(frame[:2])
14131130Szelenkov@nginx.com
14141130Szelenkov@nginx.com        time.sleep(2)
14151130Szelenkov@nginx.com
14161130Szelenkov@nginx.com        frame = self.ws.frame_read(sock)
14171130Szelenkov@nginx.com        self.check_frame(frame, True, self.ws.OP_PING, '')  # PING frame
14181130Szelenkov@nginx.com
14191130Szelenkov@nginx.com        sock.close()
1420