xref: /freebsd/contrib/llvm-project/lldb/source/API/SBCommandInterpreter.cpp (revision 700637cbb5e582861067a11aaca4d053546871d2)
1 //===-- SBCommandInterpreter.cpp ------------------------------------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 
9 #include "lldb/Utility/StructuredData.h"
10 #include "lldb/lldb-types.h"
11 
12 #include "lldb/Interpreter/CommandInterpreter.h"
13 #include "lldb/Interpreter/CommandObjectMultiword.h"
14 #include "lldb/Interpreter/CommandReturnObject.h"
15 #include "lldb/Target/Target.h"
16 #include "lldb/Utility/Instrumentation.h"
17 #include "lldb/Utility/Listener.h"
18 
19 #include "lldb/API/SBBroadcaster.h"
20 #include "lldb/API/SBCommandInterpreter.h"
21 #include "lldb/API/SBCommandInterpreterRunOptions.h"
22 #include "lldb/API/SBCommandReturnObject.h"
23 #include "lldb/API/SBEvent.h"
24 #include "lldb/API/SBExecutionContext.h"
25 #include "lldb/API/SBListener.h"
26 #include "lldb/API/SBProcess.h"
27 #include "lldb/API/SBStream.h"
28 #include "lldb/API/SBStringList.h"
29 #include "lldb/API/SBTarget.h"
30 
31 #include <memory>
32 #include <optional>
33 
34 using namespace lldb;
35 using namespace lldb_private;
36 
37 namespace lldb_private {
38 class CommandPluginInterfaceImplementation : public CommandObjectParsed {
39 public:
CommandPluginInterfaceImplementation(CommandInterpreter & interpreter,const char * name,lldb::SBCommandPluginInterface * backend,const char * help=nullptr,const char * syntax=nullptr,uint32_t flags=0,const char * auto_repeat_command="")40   CommandPluginInterfaceImplementation(CommandInterpreter &interpreter,
41                                        const char *name,
42                                        lldb::SBCommandPluginInterface *backend,
43                                        const char *help = nullptr,
44                                        const char *syntax = nullptr,
45                                        uint32_t flags = 0,
46                                        const char *auto_repeat_command = "")
47       : CommandObjectParsed(interpreter, name, help, syntax, flags),
48         m_backend(backend) {
49     m_auto_repeat_command =
50         auto_repeat_command == nullptr
51             ? std::nullopt
52             : std::optional<std::string>(auto_repeat_command);
53     // We don't know whether any given command coming from this interface takes
54     // arguments or not so here we're just disabling the basic args check.
55     CommandArgumentData none_arg{eArgTypeNone, eArgRepeatStar};
56     m_arguments.push_back({none_arg});
57   }
58 
IsRemovable() const59   bool IsRemovable() const override { return true; }
60 
61   /// More documentation is available in lldb::CommandObject::GetRepeatCommand,
62   /// but in short, if std::nullopt is returned, the previous command will be
63   /// repeated, and if an empty string is returned, no commands will be
64   /// executed.
GetRepeatCommand(Args & current_command_args,uint32_t index)65   std::optional<std::string> GetRepeatCommand(Args &current_command_args,
66                                               uint32_t index) override {
67     if (!m_auto_repeat_command)
68       return std::nullopt;
69     else
70       return m_auto_repeat_command;
71   }
72 
73 protected:
DoExecute(Args & command,CommandReturnObject & result)74   void DoExecute(Args &command, CommandReturnObject &result) override {
75     SBCommandReturnObject sb_return(result);
76     SBCommandInterpreter sb_interpreter(&m_interpreter);
77     SBDebugger debugger_sb(m_interpreter.GetDebugger().shared_from_this());
78     m_backend->DoExecute(debugger_sb, command.GetArgumentVector(), sb_return);
79   }
80   lldb::SBCommandPluginInterface *m_backend;
81   std::optional<std::string> m_auto_repeat_command;
82 };
83 } // namespace lldb_private
84 
SBCommandInterpreter()85 SBCommandInterpreter::SBCommandInterpreter() : m_opaque_ptr() {
86   LLDB_INSTRUMENT_VA(this);
87 }
88 
SBCommandInterpreter(CommandInterpreter * interpreter)89 SBCommandInterpreter::SBCommandInterpreter(CommandInterpreter *interpreter)
90     : m_opaque_ptr(interpreter) {
91   LLDB_INSTRUMENT_VA(this, interpreter);
92 }
93 
SBCommandInterpreter(const SBCommandInterpreter & rhs)94 SBCommandInterpreter::SBCommandInterpreter(const SBCommandInterpreter &rhs)
95     : m_opaque_ptr(rhs.m_opaque_ptr) {
96   LLDB_INSTRUMENT_VA(this, rhs);
97 }
98 
99 SBCommandInterpreter::~SBCommandInterpreter() = default;
100 
101 const SBCommandInterpreter &
operator =(const SBCommandInterpreter & rhs)102 SBCommandInterpreter::operator=(const SBCommandInterpreter &rhs) {
103   LLDB_INSTRUMENT_VA(this, rhs);
104 
105   m_opaque_ptr = rhs.m_opaque_ptr;
106   return *this;
107 }
108 
IsValid() const109 bool SBCommandInterpreter::IsValid() const {
110   LLDB_INSTRUMENT_VA(this);
111   return this->operator bool();
112 }
operator bool() const113 SBCommandInterpreter::operator bool() const {
114   LLDB_INSTRUMENT_VA(this);
115 
116   return m_opaque_ptr != nullptr;
117 }
118 
CommandExists(const char * cmd)119 bool SBCommandInterpreter::CommandExists(const char *cmd) {
120   LLDB_INSTRUMENT_VA(this, cmd);
121 
122   return (((cmd != nullptr) && IsValid()) ? m_opaque_ptr->CommandExists(cmd)
123                                           : false);
124 }
125 
UserCommandExists(const char * cmd)126 bool SBCommandInterpreter::UserCommandExists(const char *cmd) {
127   LLDB_INSTRUMENT_VA(this, cmd);
128 
129   return (((cmd != nullptr) && IsValid()) ? m_opaque_ptr->UserCommandExists(cmd)
130                                           : false);
131 }
132 
AliasExists(const char * cmd)133 bool SBCommandInterpreter::AliasExists(const char *cmd) {
134   LLDB_INSTRUMENT_VA(this, cmd);
135 
136   return (((cmd != nullptr) && IsValid()) ? m_opaque_ptr->AliasExists(cmd)
137                                           : false);
138 }
139 
IsActive()140 bool SBCommandInterpreter::IsActive() {
141   LLDB_INSTRUMENT_VA(this);
142 
143   return (IsValid() ? m_opaque_ptr->IsActive() : false);
144 }
145 
WasInterrupted() const146 bool SBCommandInterpreter::WasInterrupted() const {
147   LLDB_INSTRUMENT_VA(this);
148 
149   return (IsValid() ? m_opaque_ptr->GetDebugger().InterruptRequested() : false);
150 }
151 
InterruptCommand()152 bool SBCommandInterpreter::InterruptCommand() {
153   LLDB_INSTRUMENT_VA(this);
154 
155   return (IsValid() ? m_opaque_ptr->InterruptCommand() : false);
156 }
157 
GetIOHandlerControlSequence(char ch)158 const char *SBCommandInterpreter::GetIOHandlerControlSequence(char ch) {
159   LLDB_INSTRUMENT_VA(this, ch);
160 
161   if (!IsValid())
162     return nullptr;
163 
164   return ConstString(
165              m_opaque_ptr->GetDebugger().GetTopIOHandlerControlSequence(ch))
166       .GetCString();
167 }
168 
169 lldb::ReturnStatus
HandleCommand(const char * command_line,SBCommandReturnObject & result,bool add_to_history)170 SBCommandInterpreter::HandleCommand(const char *command_line,
171                                     SBCommandReturnObject &result,
172                                     bool add_to_history) {
173   LLDB_INSTRUMENT_VA(this, command_line, result, add_to_history);
174 
175   SBExecutionContext sb_exe_ctx;
176   return HandleCommand(command_line, sb_exe_ctx, result, add_to_history);
177 }
178 
HandleCommand(const char * command_line,SBExecutionContext & override_context,SBCommandReturnObject & result,bool add_to_history)179 lldb::ReturnStatus SBCommandInterpreter::HandleCommand(
180     const char *command_line, SBExecutionContext &override_context,
181     SBCommandReturnObject &result, bool add_to_history) {
182   LLDB_INSTRUMENT_VA(this, command_line, override_context, result,
183                      add_to_history);
184 
185   result.Clear();
186   if (command_line && IsValid()) {
187     result.ref().SetInteractive(false);
188     auto do_add_to_history = add_to_history ? eLazyBoolYes : eLazyBoolNo;
189     if (override_context.get())
190       m_opaque_ptr->HandleCommand(command_line, do_add_to_history,
191                                   override_context.get()->Lock(true),
192                                   result.ref());
193     else
194       m_opaque_ptr->HandleCommand(command_line, do_add_to_history,
195                                   result.ref());
196   } else {
197     result->AppendError(
198         "SBCommandInterpreter or the command line is not valid");
199   }
200 
201   return result.GetStatus();
202 }
203 
HandleCommandsFromFile(lldb::SBFileSpec & file,lldb::SBExecutionContext & override_context,lldb::SBCommandInterpreterRunOptions & options,lldb::SBCommandReturnObject result)204 void SBCommandInterpreter::HandleCommandsFromFile(
205     lldb::SBFileSpec &file, lldb::SBExecutionContext &override_context,
206     lldb::SBCommandInterpreterRunOptions &options,
207     lldb::SBCommandReturnObject result) {
208   LLDB_INSTRUMENT_VA(this, file, override_context, options, result);
209 
210   if (!IsValid()) {
211     result->AppendError("SBCommandInterpreter is not valid.");
212     return;
213   }
214 
215   if (!file.IsValid()) {
216     SBStream s;
217     file.GetDescription(s);
218     result->AppendErrorWithFormat("File is not valid: %s.", s.GetData());
219   }
220 
221   FileSpec tmp_spec = file.ref();
222   if (override_context.get())
223     m_opaque_ptr->HandleCommandsFromFile(tmp_spec,
224                                          override_context.get()->Lock(true),
225                                          options.ref(), result.ref());
226 
227   else
228     m_opaque_ptr->HandleCommandsFromFile(tmp_spec, options.ref(), result.ref());
229 }
230 
HandleCompletion(const char * current_line,const char * cursor,const char * last_char,int match_start_point,int max_return_elements,SBStringList & matches)231 int SBCommandInterpreter::HandleCompletion(
232     const char *current_line, const char *cursor, const char *last_char,
233     int match_start_point, int max_return_elements, SBStringList &matches) {
234   LLDB_INSTRUMENT_VA(this, current_line, cursor, last_char, match_start_point,
235                      max_return_elements, matches);
236 
237   SBStringList dummy_descriptions;
238   return HandleCompletionWithDescriptions(
239       current_line, cursor, last_char, match_start_point, max_return_elements,
240       matches, dummy_descriptions);
241 }
242 
HandleCompletionWithDescriptions(const char * current_line,const char * cursor,const char * last_char,int match_start_point,int max_return_elements,SBStringList & matches,SBStringList & descriptions)243 int SBCommandInterpreter::HandleCompletionWithDescriptions(
244     const char *current_line, const char *cursor, const char *last_char,
245     int match_start_point, int max_return_elements, SBStringList &matches,
246     SBStringList &descriptions) {
247   LLDB_INSTRUMENT_VA(this, current_line, cursor, last_char, match_start_point,
248                      max_return_elements, matches, descriptions);
249 
250   // Sanity check the arguments that are passed in: cursor & last_char have to
251   // be within the current_line.
252   if (current_line == nullptr || cursor == nullptr || last_char == nullptr)
253     return 0;
254 
255   if (cursor < current_line || last_char < current_line)
256     return 0;
257 
258   size_t current_line_size = strlen(current_line);
259   if (cursor - current_line > static_cast<ptrdiff_t>(current_line_size) ||
260       last_char - current_line > static_cast<ptrdiff_t>(current_line_size))
261     return 0;
262 
263   if (!IsValid())
264     return 0;
265 
266   if (max_return_elements == 0)
267     return 0;
268 
269   lldb_private::StringList lldb_matches, lldb_descriptions;
270   CompletionResult result;
271   CompletionRequest request(current_line, cursor - current_line, result);
272   if (max_return_elements > 0)
273     request.SetMaxReturnElements(max_return_elements);
274   m_opaque_ptr->HandleCompletion(request);
275   result.GetMatches(lldb_matches);
276   result.GetDescriptions(lldb_descriptions);
277 
278   // limit the matches to the max_return_elements if necessary
279   if (max_return_elements > 0 &&
280       lldb_matches.GetSize() > static_cast<size_t>(max_return_elements)) {
281     lldb_matches.SetSize(max_return_elements);
282     lldb_descriptions.SetSize(max_return_elements);
283   }
284   int number_of_matches = lldb_matches.GetSize();
285 
286   // Make the result array indexed from 1 again by adding the 'common prefix'
287   // of all completions as element 0. This is done to emulate the old API.
288   if (request.GetParsedLine().GetArgumentCount() == 0) {
289     // If we got an empty string, insert nothing.
290     lldb_matches.InsertStringAtIndex(0, "");
291     lldb_descriptions.InsertStringAtIndex(0, "");
292   } else {
293     // Now figure out if there is a common substring, and if so put that in
294     // element 0, otherwise put an empty string in element 0.
295     std::string command_partial_str = request.GetCursorArgumentPrefix().str();
296 
297     std::string common_prefix = lldb_matches.LongestCommonPrefix();
298     const size_t partial_name_len = command_partial_str.size();
299     common_prefix.erase(0, partial_name_len);
300 
301     // If we matched a unique single command, add a space... Only do this if
302     // the completer told us this was a complete word, however...
303     if (lldb_matches.GetSize() == 1) {
304       char quote_char = request.GetParsedArg().GetQuoteChar();
305       common_prefix =
306           Args::EscapeLLDBCommandArgument(common_prefix, quote_char);
307       if (request.GetParsedArg().IsQuoted())
308         common_prefix.push_back(quote_char);
309       common_prefix.push_back(' ');
310     }
311     lldb_matches.InsertStringAtIndex(0, common_prefix.c_str());
312     lldb_descriptions.InsertStringAtIndex(0, "");
313   }
314 
315   SBStringList temp_matches_list(&lldb_matches);
316   matches.AppendList(temp_matches_list);
317   SBStringList temp_descriptions_list(&lldb_descriptions);
318   descriptions.AppendList(temp_descriptions_list);
319   return number_of_matches;
320 }
321 
HandleCompletionWithDescriptions(const char * current_line,uint32_t cursor_pos,int match_start_point,int max_return_elements,SBStringList & matches,SBStringList & descriptions)322 int SBCommandInterpreter::HandleCompletionWithDescriptions(
323     const char *current_line, uint32_t cursor_pos, int match_start_point,
324     int max_return_elements, SBStringList &matches,
325     SBStringList &descriptions) {
326   LLDB_INSTRUMENT_VA(this, current_line, cursor_pos, match_start_point,
327                      max_return_elements, matches, descriptions);
328 
329   const char *cursor = current_line + cursor_pos;
330   const char *last_char = current_line + strlen(current_line);
331   return HandleCompletionWithDescriptions(
332       current_line, cursor, last_char, match_start_point, max_return_elements,
333       matches, descriptions);
334 }
335 
HandleCompletion(const char * current_line,uint32_t cursor_pos,int match_start_point,int max_return_elements,lldb::SBStringList & matches)336 int SBCommandInterpreter::HandleCompletion(const char *current_line,
337                                            uint32_t cursor_pos,
338                                            int match_start_point,
339                                            int max_return_elements,
340                                            lldb::SBStringList &matches) {
341   LLDB_INSTRUMENT_VA(this, current_line, cursor_pos, match_start_point,
342                      max_return_elements, matches);
343 
344   const char *cursor = current_line + cursor_pos;
345   const char *last_char = current_line + strlen(current_line);
346   return HandleCompletion(current_line, cursor, last_char, match_start_point,
347                           max_return_elements, matches);
348 }
349 
HasCommands()350 bool SBCommandInterpreter::HasCommands() {
351   LLDB_INSTRUMENT_VA(this);
352 
353   return (IsValid() ? m_opaque_ptr->HasCommands() : false);
354 }
355 
HasAliases()356 bool SBCommandInterpreter::HasAliases() {
357   LLDB_INSTRUMENT_VA(this);
358 
359   return (IsValid() ? m_opaque_ptr->HasAliases() : false);
360 }
361 
HasAliasOptions()362 bool SBCommandInterpreter::HasAliasOptions() {
363   LLDB_INSTRUMENT_VA(this);
364 
365   return (IsValid() ? m_opaque_ptr->HasAliasOptions() : false);
366 }
367 
IsInteractive()368 bool SBCommandInterpreter::IsInteractive() {
369   LLDB_INSTRUMENT_VA(this);
370 
371   return (IsValid() ? m_opaque_ptr->IsInteractive() : false);
372 }
373 
GetProcess()374 SBProcess SBCommandInterpreter::GetProcess() {
375   LLDB_INSTRUMENT_VA(this);
376 
377   SBProcess sb_process;
378   ProcessSP process_sp;
379   if (IsValid()) {
380     TargetSP target_sp(m_opaque_ptr->GetDebugger().GetSelectedTarget());
381     if (target_sp) {
382       std::lock_guard<std::recursive_mutex> guard(target_sp->GetAPIMutex());
383       process_sp = target_sp->GetProcessSP();
384       sb_process.SetSP(process_sp);
385     }
386   }
387 
388   return sb_process;
389 }
390 
GetDebugger()391 SBDebugger SBCommandInterpreter::GetDebugger() {
392   LLDB_INSTRUMENT_VA(this);
393 
394   SBDebugger sb_debugger;
395   if (IsValid())
396     sb_debugger.reset(m_opaque_ptr->GetDebugger().shared_from_this());
397 
398   return sb_debugger;
399 }
400 
GetPromptOnQuit()401 bool SBCommandInterpreter::GetPromptOnQuit() {
402   LLDB_INSTRUMENT_VA(this);
403 
404   return (IsValid() ? m_opaque_ptr->GetPromptOnQuit() : false);
405 }
406 
SetPromptOnQuit(bool b)407 void SBCommandInterpreter::SetPromptOnQuit(bool b) {
408   LLDB_INSTRUMENT_VA(this, b);
409 
410   if (IsValid())
411     m_opaque_ptr->SetPromptOnQuit(b);
412 }
413 
AllowExitCodeOnQuit(bool allow)414 void SBCommandInterpreter::AllowExitCodeOnQuit(bool allow) {
415   LLDB_INSTRUMENT_VA(this, allow);
416 
417   if (m_opaque_ptr)
418     m_opaque_ptr->AllowExitCodeOnQuit(allow);
419 }
420 
HasCustomQuitExitCode()421 bool SBCommandInterpreter::HasCustomQuitExitCode() {
422   LLDB_INSTRUMENT_VA(this);
423 
424   bool exited = false;
425   if (m_opaque_ptr)
426     m_opaque_ptr->GetQuitExitCode(exited);
427   return exited;
428 }
429 
GetQuitStatus()430 int SBCommandInterpreter::GetQuitStatus() {
431   LLDB_INSTRUMENT_VA(this);
432 
433   bool exited = false;
434   return (m_opaque_ptr ? m_opaque_ptr->GetQuitExitCode(exited) : 0);
435 }
436 
ResolveCommand(const char * command_line,SBCommandReturnObject & result)437 void SBCommandInterpreter::ResolveCommand(const char *command_line,
438                                           SBCommandReturnObject &result) {
439   LLDB_INSTRUMENT_VA(this, command_line, result);
440 
441   result.Clear();
442   if (command_line && IsValid()) {
443     m_opaque_ptr->ResolveCommand(command_line, result.ref());
444   } else {
445     result->AppendError(
446         "SBCommandInterpreter or the command line is not valid");
447   }
448 }
449 
get()450 CommandInterpreter *SBCommandInterpreter::get() { return m_opaque_ptr; }
451 
ref()452 CommandInterpreter &SBCommandInterpreter::ref() {
453   assert(m_opaque_ptr);
454   return *m_opaque_ptr;
455 }
456 
reset(lldb_private::CommandInterpreter * interpreter)457 void SBCommandInterpreter::reset(
458     lldb_private::CommandInterpreter *interpreter) {
459   m_opaque_ptr = interpreter;
460 }
461 
SourceInitFileInGlobalDirectory(SBCommandReturnObject & result)462 void SBCommandInterpreter::SourceInitFileInGlobalDirectory(
463     SBCommandReturnObject &result) {
464   LLDB_INSTRUMENT_VA(this, result);
465 
466   result.Clear();
467   if (IsValid()) {
468     TargetSP target_sp(m_opaque_ptr->GetDebugger().GetSelectedTarget());
469     std::unique_lock<std::recursive_mutex> lock;
470     if (target_sp)
471       lock = std::unique_lock<std::recursive_mutex>(target_sp->GetAPIMutex());
472     m_opaque_ptr->SourceInitFileGlobal(result.ref());
473   } else {
474     result->AppendError("SBCommandInterpreter is not valid");
475   }
476 }
477 
SourceInitFileInHomeDirectory(SBCommandReturnObject & result)478 void SBCommandInterpreter::SourceInitFileInHomeDirectory(
479     SBCommandReturnObject &result) {
480   LLDB_INSTRUMENT_VA(this, result);
481 
482   SourceInitFileInHomeDirectory(result, /*is_repl=*/false);
483 }
484 
SourceInitFileInHomeDirectory(SBCommandReturnObject & result,bool is_repl)485 void SBCommandInterpreter::SourceInitFileInHomeDirectory(
486     SBCommandReturnObject &result, bool is_repl) {
487   LLDB_INSTRUMENT_VA(this, result, is_repl);
488 
489   result.Clear();
490   if (IsValid()) {
491     TargetSP target_sp(m_opaque_ptr->GetDebugger().GetSelectedTarget());
492     std::unique_lock<std::recursive_mutex> lock;
493     if (target_sp)
494       lock = std::unique_lock<std::recursive_mutex>(target_sp->GetAPIMutex());
495     m_opaque_ptr->SourceInitFileHome(result.ref(), is_repl);
496   } else {
497     result->AppendError("SBCommandInterpreter is not valid");
498   }
499 }
500 
SourceInitFileInCurrentWorkingDirectory(SBCommandReturnObject & result)501 void SBCommandInterpreter::SourceInitFileInCurrentWorkingDirectory(
502     SBCommandReturnObject &result) {
503   LLDB_INSTRUMENT_VA(this, result);
504 
505   result.Clear();
506   if (IsValid()) {
507     TargetSP target_sp(m_opaque_ptr->GetDebugger().GetSelectedTarget());
508     std::unique_lock<std::recursive_mutex> lock;
509     if (target_sp)
510       lock = std::unique_lock<std::recursive_mutex>(target_sp->GetAPIMutex());
511     m_opaque_ptr->SourceInitFileCwd(result.ref());
512   } else {
513     result->AppendError("SBCommandInterpreter is not valid");
514   }
515 }
516 
GetBroadcaster()517 SBBroadcaster SBCommandInterpreter::GetBroadcaster() {
518   LLDB_INSTRUMENT_VA(this);
519 
520   SBBroadcaster broadcaster(m_opaque_ptr, false);
521 
522   return broadcaster;
523 }
524 
GetBroadcasterClass()525 const char *SBCommandInterpreter::GetBroadcasterClass() {
526   LLDB_INSTRUMENT();
527 
528   return ConstString(CommandInterpreter::GetStaticBroadcasterClass())
529       .AsCString();
530 }
531 
GetArgumentTypeAsCString(const lldb::CommandArgumentType arg_type)532 const char *SBCommandInterpreter::GetArgumentTypeAsCString(
533     const lldb::CommandArgumentType arg_type) {
534   LLDB_INSTRUMENT_VA(arg_type);
535 
536   return ConstString(CommandObject::GetArgumentTypeAsCString(arg_type))
537       .GetCString();
538 }
539 
GetArgumentDescriptionAsCString(const lldb::CommandArgumentType arg_type)540 const char *SBCommandInterpreter::GetArgumentDescriptionAsCString(
541     const lldb::CommandArgumentType arg_type) {
542   LLDB_INSTRUMENT_VA(arg_type);
543 
544   return ConstString(CommandObject::GetArgumentDescriptionAsCString(arg_type))
545       .GetCString();
546 }
547 
EventIsCommandInterpreterEvent(const lldb::SBEvent & event)548 bool SBCommandInterpreter::EventIsCommandInterpreterEvent(
549     const lldb::SBEvent &event) {
550   LLDB_INSTRUMENT_VA(event);
551 
552   return event.GetBroadcasterClass() ==
553          SBCommandInterpreter::GetBroadcasterClass();
554 }
555 
SetCommandOverrideCallback(const char * command_name,lldb::CommandOverrideCallback callback,void * baton)556 bool SBCommandInterpreter::SetCommandOverrideCallback(
557     const char *command_name, lldb::CommandOverrideCallback callback,
558     void *baton) {
559   LLDB_INSTRUMENT_VA(this, command_name, callback, baton);
560 
561   if (command_name && command_name[0] && IsValid()) {
562     llvm::StringRef command_name_str = command_name;
563     CommandObject *cmd_obj =
564         m_opaque_ptr->GetCommandObjectForCommand(command_name_str);
565     if (cmd_obj) {
566       assert(command_name_str.empty());
567       cmd_obj->SetOverrideCallback(callback, baton);
568       return true;
569     }
570   }
571   return false;
572 }
573 
GetStatistics()574 SBStructuredData SBCommandInterpreter::GetStatistics() {
575   LLDB_INSTRUMENT_VA(this);
576 
577   SBStructuredData data;
578   if (!IsValid())
579     return data;
580 
581   std::string json_str =
582       llvm::formatv("{0:2}", m_opaque_ptr->GetStatistics()).str();
583   data.m_impl_up->SetObjectSP(StructuredData::ParseJSON(json_str));
584   return data;
585 }
586 
GetTranscript()587 SBStructuredData SBCommandInterpreter::GetTranscript() {
588   LLDB_INSTRUMENT_VA(this);
589 
590   SBStructuredData data;
591   if (IsValid())
592     // A deep copy is performed by `std::make_shared` on the
593     // `StructuredData::Array`, via its implicitly-declared copy constructor.
594     // This ensures thread-safety between the user changing the returned
595     // `SBStructuredData` and the `CommandInterpreter` changing its internal
596     // `m_transcript`.
597     data.m_impl_up->SetObjectSP(
598         std::make_shared<StructuredData::Array>(m_opaque_ptr->GetTranscript()));
599   return data;
600 }
601 
AddMultiwordCommand(const char * name,const char * help)602 lldb::SBCommand SBCommandInterpreter::AddMultiwordCommand(const char *name,
603                                                           const char *help) {
604   LLDB_INSTRUMENT_VA(this, name, help);
605 
606   lldb::CommandObjectSP new_command_sp(
607       new CommandObjectMultiword(*m_opaque_ptr, name, help));
608   new_command_sp->GetAsMultiwordCommand()->SetRemovable(true);
609   Status add_error = m_opaque_ptr->AddUserCommand(name, new_command_sp, true);
610   if (add_error.Success())
611     return lldb::SBCommand(new_command_sp);
612   return lldb::SBCommand();
613 }
614 
AddCommand(const char * name,lldb::SBCommandPluginInterface * impl,const char * help)615 lldb::SBCommand SBCommandInterpreter::AddCommand(
616     const char *name, lldb::SBCommandPluginInterface *impl, const char *help) {
617   LLDB_INSTRUMENT_VA(this, name, impl, help);
618 
619   return AddCommand(name, impl, help, /*syntax=*/nullptr,
620                     /*auto_repeat_command=*/"");
621 }
622 
623 lldb::SBCommand
AddCommand(const char * name,lldb::SBCommandPluginInterface * impl,const char * help,const char * syntax)624 SBCommandInterpreter::AddCommand(const char *name,
625                                  lldb::SBCommandPluginInterface *impl,
626                                  const char *help, const char *syntax) {
627   LLDB_INSTRUMENT_VA(this, name, impl, help, syntax);
628   return AddCommand(name, impl, help, syntax, /*auto_repeat_command=*/"");
629 }
630 
AddCommand(const char * name,lldb::SBCommandPluginInterface * impl,const char * help,const char * syntax,const char * auto_repeat_command)631 lldb::SBCommand SBCommandInterpreter::AddCommand(
632     const char *name, lldb::SBCommandPluginInterface *impl, const char *help,
633     const char *syntax, const char *auto_repeat_command) {
634   LLDB_INSTRUMENT_VA(this, name, impl, help, syntax, auto_repeat_command);
635 
636   lldb::CommandObjectSP new_command_sp;
637   new_command_sp = std::make_shared<CommandPluginInterfaceImplementation>(
638       *m_opaque_ptr, name, impl, help, syntax, /*flags=*/0,
639       auto_repeat_command);
640 
641   Status add_error = m_opaque_ptr->AddUserCommand(name, new_command_sp, true);
642   if (add_error.Success())
643     return lldb::SBCommand(new_command_sp);
644   return lldb::SBCommand();
645 }
646 
SBCommand()647 SBCommand::SBCommand() { LLDB_INSTRUMENT_VA(this); }
648 
SBCommand(lldb::CommandObjectSP cmd_sp)649 SBCommand::SBCommand(lldb::CommandObjectSP cmd_sp) : m_opaque_sp(cmd_sp) {}
650 
IsValid()651 bool SBCommand::IsValid() {
652   LLDB_INSTRUMENT_VA(this);
653   return this->operator bool();
654 }
operator bool() const655 SBCommand::operator bool() const {
656   LLDB_INSTRUMENT_VA(this);
657 
658   return m_opaque_sp.get() != nullptr;
659 }
660 
GetName()661 const char *SBCommand::GetName() {
662   LLDB_INSTRUMENT_VA(this);
663 
664   return (IsValid() ? ConstString(m_opaque_sp->GetCommandName()).AsCString()
665                     : nullptr);
666 }
667 
GetHelp()668 const char *SBCommand::GetHelp() {
669   LLDB_INSTRUMENT_VA(this);
670 
671   return (IsValid() ? ConstString(m_opaque_sp->GetHelp()).AsCString()
672                     : nullptr);
673 }
674 
GetHelpLong()675 const char *SBCommand::GetHelpLong() {
676   LLDB_INSTRUMENT_VA(this);
677 
678   return (IsValid() ? ConstString(m_opaque_sp->GetHelpLong()).AsCString()
679                     : nullptr);
680 }
681 
SetHelp(const char * help)682 void SBCommand::SetHelp(const char *help) {
683   LLDB_INSTRUMENT_VA(this, help);
684 
685   if (IsValid())
686     m_opaque_sp->SetHelp(help);
687 }
688 
SetHelpLong(const char * help)689 void SBCommand::SetHelpLong(const char *help) {
690   LLDB_INSTRUMENT_VA(this, help);
691 
692   if (IsValid())
693     m_opaque_sp->SetHelpLong(help);
694 }
695 
AddMultiwordCommand(const char * name,const char * help)696 lldb::SBCommand SBCommand::AddMultiwordCommand(const char *name,
697                                                const char *help) {
698   LLDB_INSTRUMENT_VA(this, name, help);
699 
700   if (!IsValid())
701     return lldb::SBCommand();
702   if (!m_opaque_sp->IsMultiwordObject())
703     return lldb::SBCommand();
704   CommandObjectMultiword *new_command = new CommandObjectMultiword(
705       m_opaque_sp->GetCommandInterpreter(), name, help);
706   new_command->SetRemovable(true);
707   lldb::CommandObjectSP new_command_sp(new_command);
708   if (new_command_sp && m_opaque_sp->LoadSubCommand(name, new_command_sp))
709     return lldb::SBCommand(new_command_sp);
710   return lldb::SBCommand();
711 }
712 
AddCommand(const char * name,lldb::SBCommandPluginInterface * impl,const char * help)713 lldb::SBCommand SBCommand::AddCommand(const char *name,
714                                       lldb::SBCommandPluginInterface *impl,
715                                       const char *help) {
716   LLDB_INSTRUMENT_VA(this, name, impl, help);
717   return AddCommand(name, impl, help, /*syntax=*/nullptr,
718                     /*auto_repeat_command=*/"");
719 }
720 
AddCommand(const char * name,lldb::SBCommandPluginInterface * impl,const char * help,const char * syntax)721 lldb::SBCommand SBCommand::AddCommand(const char *name,
722                                       lldb::SBCommandPluginInterface *impl,
723                                       const char *help, const char *syntax) {
724   LLDB_INSTRUMENT_VA(this, name, impl, help, syntax);
725   return AddCommand(name, impl, help, syntax, /*auto_repeat_command=*/"");
726 }
727 
AddCommand(const char * name,lldb::SBCommandPluginInterface * impl,const char * help,const char * syntax,const char * auto_repeat_command)728 lldb::SBCommand SBCommand::AddCommand(const char *name,
729                                       lldb::SBCommandPluginInterface *impl,
730                                       const char *help, const char *syntax,
731                                       const char *auto_repeat_command) {
732   LLDB_INSTRUMENT_VA(this, name, impl, help, syntax, auto_repeat_command);
733 
734   if (!IsValid())
735     return lldb::SBCommand();
736   if (!m_opaque_sp->IsMultiwordObject())
737     return lldb::SBCommand();
738   lldb::CommandObjectSP new_command_sp;
739   new_command_sp = std::make_shared<CommandPluginInterfaceImplementation>(
740       m_opaque_sp->GetCommandInterpreter(), name, impl, help, syntax,
741       /*flags=*/0, auto_repeat_command);
742   if (new_command_sp && m_opaque_sp->LoadSubCommand(name, new_command_sp))
743     return lldb::SBCommand(new_command_sp);
744   return lldb::SBCommand();
745 }
746 
GetFlags()747 uint32_t SBCommand::GetFlags() {
748   LLDB_INSTRUMENT_VA(this);
749 
750   return (IsValid() ? m_opaque_sp->GetFlags().Get() : 0);
751 }
752 
SetFlags(uint32_t flags)753 void SBCommand::SetFlags(uint32_t flags) {
754   LLDB_INSTRUMENT_VA(this, flags);
755 
756   if (IsValid())
757     m_opaque_sp->GetFlags().Set(flags);
758 }
759 
SetPrintCallback(lldb::SBCommandPrintCallback callback,void * baton)760 void SBCommandInterpreter::SetPrintCallback(
761     lldb::SBCommandPrintCallback callback, void *baton) {
762   LLDB_INSTRUMENT_VA(this, callback, baton);
763 
764   if (m_opaque_ptr)
765     m_opaque_ptr->SetPrintCallback(
766         [callback, baton](lldb_private::CommandReturnObject &result) {
767           SBCommandReturnObject sb_result(result);
768           return callback(sb_result, baton);
769         });
770 }
771