1 // Copyright 2018 The Abseil Authors.
2 //
3 // Licensed under the Apache License, Version 2.0 (the "License");
4 // you may not use this file except in compliance with the License.
5 // You may obtain a copy of the License at
6 //
7 // https://www.apache.org/licenses/LICENSE-2.0
8 //
9 // Unless required by applicable law or agreed to in writing, software
10 // distributed under the License is distributed on an "AS IS" BASIS,
11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 // See the License for the specific language governing permissions and
13 // limitations under the License.
14 //
15 // -----------------------------------------------------------------------------
16 // File: leak_check.h
17 // -----------------------------------------------------------------------------
18 //
19 // This file contains functions that affect leak checking behavior within
20 // targets built with the LeakSanitizer (LSan), a memory leak detector that is
21 // integrated within the AddressSanitizer (ASan) as an additional component, or
22 // which can be used standalone. LSan and ASan are included (or can be provided)
23 // as additional components for most compilers such as Clang, gcc and MSVC.
24 // Note: this leak checking API is not yet supported in MSVC.
25 // Leak checking is enabled by default in all ASan builds.
26 //
27 // See https://github.com/google/sanitizers/wiki/AddressSanitizerLeakSanitizer
28 //
29 // -----------------------------------------------------------------------------
30 #ifndef ABSL_DEBUGGING_LEAK_CHECK_H_
31 #define ABSL_DEBUGGING_LEAK_CHECK_H_
32
33 #include <cstddef>
34
35 #include "absl/base/config.h"
36
37 namespace absl {
38 ABSL_NAMESPACE_BEGIN
39
40 // HaveLeakSanitizer()
41 //
42 // Returns true if a leak-checking sanitizer (either ASan or standalone LSan) is
43 // currently built into this target.
44 bool HaveLeakSanitizer();
45
46 // DoIgnoreLeak()
47 //
48 // Implements `IgnoreLeak()` below. This function should usually
49 // not be called directly; calling `IgnoreLeak()` is preferred.
50 void DoIgnoreLeak(const void* ptr);
51
52 // IgnoreLeak()
53 //
54 // Instruct the leak sanitizer to ignore leak warnings on the object referenced
55 // by the passed pointer, as well as all heap objects transitively referenced
56 // by it. The passed object pointer can point to either the beginning of the
57 // object or anywhere within it.
58 //
59 // Example:
60 //
61 // static T* obj = IgnoreLeak(new T(...));
62 //
63 // If the passed `ptr` does not point to an actively allocated object at the
64 // time `IgnoreLeak()` is called, the call is a no-op; if it is actively
65 // allocated, the object must not get deallocated later.
66 //
67 template <typename T>
IgnoreLeak(T * ptr)68 T* IgnoreLeak(T* ptr) {
69 DoIgnoreLeak(ptr);
70 return ptr;
71 }
72
73 // LeakCheckDisabler
74 //
75 // This helper class indicates that any heap allocations done in the code block
76 // covered by the scoped object, which should be allocated on the stack, will
77 // not be reported as leaks. Leak check disabling will occur within the code
78 // block and any nested function calls within the code block.
79 //
80 // Example:
81 //
82 // void Foo() {
83 // LeakCheckDisabler disabler;
84 // ... code that allocates objects whose leaks should be ignored ...
85 // }
86 //
87 // REQUIRES: Destructor runs in same thread as constructor
88 class LeakCheckDisabler {
89 public:
90 LeakCheckDisabler();
91 LeakCheckDisabler(const LeakCheckDisabler&) = delete;
92 LeakCheckDisabler& operator=(const LeakCheckDisabler&) = delete;
93 ~LeakCheckDisabler();
94 };
95
96 // RegisterLivePointers()
97 //
98 // Registers `ptr[0,size-1]` as pointers to memory that is still actively being
99 // referenced and for which leak checking should be ignored. This function is
100 // useful if you store pointers in mapped memory, for memory ranges that we know
101 // are correct but for which normal analysis would flag as leaked code.
102 void RegisterLivePointers(const void* ptr, size_t size);
103
104 // UnRegisterLivePointers()
105 //
106 // Deregisters the pointers previously marked as active in
107 // `RegisterLivePointers()`, enabling leak checking of those pointers.
108 void UnRegisterLivePointers(const void* ptr, size_t size);
109
110 ABSL_NAMESPACE_END
111 } // namespace absl
112
113 #endif // ABSL_DEBUGGING_LEAK_CHECK_H_
114