brintos

brintos / llvm-project-archived public Read only

0
0
Text · 2.5 KiB · f1640e9 Raw
59 lines · cpp
1//===-- Unittests for memalignment ----------------------------------------===//2//3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.4// See https://llvm.org/LICENSE.txt for license information.5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception6//7//===----------------------------------------------------------------------===//8 9#include "hdr/stdint_proxy.h"10#include "src/stdlib/memalignment.h"11#include "test/UnitTest/Test.h"12 13TEST(LlvmLibcMemAlignmentTest, NullPointer) {14  void *ptr = nullptr;15  EXPECT_EQ(LIBC_NAMESPACE::memalignment(ptr), static_cast<size_t>(0));16}17 18TEST(LlvmLibcMemAlignmentTest, SpecificAlignment) {19 20  // These addresses have known alignment patterns - if we can construct them21  uintptr_t addr_align2 = 0x2;   // 2-byte aligned22  uintptr_t addr_align4 = 0x4;   // 4-byte aligned23  uintptr_t addr_align8 = 0x8;   // 8-byte aligned24  uintptr_t addr_align16 = 0x10; // 16-byte aligned25  uintptr_t addr_align32 = 0x20; // 32-byte aligned26 27  void *ptr_align2 = reinterpret_cast<void *>(addr_align2);28  void *ptr_align4 = reinterpret_cast<void *>(addr_align4);29  void *ptr_align8 = reinterpret_cast<void *>(addr_align8);30  void *ptr_align16 = reinterpret_cast<void *>(addr_align16);31  void *ptr_align32 = reinterpret_cast<void *>(addr_align32);32 33  EXPECT_EQ(LIBC_NAMESPACE::memalignment(ptr_align2), static_cast<size_t>(2));34  EXPECT_EQ(LIBC_NAMESPACE::memalignment(ptr_align4), static_cast<size_t>(4));35  EXPECT_EQ(LIBC_NAMESPACE::memalignment(ptr_align8), static_cast<size_t>(8));36  EXPECT_EQ(LIBC_NAMESPACE::memalignment(ptr_align16), static_cast<size_t>(16));37  EXPECT_EQ(LIBC_NAMESPACE::memalignment(ptr_align32), static_cast<size_t>(32));38 39  uintptr_t addr_complex = 0x1234560; // 16-byte aligned (ends in 0)40  void *ptr_complex = reinterpret_cast<void *>(addr_complex);41  EXPECT_EQ(LIBC_NAMESPACE::memalignment(ptr_complex), static_cast<size_t>(32));42}43 44TEST(LlvmLibcMemAlignmentTest, AlignasSpecifiedAlignment) {45  alignas(16) static int aligned_16;46  alignas(32) static int aligned_32;47  alignas(64) static int aligned_64;48  alignas(128) static int aligned_128;49  alignas(256) static int aligned_256;50 51  EXPECT_GE(LIBC_NAMESPACE::memalignment(&aligned_16), static_cast<size_t>(16));52  EXPECT_GE(LIBC_NAMESPACE::memalignment(&aligned_32), static_cast<size_t>(32));53  EXPECT_GE(LIBC_NAMESPACE::memalignment(&aligned_64), static_cast<size_t>(64));54  EXPECT_GE(LIBC_NAMESPACE::memalignment(&aligned_128),55            static_cast<size_t>(128));56  EXPECT_GE(LIBC_NAMESPACE::memalignment(&aligned_256),57            static_cast<size_t>(256));58}59