Websockets

apps/RenderCadenceCompositor/control/HttpControlServer.h

@@ -9,8 +9,10 @@
 #include <filesystem>
 #include <functional>
 #include <map>
+#include <mutex>
 #include <string>
 #include <thread>
+#include <vector>
 
 namespace RenderCadenceCompositor
 {
@@ -89,11 +91,15 @@ public:
 	HttpResponse RouteRequestForTest(const HttpRequest& request) const;
 
 	static bool ParseHttpRequest(const std::string& rawRequest, HttpRequest& request);
+	static std::string WebSocketAcceptKey(const std::string& clientKey);
 
 private:
 	void ThreadMain();
 	bool TryAcceptClient();
 	bool HandleClient(UniqueSocket clientSocket);
+	bool HandleWebSocketClient(UniqueSocket clientSocket, const HttpRequest& request);
+	void WebSocketClientMain(UniqueSocket clientSocket);
+	void JoinFinishedClientThreads();
 	bool SendResponse(SOCKET clientSocket, const HttpResponse& response) const;
 	HttpResponse RouteRequest(const HttpRequest& request) const;
 	HttpResponse ServeGet(const HttpRequest& request) const;
@@ -107,6 +113,7 @@ private:
 	static HttpResponse TextResponse(const std::string& status, const std::string& body);
 	static HttpResponse HtmlResponse(const std::string& status, const std::string& body);
 	static std::string ActionResponse(bool ok, const std::string& error = std::string());
+	static bool SendWebSocketText(SOCKET clientSocket, const std::string& text);
 	static std::string GuessContentType(const std::filesystem::path& path);
 	static bool IsSafeRelativePath(const std::filesystem::path& path);
 	static std::string ToLower(std::string text);
@@ -117,6 +124,9 @@ private:
 	HttpControlServerCallbacks mCallbacks;
 	UniqueSocket mListenSocket;
 	std::thread mThread;
+	std::mutex mClientThreadsMutex;
+	std::vector<std::thread> mClientThreads;
+	std::vector<std::thread> mFinishedClientThreads;
 	std::atomic<bool> mRunning{ false };
 	unsigned short mPort = 0;
 	bool mWinsockStarted = false;