13cab2bb3Spatrick //===-- bytemap.h -----------------------------------------------*- C++ -*-===// 23cab2bb3Spatrick // 33cab2bb3Spatrick // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 43cab2bb3Spatrick // See https://llvm.org/LICENSE.txt for license information. 53cab2bb3Spatrick // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 63cab2bb3Spatrick // 73cab2bb3Spatrick //===----------------------------------------------------------------------===// 83cab2bb3Spatrick 93cab2bb3Spatrick #ifndef SCUDO_BYTEMAP_H_ 103cab2bb3Spatrick #define SCUDO_BYTEMAP_H_ 113cab2bb3Spatrick 123cab2bb3Spatrick #include "atomic_helpers.h" 133cab2bb3Spatrick #include "common.h" 143cab2bb3Spatrick #include "mutex.h" 153cab2bb3Spatrick 163cab2bb3Spatrick namespace scudo { 173cab2bb3Spatrick 183cab2bb3Spatrick template <uptr Size> class FlatByteMap { 193cab2bb3Spatrick public: init()20*d89ec533Spatrick void init() { DCHECK(Size == 0 || Map[0] == 0); } 213cab2bb3Spatrick unmapTestOnly()22*d89ec533Spatrick void unmapTestOnly() { memset(Map, 0, Size); } 233cab2bb3Spatrick set(uptr Index,u8 Value)243cab2bb3Spatrick void set(uptr Index, u8 Value) { 253cab2bb3Spatrick DCHECK_LT(Index, Size); 263cab2bb3Spatrick DCHECK_EQ(0U, Map[Index]); 273cab2bb3Spatrick Map[Index] = Value; 283cab2bb3Spatrick } 293cab2bb3Spatrick u8 operator[](uptr Index) { 303cab2bb3Spatrick DCHECK_LT(Index, Size); 313cab2bb3Spatrick return Map[Index]; 323cab2bb3Spatrick } 333cab2bb3Spatrick disable()343cab2bb3Spatrick void disable() {} enable()353cab2bb3Spatrick void enable() {} 363cab2bb3Spatrick 373cab2bb3Spatrick private: 38*d89ec533Spatrick u8 Map[Size] = {}; 393cab2bb3Spatrick }; 403cab2bb3Spatrick 413cab2bb3Spatrick } // namespace scudo 423cab2bb3Spatrick 433cab2bb3Spatrick #endif // SCUDO_BYTEMAP_H_ 44