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