2021-09-02 16:14:01 +08:00
|
|
|
//===-- runtime/unit.cpp --------------------------------------------------===//
|
2020-01-24 08:59:27 +08:00
|
|
|
//
|
|
|
|
// 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 "unit.h"
|
2020-07-22 08:37:35 +08:00
|
|
|
#include "environment.h"
|
2020-02-14 06:41:56 +08:00
|
|
|
#include "io-error.h"
|
2020-01-24 08:59:27 +08:00
|
|
|
#include "lock.h"
|
2020-02-14 06:41:56 +08:00
|
|
|
#include "unit-map.h"
|
2020-07-15 02:28:03 +08:00
|
|
|
#include <cstdio>
|
2021-05-24 17:11:52 +08:00
|
|
|
#include <limits>
|
2020-07-22 08:37:35 +08:00
|
|
|
#include <utility>
|
2020-01-24 08:59:27 +08:00
|
|
|
|
|
|
|
namespace Fortran::runtime::io {
|
|
|
|
|
2020-02-14 06:41:56 +08:00
|
|
|
// The per-unit data structures are created on demand so that Fortran I/O
|
|
|
|
// should work without a Fortran main program.
|
|
|
|
static Lock unitMapLock;
|
|
|
|
static UnitMap *unitMap{nullptr};
|
2020-07-04 03:38:22 +08:00
|
|
|
static ExternalFileUnit *defaultInput{nullptr};
|
2020-02-05 08:55:45 +08:00
|
|
|
static ExternalFileUnit *defaultOutput{nullptr};
|
|
|
|
|
|
|
|
void FlushOutputOnCrash(const Terminator &terminator) {
|
2020-02-14 06:41:56 +08:00
|
|
|
if (!defaultOutput) {
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
CriticalSection critical{unitMapLock};
|
2020-02-05 08:55:45 +08:00
|
|
|
if (defaultOutput) {
|
|
|
|
IoErrorHandler handler{terminator};
|
2020-03-29 12:00:16 +08:00
|
|
|
handler.HasIoStat(); // prevent nested crash if flush has error
|
2021-06-29 02:41:04 +08:00
|
|
|
defaultOutput->FlushOutput(handler);
|
2020-02-05 08:55:45 +08:00
|
|
|
}
|
|
|
|
}
|
2020-01-24 08:59:27 +08:00
|
|
|
|
2020-02-05 08:55:45 +08:00
|
|
|
ExternalFileUnit *ExternalFileUnit::LookUp(int unit) {
|
2020-02-14 06:41:56 +08:00
|
|
|
return GetUnitMap().LookUp(unit);
|
2020-01-24 08:59:27 +08:00
|
|
|
}
|
|
|
|
|
2020-02-05 08:55:45 +08:00
|
|
|
ExternalFileUnit &ExternalFileUnit::LookUpOrCrash(
|
|
|
|
int unit, const Terminator &terminator) {
|
|
|
|
ExternalFileUnit *file{LookUp(unit)};
|
2020-01-24 08:59:27 +08:00
|
|
|
if (!file) {
|
|
|
|
terminator.Crash("Not an open I/O unit number: %d", unit);
|
|
|
|
}
|
|
|
|
return *file;
|
|
|
|
}
|
|
|
|
|
2020-02-14 06:41:56 +08:00
|
|
|
ExternalFileUnit &ExternalFileUnit::LookUpOrCreate(
|
2020-07-15 02:28:03 +08:00
|
|
|
int unit, const Terminator &terminator, bool &wasExtant) {
|
2020-02-14 06:41:56 +08:00
|
|
|
return GetUnitMap().LookUpOrCreate(unit, terminator, wasExtant);
|
2020-01-24 08:59:27 +08:00
|
|
|
}
|
|
|
|
|
2021-05-06 02:33:00 +08:00
|
|
|
ExternalFileUnit &ExternalFileUnit::LookUpOrCreateAnonymous(int unit,
|
|
|
|
Direction dir, std::optional<bool> isUnformatted,
|
|
|
|
const Terminator &terminator) {
|
2020-07-15 02:28:03 +08:00
|
|
|
bool exists{false};
|
|
|
|
ExternalFileUnit &result{
|
|
|
|
GetUnitMap().LookUpOrCreate(unit, terminator, exists)};
|
|
|
|
if (!exists) {
|
|
|
|
IoErrorHandler handler{terminator};
|
2020-08-04 02:35:29 +08:00
|
|
|
result.OpenAnonymousUnit(
|
|
|
|
dir == Direction::Input ? OpenStatus::Unknown : OpenStatus::Replace,
|
|
|
|
Action::ReadWrite, Position::Rewind, Convert::Native, handler);
|
2020-07-15 02:28:03 +08:00
|
|
|
result.isUnformatted = isUnformatted;
|
|
|
|
}
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
2020-08-04 02:35:29 +08:00
|
|
|
ExternalFileUnit *ExternalFileUnit::LookUp(const char *path) {
|
|
|
|
return GetUnitMap().LookUp(path);
|
|
|
|
}
|
|
|
|
|
2020-07-15 02:28:03 +08:00
|
|
|
ExternalFileUnit &ExternalFileUnit::CreateNew(
|
|
|
|
int unit, const Terminator &terminator) {
|
|
|
|
bool wasExtant{false};
|
|
|
|
ExternalFileUnit &result{
|
|
|
|
GetUnitMap().LookUpOrCreate(unit, terminator, wasExtant)};
|
|
|
|
RUNTIME_CHECK(terminator, !wasExtant);
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
2020-02-14 06:41:56 +08:00
|
|
|
ExternalFileUnit *ExternalFileUnit::LookUpForClose(int unit) {
|
|
|
|
return GetUnitMap().LookUpForClose(unit);
|
|
|
|
}
|
|
|
|
|
2021-06-26 01:40:08 +08:00
|
|
|
ExternalFileUnit &ExternalFileUnit::NewUnit(
|
|
|
|
const Terminator &terminator, bool forChildIo) {
|
|
|
|
ExternalFileUnit &unit{GetUnitMap().NewUnit(terminator)};
|
|
|
|
unit.createdForInternalChildIo_ = forChildIo;
|
|
|
|
return unit;
|
2020-02-05 08:55:45 +08:00
|
|
|
}
|
|
|
|
|
2021-04-13 08:05:05 +08:00
|
|
|
void ExternalFileUnit::OpenUnit(std::optional<OpenStatus> status,
|
|
|
|
std::optional<Action> action, Position position, OwningPtr<char> &&newPath,
|
|
|
|
std::size_t newPathLength, Convert convert, IoErrorHandler &handler) {
|
2020-07-22 08:37:35 +08:00
|
|
|
if (executionEnvironment.conversion != Convert::Unknown) {
|
|
|
|
convert = executionEnvironment.conversion;
|
|
|
|
}
|
|
|
|
swapEndianness_ = convert == Convert::Swap ||
|
|
|
|
(convert == Convert::LittleEndian && !isHostLittleEndian) ||
|
|
|
|
(convert == Convert::BigEndian && isHostLittleEndian);
|
2020-02-05 08:55:45 +08:00
|
|
|
if (IsOpen()) {
|
2021-04-13 08:05:05 +08:00
|
|
|
bool isSamePath{newPath.get() && path() && pathLength() == newPathLength &&
|
|
|
|
std::memcmp(path(), newPath.get(), newPathLength) == 0};
|
|
|
|
if (status && *status != OpenStatus::Old && isSamePath) {
|
|
|
|
handler.SignalError("OPEN statement for connected unit may not have "
|
|
|
|
"explicit STATUS= other than 'OLD'");
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
if (!newPath.get() || isSamePath) {
|
|
|
|
// OPEN of existing unit, STATUS='OLD' or unspecified, not new FILE=
|
2020-02-05 08:55:45 +08:00
|
|
|
newPath.reset();
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
// Otherwise, OPEN on open unit with new FILE= implies CLOSE
|
2020-07-04 03:38:22 +08:00
|
|
|
DoImpliedEndfile(handler);
|
2021-06-29 02:41:04 +08:00
|
|
|
FlushOutput(handler);
|
2020-02-05 08:55:45 +08:00
|
|
|
Close(CloseStatus::Keep, handler);
|
|
|
|
}
|
|
|
|
set_path(std::move(newPath), newPathLength);
|
2021-04-13 08:05:05 +08:00
|
|
|
Open(status.value_or(OpenStatus::Unknown), action, position, handler);
|
2020-07-04 03:38:22 +08:00
|
|
|
auto totalBytes{knownSize()};
|
|
|
|
if (access == Access::Direct) {
|
|
|
|
if (!isFixedRecordLength || !recordLength) {
|
|
|
|
handler.SignalError(IostatOpenBadRecl,
|
|
|
|
"OPEN(UNIT=%d,ACCESS='DIRECT'): record length is not known",
|
|
|
|
unitNumber());
|
|
|
|
} else if (*recordLength <= 0) {
|
|
|
|
handler.SignalError(IostatOpenBadRecl,
|
|
|
|
"OPEN(UNIT=%d,ACCESS='DIRECT',RECL=%jd): record length is invalid",
|
|
|
|
unitNumber(), static_cast<std::intmax_t>(*recordLength));
|
2020-08-04 02:35:29 +08:00
|
|
|
} else if (totalBytes && (*totalBytes % *recordLength != 0)) {
|
2020-07-04 03:38:22 +08:00
|
|
|
handler.SignalError(IostatOpenBadAppend,
|
|
|
|
"OPEN(UNIT=%d,ACCESS='DIRECT',RECL=%jd): record length is not an "
|
|
|
|
"even divisor of the file size %jd",
|
|
|
|
unitNumber(), static_cast<std::intmax_t>(*recordLength),
|
|
|
|
static_cast<std::intmax_t>(*totalBytes));
|
|
|
|
}
|
|
|
|
}
|
2020-10-02 00:50:48 +08:00
|
|
|
endfileRecordNumber.reset();
|
|
|
|
currentRecordNumber = 1;
|
|
|
|
if (totalBytes && recordLength && *recordLength) {
|
|
|
|
endfileRecordNumber = 1 + (*totalBytes / *recordLength);
|
|
|
|
}
|
2020-07-04 03:38:22 +08:00
|
|
|
if (position == Position::Append) {
|
2020-10-02 00:50:48 +08:00
|
|
|
if (!endfileRecordNumber) {
|
2020-07-04 03:38:22 +08:00
|
|
|
// Fake it so that we can backspace relative from the end
|
2020-10-02 00:50:48 +08:00
|
|
|
endfileRecordNumber = std::numeric_limits<std::int64_t>::max() - 2;
|
2020-07-04 03:38:22 +08:00
|
|
|
}
|
|
|
|
currentRecordNumber = *endfileRecordNumber;
|
|
|
|
}
|
2020-02-05 08:55:45 +08:00
|
|
|
}
|
|
|
|
|
2021-04-13 08:05:05 +08:00
|
|
|
void ExternalFileUnit::OpenAnonymousUnit(std::optional<OpenStatus> status,
|
2020-08-04 02:35:29 +08:00
|
|
|
std::optional<Action> action, Position position, Convert convert,
|
|
|
|
IoErrorHandler &handler) {
|
|
|
|
// I/O to an unconnected unit reads/creates a local file, e.g. fort.7
|
|
|
|
std::size_t pathMaxLen{32};
|
|
|
|
auto path{SizedNew<char>{handler}(pathMaxLen)};
|
|
|
|
std::snprintf(path.get(), pathMaxLen, "fort.%d", unitNumber_);
|
|
|
|
OpenUnit(status, action, position, std::move(path), std::strlen(path.get()),
|
|
|
|
convert, handler);
|
|
|
|
}
|
|
|
|
|
2020-02-05 08:55:45 +08:00
|
|
|
void ExternalFileUnit::CloseUnit(CloseStatus status, IoErrorHandler &handler) {
|
2020-07-04 03:38:22 +08:00
|
|
|
DoImpliedEndfile(handler);
|
2021-06-29 02:41:04 +08:00
|
|
|
FlushOutput(handler);
|
2020-02-14 06:41:56 +08:00
|
|
|
Close(status, handler);
|
|
|
|
}
|
|
|
|
|
|
|
|
void ExternalFileUnit::DestroyClosed() {
|
2020-03-29 12:00:16 +08:00
|
|
|
GetUnitMap().DestroyClosed(*this); // destroys *this
|
2020-01-24 08:59:27 +08:00
|
|
|
}
|
|
|
|
|
2020-07-04 03:38:22 +08:00
|
|
|
bool ExternalFileUnit::SetDirection(
|
|
|
|
Direction direction, IoErrorHandler &handler) {
|
|
|
|
if (direction == Direction::Input) {
|
|
|
|
if (mayRead()) {
|
|
|
|
direction_ = Direction::Input;
|
|
|
|
return true;
|
|
|
|
} else {
|
|
|
|
handler.SignalError(IostatReadFromWriteOnly,
|
|
|
|
"READ(UNIT=%d) with ACTION='WRITE'", unitNumber());
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
if (mayWrite()) {
|
|
|
|
direction_ = Direction::Output;
|
|
|
|
return true;
|
|
|
|
} else {
|
|
|
|
handler.SignalError(IostatWriteToReadOnly,
|
|
|
|
"WRITE(UNIT=%d) with ACTION='READ'", unitNumber());
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-02-14 06:41:56 +08:00
|
|
|
UnitMap &ExternalFileUnit::GetUnitMap() {
|
|
|
|
if (unitMap) {
|
|
|
|
return *unitMap;
|
|
|
|
}
|
|
|
|
CriticalSection critical{unitMapLock};
|
|
|
|
if (unitMap) {
|
|
|
|
return *unitMap;
|
|
|
|
}
|
|
|
|
Terminator terminator{__FILE__, __LINE__};
|
2020-07-04 03:38:22 +08:00
|
|
|
IoErrorHandler handler{terminator};
|
2021-05-06 02:26:12 +08:00
|
|
|
UnitMap *newUnitMap{New<UnitMap>{terminator}().release()};
|
|
|
|
bool wasExtant{false};
|
|
|
|
ExternalFileUnit &out{newUnitMap->LookUpOrCreate(6, terminator, wasExtant)};
|
|
|
|
RUNTIME_CHECK(terminator, !wasExtant);
|
2020-01-24 08:59:27 +08:00
|
|
|
out.Predefine(1);
|
2020-07-04 03:38:22 +08:00
|
|
|
out.SetDirection(Direction::Output, handler);
|
2020-02-05 08:55:45 +08:00
|
|
|
defaultOutput = &out;
|
2021-05-06 02:26:12 +08:00
|
|
|
ExternalFileUnit &in{newUnitMap->LookUpOrCreate(5, terminator, wasExtant)};
|
|
|
|
RUNTIME_CHECK(terminator, !wasExtant);
|
2020-01-24 08:59:27 +08:00
|
|
|
in.Predefine(0);
|
2020-07-04 03:38:22 +08:00
|
|
|
in.SetDirection(Direction::Input, handler);
|
|
|
|
defaultInput = ∈
|
2020-01-24 08:59:27 +08:00
|
|
|
// TODO: Set UTF-8 mode from the environment
|
2021-05-06 02:26:12 +08:00
|
|
|
unitMap = newUnitMap;
|
2020-02-14 06:41:56 +08:00
|
|
|
return *unitMap;
|
2020-01-24 08:59:27 +08:00
|
|
|
}
|
|
|
|
|
2020-02-05 08:55:45 +08:00
|
|
|
void ExternalFileUnit::CloseAll(IoErrorHandler &handler) {
|
2020-02-14 06:41:56 +08:00
|
|
|
CriticalSection critical{unitMapLock};
|
|
|
|
if (unitMap) {
|
|
|
|
unitMap->CloseAll(handler);
|
|
|
|
FreeMemoryAndNullify(unitMap);
|
2020-01-24 08:59:27 +08:00
|
|
|
}
|
2020-02-14 06:41:56 +08:00
|
|
|
defaultOutput = nullptr;
|
2020-01-24 08:59:27 +08:00
|
|
|
}
|
|
|
|
|
2020-07-04 03:38:22 +08:00
|
|
|
void ExternalFileUnit::FlushAll(IoErrorHandler &handler) {
|
|
|
|
CriticalSection critical{unitMapLock};
|
|
|
|
if (unitMap) {
|
|
|
|
unitMap->FlushAll(handler);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-07-22 08:37:35 +08:00
|
|
|
static void SwapEndianness(
|
|
|
|
char *data, std::size_t bytes, std::size_t elementBytes) {
|
|
|
|
if (elementBytes > 1) {
|
|
|
|
auto half{elementBytes >> 1};
|
|
|
|
for (std::size_t j{0}; j + elementBytes <= bytes; j += elementBytes) {
|
|
|
|
for (std::size_t k{0}; k < half; ++k) {
|
|
|
|
std::swap(data[j + k], data[j + elementBytes - 1 - k]);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
bool ExternalFileUnit::Emit(const char *data, std::size_t bytes,
|
|
|
|
std::size_t elementBytes, IoErrorHandler &handler) {
|
2020-02-05 08:55:45 +08:00
|
|
|
auto furthestAfter{std::max(furthestPositionInRecord,
|
|
|
|
positionInRecord + static_cast<std::int64_t>(bytes))};
|
2021-08-24 04:56:22 +08:00
|
|
|
if (recordLength) {
|
|
|
|
// It is possible for recordLength to have a value now for a
|
|
|
|
// variable-length output record if the previous operation
|
|
|
|
// was a BACKSPACE.
|
|
|
|
if (!isFixedRecordLength) {
|
|
|
|
recordLength.reset();
|
|
|
|
} else if (furthestAfter > *recordLength) {
|
|
|
|
handler.SignalError(IostatRecordWriteOverrun,
|
|
|
|
"Attempt to write %zd bytes to position %jd in a fixed-size record "
|
|
|
|
"of %jd bytes",
|
|
|
|
bytes, static_cast<std::intmax_t>(positionInRecord),
|
|
|
|
static_cast<std::intmax_t>(*recordLength));
|
|
|
|
return false;
|
|
|
|
}
|
2020-02-14 06:41:56 +08:00
|
|
|
}
|
|
|
|
WriteFrame(frameOffsetInFile_, recordOffsetInFrame_ + furthestAfter, handler);
|
2020-06-19 03:19:49 +08:00
|
|
|
if (positionInRecord > furthestPositionInRecord) {
|
2020-07-04 03:38:22 +08:00
|
|
|
std::memset(Frame() + recordOffsetInFrame_ + furthestPositionInRecord, ' ',
|
2020-06-19 03:19:49 +08:00
|
|
|
positionInRecord - furthestPositionInRecord);
|
|
|
|
}
|
2020-07-22 08:37:35 +08:00
|
|
|
char *to{Frame() + recordOffsetInFrame_ + positionInRecord};
|
|
|
|
std::memcpy(to, data, bytes);
|
|
|
|
if (swapEndianness_) {
|
|
|
|
SwapEndianness(to, bytes, elementBytes);
|
|
|
|
}
|
2020-01-24 08:59:27 +08:00
|
|
|
positionInRecord += bytes;
|
|
|
|
furthestPositionInRecord = furthestAfter;
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2020-07-22 08:37:35 +08:00
|
|
|
bool ExternalFileUnit::Receive(char *data, std::size_t bytes,
|
|
|
|
std::size_t elementBytes, IoErrorHandler &handler) {
|
2020-07-04 03:38:22 +08:00
|
|
|
RUNTIME_CHECK(handler, direction_ == Direction::Input);
|
|
|
|
auto furthestAfter{std::max(furthestPositionInRecord,
|
|
|
|
positionInRecord + static_cast<std::int64_t>(bytes))};
|
|
|
|
if (furthestAfter > recordLength.value_or(furthestAfter)) {
|
|
|
|
handler.SignalError(IostatRecordReadOverrun,
|
|
|
|
"Attempt to read %zd bytes at position %jd in a record of %jd bytes",
|
|
|
|
bytes, static_cast<std::intmax_t>(positionInRecord),
|
|
|
|
static_cast<std::intmax_t>(*recordLength));
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
auto need{recordOffsetInFrame_ + furthestAfter};
|
|
|
|
auto got{ReadFrame(frameOffsetInFile_, need, handler)};
|
|
|
|
if (got >= need) {
|
|
|
|
std::memcpy(data, Frame() + recordOffsetInFrame_ + positionInRecord, bytes);
|
2020-07-22 08:37:35 +08:00
|
|
|
if (swapEndianness_) {
|
|
|
|
SwapEndianness(data, bytes, elementBytes);
|
|
|
|
}
|
2020-07-04 03:38:22 +08:00
|
|
|
positionInRecord += bytes;
|
|
|
|
furthestPositionInRecord = furthestAfter;
|
|
|
|
return true;
|
|
|
|
} else {
|
2020-10-02 00:32:48 +08:00
|
|
|
// EOF or error: can be handled & has been signaled
|
2020-07-04 03:38:22 +08:00
|
|
|
endfileRecordNumber = currentRecordNumber;
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-02-14 06:41:56 +08:00
|
|
|
std::optional<char32_t> ExternalFileUnit::GetCurrentChar(
|
|
|
|
IoErrorHandler &handler) {
|
2020-07-04 03:38:22 +08:00
|
|
|
RUNTIME_CHECK(handler, direction_ == Direction::Input);
|
|
|
|
if (const char *p{FrameNextInput(handler, 1)}) {
|
|
|
|
// TODO: UTF-8 decoding; may have to get more bytes in a loop
|
|
|
|
return *p;
|
2020-02-14 06:41:56 +08:00
|
|
|
}
|
2020-07-04 03:38:22 +08:00
|
|
|
return std::nullopt;
|
|
|
|
}
|
|
|
|
|
|
|
|
const char *ExternalFileUnit::FrameNextInput(
|
|
|
|
IoErrorHandler &handler, std::size_t bytes) {
|
2021-05-06 02:33:00 +08:00
|
|
|
RUNTIME_CHECK(handler, isUnformatted.has_value() && !*isUnformatted);
|
2020-07-04 03:38:22 +08:00
|
|
|
if (static_cast<std::int64_t>(positionInRecord + bytes) <=
|
|
|
|
recordLength.value_or(positionInRecord + bytes)) {
|
|
|
|
auto at{recordOffsetInFrame_ + positionInRecord};
|
|
|
|
auto need{static_cast<std::size_t>(at + bytes)};
|
|
|
|
auto got{ReadFrame(frameOffsetInFile_, need, handler)};
|
|
|
|
SetSequentialVariableFormattedRecordLength();
|
|
|
|
if (got >= need) {
|
|
|
|
return Frame() + at;
|
|
|
|
}
|
2020-02-14 06:41:56 +08:00
|
|
|
handler.SignalEnd();
|
2020-07-04 03:38:22 +08:00
|
|
|
endfileRecordNumber = currentRecordNumber;
|
2020-02-14 06:41:56 +08:00
|
|
|
}
|
2020-07-04 03:38:22 +08:00
|
|
|
return nullptr;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool ExternalFileUnit::SetSequentialVariableFormattedRecordLength() {
|
|
|
|
if (recordLength || access != Access::Sequential) {
|
|
|
|
return true;
|
2021-07-21 02:39:21 +08:00
|
|
|
} else if (FrameLength() > recordOffsetInFrame_) {
|
2020-07-04 03:38:22 +08:00
|
|
|
const char *record{Frame() + recordOffsetInFrame_};
|
2021-07-21 02:39:21 +08:00
|
|
|
std::size_t bytes{FrameLength() - recordOffsetInFrame_};
|
|
|
|
if (const char *nl{
|
|
|
|
reinterpret_cast<const char *>(std::memchr(record, '\n', bytes))}) {
|
2020-07-04 03:38:22 +08:00
|
|
|
recordLength = nl - record;
|
|
|
|
if (*recordLength > 0 && record[*recordLength - 1] == '\r') {
|
|
|
|
--*recordLength;
|
|
|
|
}
|
2021-08-17 04:15:01 +08:00
|
|
|
return true;
|
2020-07-04 03:38:22 +08:00
|
|
|
}
|
2020-02-14 06:41:56 +08:00
|
|
|
}
|
2021-08-17 04:15:01 +08:00
|
|
|
return false;
|
2020-02-14 06:41:56 +08:00
|
|
|
}
|
|
|
|
|
2020-02-05 08:55:45 +08:00
|
|
|
void ExternalFileUnit::SetLeftTabLimit() {
|
2020-01-24 08:59:27 +08:00
|
|
|
leftTabLimit = furthestPositionInRecord;
|
|
|
|
positionInRecord = furthestPositionInRecord;
|
|
|
|
}
|
|
|
|
|
2021-04-14 07:07:58 +08:00
|
|
|
bool ExternalFileUnit::BeginReadingRecord(IoErrorHandler &handler) {
|
2020-07-04 03:38:22 +08:00
|
|
|
RUNTIME_CHECK(handler, direction_ == Direction::Input);
|
2021-04-14 07:07:58 +08:00
|
|
|
if (!beganReadingRecord_) {
|
|
|
|
beganReadingRecord_ = true;
|
|
|
|
if (access == Access::Sequential) {
|
|
|
|
if (endfileRecordNumber && currentRecordNumber >= *endfileRecordNumber) {
|
2020-07-04 03:38:22 +08:00
|
|
|
handler.SignalEnd();
|
2021-04-14 07:07:58 +08:00
|
|
|
} else if (isFixedRecordLength) {
|
|
|
|
RUNTIME_CHECK(handler, recordLength.has_value());
|
|
|
|
auto need{
|
|
|
|
static_cast<std::size_t>(recordOffsetInFrame_ + *recordLength)};
|
|
|
|
auto got{ReadFrame(frameOffsetInFile_, need, handler)};
|
|
|
|
if (got < need) {
|
|
|
|
handler.SignalEnd();
|
|
|
|
}
|
2021-05-06 02:33:00 +08:00
|
|
|
} else {
|
|
|
|
RUNTIME_CHECK(handler, isUnformatted.has_value());
|
|
|
|
if (isUnformatted.value_or(false)) {
|
|
|
|
BeginSequentialVariableUnformattedInputRecord(handler);
|
|
|
|
} else { // formatted
|
|
|
|
BeginSequentialVariableFormattedInputRecord(handler);
|
|
|
|
}
|
2020-07-04 03:38:22 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2021-04-14 07:07:58 +08:00
|
|
|
RUNTIME_CHECK(handler,
|
|
|
|
access != Access::Sequential || recordLength.has_value() ||
|
|
|
|
handler.InError());
|
|
|
|
return !handler.InError();
|
2020-07-04 03:38:22 +08:00
|
|
|
}
|
|
|
|
|
2020-10-01 03:53:00 +08:00
|
|
|
void ExternalFileUnit::FinishReadingRecord(IoErrorHandler &handler) {
|
|
|
|
RUNTIME_CHECK(handler, direction_ == Direction::Input && beganReadingRecord_);
|
|
|
|
beganReadingRecord_ = false;
|
2021-09-10 19:07:13 +08:00
|
|
|
if (handler.InError() && handler.GetIoStat() != IostatEor) {
|
2020-10-02 00:50:48 +08:00
|
|
|
// avoid bogus crashes in END/ERR circumstances
|
|
|
|
} else if (access == Access::Sequential) {
|
2020-10-01 03:53:00 +08:00
|
|
|
RUNTIME_CHECK(handler, recordLength.has_value());
|
|
|
|
if (isFixedRecordLength) {
|
|
|
|
frameOffsetInFile_ += recordOffsetInFrame_ + *recordLength;
|
|
|
|
recordOffsetInFrame_ = 0;
|
2021-05-06 02:33:00 +08:00
|
|
|
} else {
|
|
|
|
RUNTIME_CHECK(handler, isUnformatted.has_value());
|
|
|
|
if (isUnformatted.value_or(false)) {
|
|
|
|
// Retain footer in frame for more efficient BACKSPACE
|
|
|
|
frameOffsetInFile_ += recordOffsetInFrame_ + *recordLength;
|
|
|
|
recordOffsetInFrame_ = sizeof(std::uint32_t);
|
|
|
|
recordLength.reset();
|
|
|
|
} else { // formatted
|
[flang] runtime: fix problems with I/O around EOF & delimited characters
When a WRITE overwrites an endfile record, we need to forget
that there was an endfile record. When doing a BACKSPACE
after an explicit ENDFILE statement, the position afterwards
must be upon the endfile record.
Attempts to join list-directed delimited character input across
record boundaries was due to a bad reading of the standard
and has been deleted, now that the requirements are better understood.
This problem would cause a read attempt past EOF if a delimited
character input value was at the end of a record.
It turns out that delimited list-directed (and NAMELIST) character
output is required to emit contiguous doubled instances of the
delimiter character when it appears in the output value. When
fixed-size records are being emitted, as is the case with internal
output, this is not possible when the problematic character falls
on the last position of a record. No two other Fortran compilers
do the same thing in this situation so there is no good precedent
to follow.
Because it seems least wrong, with this patch we now emit one copy
of the delimiter as the last character of the current record and
another as the first character of the next record. (The
second-least-wrong alternative might be to flag a runtime error,
but that seems harsh since it's not an explicit error in the standard,
and the output may not have to be usable later as input anyway.)
Consequently, the output is not suitable for use as list-directed or
NAMELIST input.
If a later standard were to clarify this case, this behavior will of
course change as needed to conform.
Differential Revision: https://reviews.llvm.org/D106695
2021-07-23 00:47:37 +08:00
|
|
|
if (FrameLength() > recordOffsetInFrame_ + *recordLength &&
|
|
|
|
Frame()[recordOffsetInFrame_ + *recordLength] == '\r') {
|
2021-05-06 02:33:00 +08:00
|
|
|
++recordOffsetInFrame_;
|
|
|
|
}
|
[flang] runtime: fix problems with I/O around EOF & delimited characters
When a WRITE overwrites an endfile record, we need to forget
that there was an endfile record. When doing a BACKSPACE
after an explicit ENDFILE statement, the position afterwards
must be upon the endfile record.
Attempts to join list-directed delimited character input across
record boundaries was due to a bad reading of the standard
and has been deleted, now that the requirements are better understood.
This problem would cause a read attempt past EOF if a delimited
character input value was at the end of a record.
It turns out that delimited list-directed (and NAMELIST) character
output is required to emit contiguous doubled instances of the
delimiter character when it appears in the output value. When
fixed-size records are being emitted, as is the case with internal
output, this is not possible when the problematic character falls
on the last position of a record. No two other Fortran compilers
do the same thing in this situation so there is no good precedent
to follow.
Because it seems least wrong, with this patch we now emit one copy
of the delimiter as the last character of the current record and
another as the first character of the next record. (The
second-least-wrong alternative might be to flag a runtime error,
but that seems harsh since it's not an explicit error in the standard,
and the output may not have to be usable later as input anyway.)
Consequently, the output is not suitable for use as list-directed or
NAMELIST input.
If a later standard were to clarify this case, this behavior will of
course change as needed to conform.
Differential Revision: https://reviews.llvm.org/D106695
2021-07-23 00:47:37 +08:00
|
|
|
if (FrameLength() >= recordOffsetInFrame_ &&
|
|
|
|
Frame()[recordOffsetInFrame_ + *recordLength] == '\n') {
|
2021-07-21 02:39:21 +08:00
|
|
|
++recordOffsetInFrame_;
|
|
|
|
}
|
2021-08-17 04:15:01 +08:00
|
|
|
if (!resumptionRecordNumber || mayPosition()) {
|
|
|
|
frameOffsetInFile_ += recordOffsetInFrame_ + *recordLength;
|
|
|
|
recordOffsetInFrame_ = 0;
|
|
|
|
}
|
2021-05-06 02:33:00 +08:00
|
|
|
recordLength.reset();
|
2020-10-01 03:53:00 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
++currentRecordNumber;
|
|
|
|
BeginRecord();
|
|
|
|
}
|
|
|
|
|
2020-02-05 08:55:45 +08:00
|
|
|
bool ExternalFileUnit::AdvanceRecord(IoErrorHandler &handler) {
|
2020-07-04 03:38:22 +08:00
|
|
|
if (direction_ == Direction::Input) {
|
2020-10-01 03:53:00 +08:00
|
|
|
FinishReadingRecord(handler);
|
2021-04-14 07:07:58 +08:00
|
|
|
return BeginReadingRecord(handler);
|
2020-07-04 03:38:22 +08:00
|
|
|
} else { // Direction::Output
|
2021-04-14 07:07:58 +08:00
|
|
|
bool ok{true};
|
2021-05-06 02:33:00 +08:00
|
|
|
RUNTIME_CHECK(handler, isUnformatted.has_value());
|
2020-10-02 01:59:09 +08:00
|
|
|
if (isFixedRecordLength && recordLength) {
|
|
|
|
// Pad remainder of fixed length record
|
|
|
|
if (furthestPositionInRecord < *recordLength) {
|
|
|
|
WriteFrame(
|
|
|
|
frameOffsetInFile_, recordOffsetInFrame_ + *recordLength, handler);
|
|
|
|
std::memset(Frame() + recordOffsetInFrame_ + furthestPositionInRecord,
|
2021-05-06 02:33:00 +08:00
|
|
|
isUnformatted.value_or(false) ? 0 : ' ',
|
|
|
|
*recordLength - furthestPositionInRecord);
|
2020-10-02 01:59:09 +08:00
|
|
|
}
|
|
|
|
} else {
|
|
|
|
positionInRecord = furthestPositionInRecord;
|
2021-05-06 02:33:00 +08:00
|
|
|
if (isUnformatted.value_or(false)) {
|
2020-10-02 01:59:09 +08:00
|
|
|
// Append the length of a sequential unformatted variable-length record
|
|
|
|
// as its footer, then overwrite the reserved first four bytes of the
|
|
|
|
// record with its length as its header. These four bytes were skipped
|
|
|
|
// over in BeginUnformattedIO<Output>().
|
|
|
|
// TODO: Break very large records up into subrecords with negative
|
|
|
|
// headers &/or footers
|
|
|
|
std::uint32_t length;
|
|
|
|
length = furthestPositionInRecord - sizeof length;
|
2021-05-06 02:37:49 +08:00
|
|
|
ok = ok &&
|
|
|
|
Emit(reinterpret_cast<const char *>(&length), sizeof length,
|
|
|
|
sizeof length, handler);
|
2020-10-02 01:59:09 +08:00
|
|
|
positionInRecord = 0;
|
2021-05-06 02:37:49 +08:00
|
|
|
ok = ok &&
|
|
|
|
Emit(reinterpret_cast<const char *>(&length), sizeof length,
|
|
|
|
sizeof length, handler);
|
2020-07-04 03:38:22 +08:00
|
|
|
} else {
|
2020-10-02 01:59:09 +08:00
|
|
|
// Terminate formatted variable length record
|
2021-05-06 02:37:49 +08:00
|
|
|
ok = ok && Emit("\n", 1, 1, handler); // TODO: Windows CR+LF
|
2020-02-14 06:41:56 +08:00
|
|
|
}
|
|
|
|
}
|
2021-06-29 02:41:04 +08:00
|
|
|
CommitWrites();
|
2020-07-04 03:38:22 +08:00
|
|
|
impliedEndfile_ = true;
|
2020-10-01 03:53:00 +08:00
|
|
|
++currentRecordNumber;
|
[flang] runtime: fix problems with I/O around EOF & delimited characters
When a WRITE overwrites an endfile record, we need to forget
that there was an endfile record. When doing a BACKSPACE
after an explicit ENDFILE statement, the position afterwards
must be upon the endfile record.
Attempts to join list-directed delimited character input across
record boundaries was due to a bad reading of the standard
and has been deleted, now that the requirements are better understood.
This problem would cause a read attempt past EOF if a delimited
character input value was at the end of a record.
It turns out that delimited list-directed (and NAMELIST) character
output is required to emit contiguous doubled instances of the
delimiter character when it appears in the output value. When
fixed-size records are being emitted, as is the case with internal
output, this is not possible when the problematic character falls
on the last position of a record. No two other Fortran compilers
do the same thing in this situation so there is no good precedent
to follow.
Because it seems least wrong, with this patch we now emit one copy
of the delimiter as the last character of the current record and
another as the first character of the next record. (The
second-least-wrong alternative might be to flag a runtime error,
but that seems harsh since it's not an explicit error in the standard,
and the output may not have to be usable later as input anyway.)
Consequently, the output is not suitable for use as list-directed or
NAMELIST input.
If a later standard were to clarify this case, this behavior will of
course change as needed to conform.
Differential Revision: https://reviews.llvm.org/D106695
2021-07-23 00:47:37 +08:00
|
|
|
if (endfileRecordNumber && currentRecordNumber >= *endfileRecordNumber) {
|
|
|
|
endfileRecordNumber.reset();
|
|
|
|
}
|
2021-04-14 07:07:58 +08:00
|
|
|
return ok;
|
2020-01-24 08:59:27 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-02-14 06:41:56 +08:00
|
|
|
void ExternalFileUnit::BackspaceRecord(IoErrorHandler &handler) {
|
2020-07-04 03:38:22 +08:00
|
|
|
if (access != Access::Sequential) {
|
|
|
|
handler.SignalError(IostatBackspaceNonSequential,
|
|
|
|
"BACKSPACE(UNIT=%d) on non-sequential file", unitNumber());
|
|
|
|
} else {
|
2020-10-02 00:50:48 +08:00
|
|
|
if (endfileRecordNumber && currentRecordNumber > *endfileRecordNumber) {
|
[flang] runtime: fix problems with I/O around EOF & delimited characters
When a WRITE overwrites an endfile record, we need to forget
that there was an endfile record. When doing a BACKSPACE
after an explicit ENDFILE statement, the position afterwards
must be upon the endfile record.
Attempts to join list-directed delimited character input across
record boundaries was due to a bad reading of the standard
and has been deleted, now that the requirements are better understood.
This problem would cause a read attempt past EOF if a delimited
character input value was at the end of a record.
It turns out that delimited list-directed (and NAMELIST) character
output is required to emit contiguous doubled instances of the
delimiter character when it appears in the output value. When
fixed-size records are being emitted, as is the case with internal
output, this is not possible when the problematic character falls
on the last position of a record. No two other Fortran compilers
do the same thing in this situation so there is no good precedent
to follow.
Because it seems least wrong, with this patch we now emit one copy
of the delimiter as the last character of the current record and
another as the first character of the next record. (The
second-least-wrong alternative might be to flag a runtime error,
but that seems harsh since it's not an explicit error in the standard,
and the output may not have to be usable later as input anyway.)
Consequently, the output is not suitable for use as list-directed or
NAMELIST input.
If a later standard were to clarify this case, this behavior will of
course change as needed to conform.
Differential Revision: https://reviews.llvm.org/D106695
2021-07-23 00:47:37 +08:00
|
|
|
// BACKSPACE after explicit ENDFILE
|
|
|
|
currentRecordNumber = *endfileRecordNumber;
|
2020-02-14 06:41:56 +08:00
|
|
|
} else {
|
2020-10-02 00:50:48 +08:00
|
|
|
DoImpliedEndfile(handler);
|
|
|
|
if (frameOffsetInFile_ + recordOffsetInFrame_ > 0) {
|
|
|
|
--currentRecordNumber;
|
|
|
|
if (isFixedRecordLength) {
|
|
|
|
BackspaceFixedRecord(handler);
|
|
|
|
} else {
|
2021-05-06 02:33:00 +08:00
|
|
|
RUNTIME_CHECK(handler, isUnformatted.has_value());
|
|
|
|
if (isUnformatted.value_or(false)) {
|
|
|
|
BackspaceVariableUnformattedRecord(handler);
|
|
|
|
} else {
|
|
|
|
BackspaceVariableFormattedRecord(handler);
|
|
|
|
}
|
2020-10-02 00:50:48 +08:00
|
|
|
}
|
|
|
|
}
|
2020-02-14 06:41:56 +08:00
|
|
|
}
|
2020-10-02 00:50:48 +08:00
|
|
|
BeginRecord();
|
2020-02-14 06:41:56 +08:00
|
|
|
}
|
2020-01-24 08:59:27 +08:00
|
|
|
}
|
|
|
|
|
2021-06-29 02:41:04 +08:00
|
|
|
void ExternalFileUnit::FlushOutput(IoErrorHandler &handler) {
|
|
|
|
if (!mayPosition()) {
|
|
|
|
auto frameAt{FrameAt()};
|
|
|
|
if (frameOffsetInFile_ >= frameAt &&
|
|
|
|
frameOffsetInFile_ <
|
|
|
|
static_cast<std::int64_t>(frameAt + FrameLength())) {
|
|
|
|
// A Flush() that's about to happen to a non-positionable file
|
|
|
|
// needs to advance frameOffsetInFile_ to prevent attempts at
|
|
|
|
// impossible seeks
|
|
|
|
CommitWrites();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
Flush(handler);
|
|
|
|
}
|
|
|
|
|
2020-02-05 08:55:45 +08:00
|
|
|
void ExternalFileUnit::FlushIfTerminal(IoErrorHandler &handler) {
|
|
|
|
if (isTerminal()) {
|
2021-06-29 02:41:04 +08:00
|
|
|
FlushOutput(handler);
|
2020-02-05 08:55:45 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-07-04 03:38:22 +08:00
|
|
|
void ExternalFileUnit::Endfile(IoErrorHandler &handler) {
|
|
|
|
if (access != Access::Sequential) {
|
|
|
|
handler.SignalError(IostatEndfileNonSequential,
|
|
|
|
"ENDFILE(UNIT=%d) on non-sequential file", unitNumber());
|
|
|
|
} else if (!mayWrite()) {
|
|
|
|
handler.SignalError(IostatEndfileUnwritable,
|
|
|
|
"ENDFILE(UNIT=%d) on read-only file", unitNumber());
|
2020-10-02 00:50:48 +08:00
|
|
|
} else if (endfileRecordNumber &&
|
|
|
|
currentRecordNumber > *endfileRecordNumber) {
|
|
|
|
// ENDFILE after ENDFILE
|
2020-07-04 03:38:22 +08:00
|
|
|
} else {
|
|
|
|
DoEndfile(handler);
|
[flang] runtime: fix problems with I/O around EOF & delimited characters
When a WRITE overwrites an endfile record, we need to forget
that there was an endfile record. When doing a BACKSPACE
after an explicit ENDFILE statement, the position afterwards
must be upon the endfile record.
Attempts to join list-directed delimited character input across
record boundaries was due to a bad reading of the standard
and has been deleted, now that the requirements are better understood.
This problem would cause a read attempt past EOF if a delimited
character input value was at the end of a record.
It turns out that delimited list-directed (and NAMELIST) character
output is required to emit contiguous doubled instances of the
delimiter character when it appears in the output value. When
fixed-size records are being emitted, as is the case with internal
output, this is not possible when the problematic character falls
on the last position of a record. No two other Fortran compilers
do the same thing in this situation so there is no good precedent
to follow.
Because it seems least wrong, with this patch we now emit one copy
of the delimiter as the last character of the current record and
another as the first character of the next record. (The
second-least-wrong alternative might be to flag a runtime error,
but that seems harsh since it's not an explicit error in the standard,
and the output may not have to be usable later as input anyway.)
Consequently, the output is not suitable for use as list-directed or
NAMELIST input.
If a later standard were to clarify this case, this behavior will of
course change as needed to conform.
Differential Revision: https://reviews.llvm.org/D106695
2021-07-23 00:47:37 +08:00
|
|
|
// Explicit ENDFILE leaves position *after* the endfile record
|
|
|
|
RUNTIME_CHECK(handler, endfileRecordNumber.has_value());
|
|
|
|
currentRecordNumber = *endfileRecordNumber + 1;
|
2020-07-04 03:38:22 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void ExternalFileUnit::Rewind(IoErrorHandler &handler) {
|
|
|
|
if (access == Access::Direct) {
|
|
|
|
handler.SignalError(IostatRewindNonSequential,
|
|
|
|
"REWIND(UNIT=%d) on non-sequential file", unitNumber());
|
|
|
|
} else {
|
|
|
|
DoImpliedEndfile(handler);
|
|
|
|
SetPosition(0);
|
|
|
|
currentRecordNumber = 1;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-02-05 08:55:45 +08:00
|
|
|
void ExternalFileUnit::EndIoStatement() {
|
|
|
|
io_.reset();
|
2020-01-24 08:59:27 +08:00
|
|
|
u_.emplace<std::monostate>();
|
2020-02-14 06:41:56 +08:00
|
|
|
lock_.Drop();
|
|
|
|
}
|
|
|
|
|
2020-07-04 03:38:22 +08:00
|
|
|
void ExternalFileUnit::BeginSequentialVariableUnformattedInputRecord(
|
2020-02-14 06:41:56 +08:00
|
|
|
IoErrorHandler &handler) {
|
|
|
|
std::int32_t header{0}, footer{0};
|
2020-07-04 03:38:22 +08:00
|
|
|
std::size_t need{recordOffsetInFrame_ + sizeof header};
|
|
|
|
std::size_t got{ReadFrame(frameOffsetInFile_, need, handler)};
|
2020-02-14 06:41:56 +08:00
|
|
|
// Try to emit informative errors to help debug corrupted files.
|
|
|
|
const char *error{nullptr};
|
|
|
|
if (got < need) {
|
2020-07-04 03:38:22 +08:00
|
|
|
if (got == recordOffsetInFrame_) {
|
2020-02-14 06:41:56 +08:00
|
|
|
handler.SignalEnd();
|
|
|
|
} else {
|
2020-07-04 03:38:22 +08:00
|
|
|
error = "Unformatted variable-length sequential file input failed at "
|
|
|
|
"record #%jd (file offset %jd): truncated record header";
|
2020-02-14 06:41:56 +08:00
|
|
|
}
|
|
|
|
} else {
|
2020-07-04 03:38:22 +08:00
|
|
|
std::memcpy(&header, Frame() + recordOffsetInFrame_, sizeof header);
|
|
|
|
recordLength = sizeof header + header; // does not include footer
|
|
|
|
need = recordOffsetInFrame_ + *recordLength + sizeof footer;
|
|
|
|
got = ReadFrame(frameOffsetInFile_, need, handler);
|
2020-02-14 06:41:56 +08:00
|
|
|
if (got < need) {
|
2020-07-04 03:38:22 +08:00
|
|
|
error = "Unformatted variable-length sequential file input failed at "
|
|
|
|
"record #%jd (file offset %jd): hit EOF reading record with "
|
|
|
|
"length %jd bytes";
|
2020-02-14 06:41:56 +08:00
|
|
|
} else {
|
2020-07-04 03:38:22 +08:00
|
|
|
std::memcpy(&footer, Frame() + recordOffsetInFrame_ + *recordLength,
|
|
|
|
sizeof footer);
|
2020-02-14 06:41:56 +08:00
|
|
|
if (footer != header) {
|
2020-07-04 03:38:22 +08:00
|
|
|
error = "Unformatted variable-length sequential file input failed at "
|
|
|
|
"record #%jd (file offset %jd): record header has length %jd "
|
|
|
|
"that does not match record footer (%jd)";
|
2020-02-14 06:41:56 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if (error) {
|
|
|
|
handler.SignalError(error, static_cast<std::intmax_t>(currentRecordNumber),
|
|
|
|
static_cast<std::intmax_t>(frameOffsetInFile_),
|
|
|
|
static_cast<std::intmax_t>(header), static_cast<std::intmax_t>(footer));
|
2020-07-04 03:38:22 +08:00
|
|
|
// TODO: error recovery
|
2020-02-14 06:41:56 +08:00
|
|
|
}
|
|
|
|
positionInRecord = sizeof header;
|
|
|
|
}
|
|
|
|
|
2020-07-04 03:38:22 +08:00
|
|
|
void ExternalFileUnit::BeginSequentialVariableFormattedInputRecord(
|
2020-02-14 06:41:56 +08:00
|
|
|
IoErrorHandler &handler) {
|
2020-07-04 03:38:22 +08:00
|
|
|
if (this == defaultInput && defaultOutput) {
|
2021-06-29 02:41:04 +08:00
|
|
|
defaultOutput->FlushOutput(handler);
|
2020-02-14 06:41:56 +08:00
|
|
|
}
|
2020-07-04 03:38:22 +08:00
|
|
|
std::size_t length{0};
|
|
|
|
do {
|
2021-08-17 04:15:01 +08:00
|
|
|
std::size_t need{length + 1};
|
|
|
|
length =
|
|
|
|
ReadFrame(frameOffsetInFile_, recordOffsetInFrame_ + need, handler) -
|
|
|
|
recordOffsetInFrame_;
|
2020-07-04 03:38:22 +08:00
|
|
|
if (length < need) {
|
2021-08-17 04:15:01 +08:00
|
|
|
if (length > 0) {
|
|
|
|
// final record w/o \n
|
|
|
|
recordLength = length;
|
|
|
|
} else {
|
|
|
|
handler.SignalEnd();
|
|
|
|
}
|
2020-02-14 06:41:56 +08:00
|
|
|
break;
|
|
|
|
}
|
2020-07-04 03:38:22 +08:00
|
|
|
} while (!SetSequentialVariableFormattedRecordLength());
|
|
|
|
}
|
|
|
|
|
|
|
|
void ExternalFileUnit::BackspaceFixedRecord(IoErrorHandler &handler) {
|
|
|
|
RUNTIME_CHECK(handler, recordLength.has_value());
|
|
|
|
if (frameOffsetInFile_ < *recordLength) {
|
|
|
|
handler.SignalError(IostatBackspaceAtFirstRecord);
|
|
|
|
} else {
|
|
|
|
frameOffsetInFile_ -= *recordLength;
|
2020-02-14 06:41:56 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-07-04 03:38:22 +08:00
|
|
|
void ExternalFileUnit::BackspaceVariableUnformattedRecord(
|
2020-02-14 06:41:56 +08:00
|
|
|
IoErrorHandler &handler) {
|
|
|
|
std::int32_t header{0}, footer{0};
|
2020-07-04 03:38:22 +08:00
|
|
|
auto headerBytes{static_cast<std::int64_t>(sizeof header)};
|
|
|
|
frameOffsetInFile_ += recordOffsetInFrame_;
|
|
|
|
recordOffsetInFrame_ = 0;
|
|
|
|
if (frameOffsetInFile_ <= headerBytes) {
|
|
|
|
handler.SignalError(IostatBackspaceAtFirstRecord);
|
|
|
|
return;
|
|
|
|
}
|
2020-02-14 06:41:56 +08:00
|
|
|
// Error conditions here cause crashes, not file format errors, because the
|
|
|
|
// validity of the file structure before the current record will have been
|
2020-07-04 03:38:22 +08:00
|
|
|
// checked informatively in NextSequentialVariableUnformattedInputRecord().
|
2020-02-14 06:41:56 +08:00
|
|
|
std::size_t got{
|
2020-07-04 03:38:22 +08:00
|
|
|
ReadFrame(frameOffsetInFile_ - headerBytes, headerBytes, handler)};
|
2020-02-14 06:41:56 +08:00
|
|
|
RUNTIME_CHECK(handler, got >= sizeof footer);
|
|
|
|
std::memcpy(&footer, Frame(), sizeof footer);
|
|
|
|
recordLength = footer;
|
2020-07-04 03:38:22 +08:00
|
|
|
RUNTIME_CHECK(handler, frameOffsetInFile_ >= *recordLength + 2 * headerBytes);
|
|
|
|
frameOffsetInFile_ -= *recordLength + 2 * headerBytes;
|
|
|
|
if (frameOffsetInFile_ >= headerBytes) {
|
|
|
|
frameOffsetInFile_ -= headerBytes;
|
|
|
|
recordOffsetInFrame_ = headerBytes;
|
|
|
|
}
|
|
|
|
auto need{static_cast<std::size_t>(
|
|
|
|
recordOffsetInFrame_ + sizeof header + *recordLength)};
|
|
|
|
got = ReadFrame(frameOffsetInFile_, need, handler);
|
|
|
|
RUNTIME_CHECK(handler, got >= need);
|
|
|
|
std::memcpy(&header, Frame() + recordOffsetInFrame_, sizeof header);
|
|
|
|
RUNTIME_CHECK(handler, header == *recordLength);
|
2020-02-14 06:41:56 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
// There's no portable memrchr(), unfortunately, and strrchr() would
|
|
|
|
// fail on a record with a NUL, so we have to do it the hard way.
|
|
|
|
static const char *FindLastNewline(const char *str, std::size_t length) {
|
|
|
|
for (const char *p{str + length}; p-- > str;) {
|
|
|
|
if (*p == '\n') {
|
|
|
|
return p;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return nullptr;
|
|
|
|
}
|
|
|
|
|
2020-07-04 03:38:22 +08:00
|
|
|
void ExternalFileUnit::BackspaceVariableFormattedRecord(
|
2020-02-14 06:41:56 +08:00
|
|
|
IoErrorHandler &handler) {
|
2020-07-04 03:38:22 +08:00
|
|
|
// File offset of previous record's newline
|
|
|
|
auto prevNL{
|
|
|
|
frameOffsetInFile_ + static_cast<std::int64_t>(recordOffsetInFrame_) - 1};
|
|
|
|
if (prevNL < 0) {
|
|
|
|
handler.SignalError(IostatBackspaceAtFirstRecord);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
while (true) {
|
|
|
|
if (frameOffsetInFile_ < prevNL) {
|
2020-02-14 06:41:56 +08:00
|
|
|
if (const char *p{
|
2020-07-04 03:38:22 +08:00
|
|
|
FindLastNewline(Frame(), prevNL - 1 - frameOffsetInFile_)}) {
|
2020-02-14 06:41:56 +08:00
|
|
|
recordOffsetInFrame_ = p - Frame() + 1;
|
2021-07-27 04:02:08 +08:00
|
|
|
recordLength = prevNL - (frameOffsetInFile_ + recordOffsetInFrame_);
|
2020-02-14 06:41:56 +08:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
2020-07-04 03:38:22 +08:00
|
|
|
if (frameOffsetInFile_ == 0) {
|
|
|
|
recordOffsetInFrame_ = 0;
|
2021-07-27 04:02:08 +08:00
|
|
|
recordLength = prevNL;
|
2020-07-04 03:38:22 +08:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
frameOffsetInFile_ -= std::min<std::int64_t>(frameOffsetInFile_, 1024);
|
|
|
|
auto need{static_cast<std::size_t>(prevNL + 1 - frameOffsetInFile_)};
|
|
|
|
auto got{ReadFrame(frameOffsetInFile_, need, handler)};
|
|
|
|
RUNTIME_CHECK(handler, got >= need);
|
2020-02-14 06:41:56 +08:00
|
|
|
}
|
|
|
|
RUNTIME_CHECK(handler, Frame()[recordOffsetInFrame_ + *recordLength] == '\n');
|
|
|
|
if (*recordLength > 0 &&
|
|
|
|
Frame()[recordOffsetInFrame_ + *recordLength - 1] == '\r') {
|
|
|
|
--*recordLength;
|
|
|
|
}
|
2020-01-24 08:59:27 +08:00
|
|
|
}
|
2020-07-04 03:38:22 +08:00
|
|
|
|
|
|
|
void ExternalFileUnit::DoImpliedEndfile(IoErrorHandler &handler) {
|
|
|
|
if (impliedEndfile_) {
|
|
|
|
impliedEndfile_ = false;
|
|
|
|
if (access == Access::Sequential && mayPosition()) {
|
|
|
|
DoEndfile(handler);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void ExternalFileUnit::DoEndfile(IoErrorHandler &handler) {
|
|
|
|
endfileRecordNumber = currentRecordNumber;
|
|
|
|
Truncate(frameOffsetInFile_ + recordOffsetInFrame_, handler);
|
|
|
|
BeginRecord();
|
|
|
|
impliedEndfile_ = false;
|
|
|
|
}
|
2021-06-26 01:40:08 +08:00
|
|
|
|
2021-06-29 02:41:04 +08:00
|
|
|
void ExternalFileUnit::CommitWrites() {
|
|
|
|
frameOffsetInFile_ +=
|
|
|
|
recordOffsetInFrame_ + recordLength.value_or(furthestPositionInRecord);
|
|
|
|
recordOffsetInFrame_ = 0;
|
|
|
|
BeginRecord();
|
|
|
|
}
|
|
|
|
|
2021-06-26 01:40:08 +08:00
|
|
|
ChildIo &ExternalFileUnit::PushChildIo(IoStatementState &parent) {
|
|
|
|
OwningPtr<ChildIo> current{std::move(child_)};
|
|
|
|
Terminator &terminator{parent.GetIoErrorHandler()};
|
|
|
|
OwningPtr<ChildIo> next{New<ChildIo>{terminator}(parent, std::move(current))};
|
|
|
|
child_.reset(next.release());
|
|
|
|
return *child_;
|
|
|
|
}
|
|
|
|
|
|
|
|
void ExternalFileUnit::PopChildIo(ChildIo &child) {
|
|
|
|
if (child_.get() != &child) {
|
|
|
|
child.parent().GetIoErrorHandler().Crash(
|
|
|
|
"ChildIo being popped is not top of stack");
|
|
|
|
}
|
|
|
|
child_.reset(child.AcquirePrevious().release()); // deletes top child
|
|
|
|
}
|
|
|
|
|
|
|
|
void ChildIo::EndIoStatement() {
|
|
|
|
io_.reset();
|
|
|
|
u_.emplace<std::monostate>();
|
|
|
|
}
|
|
|
|
|
|
|
|
bool ChildIo::CheckFormattingAndDirection(Terminator &terminator,
|
|
|
|
const char *what, bool unformatted, Direction direction) {
|
|
|
|
bool parentIsInput{!parent_.get_if<IoDirectionState<Direction::Output>>()};
|
2021-09-21 01:52:39 +08:00
|
|
|
bool parentIsFormatted{parentIsInput
|
|
|
|
? parent_.get_if<FormattedIoStatementState<Direction::Input>>() !=
|
|
|
|
nullptr
|
|
|
|
: parent_.get_if<FormattedIoStatementState<Direction::Output>>() !=
|
|
|
|
nullptr};
|
|
|
|
bool parentIsUnformatted{!parentIsFormatted};
|
2021-06-26 01:40:08 +08:00
|
|
|
if (unformatted != parentIsUnformatted) {
|
|
|
|
terminator.Crash("Child %s attempted on %s parent I/O unit", what,
|
|
|
|
parentIsUnformatted ? "unformatted" : "formatted");
|
|
|
|
return false;
|
|
|
|
} else if (parentIsInput != (direction == Direction::Input)) {
|
|
|
|
terminator.Crash("Child %s attempted on %s parent I/O unit", what,
|
|
|
|
parentIsInput ? "input" : "output");
|
|
|
|
return false;
|
|
|
|
} else {
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-03-29 12:00:16 +08:00
|
|
|
} // namespace Fortran::runtime::io
|