2 files changed, 4 insertions(+), 0 deletions(-)

M Local/NosSmooth.LocalCore/Network.cpp
M Local/NosSmooth.LocalCore/Network.h

@@ 31,11 31,13 @@ void Network::ReceivePacket(System::String^ packet)
 void Network::SetReceiveCallback(NetworkCallback^ callback)
 {
     IntPtr functionPointer = Marshal::GetFunctionPointerForDelegate(callback);
+    _receiveCallback = callback;
     NetworkUnmanaged::GetInstance()->SetReceiveCallback(static_cast<PacketCallback>(functionPointer.ToPointer()));
 }
 
 void Network::SetSendCallback(NetworkCallback^ callback)
 {
     IntPtr functionPointer = Marshal::GetFunctionPointerForDelegate(callback);
+    _sendCallback = callback;
     NetworkUnmanaged::GetInstance()->SetSendCallback(static_cast<PacketCallback>(functionPointer.ToPointer()));
 }=
\ No newline at end of file

@@ 39,6 39,8 @@ namespace NosSmoothCore
 		void ResetHooks();
 	private:
 		NetworkUnmanaged* _networkUnmanaged;
+		NetworkCallback^ _sendCallback;
+		NetworkCallback^ _receiveCallback;
 	};
 }