source/API/SBStream.cpp

   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/Error.h"
  13 #include "lldb/Core/Stream.h"
  14 #include "lldb/Core/StreamFile.h"
  15 #include "lldb/Core/StreamString.h"
  16
  17 using namespace lldb;
  18 using namespace lldb_private;
  19
  20 SBStream::SBStream() : m_opaque_ap(new StreamString()), m_is_file(false) {}
  21
  22 SBStream::SBStream(SBStream &&rhs)
  23     : m_opaque_ap(std::move(rhs.m_opaque_ap)), m_is_file(rhs.m_is_file) {}
  24
  25 SBStream::~SBStream() {}
  26
  27 bool SBStream::IsValid() const { return (m_opaque_ap.get() != NULL); }
  28
  29 // If this stream is not redirected to a file, it will maintain a local
  30 // cache for the stream data which can be accessed using this accessor.
  31 const char *SBStream::GetData() {
  32   if (m_is_file || m_opaque_ap.get() == NULL)
  33     return NULL;
  34
  35   return static_cast<StreamString *>(m_opaque_ap.get())->GetData();
  36 }
  37
  38 // If this stream is not redirected to a file, it will maintain a local
  39 // cache for the stream output whose length can be accessed using this
  40 // accessor.
  41 size_t SBStream::GetSize() {
  42   if (m_is_file || m_opaque_ap.get() == NULL)
  43     return 0;
  44
  45   return static_cast<StreamString *>(m_opaque_ap.get())->GetSize();
  46 }
  47
  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
  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.get()) {
  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   stream_file->GetFile().Open(path, open_options,
  75                               lldb::eFilePermissionsFileDefault);
  76
  77   m_opaque_ap.reset(stream_file);
  78
  79   if (m_opaque_ap.get()) {
  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
  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.get()) {
  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.get()) {
 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
 114 void SBStream::RedirectToFileDescriptor(int fd, bool transfer_fh_ownership) {
 115   std::string local_data;
 116   if (m_opaque_ap.get()) {
 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.get()) {
 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
 135 lldb_private::Stream *SBStream::operator->() { return m_opaque_ap.get(); }
 136
 137 lldb_private::Stream *SBStream::get() { return m_opaque_ap.get(); }
 138
 139 lldb_private::Stream &SBStream::ref() {
 140   if (m_opaque_ap.get() == NULL)
 141     m_opaque_ap.reset(new StreamString());
 142   return *m_opaque_ap.get();
 143 }
 144
 145 void SBStream::Clear() {
 146   if (m_opaque_ap.get()) {
 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 }