3d758e68a7
TestHelpers.h pulls few pieces from LLVM libc's unittest framework which aren't available on platforms like Fuchsia which use their own unittest framework. So, by moving FPExceptMatcher to a different file we can exclude LLVM libc specific pieces in a cleaner way. In a later pass, it might make more sense to rename TestHelpers.h also to FPMatcher.h. That way, we can make macros like EXPECT_FP_EQ to be equivalent to EXPECT_EQ on platforms like Fuchsia. Reviewed By: michaelrj Differential Revision: https://reviews.llvm.org/D107129
53 lines
1.4 KiB
C++
53 lines
1.4 KiB
C++
//===-- FPExceptMatchers.cpp ----------------------------------------------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "FPExceptMatcher.h"
|
|
|
|
#include <fenv.h>
|
|
#include <memory>
|
|
#include <setjmp.h>
|
|
#include <signal.h>
|
|
|
|
namespace __llvm_libc {
|
|
namespace fputil {
|
|
namespace testing {
|
|
|
|
#if defined(_WIN32)
|
|
#define sigjmp_buf jmp_buf
|
|
#define sigsetjmp(buf, save) setjmp(buf)
|
|
#define siglongjmp(buf, val) longjmp(buf, val)
|
|
#endif
|
|
|
|
static thread_local sigjmp_buf jumpBuffer;
|
|
static thread_local bool caughtExcept;
|
|
|
|
static void sigfpeHandler(int sig) {
|
|
caughtExcept = true;
|
|
siglongjmp(jumpBuffer, -1);
|
|
}
|
|
|
|
FPExceptMatcher::FPExceptMatcher(FunctionCaller *func) {
|
|
auto oldSIGFPEHandler = signal(SIGFPE, &sigfpeHandler);
|
|
std::unique_ptr<FunctionCaller> funcUP(func);
|
|
|
|
caughtExcept = false;
|
|
fenv_t oldEnv;
|
|
fegetenv(&oldEnv);
|
|
if (sigsetjmp(jumpBuffer, 1) == 0)
|
|
funcUP->call();
|
|
// We restore the previous floating point environment after
|
|
// the call to the function which can potentially raise SIGFPE.
|
|
fesetenv(&oldEnv);
|
|
signal(SIGFPE, oldSIGFPEHandler);
|
|
exceptionRaised = caughtExcept;
|
|
}
|
|
|
|
} // namespace testing
|
|
} // namespace fputil
|
|
} // namespace __llvm_libc
|