Index: third_party/tcmalloc/chromium/src/tests/tcmalloc_unittest.cc |
=================================================================== |
--- third_party/tcmalloc/chromium/src/tests/tcmalloc_unittest.cc (revision 94429) |
+++ third_party/tcmalloc/chromium/src/tests/tcmalloc_unittest.cc (working copy) |
@@ -96,16 +96,45 @@ |
// Windows doesn't define pvalloc and a few other obsolete unix |
// functions; nor does it define posix_memalign (which is not obsolete). |
-#if defined(_MSC_VER) || defined(__MINGW32__) |
+#if defined(_WIN32) |
# define cfree free // don't bother to try to test these obsolete fns |
# define valloc malloc |
# define pvalloc malloc |
// I'd like to map posix_memalign to _aligned_malloc, but _aligned_malloc |
// must be paired with _aligned_free (not normal free), which is too |
// invasive a change to how we allocate memory here. So just bail |
-# include <errno.h> |
-# define memalign(alignment, size) malloc(size) |
-# define posix_memalign(pptr, align, size) ((*(pptr)=malloc(size)) ? 0 : ENOMEM) |
+static bool kOSSupportsMemalign = false; |
+static inline void* Memalign(size_t align, size_t size) { |
+ //LOG(FATAL) << "memalign not supported on windows"; |
+ exit(1); |
+} |
+static inline int PosixMemalign(void** ptr, size_t align, size_t size) { |
+ //LOG(FATAL) << "posix_memalign not supported on windows"; |
+ exit(1); |
+} |
+ |
+// OS X defines posix_memalign in some OS versions but not others; |
+// it's confusing enough to check that it's easiest to just not to test. |
+#elif defined(__APPLE__) |
+static bool kOSSupportsMemalign = false; |
+static inline void* Memalign(size_t align, size_t size) { |
+ //LOG(FATAL) << "memalign not supported on OS X"; |
+ exit(1); |
+} |
+static inline int PosixMemalign(void** ptr, size_t align, size_t size) { |
+ //LOG(FATAL) << "posix_memalign not supported on OS X"; |
+ exit(1); |
+} |
+ |
+#else |
+static bool kOSSupportsMemalign = true; |
+static inline void* Memalign(size_t align, size_t size) { |
+ return memalign(align, size); |
+} |
+static inline int PosixMemalign(void** ptr, size_t align, size_t size) { |
+ return posix_memalign(ptr, align, size); |
+} |
+ |
#endif |
// On systems (like freebsd) that don't define MAP_ANONYMOUS, use the old |
@@ -149,7 +178,12 @@ |
static const size_t kMaxSignedSize = ((size_t(1) << (kSizeBits-1)) - 1); |
static const size_t kNotTooBig = 100000; |
-static const size_t kTooBig = kMaxSize; |
+// We want an allocation that is definitely more than main memory. OS |
+// X has special logic to discard very big allocs before even passing |
+// the request along to the user-defined memory allocator; we're not |
+// interested in testing their logic, so we have to make sure we're |
+// not *too* big. |
+static const size_t kTooBig = kMaxSize - 100000; |
static int news_handled = 0; |
@@ -246,16 +280,18 @@ |
class AllocatorState : public TestHarness { |
public: |
- explicit AllocatorState(int seed) : TestHarness(seed) { |
- CHECK_GE(FLAGS_memalign_max_fraction, 0); |
- CHECK_LE(FLAGS_memalign_max_fraction, 1); |
- CHECK_GE(FLAGS_memalign_min_fraction, 0); |
- CHECK_LE(FLAGS_memalign_min_fraction, 1); |
- double delta = FLAGS_memalign_max_fraction - FLAGS_memalign_min_fraction; |
- CHECK_GE(delta, 0); |
- memalign_fraction_ = (Uniform(10000)/10000.0 * delta + |
- FLAGS_memalign_min_fraction); |
- //fprintf(LOGSTREAM, "memalign fraction: %f\n", memalign_fraction_); |
+ explicit AllocatorState(int seed) : TestHarness(seed), memalign_fraction_(0) { |
+ if (kOSSupportsMemalign) { |
+ CHECK_GE(FLAGS_memalign_max_fraction, 0); |
+ CHECK_LE(FLAGS_memalign_max_fraction, 1); |
+ CHECK_GE(FLAGS_memalign_min_fraction, 0); |
+ CHECK_LE(FLAGS_memalign_min_fraction, 1); |
+ double delta = FLAGS_memalign_max_fraction - FLAGS_memalign_min_fraction; |
+ CHECK_GE(delta, 0); |
+ memalign_fraction_ = (Uniform(10000)/10000.0 * delta + |
+ FLAGS_memalign_min_fraction); |
+ //fprintf(LOGSTREAM, "memalign fraction: %f\n", memalign_fraction_); |
+ } |
} |
virtual ~AllocatorState() {} |
@@ -269,7 +305,7 @@ |
(size < sizeof(intptr_t) || |
alignment < FLAGS_memalign_max_alignment_ratio * size)) { |
void *result = reinterpret_cast<void*>(static_cast<intptr_t>(0x1234)); |
- int err = posix_memalign(&result, alignment, size); |
+ int err = PosixMemalign(&result, alignment, size); |
if (err != 0) { |
CHECK_EQ(err, ENOMEM); |
} |
@@ -891,44 +927,42 @@ |
static void TestSetNewMode() { |
int old_mode = tc_set_new_mode(1); |
- // DebugAllocation will try to catch huge allocations. We need to avoid this |
- // by requesting a smaller malloc block, that still can't be satisfied. |
- const size_t kHugeRequest = kTooBig - 1024; |
- |
g_old_handler = std::set_new_handler(&OnNoMemory); |
g_no_memory = false; |
- void* ret = malloc(kHugeRequest); |
+ void* ret = malloc(kTooBig); |
EXPECT_EQ(NULL, ret); |
EXPECT_TRUE(g_no_memory); |
g_old_handler = std::set_new_handler(&OnNoMemory); |
g_no_memory = false; |
- ret = calloc(1, kHugeRequest); |
+ ret = calloc(1, kTooBig); |
EXPECT_EQ(NULL, ret); |
EXPECT_TRUE(g_no_memory); |
g_old_handler = std::set_new_handler(&OnNoMemory); |
g_no_memory = false; |
- ret = realloc(NULL, kHugeRequest); |
+ ret = realloc(NULL, kTooBig); |
EXPECT_EQ(NULL, ret); |
EXPECT_TRUE(g_no_memory); |
- // Not really important, but must be small enough such that kAlignment + |
- // kHugeRequest does not overflow. |
- const int kAlignment = 1 << 5; |
+ if (kOSSupportsMemalign) { |
+ // Not really important, but must be small enough such that |
+ // kAlignment + kTooBig does not overflow. |
+ const int kAlignment = 1 << 5; |
- g_old_handler = std::set_new_handler(&OnNoMemory); |
- g_no_memory = false; |
- ret = memalign(kAlignment, kHugeRequest); |
- EXPECT_EQ(NULL, ret); |
- EXPECT_TRUE(g_no_memory); |
+ g_old_handler = std::set_new_handler(&OnNoMemory); |
+ g_no_memory = false; |
+ ret = Memalign(kAlignment, kTooBig); |
+ EXPECT_EQ(NULL, ret); |
+ EXPECT_TRUE(g_no_memory); |
- g_old_handler = std::set_new_handler(&OnNoMemory); |
- g_no_memory = false; |
- EXPECT_EQ(ENOMEM, |
- posix_memalign(&ret, kAlignment, kHugeRequest)); |
- EXPECT_EQ(NULL, ret); |
- EXPECT_TRUE(g_no_memory); |
+ g_old_handler = std::set_new_handler(&OnNoMemory); |
+ g_no_memory = false; |
+ EXPECT_EQ(ENOMEM, |
+ PosixMemalign(&ret, kAlignment, kTooBig)); |
+ EXPECT_EQ(NULL, ret); |
+ EXPECT_TRUE(g_no_memory); |
+ } |
tc_set_new_mode(old_mode); |
} |
@@ -1032,17 +1066,19 @@ |
cfree(p1); // synonym for free |
VerifyDeleteHookWasCalled(); |
- CHECK_EQ(posix_memalign(&p1, sizeof(p1), 40), 0); |
- CHECK(p1 != NULL); |
- VerifyNewHookWasCalled(); |
- free(p1); |
- VerifyDeleteHookWasCalled(); |
+ if (kOSSupportsMemalign) { |
+ CHECK_EQ(PosixMemalign(&p1, sizeof(p1), 40), 0); |
+ CHECK(p1 != NULL); |
+ VerifyNewHookWasCalled(); |
+ free(p1); |
+ VerifyDeleteHookWasCalled(); |
- p1 = memalign(sizeof(p1) * 2, 50); |
- CHECK(p1 != NULL); |
- VerifyNewHookWasCalled(); |
- free(p1); |
- VerifyDeleteHookWasCalled(); |
+ p1 = Memalign(sizeof(p1) * 2, 50); |
+ CHECK(p1 != NULL); |
+ VerifyNewHookWasCalled(); |
+ free(p1); |
+ VerifyDeleteHookWasCalled(); |
+ } |
// Windows has _aligned_malloc. Let's test that that's captured too. |
#if (defined(_MSC_VER) || defined(__MINGW32__)) && !defined(PERFTOOLS_NO_ALIGNED_MALLOC) |