1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
|
//===----------------------------------------------------------------------===//
//
// 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
/// This file implements \c OutputFile class methods.
///
//===----------------------------------------------------------------------===//
#include "llvm/Support/VirtualOutputFile.h"
#include "llvm/Support/VirtualOutputError.h"
#include "llvm/Support/raw_ostream.h"
#include "llvm/Support/raw_ostream_proxy.h"
using namespace llvm;
using namespace llvm::vfs;
char OutputFileImpl::ID = 0;
char NullOutputFileImpl::ID = 0;
void OutputFileImpl::anchor() {}
void NullOutputFileImpl::anchor() {}
class OutputFile::TrackedProxy : public raw_pwrite_stream_proxy {
public:
void resetProxy() {
TrackingPointer = nullptr;
resetProxiedOS();
}
explicit TrackedProxy(TrackedProxy *&TrackingPointer, raw_pwrite_stream &OS)
: raw_pwrite_stream_proxy(OS), TrackingPointer(TrackingPointer) {
assert(!TrackingPointer && "Expected to add a proxy");
TrackingPointer = this;
}
~TrackedProxy() override { resetProxy(); }
TrackedProxy *&TrackingPointer;
};
Expected<std::unique_ptr<raw_pwrite_stream>> OutputFile::createProxy() {
if (OpenProxy)
return make_error<OutputError>(getPath(), OutputErrorCode::has_open_proxy);
return std::make_unique<TrackedProxy>(OpenProxy, getOS());
}
Error OutputFile::keep() {
// Catch double-closing logic bugs.
if (LLVM_UNLIKELY(!Impl))
report_fatal_error(
make_error<OutputError>(getPath(), OutputErrorCode::already_closed));
// Report a fatal error if there's an open proxy and the file is being kept.
// This is safer than relying on clients to remember to flush(). Also call
// OutputFile::discard() to give the backend a chance to clean up any
// side effects (such as temporaries).
if (LLVM_UNLIKELY(OpenProxy))
report_fatal_error(joinErrors(
make_error<OutputError>(getPath(), OutputErrorCode::has_open_proxy),
discard()));
Error E = Impl->keep();
Impl = nullptr;
DiscardOnDestroyHandler = nullptr;
return E;
}
Error OutputFile::discard() {
// Catch double-closing logic bugs.
if (LLVM_UNLIKELY(!Impl))
report_fatal_error(
make_error<OutputError>(getPath(), OutputErrorCode::already_closed));
// Be lenient about open proxies since client teardown paths won't
// necessarily clean up in the right order. Reset the proxy to flush any
// current content; if there is another write, there should be quick crash on
// null dereference.
if (OpenProxy)
OpenProxy->resetProxy();
Error E = Impl->discard();
Impl = nullptr;
DiscardOnDestroyHandler = nullptr;
return E;
}
void OutputFile::destroy() {
if (!Impl)
return;
// Clean up the file. Move the discard handler into a local since discard
// will reset it.
auto DiscardHandler = std::move(DiscardOnDestroyHandler);
Error E = discard();
assert(!Impl && "Expected discard to destroy Impl");
// If there's no handler, report a fatal error.
if (LLVM_UNLIKELY(!DiscardHandler))
llvm::report_fatal_error(joinErrors(
make_error<OutputError>(getPath(), OutputErrorCode::not_closed),
std::move(E)));
else if (E)
DiscardHandler(std::move(E));
}
|
