WebSocketSession: Add challenge var

src/WebSocketSession.cpp

@@ -13,6 +13,19 @@ WebSocketSession::WebSocketSession() :
 {
 }
 
+std::string WebSocketSession::Challenge()
+{
+	std::lock_guard<std::mutex> lock(challengeMutex);
+	std::string ret(challenge);
+	return ret;
+}
+
+void WebSocketSession::SetChallenge(std::string challengeString)
+{
+	std::lock_guard<std::mutex> lock(challengeMutex);
+	challenge = challengeString;
+}
+
 uint64_t WebSocketSession::IncomingMessages()
 {
 	return incomingMessages.load();

src/WebSocketSession.h

@@ -1,5 +1,7 @@
 #pragma once
 
+#include <mutex>
+#include <string>
 #include <atomic>
 
 class WebSocketSession
@@ -7,6 +9,9 @@ class WebSocketSession
 	public:
 		WebSocketSession();
 
+		std::string Challenge();
+		void SetChallenge(std::string challenge);
+
 		uint64_t IncomingMessages();
 		void IncrementIncomingMessages();
 
@@ -29,6 +34,8 @@ class WebSocketSession
 		void SetEventSubscriptions(uint64_t subscriptions);
 
 	private:
+		std::mutex challengeMutex;
+		std::string challenge;
 		std::atomic<uint64_t> incomingMessages;
 		std::atomic<uint64_t> outgoingMessages;
 		std::atomic<uint8_t> rpcVersion;