qemu-malloc.c

   1 /*
   2  * malloc-like functions for system emulation.
   3  *
   4  * Copyright (c) 2006 Fabrice Bellard
   5  *
   6  * Permission is hereby granted, free of charge, to any person obtaining a copy
   7  * of this software and associated documentation files (the "Software"), to deal
   8  * in the Software without restriction, including without limitation the rights
   9  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  10  * copies of the Software, and to permit persons to whom the Software is
  11  * furnished to do so, subject to the following conditions:
  12  *
  13  * The above copyright notice and this permission notice shall be included in
  14  * all copies or substantial portions of the Software.
  15  *
  16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  17  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  18  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
  19  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  20  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  21  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  22  * THE SOFTWARE.
  23  */
  24 #include "qemu-common.h"
  25 #include "trace.h"
  26 #include <stdlib.h>
  27
  28 static void *oom_check(void *ptr)
  29 {
  30     if (ptr == NULL) {
  31         abort();
  32     }
  33     return ptr;
  34 }
  35
  36 void qemu_free(void *ptr)
  37 {
  38     trace_qemu_free(ptr);
  39     free(ptr);
  40 }
  41
  42 static int allow_zero_malloc(void)
  43 {
  44 #if defined(CONFIG_ZERO_MALLOC)
  45     return 1;
  46 #else
  47     return 0;
  48 #endif
  49 }
  50
  51 void *qemu_malloc(size_t size)
  52 {
  53     void *ptr;
  54     if (!size && !allow_zero_malloc()) {
  55         abort();
  56     }
  57     ptr = oom_check(malloc(size ? size : 1));
  58     trace_qemu_malloc(size, ptr);
  59     return ptr;
  60 }
  61
  62 void *qemu_realloc(void *ptr, size_t size)
  63 {
  64     void *newptr;
  65     if (!size && !allow_zero_malloc()) {
  66         abort();
  67     }
  68     newptr = oom_check(realloc(ptr, size ? size : 1));
  69     trace_qemu_realloc(ptr, size, newptr);
  70     return newptr;
  71 }
  72
  73 void *qemu_mallocz(size_t size)
  74 {
  75     if (!size && !allow_zero_malloc()) {
  76         abort();
  77     }
  78     return oom_check(calloc(1, size ? size : 1));
  79 }
  80
  81 char *qemu_strdup(const char *str)
  82 {
  83     char *ptr;
  84     size_t len = strlen(str);
  85     ptr = qemu_malloc(len + 1);
  86     memcpy(ptr, str, len + 1);
  87     return ptr;
  88 }
  89
  90 char *qemu_strndup(const char *str, size_t size)
  91 {
  92     const char *end = memchr(str, 0, size);
  93     char *new;
  94
  95     if (end) {
  96         size = end - str;
  97     }
  98
  99     new = qemu_malloc(size + 1);
 100     new[size] = 0;
 101
 102     return memcpy(new, str, size);
 103 }