1 //===-- sanitizer_termination.cc --------------------------------*- C++ -*-===//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 /// This file contains the Sanitizer termination functions CheckFailed and Die,
11 /// and the callback functionalities associated with them.
13 //===----------------------------------------------------------------------===//
15 #include "sanitizer_common.h"
16 #include "sanitizer_libc.h"
18 namespace __sanitizer {
20 static const int kMaxNumOfInternalDieCallbacks = 5;
21 static DieCallbackType InternalDieCallbacks[kMaxNumOfInternalDieCallbacks];
23 bool AddDieCallback(DieCallbackType callback) {
24 for (int i = 0; i < kMaxNumOfInternalDieCallbacks; i++) {
25 if (InternalDieCallbacks[i] == nullptr) {
26 InternalDieCallbacks[i] = callback;
33 bool RemoveDieCallback(DieCallbackType callback) {
34 for (int i = 0; i < kMaxNumOfInternalDieCallbacks; i++) {
35 if (InternalDieCallbacks[i] == callback) {
36 internal_memmove(&InternalDieCallbacks[i], &InternalDieCallbacks[i + 1],
37 sizeof(InternalDieCallbacks[0]) *
38 (kMaxNumOfInternalDieCallbacks - i - 1));
39 InternalDieCallbacks[kMaxNumOfInternalDieCallbacks - 1] = nullptr;
46 static DieCallbackType UserDieCallback;
47 void SetUserDieCallback(DieCallbackType callback) {
48 UserDieCallback = callback;
54 for (int i = kMaxNumOfInternalDieCallbacks - 1; i >= 0; i--) {
55 if (InternalDieCallbacks[i])
56 InternalDieCallbacks[i]();
58 if (common_flags()->abort_on_error)
60 internal__exit(common_flags()->exitcode);
63 static CheckFailedCallbackType CheckFailedCallback;
64 void SetCheckFailedCallback(CheckFailedCallbackType callback) {
65 CheckFailedCallback = callback;
68 const int kSecondsToSleepWhenRecursiveCheckFailed = 2;
70 void NORETURN CheckFailed(const char *file, int line, const char *cond,
72 static atomic_uint32_t num_calls;
73 if (atomic_fetch_add(&num_calls, 1, memory_order_relaxed) > 10) {
74 SleepForSeconds(kSecondsToSleepWhenRecursiveCheckFailed);
78 if (CheckFailedCallback) {
79 CheckFailedCallback(file, line, cond, v1, v2);
81 Report("Sanitizer CHECK failed: %s:%d %s (%lld, %lld)\n", file, line, cond,
86 } // namespace __sanitizer