| David Blaikie | ef54338 | 2018-11-17 18:03:47 +0000 | [diff] [blame] | 1 | //===- BuryPointer.cpp - Memory Manipulation/Leak ---------------*- C++ -*-===// | 
|  | 2 | // | 
| Chandler Carruth | 2946cd7 | 2019-01-19 08:50:56 +0000 | [diff] [blame] | 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 | 
| David Blaikie | ef54338 | 2018-11-17 18:03:47 +0000 | [diff] [blame] | 6 | // | 
|  | 7 | //===----------------------------------------------------------------------===// | 
|  | 8 |  | 
|  | 9 | #include "llvm/Support/BuryPointer.h" | 
|  | 10 | #include "llvm/Support/Compiler.h" | 
|  | 11 | #include <atomic> | 
|  | 12 |  | 
|  | 13 | namespace llvm { | 
|  | 14 |  | 
|  | 15 | void BuryPointer(const void *Ptr) { | 
|  | 16 | // This function may be called only a small fixed amount of times per each | 
|  | 17 | // invocation, otherwise we do actually have a leak which we want to report. | 
|  | 18 | // If this function is called more than kGraveYardMaxSize times, the pointers | 
|  | 19 | // will not be properly buried and a leak detector will report a leak, which | 
|  | 20 | // is what we want in such case. | 
|  | 21 | static const size_t kGraveYardMaxSize = 16; | 
|  | 22 | LLVM_ATTRIBUTE_UNUSED static const void *GraveYard[kGraveYardMaxSize]; | 
|  | 23 | static std::atomic<unsigned> GraveYardSize; | 
|  | 24 | unsigned Idx = GraveYardSize++; | 
|  | 25 | if (Idx >= kGraveYardMaxSize) | 
|  | 26 | return; | 
|  | 27 | GraveYard[Idx] = Ptr; | 
|  | 28 | } | 
|  | 29 |  | 
|  | 30 | } |