From 8e03d9e5dde1746fba43cfe33be980fcd8303b21 Mon Sep 17 00:00:00 2001
From: fallenoak <git@fallenoak.me>
Date: Fri, 24 Feb 2023 19:41:07 -0600
Subject: [PATCH] feat(net): implement message ready callback in NetClient

---
 src/event/Types.hpp              |  2 +-
 src/net/connection/NetClient.cpp | 41 ++++++++++++++++++++++++++++++--
 src/net/connection/NetClient.hpp |  5 +++-
 3 files changed, 44 insertions(+), 4 deletions(-)

diff --git a/src/event/Types.hpp b/src/event/Types.hpp
index b042c6f..447b520 100644
--- a/src/event/Types.hpp
+++ b/src/event/Types.hpp
@@ -31,7 +31,7 @@ enum EVENTID {
     EVENT_ID_21                 = 21,
     EVENT_ID_22                 = 22,
     EVENT_ID_PAINT              = 23,
-    EVENT_ID_24                 = 24,
+    EVENT_ID_NET_DATA           = 24,
     EVENT_ID_NET_CONNECT        = 25,
     EVENT_ID_26                 = 26,
     EVENT_ID_27                 = 27,
diff --git a/src/net/connection/NetClient.cpp b/src/net/connection/NetClient.cpp
index df1b9fe..62f50c1 100644
--- a/src/net/connection/NetClient.cpp
+++ b/src/net/connection/NetClient.cpp
@@ -3,6 +3,7 @@
 #include <cstdlib>
 #include <cstring>
 #include <new>
+#include <common/DataStore.hpp>
 #include <common/Prop.hpp>
 #include <common/Time.hpp>
 #include <storm/Error.hpp>
@@ -22,6 +23,10 @@ void NETEVENTQUEUE::AddEvent(EVENTID eventId, void* conn, NetClient* client, con
     // TODO
 }
 
+void NetClient::AuthChallengeHandler(WowConnection* conn, CDataStore* msg) {
+    // TODO
+}
+
 void NetClient::Connect(const char* addrStr) {
     if (this->m_netState != NS_INITIALIZED) {
         SErrDisplayAppFatal("Expected (m_netState == NS_INITIALIZED), got %d", this->m_netState);
@@ -85,6 +90,10 @@ int32_t NetClient::Initialize() {
     return 1;
 }
 
+void NetClient::PongHandler(WowConnection* conn, CDataStore* msg) {
+    // TODO
+}
+
 void NetClient::SetLoginData(LoginData* loginData) {
     memcpy(&this->m_loginData, loginData, sizeof(this->m_loginData));
 }
@@ -120,6 +129,34 @@ void NetClient::WCDisconnected(WowConnection* conn, uint32_t timeStamp, NETCONNA
     // TODO
 }
 
-void NetClient::WCMessageReady(WowConnection *conn, uint32_t timeStamp, CDataStore* msg) {
-    // TODO
+void NetClient::WCMessageReady(WowConnection* conn, uint32_t timeStamp, CDataStore* msg) {
+    uint8_t* data;
+    msg->GetDataInSitu(reinterpret_cast<void*&>(data), msg->m_size);
+
+    // TODO increment byte counter
+    // SInterlockedExchangeAdd(this->m_bytesReceived, msg->m_size);
+
+    msg->m_read = 0;
+
+    uint16_t msgId;
+    msg->Get(msgId);
+
+    // TODO SMSG_SUSPEND_COMMS (0x50F)
+    // TODO SMSG_FORCE_SEND_QUEUED_PACKETS (0x511)
+    // TODO SMSG_REDIRECT_CLIENT (0x50D)
+
+    if (msgId == SMSG_PONG) {
+        this->PongHandler(conn, msg);
+        return;
+    } else if (msgId == SMSG_AUTH_CHALLENGE) {
+        this->AuthChallengeHandler(conn, msg);
+        return;
+    }
+
+    if (conn == this->m_serverConnection && !this->m_suspended) {
+        msg->m_read = msg->m_size;
+        this->m_netEventQueue->AddEvent(EVENT_ID_NET_DATA, conn, this, data, msg->m_size);
+    } else {
+        conn->Disconnect();
+    }
 }
diff --git a/src/net/connection/NetClient.hpp b/src/net/connection/NetClient.hpp
index 05948ab..42dec31 100644
--- a/src/net/connection/NetClient.hpp
+++ b/src/net/connection/NetClient.hpp
@@ -40,6 +40,7 @@ class NetClient : public WowConnectionResponse {
         // Member variables
         LoginData m_loginData;
         NETSTATE m_netState = NS_UNINITIALIZED;
+        bool m_suspended = false;
         MESSAGE_HANDLER m_handlers[NUM_MSG_TYPES];
         void* m_handlerParams[NUM_MSG_TYPES];
         NETEVENTQUEUE* m_netEventQueue = nullptr;
@@ -52,15 +53,17 @@ class NetClient : public WowConnectionResponse {
         SCritSect m_pingLock;
 
         // Virtual member functions
-        virtual void WCMessageReady(WowConnection *conn, uint32_t timeStamp, CDataStore* msg);
+        virtual void WCMessageReady(WowConnection* conn, uint32_t timeStamp, CDataStore* msg);
         virtual void WCConnected(WowConnection* conn, WowConnection* inbound, uint32_t timeStamp, const NETCONNADDR* addr);
         virtual void WCCantConnect(WowConnection* conn, uint32_t timeStamp, NETCONNADDR* addr);
         virtual void WCDisconnected(WowConnection* conn, uint32_t timeStamp, NETCONNADDR* addr);
 
         // Member functions
+        void AuthChallengeHandler(WowConnection* conn, CDataStore* msg);
         void Connect(const char* addrStr);
         int32_t ConnectInternal(const char* host, uint16_t port);
         int32_t Initialize();
+        void PongHandler(WowConnection* conn, CDataStore* msg);
         void SetLoginData(LoginData* loginData);
         void SetMessageHandler(NETMESSAGE msgId, MESSAGE_HANDLER handler, void* param);
 };