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, int len)
108 {
109     if (!in_string)
110         return;
111     if (len < 0)
112         len = ::strlen (in_string);
113     GetOutputStream().Printf("%*.*s\n", len, len, in_string);
114 }
115 
116 void
117 CommandReturnObject::AppendWarning (const char *in_string, int len)
118 {
119     if (!in_string)
120         return;
121     if (len < 0)
122         len = ::strlen (in_string);
123     GetErrorStream().Printf("warning: %*.*s\n", len, len, in_string);
124 }
125 
126 // Similar to AppendWarning, but do not prepend 'warning: ' to message, and
127 // don't append "\n" to the end of it.
128 
129 void
130 CommandReturnObject::AppendRawWarning (const char *in_string, int len)
131 {
132     if (!in_string)
133         return;
134     if (len < 0)
135         len = ::strlen (in_string);
136     GetErrorStream().Printf("%*.*s", len, len, in_string);
137 }
138 
139 void
140 CommandReturnObject::AppendError (const char *in_string, int len)
141 {
142     if (!in_string)
143         return;
144     if (len < 0)
145         len = ::strlen (in_string);
146     GetErrorStream().Printf ("error: %*.*s\n", len, len, in_string);
147 }
148 
149 void
150 CommandReturnObject::SetError (const Error &error, const char *fallback_error_cstr)
151 {
152     const char *error_cstr = error.AsCString();
153     if (error_cstr == NULL)
154         error_cstr = fallback_error_cstr;
155     AppendError (error_cstr);
156     SetStatus (eReturnStatusFailed);
157 }
158 // Similar to AppendError, but do not prepend 'Error: ' to message, and
159 // don't append "\n" to the end of it.
160 
161 void
162 CommandReturnObject::AppendRawError (const char *in_string, int len)
163 {
164     if (!in_string)
165         return;
166     if (len < 0)
167         len = ::strlen (in_string);
168     GetErrorStream().Printf ("%*.*s", len, len, in_string);
169 }
170 
171 void
172 CommandReturnObject::SetStatus (ReturnStatus status)
173 {
174     m_status = status;
175 }
176 
177 ReturnStatus
178 CommandReturnObject::GetStatus ()
179 {
180     return m_status;
181 }
182 
183 bool
184 CommandReturnObject::Succeeded ()
185 {
186     return m_status <= eReturnStatusSuccessContinuingResult;
187 }
188 
189 bool
190 CommandReturnObject::HasResult ()
191 {
192     return (m_status == eReturnStatusSuccessFinishResult ||
193             m_status == eReturnStatusSuccessContinuingResult);
194 }
195 
196 void
197 CommandReturnObject::Clear()
198 {
199     lldb::StreamSP stream_sp;
200     stream_sp = m_out_stream.GetStreamAtIndex (eStreamStringIndex);
201     if (stream_sp)
202         static_cast<StreamString *>(stream_sp.get())->Clear();
203     stream_sp = m_err_stream.GetStreamAtIndex (eStreamStringIndex);
204     if (stream_sp)
205         static_cast<StreamString *>(stream_sp.get())->Clear();
206     m_status = eReturnStatusStarted;
207     m_did_change_process_state = false;
208 }
209 
210 bool
211 CommandReturnObject::GetDidChangeProcessState ()
212 {
213     return m_did_change_process_state;
214 }
215 
216 void
217 CommandReturnObject::SetDidChangeProcessState (bool b)
218 {
219     m_did_change_process_state = b;
220 }
221 
222