1 // Copyright 2014 The PDFium Authors
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com
6
7 #include "core/fxcrt/fx_memory.h"
8
9 #include <stdlib.h> // For abort().
10
11 #include <iterator>
12 #include <limits>
13
14 #include "build/build_config.h"
15 #include "third_party/base/debug/alias.h"
16
17 #if BUILDFLAG(IS_WIN)
18 #include <windows.h>
19 #endif
20
FXMEM_DefaultAlloc(size_t byte_size)21 void* FXMEM_DefaultAlloc(size_t byte_size) {
22 return pdfium::internal::Alloc(byte_size, 1);
23 }
24
FXMEM_DefaultCalloc(size_t num_elems,size_t byte_size)25 void* FXMEM_DefaultCalloc(size_t num_elems, size_t byte_size) {
26 return pdfium::internal::Calloc(num_elems, byte_size);
27 }
28
FXMEM_DefaultRealloc(void * pointer,size_t new_size)29 void* FXMEM_DefaultRealloc(void* pointer, size_t new_size) {
30 return pdfium::internal::Realloc(pointer, new_size, 1);
31 }
32
FXMEM_DefaultFree(void * pointer)33 void FXMEM_DefaultFree(void* pointer) {
34 FX_Free(pointer);
35 }
36
FX_OutOfMemoryTerminate(size_t size)37 NOINLINE void FX_OutOfMemoryTerminate(size_t size) {
38 // Convince the linker this should not be folded with similar functions using
39 // Identical Code Folding.
40 static int make_this_function_aliased = 0xbd;
41 pdfium::base::debug::Alias(&make_this_function_aliased);
42
43 #if BUILDFLAG(IS_WIN)
44 // The same custom Windows exception code used in Chromium and Breakpad.
45 constexpr DWORD kOomExceptionCode = 0xe0000008;
46 ULONG_PTR exception_args[] = {size};
47 ::RaiseException(kOomExceptionCode, EXCEPTION_NONCONTINUABLE,
48 std::size(exception_args), exception_args);
49 #endif
50
51 // Terminate cleanly.
52 abort();
53 }
54
55 namespace pdfium {
56 namespace internal {
57
AllocOrDie(size_t num_members,size_t member_size)58 void* AllocOrDie(size_t num_members, size_t member_size) {
59 void* result = Alloc(num_members, member_size);
60 if (!result)
61 FX_OutOfMemoryTerminate(0); // Never returns.
62
63 return result;
64 }
65
AllocOrDie2D(size_t w,size_t h,size_t member_size)66 void* AllocOrDie2D(size_t w, size_t h, size_t member_size) {
67 if (w >= std::numeric_limits<size_t>::max() / h)
68 FX_OutOfMemoryTerminate(0); // Never returns.
69
70 return AllocOrDie(w * h, member_size);
71 }
CallocOrDie(size_t num_members,size_t member_size)72 void* CallocOrDie(size_t num_members, size_t member_size) {
73 void* result = Calloc(num_members, member_size);
74 if (!result)
75 FX_OutOfMemoryTerminate(0); // Never returns.
76
77 return result;
78 }
79
CallocOrDie2D(size_t w,size_t h,size_t member_size)80 void* CallocOrDie2D(size_t w, size_t h, size_t member_size) {
81 if (w >= std::numeric_limits<size_t>::max() / h)
82 FX_OutOfMemoryTerminate(0); // Never returns.
83
84 return CallocOrDie(w * h, member_size);
85 }
86
ReallocOrDie(void * ptr,size_t num_members,size_t member_size)87 void* ReallocOrDie(void* ptr, size_t num_members, size_t member_size) {
88 void* result = Realloc(ptr, num_members, member_size);
89 if (!result)
90 FX_OutOfMemoryTerminate(0); // Never returns.
91
92 return result;
93 }
94
StringAllocOrDie(size_t num_members,size_t member_size)95 void* StringAllocOrDie(size_t num_members, size_t member_size) {
96 void* result = StringAlloc(num_members, member_size);
97 if (!result)
98 FX_OutOfMemoryTerminate(0); // Never returns.
99
100 return result;
101 }
102 } // namespace internal
103 } // namespace pdfium
104