Merge commit 'c992cb4e42cc223f67ede0e48d7ff3f4947af0c6' as 'test/compatibility/C/uWebSockets'

test/compatibility/C/uWebSockets/src/WebSocketData.h

@@ -0,0 +1,70 @@
+/*
+ * Authored by Alex Hultman, 2018-2019.
+ * Intellectual property of third-party.
+
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+
+ *     http://www.apache.org/licenses/LICENSE-2.0
+
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#ifndef UWS_WEBSOCKETDATA_H
+#define UWS_WEBSOCKETDATA_H
+
+#include "WebSocketProtocol.h"
+#include "AsyncSocketData.h"
+#include "PerMessageDeflate.h"
+
+#include <string>
+
+namespace uWS {
+
+struct WebSocketData : AsyncSocketData<false>, WebSocketState<true> {
+    template <bool, bool> friend struct WebSocketContext;
+    template <bool, bool> friend struct WebSocket;
+private:
+    std::string fragmentBuffer;
+    int controlTipLength = 0;
+    bool isShuttingDown = 0;
+    enum CompressionStatus : char {
+        DISABLED,
+        ENABLED,
+        COMPRESSED_FRAME
+    } compressionStatus;
+
+    /* We might have a dedicated compressor */
+    DeflationStream *deflationStream = nullptr;
+
+    /* We could be a subscriber */
+    Subscriber *subscriber = nullptr;
+public:
+    WebSocketData(bool perMessageDeflate, bool slidingCompression, std::string &&backpressure) : AsyncSocketData<false>(std::move(backpressure)), WebSocketState<true>() {
+        compressionStatus = perMessageDeflate ? ENABLED : DISABLED;
+
+        /* Initialize the dedicated sliding window */
+        if (perMessageDeflate && slidingCompression) {
+            deflationStream = new DeflationStream;
+        }
+    }
+
+    ~WebSocketData() {
+        if (deflationStream) {
+            delete deflationStream;
+        }
+
+        if (subscriber) {
+            delete subscriber;
+        }
+    }
+};
+
+}
+
+#endif // UWS_WEBSOCKETDATA_H