Daniel Kiss | 163101b | 2020-09-16 23:03:19 +0200 | [diff] [blame] | 1 | // -*- C++ -*- |
| 2 | //===----------------------------------------------------------------------===// |
| 3 | // |
| 4 | // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
| 5 | // See https://llvm.org/LICENSE.txt for license information. |
| 6 | // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
| 7 | // |
| 8 | //===----------------------------------------------------------------------===// |
| 9 | |
| 10 | // Ensure that leaf function can be unwund. |
Louis Dionne | ed4a85e | 2021-06-18 13:33:14 -0400 | [diff] [blame] | 11 | // REQUIRES: linux && (target={{aarch64-.+}} || target={{x86_64-.+}}) |
Daniel Kiss | 163101b | 2020-09-16 23:03:19 +0200 | [diff] [blame] | 12 | |
| 13 | #include <assert.h> |
| 14 | #include <dlfcn.h> |
| 15 | #include <signal.h> |
| 16 | #include <stdio.h> |
| 17 | #include <stdlib.h> |
| 18 | #include <string.h> |
| 19 | #include <sys/types.h> |
| 20 | #include <unistd.h> |
| 21 | #include <unwind.h> |
| 22 | |
| 23 | _Unwind_Reason_Code frame_handler(struct _Unwind_Context* ctx, void* arg) { |
| 24 | (void)arg; |
| 25 | Dl_info info = { 0, 0, 0, 0 }; |
Daniel Kiss | 163101b | 2020-09-16 23:03:19 +0200 | [diff] [blame] | 26 | |
| 27 | // Unwind util the main is reached, above frames deeped on the platfrom and architecture. |
Ryan Prichard | a684bed | 2021-01-13 16:38:36 -0800 | [diff] [blame] | 28 | if (dladdr(reinterpret_cast<void *>(_Unwind_GetIP(ctx)), &info) && |
| 29 | info.dli_sname && !strcmp("main", info.dli_sname)) { |
Daniel Kiss | 163101b | 2020-09-16 23:03:19 +0200 | [diff] [blame] | 30 | _Exit(0); |
| 31 | } |
| 32 | return _URC_NO_REASON; |
| 33 | } |
| 34 | |
| 35 | void signal_handler(int signum) { |
| 36 | (void)signum; |
| 37 | _Unwind_Backtrace(frame_handler, NULL); |
| 38 | _Exit(-1); |
| 39 | } |
| 40 | |
Daniel Kiss | 163101b | 2020-09-16 23:03:19 +0200 | [diff] [blame] | 41 | __attribute__((noinline)) void crashing_leaf_func(void) { |
Louis Dionne | f51a154 | 2021-11-22 14:51:09 -0500 | [diff] [blame] | 42 | raise(SIGSEGV); |
Daniel Kiss | 163101b | 2020-09-16 23:03:19 +0200 | [diff] [blame] | 43 | } |
| 44 | |
Louis Dionne | cbfe017 | 2020-10-08 13:36:33 -0400 | [diff] [blame] | 45 | int main(int, char**) { |
Daniel Kiss | 163101b | 2020-09-16 23:03:19 +0200 | [diff] [blame] | 46 | signal(SIGSEGV, signal_handler); |
| 47 | crashing_leaf_func(); |
| 48 | return -2; |
Louis Dionne | f51a154 | 2021-11-22 14:51:09 -0500 | [diff] [blame] | 49 | } |