summaryrefslogtreecommitdiff
path: root/chromium/net/data/websocket/close-with-split-packet_wsh.py
blob: d5185c4114bc4e4eb2cb2124691b7dd518e9532b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
# Copyright (c) 2012 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import struct

from mod_pywebsocket import handshake
from mod_pywebsocket import stream


def web_socket_do_extra_handshake(_request):
  pass


def web_socket_transfer_data(request):
  line = request.ws_stream.receive_message()
  if line is None:
    return
  if isinstance(line, unicode):
    request.ws_stream.send_message(line, binary=False)
  else:
    request.ws_stream.send_message(line, binary=True)


def web_socket_passive_closing_handshake(request):
  code = struct.pack('!H', 1000)
  packet = stream.create_close_frame(code + 'split test'.encode('utf-8'))
  request.connection.write(packet[:1])
  request.connection.write(packet[1:])
  raise handshake.AbortedByUserException('Abort the connection')