net/data/websocket/protocol-test_wsh.py

   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
   3 # found in the LICENSE file.
   4
   5 import cgi
   6 from mod_pywebsocket import msgutil
   7
   8
   9 def web_socket_do_extra_handshake(request):
  10   r = request.ws_resource.split('?', 1)
  11   if len(r) == 1:
  12     return
  13   param = cgi.parse_qs(r[1])
  14   if 'protocol' in param:
  15     request.ws_protocol = param['protocol'][0]
  16
  17
  18 def web_socket_transfer_data(request):
  19   msgutil.send_message(request, request.ws_protocol)
  20   # Wait for a close message.
  21   unused = request.ws_stream.receive_message()