aboutsummaryrefslogtreecommitdiffstats
path: root/contrib/libs/clang18-rt/lib/sanitizer_common/sanitizer_dense_map_info.h
blob: f4640369ae588a7f79593eb558c1a7779a42e888 (plain) (blame)
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
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
//===- sanitizer_dense_map_info.h - Type traits for DenseMap ----*- 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
//
//===----------------------------------------------------------------------===//

#ifndef SANITIZER_DENSE_MAP_INFO_H
#define SANITIZER_DENSE_MAP_INFO_H

#include "sanitizer_common.h"
#include "sanitizer_internal_defs.h"
#include "sanitizer_type_traits.h"

namespace __sanitizer {

namespace detail {

/// Simplistic combination of 32-bit hash values into 32-bit hash values.
static constexpr unsigned combineHashValue(unsigned a, unsigned b) {
  u64 key = (u64)a << 32 | (u64)b;
  key += ~(key << 32);
  key ^= (key >> 22);
  key += ~(key << 13);
  key ^= (key >> 8);
  key += (key << 3);
  key ^= (key >> 15);
  key += ~(key << 27);
  key ^= (key >> 31);
  return (unsigned)key;
}

// We extend a pair to allow users to override the bucket type with their own
// implementation without requiring two members.
template <typename KeyT, typename ValueT>
struct DenseMapPair {
  KeyT first = {};
  ValueT second = {};
  constexpr DenseMapPair() = default;
  constexpr DenseMapPair(const KeyT &f, const ValueT &s)
      : first(f), second(s) {}

  template <typename KeyT2, typename ValueT2>
  constexpr DenseMapPair(KeyT2 &&f, ValueT2 &&s)
      : first(__sanitizer::forward<KeyT2>(f)),
        second(__sanitizer::forward<ValueT2>(s)) {}

  constexpr DenseMapPair(const DenseMapPair &other) = default;
  constexpr DenseMapPair &operator=(const DenseMapPair &other) = default;
  constexpr DenseMapPair(DenseMapPair &&other) = default;
  constexpr DenseMapPair &operator=(DenseMapPair &&other) = default;

  KeyT &getFirst() { return first; }
  const KeyT &getFirst() const { return first; }
  ValueT &getSecond() { return second; }
  const ValueT &getSecond() const { return second; }
};

}  // end namespace detail

template <typename T>
struct DenseMapInfo {
  // static T getEmptyKey();
  // static T getTombstoneKey();
  // static unsigned getHashValue(const T &Val);
  // static bool isEqual(const T &LHS, const T &RHS);
};

// Provide DenseMapInfo for all pointers. Come up with sentinel pointer values
// that are aligned to alignof(T) bytes, but try to avoid requiring T to be
// complete. This allows clients to instantiate DenseMap<T*, ...> with forward
// declared key types. Assume that no pointer key type requires more than 4096
// bytes of alignment.
template <typename T>
struct DenseMapInfo<T *> {
  // The following should hold, but it would require T to be complete:
  // static_assert(alignof(T) <= (1 << Log2MaxAlign),
  //               "DenseMap does not support pointer keys requiring more than "
  //               "Log2MaxAlign bits of alignment");
  static constexpr uptr Log2MaxAlign = 12;

  static constexpr T *getEmptyKey() {
    uptr Val = static_cast<uptr>(-1);
    Val <<= Log2MaxAlign;
    return reinterpret_cast<T *>(Val);
  }

  static constexpr T *getTombstoneKey() {
    uptr Val = static_cast<uptr>(-2);
    Val <<= Log2MaxAlign;
    return reinterpret_cast<T *>(Val);
  }

  static constexpr unsigned getHashValue(const T *PtrVal) {
    return (unsigned((uptr)PtrVal) >> 4) ^ (unsigned((uptr)PtrVal) >> 9);
  }

  static constexpr bool isEqual(const T *LHS, const T *RHS) {
    return LHS == RHS;
  }
};

// Provide DenseMapInfo for chars.
template <>
struct DenseMapInfo<char> {
  static constexpr char getEmptyKey() { return ~0; }
  static constexpr char getTombstoneKey() { return ~0 - 1; }
  static constexpr unsigned getHashValue(const char &Val) { return Val * 37U; }

  static constexpr bool isEqual(const char &LHS, const char &RHS) {
    return LHS == RHS;
  }
};

// Provide DenseMapInfo for unsigned chars.
template <>
struct DenseMapInfo<unsigned char> {
  static constexpr unsigned char getEmptyKey() { return ~0; }
  static constexpr unsigned char getTombstoneKey() { return ~0 - 1; }
  static constexpr unsigned getHashValue(const unsigned char &Val) {
    return Val * 37U;
  }

  static constexpr bool isEqual(const unsigned char &LHS,
                                const unsigned char &RHS) {
    return LHS == RHS;
  }
};

// Provide DenseMapInfo for unsigned shorts.
template <>
struct DenseMapInfo<unsigned short> {
  static constexpr unsigned short getEmptyKey() { return 0xFFFF; }
  static constexpr unsigned short getTombstoneKey() { return 0xFFFF - 1; }
  static constexpr unsigned getHashValue(const unsigned short &Val) {
    return Val * 37U;
  }

