| 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 |