1/*
2 * Copyright (c) 2015, Peter Thorson. All rights reserved.
3 *
4 * Redistribution and use in source and binary forms, with or without
5 * modification, are permitted provided that the following conditions are met:
6 * * Redistributions of source code must retain the above copyright
7 * notice, this list of conditions and the following disclaimer.
8 * * Redistributions in binary form must reproduce the above copyright
9 * notice, this list of conditions and the following disclaimer in the
10 * documentation and/or other materials provided with the distribution.
11 * * Neither the name of the WebSocket++ Project nor the
12 * names of its contributors may be used to endorse or promote products
13 * derived from this software without specific prior written permission.
14 *
15 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
16 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
17 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
18 * ARE DISCLAIMED. IN NO EVENT SHALL PETER THORSON BE LIABLE FOR ANY
19 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
20 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
21 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
22 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
23 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
24 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
25 */
26
27/*
28 * Author: Benjamin Sergeant
29 * Copyright (c) 2018 Machine Zone, Inc. All rights reserved.
30 *
31 * Adapted from websocketpp/extensions/permessage_deflate/enabled.hpp
32 * (same license as MZ: https://opensource.org/licenses/BSD-3-Clause)
33 *
34 * - Reused zlib compression + decompression bits.
35 * - Refactored to have 2 class for compression and decompression, to allow multi-threading
36 * and make sure that _compressBuffer is not shared between threads.
37 * - Original code wasn't working for some reason, I had to add checks
38 * for the presence of the kEmptyUncompressedBlock at the end of buffer so that servers
39 * would start accepting receiving/decoding compressed messages. Original code was probably
40 * modifying the passed in buffers before processing in enabled.hpp ?
41 * - Added more documentation.
42 *
43 * Per message Deflate RFC: https://tools.ietf.org/html/rfc7692
44 * Chrome websocket ->
45 * https://github.com/chromium/chromium/tree/2ca8c5037021c9d2ecc00b787d58a31ed8fc8bcb/net/websockets
46 *
47 */
48
49#include "IXWebSocketPerMessageDeflate.h"
50
51#include "IXUniquePtr.h"
52#include "IXWebSocketPerMessageDeflateCodec.h"
53#include "IXWebSocketPerMessageDeflateOptions.h"
54
55namespace ix
56{
57 WebSocketPerMessageDeflate::WebSocketPerMessageDeflate()
58 : _compressor(ix::make_unique<WebSocketPerMessageDeflateCompressor>())
59 , _decompressor(ix::make_unique<WebSocketPerMessageDeflateDecompressor>())
60 {
61 ;
62 }
63
64 WebSocketPerMessageDeflate::~WebSocketPerMessageDeflate()
65 {
66 ;
67 }
68
69 bool WebSocketPerMessageDeflate::init(
70 const WebSocketPerMessageDeflateOptions& perMessageDeflateOptions)
71 {
72 bool clientNoContextTakeover = perMessageDeflateOptions.getClientNoContextTakeover();
73
74 uint8_t deflateBits = perMessageDeflateOptions.getClientMaxWindowBits();
75 uint8_t inflateBits = perMessageDeflateOptions.getServerMaxWindowBits();
76
77 return _compressor->init(deflateBits, clientNoContextTakeover) &&
78 _decompressor->init(inflateBits, clientNoContextTakeover);
79 }
80
81 bool WebSocketPerMessageDeflate::compress(const IXWebSocketSendData& in, std::string& out)
82 {
83 return _compressor->compress(in, out);
84 }
85
86 bool WebSocketPerMessageDeflate::compress(const std::string& in, std::string& out)
87 {
88 return _compressor->compress(in, out);
89 }
90
91 bool WebSocketPerMessageDeflate::decompress(const std::string& in, std::string& out)
92 {
93 return _decompressor->decompress(in, out);
94 }
95
96} // namespace ix
97