OLD | NEW |
1 # Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 # Copyright (c) 2012 The Chromium Authors. All rights reserved. |
2 # Use of this source code is governed by a BSD-style license that can be | 2 # Use of this source code is governed by a BSD-style license that can be |
3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
4 | 4 |
5 import cgi | 5 import cgi |
6 from mod_pywebsocket import msgutil | 6 from mod_pywebsocket import msgutil |
7 | 7 |
8 | 8 |
9 def web_socket_do_extra_handshake(request): | 9 def web_socket_do_extra_handshake(request): |
10 r = request.ws_resource.split('?', 1) | 10 r = request.ws_resource.split('?', 1) |
11 if len(r) == 1: | 11 if len(r) == 1: |
12 return | 12 return |
13 param = cgi.parse_qs(r[1]) | 13 param = cgi.parse_qs(r[1]) |
14 if 'protocol' in param: | 14 if 'protocol' in param: |
15 request.ws_protocol = param['protocol'][0] | 15 request.ws_protocol = param['protocol'][0] |
16 | 16 |
17 | 17 |
18 def web_socket_transfer_data(request): | 18 def web_socket_transfer_data(request): |
19 msgutil.send_message(request, request.ws_protocol) | 19 msgutil.send_message(request, request.ws_protocol) |
| 20 # Wait for a close message. |
| 21 unused = request.ws_stream.receive_message() |
OLD | NEW |