memory/build/mozmemory_wrap.c

   1 /* This Source Code Form is subject to the terms of the Mozilla Public
   2  * License, v. 2.0. If a copy of the MPL was not distributed with this file,
   3  * You can obtain one at http://mozilla.org/MPL/2.0/. */
   4
   5 #include <string.h>
   6 #include "mozmemory_wrap.h"
   7 #include "mozilla/Types.h"
   8
   9 /* Declare malloc implementation functions with the right return and
  10  * argument types. */
  11 #define MALLOC_DECL(name, return_type, ...) \
  12   MOZ_MEMORY_API return_type name ## _impl(__VA_ARGS__);
  13 #include "malloc_decls.h"
  14
  15 #ifdef MOZ_WRAP_NEW_DELETE
  16 /* operator new(unsigned int) */
  17 MOZ_MEMORY_API void *
  18 mozmem_malloc_impl(_Znwj)(unsigned int size)
  19 {
  20   return malloc_impl(size);
  21 }
  22 /* operator new[](unsigned int) */
  23 MOZ_MEMORY_API void *
  24 mozmem_malloc_impl(_Znaj)(unsigned int size)
  25 {
  26   return malloc_impl(size);
  27 }
  28 /* operator delete(void*) */
  29 MOZ_MEMORY_API void
  30 mozmem_malloc_impl(_ZdlPv)(void *ptr)
  31 {
  32   free_impl(ptr);
  33 }
  34 /* operator delete[](void*) */
  35 MOZ_MEMORY_API void
  36 mozmem_malloc_impl(_ZdaPv)(void *ptr)
  37 {
  38   free_impl(ptr);
  39 }
  40 /*operator new(unsigned int, std::nothrow_t const&)*/
  41 MOZ_MEMORY_API void *
  42 mozmem_malloc_impl(_ZnwjRKSt9nothrow_t)(unsigned int size)
  43 {
  44   return malloc_impl(size);
  45 }
  46 /*operator new[](unsigned int, std::nothrow_t const&)*/
  47 MOZ_MEMORY_API void *
  48 mozmem_malloc_impl(_ZnajRKSt9nothrow_t)(unsigned int size)
  49 {
  50   return malloc_impl(size);
  51 }
  52 /* operator delete(void*, std::nothrow_t const&) */
  53 MOZ_MEMORY_API void
  54 mozmem_malloc_impl(_ZdlPvRKSt9nothrow_t)(void *ptr)
  55 {
  56   free_impl(ptr);
  57 }
  58 /* operator delete[](void*, std::nothrow_t const&) */
  59 MOZ_MEMORY_API void
  60 mozmem_malloc_impl(_ZdaPvRKSt9nothrow_t)(void *ptr)
  61 {
  62   free_impl(ptr);
  63 }
  64 #endif
  65
  66 /* strndup and strdup may be defined as macros in string.h, which would
  67  * clash with the definitions below. */
  68 #undef strndup
  69 #undef strdup
  70
  71 #ifndef XP_DARWIN
  72 MOZ_MEMORY_API char *
  73 strndup_impl(const char *src, size_t len)
  74 {
  75   char* dst = (char*) malloc_impl(len + 1);
  76   if (dst) {
  77     strncpy(dst, src, len);
  78     dst[len] = '\0';
  79   }
  80   return dst;
  81 }
  82
  83 MOZ_MEMORY_API char *
  84 strdup_impl(const char *src)
  85 {
  86   size_t len = strlen(src);
  87   return strndup_impl(src, len);
  88 }
  89 #endif /* XP_DARWIN */
  90
  91 #ifdef ANDROID
  92 #include <stdarg.h>
  93 #include <stdio.h>
  94
  95 MOZ_MEMORY_API int
  96 vasprintf_impl(char **str, const char *fmt, va_list ap)
  97 {
  98   char* ptr, *_ptr;
  99   int ret;
 100
 101   if (str == NULL || fmt == NULL) {
 102     return -1;
 103   }
 104
 105   ptr = (char*)malloc_impl(128);
 106   if (ptr == NULL) {
 107     *str = NULL;
 108     return -1;
 109   }
 110
 111   ret = vsnprintf(ptr, 128, fmt, ap);
 112   if (ret < 0) {
 113     free_impl(ptr);
 114     *str = NULL;
 115     return -1;
 116   }
 117
 118   _ptr = realloc_impl(ptr, ret + 1);
 119   if (_ptr == NULL) {
 120     free_impl(ptr);
 121     *str = NULL;
 122     return -1;
 123   }
 124
 125   *str = _ptr;
 126
 127   return ret;
 128 }
 129
 130 MOZ_MEMORY_API int
 131 asprintf_impl(char **str, const char *fmt, ...)
 132 {
 133    int ret;
 134    va_list ap;
 135    va_start(ap, fmt);
 136
 137    ret = vasprintf_impl(str, fmt, ap);
 138
 139    va_end(ap);
 140
 141    return ret;
 142 }
 143 #endif
 144
 145 #ifdef XP_WIN
 146 /*
 147  *  There's a fun allocator mismatch in (at least) the VS 2010 CRT
 148  *  (see the giant comment in $(topsrcdir)/mozglue/build/Makefile.in)
 149  *  that gets redirected here to avoid a crash on shutdown.
 150  */
 151 void
 152 dumb_free_thunk(void *ptr)
 153 {
 154   return; /* shutdown leaks that we don't care about */
 155 }
 156
 157 #include <wchar.h>
 158
 159 /*
 160  *  We also need to provide our own impl of wcsdup so that we don't ask
 161  *  the CRT for memory from its heap (which will then be unfreeable).
 162  */
 163 wchar_t *
 164 wcsdup_impl(const wchar_t *src)
 165 {
 166   size_t len = wcslen(src);
 167   wchar_t *dst = (wchar_t*) malloc_impl((len + 1) * sizeof(wchar_t));
 168   if (dst)
 169     wcsncpy(dst, src, len + 1);
 170   return dst;
 171 }
 172 #endif /* XP_WIN */