1 files changed, 137 insertions, 1 deletions
diff --git a/protocols/Steam/src/steam_server.cpp b/protocols/Steam/src/steam_server.cpp
index f8c206c0e0..24596894bf 100644
--- a/protocols/Steam/src/steam_server.cpp
+++ b/protocols/Steam/src/steam_server.cpp
@@ -21,10 +21,146 @@ void __cdecl CSteamProto::ServerThread(void *)
 {
 	// load web socket servers first if needed
 	int iTimeDiff = db_get_dw(0, STEAM_MODULE, DBKEY_HOSTS_DATE);
-	if (!db_get_dw(0, STEAM_MODULE, DBKEY_HOSTS_COUNT) || time(0) - iTimeDiff > 3600 * 24 * 7) { // once a week
+	int iHostCount = db_get_dw(0, STEAM_MODULE, DBKEY_HOSTS_COUNT);
+	if (!iHostCount || time(0) - iTimeDiff > 3600 * 24 * 7) { // once a week
 		if (!SendRequest(new GetHostsRequest(), &CSteamProto::OnGotHosts)) {
 			LoginFailed();
 			return;
 		}
 	}
+
+	srand(time(0));
+	m_hServerConn = nullptr;
+
+	CMStringA szHost;
+	do {
+		szHost.Format("Host%d", rand() % iHostCount);
+		szHost = db_get_sm(0, STEAM_MODULE, szHost);
+		szHost.Insert(0, "wss://");
+		szHost += "/cmsocket/";
+	}
+	while (ServerThreadStub(szHost));
+}
+
+bool CSteamProto::ServerThreadStub(const char *szHost)
+{
+	NLHR_PTR pReply(WebSocket_Connect(m_hNetlibUser, szHost));
+	if (pReply == nullptr) {
+		debugLogA("websocket connection failed");
+		return false;
+	}
+
+	if (pReply->resultCode != 101) {
+		debugLogA("websocket connection failed: %d", pReply->resultCode);
+		return false;
+	}
+
+	m_hServerConn = pReply->nlc;
+	debugLogA("Websocket connection succeeded");
+
+	bool bExit = false;
+	int offset = 0;
+	MBinBuffer netbuf;
+
+	while (!bExit) {
+		if (m_bTerminated)
+			break;
+
+		unsigned char buf[2048];
+		int bufSize = Netlib_Recv(m_hServerConn, (char *)buf + offset, _countof(buf) - offset, MSG_NODUMP);
+		if (bufSize == 0) {
+			debugLogA("Gateway connection gracefully closed");
+			bExit = !m_bTerminated;
+			break;
+		}
+		if (bufSize < 0) {
+			debugLogA("Gateway connection error, exiting");
+			break;
+		}
+
+		WSHeader hdr;
+		if (!WebSocket_InitHeader(hdr, buf, bufSize)) {
+			offset += bufSize;
+			continue;
+		}
+		offset = 0;
+
+		debugLogA("Got packet: buffer = %d, opcode = %d, headerSize = %d, final = %d, masked = %d", bufSize, hdr.opCode, hdr.headerSize, hdr.bIsFinal, hdr.bIsMasked);
+
+		// we have some additional data, not only opcode
+		if ((size_t)bufSize > hdr.headerSize) {
+			size_t currPacketSize = bufSize - hdr.headerSize;
+			netbuf.append(buf, bufSize);
+			while (currPacketSize < hdr.payloadSize) {
+				int result = Netlib_Recv(m_hServerConn, (char *)buf, _countof(buf), MSG_NODUMP);
+				if (result == 0) {
+					debugLogA("Gateway connection gracefully closed");
+					bExit = !m_bTerminated;
+					break;
+				}
+				if (result < 0) {
+					debugLogA("Gateway connection error, exiting");
+					break;
+				}
+				currPacketSize += result;
+				netbuf.append(buf, result);
+			}
+		}
+
+		// read all payloads from the current buffer, one by one
+		size_t prevSize = 0;
+		while (true) {
+			switch (hdr.opCode) {
+			case 0: // text packet
+			case 1: // binary packet
+			case 2: // continuation
+				if (hdr.bIsFinal) {
+					// process a packet here
+					CMStringA szJson((char *)netbuf.data() + hdr.headerSize, (int)hdr.payloadSize);
+					debugLogA("JSON received:\n%s", szJson.c_str());
+					JSONNode root = JSONNode::parse(szJson);
+	//				if (root)
+//						bExit = ProcessMessage(root);
+				}
+				break;
+
+			case 8: // close
+				debugLogA("server required to exit");
+				bExit = true; // simply reconnect, don't exit
+				break;
+
+			case 9: // ping
+				debugLogA("ping received");
+				Netlib_Send(m_hServerConn, (char *)buf + hdr.headerSize, bufSize - int(hdr.headerSize), 0);
+				break;
+			}
+
+			if (hdr.bIsFinal)
+				netbuf.remove(hdr.headerSize + hdr.payloadSize);
+
+			if (netbuf.length() == 0)
+				break;
+
+			// if we have not enough data for header, continue reading
+			if (!WebSocket_InitHeader(hdr, netbuf.data(), netbuf.length()))
+				break;
+
+			// if we have not enough data for data, continue reading
+			if (hdr.headerSize + hdr.payloadSize > netbuf.length())
+				break;
+
+			debugLogA("Got inner packet: buffer = %d, opcode = %d, headerSize = %d, payloadSize = %d, final = %d, masked = %d", netbuf.length(), hdr.opCode, hdr.headerSize, hdr.payloadSize, hdr.bIsFinal, hdr.bIsMasked);
+			if (prevSize == netbuf.length()) {
+				netbuf.remove(prevSize);
+				debugLogA("dropping current packet, exiting");
+				break;
+			}
+
+			prevSize = netbuf.length();
+		}
+	}
+
+	Netlib_CloseHandle(m_hServerConn);
+	m_hServerConn = nullptr;
+	return bExit;
 }