1 //===-- flang/unittests/Runtime/CrashHandlerFixture.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 #include "CrashHandlerFixture.h"
9 #include "../../runtime/terminator.h"
10 #include <cstdarg>
11 #include <cstdlib>
12
13 // Replaces Fortran runtime's crash handler so we can verify the crash message
CatchCrash(const char * sourceFile,int sourceLine,const char * message,va_list & ap)14 [[noreturn]] static void CatchCrash(
15 const char *sourceFile, int sourceLine, const char *message, va_list &ap) {
16 char buffer[1000];
17 std::vsnprintf(buffer, sizeof buffer, message, ap);
18 va_end(ap);
19 llvm::errs()
20 << "Test "
21 << ::testing::UnitTest::GetInstance()->current_test_info()->name()
22 << " crashed in file "
23 << (sourceFile ? sourceFile : "unknown source file") << '(' << sourceLine
24 << "): " << buffer << '\n';
25 std::exit(EXIT_FAILURE);
26 }
27
28 // Register the crash handler above when creating each unit test in this suite
SetUp()29 void CrashHandlerFixture::SetUp() {
30 static bool isCrashHanlderRegistered{false};
31
32 if (!isCrashHanlderRegistered) {
33 Fortran::runtime::Terminator::RegisterCrashHandler(CatchCrash);
34 }
35
36 isCrashHanlderRegistered = true;
37 }
38