pci: insert assert that auto-assigned-address function is single function device.

[qemu.git] / qemu-malloc.c

diff --git a/qemu-malloc.c b/qemu-malloc.c
index 3bffae1fbb0384384f5391e13766b8b9e1623afb..36b0b3641ee56a13a03f5e0479f945a9951e8843 100644 (file)
--- a/qemu-malloc.c
+++ b/qemu-malloc.c
@@ -22,10 +22,14 @@
  * THE SOFTWARE.
  */
 #include "qemu-common.h"
+#include <stdlib.h>
 
-void *get_mmap_addr(unsigned long size)
+static void *oom_check(void *ptr)
 {
-    return NULL;
+    if (ptr == NULL) {
+        abort();
+    }
+    return ptr;
 }
 
 void qemu_free(void *ptr)
@@ -33,24 +37,37 @@ void qemu_free(void *ptr)
     free(ptr);
 }
 
+static int allow_zero_malloc(void)
+{
+#if defined(CONFIG_ZERO_MALLOC)
+    return 1;
+#else
+    return 0;
+#endif
+}
+
 void *qemu_malloc(size_t size)
 {
-    return malloc(size);
+    if (!size && !allow_zero_malloc()) {
+        abort();
+    }
+    return oom_check(malloc(size ? size : 1));
 }
 
 void *qemu_realloc(void *ptr, size_t size)
 {
-    return realloc(ptr, size);
+    if (!size && !allow_zero_malloc()) {
+        abort();
+    }
+    return oom_check(realloc(ptr, size ? size : 1));
 }
 
 void *qemu_mallocz(size_t size)
 {
-    void *ptr;
-    ptr = qemu_malloc(size);
-    if (!ptr)
-        return NULL;
-    memset(ptr, 0, size);
-    return ptr;
+    if (!size && !allow_zero_malloc()) {
+        abort();
+    }
+    return oom_check(calloc(1, size ? size : 1));
 }
 
 char *qemu_strdup(const char *str)
@@ -58,8 +75,21 @@ char *qemu_strdup(const char *str)
     char *ptr;
     size_t len = strlen(str);
     ptr = qemu_malloc(len + 1);
-    if (!ptr)
-        return NULL;
-    pstrcpy(ptr, len + 1, str);
+    memcpy(ptr, str, len + 1);
     return ptr;
 }
+
+char *qemu_strndup(const char *str, size_t size)
+{
+    const char *end = memchr(str, 0, size);
+    char *new;
+
+    if (end) {
+        size = end - str;
+    }
+
+    new = qemu_malloc(size + 1);
+    new[size] = 0;
+
+    return memcpy(new, str, size);
+}