content/base/test/file_websocket_basic_wsh.py@6474c204b198 (annotated)
content/base/test/file_websocket_basic_wsh.py
Wed, 31 Dec 2014 06:09:35 +0100
- author
- Michael Schloh von Bennewitz <michael@schloh.com>
- date
- Wed, 31 Dec 2014 06:09:35 +0100
- changeset 0
- 6474c204b198
- permissions
- -rw-r--r--
Cloned upstream origin tor-browser at tor-browser-31.3.0esr-4.5-1-build1
revision ID fc1c9ff7c1b2defdbc039f12214767608f46423f for hacking purpose.
| michael@0 | 1 | from mod_pywebsocket import msgutil |
| michael@0 | 2 | |
| michael@0 | 3 | def web_socket_do_extra_handshake(request): |
| michael@0 | 4 | # must set request.ws_protocol to the selected version from ws_requested_protocols |
| michael@0 | 5 | request.ws_protocol = request.ws_requested_protocols[0] |
| michael@0 | 6 | |
| michael@0 | 7 | if (request.ws_protocol == 'error'): |
| michael@0 | 8 | raise ValueError('Error') |
| michael@0 | 9 | pass |
| michael@0 | 10 | |
| michael@0 | 11 | def web_socket_transfer_data(request): |
| michael@0 | 12 | while True: |
| michael@0 | 13 | line = msgutil.receive_message(request) |
| michael@0 | 14 | if line == 'protocol': |
| michael@0 | 15 | msgutil.send_message(request, request.ws_protocol) |
| michael@0 | 16 | continue |
| michael@0 | 17 | |
| michael@0 | 18 | if line == 'resource': |
| michael@0 | 19 | msgutil.send_message(request, request.ws_resource) |
| michael@0 | 20 | continue |
| michael@0 | 21 | |
| michael@0 | 22 | if line == 'origin': |
| michael@0 | 23 | msgutil.send_message(request, request.ws_origin) |
| michael@0 | 24 | continue |
| michael@0 | 25 | |
| michael@0 | 26 | msgutil.send_message(request, line) |
| michael@0 | 27 | |
| michael@0 | 28 | if line == 'end': |
| michael@0 | 29 | return |
