| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 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 #define _CRT_SECURE_NO_WARNINGS | 5 #define _CRT_SECURE_NO_WARNINGS |
| 6 | 6 |
| 7 #include "base/process/memory.h" | 7 #include "base/process/memory.h" |
| 8 | 8 |
| 9 #include <stddef.h> | 9 #include <stddef.h> |
| 10 | 10 |
| (...skipping 43 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 54 #endif // defined(OS_WIN) | 54 #endif // defined(OS_WIN) |
| 55 | 55 |
| 56 #if defined(OS_MACOSX) | 56 #if defined(OS_MACOSX) |
| 57 | 57 |
| 58 // For the following Mac tests: | 58 // For the following Mac tests: |
| 59 // Note that base::EnableTerminationOnHeapCorruption() is called as part of | 59 // Note that base::EnableTerminationOnHeapCorruption() is called as part of |
| 60 // test suite setup and does not need to be done again, else mach_override | 60 // test suite setup and does not need to be done again, else mach_override |
| 61 // will fail. | 61 // will fail. |
| 62 | 62 |
| 63 TEST(ProcessMemoryTest, MacTerminateOnHeapCorruption) { | 63 TEST(ProcessMemoryTest, MacTerminateOnHeapCorruption) { |
| 64 #if BUILDFLAG(USE_EXPERIMENTAL_ALLOCATOR_SHIM) | 64 #if BUILDFLAG(USE_ALLOCATOR_SHIM) |
| 65 base::allocator::InitializeAllocatorShim(); | 65 base::allocator::InitializeAllocatorShim(); |
| 66 #endif | 66 #endif |
| 67 // Assert that freeing an unallocated pointer will crash the process. | 67 // Assert that freeing an unallocated pointer will crash the process. |
| 68 char buf[9]; | 68 char buf[9]; |
| 69 asm("" : "=r" (buf)); // Prevent clang from being too smart. | 69 asm("" : "=r" (buf)); // Prevent clang from being too smart. |
| 70 #if ARCH_CPU_64_BITS | 70 #if ARCH_CPU_64_BITS |
| 71 // On 64 bit Macs, the malloc system automatically abort()s on heap corruption | 71 // On 64 bit Macs, the malloc system automatically abort()s on heap corruption |
| 72 // but does not output anything. | 72 // but does not output anything. |
| 73 ASSERT_DEATH(free(buf), ""); | 73 ASSERT_DEATH(free(buf), ""); |
| 74 #elif defined(ADDRESS_SANITIZER) | 74 #elif defined(ADDRESS_SANITIZER) |
| 75 // AddressSanitizer replaces malloc() and prints a different error message on | 75 // AddressSanitizer replaces malloc() and prints a different error message on |
| 76 // heap corruption. | 76 // heap corruption. |
| 77 ASSERT_DEATH(free(buf), "attempting free on address which " | 77 ASSERT_DEATH(free(buf), "attempting free on address which " |
| 78 "was not malloc\\(\\)-ed"); | 78 "was not malloc\\(\\)-ed"); |
| 79 #else | 79 #else |
| 80 ADD_FAILURE() << "This test is not supported in this build configuration."; | 80 ADD_FAILURE() << "This test is not supported in this build configuration."; |
| 81 #endif | 81 #endif |
| 82 | 82 |
| 83 #if BUILDFLAG(USE_EXPERIMENTAL_ALLOCATOR_SHIM) | 83 #if BUILDFLAG(USE_ALLOCATOR_SHIM) |
| 84 base::allocator::UninterceptMallocZonesForTesting(); | 84 base::allocator::UninterceptMallocZonesForTesting(); |
| 85 #endif | 85 #endif |
| 86 } | 86 } |
| 87 | 87 |
| 88 #endif // defined(OS_MACOSX) | 88 #endif // defined(OS_MACOSX) |
| 89 | 89 |
| 90 TEST(MemoryTest, AllocatorShimWorking) { | 90 TEST(MemoryTest, AllocatorShimWorking) { |
| 91 #if defined(OS_MACOSX) | 91 #if defined(OS_MACOSX) |
| 92 #if BUILDFLAG(USE_EXPERIMENTAL_ALLOCATOR_SHIM) | 92 #if BUILDFLAG(USE_ALLOCATOR_SHIM) |
| 93 base::allocator::InitializeAllocatorShim(); | 93 base::allocator::InitializeAllocatorShim(); |
| 94 #endif | 94 #endif |
| 95 base::allocator::InterceptAllocationsMac(); | 95 base::allocator::InterceptAllocationsMac(); |
| 96 #endif | 96 #endif |
| 97 ASSERT_TRUE(base::allocator::IsAllocatorInitialized()); | 97 ASSERT_TRUE(base::allocator::IsAllocatorInitialized()); |
| 98 | 98 |
| 99 #if defined(OS_MACOSX) | 99 #if defined(OS_MACOSX) |
| 100 base::allocator::UninterceptMallocZonesForTesting(); | 100 base::allocator::UninterceptMallocZonesForTesting(); |
| 101 #endif | 101 #endif |
| 102 } | 102 } |
| (...skipping 33 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 136 protected: | 136 protected: |
| 137 void* value_; | 137 void* value_; |
| 138 size_t test_size_; | 138 size_t test_size_; |
| 139 size_t insecure_test_size_; | 139 size_t insecure_test_size_; |
| 140 ssize_t signed_test_size_; | 140 ssize_t signed_test_size_; |
| 141 }; | 141 }; |
| 142 | 142 |
| 143 class OutOfMemoryDeathTest : public OutOfMemoryTest { | 143 class OutOfMemoryDeathTest : public OutOfMemoryTest { |
| 144 public: | 144 public: |
| 145 void SetUpInDeathAssert() { | 145 void SetUpInDeathAssert() { |
| 146 #if defined(OS_MACOSX) && BUILDFLAG(USE_EXPERIMENTAL_ALLOCATOR_SHIM) | 146 #if defined(OS_MACOSX) && BUILDFLAG(USE_ALLOCATOR_SHIM) |
| 147 base::allocator::InitializeAllocatorShim(); | 147 base::allocator::InitializeAllocatorShim(); |
| 148 #endif | 148 #endif |
| 149 | 149 |
| 150 // Must call EnableTerminationOnOutOfMemory() because that is called from | 150 // Must call EnableTerminationOnOutOfMemory() because that is called from |
| 151 // chrome's main function and therefore hasn't been called yet. | 151 // chrome's main function and therefore hasn't been called yet. |
| 152 // Since this call may result in another thread being created and death | 152 // Since this call may result in another thread being created and death |
| 153 // tests shouldn't be started in a multithread environment, this call | 153 // tests shouldn't be started in a multithread environment, this call |
| 154 // should be done inside of the ASSERT_DEATH. | 154 // should be done inside of the ASSERT_DEATH. |
| 155 base::EnableTerminationOnOutOfMemory(); | 155 base::EnableTerminationOnOutOfMemory(); |
| 156 } | 156 } |
| (...skipping 346 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 503 bytes = static_cast<const char*>(value_); | 503 bytes = static_cast<const char*>(value_); |
| 504 for (size_t i = 0; i < (kSafeCallocItems * kSafeCallocSize); ++i) | 504 for (size_t i = 0; i < (kSafeCallocItems * kSafeCallocSize); ++i) |
| 505 EXPECT_EQ(0, bytes[i]); | 505 EXPECT_EQ(0, bytes[i]); |
| 506 free(value_); | 506 free(value_); |
| 507 | 507 |
| 508 EXPECT_FALSE(base::UncheckedCalloc(1, test_size_, &value_)); | 508 EXPECT_FALSE(base::UncheckedCalloc(1, test_size_, &value_)); |
| 509 EXPECT_TRUE(value_ == NULL); | 509 EXPECT_TRUE(value_ == NULL); |
| 510 } | 510 } |
| 511 #endif // !defined(OS_OPENBSD) && BUILDFLAG(ENABLE_WIN_ALLOCATOR_SHIM_TESTS) && | 511 #endif // !defined(OS_OPENBSD) && BUILDFLAG(ENABLE_WIN_ALLOCATOR_SHIM_TESTS) && |
| 512 // !defined(MEMORY_TOOL_REPLACES_ALLOCATOR) | 512 // !defined(MEMORY_TOOL_REPLACES_ALLOCATOR) |
| OLD | NEW |