//===----------------------------------------------------------------------===//
//
// 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
//
//===----------------------------------------------------------------------===//
///
/// \file
/// FPExceptMatchers.cpp.
///
//===----------------------------------------------------------------------===//

#include "FPExceptMatcher.h"

#include "src/__support/macros/config.h"
#include "test/UnitTest/Test.h"

#include "hdr/types/fenv_t.h"
#include "src/__support/FPUtil/FEnvImpl.h"
#include <setjmp.h>
#if TARGET_SUPPORTS_SIGNAL_CATCHING
#include <signal.h>
#endif

#if LIBC_TEST_HAS_MATCHERS()

namespace LIBC_NAMESPACE_DECL {
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 bool caughtExcept;

#if TARGET_SUPPORTS_SIGNAL_CATCHING

static thread_local sigjmp_buf jumpBuffer;

static void sigfpeHandler([[maybe_unused]] int sig) {
  caughtExcept = true;
  siglongjmp(jumpBuffer, -1);
}

#endif // TARGET_SUPPORTS_SIGNAL_CATCHING

FPExceptMatcher::FPExceptMatcher(cpp::function_ref<void(void)> func) {
#if TARGET_SUPPORTS_SIGNAL_CATCHING
  auto *oldSIGFPEHandler = signal(SIGFPE, &sigfpeHandler);
#endif

  caughtExcept = false;
  fenv_t oldEnv;
  fputil::get_env(&oldEnv);
#if TARGET_SUPPORTS_SIGNAL_CATCHING
  if (sigsetjmp(jumpBuffer, 1) == 0)
#endif
    func();
  // We restore the previous floating point environment after
  // the call to the function which can potentially raise SIGFPE.
  fputil::set_env(&oldEnv);
#if TARGET_SUPPORTS_SIGNAL_CATCHING
  signal(SIGFPE, oldSIGFPEHandler);
#endif
  exceptionRaised = caughtExcept;
}

} // namespace testing
} // namespace LIBC_NAMESPACE_DECL

#endif // LIBC_TEST_HAS_MATCHERS()