1696bd635SAlexander Shaposhnikov //===-- CommandObjectCommands.cpp -------------------------------*- C++ -*-===// 2ebc09c36SJim Ingham // 3ebc09c36SJim Ingham // The LLVM Compiler Infrastructure 4ebc09c36SJim Ingham // 5ebc09c36SJim Ingham // This file is distributed under the University of Illinois Open Source 6ebc09c36SJim Ingham // License. See LICENSE.TXT for details. 7ebc09c36SJim Ingham // 8ebc09c36SJim Ingham //===----------------------------------------------------------------------===// 9ebc09c36SJim Ingham 10ebc09c36SJim Ingham // C Includes 11ebc09c36SJim Ingham // C++ Includes 12ebc09c36SJim Ingham // Other libraries and framework includes 130e5e5a79SGreg Clayton #include "llvm/ADT/StringRef.h" 140e5e5a79SGreg Clayton 15ebc09c36SJim Ingham // Project includes 166e3d8e7fSEugene Zelenko #include "CommandObjectCommands.h" 1746d4aa21SEnrico Granata #include "CommandObjectHelp.h" 18ebc09c36SJim Ingham #include "lldb/Core/Debugger.h" 1944d93782SGreg Clayton #include "lldb/Core/IOHandler.h" 203eb2b44dSZachary Turner #include "lldb/Host/OptionParser.h" 21de164aaaSGreg Clayton #include "lldb/Interpreter/Args.h" 227594f14fSEnrico Granata #include "lldb/Interpreter/CommandHistory.h" 23ebc09c36SJim Ingham #include "lldb/Interpreter/CommandInterpreter.h" 24de164aaaSGreg Clayton #include "lldb/Interpreter/CommandObjectRegexCommand.h" 25ebc09c36SJim Ingham #include "lldb/Interpreter/CommandReturnObject.h" 26012d4fcaSEnrico Granata #include "lldb/Interpreter/OptionValueBoolean.h" 2745d0e238SEnrico Granata #include "lldb/Interpreter/OptionValueString.h" 287594f14fSEnrico Granata #include "lldb/Interpreter/OptionValueUInt64.h" 29ebc09c36SJim Ingham #include "lldb/Interpreter/Options.h" 3099f0b8f9SEnrico Granata #include "lldb/Interpreter/ScriptInterpreter.h" 31573ab909SZachary Turner #include "lldb/Utility/StringList.h" 32ebc09c36SJim Ingham 33ebc09c36SJim Ingham using namespace lldb; 34ebc09c36SJim Ingham using namespace lldb_private; 35ebc09c36SJim Ingham 36ebc09c36SJim Ingham //------------------------------------------------------------------------- 37ebc09c36SJim Ingham // CommandObjectCommandsSource 38ebc09c36SJim Ingham //------------------------------------------------------------------------- 39ebc09c36SJim Ingham 401f0f5b5bSZachary Turner static OptionDefinition g_history_options[] = { 411f0f5b5bSZachary Turner // clang-format off 421f0f5b5bSZachary Turner { LLDB_OPT_SET_1, false, "count", 'c', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeUnsignedInteger, "How many history commands to print." }, 431f0f5b5bSZachary Turner { LLDB_OPT_SET_1, false, "start-index", 's', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeUnsignedInteger, "Index at which to start printing history commands (or end to mean tail mode)." }, 441f0f5b5bSZachary Turner { LLDB_OPT_SET_1, false, "end-index", 'e', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeUnsignedInteger, "Index at which to stop printing history commands." }, 451f0f5b5bSZachary Turner { LLDB_OPT_SET_2, false, "clear", 'C', OptionParser::eNoArgument, nullptr, nullptr, 0, eArgTypeBoolean, "Clears the current command history." }, 461f0f5b5bSZachary Turner // clang-format on 471f0f5b5bSZachary Turner }; 481f0f5b5bSZachary Turner 49b9c1b51eSKate Stone class CommandObjectCommandsHistory : public CommandObjectParsed { 505a988416SJim Ingham public: 51b9c1b51eSKate Stone CommandObjectCommandsHistory(CommandInterpreter &interpreter) 52b9c1b51eSKate Stone : CommandObjectParsed(interpreter, "command history", 53*aab5be05SJim Ingham "Dump the history of commands in this session.\n" 54*aab5be05SJim Ingham "Commands in the history list can be run again " 55*aab5be05SJim Ingham "using \"!<INDEX>\". \"!-<OFFSET>\" will re-run " 56*aab5be05SJim Ingham "the command that is <OFFSET> commands from the end" 57*aab5be05SJim Ingham " of the list (counting the current command).", 586e3d8e7fSEugene Zelenko nullptr), 59b9c1b51eSKate Stone m_options() {} 605a988416SJim Ingham 616e3d8e7fSEugene Zelenko ~CommandObjectCommandsHistory() override = default; 625a988416SJim Ingham 63b9c1b51eSKate Stone Options *GetOptions() override { return &m_options; } 645a988416SJim Ingham 655a988416SJim Ingham protected: 66b9c1b51eSKate Stone class CommandOptions : public Options { 67a5a97ebeSJim Ingham public: 68b9c1b51eSKate Stone CommandOptions() 69b9c1b51eSKate Stone : Options(), m_start_idx(0), m_stop_idx(0), m_count(0), m_clear(false) { 70a5a97ebeSJim Ingham } 71a5a97ebeSJim Ingham 726e3d8e7fSEugene Zelenko ~CommandOptions() override = default; 73a5a97ebeSJim Ingham 74fe11483bSZachary Turner Error SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg, 75b9c1b51eSKate Stone ExecutionContext *execution_context) override { 76a5a97ebeSJim Ingham Error error; 773bcdfc0eSGreg Clayton const int short_option = m_getopt_table[option_idx].val; 78a5a97ebeSJim Ingham 79b9c1b51eSKate Stone switch (short_option) { 80a5a97ebeSJim Ingham case 'c': 81fe11483bSZachary Turner error = m_count.SetValueFromString(option_arg, eVarSetOperationAssign); 82a5a97ebeSJim Ingham break; 83a5a97ebeSJim Ingham case 's': 84fe11483bSZachary Turner if (option_arg == "end") { 857594f14fSEnrico Granata m_start_idx.SetCurrentValue(UINT64_MAX); 867594f14fSEnrico Granata m_start_idx.SetOptionWasSet(); 87b9c1b51eSKate Stone } else 88fe11483bSZachary Turner error = m_start_idx.SetValueFromString(option_arg, 89b9c1b51eSKate Stone eVarSetOperationAssign); 907594f14fSEnrico Granata break; 917594f14fSEnrico Granata case 'e': 92fe11483bSZachary Turner error = 93fe11483bSZachary Turner m_stop_idx.SetValueFromString(option_arg, eVarSetOperationAssign); 947594f14fSEnrico Granata break; 9563123b64SEnrico Granata case 'C': 9663123b64SEnrico Granata m_clear.SetCurrentValue(true); 9763123b64SEnrico Granata m_clear.SetOptionWasSet(); 98a5a97ebeSJim Ingham break; 99a5a97ebeSJim Ingham default: 100b9c1b51eSKate Stone error.SetErrorStringWithFormat("unrecognized option '%c'", 101b9c1b51eSKate Stone short_option); 102a5a97ebeSJim Ingham break; 103a5a97ebeSJim Ingham } 104a5a97ebeSJim Ingham 105a5a97ebeSJim Ingham return error; 106a5a97ebeSJim Ingham } 107a5a97ebeSJim Ingham 108b9c1b51eSKate Stone void OptionParsingStarting(ExecutionContext *execution_context) override { 1097594f14fSEnrico Granata m_start_idx.Clear(); 1107594f14fSEnrico Granata m_stop_idx.Clear(); 1117594f14fSEnrico Granata m_count.Clear(); 11263123b64SEnrico Granata m_clear.Clear(); 113a5a97ebeSJim Ingham } 114a5a97ebeSJim Ingham 1151f0f5b5bSZachary Turner llvm::ArrayRef<OptionDefinition> GetDefinitions() override { 11670602439SZachary Turner return llvm::makeArrayRef(g_history_options); 1171f0f5b5bSZachary Turner } 118a5a97ebeSJim Ingham 119a5a97ebeSJim Ingham // Instance variables to hold the values for command options. 120a5a97ebeSJim Ingham 1217594f14fSEnrico Granata OptionValueUInt64 m_start_idx; 1227594f14fSEnrico Granata OptionValueUInt64 m_stop_idx; 1237594f14fSEnrico Granata OptionValueUInt64 m_count; 12463123b64SEnrico Granata OptionValueBoolean m_clear; 125a5a97ebeSJim Ingham }; 126a5a97ebeSJim Ingham 127b9c1b51eSKate Stone bool DoExecute(Args &command, CommandReturnObject &result) override { 128b9c1b51eSKate Stone if (m_options.m_clear.GetCurrentValue() && 129b9c1b51eSKate Stone m_options.m_clear.OptionWasSet()) { 1307594f14fSEnrico Granata m_interpreter.GetCommandHistory().Clear(); 1317594f14fSEnrico Granata result.SetStatus(lldb::eReturnStatusSuccessFinishNoResult); 132b9c1b51eSKate Stone } else { 133b9c1b51eSKate Stone if (m_options.m_start_idx.OptionWasSet() && 134b9c1b51eSKate Stone m_options.m_stop_idx.OptionWasSet() && 135b9c1b51eSKate Stone m_options.m_count.OptionWasSet()) { 136b9c1b51eSKate Stone result.AppendError("--count, --start-index and --end-index cannot be " 137b9c1b51eSKate Stone "all specified in the same invocation"); 1387594f14fSEnrico Granata result.SetStatus(lldb::eReturnStatusFailed); 139b9c1b51eSKate Stone } else { 140b9c1b51eSKate Stone std::pair<bool, uint64_t> start_idx( 141b9c1b51eSKate Stone m_options.m_start_idx.OptionWasSet(), 142b9c1b51eSKate Stone m_options.m_start_idx.GetCurrentValue()); 143b9c1b51eSKate Stone std::pair<bool, uint64_t> stop_idx( 144b9c1b51eSKate Stone m_options.m_stop_idx.OptionWasSet(), 145b9c1b51eSKate Stone m_options.m_stop_idx.GetCurrentValue()); 146b9c1b51eSKate Stone std::pair<bool, uint64_t> count(m_options.m_count.OptionWasSet(), 147b9c1b51eSKate Stone m_options.m_count.GetCurrentValue()); 148a5a97ebeSJim Ingham 1497594f14fSEnrico Granata const CommandHistory &history(m_interpreter.GetCommandHistory()); 1507594f14fSEnrico Granata 151b9c1b51eSKate Stone if (start_idx.first && start_idx.second == UINT64_MAX) { 152b9c1b51eSKate Stone if (count.first) { 1537594f14fSEnrico Granata start_idx.second = history.GetSize() - count.second; 1547594f14fSEnrico Granata stop_idx.second = history.GetSize() - 1; 155b9c1b51eSKate Stone } else if (stop_idx.first) { 1567594f14fSEnrico Granata start_idx.second = stop_idx.second; 1577594f14fSEnrico Granata stop_idx.second = history.GetSize() - 1; 158b9c1b51eSKate Stone } else { 1597594f14fSEnrico Granata start_idx.second = 0; 1607594f14fSEnrico Granata stop_idx.second = history.GetSize() - 1; 1617594f14fSEnrico Granata } 162b9c1b51eSKate Stone } else { 163b9c1b51eSKate Stone if (!start_idx.first && !stop_idx.first && !count.first) { 1647594f14fSEnrico Granata start_idx.second = 0; 1657594f14fSEnrico Granata stop_idx.second = history.GetSize() - 1; 166b9c1b51eSKate Stone } else if (start_idx.first) { 167b9c1b51eSKate Stone if (count.first) { 1687594f14fSEnrico Granata stop_idx.second = start_idx.second + count.second - 1; 169b9c1b51eSKate Stone } else if (!stop_idx.first) { 1707594f14fSEnrico Granata stop_idx.second = history.GetSize() - 1; 1717594f14fSEnrico Granata } 172b9c1b51eSKate Stone } else if (stop_idx.first) { 173b9c1b51eSKate Stone if (count.first) { 1747594f14fSEnrico Granata if (stop_idx.second >= count.second) 1757594f14fSEnrico Granata start_idx.second = stop_idx.second - count.second + 1; 1767594f14fSEnrico Granata else 1777594f14fSEnrico Granata start_idx.second = 0; 1787594f14fSEnrico Granata } 179b9c1b51eSKate Stone } else /* if (count.first) */ 1807594f14fSEnrico Granata { 1817594f14fSEnrico Granata start_idx.second = 0; 1827594f14fSEnrico Granata stop_idx.second = count.second - 1; 1837594f14fSEnrico Granata } 1847594f14fSEnrico Granata } 185b9c1b51eSKate Stone history.Dump(result.GetOutputStream(), start_idx.second, 186b9c1b51eSKate Stone stop_idx.second); 1877594f14fSEnrico Granata } 1887594f14fSEnrico Granata } 189a5a97ebeSJim Ingham return result.Succeeded(); 190a5a97ebeSJim Ingham } 1915a988416SJim Ingham 1925a988416SJim Ingham CommandOptions m_options; 193a5a97ebeSJim Ingham }; 194a5a97ebeSJim Ingham 195a5a97ebeSJim Ingham //------------------------------------------------------------------------- 196a5a97ebeSJim Ingham // CommandObjectCommandsSource 197a5a97ebeSJim Ingham //------------------------------------------------------------------------- 198a5a97ebeSJim Ingham 1991f0f5b5bSZachary Turner static OptionDefinition g_source_options[] = { 2001f0f5b5bSZachary Turner // clang-format off 2011f0f5b5bSZachary Turner { LLDB_OPT_SET_ALL, false, "stop-on-error", 'e', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeBoolean, "If true, stop executing commands on error." }, 2021f0f5b5bSZachary Turner { LLDB_OPT_SET_ALL, false, "stop-on-continue", 'c', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeBoolean, "If true, stop executing commands on continue." }, 2031f0f5b5bSZachary Turner { LLDB_OPT_SET_ALL, false, "silent-run", 's', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeBoolean, "If true don't echo commands while executing." }, 2041f0f5b5bSZachary Turner // clang-format on 2051f0f5b5bSZachary Turner }; 2061f0f5b5bSZachary Turner 207b9c1b51eSKate Stone class CommandObjectCommandsSource : public CommandObjectParsed { 2085a988416SJim Ingham public: 2097428a18cSKate Stone CommandObjectCommandsSource(CommandInterpreter &interpreter) 210b9c1b51eSKate Stone : CommandObjectParsed( 211b9c1b51eSKate Stone interpreter, "command source", 212b9c1b51eSKate Stone "Read and execute LLDB commands from the file <filename>.", 2136e3d8e7fSEugene Zelenko nullptr), 214b9c1b51eSKate Stone m_options() { 2155a988416SJim Ingham CommandArgumentEntry arg; 2165a988416SJim Ingham CommandArgumentData file_arg; 2175a988416SJim Ingham 2185a988416SJim Ingham // Define the first (and only) variant of this arg. 2195a988416SJim Ingham file_arg.arg_type = eArgTypeFilename; 2205a988416SJim Ingham file_arg.arg_repetition = eArgRepeatPlain; 2215a988416SJim Ingham 222b9c1b51eSKate Stone // There is only one variant this argument could be; put it into the 223b9c1b51eSKate Stone // argument entry. 2245a988416SJim Ingham arg.push_back(file_arg); 2255a988416SJim Ingham 2265a988416SJim Ingham // Push the data for the first argument into the m_arguments vector. 2275a988416SJim Ingham m_arguments.push_back(arg); 2285a988416SJim Ingham } 2295a988416SJim Ingham 2306e3d8e7fSEugene Zelenko ~CommandObjectCommandsSource() override = default; 2315a988416SJim Ingham 232b9c1b51eSKate Stone const char *GetRepeatCommand(Args ¤t_command_args, 233b9c1b51eSKate Stone uint32_t index) override { 2345a988416SJim Ingham return ""; 2355a988416SJim Ingham } 2365a988416SJim Ingham 237b9c1b51eSKate Stone int HandleArgumentCompletion(Args &input, int &cursor_index, 2385a988416SJim Ingham int &cursor_char_position, 2395a988416SJim Ingham OptionElementVector &opt_element_vector, 240b9c1b51eSKate Stone int match_start_point, int max_return_elements, 2415a988416SJim Ingham bool &word_complete, 242b9c1b51eSKate Stone StringList &matches) override { 2434574a890SZachary Turner auto completion_str = input[cursor_index].ref; 2444574a890SZachary Turner completion_str = completion_str.take_front(cursor_char_position); 2455a988416SJim Ingham 246b9c1b51eSKate Stone CommandCompletions::InvokeCommonCompletionCallbacks( 247b9c1b51eSKate Stone GetCommandInterpreter(), CommandCompletions::eDiskFileCompletion, 2484574a890SZachary Turner completion_str, match_start_point, max_return_elements, nullptr, 249b9c1b51eSKate Stone word_complete, matches); 2505a988416SJim Ingham return matches.GetSize(); 2515a988416SJim Ingham } 2525a988416SJim Ingham 253b9c1b51eSKate Stone Options *GetOptions() override { return &m_options; } 2545a988416SJim Ingham 2555a988416SJim Ingham protected: 256b9c1b51eSKate Stone class CommandOptions : public Options { 257e16c50a1SJim Ingham public: 258b9c1b51eSKate Stone CommandOptions() 259b9c1b51eSKate Stone : Options(), m_stop_on_error(true), m_silent_run(false), 260b9c1b51eSKate Stone m_stop_on_continue(true) {} 261e16c50a1SJim Ingham 2626e3d8e7fSEugene Zelenko ~CommandOptions() override = default; 263e16c50a1SJim Ingham 264fe11483bSZachary Turner Error SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg, 265b9c1b51eSKate Stone ExecutionContext *execution_context) override { 266e16c50a1SJim Ingham Error error; 2673bcdfc0eSGreg Clayton const int short_option = m_getopt_table[option_idx].val; 268e16c50a1SJim Ingham 269b9c1b51eSKate Stone switch (short_option) { 270e16c50a1SJim Ingham case 'e': 271fe11483bSZachary Turner error = m_stop_on_error.SetValueFromString(option_arg); 272e16c50a1SJim Ingham break; 273340b0309SGreg Clayton 274e16c50a1SJim Ingham case 'c': 275fe11483bSZachary Turner error = m_stop_on_continue.SetValueFromString(option_arg); 276e16c50a1SJim Ingham break; 277340b0309SGreg Clayton 27860986174SMichael Sartain case 's': 279fe11483bSZachary Turner error = m_silent_run.SetValueFromString(option_arg); 28060986174SMichael Sartain break; 281340b0309SGreg Clayton 282e16c50a1SJim Ingham default: 283b9c1b51eSKate Stone error.SetErrorStringWithFormat("unrecognized option '%c'", 284b9c1b51eSKate Stone short_option); 285e16c50a1SJim Ingham break; 286e16c50a1SJim Ingham } 287e16c50a1SJim Ingham 288e16c50a1SJim Ingham return error; 289e16c50a1SJim Ingham } 290e16c50a1SJim Ingham 291b9c1b51eSKate Stone void OptionParsingStarting(ExecutionContext *execution_context) override { 292012d4fcaSEnrico Granata m_stop_on_error.Clear(); 293340b0309SGreg Clayton m_silent_run.Clear(); 294340b0309SGreg Clayton m_stop_on_continue.Clear(); 295e16c50a1SJim Ingham } 296e16c50a1SJim Ingham 2971f0f5b5bSZachary Turner llvm::ArrayRef<OptionDefinition> GetDefinitions() override { 29870602439SZachary Turner return llvm::makeArrayRef(g_source_options); 2991f0f5b5bSZachary Turner } 300e16c50a1SJim Ingham 301e16c50a1SJim Ingham // Instance variables to hold the values for command options. 302e16c50a1SJim Ingham 303012d4fcaSEnrico Granata OptionValueBoolean m_stop_on_error; 304340b0309SGreg Clayton OptionValueBoolean m_silent_run; 305340b0309SGreg Clayton OptionValueBoolean m_stop_on_continue; 306e16c50a1SJim Ingham }; 307e16c50a1SJim Ingham 308b9c1b51eSKate Stone bool DoExecute(Args &command, CommandReturnObject &result) override { 3094574a890SZachary Turner if (command.GetArgumentCount() != 1) { 3104574a890SZachary Turner result.AppendErrorWithFormat( 3114574a890SZachary Turner "'%s' takes exactly one executable filename argument.\n", 3124574a890SZachary Turner GetCommandName().str().c_str()); 3134574a890SZachary Turner result.SetStatus(eReturnStatusFailed); 3144574a890SZachary Turner return false; 3154574a890SZachary Turner } 316ebc09c36SJim Ingham 3174574a890SZachary Turner FileSpec cmd_file(command[0].ref, true); 3186e3d8e7fSEugene Zelenko ExecutionContext *exe_ctx = nullptr; // Just use the default context. 319ebc09c36SJim Ingham 320340b0309SGreg Clayton // If any options were set, then use them 321340b0309SGreg Clayton if (m_options.m_stop_on_error.OptionWasSet() || 322340b0309SGreg Clayton m_options.m_silent_run.OptionWasSet() || 323b9c1b51eSKate Stone m_options.m_stop_on_continue.OptionWasSet()) { 324340b0309SGreg Clayton // Use user set settings 32526c7bf93SJim Ingham CommandInterpreterRunOptions options; 3264574a890SZachary Turner options.SetStopOnContinue(m_options.m_stop_on_continue.GetCurrentValue()); 32726c7bf93SJim Ingham options.SetStopOnError(m_options.m_stop_on_error.GetCurrentValue()); 3287d8555c4SJim Ingham options.SetEchoCommands(!m_options.m_silent_run.GetCurrentValue()); 3297d8555c4SJim Ingham options.SetPrintResults(!m_options.m_silent_run.GetCurrentValue()); 33026c7bf93SJim Ingham 3314574a890SZachary Turner m_interpreter.HandleCommandsFromFile(cmd_file, exe_ctx, options, result); 332b9c1b51eSKate Stone } else { 333b9c1b51eSKate Stone // No options were set, inherit any settings from nested "command 334b9c1b51eSKate Stone // source" commands, 335340b0309SGreg Clayton // or set to sane default settings... 33626c7bf93SJim Ingham CommandInterpreterRunOptions options; 3374574a890SZachary Turner m_interpreter.HandleCommandsFromFile(cmd_file, exe_ctx, options, result); 338ebc09c36SJim Ingham } 339ebc09c36SJim Ingham return result.Succeeded(); 340ebc09c36SJim Ingham } 3416e3d8e7fSEugene Zelenko 3425a988416SJim Ingham CommandOptions m_options; 343ebc09c36SJim Ingham }; 344ebc09c36SJim Ingham 345ebc09c36SJim Ingham #pragma mark CommandObjectCommandsAlias 346ebc09c36SJim Ingham //------------------------------------------------------------------------- 347ebc09c36SJim Ingham // CommandObjectCommandsAlias 348ebc09c36SJim Ingham //------------------------------------------------------------------------- 349ebc09c36SJim Ingham 3501f0f5b5bSZachary Turner static OptionDefinition g_alias_options[] = { 3511f0f5b5bSZachary Turner // clang-format off 3521f0f5b5bSZachary Turner { LLDB_OPT_SET_ALL, false, "help", 'h', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeHelpText, "Help text for this command" }, 3531f0f5b5bSZachary Turner { LLDB_OPT_SET_ALL, false, "long-help", 'H', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeHelpText, "Long help text for this command" }, 3541f0f5b5bSZachary Turner // clang-format on 3551f0f5b5bSZachary Turner }; 3561f0f5b5bSZachary Turner 357b9c1b51eSKate Stone static const char *g_python_command_instructions = 358b9c1b51eSKate Stone "Enter your Python command(s). Type 'DONE' to end.\n" 359be93a35aSEnrico Granata "You must define a Python function with this signature:\n" 36044d93782SGreg Clayton "def my_command_impl(debugger, args, result, internal_dict):\n"; 361be93a35aSEnrico Granata 362b9c1b51eSKate Stone class CommandObjectCommandsAlias : public CommandObjectRaw { 36345d0e238SEnrico Granata protected: 364b9c1b51eSKate Stone class CommandOptions : public OptionGroup { 365ebc09c36SJim Ingham public: 366b9c1b51eSKate Stone CommandOptions() : OptionGroup(), m_help(), m_long_help() {} 36745d0e238SEnrico Granata 36845d0e238SEnrico Granata ~CommandOptions() override = default; 36945d0e238SEnrico Granata 3701f0f5b5bSZachary Turner llvm::ArrayRef<OptionDefinition> GetDefinitions() override { 37170602439SZachary Turner return llvm::makeArrayRef(g_alias_options); 3721f0f5b5bSZachary Turner } 37345d0e238SEnrico Granata 3748cef4b0bSZachary Turner Error SetOptionValue(uint32_t option_idx, llvm::StringRef option_value, 375b9c1b51eSKate Stone ExecutionContext *execution_context) override { 37645d0e238SEnrico Granata Error error; 37745d0e238SEnrico Granata 3781f0f5b5bSZachary Turner const int short_option = GetDefinitions()[option_idx].short_option; 3798cef4b0bSZachary Turner std::string option_str(option_value); 38045d0e238SEnrico Granata 381b9c1b51eSKate Stone switch (short_option) { 38245d0e238SEnrico Granata case 'h': 3838cef4b0bSZachary Turner m_help.SetCurrentValue(option_str); 38445d0e238SEnrico Granata m_help.SetOptionWasSet(); 38545d0e238SEnrico Granata break; 38645d0e238SEnrico Granata 38745d0e238SEnrico Granata case 'H': 3888cef4b0bSZachary Turner m_long_help.SetCurrentValue(option_str); 38945d0e238SEnrico Granata m_long_help.SetOptionWasSet(); 39045d0e238SEnrico Granata break; 39145d0e238SEnrico Granata 39245d0e238SEnrico Granata default: 393b9c1b51eSKate Stone error.SetErrorStringWithFormat("invalid short option character '%c'", 394b9c1b51eSKate Stone short_option); 39545d0e238SEnrico Granata break; 39645d0e238SEnrico Granata } 39745d0e238SEnrico Granata 39845d0e238SEnrico Granata return error; 39945d0e238SEnrico Granata } 40045d0e238SEnrico Granata 401b9c1b51eSKate Stone void OptionParsingStarting(ExecutionContext *execution_context) override { 40245d0e238SEnrico Granata m_help.Clear(); 40345d0e238SEnrico Granata m_long_help.Clear(); 40445d0e238SEnrico Granata } 40545d0e238SEnrico Granata 40645d0e238SEnrico Granata OptionValueString m_help; 40745d0e238SEnrico Granata OptionValueString m_long_help; 40845d0e238SEnrico Granata }; 40945d0e238SEnrico Granata 41045d0e238SEnrico Granata OptionGroupOptions m_option_group; 41145d0e238SEnrico Granata CommandOptions m_command_options; 41245d0e238SEnrico Granata 41345d0e238SEnrico Granata public: 414b9c1b51eSKate Stone Options *GetOptions() override { return &m_option_group; } 41545d0e238SEnrico Granata 4167428a18cSKate Stone CommandObjectCommandsAlias(CommandInterpreter &interpreter) 417b9c1b51eSKate Stone : CommandObjectRaw( 418b9c1b51eSKate Stone interpreter, "command alias", 419a449698cSZachary Turner "Define a custom command in terms of an existing command."), 420b9c1b51eSKate Stone m_option_group(), m_command_options() { 42145d0e238SEnrico Granata m_option_group.Append(&m_command_options); 42245d0e238SEnrico Granata m_option_group.Finalize(); 42345d0e238SEnrico Granata 424ebc09c36SJim Ingham SetHelpLong( 425ea671fbdSKate Stone "'alias' allows the user to create a short-cut or abbreviation for long \ 426ea671fbdSKate Stone commands, multi-word commands, and commands that take particular options. \ 427b9c1b51eSKate Stone Below are some simple examples of how one might use the 'alias' command:" 428b9c1b51eSKate Stone R"( 429ea671fbdSKate Stone 430ea671fbdSKate Stone (lldb) command alias sc script 431ea671fbdSKate Stone 432ea671fbdSKate Stone Creates the abbreviation 'sc' for the 'script' command. 433ea671fbdSKate Stone 434ea671fbdSKate Stone (lldb) command alias bp breakpoint 435ea671fbdSKate Stone 436b9c1b51eSKate Stone )" 437b9c1b51eSKate Stone " Creates the abbreviation 'bp' for the 'breakpoint' command. Since \ 438ea671fbdSKate Stone breakpoint commands are two-word commands, the user would still need to \ 439b9c1b51eSKate Stone enter the second word after 'bp', e.g. 'bp enable' or 'bp delete'." 440b9c1b51eSKate Stone R"( 441ea671fbdSKate Stone 442ea671fbdSKate Stone (lldb) command alias bpl breakpoint list 443ea671fbdSKate Stone 444ea671fbdSKate Stone Creates the abbreviation 'bpl' for the two-word command 'breakpoint list'. 445ea671fbdSKate Stone 446b9c1b51eSKate Stone )" 447b9c1b51eSKate Stone "An alias can include some options for the command, with the values either \ 448ea671fbdSKate Stone filled in at the time the alias is created, or specified as positional \ 449ea671fbdSKate Stone arguments, to be filled in when the alias is invoked. The following example \ 450b9c1b51eSKate Stone shows how to create aliases with options:" 451b9c1b51eSKate Stone R"( 452ea671fbdSKate Stone 453ea671fbdSKate Stone (lldb) command alias bfl breakpoint set -f %1 -l %2 454ea671fbdSKate Stone 455b9c1b51eSKate Stone )" 456b9c1b51eSKate Stone " Creates the abbreviation 'bfl' (for break-file-line), with the -f and -l \ 457ea671fbdSKate Stone options already part of the alias. So if the user wants to set a breakpoint \ 458ea671fbdSKate Stone by file and line without explicitly having to use the -f and -l options, the \ 459ea671fbdSKate Stone user can now use 'bfl' instead. The '%1' and '%2' are positional placeholders \ 460ea671fbdSKate Stone for the actual arguments that will be passed when the alias command is used. \ 461ea671fbdSKate Stone The number in the placeholder refers to the position/order the actual value \ 462ea671fbdSKate Stone occupies when the alias is used. All the occurrences of '%1' in the alias \ 463ea671fbdSKate Stone will be replaced with the first argument, all the occurrences of '%2' in the \ 464ea671fbdSKate Stone alias will be replaced with the second argument, and so on. This also allows \ 465ea671fbdSKate Stone actual arguments to be used multiple times within an alias (see 'process \ 466b9c1b51eSKate Stone launch' example below)." 467b9c1b51eSKate Stone R"( 468ea671fbdSKate Stone 469b9c1b51eSKate Stone )" 470b9c1b51eSKate Stone "Note: the positional arguments must substitute as whole words in the resultant \ 471ea671fbdSKate Stone command, so you can't at present do something like this to append the file extension \ 472b9c1b51eSKate Stone \".cpp\":" 473b9c1b51eSKate Stone R"( 474ea671fbdSKate Stone 475ea671fbdSKate Stone (lldb) command alias bcppfl breakpoint set -f %1.cpp -l %2 476ea671fbdSKate Stone 477b9c1b51eSKate Stone )" 478b9c1b51eSKate Stone "For more complex aliasing, use the \"command regex\" command instead. In the \ 479ea671fbdSKate Stone 'bfl' case above, the actual file value will be filled in with the first argument \ 480ea671fbdSKate Stone following 'bfl' and the actual line number value will be filled in with the second \ 481b9c1b51eSKate Stone argument. The user would use this alias as follows:" 482b9c1b51eSKate Stone R"( 483ea671fbdSKate Stone 484ea671fbdSKate Stone (lldb) command alias bfl breakpoint set -f %1 -l %2 485ea671fbdSKate Stone (lldb) bfl my-file.c 137 486ea671fbdSKate Stone 487ea671fbdSKate Stone This would be the same as if the user had entered 'breakpoint set -f my-file.c -l 137'. 488ea671fbdSKate Stone 489ea671fbdSKate Stone Another example: 490ea671fbdSKate Stone 491ea671fbdSKate Stone (lldb) command alias pltty process launch -s -o %1 -e %1 492ea671fbdSKate Stone (lldb) pltty /dev/tty0 493ea671fbdSKate Stone 494ea671fbdSKate Stone Interpreted as 'process launch -s -o /dev/tty0 -e /dev/tty0' 495ea671fbdSKate Stone 496b9c1b51eSKate Stone )" 497b9c1b51eSKate Stone "If the user always wanted to pass the same value to a particular option, the \ 498ea671fbdSKate Stone alias could be defined with that value directly in the alias as a constant, \ 499b9c1b51eSKate Stone rather than using a positional placeholder:" 500b9c1b51eSKate Stone R"( 501ea671fbdSKate Stone 502ea671fbdSKate Stone (lldb) command alias bl3 breakpoint set -f %1 -l 3 503ea671fbdSKate Stone 504b9c1b51eSKate Stone Always sets a breakpoint on line 3 of whatever file is indicated.)"); 505ebc09c36SJim Ingham 506405fe67fSCaroline Tice CommandArgumentEntry arg1; 507405fe67fSCaroline Tice CommandArgumentEntry arg2; 508405fe67fSCaroline Tice CommandArgumentEntry arg3; 509405fe67fSCaroline Tice CommandArgumentData alias_arg; 510405fe67fSCaroline Tice CommandArgumentData cmd_arg; 511405fe67fSCaroline Tice CommandArgumentData options_arg; 512405fe67fSCaroline Tice 513405fe67fSCaroline Tice // Define the first (and only) variant of this arg. 514405fe67fSCaroline Tice alias_arg.arg_type = eArgTypeAliasName; 515405fe67fSCaroline Tice alias_arg.arg_repetition = eArgRepeatPlain; 516405fe67fSCaroline Tice 517b9c1b51eSKate Stone // There is only one variant this argument could be; put it into the 518b9c1b51eSKate Stone // argument entry. 519405fe67fSCaroline Tice arg1.push_back(alias_arg); 520405fe67fSCaroline Tice 521405fe67fSCaroline Tice // Define the first (and only) variant of this arg. 522405fe67fSCaroline Tice cmd_arg.arg_type = eArgTypeCommandName; 523405fe67fSCaroline Tice cmd_arg.arg_repetition = eArgRepeatPlain; 524405fe67fSCaroline Tice 525b9c1b51eSKate Stone // There is only one variant this argument could be; put it into the 526b9c1b51eSKate Stone // argument entry. 527405fe67fSCaroline Tice arg2.push_back(cmd_arg); 528405fe67fSCaroline Tice 529405fe67fSCaroline Tice // Define the first (and only) variant of this arg. 530405fe67fSCaroline Tice options_arg.arg_type = eArgTypeAliasOptions; 531405fe67fSCaroline Tice options_arg.arg_repetition = eArgRepeatOptional; 532405fe67fSCaroline Tice 533b9c1b51eSKate Stone // There is only one variant this argument could be; put it into the 534b9c1b51eSKate Stone // argument entry. 535405fe67fSCaroline Tice arg3.push_back(options_arg); 536405fe67fSCaroline Tice 537405fe67fSCaroline Tice // Push the data for the first argument into the m_arguments vector. 538405fe67fSCaroline Tice m_arguments.push_back(arg1); 539405fe67fSCaroline Tice m_arguments.push_back(arg2); 540405fe67fSCaroline Tice m_arguments.push_back(arg3); 541ebc09c36SJim Ingham } 542ebc09c36SJim Ingham 5436e3d8e7fSEugene Zelenko ~CommandObjectCommandsAlias() override = default; 544ebc09c36SJim Ingham 5455a988416SJim Ingham protected: 546b9c1b51eSKate Stone bool DoExecute(const char *raw_command_line, 547b9c1b51eSKate Stone CommandReturnObject &result) override { 548b9c1b51eSKate Stone if (!raw_command_line || !raw_command_line[0]) { 549d72e412fSEnrico Granata result.AppendError("'command alias' requires at least two arguments"); 55045d0e238SEnrico Granata return false; 55145d0e238SEnrico Granata } 55245d0e238SEnrico Granata 553e1cfbc79STodd Fiala ExecutionContext exe_ctx = GetCommandInterpreter().GetExecutionContext(); 554e1cfbc79STodd Fiala m_option_group.NotifyOptionParsingStarting(&exe_ctx); 55545d0e238SEnrico Granata 55645d0e238SEnrico Granata const char *remainder = nullptr; 55745d0e238SEnrico Granata 558b9c1b51eSKate Stone if (raw_command_line[0] == '-') { 55945d0e238SEnrico Granata // We have some options and these options MUST end with --. 56045d0e238SEnrico Granata const char *end_options = nullptr; 56145d0e238SEnrico Granata const char *s = raw_command_line; 562b9c1b51eSKate Stone while (s && s[0]) { 56345d0e238SEnrico Granata end_options = ::strstr(s, "--"); 564b9c1b51eSKate Stone if (end_options) { 56545d0e238SEnrico Granata end_options += 2; // Get past the "--" 566b9c1b51eSKate Stone if (::isspace(end_options[0])) { 56745d0e238SEnrico Granata remainder = end_options; 56845d0e238SEnrico Granata while (::isspace(*remainder)) 56945d0e238SEnrico Granata ++remainder; 57045d0e238SEnrico Granata break; 57145d0e238SEnrico Granata } 57245d0e238SEnrico Granata } 57345d0e238SEnrico Granata s = end_options; 57445d0e238SEnrico Granata } 57545d0e238SEnrico Granata 576b9c1b51eSKate Stone if (end_options) { 577b9c1b51eSKate Stone Args args( 578b9c1b51eSKate Stone llvm::StringRef(raw_command_line, end_options - raw_command_line)); 57945d0e238SEnrico Granata if (!ParseOptions(args, result)) 58045d0e238SEnrico Granata return false; 58145d0e238SEnrico Granata 582e1cfbc79STodd Fiala Error error(m_option_group.NotifyOptionParsingFinished(&exe_ctx)); 583b9c1b51eSKate Stone if (error.Fail()) { 58445d0e238SEnrico Granata result.AppendError(error.AsCString()); 58545d0e238SEnrico Granata result.SetStatus(eReturnStatusFailed); 58645d0e238SEnrico Granata return false; 58745d0e238SEnrico Granata } 58845d0e238SEnrico Granata } 58945d0e238SEnrico Granata } 59045d0e238SEnrico Granata if (nullptr == remainder) 59145d0e238SEnrico Granata remainder = raw_command_line; 59245d0e238SEnrico Granata 593a01bccdbSZachary Turner llvm::StringRef raw_command_string(remainder); 594a01bccdbSZachary Turner Args args(raw_command_string); 595844d2303SCaroline Tice 59611eb9c64SZachary Turner if (args.GetArgumentCount() < 2) { 597d72e412fSEnrico Granata result.AppendError("'command alias' requires at least two arguments"); 598844d2303SCaroline Tice result.SetStatus(eReturnStatusFailed); 599844d2303SCaroline Tice return false; 600844d2303SCaroline Tice } 601844d2303SCaroline Tice 602844d2303SCaroline Tice // Get the alias command. 603844d2303SCaroline Tice 6044574a890SZachary Turner auto alias_command = args[0].ref; 6054574a890SZachary Turner if (alias_command.startswith("-")) { 606d72e412fSEnrico Granata result.AppendError("aliases starting with a dash are not supported"); 607b9c1b51eSKate Stone if (alias_command == "--help" || alias_command == "--long-help") { 608b9c1b51eSKate Stone result.AppendWarning("if trying to pass options to 'command alias' add " 609b9c1b51eSKate Stone "a -- at the end of the options"); 610d72e412fSEnrico Granata } 611d72e412fSEnrico Granata result.SetStatus(eReturnStatusFailed); 612d72e412fSEnrico Granata return false; 613d72e412fSEnrico Granata } 614844d2303SCaroline Tice 615b9c1b51eSKate Stone // Strip the new alias name off 'raw_command_string' (leave it on args, 616b9c1b51eSKate Stone // which gets passed to 'Execute', which 617844d2303SCaroline Tice // does the stripping itself. 618844d2303SCaroline Tice size_t pos = raw_command_string.find(alias_command); 619b9c1b51eSKate Stone if (pos == 0) { 620844d2303SCaroline Tice raw_command_string = raw_command_string.substr(alias_command.size()); 621844d2303SCaroline Tice pos = raw_command_string.find_first_not_of(' '); 622844d2303SCaroline Tice if ((pos != std::string::npos) && (pos > 0)) 623844d2303SCaroline Tice raw_command_string = raw_command_string.substr(pos); 624b9c1b51eSKate Stone } else { 625844d2303SCaroline Tice result.AppendError("Error parsing command string. No alias created."); 626844d2303SCaroline Tice result.SetStatus(eReturnStatusFailed); 627844d2303SCaroline Tice return false; 628844d2303SCaroline Tice } 629844d2303SCaroline Tice 630844d2303SCaroline Tice // Verify that the command is alias-able. 631771ef6d4SMalcolm Parsons if (m_interpreter.CommandExists(alias_command)) { 632b9c1b51eSKate Stone result.AppendErrorWithFormat( 633b9c1b51eSKate Stone "'%s' is a permanent debugger command and cannot be redefined.\n", 6344574a890SZachary Turner args[0].c_str()); 635844d2303SCaroline Tice result.SetStatus(eReturnStatusFailed); 636844d2303SCaroline Tice return false; 637844d2303SCaroline Tice } 638844d2303SCaroline Tice 639b9c1b51eSKate Stone // Get CommandObject that is being aliased. The command name is read from 640a01bccdbSZachary Turner // the front of raw_command_string. raw_command_string is returned with the 641a01bccdbSZachary Turner // name of the command object stripped off the front. 642a01bccdbSZachary Turner llvm::StringRef original_raw_command_string = raw_command_string; 643b9c1b51eSKate Stone CommandObject *cmd_obj = 644b9c1b51eSKate Stone m_interpreter.GetCommandObjectForCommand(raw_command_string); 645844d2303SCaroline Tice 646b9c1b51eSKate Stone if (!cmd_obj) { 647b9c1b51eSKate Stone result.AppendErrorWithFormat("invalid command given to 'command alias'. " 648b9c1b51eSKate Stone "'%s' does not begin with a valid command." 649b9c1b51eSKate Stone " No alias created.", 650a01bccdbSZachary Turner original_raw_command_string.str().c_str()); 651844d2303SCaroline Tice result.SetStatus(eReturnStatusFailed); 652844d2303SCaroline Tice return false; 653b9c1b51eSKate Stone } else if (!cmd_obj->WantsRawCommandString()) { 654b9c1b51eSKate Stone // Note that args was initialized with the original command, and has not 655b9c1b51eSKate Stone // been updated to this point. 656b9c1b51eSKate Stone // Therefore can we pass it to the version of Execute that does not 657b9c1b51eSKate Stone // need/expect raw input in the alias. 6585a988416SJim Ingham return HandleAliasingNormalCommand(args, result); 659b9c1b51eSKate Stone } else { 660b9c1b51eSKate Stone return HandleAliasingRawCommand(alias_command, raw_command_string, 661b9c1b51eSKate Stone *cmd_obj, result); 6625a988416SJim Ingham } 6635a988416SJim Ingham return result.Succeeded(); 6645a988416SJim Ingham } 6655a988416SJim Ingham 666a01bccdbSZachary Turner bool HandleAliasingRawCommand(llvm::StringRef alias_command, 667a01bccdbSZachary Turner llvm::StringRef raw_command_string, 668b9c1b51eSKate Stone CommandObject &cmd_obj, 669b9c1b51eSKate Stone CommandReturnObject &result) { 670844d2303SCaroline Tice // Verify & handle any options/arguments passed to the alias command 671844d2303SCaroline Tice 672b9c1b51eSKate Stone OptionArgVectorSP option_arg_vector_sp = 673b9c1b51eSKate Stone OptionArgVectorSP(new OptionArgVector); 674844d2303SCaroline Tice 675b9c1b51eSKate Stone if (CommandObjectSP cmd_obj_sp = 676b9c1b51eSKate Stone m_interpreter.GetCommandSPExact(cmd_obj.GetCommandName(), false)) { 677a01bccdbSZachary Turner if (m_interpreter.AliasExists(alias_command) || 678a01bccdbSZachary Turner m_interpreter.UserCommandExists(alias_command)) { 679b9c1b51eSKate Stone result.AppendWarningWithFormat( 680b9c1b51eSKate Stone "Overwriting existing definition for '%s'.\n", 681a01bccdbSZachary Turner alias_command.str().c_str()); 682844d2303SCaroline Tice } 683b9c1b51eSKate Stone if (CommandAlias *alias = m_interpreter.AddAlias( 684a01bccdbSZachary Turner alias_command, cmd_obj_sp, raw_command_string)) { 68545d0e238SEnrico Granata if (m_command_options.m_help.OptionWasSet()) 68645d0e238SEnrico Granata alias->SetHelp(m_command_options.m_help.GetCurrentValue()); 68745d0e238SEnrico Granata if (m_command_options.m_long_help.OptionWasSet()) 68845d0e238SEnrico Granata alias->SetHelpLong(m_command_options.m_long_help.GetCurrentValue()); 689844d2303SCaroline Tice result.SetStatus(eReturnStatusSuccessFinishNoResult); 690b9c1b51eSKate Stone } else { 691472362e6SCaroline Tice result.AppendError("Unable to create requested alias.\n"); 692472362e6SCaroline Tice result.SetStatus(eReturnStatusFailed); 693472362e6SCaroline Tice } 694212130acSEnrico Granata 695b9c1b51eSKate Stone } else { 696212130acSEnrico Granata result.AppendError("Unable to create requested alias.\n"); 697212130acSEnrico Granata result.SetStatus(eReturnStatusFailed); 698212130acSEnrico Granata } 699212130acSEnrico Granata 700844d2303SCaroline Tice return result.Succeeded(); 701844d2303SCaroline Tice } 702ebc09c36SJim Ingham 703b9c1b51eSKate Stone bool HandleAliasingNormalCommand(Args &args, CommandReturnObject &result) { 704867b185dSCaroline Tice size_t argc = args.GetArgumentCount(); 705ebc09c36SJim Ingham 706b9c1b51eSKate Stone if (argc < 2) { 707d72e412fSEnrico Granata result.AppendError("'command alias' requires at least two arguments"); 708ebc09c36SJim Ingham result.SetStatus(eReturnStatusFailed); 709ebc09c36SJim Ingham return false; 710ebc09c36SJim Ingham } 711ebc09c36SJim Ingham 7124574a890SZachary Turner // Save these in std::strings since we're going to shift them off. 7134574a890SZachary Turner const std::string alias_command(args[0].ref); 7144574a890SZachary Turner const std::string actual_command(args[1].ref); 715ebc09c36SJim Ingham 716ebc09c36SJim Ingham args.Shift(); // Shift the alias command word off the argument vector. 717ebc09c36SJim Ingham args.Shift(); // Shift the old command word off the argument vector. 718ebc09c36SJim Ingham 719b9c1b51eSKate Stone // Verify that the command is alias'able, and get the appropriate command 720b9c1b51eSKate Stone // object. 721ebc09c36SJim Ingham 722771ef6d4SMalcolm Parsons if (m_interpreter.CommandExists(alias_command)) { 723b9c1b51eSKate Stone result.AppendErrorWithFormat( 724b9c1b51eSKate Stone "'%s' is a permanent debugger command and cannot be redefined.\n", 725ebc09c36SJim Ingham alias_command.c_str()); 726ebc09c36SJim Ingham result.SetStatus(eReturnStatusFailed); 7274574a890SZachary Turner return false; 7284574a890SZachary Turner } 7294574a890SZachary Turner 730b9c1b51eSKate Stone CommandObjectSP command_obj_sp( 731a449698cSZachary Turner m_interpreter.GetCommandSPExact(actual_command, true)); 732ebc09c36SJim Ingham CommandObjectSP subcommand_obj_sp; 733ebc09c36SJim Ingham bool use_subcommand = false; 7344574a890SZachary Turner if (!command_obj_sp) { 7354574a890SZachary Turner result.AppendErrorWithFormat("'%s' is not an existing command.\n", 7364574a890SZachary Turner actual_command.c_str()); 7374574a890SZachary Turner result.SetStatus(eReturnStatusFailed); 7384574a890SZachary Turner return false; 7394574a890SZachary Turner } 740ebc09c36SJim Ingham CommandObject *cmd_obj = command_obj_sp.get(); 7416e3d8e7fSEugene Zelenko CommandObject *sub_cmd_obj = nullptr; 742b9c1b51eSKate Stone OptionArgVectorSP option_arg_vector_sp = 743b9c1b51eSKate Stone OptionArgVectorSP(new OptionArgVector); 744ebc09c36SJim Ingham 74511eb9c64SZachary Turner while (cmd_obj->IsMultiwordObject() && !args.empty()) { 7464574a890SZachary Turner auto sub_command = args[0].ref; 74711eb9c64SZachary Turner assert(!sub_command.empty()); 7484574a890SZachary Turner subcommand_obj_sp = cmd_obj->GetSubcommandSP(sub_command); 7494574a890SZachary Turner if (!subcommand_obj_sp) { 750b9c1b51eSKate Stone result.AppendErrorWithFormat( 751b9c1b51eSKate Stone "'%s' is not a valid sub-command of '%s'. " 752f415eeb4SCaroline Tice "Unable to create alias.\n", 7534574a890SZachary Turner args[0].c_str(), actual_command.c_str()); 754ebc09c36SJim Ingham result.SetStatus(eReturnStatusFailed); 755ebc09c36SJim Ingham return false; 756ebc09c36SJim Ingham } 7574574a890SZachary Turner 7584574a890SZachary Turner sub_cmd_obj = subcommand_obj_sp.get(); 7594574a890SZachary Turner use_subcommand = true; 7604574a890SZachary Turner args.Shift(); // Shift the sub_command word off the argument vector. 7614574a890SZachary Turner cmd_obj = sub_cmd_obj; 762ebc09c36SJim Ingham } 763ebc09c36SJim Ingham 764ebc09c36SJim Ingham // Verify & handle any options/arguments passed to the alias command 765ebc09c36SJim Ingham 766212130acSEnrico Granata std::string args_string; 767212130acSEnrico Granata 76811eb9c64SZachary Turner if (!args.empty()) { 769b9c1b51eSKate Stone CommandObjectSP tmp_sp = 770b9c1b51eSKate Stone m_interpreter.GetCommandSPExact(cmd_obj->GetCommandName(), false); 771ebc09c36SJim Ingham if (use_subcommand) 7724574a890SZachary Turner tmp_sp = m_interpreter.GetCommandSPExact(sub_cmd_obj->GetCommandName(), 7734574a890SZachary Turner false); 774ca90c47eSCaroline Tice 775ca90c47eSCaroline Tice args.GetCommandString(args_string); 776867b185dSCaroline Tice } 777ebc09c36SJim Ingham 778771ef6d4SMalcolm Parsons if (m_interpreter.AliasExists(alias_command) || 779771ef6d4SMalcolm Parsons m_interpreter.UserCommandExists(alias_command)) { 780b9c1b51eSKate Stone result.AppendWarningWithFormat( 7814574a890SZachary Turner "Overwriting existing definition for '%s'.\n", alias_command.c_str()); 782ebc09c36SJim Ingham } 783ebc09c36SJim Ingham 784b9c1b51eSKate Stone if (CommandAlias *alias = m_interpreter.AddAlias( 7854574a890SZachary Turner alias_command, use_subcommand ? subcommand_obj_sp : command_obj_sp, 786771ef6d4SMalcolm Parsons args_string)) { 78745d0e238SEnrico Granata if (m_command_options.m_help.OptionWasSet()) 78845d0e238SEnrico Granata alias->SetHelp(m_command_options.m_help.GetCurrentValue()); 78945d0e238SEnrico Granata if (m_command_options.m_long_help.OptionWasSet()) 79045d0e238SEnrico Granata alias->SetHelpLong(m_command_options.m_long_help.GetCurrentValue()); 791ebc09c36SJim Ingham result.SetStatus(eReturnStatusSuccessFinishNoResult); 792b9c1b51eSKate Stone } else { 793212130acSEnrico Granata result.AppendError("Unable to create requested alias.\n"); 794212130acSEnrico Granata result.SetStatus(eReturnStatusFailed); 795212130acSEnrico Granata return false; 796212130acSEnrico Granata } 797ebc09c36SJim Ingham 798ebc09c36SJim Ingham return result.Succeeded(); 799ebc09c36SJim Ingham } 800ebc09c36SJim Ingham }; 801ebc09c36SJim Ingham 802ebc09c36SJim Ingham #pragma mark CommandObjectCommandsUnalias 803ebc09c36SJim Ingham //------------------------------------------------------------------------- 804ebc09c36SJim Ingham // CommandObjectCommandsUnalias 805ebc09c36SJim Ingham //------------------------------------------------------------------------- 806ebc09c36SJim Ingham 807b9c1b51eSKate Stone class CommandObjectCommandsUnalias : public CommandObjectParsed { 808ebc09c36SJim Ingham public: 8097428a18cSKate Stone CommandObjectCommandsUnalias(CommandInterpreter &interpreter) 810b9c1b51eSKate Stone : CommandObjectParsed( 811b9c1b51eSKate Stone interpreter, "command unalias", 812b9c1b51eSKate Stone "Delete one or more custom commands defined by 'command alias'.", 813b9c1b51eSKate Stone nullptr) { 814405fe67fSCaroline Tice CommandArgumentEntry arg; 815405fe67fSCaroline Tice CommandArgumentData alias_arg; 816405fe67fSCaroline Tice 817405fe67fSCaroline Tice // Define the first (and only) variant of this arg. 818405fe67fSCaroline Tice alias_arg.arg_type = eArgTypeAliasName; 819405fe67fSCaroline Tice alias_arg.arg_repetition = eArgRepeatPlain; 820405fe67fSCaroline Tice 821b9c1b51eSKate Stone // There is only one variant this argument could be; put it into the 822b9c1b51eSKate Stone // argument entry. 823405fe67fSCaroline Tice arg.push_back(alias_arg); 824405fe67fSCaroline Tice 825405fe67fSCaroline Tice // Push the data for the first argument into the m_arguments vector. 826405fe67fSCaroline Tice m_arguments.push_back(arg); 827ebc09c36SJim Ingham } 828ebc09c36SJim Ingham 8296e3d8e7fSEugene Zelenko ~CommandObjectCommandsUnalias() override = default; 830ebc09c36SJim Ingham 8315a988416SJim Ingham protected: 832b9c1b51eSKate Stone bool DoExecute(Args &args, CommandReturnObject &result) override { 833ebc09c36SJim Ingham CommandObject::CommandMap::iterator pos; 834ebc09c36SJim Ingham CommandObject *cmd_obj; 835ebc09c36SJim Ingham 83611eb9c64SZachary Turner if (args.empty()) { 83711eb9c64SZachary Turner result.AppendError("must call 'unalias' with a valid alias"); 83811eb9c64SZachary Turner result.SetStatus(eReturnStatusFailed); 83911eb9c64SZachary Turner return false; 84011eb9c64SZachary Turner } 84111eb9c64SZachary Turner 8424574a890SZachary Turner auto command_name = args[0].ref; 843a7015092SGreg Clayton cmd_obj = m_interpreter.GetCommandObject(command_name); 8444574a890SZachary Turner if (!cmd_obj) { 8454574a890SZachary Turner result.AppendErrorWithFormat( 8464574a890SZachary Turner "'%s' is not a known command.\nTry 'help' to see a " 8474574a890SZachary Turner "current list of commands.\n", 848867e7d17SZachary Turner args[0].c_str()); 8494574a890SZachary Turner result.SetStatus(eReturnStatusFailed); 8504574a890SZachary Turner return false; 8514574a890SZachary Turner } 8524574a890SZachary Turner 853b9c1b51eSKate Stone if (m_interpreter.CommandExists(command_name)) { 854b9c1b51eSKate Stone if (cmd_obj->IsRemovable()) { 855b9c1b51eSKate Stone result.AppendErrorWithFormat( 856b9c1b51eSKate Stone "'%s' is not an alias, it is a debugger command which can be " 857b9c1b51eSKate Stone "removed using the 'command delete' command.\n", 858867e7d17SZachary Turner args[0].c_str()); 859b9c1b51eSKate Stone } else { 860b9c1b51eSKate Stone result.AppendErrorWithFormat( 861b9c1b51eSKate Stone "'%s' is a permanent debugger command and cannot be removed.\n", 862867e7d17SZachary Turner args[0].c_str()); 863b547278cSGreg Clayton } 864ebc09c36SJim Ingham result.SetStatus(eReturnStatusFailed); 8654574a890SZachary Turner return false; 8664574a890SZachary Turner } 8674574a890SZachary Turner 868b9c1b51eSKate Stone if (!m_interpreter.RemoveAlias(command_name)) { 869a7015092SGreg Clayton if (m_interpreter.AliasExists(command_name)) 870b9c1b51eSKate Stone result.AppendErrorWithFormat( 871867e7d17SZachary Turner "Error occurred while attempting to unalias '%s'.\n", 872867e7d17SZachary Turner args[0].c_str()); 873ebc09c36SJim Ingham else 874b9c1b51eSKate Stone result.AppendErrorWithFormat("'%s' is not an existing alias.\n", 875867e7d17SZachary Turner args[0].c_str()); 876ebc09c36SJim Ingham result.SetStatus(eReturnStatusFailed); 8774574a890SZachary Turner return false; 878ebc09c36SJim Ingham } 879ebc09c36SJim Ingham 8804574a890SZachary Turner result.SetStatus(eReturnStatusSuccessFinishNoResult); 881ebc09c36SJim Ingham return result.Succeeded(); 882ebc09c36SJim Ingham } 883ebc09c36SJim Ingham }; 884ebc09c36SJim Ingham 885b547278cSGreg Clayton #pragma mark CommandObjectCommandsDelete 886b547278cSGreg Clayton //------------------------------------------------------------------------- 887b547278cSGreg Clayton // CommandObjectCommandsDelete 888b547278cSGreg Clayton //------------------------------------------------------------------------- 889b547278cSGreg Clayton 890b9c1b51eSKate Stone class CommandObjectCommandsDelete : public CommandObjectParsed { 891b547278cSGreg Clayton public: 8927428a18cSKate Stone CommandObjectCommandsDelete(CommandInterpreter &interpreter) 893b9c1b51eSKate Stone : CommandObjectParsed( 894b9c1b51eSKate Stone interpreter, "command delete", 895b9c1b51eSKate Stone "Delete one or more custom commands defined by 'command regex'.", 896b9c1b51eSKate Stone nullptr) { 897b547278cSGreg Clayton CommandArgumentEntry arg; 898b547278cSGreg Clayton CommandArgumentData alias_arg; 899b547278cSGreg Clayton 900b547278cSGreg Clayton // Define the first (and only) variant of this arg. 901b547278cSGreg Clayton alias_arg.arg_type = eArgTypeCommandName; 902b547278cSGreg Clayton alias_arg.arg_repetition = eArgRepeatPlain; 903b547278cSGreg Clayton 904b9c1b51eSKate Stone // There is only one variant this argument could be; put it into the 905b9c1b51eSKate Stone // argument entry. 906b547278cSGreg Clayton arg.push_back(alias_arg); 907b547278cSGreg Clayton 908b547278cSGreg Clayton // Push the data for the first argument into the m_arguments vector. 909b547278cSGreg Clayton m_arguments.push_back(arg); 910b547278cSGreg Clayton } 911b547278cSGreg Clayton 9126e3d8e7fSEugene Zelenko ~CommandObjectCommandsDelete() override = default; 913b547278cSGreg Clayton 914b547278cSGreg Clayton protected: 915b9c1b51eSKate Stone bool DoExecute(Args &args, CommandReturnObject &result) override { 916b547278cSGreg Clayton CommandObject::CommandMap::iterator pos; 917b547278cSGreg Clayton 91811eb9c64SZachary Turner if (args.empty()) { 91911eb9c64SZachary Turner result.AppendErrorWithFormat("must call '%s' with one or more valid user " 92011eb9c64SZachary Turner "defined regular expression command names", 921a449698cSZachary Turner GetCommandName().str().c_str()); 92211eb9c64SZachary Turner result.SetStatus(eReturnStatusFailed); 92311eb9c64SZachary Turner } 92411eb9c64SZachary Turner 9254574a890SZachary Turner auto command_name = args[0].ref; 9264574a890SZachary Turner if (!m_interpreter.CommandExists(command_name)) { 92746d4aa21SEnrico Granata StreamString error_msg_stream; 92846d4aa21SEnrico Granata const bool generate_apropos = true; 92946d4aa21SEnrico Granata const bool generate_type_lookup = false; 930b9c1b51eSKate Stone CommandObjectHelp::GenerateAdditionalHelpAvenuesMessage( 9314574a890SZachary Turner &error_msg_stream, command_name, llvm::StringRef(), llvm::StringRef(), 9324574a890SZachary Turner generate_apropos, generate_type_lookup); 933c156427dSZachary Turner result.AppendError(error_msg_stream.GetString()); 934b547278cSGreg Clayton result.SetStatus(eReturnStatusFailed); 9354574a890SZachary Turner return false; 936b547278cSGreg Clayton } 937b547278cSGreg Clayton 9384574a890SZachary Turner if (!m_interpreter.RemoveCommand(command_name)) { 9394574a890SZachary Turner result.AppendErrorWithFormat( 9404574a890SZachary Turner "'%s' is a permanent debugger command and cannot be removed.\n", 941867e7d17SZachary Turner args[0].c_str()); 9424574a890SZachary Turner result.SetStatus(eReturnStatusFailed); 9434574a890SZachary Turner return false; 9444574a890SZachary Turner } 9454574a890SZachary Turner 9464574a890SZachary Turner result.SetStatus(eReturnStatusSuccessFinishNoResult); 9474574a890SZachary Turner return true; 948b547278cSGreg Clayton } 949b547278cSGreg Clayton }; 950b547278cSGreg Clayton 951de164aaaSGreg Clayton //------------------------------------------------------------------------- 952de164aaaSGreg Clayton // CommandObjectCommandsAddRegex 953de164aaaSGreg Clayton //------------------------------------------------------------------------- 9541f0f5b5bSZachary Turner 9551f0f5b5bSZachary Turner static OptionDefinition g_regex_options[] = { 9561f0f5b5bSZachary Turner // clang-format off 9571f0f5b5bSZachary Turner { LLDB_OPT_SET_1, false, "help" , 'h', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeNone, "The help text to display for this command." }, 9581f0f5b5bSZachary Turner { LLDB_OPT_SET_1, false, "syntax", 's', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeNone, "A syntax string showing the typical usage syntax." }, 9591f0f5b5bSZachary Turner // clang-format on 9601f0f5b5bSZachary Turner }; 9611f0f5b5bSZachary Turner 9625a988416SJim Ingham #pragma mark CommandObjectCommandsAddRegex 963de164aaaSGreg Clayton 964b9c1b51eSKate Stone class CommandObjectCommandsAddRegex : public CommandObjectParsed, 965b9c1b51eSKate Stone public IOHandlerDelegateMultiline { 966de164aaaSGreg Clayton public: 9677428a18cSKate Stone CommandObjectCommandsAddRegex(CommandInterpreter &interpreter) 968b9c1b51eSKate Stone : CommandObjectParsed( 969b9c1b51eSKate Stone interpreter, "command regex", "Define a custom command in terms of " 970b9c1b51eSKate Stone "existing commands by matching " 971b9c1b51eSKate Stone "regular expressions.", 9720e5e5a79SGreg Clayton "command regex <cmd-name> [s/<regex>/<subst>/ ...]"), 973b9c1b51eSKate Stone IOHandlerDelegateMultiline("", 974b9c1b51eSKate Stone IOHandlerDelegate::Completion::LLDBCommand), 975b9c1b51eSKate Stone m_options() { 976b9c1b51eSKate Stone SetHelpLong( 977b9c1b51eSKate Stone R"( 978b9c1b51eSKate Stone )" 979b9c1b51eSKate Stone "This command allows the user to create powerful regular expression commands \ 980ea671fbdSKate Stone with substitutions. The regular expressions and substitutions are specified \ 981b9c1b51eSKate Stone using the regular expression substitution format of:" 982b9c1b51eSKate Stone R"( 983ea671fbdSKate Stone 984ea671fbdSKate Stone s/<regex>/<subst>/ 985ea671fbdSKate Stone 986b9c1b51eSKate Stone )" 987b9c1b51eSKate Stone "<regex> is a regular expression that can use parenthesis to capture regular \ 988ea671fbdSKate Stone expression input and substitute the captured matches in the output using %1 \ 989b9c1b51eSKate Stone for the first match, %2 for the second, and so on." 990b9c1b51eSKate Stone R"( 991ea671fbdSKate Stone 992b9c1b51eSKate Stone )" 993b9c1b51eSKate Stone "The regular expressions can all be specified on the command line if more than \ 994ea671fbdSKate Stone one argument is provided. If just the command name is provided on the command \ 995ea671fbdSKate Stone line, then the regular expressions and substitutions can be entered on separate \ 996b9c1b51eSKate Stone lines, followed by an empty line to terminate the command definition." 997b9c1b51eSKate Stone R"( 998ea671fbdSKate Stone 999ea671fbdSKate Stone EXAMPLES 1000ea671fbdSKate Stone 1001b9c1b51eSKate Stone )" 1002b9c1b51eSKate Stone "The following example will define a regular expression command named 'f' that \ 1003ea671fbdSKate Stone will call 'finish' if there are no arguments, or 'frame select <frame-idx>' if \ 1004b9c1b51eSKate Stone a number follows 'f':" 1005b9c1b51eSKate Stone R"( 1006ea671fbdSKate Stone 1007b9c1b51eSKate Stone (lldb) command regex f s/^$/finish/ 's/([0-9]+)/frame select %1/')"); 1008de164aaaSGreg Clayton } 1009de164aaaSGreg Clayton 10106e3d8e7fSEugene Zelenko ~CommandObjectCommandsAddRegex() override = default; 1011de164aaaSGreg Clayton 10125a988416SJim Ingham protected: 1013b9c1b51eSKate Stone void IOHandlerActivated(IOHandler &io_handler) override { 101444d93782SGreg Clayton StreamFileSP output_sp(io_handler.GetOutputStreamFile()); 1015b9c1b51eSKate Stone if (output_sp) { 1016b9c1b51eSKate Stone output_sp->PutCString("Enter one of more sed substitution commands in " 1017b9c1b51eSKate Stone "the form: 's/<regex>/<subst>/'.\nTerminate the " 1018b9c1b51eSKate Stone "substitution list with an empty line.\n"); 101944d93782SGreg Clayton output_sp->Flush(); 102044d93782SGreg Clayton } 102144d93782SGreg Clayton } 102244d93782SGreg Clayton 1023b9c1b51eSKate Stone void IOHandlerInputComplete(IOHandler &io_handler, 1024b9c1b51eSKate Stone std::string &data) override { 102544d93782SGreg Clayton io_handler.SetIsDone(true); 1026b9c1b51eSKate Stone if (m_regex_cmd_ap) { 102744d93782SGreg Clayton StringList lines; 1028b9c1b51eSKate Stone if (lines.SplitIntoLines(data)) { 102944d93782SGreg Clayton const size_t num_lines = lines.GetSize(); 103044d93782SGreg Clayton bool check_only = false; 1031b9c1b51eSKate Stone for (size_t i = 0; i < num_lines; ++i) { 103244d93782SGreg Clayton llvm::StringRef bytes_strref(lines[i]); 103344d93782SGreg Clayton Error error = AppendRegexSubstitution(bytes_strref, check_only); 1034b9c1b51eSKate Stone if (error.Fail()) { 1035b9c1b51eSKate Stone if (!m_interpreter.GetDebugger() 1036b9c1b51eSKate Stone .GetCommandInterpreter() 1037b9c1b51eSKate Stone .GetBatchCommandMode()) { 1038b9c1b51eSKate Stone StreamSP out_stream = 1039b9c1b51eSKate Stone m_interpreter.GetDebugger().GetAsyncOutputStream(); 104044d93782SGreg Clayton out_stream->Printf("error: %s\n", error.AsCString()); 104144d93782SGreg Clayton } 104244d93782SGreg Clayton } 104344d93782SGreg Clayton } 104444d93782SGreg Clayton } 1045b9c1b51eSKate Stone if (m_regex_cmd_ap->HasRegexEntries()) { 104644d93782SGreg Clayton CommandObjectSP cmd_sp(m_regex_cmd_ap.release()); 104744d93782SGreg Clayton m_interpreter.AddCommand(cmd_sp->GetCommandName(), cmd_sp, true); 104844d93782SGreg Clayton } 104944d93782SGreg Clayton } 105044d93782SGreg Clayton } 105144d93782SGreg Clayton 1052b9c1b51eSKate Stone bool DoExecute(Args &command, CommandReturnObject &result) override { 10535a988416SJim Ingham const size_t argc = command.GetArgumentCount(); 1054b9c1b51eSKate Stone if (argc == 0) { 1055b9c1b51eSKate Stone result.AppendError("usage: 'command regex <command-name> " 1056b9c1b51eSKate Stone "[s/<regex1>/<subst1>/ s/<regex2>/<subst2>/ ...]'\n"); 10570e5e5a79SGreg Clayton result.SetStatus(eReturnStatusFailed); 105811eb9c64SZachary Turner return false; 105911eb9c64SZachary Turner } 106011eb9c64SZachary Turner 10610e5e5a79SGreg Clayton Error error; 10624574a890SZachary Turner auto name = command[0].ref; 10634574a890SZachary Turner m_regex_cmd_ap = llvm::make_unique<CommandObjectRegexCommand>( 10644574a890SZachary Turner m_interpreter, name, m_options.GetHelp(), m_options.GetSyntax(), 10, 0, 10654574a890SZachary Turner true); 10660e5e5a79SGreg Clayton 1067b9c1b51eSKate Stone if (argc == 1) { 106844d93782SGreg Clayton Debugger &debugger = m_interpreter.GetDebugger(); 1069e30f11d9SKate Stone bool color_prompt = debugger.GetUseColor(); 107044d93782SGreg Clayton const bool multiple_lines = true; // Get multiple lines 1071b9c1b51eSKate Stone IOHandlerSP io_handler_sp(new IOHandlerEditline( 1072b9c1b51eSKate Stone debugger, IOHandler::Type::Other, 107373d80faaSGreg Clayton "lldb-regex", // Name of input reader for history 1074514d8cd8SZachary Turner llvm::StringRef("> "), // Prompt 1075514d8cd8SZachary Turner llvm::StringRef(), // Continuation prompt 1076b9c1b51eSKate Stone multiple_lines, color_prompt, 1077f6913cd7SGreg Clayton 0, // Don't show line numbers 107844d93782SGreg Clayton *this)); 107944d93782SGreg Clayton 1080b9c1b51eSKate Stone if (io_handler_sp) { 108144d93782SGreg Clayton debugger.PushIOHandler(io_handler_sp); 1082de164aaaSGreg Clayton result.SetStatus(eReturnStatusSuccessFinishNoResult); 1083de164aaaSGreg Clayton } 1084b9c1b51eSKate Stone } else { 108597d2c401SZachary Turner for (auto &entry : command.entries().drop_front()) { 108644d93782SGreg Clayton bool check_only = false; 108797d2c401SZachary Turner error = AppendRegexSubstitution(entry.ref, check_only); 10880e5e5a79SGreg Clayton if (error.Fail()) 10890e5e5a79SGreg Clayton break; 10900e5e5a79SGreg Clayton } 10910e5e5a79SGreg Clayton 1092b9c1b51eSKate Stone if (error.Success()) { 10930e5e5a79SGreg Clayton AddRegexCommandToInterpreter(); 10940e5e5a79SGreg Clayton } 10950e5e5a79SGreg Clayton } 1096b9c1b51eSKate Stone if (error.Fail()) { 10970e5e5a79SGreg Clayton result.AppendError(error.AsCString()); 1098de164aaaSGreg Clayton result.SetStatus(eReturnStatusFailed); 1099de164aaaSGreg Clayton } 11000e5e5a79SGreg Clayton 1101de164aaaSGreg Clayton return result.Succeeded(); 1102de164aaaSGreg Clayton } 1103de164aaaSGreg Clayton 1104b9c1b51eSKate Stone Error AppendRegexSubstitution(const llvm::StringRef ®ex_sed, 1105b9c1b51eSKate Stone bool check_only) { 11060e5e5a79SGreg Clayton Error error; 11070e5e5a79SGreg Clayton 1108b9c1b51eSKate Stone if (!m_regex_cmd_ap) { 1109b9c1b51eSKate Stone error.SetErrorStringWithFormat( 1110b9c1b51eSKate Stone "invalid regular expression command object for: '%.*s'", 1111b9c1b51eSKate Stone (int)regex_sed.size(), regex_sed.data()); 11120e5e5a79SGreg Clayton return error; 1113de164aaaSGreg Clayton } 11140e5e5a79SGreg Clayton 11150e5e5a79SGreg Clayton size_t regex_sed_size = regex_sed.size(); 11160e5e5a79SGreg Clayton 1117b9c1b51eSKate Stone if (regex_sed_size <= 1) { 1118b9c1b51eSKate Stone error.SetErrorStringWithFormat( 1119b9c1b51eSKate Stone "regular expression substitution string is too short: '%.*s'", 1120b9c1b51eSKate Stone (int)regex_sed.size(), regex_sed.data()); 11210e5e5a79SGreg Clayton return error; 11220e5e5a79SGreg Clayton } 11230e5e5a79SGreg Clayton 1124b9c1b51eSKate Stone if (regex_sed[0] != 's') { 1125b9c1b51eSKate Stone error.SetErrorStringWithFormat("regular expression substitution string " 1126b9c1b51eSKate Stone "doesn't start with 's': '%.*s'", 1127b9c1b51eSKate Stone (int)regex_sed.size(), regex_sed.data()); 11280e5e5a79SGreg Clayton return error; 11290e5e5a79SGreg Clayton } 11300e5e5a79SGreg Clayton const size_t first_separator_char_pos = 1; 11310e5e5a79SGreg Clayton // use the char that follows 's' as the regex separator character 11320e5e5a79SGreg Clayton // so we can have "s/<regex>/<subst>/" or "s|<regex>|<subst>|" 11330e5e5a79SGreg Clayton const char separator_char = regex_sed[first_separator_char_pos]; 1134b9c1b51eSKate Stone const size_t second_separator_char_pos = 1135b9c1b51eSKate Stone regex_sed.find(separator_char, first_separator_char_pos + 1); 11360e5e5a79SGreg Clayton 1137b9c1b51eSKate Stone if (second_separator_char_pos == std::string::npos) { 1138b9c1b51eSKate Stone error.SetErrorStringWithFormat( 1139b9c1b51eSKate Stone "missing second '%c' separator char after '%.*s' in '%.*s'", 11400e5e5a79SGreg Clayton separator_char, 11410e5e5a79SGreg Clayton (int)(regex_sed.size() - first_separator_char_pos - 1), 1142ea508635SGreg Clayton regex_sed.data() + (first_separator_char_pos + 1), 1143b9c1b51eSKate Stone (int)regex_sed.size(), regex_sed.data()); 11440e5e5a79SGreg Clayton return error; 11450e5e5a79SGreg Clayton } 11460e5e5a79SGreg Clayton 1147b9c1b51eSKate Stone const size_t third_separator_char_pos = 1148b9c1b51eSKate Stone regex_sed.find(separator_char, second_separator_char_pos + 1); 11490e5e5a79SGreg Clayton 1150b9c1b51eSKate Stone if (third_separator_char_pos == std::string::npos) { 1151b9c1b51eSKate Stone error.SetErrorStringWithFormat( 1152b9c1b51eSKate Stone "missing third '%c' separator char after '%.*s' in '%.*s'", 11530e5e5a79SGreg Clayton separator_char, 11540e5e5a79SGreg Clayton (int)(regex_sed.size() - second_separator_char_pos - 1), 1155ea508635SGreg Clayton regex_sed.data() + (second_separator_char_pos + 1), 1156b9c1b51eSKate Stone (int)regex_sed.size(), regex_sed.data()); 11570e5e5a79SGreg Clayton return error; 11580e5e5a79SGreg Clayton } 11590e5e5a79SGreg Clayton 1160b9c1b51eSKate Stone if (third_separator_char_pos != regex_sed_size - 1) { 11610e5e5a79SGreg Clayton // Make sure that everything that follows the last regex 11620e5e5a79SGreg Clayton // separator char 1163b9c1b51eSKate Stone if (regex_sed.find_first_not_of("\t\n\v\f\r ", 1164b9c1b51eSKate Stone third_separator_char_pos + 1) != 1165b9c1b51eSKate Stone std::string::npos) { 1166b9c1b51eSKate Stone error.SetErrorStringWithFormat( 1167b9c1b51eSKate Stone "extra data found after the '%.*s' regular expression substitution " 1168b9c1b51eSKate Stone "string: '%.*s'", 1169b9c1b51eSKate Stone (int)third_separator_char_pos + 1, regex_sed.data(), 11700e5e5a79SGreg Clayton (int)(regex_sed.size() - third_separator_char_pos - 1), 11710e5e5a79SGreg Clayton regex_sed.data() + (third_separator_char_pos + 1)); 11720e5e5a79SGreg Clayton return error; 11730e5e5a79SGreg Clayton } 1174b9c1b51eSKate Stone } else if (first_separator_char_pos + 1 == second_separator_char_pos) { 1175b9c1b51eSKate Stone error.SetErrorStringWithFormat( 1176b9c1b51eSKate Stone "<regex> can't be empty in 's%c<regex>%c<subst>%c' string: '%.*s'", 1177b9c1b51eSKate Stone separator_char, separator_char, separator_char, (int)regex_sed.size(), 11780e5e5a79SGreg Clayton regex_sed.data()); 11790e5e5a79SGreg Clayton return error; 1180b9c1b51eSKate Stone } else if (second_separator_char_pos + 1 == third_separator_char_pos) { 1181b9c1b51eSKate Stone error.SetErrorStringWithFormat( 1182b9c1b51eSKate Stone "<subst> can't be empty in 's%c<regex>%c<subst>%c' string: '%.*s'", 1183b9c1b51eSKate Stone separator_char, separator_char, separator_char, (int)regex_sed.size(), 11840e5e5a79SGreg Clayton regex_sed.data()); 11850e5e5a79SGreg Clayton return error; 11860e5e5a79SGreg Clayton } 118744d93782SGreg Clayton 1188b9c1b51eSKate Stone if (!check_only) { 1189b9c1b51eSKate Stone std::string regex(regex_sed.substr(first_separator_char_pos + 1, 1190b9c1b51eSKate Stone second_separator_char_pos - 1191b9c1b51eSKate Stone first_separator_char_pos - 1)); 1192b9c1b51eSKate Stone std::string subst(regex_sed.substr(second_separator_char_pos + 1, 1193b9c1b51eSKate Stone third_separator_char_pos - 1194b9c1b51eSKate Stone second_separator_char_pos - 1)); 1195b9c1b51eSKate Stone m_regex_cmd_ap->AddRegexCommand(regex.c_str(), subst.c_str()); 119644d93782SGreg Clayton } 11970e5e5a79SGreg Clayton return error; 1198de164aaaSGreg Clayton } 1199de164aaaSGreg Clayton 1200b9c1b51eSKate Stone void AddRegexCommandToInterpreter() { 1201b9c1b51eSKate Stone if (m_regex_cmd_ap) { 1202b9c1b51eSKate Stone if (m_regex_cmd_ap->HasRegexEntries()) { 1203de164aaaSGreg Clayton CommandObjectSP cmd_sp(m_regex_cmd_ap.release()); 1204de164aaaSGreg Clayton m_interpreter.AddCommand(cmd_sp->GetCommandName(), cmd_sp, true); 1205de164aaaSGreg Clayton } 1206de164aaaSGreg Clayton } 1207de164aaaSGreg Clayton } 1208de164aaaSGreg Clayton 1209de164aaaSGreg Clayton private: 12107b0992d9SGreg Clayton std::unique_ptr<CommandObjectRegexCommand> m_regex_cmd_ap; 1211de164aaaSGreg Clayton 1212b9c1b51eSKate Stone class CommandOptions : public Options { 1213de164aaaSGreg Clayton public: 1214b9c1b51eSKate Stone CommandOptions() : Options() {} 1215de164aaaSGreg Clayton 12166e3d8e7fSEugene Zelenko ~CommandOptions() override = default; 1217de164aaaSGreg Clayton 1218fe11483bSZachary Turner Error SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg, 1219b9c1b51eSKate Stone ExecutionContext *execution_context) override { 1220de164aaaSGreg Clayton Error error; 12213bcdfc0eSGreg Clayton const int short_option = m_getopt_table[option_idx].val; 1222de164aaaSGreg Clayton 1223b9c1b51eSKate Stone switch (short_option) { 1224de164aaaSGreg Clayton case 'h': 1225de164aaaSGreg Clayton m_help.assign(option_arg); 1226de164aaaSGreg Clayton break; 1227de164aaaSGreg Clayton case 's': 1228de164aaaSGreg Clayton m_syntax.assign(option_arg); 1229de164aaaSGreg Clayton break; 1230de164aaaSGreg Clayton default: 1231b9c1b51eSKate Stone error.SetErrorStringWithFormat("unrecognized option '%c'", 1232b9c1b51eSKate Stone short_option); 1233de164aaaSGreg Clayton break; 1234de164aaaSGreg Clayton } 1235de164aaaSGreg Clayton 1236de164aaaSGreg Clayton return error; 1237de164aaaSGreg Clayton } 1238de164aaaSGreg Clayton 1239b9c1b51eSKate Stone void OptionParsingStarting(ExecutionContext *execution_context) override { 1240de164aaaSGreg Clayton m_help.clear(); 1241de164aaaSGreg Clayton m_syntax.clear(); 1242de164aaaSGreg Clayton } 1243de164aaaSGreg Clayton 12441f0f5b5bSZachary Turner llvm::ArrayRef<OptionDefinition> GetDefinitions() override { 124570602439SZachary Turner return llvm::makeArrayRef(g_regex_options); 12461f0f5b5bSZachary Turner } 1247de164aaaSGreg Clayton 124811eb9c64SZachary Turner // TODO: Convert these functions to return StringRefs. 1249b9c1b51eSKate Stone const char *GetHelp() { 12506e3d8e7fSEugene Zelenko return (m_help.empty() ? nullptr : m_help.c_str()); 1251de164aaaSGreg Clayton } 12526e3d8e7fSEugene Zelenko 1253b9c1b51eSKate Stone const char *GetSyntax() { 12546e3d8e7fSEugene Zelenko return (m_syntax.empty() ? nullptr : m_syntax.c_str()); 1255de164aaaSGreg Clayton } 12566e3d8e7fSEugene Zelenko 1257de164aaaSGreg Clayton protected: 12586e3d8e7fSEugene Zelenko // Instance variables to hold the values for command options. 12596e3d8e7fSEugene Zelenko 1260de164aaaSGreg Clayton std::string m_help; 1261de164aaaSGreg Clayton std::string m_syntax; 1262de164aaaSGreg Clayton }; 1263de164aaaSGreg Clayton 1264b9c1b51eSKate Stone Options *GetOptions() override { return &m_options; } 1265de164aaaSGreg Clayton 12665a988416SJim Ingham CommandOptions m_options; 1267de164aaaSGreg Clayton }; 1268de164aaaSGreg Clayton 1269b9c1b51eSKate Stone class CommandObjectPythonFunction : public CommandObjectRaw { 1270223383edSEnrico Granata public: 1271b9c1b51eSKate Stone CommandObjectPythonFunction(CommandInterpreter &interpreter, std::string name, 1272b9c1b51eSKate Stone std::string funct, std::string help, 1273b9c1b51eSKate Stone ScriptedCommandSynchronicity synch) 1274a449698cSZachary Turner : CommandObjectRaw(interpreter, name), 1275b9c1b51eSKate Stone m_function_name(funct), m_synchro(synch), m_fetched_help_long(false) { 1276735152e3SEnrico Granata if (!help.empty()) 1277442f6530SZachary Turner SetHelp(help); 1278b9c1b51eSKate Stone else { 1279735152e3SEnrico Granata StreamString stream; 1280735152e3SEnrico Granata stream.Printf("For more information run 'help %s'", name.c_str()); 1281c156427dSZachary Turner SetHelp(stream.GetString()); 1282735152e3SEnrico Granata } 1283223383edSEnrico Granata } 1284223383edSEnrico Granata 12856e3d8e7fSEugene Zelenko ~CommandObjectPythonFunction() override = default; 1286223383edSEnrico Granata 1287b9c1b51eSKate Stone bool IsRemovable() const override { return true; } 12885a988416SJim Ingham 1289b9c1b51eSKate Stone const std::string &GetFunctionName() { return m_function_name; } 12905a988416SJim Ingham 1291b9c1b51eSKate Stone ScriptedCommandSynchronicity GetSynchronicity() { return m_synchro; } 12925a988416SJim Ingham 1293442f6530SZachary Turner llvm::StringRef GetHelpLong() override { 1294442f6530SZachary Turner if (m_fetched_help_long) 1295442f6530SZachary Turner return CommandObjectRaw::GetHelpLong(); 1296442f6530SZachary Turner 1297fac939e9SEnrico Granata ScriptInterpreter *scripter = m_interpreter.GetScriptInterpreter(); 1298442f6530SZachary Turner if (!scripter) 1299442f6530SZachary Turner return CommandObjectRaw::GetHelpLong(); 1300442f6530SZachary Turner 1301fac939e9SEnrico Granata std::string docstring; 1302442f6530SZachary Turner m_fetched_help_long = 1303442f6530SZachary Turner scripter->GetDocumentationForItem(m_function_name.c_str(), docstring); 1304fac939e9SEnrico Granata if (!docstring.empty()) 1305442f6530SZachary Turner SetHelpLong(docstring); 1306fac939e9SEnrico Granata return CommandObjectRaw::GetHelpLong(); 1307fac939e9SEnrico Granata } 1308fac939e9SEnrico Granata 13095a988416SJim Ingham protected: 1310b9c1b51eSKate Stone bool DoExecute(const char *raw_command_line, 1311b9c1b51eSKate Stone CommandReturnObject &result) override { 1312223383edSEnrico Granata ScriptInterpreter *scripter = m_interpreter.GetScriptInterpreter(); 1313223383edSEnrico Granata 1314223383edSEnrico Granata Error error; 1315223383edSEnrico Granata 131670f11f88SJim Ingham result.SetStatus(eReturnStatusInvalid); 131770f11f88SJim Ingham 1318b9c1b51eSKate Stone if (!scripter || 1319b9c1b51eSKate Stone !scripter->RunScriptBasedCommand(m_function_name.c_str(), 1320b9c1b51eSKate Stone raw_command_line, m_synchro, result, 1321b9c1b51eSKate Stone error, m_exe_ctx)) { 1322223383edSEnrico Granata result.AppendError(error.AsCString()); 1323223383edSEnrico Granata result.SetStatus(eReturnStatusFailed); 1324b9c1b51eSKate Stone } else { 132570f11f88SJim Ingham // Don't change the status if the command already set it... 1326b9c1b51eSKate Stone if (result.GetStatus() == eReturnStatusInvalid) { 1327c156427dSZachary Turner if (result.GetOutputData().empty()) 1328223383edSEnrico Granata result.SetStatus(eReturnStatusSuccessFinishNoResult); 132970f11f88SJim Ingham else 133070f11f88SJim Ingham result.SetStatus(eReturnStatusSuccessFinishResult); 133170f11f88SJim Ingham } 133270f11f88SJim Ingham } 1333223383edSEnrico Granata 1334223383edSEnrico Granata return result.Succeeded(); 1335223383edSEnrico Granata } 1336223383edSEnrico Granata 13376e3d8e7fSEugene Zelenko private: 13386e3d8e7fSEugene Zelenko std::string m_function_name; 13396e3d8e7fSEugene Zelenko ScriptedCommandSynchronicity m_synchro; 13406e3d8e7fSEugene Zelenko bool m_fetched_help_long; 1341223383edSEnrico Granata }; 1342223383edSEnrico Granata 1343b9c1b51eSKate Stone class CommandObjectScriptingObject : public CommandObjectRaw { 13449fe00e52SEnrico Granata public: 13459fe00e52SEnrico Granata CommandObjectScriptingObject(CommandInterpreter &interpreter, 13469fe00e52SEnrico Granata std::string name, 13470641ca1aSZachary Turner StructuredData::GenericSP cmd_obj_sp, 1348b9c1b51eSKate Stone ScriptedCommandSynchronicity synch) 1349a449698cSZachary Turner : CommandObjectRaw(interpreter, name), 1350b9c1b51eSKate Stone m_cmd_obj_sp(cmd_obj_sp), m_synchro(synch), m_fetched_help_short(false), 1351b9c1b51eSKate Stone m_fetched_help_long(false) { 13529fe00e52SEnrico Granata StreamString stream; 13539fe00e52SEnrico Granata stream.Printf("For more information run 'help %s'", name.c_str()); 1354c156427dSZachary Turner SetHelp(stream.GetString()); 1355e87764f2SEnrico Granata if (ScriptInterpreter *scripter = m_interpreter.GetScriptInterpreter()) 1356e87764f2SEnrico Granata GetFlags().Set(scripter->GetFlagsForCommandObject(cmd_obj_sp)); 13579fe00e52SEnrico Granata } 13589fe00e52SEnrico Granata 13596e3d8e7fSEugene Zelenko ~CommandObjectScriptingObject() override = default; 13609fe00e52SEnrico Granata 1361b9c1b51eSKate Stone bool IsRemovable() const override { return true; } 13629fe00e52SEnrico Granata 1363b9c1b51eSKate Stone StructuredData::GenericSP GetImplementingObject() { return m_cmd_obj_sp; } 13649fe00e52SEnrico Granata 1365b9c1b51eSKate Stone ScriptedCommandSynchronicity GetSynchronicity() { return m_synchro; } 13669fe00e52SEnrico Granata 1367442f6530SZachary Turner llvm::StringRef GetHelp() override { 1368442f6530SZachary Turner if (m_fetched_help_short) 1369442f6530SZachary Turner return CommandObjectRaw::GetHelp(); 13706f79bb2dSEnrico Granata ScriptInterpreter *scripter = m_interpreter.GetScriptInterpreter(); 1371442f6530SZachary Turner if (!scripter) 1372442f6530SZachary Turner return CommandObjectRaw::GetHelp(); 13736f79bb2dSEnrico Granata std::string docstring; 1374b9c1b51eSKate Stone m_fetched_help_short = 1375b9c1b51eSKate Stone scripter->GetShortHelpForCommandObject(m_cmd_obj_sp, docstring); 13766f79bb2dSEnrico Granata if (!docstring.empty()) 1377442f6530SZachary Turner SetHelp(docstring); 1378442f6530SZachary Turner 13796f79bb2dSEnrico Granata return CommandObjectRaw::GetHelp(); 13806f79bb2dSEnrico Granata } 13816f79bb2dSEnrico Granata 1382442f6530SZachary Turner llvm::StringRef GetHelpLong() override { 1383442f6530SZachary Turner if (m_fetched_help_long) 1384442f6530SZachary Turner return CommandObjectRaw::GetHelpLong(); 1385442f6530SZachary Turner 13866f79bb2dSEnrico Granata ScriptInterpreter *scripter = m_interpreter.GetScriptInterpreter(); 1387442f6530SZachary Turner if (!scripter) 1388442f6530SZachary Turner return CommandObjectRaw::GetHelpLong(); 1389442f6530SZachary Turner 13906f79bb2dSEnrico Granata std::string docstring; 1391b9c1b51eSKate Stone m_fetched_help_long = 1392b9c1b51eSKate Stone scripter->GetLongHelpForCommandObject(m_cmd_obj_sp, docstring); 13936f79bb2dSEnrico Granata if (!docstring.empty()) 1394442f6530SZachary Turner SetHelpLong(docstring); 13959fe00e52SEnrico Granata return CommandObjectRaw::GetHelpLong(); 13969fe00e52SEnrico Granata } 13979fe00e52SEnrico Granata 13989fe00e52SEnrico Granata protected: 1399b9c1b51eSKate Stone bool DoExecute(const char *raw_command_line, 1400b9c1b51eSKate Stone CommandReturnObject &result) override { 14019fe00e52SEnrico Granata ScriptInterpreter *scripter = m_interpreter.GetScriptInterpreter(); 14029fe00e52SEnrico Granata 14039fe00e52SEnrico Granata Error error; 14049fe00e52SEnrico Granata 14059fe00e52SEnrico Granata result.SetStatus(eReturnStatusInvalid); 14069fe00e52SEnrico Granata 1407b9c1b51eSKate Stone if (!scripter || 1408b9c1b51eSKate Stone !scripter->RunScriptBasedCommand(m_cmd_obj_sp, raw_command_line, 1409b9c1b51eSKate Stone m_synchro, result, error, m_exe_ctx)) { 14109fe00e52SEnrico Granata result.AppendError(error.AsCString()); 14119fe00e52SEnrico Granata result.SetStatus(eReturnStatusFailed); 1412b9c1b51eSKate Stone } else { 14139fe00e52SEnrico Granata // Don't change the status if the command already set it... 1414b9c1b51eSKate Stone if (result.GetStatus() == eReturnStatusInvalid) { 1415c156427dSZachary Turner if (result.GetOutputData().empty()) 14169fe00e52SEnrico Granata result.SetStatus(eReturnStatusSuccessFinishNoResult); 14179fe00e52SEnrico Granata else 14189fe00e52SEnrico Granata result.SetStatus(eReturnStatusSuccessFinishResult); 14199fe00e52SEnrico Granata } 14209fe00e52SEnrico Granata } 14219fe00e52SEnrico Granata 14229fe00e52SEnrico Granata return result.Succeeded(); 14239fe00e52SEnrico Granata } 14249fe00e52SEnrico Granata 14256e3d8e7fSEugene Zelenko private: 14266e3d8e7fSEugene Zelenko StructuredData::GenericSP m_cmd_obj_sp; 14276e3d8e7fSEugene Zelenko ScriptedCommandSynchronicity m_synchro; 14286e3d8e7fSEugene Zelenko bool m_fetched_help_short : 1; 14296e3d8e7fSEugene Zelenko bool m_fetched_help_long : 1; 14309fe00e52SEnrico Granata }; 14319fe00e52SEnrico Granata 1432a9dbf432SEnrico Granata //------------------------------------------------------------------------- 1433a9dbf432SEnrico Granata // CommandObjectCommandsScriptImport 1434a9dbf432SEnrico Granata //------------------------------------------------------------------------- 1435a9dbf432SEnrico Granata 14361f0f5b5bSZachary Turner OptionDefinition g_script_import_options[] = { 14371f0f5b5bSZachary Turner // clang-format off 14381f0f5b5bSZachary Turner { LLDB_OPT_SET_1, false, "allow-reload", 'r', OptionParser::eNoArgument, nullptr, nullptr, 0, eArgTypeNone, "Allow the script to be loaded even if it was already loaded before. This argument exists for backwards compatibility, but reloading is always allowed, whether you specify it or not." }, 14391f0f5b5bSZachary Turner // clang-format on 14401f0f5b5bSZachary Turner }; 14411f0f5b5bSZachary Turner 1442b9c1b51eSKate Stone class CommandObjectCommandsScriptImport : public CommandObjectParsed { 14435a988416SJim Ingham public: 1444b9c1b51eSKate Stone CommandObjectCommandsScriptImport(CommandInterpreter &interpreter) 1445b9c1b51eSKate Stone : CommandObjectParsed(interpreter, "command script import", 1446b9c1b51eSKate Stone "Import a scripting module in LLDB.", nullptr), 1447b9c1b51eSKate Stone m_options() { 14485a988416SJim Ingham CommandArgumentEntry arg1; 14495a988416SJim Ingham CommandArgumentData cmd_arg; 14505a988416SJim Ingham 14515a988416SJim Ingham // Define the first (and only) variant of this arg. 14525a988416SJim Ingham cmd_arg.arg_type = eArgTypeFilename; 14533b00e35bSEnrico Granata cmd_arg.arg_repetition = eArgRepeatPlus; 14545a988416SJim Ingham 1455b9c1b51eSKate Stone // There is only one variant this argument could be; put it into the 1456b9c1b51eSKate Stone // argument entry. 14575a988416SJim Ingham arg1.push_back(cmd_arg); 14585a988416SJim Ingham 14595a988416SJim Ingham // Push the data for the first argument into the m_arguments vector. 14605a988416SJim Ingham m_arguments.push_back(arg1); 14615a988416SJim Ingham } 14625a988416SJim Ingham 14636e3d8e7fSEugene Zelenko ~CommandObjectCommandsScriptImport() override = default; 14645a988416SJim Ingham 1465b9c1b51eSKate Stone int HandleArgumentCompletion(Args &input, int &cursor_index, 14665a988416SJim Ingham int &cursor_char_position, 14675a988416SJim Ingham OptionElementVector &opt_element_vector, 1468b9c1b51eSKate Stone int match_start_point, int max_return_elements, 14695a988416SJim Ingham bool &word_complete, 1470b9c1b51eSKate Stone StringList &matches) override { 1471d6a24757SZachary Turner llvm::StringRef completion_str = input[cursor_index].ref; 1472d6a24757SZachary Turner completion_str = completion_str.take_front(cursor_char_position); 14735a988416SJim Ingham 1474b9c1b51eSKate Stone CommandCompletions::InvokeCommonCompletionCallbacks( 1475b9c1b51eSKate Stone GetCommandInterpreter(), CommandCompletions::eDiskFileCompletion, 1476d6a24757SZachary Turner completion_str, match_start_point, max_return_elements, nullptr, 1477b9c1b51eSKate Stone word_complete, matches); 14785a988416SJim Ingham return matches.GetSize(); 14795a988416SJim Ingham } 14805a988416SJim Ingham 1481b9c1b51eSKate Stone Options *GetOptions() override { return &m_options; } 14825a988416SJim Ingham 14835a988416SJim Ingham protected: 1484b9c1b51eSKate Stone class CommandOptions : public Options { 14850a305db7SEnrico Granata public: 1486b9c1b51eSKate Stone CommandOptions() : Options() {} 14870a305db7SEnrico Granata 14886e3d8e7fSEugene Zelenko ~CommandOptions() override = default; 14890a305db7SEnrico Granata 1490fe11483bSZachary Turner Error SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg, 1491b9c1b51eSKate Stone ExecutionContext *execution_context) override { 14920a305db7SEnrico Granata Error error; 14933bcdfc0eSGreg Clayton const int short_option = m_getopt_table[option_idx].val; 14940a305db7SEnrico Granata 1495b9c1b51eSKate Stone switch (short_option) { 14960a305db7SEnrico Granata case 'r': 14970a305db7SEnrico Granata m_allow_reload = true; 14980a305db7SEnrico Granata break; 14990a305db7SEnrico Granata default: 1500b9c1b51eSKate Stone error.SetErrorStringWithFormat("unrecognized option '%c'", 1501b9c1b51eSKate Stone short_option); 15020a305db7SEnrico Granata break; 15030a305db7SEnrico Granata } 15040a305db7SEnrico Granata 15050a305db7SEnrico Granata return error; 15060a305db7SEnrico Granata } 15070a305db7SEnrico Granata 1508b9c1b51eSKate Stone void OptionParsingStarting(ExecutionContext *execution_context) override { 1509e0c70f1bSEnrico Granata m_allow_reload = true; 15100a305db7SEnrico Granata } 15110a305db7SEnrico Granata 15121f0f5b5bSZachary Turner llvm::ArrayRef<OptionDefinition> GetDefinitions() override { 151370602439SZachary Turner return llvm::makeArrayRef(g_script_import_options); 15141f0f5b5bSZachary Turner } 15150a305db7SEnrico Granata 15160a305db7SEnrico Granata // Instance variables to hold the values for command options. 15170a305db7SEnrico Granata 15180a305db7SEnrico Granata bool m_allow_reload; 15190a305db7SEnrico Granata }; 15200a305db7SEnrico Granata 1521b9c1b51eSKate Stone bool DoExecute(Args &command, CommandReturnObject &result) override { 1522b9c1b51eSKate Stone if (m_interpreter.GetDebugger().GetScriptLanguage() != 1523b9c1b51eSKate Stone lldb::eScriptLanguagePython) { 1524b9c1b51eSKate Stone result.AppendError("only scripting language supported for module " 1525b9c1b51eSKate Stone "importing is currently Python"); 1526a9dbf432SEnrico Granata result.SetStatus(eReturnStatusFailed); 1527a9dbf432SEnrico Granata return false; 1528a9dbf432SEnrico Granata } 1529a9dbf432SEnrico Granata 153011eb9c64SZachary Turner if (command.empty()) { 15313b00e35bSEnrico Granata result.AppendError("command script import needs one or more arguments"); 1532a9dbf432SEnrico Granata result.SetStatus(eReturnStatusFailed); 1533a9dbf432SEnrico Granata return false; 1534a9dbf432SEnrico Granata } 1535a9dbf432SEnrico Granata 153611eb9c64SZachary Turner for (auto &entry : command.entries()) { 1537a9dbf432SEnrico Granata Error error; 1538a9dbf432SEnrico Granata 1539c9d645d3SGreg Clayton const bool init_session = true; 1540b9c1b51eSKate Stone // FIXME: this is necessary because CommandObject::CheckRequirements() 154111eb9c64SZachary Turner // assumes that commands won't ever be recursively invoked, but it's 154211eb9c64SZachary Turner // actually possible to craft a Python script that does other "command 154311eb9c64SZachary Turner // script imports" in __lldb_init_module the real fix is to have recursive 154411eb9c64SZachary Turner // commands possible with a CommandInvocation object separate from the 154511eb9c64SZachary Turner // CommandObject itself, so that recursive command invocations won't stomp 154611eb9c64SZachary Turner // on each other (wrt to execution contents, options, and more) 1547078551c7SEnrico Granata m_exe_ctx.Clear(); 1548b9c1b51eSKate Stone if (m_interpreter.GetScriptInterpreter()->LoadScriptingModule( 154911eb9c64SZachary Turner entry.c_str(), m_options.m_allow_reload, init_session, error)) { 1550a9dbf432SEnrico Granata result.SetStatus(eReturnStatusSuccessFinishNoResult); 1551b9c1b51eSKate Stone } else { 1552b9c1b51eSKate Stone result.AppendErrorWithFormat("module importing failed: %s", 1553b9c1b51eSKate Stone error.AsCString()); 1554a9dbf432SEnrico Granata result.SetStatus(eReturnStatusFailed); 1555a9dbf432SEnrico Granata } 15563b00e35bSEnrico Granata } 1557a9dbf432SEnrico Granata 1558a9dbf432SEnrico Granata return result.Succeeded(); 1559a9dbf432SEnrico Granata } 15600a305db7SEnrico Granata 15615a988416SJim Ingham CommandOptions m_options; 1562a9dbf432SEnrico Granata }; 1563223383edSEnrico Granata 1564223383edSEnrico Granata //------------------------------------------------------------------------- 1565223383edSEnrico Granata // CommandObjectCommandsScriptAdd 1566223383edSEnrico Granata //------------------------------------------------------------------------- 1567223383edSEnrico Granata 15681f0f5b5bSZachary Turner static OptionEnumValueElement g_script_synchro_type[] = { 15691f0f5b5bSZachary Turner {eScriptedCommandSynchronicitySynchronous, "synchronous", 15701f0f5b5bSZachary Turner "Run synchronous"}, 15711f0f5b5bSZachary Turner {eScriptedCommandSynchronicityAsynchronous, "asynchronous", 15721f0f5b5bSZachary Turner "Run asynchronous"}, 15731f0f5b5bSZachary Turner {eScriptedCommandSynchronicityCurrentValue, "current", 15741f0f5b5bSZachary Turner "Do not alter current setting"}, 15751f0f5b5bSZachary Turner {0, nullptr, nullptr}}; 15761f0f5b5bSZachary Turner 15771f0f5b5bSZachary Turner static OptionDefinition g_script_add_options[] = { 15781f0f5b5bSZachary Turner // clang-format off 15791f0f5b5bSZachary Turner { LLDB_OPT_SET_1, false, "function", 'f', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypePythonFunction, "Name of the Python function to bind to this command name." }, 15801f0f5b5bSZachary Turner { LLDB_OPT_SET_2, false, "class", 'c', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypePythonClass, "Name of the Python class to bind to this command name." }, 15811f0f5b5bSZachary Turner { LLDB_OPT_SET_1, false, "help" , 'h', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeHelpText, "The help text to display for this command." }, 15821f0f5b5bSZachary Turner { LLDB_OPT_SET_ALL, false, "synchronicity", 's', OptionParser::eRequiredArgument, nullptr, g_script_synchro_type, 0, eArgTypeScriptedCommandSynchronicity, "Set the synchronicity of this command's executions with regard to LLDB event system." }, 15831f0f5b5bSZachary Turner // clang-format on 15841f0f5b5bSZachary Turner }; 15851f0f5b5bSZachary Turner 1586b9c1b51eSKate Stone class CommandObjectCommandsScriptAdd : public CommandObjectParsed, 1587b9c1b51eSKate Stone public IOHandlerDelegateMultiline { 15885a988416SJim Ingham public: 1589b9c1b51eSKate Stone CommandObjectCommandsScriptAdd(CommandInterpreter &interpreter) 1590b9c1b51eSKate Stone : CommandObjectParsed(interpreter, "command script add", 15915a988416SJim Ingham "Add a scripted function as an LLDB command.", 15926e3d8e7fSEugene Zelenko nullptr), 1593b9c1b51eSKate Stone IOHandlerDelegateMultiline("DONE"), m_options() { 15945a988416SJim Ingham CommandArgumentEntry arg1; 15955a988416SJim Ingham CommandArgumentData cmd_arg; 15965a988416SJim Ingham 15975a988416SJim Ingham // Define the first (and only) variant of this arg. 15985a988416SJim Ingham cmd_arg.arg_type = eArgTypeCommandName; 15995a988416SJim Ingham cmd_arg.arg_repetition = eArgRepeatPlain; 16005a988416SJim Ingham 1601b9c1b51eSKate Stone // There is only one variant this argument could be; put it into the 1602b9c1b51eSKate Stone // argument entry. 16035a988416SJim Ingham arg1.push_back(cmd_arg); 16045a988416SJim Ingham 16055a988416SJim Ingham // Push the data for the first argument into the m_arguments vector. 16065a988416SJim Ingham m_arguments.push_back(arg1); 16075a988416SJim Ingham } 16085a988416SJim Ingham 16096e3d8e7fSEugene Zelenko ~CommandObjectCommandsScriptAdd() override = default; 16105a988416SJim Ingham 1611b9c1b51eSKate Stone Options *GetOptions() override { return &m_options; } 16125a988416SJim Ingham 16135a988416SJim Ingham protected: 1614b9c1b51eSKate Stone class CommandOptions : public Options { 1615223383edSEnrico Granata public: 1616b9c1b51eSKate Stone CommandOptions() 1617b9c1b51eSKate Stone : Options(), m_class_name(), m_funct_name(), m_short_help(), 1618b9c1b51eSKate Stone m_synchronicity(eScriptedCommandSynchronicitySynchronous) {} 1619223383edSEnrico Granata 16206e3d8e7fSEugene Zelenko ~CommandOptions() override = default; 1621223383edSEnrico Granata 1622fe11483bSZachary Turner Error SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg, 1623b9c1b51eSKate Stone ExecutionContext *execution_context) override { 1624223383edSEnrico Granata Error error; 16253bcdfc0eSGreg Clayton const int short_option = m_getopt_table[option_idx].val; 1626223383edSEnrico Granata 1627b9c1b51eSKate Stone switch (short_option) { 1628223383edSEnrico Granata case 'f': 1629fe11483bSZachary Turner if (!option_arg.empty()) 1630fe11483bSZachary Turner m_funct_name = option_arg; 1631735152e3SEnrico Granata break; 16329fe00e52SEnrico Granata case 'c': 1633fe11483bSZachary Turner if (!option_arg.empty()) 1634fe11483bSZachary Turner m_class_name = option_arg; 16359fe00e52SEnrico Granata break; 1636735152e3SEnrico Granata case 'h': 1637fe11483bSZachary Turner if (!option_arg.empty()) 1638fe11483bSZachary Turner m_short_help = option_arg; 1639223383edSEnrico Granata break; 16400a305db7SEnrico Granata case 's': 1641b9c1b51eSKate Stone m_synchronicity = 1642b9c1b51eSKate Stone (ScriptedCommandSynchronicity)Args::StringToOptionEnum( 1643fe11483bSZachary Turner option_arg, GetDefinitions()[option_idx].enum_values, 0, error); 16440a305db7SEnrico Granata if (!error.Success()) 1645b9c1b51eSKate Stone error.SetErrorStringWithFormat( 1646fe11483bSZachary Turner "unrecognized value for synchronicity '%s'", 1647fe11483bSZachary Turner option_arg.str().c_str()); 16480a305db7SEnrico Granata break; 1649223383edSEnrico Granata default: 1650b9c1b51eSKate Stone error.SetErrorStringWithFormat("unrecognized option '%c'", 1651b9c1b51eSKate Stone short_option); 1652223383edSEnrico Granata break; 1653223383edSEnrico Granata } 1654223383edSEnrico Granata 1655223383edSEnrico Granata return error; 1656223383edSEnrico Granata } 1657223383edSEnrico Granata 1658b9c1b51eSKate Stone void OptionParsingStarting(ExecutionContext *execution_context) override { 16599fe00e52SEnrico Granata m_class_name.clear(); 1660735152e3SEnrico Granata m_funct_name.clear(); 1661735152e3SEnrico Granata m_short_help.clear(); 166244d93782SGreg Clayton m_synchronicity = eScriptedCommandSynchronicitySynchronous; 1663223383edSEnrico Granata } 1664223383edSEnrico Granata 16651f0f5b5bSZachary Turner llvm::ArrayRef<OptionDefinition> GetDefinitions() override { 166670602439SZachary Turner return llvm::makeArrayRef(g_script_add_options); 16671f0f5b5bSZachary Turner } 1668223383edSEnrico Granata 1669223383edSEnrico Granata // Instance variables to hold the values for command options. 1670223383edSEnrico Granata 16719fe00e52SEnrico Granata std::string m_class_name; 1672223383edSEnrico Granata std::string m_funct_name; 1673735152e3SEnrico Granata std::string m_short_help; 167444d93782SGreg Clayton ScriptedCommandSynchronicity m_synchronicity; 1675223383edSEnrico Granata }; 1676223383edSEnrico Granata 1677b9c1b51eSKate Stone void IOHandlerActivated(IOHandler &io_handler) override { 167844d93782SGreg Clayton StreamFileSP output_sp(io_handler.GetOutputStreamFile()); 1679b9c1b51eSKate Stone if (output_sp) { 168044d93782SGreg Clayton output_sp->PutCString(g_python_command_instructions); 168144d93782SGreg Clayton output_sp->Flush(); 1682223383edSEnrico Granata } 1683223383edSEnrico Granata } 1684223383edSEnrico Granata 1685b9c1b51eSKate Stone void IOHandlerInputComplete(IOHandler &io_handler, 1686b9c1b51eSKate Stone std::string &data) override { 168744d93782SGreg Clayton StreamFileSP error_sp = io_handler.GetErrorStreamFile(); 168844d93782SGreg Clayton 168944d93782SGreg Clayton ScriptInterpreter *interpreter = m_interpreter.GetScriptInterpreter(); 1690b9c1b51eSKate Stone if (interpreter) { 169144d93782SGreg Clayton 169244d93782SGreg Clayton StringList lines; 169344d93782SGreg Clayton lines.SplitIntoLines(data); 1694b9c1b51eSKate Stone if (lines.GetSize() > 0) { 1695a73b7df7SEnrico Granata std::string funct_name_str; 1696b9c1b51eSKate Stone if (interpreter->GenerateScriptAliasFunction(lines, funct_name_str)) { 1697b9c1b51eSKate Stone if (funct_name_str.empty()) { 1698b9c1b51eSKate Stone error_sp->Printf("error: unable to obtain a function name, didn't " 1699b9c1b51eSKate Stone "add python command.\n"); 170044d93782SGreg Clayton error_sp->Flush(); 1701b9c1b51eSKate Stone } else { 1702223383edSEnrico Granata // everything should be fine now, let's add this alias 1703223383edSEnrico Granata 1704b9c1b51eSKate Stone CommandObjectSP command_obj_sp(new CommandObjectPythonFunction( 1705771ef6d4SMalcolm Parsons m_interpreter, m_cmd_name, funct_name_str, m_short_help, 170644d93782SGreg Clayton m_synchronicity)); 1707223383edSEnrico Granata 1708b9c1b51eSKate Stone if (!m_interpreter.AddUserCommand(m_cmd_name, command_obj_sp, 1709b9c1b51eSKate Stone true)) { 1710b9c1b51eSKate Stone error_sp->Printf("error: unable to add selected command, didn't " 1711b9c1b51eSKate Stone "add python command.\n"); 171244d93782SGreg Clayton error_sp->Flush(); 1713223383edSEnrico Granata } 1714223383edSEnrico Granata } 1715b9c1b51eSKate Stone } else { 1716b9c1b51eSKate Stone error_sp->Printf( 1717b9c1b51eSKate Stone "error: unable to create function, didn't add python command.\n"); 171844d93782SGreg Clayton error_sp->Flush(); 171944d93782SGreg Clayton } 1720b9c1b51eSKate Stone } else { 172144d93782SGreg Clayton error_sp->Printf("error: empty function, didn't add python command.\n"); 172244d93782SGreg Clayton error_sp->Flush(); 172344d93782SGreg Clayton } 1724b9c1b51eSKate Stone } else { 1725b9c1b51eSKate Stone error_sp->Printf( 1726b9c1b51eSKate Stone "error: script interpreter missing, didn't add python command.\n"); 172744d93782SGreg Clayton error_sp->Flush(); 172844d93782SGreg Clayton } 172944d93782SGreg Clayton 173044d93782SGreg Clayton io_handler.SetIsDone(true); 173144d93782SGreg Clayton } 1732223383edSEnrico Granata 17335a988416SJim Ingham protected: 1734b9c1b51eSKate Stone bool DoExecute(Args &command, CommandReturnObject &result) override { 1735b9c1b51eSKate Stone if (m_interpreter.GetDebugger().GetScriptLanguage() != 1736b9c1b51eSKate Stone lldb::eScriptLanguagePython) { 1737b9c1b51eSKate Stone result.AppendError("only scripting language supported for scripted " 1738b9c1b51eSKate Stone "commands is currently Python"); 173999f0b8f9SEnrico Granata result.SetStatus(eReturnStatusFailed); 174099f0b8f9SEnrico Granata return false; 174199f0b8f9SEnrico Granata } 174299f0b8f9SEnrico Granata 174311eb9c64SZachary Turner if (command.GetArgumentCount() != 1) { 1744223383edSEnrico Granata result.AppendError("'command script add' requires one argument"); 1745223383edSEnrico Granata result.SetStatus(eReturnStatusFailed); 1746223383edSEnrico Granata return false; 1747223383edSEnrico Granata } 1748223383edSEnrico Granata 1749735152e3SEnrico Granata // Store the options in case we get multi-line input 17504574a890SZachary Turner m_cmd_name = command[0].ref; 1751735152e3SEnrico Granata m_short_help.assign(m_options.m_short_help); 175244d93782SGreg Clayton m_synchronicity = m_options.m_synchronicity; 1753223383edSEnrico Granata 1754b9c1b51eSKate Stone if (m_options.m_class_name.empty()) { 1755b9c1b51eSKate Stone if (m_options.m_funct_name.empty()) { 1756b9c1b51eSKate Stone m_interpreter.GetPythonCommandsFromIOHandler( 1757b9c1b51eSKate Stone " ", // Prompt 175844d93782SGreg Clayton *this, // IOHandlerDelegate 175944d93782SGreg Clayton true, // Run IOHandler in async mode 1760b9c1b51eSKate Stone nullptr); // Baton for the "io_handler" that will be passed back 1761b9c1b51eSKate Stone // into our IOHandlerDelegate functions 1762b9c1b51eSKate Stone } else { 1763b9c1b51eSKate Stone CommandObjectSP new_cmd(new CommandObjectPythonFunction( 1764b9c1b51eSKate Stone m_interpreter, m_cmd_name, m_options.m_funct_name, 1765b9c1b51eSKate Stone m_options.m_short_help, m_synchronicity)); 1766b9c1b51eSKate Stone if (m_interpreter.AddUserCommand(m_cmd_name, new_cmd, true)) { 1767223383edSEnrico Granata result.SetStatus(eReturnStatusSuccessFinishNoResult); 1768b9c1b51eSKate Stone } else { 1769223383edSEnrico Granata result.AppendError("cannot add command"); 1770223383edSEnrico Granata result.SetStatus(eReturnStatusFailed); 1771223383edSEnrico Granata } 1772223383edSEnrico Granata } 1773b9c1b51eSKate Stone } else { 1774b9c1b51eSKate Stone ScriptInterpreter *interpreter = 1775b9c1b51eSKate Stone GetCommandInterpreter().GetScriptInterpreter(); 1776b9c1b51eSKate Stone if (!interpreter) { 17779fe00e52SEnrico Granata result.AppendError("cannot find ScriptInterpreter"); 17789fe00e52SEnrico Granata result.SetStatus(eReturnStatusFailed); 17799fe00e52SEnrico Granata return false; 17809fe00e52SEnrico Granata } 17819fe00e52SEnrico Granata 1782b9c1b51eSKate Stone auto cmd_obj_sp = interpreter->CreateScriptCommandObject( 1783b9c1b51eSKate Stone m_options.m_class_name.c_str()); 1784b9c1b51eSKate Stone if (!cmd_obj_sp) { 17859fe00e52SEnrico Granata result.AppendError("cannot create helper object"); 17869fe00e52SEnrico Granata result.SetStatus(eReturnStatusFailed); 17879fe00e52SEnrico Granata return false; 17889fe00e52SEnrico Granata } 17899fe00e52SEnrico Granata 1790b9c1b51eSKate Stone CommandObjectSP new_cmd(new CommandObjectScriptingObject( 1791b9c1b51eSKate Stone m_interpreter, m_cmd_name, cmd_obj_sp, m_synchronicity)); 1792b9c1b51eSKate Stone if (m_interpreter.AddUserCommand(m_cmd_name, new_cmd, true)) { 17939fe00e52SEnrico Granata result.SetStatus(eReturnStatusSuccessFinishNoResult); 1794b9c1b51eSKate Stone } else { 17959fe00e52SEnrico Granata result.AppendError("cannot add command"); 17969fe00e52SEnrico Granata result.SetStatus(eReturnStatusFailed); 17979fe00e52SEnrico Granata } 17989fe00e52SEnrico Granata } 1799223383edSEnrico Granata 1800223383edSEnrico Granata return result.Succeeded(); 1801223383edSEnrico Granata } 18025a988416SJim Ingham 18035a988416SJim Ingham CommandOptions m_options; 180444d93782SGreg Clayton std::string m_cmd_name; 1805735152e3SEnrico Granata std::string m_short_help; 180644d93782SGreg Clayton ScriptedCommandSynchronicity m_synchronicity; 1807223383edSEnrico Granata }; 1808223383edSEnrico Granata 1809223383edSEnrico Granata //------------------------------------------------------------------------- 1810223383edSEnrico Granata // CommandObjectCommandsScriptList 1811223383edSEnrico Granata //------------------------------------------------------------------------- 1812223383edSEnrico Granata 1813b9c1b51eSKate Stone class CommandObjectCommandsScriptList : public CommandObjectParsed { 1814223383edSEnrico Granata public: 1815b9c1b51eSKate Stone CommandObjectCommandsScriptList(CommandInterpreter &interpreter) 1816b9c1b51eSKate Stone : CommandObjectParsed(interpreter, "command script list", 1817b9c1b51eSKate Stone "List defined scripted commands.", nullptr) {} 1818223383edSEnrico Granata 18196e3d8e7fSEugene Zelenko ~CommandObjectCommandsScriptList() override = default; 1820223383edSEnrico Granata 1821b9c1b51eSKate Stone bool DoExecute(Args &command, CommandReturnObject &result) override { 1822b9c1b51eSKate Stone m_interpreter.GetHelp(result, CommandInterpreter::eCommandTypesUserDef); 1823223383edSEnrico Granata 1824223383edSEnrico Granata result.SetStatus(eReturnStatusSuccessFinishResult); 1825223383edSEnrico Granata 1826223383edSEnrico Granata return true; 1827223383edSEnrico Granata } 1828223383edSEnrico Granata }; 1829223383edSEnrico Granata 1830223383edSEnrico Granata //------------------------------------------------------------------------- 1831223383edSEnrico Granata // CommandObjectCommandsScriptClear 1832223383edSEnrico Granata //------------------------------------------------------------------------- 1833223383edSEnrico Granata 1834b9c1b51eSKate Stone class CommandObjectCommandsScriptClear : public CommandObjectParsed { 1835223383edSEnrico Granata public: 1836b9c1b51eSKate Stone CommandObjectCommandsScriptClear(CommandInterpreter &interpreter) 1837b9c1b51eSKate Stone : CommandObjectParsed(interpreter, "command script clear", 1838b9c1b51eSKate Stone "Delete all scripted commands.", nullptr) {} 1839223383edSEnrico Granata 18406e3d8e7fSEugene Zelenko ~CommandObjectCommandsScriptClear() override = default; 1841223383edSEnrico Granata 18425a988416SJim Ingham protected: 1843b9c1b51eSKate Stone bool DoExecute(Args &command, CommandReturnObject &result) override { 1844223383edSEnrico Granata m_interpreter.RemoveAllUser(); 1845223383edSEnrico Granata 1846223383edSEnrico Granata result.SetStatus(eReturnStatusSuccessFinishResult); 1847223383edSEnrico Granata 1848223383edSEnrico Granata return true; 1849223383edSEnrico Granata } 1850223383edSEnrico Granata }; 1851223383edSEnrico Granata 1852223383edSEnrico Granata //------------------------------------------------------------------------- 1853223383edSEnrico Granata // CommandObjectCommandsScriptDelete 1854223383edSEnrico Granata //------------------------------------------------------------------------- 1855223383edSEnrico Granata 1856b9c1b51eSKate Stone class CommandObjectCommandsScriptDelete : public CommandObjectParsed { 1857223383edSEnrico Granata public: 1858b9c1b51eSKate Stone CommandObjectCommandsScriptDelete(CommandInterpreter &interpreter) 1859b9c1b51eSKate Stone : CommandObjectParsed(interpreter, "command script delete", 1860b9c1b51eSKate Stone "Delete a scripted command.", nullptr) { 1861223383edSEnrico Granata CommandArgumentEntry arg1; 1862223383edSEnrico Granata CommandArgumentData cmd_arg; 1863223383edSEnrico Granata 1864223383edSEnrico Granata // Define the first (and only) variant of this arg. 1865223383edSEnrico Granata cmd_arg.arg_type = eArgTypeCommandName; 1866223383edSEnrico Granata cmd_arg.arg_repetition = eArgRepeatPlain; 1867223383edSEnrico Granata 1868b9c1b51eSKate Stone // There is only one variant this argument could be; put it into the 1869b9c1b51eSKate Stone // argument entry. 1870223383edSEnrico Granata arg1.push_back(cmd_arg); 1871223383edSEnrico Granata 1872223383edSEnrico Granata // Push the data for the first argument into the m_arguments vector. 1873223383edSEnrico Granata m_arguments.push_back(arg1); 1874223383edSEnrico Granata } 1875223383edSEnrico Granata 18766e3d8e7fSEugene Zelenko ~CommandObjectCommandsScriptDelete() override = default; 1877223383edSEnrico Granata 18785a988416SJim Ingham protected: 1879b9c1b51eSKate Stone bool DoExecute(Args &command, CommandReturnObject &result) override { 1880223383edSEnrico Granata 188111eb9c64SZachary Turner if (command.GetArgumentCount() != 1) { 1882223383edSEnrico Granata result.AppendError("'command script delete' requires one argument"); 1883223383edSEnrico Granata result.SetStatus(eReturnStatusFailed); 1884223383edSEnrico Granata return false; 1885223383edSEnrico Granata } 1886223383edSEnrico Granata 18874574a890SZachary Turner auto cmd_name = command[0].ref; 1888223383edSEnrico Granata 18894574a890SZachary Turner if (cmd_name.empty() || !m_interpreter.HasUserCommands() || 18904574a890SZachary Turner !m_interpreter.UserCommandExists(cmd_name)) { 1891867e7d17SZachary Turner result.AppendErrorWithFormat("command %s not found", command[0].c_str()); 1892223383edSEnrico Granata result.SetStatus(eReturnStatusFailed); 18934574a890SZachary Turner return false; 1894223383edSEnrico Granata } 1895223383edSEnrico Granata 18964574a890SZachary Turner m_interpreter.RemoveUser(cmd_name); 18974574a890SZachary Turner result.SetStatus(eReturnStatusSuccessFinishResult); 18984574a890SZachary Turner return true; 1899223383edSEnrico Granata } 1900223383edSEnrico Granata }; 1901223383edSEnrico Granata 1902223383edSEnrico Granata #pragma mark CommandObjectMultiwordCommandsScript 1903223383edSEnrico Granata 1904223383edSEnrico Granata //------------------------------------------------------------------------- 1905223383edSEnrico Granata // CommandObjectMultiwordCommandsScript 1906223383edSEnrico Granata //------------------------------------------------------------------------- 1907223383edSEnrico Granata 1908b9c1b51eSKate Stone class CommandObjectMultiwordCommandsScript : public CommandObjectMultiword { 1909223383edSEnrico Granata public: 19107428a18cSKate Stone CommandObjectMultiwordCommandsScript(CommandInterpreter &interpreter) 1911b9c1b51eSKate Stone : CommandObjectMultiword( 1912b9c1b51eSKate Stone interpreter, "command script", "Commands for managing custom " 1913b9c1b51eSKate Stone "commands implemented by " 1914b9c1b51eSKate Stone "interpreter scripts.", 1915b9c1b51eSKate Stone "command script <subcommand> [<subcommand-options>]") { 1916b9c1b51eSKate Stone LoadSubCommand("add", CommandObjectSP( 1917b9c1b51eSKate Stone new CommandObjectCommandsScriptAdd(interpreter))); 1918b9c1b51eSKate Stone LoadSubCommand( 1919b9c1b51eSKate Stone "delete", 1920b9c1b51eSKate Stone CommandObjectSP(new CommandObjectCommandsScriptDelete(interpreter))); 1921b9c1b51eSKate Stone LoadSubCommand( 1922b9c1b51eSKate Stone "clear", 1923b9c1b51eSKate Stone CommandObjectSP(new CommandObjectCommandsScriptClear(interpreter))); 1924b9c1b51eSKate Stone LoadSubCommand("list", CommandObjectSP(new CommandObjectCommandsScriptList( 1925b9c1b51eSKate Stone interpreter))); 1926b9c1b51eSKate Stone LoadSubCommand( 1927b9c1b51eSKate Stone "import", 1928b9c1b51eSKate Stone CommandObjectSP(new CommandObjectCommandsScriptImport(interpreter))); 1929223383edSEnrico Granata } 1930223383edSEnrico Granata 19316e3d8e7fSEugene Zelenko ~CommandObjectMultiwordCommandsScript() override = default; 1932223383edSEnrico Granata }; 1933223383edSEnrico Granata 1934ebc09c36SJim Ingham #pragma mark CommandObjectMultiwordCommands 1935ebc09c36SJim Ingham 1936ebc09c36SJim Ingham //------------------------------------------------------------------------- 1937ebc09c36SJim Ingham // CommandObjectMultiwordCommands 1938ebc09c36SJim Ingham //------------------------------------------------------------------------- 1939ebc09c36SJim Ingham 1940b9c1b51eSKate Stone CommandObjectMultiwordCommands::CommandObjectMultiwordCommands( 1941b9c1b51eSKate Stone CommandInterpreter &interpreter) 1942b9c1b51eSKate Stone : CommandObjectMultiword(interpreter, "command", 1943b9c1b51eSKate Stone "Commands for managing custom LLDB commands.", 1944b9c1b51eSKate Stone "command <subcommand> [<subcommand-options>]") { 1945b9c1b51eSKate Stone LoadSubCommand("source", 1946b9c1b51eSKate Stone CommandObjectSP(new CommandObjectCommandsSource(interpreter))); 1947b9c1b51eSKate Stone LoadSubCommand("alias", 1948b9c1b51eSKate Stone CommandObjectSP(new CommandObjectCommandsAlias(interpreter))); 1949b9c1b51eSKate Stone LoadSubCommand("unalias", CommandObjectSP( 1950b9c1b51eSKate Stone new CommandObjectCommandsUnalias(interpreter))); 1951b9c1b51eSKate Stone LoadSubCommand("delete", 1952b9c1b51eSKate Stone CommandObjectSP(new CommandObjectCommandsDelete(interpreter))); 1953b9c1b51eSKate Stone LoadSubCommand( 1954b9c1b51eSKate Stone "regex", CommandObjectSP(new CommandObjectCommandsAddRegex(interpreter))); 1955b9c1b51eSKate Stone LoadSubCommand("history", CommandObjectSP( 1956b9c1b51eSKate Stone new CommandObjectCommandsHistory(interpreter))); 1957b9c1b51eSKate Stone LoadSubCommand( 1958b9c1b51eSKate Stone "script", 1959b9c1b51eSKate Stone CommandObjectSP(new CommandObjectMultiwordCommandsScript(interpreter))); 1960ebc09c36SJim Ingham } 1961ebc09c36SJim Ingham 19626e3d8e7fSEugene Zelenko CommandObjectMultiwordCommands::~CommandObjectMultiwordCommands() = default; 1963