Bemanitools v5.26 release

src/main/util/mem.c

@@ -0,0 +1,89 @@
+#define LOG_MODULE "mem"
+
+#include <windows.h>
+
+#include <stddef.h>
+#include <stdlib.h>
+#include <string.h>
+#include <stdbool.h>
+
+#include "util/log.h"
+
+void *xcalloc(size_t nbytes)
+{
+    void *mem;
+
+    mem = calloc(nbytes, 1);
+
+    if (mem == NULL) {
+        log_fatal("xmalloc(%u) failed", (uint32_t) nbytes);
+
+        return NULL;
+    }
+
+    return mem;
+}
+
+void *xmalloc(size_t nbytes)
+{
+    void *mem;
+
+    mem = malloc(nbytes);
+
+    if (mem == NULL) {
+        log_fatal("xcalloc(%u) failed", (uint32_t) nbytes);
+
+        return NULL;
+    }
+
+    return mem;
+}
+
+void *xrealloc(void *mem, size_t nbytes)
+{
+    void *newmem;
+
+    newmem = realloc(mem, nbytes);
+
+    if (newmem == NULL) {
+        log_fatal("xrealloc(%p, %u) failed", mem, (uint32_t) nbytes);
+
+        return NULL;
+    }
+
+    return newmem;
+}
+
+bool mem_nop(size_t mem_offset, size_t length)
+{
+    DWORD oldProt;
+
+    if (VirtualProtect((void*) (mem_offset), length, PAGE_EXECUTE_READWRITE,
+        &oldProt) == 0) {
+        return false;
+    }
+
+    for (size_t i = 0; i < length; i++) {
+        ((uint8_t*) mem_offset)[i] = 0x90;
+    }
+
+    if (VirtualProtect((void*) (mem_offset), length, oldProt, &oldProt) == 0) {
+        return false;
+    }
+
+    return true;
+}
+
+void* mem_find_signiture(const uint8_t* sig, uint32_t sig_len,
+    	int32_t sig_offset, void* start_addr, void* end_addr, int32_t alignment)
+{
+	size_t pos;
+
+	for (pos = (size_t) start_addr; pos < (size_t) end_addr; pos += alignment) {
+		if (!memcmp((void*) pos, sig, sig_len)) {
+			return (void*) (pos + sig_offset);
+		}
+	}
+
+	return NULL;
+}