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 96e3d8e7fSEugene Zelenko #include "CommandObjectCommands.h" 1046d4aa21SEnrico Granata #include "CommandObjectHelp.h" 119390b346SJonas Devlieghere #include "CommandObjectRegexCommand.h" 12ebc09c36SJim Ingham #include "lldb/Core/Debugger.h" 1344d93782SGreg Clayton #include "lldb/Core/IOHandler.h" 147594f14fSEnrico Granata #include "lldb/Interpreter/CommandHistory.h" 15ebc09c36SJim Ingham #include "lldb/Interpreter/CommandInterpreter.h" 16ebc09c36SJim Ingham #include "lldb/Interpreter/CommandReturnObject.h" 1747cbf4a0SPavel Labath #include "lldb/Interpreter/OptionArgParser.h" 18012d4fcaSEnrico Granata #include "lldb/Interpreter/OptionValueBoolean.h" 1945d0e238SEnrico Granata #include "lldb/Interpreter/OptionValueString.h" 207594f14fSEnrico Granata #include "lldb/Interpreter/OptionValueUInt64.h" 21ebc09c36SJim Ingham #include "lldb/Interpreter/Options.h" 2299f0b8f9SEnrico Granata #include "lldb/Interpreter/ScriptInterpreter.h" 23145d95c9SPavel Labath #include "lldb/Utility/Args.h" 24573ab909SZachary Turner #include "lldb/Utility/StringList.h" 259390b346SJonas Devlieghere #include "llvm/ADT/StringRef.h" 26ebc09c36SJim Ingham 27ebc09c36SJim Ingham using namespace lldb; 28ebc09c36SJim Ingham using namespace lldb_private; 29ebc09c36SJim Ingham 30ebc09c36SJim Ingham // CommandObjectCommandsSource 31ebc09c36SJim Ingham 3264becc11SRaphael Isemann #define LLDB_OPTIONS_source 3364becc11SRaphael Isemann #include "CommandOptions.inc" 341f0f5b5bSZachary Turner 35b9c1b51eSKate Stone class CommandObjectCommandsSource : public CommandObjectParsed { 365a988416SJim Ingham public: 377428a18cSKate Stone CommandObjectCommandsSource(CommandInterpreter &interpreter) 38b9c1b51eSKate Stone : CommandObjectParsed( 39b9c1b51eSKate Stone interpreter, "command source", 40b9c1b51eSKate Stone "Read and execute LLDB commands from the file <filename>.", 416e3d8e7fSEugene Zelenko nullptr), 42b9c1b51eSKate Stone m_options() { 435a988416SJim Ingham CommandArgumentEntry arg; 445a988416SJim Ingham CommandArgumentData file_arg; 455a988416SJim Ingham 465a988416SJim Ingham // Define the first (and only) variant of this arg. 475a988416SJim Ingham file_arg.arg_type = eArgTypeFilename; 485a988416SJim Ingham file_arg.arg_repetition = eArgRepeatPlain; 495a988416SJim Ingham 50b9c1b51eSKate Stone // There is only one variant this argument could be; put it into the 51b9c1b51eSKate Stone // argument entry. 525a988416SJim Ingham arg.push_back(file_arg); 535a988416SJim Ingham 545a988416SJim Ingham // Push the data for the first argument into the m_arguments vector. 555a988416SJim Ingham m_arguments.push_back(arg); 565a988416SJim Ingham } 575a988416SJim Ingham 586e3d8e7fSEugene Zelenko ~CommandObjectCommandsSource() override = default; 595a988416SJim Ingham 60b9c1b51eSKate Stone const char *GetRepeatCommand(Args ¤t_command_args, 61b9c1b51eSKate Stone uint32_t index) override { 625a988416SJim Ingham return ""; 635a988416SJim Ingham } 645a988416SJim Ingham 65ae34ed2cSRaphael Isemann void 66ae34ed2cSRaphael Isemann HandleArgumentCompletion(CompletionRequest &request, 672443bbd4SRaphael Isemann OptionElementVector &opt_element_vector) override { 68b9c1b51eSKate Stone CommandCompletions::InvokeCommonCompletionCallbacks( 69b9c1b51eSKate Stone GetCommandInterpreter(), CommandCompletions::eDiskFileCompletion, 70a2e76c0bSRaphael Isemann request, nullptr); 715a988416SJim Ingham } 725a988416SJim Ingham 73b9c1b51eSKate Stone Options *GetOptions() override { return &m_options; } 745a988416SJim Ingham 755a988416SJim Ingham protected: 76b9c1b51eSKate Stone class CommandOptions : public Options { 77e16c50a1SJim Ingham public: 78b9c1b51eSKate Stone CommandOptions() 79b9c1b51eSKate Stone : Options(), m_stop_on_error(true), m_silent_run(false), 80b9c1b51eSKate Stone m_stop_on_continue(true) {} 81e16c50a1SJim Ingham 826e3d8e7fSEugene Zelenko ~CommandOptions() override = default; 83e16c50a1SJim Ingham 8497206d57SZachary Turner Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg, 85b9c1b51eSKate Stone ExecutionContext *execution_context) override { 8697206d57SZachary Turner Status error; 873bcdfc0eSGreg Clayton const int short_option = m_getopt_table[option_idx].val; 88e16c50a1SJim Ingham 89b9c1b51eSKate Stone switch (short_option) { 90e16c50a1SJim Ingham case 'e': 91fe11483bSZachary Turner error = m_stop_on_error.SetValueFromString(option_arg); 92e16c50a1SJim Ingham break; 93340b0309SGreg Clayton 94e16c50a1SJim Ingham case 'c': 95fe11483bSZachary Turner error = m_stop_on_continue.SetValueFromString(option_arg); 96e16c50a1SJim Ingham break; 97340b0309SGreg Clayton 9860986174SMichael Sartain case 's': 99fe11483bSZachary Turner error = m_silent_run.SetValueFromString(option_arg); 10060986174SMichael Sartain break; 101340b0309SGreg Clayton 102e16c50a1SJim Ingham default: 10336162014SRaphael Isemann llvm_unreachable("Unimplemented option"); 104e16c50a1SJim Ingham } 105e16c50a1SJim Ingham 106e16c50a1SJim Ingham return error; 107e16c50a1SJim Ingham } 108e16c50a1SJim Ingham 109b9c1b51eSKate Stone void OptionParsingStarting(ExecutionContext *execution_context) override { 110012d4fcaSEnrico Granata m_stop_on_error.Clear(); 111340b0309SGreg Clayton m_silent_run.Clear(); 112340b0309SGreg Clayton m_stop_on_continue.Clear(); 113e16c50a1SJim Ingham } 114e16c50a1SJim Ingham 1151f0f5b5bSZachary Turner llvm::ArrayRef<OptionDefinition> GetDefinitions() override { 11670602439SZachary Turner return llvm::makeArrayRef(g_source_options); 1171f0f5b5bSZachary Turner } 118e16c50a1SJim Ingham 119e16c50a1SJim Ingham // Instance variables to hold the values for command options. 120e16c50a1SJim Ingham 121012d4fcaSEnrico Granata OptionValueBoolean m_stop_on_error; 122340b0309SGreg Clayton OptionValueBoolean m_silent_run; 123340b0309SGreg Clayton OptionValueBoolean m_stop_on_continue; 124e16c50a1SJim Ingham }; 125e16c50a1SJim Ingham 126b9c1b51eSKate Stone bool DoExecute(Args &command, CommandReturnObject &result) override { 1274574a890SZachary Turner if (command.GetArgumentCount() != 1) { 1284574a890SZachary Turner result.AppendErrorWithFormat( 1294574a890SZachary Turner "'%s' takes exactly one executable filename argument.\n", 1304574a890SZachary Turner GetCommandName().str().c_str()); 1314574a890SZachary Turner result.SetStatus(eReturnStatusFailed); 1324574a890SZachary Turner return false; 1334574a890SZachary Turner } 134ebc09c36SJim Ingham 1350d9a201eSRaphael Isemann FileSpec cmd_file(command[0].ref()); 1368f3be7a3SJonas Devlieghere FileSystem::Instance().Resolve(cmd_file); 1376e3d8e7fSEugene Zelenko ExecutionContext *exe_ctx = nullptr; // Just use the default context. 138ebc09c36SJim Ingham 139340b0309SGreg Clayton // If any options were set, then use them 140340b0309SGreg Clayton if (m_options.m_stop_on_error.OptionWasSet() || 141340b0309SGreg Clayton m_options.m_silent_run.OptionWasSet() || 142b9c1b51eSKate Stone m_options.m_stop_on_continue.OptionWasSet()) { 143340b0309SGreg Clayton // Use user set settings 14426c7bf93SJim Ingham CommandInterpreterRunOptions options; 1451c19b74cSJonas Devlieghere 1461c19b74cSJonas Devlieghere if (m_options.m_stop_on_continue.OptionWasSet()) 1471c19b74cSJonas Devlieghere options.SetStopOnContinue( 1481c19b74cSJonas Devlieghere m_options.m_stop_on_continue.GetCurrentValue()); 1491c19b74cSJonas Devlieghere 1501c19b74cSJonas Devlieghere if (m_options.m_stop_on_error.OptionWasSet()) 15126c7bf93SJim Ingham options.SetStopOnError(m_options.m_stop_on_error.GetCurrentValue()); 152c678ed77SStefan Granitz 153c678ed77SStefan Granitz // Individual silent setting is override for global command echo settings. 154c678ed77SStefan Granitz if (m_options.m_silent_run.GetCurrentValue()) { 155c678ed77SStefan Granitz options.SetSilent(true); 156c678ed77SStefan Granitz } else { 157c678ed77SStefan Granitz options.SetPrintResults(true); 158c0b48ab6SJonas Devlieghere options.SetPrintErrors(true); 159c678ed77SStefan Granitz options.SetEchoCommands(m_interpreter.GetEchoCommands()); 160c678ed77SStefan Granitz options.SetEchoCommentCommands(m_interpreter.GetEchoCommentCommands()); 161c678ed77SStefan Granitz } 16226c7bf93SJim Ingham 1634574a890SZachary Turner m_interpreter.HandleCommandsFromFile(cmd_file, exe_ctx, options, result); 164b9c1b51eSKate Stone } else { 16505097246SAdrian Prantl // No options were set, inherit any settings from nested "command source" 16605097246SAdrian Prantl // commands, or set to sane default settings... 16726c7bf93SJim Ingham CommandInterpreterRunOptions options; 1684574a890SZachary Turner m_interpreter.HandleCommandsFromFile(cmd_file, exe_ctx, options, result); 169ebc09c36SJim Ingham } 170ebc09c36SJim Ingham return result.Succeeded(); 171ebc09c36SJim Ingham } 1726e3d8e7fSEugene Zelenko 1735a988416SJim Ingham CommandOptions m_options; 174ebc09c36SJim Ingham }; 175ebc09c36SJim Ingham 176ebc09c36SJim Ingham #pragma mark CommandObjectCommandsAlias 177ebc09c36SJim Ingham // CommandObjectCommandsAlias 178ebc09c36SJim Ingham 17964becc11SRaphael Isemann #define LLDB_OPTIONS_alias 18064becc11SRaphael Isemann #include "CommandOptions.inc" 1811f0f5b5bSZachary Turner 182b9c1b51eSKate Stone static const char *g_python_command_instructions = 183b9c1b51eSKate Stone "Enter your Python command(s). Type 'DONE' to end.\n" 184be93a35aSEnrico Granata "You must define a Python function with this signature:\n" 18544d93782SGreg Clayton "def my_command_impl(debugger, args, result, internal_dict):\n"; 186be93a35aSEnrico Granata 187b9c1b51eSKate Stone class CommandObjectCommandsAlias : public CommandObjectRaw { 18845d0e238SEnrico Granata protected: 189b9c1b51eSKate Stone class CommandOptions : public OptionGroup { 190ebc09c36SJim Ingham public: 191b9c1b51eSKate Stone CommandOptions() : OptionGroup(), m_help(), m_long_help() {} 19245d0e238SEnrico Granata 19345d0e238SEnrico Granata ~CommandOptions() override = default; 19445d0e238SEnrico Granata 1951f0f5b5bSZachary Turner llvm::ArrayRef<OptionDefinition> GetDefinitions() override { 19670602439SZachary Turner return llvm::makeArrayRef(g_alias_options); 1971f0f5b5bSZachary Turner } 19845d0e238SEnrico Granata 19997206d57SZachary Turner Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_value, 200b9c1b51eSKate Stone ExecutionContext *execution_context) override { 20197206d57SZachary Turner Status error; 20245d0e238SEnrico Granata 2031f0f5b5bSZachary Turner const int short_option = GetDefinitions()[option_idx].short_option; 2048cef4b0bSZachary Turner std::string option_str(option_value); 20545d0e238SEnrico Granata 206b9c1b51eSKate Stone switch (short_option) { 20745d0e238SEnrico Granata case 'h': 2088cef4b0bSZachary Turner m_help.SetCurrentValue(option_str); 20945d0e238SEnrico Granata m_help.SetOptionWasSet(); 21045d0e238SEnrico Granata break; 21145d0e238SEnrico Granata 21245d0e238SEnrico Granata case 'H': 2138cef4b0bSZachary Turner m_long_help.SetCurrentValue(option_str); 21445d0e238SEnrico Granata m_long_help.SetOptionWasSet(); 21545d0e238SEnrico Granata break; 21645d0e238SEnrico Granata 21745d0e238SEnrico Granata default: 21836162014SRaphael Isemann llvm_unreachable("Unimplemented option"); 21945d0e238SEnrico Granata } 22045d0e238SEnrico Granata 22145d0e238SEnrico Granata return error; 22245d0e238SEnrico Granata } 22345d0e238SEnrico Granata 224b9c1b51eSKate Stone void OptionParsingStarting(ExecutionContext *execution_context) override { 22545d0e238SEnrico Granata m_help.Clear(); 22645d0e238SEnrico Granata m_long_help.Clear(); 22745d0e238SEnrico Granata } 22845d0e238SEnrico Granata 22945d0e238SEnrico Granata OptionValueString m_help; 23045d0e238SEnrico Granata OptionValueString m_long_help; 23145d0e238SEnrico Granata }; 23245d0e238SEnrico Granata 23345d0e238SEnrico Granata OptionGroupOptions m_option_group; 23445d0e238SEnrico Granata CommandOptions m_command_options; 23545d0e238SEnrico Granata 23645d0e238SEnrico Granata public: 237b9c1b51eSKate Stone Options *GetOptions() override { return &m_option_group; } 23845d0e238SEnrico Granata 2397428a18cSKate Stone CommandObjectCommandsAlias(CommandInterpreter &interpreter) 240b9c1b51eSKate Stone : CommandObjectRaw( 241b9c1b51eSKate Stone interpreter, "command alias", 242a449698cSZachary Turner "Define a custom command in terms of an existing command."), 243b9c1b51eSKate Stone m_option_group(), m_command_options() { 24445d0e238SEnrico Granata m_option_group.Append(&m_command_options); 24545d0e238SEnrico Granata m_option_group.Finalize(); 24645d0e238SEnrico Granata 247ebc09c36SJim Ingham SetHelpLong( 248ea671fbdSKate Stone "'alias' allows the user to create a short-cut or abbreviation for long \ 249ea671fbdSKate Stone commands, multi-word commands, and commands that take particular options. \ 250b9c1b51eSKate Stone Below are some simple examples of how one might use the 'alias' command:" 251b9c1b51eSKate Stone R"( 252ea671fbdSKate Stone 253ea671fbdSKate Stone (lldb) command alias sc script 254ea671fbdSKate Stone 255ea671fbdSKate Stone Creates the abbreviation 'sc' for the 'script' command. 256ea671fbdSKate Stone 257ea671fbdSKate Stone (lldb) command alias bp breakpoint 258ea671fbdSKate Stone 259b9c1b51eSKate Stone )" 260b9c1b51eSKate Stone " Creates the abbreviation 'bp' for the 'breakpoint' command. Since \ 261ea671fbdSKate Stone breakpoint commands are two-word commands, the user would still need to \ 262b9c1b51eSKate Stone enter the second word after 'bp', e.g. 'bp enable' or 'bp delete'." 263b9c1b51eSKate Stone R"( 264ea671fbdSKate Stone 265ea671fbdSKate Stone (lldb) command alias bpl breakpoint list 266ea671fbdSKate Stone 267ea671fbdSKate Stone Creates the abbreviation 'bpl' for the two-word command 'breakpoint list'. 268ea671fbdSKate Stone 269b9c1b51eSKate Stone )" 270b9c1b51eSKate Stone "An alias can include some options for the command, with the values either \ 271ea671fbdSKate Stone filled in at the time the alias is created, or specified as positional \ 272ea671fbdSKate Stone arguments, to be filled in when the alias is invoked. The following example \ 273b9c1b51eSKate Stone shows how to create aliases with options:" 274b9c1b51eSKate Stone R"( 275ea671fbdSKate Stone 276ea671fbdSKate Stone (lldb) command alias bfl breakpoint set -f %1 -l %2 277ea671fbdSKate Stone 278b9c1b51eSKate Stone )" 279b9c1b51eSKate Stone " Creates the abbreviation 'bfl' (for break-file-line), with the -f and -l \ 280ea671fbdSKate Stone options already part of the alias. So if the user wants to set a breakpoint \ 281ea671fbdSKate Stone by file and line without explicitly having to use the -f and -l options, the \ 282ea671fbdSKate Stone user can now use 'bfl' instead. The '%1' and '%2' are positional placeholders \ 283ea671fbdSKate Stone for the actual arguments that will be passed when the alias command is used. \ 284ea671fbdSKate Stone The number in the placeholder refers to the position/order the actual value \ 285ea671fbdSKate Stone occupies when the alias is used. All the occurrences of '%1' in the alias \ 286ea671fbdSKate Stone will be replaced with the first argument, all the occurrences of '%2' in the \ 287ea671fbdSKate Stone alias will be replaced with the second argument, and so on. This also allows \ 288ea671fbdSKate Stone actual arguments to be used multiple times within an alias (see 'process \ 289b9c1b51eSKate Stone launch' example below)." 290b9c1b51eSKate Stone R"( 291ea671fbdSKate Stone 292b9c1b51eSKate Stone )" 293b9c1b51eSKate Stone "Note: the positional arguments must substitute as whole words in the resultant \ 294ea671fbdSKate Stone command, so you can't at present do something like this to append the file extension \ 295b9c1b51eSKate Stone \".cpp\":" 296b9c1b51eSKate Stone R"( 297ea671fbdSKate Stone 298ea671fbdSKate Stone (lldb) command alias bcppfl breakpoint set -f %1.cpp -l %2 299ea671fbdSKate Stone 300b9c1b51eSKate Stone )" 301b9c1b51eSKate Stone "For more complex aliasing, use the \"command regex\" command instead. In the \ 302ea671fbdSKate Stone 'bfl' case above, the actual file value will be filled in with the first argument \ 303ea671fbdSKate Stone following 'bfl' and the actual line number value will be filled in with the second \ 304b9c1b51eSKate Stone argument. The user would use this alias as follows:" 305b9c1b51eSKate Stone R"( 306ea671fbdSKate Stone 307ea671fbdSKate Stone (lldb) command alias bfl breakpoint set -f %1 -l %2 308ea671fbdSKate Stone (lldb) bfl my-file.c 137 309ea671fbdSKate Stone 310ea671fbdSKate Stone This would be the same as if the user had entered 'breakpoint set -f my-file.c -l 137'. 311ea671fbdSKate Stone 312ea671fbdSKate Stone Another example: 313ea671fbdSKate Stone 314ea671fbdSKate Stone (lldb) command alias pltty process launch -s -o %1 -e %1 315ea671fbdSKate Stone (lldb) pltty /dev/tty0 316ea671fbdSKate Stone 317ea671fbdSKate Stone Interpreted as 'process launch -s -o /dev/tty0 -e /dev/tty0' 318ea671fbdSKate Stone 319b9c1b51eSKate Stone )" 320b9c1b51eSKate Stone "If the user always wanted to pass the same value to a particular option, the \ 321ea671fbdSKate Stone alias could be defined with that value directly in the alias as a constant, \ 322b9c1b51eSKate Stone rather than using a positional placeholder:" 323b9c1b51eSKate Stone R"( 324ea671fbdSKate Stone 325ea671fbdSKate Stone (lldb) command alias bl3 breakpoint set -f %1 -l 3 326ea671fbdSKate Stone 327b9c1b51eSKate Stone Always sets a breakpoint on line 3 of whatever file is indicated.)"); 328ebc09c36SJim Ingham 329405fe67fSCaroline Tice CommandArgumentEntry arg1; 330405fe67fSCaroline Tice CommandArgumentEntry arg2; 331405fe67fSCaroline Tice CommandArgumentEntry arg3; 332405fe67fSCaroline Tice CommandArgumentData alias_arg; 333405fe67fSCaroline Tice CommandArgumentData cmd_arg; 334405fe67fSCaroline Tice CommandArgumentData options_arg; 335405fe67fSCaroline Tice 336405fe67fSCaroline Tice // Define the first (and only) variant of this arg. 337405fe67fSCaroline Tice alias_arg.arg_type = eArgTypeAliasName; 338405fe67fSCaroline Tice alias_arg.arg_repetition = eArgRepeatPlain; 339405fe67fSCaroline Tice 340b9c1b51eSKate Stone // There is only one variant this argument could be; put it into the 341b9c1b51eSKate Stone // argument entry. 342405fe67fSCaroline Tice arg1.push_back(alias_arg); 343405fe67fSCaroline Tice 344405fe67fSCaroline Tice // Define the first (and only) variant of this arg. 345405fe67fSCaroline Tice cmd_arg.arg_type = eArgTypeCommandName; 346405fe67fSCaroline Tice cmd_arg.arg_repetition = eArgRepeatPlain; 347405fe67fSCaroline Tice 348b9c1b51eSKate Stone // There is only one variant this argument could be; put it into the 349b9c1b51eSKate Stone // argument entry. 350405fe67fSCaroline Tice arg2.push_back(cmd_arg); 351405fe67fSCaroline Tice 352405fe67fSCaroline Tice // Define the first (and only) variant of this arg. 353405fe67fSCaroline Tice options_arg.arg_type = eArgTypeAliasOptions; 354405fe67fSCaroline Tice options_arg.arg_repetition = eArgRepeatOptional; 355405fe67fSCaroline Tice 356b9c1b51eSKate Stone // There is only one variant this argument could be; put it into the 357b9c1b51eSKate Stone // argument entry. 358405fe67fSCaroline Tice arg3.push_back(options_arg); 359405fe67fSCaroline Tice 360405fe67fSCaroline Tice // Push the data for the first argument into the m_arguments vector. 361405fe67fSCaroline Tice m_arguments.push_back(arg1); 362405fe67fSCaroline Tice m_arguments.push_back(arg2); 363405fe67fSCaroline Tice m_arguments.push_back(arg3); 364ebc09c36SJim Ingham } 365ebc09c36SJim Ingham 3666e3d8e7fSEugene Zelenko ~CommandObjectCommandsAlias() override = default; 367ebc09c36SJim Ingham 3685a988416SJim Ingham protected: 3694d51a902SRaphael Isemann bool DoExecute(llvm::StringRef raw_command_line, 370b9c1b51eSKate Stone CommandReturnObject &result) override { 3714d51a902SRaphael Isemann if (raw_command_line.empty()) { 372d72e412fSEnrico Granata result.AppendError("'command alias' requires at least two arguments"); 37345d0e238SEnrico Granata return false; 37445d0e238SEnrico Granata } 37545d0e238SEnrico Granata 376e1cfbc79STodd Fiala ExecutionContext exe_ctx = GetCommandInterpreter().GetExecutionContext(); 377e1cfbc79STodd Fiala m_option_group.NotifyOptionParsingStarting(&exe_ctx); 37845d0e238SEnrico Granata 3793a0e1270SRaphael Isemann OptionsWithRaw args_with_suffix(raw_command_line); 38045d0e238SEnrico Granata 3813a0e1270SRaphael Isemann if (args_with_suffix.HasArgs()) 3823a0e1270SRaphael Isemann if (!ParseOptionsAndNotify(args_with_suffix.GetArgs(), result, 3833a0e1270SRaphael Isemann m_option_group, exe_ctx)) 38445d0e238SEnrico Granata return false; 38545d0e238SEnrico Granata 386daed98e5SShivam Mittal llvm::StringRef raw_command_string = args_with_suffix.GetRawPart(); 387a01bccdbSZachary Turner Args args(raw_command_string); 388844d2303SCaroline Tice 38911eb9c64SZachary Turner if (args.GetArgumentCount() < 2) { 390d72e412fSEnrico Granata result.AppendError("'command alias' requires at least two arguments"); 391844d2303SCaroline Tice result.SetStatus(eReturnStatusFailed); 392844d2303SCaroline Tice return false; 393844d2303SCaroline Tice } 394844d2303SCaroline Tice 395844d2303SCaroline Tice // Get the alias command. 396844d2303SCaroline Tice 3970d9a201eSRaphael Isemann auto alias_command = args[0].ref(); 3984574a890SZachary Turner if (alias_command.startswith("-")) { 399d72e412fSEnrico Granata result.AppendError("aliases starting with a dash are not supported"); 400b9c1b51eSKate Stone if (alias_command == "--help" || alias_command == "--long-help") { 401b9c1b51eSKate Stone result.AppendWarning("if trying to pass options to 'command alias' add " 402b9c1b51eSKate Stone "a -- at the end of the options"); 403d72e412fSEnrico Granata } 404d72e412fSEnrico Granata result.SetStatus(eReturnStatusFailed); 405d72e412fSEnrico Granata return false; 406d72e412fSEnrico Granata } 407844d2303SCaroline Tice 408b9c1b51eSKate Stone // Strip the new alias name off 'raw_command_string' (leave it on args, 40905097246SAdrian Prantl // which gets passed to 'Execute', which does the stripping itself. 410844d2303SCaroline Tice size_t pos = raw_command_string.find(alias_command); 411b9c1b51eSKate Stone if (pos == 0) { 412844d2303SCaroline Tice raw_command_string = raw_command_string.substr(alias_command.size()); 413844d2303SCaroline Tice pos = raw_command_string.find_first_not_of(' '); 414844d2303SCaroline Tice if ((pos != std::string::npos) && (pos > 0)) 415844d2303SCaroline Tice raw_command_string = raw_command_string.substr(pos); 416b9c1b51eSKate Stone } else { 417844d2303SCaroline Tice result.AppendError("Error parsing command string. No alias created."); 418844d2303SCaroline Tice result.SetStatus(eReturnStatusFailed); 419844d2303SCaroline Tice return false; 420844d2303SCaroline Tice } 421844d2303SCaroline Tice 422844d2303SCaroline Tice // Verify that the command is alias-able. 423771ef6d4SMalcolm Parsons if (m_interpreter.CommandExists(alias_command)) { 424b9c1b51eSKate Stone result.AppendErrorWithFormat( 425b9c1b51eSKate Stone "'%s' is a permanent debugger command and cannot be redefined.\n", 4264574a890SZachary Turner args[0].c_str()); 427844d2303SCaroline Tice result.SetStatus(eReturnStatusFailed); 428844d2303SCaroline Tice return false; 429844d2303SCaroline Tice } 430844d2303SCaroline Tice 431b9c1b51eSKate Stone // Get CommandObject that is being aliased. The command name is read from 432a01bccdbSZachary Turner // the front of raw_command_string. raw_command_string is returned with the 433a01bccdbSZachary Turner // name of the command object stripped off the front. 434a01bccdbSZachary Turner llvm::StringRef original_raw_command_string = raw_command_string; 435b9c1b51eSKate Stone CommandObject *cmd_obj = 436b9c1b51eSKate Stone m_interpreter.GetCommandObjectForCommand(raw_command_string); 437844d2303SCaroline Tice 438b9c1b51eSKate Stone if (!cmd_obj) { 439b9c1b51eSKate Stone result.AppendErrorWithFormat("invalid command given to 'command alias'. " 440b9c1b51eSKate Stone "'%s' does not begin with a valid command." 441b9c1b51eSKate Stone " No alias created.", 442a01bccdbSZachary Turner original_raw_command_string.str().c_str()); 443844d2303SCaroline Tice result.SetStatus(eReturnStatusFailed); 444844d2303SCaroline Tice return false; 445b9c1b51eSKate Stone } else if (!cmd_obj->WantsRawCommandString()) { 446b9c1b51eSKate Stone // Note that args was initialized with the original command, and has not 44705097246SAdrian Prantl // been updated to this point. Therefore can we pass it to the version of 44805097246SAdrian Prantl // Execute that does not need/expect raw input in the alias. 4495a988416SJim Ingham return HandleAliasingNormalCommand(args, result); 450b9c1b51eSKate Stone } else { 451b9c1b51eSKate Stone return HandleAliasingRawCommand(alias_command, raw_command_string, 452b9c1b51eSKate Stone *cmd_obj, result); 4535a988416SJim Ingham } 4545a988416SJim Ingham return result.Succeeded(); 4555a988416SJim Ingham } 4565a988416SJim Ingham 457a01bccdbSZachary Turner bool HandleAliasingRawCommand(llvm::StringRef alias_command, 458a01bccdbSZachary Turner llvm::StringRef raw_command_string, 459b9c1b51eSKate Stone CommandObject &cmd_obj, 460b9c1b51eSKate Stone CommandReturnObject &result) { 461844d2303SCaroline Tice // Verify & handle any options/arguments passed to the alias command 462844d2303SCaroline Tice 463b9c1b51eSKate Stone OptionArgVectorSP option_arg_vector_sp = 464b9c1b51eSKate Stone OptionArgVectorSP(new OptionArgVector); 465844d2303SCaroline Tice 466b9c1b51eSKate Stone if (CommandObjectSP cmd_obj_sp = 467b9c1b51eSKate Stone m_interpreter.GetCommandSPExact(cmd_obj.GetCommandName(), false)) { 468a01bccdbSZachary Turner if (m_interpreter.AliasExists(alias_command) || 469a01bccdbSZachary Turner m_interpreter.UserCommandExists(alias_command)) { 470b9c1b51eSKate Stone result.AppendWarningWithFormat( 471b9c1b51eSKate Stone "Overwriting existing definition for '%s'.\n", 472a01bccdbSZachary Turner alias_command.str().c_str()); 473844d2303SCaroline Tice } 474b9c1b51eSKate Stone if (CommandAlias *alias = m_interpreter.AddAlias( 475a01bccdbSZachary Turner alias_command, cmd_obj_sp, raw_command_string)) { 47645d0e238SEnrico Granata if (m_command_options.m_help.OptionWasSet()) 47745d0e238SEnrico Granata alias->SetHelp(m_command_options.m_help.GetCurrentValue()); 47845d0e238SEnrico Granata if (m_command_options.m_long_help.OptionWasSet()) 47945d0e238SEnrico Granata alias->SetHelpLong(m_command_options.m_long_help.GetCurrentValue()); 480844d2303SCaroline Tice result.SetStatus(eReturnStatusSuccessFinishNoResult); 481b9c1b51eSKate Stone } else { 482472362e6SCaroline Tice result.AppendError("Unable to create requested alias.\n"); 483472362e6SCaroline Tice result.SetStatus(eReturnStatusFailed); 484472362e6SCaroline Tice } 485212130acSEnrico Granata 486b9c1b51eSKate Stone } else { 487212130acSEnrico Granata result.AppendError("Unable to create requested alias.\n"); 488212130acSEnrico Granata result.SetStatus(eReturnStatusFailed); 489212130acSEnrico Granata } 490212130acSEnrico Granata 491844d2303SCaroline Tice return result.Succeeded(); 492844d2303SCaroline Tice } 493ebc09c36SJim Ingham 494b9c1b51eSKate Stone bool HandleAliasingNormalCommand(Args &args, CommandReturnObject &result) { 495867b185dSCaroline Tice size_t argc = args.GetArgumentCount(); 496ebc09c36SJim Ingham 497b9c1b51eSKate Stone if (argc < 2) { 498d72e412fSEnrico Granata result.AppendError("'command alias' requires at least two arguments"); 499ebc09c36SJim Ingham result.SetStatus(eReturnStatusFailed); 500ebc09c36SJim Ingham return false; 501ebc09c36SJim Ingham } 502ebc09c36SJim Ingham 5034574a890SZachary Turner // Save these in std::strings since we're going to shift them off. 504adcd0268SBenjamin Kramer const std::string alias_command(std::string(args[0].ref())); 505adcd0268SBenjamin Kramer const std::string actual_command(std::string(args[1].ref())); 506ebc09c36SJim Ingham 507ebc09c36SJim Ingham args.Shift(); // Shift the alias command word off the argument vector. 508ebc09c36SJim Ingham args.Shift(); // Shift the old command word off the argument vector. 509ebc09c36SJim Ingham 510b9c1b51eSKate Stone // Verify that the command is alias'able, and get the appropriate command 511b9c1b51eSKate Stone // object. 512ebc09c36SJim Ingham 513771ef6d4SMalcolm Parsons if (m_interpreter.CommandExists(alias_command)) { 514b9c1b51eSKate Stone result.AppendErrorWithFormat( 515b9c1b51eSKate Stone "'%s' is a permanent debugger command and cannot be redefined.\n", 516ebc09c36SJim Ingham alias_command.c_str()); 517ebc09c36SJim Ingham result.SetStatus(eReturnStatusFailed); 5184574a890SZachary Turner return false; 5194574a890SZachary Turner } 5204574a890SZachary Turner 521b9c1b51eSKate Stone CommandObjectSP command_obj_sp( 522a449698cSZachary Turner m_interpreter.GetCommandSPExact(actual_command, true)); 523ebc09c36SJim Ingham CommandObjectSP subcommand_obj_sp; 524ebc09c36SJim Ingham bool use_subcommand = false; 5254574a890SZachary Turner if (!command_obj_sp) { 5264574a890SZachary Turner result.AppendErrorWithFormat("'%s' is not an existing command.\n", 5274574a890SZachary Turner actual_command.c_str()); 5284574a890SZachary Turner result.SetStatus(eReturnStatusFailed); 5294574a890SZachary Turner return false; 5304574a890SZachary Turner } 531ebc09c36SJim Ingham CommandObject *cmd_obj = command_obj_sp.get(); 5326e3d8e7fSEugene Zelenko CommandObject *sub_cmd_obj = nullptr; 533b9c1b51eSKate Stone OptionArgVectorSP option_arg_vector_sp = 534b9c1b51eSKate Stone OptionArgVectorSP(new OptionArgVector); 535ebc09c36SJim Ingham 53611eb9c64SZachary Turner while (cmd_obj->IsMultiwordObject() && !args.empty()) { 5370d9a201eSRaphael Isemann auto sub_command = args[0].ref(); 53811eb9c64SZachary Turner assert(!sub_command.empty()); 5394574a890SZachary Turner subcommand_obj_sp = cmd_obj->GetSubcommandSP(sub_command); 5404574a890SZachary Turner if (!subcommand_obj_sp) { 541b9c1b51eSKate Stone result.AppendErrorWithFormat( 542b9c1b51eSKate Stone "'%s' is not a valid sub-command of '%s'. " 543f415eeb4SCaroline Tice "Unable to create alias.\n", 5444574a890SZachary Turner args[0].c_str(), actual_command.c_str()); 545ebc09c36SJim Ingham result.SetStatus(eReturnStatusFailed); 546ebc09c36SJim Ingham return false; 547ebc09c36SJim Ingham } 5484574a890SZachary Turner 5494574a890SZachary Turner sub_cmd_obj = subcommand_obj_sp.get(); 5504574a890SZachary Turner use_subcommand = true; 5514574a890SZachary Turner args.Shift(); // Shift the sub_command word off the argument vector. 5524574a890SZachary Turner cmd_obj = sub_cmd_obj; 553ebc09c36SJim Ingham } 554ebc09c36SJim Ingham 555ebc09c36SJim Ingham // Verify & handle any options/arguments passed to the alias command 556ebc09c36SJim Ingham 557212130acSEnrico Granata std::string args_string; 558212130acSEnrico Granata 55911eb9c64SZachary Turner if (!args.empty()) { 560b9c1b51eSKate Stone CommandObjectSP tmp_sp = 561b9c1b51eSKate Stone m_interpreter.GetCommandSPExact(cmd_obj->GetCommandName(), false); 562ebc09c36SJim Ingham if (use_subcommand) 5634574a890SZachary Turner tmp_sp = m_interpreter.GetCommandSPExact(sub_cmd_obj->GetCommandName(), 5644574a890SZachary Turner false); 565ca90c47eSCaroline Tice 566ca90c47eSCaroline Tice args.GetCommandString(args_string); 567867b185dSCaroline Tice } 568ebc09c36SJim Ingham 569771ef6d4SMalcolm Parsons if (m_interpreter.AliasExists(alias_command) || 570771ef6d4SMalcolm Parsons m_interpreter.UserCommandExists(alias_command)) { 571b9c1b51eSKate Stone result.AppendWarningWithFormat( 5724574a890SZachary Turner "Overwriting existing definition for '%s'.\n", alias_command.c_str()); 573ebc09c36SJim Ingham } 574ebc09c36SJim Ingham 575b9c1b51eSKate Stone if (CommandAlias *alias = m_interpreter.AddAlias( 5764574a890SZachary Turner alias_command, use_subcommand ? subcommand_obj_sp : command_obj_sp, 577771ef6d4SMalcolm Parsons args_string)) { 57845d0e238SEnrico Granata if (m_command_options.m_help.OptionWasSet()) 57945d0e238SEnrico Granata alias->SetHelp(m_command_options.m_help.GetCurrentValue()); 58045d0e238SEnrico Granata if (m_command_options.m_long_help.OptionWasSet()) 58145d0e238SEnrico Granata alias->SetHelpLong(m_command_options.m_long_help.GetCurrentValue()); 582ebc09c36SJim Ingham result.SetStatus(eReturnStatusSuccessFinishNoResult); 583b9c1b51eSKate Stone } else { 584212130acSEnrico Granata result.AppendError("Unable to create requested alias.\n"); 585212130acSEnrico Granata result.SetStatus(eReturnStatusFailed); 586212130acSEnrico Granata return false; 587212130acSEnrico Granata } 588ebc09c36SJim Ingham 589ebc09c36SJim Ingham return result.Succeeded(); 590ebc09c36SJim Ingham } 591ebc09c36SJim Ingham }; 592ebc09c36SJim Ingham 593ebc09c36SJim Ingham #pragma mark CommandObjectCommandsUnalias 594ebc09c36SJim Ingham // CommandObjectCommandsUnalias 595ebc09c36SJim Ingham 596b9c1b51eSKate Stone class CommandObjectCommandsUnalias : public CommandObjectParsed { 597ebc09c36SJim Ingham public: 5987428a18cSKate Stone CommandObjectCommandsUnalias(CommandInterpreter &interpreter) 599b9c1b51eSKate Stone : CommandObjectParsed( 600b9c1b51eSKate Stone interpreter, "command unalias", 601b9c1b51eSKate Stone "Delete one or more custom commands defined by 'command alias'.", 602b9c1b51eSKate Stone nullptr) { 603405fe67fSCaroline Tice CommandArgumentEntry arg; 604405fe67fSCaroline Tice CommandArgumentData alias_arg; 605405fe67fSCaroline Tice 606405fe67fSCaroline Tice // Define the first (and only) variant of this arg. 607405fe67fSCaroline Tice alias_arg.arg_type = eArgTypeAliasName; 608405fe67fSCaroline Tice alias_arg.arg_repetition = eArgRepeatPlain; 609405fe67fSCaroline Tice 610b9c1b51eSKate Stone // There is only one variant this argument could be; put it into the 611b9c1b51eSKate Stone // argument entry. 612405fe67fSCaroline Tice arg.push_back(alias_arg); 613405fe67fSCaroline Tice 614405fe67fSCaroline Tice // Push the data for the first argument into the m_arguments vector. 615405fe67fSCaroline Tice m_arguments.push_back(arg); 616ebc09c36SJim Ingham } 617ebc09c36SJim Ingham 6186e3d8e7fSEugene Zelenko ~CommandObjectCommandsUnalias() override = default; 619ebc09c36SJim Ingham 62031fd64acSGongyu Deng void 62131fd64acSGongyu Deng HandleArgumentCompletion(CompletionRequest &request, 62231fd64acSGongyu Deng OptionElementVector &opt_element_vector) override { 62331fd64acSGongyu Deng if (!m_interpreter.HasCommands() || request.GetCursorIndex() != 0) 62431fd64acSGongyu Deng return; 62531fd64acSGongyu Deng 62631fd64acSGongyu Deng for (const auto &ent : m_interpreter.GetAliases()) { 62731fd64acSGongyu Deng request.TryCompleteCurrentArg(ent.first, ent.second->GetHelp()); 62831fd64acSGongyu Deng } 62931fd64acSGongyu Deng } 63031fd64acSGongyu Deng 6315a988416SJim Ingham protected: 632b9c1b51eSKate Stone bool DoExecute(Args &args, CommandReturnObject &result) override { 633ebc09c36SJim Ingham CommandObject::CommandMap::iterator pos; 634ebc09c36SJim Ingham CommandObject *cmd_obj; 635ebc09c36SJim Ingham 63611eb9c64SZachary Turner if (args.empty()) { 63711eb9c64SZachary Turner result.AppendError("must call 'unalias' with a valid alias"); 63811eb9c64SZachary Turner result.SetStatus(eReturnStatusFailed); 63911eb9c64SZachary Turner return false; 64011eb9c64SZachary Turner } 64111eb9c64SZachary Turner 6420d9a201eSRaphael Isemann auto command_name = args[0].ref(); 643a7015092SGreg Clayton cmd_obj = m_interpreter.GetCommandObject(command_name); 6444574a890SZachary Turner if (!cmd_obj) { 6454574a890SZachary Turner result.AppendErrorWithFormat( 6464574a890SZachary Turner "'%s' is not a known command.\nTry 'help' to see a " 6474574a890SZachary Turner "current list of commands.\n", 648867e7d17SZachary Turner args[0].c_str()); 6494574a890SZachary Turner result.SetStatus(eReturnStatusFailed); 6504574a890SZachary Turner return false; 6514574a890SZachary Turner } 6524574a890SZachary Turner 653b9c1b51eSKate Stone if (m_interpreter.CommandExists(command_name)) { 654b9c1b51eSKate Stone if (cmd_obj->IsRemovable()) { 655b9c1b51eSKate Stone result.AppendErrorWithFormat( 656b9c1b51eSKate Stone "'%s' is not an alias, it is a debugger command which can be " 657b9c1b51eSKate Stone "removed using the 'command delete' command.\n", 658867e7d17SZachary Turner args[0].c_str()); 659b9c1b51eSKate Stone } else { 660b9c1b51eSKate Stone result.AppendErrorWithFormat( 661b9c1b51eSKate Stone "'%s' is a permanent debugger command and cannot be removed.\n", 662867e7d17SZachary Turner args[0].c_str()); 663b547278cSGreg Clayton } 664ebc09c36SJim Ingham result.SetStatus(eReturnStatusFailed); 6654574a890SZachary Turner return false; 6664574a890SZachary Turner } 6674574a890SZachary Turner 668b9c1b51eSKate Stone if (!m_interpreter.RemoveAlias(command_name)) { 669a7015092SGreg Clayton if (m_interpreter.AliasExists(command_name)) 670b9c1b51eSKate Stone result.AppendErrorWithFormat( 671867e7d17SZachary Turner "Error occurred while attempting to unalias '%s'.\n", 672867e7d17SZachary Turner args[0].c_str()); 673ebc09c36SJim Ingham else 674b9c1b51eSKate Stone result.AppendErrorWithFormat("'%s' is not an existing alias.\n", 675867e7d17SZachary Turner args[0].c_str()); 676ebc09c36SJim Ingham result.SetStatus(eReturnStatusFailed); 6774574a890SZachary Turner return false; 678ebc09c36SJim Ingham } 679ebc09c36SJim Ingham 6804574a890SZachary Turner result.SetStatus(eReturnStatusSuccessFinishNoResult); 681ebc09c36SJim Ingham return result.Succeeded(); 682ebc09c36SJim Ingham } 683ebc09c36SJim Ingham }; 684ebc09c36SJim Ingham 685b547278cSGreg Clayton #pragma mark CommandObjectCommandsDelete 686b547278cSGreg Clayton // CommandObjectCommandsDelete 687b547278cSGreg Clayton 688b9c1b51eSKate Stone class CommandObjectCommandsDelete : public CommandObjectParsed { 689b547278cSGreg Clayton public: 6907428a18cSKate Stone CommandObjectCommandsDelete(CommandInterpreter &interpreter) 691b9c1b51eSKate Stone : CommandObjectParsed( 692b9c1b51eSKate Stone interpreter, "command delete", 693b9c1b51eSKate Stone "Delete one or more custom commands defined by 'command regex'.", 694b9c1b51eSKate Stone nullptr) { 695b547278cSGreg Clayton CommandArgumentEntry arg; 696b547278cSGreg Clayton CommandArgumentData alias_arg; 697b547278cSGreg Clayton 698b547278cSGreg Clayton // Define the first (and only) variant of this arg. 699b547278cSGreg Clayton alias_arg.arg_type = eArgTypeCommandName; 700b547278cSGreg Clayton alias_arg.arg_repetition = eArgRepeatPlain; 701b547278cSGreg Clayton 702b9c1b51eSKate Stone // There is only one variant this argument could be; put it into the 703b9c1b51eSKate Stone // argument entry. 704b547278cSGreg Clayton arg.push_back(alias_arg); 705b547278cSGreg Clayton 706b547278cSGreg Clayton // Push the data for the first argument into the m_arguments vector. 707b547278cSGreg Clayton m_arguments.push_back(arg); 708b547278cSGreg Clayton } 709b547278cSGreg Clayton 7106e3d8e7fSEugene Zelenko ~CommandObjectCommandsDelete() override = default; 711b547278cSGreg Clayton 71231fd64acSGongyu Deng void 71331fd64acSGongyu Deng HandleArgumentCompletion(CompletionRequest &request, 71431fd64acSGongyu Deng OptionElementVector &opt_element_vector) override { 71531fd64acSGongyu Deng if (!m_interpreter.HasCommands() || request.GetCursorIndex() != 0) 71631fd64acSGongyu Deng return; 71731fd64acSGongyu Deng 71831fd64acSGongyu Deng for (const auto &ent : m_interpreter.GetCommands()) { 71931fd64acSGongyu Deng if (ent.second->IsRemovable()) 72031fd64acSGongyu Deng request.TryCompleteCurrentArg(ent.first, ent.second->GetHelp()); 72131fd64acSGongyu Deng } 72231fd64acSGongyu Deng } 72331fd64acSGongyu Deng 724b547278cSGreg Clayton protected: 725b9c1b51eSKate Stone bool DoExecute(Args &args, CommandReturnObject &result) override { 726b547278cSGreg Clayton CommandObject::CommandMap::iterator pos; 727b547278cSGreg Clayton 72811eb9c64SZachary Turner if (args.empty()) { 72911eb9c64SZachary Turner result.AppendErrorWithFormat("must call '%s' with one or more valid user " 73011eb9c64SZachary Turner "defined regular expression command names", 731a449698cSZachary Turner GetCommandName().str().c_str()); 73211eb9c64SZachary Turner result.SetStatus(eReturnStatusFailed); 733d77ea5b2SRaphael Isemann return false; 73411eb9c64SZachary Turner } 73511eb9c64SZachary Turner 7360d9a201eSRaphael Isemann auto command_name = args[0].ref(); 7374574a890SZachary Turner if (!m_interpreter.CommandExists(command_name)) { 73846d4aa21SEnrico Granata StreamString error_msg_stream; 739d5b44036SJonas Devlieghere const bool generate_upropos = true; 74046d4aa21SEnrico Granata const bool generate_type_lookup = false; 741b9c1b51eSKate Stone CommandObjectHelp::GenerateAdditionalHelpAvenuesMessage( 7424574a890SZachary Turner &error_msg_stream, command_name, llvm::StringRef(), llvm::StringRef(), 743d5b44036SJonas Devlieghere generate_upropos, generate_type_lookup); 744c156427dSZachary Turner result.AppendError(error_msg_stream.GetString()); 745b547278cSGreg Clayton result.SetStatus(eReturnStatusFailed); 7464574a890SZachary Turner return false; 747b547278cSGreg Clayton } 748b547278cSGreg Clayton 7494574a890SZachary Turner if (!m_interpreter.RemoveCommand(command_name)) { 7504574a890SZachary Turner result.AppendErrorWithFormat( 7514574a890SZachary Turner "'%s' is a permanent debugger command and cannot be removed.\n", 752867e7d17SZachary Turner args[0].c_str()); 7534574a890SZachary Turner result.SetStatus(eReturnStatusFailed); 7544574a890SZachary Turner return false; 7554574a890SZachary Turner } 7564574a890SZachary Turner 7574574a890SZachary Turner result.SetStatus(eReturnStatusSuccessFinishNoResult); 7584574a890SZachary Turner return true; 759b547278cSGreg Clayton } 760b547278cSGreg Clayton }; 761b547278cSGreg Clayton 762de164aaaSGreg Clayton // CommandObjectCommandsAddRegex 7631f0f5b5bSZachary Turner 76464becc11SRaphael Isemann #define LLDB_OPTIONS_regex 76564becc11SRaphael Isemann #include "CommandOptions.inc" 7661f0f5b5bSZachary Turner 7675a988416SJim Ingham #pragma mark CommandObjectCommandsAddRegex 768de164aaaSGreg Clayton 769b9c1b51eSKate Stone class CommandObjectCommandsAddRegex : public CommandObjectParsed, 770b9c1b51eSKate Stone public IOHandlerDelegateMultiline { 771de164aaaSGreg Clayton public: 7727428a18cSKate Stone CommandObjectCommandsAddRegex(CommandInterpreter &interpreter) 773b9c1b51eSKate Stone : CommandObjectParsed( 774a925974bSAdrian Prantl interpreter, "command regex", 775a925974bSAdrian Prantl "Define a custom command in terms of " 776b9c1b51eSKate Stone "existing commands by matching " 777b9c1b51eSKate Stone "regular expressions.", 7780e5e5a79SGreg Clayton "command regex <cmd-name> [s/<regex>/<subst>/ ...]"), 779b9c1b51eSKate Stone IOHandlerDelegateMultiline("", 780b9c1b51eSKate Stone IOHandlerDelegate::Completion::LLDBCommand), 781b9c1b51eSKate Stone m_options() { 782b9c1b51eSKate Stone SetHelpLong( 783b9c1b51eSKate Stone R"( 784b9c1b51eSKate Stone )" 785b9c1b51eSKate Stone "This command allows the user to create powerful regular expression commands \ 786ea671fbdSKate Stone with substitutions. The regular expressions and substitutions are specified \ 787b9c1b51eSKate Stone using the regular expression substitution format of:" 788b9c1b51eSKate Stone R"( 789ea671fbdSKate Stone 790ea671fbdSKate Stone s/<regex>/<subst>/ 791ea671fbdSKate Stone 792b9c1b51eSKate Stone )" 793b9c1b51eSKate Stone "<regex> is a regular expression that can use parenthesis to capture regular \ 794ea671fbdSKate Stone expression input and substitute the captured matches in the output using %1 \ 795b9c1b51eSKate Stone for the first match, %2 for the second, and so on." 796b9c1b51eSKate Stone R"( 797ea671fbdSKate Stone 798b9c1b51eSKate Stone )" 799b9c1b51eSKate Stone "The regular expressions can all be specified on the command line if more than \ 800ea671fbdSKate Stone one argument is provided. If just the command name is provided on the command \ 801ea671fbdSKate Stone line, then the regular expressions and substitutions can be entered on separate \ 802b9c1b51eSKate Stone lines, followed by an empty line to terminate the command definition." 803b9c1b51eSKate Stone R"( 804ea671fbdSKate Stone 805ea671fbdSKate Stone EXAMPLES 806ea671fbdSKate Stone 807b9c1b51eSKate Stone )" 808b9c1b51eSKate Stone "The following example will define a regular expression command named 'f' that \ 809ea671fbdSKate Stone will call 'finish' if there are no arguments, or 'frame select <frame-idx>' if \ 810b9c1b51eSKate Stone a number follows 'f':" 811b9c1b51eSKate Stone R"( 812ea671fbdSKate Stone 813b9c1b51eSKate Stone (lldb) command regex f s/^$/finish/ 's/([0-9]+)/frame select %1/')"); 814de164aaaSGreg Clayton } 815de164aaaSGreg Clayton 8166e3d8e7fSEugene Zelenko ~CommandObjectCommandsAddRegex() override = default; 817de164aaaSGreg Clayton 8185a988416SJim Ingham protected: 8190affb582SDave Lee void IOHandlerActivated(IOHandler &io_handler, bool interactive) override { 8207ca15ba7SLawrence D'Anna StreamFileSP output_sp(io_handler.GetOutputStreamFileSP()); 8210affb582SDave Lee if (output_sp && interactive) { 8220affb582SDave Lee output_sp->PutCString("Enter one or more sed substitution commands in " 823b9c1b51eSKate Stone "the form: 's/<regex>/<subst>/'.\nTerminate the " 824b9c1b51eSKate Stone "substitution list with an empty line.\n"); 82544d93782SGreg Clayton output_sp->Flush(); 82644d93782SGreg Clayton } 82744d93782SGreg Clayton } 82844d93782SGreg Clayton 829b9c1b51eSKate Stone void IOHandlerInputComplete(IOHandler &io_handler, 830b9c1b51eSKate Stone std::string &data) override { 83144d93782SGreg Clayton io_handler.SetIsDone(true); 832d5b44036SJonas Devlieghere if (m_regex_cmd_up) { 83344d93782SGreg Clayton StringList lines; 834b9c1b51eSKate Stone if (lines.SplitIntoLines(data)) { 83544d93782SGreg Clayton bool check_only = false; 8364c78b788SRaphael Isemann for (const std::string &line : lines) { 8374c78b788SRaphael Isemann Status error = AppendRegexSubstitution(line, check_only); 838b9c1b51eSKate Stone if (error.Fail()) { 83957179860SJonas Devlieghere if (!GetDebugger().GetCommandInterpreter().GetBatchCommandMode()) { 84057179860SJonas Devlieghere StreamSP out_stream = GetDebugger().GetAsyncOutputStream(); 84144d93782SGreg Clayton out_stream->Printf("error: %s\n", error.AsCString()); 84244d93782SGreg Clayton } 84344d93782SGreg Clayton } 84444d93782SGreg Clayton } 84544d93782SGreg Clayton } 846d5b44036SJonas Devlieghere if (m_regex_cmd_up->HasRegexEntries()) { 847d5b44036SJonas Devlieghere CommandObjectSP cmd_sp(m_regex_cmd_up.release()); 84844d93782SGreg Clayton m_interpreter.AddCommand(cmd_sp->GetCommandName(), cmd_sp, true); 84944d93782SGreg Clayton } 85044d93782SGreg Clayton } 85144d93782SGreg Clayton } 85244d93782SGreg Clayton 853b9c1b51eSKate Stone bool DoExecute(Args &command, CommandReturnObject &result) override { 8545a988416SJim Ingham const size_t argc = command.GetArgumentCount(); 855b9c1b51eSKate Stone if (argc == 0) { 856b9c1b51eSKate Stone result.AppendError("usage: 'command regex <command-name> " 857b9c1b51eSKate Stone "[s/<regex1>/<subst1>/ s/<regex2>/<subst2>/ ...]'\n"); 8580e5e5a79SGreg Clayton result.SetStatus(eReturnStatusFailed); 85911eb9c64SZachary Turner return false; 86011eb9c64SZachary Turner } 86111eb9c64SZachary Turner 86297206d57SZachary Turner Status error; 8630d9a201eSRaphael Isemann auto name = command[0].ref(); 864a8f3ae7cSJonas Devlieghere m_regex_cmd_up = std::make_unique<CommandObjectRegexCommand>( 8654574a890SZachary Turner m_interpreter, name, m_options.GetHelp(), m_options.GetSyntax(), 10, 0, 8664574a890SZachary Turner true); 8670e5e5a79SGreg Clayton 868b9c1b51eSKate Stone if (argc == 1) { 86957179860SJonas Devlieghere Debugger &debugger = GetDebugger(); 870e30f11d9SKate Stone bool color_prompt = debugger.GetUseColor(); 87144d93782SGreg Clayton const bool multiple_lines = true; // Get multiple lines 872b9c1b51eSKate Stone IOHandlerSP io_handler_sp(new IOHandlerEditline( 873b9c1b51eSKate Stone debugger, IOHandler::Type::Other, 87473d80faaSGreg Clayton "lldb-regex", // Name of input reader for history 875514d8cd8SZachary Turner llvm::StringRef("> "), // Prompt 876514d8cd8SZachary Turner llvm::StringRef(), // Continuation prompt 877b9c1b51eSKate Stone multiple_lines, color_prompt, 878f6913cd7SGreg Clayton 0, // Don't show line numbers 879d77c2e09SJonas Devlieghere *this, nullptr)); 88044d93782SGreg Clayton 881b9c1b51eSKate Stone if (io_handler_sp) { 8827ce2de2cSJonas Devlieghere debugger.RunIOHandlerAsync(io_handler_sp); 883de164aaaSGreg Clayton result.SetStatus(eReturnStatusSuccessFinishNoResult); 884de164aaaSGreg Clayton } 885b9c1b51eSKate Stone } else { 88697d2c401SZachary Turner for (auto &entry : command.entries().drop_front()) { 88744d93782SGreg Clayton bool check_only = false; 8880d9a201eSRaphael Isemann error = AppendRegexSubstitution(entry.ref(), check_only); 8890e5e5a79SGreg Clayton if (error.Fail()) 8900e5e5a79SGreg Clayton break; 8910e5e5a79SGreg Clayton } 8920e5e5a79SGreg Clayton 893b9c1b51eSKate Stone if (error.Success()) { 8940e5e5a79SGreg Clayton AddRegexCommandToInterpreter(); 8950e5e5a79SGreg Clayton } 8960e5e5a79SGreg Clayton } 897b9c1b51eSKate Stone if (error.Fail()) { 8980e5e5a79SGreg Clayton result.AppendError(error.AsCString()); 899de164aaaSGreg Clayton result.SetStatus(eReturnStatusFailed); 900de164aaaSGreg Clayton } 9010e5e5a79SGreg Clayton 902de164aaaSGreg Clayton return result.Succeeded(); 903de164aaaSGreg Clayton } 904de164aaaSGreg Clayton 90597206d57SZachary Turner Status AppendRegexSubstitution(const llvm::StringRef ®ex_sed, 906b9c1b51eSKate Stone bool check_only) { 90797206d57SZachary Turner Status error; 9080e5e5a79SGreg Clayton 909d5b44036SJonas Devlieghere if (!m_regex_cmd_up) { 910b9c1b51eSKate Stone error.SetErrorStringWithFormat( 911b9c1b51eSKate Stone "invalid regular expression command object for: '%.*s'", 912b9c1b51eSKate Stone (int)regex_sed.size(), regex_sed.data()); 9130e5e5a79SGreg Clayton return error; 914de164aaaSGreg Clayton } 9150e5e5a79SGreg Clayton 9160e5e5a79SGreg Clayton size_t regex_sed_size = regex_sed.size(); 9170e5e5a79SGreg Clayton 918b9c1b51eSKate Stone if (regex_sed_size <= 1) { 919b9c1b51eSKate Stone error.SetErrorStringWithFormat( 920b9c1b51eSKate Stone "regular expression substitution string is too short: '%.*s'", 921b9c1b51eSKate Stone (int)regex_sed.size(), regex_sed.data()); 9220e5e5a79SGreg Clayton return error; 9230e5e5a79SGreg Clayton } 9240e5e5a79SGreg Clayton 925b9c1b51eSKate Stone if (regex_sed[0] != 's') { 926b9c1b51eSKate Stone error.SetErrorStringWithFormat("regular expression substitution string " 927b9c1b51eSKate Stone "doesn't start with 's': '%.*s'", 928b9c1b51eSKate Stone (int)regex_sed.size(), regex_sed.data()); 9290e5e5a79SGreg Clayton return error; 9300e5e5a79SGreg Clayton } 9310e5e5a79SGreg Clayton const size_t first_separator_char_pos = 1; 93205097246SAdrian Prantl // use the char that follows 's' as the regex separator character so we can 93305097246SAdrian Prantl // have "s/<regex>/<subst>/" or "s|<regex>|<subst>|" 9340e5e5a79SGreg Clayton const char separator_char = regex_sed[first_separator_char_pos]; 935b9c1b51eSKate Stone const size_t second_separator_char_pos = 936b9c1b51eSKate Stone regex_sed.find(separator_char, first_separator_char_pos + 1); 9370e5e5a79SGreg Clayton 938b9c1b51eSKate Stone if (second_separator_char_pos == std::string::npos) { 939b9c1b51eSKate Stone error.SetErrorStringWithFormat( 940b9c1b51eSKate Stone "missing second '%c' separator char after '%.*s' in '%.*s'", 9410e5e5a79SGreg Clayton separator_char, 9420e5e5a79SGreg Clayton (int)(regex_sed.size() - first_separator_char_pos - 1), 943ea508635SGreg Clayton regex_sed.data() + (first_separator_char_pos + 1), 944b9c1b51eSKate Stone (int)regex_sed.size(), regex_sed.data()); 9450e5e5a79SGreg Clayton return error; 9460e5e5a79SGreg Clayton } 9470e5e5a79SGreg Clayton 948b9c1b51eSKate Stone const size_t third_separator_char_pos = 949b9c1b51eSKate Stone regex_sed.find(separator_char, second_separator_char_pos + 1); 9500e5e5a79SGreg Clayton 951b9c1b51eSKate Stone if (third_separator_char_pos == std::string::npos) { 952b9c1b51eSKate Stone error.SetErrorStringWithFormat( 953b9c1b51eSKate Stone "missing third '%c' separator char after '%.*s' in '%.*s'", 9540e5e5a79SGreg Clayton separator_char, 9550e5e5a79SGreg Clayton (int)(regex_sed.size() - second_separator_char_pos - 1), 956ea508635SGreg Clayton regex_sed.data() + (second_separator_char_pos + 1), 957b9c1b51eSKate Stone (int)regex_sed.size(), regex_sed.data()); 9580e5e5a79SGreg Clayton return error; 9590e5e5a79SGreg Clayton } 9600e5e5a79SGreg Clayton 961b9c1b51eSKate Stone if (third_separator_char_pos != regex_sed_size - 1) { 96205097246SAdrian Prantl // Make sure that everything that follows the last regex separator char 963b9c1b51eSKate Stone if (regex_sed.find_first_not_of("\t\n\v\f\r ", 964b9c1b51eSKate Stone third_separator_char_pos + 1) != 965b9c1b51eSKate Stone std::string::npos) { 966b9c1b51eSKate Stone error.SetErrorStringWithFormat( 967b9c1b51eSKate Stone "extra data found after the '%.*s' regular expression substitution " 968b9c1b51eSKate Stone "string: '%.*s'", 969b9c1b51eSKate Stone (int)third_separator_char_pos + 1, regex_sed.data(), 9700e5e5a79SGreg Clayton (int)(regex_sed.size() - third_separator_char_pos - 1), 9710e5e5a79SGreg Clayton regex_sed.data() + (third_separator_char_pos + 1)); 9720e5e5a79SGreg Clayton return error; 9730e5e5a79SGreg Clayton } 974b9c1b51eSKate Stone } else if (first_separator_char_pos + 1 == second_separator_char_pos) { 975b9c1b51eSKate Stone error.SetErrorStringWithFormat( 976b9c1b51eSKate Stone "<regex> can't be empty in 's%c<regex>%c<subst>%c' string: '%.*s'", 977b9c1b51eSKate Stone separator_char, separator_char, separator_char, (int)regex_sed.size(), 9780e5e5a79SGreg Clayton regex_sed.data()); 9790e5e5a79SGreg Clayton return error; 980b9c1b51eSKate Stone } else if (second_separator_char_pos + 1 == third_separator_char_pos) { 981b9c1b51eSKate Stone error.SetErrorStringWithFormat( 982b9c1b51eSKate Stone "<subst> can't be empty in 's%c<regex>%c<subst>%c' string: '%.*s'", 983b9c1b51eSKate Stone separator_char, separator_char, separator_char, (int)regex_sed.size(), 9840e5e5a79SGreg Clayton regex_sed.data()); 9850e5e5a79SGreg Clayton return error; 9860e5e5a79SGreg Clayton } 98744d93782SGreg Clayton 988b9c1b51eSKate Stone if (!check_only) { 989adcd0268SBenjamin Kramer std::string regex(std::string(regex_sed.substr( 990adcd0268SBenjamin Kramer first_separator_char_pos + 1, 991adcd0268SBenjamin Kramer second_separator_char_pos - first_separator_char_pos - 1))); 992adcd0268SBenjamin Kramer std::string subst(std::string(regex_sed.substr( 993adcd0268SBenjamin Kramer second_separator_char_pos + 1, 994adcd0268SBenjamin Kramer third_separator_char_pos - second_separator_char_pos - 1))); 99543224195SRaphael Isemann m_regex_cmd_up->AddRegexCommand(regex, subst); 99644d93782SGreg Clayton } 9970e5e5a79SGreg Clayton return error; 998de164aaaSGreg Clayton } 999de164aaaSGreg Clayton 1000b9c1b51eSKate Stone void AddRegexCommandToInterpreter() { 1001d5b44036SJonas Devlieghere if (m_regex_cmd_up) { 1002d5b44036SJonas Devlieghere if (m_regex_cmd_up->HasRegexEntries()) { 1003d5b44036SJonas Devlieghere CommandObjectSP cmd_sp(m_regex_cmd_up.release()); 1004de164aaaSGreg Clayton m_interpreter.AddCommand(cmd_sp->GetCommandName(), cmd_sp, true); 1005de164aaaSGreg Clayton } 1006de164aaaSGreg Clayton } 1007de164aaaSGreg Clayton } 1008de164aaaSGreg Clayton 1009de164aaaSGreg Clayton private: 1010d5b44036SJonas Devlieghere std::unique_ptr<CommandObjectRegexCommand> m_regex_cmd_up; 1011de164aaaSGreg Clayton 1012b9c1b51eSKate Stone class CommandOptions : public Options { 1013de164aaaSGreg Clayton public: 1014b9c1b51eSKate Stone CommandOptions() : Options() {} 1015de164aaaSGreg Clayton 10166e3d8e7fSEugene Zelenko ~CommandOptions() override = default; 1017de164aaaSGreg Clayton 101897206d57SZachary Turner Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg, 1019b9c1b51eSKate Stone ExecutionContext *execution_context) override { 102097206d57SZachary Turner Status error; 10213bcdfc0eSGreg Clayton const int short_option = m_getopt_table[option_idx].val; 1022de164aaaSGreg Clayton 1023b9c1b51eSKate Stone switch (short_option) { 1024de164aaaSGreg Clayton case 'h': 1025adcd0268SBenjamin Kramer m_help.assign(std::string(option_arg)); 1026de164aaaSGreg Clayton break; 1027de164aaaSGreg Clayton case 's': 1028adcd0268SBenjamin Kramer m_syntax.assign(std::string(option_arg)); 1029de164aaaSGreg Clayton break; 1030de164aaaSGreg Clayton default: 103136162014SRaphael Isemann llvm_unreachable("Unimplemented option"); 1032de164aaaSGreg Clayton } 1033de164aaaSGreg Clayton 1034de164aaaSGreg Clayton return error; 1035de164aaaSGreg Clayton } 1036de164aaaSGreg Clayton 1037b9c1b51eSKate Stone void OptionParsingStarting(ExecutionContext *execution_context) override { 1038de164aaaSGreg Clayton m_help.clear(); 1039de164aaaSGreg Clayton m_syntax.clear(); 1040de164aaaSGreg Clayton } 1041de164aaaSGreg Clayton 10421f0f5b5bSZachary Turner llvm::ArrayRef<OptionDefinition> GetDefinitions() override { 104370602439SZachary Turner return llvm::makeArrayRef(g_regex_options); 10441f0f5b5bSZachary Turner } 1045de164aaaSGreg Clayton 1046daed98e5SShivam Mittal llvm::StringRef GetHelp() { return m_help; } 10476e3d8e7fSEugene Zelenko 1048daed98e5SShivam Mittal llvm::StringRef GetSyntax() { return m_syntax; } 10496e3d8e7fSEugene Zelenko 1050de164aaaSGreg Clayton protected: 10516e3d8e7fSEugene Zelenko // Instance variables to hold the values for command options. 10526e3d8e7fSEugene Zelenko 1053de164aaaSGreg Clayton std::string m_help; 1054de164aaaSGreg Clayton std::string m_syntax; 1055de164aaaSGreg Clayton }; 1056de164aaaSGreg Clayton 1057b9c1b51eSKate Stone Options *GetOptions() override { return &m_options; } 1058de164aaaSGreg Clayton 10595a988416SJim Ingham CommandOptions m_options; 1060de164aaaSGreg Clayton }; 1061de164aaaSGreg Clayton 1062b9c1b51eSKate Stone class CommandObjectPythonFunction : public CommandObjectRaw { 1063223383edSEnrico Granata public: 1064b9c1b51eSKate Stone CommandObjectPythonFunction(CommandInterpreter &interpreter, std::string name, 1065b9c1b51eSKate Stone std::string funct, std::string help, 1066b9c1b51eSKate Stone ScriptedCommandSynchronicity synch) 1067a925974bSAdrian Prantl : CommandObjectRaw(interpreter, name), m_function_name(funct), 1068a925974bSAdrian Prantl m_synchro(synch), m_fetched_help_long(false) { 1069735152e3SEnrico Granata if (!help.empty()) 1070442f6530SZachary Turner SetHelp(help); 1071b9c1b51eSKate Stone else { 1072735152e3SEnrico Granata StreamString stream; 1073735152e3SEnrico Granata stream.Printf("For more information run 'help %s'", name.c_str()); 1074c156427dSZachary Turner SetHelp(stream.GetString()); 1075735152e3SEnrico Granata } 1076223383edSEnrico Granata } 1077223383edSEnrico Granata 10786e3d8e7fSEugene Zelenko ~CommandObjectPythonFunction() override = default; 1079223383edSEnrico Granata 1080b9c1b51eSKate Stone bool IsRemovable() const override { return true; } 10815a988416SJim Ingham 1082b9c1b51eSKate Stone const std::string &GetFunctionName() { return m_function_name; } 10835a988416SJim Ingham 1084b9c1b51eSKate Stone ScriptedCommandSynchronicity GetSynchronicity() { return m_synchro; } 10855a988416SJim Ingham 1086442f6530SZachary Turner llvm::StringRef GetHelpLong() override { 1087442f6530SZachary Turner if (m_fetched_help_long) 1088442f6530SZachary Turner return CommandObjectRaw::GetHelpLong(); 1089442f6530SZachary Turner 10902b29b432SJonas Devlieghere ScriptInterpreter *scripter = GetDebugger().GetScriptInterpreter(); 1091442f6530SZachary Turner if (!scripter) 1092442f6530SZachary Turner return CommandObjectRaw::GetHelpLong(); 1093442f6530SZachary Turner 1094fac939e9SEnrico Granata std::string docstring; 1095442f6530SZachary Turner m_fetched_help_long = 1096442f6530SZachary Turner scripter->GetDocumentationForItem(m_function_name.c_str(), docstring); 1097fac939e9SEnrico Granata if (!docstring.empty()) 1098442f6530SZachary Turner SetHelpLong(docstring); 1099fac939e9SEnrico Granata return CommandObjectRaw::GetHelpLong(); 1100fac939e9SEnrico Granata } 1101fac939e9SEnrico Granata 11025a988416SJim Ingham protected: 11034d51a902SRaphael Isemann bool DoExecute(llvm::StringRef raw_command_line, 1104b9c1b51eSKate Stone CommandReturnObject &result) override { 11052b29b432SJonas Devlieghere ScriptInterpreter *scripter = GetDebugger().GetScriptInterpreter(); 1106223383edSEnrico Granata 110797206d57SZachary Turner Status error; 1108223383edSEnrico Granata 110970f11f88SJim Ingham result.SetStatus(eReturnStatusInvalid); 111070f11f88SJim Ingham 1111a925974bSAdrian Prantl if (!scripter || !scripter->RunScriptBasedCommand( 1112a925974bSAdrian Prantl m_function_name.c_str(), raw_command_line, m_synchro, 1113a925974bSAdrian Prantl result, error, m_exe_ctx)) { 1114223383edSEnrico Granata result.AppendError(error.AsCString()); 1115223383edSEnrico Granata result.SetStatus(eReturnStatusFailed); 1116b9c1b51eSKate Stone } else { 111770f11f88SJim Ingham // Don't change the status if the command already set it... 1118b9c1b51eSKate Stone if (result.GetStatus() == eReturnStatusInvalid) { 1119c156427dSZachary Turner if (result.GetOutputData().empty()) 1120223383edSEnrico Granata result.SetStatus(eReturnStatusSuccessFinishNoResult); 112170f11f88SJim Ingham else 112270f11f88SJim Ingham result.SetStatus(eReturnStatusSuccessFinishResult); 112370f11f88SJim Ingham } 112470f11f88SJim Ingham } 1125223383edSEnrico Granata 1126223383edSEnrico Granata return result.Succeeded(); 1127223383edSEnrico Granata } 1128223383edSEnrico Granata 11296e3d8e7fSEugene Zelenko private: 11306e3d8e7fSEugene Zelenko std::string m_function_name; 11316e3d8e7fSEugene Zelenko ScriptedCommandSynchronicity m_synchro; 11326e3d8e7fSEugene Zelenko bool m_fetched_help_long; 1133223383edSEnrico Granata }; 1134223383edSEnrico Granata 1135b9c1b51eSKate Stone class CommandObjectScriptingObject : public CommandObjectRaw { 11369fe00e52SEnrico Granata public: 11379fe00e52SEnrico Granata CommandObjectScriptingObject(CommandInterpreter &interpreter, 11389fe00e52SEnrico Granata std::string name, 11390641ca1aSZachary Turner StructuredData::GenericSP cmd_obj_sp, 1140b9c1b51eSKate Stone ScriptedCommandSynchronicity synch) 1141a925974bSAdrian Prantl : CommandObjectRaw(interpreter, name), m_cmd_obj_sp(cmd_obj_sp), 1142a925974bSAdrian Prantl m_synchro(synch), m_fetched_help_short(false), 1143b9c1b51eSKate Stone m_fetched_help_long(false) { 11449fe00e52SEnrico Granata StreamString stream; 11459fe00e52SEnrico Granata stream.Printf("For more information run 'help %s'", name.c_str()); 1146c156427dSZachary Turner SetHelp(stream.GetString()); 11472b29b432SJonas Devlieghere if (ScriptInterpreter *scripter = GetDebugger().GetScriptInterpreter()) 1148e87764f2SEnrico Granata GetFlags().Set(scripter->GetFlagsForCommandObject(cmd_obj_sp)); 11499fe00e52SEnrico Granata } 11509fe00e52SEnrico Granata 11516e3d8e7fSEugene Zelenko ~CommandObjectScriptingObject() override = default; 11529fe00e52SEnrico Granata 1153b9c1b51eSKate Stone bool IsRemovable() const override { return true; } 11549fe00e52SEnrico Granata 1155b9c1b51eSKate Stone ScriptedCommandSynchronicity GetSynchronicity() { return m_synchro; } 11569fe00e52SEnrico Granata 1157442f6530SZachary Turner llvm::StringRef GetHelp() override { 1158442f6530SZachary Turner if (m_fetched_help_short) 1159442f6530SZachary Turner return CommandObjectRaw::GetHelp(); 11602b29b432SJonas Devlieghere ScriptInterpreter *scripter = GetDebugger().GetScriptInterpreter(); 1161442f6530SZachary Turner if (!scripter) 1162442f6530SZachary Turner return CommandObjectRaw::GetHelp(); 11636f79bb2dSEnrico Granata std::string docstring; 1164b9c1b51eSKate Stone m_fetched_help_short = 1165b9c1b51eSKate Stone scripter->GetShortHelpForCommandObject(m_cmd_obj_sp, docstring); 11666f79bb2dSEnrico Granata if (!docstring.empty()) 1167442f6530SZachary Turner SetHelp(docstring); 1168442f6530SZachary Turner 11696f79bb2dSEnrico Granata return CommandObjectRaw::GetHelp(); 11706f79bb2dSEnrico Granata } 11716f79bb2dSEnrico Granata 1172442f6530SZachary Turner llvm::StringRef GetHelpLong() override { 1173442f6530SZachary Turner if (m_fetched_help_long) 1174442f6530SZachary Turner return CommandObjectRaw::GetHelpLong(); 1175442f6530SZachary Turner 11762b29b432SJonas Devlieghere ScriptInterpreter *scripter = GetDebugger().GetScriptInterpreter(); 1177442f6530SZachary Turner if (!scripter) 1178442f6530SZachary Turner return CommandObjectRaw::GetHelpLong(); 1179442f6530SZachary Turner 11806f79bb2dSEnrico Granata std::string docstring; 1181b9c1b51eSKate Stone m_fetched_help_long = 1182b9c1b51eSKate Stone scripter->GetLongHelpForCommandObject(m_cmd_obj_sp, docstring); 11836f79bb2dSEnrico Granata if (!docstring.empty()) 1184442f6530SZachary Turner SetHelpLong(docstring); 11859fe00e52SEnrico Granata return CommandObjectRaw::GetHelpLong(); 11869fe00e52SEnrico Granata } 11879fe00e52SEnrico Granata 11889fe00e52SEnrico Granata protected: 11894d51a902SRaphael Isemann bool DoExecute(llvm::StringRef raw_command_line, 1190b9c1b51eSKate Stone CommandReturnObject &result) override { 11912b29b432SJonas Devlieghere ScriptInterpreter *scripter = GetDebugger().GetScriptInterpreter(); 11929fe00e52SEnrico Granata 119397206d57SZachary Turner Status error; 11949fe00e52SEnrico Granata 11959fe00e52SEnrico Granata result.SetStatus(eReturnStatusInvalid); 11969fe00e52SEnrico Granata 1197b9c1b51eSKate Stone if (!scripter || 1198b9c1b51eSKate Stone !scripter->RunScriptBasedCommand(m_cmd_obj_sp, raw_command_line, 1199b9c1b51eSKate Stone m_synchro, result, error, m_exe_ctx)) { 12009fe00e52SEnrico Granata result.AppendError(error.AsCString()); 12019fe00e52SEnrico Granata result.SetStatus(eReturnStatusFailed); 1202b9c1b51eSKate Stone } else { 12039fe00e52SEnrico Granata // Don't change the status if the command already set it... 1204b9c1b51eSKate Stone if (result.GetStatus() == eReturnStatusInvalid) { 1205c156427dSZachary Turner if (result.GetOutputData().empty()) 12069fe00e52SEnrico Granata result.SetStatus(eReturnStatusSuccessFinishNoResult); 12079fe00e52SEnrico Granata else 12089fe00e52SEnrico Granata result.SetStatus(eReturnStatusSuccessFinishResult); 12099fe00e52SEnrico Granata } 12109fe00e52SEnrico Granata } 12119fe00e52SEnrico Granata 12129fe00e52SEnrico Granata return result.Succeeded(); 12139fe00e52SEnrico Granata } 12149fe00e52SEnrico Granata 12156e3d8e7fSEugene Zelenko private: 12166e3d8e7fSEugene Zelenko StructuredData::GenericSP m_cmd_obj_sp; 12176e3d8e7fSEugene Zelenko ScriptedCommandSynchronicity m_synchro; 12186e3d8e7fSEugene Zelenko bool m_fetched_help_short : 1; 12196e3d8e7fSEugene Zelenko bool m_fetched_help_long : 1; 12209fe00e52SEnrico Granata }; 12219fe00e52SEnrico Granata 1222a9dbf432SEnrico Granata // CommandObjectCommandsScriptImport 122364becc11SRaphael Isemann #define LLDB_OPTIONS_script_import 122464becc11SRaphael Isemann #include "CommandOptions.inc" 12251f0f5b5bSZachary Turner 1226b9c1b51eSKate Stone class CommandObjectCommandsScriptImport : public CommandObjectParsed { 12275a988416SJim Ingham public: 1228b9c1b51eSKate Stone CommandObjectCommandsScriptImport(CommandInterpreter &interpreter) 1229b9c1b51eSKate Stone : CommandObjectParsed(interpreter, "command script import", 1230b9c1b51eSKate Stone "Import a scripting module in LLDB.", nullptr), 1231b9c1b51eSKate Stone m_options() { 12325a988416SJim Ingham CommandArgumentEntry arg1; 12335a988416SJim Ingham CommandArgumentData cmd_arg; 12345a988416SJim Ingham 12355a988416SJim Ingham // Define the first (and only) variant of this arg. 12365a988416SJim Ingham cmd_arg.arg_type = eArgTypeFilename; 12373b00e35bSEnrico Granata cmd_arg.arg_repetition = eArgRepeatPlus; 12385a988416SJim Ingham 1239b9c1b51eSKate Stone // There is only one variant this argument could be; put it into the 1240b9c1b51eSKate Stone // argument entry. 12415a988416SJim Ingham arg1.push_back(cmd_arg); 12425a988416SJim Ingham 12435a988416SJim Ingham // Push the data for the first argument into the m_arguments vector. 12445a988416SJim Ingham m_arguments.push_back(arg1); 12455a988416SJim Ingham } 12465a988416SJim Ingham 12476e3d8e7fSEugene Zelenko ~CommandObjectCommandsScriptImport() override = default; 12485a988416SJim Ingham 1249ae34ed2cSRaphael Isemann void 1250ae34ed2cSRaphael Isemann HandleArgumentCompletion(CompletionRequest &request, 12512443bbd4SRaphael Isemann OptionElementVector &opt_element_vector) override { 1252b9c1b51eSKate Stone CommandCompletions::InvokeCommonCompletionCallbacks( 1253b9c1b51eSKate Stone GetCommandInterpreter(), CommandCompletions::eDiskFileCompletion, 1254a2e76c0bSRaphael Isemann request, nullptr); 12555a988416SJim Ingham } 12565a988416SJim Ingham 1257b9c1b51eSKate Stone Options *GetOptions() override { return &m_options; } 12585a988416SJim Ingham 12595a988416SJim Ingham protected: 1260b9c1b51eSKate Stone class CommandOptions : public Options { 12610a305db7SEnrico Granata public: 1262b9c1b51eSKate Stone CommandOptions() : Options() {} 12630a305db7SEnrico Granata 12646e3d8e7fSEugene Zelenko ~CommandOptions() override = default; 12650a305db7SEnrico Granata 126697206d57SZachary Turner Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg, 1267b9c1b51eSKate Stone ExecutionContext *execution_context) override { 126897206d57SZachary Turner Status error; 12693bcdfc0eSGreg Clayton const int short_option = m_getopt_table[option_idx].val; 12700a305db7SEnrico Granata 1271b9c1b51eSKate Stone switch (short_option) { 12720a305db7SEnrico Granata case 'r': 127315625112SJonas Devlieghere // NO-OP 12740a305db7SEnrico Granata break; 1275*00bb397bSJonas Devlieghere case 'c': 1276*00bb397bSJonas Devlieghere relative_to_command_file = true; 1277*00bb397bSJonas Devlieghere break; 12780a305db7SEnrico Granata default: 127936162014SRaphael Isemann llvm_unreachable("Unimplemented option"); 12800a305db7SEnrico Granata } 12810a305db7SEnrico Granata 12820a305db7SEnrico Granata return error; 12830a305db7SEnrico Granata } 12840a305db7SEnrico Granata 1285b9c1b51eSKate Stone void OptionParsingStarting(ExecutionContext *execution_context) override { 1286*00bb397bSJonas Devlieghere relative_to_command_file = false; 12870a305db7SEnrico Granata } 12880a305db7SEnrico Granata 12891f0f5b5bSZachary Turner llvm::ArrayRef<OptionDefinition> GetDefinitions() override { 129070602439SZachary Turner return llvm::makeArrayRef(g_script_import_options); 12911f0f5b5bSZachary Turner } 1292*00bb397bSJonas Devlieghere bool relative_to_command_file = false; 12930a305db7SEnrico Granata }; 12940a305db7SEnrico Granata 1295b9c1b51eSKate Stone bool DoExecute(Args &command, CommandReturnObject &result) override { 129611eb9c64SZachary Turner if (command.empty()) { 12973b00e35bSEnrico Granata result.AppendError("command script import needs one or more arguments"); 1298a9dbf432SEnrico Granata result.SetStatus(eReturnStatusFailed); 1299a9dbf432SEnrico Granata return false; 1300a9dbf432SEnrico Granata } 1301a9dbf432SEnrico Granata 1302*00bb397bSJonas Devlieghere FileSpec source_dir = {}; 1303*00bb397bSJonas Devlieghere if (m_options.relative_to_command_file) { 1304*00bb397bSJonas Devlieghere source_dir = GetDebugger().GetCommandInterpreter().GetCurrentSourceDir(); 1305*00bb397bSJonas Devlieghere if (!source_dir) { 1306*00bb397bSJonas Devlieghere result.AppendError("command script import -c can only be specified " 1307*00bb397bSJonas Devlieghere "from a command file"); 1308*00bb397bSJonas Devlieghere result.SetStatus(eReturnStatusFailed); 1309*00bb397bSJonas Devlieghere return false; 1310*00bb397bSJonas Devlieghere } 1311*00bb397bSJonas Devlieghere } 1312*00bb397bSJonas Devlieghere 131311eb9c64SZachary Turner for (auto &entry : command.entries()) { 131497206d57SZachary Turner Status error; 1315a9dbf432SEnrico Granata 1316c9d645d3SGreg Clayton const bool init_session = true; 1317b9c1b51eSKate Stone // FIXME: this is necessary because CommandObject::CheckRequirements() 131811eb9c64SZachary Turner // assumes that commands won't ever be recursively invoked, but it's 131911eb9c64SZachary Turner // actually possible to craft a Python script that does other "command 132005097246SAdrian Prantl // script imports" in __lldb_init_module the real fix is to have 132105097246SAdrian Prantl // recursive commands possible with a CommandInvocation object separate 132205097246SAdrian Prantl // from the CommandObject itself, so that recursive command invocations 132305097246SAdrian Prantl // won't stomp on each other (wrt to execution contents, options, and 132405097246SAdrian Prantl // more) 1325078551c7SEnrico Granata m_exe_ctx.Clear(); 13262b29b432SJonas Devlieghere if (GetDebugger().GetScriptInterpreter()->LoadScriptingModule( 1327*00bb397bSJonas Devlieghere entry.c_str(), init_session, error, nullptr, source_dir)) { 1328a9dbf432SEnrico Granata result.SetStatus(eReturnStatusSuccessFinishNoResult); 1329b9c1b51eSKate Stone } else { 1330b9c1b51eSKate Stone result.AppendErrorWithFormat("module importing failed: %s", 1331b9c1b51eSKate Stone error.AsCString()); 1332a9dbf432SEnrico Granata result.SetStatus(eReturnStatusFailed); 1333a9dbf432SEnrico Granata } 13343b00e35bSEnrico Granata } 1335a9dbf432SEnrico Granata 1336a9dbf432SEnrico Granata return result.Succeeded(); 1337a9dbf432SEnrico Granata } 13380a305db7SEnrico Granata 13395a988416SJim Ingham CommandOptions m_options; 1340a9dbf432SEnrico Granata }; 1341223383edSEnrico Granata 1342223383edSEnrico Granata // CommandObjectCommandsScriptAdd 13438fe53c49STatyana Krasnukha static constexpr OptionEnumValueElement g_script_synchro_type[] = { 1344e063ecccSJonas Devlieghere { 1345e063ecccSJonas Devlieghere eScriptedCommandSynchronicitySynchronous, 1346e063ecccSJonas Devlieghere "synchronous", 1347e063ecccSJonas Devlieghere "Run synchronous", 1348e063ecccSJonas Devlieghere }, 1349e063ecccSJonas Devlieghere { 1350e063ecccSJonas Devlieghere eScriptedCommandSynchronicityAsynchronous, 1351e063ecccSJonas Devlieghere "asynchronous", 1352e063ecccSJonas Devlieghere "Run asynchronous", 1353e063ecccSJonas Devlieghere }, 1354e063ecccSJonas Devlieghere { 1355e063ecccSJonas Devlieghere eScriptedCommandSynchronicityCurrentValue, 1356e063ecccSJonas Devlieghere "current", 1357e063ecccSJonas Devlieghere "Do not alter current setting", 1358e063ecccSJonas Devlieghere }, 1359e063ecccSJonas Devlieghere }; 13601f0f5b5bSZachary Turner 13618fe53c49STatyana Krasnukha static constexpr OptionEnumValues ScriptSynchroType() { 13628fe53c49STatyana Krasnukha return OptionEnumValues(g_script_synchro_type); 13638fe53c49STatyana Krasnukha } 13648fe53c49STatyana Krasnukha 136564becc11SRaphael Isemann #define LLDB_OPTIONS_script_add 136664becc11SRaphael Isemann #include "CommandOptions.inc" 13671f0f5b5bSZachary Turner 1368b9c1b51eSKate Stone class CommandObjectCommandsScriptAdd : public CommandObjectParsed, 1369b9c1b51eSKate Stone public IOHandlerDelegateMultiline { 13705a988416SJim Ingham public: 1371b9c1b51eSKate Stone CommandObjectCommandsScriptAdd(CommandInterpreter &interpreter) 1372b9c1b51eSKate Stone : CommandObjectParsed(interpreter, "command script add", 13735a988416SJim Ingham "Add a scripted function as an LLDB command.", 13746e3d8e7fSEugene Zelenko nullptr), 1375b9c1b51eSKate Stone IOHandlerDelegateMultiline("DONE"), m_options() { 13765a988416SJim Ingham CommandArgumentEntry arg1; 13775a988416SJim Ingham CommandArgumentData cmd_arg; 13785a988416SJim Ingham 13795a988416SJim Ingham // Define the first (and only) variant of this arg. 13805a988416SJim Ingham cmd_arg.arg_type = eArgTypeCommandName; 13815a988416SJim Ingham cmd_arg.arg_repetition = eArgRepeatPlain; 13825a988416SJim Ingham 1383b9c1b51eSKate Stone // There is only one variant this argument could be; put it into the 1384b9c1b51eSKate Stone // argument entry. 13855a988416SJim Ingham arg1.push_back(cmd_arg); 13865a988416SJim Ingham 13875a988416SJim Ingham // Push the data for the first argument into the m_arguments vector. 13885a988416SJim Ingham m_arguments.push_back(arg1); 13895a988416SJim Ingham } 13905a988416SJim Ingham 13916e3d8e7fSEugene Zelenko ~CommandObjectCommandsScriptAdd() override = default; 13925a988416SJim Ingham 1393b9c1b51eSKate Stone Options *GetOptions() override { return &m_options; } 13945a988416SJim Ingham 13955a988416SJim Ingham protected: 1396b9c1b51eSKate Stone class CommandOptions : public Options { 1397223383edSEnrico Granata public: 1398b9c1b51eSKate Stone CommandOptions() 1399b9c1b51eSKate Stone : Options(), m_class_name(), m_funct_name(), m_short_help(), 1400b9c1b51eSKate Stone m_synchronicity(eScriptedCommandSynchronicitySynchronous) {} 1401223383edSEnrico Granata 14026e3d8e7fSEugene Zelenko ~CommandOptions() override = default; 1403223383edSEnrico Granata 140497206d57SZachary Turner Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg, 1405b9c1b51eSKate Stone ExecutionContext *execution_context) override { 140697206d57SZachary Turner Status error; 14073bcdfc0eSGreg Clayton const int short_option = m_getopt_table[option_idx].val; 1408223383edSEnrico Granata 1409b9c1b51eSKate Stone switch (short_option) { 1410223383edSEnrico Granata case 'f': 1411fe11483bSZachary Turner if (!option_arg.empty()) 1412adcd0268SBenjamin Kramer m_funct_name = std::string(option_arg); 1413735152e3SEnrico Granata break; 14149fe00e52SEnrico Granata case 'c': 1415fe11483bSZachary Turner if (!option_arg.empty()) 1416adcd0268SBenjamin Kramer m_class_name = std::string(option_arg); 14179fe00e52SEnrico Granata break; 1418735152e3SEnrico Granata case 'h': 1419fe11483bSZachary Turner if (!option_arg.empty()) 1420adcd0268SBenjamin Kramer m_short_help = std::string(option_arg); 1421223383edSEnrico Granata break; 14220a305db7SEnrico Granata case 's': 1423b9c1b51eSKate Stone m_synchronicity = 142447cbf4a0SPavel Labath (ScriptedCommandSynchronicity)OptionArgParser::ToOptionEnum( 1425fe11483bSZachary Turner option_arg, GetDefinitions()[option_idx].enum_values, 0, error); 14260a305db7SEnrico Granata if (!error.Success()) 1427b9c1b51eSKate Stone error.SetErrorStringWithFormat( 1428fe11483bSZachary Turner "unrecognized value for synchronicity '%s'", 1429fe11483bSZachary Turner option_arg.str().c_str()); 14300a305db7SEnrico Granata break; 1431223383edSEnrico Granata default: 143236162014SRaphael Isemann llvm_unreachable("Unimplemented option"); 1433223383edSEnrico Granata } 1434223383edSEnrico Granata 1435223383edSEnrico Granata return error; 1436223383edSEnrico Granata } 1437223383edSEnrico Granata 1438b9c1b51eSKate Stone void OptionParsingStarting(ExecutionContext *execution_context) override { 14399fe00e52SEnrico Granata m_class_name.clear(); 1440735152e3SEnrico Granata m_funct_name.clear(); 1441735152e3SEnrico Granata m_short_help.clear(); 144244d93782SGreg Clayton m_synchronicity = eScriptedCommandSynchronicitySynchronous; 1443223383edSEnrico Granata } 1444223383edSEnrico Granata 14451f0f5b5bSZachary Turner llvm::ArrayRef<OptionDefinition> GetDefinitions() override { 144670602439SZachary Turner return llvm::makeArrayRef(g_script_add_options); 14471f0f5b5bSZachary Turner } 1448223383edSEnrico Granata 1449223383edSEnrico Granata // Instance variables to hold the values for command options. 1450223383edSEnrico Granata 14519fe00e52SEnrico Granata std::string m_class_name; 1452223383edSEnrico Granata std::string m_funct_name; 1453735152e3SEnrico Granata std::string m_short_help; 145444d93782SGreg Clayton ScriptedCommandSynchronicity m_synchronicity; 1455223383edSEnrico Granata }; 1456223383edSEnrico Granata 14570affb582SDave Lee void IOHandlerActivated(IOHandler &io_handler, bool interactive) override { 14587ca15ba7SLawrence D'Anna StreamFileSP output_sp(io_handler.GetOutputStreamFileSP()); 14590affb582SDave Lee if (output_sp && interactive) { 146044d93782SGreg Clayton output_sp->PutCString(g_python_command_instructions); 146144d93782SGreg Clayton output_sp->Flush(); 1462223383edSEnrico Granata } 1463223383edSEnrico Granata } 1464223383edSEnrico Granata 1465b9c1b51eSKate Stone void IOHandlerInputComplete(IOHandler &io_handler, 1466b9c1b51eSKate Stone std::string &data) override { 14677ca15ba7SLawrence D'Anna StreamFileSP error_sp = io_handler.GetErrorStreamFileSP(); 146844d93782SGreg Clayton 14692b29b432SJonas Devlieghere ScriptInterpreter *interpreter = GetDebugger().GetScriptInterpreter(); 1470b9c1b51eSKate Stone if (interpreter) { 147144d93782SGreg Clayton 147244d93782SGreg Clayton StringList lines; 147344d93782SGreg Clayton lines.SplitIntoLines(data); 1474b9c1b51eSKate Stone if (lines.GetSize() > 0) { 1475a73b7df7SEnrico Granata std::string funct_name_str; 1476b9c1b51eSKate Stone if (interpreter->GenerateScriptAliasFunction(lines, funct_name_str)) { 1477b9c1b51eSKate Stone if (funct_name_str.empty()) { 1478b9c1b51eSKate Stone error_sp->Printf("error: unable to obtain a function name, didn't " 1479b9c1b51eSKate Stone "add python command.\n"); 148044d93782SGreg Clayton error_sp->Flush(); 1481b9c1b51eSKate Stone } else { 1482223383edSEnrico Granata // everything should be fine now, let's add this alias 1483223383edSEnrico Granata 1484b9c1b51eSKate Stone CommandObjectSP command_obj_sp(new CommandObjectPythonFunction( 1485771ef6d4SMalcolm Parsons m_interpreter, m_cmd_name, funct_name_str, m_short_help, 148644d93782SGreg Clayton m_synchronicity)); 1487223383edSEnrico Granata 1488b9c1b51eSKate Stone if (!m_interpreter.AddUserCommand(m_cmd_name, command_obj_sp, 1489b9c1b51eSKate Stone true)) { 1490b9c1b51eSKate Stone error_sp->Printf("error: unable to add selected command, didn't " 1491b9c1b51eSKate Stone "add python command.\n"); 149244d93782SGreg Clayton error_sp->Flush(); 1493223383edSEnrico Granata } 1494223383edSEnrico Granata } 1495b9c1b51eSKate Stone } else { 1496b9c1b51eSKate Stone error_sp->Printf( 1497b9c1b51eSKate Stone "error: unable to create function, didn't add python command.\n"); 149844d93782SGreg Clayton error_sp->Flush(); 149944d93782SGreg Clayton } 1500b9c1b51eSKate Stone } else { 150144d93782SGreg Clayton error_sp->Printf("error: empty function, didn't add python command.\n"); 150244d93782SGreg Clayton error_sp->Flush(); 150344d93782SGreg Clayton } 1504b9c1b51eSKate Stone } else { 1505b9c1b51eSKate Stone error_sp->Printf( 1506b9c1b51eSKate Stone "error: script interpreter missing, didn't add python command.\n"); 150744d93782SGreg Clayton error_sp->Flush(); 150844d93782SGreg Clayton } 150944d93782SGreg Clayton 151044d93782SGreg Clayton io_handler.SetIsDone(true); 151144d93782SGreg Clayton } 1512223383edSEnrico Granata 1513b9c1b51eSKate Stone bool DoExecute(Args &command, CommandReturnObject &result) override { 151457179860SJonas Devlieghere if (GetDebugger().GetScriptLanguage() != lldb::eScriptLanguagePython) { 1515b9c1b51eSKate Stone result.AppendError("only scripting language supported for scripted " 1516b9c1b51eSKate Stone "commands is currently Python"); 151799f0b8f9SEnrico Granata result.SetStatus(eReturnStatusFailed); 151899f0b8f9SEnrico Granata return false; 151999f0b8f9SEnrico Granata } 152099f0b8f9SEnrico Granata 152111eb9c64SZachary Turner if (command.GetArgumentCount() != 1) { 1522223383edSEnrico Granata result.AppendError("'command script add' requires one argument"); 1523223383edSEnrico Granata result.SetStatus(eReturnStatusFailed); 1524223383edSEnrico Granata return false; 1525223383edSEnrico Granata } 1526223383edSEnrico Granata 1527735152e3SEnrico Granata // Store the options in case we get multi-line input 1528adcd0268SBenjamin Kramer m_cmd_name = std::string(command[0].ref()); 1529735152e3SEnrico Granata m_short_help.assign(m_options.m_short_help); 153044d93782SGreg Clayton m_synchronicity = m_options.m_synchronicity; 1531223383edSEnrico Granata 1532b9c1b51eSKate Stone if (m_options.m_class_name.empty()) { 1533b9c1b51eSKate Stone if (m_options.m_funct_name.empty()) { 1534b9c1b51eSKate Stone m_interpreter.GetPythonCommandsFromIOHandler( 1535b9c1b51eSKate Stone " ", // Prompt 1536a6faf851SJonas Devlieghere *this); // IOHandlerDelegate 1537b9c1b51eSKate Stone } else { 1538b9c1b51eSKate Stone CommandObjectSP new_cmd(new CommandObjectPythonFunction( 1539b9c1b51eSKate Stone m_interpreter, m_cmd_name, m_options.m_funct_name, 1540b9c1b51eSKate Stone m_options.m_short_help, m_synchronicity)); 1541b9c1b51eSKate Stone if (m_interpreter.AddUserCommand(m_cmd_name, new_cmd, true)) { 1542223383edSEnrico Granata result.SetStatus(eReturnStatusSuccessFinishNoResult); 1543b9c1b51eSKate Stone } else { 1544223383edSEnrico Granata result.AppendError("cannot add command"); 1545223383edSEnrico Granata result.SetStatus(eReturnStatusFailed); 1546223383edSEnrico Granata } 1547223383edSEnrico Granata } 1548b9c1b51eSKate Stone } else { 15492b29b432SJonas Devlieghere ScriptInterpreter *interpreter = GetDebugger().GetScriptInterpreter(); 1550b9c1b51eSKate Stone if (!interpreter) { 15519fe00e52SEnrico Granata result.AppendError("cannot find ScriptInterpreter"); 15529fe00e52SEnrico Granata result.SetStatus(eReturnStatusFailed); 15539fe00e52SEnrico Granata return false; 15549fe00e52SEnrico Granata } 15559fe00e52SEnrico Granata 1556b9c1b51eSKate Stone auto cmd_obj_sp = interpreter->CreateScriptCommandObject( 1557b9c1b51eSKate Stone m_options.m_class_name.c_str()); 1558b9c1b51eSKate Stone if (!cmd_obj_sp) { 15599fe00e52SEnrico Granata result.AppendError("cannot create helper object"); 15609fe00e52SEnrico Granata result.SetStatus(eReturnStatusFailed); 15619fe00e52SEnrico Granata return false; 15629fe00e52SEnrico Granata } 15639fe00e52SEnrico Granata 1564b9c1b51eSKate Stone CommandObjectSP new_cmd(new CommandObjectScriptingObject( 1565b9c1b51eSKate Stone m_interpreter, m_cmd_name, cmd_obj_sp, m_synchronicity)); 1566b9c1b51eSKate Stone if (m_interpreter.AddUserCommand(m_cmd_name, new_cmd, true)) { 15679fe00e52SEnrico Granata result.SetStatus(eReturnStatusSuccessFinishNoResult); 1568b9c1b51eSKate Stone } else { 15699fe00e52SEnrico Granata result.AppendError("cannot add command"); 15709fe00e52SEnrico Granata result.SetStatus(eReturnStatusFailed); 15719fe00e52SEnrico Granata } 15729fe00e52SEnrico Granata } 1573223383edSEnrico Granata 1574223383edSEnrico Granata return result.Succeeded(); 1575223383edSEnrico Granata } 15765a988416SJim Ingham 15775a988416SJim Ingham CommandOptions m_options; 157844d93782SGreg Clayton std::string m_cmd_name; 1579735152e3SEnrico Granata std::string m_short_help; 158044d93782SGreg Clayton ScriptedCommandSynchronicity m_synchronicity; 1581223383edSEnrico Granata }; 1582223383edSEnrico Granata 1583223383edSEnrico Granata // CommandObjectCommandsScriptList 1584223383edSEnrico Granata 1585b9c1b51eSKate Stone class CommandObjectCommandsScriptList : public CommandObjectParsed { 1586223383edSEnrico Granata public: 1587b9c1b51eSKate Stone CommandObjectCommandsScriptList(CommandInterpreter &interpreter) 1588b9c1b51eSKate Stone : CommandObjectParsed(interpreter, "command script list", 1589b9c1b51eSKate Stone "List defined scripted commands.", nullptr) {} 1590223383edSEnrico Granata 15916e3d8e7fSEugene Zelenko ~CommandObjectCommandsScriptList() override = default; 1592223383edSEnrico Granata 1593b9c1b51eSKate Stone bool DoExecute(Args &command, CommandReturnObject &result) override { 1594d77ea5b2SRaphael Isemann if (command.GetArgumentCount() != 0) { 1595d77ea5b2SRaphael Isemann result.AppendError("'command script list' doesn't take any arguments"); 1596d77ea5b2SRaphael Isemann result.SetStatus(eReturnStatusFailed); 1597d77ea5b2SRaphael Isemann return false; 1598d77ea5b2SRaphael Isemann } 1599d77ea5b2SRaphael Isemann 1600b9c1b51eSKate Stone m_interpreter.GetHelp(result, CommandInterpreter::eCommandTypesUserDef); 1601223383edSEnrico Granata 1602223383edSEnrico Granata result.SetStatus(eReturnStatusSuccessFinishResult); 1603223383edSEnrico Granata 1604223383edSEnrico Granata return true; 1605223383edSEnrico Granata } 1606223383edSEnrico Granata }; 1607223383edSEnrico Granata 1608223383edSEnrico Granata // CommandObjectCommandsScriptClear 1609223383edSEnrico Granata 1610b9c1b51eSKate Stone class CommandObjectCommandsScriptClear : public CommandObjectParsed { 1611223383edSEnrico Granata public: 1612b9c1b51eSKate Stone CommandObjectCommandsScriptClear(CommandInterpreter &interpreter) 1613b9c1b51eSKate Stone : CommandObjectParsed(interpreter, "command script clear", 1614b9c1b51eSKate Stone "Delete all scripted commands.", nullptr) {} 1615223383edSEnrico Granata 16166e3d8e7fSEugene Zelenko ~CommandObjectCommandsScriptClear() override = default; 1617223383edSEnrico Granata 16185a988416SJim Ingham protected: 1619b9c1b51eSKate Stone bool DoExecute(Args &command, CommandReturnObject &result) override { 1620d77ea5b2SRaphael Isemann if (command.GetArgumentCount() != 0) { 1621d77ea5b2SRaphael Isemann result.AppendError("'command script clear' doesn't take any arguments"); 1622d77ea5b2SRaphael Isemann result.SetStatus(eReturnStatusFailed); 1623d77ea5b2SRaphael Isemann return false; 1624d77ea5b2SRaphael Isemann } 1625d77ea5b2SRaphael Isemann 1626223383edSEnrico Granata m_interpreter.RemoveAllUser(); 1627223383edSEnrico Granata 1628223383edSEnrico Granata result.SetStatus(eReturnStatusSuccessFinishResult); 1629223383edSEnrico Granata 1630223383edSEnrico Granata return true; 1631223383edSEnrico Granata } 1632223383edSEnrico Granata }; 1633223383edSEnrico Granata 1634223383edSEnrico Granata // CommandObjectCommandsScriptDelete 1635223383edSEnrico Granata 1636b9c1b51eSKate Stone class CommandObjectCommandsScriptDelete : public CommandObjectParsed { 1637223383edSEnrico Granata public: 1638b9c1b51eSKate Stone CommandObjectCommandsScriptDelete(CommandInterpreter &interpreter) 1639b9c1b51eSKate Stone : CommandObjectParsed(interpreter, "command script delete", 1640b9c1b51eSKate Stone "Delete a scripted command.", nullptr) { 1641223383edSEnrico Granata CommandArgumentEntry arg1; 1642223383edSEnrico Granata CommandArgumentData cmd_arg; 1643223383edSEnrico Granata 1644223383edSEnrico Granata // Define the first (and only) variant of this arg. 1645223383edSEnrico Granata cmd_arg.arg_type = eArgTypeCommandName; 1646223383edSEnrico Granata cmd_arg.arg_repetition = eArgRepeatPlain; 1647223383edSEnrico Granata 1648b9c1b51eSKate Stone // There is only one variant this argument could be; put it into the 1649b9c1b51eSKate Stone // argument entry. 1650223383edSEnrico Granata arg1.push_back(cmd_arg); 1651223383edSEnrico Granata 1652223383edSEnrico Granata // Push the data for the first argument into the m_arguments vector. 1653223383edSEnrico Granata m_arguments.push_back(arg1); 1654223383edSEnrico Granata } 1655223383edSEnrico Granata 16566e3d8e7fSEugene Zelenko ~CommandObjectCommandsScriptDelete() override = default; 1657223383edSEnrico Granata 16582e8f304fSGongyu Deng void 16592e8f304fSGongyu Deng HandleArgumentCompletion(CompletionRequest &request, 16602e8f304fSGongyu Deng OptionElementVector &opt_element_vector) override { 16612e8f304fSGongyu Deng if (!m_interpreter.HasCommands() || request.GetCursorIndex() != 0) 16622e8f304fSGongyu Deng return; 16632e8f304fSGongyu Deng 16642ebe30c6SRaphael Isemann for (const auto &c : m_interpreter.GetUserCommands()) 16652ebe30c6SRaphael Isemann request.TryCompleteCurrentArg(c.first, c.second->GetHelp()); 16662e8f304fSGongyu Deng } 16672e8f304fSGongyu Deng 16685a988416SJim Ingham protected: 1669b9c1b51eSKate Stone bool DoExecute(Args &command, CommandReturnObject &result) override { 1670223383edSEnrico Granata 167111eb9c64SZachary Turner if (command.GetArgumentCount() != 1) { 1672223383edSEnrico Granata result.AppendError("'command script delete' requires one argument"); 1673223383edSEnrico Granata result.SetStatus(eReturnStatusFailed); 1674223383edSEnrico Granata return false; 1675223383edSEnrico Granata } 1676223383edSEnrico Granata 16770d9a201eSRaphael Isemann auto cmd_name = command[0].ref(); 1678223383edSEnrico Granata 16794574a890SZachary Turner if (cmd_name.empty() || !m_interpreter.HasUserCommands() || 16804574a890SZachary Turner !m_interpreter.UserCommandExists(cmd_name)) { 1681867e7d17SZachary Turner result.AppendErrorWithFormat("command %s not found", command[0].c_str()); 1682223383edSEnrico Granata result.SetStatus(eReturnStatusFailed); 16834574a890SZachary Turner return false; 1684223383edSEnrico Granata } 1685223383edSEnrico Granata 16864574a890SZachary Turner m_interpreter.RemoveUser(cmd_name); 16874574a890SZachary Turner result.SetStatus(eReturnStatusSuccessFinishResult); 16884574a890SZachary Turner return true; 1689223383edSEnrico Granata } 1690223383edSEnrico Granata }; 1691223383edSEnrico Granata 1692223383edSEnrico Granata #pragma mark CommandObjectMultiwordCommandsScript 1693223383edSEnrico Granata 1694223383edSEnrico Granata // CommandObjectMultiwordCommandsScript 1695223383edSEnrico Granata 1696b9c1b51eSKate Stone class CommandObjectMultiwordCommandsScript : public CommandObjectMultiword { 1697223383edSEnrico Granata public: 16987428a18cSKate Stone CommandObjectMultiwordCommandsScript(CommandInterpreter &interpreter) 1699b9c1b51eSKate Stone : CommandObjectMultiword( 1700a925974bSAdrian Prantl interpreter, "command script", 1701a925974bSAdrian Prantl "Commands for managing custom " 1702b9c1b51eSKate Stone "commands implemented by " 1703b9c1b51eSKate Stone "interpreter scripts.", 1704b9c1b51eSKate Stone "command script <subcommand> [<subcommand-options>]") { 1705b9c1b51eSKate Stone LoadSubCommand("add", CommandObjectSP( 1706b9c1b51eSKate Stone new CommandObjectCommandsScriptAdd(interpreter))); 1707b9c1b51eSKate Stone LoadSubCommand( 1708b9c1b51eSKate Stone "delete", 1709b9c1b51eSKate Stone CommandObjectSP(new CommandObjectCommandsScriptDelete(interpreter))); 1710b9c1b51eSKate Stone LoadSubCommand( 1711b9c1b51eSKate Stone "clear", 1712b9c1b51eSKate Stone CommandObjectSP(new CommandObjectCommandsScriptClear(interpreter))); 1713b9c1b51eSKate Stone LoadSubCommand("list", CommandObjectSP(new CommandObjectCommandsScriptList( 1714b9c1b51eSKate Stone interpreter))); 1715b9c1b51eSKate Stone LoadSubCommand( 1716b9c1b51eSKate Stone "import", 1717b9c1b51eSKate Stone CommandObjectSP(new CommandObjectCommandsScriptImport(interpreter))); 1718223383edSEnrico Granata } 1719223383edSEnrico Granata 17206e3d8e7fSEugene Zelenko ~CommandObjectMultiwordCommandsScript() override = default; 1721223383edSEnrico Granata }; 1722223383edSEnrico Granata 1723ebc09c36SJim Ingham #pragma mark CommandObjectMultiwordCommands 1724ebc09c36SJim Ingham 1725ebc09c36SJim Ingham // CommandObjectMultiwordCommands 1726ebc09c36SJim Ingham 1727b9c1b51eSKate Stone CommandObjectMultiwordCommands::CommandObjectMultiwordCommands( 1728b9c1b51eSKate Stone CommandInterpreter &interpreter) 1729b9c1b51eSKate Stone : CommandObjectMultiword(interpreter, "command", 1730b9c1b51eSKate Stone "Commands for managing custom LLDB commands.", 1731b9c1b51eSKate Stone "command <subcommand> [<subcommand-options>]") { 1732b9c1b51eSKate Stone LoadSubCommand("source", 1733b9c1b51eSKate Stone CommandObjectSP(new CommandObjectCommandsSource(interpreter))); 1734b9c1b51eSKate Stone LoadSubCommand("alias", 1735b9c1b51eSKate Stone CommandObjectSP(new CommandObjectCommandsAlias(interpreter))); 1736b9c1b51eSKate Stone LoadSubCommand("unalias", CommandObjectSP( 1737b9c1b51eSKate Stone new CommandObjectCommandsUnalias(interpreter))); 1738b9c1b51eSKate Stone LoadSubCommand("delete", 1739b9c1b51eSKate Stone CommandObjectSP(new CommandObjectCommandsDelete(interpreter))); 1740b9c1b51eSKate Stone LoadSubCommand( 1741b9c1b51eSKate Stone "regex", CommandObjectSP(new CommandObjectCommandsAddRegex(interpreter))); 1742b9c1b51eSKate Stone LoadSubCommand( 1743b9c1b51eSKate Stone "script", 1744b9c1b51eSKate Stone CommandObjectSP(new CommandObjectMultiwordCommandsScript(interpreter))); 1745ebc09c36SJim Ingham } 1746ebc09c36SJim Ingham 17476e3d8e7fSEugene Zelenko CommandObjectMultiwordCommands::~CommandObjectMultiwordCommands() = default; 1748