  static constexpr bool isEqual(const unsigned short &LHS,
                                const unsigned short &RHS) {
    return LHS == RHS;
  }
};

// Provide DenseMapInfo for unsigned ints.
template <>
struct DenseMapInfo<unsigned> {
  static constexpr unsigned getEmptyKey() { return ~0U; }
  static constexpr unsigned getTombstoneKey() { return ~0U - 1; }
  static constexpr unsigned getHashValue(const unsigned &Val) {
    return Val * 37U;
  }

  static constexpr bool isEqual(const unsigned &LHS, const unsigned &RHS) {
    return LHS == RHS;
  }
};

// Provide DenseMapInfo for unsigned longs.
template <>
struct DenseMapInfo<unsigned long> {
  static constexpr unsigned long getEmptyKey() { return ~0UL; }
  static constexpr unsigned long getTombstoneKey() { return ~0UL - 1L; }

  static constexpr unsigned getHashValue(const unsigned long &Val) {
    return (unsigned)(Val * 37UL);
  }

  static constexpr bool isEqual(const unsigned long &LHS,
                                const unsigned long &RHS) {
    return LHS == RHS;
  }
};

// Provide DenseMapInfo for unsigned long longs.
template <>
struct DenseMapInfo<unsigned long long> {
  static constexpr unsigned long long getEmptyKey() { return ~0ULL; }
  static constexpr unsigned long long getTombstoneKey() { return ~0ULL - 1ULL; }

  static constexpr unsigned getHashValue(const unsigned long long &Val) {
    return (unsigned)(Val * 37ULL);
  }

  static constexpr bool isEqual(const unsigned long long &LHS,
                                const unsigned long long &RHS) {
    return LHS == RHS;
  }
};

// Provide DenseMapInfo for shorts.
template <>
struct DenseMapInfo<short> {
  static constexpr short getEmptyKey() { return 0x7FFF; }
  static constexpr short getTombstoneKey() { return -0x7FFF - 1; }
  static constexpr unsigned getHashValue(const short &Val) { return Val * 37U; }
  static constexpr bool isEqual(const short &LHS, const short &RHS) {
    return LHS == RHS;
  }
};

// Provide DenseMapInfo for ints.
template <>
struct DenseMapInfo<int> {
  static constexpr int getEmptyKey() { return 0x7fffffff; }
  static constexpr int getTombstoneKey() { return -0x7fffffff - 1; }
  static constexpr unsigned getHashValue(const int &Val) {
    return (unsigned)(Val * 37U);
  }

  static constexpr bool isEqual(const int &LHS, const int &RHS) {
    return LHS == RHS;
  }
};

// Provide DenseMapInfo for longs.
template <>
struct DenseMapInfo<long> {
  static constexpr long getEmptyKey() {
    return (1UL << (sizeof(long) * 8 - 1)) - 1UL;
  }

  static constexpr long getTombstoneKey() { return getEmptyKey() - 1L; }

  static constexpr unsigned getHashValue(const long &Val) {
    return (unsigned)(Val * 37UL);
  }

  static constexpr bool isEqual(const long &LHS, const long &RHS) {
    return LHS == RHS;
  }
};

// Provide DenseMapInfo for long longs.
template <>
struct DenseMapInfo<long long> {
  static constexpr long long getEmptyKey() { return 0x7fffffffffffffffLL; }
  static constexpr long long getTombstoneKey() {
    return -0x7fffffffffffffffLL - 1;
  }

  static constexpr unsigned getHashValue(const long long &Val) {
    return (unsigned)(Val * 37ULL);
  }

  static constexpr bool isEqual(const long long &LHS, const long long &RHS) {
    return LHS == RHS;
  }
};

// Provide DenseMapInfo for all pairs whose members have info.
template <typename T, typename U>
struct DenseMapInfo<detail::DenseMapPair<T, U>> {
  using Pair = detail::DenseMapPair<T, U>;
  using FirstInfo = DenseMapInfo<T>;
  using SecondInfo = DenseMapInfo<U>;

  static constexpr Pair getEmptyKey() {
    return detail::DenseMapPair<T, U>(FirstInfo::getEmptyKey(),
                                      SecondInfo::getEmptyKey());
  }

  static constexpr Pair getTombstoneKey() {
    return detail::DenseMapPair<T, U>(FirstInfo::getTombstoneKey(),
                                      SecondInfo::getTombstoneKey());
  }

  static constexpr unsigned getHashValue(const Pair &PairVal) {
    return detail::combineHashValue(FirstInfo::getHashValue(PairVal.first),
                                    SecondInfo::getHashValue(PairVal.second));
  }

  static constexpr bool isEqual(const Pair &LHS, const Pair &RHS) {
    return FirstInfo::isEqual(LHS.first, RHS.first) &&
           SecondInfo::isEqual(LHS.second, RHS.second);
  }
};

}  // namespace __sanitizer

#endif  // SANITIZER_DENSE_MAP_INFO_H