Implement sPAPR Virtual LAN (ibmveth)

[qemu.git] / qemu-malloc.c

diff --git a/qemu-malloc.c b/qemu-malloc.c
index e9e49cb8f2bdd2e6246f5f1acbc816ccdd7552f8..b9b38514ac1ebb0cbb24d208da0c3c15d021c6a7 100644 (file)
--- a/qemu-malloc.c
+++ b/qemu-malloc.c
@@ -22,40 +22,54 @@
  * THE SOFTWARE.
  */
 #include "qemu-common.h"
+#include "trace.h"
 #include <stdlib.h>
 
-static void *oom_check(void *ptr)
-{
-    if (ptr == NULL)
-        abort();
-    return ptr;
-}
-
-void *get_mmap_addr(unsigned long size)
+void qemu_free(void *ptr)
 {
-    return NULL;
+    trace_qemu_free(ptr);
+    free(ptr);
 }
 
-void qemu_free(void *ptr)
+static int allow_zero_malloc(void)
 {
-    free(ptr);
+#if defined(CONFIG_ZERO_MALLOC)
+    return 1;
+#else
+    return 0;
+#endif
 }
 
 void *qemu_malloc(size_t size)
 {
-    return oom_check(malloc(size));
+    void *ptr;
+    if (!size && !allow_zero_malloc()) {
+        abort();
+    }
+    ptr = qemu_oom_check(malloc(size ? size : 1));
+    trace_qemu_malloc(size, ptr);
+    return ptr;
 }
 
 void *qemu_realloc(void *ptr, size_t size)
 {
-    return oom_check(realloc(ptr, size));
+    void *newptr;
+    if (!size && !allow_zero_malloc()) {
+        abort();
+    }
+    newptr = qemu_oom_check(realloc(ptr, size ? size : 1));
+    trace_qemu_realloc(ptr, size, newptr);
+    return newptr;
 }
 
 void *qemu_mallocz(size_t size)
 {
     void *ptr;
-    ptr = qemu_malloc(size);
-    memset(ptr, 0, size);
+    if (!size && !allow_zero_malloc()) {
+        abort();
+    }
+    ptr = qemu_oom_check(calloc(1, size ? size : 1));
+    trace_qemu_malloc(size, ptr);
     return ptr;
 }
 
@@ -73,8 +87,9 @@ char *qemu_strndup(const char *str, size_t size)
     const char *end = memchr(str, 0, size);
     char *new;
 
-    if (end)
+    if (end) {
         size = end - str;
+    }
 
     new = qemu_malloc(size + 1);
     new[size] = 0;