flang/unittests/Runtime/CrashHandlerFixture.cpp

   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
  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
  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 }