180814287SRaphael Isemann //===-- CommandObjectCommands.cpp -----------------------------------------===// 2ebc09c36SJim Ingham // 32946cd70SChandler Carruth // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 42946cd70SChandler Carruth // See https://llvm.org/LICENSE.txt for license information. 52946cd70SChandler Carruth // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6ebc09c36SJim Ingham // 7ebc09c36SJim Ingham //===----------------------------------------------------------------------===// 8ebc09c36SJim Ingham 90e5e5a79SGreg Clayton #include "llvm/ADT/StringRef.h" 100e5e5a79SGreg Clayton 116e3d8e7fSEugene Zelenko #include "CommandObjectCommands.h" 1246d4aa21SEnrico Granata #include "CommandObjectHelp.h" 13ebc09c36SJim Ingham #include "lldb/Core/Debugger.h" 1444d93782SGreg Clayton #include "lldb/Core/IOHandler.h" 157594f14fSEnrico Granata #include "lldb/Interpreter/CommandHistory.h" 16ebc09c36SJim Ingham #include "lldb/Interpreter/CommandInterpreter.h" 17de164aaaSGreg Clayton #include "lldb/Interpreter/CommandObjectRegexCommand.h" 18ebc09c36SJim Ingham #include "lldb/Interpreter/CommandReturnObject.h" 1947cbf4a0SPavel Labath #include "lldb/Interpreter/OptionArgParser.h" 20012d4fcaSEnrico Granata #include "lldb/Interpreter/OptionValueBoolean.h" 2145d0e238SEnrico Granata #include "lldb/Interpreter/OptionValueString.h" 227594f14fSEnrico Granata #include "lldb/Interpreter/OptionValueUInt64.h" 23ebc09c36SJim Ingham #include "lldb/Interpreter/Options.h" 2499f0b8f9SEnrico Granata #include "lldb/Interpreter/ScriptInterpreter.h" 25145d95c9SPavel Labath #include "lldb/Utility/Args.h" 26573ab909SZachary Turner #include "lldb/Utility/StringList.h" 27ebc09c36SJim Ingham 28ebc09c36SJim Ingham using namespace lldb; 29ebc09c36SJim Ingham using namespace lldb_private; 30ebc09c36SJim Ingham 31ebc09c36SJim Ingham // CommandObjectCommandsSource 32ebc09c36SJim Ingham 3364becc11SRaphael Isemann #define LLDB_OPTIONS_source 3464becc11SRaphael Isemann #include "CommandOptions.inc" 351f0f5b5bSZachary Turner 36b9c1b51eSKate Stone class CommandObjectCommandsSource : public CommandObjectParsed { 375a988416SJim Ingham public: 387428a18cSKate Stone CommandObjectCommandsSource(CommandInterpreter &interpreter) 39b9c1b51eSKate Stone : CommandObjectParsed( 40b9c1b51eSKate Stone interpreter, "command source", 41b9c1b51eSKate Stone "Read and execute LLDB commands from the file <filename>.", 426e3d8e7fSEugene Zelenko nullptr), 43b9c1b51eSKate Stone m_options() { 445a988416SJim Ingham CommandArgumentEntry arg; 455a988416SJim Ingham CommandArgumentData file_arg; 465a988416SJim Ingham 475a988416SJim Ingham // Define the first (and only) variant of this arg. 485a988416SJim Ingham file_arg.arg_type = eArgTypeFilename; 495a988416SJim Ingham file_arg.arg_repetition = eArgRepeatPlain; 505a988416SJim Ingham 51b9c1b51eSKate Stone // There is only one variant this argument could be; put it into the 52b9c1b51eSKate Stone // argument entry. 535a988416SJim Ingham arg.push_back(file_arg); 545a988416SJim Ingham 555a988416SJim Ingham // Push the data for the first argument into the m_arguments vector. 565a988416SJim Ingham m_arguments.push_back(arg); 575a988416SJim Ingham } 585a988416SJim Ingham 596e3d8e7fSEugene Zelenko ~CommandObjectCommandsSource() override = default; 605a988416SJim Ingham 61b9c1b51eSKate Stone const char *GetRepeatCommand(Args ¤t_command_args, 62b9c1b51eSKate Stone uint32_t index) override { 635a988416SJim Ingham return ""; 645a988416SJim Ingham } 655a988416SJim Ingham 66ae34ed2cSRaphael Isemann void 67ae34ed2cSRaphael Isemann HandleArgumentCompletion(CompletionRequest &request, 682443bbd4SRaphael Isemann OptionElementVector &opt_element_vector) override { 69b9c1b51eSKate Stone CommandCompletions::InvokeCommonCompletionCallbacks( 70b9c1b51eSKate Stone GetCommandInterpreter(), CommandCompletions::eDiskFileCompletion, 71a2e76c0bSRaphael Isemann request, nullptr); 725a988416SJim Ingham } 735a988416SJim Ingham 74b9c1b51eSKate Stone Options *GetOptions() override { return &m_options; } 755a988416SJim Ingham 765a988416SJim Ingham protected: 77b9c1b51eSKate Stone class CommandOptions : public Options { 78e16c50a1SJim Ingham public: 79b9c1b51eSKate Stone CommandOptions() 80b9c1b51eSKate Stone : Options(), m_stop_on_error(true), m_silent_run(false), 81b9c1b51eSKate Stone m_stop_on_continue(true) {} 82e16c50a1SJim Ingham 836e3d8e7fSEugene Zelenko ~CommandOptions() override = default; 84e16c50a1SJim Ingham 8597206d57SZachary Turner Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg, 86b9c1b51eSKate Stone ExecutionContext *execution_context) override { 8797206d57SZachary Turner Status error; 883bcdfc0eSGreg Clayton const int short_option = m_getopt_table[option_idx].val; 89e16c50a1SJim Ingham 90b9c1b51eSKate Stone switch (short_option) { 91e16c50a1SJim Ingham case 'e': 92fe11483bSZachary Turner error = m_stop_on_error.SetValueFromString(option_arg); 93e16c50a1SJim Ingham break; 94340b0309SGreg Clayton 95e16c50a1SJim Ingham case 'c': 96fe11483bSZachary Turner error = m_stop_on_continue.SetValueFromString(option_arg); 97e16c50a1SJim Ingham break; 98340b0309SGreg Clayton 9960986174SMichael Sartain case 's': 100fe11483bSZachary Turner error = m_silent_run.SetValueFromString(option_arg); 10160986174SMichael Sartain break; 102340b0309SGreg Clayton 103e16c50a1SJim Ingham default: 10436162014SRaphael Isemann llvm_unreachable("Unimplemented option"); 105e16c50a1SJim Ingham } 106e16c50a1SJim Ingham 107e16c50a1SJim Ingham return error; 108e16c50a1SJim Ingham } 109e16c50a1SJim Ingham 110b9c1b51eSKate Stone void OptionParsingStarting(ExecutionContext *execution_context) override { 111012d4fcaSEnrico Granata m_stop_on_error.Clear(); 112340b0309SGreg Clayton m_silent_run.Clear(); 113340b0309SGreg Clayton m_stop_on_continue.Clear(); 114e16c50a1SJim Ingham } 115e16c50a1SJim Ingham 1161f0f5b5bSZachary Turner llvm::ArrayRef<OptionDefinition> GetDefinitions() override { 11770602439SZachary Turner return llvm::makeArrayRef(g_source_options); 1181f0f5b5bSZachary Turner } 119e16c50a1SJim Ingham 120e16c50a1SJim Ingham // Instance variables to hold the values for command options. 121e16c50a1SJim Ingham 122012d4fcaSEnrico Granata OptionValueBoolean m_stop_on_error; 123340b0309SGreg Clayton OptionValueBoolean m_silent_run; 124340b0309SGreg Clayton OptionValueBoolean m_stop_on_continue; 125e16c50a1SJim Ingham }; 126e16c50a1SJim Ingham 127b9c1b51eSKate Stone bool DoExecute(Args &command, CommandReturnObject &result) override { 1284574a890SZachary Turner if (command.GetArgumentCount() != 1) { 1294574a890SZachary Turner result.AppendErrorWithFormat( 1304574a890SZachary Turner "'%s' takes exactly one executable filename argument.\n", 1314574a890SZachary Turner GetCommandName().str().c_str()); 1324574a890SZachary Turner result.SetStatus(eReturnStatusFailed); 1334574a890SZachary Turner return false; 1344574a890SZachary Turner } 135ebc09c36SJim Ingham 1360d9a201eSRaphael Isemann FileSpec cmd_file(command[0].ref()); 1378f3be7a3SJonas Devlieghere FileSystem::Instance().Resolve(cmd_file); 1386e3d8e7fSEugene Zelenko ExecutionContext *exe_ctx = nullptr; // Just use the default context. 139ebc09c36SJim Ingham 140340b0309SGreg Clayton // If any options were set, then use them 141340b0309SGreg Clayton if (m_options.m_stop_on_error.OptionWasSet() || 142340b0309SGreg Clayton m_options.m_silent_run.OptionWasSet() || 143b9c1b51eSKate Stone m_options.m_stop_on_continue.OptionWasSet()) { 144340b0309SGreg Clayton // Use user set settings 14526c7bf93SJim Ingham CommandInterpreterRunOptions options; 1461c19b74cSJonas Devlieghere 1471c19b74cSJonas Devlieghere if (m_options.m_stop_on_continue.OptionWasSet()) 1481c19b74cSJonas Devlieghere options.SetStopOnContinue( 1491c19b74cSJonas Devlieghere m_options.m_stop_on_continue.GetCurrentValue()); 1501c19b74cSJonas Devlieghere 1511c19b74cSJonas Devlieghere if (m_options.m_stop_on_error.OptionWasSet()) 15226c7bf93SJim Ingham options.SetStopOnError(m_options.m_stop_on_error.GetCurrentValue()); 153c678ed77SStefan Granitz 154c678ed77SStefan Granitz // Individual silent setting is override for global command echo settings. 155c678ed77SStefan Granitz if (m_options.m_silent_run.GetCurrentValue()) { 156c678ed77SStefan Granitz options.SetSilent(true); 157c678ed77SStefan Granitz } else { 158c678ed77SStefan Granitz options.SetPrintResults(true); 159c0b48ab6SJonas Devlieghere options.SetPrintErrors(true); 160c678ed77SStefan Granitz options.SetEchoCommands(m_interpreter.GetEchoCommands()); 161c678ed77SStefan Granitz options.SetEchoCommentCommands(m_interpreter.GetEchoCommentCommands()); 162c678ed77SStefan Granitz } 16326c7bf93SJim Ingham 1644574a890SZachary Turner m_interpreter.HandleCommandsFromFile(cmd_file, exe_ctx, options, result); 165b9c1b51eSKate Stone } else { 16605097246SAdrian Prantl // No options were set, inherit any settings from nested "command source" 16705097246SAdrian Prantl // commands, or set to sane default settings... 16826c7bf93SJim Ingham CommandInterpreterRunOptions options; 1694574a890SZachary Turner m_interpreter.HandleCommandsFromFile(cmd_file, exe_ctx, options, result); 170ebc09c36SJim Ingham } 171ebc09c36SJim Ingham return result.Succeeded(); 172ebc09c36SJim Ingham } 1736e3d8e7fSEugene Zelenko 1745a988416SJim Ingham CommandOptions m_options; 175ebc09c36SJim Ingham }; 176ebc09c36SJim Ingham 177ebc09c36SJim Ingham #pragma mark CommandObjectCommandsAlias 178ebc09c36SJim Ingham // CommandObjectCommandsAlias 179ebc09c36SJim Ingham 18064becc11SRaphael Isemann #define LLDB_OPTIONS_alias 18164becc11SRaphael Isemann #include "CommandOptions.inc" 1821f0f5b5bSZachary Turner 183b9c1b51eSKate Stone static const char *g_python_command_instructions = 184b9c1b51eSKate Stone "Enter your Python command(s). Type 'DONE' to end.\n" 185be93a35aSEnrico Granata "You must define a Python function with this signature:\n" 18644d93782SGreg Clayton "def my_command_impl(debugger, args, result, internal_dict):\n"; 187be93a35aSEnrico Granata 188b9c1b51eSKate Stone class CommandObjectCommandsAlias : public CommandObjectRaw { 18945d0e238SEnrico Granata protected: 190b9c1b51eSKate Stone class CommandOptions : public OptionGroup { 191ebc09c36SJim Ingham public: 192b9c1b51eSKate Stone CommandOptions() : OptionGroup(), m_help(), m_long_help() {} 19345d0e238SEnrico Granata 19445d0e238SEnrico Granata ~CommandOptions() override = default; 19545d0e238SEnrico Granata 1961f0f5b5bSZachary Turner llvm::ArrayRef<OptionDefinition> GetDefinitions() override { 19770602439SZachary Turner return llvm::makeArrayRef(g_alias_options); 1981f0f5b5bSZachary Turner } 19945d0e238SEnrico Granata 20097206d57SZachary Turner Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_value, 201b9c1b51eSKate Stone ExecutionContext *execution_context) override { 20297206d57SZachary Turner Status error; 20345d0e238SEnrico Granata 2041f0f5b5bSZachary Turner const int short_option = GetDefinitions()[option_idx].short_option; 2058cef4b0bSZachary Turner std::string option_str(option_value); 20645d0e238SEnrico Granata 207b9c1b51eSKate Stone switch (short_option) { 20845d0e238SEnrico Granata case 'h': 2098cef4b0bSZachary Turner m_help.SetCurrentValue(option_str); 21045d0e238SEnrico Granata m_help.SetOptionWasSet(); 21145d0e238SEnrico Granata break; 21245d0e238SEnrico Granata 21345d0e238SEnrico Granata case 'H': 2148cef4b0bSZachary Turner m_long_help.SetCurrentValue(option_str); 21545d0e238SEnrico Granata m_long_help.SetOptionWasSet(); 21645d0e238SEnrico Granata break; 21745d0e238SEnrico Granata 21845d0e238SEnrico Granata default: 21936162014SRaphael Isemann llvm_unreachable("Unimplemented option"); 22045d0e238SEnrico Granata } 22145d0e238SEnrico Granata 22245d0e238SEnrico Granata return error; 22345d0e238SEnrico Granata } 22445d0e238SEnrico Granata 225b9c1b51eSKate Stone void OptionParsingStarting(ExecutionContext *execution_context) override { 22645d0e238SEnrico Granata m_help.Clear(); 22745d0e238SEnrico Granata m_long_help.Clear(); 22845d0e238SEnrico Granata } 22945d0e238SEnrico Granata 23045d0e238SEnrico Granata OptionValueString m_help; 23145d0e238SEnrico Granata OptionValueString m_long_help; 23245d0e238SEnrico Granata }; 23345d0e238SEnrico Granata 23445d0e238SEnrico Granata OptionGroupOptions m_option_group; 23545d0e238SEnrico Granata CommandOptions m_command_options; 23645d0e238SEnrico Granata 23745d0e238SEnrico Granata public: 238b9c1b51eSKate Stone Options *GetOptions() override { return &m_option_group; } 23945d0e238SEnrico Granata 2407428a18cSKate Stone CommandObjectCommandsAlias(CommandInterpreter &interpreter) 241b9c1b51eSKate Stone : CommandObjectRaw( 242b9c1b51eSKate Stone interpreter, "command alias", 243a449698cSZachary Turner "Define a custom command in terms of an existing command."), 244b9c1b51eSKate Stone m_option_group(), m_command_options() { 24545d0e238SEnrico Granata m_option_group.Append(&m_command_options); 24645d0e238SEnrico Granata m_option_group.Finalize(); 24745d0e238SEnrico Granata 248ebc09c36SJim Ingham SetHelpLong( 249ea671fbdSKate Stone "'alias' allows the user to create a short-cut or abbreviation for long \ 250ea671fbdSKate Stone commands, multi-word commands, and commands that take particular options. \ 251b9c1b51eSKate Stone Below are some simple examples of how one might use the 'alias' command:" 252b9c1b51eSKate Stone R"( 253ea671fbdSKate Stone 254ea671fbdSKate Stone (lldb) command alias sc script 255ea671fbdSKate Stone 256ea671fbdSKate Stone Creates the abbreviation 'sc' for the 'script' command. 257ea671fbdSKate Stone 258ea671fbdSKate Stone (lldb) command alias bp breakpoint 259ea671fbdSKate Stone 260b9c1b51eSKate Stone )" 261b9c1b51eSKate Stone " Creates the abbreviation 'bp' for the 'breakpoint' command. Since \ 262ea671fbdSKate Stone breakpoint commands are two-word commands, the user would still need to \ 263b9c1b51eSKate Stone enter the second word after 'bp', e.g. 'bp enable' or 'bp delete'." 264b9c1b51eSKate Stone R"( 265ea671fbdSKate Stone 266ea671fbdSKate Stone (lldb) command alias bpl breakpoint list 267ea671fbdSKate Stone 268ea671fbdSKate Stone Creates the abbreviation 'bpl' for the two-word command 'breakpoint list'. 269ea671fbdSKate Stone 270b9c1b51eSKate Stone )" 271b9c1b51eSKate Stone "An alias can include some options for the command, with the values either \ 272ea671fbdSKate Stone filled in at the time the alias is created, or specified as positional \ 273ea671fbdSKate Stone arguments, to be filled in when the alias is invoked. The following example \ 274b9c1b51eSKate Stone shows how to create aliases with options:" 275b9c1b51eSKate Stone R"( 276ea671fbdSKate Stone 277ea671fbdSKate Stone (lldb) command alias bfl breakpoint set -f %1 -l %2 278ea671fbdSKate Stone 279b9c1b51eSKate Stone )" 280b9c1b51eSKate Stone " Creates the abbreviation 'bfl' (for break-file-line), with the -f and -l \ 281ea671fbdSKate Stone options already part of the alias. So if the user wants to set a breakpoint \ 282ea671fbdSKate Stone by file and line without explicitly having to use the -f and -l options, the \ 283ea671fbdSKate Stone user can now use 'bfl' instead. The '%1' and '%2' are positional placeholders \ 284ea671fbdSKate Stone for the actual arguments that will be passed when the alias command is used. \ 285ea671fbdSKate Stone The number in the placeholder refers to the position/order the actual value \ 286ea671fbdSKate Stone occupies when the alias is used. All the occurrences of '%1' in the alias \ 287ea671fbdSKate Stone will be replaced with the first argument, all the occurrences of '%2' in the \ 288ea671fbdSKate Stone alias will be replaced with the second argument, and so on. This also allows \ 289ea671fbdSKate Stone actual arguments to be used multiple times within an alias (see 'process \ 290b9c1b51eSKate Stone launch' example below)." 291b9c1b51eSKate Stone R"( 292ea671fbdSKate Stone 293b9c1b51eSKate Stone )" 294b9c1b51eSKate Stone "Note: the positional arguments must substitute as whole words in the resultant \ 295ea671fbdSKate Stone command, so you can't at present do something like this to append the file extension \ 296b9c1b51eSKate Stone \".cpp\":" 297b9c1b51eSKate Stone R"( 298ea671fbdSKate Stone 299ea671fbdSKate Stone (lldb) command alias bcppfl breakpoint set -f %1.cpp -l %2 300ea671fbdSKate Stone 301b9c1b51eSKate Stone )" 302b9c1b51eSKate Stone "For more complex aliasing, use the \"command regex\" command instead. In the \ 303ea671fbdSKate Stone 'bfl' case above, the actual file value will be filled in with the first argument \ 304ea671fbdSKate Stone following 'bfl' and the actual line number value will be filled in with the second \ 305b9c1b51eSKate Stone argument. The user would use this alias as follows:" 306b9c1b51eSKate Stone R"( 307ea671fbdSKate Stone 308ea671fbdSKate Stone (lldb) command alias bfl breakpoint set -f %1 -l %2 309ea671fbdSKate Stone (lldb) bfl my-file.c 137 310ea671fbdSKate Stone 311ea671fbdSKate Stone This would be the same as if the user had entered 'breakpoint set -f my-file.c -l 137'. 312ea671fbdSKate Stone 313ea671fbdSKate Stone Another example: 314ea671fbdSKate Stone 315ea671fbdSKate Stone (lldb) command alias pltty process launch -s -o %1 -e %1 316ea671fbdSKate Stone (lldb) pltty /dev/tty0 317ea671fbdSKate Stone 318ea671fbdSKate Stone Interpreted as 'process launch -s -o /dev/tty0 -e /dev/tty0' 319ea671fbdSKate Stone 320b9c1b51eSKate Stone )" 321b9c1b51eSKate Stone "If the user always wanted to pass the same value to a particular option, the \ 322ea671fbdSKate Stone alias could be defined with that value directly in the alias as a constant, \ 323b9c1b51eSKate Stone rather than using a positional placeholder:" 324b9c1b51eSKate Stone R"( 325ea671fbdSKate Stone 326ea671fbdSKate Stone (lldb) command alias bl3 breakpoint set -f %1 -l 3 327ea671fbdSKate Stone 328b9c1b51eSKate Stone Always sets a breakpoint on line 3 of whatever file is indicated.)"); 329ebc09c36SJim Ingham 330405fe67fSCaroline Tice CommandArgumentEntry arg1; 331405fe67fSCaroline Tice CommandArgumentEntry arg2; 332405fe67fSCaroline Tice CommandArgumentEntry arg3; 333405fe67fSCaroline Tice CommandArgumentData alias_arg; 334405fe67fSCaroline Tice CommandArgumentData cmd_arg; 335405fe67fSCaroline Tice CommandArgumentData options_arg; 336405fe67fSCaroline Tice 337405fe67fSCaroline Tice // Define the first (and only) variant of this arg. 338405fe67fSCaroline Tice alias_arg.arg_type = eArgTypeAliasName; 339405fe67fSCaroline Tice alias_arg.arg_repetition = eArgRepeatPlain; 340405fe67fSCaroline Tice 341b9c1b51eSKate Stone // There is only one variant this argument could be; put it into the 342b9c1b51eSKate Stone // argument entry. 343405fe67fSCaroline Tice arg1.push_back(alias_arg); 344405fe67fSCaroline Tice 345405fe67fSCaroline Tice // Define the first (and only) variant of this arg. 346405fe67fSCaroline Tice cmd_arg.arg_type = eArgTypeCommandName; 347405fe67fSCaroline Tice cmd_arg.arg_repetition = eArgRepeatPlain; 348405fe67fSCaroline Tice 349b9c1b51eSKate Stone // There is only one variant this argument could be; put it into the 350b9c1b51eSKate Stone // argument entry. 351405fe67fSCaroline Tice arg2.push_back(cmd_arg); 352405fe67fSCaroline Tice 353405fe67fSCaroline Tice // Define the first (and only) variant of this arg. 354405fe67fSCaroline Tice options_arg.arg_type = eArgTypeAliasOptions; 355405fe67fSCaroline Tice options_arg.arg_repetition = eArgRepeatOptional; 356405fe67fSCaroline Tice 357b9c1b51eSKate Stone // There is only one variant this argument could be; put it into the 358b9c1b51eSKate Stone // argument entry. 359405fe67fSCaroline Tice arg3.push_back(options_arg); 360405fe67fSCaroline Tice 361405fe67fSCaroline Tice // Push the data for the first argument into the m_arguments vector. 362405fe67fSCaroline Tice m_arguments.push_back(arg1); 363405fe67fSCaroline Tice m_arguments.push_back(arg2); 364405fe67fSCaroline Tice m_arguments.push_back(arg3); 365ebc09c36SJim Ingham } 366ebc09c36SJim Ingham 3676e3d8e7fSEugene Zelenko ~CommandObjectCommandsAlias() override = default; 368ebc09c36SJim Ingham 3695a988416SJim Ingham protected: 3704d51a902SRaphael Isemann bool DoExecute(llvm::StringRef raw_command_line, 371b9c1b51eSKate Stone CommandReturnObject &result) override { 3724d51a902SRaphael Isemann if (raw_command_line.empty()) { 373d72e412fSEnrico Granata result.AppendError("'command alias' requires at least two arguments"); 37445d0e238SEnrico Granata return false; 37545d0e238SEnrico Granata } 37645d0e238SEnrico Granata 377e1cfbc79STodd Fiala ExecutionContext exe_ctx = GetCommandInterpreter().GetExecutionContext(); 378e1cfbc79STodd Fiala m_option_group.NotifyOptionParsingStarting(&exe_ctx); 37945d0e238SEnrico Granata 3803a0e1270SRaphael Isemann OptionsWithRaw args_with_suffix(raw_command_line); 38145d0e238SEnrico Granata 3823a0e1270SRaphael Isemann if (args_with_suffix.HasArgs()) 3833a0e1270SRaphael Isemann if (!ParseOptionsAndNotify(args_with_suffix.GetArgs(), result, 3843a0e1270SRaphael Isemann m_option_group, exe_ctx)) 38545d0e238SEnrico Granata return false; 38645d0e238SEnrico Granata 387daed98e5SShivam Mittal llvm::StringRef raw_command_string = args_with_suffix.GetRawPart(); 388a01bccdbSZachary Turner Args args(raw_command_string); 389844d2303SCaroline Tice 39011eb9c64SZachary Turner if (args.GetArgumentCount() < 2) { 391d72e412fSEnrico Granata result.AppendError("'command alias' requires at least two arguments"); 392844d2303SCaroline Tice result.SetStatus(eReturnStatusFailed); 393844d2303SCaroline Tice return false; 394844d2303SCaroline Tice } 395844d2303SCaroline Tice 396844d2303SCaroline Tice // Get the alias command. 397844d2303SCaroline Tice 3980d9a201eSRaphael Isemann auto alias_command = args[0].ref(); 3994574a890SZachary Turner if (alias_command.startswith("-")) { 400d72e412fSEnrico Granata result.AppendError("aliases starting with a dash are not supported"); 401b9c1b51eSKate Stone if (alias_command == "--help" || alias_command == "--long-help") { 402b9c1b51eSKate Stone result.AppendWarning("if trying to pass options to 'command alias' add " 403b9c1b51eSKate Stone "a -- at the end of the options"); 404d72e412fSEnrico Granata } 405d72e412fSEnrico Granata result.SetStatus(eReturnStatusFailed); 406d72e412fSEnrico Granata return false; 407d72e412fSEnrico Granata } 408844d2303SCaroline Tice 409b9c1b51eSKate Stone // Strip the new alias name off 'raw_command_string' (leave it on args, 41005097246SAdrian Prantl // which gets passed to 'Execute', which does the stripping itself. 411844d2303SCaroline Tice size_t pos = raw_command_string.find(alias_command); 412b9c1b51eSKate Stone if (pos == 0) { 413844d2303SCaroline Tice raw_command_string = raw_command_string.substr(alias_command.size()); 414844d2303SCaroline Tice pos = raw_command_string.find_first_not_of(' '); 415844d2303SCaroline Tice if ((pos != std::string::npos) && (pos > 0)) 416844d2303SCaroline Tice raw_command_string = raw_command_string.substr(pos); 417b9c1b51eSKate Stone } else { 418844d2303SCaroline Tice result.AppendError("Error parsing command string. No alias created."); 419844d2303SCaroline Tice result.SetStatus(eReturnStatusFailed); 420844d2303SCaroline Tice return false; 421844d2303SCaroline Tice } 422844d2303SCaroline Tice 423844d2303SCaroline Tice // Verify that the command is alias-able. 424771ef6d4SMalcolm Parsons if (m_interpreter.CommandExists(alias_command)) { 425b9c1b51eSKate Stone result.AppendErrorWithFormat( 426b9c1b51eSKate Stone "'%s' is a permanent debugger command and cannot be redefined.\n", 4274574a890SZachary Turner args[0].c_str()); 428844d2303SCaroline Tice result.SetStatus(eReturnStatusFailed); 429844d2303SCaroline Tice return false; 430844d2303SCaroline Tice } 431844d2303SCaroline Tice 432b9c1b51eSKate Stone // Get CommandObject that is being aliased. The command name is read from 433a01bccdbSZachary Turner // the front of raw_command_string. raw_command_string is returned with the 434a01bccdbSZachary Turner // name of the command object stripped off the front. 435a01bccdbSZachary Turner llvm::StringRef original_raw_command_string = raw_command_string; 436b9c1b51eSKate Stone CommandObject *cmd_obj = 437b9c1b51eSKate Stone m_interpreter.GetCommandObjectForCommand(raw_command_string); 438844d2303SCaroline Tice 439b9c1b51eSKate Stone if (!cmd_obj) { 440b9c1b51eSKate Stone result.AppendErrorWithFormat("invalid command given to 'command alias'. " 441b9c1b51eSKate Stone "'%s' does not begin with a valid command." 442b9c1b51eSKate Stone " No alias created.", 443a01bccdbSZachary Turner original_raw_command_string.str().c_str()); 444844d2303SCaroline Tice result.SetStatus(eReturnStatusFailed); 445844d2303SCaroline Tice return false; 446b9c1b51eSKate Stone } else if (!cmd_obj->WantsRawCommandString()) { 447b9c1b51eSKate Stone // Note that args was initialized with the original command, and has not 44805097246SAdrian Prantl // been updated to this point. Therefore can we pass it to the version of 44905097246SAdrian Prantl // Execute that does not need/expect raw input in the alias. 4505a988416SJim Ingham return HandleAliasingNormalCommand(args, result); 451b9c1b51eSKate Stone } else { 452b9c1b51eSKate Stone return HandleAliasingRawCommand(alias_command, raw_command_string, 453b9c1b51eSKate Stone *cmd_obj, result); 4545a988416SJim Ingham } 4555a988416SJim Ingham return result.Succeeded(); 4565a988416SJim Ingham } 4575a988416SJim Ingham 458a01bccdbSZachary Turner bool HandleAliasingRawCommand(llvm::StringRef alias_command, 459a01bccdbSZachary Turner llvm::StringRef raw_command_string, 460b9c1b51eSKate Stone CommandObject &cmd_obj, 461b9c1b51eSKate Stone CommandReturnObject &result) { 462844d2303SCaroline Tice // Verify & handle any options/arguments passed to the alias command 463844d2303SCaroline Tice 464b9c1b51eSKate Stone OptionArgVectorSP option_arg_vector_sp = 465b9c1b51eSKate Stone OptionArgVectorSP(new OptionArgVector); 466844d2303SCaroline Tice 467b9c1b51eSKate Stone if (CommandObjectSP cmd_obj_sp = 468b9c1b51eSKate Stone m_interpreter.GetCommandSPExact(cmd_obj.GetCommandName(), false)) { 469a01bccdbSZachary Turner if (m_interpreter.AliasExists(alias_command) || 470a01bccdbSZachary Turner m_interpreter.UserCommandExists(alias_command)) { 471b9c1b51eSKate Stone result.AppendWarningWithFormat( 472b9c1b51eSKate Stone "Overwriting existing definition for '%s'.\n", 473a01bccdbSZachary Turner alias_command.str().c_str()); 474844d2303SCaroline Tice } 475b9c1b51eSKate Stone if (CommandAlias *alias = m_interpreter.AddAlias( 476a01bccdbSZachary Turner alias_command, cmd_obj_sp, raw_command_string)) { 47745d0e238SEnrico Granata if (m_command_options.m_help.OptionWasSet()) 47845d0e238SEnrico Granata alias->SetHelp(m_command_options.m_help.GetCurrentValue()); 47945d0e238SEnrico Granata if (m_command_options.m_long_help.OptionWasSet()) 48045d0e238SEnrico Granata alias->SetHelpLong(m_command_options.m_long_help.GetCurrentValue()); 481844d2303SCaroline Tice result.SetStatus(eReturnStatusSuccessFinishNoResult); 482b9c1b51eSKate Stone } else { 483472362e6SCaroline Tice result.AppendError("Unable to create requested alias.\n"); 484472362e6SCaroline Tice result.SetStatus(eReturnStatusFailed); 485472362e6SCaroline Tice } 486212130acSEnrico Granata 487b9c1b51eSKate Stone } else { 488212130acSEnrico Granata result.AppendError("Unable to create requested alias.\n"); 489212130acSEnrico Granata result.SetStatus(eReturnStatusFailed); 490212130acSEnrico Granata } 491212130acSEnrico Granata 492844d2303SCaroline Tice return result.Succeeded(); 493844d2303SCaroline Tice } 494ebc09c36SJim Ingham 495b9c1b51eSKate Stone bool HandleAliasingNormalCommand(Args &args, CommandReturnObject &result) { 496867b185dSCaroline Tice size_t argc = args.GetArgumentCount(); 497ebc09c36SJim Ingham 498b9c1b51eSKate Stone if (argc < 2) { 499d72e412fSEnrico Granata result.AppendError("'command alias' requires at least two arguments"); 500ebc09c36SJim Ingham result.SetStatus(eReturnStatusFailed); 501ebc09c36SJim Ingham return false; 502ebc09c36SJim Ingham } 503ebc09c36SJim Ingham 5044574a890SZachary Turner // Save these in std::strings since we're going to shift them off. 505adcd0268SBenjamin Kramer const std::string alias_command(std::string(args[0].ref())); 506adcd0268SBenjamin Kramer const std::string actual_command(std::string(args[1].ref())); 507ebc09c36SJim Ingham 508ebc09c36SJim Ingham args.Shift(); // Shift the alias command word off the argument vector. 509ebc09c36SJim Ingham args.Shift(); // Shift the old command word off the argument vector. 510ebc09c36SJim Ingham 511b9c1b51eSKate Stone // Verify that the command is alias'able, and get the appropriate command 512b9c1b51eSKate Stone // object. 513ebc09c36SJim Ingham 514771ef6d4SMalcolm Parsons if (m_interpreter.CommandExists(alias_command)) { 515b9c1b51eSKate Stone result.AppendErrorWithFormat( 516b9c1b51eSKate Stone "'%s' is a permanent debugger command and cannot be redefined.\n", 517ebc09c36SJim Ingham alias_command.c_str()); 518ebc09c36SJim Ingham result.SetStatus(eReturnStatusFailed); 5194574a890SZachary Turner return false; 5204574a890SZachary Turner } 5214574a890SZachary Turner 522b9c1b51eSKate Stone CommandObjectSP command_obj_sp( 523a449698cSZachary Turner m_interpreter.GetCommandSPExact(actual_command, true)); 524ebc09c36SJim Ingham CommandObjectSP subcommand_obj_sp; 525ebc09c36SJim Ingham bool use_subcommand = false; 5264574a890SZachary Turner if (!command_obj_sp) { 5274574a890SZachary Turner result.AppendErrorWithFormat("'%s' is not an existing command.\n", 5284574a890SZachary Turner actual_command.c_str()); 5294574a890SZachary Turner result.SetStatus(eReturnStatusFailed); 5304574a890SZachary Turner return false; 5314574a890SZachary Turner } 532ebc09c36SJim Ingham CommandObject *cmd_obj = command_obj_sp.get(); 5336e3d8e7fSEugene Zelenko CommandObject *sub_cmd_obj = nullptr; 534b9c1b51eSKate Stone OptionArgVectorSP option_arg_vector_sp = 535b9c1b51eSKate Stone OptionArgVectorSP(new OptionArgVector); 536ebc09c36SJim Ingham 53711eb9c64SZachary Turner while (cmd_obj->IsMultiwordObject() && !args.empty()) { 5380d9a201eSRaphael Isemann auto sub_command = args[0].ref(); 53911eb9c64SZachary Turner assert(!sub_command.empty()); 5404574a890SZachary Turner subcommand_obj_sp = cmd_obj->GetSubcommandSP(sub_command); 5414574a890SZachary Turner if (!subcommand_obj_sp) { 542b9c1b51eSKate Stone result.AppendErrorWithFormat( 543b9c1b51eSKate Stone "'%s' is not a valid sub-command of '%s'. " 544f415eeb4SCaroline Tice "Unable to create alias.\n", 5454574a890SZachary Turner args[0].c_str(), actual_command.c_str()); 546ebc09c36SJim Ingham result.SetStatus(eReturnStatusFailed); 547ebc09c36SJim Ingham return false; 548ebc09c36SJim Ingham } 5494574a890SZachary Turner 5504574a890SZachary Turner sub_cmd_obj = subcommand_obj_sp.get(); 5514574a890SZachary Turner use_subcommand = true; 5524574a890SZachary Turner args.Shift(); // Shift the sub_command word off the argument vector. 5534574a890SZachary Turner cmd_obj = sub_cmd_obj; 554ebc09c36SJim Ingham } 555ebc09c36SJim Ingham 556ebc09c36SJim Ingham // Verify & handle any options/arguments passed to the alias command 557ebc09c36SJim Ingham 558212130acSEnrico Granata std::string args_string; 559212130acSEnrico Granata 56011eb9c64SZachary Turner if (!args.empty()) { 561b9c1b51eSKate Stone CommandObjectSP tmp_sp = 562b9c1b51eSKate Stone m_interpreter.GetCommandSPExact(cmd_obj->GetCommandName(), false); 563ebc09c36SJim Ingham if (use_subcommand) 5644574a890SZachary Turner tmp_sp = m_interpreter.GetCommandSPExact(sub_cmd_obj->GetCommandName(), 5654574a890SZachary Turner false); 566ca90c47eSCaroline Tice 567ca90c47eSCaroline Tice args.GetCommandString(args_string); 568867b185dSCaroline Tice } 569ebc09c36SJim Ingham 570771ef6d4SMalcolm Parsons if (m_interpreter.AliasExists(alias_command) || 571771ef6d4SMalcolm Parsons m_interpreter.UserCommandExists(alias_command)) { 572b9c1b51eSKate Stone result.AppendWarningWithFormat( 5734574a890SZachary Turner "Overwriting existing definition for '%s'.\n", alias_command.c_str()); 574ebc09c36SJim Ingham } 575ebc09c36SJim Ingham 576b9c1b51eSKate Stone if (CommandAlias *alias = m_interpreter.AddAlias( 5774574a890SZachary Turner alias_command, use_subcommand ? subcommand_obj_sp : command_obj_sp, 578771ef6d4SMalcolm Parsons args_string)) { 57945d0e238SEnrico Granata if (m_command_options.m_help.OptionWasSet()) 58045d0e238SEnrico Granata alias->SetHelp(m_command_options.m_help.GetCurrentValue()); 58145d0e238SEnrico Granata if (m_command_options.m_long_help.OptionWasSet()) 58245d0e238SEnrico Granata alias->SetHelpLong(m_command_options.m_long_help.GetCurrentValue()); 583ebc09c36SJim Ingham result.SetStatus(eReturnStatusSuccessFinishNoResult); 584b9c1b51eSKate Stone } else { 585212130acSEnrico Granata result.AppendError("Unable to create requested alias.\n"); 586212130acSEnrico Granata result.SetStatus(eReturnStatusFailed); 587212130acSEnrico Granata return false; 588212130acSEnrico Granata } 589ebc09c36SJim Ingham 590ebc09c36SJim Ingham return result.Succeeded(); 591ebc09c36SJim Ingham } 592ebc09c36SJim Ingham }; 593ebc09c36SJim Ingham 594ebc09c36SJim Ingham #pragma mark CommandObjectCommandsUnalias 595ebc09c36SJim Ingham // CommandObjectCommandsUnalias 596ebc09c36SJim Ingham 597b9c1b51eSKate Stone class CommandObjectCommandsUnalias : public CommandObjectParsed { 598ebc09c36SJim Ingham public: 5997428a18cSKate Stone CommandObjectCommandsUnalias(CommandInterpreter &interpreter) 600b9c1b51eSKate Stone : CommandObjectParsed( 601b9c1b51eSKate Stone interpreter, "command unalias", 602b9c1b51eSKate Stone "Delete one or more custom commands defined by 'command alias'.", 603b9c1b51eSKate Stone nullptr) { 604405fe67fSCaroline Tice CommandArgumentEntry arg; 605405fe67fSCaroline Tice CommandArgumentData alias_arg; 606405fe67fSCaroline Tice 607405fe67fSCaroline Tice // Define the first (and only) variant of this arg. 608405fe67fSCaroline Tice alias_arg.arg_type = eArgTypeAliasName; 609405fe67fSCaroline Tice alias_arg.arg_repetition = eArgRepeatPlain; 610405fe67fSCaroline Tice 611b9c1b51eSKate Stone // There is only one variant this argument could be; put it into the 612b9c1b51eSKate Stone // argument entry. 613405fe67fSCaroline Tice arg.push_back(alias_arg); 614405fe67fSCaroline Tice 615405fe67fSCaroline Tice // Push the data for the first argument into the m_arguments vector. 616405fe67fSCaroline Tice m_arguments.push_back(arg); 617ebc09c36SJim Ingham } 618ebc09c36SJim Ingham 6196e3d8e7fSEugene Zelenko ~CommandObjectCommandsUnalias() override = default; 620ebc09c36SJim Ingham 621*31fd64acSGongyu Deng void 622*31fd64acSGongyu Deng HandleArgumentCompletion(CompletionRequest &request, 623*31fd64acSGongyu Deng OptionElementVector &opt_element_vector) override { 624*31fd64acSGongyu Deng if (!m_interpreter.HasCommands() || request.GetCursorIndex() != 0) 625*31fd64acSGongyu Deng return; 626*31fd64acSGongyu Deng 627*31fd64acSGongyu Deng for (const auto &ent : m_interpreter.GetAliases()) { 628*31fd64acSGongyu Deng request.TryCompleteCurrentArg(ent.first, ent.second->GetHelp()); 629*31fd64acSGongyu Deng } 630*31fd64acSGongyu Deng } 631*31fd64acSGongyu Deng 6325a988416SJim Ingham protected: 633b9c1b51eSKate Stone bool DoExecute(Args &args, CommandReturnObject &result) override { 634ebc09c36SJim Ingham CommandObject::CommandMap::iterator pos; 635ebc09c36SJim Ingham CommandObject *cmd_obj; 636ebc09c36SJim Ingham 63711eb9c64SZachary Turner if (args.empty()) { 63811eb9c64SZachary Turner result.AppendError("must call 'unalias' with a valid alias"); 63911eb9c64SZachary Turner result.SetStatus(eReturnStatusFailed); 64011eb9c64SZachary Turner return false; 64111eb9c64SZachary Turner } 64211eb9c64SZachary Turner 6430d9a201eSRaphael Isemann auto command_name = args[0].ref(); 644a7015092SGreg Clayton cmd_obj = m_interpreter.GetCommandObject(command_name); 6454574a890SZachary Turner if (!cmd_obj) { 6464574a890SZachary Turner result.AppendErrorWithFormat( 6474574a890SZachary Turner "'%s' is not a known command.\nTry 'help' to see a " 6484574a890SZachary Turner "current list of commands.\n", 649867e7d17SZachary Turner args[0].c_str()); 6504574a890SZachary Turner result.SetStatus(eReturnStatusFailed); 6514574a890SZachary Turner return false; 6524574a890SZachary Turner } 6534574a890SZachary Turner 654b9c1b51eSKate Stone if (m_interpreter.CommandExists(command_name)) { 655b9c1b51eSKate Stone if (cmd_obj->IsRemovable()) { 656b9c1b51eSKate Stone result.AppendErrorWithFormat( 657b9c1b51eSKate Stone "'%s' is not an alias, it is a debugger command which can be " 658b9c1b51eSKate Stone "removed using the 'command delete' command.\n", 659867e7d17SZachary Turner args[0].c_str()); 660b9c1b51eSKate Stone } else { 661b9c1b51eSKate Stone result.AppendErrorWithFormat( 662b9c1b51eSKate Stone "'%s' is a permanent debugger command and cannot be removed.\n", 663867e7d17SZachary Turner args[0].c_str()); 664b547278cSGreg Clayton } 665ebc09c36SJim Ingham result.SetStatus(eReturnStatusFailed); 6664574a890SZachary Turner return false; 6674574a890SZachary Turner } 6684574a890SZachary Turner 669b9c1b51eSKate Stone if (!m_interpreter.RemoveAlias(command_name)) { 670a7015092SGreg Clayton if (m_interpreter.AliasExists(command_name)) 671b9c1b51eSKate Stone result.AppendErrorWithFormat( 672867e7d17SZachary Turner "Error occurred while attempting to unalias '%s'.\n", 673867e7d17SZachary Turner args[0].c_str()); 674ebc09c36SJim Ingham else 675b9c1b51eSKate Stone result.AppendErrorWithFormat("'%s' is not an existing alias.\n", 676867e7d17SZachary Turner args[0].c_str()); 677ebc09c36SJim Ingham result.SetStatus(eReturnStatusFailed); 6784574a890SZachary Turner return false; 679ebc09c36SJim Ingham } 680ebc09c36SJim Ingham 6814574a890SZachary Turner result.SetStatus(eReturnStatusSuccessFinishNoResult); 682ebc09c36SJim Ingham return result.Succeeded(); 683ebc09c36SJim Ingham } 684ebc09c36SJim Ingham }; 685ebc09c36SJim Ingham 686b547278cSGreg Clayton #pragma mark CommandObjectCommandsDelete 687b547278cSGreg Clayton // CommandObjectCommandsDelete 688b547278cSGreg Clayton 689b9c1b51eSKate Stone class CommandObjectCommandsDelete : public CommandObjectParsed { 690b547278cSGreg Clayton public: 6917428a18cSKate Stone CommandObjectCommandsDelete(CommandInterpreter &interpreter) 692b9c1b51eSKate Stone : CommandObjectParsed( 693b9c1b51eSKate Stone interpreter, "command delete", 694b9c1b51eSKate Stone "Delete one or more custom commands defined by 'command regex'.", 695b9c1b51eSKate Stone nullptr) { 696b547278cSGreg Clayton CommandArgumentEntry arg; 697b547278cSGreg Clayton CommandArgumentData alias_arg; 698b547278cSGreg Clayton 699b547278cSGreg Clayton // Define the first (and only) variant of this arg. 700b547278cSGreg Clayton alias_arg.arg_type = eArgTypeCommandName; 701b547278cSGreg Clayton alias_arg.arg_repetition = eArgRepeatPlain; 702b547278cSGreg Clayton 703b9c1b51eSKate Stone // There is only one variant this argument could be; put it into the 704b9c1b51eSKate Stone // argument entry. 705b547278cSGreg Clayton arg.push_back(alias_arg); 706b547278cSGreg Clayton 707b547278cSGreg Clayton // Push the data for the first argument into the m_arguments vector. 708b547278cSGreg Clayton m_arguments.push_back(arg); 709b547278cSGreg Clayton } 710b547278cSGreg Clayton 7116e3d8e7fSEugene Zelenko ~CommandObjectCommandsDelete() override = default; 712b547278cSGreg Clayton 713*31fd64acSGongyu Deng void 714*31fd64acSGongyu Deng HandleArgumentCompletion(CompletionRequest &request, 715*31fd64acSGongyu Deng OptionElementVector &opt_element_vector) override { 716*31fd64acSGongyu Deng if (!m_interpreter.HasCommands() || request.GetCursorIndex() != 0) 717*31fd64acSGongyu Deng return; 718*31fd64acSGongyu Deng 719*31fd64acSGongyu Deng for (const auto &ent : m_interpreter.GetCommands()) { 720*31fd64acSGongyu Deng if (ent.second->IsRemovable()) 721*31fd64acSGongyu Deng request.TryCompleteCurrentArg(ent.first, ent.second->GetHelp()); 722*31fd64acSGongyu Deng } 723*31fd64acSGongyu Deng } 724*31fd64acSGongyu Deng 725b547278cSGreg Clayton protected: 726b9c1b51eSKate Stone bool DoExecute(Args &args, CommandReturnObject &result) override { 727b547278cSGreg Clayton CommandObject::CommandMap::iterator pos; 728b547278cSGreg Clayton 72911eb9c64SZachary Turner if (args.empty()) { 73011eb9c64SZachary Turner result.AppendErrorWithFormat("must call '%s' with one or more valid user " 73111eb9c64SZachary Turner "defined regular expression command names", 732a449698cSZachary Turner GetCommandName().str().c_str()); 73311eb9c64SZachary Turner result.SetStatus(eReturnStatusFailed); 734d77ea5b2SRaphael Isemann return false; 73511eb9c64SZachary Turner } 73611eb9c64SZachary Turner 7370d9a201eSRaphael Isemann auto command_name = args[0].ref(); 7384574a890SZachary Turner if (!m_interpreter.CommandExists(command_name)) { 73946d4aa21SEnrico Granata StreamString error_msg_stream; 740d5b44036SJonas Devlieghere const bool generate_upropos = true; 74146d4aa21SEnrico Granata const bool generate_type_lookup = false; 742b9c1b51eSKate Stone CommandObjectHelp::GenerateAdditionalHelpAvenuesMessage( 7434574a890SZachary Turner &error_msg_stream, command_name, llvm::StringRef(), llvm::StringRef(), 744d5b44036SJonas Devlieghere generate_upropos, generate_type_lookup); 745c156427dSZachary Turner result.AppendError(error_msg_stream.GetString()); 746b547278cSGreg Clayton result.SetStatus(eReturnStatusFailed); 7474574a890SZachary Turner return false; 748b547278cSGreg Clayton } 749b547278cSGreg Clayton 7504574a890SZachary Turner if (!m_interpreter.RemoveCommand(command_name)) { 7514574a890SZachary Turner result.AppendErrorWithFormat( 7524574a890SZachary Turner "'%s' is a permanent debugger command and cannot be removed.\n", 753867e7d17SZachary Turner args[0].c_str()); 7544574a890SZachary Turner result.SetStatus(eReturnStatusFailed); 7554574a890SZachary Turner return false; 7564574a890SZachary Turner } 7574574a890SZachary Turner 7584574a890SZachary Turner result.SetStatus(eReturnStatusSuccessFinishNoResult); 7594574a890SZachary Turner return true; 760b547278cSGreg Clayton } 761b547278cSGreg Clayton }; 762b547278cSGreg Clayton 763de164aaaSGreg Clayton // CommandObjectCommandsAddRegex 7641f0f5b5bSZachary Turner 76564becc11SRaphael Isemann #define LLDB_OPTIONS_regex 76664becc11SRaphael Isemann #include "CommandOptions.inc" 7671f0f5b5bSZachary Turner 7685a988416SJim Ingham #pragma mark CommandObjectCommandsAddRegex 769de164aaaSGreg Clayton 770b9c1b51eSKate Stone class CommandObjectCommandsAddRegex : public CommandObjectParsed, 771b9c1b51eSKate Stone public IOHandlerDelegateMultiline { 772de164aaaSGreg Clayton public: 7737428a18cSKate Stone CommandObjectCommandsAddRegex(CommandInterpreter &interpreter) 774b9c1b51eSKate Stone : CommandObjectParsed( 775a925974bSAdrian Prantl interpreter, "command regex", 776a925974bSAdrian Prantl "Define a custom command in terms of " 777b9c1b51eSKate Stone "existing commands by matching " 778b9c1b51eSKate Stone "regular expressions.", 7790e5e5a79SGreg Clayton "command regex <cmd-name> [s/<regex>/<subst>/ ...]"), 780b9c1b51eSKate Stone IOHandlerDelegateMultiline("", 781b9c1b51eSKate Stone IOHandlerDelegate::Completion::LLDBCommand), 782b9c1b51eSKate Stone m_options() { 783b9c1b51eSKate Stone SetHelpLong( 784b9c1b51eSKate Stone R"( 785b9c1b51eSKate Stone )" 786b9c1b51eSKate Stone "This command allows the user to create powerful regular expression commands \ 787ea671fbdSKate Stone with substitutions. The regular expressions and substitutions are specified \ 788b9c1b51eSKate Stone using the regular expression substitution format of:" 789b9c1b51eSKate Stone R"( 790ea671fbdSKate Stone 791ea671fbdSKate Stone s/<regex>/<subst>/ 792ea671fbdSKate Stone 793b9c1b51eSKate Stone )" 794b9c1b51eSKate Stone "<regex> is a regular expression that can use parenthesis to capture regular \ 795ea671fbdSKate Stone expression input and substitute the captured matches in the output using %1 \ 796b9c1b51eSKate Stone for the first match, %2 for the second, and so on." 797b9c1b51eSKate Stone R"( 798ea671fbdSKate Stone 799b9c1b51eSKate Stone )" 800b9c1b51eSKate Stone "The regular expressions can all be specified on the command line if more than \ 801ea671fbdSKate Stone one argument is provided. If just the command name is provided on the command \ 802ea671fbdSKate Stone line, then the regular expressions and substitutions can be entered on separate \ 803b9c1b51eSKate Stone lines, followed by an empty line to terminate the command definition." 804b9c1b51eSKate Stone R"( 805ea671fbdSKate Stone 806ea671fbdSKate Stone EXAMPLES 807ea671fbdSKate Stone 808b9c1b51eSKate Stone )" 809b9c1b51eSKate Stone "The following example will define a regular expression command named 'f' that \ 810ea671fbdSKate Stone will call 'finish' if there are no arguments, or 'frame select <frame-idx>' if \ 811b9c1b51eSKate Stone a number follows 'f':" 812b9c1b51eSKate Stone R"( 813ea671fbdSKate Stone 814b9c1b51eSKate Stone (lldb) command regex f s/^$/finish/ 's/([0-9]+)/frame select %1/')"); 815de164aaaSGreg Clayton } 816de164aaaSGreg Clayton 8176e3d8e7fSEugene Zelenko ~CommandObjectCommandsAddRegex() override = default; 818de164aaaSGreg Clayton 8195a988416SJim Ingham protected: 8200affb582SDave Lee void IOHandlerActivated(IOHandler &io_handler, bool interactive) override { 8217ca15ba7SLawrence D'Anna StreamFileSP output_sp(io_handler.GetOutputStreamFileSP()); 8220affb582SDave Lee if (output_sp && interactive) { 8230affb582SDave Lee output_sp->PutCString("Enter one or more sed substitution commands in " 824b9c1b51eSKate Stone "the form: 's/<regex>/<subst>/'.\nTerminate the " 825b9c1b51eSKate Stone "substitution list with an empty line.\n"); 82644d93782SGreg Clayton output_sp->Flush(); 82744d93782SGreg Clayton } 82844d93782SGreg Clayton } 82944d93782SGreg Clayton 830b9c1b51eSKate Stone void IOHandlerInputComplete(IOHandler &io_handler, 831b9c1b51eSKate Stone std::string &data) override { 83244d93782SGreg Clayton io_handler.SetIsDone(true); 833d5b44036SJonas Devlieghere if (m_regex_cmd_up) { 83444d93782SGreg Clayton StringList lines; 835b9c1b51eSKate Stone if (lines.SplitIntoLines(data)) { 83644d93782SGreg Clayton bool check_only = false; 8374c78b788SRaphael Isemann for (const std::string &line : lines) { 8384c78b788SRaphael Isemann Status error = AppendRegexSubstitution(line, check_only); 839b9c1b51eSKate Stone if (error.Fail()) { 84057179860SJonas Devlieghere if (!GetDebugger().GetCommandInterpreter().GetBatchCommandMode()) { 84157179860SJonas Devlieghere StreamSP out_stream = GetDebugger().GetAsyncOutputStream(); 84244d93782SGreg Clayton out_stream->Printf("error: %s\n", error.AsCString()); 84344d93782SGreg Clayton } 84444d93782SGreg Clayton } 84544d93782SGreg Clayton } 84644d93782SGreg Clayton } 847d5b44036SJonas Devlieghere if (m_regex_cmd_up->HasRegexEntries()) { 848d5b44036SJonas Devlieghere CommandObjectSP cmd_sp(m_regex_cmd_up.release()); 84944d93782SGreg Clayton m_interpreter.AddCommand(cmd_sp->GetCommandName(), cmd_sp, true); 85044d93782SGreg Clayton } 85144d93782SGreg Clayton } 85244d93782SGreg Clayton } 85344d93782SGreg Clayton 854b9c1b51eSKate Stone bool DoExecute(Args &command, CommandReturnObject &result) override { 8555a988416SJim Ingham const size_t argc = command.GetArgumentCount(); 856b9c1b51eSKate Stone if (argc == 0) { 857b9c1b51eSKate Stone result.AppendError("usage: 'command regex <command-name> " 858b9c1b51eSKate Stone "[s/<regex1>/<subst1>/ s/<regex2>/<subst2>/ ...]'\n"); 8590e5e5a79SGreg Clayton result.SetStatus(eReturnStatusFailed); 86011eb9c64SZachary Turner return false; 86111eb9c64SZachary Turner } 86211eb9c64SZachary Turner 86397206d57SZachary Turner Status error; 8640d9a201eSRaphael Isemann auto name = command[0].ref(); 865a8f3ae7cSJonas Devlieghere m_regex_cmd_up = std::make_unique<CommandObjectRegexCommand>( 8664574a890SZachary Turner m_interpreter, name, m_options.GetHelp(), m_options.GetSyntax(), 10, 0, 8674574a890SZachary Turner true); 8680e5e5a79SGreg Clayton 869b9c1b51eSKate Stone if (argc == 1) { 87057179860SJonas Devlieghere Debugger &debugger = GetDebugger(); 871e30f11d9SKate Stone bool color_prompt = debugger.GetUseColor(); 87244d93782SGreg Clayton const bool multiple_lines = true; // Get multiple lines 873b9c1b51eSKate Stone IOHandlerSP io_handler_sp(new IOHandlerEditline( 874b9c1b51eSKate Stone debugger, IOHandler::Type::Other, 87573d80faaSGreg Clayton "lldb-regex", // Name of input reader for history 876514d8cd8SZachary Turner llvm::StringRef("> "), // Prompt 877514d8cd8SZachary Turner llvm::StringRef(), // Continuation prompt 878b9c1b51eSKate Stone multiple_lines, color_prompt, 879f6913cd7SGreg Clayton 0, // Don't show line numbers 880d77c2e09SJonas Devlieghere *this, nullptr)); 88144d93782SGreg Clayton 882b9c1b51eSKate Stone if (io_handler_sp) { 8837ce2de2cSJonas Devlieghere debugger.RunIOHandlerAsync(io_handler_sp); 884de164aaaSGreg Clayton result.SetStatus(eReturnStatusSuccessFinishNoResult); 885de164aaaSGreg Clayton } 886b9c1b51eSKate Stone } else { 88797d2c401SZachary Turner for (auto &entry : command.entries().drop_front()) { 88844d93782SGreg Clayton bool check_only = false; 8890d9a201eSRaphael Isemann error = AppendRegexSubstitution(entry.ref(), check_only); 8900e5e5a79SGreg Clayton if (error.Fail()) 8910e5e5a79SGreg Clayton break; 8920e5e5a79SGreg Clayton } 8930e5e5a79SGreg Clayton 894b9c1b51eSKate Stone if (error.Success()) { 8950e5e5a79SGreg Clayton AddRegexCommandToInterpreter(); 8960e5e5a79SGreg Clayton } 8970e5e5a79SGreg Clayton } 898b9c1b51eSKate Stone if (error.Fail()) { 8990e5e5a79SGreg Clayton result.AppendError(error.AsCString()); 900de164aaaSGreg Clayton result.SetStatus(eReturnStatusFailed); 901de164aaaSGreg Clayton } 9020e5e5a79SGreg Clayton 903de164aaaSGreg Clayton return result.Succeeded(); 904de164aaaSGreg Clayton } 905de164aaaSGreg Clayton 90697206d57SZachary Turner Status AppendRegexSubstitution(const llvm::StringRef ®ex_sed, 907b9c1b51eSKate Stone bool check_only) { 90897206d57SZachary Turner Status error; 9090e5e5a79SGreg Clayton 910d5b44036SJonas Devlieghere if (!m_regex_cmd_up) { 911b9c1b51eSKate Stone error.SetErrorStringWithFormat( 912b9c1b51eSKate Stone "invalid regular expression command object for: '%.*s'", 913b9c1b51eSKate Stone (int)regex_sed.size(), regex_sed.data()); 9140e5e5a79SGreg Clayton return error; 915de164aaaSGreg Clayton } 9160e5e5a79SGreg Clayton 9170e5e5a79SGreg Clayton size_t regex_sed_size = regex_sed.size(); 9180e5e5a79SGreg Clayton 919b9c1b51eSKate Stone if (regex_sed_size <= 1) { 920b9c1b51eSKate Stone error.SetErrorStringWithFormat( 921b9c1b51eSKate Stone "regular expression substitution string is too short: '%.*s'", 922b9c1b51eSKate Stone (int)regex_sed.size(), regex_sed.data()); 9230e5e5a79SGreg Clayton return error; 9240e5e5a79SGreg Clayton } 9250e5e5a79SGreg Clayton 926b9c1b51eSKate Stone if (regex_sed[0] != 's') { 927b9c1b51eSKate Stone error.SetErrorStringWithFormat("regular expression substitution string " 928b9c1b51eSKate Stone "doesn't start with 's': '%.*s'", 929b9c1b51eSKate Stone (int)regex_sed.size(), regex_sed.data()); 9300e5e5a79SGreg Clayton return error; 9310e5e5a79SGreg Clayton } 9320e5e5a79SGreg Clayton const size_t first_separator_char_pos = 1; 93305097246SAdrian Prantl // use the char that follows 's' as the regex separator character so we can 93405097246SAdrian Prantl // have "s/<regex>/<subst>/" or "s|<regex>|<subst>|" 9350e5e5a79SGreg Clayton const char separator_char = regex_sed[first_separator_char_pos]; 936b9c1b51eSKate Stone const size_t second_separator_char_pos = 937b9c1b51eSKate Stone regex_sed.find(separator_char, first_separator_char_pos + 1); 9380e5e5a79SGreg Clayton 939b9c1b51eSKate Stone if (second_separator_char_pos == std::string::npos) { 940b9c1b51eSKate Stone error.SetErrorStringWithFormat( 941b9c1b51eSKate Stone "missing second '%c' separator char after '%.*s' in '%.*s'", 9420e5e5a79SGreg Clayton separator_char, 9430e5e5a79SGreg Clayton (int)(regex_sed.size() - first_separator_char_pos - 1), 944ea508635SGreg Clayton regex_sed.data() + (first_separator_char_pos + 1), 945b9c1b51eSKate Stone (int)regex_sed.size(), regex_sed.data()); 9460e5e5a79SGreg Clayton return error; 9470e5e5a79SGreg Clayton } 9480e5e5a79SGreg Clayton 949b9c1b51eSKate Stone const size_t third_separator_char_pos = 950b9c1b51eSKate Stone regex_sed.find(separator_char, second_separator_char_pos + 1); 9510e5e5a79SGreg Clayton 952b9c1b51eSKate Stone if (third_separator_char_pos == std::string::npos) { 953b9c1b51eSKate Stone error.SetErrorStringWithFormat( 954b9c1b51eSKate Stone "missing third '%c' separator char after '%.*s' in '%.*s'", 9550e5e5a79SGreg Clayton separator_char, 9560e5e5a79SGreg Clayton (int)(regex_sed.size() - second_separator_char_pos - 1), 957ea508635SGreg Clayton regex_sed.data() + (second_separator_char_pos + 1), 958b9c1b51eSKate Stone (int)regex_sed.size(), regex_sed.data()); 9590e5e5a79SGreg Clayton return error; 9600e5e5a79SGreg Clayton } 9610e5e5a79SGreg Clayton 962b9c1b51eSKate Stone if (third_separator_char_pos != regex_sed_size - 1) { 96305097246SAdrian Prantl // Make sure that everything that follows the last regex separator char 964b9c1b51eSKate Stone if (regex_sed.find_first_not_of("\t\n\v\f\r ", 965b9c1b51eSKate Stone third_separator_char_pos + 1) != 966b9c1b51eSKate Stone std::string::npos) { 967b9c1b51eSKate Stone error.SetErrorStringWithFormat( 968b9c1b51eSKate Stone "extra data found after the '%.*s' regular expression substitution " 969b9c1b51eSKate Stone "string: '%.*s'", 970b9c1b51eSKate Stone (int)third_separator_char_pos + 1, regex_sed.data(), 9710e5e5a79SGreg Clayton (int)(regex_sed.size() - third_separator_char_pos - 1), 9720e5e5a79SGreg Clayton regex_sed.data() + (third_separator_char_pos + 1)); 9730e5e5a79SGreg Clayton return error; 9740e5e5a79SGreg Clayton } 975b9c1b51eSKate Stone } else if (first_separator_char_pos + 1 == second_separator_char_pos) { 976b9c1b51eSKate Stone error.SetErrorStringWithFormat( 977b9c1b51eSKate Stone "<regex> can't be empty in 's%c<regex>%c<subst>%c' string: '%.*s'", 978b9c1b51eSKate Stone separator_char, separator_char, separator_char, (int)regex_sed.size(), 9790e5e5a79SGreg Clayton regex_sed.data()); 9800e5e5a79SGreg Clayton return error; 981b9c1b51eSKate Stone } else if (second_separator_char_pos + 1 == third_separator_char_pos) { 982b9c1b51eSKate Stone error.SetErrorStringWithFormat( 983b9c1b51eSKate Stone "<subst> can't be empty in 's%c<regex>%c<subst>%c' string: '%.*s'", 984b9c1b51eSKate Stone separator_char, separator_char, separator_char, (int)regex_sed.size(), 9850e5e5a79SGreg Clayton regex_sed.data()); 9860e5e5a79SGreg Clayton return error; 9870e5e5a79SGreg Clayton } 98844d93782SGreg Clayton 989b9c1b51eSKate Stone if (!check_only) { 990adcd0268SBenjamin Kramer std::string regex(std::string(regex_sed.substr( 991adcd0268SBenjamin Kramer first_separator_char_pos + 1, 992adcd0268SBenjamin Kramer second_separator_char_pos - first_separator_char_pos - 1))); 993adcd0268SBenjamin Kramer std::string subst(std::string(regex_sed.substr( 994adcd0268SBenjamin Kramer second_separator_char_pos + 1, 995adcd0268SBenjamin Kramer third_separator_char_pos - second_separator_char_pos - 1))); 99643224195SRaphael Isemann m_regex_cmd_up->AddRegexCommand(regex, subst); 99744d93782SGreg Clayton } 9980e5e5a79SGreg Clayton return error; 999de164aaaSGreg Clayton } 1000de164aaaSGreg Clayton 1001b9c1b51eSKate Stone void AddRegexCommandToInterpreter() { 1002d5b44036SJonas Devlieghere if (m_regex_cmd_up) { 1003d5b44036SJonas Devlieghere if (m_regex_cmd_up->HasRegexEntries()) { 1004d5b44036SJonas Devlieghere CommandObjectSP cmd_sp(m_regex_cmd_up.release()); 1005de164aaaSGreg Clayton m_interpreter.AddCommand(cmd_sp->GetCommandName(), cmd_sp, true); 1006de164aaaSGreg Clayton } 1007de164aaaSGreg Clayton } 1008de164aaaSGreg Clayton } 1009de164aaaSGreg Clayton 1010de164aaaSGreg Clayton private: 1011d5b44036SJonas Devlieghere std::unique_ptr<CommandObjectRegexCommand> m_regex_cmd_up; 1012de164aaaSGreg Clayton 1013b9c1b51eSKate Stone class CommandOptions : public Options { 1014de164aaaSGreg Clayton public: 1015b9c1b51eSKate Stone CommandOptions() : Options() {} 1016de164aaaSGreg Clayton 10176e3d8e7fSEugene Zelenko ~CommandOptions() override = default; 1018de164aaaSGreg Clayton 101997206d57SZachary Turner Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg, 1020b9c1b51eSKate Stone ExecutionContext *execution_context) override { 102197206d57SZachary Turner Status error; 10223bcdfc0eSGreg Clayton const int short_option = m_getopt_table[option_idx].val; 1023de164aaaSGreg Clayton 1024b9c1b51eSKate Stone switch (short_option) { 1025de164aaaSGreg Clayton case 'h': 1026adcd0268SBenjamin Kramer m_help.assign(std::string(option_arg)); 1027de164aaaSGreg Clayton break; 1028de164aaaSGreg Clayton case 's': 1029adcd0268SBenjamin Kramer m_syntax.assign(std::string(option_arg)); 1030de164aaaSGreg Clayton break; 1031de164aaaSGreg Clayton default: 103236162014SRaphael Isemann llvm_unreachable("Unimplemented option"); 1033de164aaaSGreg Clayton } 1034de164aaaSGreg Clayton 1035de164aaaSGreg Clayton return error; 1036de164aaaSGreg Clayton } 1037de164aaaSGreg Clayton 1038b9c1b51eSKate Stone void OptionParsingStarting(ExecutionContext *execution_context) override { 1039de164aaaSGreg Clayton m_help.clear(); 1040de164aaaSGreg Clayton m_syntax.clear(); 1041de164aaaSGreg Clayton } 1042de164aaaSGreg Clayton 10431f0f5b5bSZachary Turner llvm::ArrayRef<OptionDefinition> GetDefinitions() override { 104470602439SZachary Turner return llvm::makeArrayRef(g_regex_options); 10451f0f5b5bSZachary Turner } 1046de164aaaSGreg Clayton 1047daed98e5SShivam Mittal llvm::StringRef GetHelp() { return m_help; } 10486e3d8e7fSEugene Zelenko 1049daed98e5SShivam Mittal llvm::StringRef GetSyntax() { return m_syntax; } 10506e3d8e7fSEugene Zelenko 1051de164aaaSGreg Clayton protected: 10526e3d8e7fSEugene Zelenko // Instance variables to hold the values for command options. 10536e3d8e7fSEugene Zelenko 1054de164aaaSGreg Clayton std::string m_help; 1055de164aaaSGreg Clayton std::string m_syntax; 1056de164aaaSGreg Clayton }; 1057de164aaaSGreg Clayton 1058b9c1b51eSKate Stone Options *GetOptions() override { return &m_options; } 1059de164aaaSGreg Clayton 10605a988416SJim Ingham CommandOptions m_options; 1061de164aaaSGreg Clayton }; 1062de164aaaSGreg Clayton 1063b9c1b51eSKate Stone class CommandObjectPythonFunction : public CommandObjectRaw { 1064223383edSEnrico Granata public: 1065b9c1b51eSKate Stone CommandObjectPythonFunction(CommandInterpreter &interpreter, std::string name, 1066b9c1b51eSKate Stone std::string funct, std::string help, 1067b9c1b51eSKate Stone ScriptedCommandSynchronicity synch) 1068a925974bSAdrian Prantl : CommandObjectRaw(interpreter, name), m_function_name(funct), 1069a925974bSAdrian Prantl m_synchro(synch), m_fetched_help_long(false) { 1070735152e3SEnrico Granata if (!help.empty()) 1071442f6530SZachary Turner SetHelp(help); 1072b9c1b51eSKate Stone else { 1073735152e3SEnrico Granata StreamString stream; 1074735152e3SEnrico Granata stream.Printf("For more information run 'help %s'", name.c_str()); 1075c156427dSZachary Turner SetHelp(stream.GetString()); 1076735152e3SEnrico Granata } 1077223383edSEnrico Granata } 1078223383edSEnrico Granata 10796e3d8e7fSEugene Zelenko ~CommandObjectPythonFunction() override = default; 1080223383edSEnrico Granata 1081b9c1b51eSKate Stone bool IsRemovable() const override { return true; } 10825a988416SJim Ingham 1083b9c1b51eSKate Stone const std::string &GetFunctionName() { return m_function_name; } 10845a988416SJim Ingham 1085b9c1b51eSKate Stone ScriptedCommandSynchronicity GetSynchronicity() { return m_synchro; } 10865a988416SJim Ingham 1087442f6530SZachary Turner llvm::StringRef GetHelpLong() override { 1088442f6530SZachary Turner if (m_fetched_help_long) 1089442f6530SZachary Turner return CommandObjectRaw::GetHelpLong(); 1090442f6530SZachary Turner 10912b29b432SJonas Devlieghere ScriptInterpreter *scripter = GetDebugger().GetScriptInterpreter(); 1092442f6530SZachary Turner if (!scripter) 1093442f6530SZachary Turner return CommandObjectRaw::GetHelpLong(); 1094442f6530SZachary Turner 1095fac939e9SEnrico Granata std::string docstring; 1096442f6530SZachary Turner m_fetched_help_long = 1097442f6530SZachary Turner scripter->GetDocumentationForItem(m_function_name.c_str(), docstring); 1098fac939e9SEnrico Granata if (!docstring.empty()) 1099442f6530SZachary Turner SetHelpLong(docstring); 1100fac939e9SEnrico Granata return CommandObjectRaw::GetHelpLong(); 1101fac939e9SEnrico Granata } 1102fac939e9SEnrico Granata 11035a988416SJim Ingham protected: 11044d51a902SRaphael Isemann bool DoExecute(llvm::StringRef raw_command_line, 1105b9c1b51eSKate Stone CommandReturnObject &result) override { 11062b29b432SJonas Devlieghere ScriptInterpreter *scripter = GetDebugger().GetScriptInterpreter(); 1107223383edSEnrico Granata 110897206d57SZachary Turner Status error; 1109223383edSEnrico Granata 111070f11f88SJim Ingham result.SetStatus(eReturnStatusInvalid); 111170f11f88SJim Ingham 1112a925974bSAdrian Prantl if (!scripter || !scripter->RunScriptBasedCommand( 1113a925974bSAdrian Prantl m_function_name.c_str(), raw_command_line, m_synchro, 1114a925974bSAdrian Prantl result, error, m_exe_ctx)) { 1115223383edSEnrico Granata result.AppendError(error.AsCString()); 1116223383edSEnrico Granata result.SetStatus(eReturnStatusFailed); 1117b9c1b51eSKate Stone } else { 111870f11f88SJim Ingham // Don't change the status if the command already set it... 1119b9c1b51eSKate Stone if (result.GetStatus() == eReturnStatusInvalid) { 1120c156427dSZachary Turner if (result.GetOutputData().empty()) 1121223383edSEnrico Granata result.SetStatus(eReturnStatusSuccessFinishNoResult); 112270f11f88SJim Ingham else 112370f11f88SJim Ingham result.SetStatus(eReturnStatusSuccessFinishResult); 112470f11f88SJim Ingham } 112570f11f88SJim Ingham } 1126223383edSEnrico Granata 1127223383edSEnrico Granata return result.Succeeded(); 1128223383edSEnrico Granata } 1129223383edSEnrico Granata 11306e3d8e7fSEugene Zelenko private: 11316e3d8e7fSEugene Zelenko std::string m_function_name; 11326e3d8e7fSEugene Zelenko ScriptedCommandSynchronicity m_synchro; 11336e3d8e7fSEugene Zelenko bool m_fetched_help_long; 1134223383edSEnrico Granata }; 1135223383edSEnrico Granata 1136b9c1b51eSKate Stone class CommandObjectScriptingObject : public CommandObjectRaw { 11379fe00e52SEnrico Granata public: 11389fe00e52SEnrico Granata CommandObjectScriptingObject(CommandInterpreter &interpreter, 11399fe00e52SEnrico Granata std::string name, 11400641ca1aSZachary Turner StructuredData::GenericSP cmd_obj_sp, 1141b9c1b51eSKate Stone ScriptedCommandSynchronicity synch) 1142a925974bSAdrian Prantl : CommandObjectRaw(interpreter, name), m_cmd_obj_sp(cmd_obj_sp), 1143a925974bSAdrian Prantl m_synchro(synch), m_fetched_help_short(false), 1144b9c1b51eSKate Stone m_fetched_help_long(false) { 11459fe00e52SEnrico Granata StreamString stream; 11469fe00e52SEnrico Granata stream.Printf("For more information run 'help %s'", name.c_str()); 1147c156427dSZachary Turner SetHelp(stream.GetString()); 11482b29b432SJonas Devlieghere if (ScriptInterpreter *scripter = GetDebugger().GetScriptInterpreter()) 1149e87764f2SEnrico Granata GetFlags().Set(scripter->GetFlagsForCommandObject(cmd_obj_sp)); 11509fe00e52SEnrico Granata } 11519fe00e52SEnrico Granata 11526e3d8e7fSEugene Zelenko ~CommandObjectScriptingObject() override = default; 11539fe00e52SEnrico Granata 1154b9c1b51eSKate Stone bool IsRemovable() const override { return true; } 11559fe00e52SEnrico Granata 1156b9c1b51eSKate Stone ScriptedCommandSynchronicity GetSynchronicity() { return m_synchro; } 11579fe00e52SEnrico Granata 1158442f6530SZachary Turner llvm::StringRef GetHelp() override { 1159442f6530SZachary Turner if (m_fetched_help_short) 1160442f6530SZachary Turner return CommandObjectRaw::GetHelp(); 11612b29b432SJonas Devlieghere ScriptInterpreter *scripter = GetDebugger().GetScriptInterpreter(); 1162442f6530SZachary Turner if (!scripter) 1163442f6530SZachary Turner return CommandObjectRaw::GetHelp(); 11646f79bb2dSEnrico Granata std::string docstring; 1165b9c1b51eSKate Stone m_fetched_help_short = 1166b9c1b51eSKate Stone scripter->GetShortHelpForCommandObject(m_cmd_obj_sp, docstring); 11676f79bb2dSEnrico Granata if (!docstring.empty()) 1168442f6530SZachary Turner SetHelp(docstring); 1169442f6530SZachary Turner 11706f79bb2dSEnrico Granata return CommandObjectRaw::GetHelp(); 11716f79bb2dSEnrico Granata } 11726f79bb2dSEnrico Granata 1173442f6530SZachary Turner llvm::StringRef GetHelpLong() override { 1174442f6530SZachary Turner if (m_fetched_help_long) 1175442f6530SZachary Turner return CommandObjectRaw::GetHelpLong(); 1176442f6530SZachary Turner 11772b29b432SJonas Devlieghere ScriptInterpreter *scripter = GetDebugger().GetScriptInterpreter(); 1178442f6530SZachary Turner if (!scripter) 1179442f6530SZachary Turner return CommandObjectRaw::GetHelpLong(); 1180442f6530SZachary Turner 11816f79bb2dSEnrico Granata std::string docstring; 1182b9c1b51eSKate Stone m_fetched_help_long = 1183b9c1b51eSKate Stone scripter->GetLongHelpForCommandObject(m_cmd_obj_sp, docstring); 11846f79bb2dSEnrico Granata if (!docstring.empty()) 1185442f6530SZachary Turner SetHelpLong(docstring); 11869fe00e52SEnrico Granata return CommandObjectRaw::GetHelpLong(); 11879fe00e52SEnrico Granata } 11889fe00e52SEnrico Granata 11899fe00e52SEnrico Granata protected: 11904d51a902SRaphael Isemann bool DoExecute(llvm::StringRef raw_command_line, 1191b9c1b51eSKate Stone CommandReturnObject &result) override { 11922b29b432SJonas Devlieghere ScriptInterpreter *scripter = GetDebugger().GetScriptInterpreter(); 11939fe00e52SEnrico Granata 119497206d57SZachary Turner Status error; 11959fe00e52SEnrico Granata 11969fe00e52SEnrico Granata result.SetStatus(eReturnStatusInvalid); 11979fe00e52SEnrico Granata 1198b9c1b51eSKate Stone if (!scripter || 1199b9c1b51eSKate Stone !scripter->RunScriptBasedCommand(m_cmd_obj_sp, raw_command_line, 1200b9c1b51eSKate Stone m_synchro, result, error, m_exe_ctx)) { 12019fe00e52SEnrico Granata result.AppendError(error.AsCString()); 12029fe00e52SEnrico Granata result.SetStatus(eReturnStatusFailed); 1203b9c1b51eSKate Stone } else { 12049fe00e52SEnrico Granata // Don't change the status if the command already set it... 1205b9c1b51eSKate Stone if (result.GetStatus() == eReturnStatusInvalid) { 1206c156427dSZachary Turner if (result.GetOutputData().empty()) 12079fe00e52SEnrico Granata result.SetStatus(eReturnStatusSuccessFinishNoResult); 12089fe00e52SEnrico Granata else 12099fe00e52SEnrico Granata result.SetStatus(eReturnStatusSuccessFinishResult); 12109fe00e52SEnrico Granata } 12119fe00e52SEnrico Granata } 12129fe00e52SEnrico Granata 12139fe00e52SEnrico Granata return result.Succeeded(); 12149fe00e52SEnrico Granata } 12159fe00e52SEnrico Granata 12166e3d8e7fSEugene Zelenko private: 12176e3d8e7fSEugene Zelenko StructuredData::GenericSP m_cmd_obj_sp; 12186e3d8e7fSEugene Zelenko ScriptedCommandSynchronicity m_synchro; 12196e3d8e7fSEugene Zelenko bool m_fetched_help_short : 1; 12206e3d8e7fSEugene Zelenko bool m_fetched_help_long : 1; 12219fe00e52SEnrico Granata }; 12229fe00e52SEnrico Granata 1223a9dbf432SEnrico Granata // CommandObjectCommandsScriptImport 122464becc11SRaphael Isemann #define LLDB_OPTIONS_script_import 122564becc11SRaphael Isemann #include "CommandOptions.inc" 12261f0f5b5bSZachary Turner 1227b9c1b51eSKate Stone class CommandObjectCommandsScriptImport : public CommandObjectParsed { 12285a988416SJim Ingham public: 1229b9c1b51eSKate Stone CommandObjectCommandsScriptImport(CommandInterpreter &interpreter) 1230b9c1b51eSKate Stone : CommandObjectParsed(interpreter, "command script import", 1231b9c1b51eSKate Stone "Import a scripting module in LLDB.", nullptr), 1232b9c1b51eSKate Stone m_options() { 12335a988416SJim Ingham CommandArgumentEntry arg1; 12345a988416SJim Ingham CommandArgumentData cmd_arg; 12355a988416SJim Ingham 12365a988416SJim Ingham // Define the first (and only) variant of this arg. 12375a988416SJim Ingham cmd_arg.arg_type = eArgTypeFilename; 12383b00e35bSEnrico Granata cmd_arg.arg_repetition = eArgRepeatPlus; 12395a988416SJim Ingham 1240b9c1b51eSKate Stone // There is only one variant this argument could be; put it into the 1241b9c1b51eSKate Stone // argument entry. 12425a988416SJim Ingham arg1.push_back(cmd_arg); 12435a988416SJim Ingham 12445a988416SJim Ingham // Push the data for the first argument into the m_arguments vector. 12455a988416SJim Ingham m_arguments.push_back(arg1); 12465a988416SJim Ingham } 12475a988416SJim Ingham 12486e3d8e7fSEugene Zelenko ~CommandObjectCommandsScriptImport() override = default; 12495a988416SJim Ingham 1250ae34ed2cSRaphael Isemann void 1251ae34ed2cSRaphael Isemann HandleArgumentCompletion(CompletionRequest &request, 12522443bbd4SRaphael Isemann OptionElementVector &opt_element_vector) override { 1253b9c1b51eSKate Stone CommandCompletions::InvokeCommonCompletionCallbacks( 1254b9c1b51eSKate Stone GetCommandInterpreter(), CommandCompletions::eDiskFileCompletion, 1255a2e76c0bSRaphael Isemann request, nullptr); 12565a988416SJim Ingham } 12575a988416SJim Ingham 1258b9c1b51eSKate Stone Options *GetOptions() override { return &m_options; } 12595a988416SJim Ingham 12605a988416SJim Ingham protected: 1261b9c1b51eSKate Stone class CommandOptions : public Options { 12620a305db7SEnrico Granata public: 1263b9c1b51eSKate Stone CommandOptions() : Options() {} 12640a305db7SEnrico Granata 12656e3d8e7fSEugene Zelenko ~CommandOptions() override = default; 12660a305db7SEnrico Granata 126797206d57SZachary Turner Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg, 1268b9c1b51eSKate Stone ExecutionContext *execution_context) override { 126997206d57SZachary Turner Status error; 12703bcdfc0eSGreg Clayton const int short_option = m_getopt_table[option_idx].val; 12710a305db7SEnrico Granata 1272b9c1b51eSKate Stone switch (short_option) { 12730a305db7SEnrico Granata case 'r': 127415625112SJonas Devlieghere // NO-OP 12750a305db7SEnrico Granata break; 12760a305db7SEnrico Granata default: 127736162014SRaphael Isemann llvm_unreachable("Unimplemented option"); 12780a305db7SEnrico Granata } 12790a305db7SEnrico Granata 12800a305db7SEnrico Granata return error; 12810a305db7SEnrico Granata } 12820a305db7SEnrico Granata 1283b9c1b51eSKate Stone void OptionParsingStarting(ExecutionContext *execution_context) override { 12840a305db7SEnrico Granata } 12850a305db7SEnrico Granata 12861f0f5b5bSZachary Turner llvm::ArrayRef<OptionDefinition> GetDefinitions() override { 128770602439SZachary Turner return llvm::makeArrayRef(g_script_import_options); 12881f0f5b5bSZachary Turner } 12890a305db7SEnrico Granata }; 12900a305db7SEnrico Granata 1291b9c1b51eSKate Stone bool DoExecute(Args &command, CommandReturnObject &result) override { 129211eb9c64SZachary Turner if (command.empty()) { 12933b00e35bSEnrico Granata result.AppendError("command script import needs one or more arguments"); 1294a9dbf432SEnrico Granata result.SetStatus(eReturnStatusFailed); 1295a9dbf432SEnrico Granata return false; 1296a9dbf432SEnrico Granata } 1297a9dbf432SEnrico Granata 129811eb9c64SZachary Turner for (auto &entry : command.entries()) { 129997206d57SZachary Turner Status error; 1300a9dbf432SEnrico Granata 1301c9d645d3SGreg Clayton const bool init_session = true; 1302b9c1b51eSKate Stone // FIXME: this is necessary because CommandObject::CheckRequirements() 130311eb9c64SZachary Turner // assumes that commands won't ever be recursively invoked, but it's 130411eb9c64SZachary Turner // actually possible to craft a Python script that does other "command 130505097246SAdrian Prantl // script imports" in __lldb_init_module the real fix is to have 130605097246SAdrian Prantl // recursive commands possible with a CommandInvocation object separate 130705097246SAdrian Prantl // from the CommandObject itself, so that recursive command invocations 130805097246SAdrian Prantl // won't stomp on each other (wrt to execution contents, options, and 130905097246SAdrian Prantl // more) 1310078551c7SEnrico Granata m_exe_ctx.Clear(); 13112b29b432SJonas Devlieghere if (GetDebugger().GetScriptInterpreter()->LoadScriptingModule( 131215625112SJonas Devlieghere entry.c_str(), init_session, error)) { 1313a9dbf432SEnrico Granata result.SetStatus(eReturnStatusSuccessFinishNoResult); 1314b9c1b51eSKate Stone } else { 1315b9c1b51eSKate Stone result.AppendErrorWithFormat("module importing failed: %s", 1316b9c1b51eSKate Stone error.AsCString()); 1317a9dbf432SEnrico Granata result.SetStatus(eReturnStatusFailed); 1318a9dbf432SEnrico Granata } 13193b00e35bSEnrico Granata } 1320a9dbf432SEnrico Granata 1321a9dbf432SEnrico Granata return result.Succeeded(); 1322a9dbf432SEnrico Granata } 13230a305db7SEnrico Granata 13245a988416SJim Ingham CommandOptions m_options; 1325a9dbf432SEnrico Granata }; 1326223383edSEnrico Granata 1327223383edSEnrico Granata // CommandObjectCommandsScriptAdd 13288fe53c49STatyana Krasnukha static constexpr OptionEnumValueElement g_script_synchro_type[] = { 1329e063ecccSJonas Devlieghere { 1330e063ecccSJonas Devlieghere eScriptedCommandSynchronicitySynchronous, 1331e063ecccSJonas Devlieghere "synchronous", 1332e063ecccSJonas Devlieghere "Run synchronous", 1333e063ecccSJonas Devlieghere }, 1334e063ecccSJonas Devlieghere { 1335e063ecccSJonas Devlieghere eScriptedCommandSynchronicityAsynchronous, 1336e063ecccSJonas Devlieghere "asynchronous", 1337e063ecccSJonas Devlieghere "Run asynchronous", 1338e063ecccSJonas Devlieghere }, 1339e063ecccSJonas Devlieghere { 1340e063ecccSJonas Devlieghere eScriptedCommandSynchronicityCurrentValue, 1341e063ecccSJonas Devlieghere "current", 1342e063ecccSJonas Devlieghere "Do not alter current setting", 1343e063ecccSJonas Devlieghere }, 1344e063ecccSJonas Devlieghere }; 13451f0f5b5bSZachary Turner 13468fe53c49STatyana Krasnukha static constexpr OptionEnumValues ScriptSynchroType() { 13478fe53c49STatyana Krasnukha return OptionEnumValues(g_script_synchro_type); 13488fe53c49STatyana Krasnukha } 13498fe53c49STatyana Krasnukha 135064becc11SRaphael Isemann #define LLDB_OPTIONS_script_add 135164becc11SRaphael Isemann #include "CommandOptions.inc" 13521f0f5b5bSZachary Turner 1353b9c1b51eSKate Stone class CommandObjectCommandsScriptAdd : public CommandObjectParsed, 1354b9c1b51eSKate Stone public IOHandlerDelegateMultiline { 13555a988416SJim Ingham public: 1356b9c1b51eSKate Stone CommandObjectCommandsScriptAdd(CommandInterpreter &interpreter) 1357b9c1b51eSKate Stone : CommandObjectParsed(interpreter, "command script add", 13585a988416SJim Ingham "Add a scripted function as an LLDB command.", 13596e3d8e7fSEugene Zelenko nullptr), 1360b9c1b51eSKate Stone IOHandlerDelegateMultiline("DONE"), m_options() { 13615a988416SJim Ingham CommandArgumentEntry arg1; 13625a988416SJim Ingham CommandArgumentData cmd_arg; 13635a988416SJim Ingham 13645a988416SJim Ingham // Define the first (and only) variant of this arg. 13655a988416SJim Ingham cmd_arg.arg_type = eArgTypeCommandName; 13665a988416SJim Ingham cmd_arg.arg_repetition = eArgRepeatPlain; 13675a988416SJim Ingham 1368b9c1b51eSKate Stone // There is only one variant this argument could be; put it into the 1369b9c1b51eSKate Stone // argument entry. 13705a988416SJim Ingham arg1.push_back(cmd_arg); 13715a988416SJim Ingham 13725a988416SJim Ingham // Push the data for the first argument into the m_arguments vector. 13735a988416SJim Ingham m_arguments.push_back(arg1); 13745a988416SJim Ingham } 13755a988416SJim Ingham 13766e3d8e7fSEugene Zelenko ~CommandObjectCommandsScriptAdd() override = default; 13775a988416SJim Ingham 1378b9c1b51eSKate Stone Options *GetOptions() override { return &m_options; } 13795a988416SJim Ingham 13805a988416SJim Ingham protected: 1381b9c1b51eSKate Stone class CommandOptions : public Options { 1382223383edSEnrico Granata public: 1383b9c1b51eSKate Stone CommandOptions() 1384b9c1b51eSKate Stone : Options(), m_class_name(), m_funct_name(), m_short_help(), 1385b9c1b51eSKate Stone m_synchronicity(eScriptedCommandSynchronicitySynchronous) {} 1386223383edSEnrico Granata 13876e3d8e7fSEugene Zelenko ~CommandOptions() override = default; 1388223383edSEnrico Granata 138997206d57SZachary Turner Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg, 1390b9c1b51eSKate Stone ExecutionContext *execution_context) override { 139197206d57SZachary Turner Status error; 13923bcdfc0eSGreg Clayton const int short_option = m_getopt_table[option_idx].val; 1393223383edSEnrico Granata 1394b9c1b51eSKate Stone switch (short_option) { 1395223383edSEnrico Granata case 'f': 1396fe11483bSZachary Turner if (!option_arg.empty()) 1397adcd0268SBenjamin Kramer m_funct_name = std::string(option_arg); 1398735152e3SEnrico Granata break; 13999fe00e52SEnrico Granata case 'c': 1400fe11483bSZachary Turner if (!option_arg.empty()) 1401adcd0268SBenjamin Kramer m_class_name = std::string(option_arg); 14029fe00e52SEnrico Granata break; 1403735152e3SEnrico Granata case 'h': 1404fe11483bSZachary Turner if (!option_arg.empty()) 1405adcd0268SBenjamin Kramer m_short_help = std::string(option_arg); 1406223383edSEnrico Granata break; 14070a305db7SEnrico Granata case 's': 1408b9c1b51eSKate Stone m_synchronicity = 140947cbf4a0SPavel Labath (ScriptedCommandSynchronicity)OptionArgParser::ToOptionEnum( 1410fe11483bSZachary Turner option_arg, GetDefinitions()[option_idx].enum_values, 0, error); 14110a305db7SEnrico Granata if (!error.Success()) 1412b9c1b51eSKate Stone error.SetErrorStringWithFormat( 1413fe11483bSZachary Turner "unrecognized value for synchronicity '%s'", 1414fe11483bSZachary Turner option_arg.str().c_str()); 14150a305db7SEnrico Granata break; 1416223383edSEnrico Granata default: 141736162014SRaphael Isemann llvm_unreachable("Unimplemented option"); 1418223383edSEnrico Granata } 1419223383edSEnrico Granata 1420223383edSEnrico Granata return error; 1421223383edSEnrico Granata } 1422223383edSEnrico Granata 1423b9c1b51eSKate Stone void OptionParsingStarting(ExecutionContext *execution_context) override { 14249fe00e52SEnrico Granata m_class_name.clear(); 1425735152e3SEnrico Granata m_funct_name.clear(); 1426735152e3SEnrico Granata m_short_help.clear(); 142744d93782SGreg Clayton m_synchronicity = eScriptedCommandSynchronicitySynchronous; 1428223383edSEnrico Granata } 1429223383edSEnrico Granata 14301f0f5b5bSZachary Turner llvm::ArrayRef<OptionDefinition> GetDefinitions() override { 143170602439SZachary Turner return llvm::makeArrayRef(g_script_add_options); 14321f0f5b5bSZachary Turner } 1433223383edSEnrico Granata 1434223383edSEnrico Granata // Instance variables to hold the values for command options. 1435223383edSEnrico Granata 14369fe00e52SEnrico Granata std::string m_class_name; 1437223383edSEnrico Granata std::string m_funct_name; 1438735152e3SEnrico Granata std::string m_short_help; 143944d93782SGreg Clayton ScriptedCommandSynchronicity m_synchronicity; 1440223383edSEnrico Granata }; 1441223383edSEnrico Granata 14420affb582SDave Lee void IOHandlerActivated(IOHandler &io_handler, bool interactive) override { 14437ca15ba7SLawrence D'Anna StreamFileSP output_sp(io_handler.GetOutputStreamFileSP()); 14440affb582SDave Lee if (output_sp && interactive) { 144544d93782SGreg Clayton output_sp->PutCString(g_python_command_instructions); 144644d93782SGreg Clayton output_sp->Flush(); 1447223383edSEnrico Granata } 1448223383edSEnrico Granata } 1449223383edSEnrico Granata 1450b9c1b51eSKate Stone void IOHandlerInputComplete(IOHandler &io_handler, 1451b9c1b51eSKate Stone std::string &data) override { 14527ca15ba7SLawrence D'Anna StreamFileSP error_sp = io_handler.GetErrorStreamFileSP(); 145344d93782SGreg Clayton 14542b29b432SJonas Devlieghere ScriptInterpreter *interpreter = GetDebugger().GetScriptInterpreter(); 1455b9c1b51eSKate Stone if (interpreter) { 145644d93782SGreg Clayton 145744d93782SGreg Clayton StringList lines; 145844d93782SGreg Clayton lines.SplitIntoLines(data); 1459b9c1b51eSKate Stone if (lines.GetSize() > 0) { 1460a73b7df7SEnrico Granata std::string funct_name_str; 1461b9c1b51eSKate Stone if (interpreter->GenerateScriptAliasFunction(lines, funct_name_str)) { 1462b9c1b51eSKate Stone if (funct_name_str.empty()) { 1463b9c1b51eSKate Stone error_sp->Printf("error: unable to obtain a function name, didn't " 1464b9c1b51eSKate Stone "add python command.\n"); 146544d93782SGreg Clayton error_sp->Flush(); 1466b9c1b51eSKate Stone } else { 1467223383edSEnrico Granata // everything should be fine now, let's add this alias 1468223383edSEnrico Granata 1469b9c1b51eSKate Stone CommandObjectSP command_obj_sp(new CommandObjectPythonFunction( 1470771ef6d4SMalcolm Parsons m_interpreter, m_cmd_name, funct_name_str, m_short_help, 147144d93782SGreg Clayton m_synchronicity)); 1472223383edSEnrico Granata 1473b9c1b51eSKate Stone if (!m_interpreter.AddUserCommand(m_cmd_name, command_obj_sp, 1474b9c1b51eSKate Stone true)) { 1475b9c1b51eSKate Stone error_sp->Printf("error: unable to add selected command, didn't " 1476b9c1b51eSKate Stone "add python command.\n"); 147744d93782SGreg Clayton error_sp->Flush(); 1478223383edSEnrico Granata } 1479223383edSEnrico Granata } 1480b9c1b51eSKate Stone } else { 1481b9c1b51eSKate Stone error_sp->Printf( 1482b9c1b51eSKate Stone "error: unable to create function, didn't add python command.\n"); 148344d93782SGreg Clayton error_sp->Flush(); 148444d93782SGreg Clayton } 1485b9c1b51eSKate Stone } else { 148644d93782SGreg Clayton error_sp->Printf("error: empty function, didn't add python command.\n"); 148744d93782SGreg Clayton error_sp->Flush(); 148844d93782SGreg Clayton } 1489b9c1b51eSKate Stone } else { 1490b9c1b51eSKate Stone error_sp->Printf( 1491b9c1b51eSKate Stone "error: script interpreter missing, didn't add python command.\n"); 149244d93782SGreg Clayton error_sp->Flush(); 149344d93782SGreg Clayton } 149444d93782SGreg Clayton 149544d93782SGreg Clayton io_handler.SetIsDone(true); 149644d93782SGreg Clayton } 1497223383edSEnrico Granata 1498b9c1b51eSKate Stone bool DoExecute(Args &command, CommandReturnObject &result) override { 149957179860SJonas Devlieghere if (GetDebugger().GetScriptLanguage() != lldb::eScriptLanguagePython) { 1500b9c1b51eSKate Stone result.AppendError("only scripting language supported for scripted " 1501b9c1b51eSKate Stone "commands is currently Python"); 150299f0b8f9SEnrico Granata result.SetStatus(eReturnStatusFailed); 150399f0b8f9SEnrico Granata return false; 150499f0b8f9SEnrico Granata } 150599f0b8f9SEnrico Granata 150611eb9c64SZachary Turner if (command.GetArgumentCount() != 1) { 1507223383edSEnrico Granata result.AppendError("'command script add' requires one argument"); 1508223383edSEnrico Granata result.SetStatus(eReturnStatusFailed); 1509223383edSEnrico Granata return false; 1510223383edSEnrico Granata } 1511223383edSEnrico Granata 1512735152e3SEnrico Granata // Store the options in case we get multi-line input 1513adcd0268SBenjamin Kramer m_cmd_name = std::string(command[0].ref()); 1514735152e3SEnrico Granata m_short_help.assign(m_options.m_short_help); 151544d93782SGreg Clayton m_synchronicity = m_options.m_synchronicity; 1516223383edSEnrico Granata 1517b9c1b51eSKate Stone if (m_options.m_class_name.empty()) { 1518b9c1b51eSKate Stone if (m_options.m_funct_name.empty()) { 1519b9c1b51eSKate Stone m_interpreter.GetPythonCommandsFromIOHandler( 1520b9c1b51eSKate Stone " ", // Prompt 1521a6faf851SJonas Devlieghere *this); // IOHandlerDelegate 1522b9c1b51eSKate Stone } else { 1523b9c1b51eSKate Stone CommandObjectSP new_cmd(new CommandObjectPythonFunction( 1524b9c1b51eSKate Stone m_interpreter, m_cmd_name, m_options.m_funct_name, 1525b9c1b51eSKate Stone m_options.m_short_help, m_synchronicity)); 1526b9c1b51eSKate Stone if (m_interpreter.AddUserCommand(m_cmd_name, new_cmd, true)) { 1527223383edSEnrico Granata result.SetStatus(eReturnStatusSuccessFinishNoResult); 1528b9c1b51eSKate Stone } else { 1529223383edSEnrico Granata result.AppendError("cannot add command"); 1530223383edSEnrico Granata result.SetStatus(eReturnStatusFailed); 1531223383edSEnrico Granata } 1532223383edSEnrico Granata } 1533b9c1b51eSKate Stone } else { 15342b29b432SJonas Devlieghere ScriptInterpreter *interpreter = GetDebugger().GetScriptInterpreter(); 1535b9c1b51eSKate Stone if (!interpreter) { 15369fe00e52SEnrico Granata result.AppendError("cannot find ScriptInterpreter"); 15379fe00e52SEnrico Granata result.SetStatus(eReturnStatusFailed); 15389fe00e52SEnrico Granata return false; 15399fe00e52SEnrico Granata } 15409fe00e52SEnrico Granata 1541b9c1b51eSKate Stone auto cmd_obj_sp = interpreter->CreateScriptCommandObject( 1542b9c1b51eSKate Stone m_options.m_class_name.c_str()); 1543b9c1b51eSKate Stone if (!cmd_obj_sp) { 15449fe00e52SEnrico Granata result.AppendError("cannot create helper object"); 15459fe00e52SEnrico Granata result.SetStatus(eReturnStatusFailed); 15469fe00e52SEnrico Granata return false; 15479fe00e52SEnrico Granata } 15489fe00e52SEnrico Granata 1549b9c1b51eSKate Stone CommandObjectSP new_cmd(new CommandObjectScriptingObject( 1550b9c1b51eSKate Stone m_interpreter, m_cmd_name, cmd_obj_sp, m_synchronicity)); 1551b9c1b51eSKate Stone if (m_interpreter.AddUserCommand(m_cmd_name, new_cmd, true)) { 15529fe00e52SEnrico Granata result.SetStatus(eReturnStatusSuccessFinishNoResult); 1553b9c1b51eSKate Stone } else { 15549fe00e52SEnrico Granata result.AppendError("cannot add command"); 15559fe00e52SEnrico Granata result.SetStatus(eReturnStatusFailed); 15569fe00e52SEnrico Granata } 15579fe00e52SEnrico Granata } 1558223383edSEnrico Granata 1559223383edSEnrico Granata return result.Succeeded(); 1560223383edSEnrico Granata } 15615a988416SJim Ingham 15625a988416SJim Ingham CommandOptions m_options; 156344d93782SGreg Clayton std::string m_cmd_name; 1564735152e3SEnrico Granata std::string m_short_help; 156544d93782SGreg Clayton ScriptedCommandSynchronicity m_synchronicity; 1566223383edSEnrico Granata }; 1567223383edSEnrico Granata 1568223383edSEnrico Granata // CommandObjectCommandsScriptList 1569223383edSEnrico Granata 1570b9c1b51eSKate Stone class CommandObjectCommandsScriptList : public CommandObjectParsed { 1571223383edSEnrico Granata public: 1572b9c1b51eSKate Stone CommandObjectCommandsScriptList(CommandInterpreter &interpreter) 1573b9c1b51eSKate Stone : CommandObjectParsed(interpreter, "command script list", 1574b9c1b51eSKate Stone "List defined scripted commands.", nullptr) {} 1575223383edSEnrico Granata 15766e3d8e7fSEugene Zelenko ~CommandObjectCommandsScriptList() override = default; 1577223383edSEnrico Granata 1578b9c1b51eSKate Stone bool DoExecute(Args &command, CommandReturnObject &result) override { 1579d77ea5b2SRaphael Isemann if (command.GetArgumentCount() != 0) { 1580d77ea5b2SRaphael Isemann result.AppendError("'command script list' doesn't take any arguments"); 1581d77ea5b2SRaphael Isemann result.SetStatus(eReturnStatusFailed); 1582d77ea5b2SRaphael Isemann return false; 1583d77ea5b2SRaphael Isemann } 1584d77ea5b2SRaphael Isemann 1585b9c1b51eSKate Stone m_interpreter.GetHelp(result, CommandInterpreter::eCommandTypesUserDef); 1586223383edSEnrico Granata 1587223383edSEnrico Granata result.SetStatus(eReturnStatusSuccessFinishResult); 1588223383edSEnrico Granata 1589223383edSEnrico Granata return true; 1590223383edSEnrico Granata } 1591223383edSEnrico Granata }; 1592223383edSEnrico Granata 1593223383edSEnrico Granata // CommandObjectCommandsScriptClear 1594223383edSEnrico Granata 1595b9c1b51eSKate Stone class CommandObjectCommandsScriptClear : public CommandObjectParsed { 1596223383edSEnrico Granata public: 1597b9c1b51eSKate Stone CommandObjectCommandsScriptClear(CommandInterpreter &interpreter) 1598b9c1b51eSKate Stone : CommandObjectParsed(interpreter, "command script clear", 1599b9c1b51eSKate Stone "Delete all scripted commands.", nullptr) {} 1600223383edSEnrico Granata 16016e3d8e7fSEugene Zelenko ~CommandObjectCommandsScriptClear() override = default; 1602223383edSEnrico Granata 16035a988416SJim Ingham protected: 1604b9c1b51eSKate Stone bool DoExecute(Args &command, CommandReturnObject &result) override { 1605d77ea5b2SRaphael Isemann if (command.GetArgumentCount() != 0) { 1606d77ea5b2SRaphael Isemann result.AppendError("'command script clear' doesn't take any arguments"); 1607d77ea5b2SRaphael Isemann result.SetStatus(eReturnStatusFailed); 1608d77ea5b2SRaphael Isemann return false; 1609d77ea5b2SRaphael Isemann } 1610d77ea5b2SRaphael Isemann 1611223383edSEnrico Granata m_interpreter.RemoveAllUser(); 1612223383edSEnrico Granata 1613223383edSEnrico Granata result.SetStatus(eReturnStatusSuccessFinishResult); 1614223383edSEnrico Granata 1615223383edSEnrico Granata return true; 1616223383edSEnrico Granata } 1617223383edSEnrico Granata }; 1618223383edSEnrico Granata 1619223383edSEnrico Granata // CommandObjectCommandsScriptDelete 1620223383edSEnrico Granata 1621b9c1b51eSKate Stone class CommandObjectCommandsScriptDelete : public CommandObjectParsed { 1622223383edSEnrico Granata public: 1623b9c1b51eSKate Stone CommandObjectCommandsScriptDelete(CommandInterpreter &interpreter) 1624b9c1b51eSKate Stone : CommandObjectParsed(interpreter, "command script delete", 1625b9c1b51eSKate Stone "Delete a scripted command.", nullptr) { 1626223383edSEnrico Granata CommandArgumentEntry arg1; 1627223383edSEnrico Granata CommandArgumentData cmd_arg; 1628223383edSEnrico Granata 1629223383edSEnrico Granata // Define the first (and only) variant of this arg. 1630223383edSEnrico Granata cmd_arg.arg_type = eArgTypeCommandName; 1631223383edSEnrico Granata cmd_arg.arg_repetition = eArgRepeatPlain; 1632223383edSEnrico Granata 1633b9c1b51eSKate Stone // There is only one variant this argument could be; put it into the 1634b9c1b51eSKate Stone // argument entry. 1635223383edSEnrico Granata arg1.push_back(cmd_arg); 1636223383edSEnrico Granata 1637223383edSEnrico Granata // Push the data for the first argument into the m_arguments vector. 1638223383edSEnrico Granata m_arguments.push_back(arg1); 1639223383edSEnrico Granata } 1640223383edSEnrico Granata 16416e3d8e7fSEugene Zelenko ~CommandObjectCommandsScriptDelete() override = default; 1642223383edSEnrico Granata 16432e8f304fSGongyu Deng void 16442e8f304fSGongyu Deng HandleArgumentCompletion(CompletionRequest &request, 16452e8f304fSGongyu Deng OptionElementVector &opt_element_vector) override { 16462e8f304fSGongyu Deng if (!m_interpreter.HasCommands() || request.GetCursorIndex() != 0) 16472e8f304fSGongyu Deng return; 16482e8f304fSGongyu Deng 16492ebe30c6SRaphael Isemann for (const auto &c : m_interpreter.GetUserCommands()) 16502ebe30c6SRaphael Isemann request.TryCompleteCurrentArg(c.first, c.second->GetHelp()); 16512e8f304fSGongyu Deng } 16522e8f304fSGongyu Deng 16535a988416SJim Ingham protected: 1654b9c1b51eSKate Stone bool DoExecute(Args &command, CommandReturnObject &result) override { 1655223383edSEnrico Granata 165611eb9c64SZachary Turner if (command.GetArgumentCount() != 1) { 1657223383edSEnrico Granata result.AppendError("'command script delete' requires one argument"); 1658223383edSEnrico Granata result.SetStatus(eReturnStatusFailed); 1659223383edSEnrico Granata return false; 1660223383edSEnrico Granata } 1661223383edSEnrico Granata 16620d9a201eSRaphael Isemann auto cmd_name = command[0].ref(); 1663223383edSEnrico Granata 16644574a890SZachary Turner if (cmd_name.empty() || !m_interpreter.HasUserCommands() || 16654574a890SZachary Turner !m_interpreter.UserCommandExists(cmd_name)) { 1666867e7d17SZachary Turner result.AppendErrorWithFormat("command %s not found", command[0].c_str()); 1667223383edSEnrico Granata result.SetStatus(eReturnStatusFailed); 16684574a890SZachary Turner return false; 1669223383edSEnrico Granata } 1670223383edSEnrico Granata 16714574a890SZachary Turner m_interpreter.RemoveUser(cmd_name); 16724574a890SZachary Turner result.SetStatus(eReturnStatusSuccessFinishResult); 16734574a890SZachary Turner return true; 1674223383edSEnrico Granata } 1675223383edSEnrico Granata }; 1676223383edSEnrico Granata 1677223383edSEnrico Granata #pragma mark CommandObjectMultiwordCommandsScript 1678223383edSEnrico Granata 1679223383edSEnrico Granata // CommandObjectMultiwordCommandsScript 1680223383edSEnrico Granata 1681b9c1b51eSKate Stone class CommandObjectMultiwordCommandsScript : public CommandObjectMultiword { 1682223383edSEnrico Granata public: 16837428a18cSKate Stone CommandObjectMultiwordCommandsScript(CommandInterpreter &interpreter) 1684b9c1b51eSKate Stone : CommandObjectMultiword( 1685a925974bSAdrian Prantl interpreter, "command script", 1686a925974bSAdrian Prantl "Commands for managing custom " 1687b9c1b51eSKate Stone "commands implemented by " 1688b9c1b51eSKate Stone "interpreter scripts.", 1689b9c1b51eSKate Stone "command script <subcommand> [<subcommand-options>]") { 1690b9c1b51eSKate Stone LoadSubCommand("add", CommandObjectSP( 1691b9c1b51eSKate Stone new CommandObjectCommandsScriptAdd(interpreter))); 1692b9c1b51eSKate Stone LoadSubCommand( 1693b9c1b51eSKate Stone "delete", 1694b9c1b51eSKate Stone CommandObjectSP(new CommandObjectCommandsScriptDelete(interpreter))); 1695b9c1b51eSKate Stone LoadSubCommand( 1696b9c1b51eSKate Stone "clear", 1697b9c1b51eSKate Stone CommandObjectSP(new CommandObjectCommandsScriptClear(interpreter))); 1698b9c1b51eSKate Stone LoadSubCommand("list", CommandObjectSP(new CommandObjectCommandsScriptList( 1699b9c1b51eSKate Stone interpreter))); 1700b9c1b51eSKate Stone LoadSubCommand( 1701b9c1b51eSKate Stone "import", 1702b9c1b51eSKate Stone CommandObjectSP(new CommandObjectCommandsScriptImport(interpreter))); 1703223383edSEnrico Granata } 1704223383edSEnrico Granata 17056e3d8e7fSEugene Zelenko ~CommandObjectMultiwordCommandsScript() override = default; 1706223383edSEnrico Granata }; 1707223383edSEnrico Granata 1708ebc09c36SJim Ingham #pragma mark CommandObjectMultiwordCommands 1709ebc09c36SJim Ingham 1710ebc09c36SJim Ingham // CommandObjectMultiwordCommands 1711ebc09c36SJim Ingham 1712b9c1b51eSKate Stone CommandObjectMultiwordCommands::CommandObjectMultiwordCommands( 1713b9c1b51eSKate Stone CommandInterpreter &interpreter) 1714b9c1b51eSKate Stone : CommandObjectMultiword(interpreter, "command", 1715b9c1b51eSKate Stone "Commands for managing custom LLDB commands.", 1716b9c1b51eSKate Stone "command <subcommand> [<subcommand-options>]") { 1717b9c1b51eSKate Stone LoadSubCommand("source", 1718b9c1b51eSKate Stone CommandObjectSP(new CommandObjectCommandsSource(interpreter))); 1719b9c1b51eSKate Stone LoadSubCommand("alias", 1720b9c1b51eSKate Stone CommandObjectSP(new CommandObjectCommandsAlias(interpreter))); 1721b9c1b51eSKate Stone LoadSubCommand("unalias", CommandObjectSP( 1722b9c1b51eSKate Stone new CommandObjectCommandsUnalias(interpreter))); 1723b9c1b51eSKate Stone LoadSubCommand("delete", 1724b9c1b51eSKate Stone CommandObjectSP(new CommandObjectCommandsDelete(interpreter))); 1725b9c1b51eSKate Stone LoadSubCommand( 1726b9c1b51eSKate Stone "regex", CommandObjectSP(new CommandObjectCommandsAddRegex(interpreter))); 1727b9c1b51eSKate Stone LoadSubCommand( 1728b9c1b51eSKate Stone "script", 1729b9c1b51eSKate Stone CommandObjectSP(new CommandObjectMultiwordCommandsScript(interpreter))); 1730ebc09c36SJim Ingham } 1731ebc09c36SJim Ingham 17326e3d8e7fSEugene Zelenko CommandObjectMultiwordCommands::~CommandObjectMultiwordCommands() = default; 1733