xref: /llvm-project/compiler-rt/lib/rtsan/rtsan_context.h (revision 10d43061aa1c95facea397a900d9ce4f65fa03da)
1 //===--- rtsan_context.h - Realtime Sanitizer -------------------*- 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 //===----------------------------------------------------------------------===//
10 
11 #pragma once
12 
13 namespace __rtsan {
14 
15 class Context {
16 public:
17   Context();
18 
19   void RealtimePush();
20   void RealtimePop();
21 
22   void BypassPush();
23   void BypassPop();
24 
25   bool InRealtimeContext() const;
26   bool IsBypassed() const;
27 
28   Context(const Context &) = delete;
29   Context(Context &&) = delete;
30   Context &operator=(const Context &) = delete;
31   Context &operator=(Context &&) = delete;
32 
33 private:
34   int realtime_depth_{0};
35   int bypass_depth_{0};
36 };
37 
38 class ScopedBypass {
39 public:
40   [[nodiscard]] explicit ScopedBypass(Context &context) : context_(context) {
41     context_.BypassPush();
42   }
43 
44   ~ScopedBypass() { context_.BypassPop(); }
45 
46   ScopedBypass(const ScopedBypass &) = delete;
47   ScopedBypass &operator=(const ScopedBypass &) = delete;
48   ScopedBypass(ScopedBypass &&) = delete;
49   ScopedBypass &operator=(ScopedBypass &&) = delete;
50 
51 private:
52   Context &context_;
53 };
54 
55 Context &GetContextForThisThread();
56 } // namespace __rtsan
57