Add HandleInventoryCheck

CubeModLoader/DLL.h

@@ -17,6 +17,7 @@ class DLL
         // Callbacks
         FARPROC HandleChat;
         FARPROC HandleP2PRequest;
+        FARPROC HandleInventoryCheck;
 
         DLL(std::string fileName);
         HMODULE Load();

CubeModLoader/callbacks/CheckInventoryFullHandler.h

@@ -0,0 +1,60 @@
+int CheckInventoryFullHandler(void* player, void* item) {
+    for (DLL* dll: modDLLs) {
+        if (dll->HandleInventoryCheck) {
+            if ( int result = ((int(*)(void*, void*))dll->HandleInventoryCheck)(player, item)  ){
+                return result;
+            }
+        }
+    }
+    return 0;
+}
+void* CheckInventoryFullHandler_ptr = (void*)&CheckInventoryFullHandler;
+
+void* ASMCheckInventoryFullHandler_jmpback;
+void* ASMCheckInventoryFullHandler_retn;
+void no_optimize ASMCheckInventoryFullHandler() {
+    asm(PUSH_ALL
+
+        PREPARE_STACK
+        "sub rsp, 0x0F \n"
+        "call [CheckInventoryFullHandler_ptr] \n"
+        "add rsp, 0x0F \n"
+
+        RESTORE_STACK
+
+        // Did the handler return 1? true
+        "cmp eax, 1 \n"
+        "je 1f \n"
+
+        // Did the handler return 2? false
+        "cmp eax, 2 \n"
+        "je 2f \n"
+
+        // Otherwise? Do nothing
+        POP_ALL
+
+        // original code
+        "mov     [rsp+0x20], rbp \n"
+        "push    r12 \n"
+        "push    r14 \n"
+        "push    r15 \n"
+        "sub     rsp, 0x20 \n"
+        "jmp [ASMCheckInventoryFullHandler_jmpback] \n"
+
+
+        "1: \n"  //full
+        POP_ALL
+        "xor al,al \n"
+        "jmp [ASMCheckInventoryFullHandler_retn] \n"
+
+        "2: \n" //not
+        POP_ALL
+        "mov al,1 \n"
+        "jmp [ASMCheckInventoryFullHandler_retn] \n"
+       );
+}
+void SetupCheckInventoryFullHandler() {
+    WriteFarJMP(base+0x50670, (void*)&ASMCheckInventoryFullHandler);
+    ASMCheckInventoryFullHandler_jmpback = (void*)base+0x5067F;
+    ASMCheckInventoryFullHandler_retn = (void*)base+0x507A0;
+}

CubeModLoader/main.cpp

@@ -51,10 +51,12 @@ void WriteFarJMP(void* source, void* destination) {
 
 #include "callbacks/ChatHandler.h"
 #include "callbacks/P2PRequestHandler.h"
+#include "callbacks/CheckInventoryFullHandler.h"
 
 void SetupHandlers() {
     SetupChatHandler();
     SetupP2PRequestHandler();
+    SetupCheckInventoryFullHandler();
 }
 
 void Popup(const char* title, char* msg ){
@@ -93,6 +95,7 @@ extern "C" __declspec(dllexport) BOOL APIENTRY DllMain(HINSTANCE hinstDLL, DWORD
             IMPORT(dll, ModInitialize);
             IMPORT(dll, HandleChat);
             IMPORT(dll, HandleP2PRequest);
+            IMPORT(dll, HandleInventoryCheck);
         }
 
         // Ensure version compatibility