1 //===-- CommandInterpreter.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 <chrono>
10 #include <cstdlib>
11 #include <limits>
12 #include <memory>
13 #include <optional>
14 #include <string>
15 #include <vector>
16
17 #include "Commands/CommandObjectApropos.h"
18 #include "Commands/CommandObjectBreakpoint.h"
19 #include "Commands/CommandObjectCommands.h"
20 #include "Commands/CommandObjectDWIMPrint.h"
21 #include "Commands/CommandObjectDiagnostics.h"
22 #include "Commands/CommandObjectDisassemble.h"
23 #include "Commands/CommandObjectExpression.h"
24 #include "Commands/CommandObjectFrame.h"
25 #include "Commands/CommandObjectGUI.h"
26 #include "Commands/CommandObjectHelp.h"
27 #include "Commands/CommandObjectLanguage.h"
28 #include "Commands/CommandObjectLog.h"
29 #include "Commands/CommandObjectMemory.h"
30 #include "Commands/CommandObjectPlatform.h"
31 #include "Commands/CommandObjectPlugin.h"
32 #include "Commands/CommandObjectProcess.h"
33 #include "Commands/CommandObjectProtocolServer.h"
34 #include "Commands/CommandObjectQuit.h"
35 #include "Commands/CommandObjectRegexCommand.h"
36 #include "Commands/CommandObjectRegister.h"
37 #include "Commands/CommandObjectScripting.h"
38 #include "Commands/CommandObjectSession.h"
39 #include "Commands/CommandObjectSettings.h"
40 #include "Commands/CommandObjectSource.h"
41 #include "Commands/CommandObjectStats.h"
42 #include "Commands/CommandObjectTarget.h"
43 #include "Commands/CommandObjectThread.h"
44 #include "Commands/CommandObjectTrace.h"
45 #include "Commands/CommandObjectType.h"
46 #include "Commands/CommandObjectVersion.h"
47 #include "Commands/CommandObjectWatchpoint.h"
48
49 #include "lldb/Core/Debugger.h"
50 #include "lldb/Core/Module.h"
51 #include "lldb/Core/PluginManager.h"
52 #include "lldb/Core/Telemetry.h"
53 #include "lldb/Host/StreamFile.h"
54 #include "lldb/Utility/ErrorMessages.h"
55 #include "lldb/Utility/LLDBLog.h"
56 #include "lldb/Utility/Log.h"
57 #include "lldb/Utility/State.h"
58 #include "lldb/Utility/Stream.h"
59 #include "lldb/Utility/StructuredData.h"
60 #include "lldb/Utility/Timer.h"
61
62 #include "lldb/Host/Config.h"
63 #include "lldb/lldb-forward.h"
64 #if LLDB_ENABLE_LIBEDIT
65 #include "lldb/Host/Editline.h"
66 #endif
67 #include "lldb/Host/File.h"
68 #include "lldb/Host/FileCache.h"
69 #include "lldb/Host/Host.h"
70 #include "lldb/Host/HostInfo.h"
71
72 #include "lldb/Interpreter/CommandCompletions.h"
73 #include "lldb/Interpreter/CommandInterpreter.h"
74 #include "lldb/Interpreter/CommandReturnObject.h"
75 #include "lldb/Interpreter/OptionValueProperties.h"
76 #include "lldb/Interpreter/Options.h"
77 #include "lldb/Interpreter/Property.h"
78 #include "lldb/Utility/Args.h"
79
80 #include "lldb/Target/Language.h"
81 #include "lldb/Target/Process.h"
82 #include "lldb/Target/StopInfo.h"
83 #include "lldb/Target/TargetList.h"
84 #include "lldb/Target/Thread.h"
85 #include "lldb/Target/UnixSignals.h"
86
87 #include "llvm/ADT/STLExtras.h"
88 #include "llvm/ADT/ScopeExit.h"
89 #include "llvm/ADT/SmallString.h"
90 #include "llvm/Support/FormatAdapters.h"
91 #include "llvm/Support/Path.h"
92 #include "llvm/Support/PrettyStackTrace.h"
93 #include "llvm/Support/ScopedPrinter.h"
94 #include "llvm/Telemetry/Telemetry.h"
95
96 #if defined(__APPLE__)
97 #include <TargetConditionals.h>
98 #endif
99
100 using namespace lldb;
101 using namespace lldb_private;
102
103 static const char *k_white_space = " \t\v";
104
105 static constexpr const char *InitFileWarning =
106 "There is a .lldbinit file in the current directory which is not being "
107 "read.\n"
108 "To silence this warning without sourcing in the local .lldbinit,\n"
109 "add the following to the lldbinit file in your home directory:\n"
110 " settings set target.load-cwd-lldbinit false\n"
111 "To allow lldb to source .lldbinit files in the current working "
112 "directory,\n"
113 "set the value of this variable to true. Only do so if you understand "
114 "and\n"
115 "accept the security risk.";
116
117 const char *CommandInterpreter::g_no_argument = "<no-argument>";
118 const char *CommandInterpreter::g_need_argument = "<need-argument>";
119 const char *CommandInterpreter::g_argument = "<argument>";
120
121 #define LLDB_PROPERTIES_interpreter
122 #include "InterpreterProperties.inc"
123
124 enum {
125 #define LLDB_PROPERTIES_interpreter
126 #include "InterpreterPropertiesEnum.inc"
127 };
128
GetStaticBroadcasterClass()129 llvm::StringRef CommandInterpreter::GetStaticBroadcasterClass() {
130 static constexpr llvm::StringLiteral class_name("lldb.commandInterpreter");
131 return class_name;
132 }
133
CommandInterpreter(Debugger & debugger,bool synchronous_execution)134 CommandInterpreter::CommandInterpreter(Debugger &debugger,
135 bool synchronous_execution)
136 : Broadcaster(debugger.GetBroadcasterManager(),
137 CommandInterpreter::GetStaticBroadcasterClass().str()),
138 Properties(
139 OptionValuePropertiesSP(new OptionValueProperties("interpreter"))),
140 IOHandlerDelegate(IOHandlerDelegate::Completion::LLDBCommand),
141 m_debugger(debugger), m_synchronous_execution(true),
142 m_skip_lldbinit_files(false), m_skip_app_init_files(false),
143 m_comment_char('#'), m_batch_command_mode(false),
144 m_truncation_warning(eNoOmission), m_max_depth_warning(eNoOmission),
145 m_command_source_depth(0) {
146 SetEventName(eBroadcastBitThreadShouldExit, "thread-should-exit");
147 SetEventName(eBroadcastBitResetPrompt, "reset-prompt");
148 SetEventName(eBroadcastBitQuitCommandReceived, "quit");
149 SetSynchronous(synchronous_execution);
150 CheckInWithManager();
151 m_collection_sp->Initialize(g_interpreter_properties);
152 }
153
GetExpandRegexAliases() const154 bool CommandInterpreter::GetExpandRegexAliases() const {
155 const uint32_t idx = ePropertyExpandRegexAliases;
156 return GetPropertyAtIndexAs<bool>(
157 idx, g_interpreter_properties[idx].default_uint_value != 0);
158 }
159
GetPromptOnQuit() const160 bool CommandInterpreter::GetPromptOnQuit() const {
161 const uint32_t idx = ePropertyPromptOnQuit;
162 return GetPropertyAtIndexAs<bool>(
163 idx, g_interpreter_properties[idx].default_uint_value != 0);
164 }
165
SetPromptOnQuit(bool enable)166 void CommandInterpreter::SetPromptOnQuit(bool enable) {
167 const uint32_t idx = ePropertyPromptOnQuit;
168 SetPropertyAtIndex(idx, enable);
169 }
170
GetSaveTranscript() const171 bool CommandInterpreter::GetSaveTranscript() const {
172 const uint32_t idx = ePropertySaveTranscript;
173 return GetPropertyAtIndexAs<bool>(
174 idx, g_interpreter_properties[idx].default_uint_value != 0);
175 }
176
SetSaveTranscript(bool enable)177 void CommandInterpreter::SetSaveTranscript(bool enable) {
178 const uint32_t idx = ePropertySaveTranscript;
179 SetPropertyAtIndex(idx, enable);
180 }
181
GetSaveSessionOnQuit() const182 bool CommandInterpreter::GetSaveSessionOnQuit() const {
183 const uint32_t idx = ePropertySaveSessionOnQuit;
184 return GetPropertyAtIndexAs<bool>(
185 idx, g_interpreter_properties[idx].default_uint_value != 0);
186 }
187
SetSaveSessionOnQuit(bool enable)188 void CommandInterpreter::SetSaveSessionOnQuit(bool enable) {
189 const uint32_t idx = ePropertySaveSessionOnQuit;
190 SetPropertyAtIndex(idx, enable);
191 }
192
GetOpenTranscriptInEditor() const193 bool CommandInterpreter::GetOpenTranscriptInEditor() const {
194 const uint32_t idx = ePropertyOpenTranscriptInEditor;
195 return GetPropertyAtIndexAs<bool>(
196 idx, g_interpreter_properties[idx].default_uint_value != 0);
197 }
198
SetOpenTranscriptInEditor(bool enable)199 void CommandInterpreter::SetOpenTranscriptInEditor(bool enable) {
200 const uint32_t idx = ePropertyOpenTranscriptInEditor;
201 SetPropertyAtIndex(idx, enable);
202 }
203
GetSaveSessionDirectory() const204 FileSpec CommandInterpreter::GetSaveSessionDirectory() const {
205 const uint32_t idx = ePropertySaveSessionDirectory;
206 return GetPropertyAtIndexAs<FileSpec>(idx, {});
207 }
208
SetSaveSessionDirectory(llvm::StringRef path)209 void CommandInterpreter::SetSaveSessionDirectory(llvm::StringRef path) {
210 const uint32_t idx = ePropertySaveSessionDirectory;
211 SetPropertyAtIndex(idx, path);
212 }
213
GetEchoCommands() const214 bool CommandInterpreter::GetEchoCommands() const {
215 const uint32_t idx = ePropertyEchoCommands;
216 return GetPropertyAtIndexAs<bool>(
217 idx, g_interpreter_properties[idx].default_uint_value != 0);
218 }
219
SetEchoCommands(bool enable)220 void CommandInterpreter::SetEchoCommands(bool enable) {
221 const uint32_t idx = ePropertyEchoCommands;
222 SetPropertyAtIndex(idx, enable);
223 }
224
GetEchoCommentCommands() const225 bool CommandInterpreter::GetEchoCommentCommands() const {
226 const uint32_t idx = ePropertyEchoCommentCommands;
227 return GetPropertyAtIndexAs<bool>(
228 idx, g_interpreter_properties[idx].default_uint_value != 0);
229 }
230
SetEchoCommentCommands(bool enable)231 void CommandInterpreter::SetEchoCommentCommands(bool enable) {
232 const uint32_t idx = ePropertyEchoCommentCommands;
233 SetPropertyAtIndex(idx, enable);
234 }
235
AllowExitCodeOnQuit(bool allow)236 void CommandInterpreter::AllowExitCodeOnQuit(bool allow) {
237 m_allow_exit_code = allow;
238 if (!allow)
239 m_quit_exit_code.reset();
240 }
241
SetQuitExitCode(int exit_code)242 bool CommandInterpreter::SetQuitExitCode(int exit_code) {
243 if (!m_allow_exit_code)
244 return false;
245 m_quit_exit_code = exit_code;
246 return true;
247 }
248
GetQuitExitCode(bool & exited) const249 int CommandInterpreter::GetQuitExitCode(bool &exited) const {
250 exited = m_quit_exit_code.has_value();
251 if (exited)
252 return *m_quit_exit_code;
253 return 0;
254 }
255
ResolveCommand(const char * command_line,CommandReturnObject & result)256 void CommandInterpreter::ResolveCommand(const char *command_line,
257 CommandReturnObject &result) {
258 std::string command = command_line;
259 if (ResolveCommandImpl(command, result) != nullptr) {
260 result.AppendMessageWithFormat("%s", command.c_str());
261 result.SetStatus(eReturnStatusSuccessFinishResult);
262 }
263 }
264
GetStopCmdSourceOnError() const265 bool CommandInterpreter::GetStopCmdSourceOnError() const {
266 const uint32_t idx = ePropertyStopCmdSourceOnError;
267 return GetPropertyAtIndexAs<bool>(
268 idx, g_interpreter_properties[idx].default_uint_value != 0);
269 }
270
GetSpaceReplPrompts() const271 bool CommandInterpreter::GetSpaceReplPrompts() const {
272 const uint32_t idx = ePropertySpaceReplPrompts;
273 return GetPropertyAtIndexAs<bool>(
274 idx, g_interpreter_properties[idx].default_uint_value != 0);
275 }
276
GetRepeatPreviousCommand() const277 bool CommandInterpreter::GetRepeatPreviousCommand() const {
278 const uint32_t idx = ePropertyRepeatPreviousCommand;
279 return GetPropertyAtIndexAs<bool>(
280 idx, g_interpreter_properties[idx].default_uint_value != 0);
281 }
282
GetRequireCommandOverwrite() const283 bool CommandInterpreter::GetRequireCommandOverwrite() const {
284 const uint32_t idx = ePropertyRequireCommandOverwrite;
285 return GetPropertyAtIndexAs<bool>(
286 idx, g_interpreter_properties[idx].default_uint_value != 0);
287 }
288
Initialize()289 void CommandInterpreter::Initialize() {
290 LLDB_SCOPED_TIMER();
291
292 LoadCommandDictionary();
293
294 // An alias arguments vector to reuse - reset it before use...
295 OptionArgVectorSP alias_arguments_vector_sp(new OptionArgVector);
296
297 // Set up some initial aliases.
298 CommandObjectSP cmd_obj_sp = GetCommandSPExact("quit");
299 if (cmd_obj_sp) {
300 AddAlias("q", cmd_obj_sp);
301 AddAlias("exit", cmd_obj_sp);
302 }
303
304 cmd_obj_sp = GetCommandSPExact("_regexp-attach");
305 if (cmd_obj_sp)
306 AddAlias("attach", cmd_obj_sp)->SetSyntax(cmd_obj_sp->GetSyntax());
307
308 cmd_obj_sp = GetCommandSPExact("process detach");
309 if (cmd_obj_sp) {
310 AddAlias("detach", cmd_obj_sp);
311 }
312
313 cmd_obj_sp = GetCommandSPExact("process continue");
314 if (cmd_obj_sp) {
315 AddAlias("c", cmd_obj_sp);
316 AddAlias("continue", cmd_obj_sp);
317 }
318
319 cmd_obj_sp = GetCommandSPExact("_regexp-break");
320 if (cmd_obj_sp)
321 AddAlias("b", cmd_obj_sp)->SetSyntax(cmd_obj_sp->GetSyntax());
322
323 cmd_obj_sp = GetCommandSPExact("_regexp-tbreak");
324 if (cmd_obj_sp)
325 AddAlias("tbreak", cmd_obj_sp)->SetSyntax(cmd_obj_sp->GetSyntax());
326
327 cmd_obj_sp = GetCommandSPExact("thread step-inst");
328 if (cmd_obj_sp) {
329 AddAlias("stepi", cmd_obj_sp);
330 AddAlias("si", cmd_obj_sp);
331 }
332
333 cmd_obj_sp = GetCommandSPExact("thread step-inst-over");
334 if (cmd_obj_sp) {
335 AddAlias("nexti", cmd_obj_sp);
336 AddAlias("ni", cmd_obj_sp);
337 }
338
339 cmd_obj_sp = GetCommandSPExact("thread step-in");
340 if (cmd_obj_sp) {
341 AddAlias("s", cmd_obj_sp);
342 AddAlias("step", cmd_obj_sp);
343 CommandAlias *sif_alias = AddAlias(
344 "sif", cmd_obj_sp, "--end-linenumber block --step-in-target %1");
345 if (sif_alias) {
346 sif_alias->SetHelp("Step through the current block, stopping if you step "
347 "directly into a function whose name matches the "
348 "TargetFunctionName.");
349 sif_alias->SetSyntax("sif <TargetFunctionName>");
350 }
351 }
352
353 cmd_obj_sp = GetCommandSPExact("thread step-over");
354 if (cmd_obj_sp) {
355 AddAlias("n", cmd_obj_sp);
356 AddAlias("next", cmd_obj_sp);
357 }
358
359 cmd_obj_sp = GetCommandSPExact("thread step-out");
360 if (cmd_obj_sp) {
361 AddAlias("finish", cmd_obj_sp);
362 }
363
364 cmd_obj_sp = GetCommandSPExact("frame select");
365 if (cmd_obj_sp) {
366 AddAlias("f", cmd_obj_sp);
367 }
368
369 cmd_obj_sp = GetCommandSPExact("thread select");
370 if (cmd_obj_sp) {
371 AddAlias("t", cmd_obj_sp);
372 }
373
374 cmd_obj_sp = GetCommandSPExact("_regexp-jump");
375 if (cmd_obj_sp) {
376 AddAlias("j", cmd_obj_sp)->SetSyntax(cmd_obj_sp->GetSyntax());
377 AddAlias("jump", cmd_obj_sp)->SetSyntax(cmd_obj_sp->GetSyntax());
378 }
379
380 cmd_obj_sp = GetCommandSPExact("_regexp-list");
381 if (cmd_obj_sp) {
382 AddAlias("l", cmd_obj_sp)->SetSyntax(cmd_obj_sp->GetSyntax());
383 AddAlias("list", cmd_obj_sp)->SetSyntax(cmd_obj_sp->GetSyntax());
384 }
385
386 cmd_obj_sp = GetCommandSPExact("_regexp-env");
387 if (cmd_obj_sp)
388 AddAlias("env", cmd_obj_sp)->SetSyntax(cmd_obj_sp->GetSyntax());
389
390 cmd_obj_sp = GetCommandSPExact("memory read");
391 if (cmd_obj_sp)
392 AddAlias("x", cmd_obj_sp);
393
394 cmd_obj_sp = GetCommandSPExact("_regexp-up");
395 if (cmd_obj_sp)
396 AddAlias("up", cmd_obj_sp)->SetSyntax(cmd_obj_sp->GetSyntax());
397
398 cmd_obj_sp = GetCommandSPExact("_regexp-down");
399 if (cmd_obj_sp)
400 AddAlias("down", cmd_obj_sp)->SetSyntax(cmd_obj_sp->GetSyntax());
401
402 cmd_obj_sp = GetCommandSPExact("_regexp-display");
403 if (cmd_obj_sp)
404 AddAlias("display", cmd_obj_sp)->SetSyntax(cmd_obj_sp->GetSyntax());
405
406 cmd_obj_sp = GetCommandSPExact("disassemble");
407 if (cmd_obj_sp)
408 AddAlias("dis", cmd_obj_sp);
409
410 cmd_obj_sp = GetCommandSPExact("disassemble");
411 if (cmd_obj_sp)
412 AddAlias("di", cmd_obj_sp);
413
414 cmd_obj_sp = GetCommandSPExact("_regexp-undisplay");
415 if (cmd_obj_sp)
416 AddAlias("undisplay", cmd_obj_sp)->SetSyntax(cmd_obj_sp->GetSyntax());
417
418 cmd_obj_sp = GetCommandSPExact("_regexp-bt");
419 if (cmd_obj_sp)
420 AddAlias("bt", cmd_obj_sp)->SetSyntax(cmd_obj_sp->GetSyntax());
421
422 cmd_obj_sp = GetCommandSPExact("target create");
423 if (cmd_obj_sp)
424 AddAlias("file", cmd_obj_sp);
425
426 cmd_obj_sp = GetCommandSPExact("target modules");
427 if (cmd_obj_sp)
428 AddAlias("image", cmd_obj_sp);
429
430 alias_arguments_vector_sp = std::make_shared<OptionArgVector>();
431
432 cmd_obj_sp = GetCommandSPExact("dwim-print");
433 if (cmd_obj_sp) {
434 AddAlias("p", cmd_obj_sp, "--")->SetHelpLong("");
435 AddAlias("print", cmd_obj_sp, "--")->SetHelpLong("");
436 if (auto *po = AddAlias("po", cmd_obj_sp, "-O --")) {
437 po->SetHelp("Evaluate an expression on the current thread. Displays any "
438 "returned value with formatting "
439 "controlled by the type's author.");
440 po->SetHelpLong("");
441 }
442 }
443
444 cmd_obj_sp = GetCommandSPExact("expression");
445 if (cmd_obj_sp) {
446 // Ensure `e` runs `expression`.
447 AddAlias("e", cmd_obj_sp);
448 AddAlias("call", cmd_obj_sp, "--")->SetHelpLong("");
449 CommandAlias *parray_alias =
450 AddAlias("parray", cmd_obj_sp, "--element-count %1 --");
451 if (parray_alias) {
452 parray_alias->SetHelp(
453 "parray <COUNT> <EXPRESSION> -- lldb will evaluate EXPRESSION "
454 "to get a typed-pointer-to-an-array in memory, and will display "
455 "COUNT elements of that type from the array.");
456 parray_alias->SetHelpLong("");
457 }
458 CommandAlias *poarray_alias = AddAlias(
459 "poarray", cmd_obj_sp, "--object-description --element-count %1 --");
460 if (poarray_alias) {
461 poarray_alias->SetHelp(
462 "poarray <COUNT> <EXPRESSION> -- lldb will "
463 "evaluate EXPRESSION to get the address of an array of COUNT "
464 "objects in memory, and will call po on them.");
465 poarray_alias->SetHelpLong("");
466 }
467 }
468
469 cmd_obj_sp = GetCommandSPExact("platform shell");
470 if (cmd_obj_sp) {
471 CommandAlias *shell_alias = AddAlias("shell", cmd_obj_sp, " --host --");
472 if (shell_alias) {
473 shell_alias->SetHelp("Run a shell command on the host.");
474 shell_alias->SetHelpLong("");
475 shell_alias->SetSyntax("shell <shell-command>");
476 }
477 }
478
479 cmd_obj_sp = GetCommandSPExact("process kill");
480 if (cmd_obj_sp) {
481 AddAlias("kill", cmd_obj_sp);
482 }
483
484 cmd_obj_sp = GetCommandSPExact("process launch");
485 if (cmd_obj_sp) {
486 alias_arguments_vector_sp = std::make_shared<OptionArgVector>();
487 #if defined(__APPLE__)
488 #if TARGET_OS_IPHONE
489 AddAlias("r", cmd_obj_sp, "--");
490 AddAlias("run", cmd_obj_sp, "--");
491 #else
492 AddAlias("r", cmd_obj_sp, "--shell-expand-args true --");
493 AddAlias("run", cmd_obj_sp, "--shell-expand-args true --");
494 #endif
495 #else
496 StreamString defaultshell;
497 defaultshell.Printf("--shell=%s --",
498 HostInfo::GetDefaultShell().GetPath().c_str());
499 AddAlias("r", cmd_obj_sp, defaultshell.GetString());
500 AddAlias("run", cmd_obj_sp, defaultshell.GetString());
501 #endif
502 }
503
504 cmd_obj_sp = GetCommandSPExact("target symbols add");
505 if (cmd_obj_sp) {
506 AddAlias("add-dsym", cmd_obj_sp);
507 }
508
509 cmd_obj_sp = GetCommandSPExact("breakpoint set");
510 if (cmd_obj_sp) {
511 AddAlias("rbreak", cmd_obj_sp, "--func-regex %1");
512 }
513
514 cmd_obj_sp = GetCommandSPExact("frame variable");
515 if (cmd_obj_sp) {
516 AddAlias("v", cmd_obj_sp);
517 AddAlias("var", cmd_obj_sp);
518 AddAlias("vo", cmd_obj_sp, "--object-description");
519 }
520
521 cmd_obj_sp = GetCommandSPExact("register");
522 if (cmd_obj_sp) {
523 AddAlias("re", cmd_obj_sp);
524 }
525
526 cmd_obj_sp = GetCommandSPExact("scripting run");
527 if (cmd_obj_sp) {
528 AddAlias("script", cmd_obj_sp);
529 }
530
531 cmd_obj_sp = GetCommandSPExact("session history");
532 if (cmd_obj_sp) {
533 AddAlias("history", cmd_obj_sp);
534 }
535
536 cmd_obj_sp = GetCommandSPExact("help");
537 if (cmd_obj_sp) {
538 AddAlias("h", cmd_obj_sp);
539 }
540 }
541
Clear()542 void CommandInterpreter::Clear() { m_command_io_handler_sp.reset(); }
543
ProcessEmbeddedScriptCommands(const char * arg)544 const char *CommandInterpreter::ProcessEmbeddedScriptCommands(const char *arg) {
545 // This function has not yet been implemented.
546
547 // Look for any embedded script command
548 // If found,
549 // get interpreter object from the command dictionary,
550 // call execute_one_command on it,
551 // get the results as a string,
552 // substitute that string for current stuff.
553
554 return arg;
555 }
556
557 #define REGISTER_COMMAND_OBJECT(NAME, CLASS) \
558 m_command_dict[NAME] = std::make_shared<CLASS>(*this);
559
LoadCommandDictionary()560 void CommandInterpreter::LoadCommandDictionary() {
561 LLDB_SCOPED_TIMER();
562
563 REGISTER_COMMAND_OBJECT("apropos", CommandObjectApropos);
564 REGISTER_COMMAND_OBJECT("breakpoint", CommandObjectMultiwordBreakpoint);
565 REGISTER_COMMAND_OBJECT("command", CommandObjectMultiwordCommands);
566 REGISTER_COMMAND_OBJECT("diagnostics", CommandObjectDiagnostics);
567 REGISTER_COMMAND_OBJECT("disassemble", CommandObjectDisassemble);
568 REGISTER_COMMAND_OBJECT("dwim-print", CommandObjectDWIMPrint);
569 REGISTER_COMMAND_OBJECT("expression", CommandObjectExpression);
570 REGISTER_COMMAND_OBJECT("frame", CommandObjectMultiwordFrame);
571 REGISTER_COMMAND_OBJECT("gui", CommandObjectGUI);
572 REGISTER_COMMAND_OBJECT("help", CommandObjectHelp);
573 REGISTER_COMMAND_OBJECT("log", CommandObjectLog);
574 REGISTER_COMMAND_OBJECT("memory", CommandObjectMemory);
575 REGISTER_COMMAND_OBJECT("platform", CommandObjectPlatform);
576 REGISTER_COMMAND_OBJECT("plugin", CommandObjectPlugin);
577 REGISTER_COMMAND_OBJECT("process", CommandObjectMultiwordProcess);
578 REGISTER_COMMAND_OBJECT("protocol-server", CommandObjectProtocolServer);
579 REGISTER_COMMAND_OBJECT("quit", CommandObjectQuit);
580 REGISTER_COMMAND_OBJECT("register", CommandObjectRegister);
581 REGISTER_COMMAND_OBJECT("scripting", CommandObjectMultiwordScripting);
582 REGISTER_COMMAND_OBJECT("settings", CommandObjectMultiwordSettings);
583 REGISTER_COMMAND_OBJECT("session", CommandObjectSession);
584 REGISTER_COMMAND_OBJECT("source", CommandObjectMultiwordSource);
585 REGISTER_COMMAND_OBJECT("statistics", CommandObjectStats);
586 REGISTER_COMMAND_OBJECT("target", CommandObjectMultiwordTarget);
587 REGISTER_COMMAND_OBJECT("thread", CommandObjectMultiwordThread);
588 REGISTER_COMMAND_OBJECT("trace", CommandObjectTrace);
589 REGISTER_COMMAND_OBJECT("type", CommandObjectType);
590 REGISTER_COMMAND_OBJECT("version", CommandObjectVersion);
591 REGISTER_COMMAND_OBJECT("watchpoint", CommandObjectMultiwordWatchpoint);
592 REGISTER_COMMAND_OBJECT("language", CommandObjectLanguage);
593
594 // clang-format off
595 const char *break_regexes[][2] = {
596 {"^(.*[^[:space:]])[[:space:]]*:[[:space:]]*([[:digit:]]+)[[:space:]]*:[[:space:]]*([[:digit:]]+)[[:space:]]*$",
597 "breakpoint set --file '%1' --line %2 --column %3"},
598 {"^(.*[^[:space:]])[[:space:]]*:[[:space:]]*([[:digit:]]+)[[:space:]]*$",
599 "breakpoint set --file '%1' --line %2"},
600 {"^/([^/]+)/$", "breakpoint set --source-pattern-regexp '%1'"},
601 {"^([[:digit:]]+)[[:space:]]*$", "breakpoint set --line %1"},
602 {"^\\*?(0x[[:xdigit:]]+)[[:space:]]*$", "breakpoint set --address %1"},
603 {"^[\"']?([-+]?\\[.*\\])[\"']?[[:space:]]*$",
604 "breakpoint set --name '%1'"},
605 {"^(-.*)$", "breakpoint set %1"},
606 {"^(.*[^[:space:]])`(.*[^[:space:]])[[:space:]]*$",
607 "breakpoint set --name '%2' --shlib '%1'"},
608 {"^\\&(.*[^[:space:]])[[:space:]]*$",
609 "breakpoint set --name '%1' --skip-prologue=0"},
610 {"^[\"']?(.*[^[:space:]\"'])[\"']?[[:space:]]*$",
611 "breakpoint set --name '%1'"}};
612 // clang-format on
613
614 size_t num_regexes = std::size(break_regexes);
615
616 std::unique_ptr<CommandObjectRegexCommand> break_regex_cmd_up(
617 new CommandObjectRegexCommand(
618 *this, "_regexp-break",
619 "Set a breakpoint using one of several shorthand formats.",
620 "\n"
621 "_regexp-break <filename>:<linenum>:<colnum>\n"
622 " main.c:12:21 // Break at line 12 and column "
623 "21 of main.c\n\n"
624 "_regexp-break <filename>:<linenum>\n"
625 " main.c:12 // Break at line 12 of "
626 "main.c\n\n"
627 "_regexp-break <linenum>\n"
628 " 12 // Break at line 12 of current "
629 "file\n\n"
630 "_regexp-break 0x<address>\n"
631 " 0x1234000 // Break at address "
632 "0x1234000\n\n"
633 "_regexp-break <name>\n"
634 " main // Break in 'main' after the "
635 "prologue\n\n"
636 "_regexp-break &<name>\n"
637 " &main // Break at first instruction "
638 "in 'main'\n\n"
639 "_regexp-break <module>`<name>\n"
640 " libc.so`malloc // Break in 'malloc' from "
641 "'libc.so'\n\n"
642 "_regexp-break /<source-regex>/\n"
643 " /break here/ // Break on source lines in "
644 "current file\n"
645 " // containing text 'break "
646 "here'.\n",
647 lldb::eSymbolCompletion | lldb::eSourceFileCompletion, false));
648
649 if (break_regex_cmd_up) {
650 bool success = true;
651 for (size_t i = 0; i < num_regexes; i++) {
652 success = break_regex_cmd_up->AddRegexCommand(break_regexes[i][0],
653 break_regexes[i][1]);
654 if (!success)
655 break;
656 }
657 success =
658 break_regex_cmd_up->AddRegexCommand("^$", "breakpoint list --full");
659
660 if (success) {
661 CommandObjectSP break_regex_cmd_sp(break_regex_cmd_up.release());
662 m_command_dict[std::string(break_regex_cmd_sp->GetCommandName())] =
663 break_regex_cmd_sp;
664 }
665 }
666
667 std::unique_ptr<CommandObjectRegexCommand> tbreak_regex_cmd_up(
668 new CommandObjectRegexCommand(
669 *this, "_regexp-tbreak",
670 "Set a one-shot breakpoint using one of several shorthand formats.",
671 "\n"
672 "_regexp-break <filename>:<linenum>:<colnum>\n"
673 " main.c:12:21 // Break at line 12 and column "
674 "21 of main.c\n\n"
675 "_regexp-break <filename>:<linenum>\n"
676 " main.c:12 // Break at line 12 of "
677 "main.c\n\n"
678 "_regexp-break <linenum>\n"
679 " 12 // Break at line 12 of current "
680 "file\n\n"
681 "_regexp-break 0x<address>\n"
682 " 0x1234000 // Break at address "
683 "0x1234000\n\n"
684 "_regexp-break <name>\n"
685 " main // Break in 'main' after the "
686 "prologue\n\n"
687 "_regexp-break &<name>\n"
688 " &main // Break at first instruction "
689 "in 'main'\n\n"
690 "_regexp-break <module>`<name>\n"
691 " libc.so`malloc // Break in 'malloc' from "
692 "'libc.so'\n\n"
693 "_regexp-break /<source-regex>/\n"
694 " /break here/ // Break on source lines in "
695 "current file\n"
696 " // containing text 'break "
697 "here'.\n",
698 lldb::eSymbolCompletion | lldb::eSourceFileCompletion, false));
699
700 if (tbreak_regex_cmd_up) {
701 bool success = true;
702 for (size_t i = 0; i < num_regexes; i++) {
703 std::string command = break_regexes[i][1];
704 command += " -o 1";
705 success =
706 tbreak_regex_cmd_up->AddRegexCommand(break_regexes[i][0], command);
707 if (!success)
708 break;
709 }
710 success =
711 tbreak_regex_cmd_up->AddRegexCommand("^$", "breakpoint list --full");
712
713 if (success) {
714 CommandObjectSP tbreak_regex_cmd_sp(tbreak_regex_cmd_up.release());
715 m_command_dict[std::string(tbreak_regex_cmd_sp->GetCommandName())] =
716 tbreak_regex_cmd_sp;
717 }
718 }
719
720 std::unique_ptr<CommandObjectRegexCommand> attach_regex_cmd_up(
721 new CommandObjectRegexCommand(
722 *this, "_regexp-attach", "Attach to process by ID or name.",
723 "_regexp-attach <pid> | <process-name>", 0, false));
724 if (attach_regex_cmd_up) {
725 if (attach_regex_cmd_up->AddRegexCommand("^([0-9]+)[[:space:]]*$",
726 "process attach --pid %1") &&
727 attach_regex_cmd_up->AddRegexCommand(
728 "^(-.*|.* -.*)$", "process attach %1") && // Any options that are
729 // specified get passed to
730 // 'process attach'
731 attach_regex_cmd_up->AddRegexCommand("^(.+)$",
732 "process attach --name '%1'") &&
733 attach_regex_cmd_up->AddRegexCommand("^$", "process attach")) {
734 CommandObjectSP attach_regex_cmd_sp(attach_regex_cmd_up.release());
735 m_command_dict[std::string(attach_regex_cmd_sp->GetCommandName())] =
736 attach_regex_cmd_sp;
737 }
738 }
739
740 std::unique_ptr<CommandObjectRegexCommand> down_regex_cmd_up(
741 new CommandObjectRegexCommand(*this, "_regexp-down",
742 "Select a newer stack frame. Defaults to "
743 "moving one frame, a numeric argument can "
744 "specify an arbitrary number.",
745 "_regexp-down [<count>]", 0, false));
746 if (down_regex_cmd_up) {
747 if (down_regex_cmd_up->AddRegexCommand("^$", "frame select -r -1") &&
748 down_regex_cmd_up->AddRegexCommand("^([0-9]+)$",
749 "frame select -r -%1")) {
750 CommandObjectSP down_regex_cmd_sp(down_regex_cmd_up.release());
751 m_command_dict[std::string(down_regex_cmd_sp->GetCommandName())] =
752 down_regex_cmd_sp;
753 }
754 }
755
756 std::unique_ptr<CommandObjectRegexCommand> up_regex_cmd_up(
757 new CommandObjectRegexCommand(
758 *this, "_regexp-up",
759 "Select an older stack frame. Defaults to moving one "
760 "frame, a numeric argument can specify an arbitrary number.",
761 "_regexp-up [<count>]", 0, false));
762 if (up_regex_cmd_up) {
763 if (up_regex_cmd_up->AddRegexCommand("^$", "frame select -r 1") &&
764 up_regex_cmd_up->AddRegexCommand("^([0-9]+)$", "frame select -r %1")) {
765 CommandObjectSP up_regex_cmd_sp(up_regex_cmd_up.release());
766 m_command_dict[std::string(up_regex_cmd_sp->GetCommandName())] =
767 up_regex_cmd_sp;
768 }
769 }
770
771 std::unique_ptr<CommandObjectRegexCommand> display_regex_cmd_up(
772 new CommandObjectRegexCommand(
773 *this, "_regexp-display",
774 "Evaluate an expression at every stop (see 'help target stop-hook'.)",
775 "_regexp-display expression", 0, false));
776 if (display_regex_cmd_up) {
777 if (display_regex_cmd_up->AddRegexCommand(
778 "^(.+)$", "target stop-hook add -o \"expr -- %1\"")) {
779 CommandObjectSP display_regex_cmd_sp(display_regex_cmd_up.release());
780 m_command_dict[std::string(display_regex_cmd_sp->GetCommandName())] =
781 display_regex_cmd_sp;
782 }
783 }
784
785 std::unique_ptr<CommandObjectRegexCommand> undisplay_regex_cmd_up(
786 new CommandObjectRegexCommand(*this, "_regexp-undisplay",
787 "Stop displaying expression at every "
788 "stop (specified by stop-hook index.)",
789 "_regexp-undisplay stop-hook-number", 0,
790 false));
791 if (undisplay_regex_cmd_up) {
792 if (undisplay_regex_cmd_up->AddRegexCommand("^([0-9]+)$",
793 "target stop-hook delete %1")) {
794 CommandObjectSP undisplay_regex_cmd_sp(undisplay_regex_cmd_up.release());
795 m_command_dict[std::string(undisplay_regex_cmd_sp->GetCommandName())] =
796 undisplay_regex_cmd_sp;
797 }
798 }
799
800 std::unique_ptr<CommandObjectRegexCommand> connect_gdb_remote_cmd_up(
801 new CommandObjectRegexCommand(
802 *this, "gdb-remote",
803 "Connect to a process via remote GDB server.\n"
804 "If no host is specified, localhost is assumed.\n"
805 "gdb-remote is an abbreviation for 'process connect --plugin "
806 "gdb-remote connect://<hostname>:<port>'\n",
807 "gdb-remote [<hostname>:]<portnum>", 0, false));
808 if (connect_gdb_remote_cmd_up) {
809 if (connect_gdb_remote_cmd_up->AddRegexCommand(
810 "^([^:]+|\\[[0-9a-fA-F:]+.*\\]):([0-9]+)$",
811 "process connect --plugin gdb-remote connect://%1:%2") &&
812 connect_gdb_remote_cmd_up->AddRegexCommand(
813 "^([[:digit:]]+)$",
814 "process connect --plugin gdb-remote connect://localhost:%1")) {
815 CommandObjectSP command_sp(connect_gdb_remote_cmd_up.release());
816 m_command_dict[std::string(command_sp->GetCommandName())] = command_sp;
817 }
818 }
819
820 std::unique_ptr<CommandObjectRegexCommand> connect_kdp_remote_cmd_up(
821 new CommandObjectRegexCommand(
822 *this, "kdp-remote",
823 "Connect to a process via remote KDP server.\n"
824 "If no UDP port is specified, port 41139 is assumed.\n"
825 "kdp-remote is an abbreviation for 'process connect --plugin "
826 "kdp-remote udp://<hostname>:<port>'\n",
827 "kdp-remote <hostname>[:<portnum>]", 0, false));
828 if (connect_kdp_remote_cmd_up) {
829 if (connect_kdp_remote_cmd_up->AddRegexCommand(
830 "^([^:]+:[[:digit:]]+)$",
831 "process connect --plugin kdp-remote udp://%1") &&
832 connect_kdp_remote_cmd_up->AddRegexCommand(
833 "^(.+)$", "process connect --plugin kdp-remote udp://%1:41139")) {
834 CommandObjectSP command_sp(connect_kdp_remote_cmd_up.release());
835 m_command_dict[std::string(command_sp->GetCommandName())] = command_sp;
836 }
837 }
838
839 std::unique_ptr<CommandObjectRegexCommand> bt_regex_cmd_up(
840 new CommandObjectRegexCommand(
841 *this, "_regexp-bt",
842 "Show backtrace of the current thread's call stack. Any numeric "
843 "argument displays at most that many frames. The argument 'all' "
844 "displays all threads. Use 'settings set frame-format' to customize "
845 "the printing of individual frames and 'settings set thread-format' "
846 "to customize the thread header. Frame recognizers may filter the "
847 "list. Use 'thread backtrace -u (--unfiltered)' to see them all.",
848 "bt [<digit> | all]", 0, false));
849 if (bt_regex_cmd_up) {
850 // accept but don't document "bt -c <number>" -- before bt was a regex
851 // command if you wanted to backtrace three frames you would do "bt -c 3"
852 // but the intention is to have this emulate the gdb "bt" command and so
853 // now "bt 3" is the preferred form, in line with gdb.
854 if (bt_regex_cmd_up->AddRegexCommand("^([[:digit:]]+)[[:space:]]*$",
855 "thread backtrace -c %1") &&
856 bt_regex_cmd_up->AddRegexCommand("^(-[^[:space:]].*)$",
857 "thread backtrace %1") &&
858 bt_regex_cmd_up->AddRegexCommand("^all[[:space:]]*$",
859 "thread backtrace all") &&
860 bt_regex_cmd_up->AddRegexCommand("^[[:space:]]*$",
861 "thread backtrace")) {
862 CommandObjectSP command_sp(bt_regex_cmd_up.release());
863 m_command_dict[std::string(command_sp->GetCommandName())] = command_sp;
864 }
865 }
866
867 std::unique_ptr<CommandObjectRegexCommand> list_regex_cmd_up(
868 new CommandObjectRegexCommand(
869 *this, "_regexp-list",
870 "List relevant source code using one of several shorthand formats.",
871 "\n"
872 "_regexp-list <file>:<line> // List around specific file/line\n"
873 "_regexp-list <line> // List current file around specified "
874 "line\n"
875 "_regexp-list <function-name> // List specified function\n"
876 "_regexp-list 0x<address> // List around specified address\n"
877 "_regexp-list -[<count>] // List previous <count> lines\n"
878 "_regexp-list // List subsequent lines",
879 lldb::eSourceFileCompletion, false));
880 if (list_regex_cmd_up) {
881 if (list_regex_cmd_up->AddRegexCommand("^([0-9]+)[[:space:]]*$",
882 "source list --line %1") &&
883 list_regex_cmd_up->AddRegexCommand(
884 "^(.*[^[:space:]])[[:space:]]*:[[:space:]]*([[:digit:]]+)[[:space:]"
885 "]*$",
886 "source list --file '%1' --line %2") &&
887 list_regex_cmd_up->AddRegexCommand(
888 "^\\*?(0x[[:xdigit:]]+)[[:space:]]*$",
889 "source list --address %1") &&
890 list_regex_cmd_up->AddRegexCommand("^-[[:space:]]*$",
891 "source list --reverse") &&
892 list_regex_cmd_up->AddRegexCommand(
893 "^-([[:digit:]]+)[[:space:]]*$",
894 "source list --reverse --count %1") &&
895 list_regex_cmd_up->AddRegexCommand("^(.+)$",
896 "source list --name \"%1\"") &&
897 list_regex_cmd_up->AddRegexCommand("^$", "source list")) {
898 CommandObjectSP list_regex_cmd_sp(list_regex_cmd_up.release());
899 m_command_dict[std::string(list_regex_cmd_sp->GetCommandName())] =
900 list_regex_cmd_sp;
901 }
902 }
903
904 std::unique_ptr<CommandObjectRegexCommand> env_regex_cmd_up(
905 new CommandObjectRegexCommand(
906 *this, "_regexp-env",
907 "Shorthand for viewing and setting environment variables.",
908 "\n"
909 "_regexp-env // Show environment\n"
910 "_regexp-env <name>=<value> // Set an environment variable",
911 0, false));
912 if (env_regex_cmd_up) {
913 if (env_regex_cmd_up->AddRegexCommand("^$",
914 "settings show target.env-vars") &&
915 env_regex_cmd_up->AddRegexCommand("^([A-Za-z_][A-Za-z_0-9]*=.*)$",
916 "settings set target.env-vars %1")) {
917 CommandObjectSP env_regex_cmd_sp(env_regex_cmd_up.release());
918 m_command_dict[std::string(env_regex_cmd_sp->GetCommandName())] =
919 env_regex_cmd_sp;
920 }
921 }
922
923 std::unique_ptr<CommandObjectRegexCommand> jump_regex_cmd_up(
924 new CommandObjectRegexCommand(
925 *this, "_regexp-jump", "Set the program counter to a new address.",
926 "\n"
927 "_regexp-jump <line>\n"
928 "_regexp-jump +<line-offset> | -<line-offset>\n"
929 "_regexp-jump <file>:<line>\n"
930 "_regexp-jump *<addr>\n",
931 0, false));
932 if (jump_regex_cmd_up) {
933 if (jump_regex_cmd_up->AddRegexCommand("^\\*(.*)$",
934 "thread jump --addr %1") &&
935 jump_regex_cmd_up->AddRegexCommand("^([0-9]+)$",
936 "thread jump --line %1") &&
937 jump_regex_cmd_up->AddRegexCommand("^([^:]+):([0-9]+)$",
938 "thread jump --file %1 --line %2") &&
939 jump_regex_cmd_up->AddRegexCommand("^([+\\-][0-9]+)$",
940 "thread jump --by %1")) {
941 CommandObjectSP jump_regex_cmd_sp(jump_regex_cmd_up.release());
942 m_command_dict[std::string(jump_regex_cmd_sp->GetCommandName())] =
943 jump_regex_cmd_sp;
944 }
945 }
946 }
947
GetCommandNamesMatchingPartialString(const char * cmd_str,bool include_aliases,StringList & matches,StringList & descriptions)948 int CommandInterpreter::GetCommandNamesMatchingPartialString(
949 const char *cmd_str, bool include_aliases, StringList &matches,
950 StringList &descriptions) {
951 AddNamesMatchingPartialString(m_command_dict, cmd_str, matches,
952 &descriptions);
953
954 if (include_aliases) {
955 AddNamesMatchingPartialString(m_alias_dict, cmd_str, matches,
956 &descriptions);
957 }
958
959 return matches.GetSize();
960 }
961
962 CommandObjectMultiword *
VerifyUserMultiwordCmdPath(Args & path,bool leaf_is_command,Status & result)963 CommandInterpreter::VerifyUserMultiwordCmdPath(Args &path, bool leaf_is_command,
964 Status &result) {
965 result.Clear();
966
967 auto get_multi_or_report_error =
968 [&result](CommandObjectSP cmd_sp,
969 const char *name) -> CommandObjectMultiword * {
970 if (!cmd_sp) {
971 result = Status::FromErrorStringWithFormat(
972 "Path component: '%s' not found", name);
973 return nullptr;
974 }
975 if (!cmd_sp->IsUserCommand()) {
976 result = Status::FromErrorStringWithFormat(
977 "Path component: '%s' is not a user "
978 "command",
979 name);
980 return nullptr;
981 }
982 CommandObjectMultiword *cmd_as_multi = cmd_sp->GetAsMultiwordCommand();
983 if (!cmd_as_multi) {
984 result = Status::FromErrorStringWithFormat(
985 "Path component: '%s' is not a container "
986 "command",
987 name);
988 return nullptr;
989 }
990 return cmd_as_multi;
991 };
992
993 size_t num_args = path.GetArgumentCount();
994 if (num_args == 0) {
995 result = Status::FromErrorString("empty command path");
996 return nullptr;
997 }
998
999 if (num_args == 1 && leaf_is_command) {
1000 // We just got a leaf command to be added to the root. That's not an error,
1001 // just return null for the container.
1002 return nullptr;
1003 }
1004
1005 // Start by getting the root command from the interpreter.
1006 const char *cur_name = path.GetArgumentAtIndex(0);
1007 CommandObjectSP cur_cmd_sp = GetCommandSPExact(cur_name);
1008 CommandObjectMultiword *cur_as_multi =
1009 get_multi_or_report_error(cur_cmd_sp, cur_name);
1010 if (cur_as_multi == nullptr)
1011 return nullptr;
1012
1013 size_t num_path_elements = num_args - (leaf_is_command ? 1 : 0);
1014 for (size_t cursor = 1; cursor < num_path_elements && cur_as_multi != nullptr;
1015 cursor++) {
1016 cur_name = path.GetArgumentAtIndex(cursor);
1017 cur_cmd_sp = cur_as_multi->GetSubcommandSPExact(cur_name);
1018 cur_as_multi = get_multi_or_report_error(cur_cmd_sp, cur_name);
1019 }
1020 return cur_as_multi;
1021 }
1022
GetFrameLanguageCommand() const1023 CommandObjectSP CommandInterpreter::GetFrameLanguageCommand() const {
1024 auto frame_sp = GetExecutionContext().GetFrameSP();
1025 if (!frame_sp)
1026 return {};
1027 auto frame_language =
1028 Language::GetPrimaryLanguage(frame_sp->GuessLanguage().AsLanguageType());
1029
1030 auto it = m_command_dict.find("language");
1031 if (it == m_command_dict.end())
1032 return {};
1033 // The root "language" command.
1034 CommandObjectSP language_cmd_sp = it->second;
1035
1036 auto *plugin = Language::FindPlugin(frame_language);
1037 if (!plugin)
1038 return {};
1039 // "cplusplus", "objc", etc.
1040 auto lang_name = plugin->GetPluginName();
1041
1042 return language_cmd_sp->GetSubcommandSPExact(lang_name);
1043 }
1044
1045 CommandObjectSP
GetCommandSP(llvm::StringRef cmd_str,bool include_aliases,bool exact,StringList * matches,StringList * descriptions) const1046 CommandInterpreter::GetCommandSP(llvm::StringRef cmd_str, bool include_aliases,
1047 bool exact, StringList *matches,
1048 StringList *descriptions) const {
1049 CommandObjectSP command_sp;
1050
1051 std::string cmd = std::string(cmd_str);
1052
1053 if (HasCommands()) {
1054 auto pos = m_command_dict.find(cmd);
1055 if (pos != m_command_dict.end())
1056 command_sp = pos->second;
1057 }
1058
1059 if (include_aliases && HasAliases()) {
1060 auto alias_pos = m_alias_dict.find(cmd);
1061 if (alias_pos != m_alias_dict.end())
1062 command_sp = alias_pos->second;
1063 }
1064
1065 if (HasUserCommands()) {
1066 auto pos = m_user_dict.find(cmd);
1067 if (pos != m_user_dict.end())
1068 command_sp = pos->second;
1069 }
1070
1071 if (HasUserMultiwordCommands()) {
1072 auto pos = m_user_mw_dict.find(cmd);
1073 if (pos != m_user_mw_dict.end())
1074 command_sp = pos->second;
1075 }
1076
1077 if (!exact && !command_sp) {
1078 // We will only get into here if we didn't find any exact matches.
1079
1080 CommandObjectSP user_match_sp, user_mw_match_sp, alias_match_sp,
1081 real_match_sp;
1082
1083 StringList local_matches;
1084 if (matches == nullptr)
1085 matches = &local_matches;
1086
1087 unsigned int num_cmd_matches = 0;
1088 unsigned int num_alias_matches = 0;
1089 unsigned int num_user_matches = 0;
1090 unsigned int num_user_mw_matches = 0;
1091
1092 // Look through the command dictionaries one by one, and if we get only one
1093 // match from any of them in toto, then return that, otherwise return an
1094 // empty CommandObjectSP and the list of matches.
1095
1096 if (HasCommands()) {
1097 num_cmd_matches = AddNamesMatchingPartialString(m_command_dict, cmd_str,
1098 *matches, descriptions);
1099 }
1100
1101 if (num_cmd_matches == 1) {
1102 cmd.assign(matches->GetStringAtIndex(0));
1103 auto pos = m_command_dict.find(cmd);
1104 if (pos != m_command_dict.end())
1105 real_match_sp = pos->second;
1106 }
1107
1108 if (include_aliases && HasAliases()) {
1109 num_alias_matches = AddNamesMatchingPartialString(m_alias_dict, cmd_str,
1110 *matches, descriptions);
1111 }
1112
1113 if (num_alias_matches == 1) {
1114 cmd.assign(matches->GetStringAtIndex(num_cmd_matches));
1115 auto alias_pos = m_alias_dict.find(cmd);
1116 if (alias_pos != m_alias_dict.end())
1117 alias_match_sp = alias_pos->second;
1118 }
1119
1120 if (HasUserCommands()) {
1121 num_user_matches = AddNamesMatchingPartialString(m_user_dict, cmd_str,
1122 *matches, descriptions);
1123 }
1124
1125 if (num_user_matches == 1) {
1126 cmd.assign(
1127 matches->GetStringAtIndex(num_cmd_matches + num_alias_matches));
1128
1129 auto pos = m_user_dict.find(cmd);
1130 if (pos != m_user_dict.end())
1131 user_match_sp = pos->second;
1132 }
1133
1134 if (HasUserMultiwordCommands()) {
1135 num_user_mw_matches = AddNamesMatchingPartialString(
1136 m_user_mw_dict, cmd_str, *matches, descriptions);
1137 }
1138
1139 if (num_user_mw_matches == 1) {
1140 cmd.assign(matches->GetStringAtIndex(num_cmd_matches + num_alias_matches +
1141 num_user_matches));
1142
1143 auto pos = m_user_mw_dict.find(cmd);
1144 if (pos != m_user_mw_dict.end())
1145 user_mw_match_sp = pos->second;
1146 }
1147
1148 // If we got exactly one match, return that, otherwise return the match
1149 // list.
1150
1151 if (num_user_matches + num_user_mw_matches + num_cmd_matches +
1152 num_alias_matches ==
1153 1) {
1154 if (num_cmd_matches)
1155 return real_match_sp;
1156 else if (num_alias_matches)
1157 return alias_match_sp;
1158 else if (num_user_mw_matches)
1159 return user_mw_match_sp;
1160 else
1161 return user_match_sp;
1162 }
1163 }
1164
1165 // When no single match is found, attempt to resolve the command as a language
1166 // plugin subcommand.
1167 if (!command_sp) {
1168 // The `language` subcommand ("language objc", "language cplusplus", etc).
1169 CommandObjectMultiword *lang_subcmd = nullptr;
1170 if (auto lang_subcmd_sp = GetFrameLanguageCommand()) {
1171 lang_subcmd = lang_subcmd_sp->GetAsMultiwordCommand();
1172 command_sp = lang_subcmd_sp->GetSubcommandSPExact(cmd_str);
1173 }
1174
1175 if (!command_sp && !exact && lang_subcmd) {
1176 StringList lang_matches;
1177 AddNamesMatchingPartialString(lang_subcmd->GetSubcommandDictionary(),
1178 cmd_str, lang_matches, descriptions);
1179 if (matches)
1180 matches->AppendList(lang_matches);
1181 if (lang_matches.GetSize() == 1) {
1182 const auto &lang_dict = lang_subcmd->GetSubcommandDictionary();
1183 auto pos = lang_dict.find(lang_matches[0]);
1184 if (pos != lang_dict.end())
1185 return pos->second;
1186 }
1187 }
1188 }
1189
1190 if (matches && command_sp) {
1191 matches->AppendString(cmd_str);
1192 if (descriptions)
1193 descriptions->AppendString(command_sp->GetHelp());
1194 }
1195
1196 return command_sp;
1197 }
1198
AddCommand(llvm::StringRef name,const lldb::CommandObjectSP & cmd_sp,bool can_replace)1199 bool CommandInterpreter::AddCommand(llvm::StringRef name,
1200 const lldb::CommandObjectSP &cmd_sp,
1201 bool can_replace) {
1202 if (cmd_sp.get())
1203 lldbassert((this == &cmd_sp->GetCommandInterpreter()) &&
1204 "tried to add a CommandObject from a different interpreter");
1205
1206 if (name.empty())
1207 return false;
1208
1209 cmd_sp->SetIsUserCommand(false);
1210
1211 std::string name_sstr(name);
1212 auto name_iter = m_command_dict.find(name_sstr);
1213 if (name_iter != m_command_dict.end()) {
1214 if (!can_replace || !name_iter->second->IsRemovable())
1215 return false;
1216 name_iter->second = cmd_sp;
1217 } else {
1218 m_command_dict[name_sstr] = cmd_sp;
1219 }
1220 return true;
1221 }
1222
AddUserCommand(llvm::StringRef name,const lldb::CommandObjectSP & cmd_sp,bool can_replace)1223 Status CommandInterpreter::AddUserCommand(llvm::StringRef name,
1224 const lldb::CommandObjectSP &cmd_sp,
1225 bool can_replace) {
1226 Status result;
1227 if (cmd_sp.get())
1228 lldbassert((this == &cmd_sp->GetCommandInterpreter()) &&
1229 "tried to add a CommandObject from a different interpreter");
1230 if (name.empty()) {
1231 result = Status::FromErrorString(
1232 "can't use the empty string for a command name");
1233 return result;
1234 }
1235 // do not allow replacement of internal commands
1236 if (CommandExists(name)) {
1237 result = Status::FromErrorString("can't replace builtin command");
1238 return result;
1239 }
1240
1241 if (UserCommandExists(name)) {
1242 if (!can_replace) {
1243 result = Status::FromErrorStringWithFormatv(
1244 "user command \"{0}\" already exists and force replace was not set "
1245 "by --overwrite or 'settings set interpreter.require-overwrite "
1246 "false'",
1247 name);
1248 return result;
1249 }
1250 if (cmd_sp->IsMultiwordObject()) {
1251 if (!m_user_mw_dict[std::string(name)]->IsRemovable()) {
1252 result = Status::FromErrorString(
1253 "can't replace explicitly non-removable multi-word command");
1254 return result;
1255 }
1256 } else {
1257 if (!m_user_dict[std::string(name)]->IsRemovable()) {
1258 result = Status::FromErrorString(
1259 "can't replace explicitly non-removable command");
1260 return result;
1261 }
1262 }
1263 }
1264
1265 cmd_sp->SetIsUserCommand(true);
1266
1267 if (cmd_sp->IsMultiwordObject())
1268 m_user_mw_dict[std::string(name)] = cmd_sp;
1269 else
1270 m_user_dict[std::string(name)] = cmd_sp;
1271 return result;
1272 }
1273
1274 CommandObjectSP
GetCommandSPExact(llvm::StringRef cmd_str,bool include_aliases) const1275 CommandInterpreter::GetCommandSPExact(llvm::StringRef cmd_str,
1276 bool include_aliases) const {
1277 // Break up the command string into words, in case it's a multi-word command.
1278 Args cmd_words(cmd_str);
1279
1280 if (cmd_str.empty())
1281 return {};
1282
1283 if (cmd_words.GetArgumentCount() == 1)
1284 return GetCommandSP(cmd_str, include_aliases, true);
1285
1286 // We have a multi-word command (seemingly), so we need to do more work.
1287 // First, get the cmd_obj_sp for the first word in the command.
1288 CommandObjectSP cmd_obj_sp =
1289 GetCommandSP(cmd_words.GetArgumentAtIndex(0), include_aliases, true);
1290 if (!cmd_obj_sp)
1291 return {};
1292
1293 // Loop through the rest of the words in the command (everything passed in
1294 // was supposed to be part of a command name), and find the appropriate
1295 // sub-command SP for each command word....
1296 size_t end = cmd_words.GetArgumentCount();
1297 for (size_t i = 1; i < end; ++i) {
1298 if (!cmd_obj_sp->IsMultiwordObject()) {
1299 // We have more words in the command name, but we don't have a
1300 // multiword object. Fail and return.
1301 return {};
1302 }
1303
1304 cmd_obj_sp = cmd_obj_sp->GetSubcommandSP(cmd_words.GetArgumentAtIndex(i));
1305 if (!cmd_obj_sp) {
1306 // The sub-command name was invalid. Fail and return.
1307 return {};
1308 }
1309 }
1310
1311 // We successfully looped through all the command words and got valid
1312 // command objects for them.
1313 return cmd_obj_sp;
1314 }
1315
1316 CommandObject *
GetCommandObject(llvm::StringRef cmd_str,StringList * matches,StringList * descriptions) const1317 CommandInterpreter::GetCommandObject(llvm::StringRef cmd_str,
1318 StringList *matches,
1319 StringList *descriptions) const {
1320 // Try to find a match among commands and aliases. Allowing inexact matches,
1321 // but perferring exact matches.
1322 return GetCommandSP(cmd_str, /*include_aliases=*/true, /*exact=*/false,
1323 matches, descriptions)
1324 .get();
1325 }
1326
GetUserCommandObject(llvm::StringRef cmd,StringList * matches,StringList * descriptions) const1327 CommandObject *CommandInterpreter::GetUserCommandObject(
1328 llvm::StringRef cmd, StringList *matches, StringList *descriptions) const {
1329 std::string cmd_str(cmd);
1330 auto find_exact = [&](const CommandObject::CommandMap &map) {
1331 auto found_elem = map.find(cmd);
1332 if (found_elem == map.end())
1333 return (CommandObject *)nullptr;
1334 CommandObject *exact_cmd = found_elem->second.get();
1335 if (exact_cmd) {
1336 if (matches)
1337 matches->AppendString(exact_cmd->GetCommandName());
1338 if (descriptions)
1339 descriptions->AppendString(exact_cmd->GetHelp());
1340 return exact_cmd;
1341 }
1342 return (CommandObject *)nullptr;
1343 };
1344
1345 CommandObject *exact_cmd = find_exact(GetUserCommands());
1346 if (exact_cmd)
1347 return exact_cmd;
1348
1349 exact_cmd = find_exact(GetUserMultiwordCommands());
1350 if (exact_cmd)
1351 return exact_cmd;
1352
1353 // We didn't have an exact command, so now look for partial matches.
1354 StringList tmp_list;
1355 StringList *matches_ptr = matches ? matches : &tmp_list;
1356 AddNamesMatchingPartialString(GetUserCommands(), cmd_str, *matches_ptr);
1357 AddNamesMatchingPartialString(GetUserMultiwordCommands(), cmd_str,
1358 *matches_ptr);
1359
1360 return {};
1361 }
1362
GetAliasCommandObject(llvm::StringRef cmd,StringList * matches,StringList * descriptions) const1363 CommandObject *CommandInterpreter::GetAliasCommandObject(
1364 llvm::StringRef cmd, StringList *matches, StringList *descriptions) const {
1365 auto find_exact =
1366 [&](const CommandObject::CommandMap &map) -> CommandObject * {
1367 auto found_elem = map.find(cmd);
1368 if (found_elem == map.end())
1369 return (CommandObject *)nullptr;
1370 CommandObject *exact_cmd = found_elem->second.get();
1371 if (!exact_cmd)
1372 return nullptr;
1373
1374 if (matches)
1375 matches->AppendString(exact_cmd->GetCommandName());
1376
1377 if (descriptions)
1378 descriptions->AppendString(exact_cmd->GetHelp());
1379
1380 return exact_cmd;
1381 return nullptr;
1382 };
1383
1384 CommandObject *exact_cmd = find_exact(GetAliases());
1385 if (exact_cmd)
1386 return exact_cmd;
1387
1388 // We didn't have an exact command, so now look for partial matches.
1389 StringList tmp_list;
1390 StringList *matches_ptr = matches ? matches : &tmp_list;
1391 AddNamesMatchingPartialString(GetAliases(), cmd, *matches_ptr);
1392
1393 return {};
1394 }
1395
CommandExists(llvm::StringRef cmd) const1396 bool CommandInterpreter::CommandExists(llvm::StringRef cmd) const {
1397 return m_command_dict.find(cmd) != m_command_dict.end();
1398 }
1399
GetAliasFullName(llvm::StringRef cmd,std::string & full_name) const1400 bool CommandInterpreter::GetAliasFullName(llvm::StringRef cmd,
1401 std::string &full_name) const {
1402 bool exact_match = (m_alias_dict.find(cmd) != m_alias_dict.end());
1403 if (exact_match) {
1404 full_name.assign(std::string(cmd));
1405 return exact_match;
1406 } else {
1407 StringList matches;
1408 size_t num_alias_matches;
1409 num_alias_matches =
1410 AddNamesMatchingPartialString(m_alias_dict, cmd, matches);
1411 if (num_alias_matches == 1) {
1412 // Make sure this isn't shadowing a command in the regular command space:
1413 StringList regular_matches;
1414 const bool include_aliases = false;
1415 const bool exact = false;
1416 CommandObjectSP cmd_obj_sp(
1417 GetCommandSP(cmd, include_aliases, exact, ®ular_matches));
1418 if (cmd_obj_sp || regular_matches.GetSize() > 0)
1419 return false;
1420 else {
1421 full_name.assign(matches.GetStringAtIndex(0));
1422 return true;
1423 }
1424 } else
1425 return false;
1426 }
1427 }
1428
AliasExists(llvm::StringRef cmd) const1429 bool CommandInterpreter::AliasExists(llvm::StringRef cmd) const {
1430 return m_alias_dict.find(cmd) != m_alias_dict.end();
1431 }
1432
UserCommandExists(llvm::StringRef cmd) const1433 bool CommandInterpreter::UserCommandExists(llvm::StringRef cmd) const {
1434 return m_user_dict.find(cmd) != m_user_dict.end();
1435 }
1436
UserMultiwordCommandExists(llvm::StringRef cmd) const1437 bool CommandInterpreter::UserMultiwordCommandExists(llvm::StringRef cmd) const {
1438 return m_user_mw_dict.find(cmd) != m_user_mw_dict.end();
1439 }
1440
1441 CommandAlias *
AddAlias(llvm::StringRef alias_name,lldb::CommandObjectSP & command_obj_sp,llvm::StringRef args_string)1442 CommandInterpreter::AddAlias(llvm::StringRef alias_name,
1443 lldb::CommandObjectSP &command_obj_sp,
1444 llvm::StringRef args_string) {
1445 if (command_obj_sp.get())
1446 lldbassert((this == &command_obj_sp->GetCommandInterpreter()) &&
1447 "tried to add a CommandObject from a different interpreter");
1448
1449 std::unique_ptr<CommandAlias> command_alias_up(
1450 new CommandAlias(*this, command_obj_sp, args_string, alias_name));
1451
1452 if (command_alias_up && command_alias_up->IsValid()) {
1453 m_alias_dict[std::string(alias_name)] =
1454 CommandObjectSP(command_alias_up.get());
1455 return command_alias_up.release();
1456 }
1457
1458 return nullptr;
1459 }
1460
RemoveAlias(llvm::StringRef alias_name)1461 bool CommandInterpreter::RemoveAlias(llvm::StringRef alias_name) {
1462 auto pos = m_alias_dict.find(alias_name);
1463 if (pos != m_alias_dict.end()) {
1464 m_alias_dict.erase(pos);
1465 return true;
1466 }
1467 return false;
1468 }
1469
RemoveCommand(llvm::StringRef cmd,bool force)1470 bool CommandInterpreter::RemoveCommand(llvm::StringRef cmd, bool force) {
1471 auto pos = m_command_dict.find(cmd);
1472 if (pos != m_command_dict.end()) {
1473 if (force || pos->second->IsRemovable()) {
1474 // Only regular expression objects or python commands are removable under
1475 // normal circumstances.
1476 m_command_dict.erase(pos);
1477 return true;
1478 }
1479 }
1480 return false;
1481 }
1482
RemoveUser(llvm::StringRef user_name)1483 bool CommandInterpreter::RemoveUser(llvm::StringRef user_name) {
1484 CommandObject::CommandMap::iterator pos = m_user_dict.find(user_name);
1485 if (pos != m_user_dict.end()) {
1486 m_user_dict.erase(pos);
1487 return true;
1488 }
1489 return false;
1490 }
1491
RemoveUserMultiword(llvm::StringRef multi_name)1492 bool CommandInterpreter::RemoveUserMultiword(llvm::StringRef multi_name) {
1493 CommandObject::CommandMap::iterator pos = m_user_mw_dict.find(multi_name);
1494 if (pos != m_user_mw_dict.end()) {
1495 m_user_mw_dict.erase(pos);
1496 return true;
1497 }
1498 return false;
1499 }
1500
GetHelp(CommandReturnObject & result,uint32_t cmd_types)1501 void CommandInterpreter::GetHelp(CommandReturnObject &result,
1502 uint32_t cmd_types) {
1503 llvm::StringRef help_prologue(GetDebugger().GetIOHandlerHelpPrologue());
1504 if (!help_prologue.empty()) {
1505 OutputFormattedHelpText(result.GetOutputStream(), llvm::StringRef(),
1506 help_prologue);
1507 }
1508
1509 CommandObject::CommandMap::const_iterator pos;
1510 size_t max_len = FindLongestCommandWord(m_command_dict);
1511
1512 if ((cmd_types & eCommandTypesBuiltin) == eCommandTypesBuiltin) {
1513 result.AppendMessage("Debugger commands:");
1514 result.AppendMessage("");
1515
1516 for (pos = m_command_dict.begin(); pos != m_command_dict.end(); ++pos) {
1517 if (!(cmd_types & eCommandTypesHidden) &&
1518 (pos->first.compare(0, 1, "_") == 0))
1519 continue;
1520
1521 OutputFormattedHelpText(result.GetOutputStream(), pos->first, "--",
1522 pos->second->GetHelp(), max_len);
1523 }
1524 result.AppendMessage("");
1525 }
1526
1527 if (!m_alias_dict.empty() &&
1528 ((cmd_types & eCommandTypesAliases) == eCommandTypesAliases)) {
1529 result.AppendMessageWithFormat(
1530 "Current command abbreviations "
1531 "(type '%shelp command alias' for more info):\n",
1532 GetCommandPrefix());
1533 result.AppendMessage("");
1534 max_len = FindLongestCommandWord(m_alias_dict);
1535
1536 for (auto alias_pos = m_alias_dict.begin(); alias_pos != m_alias_dict.end();
1537 ++alias_pos) {
1538 OutputFormattedHelpText(result.GetOutputStream(), alias_pos->first, "--",
1539 alias_pos->second->GetHelp(), max_len);
1540 }
1541 result.AppendMessage("");
1542 }
1543
1544 if (!m_user_dict.empty() &&
1545 ((cmd_types & eCommandTypesUserDef) == eCommandTypesUserDef)) {
1546 result.AppendMessage("Current user-defined commands:");
1547 result.AppendMessage("");
1548 max_len = FindLongestCommandWord(m_user_dict);
1549 for (pos = m_user_dict.begin(); pos != m_user_dict.end(); ++pos) {
1550 OutputFormattedHelpText(result.GetOutputStream(), pos->first, "--",
1551 pos->second->GetHelp(), max_len);
1552 }
1553 result.AppendMessage("");
1554 }
1555
1556 if (!m_user_mw_dict.empty() &&
1557 ((cmd_types & eCommandTypesUserMW) == eCommandTypesUserMW)) {
1558 result.AppendMessage("Current user-defined container commands:");
1559 result.AppendMessage("");
1560 max_len = FindLongestCommandWord(m_user_mw_dict);
1561 for (pos = m_user_mw_dict.begin(); pos != m_user_mw_dict.end(); ++pos) {
1562 OutputFormattedHelpText(result.GetOutputStream(), pos->first, "--",
1563 pos->second->GetHelp(), max_len);
1564 }
1565 result.AppendMessage("");
1566 }
1567
1568 result.AppendMessageWithFormat(
1569 "For more information on any command, type '%shelp <command-name>'.\n",
1570 GetCommandPrefix());
1571 }
1572
GetCommandObjectForCommand(llvm::StringRef & command_string)1573 CommandObject *CommandInterpreter::GetCommandObjectForCommand(
1574 llvm::StringRef &command_string) {
1575 // This function finds the final, lowest-level, alias-resolved command object
1576 // whose 'Execute' function will eventually be invoked by the given command
1577 // line.
1578
1579 CommandObject *cmd_obj = nullptr;
1580 size_t start = command_string.find_first_not_of(k_white_space);
1581 size_t end = 0;
1582 bool done = false;
1583 while (!done) {
1584 if (start != std::string::npos) {
1585 // Get the next word from command_string.
1586 end = command_string.find_first_of(k_white_space, start);
1587 if (end == std::string::npos)
1588 end = command_string.size();
1589 std::string cmd_word =
1590 std::string(command_string.substr(start, end - start));
1591
1592 if (cmd_obj == nullptr)
1593 // Since cmd_obj is NULL we are on our first time through this loop.
1594 // Check to see if cmd_word is a valid command or alias.
1595 cmd_obj = GetCommandObject(cmd_word);
1596 else if (cmd_obj->IsMultiwordObject()) {
1597 // Our current object is a multi-word object; see if the cmd_word is a
1598 // valid sub-command for our object.
1599 CommandObject *sub_cmd_obj =
1600 cmd_obj->GetSubcommandObject(cmd_word.c_str());
1601 if (sub_cmd_obj)
1602 cmd_obj = sub_cmd_obj;
1603 else // cmd_word was not a valid sub-command word, so we are done
1604 done = true;
1605 } else
1606 // We have a cmd_obj and it is not a multi-word object, so we are done.
1607 done = true;
1608
1609 // If we didn't find a valid command object, or our command object is not
1610 // a multi-word object, or we are at the end of the command_string, then
1611 // we are done. Otherwise, find the start of the next word.
1612
1613 if (!cmd_obj || !cmd_obj->IsMultiwordObject() ||
1614 end >= command_string.size())
1615 done = true;
1616 else
1617 start = command_string.find_first_not_of(k_white_space, end);
1618 } else
1619 // Unable to find any more words.
1620 done = true;
1621 }
1622
1623 command_string = command_string.substr(end);
1624 return cmd_obj;
1625 }
1626
1627 static const char *k_valid_command_chars =
1628 "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789-_";
StripLeadingSpaces(std::string & s)1629 static void StripLeadingSpaces(std::string &s) {
1630 if (!s.empty()) {
1631 size_t pos = s.find_first_not_of(k_white_space);
1632 if (pos == std::string::npos)
1633 s.clear();
1634 else if (pos == 0)
1635 return;
1636 s.erase(0, pos);
1637 }
1638 }
1639
FindArgumentTerminator(const std::string & s)1640 static size_t FindArgumentTerminator(const std::string &s) {
1641 const size_t s_len = s.size();
1642 size_t offset = 0;
1643 while (offset < s_len) {
1644 size_t pos = s.find("--", offset);
1645 if (pos == std::string::npos)
1646 break;
1647 if (pos > 0) {
1648 if (llvm::isSpace(s[pos - 1])) {
1649 // Check if the string ends "\s--" (where \s is a space character) or
1650 // if we have "\s--\s".
1651 if ((pos + 2 >= s_len) || llvm::isSpace(s[pos + 2])) {
1652 return pos;
1653 }
1654 }
1655 }
1656 offset = pos + 2;
1657 }
1658 return std::string::npos;
1659 }
1660
ExtractCommand(std::string & command_string,std::string & command,std::string & suffix,char & quote_char)1661 static bool ExtractCommand(std::string &command_string, std::string &command,
1662 std::string &suffix, char "e_char) {
1663 command.clear();
1664 suffix.clear();
1665 StripLeadingSpaces(command_string);
1666
1667 bool result = false;
1668 quote_char = '\0';
1669
1670 if (!command_string.empty()) {
1671 const char first_char = command_string[0];
1672 if (first_char == '\'' || first_char == '"') {
1673 quote_char = first_char;
1674 const size_t end_quote_pos = command_string.find(quote_char, 1);
1675 if (end_quote_pos == std::string::npos) {
1676 command.swap(command_string);
1677 command_string.erase();
1678 } else {
1679 command.assign(command_string, 1, end_quote_pos - 1);
1680 if (end_quote_pos + 1 < command_string.size())
1681 command_string.erase(0, command_string.find_first_not_of(
1682 k_white_space, end_quote_pos + 1));
1683 else
1684 command_string.erase();
1685 }
1686 } else {
1687 const size_t first_space_pos =
1688 command_string.find_first_of(k_white_space);
1689 if (first_space_pos == std::string::npos) {
1690 command.swap(command_string);
1691 command_string.erase();
1692 } else {
1693 command.assign(command_string, 0, first_space_pos);
1694 command_string.erase(0, command_string.find_first_not_of(
1695 k_white_space, first_space_pos));
1696 }
1697 }
1698 result = true;
1699 }
1700
1701 if (!command.empty()) {
1702 // actual commands can't start with '-' or '_'
1703 if (command[0] != '-' && command[0] != '_') {
1704 size_t pos = command.find_first_not_of(k_valid_command_chars);
1705 if (pos > 0 && pos != std::string::npos) {
1706 suffix.assign(command.begin() + pos, command.end());
1707 command.erase(pos);
1708 }
1709 }
1710 }
1711
1712 return result;
1713 }
1714
BuildAliasResult(llvm::StringRef alias_name,std::string & raw_input_string,std::string & alias_result,CommandReturnObject & result)1715 CommandObject *CommandInterpreter::BuildAliasResult(
1716 llvm::StringRef alias_name, std::string &raw_input_string,
1717 std::string &alias_result, CommandReturnObject &result) {
1718 CommandObject *alias_cmd_obj = nullptr;
1719 Args cmd_args(raw_input_string);
1720 alias_cmd_obj = GetCommandObject(alias_name);
1721 StreamString result_str;
1722
1723 if (!alias_cmd_obj || !alias_cmd_obj->IsAlias()) {
1724 alias_result.clear();
1725 return alias_cmd_obj;
1726 }
1727 std::pair<CommandObjectSP, OptionArgVectorSP> desugared =
1728 ((CommandAlias *)alias_cmd_obj)->Desugar();
1729 OptionArgVectorSP option_arg_vector_sp = desugared.second;
1730 alias_cmd_obj = desugared.first.get();
1731 std::string alias_name_str = std::string(alias_name);
1732 if ((cmd_args.GetArgumentCount() == 0) ||
1733 (alias_name_str != cmd_args.GetArgumentAtIndex(0)))
1734 cmd_args.Unshift(alias_name_str);
1735
1736 result_str.Printf("%s", alias_cmd_obj->GetCommandName().str().c_str());
1737
1738 if (!option_arg_vector_sp.get()) {
1739 alias_result = std::string(result_str.GetString());
1740 return alias_cmd_obj;
1741 }
1742 OptionArgVector *option_arg_vector = option_arg_vector_sp.get();
1743
1744 int value_type;
1745 std::string option;
1746 std::string value;
1747 for (const auto &entry : *option_arg_vector) {
1748 std::tie(option, value_type, value) = entry;
1749 if (option == g_argument) {
1750 result_str.Printf(" %s", value.c_str());
1751 continue;
1752 }
1753
1754 result_str.Printf(" %s", option.c_str());
1755 if (value_type == OptionParser::eNoArgument)
1756 continue;
1757
1758 if (value_type != OptionParser::eOptionalArgument)
1759 result_str.Printf(" ");
1760 int index = GetOptionArgumentPosition(value.c_str());
1761 if (index == 0)
1762 result_str.Printf("%s", value.c_str());
1763 else if (static_cast<size_t>(index) >= cmd_args.GetArgumentCount()) {
1764
1765 result.AppendErrorWithFormat("Not enough arguments provided; you "
1766 "need at least %d arguments to use "
1767 "this alias.\n",
1768 index);
1769 return nullptr;
1770 } else {
1771 const Args::ArgEntry &entry = cmd_args[index];
1772 size_t strpos = raw_input_string.find(entry.c_str());
1773 const char quote_char = entry.GetQuoteChar();
1774 if (strpos != std::string::npos) {
1775 const size_t start_fudge = quote_char == '\0' ? 0 : 1;
1776 const size_t len_fudge = quote_char == '\0' ? 0 : 2;
1777
1778 // Make sure we aren't going outside the bounds of the cmd string:
1779 if (strpos < start_fudge) {
1780 result.AppendError("Unmatched quote at command beginning.");
1781 return nullptr;
1782 }
1783 llvm::StringRef arg_text = entry.ref();
1784 if (strpos - start_fudge + arg_text.size() + len_fudge >
1785 raw_input_string.size()) {
1786 result.AppendError("Unmatched quote at command end.");
1787 return nullptr;
1788 }
1789 raw_input_string = raw_input_string.erase(
1790 strpos - start_fudge,
1791 strlen(cmd_args.GetArgumentAtIndex(index)) + len_fudge);
1792 }
1793 if (quote_char == '\0')
1794 result_str.Printf("%s", cmd_args.GetArgumentAtIndex(index));
1795 else
1796 result_str.Printf("%c%s%c", quote_char, entry.c_str(), quote_char);
1797 }
1798 }
1799
1800 alias_result = std::string(result_str.GetString());
1801 return alias_cmd_obj;
1802 }
1803
PreprocessCommand(std::string & command)1804 Status CommandInterpreter::PreprocessCommand(std::string &command) {
1805 // The command preprocessor needs to do things to the command line before any
1806 // parsing of arguments or anything else is done. The only current stuff that
1807 // gets preprocessed is anything enclosed in backtick ('`') characters is
1808 // evaluated as an expression and the result of the expression must be a
1809 // scalar that can be substituted into the command. An example would be:
1810 // (lldb) memory read `$rsp + 20`
1811 Status error; // Status for any expressions that might not evaluate
1812 size_t start_backtick;
1813 size_t pos = 0;
1814 while ((start_backtick = command.find('`', pos)) != std::string::npos) {
1815 // Stop if an error was encountered during the previous iteration.
1816 if (error.Fail())
1817 break;
1818
1819 if (start_backtick > 0 && command[start_backtick - 1] == '\\') {
1820 // The backtick was preceded by a '\' character, remove the slash and
1821 // don't treat the backtick as the start of an expression.
1822 command.erase(start_backtick - 1, 1);
1823 // No need to add one to start_backtick since we just deleted a char.
1824 pos = start_backtick;
1825 continue;
1826 }
1827
1828 const size_t expr_content_start = start_backtick + 1;
1829 const size_t end_backtick = command.find('`', expr_content_start);
1830
1831 if (end_backtick == std::string::npos) {
1832 // Stop if there's no end backtick.
1833 break;
1834 }
1835
1836 if (end_backtick == expr_content_start) {
1837 // Skip over empty expression. (two backticks in a row)
1838 command.erase(start_backtick, 2);
1839 continue;
1840 }
1841
1842 std::string expr_str(command, expr_content_start,
1843 end_backtick - expr_content_start);
1844 error = PreprocessToken(expr_str);
1845 // We always stop at the first error:
1846 if (error.Fail())
1847 break;
1848
1849 command.erase(start_backtick, end_backtick - start_backtick + 1);
1850 command.insert(start_backtick, std::string(expr_str));
1851 pos = start_backtick + expr_str.size();
1852 }
1853 return error;
1854 }
1855
PreprocessToken(std::string & expr_str)1856 Status CommandInterpreter::PreprocessToken(std::string &expr_str) {
1857 Status error;
1858 ExecutionContext exe_ctx(GetExecutionContext());
1859
1860 // Get a dummy target to allow for calculator mode while processing
1861 // backticks. This also helps break the infinite loop caused when target is
1862 // null.
1863 Target *exe_target = exe_ctx.GetTargetPtr();
1864 Target &target = exe_target ? *exe_target : m_debugger.GetDummyTarget();
1865
1866 ValueObjectSP expr_result_valobj_sp;
1867
1868 EvaluateExpressionOptions options;
1869 options.SetCoerceToId(false);
1870 options.SetUnwindOnError(true);
1871 options.SetIgnoreBreakpoints(true);
1872 options.SetKeepInMemory(false);
1873 options.SetTryAllThreads(true);
1874 options.SetTimeout(std::nullopt);
1875
1876 ExpressionResults expr_result = target.EvaluateExpression(
1877 expr_str.c_str(), exe_ctx.GetFramePtr(), expr_result_valobj_sp, options);
1878
1879 if (expr_result == eExpressionCompleted) {
1880 Scalar scalar;
1881 if (expr_result_valobj_sp)
1882 expr_result_valobj_sp =
1883 expr_result_valobj_sp->GetQualifiedRepresentationIfAvailable(
1884 expr_result_valobj_sp->GetDynamicValueType(), true);
1885 if (expr_result_valobj_sp->ResolveValue(scalar)) {
1886
1887 StreamString value_strm;
1888 const bool show_type = false;
1889 scalar.GetValue(value_strm, show_type);
1890 size_t value_string_size = value_strm.GetSize();
1891 if (value_string_size) {
1892 expr_str = value_strm.GetData();
1893 } else {
1894 error =
1895 Status::FromErrorStringWithFormat("expression value didn't result "
1896 "in a scalar value for the "
1897 "expression '%s'",
1898 expr_str.c_str());
1899 }
1900 } else {
1901 error =
1902 Status::FromErrorStringWithFormat("expression value didn't result "
1903 "in a scalar value for the "
1904 "expression '%s'",
1905 expr_str.c_str());
1906 }
1907 return error;
1908 }
1909
1910 // If we have an error from the expression evaluation it will be in the
1911 // ValueObject error, which won't be success and we will just report it.
1912 // But if for some reason we didn't get a value object at all, then we will
1913 // make up some helpful errors from the expression result.
1914 if (expr_result_valobj_sp)
1915 error = expr_result_valobj_sp->GetError().Clone();
1916
1917 if (error.Success()) {
1918 std::string result = lldb_private::toString(expr_result) +
1919 "for the expression '" + expr_str + "'";
1920 error = Status(result);
1921 }
1922 return error;
1923 }
1924
HandleCommand(const char * command_line,LazyBool lazy_add_to_history,const ExecutionContext & override_context,CommandReturnObject & result)1925 bool CommandInterpreter::HandleCommand(const char *command_line,
1926 LazyBool lazy_add_to_history,
1927 const ExecutionContext &override_context,
1928 CommandReturnObject &result) {
1929
1930 OverrideExecutionContext(override_context);
1931 bool status = HandleCommand(command_line, lazy_add_to_history, result);
1932 RestoreExecutionContext();
1933 return status;
1934 }
1935
HandleCommand(const char * command_line,LazyBool lazy_add_to_history,CommandReturnObject & result,bool force_repeat_command)1936 bool CommandInterpreter::HandleCommand(const char *command_line,
1937 LazyBool lazy_add_to_history,
1938 CommandReturnObject &result,
1939 bool force_repeat_command) {
1940 // These are assigned later in the function but they must be declared before
1941 // the ScopedDispatcher object because we need their destructions to occur
1942 // after the dispatcher's dtor call, which may reference them.
1943 // TODO: This function could be refactored?
1944 std::string parsed_command_args;
1945 CommandObject *cmd_obj = nullptr;
1946
1947 telemetry::ScopedDispatcher<telemetry::CommandInfo> helper(&m_debugger);
1948 const bool detailed_command_telemetry =
1949 telemetry::TelemetryManager::GetInstance()
1950 ->GetConfig()
1951 ->detailed_command_telemetry;
1952 const int command_id = telemetry::CommandInfo::GetNextID();
1953
1954 std::string command_string(command_line);
1955 std::string original_command_string(command_string);
1956 std::string real_original_command_string(command_string);
1957
1958 helper.DispatchNow([&](lldb_private::telemetry::CommandInfo *info) {
1959 info->command_id = command_id;
1960 if (Target *target = GetExecutionContext().GetTargetPtr()) {
1961 // If we have a target attached to this command, then get the UUID.
1962 info->target_uuid = target->GetExecutableModule() != nullptr
1963 ? target->GetExecutableModule()->GetUUID()
1964 : UUID();
1965 }
1966 if (detailed_command_telemetry)
1967 info->original_command = original_command_string;
1968 // The rest (eg., command_name, args, etc) hasn't been parsed yet;
1969 // Those will be collected by the on-exit-callback.
1970 });
1971
1972 helper.DispatchOnExit([&cmd_obj, &parsed_command_args, &result,
1973 detailed_command_telemetry, command_id](
1974 lldb_private::telemetry::CommandInfo *info) {
1975 // TODO: this is logging the time the command-handler finishes.
1976 // But we may want a finer-grain durations too?
1977 // (ie., the execute_time recorded below?)
1978 info->command_id = command_id;
1979 llvm::StringRef command_name =
1980 cmd_obj ? cmd_obj->GetCommandName() : "<not found>";
1981 info->command_name = command_name.str();
1982 info->ret_status = result.GetStatus();
1983 if (std::string error_str = result.GetErrorString(); !error_str.empty())
1984 info->error_data = std::move(error_str);
1985
1986 if (detailed_command_telemetry)
1987 info->args = parsed_command_args;
1988 });
1989
1990 Log *log = GetLog(LLDBLog::Commands);
1991 LLDB_LOGF(log, "Processing command: %s", command_line);
1992 LLDB_SCOPED_TIMERF("Processing command: %s.", command_line);
1993
1994 // Set the command in the CommandReturnObject here so that it's there even if
1995 // the command is interrupted.
1996 result.SetCommand(command_line);
1997
1998 if (INTERRUPT_REQUESTED(GetDebugger(), "Interrupted initiating command")) {
1999 result.AppendError("... Interrupted");
2000 return false;
2001 }
2002
2003 bool add_to_history;
2004 if (lazy_add_to_history == eLazyBoolCalculate)
2005 add_to_history = (m_command_source_depth == 0);
2006 else
2007 add_to_history = (lazy_add_to_history == eLazyBoolYes);
2008
2009 // The same `transcript_item` will be used below to add output and error of
2010 // the command.
2011 StructuredData::DictionarySP transcript_item;
2012 if (GetSaveTranscript()) {
2013 m_transcript_stream << "(lldb) " << command_line << '\n';
2014
2015 transcript_item = std::make_shared<StructuredData::Dictionary>();
2016 transcript_item->AddStringItem("command", command_line);
2017 transcript_item->AddIntegerItem(
2018 "timestampInEpochSeconds",
2019 std::chrono::duration_cast<std::chrono::seconds>(
2020 std::chrono::system_clock::now().time_since_epoch())
2021 .count());
2022 m_transcript.AddItem(transcript_item);
2023 }
2024
2025 bool empty_command = false;
2026 bool comment_command = false;
2027 if (command_string.empty())
2028 empty_command = true;
2029 else {
2030 const char *k_space_characters = "\t\n\v\f\r ";
2031
2032 size_t non_space = command_string.find_first_not_of(k_space_characters);
2033 // Check for empty line or comment line (lines whose first non-space
2034 // character is the comment character for this interpreter)
2035 if (non_space == std::string::npos)
2036 empty_command = true;
2037 else if (command_string[non_space] == m_comment_char)
2038 comment_command = true;
2039 else if (command_string[non_space] == CommandHistory::g_repeat_char) {
2040 llvm::StringRef search_str(command_string);
2041 search_str = search_str.drop_front(non_space);
2042 if (auto hist_str = m_command_history.FindString(search_str)) {
2043 add_to_history = false;
2044 command_string = std::string(*hist_str);
2045 original_command_string = std::string(*hist_str);
2046 } else {
2047 result.AppendErrorWithFormat("Could not find entry: %s in history",
2048 command_string.c_str());
2049 return false;
2050 }
2051 }
2052 }
2053
2054 if (empty_command) {
2055 if (!GetRepeatPreviousCommand()) {
2056 result.SetStatus(eReturnStatusSuccessFinishNoResult);
2057 return true;
2058 }
2059
2060 if (m_command_history.IsEmpty()) {
2061 result.AppendError("empty command");
2062 return false;
2063 }
2064
2065 command_line = m_repeat_command.c_str();
2066 command_string = command_line;
2067 original_command_string = command_line;
2068 if (m_repeat_command.empty()) {
2069 result.AppendError("No auto repeat.");
2070 return false;
2071 }
2072
2073 add_to_history = false;
2074 } else if (comment_command) {
2075 result.SetStatus(eReturnStatusSuccessFinishNoResult);
2076 return true;
2077 }
2078
2079 // Phase 1.
2080
2081 // Before we do ANY kind of argument processing, we need to figure out what
2082 // the real/final command object is for the specified command. This gets
2083 // complicated by the fact that the user could have specified an alias, and,
2084 // in translating the alias, there may also be command options and/or even
2085 // data (including raw text strings) that need to be found and inserted into
2086 // the command line as part of the translation. So this first step is plain
2087 // look-up and replacement, resulting in:
2088 // 1. the command object whose Execute method will actually be called
2089 // 2. a revised command string, with all substitutions and replacements
2090 // taken care of
2091 // From 1 above, we can determine whether the Execute function wants raw
2092 // input or not.
2093
2094 cmd_obj = ResolveCommandImpl(command_string, result);
2095
2096 // We have to preprocess the whole command string for Raw commands, since we
2097 // don't know the structure of the command. For parsed commands, we only
2098 // treat backticks as quote characters specially.
2099 // FIXME: We probably want to have raw commands do their own preprocessing.
2100 // For instance, I don't think people expect substitution in expr expressions.
2101 if (cmd_obj && cmd_obj->WantsRawCommandString()) {
2102 Status error(PreprocessCommand(command_string));
2103
2104 if (error.Fail()) {
2105 result.AppendError(error.AsCString());
2106 return false;
2107 }
2108 }
2109
2110 // Although the user may have abbreviated the command, the command_string now
2111 // has the command expanded to the full name. For example, if the input was
2112 // "br s -n main", command_string is now "breakpoint set -n main".
2113 if (log) {
2114 llvm::StringRef command_name =
2115 cmd_obj ? cmd_obj->GetCommandName() : "<not found>";
2116 LLDB_LOGF(log, "HandleCommand, cmd_obj : '%s'", command_name.str().c_str());
2117 LLDB_LOGF(log, "HandleCommand, (revised) command_string: '%s'",
2118 command_string.c_str());
2119 const bool wants_raw_input =
2120 (cmd_obj != nullptr) ? cmd_obj->WantsRawCommandString() : false;
2121 LLDB_LOGF(log, "HandleCommand, wants_raw_input:'%s'",
2122 wants_raw_input ? "True" : "False");
2123 }
2124
2125 // Phase 2.
2126 // Take care of things like setting up the history command & calling the
2127 // appropriate Execute method on the CommandObject, with the appropriate
2128 // arguments.
2129 StatsDuration execute_time;
2130 if (cmd_obj != nullptr) {
2131 bool generate_repeat_command = add_to_history;
2132 // If we got here when empty_command was true, then this command is a
2133 // stored "repeat command" which we should give a chance to produce it's
2134 // repeat command, even though we don't add repeat commands to the history.
2135 generate_repeat_command |= empty_command;
2136 // For `command regex`, the regex command (ex `bt`) is added to history, but
2137 // the resolved command (ex `thread backtrace`) is _not_ added to history.
2138 // However, the resolved command must be given the opportunity to provide a
2139 // repeat command. `force_repeat_command` supports this case.
2140 generate_repeat_command |= force_repeat_command;
2141 if (generate_repeat_command) {
2142 Args command_args(command_string);
2143 std::optional<std::string> repeat_command =
2144 cmd_obj->GetRepeatCommand(command_args, 0);
2145 if (repeat_command) {
2146 LLDB_LOGF(log, "Repeat command: %s", repeat_command->data());
2147 m_repeat_command.assign(*repeat_command);
2148 } else {
2149 m_repeat_command.assign(original_command_string);
2150 }
2151 }
2152
2153 if (add_to_history)
2154 m_command_history.AppendString(original_command_string);
2155
2156 const std::size_t actual_cmd_name_len = cmd_obj->GetCommandName().size();
2157 if (actual_cmd_name_len < command_string.length())
2158 parsed_command_args = command_string.substr(actual_cmd_name_len);
2159
2160 // Remove any initial spaces
2161 size_t pos = parsed_command_args.find_first_not_of(k_white_space);
2162 if (pos != 0 && pos != std::string::npos)
2163 parsed_command_args.erase(0, pos);
2164
2165 LLDB_LOGF(
2166 log, "HandleCommand, command line after removing command name(s): '%s'",
2167 parsed_command_args.c_str());
2168
2169 // To test whether or not transcript should be saved, `transcript_item` is
2170 // used instead of `GetSaveTranscript()`. This is because the latter will
2171 // fail when the command is "settings set interpreter.save-transcript true".
2172 if (transcript_item) {
2173 transcript_item->AddStringItem("commandName", cmd_obj->GetCommandName());
2174 transcript_item->AddStringItem("commandArguments", parsed_command_args);
2175 }
2176
2177 ElapsedTime elapsed(execute_time);
2178 cmd_obj->SetOriginalCommandString(real_original_command_string);
2179 // Set the indent to the position of the command in the command line.
2180 pos = real_original_command_string.rfind(parsed_command_args);
2181 std::optional<uint16_t> indent;
2182 if (pos != std::string::npos)
2183 indent = pos;
2184 result.SetDiagnosticIndent(indent);
2185 cmd_obj->Execute(parsed_command_args.c_str(), result);
2186 }
2187
2188 LLDB_LOGF(log, "HandleCommand, command %s",
2189 (result.Succeeded() ? "succeeded" : "did not succeed"));
2190
2191 // To test whether or not transcript should be saved, `transcript_item` is
2192 // used instead of `GetSaveTrasncript()`. This is because the latter will
2193 // fail when the command is "settings set interpreter.save-transcript true".
2194 if (transcript_item) {
2195 m_transcript_stream << result.GetOutputString();
2196 m_transcript_stream << result.GetErrorString();
2197
2198 transcript_item->AddStringItem("output", result.GetOutputString());
2199 transcript_item->AddStringItem("error", result.GetErrorString());
2200 transcript_item->AddFloatItem("durationInSeconds",
2201 execute_time.get().count());
2202 }
2203
2204 return result.Succeeded();
2205 }
2206
HandleCompletionMatches(CompletionRequest & request)2207 void CommandInterpreter::HandleCompletionMatches(CompletionRequest &request) {
2208 bool look_for_subcommand = false;
2209
2210 // For any of the command completions a unique match will be a complete word.
2211
2212 if (request.GetParsedLine().GetArgumentCount() == 0) {
2213 // We got nothing on the command line, so return the list of commands
2214 bool include_aliases = true;
2215 StringList new_matches, descriptions;
2216 GetCommandNamesMatchingPartialString("", include_aliases, new_matches,
2217 descriptions);
2218 request.AddCompletions(new_matches, descriptions);
2219 } else if (request.GetCursorIndex() == 0) {
2220 // The cursor is in the first argument, so just do a lookup in the
2221 // dictionary.
2222 StringList new_matches, new_descriptions;
2223 CommandObject *cmd_obj =
2224 GetCommandObject(request.GetParsedLine().GetArgumentAtIndex(0),
2225 &new_matches, &new_descriptions);
2226
2227 if (new_matches.GetSize() && cmd_obj && cmd_obj->IsMultiwordObject() &&
2228 new_matches.GetStringAtIndex(0) != nullptr &&
2229 strcmp(request.GetParsedLine().GetArgumentAtIndex(0),
2230 new_matches.GetStringAtIndex(0)) == 0) {
2231 if (request.GetParsedLine().GetArgumentCount() != 1) {
2232 look_for_subcommand = true;
2233 new_matches.DeleteStringAtIndex(0);
2234 new_descriptions.DeleteStringAtIndex(0);
2235 request.AppendEmptyArgument();
2236 }
2237 }
2238 request.AddCompletions(new_matches, new_descriptions);
2239 }
2240
2241 if (request.GetCursorIndex() > 0 || look_for_subcommand) {
2242 // We are completing further on into a commands arguments, so find the
2243 // command and tell it to complete the command. First see if there is a
2244 // matching initial command:
2245 CommandObject *command_object =
2246 GetCommandObject(request.GetParsedLine().GetArgumentAtIndex(0));
2247 if (command_object) {
2248 request.ShiftArguments();
2249 command_object->HandleCompletion(request);
2250 }
2251 }
2252 }
2253
HandleCompletion(CompletionRequest & request)2254 void CommandInterpreter::HandleCompletion(CompletionRequest &request) {
2255
2256 // Don't complete comments, and if the line we are completing is just the
2257 // history repeat character, substitute the appropriate history line.
2258 llvm::StringRef first_arg = request.GetParsedLine().GetArgumentAtIndex(0);
2259
2260 if (!first_arg.empty()) {
2261 if (first_arg.front() == m_comment_char)
2262 return;
2263 if (first_arg.front() == CommandHistory::g_repeat_char) {
2264 if (auto hist_str = m_command_history.FindString(first_arg))
2265 request.AddCompletion(*hist_str, "Previous command history event",
2266 CompletionMode::RewriteLine);
2267 return;
2268 }
2269 }
2270
2271 HandleCompletionMatches(request);
2272 }
2273
2274 std::optional<std::string>
GetAutoSuggestionForCommand(llvm::StringRef line)2275 CommandInterpreter::GetAutoSuggestionForCommand(llvm::StringRef line) {
2276 if (line.empty())
2277 return std::nullopt;
2278 const size_t s = m_command_history.GetSize();
2279 for (int i = s - 1; i >= 0; --i) {
2280 llvm::StringRef entry = m_command_history.GetStringAtIndex(i);
2281 if (entry.consume_front(line))
2282 return entry.str();
2283 }
2284 return std::nullopt;
2285 }
2286
UpdatePrompt(llvm::StringRef new_prompt)2287 void CommandInterpreter::UpdatePrompt(llvm::StringRef new_prompt) {
2288 EventSP prompt_change_event_sp(
2289 new Event(eBroadcastBitResetPrompt, new EventDataBytes(new_prompt)));
2290
2291 BroadcastEvent(prompt_change_event_sp);
2292 if (m_command_io_handler_sp)
2293 m_command_io_handler_sp->SetPrompt(new_prompt);
2294 }
2295
UpdateUseColor(bool use_color)2296 void CommandInterpreter::UpdateUseColor(bool use_color) {
2297 if (m_command_io_handler_sp)
2298 m_command_io_handler_sp->SetUseColor(use_color);
2299 }
2300
Confirm(llvm::StringRef message,bool default_answer)2301 bool CommandInterpreter::Confirm(llvm::StringRef message, bool default_answer) {
2302 // Check AutoConfirm first:
2303 if (m_debugger.GetAutoConfirm())
2304 return default_answer;
2305
2306 IOHandlerConfirm *confirm =
2307 new IOHandlerConfirm(m_debugger, message, default_answer);
2308 IOHandlerSP io_handler_sp(confirm);
2309 m_debugger.RunIOHandlerSync(io_handler_sp);
2310 return confirm->GetResponse();
2311 }
2312
2313 const CommandAlias *
GetAlias(llvm::StringRef alias_name) const2314 CommandInterpreter::GetAlias(llvm::StringRef alias_name) const {
2315 OptionArgVectorSP ret_val;
2316
2317 auto pos = m_alias_dict.find(alias_name);
2318 if (pos != m_alias_dict.end())
2319 return (CommandAlias *)pos->second.get();
2320
2321 return nullptr;
2322 }
2323
HasCommands() const2324 bool CommandInterpreter::HasCommands() const {
2325 return (!m_command_dict.empty());
2326 }
2327
HasAliases() const2328 bool CommandInterpreter::HasAliases() const { return (!m_alias_dict.empty()); }
2329
HasUserCommands() const2330 bool CommandInterpreter::HasUserCommands() const {
2331 return (!m_user_dict.empty());
2332 }
2333
HasUserMultiwordCommands() const2334 bool CommandInterpreter::HasUserMultiwordCommands() const {
2335 return (!m_user_mw_dict.empty());
2336 }
2337
HasAliasOptions() const2338 bool CommandInterpreter::HasAliasOptions() const { return HasAliases(); }
2339
BuildAliasCommandArgs(CommandObject * alias_cmd_obj,const char * alias_name,Args & cmd_args,std::string & raw_input_string,CommandReturnObject & result)2340 void CommandInterpreter::BuildAliasCommandArgs(CommandObject *alias_cmd_obj,
2341 const char *alias_name,
2342 Args &cmd_args,
2343 std::string &raw_input_string,
2344 CommandReturnObject &result) {
2345 OptionArgVectorSP option_arg_vector_sp =
2346 GetAlias(alias_name)->GetOptionArguments();
2347
2348 bool wants_raw_input = alias_cmd_obj->WantsRawCommandString();
2349
2350 // Make sure that the alias name is the 0th element in cmd_args
2351 std::string alias_name_str = alias_name;
2352 if (alias_name_str != cmd_args.GetArgumentAtIndex(0))
2353 cmd_args.Unshift(alias_name_str);
2354
2355 Args new_args(alias_cmd_obj->GetCommandName());
2356 if (new_args.GetArgumentCount() == 2)
2357 new_args.Shift();
2358
2359 if (option_arg_vector_sp.get()) {
2360 if (wants_raw_input) {
2361 // We have a command that both has command options and takes raw input.
2362 // Make *sure* it has a " -- " in the right place in the
2363 // raw_input_string.
2364 size_t pos = raw_input_string.find(" -- ");
2365 if (pos == std::string::npos) {
2366 // None found; assume it goes at the beginning of the raw input string
2367 raw_input_string.insert(0, " -- ");
2368 }
2369 }
2370
2371 OptionArgVector *option_arg_vector = option_arg_vector_sp.get();
2372 const size_t old_size = cmd_args.GetArgumentCount();
2373 std::vector<bool> used(old_size + 1, false);
2374
2375 used[0] = true;
2376
2377 int value_type;
2378 std::string option;
2379 std::string value;
2380 for (const auto &option_entry : *option_arg_vector) {
2381 std::tie(option, value_type, value) = option_entry;
2382 if (option == g_argument) {
2383 if (!wants_raw_input || (value != "--")) {
2384 // Since we inserted this above, make sure we don't insert it twice
2385 new_args.AppendArgument(value);
2386 }
2387 continue;
2388 }
2389
2390 if (value_type != OptionParser::eOptionalArgument)
2391 new_args.AppendArgument(option);
2392
2393 if (value == g_no_argument)
2394 continue;
2395
2396 int index = GetOptionArgumentPosition(value.c_str());
2397 if (index == 0) {
2398 // value was NOT a positional argument; must be a real value
2399 if (value_type != OptionParser::eOptionalArgument)
2400 new_args.AppendArgument(value);
2401 else {
2402 new_args.AppendArgument(option + value);
2403 }
2404
2405 } else if (static_cast<size_t>(index) >= cmd_args.GetArgumentCount()) {
2406 result.AppendErrorWithFormat("Not enough arguments provided; you "
2407 "need at least %d arguments to use "
2408 "this alias.\n",
2409 index);
2410 return;
2411 } else {
2412 // Find and remove cmd_args.GetArgumentAtIndex(i) from raw_input_string
2413 size_t strpos =
2414 raw_input_string.find(cmd_args.GetArgumentAtIndex(index));
2415 if (strpos != std::string::npos) {
2416 raw_input_string = raw_input_string.erase(
2417 strpos, strlen(cmd_args.GetArgumentAtIndex(index)));
2418 }
2419
2420 if (value_type != OptionParser::eOptionalArgument)
2421 new_args.AppendArgument(cmd_args.GetArgumentAtIndex(index));
2422 else {
2423 new_args.AppendArgument(option + cmd_args.GetArgumentAtIndex(index));
2424 }
2425 used[index] = true;
2426 }
2427 }
2428
2429 for (auto entry : llvm::enumerate(cmd_args.entries())) {
2430 if (!used[entry.index()] && !wants_raw_input)
2431 new_args.AppendArgument(entry.value().ref());
2432 }
2433
2434 cmd_args.Clear();
2435 cmd_args.SetArguments(new_args.GetArgumentCount(),
2436 new_args.GetConstArgumentVector());
2437 } else {
2438 result.SetStatus(eReturnStatusSuccessFinishNoResult);
2439 // This alias was not created with any options; nothing further needs to be
2440 // done, unless it is a command that wants raw input, in which case we need
2441 // to clear the rest of the data from cmd_args, since its in the raw input
2442 // string.
2443 if (wants_raw_input) {
2444 cmd_args.Clear();
2445 cmd_args.SetArguments(new_args.GetArgumentCount(),
2446 new_args.GetConstArgumentVector());
2447 }
2448 return;
2449 }
2450
2451 result.SetStatus(eReturnStatusSuccessFinishNoResult);
2452 }
2453
GetOptionArgumentPosition(const char * in_string)2454 int CommandInterpreter::GetOptionArgumentPosition(const char *in_string) {
2455 int position = 0; // Any string that isn't an argument position, i.e. '%'
2456 // followed by an integer, gets a position
2457 // of zero.
2458
2459 const char *cptr = in_string;
2460
2461 // Does it start with '%'
2462 if (cptr[0] == '%') {
2463 ++cptr;
2464
2465 // Is the rest of it entirely digits?
2466 if (isdigit(cptr[0])) {
2467 const char *start = cptr;
2468 while (isdigit(cptr[0]))
2469 ++cptr;
2470
2471 // We've gotten to the end of the digits; are we at the end of the
2472 // string?
2473 if (cptr[0] == '\0')
2474 position = atoi(start);
2475 }
2476 }
2477
2478 return position;
2479 }
2480
GetHomeInitFile(llvm::SmallVectorImpl<char> & init_file,llvm::StringRef suffix={})2481 static void GetHomeInitFile(llvm::SmallVectorImpl<char> &init_file,
2482 llvm::StringRef suffix = {}) {
2483 std::string init_file_name = ".lldbinit";
2484 if (!suffix.empty()) {
2485 init_file_name.append("-");
2486 init_file_name.append(suffix.str());
2487 }
2488
2489 FileSystem::Instance().GetHomeDirectory(init_file);
2490 llvm::sys::path::append(init_file, init_file_name);
2491
2492 FileSystem::Instance().Resolve(init_file);
2493 }
2494
GetHomeREPLInitFile(llvm::SmallVectorImpl<char> & init_file,LanguageType language)2495 static void GetHomeREPLInitFile(llvm::SmallVectorImpl<char> &init_file,
2496 LanguageType language) {
2497 if (language == eLanguageTypeUnknown) {
2498 LanguageSet repl_languages = Language::GetLanguagesSupportingREPLs();
2499 if (auto main_repl_language = repl_languages.GetSingularLanguage())
2500 language = *main_repl_language;
2501 else
2502 return;
2503 }
2504
2505 std::string init_file_name =
2506 (llvm::Twine(".lldbinit-") +
2507 llvm::Twine(Language::GetNameForLanguageType(language)) +
2508 llvm::Twine("-repl"))
2509 .str();
2510 FileSystem::Instance().GetHomeDirectory(init_file);
2511 llvm::sys::path::append(init_file, init_file_name);
2512 FileSystem::Instance().Resolve(init_file);
2513 }
2514
GetCwdInitFile(llvm::SmallVectorImpl<char> & init_file)2515 static void GetCwdInitFile(llvm::SmallVectorImpl<char> &init_file) {
2516 llvm::StringRef s = ".lldbinit";
2517 init_file.assign(s.begin(), s.end());
2518 FileSystem::Instance().Resolve(init_file);
2519 }
2520
SourceInitFile(FileSpec file,CommandReturnObject & result)2521 void CommandInterpreter::SourceInitFile(FileSpec file,
2522 CommandReturnObject &result) {
2523 assert(!m_skip_lldbinit_files);
2524
2525 if (!FileSystem::Instance().Exists(file)) {
2526 result.SetStatus(eReturnStatusSuccessFinishNoResult);
2527 return;
2528 }
2529
2530 // Use HandleCommand to 'source' the given file; this will do the actual
2531 // broadcasting of the commands back to any appropriate listener (see
2532 // CommandObjectSource::Execute for more details).
2533 const bool saved_batch = SetBatchCommandMode(true);
2534 CommandInterpreterRunOptions options;
2535 options.SetSilent(true);
2536 options.SetPrintErrors(true);
2537 options.SetStopOnError(false);
2538 options.SetStopOnContinue(true);
2539 HandleCommandsFromFile(file, options, result);
2540 SetBatchCommandMode(saved_batch);
2541 }
2542
SourceInitFileCwd(CommandReturnObject & result)2543 void CommandInterpreter::SourceInitFileCwd(CommandReturnObject &result) {
2544 if (m_skip_lldbinit_files) {
2545 result.SetStatus(eReturnStatusSuccessFinishNoResult);
2546 return;
2547 }
2548
2549 llvm::SmallString<128> init_file;
2550 GetCwdInitFile(init_file);
2551 if (!FileSystem::Instance().Exists(init_file)) {
2552 result.SetStatus(eReturnStatusSuccessFinishNoResult);
2553 return;
2554 }
2555
2556 LoadCWDlldbinitFile should_load =
2557 Target::GetGlobalProperties().GetLoadCWDlldbinitFile();
2558
2559 switch (should_load) {
2560 case eLoadCWDlldbinitFalse:
2561 result.SetStatus(eReturnStatusSuccessFinishNoResult);
2562 break;
2563 case eLoadCWDlldbinitTrue:
2564 SourceInitFile(FileSpec(init_file.str()), result);
2565 break;
2566 case eLoadCWDlldbinitWarn: {
2567 llvm::SmallString<128> home_init_file;
2568 GetHomeInitFile(home_init_file);
2569 if (llvm::sys::path::parent_path(init_file) ==
2570 llvm::sys::path::parent_path(home_init_file)) {
2571 result.SetStatus(eReturnStatusSuccessFinishNoResult);
2572 } else {
2573 result.AppendError(InitFileWarning);
2574 }
2575 }
2576 }
2577 }
2578
2579 /// We will first see if there is an application specific ".lldbinit" file
2580 /// whose name is "~/.lldbinit" followed by a "-" and the name of the program.
2581 /// If this file doesn't exist, we fall back to the REPL init file or the
2582 /// default home init file in "~/.lldbinit".
SourceInitFileHome(CommandReturnObject & result,bool is_repl)2583 void CommandInterpreter::SourceInitFileHome(CommandReturnObject &result,
2584 bool is_repl) {
2585 if (m_skip_lldbinit_files) {
2586 result.SetStatus(eReturnStatusSuccessFinishNoResult);
2587 return;
2588 }
2589
2590 llvm::SmallString<128> init_file;
2591
2592 if (is_repl)
2593 GetHomeREPLInitFile(init_file, GetDebugger().GetREPLLanguage());
2594
2595 if (init_file.empty())
2596 GetHomeInitFile(init_file);
2597
2598 if (!m_skip_app_init_files) {
2599 llvm::StringRef program_name =
2600 HostInfo::GetProgramFileSpec().GetFilename().GetStringRef();
2601 llvm::SmallString<128> program_init_file;
2602 GetHomeInitFile(program_init_file, program_name);
2603 if (FileSystem::Instance().Exists(program_init_file))
2604 init_file = program_init_file;
2605 }
2606
2607 SourceInitFile(FileSpec(init_file.str()), result);
2608 }
2609
SourceInitFileGlobal(CommandReturnObject & result)2610 void CommandInterpreter::SourceInitFileGlobal(CommandReturnObject &result) {
2611 #ifdef LLDB_GLOBAL_INIT_DIRECTORY
2612 if (!m_skip_lldbinit_files) {
2613 FileSpec init_file(LLDB_GLOBAL_INIT_DIRECTORY);
2614 if (init_file)
2615 init_file.MakeAbsolute(HostInfo::GetShlibDir());
2616
2617 init_file.AppendPathComponent("lldbinit");
2618 SourceInitFile(init_file, result);
2619 return;
2620 }
2621 #endif
2622 result.SetStatus(eReturnStatusSuccessFinishNoResult);
2623 }
2624
GetCommandPrefix()2625 const char *CommandInterpreter::GetCommandPrefix() {
2626 const char *prefix = GetDebugger().GetIOHandlerCommandPrefix();
2627 return prefix == nullptr ? "" : prefix;
2628 }
2629
GetPlatform(bool prefer_target_platform)2630 PlatformSP CommandInterpreter::GetPlatform(bool prefer_target_platform) {
2631 PlatformSP platform_sp;
2632 if (prefer_target_platform) {
2633 ExecutionContext exe_ctx(GetExecutionContext());
2634 Target *target = exe_ctx.GetTargetPtr();
2635 if (target)
2636 platform_sp = target->GetPlatform();
2637 }
2638
2639 if (!platform_sp)
2640 platform_sp = m_debugger.GetPlatformList().GetSelectedPlatform();
2641 return platform_sp;
2642 }
2643
DidProcessStopAbnormally() const2644 bool CommandInterpreter::DidProcessStopAbnormally() const {
2645 auto exe_ctx = GetExecutionContext();
2646 TargetSP target_sp = exe_ctx.GetTargetSP();
2647 if (!target_sp)
2648 return false;
2649
2650 ProcessSP process_sp(target_sp->GetProcessSP());
2651 if (!process_sp)
2652 return false;
2653
2654 if (eStateStopped != process_sp->GetState())
2655 return false;
2656
2657 for (const auto &thread_sp : process_sp->GetThreadList().Threads()) {
2658 StopInfoSP stop_info = thread_sp->GetStopInfo();
2659 if (!stop_info) {
2660 // If there's no stop_info, keep iterating through the other threads;
2661 // it's enough that any thread has got a stop_info that indicates
2662 // an abnormal stop, to consider the process to be stopped abnormally.
2663 continue;
2664 }
2665
2666 const StopReason reason = stop_info->GetStopReason();
2667 if (reason == eStopReasonException ||
2668 reason == eStopReasonInstrumentation ||
2669 reason == eStopReasonProcessorTrace || reason == eStopReasonInterrupt ||
2670 reason == eStopReasonHistoryBoundary)
2671 return true;
2672
2673 if (reason == eStopReasonSignal) {
2674 const auto stop_signal = static_cast<int32_t>(stop_info->GetValue());
2675 UnixSignalsSP signals_sp = process_sp->GetUnixSignals();
2676 if (!signals_sp || !signals_sp->SignalIsValid(stop_signal))
2677 // The signal is unknown, treat it as abnormal.
2678 return true;
2679
2680 const auto sigint_num = signals_sp->GetSignalNumberFromName("SIGINT");
2681 const auto sigstop_num = signals_sp->GetSignalNumberFromName("SIGSTOP");
2682 if ((stop_signal != sigint_num) && (stop_signal != sigstop_num))
2683 // The signal very likely implies a crash.
2684 return true;
2685 }
2686 }
2687
2688 return false;
2689 }
2690
HandleCommands(const StringList & commands,const ExecutionContext & override_context,const CommandInterpreterRunOptions & options,CommandReturnObject & result)2691 void CommandInterpreter::HandleCommands(
2692 const StringList &commands, const ExecutionContext &override_context,
2693 const CommandInterpreterRunOptions &options, CommandReturnObject &result) {
2694
2695 OverrideExecutionContext(override_context);
2696 HandleCommands(commands, options, result);
2697 RestoreExecutionContext();
2698 }
2699
HandleCommands(const StringList & commands,const CommandInterpreterRunOptions & options,CommandReturnObject & result)2700 void CommandInterpreter::HandleCommands(
2701 const StringList &commands, const CommandInterpreterRunOptions &options,
2702 CommandReturnObject &result) {
2703 size_t num_lines = commands.GetSize();
2704
2705 // If we are going to continue past a "continue" then we need to run the
2706 // commands synchronously. Make sure you reset this value anywhere you return
2707 // from the function.
2708
2709 bool old_async_execution = m_debugger.GetAsyncExecution();
2710
2711 if (!options.GetStopOnContinue()) {
2712 m_debugger.SetAsyncExecution(false);
2713 }
2714
2715 for (size_t idx = 0; idx < num_lines; idx++) {
2716 const char *cmd = commands.GetStringAtIndex(idx);
2717 if (cmd[0] == '\0')
2718 continue;
2719
2720 if (options.GetEchoCommands()) {
2721 // TODO: Add Stream support.
2722 result.AppendMessageWithFormat("%s %s\n",
2723 m_debugger.GetPrompt().str().c_str(), cmd);
2724 }
2725
2726 CommandReturnObject tmp_result(m_debugger.GetUseColor());
2727 tmp_result.SetInteractive(result.GetInteractive());
2728 tmp_result.SetSuppressImmediateOutput(true);
2729
2730 // We might call into a regex or alias command, in which case the
2731 // add_to_history will get lost. This m_command_source_depth dingus is the
2732 // way we turn off adding to the history in that case, so set it up here.
2733 if (!options.GetAddToHistory())
2734 m_command_source_depth++;
2735 bool success = HandleCommand(cmd, options.m_add_to_history, tmp_result);
2736 if (!options.GetAddToHistory())
2737 m_command_source_depth--;
2738
2739 if (options.GetPrintResults()) {
2740 if (tmp_result.Succeeded())
2741 result.AppendMessage(tmp_result.GetOutputString());
2742 }
2743
2744 if (!success || !tmp_result.Succeeded()) {
2745 std::string error_msg = tmp_result.GetErrorString();
2746 if (error_msg.empty())
2747 error_msg = "<unknown error>.\n";
2748 if (options.GetStopOnError()) {
2749 result.AppendErrorWithFormatv("Aborting reading of commands after "
2750 "command #{0}: '{1}' failed with {2}",
2751 (uint64_t)idx, cmd, error_msg);
2752 m_debugger.SetAsyncExecution(old_async_execution);
2753 return;
2754 }
2755 if (options.GetPrintResults()) {
2756 result.AppendMessageWithFormatv("Command #{0} '{1}' failed with {2}",
2757 (uint64_t)idx + 1, cmd, error_msg);
2758 }
2759 }
2760
2761 if (result.GetImmediateOutputStream())
2762 result.GetImmediateOutputStream()->Flush();
2763
2764 if (result.GetImmediateErrorStream())
2765 result.GetImmediateErrorStream()->Flush();
2766
2767 // N.B. Can't depend on DidChangeProcessState, because the state coming
2768 // into the command execution could be running (for instance in Breakpoint
2769 // Commands. So we check the return value to see if it is has running in
2770 // it.
2771 if ((tmp_result.GetStatus() == eReturnStatusSuccessContinuingNoResult) ||
2772 (tmp_result.GetStatus() == eReturnStatusSuccessContinuingResult)) {
2773 if (options.GetStopOnContinue()) {
2774 // If we caused the target to proceed, and we're going to stop in that
2775 // case, set the status in our real result before returning. This is
2776 // an error if the continue was not the last command in the set of
2777 // commands to be run.
2778 if (idx != num_lines - 1)
2779 result.AppendErrorWithFormat(
2780 "Aborting reading of commands after command #%" PRIu64
2781 ": '%s' continued the target.\n",
2782 (uint64_t)idx + 1, cmd);
2783 else
2784 result.AppendMessageWithFormat("Command #%" PRIu64
2785 " '%s' continued the target.\n",
2786 (uint64_t)idx + 1, cmd);
2787
2788 result.SetStatus(tmp_result.GetStatus());
2789 m_debugger.SetAsyncExecution(old_async_execution);
2790
2791 return;
2792 }
2793 }
2794
2795 // Also check for "stop on crash here:
2796 if (tmp_result.GetDidChangeProcessState() && options.GetStopOnCrash() &&
2797 DidProcessStopAbnormally()) {
2798 if (idx != num_lines - 1)
2799 result.AppendErrorWithFormat(
2800 "Aborting reading of commands after command #%" PRIu64
2801 ": '%s' stopped with a signal or exception.\n",
2802 (uint64_t)idx + 1, cmd);
2803 else
2804 result.AppendMessageWithFormat(
2805 "Command #%" PRIu64 " '%s' stopped with a signal or exception.\n",
2806 (uint64_t)idx + 1, cmd);
2807
2808 result.SetStatus(tmp_result.GetStatus());
2809 m_debugger.SetAsyncExecution(old_async_execution);
2810
2811 return;
2812 }
2813 }
2814
2815 result.SetStatus(eReturnStatusSuccessFinishResult);
2816 m_debugger.SetAsyncExecution(old_async_execution);
2817 }
2818
2819 // Make flags that we can pass into the IOHandler so our delegates can do the
2820 // right thing
2821 enum {
2822 eHandleCommandFlagStopOnContinue = (1u << 0),
2823 eHandleCommandFlagStopOnError = (1u << 1),
2824 eHandleCommandFlagEchoCommand = (1u << 2),
2825 eHandleCommandFlagEchoCommentCommand = (1u << 3),
2826 eHandleCommandFlagPrintResult = (1u << 4),
2827 eHandleCommandFlagPrintErrors = (1u << 5),
2828 eHandleCommandFlagStopOnCrash = (1u << 6),
2829 eHandleCommandFlagAllowRepeats = (1u << 7)
2830 };
2831
HandleCommandsFromFile(FileSpec & cmd_file,const ExecutionContext & context,const CommandInterpreterRunOptions & options,CommandReturnObject & result)2832 void CommandInterpreter::HandleCommandsFromFile(
2833 FileSpec &cmd_file, const ExecutionContext &context,
2834 const CommandInterpreterRunOptions &options, CommandReturnObject &result) {
2835 OverrideExecutionContext(context);
2836 HandleCommandsFromFile(cmd_file, options, result);
2837 RestoreExecutionContext();
2838 }
2839
HandleCommandsFromFile(FileSpec & cmd_file,const CommandInterpreterRunOptions & options,CommandReturnObject & result)2840 void CommandInterpreter::HandleCommandsFromFile(
2841 FileSpec &cmd_file, const CommandInterpreterRunOptions &options,
2842 CommandReturnObject &result) {
2843 if (!FileSystem::Instance().Exists(cmd_file)) {
2844 result.AppendErrorWithFormat(
2845 "Error reading commands from file %s - file not found.\n",
2846 cmd_file.GetFilename().AsCString("<Unknown>"));
2847 return;
2848 }
2849
2850 std::string cmd_file_path = cmd_file.GetPath();
2851 auto input_file_up =
2852 FileSystem::Instance().Open(cmd_file, File::eOpenOptionReadOnly);
2853 if (!input_file_up) {
2854 std::string error = llvm::toString(input_file_up.takeError());
2855 result.AppendErrorWithFormatv(
2856 "error: an error occurred read file '{0}': {1}\n", cmd_file_path,
2857 llvm::fmt_consume(input_file_up.takeError()));
2858 return;
2859 }
2860 FileSP input_file_sp = FileSP(std::move(input_file_up.get()));
2861
2862 Debugger &debugger = GetDebugger();
2863
2864 uint32_t flags = 0;
2865
2866 if (options.m_stop_on_continue == eLazyBoolCalculate) {
2867 if (m_command_source_flags.empty()) {
2868 // Stop on continue by default
2869 flags |= eHandleCommandFlagStopOnContinue;
2870 } else if (m_command_source_flags.back() &
2871 eHandleCommandFlagStopOnContinue) {
2872 flags |= eHandleCommandFlagStopOnContinue;
2873 }
2874 } else if (options.m_stop_on_continue == eLazyBoolYes) {
2875 flags |= eHandleCommandFlagStopOnContinue;
2876 }
2877
2878 if (options.m_stop_on_error == eLazyBoolCalculate) {
2879 if (m_command_source_flags.empty()) {
2880 if (GetStopCmdSourceOnError())
2881 flags |= eHandleCommandFlagStopOnError;
2882 } else if (m_command_source_flags.back() & eHandleCommandFlagStopOnError) {
2883 flags |= eHandleCommandFlagStopOnError;
2884 }
2885 } else if (options.m_stop_on_error == eLazyBoolYes) {
2886 flags |= eHandleCommandFlagStopOnError;
2887 }
2888
2889 // stop-on-crash can only be set, if it is present in all levels of
2890 // pushed flag sets.
2891 if (options.GetStopOnCrash()) {
2892 if (m_command_source_flags.empty()) {
2893 flags |= eHandleCommandFlagStopOnCrash;
2894 } else if (m_command_source_flags.back() & eHandleCommandFlagStopOnCrash) {
2895 flags |= eHandleCommandFlagStopOnCrash;
2896 }
2897 }
2898
2899 if (options.m_echo_commands == eLazyBoolCalculate) {
2900 if (m_command_source_flags.empty()) {
2901 // Echo command by default
2902 flags |= eHandleCommandFlagEchoCommand;
2903 } else if (m_command_source_flags.back() & eHandleCommandFlagEchoCommand) {
2904 flags |= eHandleCommandFlagEchoCommand;
2905 }
2906 } else if (options.m_echo_commands == eLazyBoolYes) {
2907 flags |= eHandleCommandFlagEchoCommand;
2908 }
2909
2910 // We will only ever ask for this flag, if we echo commands in general.
2911 if (options.m_echo_comment_commands == eLazyBoolCalculate) {
2912 if (m_command_source_flags.empty()) {
2913 // Echo comments by default
2914 flags |= eHandleCommandFlagEchoCommentCommand;
2915 } else if (m_command_source_flags.back() &
2916 eHandleCommandFlagEchoCommentCommand) {
2917 flags |= eHandleCommandFlagEchoCommentCommand;
2918 }
2919 } else if (options.m_echo_comment_commands == eLazyBoolYes) {
2920 flags |= eHandleCommandFlagEchoCommentCommand;
2921 }
2922
2923 if (options.m_print_results == eLazyBoolCalculate) {
2924 if (m_command_source_flags.empty()) {
2925 // Print output by default
2926 flags |= eHandleCommandFlagPrintResult;
2927 } else if (m_command_source_flags.back() & eHandleCommandFlagPrintResult) {
2928 flags |= eHandleCommandFlagPrintResult;
2929 }
2930 } else if (options.m_print_results == eLazyBoolYes) {
2931 flags |= eHandleCommandFlagPrintResult;
2932 }
2933
2934 if (options.m_print_errors == eLazyBoolCalculate) {
2935 if (m_command_source_flags.empty()) {
2936 // Print output by default
2937 flags |= eHandleCommandFlagPrintErrors;
2938 } else if (m_command_source_flags.back() & eHandleCommandFlagPrintErrors) {
2939 flags |= eHandleCommandFlagPrintErrors;
2940 }
2941 } else if (options.m_print_errors == eLazyBoolYes) {
2942 flags |= eHandleCommandFlagPrintErrors;
2943 }
2944
2945 if (flags & eHandleCommandFlagPrintResult) {
2946 debugger.GetOutputFileSP()->Printf("Executing commands in '%s'.\n",
2947 cmd_file_path.c_str());
2948 }
2949
2950 // Used for inheriting the right settings when "command source" might
2951 // have nested "command source" commands
2952 lldb::LockableStreamFileSP empty_stream_sp;
2953 m_command_source_flags.push_back(flags);
2954 IOHandlerSP io_handler_sp(new IOHandlerEditline(
2955 debugger, IOHandler::Type::CommandInterpreter, input_file_sp,
2956 empty_stream_sp, // Pass in an empty stream so we inherit the top
2957 // input reader output stream
2958 empty_stream_sp, // Pass in an empty stream so we inherit the top
2959 // input reader error stream
2960 flags,
2961 nullptr, // Pass in NULL for "editline_name" so no history is saved,
2962 // or written
2963 debugger.GetPrompt(), llvm::StringRef(),
2964 false, // Not multi-line
2965 debugger.GetUseColor(), 0, *this));
2966 const bool old_async_execution = debugger.GetAsyncExecution();
2967
2968 // Set synchronous execution if we are not stopping on continue
2969 if ((flags & eHandleCommandFlagStopOnContinue) == 0)
2970 debugger.SetAsyncExecution(false);
2971
2972 m_command_source_depth++;
2973 m_command_source_dirs.push_back(cmd_file.CopyByRemovingLastPathComponent());
2974
2975 debugger.RunIOHandlerSync(io_handler_sp);
2976 if (!m_command_source_flags.empty())
2977 m_command_source_flags.pop_back();
2978
2979 m_command_source_dirs.pop_back();
2980 m_command_source_depth--;
2981
2982 result.SetStatus(eReturnStatusSuccessFinishNoResult);
2983 debugger.SetAsyncExecution(old_async_execution);
2984 }
2985
GetSynchronous()2986 bool CommandInterpreter::GetSynchronous() { return m_synchronous_execution; }
2987
SetSynchronous(bool value)2988 void CommandInterpreter::SetSynchronous(bool value) {
2989 m_synchronous_execution = value;
2990 }
2991
OutputFormattedHelpText(Stream & strm,llvm::StringRef prefix,llvm::StringRef help_text)2992 void CommandInterpreter::OutputFormattedHelpText(Stream &strm,
2993 llvm::StringRef prefix,
2994 llvm::StringRef help_text) {
2995 const uint32_t max_columns = m_debugger.GetTerminalWidth();
2996
2997 size_t line_width_max = max_columns - prefix.size();
2998 if (line_width_max < 16)
2999 line_width_max = help_text.size() + prefix.size();
3000
3001 strm.IndentMore(prefix.size());
3002 bool prefixed_yet = false;
3003 // Even if we have no help text we still want to emit the command name.
3004 if (help_text.empty())
3005 help_text = "No help text";
3006 while (!help_text.empty()) {
3007 // Prefix the first line, indent subsequent lines to line up
3008 if (!prefixed_yet) {
3009 strm << prefix;
3010 prefixed_yet = true;
3011 } else
3012 strm.Indent();
3013
3014 // Never print more than the maximum on one line.
3015 llvm::StringRef this_line = help_text.substr(0, line_width_max);
3016
3017 // Always break on an explicit newline.
3018 std::size_t first_newline = this_line.find_first_of("\n");
3019
3020 // Don't break on space/tab unless the text is too long to fit on one line.
3021 std::size_t last_space = llvm::StringRef::npos;
3022 if (this_line.size() != help_text.size())
3023 last_space = this_line.find_last_of(" \t");
3024
3025 // Break at whichever condition triggered first.
3026 this_line = this_line.substr(0, std::min(first_newline, last_space));
3027 strm.PutCString(this_line);
3028 strm.EOL();
3029
3030 // Remove whitespace / newlines after breaking.
3031 help_text = help_text.drop_front(this_line.size()).ltrim();
3032 }
3033 strm.IndentLess(prefix.size());
3034 }
3035
OutputFormattedHelpText(Stream & strm,llvm::StringRef word_text,llvm::StringRef separator,llvm::StringRef help_text,size_t max_word_len)3036 void CommandInterpreter::OutputFormattedHelpText(Stream &strm,
3037 llvm::StringRef word_text,
3038 llvm::StringRef separator,
3039 llvm::StringRef help_text,
3040 size_t max_word_len) {
3041 StreamString prefix_stream;
3042 prefix_stream.Printf(" %-*s %*s ", (int)max_word_len, word_text.data(),
3043 (int)separator.size(), separator.data());
3044 OutputFormattedHelpText(strm, prefix_stream.GetString(), help_text);
3045 }
3046
OutputHelpText(Stream & strm,llvm::StringRef word_text,llvm::StringRef separator,llvm::StringRef help_text,uint32_t max_word_len)3047 void CommandInterpreter::OutputHelpText(Stream &strm, llvm::StringRef word_text,
3048 llvm::StringRef separator,
3049 llvm::StringRef help_text,
3050 uint32_t max_word_len) {
3051 int indent_size = max_word_len + separator.size() + 2;
3052
3053 strm.IndentMore(indent_size);
3054
3055 StreamString text_strm;
3056 text_strm.Printf("%-*s ", (int)max_word_len, word_text.data());
3057 text_strm << separator << " " << help_text;
3058
3059 const uint32_t max_columns = m_debugger.GetTerminalWidth();
3060
3061 llvm::StringRef text = text_strm.GetString();
3062
3063 uint32_t chars_left = max_columns;
3064
3065 auto nextWordLength = [](llvm::StringRef S) {
3066 size_t pos = S.find(' ');
3067 return pos == llvm::StringRef::npos ? S.size() : pos;
3068 };
3069
3070 while (!text.empty()) {
3071 if (text.front() == '\n' ||
3072 (text.front() == ' ' && nextWordLength(text.ltrim(' ')) > chars_left)) {
3073 strm.EOL();
3074 strm.Indent();
3075 chars_left = max_columns - indent_size;
3076 if (text.front() == '\n')
3077 text = text.drop_front();
3078 else
3079 text = text.ltrim(' ');
3080 } else {
3081 strm.PutChar(text.front());
3082 --chars_left;
3083 text = text.drop_front();
3084 }
3085 }
3086
3087 strm.EOL();
3088 strm.IndentLess(indent_size);
3089 }
3090
FindCommandsForApropos(llvm::StringRef search_word,StringList & commands_found,StringList & commands_help,const CommandObject::CommandMap & command_map)3091 void CommandInterpreter::FindCommandsForApropos(
3092 llvm::StringRef search_word, StringList &commands_found,
3093 StringList &commands_help, const CommandObject::CommandMap &command_map) {
3094 for (const auto &pair : command_map) {
3095 llvm::StringRef command_name = pair.first;
3096 CommandObject *cmd_obj = pair.second.get();
3097
3098 const bool search_short_help = true;
3099 const bool search_long_help = false;
3100 const bool search_syntax = false;
3101 const bool search_options = false;
3102 if (command_name.contains_insensitive(search_word) ||
3103 cmd_obj->HelpTextContainsWord(search_word, search_short_help,
3104 search_long_help, search_syntax,
3105 search_options)) {
3106 commands_found.AppendString(command_name);
3107 commands_help.AppendString(cmd_obj->GetHelp());
3108 }
3109
3110 if (auto *multiword_cmd = cmd_obj->GetAsMultiwordCommand()) {
3111 StringList subcommands_found;
3112 FindCommandsForApropos(search_word, subcommands_found, commands_help,
3113 multiword_cmd->GetSubcommandDictionary());
3114 for (const auto &subcommand_name : subcommands_found) {
3115 std::string qualified_name =
3116 (command_name + " " + subcommand_name).str();
3117 commands_found.AppendString(qualified_name);
3118 }
3119 }
3120 }
3121 }
3122
FindCommandsForApropos(llvm::StringRef search_word,StringList & commands_found,StringList & commands_help,bool search_builtin_commands,bool search_user_commands,bool search_alias_commands,bool search_user_mw_commands)3123 void CommandInterpreter::FindCommandsForApropos(llvm::StringRef search_word,
3124 StringList &commands_found,
3125 StringList &commands_help,
3126 bool search_builtin_commands,
3127 bool search_user_commands,
3128 bool search_alias_commands,
3129 bool search_user_mw_commands) {
3130 CommandObject::CommandMap::const_iterator pos;
3131
3132 if (search_builtin_commands)
3133 FindCommandsForApropos(search_word, commands_found, commands_help,
3134 m_command_dict);
3135
3136 if (search_user_commands)
3137 FindCommandsForApropos(search_word, commands_found, commands_help,
3138 m_user_dict);
3139
3140 if (search_user_mw_commands)
3141 FindCommandsForApropos(search_word, commands_found, commands_help,
3142 m_user_mw_dict);
3143
3144 if (search_alias_commands)
3145 FindCommandsForApropos(search_word, commands_found, commands_help,
3146 m_alias_dict);
3147 }
3148
GetExecutionContext() const3149 ExecutionContext CommandInterpreter::GetExecutionContext() const {
3150 return !m_overriden_exe_contexts.empty()
3151 ? m_overriden_exe_contexts.top()
3152 : m_debugger.GetSelectedExecutionContext();
3153 }
3154
OverrideExecutionContext(const ExecutionContext & override_context)3155 void CommandInterpreter::OverrideExecutionContext(
3156 const ExecutionContext &override_context) {
3157 m_overriden_exe_contexts.push(override_context);
3158 }
3159
RestoreExecutionContext()3160 void CommandInterpreter::RestoreExecutionContext() {
3161 if (!m_overriden_exe_contexts.empty())
3162 m_overriden_exe_contexts.pop();
3163 }
3164
GetProcessOutput()3165 void CommandInterpreter::GetProcessOutput() {
3166 if (ProcessSP process_sp = GetExecutionContext().GetProcessSP())
3167 m_debugger.FlushProcessOutput(*process_sp, /*flush_stdout*/ true,
3168 /*flush_stderr*/ true);
3169 }
3170
StartHandlingCommand()3171 void CommandInterpreter::StartHandlingCommand() {
3172 auto idle_state = CommandHandlingState::eIdle;
3173 if (m_command_state.compare_exchange_strong(
3174 idle_state, CommandHandlingState::eInProgress))
3175 lldbassert(m_iohandler_nesting_level == 0);
3176 else
3177 lldbassert(m_iohandler_nesting_level > 0);
3178 ++m_iohandler_nesting_level;
3179 }
3180
FinishHandlingCommand()3181 void CommandInterpreter::FinishHandlingCommand() {
3182 lldbassert(m_iohandler_nesting_level > 0);
3183 if (--m_iohandler_nesting_level == 0) {
3184 auto prev_state = m_command_state.exchange(CommandHandlingState::eIdle);
3185 lldbassert(prev_state != CommandHandlingState::eIdle);
3186 }
3187 }
3188
InterruptCommand()3189 bool CommandInterpreter::InterruptCommand() {
3190 auto in_progress = CommandHandlingState::eInProgress;
3191 return m_command_state.compare_exchange_strong(
3192 in_progress, CommandHandlingState::eInterrupted);
3193 }
3194
WasInterrupted() const3195 bool CommandInterpreter::WasInterrupted() const {
3196 if (!m_debugger.IsIOHandlerThreadCurrentThread())
3197 return false;
3198
3199 bool was_interrupted =
3200 (m_command_state == CommandHandlingState::eInterrupted);
3201 lldbassert(!was_interrupted || m_iohandler_nesting_level > 0);
3202 return was_interrupted;
3203 }
3204
PrintCommandOutput(IOHandler & io_handler,llvm::StringRef str,bool is_stdout)3205 void CommandInterpreter::PrintCommandOutput(IOHandler &io_handler,
3206 llvm::StringRef str,
3207 bool is_stdout) {
3208
3209 lldb::LockableStreamFileSP stream = is_stdout
3210 ? io_handler.GetOutputStreamFileSP()
3211 : io_handler.GetErrorStreamFileSP();
3212 // Split the output into lines and poll for interrupt requests
3213 bool had_output = !str.empty();
3214 while (!str.empty()) {
3215 llvm::StringRef line;
3216 std::tie(line, str) = str.split('\n');
3217 {
3218 LockedStreamFile stream_file = stream->Lock();
3219 stream_file.Write(line.data(), line.size());
3220 stream_file.Write("\n", 1);
3221 }
3222 }
3223
3224 LockedStreamFile stream_file = stream->Lock();
3225 if (had_output &&
3226 INTERRUPT_REQUESTED(GetDebugger(), "Interrupted dumping command output"))
3227 stream_file.Printf("\n... Interrupted.\n");
3228 stream_file.Flush();
3229 }
3230
EchoCommandNonInteractive(llvm::StringRef line,const Flags & io_handler_flags) const3231 bool CommandInterpreter::EchoCommandNonInteractive(
3232 llvm::StringRef line, const Flags &io_handler_flags) const {
3233 if (!io_handler_flags.Test(eHandleCommandFlagEchoCommand))
3234 return false;
3235
3236 llvm::StringRef command = line.trim();
3237 if (command.empty())
3238 return true;
3239
3240 if (command.front() == m_comment_char)
3241 return io_handler_flags.Test(eHandleCommandFlagEchoCommentCommand);
3242
3243 return true;
3244 }
3245
IOHandlerInputComplete(IOHandler & io_handler,std::string & line)3246 void CommandInterpreter::IOHandlerInputComplete(IOHandler &io_handler,
3247 std::string &line) {
3248 // If we were interrupted, bail out...
3249 if (WasInterrupted())
3250 return;
3251
3252 const bool is_interactive = io_handler.GetIsInteractive();
3253 const bool allow_repeats =
3254 io_handler.GetFlags().Test(eHandleCommandFlagAllowRepeats);
3255
3256 if (!is_interactive && !allow_repeats) {
3257 // When we are not interactive, don't execute blank lines. This will happen
3258 // sourcing a commands file. We don't want blank lines to repeat the
3259 // previous command and cause any errors to occur (like redefining an
3260 // alias, get an error and stop parsing the commands file).
3261 // But obey the AllowRepeats flag if the user has set it.
3262 if (line.empty())
3263 return;
3264 }
3265 if (!is_interactive) {
3266 // When using a non-interactive file handle (like when sourcing commands
3267 // from a file) we need to echo the command out so we don't just see the
3268 // command output and no command...
3269 if (EchoCommandNonInteractive(line, io_handler.GetFlags())) {
3270 LockedStreamFile locked_stream =
3271 io_handler.GetOutputStreamFileSP()->Lock();
3272 locked_stream.Printf("%s%s\n", io_handler.GetPrompt(), line.c_str());
3273 }
3274 }
3275
3276 StartHandlingCommand();
3277
3278 ExecutionContext exe_ctx = m_debugger.GetSelectedExecutionContext();
3279 bool pushed_exe_ctx = false;
3280 if (exe_ctx.HasTargetScope()) {
3281 OverrideExecutionContext(exe_ctx);
3282 pushed_exe_ctx = true;
3283 }
3284 auto finalize = llvm::make_scope_exit([this, pushed_exe_ctx]() {
3285 if (pushed_exe_ctx)
3286 RestoreExecutionContext();
3287 });
3288
3289 lldb_private::CommandReturnObject result(m_debugger.GetUseColor());
3290 HandleCommand(line.c_str(), eLazyBoolCalculate, result);
3291
3292 // Now emit the command output text from the command we just executed
3293 if ((result.Succeeded() &&
3294 io_handler.GetFlags().Test(eHandleCommandFlagPrintResult)) ||
3295 io_handler.GetFlags().Test(eHandleCommandFlagPrintErrors)) {
3296 auto DefaultPrintCallback = [&](const CommandReturnObject &result) {
3297 // Display any inline diagnostics first.
3298 const bool inline_diagnostics = !result.GetImmediateErrorStream() &&
3299 GetDebugger().GetShowInlineDiagnostics();
3300 if (inline_diagnostics) {
3301 unsigned prompt_len = m_debugger.GetPrompt().size();
3302 if (auto indent = result.GetDiagnosticIndent()) {
3303 std::string diags =
3304 result.GetInlineDiagnosticString(prompt_len + *indent);
3305 PrintCommandOutput(io_handler, diags, true);
3306 }
3307 }
3308
3309 // Display any STDOUT/STDERR _prior_ to emitting the command result text.
3310 GetProcessOutput();
3311
3312 if (!result.GetImmediateOutputStream()) {
3313 llvm::StringRef output = result.GetOutputString();
3314 PrintCommandOutput(io_handler, output, true);
3315 }
3316
3317 // Now emit the command error text from the command we just executed.
3318 if (!result.GetImmediateErrorStream()) {
3319 std::string error = result.GetErrorString(!inline_diagnostics);
3320 PrintCommandOutput(io_handler, error, false);
3321 }
3322 };
3323
3324 if (m_print_callback) {
3325 const auto callback_result = m_print_callback(result);
3326 if (callback_result == eCommandReturnObjectPrintCallbackSkipped)
3327 DefaultPrintCallback(result);
3328 } else {
3329 DefaultPrintCallback(result);
3330 }
3331 }
3332
3333 FinishHandlingCommand();
3334
3335 switch (result.GetStatus()) {
3336 case eReturnStatusInvalid:
3337 case eReturnStatusSuccessFinishNoResult:
3338 case eReturnStatusSuccessFinishResult:
3339 case eReturnStatusStarted:
3340 break;
3341
3342 case eReturnStatusSuccessContinuingNoResult:
3343 case eReturnStatusSuccessContinuingResult:
3344 if (io_handler.GetFlags().Test(eHandleCommandFlagStopOnContinue))
3345 io_handler.SetIsDone(true);
3346 break;
3347
3348 case eReturnStatusFailed:
3349 m_result.IncrementNumberOfErrors();
3350 if (io_handler.GetFlags().Test(eHandleCommandFlagStopOnError)) {
3351 m_result.SetResult(lldb::eCommandInterpreterResultCommandError);
3352 io_handler.SetIsDone(true);
3353 }
3354 break;
3355
3356 case eReturnStatusQuit:
3357 m_result.SetResult(lldb::eCommandInterpreterResultQuitRequested);
3358 io_handler.SetIsDone(true);
3359 break;
3360 }
3361
3362 // Finally, if we're going to stop on crash, check that here:
3363 if (m_result.IsResult(lldb::eCommandInterpreterResultSuccess) &&
3364 result.GetDidChangeProcessState() &&
3365 io_handler.GetFlags().Test(eHandleCommandFlagStopOnCrash) &&
3366 DidProcessStopAbnormally()) {
3367 io_handler.SetIsDone(true);
3368 m_result.SetResult(lldb::eCommandInterpreterResultInferiorCrash);
3369 }
3370 }
3371
IOHandlerInterrupt(IOHandler & io_handler)3372 bool CommandInterpreter::IOHandlerInterrupt(IOHandler &io_handler) {
3373 ExecutionContext exe_ctx(GetExecutionContext());
3374 Process *process = exe_ctx.GetProcessPtr();
3375
3376 if (InterruptCommand())
3377 return true;
3378
3379 if (process) {
3380 StateType state = process->GetState();
3381 if (StateIsRunningState(state)) {
3382 process->Halt();
3383 return true; // Don't do any updating when we are running
3384 }
3385 }
3386
3387 ScriptInterpreter *script_interpreter =
3388 m_debugger.GetScriptInterpreter(false);
3389 if (script_interpreter) {
3390 if (script_interpreter->Interrupt())
3391 return true;
3392 }
3393 return false;
3394 }
3395
SaveTranscript(CommandReturnObject & result,std::optional<std::string> output_file)3396 bool CommandInterpreter::SaveTranscript(
3397 CommandReturnObject &result, std::optional<std::string> output_file) {
3398 if (output_file == std::nullopt || output_file->empty()) {
3399 std::string now = llvm::to_string(std::chrono::system_clock::now());
3400 llvm::replace(now, ' ', '_');
3401 // Can't have file name with colons on Windows
3402 llvm::replace(now, ':', '-');
3403 const std::string file_name = "lldb_session_" + now + ".log";
3404
3405 FileSpec save_location = GetSaveSessionDirectory();
3406
3407 if (!save_location)
3408 save_location = HostInfo::GetGlobalTempDir();
3409
3410 FileSystem::Instance().Resolve(save_location);
3411 save_location.AppendPathComponent(file_name);
3412 output_file = save_location.GetPath();
3413 }
3414
3415 auto error_out = [&](llvm::StringRef error_message, std::string description) {
3416 LLDB_LOG(GetLog(LLDBLog::Commands), "{0} ({1}:{2})", error_message,
3417 output_file, description);
3418 result.AppendErrorWithFormatv(
3419 "Failed to save session's transcripts to {0}!", *output_file);
3420 return false;
3421 };
3422
3423 File::OpenOptions flags = File::eOpenOptionWriteOnly |
3424 File::eOpenOptionCanCreate |
3425 File::eOpenOptionTruncate;
3426
3427 auto opened_file = FileSystem::Instance().Open(FileSpec(*output_file), flags);
3428
3429 if (!opened_file)
3430 return error_out("Unable to create file",
3431 llvm::toString(opened_file.takeError()));
3432
3433 FileUP file = std::move(opened_file.get());
3434
3435 size_t byte_size = m_transcript_stream.GetSize();
3436
3437 Status error = file->Write(m_transcript_stream.GetData(), byte_size);
3438
3439 if (error.Fail() || byte_size != m_transcript_stream.GetSize())
3440 return error_out("Unable to write to destination file",
3441 "Bytes written do not match transcript size.");
3442
3443 result.SetStatus(eReturnStatusSuccessFinishNoResult);
3444 result.AppendMessageWithFormat("Session's transcripts saved to %s\n",
3445 output_file->c_str());
3446 if (!GetSaveTranscript())
3447 result.AppendError(
3448 "Note: the setting interpreter.save-transcript is set to false, so the "
3449 "transcript might not have been recorded.");
3450
3451 if (GetOpenTranscriptInEditor() && Host::IsInteractiveGraphicSession()) {
3452 const FileSpec file_spec;
3453 error = file->GetFileSpec(const_cast<FileSpec &>(file_spec));
3454 if (error.Success()) {
3455 if (llvm::Error e = Host::OpenFileInExternalEditor(
3456 m_debugger.GetExternalEditor(), file_spec, 1))
3457 result.AppendError(llvm::toString(std::move(e)));
3458 }
3459 }
3460
3461 return true;
3462 }
3463
IsInteractive()3464 bool CommandInterpreter::IsInteractive() {
3465 return (GetIOHandler() ? GetIOHandler()->GetIsInteractive() : false);
3466 }
3467
GetCurrentSourceDir()3468 FileSpec CommandInterpreter::GetCurrentSourceDir() {
3469 if (m_command_source_dirs.empty())
3470 return {};
3471 return m_command_source_dirs.back();
3472 }
3473
GetLLDBCommandsFromIOHandler(const char * prompt,IOHandlerDelegate & delegate,void * baton)3474 void CommandInterpreter::GetLLDBCommandsFromIOHandler(
3475 const char *prompt, IOHandlerDelegate &delegate, void *baton) {
3476 Debugger &debugger = GetDebugger();
3477 IOHandlerSP io_handler_sp(
3478 new IOHandlerEditline(debugger, IOHandler::Type::CommandList,
3479 "lldb", // Name of input reader for history
3480 llvm::StringRef(prompt), // Prompt
3481 llvm::StringRef(), // Continuation prompt
3482 true, // Get multiple lines
3483 debugger.GetUseColor(),
3484 0, // Don't show line numbers
3485 delegate)); // IOHandlerDelegate
3486
3487 if (io_handler_sp) {
3488 io_handler_sp->SetUserData(baton);
3489 debugger.RunIOHandlerAsync(io_handler_sp);
3490 }
3491 }
3492
GetPythonCommandsFromIOHandler(const char * prompt,IOHandlerDelegate & delegate,void * baton)3493 void CommandInterpreter::GetPythonCommandsFromIOHandler(
3494 const char *prompt, IOHandlerDelegate &delegate, void *baton) {
3495 Debugger &debugger = GetDebugger();
3496 IOHandlerSP io_handler_sp(
3497 new IOHandlerEditline(debugger, IOHandler::Type::PythonCode,
3498 "lldb-python", // Name of input reader for history
3499 llvm::StringRef(prompt), // Prompt
3500 llvm::StringRef(), // Continuation prompt
3501 true, // Get multiple lines
3502 debugger.GetUseColor(),
3503 0, // Don't show line numbers
3504 delegate)); // IOHandlerDelegate
3505
3506 if (io_handler_sp) {
3507 io_handler_sp->SetUserData(baton);
3508 debugger.RunIOHandlerAsync(io_handler_sp);
3509 }
3510 }
3511
IsActive()3512 bool CommandInterpreter::IsActive() {
3513 return m_debugger.IsTopIOHandler(m_command_io_handler_sp);
3514 }
3515
3516 lldb::IOHandlerSP
GetIOHandler(bool force_create,CommandInterpreterRunOptions * options)3517 CommandInterpreter::GetIOHandler(bool force_create,
3518 CommandInterpreterRunOptions *options) {
3519 // Always re-create the IOHandlerEditline in case the input changed. The old
3520 // instance might have had a non-interactive input and now it does or vice
3521 // versa.
3522 if (force_create || !m_command_io_handler_sp) {
3523 // Always re-create the IOHandlerEditline in case the input changed. The
3524 // old instance might have had a non-interactive input and now it does or
3525 // vice versa.
3526 uint32_t flags = 0;
3527
3528 if (options) {
3529 if (options->m_stop_on_continue == eLazyBoolYes)
3530 flags |= eHandleCommandFlagStopOnContinue;
3531 if (options->m_stop_on_error == eLazyBoolYes)
3532 flags |= eHandleCommandFlagStopOnError;
3533 if (options->m_stop_on_crash == eLazyBoolYes)
3534 flags |= eHandleCommandFlagStopOnCrash;
3535 if (options->m_echo_commands != eLazyBoolNo)
3536 flags |= eHandleCommandFlagEchoCommand;
3537 if (options->m_echo_comment_commands != eLazyBoolNo)
3538 flags |= eHandleCommandFlagEchoCommentCommand;
3539 if (options->m_print_results != eLazyBoolNo)
3540 flags |= eHandleCommandFlagPrintResult;
3541 if (options->m_print_errors != eLazyBoolNo)
3542 flags |= eHandleCommandFlagPrintErrors;
3543 if (options->m_allow_repeats == eLazyBoolYes)
3544 flags |= eHandleCommandFlagAllowRepeats;
3545 } else {
3546 flags = eHandleCommandFlagEchoCommand | eHandleCommandFlagPrintResult |
3547 eHandleCommandFlagPrintErrors;
3548 }
3549
3550 m_command_io_handler_sp = std::make_shared<IOHandlerEditline>(
3551 m_debugger, IOHandler::Type::CommandInterpreter,
3552 m_debugger.GetInputFileSP(), m_debugger.GetOutputStreamSP(),
3553 m_debugger.GetErrorStreamSP(), flags, "lldb", m_debugger.GetPrompt(),
3554 llvm::StringRef(), // Continuation prompt
3555 false, // Don't enable multiple line input, just single line commands
3556 m_debugger.GetUseColor(),
3557 0, // Don't show line numbers
3558 *this); // IOHandlerDelegate
3559 }
3560 return m_command_io_handler_sp;
3561 }
3562
RunCommandInterpreter(CommandInterpreterRunOptions & options)3563 CommandInterpreterRunResult CommandInterpreter::RunCommandInterpreter(
3564 CommandInterpreterRunOptions &options) {
3565 // Always re-create the command interpreter when we run it in case any file
3566 // handles have changed.
3567 bool force_create = true;
3568 m_debugger.RunIOHandlerAsync(GetIOHandler(force_create, &options));
3569 m_result = CommandInterpreterRunResult();
3570
3571 if (options.GetAutoHandleEvents())
3572 m_debugger.StartEventHandlerThread();
3573
3574 if (options.GetSpawnThread()) {
3575 m_debugger.StartIOHandlerThread();
3576 } else {
3577 // If the current thread is not managed by a host thread, we won't detect
3578 // that this IS the CommandInterpreter IOHandler thread, so make it so:
3579 HostThread new_io_handler_thread(Host::GetCurrentThread());
3580 HostThread old_io_handler_thread =
3581 m_debugger.SetIOHandlerThread(new_io_handler_thread);
3582 m_debugger.RunIOHandlers();
3583 m_debugger.SetIOHandlerThread(old_io_handler_thread);
3584
3585 if (options.GetAutoHandleEvents())
3586 m_debugger.StopEventHandlerThread();
3587 }
3588
3589 return m_result;
3590 }
3591
3592 CommandObject *
ResolveCommandImpl(std::string & command_line,CommandReturnObject & result)3593 CommandInterpreter::ResolveCommandImpl(std::string &command_line,
3594 CommandReturnObject &result) {
3595 std::string scratch_command(command_line); // working copy so we don't modify
3596 // command_line unless we succeed
3597 CommandObject *cmd_obj = nullptr;
3598 StreamString revised_command_line;
3599 bool wants_raw_input = false;
3600 std::string next_word;
3601 StringList matches;
3602 bool done = false;
3603
3604 auto build_alias_cmd = [&](std::string &full_name) {
3605 revised_command_line.Clear();
3606 matches.Clear();
3607 std::string alias_result;
3608 cmd_obj =
3609 BuildAliasResult(full_name, scratch_command, alias_result, result);
3610 revised_command_line.Printf("%s", alias_result.c_str());
3611 if (cmd_obj) {
3612 wants_raw_input = cmd_obj->WantsRawCommandString();
3613 }
3614 };
3615
3616 while (!done) {
3617 char quote_char = '\0';
3618 std::string suffix;
3619 ExtractCommand(scratch_command, next_word, suffix, quote_char);
3620 if (cmd_obj == nullptr) {
3621 std::string full_name;
3622 bool is_alias = GetAliasFullName(next_word, full_name);
3623 cmd_obj = GetCommandObject(next_word, &matches);
3624 bool is_real_command =
3625 (!is_alias) || (cmd_obj != nullptr && !cmd_obj->IsAlias());
3626 if (!is_real_command) {
3627 build_alias_cmd(full_name);
3628 } else {
3629 if (cmd_obj) {
3630 llvm::StringRef cmd_name = cmd_obj->GetCommandName();
3631 revised_command_line.Printf("%s", cmd_name.str().c_str());
3632 wants_raw_input = cmd_obj->WantsRawCommandString();
3633 } else {
3634 revised_command_line.Printf("%s", next_word.c_str());
3635 }
3636 }
3637 } else {
3638 if (cmd_obj->IsMultiwordObject()) {
3639 CommandObject *sub_cmd_obj =
3640 cmd_obj->GetSubcommandObject(next_word.c_str());
3641 if (sub_cmd_obj) {
3642 // The subcommand's name includes the parent command's name, so
3643 // restart rather than append to the revised_command_line.
3644 llvm::StringRef sub_cmd_name = sub_cmd_obj->GetCommandName();
3645 revised_command_line.Clear();
3646 revised_command_line.Printf("%s", sub_cmd_name.str().c_str());
3647 cmd_obj = sub_cmd_obj;
3648 wants_raw_input = cmd_obj->WantsRawCommandString();
3649 } else {
3650 if (quote_char)
3651 revised_command_line.Printf(" %c%s%s%c", quote_char,
3652 next_word.c_str(), suffix.c_str(),
3653 quote_char);
3654 else
3655 revised_command_line.Printf(" %s%s", next_word.c_str(),
3656 suffix.c_str());
3657 done = true;
3658 }
3659 } else {
3660 if (quote_char)
3661 revised_command_line.Printf(" %c%s%s%c", quote_char,
3662 next_word.c_str(), suffix.c_str(),
3663 quote_char);
3664 else
3665 revised_command_line.Printf(" %s%s", next_word.c_str(),
3666 suffix.c_str());
3667 done = true;
3668 }
3669 }
3670
3671 if (cmd_obj == nullptr) {
3672 const size_t num_matches = matches.GetSize();
3673 if (matches.GetSize() > 1) {
3674 StringList alias_matches;
3675 GetAliasCommandObject(next_word, &alias_matches);
3676
3677 if (alias_matches.GetSize() == 1) {
3678 std::string full_name;
3679 GetAliasFullName(alias_matches.GetStringAtIndex(0), full_name);
3680 build_alias_cmd(full_name);
3681 done = static_cast<bool>(cmd_obj);
3682 } else {
3683 StreamString error_msg;
3684 error_msg.Printf("Ambiguous command '%s'. Possible matches:\n",
3685 next_word.c_str());
3686
3687 for (uint32_t i = 0; i < num_matches; ++i) {
3688 error_msg.Printf("\t%s\n", matches.GetStringAtIndex(i));
3689 }
3690 result.AppendRawError(error_msg.GetString());
3691 }
3692 } else {
3693 // We didn't have only one match, otherwise we wouldn't get here.
3694 lldbassert(num_matches == 0);
3695 result.AppendErrorWithFormat("'%s' is not a valid command.\n",
3696 next_word.c_str());
3697 }
3698 if (!done)
3699 return nullptr;
3700 }
3701
3702 if (cmd_obj->IsMultiwordObject()) {
3703 if (!suffix.empty()) {
3704 result.AppendErrorWithFormat(
3705 "command '%s' did not recognize '%s%s%s' as valid (subcommand "
3706 "might be invalid).\n",
3707 cmd_obj->GetCommandName().str().c_str(),
3708 next_word.empty() ? "" : next_word.c_str(),
3709 next_word.empty() ? " -- " : " ", suffix.c_str());
3710 return nullptr;
3711 }
3712 } else {
3713 // If we found a normal command, we are done
3714 done = true;
3715 if (!suffix.empty()) {
3716 switch (suffix[0]) {
3717 case '/':
3718 // GDB format suffixes
3719 {
3720 Options *command_options = cmd_obj->GetOptions();
3721 if (command_options &&
3722 command_options->SupportsLongOption("gdb-format")) {
3723 std::string gdb_format_option("--gdb-format=");
3724 gdb_format_option += (suffix.c_str() + 1);
3725
3726 std::string cmd = std::string(revised_command_line.GetString());
3727 size_t arg_terminator_idx = FindArgumentTerminator(cmd);
3728 if (arg_terminator_idx != std::string::npos) {
3729 // Insert the gdb format option before the "--" that terminates
3730 // options
3731 gdb_format_option.append(1, ' ');
3732 cmd.insert(arg_terminator_idx, gdb_format_option);
3733 revised_command_line.Clear();
3734 revised_command_line.PutCString(cmd);
3735 } else
3736 revised_command_line.Printf(" %s", gdb_format_option.c_str());
3737
3738 if (wants_raw_input &&
3739 FindArgumentTerminator(cmd) == std::string::npos)
3740 revised_command_line.PutCString(" --");
3741 } else {
3742 result.AppendErrorWithFormat(
3743 "the '%s' command doesn't support the --gdb-format option\n",
3744 cmd_obj->GetCommandName().str().c_str());
3745 return nullptr;
3746 }
3747 }
3748 break;
3749
3750 default:
3751 result.AppendErrorWithFormat(
3752 "unknown command shorthand suffix: '%s'\n", suffix.c_str());
3753 return nullptr;
3754 }
3755 }
3756 }
3757 if (scratch_command.empty())
3758 done = true;
3759 }
3760
3761 if (!scratch_command.empty())
3762 revised_command_line.Printf(" %s", scratch_command.c_str());
3763
3764 if (cmd_obj != nullptr)
3765 command_line = std::string(revised_command_line.GetString());
3766
3767 return cmd_obj;
3768 }
3769
GetStatistics()3770 llvm::json::Value CommandInterpreter::GetStatistics() {
3771 llvm::json::Object stats;
3772 for (const auto &command_usage : m_command_usages)
3773 stats.try_emplace(command_usage.getKey(), command_usage.getValue());
3774 return stats;
3775 }
3776
GetTranscript() const3777 const StructuredData::Array &CommandInterpreter::GetTranscript() const {
3778 return m_transcript;
3779 }
3780
SetPrintCallback(CommandReturnObjectCallback callback)3781 void CommandInterpreter::SetPrintCallback(
3782 CommandReturnObjectCallback callback) {
3783 m_print_callback = callback;
3784 }
3785