10b57cec5SDimitry Andric //===-- bytemap.h -----------------------------------------------*- C++ -*-===// 20b57cec5SDimitry Andric // 30b57cec5SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 40b57cec5SDimitry Andric // See https://llvm.org/LICENSE.txt for license information. 50b57cec5SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 60b57cec5SDimitry Andric // 70b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 80b57cec5SDimitry Andric 90b57cec5SDimitry Andric #ifndef SCUDO_BYTEMAP_H_ 100b57cec5SDimitry Andric #define SCUDO_BYTEMAP_H_ 110b57cec5SDimitry Andric 120b57cec5SDimitry Andric #include "atomic_helpers.h" 130b57cec5SDimitry Andric #include "common.h" 140b57cec5SDimitry Andric #include "mutex.h" 150b57cec5SDimitry Andric 160b57cec5SDimitry Andric namespace scudo { 170b57cec5SDimitry Andric 180b57cec5SDimitry Andric template <uptr Size> class FlatByteMap { 190b57cec5SDimitry Andric public: init()20*fe6060f1SDimitry Andric void init() { DCHECK(Size == 0 || Map[0] == 0); } 210b57cec5SDimitry Andric unmapTestOnly()22*fe6060f1SDimitry Andric void unmapTestOnly() { memset(Map, 0, Size); } 230b57cec5SDimitry Andric set(uptr Index,u8 Value)240b57cec5SDimitry Andric void set(uptr Index, u8 Value) { 250b57cec5SDimitry Andric DCHECK_LT(Index, Size); 260b57cec5SDimitry Andric DCHECK_EQ(0U, Map[Index]); 270b57cec5SDimitry Andric Map[Index] = Value; 280b57cec5SDimitry Andric } 290b57cec5SDimitry Andric u8 operator[](uptr Index) { 300b57cec5SDimitry Andric DCHECK_LT(Index, Size); 310b57cec5SDimitry Andric return Map[Index]; 320b57cec5SDimitry Andric } 330b57cec5SDimitry Andric disable()34480093f4SDimitry Andric void disable() {} enable()35480093f4SDimitry Andric void enable() {} 36480093f4SDimitry Andric 370b57cec5SDimitry Andric private: 38*fe6060f1SDimitry Andric u8 Map[Size] = {}; 390b57cec5SDimitry Andric }; 400b57cec5SDimitry Andric 410b57cec5SDimitry Andric } // namespace scudo 420b57cec5SDimitry Andric 430b57cec5SDimitry Andric #endif // SCUDO_BYTEMAP_H_ 44