net/data/websocket/close-with-split-packet_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 struct
   6
   7 from mod_pywebsocket import handshake
   8 from mod_pywebsocket import stream
   9
  10
  11 def web_socket_do_extra_handshake(_request):
  12   pass
  13
  14
  15 def web_socket_transfer_data(request):
  16   # Just waiting...
  17   request.ws_stream.receive_message()
  18
  19
  20 def web_socket_passive_closing_handshake(request):
  21   code = struct.pack('!H', 3004)
  22   packet = stream.create_close_frame(code + 'split test'.encode('utf-8'))
  23   request.connection.write(packet[:1])
  24   request.connection.write(packet[1:])
  25   raise handshake.AbortedByUserException('Abort the connection')