1 //===-- SBStream.cpp ----------------------------------------*- C++ -*-===//
2 //
3 // The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9
10 #include "lldb/API/SBStream.h"
11
12 #include "lldb/Core/StreamFile.h"
13 #include "lldb/Host/FileSystem.h"
14 #include "lldb/Utility/Status.h"
15 #include "lldb/Utility/Stream.h"
16 #include "lldb/Utility/StreamString.h"
17
18 using namespace lldb;
19 using namespace lldb_private;
20
SBStream()21 SBStream::SBStream() : m_opaque_ap(new StreamString()), m_is_file(false) {}
22
SBStream(SBStream && rhs)23 SBStream::SBStream(SBStream &&rhs)
24 : m_opaque_ap(std::move(rhs.m_opaque_ap)), m_is_file(rhs.m_is_file) {}
25
~SBStream()26 SBStream::~SBStream() {}
27
IsValid() const28 bool SBStream::IsValid() const { return (m_opaque_ap != NULL); }
29
30 // If this stream is not redirected to a file, it will maintain a local cache
31 // for the stream data which can be accessed using this accessor.
GetData()32 const char *SBStream::GetData() {
33 if (m_is_file || m_opaque_ap == NULL)
34 return NULL;
35
36 return static_cast<StreamString *>(m_opaque_ap.get())->GetData();
37 }
38
39 // If this stream is not redirected to a file, it will maintain a local cache
40 // for the stream output whose length can be accessed using this accessor.
GetSize()41 size_t SBStream::GetSize() {
42 if (m_is_file || m_opaque_ap == NULL)
43 return 0;
44
45 return static_cast<StreamString *>(m_opaque_ap.get())->GetSize();
46 }
47
Printf(const char * format,...)48 void SBStream::Printf(const char *format, ...) {
49 if (!format)
50 return;
51 va_list args;
52 va_start(args, format);
53 ref().PrintfVarArg(format, args);
54 va_end(args);
55 }
56
RedirectToFile(const char * path,bool append)57 void SBStream::RedirectToFile(const char *path, bool append) {
58 if (path == nullptr)
59 return;
60
61 std::string local_data;
62 if (m_opaque_ap) {
63 // See if we have any locally backed data. If so, copy it so we can then
64 // redirect it to the file so we don't lose the data
65 if (!m_is_file)
66 local_data = static_cast<StreamString *>(m_opaque_ap.get())->GetString();
67 }
68 StreamFile *stream_file = new StreamFile;
69 uint32_t open_options = File::eOpenOptionWrite | File::eOpenOptionCanCreate;
70 if (append)
71 open_options |= File::eOpenOptionAppend;
72 else
73 open_options |= File::eOpenOptionTruncate;
74
75 FileSystem::Instance().Open(stream_file->GetFile(), FileSpec(path),
76 open_options);
77 m_opaque_ap.reset(stream_file);
78
79 if (m_opaque_ap) {
80 m_is_file = true;
81
82 // If we had any data locally in our StreamString, then pass that along to
83 // the to new file we are redirecting to.
84 if (!local_data.empty())
85 m_opaque_ap->Write(&local_data[0], local_data.size());
86 } else
87 m_is_file = false;
88 }
89
RedirectToFileHandle(FILE * fh,bool transfer_fh_ownership)90 void SBStream::RedirectToFileHandle(FILE *fh, bool transfer_fh_ownership) {
91 if (fh == nullptr)
92 return;
93
94 std::string local_data;
95 if (m_opaque_ap) {
96 // See if we have any locally backed data. If so, copy it so we can then
97 // redirect it to the file so we don't lose the data
98 if (!m_is_file)
99 local_data = static_cast<StreamString *>(m_opaque_ap.get())->GetString();
100 }
101 m_opaque_ap.reset(new StreamFile(fh, transfer_fh_ownership));
102
103 if (m_opaque_ap) {
104 m_is_file = true;
105
106 // If we had any data locally in our StreamString, then pass that along to
107 // the to new file we are redirecting to.
108 if (!local_data.empty())
109 m_opaque_ap->Write(&local_data[0], local_data.size());
110 } else
111 m_is_file = false;
112 }
113
RedirectToFileDescriptor(int fd,bool transfer_fh_ownership)114 void SBStream::RedirectToFileDescriptor(int fd, bool transfer_fh_ownership) {
115 std::string local_data;
116 if (m_opaque_ap) {
117 // See if we have any locally backed data. If so, copy it so we can then
118 // redirect it to the file so we don't lose the data
119 if (!m_is_file)
120 local_data = static_cast<StreamString *>(m_opaque_ap.get())->GetString();
121 }
122
123 m_opaque_ap.reset(new StreamFile(::fdopen(fd, "w"), transfer_fh_ownership));
124 if (m_opaque_ap) {
125 m_is_file = true;
126
127 // If we had any data locally in our StreamString, then pass that along to
128 // the to new file we are redirecting to.
129 if (!local_data.empty())
130 m_opaque_ap->Write(&local_data[0], local_data.size());
131 } else
132 m_is_file = false;
133 }
134
operator ->()135 lldb_private::Stream *SBStream::operator->() { return m_opaque_ap.get(); }
136
get()137 lldb_private::Stream *SBStream::get() { return m_opaque_ap.get(); }
138
ref()139 lldb_private::Stream &SBStream::ref() {
140 if (m_opaque_ap == NULL)
141 m_opaque_ap.reset(new StreamString());
142 return *m_opaque_ap;
143 }
144
Clear()145 void SBStream::Clear() {
146 if (m_opaque_ap) {
147 // See if we have any locally backed data. If so, copy it so we can then
148 // redirect it to the file so we don't lose the data
149 if (m_is_file)
150 m_opaque_ap.reset();
151 else
152 static_cast<StreamString *>(m_opaque_ap.get())->Clear();
153 }
154 }
155