// Copyright 2017 The Abseil Authors.

//

// Licensed under the Apache License, Version 2.0 (the "License");

// you may not use this file except in compliance with the License.

// You may obtain a copy of the License at

//

//      https://www.apache.org/licenses/LICENSE-2.0

//

// Unless required by applicable law or agreed to in writing, software

// distributed under the License is distributed on an "AS IS" BASIS,

// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

// See the License for the specific language governing permissions and

// limitations under the License.

//

// Wrappers around lsan_interface functions.

//

// These are always-available run-time functions manipulating the LeakSanitizer,

// even when the lsan_interface (and LeakSanitizer) is not available. When

// LeakSanitizer is not linked in, these functions become no-op stubs.

#include "absl/debugging/leak_check.h"

#include "absl/base/attributes.h"

#include "absl/base/config.h"

#if defined(ABSL_HAVE_LEAK_SANITIZER)

#include <sanitizer/lsan_interface.h>

#if ABSL_HAVE_ATTRIBUTE_WEAK

extern "C" ABSL_ATTRIBUTE_WEAK int __lsan_is_turned_off() { return 0; }

#endif

namespace absl {

ABSL_NAMESPACE_BEGIN

bool HaveLeakSanitizer() { return true; }

#if ABSL_HAVE_ATTRIBUTE_WEAK

bool LeakCheckerIsActive() {

  return __lsan_is_turned_off() == 0;

}

#else

bool LeakCheckerIsActive() { return true; }

#endif

bool FindAndReportLeaks() { return __lsan_do_recoverable_leak_check() != 0; }

void DoIgnoreLeak(const void* ptr) { __lsan_ignore_object(ptr); }

void RegisterLivePointers(const void* ptr, size_t size) {

  __lsan_register_root_region(ptr, size);

}

void UnRegisterLivePointers(const void* ptr, size_t size) {

  __lsan_unregister_root_region(ptr, size);

}

LeakCheckDisabler::LeakCheckDisabler() { __lsan_disable(); }

LeakCheckDisabler::~LeakCheckDisabler() { __lsan_enable(); }

ABSL_NAMESPACE_END

}  // namespace absl

#else  // defined(ABSL_HAVE_LEAK_SANITIZER)

namespace absl {

ABSL_NAMESPACE_BEGIN

bool HaveLeakSanitizer() { return false; }

bool LeakCheckerIsActive() { return false; }

void DoIgnoreLeak(const void*) { }

void RegisterLivePointers(const void*, size_t) { }

void UnRegisterLivePointers(const void*, size_t) { }

LeakCheckDisabler::LeakCheckDisabler() = default;

LeakCheckDisabler::~LeakCheckDisabler() = default;

ABSL_NAMESPACE_END

}  // namespace absl

#endif  // defined(ABSL_HAVE_LEAK_SANITIZER)