1 //===-- CommandReturnObject.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/Interpreter/CommandReturnObject.h"
11 
12 // C Includes
13 // C++ Includes
14 // Other libraries and framework includes
15 // Project includes
16 #include "lldb/Core/Error.h"
17 #include "lldb/Core/StreamString.h"
18 
19 using namespace lldb;
20 using namespace lldb_private;
21 
22 static void
23 DumpStringToStreamWithNewline (Stream &strm, const std::string &s, bool add_newline_if_empty)
24 {
25     bool add_newline = false;
26     if (s.empty())
27     {
28         add_newline = add_newline_if_empty;
29     }
30     else
31     {
32         // We already checked for empty above, now make sure there is a newline
33         // in the error, and if there isn't one, add one.
34         strm.Write(s.c_str(), s.size());
35 
36         const char last_char = *s.rbegin();
37         add_newline = last_char != '\n' && last_char != '\r';
38 
39     }
40     if (add_newline)
41         strm.EOL();
42 }
43 
44 
45 CommandReturnObject::CommandReturnObject () :
46     m_out_stream (),
47     m_err_stream (),
48     m_status (eReturnStatusStarted),
49     m_did_change_process_state (false)
50 {
51 }
52 
53 CommandReturnObject::~CommandReturnObject ()
54 {
55 }
56 
57 void
58 CommandReturnObject::AppendErrorWithFormat (const char *format, ...)
59 {
60     if (!format)
61         return;
62     va_list args;
63     va_start (args, format);
64     StreamString sstrm;
65     sstrm.PrintfVarArg(format, args);
66     va_end (args);
67 
68 
69     const std::string &s = sstrm.GetString();
70     if (!s.empty())
71     {
72         Stream &error_strm = GetErrorStream();
73         error_strm.PutCString ("error: ");
74         DumpStringToStreamWithNewline (error_strm, s, false);
75     }
76 }
77 
78 void
79 CommandReturnObject::AppendMessageWithFormat (const char *format, ...)
80 {
81     if (!format)
82         return;
83     va_list args;
84     va_start (args, format);
85     StreamString sstrm;
86     sstrm.PrintfVarArg(format, args);
87     va_end (args);
88 
89     GetOutputStream().Printf("%s", sstrm.GetData());
90 }
91 
92 void
93 CommandReturnObject::AppendWarningWithFormat (const char *format, ...)
94 {
95     if (!format)
96         return;
97     va_list args;
98     va_start (args, format);
99     StreamString sstrm;
100     sstrm.PrintfVarArg(format, args);
101     va_end (args);
102 
103     GetErrorStream().Printf("warning: %s", sstrm.GetData());
104 }
105 
106 void
107 CommandReturnObject::AppendMessage (const char *in_string)
108 {
109     if (!in_string || *in_string == '\0')
110         return;
111     GetOutputStream().Printf("%s\n", in_string);
112 }
113 
114 void
115 CommandReturnObject::AppendWarning (const char *in_string)
116 {
117     if (!in_string || *in_string == '\0')
118         return;
119     GetErrorStream().Printf("warning: %s\n", in_string);
120 }
121 
122 // Similar to AppendWarning, but do not prepend 'warning: ' to message, and
123 // don't append "\n" to the end of it.
124 
125 void
126 CommandReturnObject::AppendRawWarning (const char *in_string, int len)
127 {
128     if (!in_string)
129         return;
130     if (len < 0)
131         len = ::strlen (in_string);
132     GetErrorStream().Printf("%*.*s", len, len, in_string);
133 }
134 
135 void
136 CommandReturnObject::AppendError (const char *in_string)
137 {
138     if (!in_string || *in_string == '\0')
139         return;
140     GetErrorStream().Printf ("error: %s\n", in_string);
141 }
142 
143 void
144 CommandReturnObject::SetError (const Error &error, const char *fallback_error_cstr)
145 {
146     const char *error_cstr = error.AsCString();
147     if (error_cstr == NULL)
148         error_cstr = fallback_error_cstr;
149     AppendError (error_cstr);
150     SetStatus (eReturnStatusFailed);
151 }
152 // Similar to AppendError, but do not prepend 'Error: ' to message, and
153 // don't append "\n" to the end of it.
154 
155 void
156 CommandReturnObject::AppendRawError (const char *in_string, int len)
157 {
158     if (!in_string)
159         return;
160     if (len < 0)
161         len = ::strlen (in_string);
162     GetErrorStream().Printf ("%*.*s", len, len, in_string);
163 }
164 
165 void
166 CommandReturnObject::SetStatus (ReturnStatus status)
167 {
168     m_status = status;
169 }
170 
171 ReturnStatus
172 CommandReturnObject::GetStatus ()
173 {
174     return m_status;
175 }
176 
177 bool
178 CommandReturnObject::Succeeded ()
179 {
180     return m_status <= eReturnStatusSuccessContinuingResult;
181 }
182 
183 bool
184 CommandReturnObject::HasResult ()
185 {
186     return (m_status == eReturnStatusSuccessFinishResult ||
187             m_status == eReturnStatusSuccessContinuingResult);
188 }
189 
190 void
191 CommandReturnObject::Clear()
192 {
193     lldb::StreamSP stream_sp;
194     stream_sp = m_out_stream.GetStreamAtIndex (eStreamStringIndex);
195     if (stream_sp)
196         static_cast<StreamString *>(stream_sp.get())->Clear();
197     stream_sp = m_err_stream.GetStreamAtIndex (eStreamStringIndex);
198     if (stream_sp)
199         static_cast<StreamString *>(stream_sp.get())->Clear();
200     m_status = eReturnStatusStarted;
201     m_did_change_process_state = false;
202 }
203 
204 bool
205 CommandReturnObject::GetDidChangeProcessState ()
206 {
207     return m_did_change_process_state;
208 }
209 
210 void
211 CommandReturnObject::SetDidChangeProcessState (bool b)
212 {
213     m_did_change_process_state = b;
214 }
215 
216