llvm-project/flang/runtime/stop.cpp

128 lines
3.4 KiB
C++

//===-- runtime/stop.cpp ----------------------------------------*- C++ -*-===//
//
// 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 "stop.h"
#include "file.h"
#include "io-error.h"
#include "terminator.h"
#include "unit.h"
#include <cfenv>
#include <cstdio>
#include <cstdlib>
extern "C" {
static void DescribeIEEESignaledExceptions() {
#ifdef fetestexcept // a macro in some environments; omit std::
auto excepts{fetestexcept(FE_ALL_EXCEPT)};
#else
auto excepts{std::fetestexcept(FE_ALL_EXCEPT)};
#endif
if (excepts) {
std::fputs("IEEE arithmetic exceptions signaled:", stderr);
if (excepts & FE_DIVBYZERO) {
std::fputs(" DIVBYZERO", stderr);
}
if (excepts & FE_INEXACT) {
std::fputs(" INEXACT", stderr);
}
if (excepts & FE_INVALID) {
std::fputs(" INVALID", stderr);
}
if (excepts & FE_OVERFLOW) {
std::fputs(" OVERFLOW", stderr);
}
if (excepts & FE_UNDERFLOW) {
std::fputs(" UNDERFLOW", stderr);
}
std::fputc('\n', stderr);
}
}
static void CloseAllExternalUnits(const char *why) {
Fortran::runtime::io::IoErrorHandler handler{why};
Fortran::runtime::io::ExternalFileUnit::CloseAll(handler);
}
[[noreturn]] void RTNAME(StopStatement)(
int code, bool isErrorStop, bool quiet) {
CloseAllExternalUnits("STOP statement");
if (!quiet) {
std::fprintf(stderr, "Fortran %s", isErrorStop ? "ERROR STOP" : "STOP");
if (code != EXIT_SUCCESS) {
std::fprintf(stderr, ": code %d\n", code);
}
std::fputc('\n', stderr);
DescribeIEEESignaledExceptions();
}
std::exit(code);
}
[[noreturn]] void RTNAME(StopStatementText)(
const char *code, std::size_t length, bool isErrorStop, bool quiet) {
CloseAllExternalUnits("STOP statement");
if (!quiet) {
std::fprintf(stderr, "Fortran %s: %.*s\n",
isErrorStop ? "ERROR STOP" : "STOP", static_cast<int>(length), code);
DescribeIEEESignaledExceptions();
}
std::exit(EXIT_FAILURE);
}
static bool StartPause() {
if (Fortran::runtime::io::IsATerminal(0)) {
Fortran::runtime::io::IoErrorHandler handler{"PAUSE statement"};
Fortran::runtime::io::ExternalFileUnit::FlushAll(handler);
return true;
}
return false;
}
static void EndPause() {
std::fflush(nullptr);
if (std::fgetc(stdin) == EOF) {
CloseAllExternalUnits("PAUSE statement");
std::exit(EXIT_SUCCESS);
}
}
void RTNAME(PauseStatement)() {
if (StartPause()) {
std::fputs("Fortran PAUSE: hit RETURN to continue:", stderr);
EndPause();
}
}
void RTNAME(PauseStatementInt)(int code) {
if (StartPause()) {
std::fprintf(stderr, "Fortran PAUSE %d: hit RETURN to continue:", code);
EndPause();
}
}
void RTNAME(PauseStatementText)(const char *code, std::size_t length) {
if (StartPause()) {
std::fprintf(stderr,
"Fortran PAUSE %.*s: hit RETURN to continue:", static_cast<int>(length),
code);
EndPause();
}
}
[[noreturn]] void RTNAME(FailImageStatement)() {
Fortran::runtime::NotifyOtherImagesOfFailImageStatement();
CloseAllExternalUnits("FAIL IMAGE statement");
std::exit(EXIT_FAILURE);
}
[[noreturn]] void RTNAME(ProgramEndStatement)() {
CloseAllExternalUnits("END statement");
std::exit(EXIT_SUCCESS);
}
}