| //===-- alignment.cpp -------------------------------------------*- C++ -*-===// |
| // |
| // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
| // See https://llvm.org/LICENSE.txt for license information. |
| // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
| // |
| //===----------------------------------------------------------------------===// |
| |
| #include "gwp_asan/tests/harness.h" |
| #include "gwp_asan/utilities.h" |
| |
| TEST(AlignmentTest, PowerOfTwo) { |
| std::vector<std::pair<size_t, size_t>> AskedSizeToAlignedSize = { |
| {1, 1}, {2, 2}, {3, 4}, {4, 4}, {5, 8}, {7, 8}, |
| {8, 8}, {9, 16}, {15, 16}, {16, 16}, {17, 32}, {31, 32}, |
| {32, 32}, {33, 48}, {4095, 4096}, {4096, 4096}, |
| }; |
| |
| for (const auto &KV : AskedSizeToAlignedSize) { |
| EXPECT_EQ(KV.second, |
| gwp_asan::rightAlignedAllocationSize( |
| KV.first, gwp_asan::AlignmentStrategy::POWER_OF_TWO)); |
| } |
| } |
| |
| TEST(AlignmentTest, AlignBionic) { |
| std::vector<std::pair<size_t, size_t>> AskedSizeToAlignedSize = { |
| {1, 8}, {2, 8}, {3, 8}, {4, 8}, {5, 8}, {7, 8}, |
| {8, 8}, {9, 16}, {15, 16}, {16, 16}, {17, 24}, {31, 32}, |
| {32, 32}, {33, 40}, {4095, 4096}, {4096, 4096}, |
| }; |
| |
| for (const auto &KV : AskedSizeToAlignedSize) { |
| EXPECT_EQ(KV.second, gwp_asan::rightAlignedAllocationSize( |
| KV.first, gwp_asan::AlignmentStrategy::BIONIC)); |
| } |
| } |
| |
| TEST(AlignmentTest, PerfectAlignment) { |
| for (size_t i = 1; i <= 4096; ++i) { |
| EXPECT_EQ(i, gwp_asan::rightAlignedAllocationSize( |
| i, gwp_asan::AlignmentStrategy::PERFECT)); |
| } |
| } |