1 | //===-- CommandObject.cpp -------------------------------------------------===// |
---|---|
2 | // |
3 | // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
4 | // See https://llvm.org/LICENSE.txt for license information. |
5 | // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
6 | // |
7 | //===----------------------------------------------------------------------===// |
8 | |
9 | #include "lldb/Interpreter/CommandObject.h" |
10 | |
11 | #include <map> |
12 | #include <sstream> |
13 | #include <string> |
14 | |
15 | #include <cctype> |
16 | #include <cstdlib> |
17 | |
18 | #include "lldb/Core/Address.h" |
19 | #include "lldb/Interpreter/CommandOptionArgumentTable.h" |
20 | #include "lldb/Interpreter/Options.h" |
21 | #include "lldb/Utility/ArchSpec.h" |
22 | #include "llvm/ADT/ScopeExit.h" |
23 | |
24 | // These are for the Sourcename completers. |
25 | // FIXME: Make a separate file for the completers. |
26 | #include "lldb/DataFormatters/FormatManager.h" |
27 | #include "lldb/Target/Process.h" |
28 | #include "lldb/Target/Target.h" |
29 | #include "lldb/Utility/FileSpec.h" |
30 | #include "lldb/Utility/FileSpecList.h" |
31 | |
32 | #include "lldb/Target/Language.h" |
33 | |
34 | #include "lldb/Interpreter/CommandInterpreter.h" |
35 | #include "lldb/Interpreter/CommandReturnObject.h" |
36 | |
37 | using namespace lldb; |
38 | using namespace lldb_private; |
39 | |
40 | // CommandObject |
41 | |
42 | CommandObject::CommandObject(CommandInterpreter &interpreter, |
43 | llvm::StringRef name, llvm::StringRef help, |
44 | llvm::StringRef syntax, uint32_t flags) |
45 | : m_interpreter(interpreter), m_cmd_name(std::string(name)), |
46 | m_flags(flags), m_deprecated_command_override_callback(nullptr), |
47 | m_command_override_callback(nullptr), m_command_override_baton(nullptr) { |
48 | m_cmd_help_short = std::string(help); |
49 | m_cmd_syntax = std::string(syntax); |
50 | } |
51 | |
52 | Debugger &CommandObject::GetDebugger() { return m_interpreter.GetDebugger(); } |
53 | |
54 | llvm::StringRef CommandObject::GetHelp() { return m_cmd_help_short; } |
55 | |
56 | llvm::StringRef CommandObject::GetHelpLong() { return m_cmd_help_long; } |
57 | |
58 | llvm::StringRef CommandObject::GetSyntax() { |
59 | if (!m_cmd_syntax.empty()) |
60 | return m_cmd_syntax; |
61 | |
62 | StreamString syntax_str; |
63 | syntax_str.PutCString(cstr: GetCommandName()); |
64 | |
65 | if (!IsDashDashCommand() && GetOptions() != nullptr) |
66 | syntax_str.PutCString(cstr: " <cmd-options>"); |
67 | |
68 | if (!m_arguments.empty()) { |
69 | syntax_str.PutCString(cstr: " "); |
70 | |
71 | if (!IsDashDashCommand() && WantsRawCommandString() && GetOptions() && |
72 | GetOptions()->NumCommandOptions()) |
73 | syntax_str.PutCString(cstr: "-- "); |
74 | GetFormattedCommandArguments(str&: syntax_str); |
75 | } |
76 | m_cmd_syntax = std::string(syntax_str.GetString()); |
77 | |
78 | return m_cmd_syntax; |
79 | } |
80 | |
81 | llvm::StringRef CommandObject::GetCommandName() const { return m_cmd_name; } |
82 | |
83 | void CommandObject::SetCommandName(llvm::StringRef name) { |
84 | m_cmd_name = std::string(name); |
85 | } |
86 | |
87 | void CommandObject::SetHelp(llvm::StringRef str) { |
88 | m_cmd_help_short = std::string(str); |
89 | } |
90 | |
91 | void CommandObject::SetHelpLong(llvm::StringRef str) { |
92 | m_cmd_help_long = std::string(str); |
93 | } |
94 | |
95 | void CommandObject::SetSyntax(llvm::StringRef str) { |
96 | m_cmd_syntax = std::string(str); |
97 | } |
98 | |
99 | Options *CommandObject::GetOptions() { |
100 | // By default commands don't have options unless this virtual function is |
101 | // overridden by base classes. |
102 | return nullptr; |
103 | } |
104 | |
105 | bool CommandObject::ParseOptions(Args &args, CommandReturnObject &result) { |
106 | // See if the subclass has options? |
107 | Options *options = GetOptions(); |
108 | if (options != nullptr) { |
109 | Status error; |
110 | |
111 | auto exe_ctx = GetCommandInterpreter().GetExecutionContext(); |
112 | options->NotifyOptionParsingStarting(execution_context: &exe_ctx); |
113 | |
114 | const bool require_validation = true; |
115 | llvm::Expected<Args> args_or = options->Parse( |
116 | args, execution_context: &exe_ctx, platform_sp: GetCommandInterpreter().GetPlatform(prefer_target_platform: true), |
117 | require_validation); |
118 | |
119 | if (args_or) { |
120 | args = std::move(*args_or); |
121 | error = options->NotifyOptionParsingFinished(execution_context: &exe_ctx); |
122 | } else { |
123 | error = Status::FromError(error: args_or.takeError()); |
124 | } |
125 | |
126 | if (error.Fail()) { |
127 | result.SetError(error.takeError()); |
128 | result.SetStatus(eReturnStatusFailed); |
129 | return false; |
130 | } |
131 | |
132 | if (llvm::Error error = options->VerifyOptions()) { |
133 | result.SetError(std::move(error)); |
134 | result.SetStatus(eReturnStatusFailed); |
135 | return false; |
136 | } |
137 | |
138 | result.SetStatus(eReturnStatusSuccessFinishNoResult); |
139 | return true; |
140 | } |
141 | return true; |
142 | } |
143 | |
144 | bool CommandObject::CheckRequirements(CommandReturnObject &result) { |
145 | // Nothing should be stored in m_exe_ctx between running commands as |
146 | // m_exe_ctx has shared pointers to the target, process, thread and frame and |
147 | // we don't want any CommandObject instances to keep any of these objects |
148 | // around longer than for a single command. Every command should call |
149 | // CommandObject::Cleanup() after it has completed. |
150 | assert(!m_exe_ctx.GetTargetPtr()); |
151 | assert(!m_exe_ctx.GetProcessPtr()); |
152 | assert(!m_exe_ctx.GetThreadPtr()); |
153 | assert(!m_exe_ctx.GetFramePtr()); |
154 | |
155 | // Lock down the interpreter's execution context prior to running the command |
156 | // so we guarantee the selected target, process, thread and frame can't go |
157 | // away during the execution |
158 | m_exe_ctx = m_interpreter.GetExecutionContext(); |
159 | |
160 | const uint32_t flags = GetFlags().Get(); |
161 | if (flags & (eCommandRequiresTarget | eCommandRequiresProcess | |
162 | eCommandRequiresThread | eCommandRequiresFrame | |
163 | eCommandTryTargetAPILock)) { |
164 | |
165 | if ((flags & eCommandRequiresTarget) && !m_exe_ctx.HasTargetScope()) { |
166 | result.AppendError(in_string: GetInvalidTargetDescription()); |
167 | return false; |
168 | } |
169 | |
170 | if ((flags & eCommandRequiresProcess) && !m_exe_ctx.HasProcessScope()) { |
171 | if (!m_exe_ctx.HasTargetScope()) |
172 | result.AppendError(in_string: GetInvalidTargetDescription()); |
173 | else |
174 | result.AppendError(in_string: GetInvalidProcessDescription()); |
175 | return false; |
176 | } |
177 | |
178 | if ((flags & eCommandRequiresThread) && !m_exe_ctx.HasThreadScope()) { |
179 | if (!m_exe_ctx.HasTargetScope()) |
180 | result.AppendError(in_string: GetInvalidTargetDescription()); |
181 | else if (!m_exe_ctx.HasProcessScope()) |
182 | result.AppendError(in_string: GetInvalidProcessDescription()); |
183 | else |
184 | result.AppendError(in_string: GetInvalidThreadDescription()); |
185 | return false; |
186 | } |
187 | |
188 | if ((flags & eCommandRequiresFrame) && !m_exe_ctx.HasFrameScope()) { |
189 | if (!m_exe_ctx.HasTargetScope()) |
190 | result.AppendError(in_string: GetInvalidTargetDescription()); |
191 | else if (!m_exe_ctx.HasProcessScope()) |
192 | result.AppendError(in_string: GetInvalidProcessDescription()); |
193 | else if (!m_exe_ctx.HasThreadScope()) |
194 | result.AppendError(in_string: GetInvalidThreadDescription()); |
195 | else |
196 | result.AppendError(in_string: GetInvalidFrameDescription()); |
197 | return false; |
198 | } |
199 | |
200 | if ((flags & eCommandRequiresRegContext) && |
201 | (m_exe_ctx.GetRegisterContext() == nullptr)) { |
202 | result.AppendError(in_string: GetInvalidRegContextDescription()); |
203 | return false; |
204 | } |
205 | |
206 | if (flags & eCommandTryTargetAPILock) { |
207 | Target *target = m_exe_ctx.GetTargetPtr(); |
208 | if (target) |
209 | m_api_locker = |
210 | std::unique_lock<std::recursive_mutex>(target->GetAPIMutex()); |
211 | } |
212 | } |
213 | |
214 | if (GetFlags().AnySet(mask: eCommandProcessMustBeLaunched | |
215 | eCommandProcessMustBePaused)) { |
216 | Process *process = m_interpreter.GetExecutionContext().GetProcessPtr(); |
217 | if (process == nullptr) { |
218 | // A process that is not running is considered paused. |
219 | if (GetFlags().Test(bit: eCommandProcessMustBeLaunched)) { |
220 | result.AppendError(in_string: "Process must exist."); |
221 | return false; |
222 | } |
223 | } else { |
224 | StateType state = process->GetState(); |
225 | switch (state) { |
226 | case eStateInvalid: |
227 | case eStateSuspended: |
228 | case eStateCrashed: |
229 | case eStateStopped: |
230 | break; |
231 | |
232 | case eStateConnected: |
233 | case eStateAttaching: |
234 | case eStateLaunching: |
235 | case eStateDetached: |
236 | case eStateExited: |
237 | case eStateUnloaded: |
238 | if (GetFlags().Test(bit: eCommandProcessMustBeLaunched)) { |
239 | result.AppendError(in_string: "Process must be launched."); |
240 | return false; |
241 | } |
242 | break; |
243 | |
244 | case eStateRunning: |
245 | case eStateStepping: |
246 | if (GetFlags().Test(bit: eCommandProcessMustBePaused)) { |
247 | result.AppendError(in_string: "Process is running. Use 'process interrupt' to " |
248 | "pause execution."); |
249 | return false; |
250 | } |
251 | } |
252 | } |
253 | } |
254 | |
255 | if (GetFlags().Test(bit: eCommandProcessMustBeTraced)) { |
256 | Target *target = m_exe_ctx.GetTargetPtr(); |
257 | if (target && !target->GetTrace()) { |
258 | result.AppendError(in_string: "Process is not being traced."); |
259 | return false; |
260 | } |
261 | } |
262 | |
263 | return true; |
264 | } |
265 | |
266 | void CommandObject::Cleanup() { |
267 | m_exe_ctx.Clear(); |
268 | if (m_api_locker.owns_lock()) |
269 | m_api_locker.unlock(); |
270 | } |
271 | |
272 | void CommandObject::HandleCompletion(CompletionRequest &request) { |
273 | |
274 | m_exe_ctx = m_interpreter.GetExecutionContext(); |
275 | auto reset_ctx = llvm::make_scope_exit(F: [this]() { Cleanup(); }); |
276 | |
277 | // Default implementation of WantsCompletion() is !WantsRawCommandString(). |
278 | // Subclasses who want raw command string but desire, for example, argument |
279 | // completion should override WantsCompletion() to return true, instead. |
280 | if (WantsRawCommandString() && !WantsCompletion()) { |
281 | // FIXME: Abstract telling the completion to insert the completion |
282 | // character. |
283 | return; |
284 | } else { |
285 | // Can we do anything generic with the options? |
286 | Options *cur_options = GetOptions(); |
287 | OptionElementVector opt_element_vector; |
288 | |
289 | if (cur_options != nullptr) { |
290 | opt_element_vector = cur_options->ParseForCompletion( |
291 | args: request.GetParsedLine(), cursor_index: request.GetCursorIndex()); |
292 | |
293 | bool handled_by_options = cur_options->HandleOptionCompletion( |
294 | request, option_map&: opt_element_vector, interpreter&: GetCommandInterpreter()); |
295 | if (handled_by_options) |
296 | return; |
297 | } |
298 | |
299 | // If we got here, the last word is not an option or an option argument. |
300 | HandleArgumentCompletion(request, opt_element_vector); |
301 | } |
302 | } |
303 | |
304 | void CommandObject::HandleArgumentCompletion( |
305 | CompletionRequest &request, OptionElementVector &opt_element_vector) { |
306 | size_t num_arg_entries = GetNumArgumentEntries(); |
307 | if (num_arg_entries != 1) |
308 | return; |
309 | |
310 | CommandArgumentEntry *entry_ptr = GetArgumentEntryAtIndex(idx: 0); |
311 | if (!entry_ptr) { |
312 | assert(entry_ptr && "We said there was one entry, but there wasn't."); |
313 | return; // Not worth crashing if asserts are off... |
314 | } |
315 | |
316 | CommandArgumentEntry &entry = *entry_ptr; |
317 | // For now, we only handle the simple case of one homogenous argument type. |
318 | if (entry.size() != 1) |
319 | return; |
320 | |
321 | // Look up the completion type, and if it has one, invoke it: |
322 | const CommandObject::ArgumentTableEntry *arg_entry = |
323 | FindArgumentDataByType(arg_type: entry[0].arg_type); |
324 | const ArgumentRepetitionType repeat = entry[0].arg_repetition; |
325 | |
326 | if (arg_entry == nullptr || arg_entry->completion_type == lldb::eNoCompletion) |
327 | return; |
328 | |
329 | // FIXME: This should be handled higher in the Command Parser. |
330 | // Check the case where this command only takes one argument, and don't do |
331 | // the completion if we aren't on the first entry: |
332 | if (repeat == eArgRepeatPlain && request.GetCursorIndex() != 0) |
333 | return; |
334 | |
335 | lldb_private::CommandCompletions::InvokeCommonCompletionCallbacks( |
336 | interpreter&: GetCommandInterpreter(), completion_mask: arg_entry->completion_type, request, searcher: nullptr); |
337 | |
338 | } |
339 | |
340 | bool CommandObject::HelpTextContainsWord(llvm::StringRef search_word, |
341 | bool search_short_help, |
342 | bool search_long_help, |
343 | bool search_syntax, |
344 | bool search_options) { |
345 | bool found_word = false; |
346 | |
347 | llvm::StringRef short_help = GetHelp(); |
348 | llvm::StringRef long_help = GetHelpLong(); |
349 | llvm::StringRef syntax_help = GetSyntax(); |
350 | |
351 | if (search_short_help && short_help.contains_insensitive(Other: search_word)) |
352 | found_word = true; |
353 | else if (search_long_help && long_help.contains_insensitive(Other: search_word)) |
354 | found_word = true; |
355 | else if (search_syntax && syntax_help.contains_insensitive(Other: search_word)) |
356 | found_word = true; |
357 | |
358 | if (!found_word && search_options && GetOptions() != nullptr) { |
359 | StreamString usage_help; |
360 | GetOptions()->GenerateOptionUsage( |
361 | strm&: usage_help, cmd&: *this, |
362 | screen_width: GetCommandInterpreter().GetDebugger().GetTerminalWidth()); |
363 | if (!usage_help.Empty()) { |
364 | llvm::StringRef usage_text = usage_help.GetString(); |
365 | if (usage_text.contains_insensitive(Other: search_word)) |
366 | found_word = true; |
367 | } |
368 | } |
369 | |
370 | return found_word; |
371 | } |
372 | |
373 | bool CommandObject::ParseOptionsAndNotify(Args &args, |
374 | CommandReturnObject &result, |
375 | OptionGroupOptions &group_options, |
376 | ExecutionContext &exe_ctx) { |
377 | if (!ParseOptions(args, result)) |
378 | return false; |
379 | |
380 | Status error(group_options.NotifyOptionParsingFinished(execution_context: &exe_ctx)); |
381 | if (error.Fail()) { |
382 | result.AppendError(in_string: error.AsCString()); |
383 | return false; |
384 | } |
385 | return true; |
386 | } |
387 | |
388 | void CommandObject::AddSimpleArgumentList( |
389 | CommandArgumentType arg_type, ArgumentRepetitionType repetition_type) { |
390 | |
391 | CommandArgumentEntry arg_entry; |
392 | CommandArgumentData simple_arg; |
393 | |
394 | // Define the first (and only) variant of this arg. |
395 | simple_arg.arg_type = arg_type; |
396 | simple_arg.arg_repetition = repetition_type; |
397 | |
398 | // There is only one variant this argument could be; put it into the argument |
399 | // entry. |
400 | arg_entry.push_back(x: simple_arg); |
401 | |
402 | // Push the data for the first argument into the m_arguments vector. |
403 | m_arguments.push_back(x: arg_entry); |
404 | } |
405 | |
406 | int CommandObject::GetNumArgumentEntries() { return m_arguments.size(); } |
407 | |
408 | CommandObject::CommandArgumentEntry * |
409 | CommandObject::GetArgumentEntryAtIndex(int idx) { |
410 | if (static_cast<size_t>(idx) < m_arguments.size()) |
411 | return &(m_arguments[idx]); |
412 | |
413 | return nullptr; |
414 | } |
415 | |
416 | const CommandObject::ArgumentTableEntry * |
417 | CommandObject::FindArgumentDataByType(CommandArgumentType arg_type) { |
418 | for (int i = 0; i < eArgTypeLastArg; ++i) |
419 | if (g_argument_table[i].arg_type == arg_type) |
420 | return &(g_argument_table[i]); |
421 | |
422 | return nullptr; |
423 | } |
424 | |
425 | void CommandObject::GetArgumentHelp(Stream &str, CommandArgumentType arg_type, |
426 | CommandInterpreter &interpreter) { |
427 | const ArgumentTableEntry *entry = &(g_argument_table[arg_type]); |
428 | |
429 | // The table is *supposed* to be kept in arg_type order, but someone *could* |
430 | // have messed it up... |
431 | |
432 | if (entry->arg_type != arg_type) |
433 | entry = CommandObject::FindArgumentDataByType(arg_type); |
434 | |
435 | if (!entry) |
436 | return; |
437 | |
438 | StreamString name_str; |
439 | name_str.Printf(format: "<%s>", entry->arg_name); |
440 | |
441 | if (entry->help_function) { |
442 | llvm::StringRef help_text = entry->help_function(); |
443 | if (!entry->help_function.self_formatting) { |
444 | interpreter.OutputFormattedHelpText(stream&: str, command_word: name_str.GetString(), separator: "--", |
445 | help_text, max_word_len: name_str.GetSize()); |
446 | } else { |
447 | interpreter.OutputHelpText(stream&: str, command_word: name_str.GetString(), separator: "--", help_text, |
448 | max_word_len: name_str.GetSize()); |
449 | } |
450 | } else { |
451 | interpreter.OutputFormattedHelpText(stream&: str, command_word: name_str.GetString(), separator: "--", |
452 | help_text: entry->help_text, max_word_len: name_str.GetSize()); |
453 | |
454 | // Print enum values and their description if any. |
455 | OptionEnumValues enum_values = g_argument_table[arg_type].enum_values; |
456 | if (!enum_values.empty()) { |
457 | str.EOL(); |
458 | size_t longest = 0; |
459 | for (const OptionEnumValueElement &element : enum_values) |
460 | longest = |
461 | std::max(a: longest, b: llvm::StringRef(element.string_value).size()); |
462 | str.IndentMore(amount: 5); |
463 | for (const OptionEnumValueElement &element : enum_values) { |
464 | str.Indent(); |
465 | interpreter.OutputHelpText(stream&: str, command_word: element.string_value, separator: ":", |
466 | help_text: element.usage, max_word_len: longest); |
467 | } |
468 | str.IndentLess(amount: 5); |
469 | str.EOL(); |
470 | } |
471 | } |
472 | } |
473 | |
474 | const char *CommandObject::GetArgumentName(CommandArgumentType arg_type) { |
475 | const ArgumentTableEntry *entry = &(g_argument_table[arg_type]); |
476 | |
477 | // The table is *supposed* to be kept in arg_type order, but someone *could* |
478 | // have messed it up... |
479 | |
480 | if (entry->arg_type != arg_type) |
481 | entry = CommandObject::FindArgumentDataByType(arg_type); |
482 | |
483 | if (entry) |
484 | return entry->arg_name; |
485 | |
486 | return nullptr; |
487 | } |
488 | |
489 | bool CommandObject::IsPairType(ArgumentRepetitionType arg_repeat_type) { |
490 | return (arg_repeat_type == eArgRepeatPairPlain) || |
491 | (arg_repeat_type == eArgRepeatPairOptional) || |
492 | (arg_repeat_type == eArgRepeatPairPlus) || |
493 | (arg_repeat_type == eArgRepeatPairStar) || |
494 | (arg_repeat_type == eArgRepeatPairRange) || |
495 | (arg_repeat_type == eArgRepeatPairRangeOptional); |
496 | } |
497 | |
498 | std::optional<ArgumentRepetitionType> |
499 | CommandObject::ArgRepetitionFromString(llvm::StringRef string) { |
500 | return llvm::StringSwitch<ArgumentRepetitionType>(string) |
501 | .Case(S: "plain", Value: eArgRepeatPlain) |
502 | .Case(S: "optional", Value: eArgRepeatOptional) |
503 | .Case(S: "plus", Value: eArgRepeatPlus) |
504 | .Case(S: "star", Value: eArgRepeatStar) |
505 | .Case(S: "range", Value: eArgRepeatRange) |
506 | .Case(S: "pair-plain", Value: eArgRepeatPairPlain) |
507 | .Case(S: "pair-optional", Value: eArgRepeatPairOptional) |
508 | .Case(S: "pair-plus", Value: eArgRepeatPairPlus) |
509 | .Case(S: "pair-star", Value: eArgRepeatPairStar) |
510 | .Case(S: "pair-range", Value: eArgRepeatPairRange) |
511 | .Case(S: "pair-range-optional", Value: eArgRepeatPairRangeOptional) |
512 | .Default(Value: {}); |
513 | } |
514 | |
515 | static CommandObject::CommandArgumentEntry |
516 | OptSetFiltered(uint32_t opt_set_mask, |
517 | CommandObject::CommandArgumentEntry &cmd_arg_entry) { |
518 | CommandObject::CommandArgumentEntry ret_val; |
519 | for (unsigned i = 0; i < cmd_arg_entry.size(); ++i) |
520 | if (opt_set_mask & cmd_arg_entry[i].arg_opt_set_association) |
521 | ret_val.push_back(x: cmd_arg_entry[i]); |
522 | return ret_val; |
523 | } |
524 | |
525 | // Default parameter value of opt_set_mask is LLDB_OPT_SET_ALL, which means |
526 | // take all the argument data into account. On rare cases where some argument |
527 | // sticks with certain option sets, this function returns the option set |
528 | // filtered args. |
529 | void CommandObject::GetFormattedCommandArguments(Stream &str, |
530 | uint32_t opt_set_mask) { |
531 | int num_args = m_arguments.size(); |
532 | for (int i = 0; i < num_args; ++i) { |
533 | if (i > 0) |
534 | str.Printf(format: " "); |
535 | CommandArgumentEntry arg_entry = |
536 | opt_set_mask == LLDB_OPT_SET_ALL |
537 | ? m_arguments[i] |
538 | : OptSetFiltered(opt_set_mask, cmd_arg_entry&: m_arguments[i]); |
539 | // This argument is not associated with the current option set, so skip it. |
540 | if (arg_entry.empty()) |
541 | continue; |
542 | int num_alternatives = arg_entry.size(); |
543 | |
544 | if ((num_alternatives == 2) && IsPairType(arg_repeat_type: arg_entry[0].arg_repetition)) { |
545 | const char *first_name = GetArgumentName(arg_type: arg_entry[0].arg_type); |
546 | const char *second_name = GetArgumentName(arg_type: arg_entry[1].arg_type); |
547 | switch (arg_entry[0].arg_repetition) { |
548 | case eArgRepeatPairPlain: |
549 | str.Printf(format: "<%s> <%s>", first_name, second_name); |
550 | break; |
551 | case eArgRepeatPairOptional: |
552 | str.Printf(format: "[<%s> <%s>]", first_name, second_name); |
553 | break; |
554 | case eArgRepeatPairPlus: |
555 | str.Printf(format: "<%s> <%s> [<%s> <%s> [...]]", first_name, second_name, |
556 | first_name, second_name); |
557 | break; |
558 | case eArgRepeatPairStar: |
559 | str.Printf(format: "[<%s> <%s> [<%s> <%s> [...]]]", first_name, second_name, |
560 | first_name, second_name); |
561 | break; |
562 | case eArgRepeatPairRange: |
563 | str.Printf(format: "<%s_1> <%s_1> ... <%s_n> <%s_n>", first_name, second_name, |
564 | first_name, second_name); |
565 | break; |
566 | case eArgRepeatPairRangeOptional: |
567 | str.Printf(format: "[<%s_1> <%s_1> ... <%s_n> <%s_n>]", first_name, second_name, |
568 | first_name, second_name); |
569 | break; |
570 | // Explicitly test for all the rest of the cases, so if new types get |
571 | // added we will notice the missing case statement(s). |
572 | case eArgRepeatPlain: |
573 | case eArgRepeatOptional: |
574 | case eArgRepeatPlus: |
575 | case eArgRepeatStar: |
576 | case eArgRepeatRange: |
577 | // These should not be reached, as they should fail the IsPairType test |
578 | // above. |
579 | break; |
580 | } |
581 | } else { |
582 | StreamString names; |
583 | for (int j = 0; j < num_alternatives; ++j) { |
584 | if (j > 0) |
585 | names.Printf(format: " | "); |
586 | names.Printf(format: "%s", GetArgumentName(arg_type: arg_entry[j].arg_type)); |
587 | } |
588 | |
589 | std::string name_str = std::string(names.GetString()); |
590 | switch (arg_entry[0].arg_repetition) { |
591 | case eArgRepeatPlain: |
592 | str.Printf(format: "<%s>", name_str.c_str()); |
593 | break; |
594 | case eArgRepeatPlus: |
595 | str.Printf(format: "<%s> [<%s> [...]]", name_str.c_str(), name_str.c_str()); |
596 | break; |
597 | case eArgRepeatStar: |
598 | str.Printf(format: "[<%s> [<%s> [...]]]", name_str.c_str(), name_str.c_str()); |
599 | break; |
600 | case eArgRepeatOptional: |
601 | str.Printf(format: "[<%s>]", name_str.c_str()); |
602 | break; |
603 | case eArgRepeatRange: |
604 | str.Printf(format: "<%s_1> .. <%s_n>", name_str.c_str(), name_str.c_str()); |
605 | break; |
606 | // Explicitly test for all the rest of the cases, so if new types get |
607 | // added we will notice the missing case statement(s). |
608 | case eArgRepeatPairPlain: |
609 | case eArgRepeatPairOptional: |
610 | case eArgRepeatPairPlus: |
611 | case eArgRepeatPairStar: |
612 | case eArgRepeatPairRange: |
613 | case eArgRepeatPairRangeOptional: |
614 | // These should not be hit, as they should pass the IsPairType test |
615 | // above, and control should have gone into the other branch of the if |
616 | // statement. |
617 | break; |
618 | } |
619 | } |
620 | } |
621 | } |
622 | |
623 | CommandArgumentType |
624 | CommandObject::LookupArgumentName(llvm::StringRef arg_name) { |
625 | CommandArgumentType return_type = eArgTypeLastArg; |
626 | |
627 | arg_name = arg_name.ltrim(Char: '<').rtrim(Char: '>'); |
628 | |
629 | for (int i = 0; i < eArgTypeLastArg; ++i) |
630 | if (arg_name == g_argument_table[i].arg_name) |
631 | return_type = g_argument_table[i].arg_type; |
632 | |
633 | return return_type; |
634 | } |
635 | |
636 | void CommandObject::FormatLongHelpText(Stream &output_strm, |
637 | llvm::StringRef long_help) { |
638 | CommandInterpreter &interpreter = GetCommandInterpreter(); |
639 | std::stringstream lineStream{std::string(long_help)}; |
640 | std::string line; |
641 | while (std::getline(is&: lineStream, str&: line)) { |
642 | if (line.empty()) { |
643 | output_strm << "\n"; |
644 | continue; |
645 | } |
646 | size_t result = line.find_first_not_of(s: " \t"); |
647 | if (result == std::string::npos) { |
648 | result = 0; |
649 | } |
650 | std::string whitespace_prefix = line.substr(pos: 0, n: result); |
651 | std::string remainder = line.substr(pos: result); |
652 | interpreter.OutputFormattedHelpText(strm&: output_strm, prefix: whitespace_prefix, |
653 | help_text: remainder); |
654 | } |
655 | } |
656 | |
657 | void CommandObject::GenerateHelpText(CommandReturnObject &result) { |
658 | GenerateHelpText(result&: result.GetOutputStream()); |
659 | |
660 | result.SetStatus(eReturnStatusSuccessFinishNoResult); |
661 | } |
662 | |
663 | void CommandObject::GenerateHelpText(Stream &output_strm) { |
664 | CommandInterpreter &interpreter = GetCommandInterpreter(); |
665 | std::string help_text(GetHelp()); |
666 | if (WantsRawCommandString()) { |
667 | help_text.append(s: " Expects 'raw' input (see 'help raw-input'.)"); |
668 | } |
669 | interpreter.OutputFormattedHelpText(strm&: output_strm, prefix: "", help_text); |
670 | output_strm << "\nSyntax: "<< GetSyntax() << "\n"; |
671 | Options *options = GetOptions(); |
672 | if (options != nullptr) { |
673 | options->GenerateOptionUsage( |
674 | strm&: output_strm, cmd&: *this, |
675 | screen_width: GetCommandInterpreter().GetDebugger().GetTerminalWidth()); |
676 | } |
677 | llvm::StringRef long_help = GetHelpLong(); |
678 | if (!long_help.empty()) { |
679 | FormatLongHelpText(output_strm, long_help); |
680 | } |
681 | if (!IsDashDashCommand() && options && options->NumCommandOptions() > 0) { |
682 | if (WantsRawCommandString() && !WantsCompletion()) { |
683 | // Emit the message about using ' -- ' between the end of the command |
684 | // options and the raw input conditionally, i.e., only if the command |
685 | // object does not want completion. |
686 | interpreter.OutputFormattedHelpText( |
687 | stream&: output_strm, command_word: "", separator: "", |
688 | help_text: "\nImportant Note: Because this command takes 'raw' input, if you " |
689 | "use any command options" |
690 | " you must use ' -- ' between the end of the command options and the " |
691 | "beginning of the raw input.", |
692 | max_word_len: 1); |
693 | } else if (GetNumArgumentEntries() > 0) { |
694 | // Also emit a warning about using "--" in case you are using a command |
695 | // that takes options and arguments. |
696 | interpreter.OutputFormattedHelpText( |
697 | stream&: output_strm, command_word: "", separator: "", |
698 | help_text: "\nThis command takes options and free-form arguments. If your " |
699 | "arguments resemble" |
700 | " option specifiers (i.e., they start with a - or --), you must use " |
701 | "' -- ' between" |
702 | " the end of the command options and the beginning of the arguments.", |
703 | max_word_len: 1); |
704 | } |
705 | } |
706 | } |
707 | |
708 | void CommandObject::AddIDsArgumentData(CommandObject::IDType type) { |
709 | CommandArgumentEntry arg; |
710 | CommandArgumentData id_arg; |
711 | CommandArgumentData id_range_arg; |
712 | |
713 | // Create the first variant for the first (and only) argument for this |
714 | // command. |
715 | switch (type) { |
716 | case eBreakpointArgs: |
717 | id_arg.arg_type = eArgTypeBreakpointID; |
718 | id_range_arg.arg_type = eArgTypeBreakpointIDRange; |
719 | break; |
720 | case eWatchpointArgs: |
721 | id_arg.arg_type = eArgTypeWatchpointID; |
722 | id_range_arg.arg_type = eArgTypeWatchpointIDRange; |
723 | break; |
724 | } |
725 | id_arg.arg_repetition = eArgRepeatOptional; |
726 | id_range_arg.arg_repetition = eArgRepeatOptional; |
727 | |
728 | // The first (and only) argument for this command could be either an id or an |
729 | // id_range. Push both variants into the entry for the first argument for |
730 | // this command. |
731 | arg.push_back(x: id_arg); |
732 | arg.push_back(x: id_range_arg); |
733 | m_arguments.push_back(x: arg); |
734 | } |
735 | |
736 | const char *CommandObject::GetArgumentTypeAsCString( |
737 | const lldb::CommandArgumentType arg_type) { |
738 | assert(arg_type < eArgTypeLastArg && |
739 | "Invalid argument type passed to GetArgumentTypeAsCString"); |
740 | return g_argument_table[arg_type].arg_name; |
741 | } |
742 | |
743 | const char *CommandObject::GetArgumentDescriptionAsCString( |
744 | const lldb::CommandArgumentType arg_type) { |
745 | assert(arg_type < eArgTypeLastArg && |
746 | "Invalid argument type passed to GetArgumentDescriptionAsCString"); |
747 | return g_argument_table[arg_type].help_text; |
748 | } |
749 | |
750 | Target &CommandObject::GetDummyTarget() { |
751 | return m_interpreter.GetDebugger().GetDummyTarget(); |
752 | } |
753 | |
754 | Target &CommandObject::GetTarget() { |
755 | // Prefer the frozen execution context in the command object. |
756 | if (Target *target = m_exe_ctx.GetTargetPtr()) |
757 | return *target; |
758 | |
759 | // Fallback to the command interpreter's execution context in case we get |
760 | // called after DoExecute has finished. For example, when doing multi-line |
761 | // expression that uses an input reader or breakpoint callbacks. |
762 | if (Target *target = m_interpreter.GetExecutionContext().GetTargetPtr()) |
763 | return *target; |
764 | |
765 | // Finally, if we have no other target, get the selected target. |
766 | if (TargetSP target_sp = m_interpreter.GetDebugger().GetSelectedTarget()) |
767 | return *target_sp; |
768 | |
769 | // We only have the dummy target. |
770 | return GetDummyTarget(); |
771 | } |
772 | |
773 | Thread *CommandObject::GetDefaultThread() { |
774 | Thread *thread_to_use = m_exe_ctx.GetThreadPtr(); |
775 | if (thread_to_use) |
776 | return thread_to_use; |
777 | |
778 | Process *process = m_exe_ctx.GetProcessPtr(); |
779 | if (!process) { |
780 | Target *target = m_exe_ctx.GetTargetPtr(); |
781 | if (!target) { |
782 | target = m_interpreter.GetDebugger().GetSelectedTarget().get(); |
783 | } |
784 | if (target) |
785 | process = target->GetProcessSP().get(); |
786 | } |
787 | |
788 | if (process) |
789 | return process->GetThreadList().GetSelectedThread().get(); |
790 | else |
791 | return nullptr; |
792 | } |
793 | |
794 | void CommandObjectParsed::Execute(const char *args_string, |
795 | CommandReturnObject &result) { |
796 | bool handled = false; |
797 | Args cmd_args(args_string); |
798 | if (HasOverrideCallback()) { |
799 | Args full_args(GetCommandName()); |
800 | full_args.AppendArguments(rhs: cmd_args); |
801 | handled = |
802 | InvokeOverrideCallback(argv: full_args.GetConstArgumentVector(), result); |
803 | } |
804 | if (!handled) { |
805 | for (auto entry : llvm::enumerate(First: cmd_args.entries())) { |
806 | const Args::ArgEntry &value = entry.value(); |
807 | if (!value.ref().empty() && value.GetQuoteChar() == '`') { |
808 | // We have to put the backtick back in place for PreprocessCommand. |
809 | std::string opt_string = value.c_str(); |
810 | Status error; |
811 | error = m_interpreter.PreprocessToken(token&: opt_string); |
812 | if (error.Success()) |
813 | cmd_args.ReplaceArgumentAtIndex(idx: entry.index(), arg_str: opt_string); |
814 | } |
815 | } |
816 | |
817 | if (CheckRequirements(result)) { |
818 | if (ParseOptions(args&: cmd_args, result)) { |
819 | // Call the command-specific version of 'Execute', passing it the |
820 | // already processed arguments. |
821 | if (cmd_args.GetArgumentCount() != 0 && m_arguments.empty()) { |
822 | result.AppendErrorWithFormatv(format: "'{0}' doesn't take any arguments.", |
823 | args: GetCommandName()); |
824 | Cleanup(); |
825 | return; |
826 | } |
827 | m_interpreter.IncreaseCommandUsage(cmd_obj: *this); |
828 | DoExecute(command&: cmd_args, result); |
829 | } |
830 | } |
831 | |
832 | Cleanup(); |
833 | } |
834 | } |
835 | |
836 | void CommandObjectRaw::Execute(const char *args_string, |
837 | CommandReturnObject &result) { |
838 | bool handled = false; |
839 | if (HasOverrideCallback()) { |
840 | std::string full_command(GetCommandName()); |
841 | full_command += ' '; |
842 | full_command += args_string; |
843 | const char *argv[2] = {nullptr, nullptr}; |
844 | argv[0] = full_command.c_str(); |
845 | handled = InvokeOverrideCallback(argv, result); |
846 | } |
847 | if (!handled) { |
848 | if (CheckRequirements(result)) |
849 | DoExecute(command: args_string, result); |
850 | |
851 | Cleanup(); |
852 | } |
853 | } |
854 |
Definitions
- CommandObject
- GetDebugger
- GetHelp
- GetHelpLong
- GetSyntax
- GetCommandName
- SetCommandName
- SetHelp
- SetHelpLong
- SetSyntax
- GetOptions
- ParseOptions
- CheckRequirements
- Cleanup
- HandleCompletion
- HandleArgumentCompletion
- HelpTextContainsWord
- ParseOptionsAndNotify
- AddSimpleArgumentList
- GetNumArgumentEntries
- GetArgumentEntryAtIndex
- FindArgumentDataByType
- GetArgumentHelp
- GetArgumentName
- IsPairType
- ArgRepetitionFromString
- OptSetFiltered
- GetFormattedCommandArguments
- LookupArgumentName
- FormatLongHelpText
- GenerateHelpText
- GenerateHelpText
- AddIDsArgumentData
- GetArgumentTypeAsCString
- GetArgumentDescriptionAsCString
- GetDummyTarget
- GetTarget
- GetDefaultThread
- Execute
Improve your Profiling and Debugging skills
Find out more