From d4c2137eaad3cdbcada9aa33fcd713dc2c711458 Mon Sep 17 00:00:00 2001
From: Francesco Cozzuto <francesco.cozzuto@gmail.com>
Date: Sat, 24 Jan 2026 12:12:48 +0100
Subject: [PATCH] Add allocation failure injection

---
 quakey/src/quakey.c | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/quakey/src/quakey.c b/quakey/src/quakey.c
index ac39b19..c57682e 100644
--- a/quakey/src/quakey.c
+++ b/quakey/src/quakey.c
@@ -4335,15 +4335,32 @@ BOOL mock_MoveFileExW(WCHAR *lpExistingFileName, WCHAR *lpNewFileName, DWORD dwF
 
 void *mock_malloc(size_t size)
 {
+    Host *host = host___;
+    if (host == NULL)
+        abort_("Call to mock_malloc() with no node scheduled\n");
+#ifdef FAULT_INJECTION
+    if ((sim_random(host->sim) % 1000) == 0)
+        return NULL;
+#endif
     return malloc(size);
 }
 
 void *mock_realloc(void *ptr, size_t size)
 {
+    Host *host = host___;
+    if (host == NULL)
+        abort_("Call to mock_realloc() with no node scheduled\n");
+#ifdef FAULT_INJECTION
+    if ((sim_random(host->sim) % 1000) == 0)
+        return NULL;
+#endif
     return realloc(ptr, size);
 }
 
 void mock_free(void *ptr)
 {
+    Host *host = host___;
+    if (host == NULL)
+        abort_("Call to mock_free() with no node scheduled\n");
     free(ptr);
 }