1 //===-- sanitizer_termination.cpp -------------------------------*- 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 /// This file contains the Sanitizer termination functions CheckFailed and Die,
10 /// and the callback functionalities associated with them.
11 ///
12 //===----------------------------------------------------------------------===//
13
14 #include "sanitizer_common.h"
15 #include "sanitizer_libc.h"
16
17 namespace __sanitizer {
18
19 static const int kMaxNumOfInternalDieCallbacks = 5;
20 static DieCallbackType InternalDieCallbacks[kMaxNumOfInternalDieCallbacks];
21
AddDieCallback(DieCallbackType callback)22 bool AddDieCallback(DieCallbackType callback) {
23 for (int i = 0; i < kMaxNumOfInternalDieCallbacks; i++) {
24 if (InternalDieCallbacks[i] == nullptr) {
25 InternalDieCallbacks[i] = callback;
26 return true;
27 }
28 }
29 return false;
30 }
31
RemoveDieCallback(DieCallbackType callback)32 bool RemoveDieCallback(DieCallbackType callback) {
33 for (int i = 0; i < kMaxNumOfInternalDieCallbacks; i++) {
34 if (InternalDieCallbacks[i] == callback) {
35 internal_memmove(&InternalDieCallbacks[i], &InternalDieCallbacks[i + 1],
36 sizeof(InternalDieCallbacks[0]) *
37 (kMaxNumOfInternalDieCallbacks - i - 1));
38 InternalDieCallbacks[kMaxNumOfInternalDieCallbacks - 1] = nullptr;
39 return true;
40 }
41 }
42 return false;
43 }
44
45 static DieCallbackType UserDieCallback;
SetUserDieCallback(DieCallbackType callback)46 void SetUserDieCallback(DieCallbackType callback) {
47 UserDieCallback = callback;
48 }
49
Die()50 void NORETURN Die() {
51 if (UserDieCallback)
52 UserDieCallback();
53 for (int i = kMaxNumOfInternalDieCallbacks - 1; i >= 0; i--) {
54 if (InternalDieCallbacks[i])
55 InternalDieCallbacks[i]();
56 }
57 if (common_flags()->abort_on_error)
58 Abort();
59 internal__exit(common_flags()->exitcode);
60 }
61
62 static CheckFailedCallbackType CheckFailedCallback;
SetCheckFailedCallback(CheckFailedCallbackType callback)63 void SetCheckFailedCallback(CheckFailedCallbackType callback) {
64 CheckFailedCallback = callback;
65 }
66
67 const int kSecondsToSleepWhenRecursiveCheckFailed = 2;
68
CheckFailed(const char * file,int line,const char * cond,u64 v1,u64 v2)69 void NORETURN CheckFailed(const char *file, int line, const char *cond,
70 u64 v1, u64 v2) {
71 static atomic_uint32_t num_calls;
72 if (atomic_fetch_add(&num_calls, 1, memory_order_relaxed) > 10) {
73 SleepForSeconds(kSecondsToSleepWhenRecursiveCheckFailed);
74 Trap();
75 }
76
77 if (CheckFailedCallback) {
78 CheckFailedCallback(file, line, cond, v1, v2);
79 }
80 Report("Sanitizer CHECK failed: %s:%d %s (%lld, %lld)\n", file, line, cond,
81 v1, v2);
82 Die();
83 }
84
85 } // namespace __sanitizer
86
87 using namespace __sanitizer;
88
89 extern "C" {
90 SANITIZER_INTERFACE_ATTRIBUTE
__sanitizer_set_death_callback(void (* callback)(void))91 void __sanitizer_set_death_callback(void (*callback)(void)) {
92 SetUserDieCallback(callback);
93 }
94 } // extern "C"
95