File: allocator_config_test.cpp

package info (click to toggle)
swiftlang 6.1.3-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 2,791,532 kB
  • sloc: cpp: 9,901,743; ansic: 2,201,431; asm: 1,091,827; python: 308,252; objc: 82,166; f90: 80,126; lisp: 38,358; pascal: 25,559; sh: 20,429; ml: 5,058; perl: 4,745; makefile: 4,484; awk: 3,535; javascript: 3,018; xml: 918; fortran: 664; cs: 573; ruby: 396
file content (119 lines) | stat: -rw-r--r-- 4,344 bytes parent folder | download | duplicates (17)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
//===-- allocator_config_test.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 "tests/scudo_unit_test.h"

#include "allocator_config.h"
#include "allocator_config_wrapper.h"
#include "common.h"
#include "secondary.h"

#include <type_traits>

struct TestBaseConfig {
  template <typename> using TSDRegistryT = void;
  template <typename> using PrimaryT = void;
  template <typename> using SecondaryT = void;
};

struct TestBaseConfigEnableOptionalFlag : public TestBaseConfig {
  static const bool MaySupportMemoryTagging = true;
  // Use the getter to avoid the test to `use` the address of static const
  // variable (which requires additional explicit definition).
  static bool getMaySupportMemoryTagging() { return MaySupportMemoryTagging; }
};

struct TestBasePrimaryConfig {
  using SizeClassMap = void;
  static const scudo::uptr RegionSizeLog = 18U;
  static const scudo::uptr GroupSizeLog = 18U;
  static const scudo::s32 MinReleaseToOsIntervalMs = INT32_MIN;
  static const scudo::s32 MaxReleaseToOsIntervalMs = INT32_MAX;
  typedef scudo::uptr CompactPtrT;
  static const scudo::uptr CompactPtrScale = 0;
  static const scudo::uptr MapSizeIncrement = 1UL << 18;
};

struct TestPrimaryConfig : public TestBaseConfig {
  struct Primary : TestBasePrimaryConfig {};
};

struct TestPrimaryConfigEnableOptionalFlag : public TestBaseConfig {
  struct Primary : TestBasePrimaryConfig {
    static const bool EnableRandomOffset = true;
    static bool getEnableRandomOffset() { return EnableRandomOffset; }
  };
};

struct TestPrimaryConfigEnableOptionalType : public TestBaseConfig {
  struct DummyConditionVariable {};

  struct Primary : TestBasePrimaryConfig {
    using ConditionVariableT = DummyConditionVariable;
  };
};

struct TestSecondaryConfig : public TestPrimaryConfig {
  struct Secondary {
    template <typename Config>
    using CacheT = scudo::MapAllocatorNoCache<Config>;
  };
};

struct TestSecondaryCacheConfigEnableOptionalFlag : public TestPrimaryConfig {
  struct Secondary {
    struct Cache {
      static const scudo::u32 EntriesArraySize = 256U;
      static scudo::u32 getEntriesArraySize() { return EntriesArraySize; }
    };
    template <typename T> using CacheT = scudo::MapAllocatorCache<T>;
  };
};

TEST(ScudoAllocatorConfigTest, VerifyOptionalFlags) {
  // Test the top level allocator optional config.
  //
  // `MaySupportMemoryTagging` is default off.
  EXPECT_FALSE(scudo::BaseConfig<TestBaseConfig>::getMaySupportMemoryTagging());
  EXPECT_EQ(scudo::BaseConfig<
                TestBaseConfigEnableOptionalFlag>::getMaySupportMemoryTagging(),
            TestBaseConfigEnableOptionalFlag::getMaySupportMemoryTagging());

  // Test primary optional config.
  //
  // `EnableRandomeOffset` is default off.
  EXPECT_FALSE(
      scudo::PrimaryConfig<TestPrimaryConfig>::getEnableRandomOffset());
  EXPECT_EQ(
      scudo::PrimaryConfig<
          TestPrimaryConfigEnableOptionalFlag>::getEnableRandomOffset(),
      TestPrimaryConfigEnableOptionalFlag::Primary::getEnableRandomOffset());

  // `ConditionVariableT` is default off.
  EXPECT_FALSE(
      scudo::PrimaryConfig<TestPrimaryConfig>::hasConditionVariableT());
  EXPECT_TRUE(scudo::PrimaryConfig<
              TestPrimaryConfigEnableOptionalType>::hasConditionVariableT());
  EXPECT_TRUE((std::is_same_v<
               typename scudo::PrimaryConfig<
                   TestPrimaryConfigEnableOptionalType>::ConditionVariableT,
               typename TestPrimaryConfigEnableOptionalType::Primary::
                   ConditionVariableT>));

  // Test secondary cache optional config.
  using NoCacheConfig =
      scudo::SecondaryConfig<TestSecondaryConfig>::CacheConfig;
  // `EntriesArraySize` is default 0.
  EXPECT_EQ(NoCacheConfig::getEntriesArraySize(), 0U);

  using CacheConfig = scudo::SecondaryConfig<
      TestSecondaryCacheConfigEnableOptionalFlag>::CacheConfig;
  EXPECT_EQ(CacheConfig::getEntriesArraySize(),
            TestSecondaryCacheConfigEnableOptionalFlag::Secondary::Cache::
                getEntriesArraySize());
}