skia/ext/SkMemory_new_handler.cpp

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include <stdio.h>
   6 #include <stdlib.h>
   7 #include <new>
   8
   9 #include "base/process/memory.h"
  10
  11 #include "third_party/skia/include/core/SkTypes.h"
  12 #include "third_party/skia/include/core/SkThread.h"
  13
  14 // This implementation of sk_malloc_flags() and friends is identical to
  15 // SkMemory_malloc.cpp, except that it disables the CRT's new_handler during
  16 // malloc() and calloc() when SK_MALLOC_THROW is not set (because our normal
  17 // new_handler itself will crash on failure when using tcmalloc).
  18
  19 SK_DECLARE_STATIC_MUTEX(gSkNewHandlerMutex);
  20
  21 static inline void* throw_on_failure(size_t size, void* p) {
  22     if (size > 0 && p == NULL) {
  23         // If we've got a NULL here, the only reason we should have failed is running out of RAM.
  24         sk_out_of_memory();
  25     }
  26     return p;
  27 }
  28
  29 void sk_throw() {
  30     SkASSERT(!"sk_throw");
  31     abort();
  32 }
  33
  34 void sk_out_of_memory(void) {
  35     SkASSERT(!"sk_out_of_memory");
  36     abort();
  37 }
  38
  39 void* sk_realloc_throw(void* addr, size_t size) {
  40     return throw_on_failure(size, realloc(addr, size));
  41 }
  42
  43 void sk_free(void* p) {
  44     if (p) {
  45         free(p);
  46     }
  47 }
  48
  49 void* sk_malloc_throw(size_t size) {
  50     return throw_on_failure(size, malloc(size));
  51 }
  52
  53 static void* sk_malloc_nothrow(size_t size) {
  54     // TODO(b.kelemen): we should always use UncheckedMalloc but currently it
  55     // doesn't work as intended everywhere.
  56 #if  defined(LIBC_GLIBC) || defined(USE_TCMALLOC) || \
  57      (defined(OS_MACOSX) && !defined(OS_IOS)) || defined(OS_ANDROID)
  58     void* result;
  59     // It's the responsibility of the caller to check the return value.
  60     ignore_result(base::UncheckedMalloc(size, &result));
  61     return result;
  62 #else
  63     // This is not really thread safe.  It only won't collide with itself, but we're totally
  64     // unprotected from races with other code that calls set_new_handler.
  65     SkAutoMutexAcquire lock(gSkNewHandlerMutex);
  66     std::new_handler old_handler = std::set_new_handler(NULL);
  67     void* p = malloc(size);
  68     std::set_new_handler(old_handler);
  69     return p;
  70 #endif
  71 }
  72
  73 void* sk_malloc_flags(size_t size, unsigned flags) {
  74     if (flags & SK_MALLOC_THROW) {
  75         return sk_malloc_throw(size);
  76     }
  77     return sk_malloc_nothrow(size);
  78 }
  79
  80 void* sk_calloc_throw(size_t size) {
  81     return throw_on_failure(size, calloc(size, 1));
  82 }
  83
  84 void* sk_calloc(size_t size) {
  85     // TODO(b.kelemen): we should always use UncheckedCalloc but currently it
  86     // doesn't work as intended everywhere.
  87 #if  defined(LIBC_GLIBC) || defined(USE_TCMALLOC) || \
  88      (defined(OS_MACOSX) && !defined(OS_IOS)) || defined(OS_ANDROID)
  89     void* result;
  90     // It's the responsibility of the caller to check the return value.
  91     ignore_result(base::UncheckedCalloc(size, 1, &result));
  92     return result;
  93 #else
  94     SkAutoMutexAcquire lock(gSkNewHandlerMutex);
  95     std::new_handler old_handler = std::set_new_handler(NULL);
  96     void* p = calloc(size, 1);
  97     std::set_new_handler(old_handler);
  98     return p;
  99 #endif
 100 }