diff --git a/src/net/connection/NetClient.cpp b/src/net/connection/NetClient.cpp index 81e24cf..f8c148b 100644 --- a/src/net/connection/NetClient.cpp +++ b/src/net/connection/NetClient.cpp @@ -381,6 +381,10 @@ void NetClient::SetMessageHandler(NETMESSAGE msgId, MESSAGE_HANDLER handler, voi this->m_handlerParams[msgId] = param; } +void NetClient::SetObjMgr(ClntObjMgr* objMgr) { + this->m_objMgr = objMgr; +} + void NetClient::WCCantConnect(WowConnection* conn, uint32_t timeStamp, NETCONNADDR* addr) { if (conn == this->m_redirectConnection) { // TODO diff --git a/src/net/connection/NetClient.hpp b/src/net/connection/NetClient.hpp index f9f0458..28dcc65 100644 --- a/src/net/connection/NetClient.hpp +++ b/src/net/connection/NetClient.hpp @@ -10,6 +10,7 @@ #include class CDataStore; +class ClntObjMgr; class NetClient; class WowConnection; @@ -80,6 +81,7 @@ class NetClient : public WowConnectionResponse { void SetDelete(); void SetLoginData(LoginData* loginData); void SetMessageHandler(NETMESSAGE msgId, MESSAGE_HANDLER handler, void* param); + void SetObjMgr(ClntObjMgr* objMgr); private: // Static variables @@ -105,6 +107,7 @@ class NetClient : public WowConnectionResponse { uint32_t m_bytesReceived = 0; uint32_t m_connectedTimestamp = 0; SCritSect m_pingLock; + ClntObjMgr* m_objMgr = nullptr; }; #endif