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 3 // found in the LICENSE file. 4 5 #include "base/memory/aligned_memory.h" 6 #include "base/memory/scoped_ptr.h" 7 #include "testing/gtest/include/gtest/gtest.h" 8 9 #define EXPECT_ALIGNED(ptr, align) \ 10 EXPECT_EQ(0u, reinterpret_cast<uintptr_t>(ptr) & (align - 1)) 11 12 namespace { 13 14 using base::AlignedMemory; 15 16 TEST(AlignedMemoryTest, StaticAlignment) { 17 static AlignedMemory<8, 8> raw8; 18 static AlignedMemory<8, 16> raw16; 19 static AlignedMemory<8, 256> raw256; 20 static AlignedMemory<8, 4096> raw4096; 21 22 EXPECT_EQ(8u, ALIGNOF(raw8)); 23 EXPECT_EQ(16u, ALIGNOF(raw16)); 24 EXPECT_EQ(256u, ALIGNOF(raw256)); 25 EXPECT_EQ(4096u, ALIGNOF(raw4096)); 26 27 EXPECT_ALIGNED(raw8.void_data(), 8); 28 EXPECT_ALIGNED(raw16.void_data(), 16); 29 EXPECT_ALIGNED(raw256.void_data(), 256); 30 EXPECT_ALIGNED(raw4096.void_data(), 4096); 31 } 32 33 TEST(AlignedMemoryTest, StackAlignment) { 34 AlignedMemory<8, 8> raw8; 35 AlignedMemory<8, 16> raw16; 36 AlignedMemory<8, 128> raw128; 37 38 EXPECT_EQ(8u, ALIGNOF(raw8)); 39 EXPECT_EQ(16u, ALIGNOF(raw16)); 40 EXPECT_EQ(128u, ALIGNOF(raw128)); 41 42 EXPECT_ALIGNED(raw8.void_data(), 8); 43 EXPECT_ALIGNED(raw16.void_data(), 16); 44 45 // TODO(ios): __attribute__((aligned(X))) with X >= 128 does not works on 46 // the stack when building for arm64 on iOS, http://crbug.com/349003 47 #if !(defined(OS_IOS) && defined(ARCH_CPU_ARM64)) 48 EXPECT_ALIGNED(raw128.void_data(), 128); 49 50 // NaCl x86-64 compiler emits non-validating instructions for >128 51 // bytes alignment. 52 // http://www.chromium.org/nativeclient/design-documents/nacl-sfi-model-on-x86-64-systems 53 // TODO(hamaji): Ideally, NaCl compiler for x86-64 should workaround 54 // this limitation and this #if should be removed. 55 // https://code.google.com/p/nativeclient/issues/detail?id=3463 56 #if !(defined(OS_NACL) && defined(ARCH_CPU_X86_64)) 57 AlignedMemory<8, 256> raw256; 58 EXPECT_EQ(256u, ALIGNOF(raw256)); 59 EXPECT_ALIGNED(raw256.void_data(), 256); 60 61 // TODO(ios): This test hits an armv7 bug in clang. crbug.com/138066 62 #if !(defined(OS_IOS) && defined(ARCH_CPU_ARM_FAMILY)) 63 AlignedMemory<8, 4096> raw4096; 64 EXPECT_EQ(4096u, ALIGNOF(raw4096)); 65 EXPECT_ALIGNED(raw4096.void_data(), 4096); 66 #endif // !(defined(OS_IOS) && defined(ARCH_CPU_ARM_FAMILY)) 67 #endif // !(defined(OS_NACL) && defined(ARCH_CPU_X86_64)) 68 #endif // !(defined(OS_IOS) && defined(ARCH_CPU_ARM64)) 69 } 70 71 TEST(AlignedMemoryTest, DynamicAllocation) { 72 void* p = base::AlignedAlloc(8, 8); 73 EXPECT_TRUE(p); 74 EXPECT_ALIGNED(p, 8); 75 base::AlignedFree(p); 76 77 p = base::AlignedAlloc(8, 16); 78 EXPECT_TRUE(p); 79 EXPECT_ALIGNED(p, 16); 80 base::AlignedFree(p); 81 82 p = base::AlignedAlloc(8, 256); 83 EXPECT_TRUE(p); 84 EXPECT_ALIGNED(p, 256); 85 base::AlignedFree(p); 86 87 p = base::AlignedAlloc(8, 4096); 88 EXPECT_TRUE(p); 89 EXPECT_ALIGNED(p, 4096); 90 base::AlignedFree(p); 91 } 92 93 TEST(AlignedMemoryTest, ScopedDynamicAllocation) { 94 scoped_ptr<float, base::AlignedFreeDeleter> p( 95 static_cast<float*>(base::AlignedAlloc(8, 8))); 96 EXPECT_TRUE(p.get()); 97 EXPECT_ALIGNED(p.get(), 8); 98 } 99 100 } // namespace 101