Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(274)

Side by Side Diff: base/process/memory_win.cc

Issue 833923003: Revert of Fix EnableTerminationOnOutOfMemory for malloc (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Created 5 years, 11 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
« no previous file with comments | « no previous file | no next file » | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
1 // Copyright (c) 2013 The Chromium Authors. All rights reserved. 1 // Copyright (c) 2013 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be 2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file. 3 // found in the LICENSE file.
4 4
5 #include "base/process/memory.h" 5 #include "base/process/memory.h"
6 6
7 #include <new.h>
8 #include <psapi.h> 7 #include <psapi.h>
9 8
10 #include "base/logging.h" 9 #include "base/logging.h"
11 #include "base/memory/scoped_ptr.h" 10 #include "base/memory/scoped_ptr.h"
12 11
13 namespace base { 12 namespace base {
14 13
15 namespace { 14 namespace {
16 15
17 int OnNoMemory(size_t) { 16 void OnNoMemory() {
18 // Kill the process. This is important for security since most of code 17 // Kill the process. This is important for security, since WebKit doesn't
19 // does not check the result of memory allocation. 18 // NULL-check many memory allocations. If a malloc fails, returns NULL, and
19 // the buffer is then used, it provides a handy mapping of memory starting at
20 // address 0 for an attacker to utilize.
20 __debugbreak(); 21 __debugbreak();
21 _exit(1); 22 _exit(1);
22 return 0;
23 } 23 }
24 24
25 // HeapSetInformation function pointer. 25 // HeapSetInformation function pointer.
26 typedef BOOL (WINAPI* HeapSetFn)(HANDLE, HEAP_INFORMATION_CLASS, PVOID, SIZE_T); 26 typedef BOOL (WINAPI* HeapSetFn)(HANDLE, HEAP_INFORMATION_CLASS, PVOID, SIZE_T);
27 27
28 } // namespace 28 } // namespace
29 29
30 bool EnableLowFragmentationHeap() { 30 bool EnableLowFragmentationHeap() {
31 HMODULE kernel32 = GetModuleHandle(L"kernel32.dll"); 31 HMODULE kernel32 = GetModuleHandle(L"kernel32.dll");
32 HeapSetFn heap_set = reinterpret_cast<HeapSetFn>(GetProcAddress( 32 HeapSetFn heap_set = reinterpret_cast<HeapSetFn>(GetProcAddress(
(...skipping 28 matching lines...) Expand all
61 } 61 }
62 return true; 62 return true;
63 } 63 }
64 64
65 void EnableTerminationOnHeapCorruption() { 65 void EnableTerminationOnHeapCorruption() {
66 // Ignore the result code. Supported on XP SP3 and Vista. 66 // Ignore the result code. Supported on XP SP3 and Vista.
67 HeapSetInformation(NULL, HeapEnableTerminationOnCorruption, NULL, 0); 67 HeapSetInformation(NULL, HeapEnableTerminationOnCorruption, NULL, 0);
68 } 68 }
69 69
70 void EnableTerminationOnOutOfMemory() { 70 void EnableTerminationOnOutOfMemory() {
71 _set_new_handler(&OnNoMemory); 71 std::set_new_handler(&OnNoMemory);
72 _set_new_mode(1);
73 } 72 }
74 73
75 HMODULE GetModuleFromAddress(void* address) { 74 HMODULE GetModuleFromAddress(void* address) {
76 HMODULE instance = NULL; 75 HMODULE instance = NULL;
77 if (!::GetModuleHandleExA(GET_MODULE_HANDLE_EX_FLAG_FROM_ADDRESS | 76 if (!::GetModuleHandleExA(GET_MODULE_HANDLE_EX_FLAG_FROM_ADDRESS |
78 GET_MODULE_HANDLE_EX_FLAG_UNCHANGED_REFCOUNT, 77 GET_MODULE_HANDLE_EX_FLAG_UNCHANGED_REFCOUNT,
79 static_cast<char*>(address), 78 static_cast<char*>(address),
80 &instance)) { 79 &instance)) {
81 NOTREACHED(); 80 NOTREACHED();
82 } 81 }
83 return instance; 82 return instance;
84 } 83 }
85 84
86 // TODO(b.kelemen): implement it with the required semantics. On Linux this is 85 // TODO(b.kelemen): implement it with the required semantics. On Linux this is
87 // implemented with a weak symbol that is overridden by tcmalloc. This is 86 // implemented with a weak symbol that is overridden by tcmalloc. This is
88 // neccessary because base cannot have a direct dependency on tcmalloc. Since 87 // neccessary because base cannot have a direct dependency on tcmalloc. Since
89 // weak symbols are not supported on Windows this will involve some build time 88 // weak symbols are not supported on Windows this will involve some build time
90 // magic, much like what is done for libcrt in order to override the allocation 89 // magic, much like what is done for libcrt in order to override the allocation
91 // functions. 90 // functions.
92 bool UncheckedMalloc(size_t size, void** result) { 91 bool UncheckedMalloc(size_t size, void** result) {
93 *result = malloc(size); 92 *result = malloc(size);
94 return *result != NULL; 93 return *result != NULL;
95 } 94 }
96 95
97 } // namespace base 96 } // namespace base
OLDNEW
« no previous file with comments | « no previous file | no next file » | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698