compiler-rt/lib/sanitizer_common/sanitizer_chained_origin_depot.h

   1 //===-- sanitizer_chained_origin_depot.h ------------------------*- C++ -*-===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8 //
   9 // A storage for chained origins.
  10 //===----------------------------------------------------------------------===//
  11
  12 #ifndef SANITIZER_CHAINED_ORIGIN_DEPOT_H
  13 #define SANITIZER_CHAINED_ORIGIN_DEPOT_H
  14
  15 #include "sanitizer_common.h"
  16
  17 namespace __sanitizer {
  18
  19 class ChainedOriginDepot {
  20  public:
  21   ChainedOriginDepot();
  22
  23   // Gets the statistic of the origin chain storage.
  24   StackDepotStats GetStats() const;
  25
  26   // Stores a chain with StackDepot ID here_id and previous chain ID prev_id.
  27   // If successful, returns true and the new chain id new_id.
  28   // If the same element already exists, returns false and sets new_id to the
  29   // existing ID.
  30   bool Put(u32 here_id, u32 prev_id, u32 *new_id);
  31
  32   // Retrieves the stored StackDepot ID for the given origin ID.
  33   u32 Get(u32 id, u32 *other);
  34
  35   void LockBeforeFork();
  36   void UnlockAfterFork(bool fork_child);
  37   void TestOnlyUnmap();
  38
  39  private:
  40   ChainedOriginDepot(const ChainedOriginDepot &) = delete;
  41   void operator=(const ChainedOriginDepot &) = delete;
  42 };
  43
  44 }  // namespace __sanitizer
  45
  46 #endif  // SANITIZER_CHAINED_ORIGIN_DEPOT_H