diff options
Diffstat (limited to 'dom/base/test/file_websocket_permessage_deflate_rejected_wsh.py')
-rw-r--r-- | dom/base/test/file_websocket_permessage_deflate_rejected_wsh.py | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/dom/base/test/file_websocket_permessage_deflate_rejected_wsh.py b/dom/base/test/file_websocket_permessage_deflate_rejected_wsh.py new file mode 100644 index 000000000..b21da97cd --- /dev/null +++ b/dom/base/test/file_websocket_permessage_deflate_rejected_wsh.py @@ -0,0 +1,23 @@ +from mod_pywebsocket import msgutil +from mod_pywebsocket import common + +def web_socket_do_extra_handshake(request): + deflate_removed = False + + if request.ws_extension_processors is not None: + for extension_processor in request.ws_extension_processors: + if extension_processor.name() == "deflate": + request.ws_extension_processors.remove(extension_processor) + deflate_removed = True + + if deflate_removed is False: + raise ValueError('deflate extension processor not found') + +def web_socket_transfer_data(request): + while True: + rcvd = msgutil.receive_message(request) + opcode = request.ws_stream.get_last_received_opcode() + if (opcode == common.OPCODE_BINARY): + msgutil.send_message(request, rcvd, binary=True) + elif (opcode == common.OPCODE_TEXT): + msgutil.send_message(request, rcvd) |