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 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202
|
//===-- SBStream.cpp ----------------------------------------*- C++ -*-===//
//
// The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
#include "lldb/API/SBStream.h"
#include "lldb/Core/Error.h"
#include "lldb/Core/Stream.h"
#include "lldb/Core/StreamFile.h"
#include "lldb/Core/StreamString.h"
using namespace lldb;
using namespace lldb_private;
SBStream::SBStream () :
m_opaque_ap (new StreamString()),
m_is_file (false)
{
}
SBStream::SBStream (SBStream &&rhs) :
m_opaque_ap (std::move(rhs.m_opaque_ap)),
m_is_file (rhs.m_is_file)
{
}
SBStream::~SBStream ()
{
}
bool
SBStream::IsValid() const
{
return (m_opaque_ap.get() != NULL);
}
// If this stream is not redirected to a file, it will maintain a local
// cache for the stream data which can be accessed using this accessor.
const char *
SBStream::GetData ()
{
if (m_is_file || m_opaque_ap.get() == NULL)
return NULL;
return static_cast<StreamString *>(m_opaque_ap.get())->GetData();
}
// If this stream is not redirected to a file, it will maintain a local
// cache for the stream output whose length can be accessed using this
// accessor.
size_t
SBStream::GetSize()
{
if (m_is_file || m_opaque_ap.get() == NULL)
return 0;
return static_cast<StreamString *>(m_opaque_ap.get())->GetSize();
}
void
SBStream::Printf (const char *format, ...)
{
if (!format)
return;
va_list args;
va_start (args, format);
ref().PrintfVarArg (format, args);
va_end (args);
}
void
SBStream::RedirectToFile (const char *path, bool append)
{
if (path == nullptr)
return;
std::string local_data;
if (m_opaque_ap.get())
{
// See if we have any locally backed data. If so, copy it so we can then
// redirect it to the file so we don't lose the data
if (!m_is_file)
local_data.swap(static_cast<StreamString *>(m_opaque_ap.get())->GetString());
}
StreamFile *stream_file = new StreamFile;
uint32_t open_options = File::eOpenOptionWrite | File::eOpenOptionCanCreate;
if (append)
open_options |= File::eOpenOptionAppend;
else
open_options |= File::eOpenOptionTruncate;
stream_file->GetFile().Open (path, open_options, lldb::eFilePermissionsFileDefault);
m_opaque_ap.reset (stream_file);
if (m_opaque_ap.get())
{
m_is_file = true;
// If we had any data locally in our StreamString, then pass that along to
// the to new file we are redirecting to.
if (!local_data.empty())
m_opaque_ap->Write (&local_data[0], local_data.size());
}
else
m_is_file = false;
}
void
SBStream::RedirectToFileHandle (FILE *fh, bool transfer_fh_ownership)
{
if (fh == nullptr)
return;
std::string local_data;
if (m_opaque_ap.get())
{
// See if we have any locally backed data. If so, copy it so we can then
// redirect it to the file so we don't lose the data
if (!m_is_file)
local_data.swap(static_cast<StreamString *>(m_opaque_ap.get())->GetString());
}
m_opaque_ap.reset (new StreamFile (fh, transfer_fh_ownership));
if (m_opaque_ap.get())
{
m_is_file = true;
// If we had any data locally in our StreamString, then pass that along to
// the to new file we are redirecting to.
if (!local_data.empty())
m_opaque_ap->Write (&local_data[0], local_data.size());
}
else
m_is_file = false;
}
void
SBStream::RedirectToFileDescriptor (int fd, bool transfer_fh_ownership)
{
std::string local_data;
if (m_opaque_ap.get())
{
// See if we have any locally backed data. If so, copy it so we can then
// redirect it to the file so we don't lose the data
if (!m_is_file)
local_data.swap(static_cast<StreamString *>(m_opaque_ap.get())->GetString());
}
m_opaque_ap.reset (new StreamFile (::fdopen (fd, "w"), transfer_fh_ownership));
if (m_opaque_ap.get())
{
m_is_file = true;
// If we had any data locally in our StreamString, then pass that along to
// the to new file we are redirecting to.
if (!local_data.empty())
m_opaque_ap->Write (&local_data[0], local_data.size());
}
else
m_is_file = false;
}
lldb_private::Stream *
SBStream::operator->()
{
return m_opaque_ap.get();
}
lldb_private::Stream *
SBStream::get()
{
return m_opaque_ap.get();
}
lldb_private::Stream &
SBStream::ref()
{
if (m_opaque_ap.get() == NULL)
m_opaque_ap.reset (new StreamString());
return *m_opaque_ap.get();
}
void
SBStream::Clear ()
{
if (m_opaque_ap.get())
{
// See if we have any locally backed data. If so, copy it so we can then
// redirect it to the file so we don't lose the data
if (m_is_file)
m_opaque_ap.reset();
else
static_cast<StreamString *>(m_opaque_ap.get())->GetString().clear();
}
}
|