1ebc09c36SJim Ingham //===-- CommandObjectSource.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 1093a64300SDaniel Malea #include "lldb/lldb-python.h" 1193a64300SDaniel Malea 12ebc09c36SJim Ingham #include "CommandObjectCommands.h" 13ebc09c36SJim Ingham 14ebc09c36SJim Ingham // C Includes 15ebc09c36SJim Ingham // C++ Includes 16ebc09c36SJim Ingham // Other libraries and framework includes 170e5e5a79SGreg Clayton #include "llvm/ADT/StringRef.h" 180e5e5a79SGreg Clayton 19ebc09c36SJim Ingham // Project includes 20ebc09c36SJim Ingham #include "lldb/Core/Debugger.h" 21*44d93782SGreg Clayton #include "lldb/Core/IOHandler.h" 22be93a35aSEnrico Granata #include "lldb/Core/StringList.h" 23de164aaaSGreg Clayton #include "lldb/Interpreter/Args.h" 247594f14fSEnrico Granata #include "lldb/Interpreter/CommandHistory.h" 25ebc09c36SJim Ingham #include "lldb/Interpreter/CommandInterpreter.h" 26de164aaaSGreg Clayton #include "lldb/Interpreter/CommandObjectRegexCommand.h" 27ebc09c36SJim Ingham #include "lldb/Interpreter/CommandReturnObject.h" 28012d4fcaSEnrico Granata #include "lldb/Interpreter/OptionValueBoolean.h" 297594f14fSEnrico Granata #include "lldb/Interpreter/OptionValueUInt64.h" 30ebc09c36SJim Ingham #include "lldb/Interpreter/Options.h" 3199f0b8f9SEnrico Granata #include "lldb/Interpreter/ScriptInterpreter.h" 3299f0b8f9SEnrico Granata #include "lldb/Interpreter/ScriptInterpreterPython.h" 33ebc09c36SJim Ingham 34ebc09c36SJim Ingham using namespace lldb; 35ebc09c36SJim Ingham using namespace lldb_private; 36ebc09c36SJim Ingham 37ebc09c36SJim Ingham //------------------------------------------------------------------------- 38ebc09c36SJim Ingham // CommandObjectCommandsSource 39ebc09c36SJim Ingham //------------------------------------------------------------------------- 40ebc09c36SJim Ingham 415a988416SJim Ingham class CommandObjectCommandsHistory : public CommandObjectParsed 42a5a97ebeSJim Ingham { 435a988416SJim Ingham public: 445a988416SJim Ingham CommandObjectCommandsHistory(CommandInterpreter &interpreter) : 455a988416SJim Ingham CommandObjectParsed (interpreter, 465a988416SJim Ingham "command history", 475a988416SJim Ingham "Dump the history of commands in this session.", 485a988416SJim Ingham NULL), 495a988416SJim Ingham m_options (interpreter) 505a988416SJim Ingham { 515a988416SJim Ingham } 525a988416SJim Ingham 535a988416SJim Ingham ~CommandObjectCommandsHistory () {} 545a988416SJim Ingham 555a988416SJim Ingham virtual Options * 565a988416SJim Ingham GetOptions () 575a988416SJim Ingham { 585a988416SJim Ingham return &m_options; 595a988416SJim Ingham } 605a988416SJim Ingham 615a988416SJim Ingham protected: 62a5a97ebeSJim Ingham 63a5a97ebeSJim Ingham class CommandOptions : public Options 64a5a97ebeSJim Ingham { 65a5a97ebeSJim Ingham public: 66a5a97ebeSJim Ingham 67a5a97ebeSJim Ingham CommandOptions (CommandInterpreter &interpreter) : 687594f14fSEnrico Granata Options (interpreter), 697594f14fSEnrico Granata m_start_idx(0), 707594f14fSEnrico Granata m_stop_idx(0), 717594f14fSEnrico Granata m_count(0), 7263123b64SEnrico Granata m_clear(false) 73a5a97ebeSJim Ingham { 74a5a97ebeSJim Ingham } 75a5a97ebeSJim Ingham 76a5a97ebeSJim Ingham virtual 77a5a97ebeSJim Ingham ~CommandOptions (){} 78a5a97ebeSJim Ingham 79a5a97ebeSJim Ingham virtual Error 80a5a97ebeSJim Ingham SetOptionValue (uint32_t option_idx, const char *option_arg) 81a5a97ebeSJim Ingham { 82a5a97ebeSJim Ingham Error error; 833bcdfc0eSGreg Clayton const int short_option = m_getopt_table[option_idx].val; 84a5a97ebeSJim Ingham 85a5a97ebeSJim Ingham switch (short_option) 86a5a97ebeSJim Ingham { 87a5a97ebeSJim Ingham case 'c': 887594f14fSEnrico Granata error = m_count.SetValueFromCString(option_arg,eVarSetOperationAssign); 89a5a97ebeSJim Ingham break; 90a5a97ebeSJim Ingham case 's': 917594f14fSEnrico Granata if (option_arg && strcmp("end", option_arg) == 0) 927594f14fSEnrico Granata { 937594f14fSEnrico Granata m_start_idx.SetCurrentValue(UINT64_MAX); 947594f14fSEnrico Granata m_start_idx.SetOptionWasSet(); 957594f14fSEnrico Granata } 967594f14fSEnrico Granata else 977594f14fSEnrico Granata error = m_start_idx.SetValueFromCString(option_arg,eVarSetOperationAssign); 987594f14fSEnrico Granata break; 997594f14fSEnrico Granata case 'e': 1007594f14fSEnrico Granata error = m_stop_idx.SetValueFromCString(option_arg,eVarSetOperationAssign); 1017594f14fSEnrico Granata break; 10263123b64SEnrico Granata case 'C': 10363123b64SEnrico Granata m_clear.SetCurrentValue(true); 10463123b64SEnrico Granata m_clear.SetOptionWasSet(); 105a5a97ebeSJim Ingham break; 106a5a97ebeSJim Ingham default: 10786edbf41SGreg Clayton error.SetErrorStringWithFormat ("unrecognized option '%c'", short_option); 108a5a97ebeSJim Ingham break; 109a5a97ebeSJim Ingham } 110a5a97ebeSJim Ingham 111a5a97ebeSJim Ingham return error; 112a5a97ebeSJim Ingham } 113a5a97ebeSJim Ingham 114a5a97ebeSJim Ingham void 115a5a97ebeSJim Ingham OptionParsingStarting () 116a5a97ebeSJim Ingham { 1177594f14fSEnrico Granata m_start_idx.Clear(); 1187594f14fSEnrico Granata m_stop_idx.Clear(); 1197594f14fSEnrico Granata m_count.Clear(); 12063123b64SEnrico Granata m_clear.Clear(); 121a5a97ebeSJim Ingham } 122a5a97ebeSJim Ingham 123a5a97ebeSJim Ingham const OptionDefinition* 124a5a97ebeSJim Ingham GetDefinitions () 125a5a97ebeSJim Ingham { 126a5a97ebeSJim Ingham return g_option_table; 127a5a97ebeSJim Ingham } 128a5a97ebeSJim Ingham 129a5a97ebeSJim Ingham // Options table: Required for subclasses of Options. 130a5a97ebeSJim Ingham 131a5a97ebeSJim Ingham static OptionDefinition g_option_table[]; 132a5a97ebeSJim Ingham 133a5a97ebeSJim Ingham // Instance variables to hold the values for command options. 134a5a97ebeSJim Ingham 1357594f14fSEnrico Granata OptionValueUInt64 m_start_idx; 1367594f14fSEnrico Granata OptionValueUInt64 m_stop_idx; 1377594f14fSEnrico Granata OptionValueUInt64 m_count; 13863123b64SEnrico Granata OptionValueBoolean m_clear; 139a5a97ebeSJim Ingham }; 140a5a97ebeSJim Ingham 141a5a97ebeSJim Ingham bool 1425a988416SJim Ingham DoExecute (Args& command, CommandReturnObject &result) 143a5a97ebeSJim Ingham { 14463123b64SEnrico Granata if (m_options.m_clear.GetCurrentValue() && m_options.m_clear.OptionWasSet()) 1457594f14fSEnrico Granata { 1467594f14fSEnrico Granata m_interpreter.GetCommandHistory().Clear(); 1477594f14fSEnrico Granata result.SetStatus(lldb::eReturnStatusSuccessFinishNoResult); 1487594f14fSEnrico Granata } 1497594f14fSEnrico Granata else 1507594f14fSEnrico Granata { 1517594f14fSEnrico Granata if (m_options.m_start_idx.OptionWasSet() && m_options.m_stop_idx.OptionWasSet() && m_options.m_count.OptionWasSet()) 1527594f14fSEnrico Granata { 1537594f14fSEnrico Granata result.AppendError("--count, --start-index and --end-index cannot be all specified in the same invocation"); 1547594f14fSEnrico Granata result.SetStatus(lldb::eReturnStatusFailed); 1557594f14fSEnrico Granata } 1567594f14fSEnrico Granata else 1577594f14fSEnrico Granata { 15884400ec7SVirgile Bello std::pair<bool,uint64_t> start_idx(m_options.m_start_idx.OptionWasSet(),m_options.m_start_idx.GetCurrentValue()); 15984400ec7SVirgile Bello std::pair<bool,uint64_t> stop_idx(m_options.m_stop_idx.OptionWasSet(),m_options.m_stop_idx.GetCurrentValue()); 16084400ec7SVirgile Bello std::pair<bool,uint64_t> count(m_options.m_count.OptionWasSet(),m_options.m_count.GetCurrentValue()); 161a5a97ebeSJim Ingham 1627594f14fSEnrico Granata const CommandHistory& history(m_interpreter.GetCommandHistory()); 1637594f14fSEnrico Granata 1647594f14fSEnrico Granata if (start_idx.first && start_idx.second == UINT64_MAX) 1657594f14fSEnrico Granata { 1667594f14fSEnrico Granata if (count.first) 1677594f14fSEnrico Granata { 1687594f14fSEnrico Granata start_idx.second = history.GetSize() - count.second; 1697594f14fSEnrico Granata stop_idx.second = history.GetSize() - 1; 1707594f14fSEnrico Granata } 1717594f14fSEnrico Granata else if (stop_idx.first) 1727594f14fSEnrico Granata { 1737594f14fSEnrico Granata start_idx.second = stop_idx.second; 1747594f14fSEnrico Granata stop_idx.second = history.GetSize() - 1; 1757594f14fSEnrico Granata } 1767594f14fSEnrico Granata else 1777594f14fSEnrico Granata { 1787594f14fSEnrico Granata start_idx.second = 0; 1797594f14fSEnrico Granata stop_idx.second = history.GetSize() - 1; 1807594f14fSEnrico Granata } 1817594f14fSEnrico Granata } 1827594f14fSEnrico Granata else 1837594f14fSEnrico Granata { 1847594f14fSEnrico Granata if (!start_idx.first && !stop_idx.first && !count.first) 1857594f14fSEnrico Granata { 1867594f14fSEnrico Granata start_idx.second = 0; 1877594f14fSEnrico Granata stop_idx.second = history.GetSize() - 1; 1887594f14fSEnrico Granata } 1897594f14fSEnrico Granata else if (start_idx.first) 1907594f14fSEnrico Granata { 1917594f14fSEnrico Granata if (count.first) 1927594f14fSEnrico Granata { 1937594f14fSEnrico Granata stop_idx.second = start_idx.second + count.second - 1; 1947594f14fSEnrico Granata } 1957594f14fSEnrico Granata else if (!stop_idx.first) 1967594f14fSEnrico Granata { 1977594f14fSEnrico Granata stop_idx.second = history.GetSize() - 1; 1987594f14fSEnrico Granata } 1997594f14fSEnrico Granata } 2007594f14fSEnrico Granata else if (stop_idx.first) 2017594f14fSEnrico Granata { 2027594f14fSEnrico Granata if (count.first) 2037594f14fSEnrico Granata { 2047594f14fSEnrico Granata if (stop_idx.second >= count.second) 2057594f14fSEnrico Granata start_idx.second = stop_idx.second - count.second + 1; 2067594f14fSEnrico Granata else 2077594f14fSEnrico Granata start_idx.second = 0; 2087594f14fSEnrico Granata } 2097594f14fSEnrico Granata } 2107594f14fSEnrico Granata else /* if (count.first) */ 2117594f14fSEnrico Granata { 2127594f14fSEnrico Granata start_idx.second = 0; 2137594f14fSEnrico Granata stop_idx.second = count.second - 1; 2147594f14fSEnrico Granata } 2157594f14fSEnrico Granata } 2167594f14fSEnrico Granata history.Dump(result.GetOutputStream(), start_idx.second, stop_idx.second); 2177594f14fSEnrico Granata } 2187594f14fSEnrico Granata } 219a5a97ebeSJim Ingham return result.Succeeded(); 220a5a97ebeSJim Ingham 221a5a97ebeSJim Ingham } 2225a988416SJim Ingham 2235a988416SJim Ingham CommandOptions m_options; 224a5a97ebeSJim Ingham }; 225a5a97ebeSJim Ingham 226a5a97ebeSJim Ingham OptionDefinition 227a5a97ebeSJim Ingham CommandObjectCommandsHistory::CommandOptions::g_option_table[] = 228a5a97ebeSJim Ingham { 229e2607b50SVirgile Bello { LLDB_OPT_SET_1, false, "count", 'c', OptionParser::eRequiredArgument, NULL, 0, eArgTypeUnsignedInteger, "How many history commands to print."}, 230e2607b50SVirgile Bello { LLDB_OPT_SET_1, false, "start-index", 's', OptionParser::eRequiredArgument, NULL, 0, eArgTypeUnsignedInteger, "Index at which to start printing history commands (or end to mean tail mode)."}, 231e2607b50SVirgile Bello { LLDB_OPT_SET_1, false, "end-index", 'e', OptionParser::eRequiredArgument, NULL, 0, eArgTypeUnsignedInteger, "Index at which to stop printing history commands."}, 232e2607b50SVirgile Bello { LLDB_OPT_SET_2, false, "clear", 'C', OptionParser::eNoArgument, NULL, 0, eArgTypeBoolean, "Clears the current command history."}, 233a5a97ebeSJim Ingham { 0, false, NULL, 0, 0, NULL, 0, eArgTypeNone, NULL } 234a5a97ebeSJim Ingham }; 235a5a97ebeSJim Ingham 236a5a97ebeSJim Ingham 237a5a97ebeSJim Ingham //------------------------------------------------------------------------- 238a5a97ebeSJim Ingham // CommandObjectCommandsSource 239a5a97ebeSJim Ingham //------------------------------------------------------------------------- 240a5a97ebeSJim Ingham 2415a988416SJim Ingham class CommandObjectCommandsSource : public CommandObjectParsed 242ebc09c36SJim Ingham { 2435a988416SJim Ingham public: 2445a988416SJim Ingham CommandObjectCommandsSource(CommandInterpreter &interpreter) : 2455a988416SJim Ingham CommandObjectParsed (interpreter, 2465a988416SJim Ingham "command source", 2475a988416SJim Ingham "Read in debugger commands from the file <filename> and execute them.", 2485a988416SJim Ingham NULL), 2495a988416SJim Ingham m_options (interpreter) 2505a988416SJim Ingham { 2515a988416SJim Ingham CommandArgumentEntry arg; 2525a988416SJim Ingham CommandArgumentData file_arg; 2535a988416SJim Ingham 2545a988416SJim Ingham // Define the first (and only) variant of this arg. 2555a988416SJim Ingham file_arg.arg_type = eArgTypeFilename; 2565a988416SJim Ingham file_arg.arg_repetition = eArgRepeatPlain; 2575a988416SJim Ingham 2585a988416SJim Ingham // There is only one variant this argument could be; put it into the argument entry. 2595a988416SJim Ingham arg.push_back (file_arg); 2605a988416SJim Ingham 2615a988416SJim Ingham // Push the data for the first argument into the m_arguments vector. 2625a988416SJim Ingham m_arguments.push_back (arg); 2635a988416SJim Ingham } 2645a988416SJim Ingham 2655a988416SJim Ingham ~CommandObjectCommandsSource () {} 2665a988416SJim Ingham 2675a988416SJim Ingham virtual const char* 2685a988416SJim Ingham GetRepeatCommand (Args ¤t_command_args, uint32_t index) 2695a988416SJim Ingham { 2705a988416SJim Ingham return ""; 2715a988416SJim Ingham } 2725a988416SJim Ingham 273c7bece56SGreg Clayton virtual int 2745a988416SJim Ingham HandleArgumentCompletion (Args &input, 2755a988416SJim Ingham int &cursor_index, 2765a988416SJim Ingham int &cursor_char_position, 2775a988416SJim Ingham OptionElementVector &opt_element_vector, 2785a988416SJim Ingham int match_start_point, 2795a988416SJim Ingham int max_return_elements, 2805a988416SJim Ingham bool &word_complete, 2815a988416SJim Ingham StringList &matches) 2825a988416SJim Ingham { 2835a988416SJim Ingham std::string completion_str (input.GetArgumentAtIndex(cursor_index)); 2845a988416SJim Ingham completion_str.erase (cursor_char_position); 2855a988416SJim Ingham 2865a988416SJim Ingham CommandCompletions::InvokeCommonCompletionCallbacks (m_interpreter, 2875a988416SJim Ingham CommandCompletions::eDiskFileCompletion, 2885a988416SJim Ingham completion_str.c_str(), 2895a988416SJim Ingham match_start_point, 2905a988416SJim Ingham max_return_elements, 2915a988416SJim Ingham NULL, 2925a988416SJim Ingham word_complete, 2935a988416SJim Ingham matches); 2945a988416SJim Ingham return matches.GetSize(); 2955a988416SJim Ingham } 2965a988416SJim Ingham 2975a988416SJim Ingham virtual Options * 2985a988416SJim Ingham GetOptions () 2995a988416SJim Ingham { 3005a988416SJim Ingham return &m_options; 3015a988416SJim Ingham } 3025a988416SJim Ingham 3035a988416SJim Ingham protected: 304e16c50a1SJim Ingham 305e16c50a1SJim Ingham class CommandOptions : public Options 306e16c50a1SJim Ingham { 307e16c50a1SJim Ingham public: 308e16c50a1SJim Ingham 309eb0103f2SGreg Clayton CommandOptions (CommandInterpreter &interpreter) : 310012d4fcaSEnrico Granata Options (interpreter), 311012d4fcaSEnrico Granata m_stop_on_error (true) 312eb0103f2SGreg Clayton { 313eb0103f2SGreg Clayton } 314e16c50a1SJim Ingham 315e16c50a1SJim Ingham virtual 316e16c50a1SJim Ingham ~CommandOptions (){} 317e16c50a1SJim Ingham 318e16c50a1SJim Ingham virtual Error 319f6b8b581SGreg Clayton SetOptionValue (uint32_t option_idx, const char *option_arg) 320e16c50a1SJim Ingham { 321e16c50a1SJim Ingham Error error; 3223bcdfc0eSGreg Clayton const int short_option = m_getopt_table[option_idx].val; 323e16c50a1SJim Ingham bool success; 324e16c50a1SJim Ingham 325e16c50a1SJim Ingham switch (short_option) 326e16c50a1SJim Ingham { 327e16c50a1SJim Ingham case 'e': 32815571f15SEnrico Granata error = m_stop_on_error.SetValueFromCString(option_arg); 329e16c50a1SJim Ingham break; 330e16c50a1SJim Ingham case 'c': 331e16c50a1SJim Ingham m_stop_on_continue = Args::StringToBoolean(option_arg, true, &success); 332e16c50a1SJim Ingham if (!success) 33386edbf41SGreg Clayton error.SetErrorStringWithFormat("invalid value for stop-on-continue: %s", option_arg); 334e16c50a1SJim Ingham break; 33560986174SMichael Sartain case 's': 33660986174SMichael Sartain m_silent_run = Args::StringToBoolean(option_arg, true, &success); 33760986174SMichael Sartain if (!success) 33860986174SMichael Sartain error.SetErrorStringWithFormat("invalid value for silent-run: %s", option_arg); 33960986174SMichael Sartain break; 340e16c50a1SJim Ingham default: 34186edbf41SGreg Clayton error.SetErrorStringWithFormat ("unrecognized option '%c'", short_option); 342e16c50a1SJim Ingham break; 343e16c50a1SJim Ingham } 344e16c50a1SJim Ingham 345e16c50a1SJim Ingham return error; 346e16c50a1SJim Ingham } 347e16c50a1SJim Ingham 348e16c50a1SJim Ingham void 349f6b8b581SGreg Clayton OptionParsingStarting () 350e16c50a1SJim Ingham { 351012d4fcaSEnrico Granata m_stop_on_error.Clear(); 35260986174SMichael Sartain m_silent_run = false; 353e16c50a1SJim Ingham m_stop_on_continue = true; 354e16c50a1SJim Ingham } 355e16c50a1SJim Ingham 356e0d378b3SGreg Clayton const OptionDefinition* 357e16c50a1SJim Ingham GetDefinitions () 358e16c50a1SJim Ingham { 359e16c50a1SJim Ingham return g_option_table; 360e16c50a1SJim Ingham } 361e16c50a1SJim Ingham 362e16c50a1SJim Ingham // Options table: Required for subclasses of Options. 363e16c50a1SJim Ingham 364e0d378b3SGreg Clayton static OptionDefinition g_option_table[]; 365e16c50a1SJim Ingham 366e16c50a1SJim Ingham // Instance variables to hold the values for command options. 367e16c50a1SJim Ingham 368012d4fcaSEnrico Granata OptionValueBoolean m_stop_on_error; 36960986174SMichael Sartain bool m_silent_run; 370e16c50a1SJim Ingham bool m_stop_on_continue; 371e16c50a1SJim Ingham }; 372e16c50a1SJim Ingham 373ebc09c36SJim Ingham bool 3745a988416SJim Ingham DoExecute(Args& command, CommandReturnObject &result) 375ebc09c36SJim Ingham { 376c7bece56SGreg Clayton const size_t argc = command.GetArgumentCount(); 377ebc09c36SJim Ingham if (argc == 1) 378ebc09c36SJim Ingham { 3795a988416SJim Ingham const char *filename = command.GetArgumentAtIndex(0); 380ebc09c36SJim Ingham 381*44d93782SGreg Clayton if (!m_interpreter.GetDebugger().GetCommandInterpreter().GetBatchCommandMode()) 382ebc09c36SJim Ingham result.AppendMessageWithFormat ("Executing commands in '%s'.\n", filename); 383ebc09c36SJim Ingham 3841ee3853fSJohnny Chen FileSpec cmd_file (filename, true); 385e16c50a1SJim Ingham ExecutionContext *exe_ctx = NULL; // Just use the default context. 38660986174SMichael Sartain bool echo_commands = !m_options.m_silent_run; 387e16c50a1SJim Ingham bool print_results = true; 388012d4fcaSEnrico Granata bool stop_on_error = m_options.m_stop_on_error.OptionWasSet() ? (bool)m_options.m_stop_on_error : m_interpreter.GetStopCmdSourceOnError(); 389ebc09c36SJim Ingham 390e16c50a1SJim Ingham m_interpreter.HandleCommandsFromFile (cmd_file, 391e16c50a1SJim Ingham exe_ctx, 392e16c50a1SJim Ingham m_options.m_stop_on_continue, 393012d4fcaSEnrico Granata stop_on_error, 394e16c50a1SJim Ingham echo_commands, 395e16c50a1SJim Ingham print_results, 3965f5ab602SEnrico Granata eLazyBoolCalculate, 397e16c50a1SJim Ingham result); 398ebc09c36SJim Ingham } 399ebc09c36SJim Ingham else 400ebc09c36SJim Ingham { 401ebc09c36SJim Ingham result.AppendErrorWithFormat("'%s' takes exactly one executable filename argument.\n", GetCommandName()); 402ebc09c36SJim Ingham result.SetStatus (eReturnStatusFailed); 403ebc09c36SJim Ingham } 404ebc09c36SJim Ingham return result.Succeeded(); 405ebc09c36SJim Ingham 406ebc09c36SJim Ingham } 4075a988416SJim Ingham CommandOptions m_options; 408ebc09c36SJim Ingham }; 409ebc09c36SJim Ingham 410e0d378b3SGreg Clayton OptionDefinition 411e16c50a1SJim Ingham CommandObjectCommandsSource::CommandOptions::g_option_table[] = 412e16c50a1SJim Ingham { 413e2607b50SVirgile Bello { LLDB_OPT_SET_ALL, false, "stop-on-error", 'e', OptionParser::eRequiredArgument, NULL, 0, eArgTypeBoolean, "If true, stop executing commands on error."}, 414e2607b50SVirgile Bello { LLDB_OPT_SET_ALL, false, "stop-on-continue", 'c', OptionParser::eRequiredArgument, NULL, 0, eArgTypeBoolean, "If true, stop executing commands on continue."}, 415e2607b50SVirgile Bello { LLDB_OPT_SET_ALL, false, "silent-run", 's', OptionParser::eRequiredArgument, NULL, 0, eArgTypeBoolean, "If true don't echo commands while executing."}, 416e16c50a1SJim Ingham { 0, false, NULL, 0, 0, NULL, 0, eArgTypeNone, NULL } 417e16c50a1SJim Ingham }; 418e16c50a1SJim Ingham 419ebc09c36SJim Ingham #pragma mark CommandObjectCommandsAlias 420ebc09c36SJim Ingham //------------------------------------------------------------------------- 421ebc09c36SJim Ingham // CommandObjectCommandsAlias 422ebc09c36SJim Ingham //------------------------------------------------------------------------- 423ebc09c36SJim Ingham 424be93a35aSEnrico Granata static const char *g_python_command_instructions = "Enter your Python command(s). Type 'DONE' to end.\n" 425be93a35aSEnrico Granata "You must define a Python function with this signature:\n" 426*44d93782SGreg Clayton "def my_command_impl(debugger, args, result, internal_dict):\n"; 427be93a35aSEnrico Granata 428be93a35aSEnrico Granata 4295a988416SJim Ingham class CommandObjectCommandsAlias : public CommandObjectRaw 430ebc09c36SJim Ingham { 431be93a35aSEnrico Granata 432be93a35aSEnrico Granata 433ebc09c36SJim Ingham public: 434a7015092SGreg Clayton CommandObjectCommandsAlias (CommandInterpreter &interpreter) : 4355a988416SJim Ingham CommandObjectRaw (interpreter, 4360e5e5a79SGreg Clayton "command alias", 437e3d26315SCaroline Tice "Allow users to define their own debugger command abbreviations.", 438405fe67fSCaroline Tice NULL) 439ebc09c36SJim Ingham { 440ebc09c36SJim Ingham SetHelpLong( 441ebc09c36SJim Ingham "'alias' allows the user to create a short-cut or abbreviation for long \n\ 442ebc09c36SJim Ingham commands, multi-word commands, and commands that take particular options. \n\ 443ebc09c36SJim Ingham Below are some simple examples of how one might use the 'alias' command: \n\ 44469c12ccbSJason Molenda \n 'command alias sc script' // Creates the abbreviation 'sc' for the 'script' \n\ 445ebc09c36SJim Ingham // command. \n\ 44669c12ccbSJason Molenda 'command alias bp breakpoint' // Creates the abbreviation 'bp' for the 'breakpoint' \n\ 447ebc09c36SJim Ingham // command. Since breakpoint commands are two-word \n\ 448ebc09c36SJim Ingham // commands, the user will still need to enter the \n\ 449ebc09c36SJim Ingham // second word after 'bp', e.g. 'bp enable' or \n\ 450ebc09c36SJim Ingham // 'bp delete'. \n\ 45169c12ccbSJason Molenda 'command alias bpl breakpoint list' // Creates the abbreviation 'bpl' for the \n\ 452ebc09c36SJim Ingham // two-word command 'breakpoint list'. \n\ 453ebc09c36SJim Ingham \nAn alias can include some options for the command, with the values either \n\ 454ebc09c36SJim Ingham filled in at the time the alias is created, or specified as positional \n\ 455ebc09c36SJim Ingham arguments, to be filled in when the alias is invoked. The following example \n\ 456ebc09c36SJim Ingham shows how to create aliases with options: \n\ 457ebc09c36SJim Ingham \n\ 45869c12ccbSJason Molenda 'command alias bfl breakpoint set -f %1 -l %2' \n\ 459ebc09c36SJim Ingham \nThis creates the abbreviation 'bfl' (for break-file-line), with the -f and -l \n\ 460ebc09c36SJim Ingham options already part of the alias. So if the user wants to set a breakpoint \n\ 461ebc09c36SJim Ingham by file and line without explicitly having to use the -f and -l options, the \n\ 462ebc09c36SJim Ingham user can now use 'bfl' instead. The '%1' and '%2' are positional placeholders \n\ 463ebc09c36SJim Ingham for the actual arguments that will be passed when the alias command is used. \n\ 464ebc09c36SJim Ingham The number in the placeholder refers to the position/order the actual value \n\ 46581ded935SJim Ingham occupies when the alias is used. All the occurrences of '%1' in the alias \n\ 466ebc09c36SJim Ingham will be replaced with the first argument, all the occurrences of '%2' in the \n\ 467ebc09c36SJim Ingham alias will be replaced with the second argument, and so on. This also allows \n\ 468ebc09c36SJim Ingham actual arguments to be used multiple times within an alias (see 'process \n\ 46981ded935SJim Ingham launch' example below). \n\ 47081ded935SJim Ingham Note: the positional arguments must substitute as whole words in the resultant\n\ 47181ded935SJim Ingham command, so you can't at present do something like:\n\ 47281ded935SJim Ingham \n\ 47369c12ccbSJason Molenda command alias bcppfl breakpoint set -f %1.cpp -l %2\n\ 47481ded935SJim Ingham \n\ 47581ded935SJim Ingham to get the file extension \".cpp\" automatically appended. For more complex\n\ 47681ded935SJim Ingham aliasing, use the \"command regex\" command instead.\n\ 47781ded935SJim Ingham \nSo in the 'bfl' case, the actual file value will be \n\ 478ebc09c36SJim Ingham filled in with the first argument following 'bfl' and the actual line number \n\ 479ebc09c36SJim Ingham value will be filled in with the second argument. The user would use this \n\ 480ebc09c36SJim Ingham alias as follows: \n\ 48169c12ccbSJason Molenda \n (lldb) command alias bfl breakpoint set -f %1 -l %2 \n\ 482ebc09c36SJim Ingham <... some time later ...> \n\ 48309799af6SCaroline Tice (lldb) bfl my-file.c 137 \n\ 484ebc09c36SJim Ingham \nThis would be the same as if the user had entered \n\ 485ebc09c36SJim Ingham 'breakpoint set -f my-file.c -l 137'. \n\ 486ebc09c36SJim Ingham \nAnother example: \n\ 48769c12ccbSJason Molenda \n (lldb) command alias pltty process launch -s -o %1 -e %1 \n\ 48809799af6SCaroline Tice (lldb) pltty /dev/tty0 \n\ 489ebc09c36SJim Ingham // becomes 'process launch -s -o /dev/tty0 -e /dev/tty0' \n\ 490ebc09c36SJim Ingham \nIf the user always wanted to pass the same value to a particular option, the \n\ 491ebc09c36SJim Ingham alias could be defined with that value directly in the alias as a constant, \n\ 492ebc09c36SJim Ingham rather than using a positional placeholder: \n\ 49369c12ccbSJason Molenda \n command alias bl3 breakpoint set -f %1 -l 3 // Always sets a breakpoint on line \n\ 494ebc09c36SJim Ingham // 3 of whatever file is indicated. \n"); 495ebc09c36SJim Ingham 496405fe67fSCaroline Tice CommandArgumentEntry arg1; 497405fe67fSCaroline Tice CommandArgumentEntry arg2; 498405fe67fSCaroline Tice CommandArgumentEntry arg3; 499405fe67fSCaroline Tice CommandArgumentData alias_arg; 500405fe67fSCaroline Tice CommandArgumentData cmd_arg; 501405fe67fSCaroline Tice CommandArgumentData options_arg; 502405fe67fSCaroline Tice 503405fe67fSCaroline Tice // Define the first (and only) variant of this arg. 504405fe67fSCaroline Tice alias_arg.arg_type = eArgTypeAliasName; 505405fe67fSCaroline Tice alias_arg.arg_repetition = eArgRepeatPlain; 506405fe67fSCaroline Tice 507405fe67fSCaroline Tice // There is only one variant this argument could be; put it into the argument entry. 508405fe67fSCaroline Tice arg1.push_back (alias_arg); 509405fe67fSCaroline Tice 510405fe67fSCaroline Tice // Define the first (and only) variant of this arg. 511405fe67fSCaroline Tice cmd_arg.arg_type = eArgTypeCommandName; 512405fe67fSCaroline Tice cmd_arg.arg_repetition = eArgRepeatPlain; 513405fe67fSCaroline Tice 514405fe67fSCaroline Tice // There is only one variant this argument could be; put it into the argument entry. 515405fe67fSCaroline Tice arg2.push_back (cmd_arg); 516405fe67fSCaroline Tice 517405fe67fSCaroline Tice // Define the first (and only) variant of this arg. 518405fe67fSCaroline Tice options_arg.arg_type = eArgTypeAliasOptions; 519405fe67fSCaroline Tice options_arg.arg_repetition = eArgRepeatOptional; 520405fe67fSCaroline Tice 521405fe67fSCaroline Tice // There is only one variant this argument could be; put it into the argument entry. 522405fe67fSCaroline Tice arg3.push_back (options_arg); 523405fe67fSCaroline Tice 524405fe67fSCaroline Tice // Push the data for the first argument into the m_arguments vector. 525405fe67fSCaroline Tice m_arguments.push_back (arg1); 526405fe67fSCaroline Tice m_arguments.push_back (arg2); 527405fe67fSCaroline Tice m_arguments.push_back (arg3); 528ebc09c36SJim Ingham } 529ebc09c36SJim Ingham 530ebc09c36SJim Ingham ~CommandObjectCommandsAlias () 531ebc09c36SJim Ingham { 532ebc09c36SJim Ingham } 533ebc09c36SJim Ingham 5345a988416SJim Ingham protected: 5355a988416SJim Ingham virtual bool 5365a988416SJim Ingham DoExecute (const char *raw_command_line, CommandReturnObject &result) 537844d2303SCaroline Tice { 538844d2303SCaroline Tice Args args (raw_command_line); 539844d2303SCaroline Tice std::string raw_command_string (raw_command_line); 540844d2303SCaroline Tice 541844d2303SCaroline Tice size_t argc = args.GetArgumentCount(); 542844d2303SCaroline Tice 543844d2303SCaroline Tice if (argc < 2) 544844d2303SCaroline Tice { 545844d2303SCaroline Tice result.AppendError ("'alias' requires at least two arguments"); 546844d2303SCaroline Tice result.SetStatus (eReturnStatusFailed); 547844d2303SCaroline Tice return false; 548844d2303SCaroline Tice } 549844d2303SCaroline Tice 550844d2303SCaroline Tice // Get the alias command. 551844d2303SCaroline Tice 552844d2303SCaroline Tice const std::string alias_command = args.GetArgumentAtIndex (0); 553844d2303SCaroline Tice 554844d2303SCaroline Tice // Strip the new alias name off 'raw_command_string' (leave it on args, which gets passed to 'Execute', which 555844d2303SCaroline Tice // does the stripping itself. 556844d2303SCaroline Tice size_t pos = raw_command_string.find (alias_command); 557844d2303SCaroline Tice if (pos == 0) 558844d2303SCaroline Tice { 559844d2303SCaroline Tice raw_command_string = raw_command_string.substr (alias_command.size()); 560844d2303SCaroline Tice pos = raw_command_string.find_first_not_of (' '); 561844d2303SCaroline Tice if ((pos != std::string::npos) && (pos > 0)) 562844d2303SCaroline Tice raw_command_string = raw_command_string.substr (pos); 563844d2303SCaroline Tice } 564844d2303SCaroline Tice else 565844d2303SCaroline Tice { 566844d2303SCaroline Tice result.AppendError ("Error parsing command string. No alias created."); 567844d2303SCaroline Tice result.SetStatus (eReturnStatusFailed); 568844d2303SCaroline Tice return false; 569844d2303SCaroline Tice } 570844d2303SCaroline Tice 571844d2303SCaroline Tice 572844d2303SCaroline Tice // Verify that the command is alias-able. 573844d2303SCaroline Tice if (m_interpreter.CommandExists (alias_command.c_str())) 574844d2303SCaroline Tice { 575844d2303SCaroline Tice result.AppendErrorWithFormat ("'%s' is a permanent debugger command and cannot be redefined.\n", 576844d2303SCaroline Tice alias_command.c_str()); 577844d2303SCaroline Tice result.SetStatus (eReturnStatusFailed); 578844d2303SCaroline Tice return false; 579844d2303SCaroline Tice } 580844d2303SCaroline Tice 581844d2303SCaroline Tice // Get CommandObject that is being aliased. The command name is read from the front of raw_command_string. 582844d2303SCaroline Tice // raw_command_string is returned with the name of the command object stripped off the front. 583844d2303SCaroline Tice CommandObject *cmd_obj = m_interpreter.GetCommandObjectForCommand (raw_command_string); 584844d2303SCaroline Tice 585844d2303SCaroline Tice if (!cmd_obj) 586844d2303SCaroline Tice { 58786edbf41SGreg Clayton result.AppendErrorWithFormat ("invalid command given to 'alias'. '%s' does not begin with a valid command." 588844d2303SCaroline Tice " No alias created.", raw_command_string.c_str()); 589844d2303SCaroline Tice result.SetStatus (eReturnStatusFailed); 590844d2303SCaroline Tice return false; 591844d2303SCaroline Tice } 592844d2303SCaroline Tice else if (!cmd_obj->WantsRawCommandString ()) 593844d2303SCaroline Tice { 594844d2303SCaroline Tice // Note that args was initialized with the original command, and has not been updated to this point. 595844d2303SCaroline Tice // Therefore can we pass it to the version of Execute that does not need/expect raw input in the alias. 5965a988416SJim Ingham return HandleAliasingNormalCommand (args, result); 597844d2303SCaroline Tice } 598844d2303SCaroline Tice else 599844d2303SCaroline Tice { 6005a988416SJim Ingham return HandleAliasingRawCommand (alias_command, raw_command_string, *cmd_obj, result); 6015a988416SJim Ingham } 6025a988416SJim Ingham return result.Succeeded(); 6035a988416SJim Ingham } 6045a988416SJim Ingham 6055a988416SJim Ingham bool 6065a988416SJim Ingham HandleAliasingRawCommand (const std::string &alias_command, std::string &raw_command_string, CommandObject &cmd_obj, CommandReturnObject &result) 6075a988416SJim Ingham { 608844d2303SCaroline Tice // Verify & handle any options/arguments passed to the alias command 609844d2303SCaroline Tice 610844d2303SCaroline Tice OptionArgVectorSP option_arg_vector_sp = OptionArgVectorSP (new OptionArgVector); 611844d2303SCaroline Tice OptionArgVector *option_arg_vector = option_arg_vector_sp.get(); 612844d2303SCaroline Tice 6135a988416SJim Ingham CommandObjectSP cmd_obj_sp = m_interpreter.GetCommandSPExact (cmd_obj.GetCommandName(), false); 614844d2303SCaroline Tice 615ca90c47eSCaroline Tice if (!m_interpreter.ProcessAliasOptionsArgs (cmd_obj_sp, raw_command_string.c_str(), option_arg_vector_sp)) 616844d2303SCaroline Tice { 617844d2303SCaroline Tice result.AppendError ("Unable to create requested alias.\n"); 618ca90c47eSCaroline Tice result.SetStatus (eReturnStatusFailed); 619844d2303SCaroline Tice return false; 620844d2303SCaroline Tice } 621844d2303SCaroline Tice 622844d2303SCaroline Tice // Create the alias 623844d2303SCaroline Tice if (m_interpreter.AliasExists (alias_command.c_str()) 624844d2303SCaroline Tice || m_interpreter.UserCommandExists (alias_command.c_str())) 625844d2303SCaroline Tice { 626844d2303SCaroline Tice OptionArgVectorSP temp_option_arg_sp (m_interpreter.GetAliasOptions (alias_command.c_str())); 627844d2303SCaroline Tice if (temp_option_arg_sp.get()) 628844d2303SCaroline Tice { 629844d2303SCaroline Tice if (option_arg_vector->size() == 0) 630844d2303SCaroline Tice m_interpreter.RemoveAliasOptions (alias_command.c_str()); 631844d2303SCaroline Tice } 632844d2303SCaroline Tice result.AppendWarningWithFormat ("Overwriting existing definition for '%s'.\n", 633844d2303SCaroline Tice alias_command.c_str()); 634844d2303SCaroline Tice } 635844d2303SCaroline Tice 636472362e6SCaroline Tice if (cmd_obj_sp) 637472362e6SCaroline Tice { 638844d2303SCaroline Tice m_interpreter.AddAlias (alias_command.c_str(), cmd_obj_sp); 639844d2303SCaroline Tice if (option_arg_vector->size() > 0) 640844d2303SCaroline Tice m_interpreter.AddOrReplaceAliasOptions (alias_command.c_str(), option_arg_vector_sp); 641844d2303SCaroline Tice result.SetStatus (eReturnStatusSuccessFinishNoResult); 642844d2303SCaroline Tice } 643472362e6SCaroline Tice else 644472362e6SCaroline Tice { 645472362e6SCaroline Tice result.AppendError ("Unable to create requested alias.\n"); 646472362e6SCaroline Tice result.SetStatus (eReturnStatusFailed); 647472362e6SCaroline Tice } 648844d2303SCaroline Tice return result.Succeeded (); 649844d2303SCaroline Tice } 650ebc09c36SJim Ingham 651ebc09c36SJim Ingham bool 6525a988416SJim Ingham HandleAliasingNormalCommand (Args& args, CommandReturnObject &result) 653ebc09c36SJim Ingham { 654867b185dSCaroline Tice size_t argc = args.GetArgumentCount(); 655ebc09c36SJim Ingham 656ebc09c36SJim Ingham if (argc < 2) 657ebc09c36SJim Ingham { 658ebc09c36SJim Ingham result.AppendError ("'alias' requires at least two arguments"); 659ebc09c36SJim Ingham result.SetStatus (eReturnStatusFailed); 660ebc09c36SJim Ingham return false; 661ebc09c36SJim Ingham } 662ebc09c36SJim Ingham 663ebc09c36SJim Ingham const std::string alias_command = args.GetArgumentAtIndex(0); 664ebc09c36SJim Ingham const std::string actual_command = args.GetArgumentAtIndex(1); 665ebc09c36SJim Ingham 666ebc09c36SJim Ingham args.Shift(); // Shift the alias command word off the argument vector. 667ebc09c36SJim Ingham args.Shift(); // Shift the old command word off the argument vector. 668ebc09c36SJim Ingham 669ebc09c36SJim Ingham // Verify that the command is alias'able, and get the appropriate command object. 670ebc09c36SJim Ingham 671a7015092SGreg Clayton if (m_interpreter.CommandExists (alias_command.c_str())) 672ebc09c36SJim Ingham { 673ebc09c36SJim Ingham result.AppendErrorWithFormat ("'%s' is a permanent debugger command and cannot be redefined.\n", 674ebc09c36SJim Ingham alias_command.c_str()); 675ebc09c36SJim Ingham result.SetStatus (eReturnStatusFailed); 676ebc09c36SJim Ingham } 677ebc09c36SJim Ingham else 678ebc09c36SJim Ingham { 679a7015092SGreg Clayton CommandObjectSP command_obj_sp(m_interpreter.GetCommandSPExact (actual_command.c_str(), true)); 680ebc09c36SJim Ingham CommandObjectSP subcommand_obj_sp; 681ebc09c36SJim Ingham bool use_subcommand = false; 682ebc09c36SJim Ingham if (command_obj_sp.get()) 683ebc09c36SJim Ingham { 684ebc09c36SJim Ingham CommandObject *cmd_obj = command_obj_sp.get(); 685c982c768SGreg Clayton CommandObject *sub_cmd_obj = NULL; 686ebc09c36SJim Ingham OptionArgVectorSP option_arg_vector_sp = OptionArgVectorSP (new OptionArgVector); 687ebc09c36SJim Ingham OptionArgVector *option_arg_vector = option_arg_vector_sp.get(); 688ebc09c36SJim Ingham 689844d2303SCaroline Tice while (cmd_obj->IsMultiwordObject() && args.GetArgumentCount() > 0) 690ebc09c36SJim Ingham { 691ebc09c36SJim Ingham if (argc >= 3) 692ebc09c36SJim Ingham { 693ebc09c36SJim Ingham const std::string sub_command = args.GetArgumentAtIndex(0); 694ebc09c36SJim Ingham assert (sub_command.length() != 0); 695998255bfSGreg Clayton subcommand_obj_sp = cmd_obj->GetSubcommandSP (sub_command.c_str()); 696ebc09c36SJim Ingham if (subcommand_obj_sp.get()) 697ebc09c36SJim Ingham { 698ebc09c36SJim Ingham sub_cmd_obj = subcommand_obj_sp.get(); 699ebc09c36SJim Ingham use_subcommand = true; 700ebc09c36SJim Ingham args.Shift(); // Shift the sub_command word off the argument vector. 701844d2303SCaroline Tice cmd_obj = sub_cmd_obj; 702ebc09c36SJim Ingham } 703ebc09c36SJim Ingham else 704ebc09c36SJim Ingham { 705f415eeb4SCaroline Tice result.AppendErrorWithFormat("'%s' is not a valid sub-command of '%s'. " 706f415eeb4SCaroline Tice "Unable to create alias.\n", 707f415eeb4SCaroline Tice sub_command.c_str(), actual_command.c_str()); 708ebc09c36SJim Ingham result.SetStatus (eReturnStatusFailed); 709ebc09c36SJim Ingham return false; 710ebc09c36SJim Ingham } 711ebc09c36SJim Ingham } 712ebc09c36SJim Ingham } 713ebc09c36SJim Ingham 714ebc09c36SJim Ingham // Verify & handle any options/arguments passed to the alias command 715ebc09c36SJim Ingham 716ebc09c36SJim Ingham if (args.GetArgumentCount () > 0) 717ebc09c36SJim Ingham { 718ca90c47eSCaroline Tice CommandObjectSP tmp_sp = m_interpreter.GetCommandSPExact (cmd_obj->GetCommandName(), false); 719ebc09c36SJim Ingham if (use_subcommand) 720ca90c47eSCaroline Tice tmp_sp = m_interpreter.GetCommandSPExact (sub_cmd_obj->GetCommandName(), false); 721ca90c47eSCaroline Tice 722ca90c47eSCaroline Tice std::string args_string; 723ca90c47eSCaroline Tice args.GetCommandString (args_string); 724ca90c47eSCaroline Tice 725ca90c47eSCaroline Tice if (!m_interpreter.ProcessAliasOptionsArgs (tmp_sp, args_string.c_str(), option_arg_vector_sp)) 726ebc09c36SJim Ingham { 727ca90c47eSCaroline Tice result.AppendError ("Unable to create requested alias.\n"); 728ca90c47eSCaroline Tice result.SetStatus (eReturnStatusFailed); 729e7941795SCaroline Tice return false; 730867b185dSCaroline Tice } 731867b185dSCaroline Tice } 732867b185dSCaroline Tice 733ebc09c36SJim Ingham // Create the alias. 734ebc09c36SJim Ingham 735a7015092SGreg Clayton if (m_interpreter.AliasExists (alias_command.c_str()) 736a7015092SGreg Clayton || m_interpreter.UserCommandExists (alias_command.c_str())) 737ebc09c36SJim Ingham { 738a7015092SGreg Clayton OptionArgVectorSP tmp_option_arg_sp (m_interpreter.GetAliasOptions (alias_command.c_str())); 739ebc09c36SJim Ingham if (tmp_option_arg_sp.get()) 740ebc09c36SJim Ingham { 741ebc09c36SJim Ingham if (option_arg_vector->size() == 0) 742a7015092SGreg Clayton m_interpreter.RemoveAliasOptions (alias_command.c_str()); 743ebc09c36SJim Ingham } 744ebc09c36SJim Ingham result.AppendWarningWithFormat ("Overwriting existing definition for '%s'.\n", 745ebc09c36SJim Ingham alias_command.c_str()); 746ebc09c36SJim Ingham } 747ebc09c36SJim Ingham 748ebc09c36SJim Ingham if (use_subcommand) 749a7015092SGreg Clayton m_interpreter.AddAlias (alias_command.c_str(), subcommand_obj_sp); 750ebc09c36SJim Ingham else 751a7015092SGreg Clayton m_interpreter.AddAlias (alias_command.c_str(), command_obj_sp); 752ebc09c36SJim Ingham if (option_arg_vector->size() > 0) 753a7015092SGreg Clayton m_interpreter.AddOrReplaceAliasOptions (alias_command.c_str(), option_arg_vector_sp); 754ebc09c36SJim Ingham result.SetStatus (eReturnStatusSuccessFinishNoResult); 755ebc09c36SJim Ingham } 756ebc09c36SJim Ingham else 757ebc09c36SJim Ingham { 758ebc09c36SJim Ingham result.AppendErrorWithFormat ("'%s' is not an existing command.\n", actual_command.c_str()); 759ebc09c36SJim Ingham result.SetStatus (eReturnStatusFailed); 760e7941795SCaroline Tice return false; 761ebc09c36SJim Ingham } 762ebc09c36SJim Ingham } 763ebc09c36SJim Ingham 764ebc09c36SJim Ingham return result.Succeeded(); 765ebc09c36SJim Ingham } 7665a988416SJim Ingham 767ebc09c36SJim Ingham }; 768ebc09c36SJim Ingham 769ebc09c36SJim Ingham #pragma mark CommandObjectCommandsUnalias 770ebc09c36SJim Ingham //------------------------------------------------------------------------- 771ebc09c36SJim Ingham // CommandObjectCommandsUnalias 772ebc09c36SJim Ingham //------------------------------------------------------------------------- 773ebc09c36SJim Ingham 7745a988416SJim Ingham class CommandObjectCommandsUnalias : public CommandObjectParsed 775ebc09c36SJim Ingham { 776ebc09c36SJim Ingham public: 777a7015092SGreg Clayton CommandObjectCommandsUnalias (CommandInterpreter &interpreter) : 7785a988416SJim Ingham CommandObjectParsed (interpreter, 7790e5e5a79SGreg Clayton "command unalias", 78086ddae50SCaroline Tice "Allow the user to remove/delete a user-defined command abbreviation.", 781405fe67fSCaroline Tice NULL) 782ebc09c36SJim Ingham { 783405fe67fSCaroline Tice CommandArgumentEntry arg; 784405fe67fSCaroline Tice CommandArgumentData alias_arg; 785405fe67fSCaroline Tice 786405fe67fSCaroline Tice // Define the first (and only) variant of this arg. 787405fe67fSCaroline Tice alias_arg.arg_type = eArgTypeAliasName; 788405fe67fSCaroline Tice alias_arg.arg_repetition = eArgRepeatPlain; 789405fe67fSCaroline Tice 790405fe67fSCaroline Tice // There is only one variant this argument could be; put it into the argument entry. 791405fe67fSCaroline Tice arg.push_back (alias_arg); 792405fe67fSCaroline Tice 793405fe67fSCaroline Tice // Push the data for the first argument into the m_arguments vector. 794405fe67fSCaroline Tice m_arguments.push_back (arg); 795ebc09c36SJim Ingham } 796ebc09c36SJim Ingham 797ebc09c36SJim Ingham ~CommandObjectCommandsUnalias() 798ebc09c36SJim Ingham { 799ebc09c36SJim Ingham } 800ebc09c36SJim Ingham 8015a988416SJim Ingham protected: 802ebc09c36SJim Ingham bool 8035a988416SJim Ingham DoExecute (Args& args, CommandReturnObject &result) 804ebc09c36SJim Ingham { 805ebc09c36SJim Ingham CommandObject::CommandMap::iterator pos; 806ebc09c36SJim Ingham CommandObject *cmd_obj; 807ebc09c36SJim Ingham 808ebc09c36SJim Ingham if (args.GetArgumentCount() != 0) 809ebc09c36SJim Ingham { 810ebc09c36SJim Ingham const char *command_name = args.GetArgumentAtIndex(0); 811a7015092SGreg Clayton cmd_obj = m_interpreter.GetCommandObject(command_name); 812ebc09c36SJim Ingham if (cmd_obj) 813ebc09c36SJim Ingham { 814a7015092SGreg Clayton if (m_interpreter.CommandExists (command_name)) 815ebc09c36SJim Ingham { 816ebc09c36SJim Ingham result.AppendErrorWithFormat ("'%s' is a permanent debugger command and cannot be removed.\n", 817ebc09c36SJim Ingham command_name); 818ebc09c36SJim Ingham result.SetStatus (eReturnStatusFailed); 819ebc09c36SJim Ingham } 820ebc09c36SJim Ingham else 821ebc09c36SJim Ingham { 822ebc09c36SJim Ingham 823a7015092SGreg Clayton if (m_interpreter.RemoveAlias (command_name) == false) 824ebc09c36SJim Ingham { 825a7015092SGreg Clayton if (m_interpreter.AliasExists (command_name)) 826ebc09c36SJim Ingham result.AppendErrorWithFormat ("Error occurred while attempting to unalias '%s'.\n", 827ebc09c36SJim Ingham command_name); 828ebc09c36SJim Ingham else 829ebc09c36SJim Ingham result.AppendErrorWithFormat ("'%s' is not an existing alias.\n", command_name); 830ebc09c36SJim Ingham result.SetStatus (eReturnStatusFailed); 831ebc09c36SJim Ingham } 832ebc09c36SJim Ingham else 833ebc09c36SJim Ingham result.SetStatus (eReturnStatusSuccessFinishNoResult); 834ebc09c36SJim Ingham } 835ebc09c36SJim Ingham } 836ebc09c36SJim Ingham else 837ebc09c36SJim Ingham { 838ebc09c36SJim Ingham result.AppendErrorWithFormat ("'%s' is not a known command.\nTry 'help' to see a " 839ebc09c36SJim Ingham "current list of commands.\n", 840ebc09c36SJim Ingham command_name); 841ebc09c36SJim Ingham result.SetStatus (eReturnStatusFailed); 842ebc09c36SJim Ingham } 843ebc09c36SJim Ingham } 844ebc09c36SJim Ingham else 845ebc09c36SJim Ingham { 846ebc09c36SJim Ingham result.AppendError ("must call 'unalias' with a valid alias"); 847ebc09c36SJim Ingham result.SetStatus (eReturnStatusFailed); 848ebc09c36SJim Ingham } 849ebc09c36SJim Ingham 850ebc09c36SJim Ingham return result.Succeeded(); 851ebc09c36SJim Ingham } 852ebc09c36SJim Ingham }; 853ebc09c36SJim Ingham 854de164aaaSGreg Clayton //------------------------------------------------------------------------- 855de164aaaSGreg Clayton // CommandObjectCommandsAddRegex 856de164aaaSGreg Clayton //------------------------------------------------------------------------- 8575a988416SJim Ingham #pragma mark CommandObjectCommandsAddRegex 858de164aaaSGreg Clayton 859*44d93782SGreg Clayton class CommandObjectCommandsAddRegex : 860*44d93782SGreg Clayton public CommandObjectParsed, 861*44d93782SGreg Clayton public IOHandlerDelegate 862de164aaaSGreg Clayton { 863de164aaaSGreg Clayton public: 864de164aaaSGreg Clayton CommandObjectCommandsAddRegex (CommandInterpreter &interpreter) : 8655a988416SJim Ingham CommandObjectParsed (interpreter, 8660e5e5a79SGreg Clayton "command regex", 867de164aaaSGreg Clayton "Allow the user to create a regular expression command.", 8680e5e5a79SGreg Clayton "command regex <cmd-name> [s/<regex>/<subst>/ ...]"), 869*44d93782SGreg Clayton IOHandlerDelegate(IOHandlerDelegate::Completion::LLDBCommand), 870de164aaaSGreg Clayton m_options (interpreter) 871de164aaaSGreg Clayton { 8720e5e5a79SGreg Clayton SetHelpLong( 8730e5e5a79SGreg Clayton "This command allows the user to create powerful regular expression commands\n" 8740e5e5a79SGreg Clayton "with substitutions. The regular expressions and substitutions are specified\n" 8750e5e5a79SGreg Clayton "using the regular exression substitution format of:\n" 8760e5e5a79SGreg Clayton "\n" 8770e5e5a79SGreg Clayton " s/<regex>/<subst>/\n" 8780e5e5a79SGreg Clayton "\n" 8790e5e5a79SGreg Clayton "<regex> is a regular expression that can use parenthesis to capture regular\n" 8800e5e5a79SGreg Clayton "expression input and substitute the captured matches in the output using %1\n" 8810e5e5a79SGreg Clayton "for the first match, %2 for the second, and so on.\n" 8820e5e5a79SGreg Clayton "\n" 8830e5e5a79SGreg Clayton "The regular expressions can all be specified on the command line if more than\n" 8840e5e5a79SGreg Clayton "one argument is provided. If just the command name is provided on the command\n" 8850e5e5a79SGreg Clayton "line, then the regular expressions and substitutions can be entered on separate\n" 8860e5e5a79SGreg Clayton " lines, followed by an empty line to terminate the command definition.\n" 8870e5e5a79SGreg Clayton "\n" 8880e5e5a79SGreg Clayton "EXAMPLES\n" 8890e5e5a79SGreg Clayton "\n" 890adc43c99SSean Callanan "The following example will define a regular expression command named 'f' that\n" 8910e5e5a79SGreg Clayton "will call 'finish' if there are no arguments, or 'frame select <frame-idx>' if\n" 8920e5e5a79SGreg Clayton "a number follows 'f':\n" 893adc43c99SSean Callanan "\n" 8940e5e5a79SGreg Clayton " (lldb) command regex f s/^$/finish/ 's/([0-9]+)/frame select %1/'\n" 895adc43c99SSean Callanan "\n" 8960e5e5a79SGreg Clayton ); 897de164aaaSGreg Clayton } 898de164aaaSGreg Clayton 899de164aaaSGreg Clayton ~CommandObjectCommandsAddRegex() 900de164aaaSGreg Clayton { 901de164aaaSGreg Clayton } 902de164aaaSGreg Clayton 903de164aaaSGreg Clayton 9045a988416SJim Ingham protected: 905*44d93782SGreg Clayton 906*44d93782SGreg Clayton virtual void 907*44d93782SGreg Clayton IOHandlerActivated (IOHandler &io_handler) 908*44d93782SGreg Clayton { 909*44d93782SGreg Clayton StreamFileSP output_sp(io_handler.GetOutputStreamFile()); 910*44d93782SGreg Clayton if (output_sp) 911*44d93782SGreg Clayton { 912*44d93782SGreg Clayton output_sp->PutCString("Enter one of more sed substitution commands in the form: 's/<regex>/<subst>/'.\nTerminate the substitution list with an empty line.\n"); 913*44d93782SGreg Clayton output_sp->Flush(); 914*44d93782SGreg Clayton } 915*44d93782SGreg Clayton } 916*44d93782SGreg Clayton 917*44d93782SGreg Clayton virtual void 918*44d93782SGreg Clayton IOHandlerInputComplete (IOHandler &io_handler, std::string &data) 919*44d93782SGreg Clayton { 920*44d93782SGreg Clayton io_handler.SetIsDone(true); 921*44d93782SGreg Clayton if (m_regex_cmd_ap.get()) 922*44d93782SGreg Clayton { 923*44d93782SGreg Clayton StringList lines; 924*44d93782SGreg Clayton if (lines.SplitIntoLines (data)) 925*44d93782SGreg Clayton { 926*44d93782SGreg Clayton const size_t num_lines = lines.GetSize(); 927*44d93782SGreg Clayton bool check_only = false; 928*44d93782SGreg Clayton for (size_t i=0; i<num_lines; ++i) 929*44d93782SGreg Clayton { 930*44d93782SGreg Clayton printf ("regex[%zu] = %s\n", i, lines[i].c_str()); 931*44d93782SGreg Clayton llvm::StringRef bytes_strref (lines[i]); 932*44d93782SGreg Clayton Error error = AppendRegexSubstitution (bytes_strref, check_only); 933*44d93782SGreg Clayton if (error.Fail()) 934*44d93782SGreg Clayton { 935*44d93782SGreg Clayton if (!m_interpreter.GetDebugger().GetCommandInterpreter().GetBatchCommandMode()) 936*44d93782SGreg Clayton { 937*44d93782SGreg Clayton StreamSP out_stream = m_interpreter.GetDebugger().GetAsyncOutputStream(); 938*44d93782SGreg Clayton out_stream->Printf("error: %s\n", error.AsCString()); 939*44d93782SGreg Clayton } 940*44d93782SGreg Clayton } 941*44d93782SGreg Clayton } 942*44d93782SGreg Clayton } 943*44d93782SGreg Clayton if (m_regex_cmd_ap->HasRegexEntries()) 944*44d93782SGreg Clayton { 945*44d93782SGreg Clayton CommandObjectSP cmd_sp (m_regex_cmd_ap.release()); 946*44d93782SGreg Clayton m_interpreter.AddCommand(cmd_sp->GetCommandName(), cmd_sp, true); 947*44d93782SGreg Clayton } 948*44d93782SGreg Clayton } 949*44d93782SGreg Clayton } 950*44d93782SGreg Clayton 951*44d93782SGreg Clayton virtual LineStatus 952*44d93782SGreg Clayton IOHandlerLinesUpdated (IOHandler &io_handler, 953*44d93782SGreg Clayton StringList &lines, 954*44d93782SGreg Clayton uint32_t line_idx, 955*44d93782SGreg Clayton Error &error) 956*44d93782SGreg Clayton { 957*44d93782SGreg Clayton if (line_idx == UINT32_MAX) 958*44d93782SGreg Clayton { 959*44d93782SGreg Clayton // Return true to indicate we are done getting lines (this 960*44d93782SGreg Clayton // is a "fake" line - the real terminating blank line was 961*44d93782SGreg Clayton // removed during a previous call with the code below) 962*44d93782SGreg Clayton error.Clear(); 963*44d93782SGreg Clayton return LineStatus::Done; 964*44d93782SGreg Clayton } 965*44d93782SGreg Clayton else 966*44d93782SGreg Clayton { 967*44d93782SGreg Clayton const size_t num_lines = lines.GetSize(); 968*44d93782SGreg Clayton if (line_idx + 1 == num_lines) 969*44d93782SGreg Clayton { 970*44d93782SGreg Clayton // The last line was edited, if this line is empty, then we are done 971*44d93782SGreg Clayton // getting our multiple lines. 972*44d93782SGreg Clayton if (lines[line_idx].empty()) 973*44d93782SGreg Clayton { 974*44d93782SGreg Clayton // Remove the last empty line from "lines" so it doesn't appear 975*44d93782SGreg Clayton // in our final expression and return true to indicate we are done 976*44d93782SGreg Clayton // getting lines 977*44d93782SGreg Clayton lines.PopBack(); 978*44d93782SGreg Clayton return LineStatus::Done; 979*44d93782SGreg Clayton } 980*44d93782SGreg Clayton } 981*44d93782SGreg Clayton // Check the current line to make sure it is formatted correctly 982*44d93782SGreg Clayton bool check_only = true; 983*44d93782SGreg Clayton llvm::StringRef regex_sed(lines[line_idx]); 984*44d93782SGreg Clayton error = AppendRegexSubstitution (regex_sed, check_only); 985*44d93782SGreg Clayton if (error.Fail()) 986*44d93782SGreg Clayton { 987*44d93782SGreg Clayton return LineStatus::Error; 988*44d93782SGreg Clayton } 989*44d93782SGreg Clayton else 990*44d93782SGreg Clayton { 991*44d93782SGreg Clayton return LineStatus::Success; 992*44d93782SGreg Clayton } 993*44d93782SGreg Clayton } 994*44d93782SGreg Clayton } 995*44d93782SGreg Clayton 996de164aaaSGreg Clayton bool 9975a988416SJim Ingham DoExecute (Args& command, CommandReturnObject &result) 998de164aaaSGreg Clayton { 9995a988416SJim Ingham const size_t argc = command.GetArgumentCount(); 10000e5e5a79SGreg Clayton if (argc == 0) 1001de164aaaSGreg Clayton { 100269c12ccbSJason Molenda result.AppendError ("usage: 'command regex <command-name> [s/<regex1>/<subst1>/ s/<regex2>/<subst2>/ ...]'\n"); 10030e5e5a79SGreg Clayton result.SetStatus (eReturnStatusFailed); 10040e5e5a79SGreg Clayton } 10050e5e5a79SGreg Clayton else 10060e5e5a79SGreg Clayton { 10070e5e5a79SGreg Clayton Error error; 10085a988416SJim Ingham const char *name = command.GetArgumentAtIndex(0); 1009de164aaaSGreg Clayton m_regex_cmd_ap.reset (new CommandObjectRegexCommand (m_interpreter, 1010de164aaaSGreg Clayton name, 1011de164aaaSGreg Clayton m_options.GetHelp (), 1012de164aaaSGreg Clayton m_options.GetSyntax (), 1013de164aaaSGreg Clayton 10)); 10140e5e5a79SGreg Clayton 10150e5e5a79SGreg Clayton if (argc == 1) 10160e5e5a79SGreg Clayton { 1017*44d93782SGreg Clayton Debugger &debugger = m_interpreter.GetDebugger(); 1018*44d93782SGreg Clayton const bool multiple_lines = true; // Get multiple lines 1019*44d93782SGreg Clayton IOHandlerSP io_handler_sp (new IOHandlerEditline (debugger, 1020*44d93782SGreg Clayton "lldb", // Name of input reader for history 1021*44d93782SGreg Clayton "\033[K> ", // Prompt and clear line 1022*44d93782SGreg Clayton multiple_lines, 1023*44d93782SGreg Clayton *this)); 1024*44d93782SGreg Clayton 1025*44d93782SGreg Clayton if (io_handler_sp) 1026de164aaaSGreg Clayton { 1027*44d93782SGreg Clayton debugger.PushIOHandler(io_handler_sp); 1028de164aaaSGreg Clayton result.SetStatus (eReturnStatusSuccessFinishNoResult); 1029de164aaaSGreg Clayton } 1030de164aaaSGreg Clayton } 1031de164aaaSGreg Clayton else 1032de164aaaSGreg Clayton { 10330e5e5a79SGreg Clayton for (size_t arg_idx = 1; arg_idx < argc; ++arg_idx) 10340e5e5a79SGreg Clayton { 10355a988416SJim Ingham llvm::StringRef arg_strref (command.GetArgumentAtIndex(arg_idx)); 1036*44d93782SGreg Clayton bool check_only = false; 1037*44d93782SGreg Clayton error = AppendRegexSubstitution (arg_strref, check_only); 10380e5e5a79SGreg Clayton if (error.Fail()) 10390e5e5a79SGreg Clayton break; 10400e5e5a79SGreg Clayton } 10410e5e5a79SGreg Clayton 10420e5e5a79SGreg Clayton if (error.Success()) 10430e5e5a79SGreg Clayton { 10440e5e5a79SGreg Clayton AddRegexCommandToInterpreter(); 10450e5e5a79SGreg Clayton } 10460e5e5a79SGreg Clayton } 10470e5e5a79SGreg Clayton if (error.Fail()) 10480e5e5a79SGreg Clayton { 10490e5e5a79SGreg Clayton result.AppendError (error.AsCString()); 1050de164aaaSGreg Clayton result.SetStatus (eReturnStatusFailed); 1051de164aaaSGreg Clayton } 10520e5e5a79SGreg Clayton } 10530e5e5a79SGreg Clayton 1054de164aaaSGreg Clayton return result.Succeeded(); 1055de164aaaSGreg Clayton } 1056de164aaaSGreg Clayton 10570e5e5a79SGreg Clayton Error 1058*44d93782SGreg Clayton AppendRegexSubstitution (const llvm::StringRef ®ex_sed, bool check_only) 1059de164aaaSGreg Clayton { 10600e5e5a79SGreg Clayton Error error; 10610e5e5a79SGreg Clayton 10620e5e5a79SGreg Clayton if (m_regex_cmd_ap.get() == NULL) 1063de164aaaSGreg Clayton { 10640e5e5a79SGreg Clayton error.SetErrorStringWithFormat("invalid regular expression command object for: '%.*s'", 10650e5e5a79SGreg Clayton (int)regex_sed.size(), 10660e5e5a79SGreg Clayton regex_sed.data()); 10670e5e5a79SGreg Clayton return error; 1068de164aaaSGreg Clayton } 10690e5e5a79SGreg Clayton 10700e5e5a79SGreg Clayton size_t regex_sed_size = regex_sed.size(); 10710e5e5a79SGreg Clayton 10720e5e5a79SGreg Clayton if (regex_sed_size <= 1) 10730e5e5a79SGreg Clayton { 10740e5e5a79SGreg Clayton error.SetErrorStringWithFormat("regular expression substitution string is too short: '%.*s'", 10750e5e5a79SGreg Clayton (int)regex_sed.size(), 10760e5e5a79SGreg Clayton regex_sed.data()); 10770e5e5a79SGreg Clayton return error; 10780e5e5a79SGreg Clayton } 10790e5e5a79SGreg Clayton 10800e5e5a79SGreg Clayton if (regex_sed[0] != 's') 10810e5e5a79SGreg Clayton { 10820e5e5a79SGreg Clayton error.SetErrorStringWithFormat("regular expression substitution string doesn't start with 's': '%.*s'", 10830e5e5a79SGreg Clayton (int)regex_sed.size(), 10840e5e5a79SGreg Clayton regex_sed.data()); 10850e5e5a79SGreg Clayton return error; 10860e5e5a79SGreg Clayton } 10870e5e5a79SGreg Clayton const size_t first_separator_char_pos = 1; 10880e5e5a79SGreg Clayton // use the char that follows 's' as the regex separator character 10890e5e5a79SGreg Clayton // so we can have "s/<regex>/<subst>/" or "s|<regex>|<subst>|" 10900e5e5a79SGreg Clayton const char separator_char = regex_sed[first_separator_char_pos]; 10910e5e5a79SGreg Clayton const size_t second_separator_char_pos = regex_sed.find (separator_char, first_separator_char_pos + 1); 10920e5e5a79SGreg Clayton 10930e5e5a79SGreg Clayton if (second_separator_char_pos == std::string::npos) 10940e5e5a79SGreg Clayton { 10950e5e5a79SGreg Clayton error.SetErrorStringWithFormat("missing second '%c' separator char after '%.*s'", 10960e5e5a79SGreg Clayton separator_char, 10970e5e5a79SGreg Clayton (int)(regex_sed.size() - first_separator_char_pos - 1), 10980e5e5a79SGreg Clayton regex_sed.data() + (first_separator_char_pos + 1)); 10990e5e5a79SGreg Clayton return error; 11000e5e5a79SGreg Clayton } 11010e5e5a79SGreg Clayton 11020e5e5a79SGreg Clayton const size_t third_separator_char_pos = regex_sed.find (separator_char, second_separator_char_pos + 1); 11030e5e5a79SGreg Clayton 11040e5e5a79SGreg Clayton if (third_separator_char_pos == std::string::npos) 11050e5e5a79SGreg Clayton { 11060e5e5a79SGreg Clayton error.SetErrorStringWithFormat("missing third '%c' separator char after '%.*s'", 11070e5e5a79SGreg Clayton separator_char, 11080e5e5a79SGreg Clayton (int)(regex_sed.size() - second_separator_char_pos - 1), 11090e5e5a79SGreg Clayton regex_sed.data() + (second_separator_char_pos + 1)); 11100e5e5a79SGreg Clayton return error; 11110e5e5a79SGreg Clayton } 11120e5e5a79SGreg Clayton 11130e5e5a79SGreg Clayton if (third_separator_char_pos != regex_sed_size - 1) 11140e5e5a79SGreg Clayton { 11150e5e5a79SGreg Clayton // Make sure that everything that follows the last regex 11160e5e5a79SGreg Clayton // separator char 11170e5e5a79SGreg Clayton if (regex_sed.find_first_not_of("\t\n\v\f\r ", third_separator_char_pos + 1) != std::string::npos) 11180e5e5a79SGreg Clayton { 11190e5e5a79SGreg Clayton error.SetErrorStringWithFormat("extra data found after the '%.*s' regular expression substitution string: '%.*s'", 11200e5e5a79SGreg Clayton (int)third_separator_char_pos + 1, 11210e5e5a79SGreg Clayton regex_sed.data(), 11220e5e5a79SGreg Clayton (int)(regex_sed.size() - third_separator_char_pos - 1), 11230e5e5a79SGreg Clayton regex_sed.data() + (third_separator_char_pos + 1)); 11240e5e5a79SGreg Clayton return error; 11250e5e5a79SGreg Clayton } 11260e5e5a79SGreg Clayton 11270e5e5a79SGreg Clayton } 11280e5e5a79SGreg Clayton else if (first_separator_char_pos + 1 == second_separator_char_pos) 11290e5e5a79SGreg Clayton { 11300e5e5a79SGreg Clayton error.SetErrorStringWithFormat("<regex> can't be empty in 's%c<regex>%c<subst>%c' string: '%.*s'", 11310e5e5a79SGreg Clayton separator_char, 11320e5e5a79SGreg Clayton separator_char, 11330e5e5a79SGreg Clayton separator_char, 11340e5e5a79SGreg Clayton (int)regex_sed.size(), 11350e5e5a79SGreg Clayton regex_sed.data()); 11360e5e5a79SGreg Clayton return error; 11370e5e5a79SGreg Clayton } 11380e5e5a79SGreg Clayton else if (second_separator_char_pos + 1 == third_separator_char_pos) 11390e5e5a79SGreg Clayton { 11400e5e5a79SGreg Clayton error.SetErrorStringWithFormat("<subst> can't be empty in 's%c<regex>%c<subst>%c' string: '%.*s'", 11410e5e5a79SGreg Clayton separator_char, 11420e5e5a79SGreg Clayton separator_char, 11430e5e5a79SGreg Clayton separator_char, 11440e5e5a79SGreg Clayton (int)regex_sed.size(), 11450e5e5a79SGreg Clayton regex_sed.data()); 11460e5e5a79SGreg Clayton return error; 11470e5e5a79SGreg Clayton } 1148*44d93782SGreg Clayton 1149*44d93782SGreg Clayton if (check_only == false) 1150*44d93782SGreg Clayton { 11510e5e5a79SGreg Clayton std::string regex(regex_sed.substr(first_separator_char_pos + 1, second_separator_char_pos - first_separator_char_pos - 1)); 11520e5e5a79SGreg Clayton std::string subst(regex_sed.substr(second_separator_char_pos + 1, third_separator_char_pos - second_separator_char_pos - 1)); 11530e5e5a79SGreg Clayton m_regex_cmd_ap->AddRegexCommand (regex.c_str(), 11540e5e5a79SGreg Clayton subst.c_str()); 1155*44d93782SGreg Clayton } 11560e5e5a79SGreg Clayton return error; 1157de164aaaSGreg Clayton } 1158de164aaaSGreg Clayton 1159de164aaaSGreg Clayton void 11600e5e5a79SGreg Clayton AddRegexCommandToInterpreter() 1161de164aaaSGreg Clayton { 1162de164aaaSGreg Clayton if (m_regex_cmd_ap.get()) 1163de164aaaSGreg Clayton { 1164de164aaaSGreg Clayton if (m_regex_cmd_ap->HasRegexEntries()) 1165de164aaaSGreg Clayton { 1166de164aaaSGreg Clayton CommandObjectSP cmd_sp (m_regex_cmd_ap.release()); 1167de164aaaSGreg Clayton m_interpreter.AddCommand(cmd_sp->GetCommandName(), cmd_sp, true); 1168de164aaaSGreg Clayton } 1169de164aaaSGreg Clayton } 1170de164aaaSGreg Clayton } 1171de164aaaSGreg Clayton 1172de164aaaSGreg Clayton private: 11737b0992d9SGreg Clayton std::unique_ptr<CommandObjectRegexCommand> m_regex_cmd_ap; 1174de164aaaSGreg Clayton 1175de164aaaSGreg Clayton class CommandOptions : public Options 1176de164aaaSGreg Clayton { 1177de164aaaSGreg Clayton public: 1178de164aaaSGreg Clayton 1179de164aaaSGreg Clayton CommandOptions (CommandInterpreter &interpreter) : 1180de164aaaSGreg Clayton Options (interpreter) 1181de164aaaSGreg Clayton { 1182de164aaaSGreg Clayton } 1183de164aaaSGreg Clayton 1184de164aaaSGreg Clayton virtual 1185de164aaaSGreg Clayton ~CommandOptions (){} 1186de164aaaSGreg Clayton 1187de164aaaSGreg Clayton virtual Error 1188de164aaaSGreg Clayton SetOptionValue (uint32_t option_idx, const char *option_arg) 1189de164aaaSGreg Clayton { 1190de164aaaSGreg Clayton Error error; 11913bcdfc0eSGreg Clayton const int short_option = m_getopt_table[option_idx].val; 1192de164aaaSGreg Clayton 1193de164aaaSGreg Clayton switch (short_option) 1194de164aaaSGreg Clayton { 1195de164aaaSGreg Clayton case 'h': 1196de164aaaSGreg Clayton m_help.assign (option_arg); 1197de164aaaSGreg Clayton break; 1198de164aaaSGreg Clayton case 's': 1199de164aaaSGreg Clayton m_syntax.assign (option_arg); 1200de164aaaSGreg Clayton break; 1201de164aaaSGreg Clayton 1202de164aaaSGreg Clayton default: 120386edbf41SGreg Clayton error.SetErrorStringWithFormat ("unrecognized option '%c'", short_option); 1204de164aaaSGreg Clayton break; 1205de164aaaSGreg Clayton } 1206de164aaaSGreg Clayton 1207de164aaaSGreg Clayton return error; 1208de164aaaSGreg Clayton } 1209de164aaaSGreg Clayton 1210de164aaaSGreg Clayton void 1211de164aaaSGreg Clayton OptionParsingStarting () 1212de164aaaSGreg Clayton { 1213de164aaaSGreg Clayton m_help.clear(); 1214de164aaaSGreg Clayton m_syntax.clear(); 1215de164aaaSGreg Clayton } 1216de164aaaSGreg Clayton 1217de164aaaSGreg Clayton const OptionDefinition* 1218de164aaaSGreg Clayton GetDefinitions () 1219de164aaaSGreg Clayton { 1220de164aaaSGreg Clayton return g_option_table; 1221de164aaaSGreg Clayton } 1222de164aaaSGreg Clayton 1223de164aaaSGreg Clayton // Options table: Required for subclasses of Options. 1224de164aaaSGreg Clayton 1225de164aaaSGreg Clayton static OptionDefinition g_option_table[]; 1226de164aaaSGreg Clayton 1227de164aaaSGreg Clayton const char * 1228de164aaaSGreg Clayton GetHelp () 1229de164aaaSGreg Clayton { 1230de164aaaSGreg Clayton if (m_help.empty()) 1231de164aaaSGreg Clayton return NULL; 1232de164aaaSGreg Clayton return m_help.c_str(); 1233de164aaaSGreg Clayton } 1234de164aaaSGreg Clayton const char * 1235de164aaaSGreg Clayton GetSyntax () 1236de164aaaSGreg Clayton { 1237de164aaaSGreg Clayton if (m_syntax.empty()) 1238de164aaaSGreg Clayton return NULL; 1239de164aaaSGreg Clayton return m_syntax.c_str(); 1240de164aaaSGreg Clayton } 1241de164aaaSGreg Clayton // Instance variables to hold the values for command options. 1242de164aaaSGreg Clayton protected: 1243de164aaaSGreg Clayton std::string m_help; 1244de164aaaSGreg Clayton std::string m_syntax; 1245de164aaaSGreg Clayton }; 1246de164aaaSGreg Clayton 1247de164aaaSGreg Clayton virtual Options * 1248de164aaaSGreg Clayton GetOptions () 1249de164aaaSGreg Clayton { 1250de164aaaSGreg Clayton return &m_options; 1251de164aaaSGreg Clayton } 1252de164aaaSGreg Clayton 12535a988416SJim Ingham CommandOptions m_options; 1254de164aaaSGreg Clayton }; 1255de164aaaSGreg Clayton 1256de164aaaSGreg Clayton OptionDefinition 1257de164aaaSGreg Clayton CommandObjectCommandsAddRegex::CommandOptions::g_option_table[] = 1258de164aaaSGreg Clayton { 1259e2607b50SVirgile Bello { LLDB_OPT_SET_1, false, "help" , 'h', OptionParser::eRequiredArgument, NULL, 0, eArgTypeNone, "The help text to display for this command."}, 1260e2607b50SVirgile Bello { LLDB_OPT_SET_1, false, "syntax", 's', OptionParser::eRequiredArgument, NULL, 0, eArgTypeNone, "A syntax string showing the typical usage syntax."}, 1261de164aaaSGreg Clayton { 0 , false, NULL , 0 , 0 , NULL, 0, eArgTypeNone, NULL } 1262de164aaaSGreg Clayton }; 1263de164aaaSGreg Clayton 1264de164aaaSGreg Clayton 12655a988416SJim Ingham class CommandObjectPythonFunction : public CommandObjectRaw 1266223383edSEnrico Granata { 1267223383edSEnrico Granata private: 1268223383edSEnrico Granata std::string m_function_name; 12690a305db7SEnrico Granata ScriptedCommandSynchronicity m_synchro; 1270fac939e9SEnrico Granata bool m_fetched_help_long; 1271223383edSEnrico Granata 1272223383edSEnrico Granata public: 1273223383edSEnrico Granata 1274223383edSEnrico Granata CommandObjectPythonFunction (CommandInterpreter &interpreter, 1275223383edSEnrico Granata std::string name, 12760a305db7SEnrico Granata std::string funct, 12770a305db7SEnrico Granata ScriptedCommandSynchronicity synch) : 12785a988416SJim Ingham CommandObjectRaw (interpreter, 1279223383edSEnrico Granata name.c_str(), 1280223383edSEnrico Granata (std::string("Run Python function ") + funct).c_str(), 1281223383edSEnrico Granata NULL), 12820a305db7SEnrico Granata m_function_name(funct), 1283fac939e9SEnrico Granata m_synchro(synch), 1284fac939e9SEnrico Granata m_fetched_help_long(false) 1285223383edSEnrico Granata { 1286223383edSEnrico Granata } 1287223383edSEnrico Granata 1288223383edSEnrico Granata virtual 1289223383edSEnrico Granata ~CommandObjectPythonFunction () 1290223383edSEnrico Granata { 1291223383edSEnrico Granata } 1292223383edSEnrico Granata 1293223383edSEnrico Granata virtual bool 12943a18e319SGreg Clayton IsRemovable () const 12955a988416SJim Ingham { 12965a988416SJim Ingham return true; 12975a988416SJim Ingham } 12985a988416SJim Ingham 12995a988416SJim Ingham const std::string& 13005a988416SJim Ingham GetFunctionName () 13015a988416SJim Ingham { 13025a988416SJim Ingham return m_function_name; 13035a988416SJim Ingham } 13045a988416SJim Ingham 13055a988416SJim Ingham ScriptedCommandSynchronicity 13065a988416SJim Ingham GetSynchronicity () 13075a988416SJim Ingham { 13085a988416SJim Ingham return m_synchro; 13095a988416SJim Ingham } 13105a988416SJim Ingham 1311fac939e9SEnrico Granata virtual const char * 1312fac939e9SEnrico Granata GetHelpLong () 1313fac939e9SEnrico Granata { 1314fac939e9SEnrico Granata if (!m_fetched_help_long) 1315fac939e9SEnrico Granata { 1316fac939e9SEnrico Granata ScriptInterpreter* scripter = m_interpreter.GetScriptInterpreter(); 1317fac939e9SEnrico Granata if (scripter) 1318fac939e9SEnrico Granata { 1319fac939e9SEnrico Granata std::string docstring; 1320fac939e9SEnrico Granata m_fetched_help_long = scripter->GetDocumentationForItem(m_function_name.c_str(),docstring); 1321fac939e9SEnrico Granata if (!docstring.empty()) 1322fac939e9SEnrico Granata SetHelpLong(docstring); 1323fac939e9SEnrico Granata } 1324fac939e9SEnrico Granata } 1325fac939e9SEnrico Granata return CommandObjectRaw::GetHelpLong(); 1326fac939e9SEnrico Granata } 1327fac939e9SEnrico Granata 13285a988416SJim Ingham protected: 13295a988416SJim Ingham virtual bool 13305a988416SJim Ingham DoExecute (const char *raw_command_line, CommandReturnObject &result) 1331223383edSEnrico Granata { 1332223383edSEnrico Granata ScriptInterpreter* scripter = m_interpreter.GetScriptInterpreter(); 1333223383edSEnrico Granata 1334223383edSEnrico Granata Error error; 1335223383edSEnrico Granata 133670f11f88SJim Ingham result.SetStatus(eReturnStatusInvalid); 133770f11f88SJim Ingham 1338223383edSEnrico Granata if (!scripter || scripter->RunScriptBasedCommand(m_function_name.c_str(), 1339223383edSEnrico Granata raw_command_line, 13400a305db7SEnrico Granata m_synchro, 1341223383edSEnrico Granata result, 1342223383edSEnrico Granata error) == false) 1343223383edSEnrico Granata { 1344223383edSEnrico Granata result.AppendError(error.AsCString()); 1345223383edSEnrico Granata result.SetStatus(eReturnStatusFailed); 1346223383edSEnrico Granata } 1347223383edSEnrico Granata else 134870f11f88SJim Ingham { 134970f11f88SJim Ingham // Don't change the status if the command already set it... 135070f11f88SJim Ingham if (result.GetStatus() == eReturnStatusInvalid) 135170f11f88SJim Ingham { 13529a71a7d8SDaniel Malea if (result.GetOutputData() == NULL || result.GetOutputData()[0] == '\0') 1353223383edSEnrico Granata result.SetStatus(eReturnStatusSuccessFinishNoResult); 135470f11f88SJim Ingham else 135570f11f88SJim Ingham result.SetStatus(eReturnStatusSuccessFinishResult); 135670f11f88SJim Ingham } 135770f11f88SJim Ingham } 1358223383edSEnrico Granata 1359223383edSEnrico Granata return result.Succeeded(); 1360223383edSEnrico Granata } 1361223383edSEnrico Granata 1362223383edSEnrico Granata }; 1363223383edSEnrico Granata 1364a9dbf432SEnrico Granata //------------------------------------------------------------------------- 1365a9dbf432SEnrico Granata // CommandObjectCommandsScriptImport 1366a9dbf432SEnrico Granata //------------------------------------------------------------------------- 1367a9dbf432SEnrico Granata 13685a988416SJim Ingham class CommandObjectCommandsScriptImport : public CommandObjectParsed 1369a9dbf432SEnrico Granata { 13705a988416SJim Ingham public: 13715a988416SJim Ingham CommandObjectCommandsScriptImport (CommandInterpreter &interpreter) : 13725a988416SJim Ingham CommandObjectParsed (interpreter, 13735a988416SJim Ingham "command script import", 13745a988416SJim Ingham "Import a scripting module in LLDB.", 13755a988416SJim Ingham NULL), 13765a988416SJim Ingham m_options(interpreter) 13775a988416SJim Ingham { 13785a988416SJim Ingham CommandArgumentEntry arg1; 13795a988416SJim Ingham CommandArgumentData cmd_arg; 13805a988416SJim Ingham 13815a988416SJim Ingham // Define the first (and only) variant of this arg. 13825a988416SJim Ingham cmd_arg.arg_type = eArgTypeFilename; 13835a988416SJim Ingham cmd_arg.arg_repetition = eArgRepeatPlain; 13845a988416SJim Ingham 13855a988416SJim Ingham // There is only one variant this argument could be; put it into the argument entry. 13865a988416SJim Ingham arg1.push_back (cmd_arg); 13875a988416SJim Ingham 13885a988416SJim Ingham // Push the data for the first argument into the m_arguments vector. 13895a988416SJim Ingham m_arguments.push_back (arg1); 13905a988416SJim Ingham } 13915a988416SJim Ingham 13925a988416SJim Ingham ~CommandObjectCommandsScriptImport () 13935a988416SJim Ingham { 13945a988416SJim Ingham } 13955a988416SJim Ingham 1396c7bece56SGreg Clayton virtual int 13975a988416SJim Ingham HandleArgumentCompletion (Args &input, 13985a988416SJim Ingham int &cursor_index, 13995a988416SJim Ingham int &cursor_char_position, 14005a988416SJim Ingham OptionElementVector &opt_element_vector, 14015a988416SJim Ingham int match_start_point, 14025a988416SJim Ingham int max_return_elements, 14035a988416SJim Ingham bool &word_complete, 14045a988416SJim Ingham StringList &matches) 14055a988416SJim Ingham { 14065a988416SJim Ingham std::string completion_str (input.GetArgumentAtIndex(cursor_index)); 14075a988416SJim Ingham completion_str.erase (cursor_char_position); 14085a988416SJim Ingham 14095a988416SJim Ingham CommandCompletions::InvokeCommonCompletionCallbacks (m_interpreter, 14105a988416SJim Ingham CommandCompletions::eDiskFileCompletion, 14115a988416SJim Ingham completion_str.c_str(), 14125a988416SJim Ingham match_start_point, 14135a988416SJim Ingham max_return_elements, 14145a988416SJim Ingham NULL, 14155a988416SJim Ingham word_complete, 14165a988416SJim Ingham matches); 14175a988416SJim Ingham return matches.GetSize(); 14185a988416SJim Ingham } 14195a988416SJim Ingham 14205a988416SJim Ingham virtual Options * 14215a988416SJim Ingham GetOptions () 14225a988416SJim Ingham { 14235a988416SJim Ingham return &m_options; 14245a988416SJim Ingham } 14255a988416SJim Ingham 14265a988416SJim Ingham protected: 14270a305db7SEnrico Granata 14280a305db7SEnrico Granata class CommandOptions : public Options 14290a305db7SEnrico Granata { 14300a305db7SEnrico Granata public: 14310a305db7SEnrico Granata 14320a305db7SEnrico Granata CommandOptions (CommandInterpreter &interpreter) : 14330a305db7SEnrico Granata Options (interpreter) 14340a305db7SEnrico Granata { 14350a305db7SEnrico Granata } 14360a305db7SEnrico Granata 14370a305db7SEnrico Granata virtual 14380a305db7SEnrico Granata ~CommandOptions (){} 14390a305db7SEnrico Granata 14400a305db7SEnrico Granata virtual Error 14410a305db7SEnrico Granata SetOptionValue (uint32_t option_idx, const char *option_arg) 14420a305db7SEnrico Granata { 14430a305db7SEnrico Granata Error error; 14443bcdfc0eSGreg Clayton const int short_option = m_getopt_table[option_idx].val; 14450a305db7SEnrico Granata 14460a305db7SEnrico Granata switch (short_option) 14470a305db7SEnrico Granata { 14480a305db7SEnrico Granata case 'r': 14490a305db7SEnrico Granata m_allow_reload = true; 14500a305db7SEnrico Granata break; 14510a305db7SEnrico Granata default: 14520a305db7SEnrico Granata error.SetErrorStringWithFormat ("unrecognized option '%c'", short_option); 14530a305db7SEnrico Granata break; 14540a305db7SEnrico Granata } 14550a305db7SEnrico Granata 14560a305db7SEnrico Granata return error; 14570a305db7SEnrico Granata } 14580a305db7SEnrico Granata 14590a305db7SEnrico Granata void 14600a305db7SEnrico Granata OptionParsingStarting () 14610a305db7SEnrico Granata { 1462e0c70f1bSEnrico Granata m_allow_reload = true; 14630a305db7SEnrico Granata } 14640a305db7SEnrico Granata 14650a305db7SEnrico Granata const OptionDefinition* 14660a305db7SEnrico Granata GetDefinitions () 14670a305db7SEnrico Granata { 14680a305db7SEnrico Granata return g_option_table; 14690a305db7SEnrico Granata } 14700a305db7SEnrico Granata 14710a305db7SEnrico Granata // Options table: Required for subclasses of Options. 14720a305db7SEnrico Granata 14730a305db7SEnrico Granata static OptionDefinition g_option_table[]; 14740a305db7SEnrico Granata 14750a305db7SEnrico Granata // Instance variables to hold the values for command options. 14760a305db7SEnrico Granata 14770a305db7SEnrico Granata bool m_allow_reload; 14780a305db7SEnrico Granata }; 14790a305db7SEnrico Granata 1480a9dbf432SEnrico Granata bool 14815a988416SJim Ingham DoExecute (Args& command, CommandReturnObject &result) 1482a9dbf432SEnrico Granata { 1483a9dbf432SEnrico Granata 1484a9dbf432SEnrico Granata if (m_interpreter.GetDebugger().GetScriptLanguage() != lldb::eScriptLanguagePython) 1485a9dbf432SEnrico Granata { 1486a9dbf432SEnrico Granata result.AppendError ("only scripting language supported for module importing is currently Python"); 1487a9dbf432SEnrico Granata result.SetStatus (eReturnStatusFailed); 1488a9dbf432SEnrico Granata return false; 1489a9dbf432SEnrico Granata } 1490a9dbf432SEnrico Granata 14915a988416SJim Ingham size_t argc = command.GetArgumentCount(); 1492a9dbf432SEnrico Granata 1493a9dbf432SEnrico Granata if (argc != 1) 1494a9dbf432SEnrico Granata { 1495a9dbf432SEnrico Granata result.AppendError ("'command script import' requires one argument"); 1496a9dbf432SEnrico Granata result.SetStatus (eReturnStatusFailed); 1497a9dbf432SEnrico Granata return false; 1498a9dbf432SEnrico Granata } 1499a9dbf432SEnrico Granata 15005a988416SJim Ingham std::string path = command.GetArgumentAtIndex(0); 1501a9dbf432SEnrico Granata Error error; 1502a9dbf432SEnrico Granata 1503c9d645d3SGreg Clayton const bool init_session = true; 1504078551c7SEnrico Granata // FIXME: this is necessary because CommandObject::CheckRequirements() assumes that 1505078551c7SEnrico Granata // commands won't ever be recursively invoked, but it's actually possible to craft 1506078551c7SEnrico Granata // a Python script that does other "command script imports" in __lldb_init_module 1507078551c7SEnrico Granata // the real fix is to have recursive commands possible with a CommandInvocation object 1508078551c7SEnrico Granata // separate from the CommandObject itself, so that recursive command invocations 1509078551c7SEnrico Granata // won't stomp on each other (wrt to execution contents, options, and more) 1510078551c7SEnrico Granata m_exe_ctx.Clear(); 1511a9dbf432SEnrico Granata if (m_interpreter.GetScriptInterpreter()->LoadScriptingModule(path.c_str(), 15120a305db7SEnrico Granata m_options.m_allow_reload, 1513c9d645d3SGreg Clayton init_session, 1514a9dbf432SEnrico Granata error)) 1515a9dbf432SEnrico Granata { 1516a9dbf432SEnrico Granata result.SetStatus (eReturnStatusSuccessFinishNoResult); 1517a9dbf432SEnrico Granata } 1518a9dbf432SEnrico Granata else 1519a9dbf432SEnrico Granata { 1520a9dbf432SEnrico Granata result.AppendErrorWithFormat("module importing failed: %s", error.AsCString()); 1521a9dbf432SEnrico Granata result.SetStatus (eReturnStatusFailed); 1522a9dbf432SEnrico Granata } 1523a9dbf432SEnrico Granata 1524a9dbf432SEnrico Granata return result.Succeeded(); 1525a9dbf432SEnrico Granata } 15260a305db7SEnrico Granata 15275a988416SJim Ingham CommandOptions m_options; 1528a9dbf432SEnrico Granata }; 1529223383edSEnrico Granata 15300a305db7SEnrico Granata OptionDefinition 15310a305db7SEnrico Granata CommandObjectCommandsScriptImport::CommandOptions::g_option_table[] = 15320a305db7SEnrico Granata { 1533e2607b50SVirgile Bello { LLDB_OPT_SET_1, false, "allow-reload", 'r', OptionParser::eNoArgument, NULL, 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."}, 15340a305db7SEnrico Granata { 0, false, NULL, 0, 0, NULL, 0, eArgTypeNone, NULL } 15350a305db7SEnrico Granata }; 15360a305db7SEnrico Granata 15370a305db7SEnrico Granata 1538223383edSEnrico Granata //------------------------------------------------------------------------- 1539223383edSEnrico Granata // CommandObjectCommandsScriptAdd 1540223383edSEnrico Granata //------------------------------------------------------------------------- 1541223383edSEnrico Granata 1542*44d93782SGreg Clayton class CommandObjectCommandsScriptAdd : 1543*44d93782SGreg Clayton public CommandObjectParsed, 1544*44d93782SGreg Clayton public IOHandlerDelegateMultiline 1545223383edSEnrico Granata { 15465a988416SJim Ingham public: 15475a988416SJim Ingham CommandObjectCommandsScriptAdd(CommandInterpreter &interpreter) : 15485a988416SJim Ingham CommandObjectParsed (interpreter, 15495a988416SJim Ingham "command script add", 15505a988416SJim Ingham "Add a scripted function as an LLDB command.", 15515a988416SJim Ingham NULL), 1552*44d93782SGreg Clayton IOHandlerDelegateMultiline ("DONE"), 15535a988416SJim Ingham m_options (interpreter) 15545a988416SJim Ingham { 15555a988416SJim Ingham CommandArgumentEntry arg1; 15565a988416SJim Ingham CommandArgumentData cmd_arg; 15575a988416SJim Ingham 15585a988416SJim Ingham // Define the first (and only) variant of this arg. 15595a988416SJim Ingham cmd_arg.arg_type = eArgTypeCommandName; 15605a988416SJim Ingham cmd_arg.arg_repetition = eArgRepeatPlain; 15615a988416SJim Ingham 15625a988416SJim Ingham // There is only one variant this argument could be; put it into the argument entry. 15635a988416SJim Ingham arg1.push_back (cmd_arg); 15645a988416SJim Ingham 15655a988416SJim Ingham // Push the data for the first argument into the m_arguments vector. 15665a988416SJim Ingham m_arguments.push_back (arg1); 15675a988416SJim Ingham } 15685a988416SJim Ingham 15695a988416SJim Ingham ~CommandObjectCommandsScriptAdd () 15705a988416SJim Ingham { 15715a988416SJim Ingham } 15725a988416SJim Ingham 15735a988416SJim Ingham virtual Options * 15745a988416SJim Ingham GetOptions () 15755a988416SJim Ingham { 15765a988416SJim Ingham return &m_options; 15775a988416SJim Ingham } 15785a988416SJim Ingham 15795a988416SJim Ingham protected: 1580223383edSEnrico Granata 1581223383edSEnrico Granata class CommandOptions : public Options 1582223383edSEnrico Granata { 1583223383edSEnrico Granata public: 1584223383edSEnrico Granata 1585223383edSEnrico Granata CommandOptions (CommandInterpreter &interpreter) : 1586223383edSEnrico Granata Options (interpreter) 1587223383edSEnrico Granata { 1588223383edSEnrico Granata } 1589223383edSEnrico Granata 1590223383edSEnrico Granata virtual 1591223383edSEnrico Granata ~CommandOptions (){} 1592223383edSEnrico Granata 1593223383edSEnrico Granata virtual Error 1594223383edSEnrico Granata SetOptionValue (uint32_t option_idx, const char *option_arg) 1595223383edSEnrico Granata { 1596223383edSEnrico Granata Error error; 15973bcdfc0eSGreg Clayton const int short_option = m_getopt_table[option_idx].val; 1598223383edSEnrico Granata 1599223383edSEnrico Granata switch (short_option) 1600223383edSEnrico Granata { 1601223383edSEnrico Granata case 'f': 1602223383edSEnrico Granata m_funct_name = std::string(option_arg); 1603223383edSEnrico Granata break; 16040a305db7SEnrico Granata case 's': 1605*44d93782SGreg Clayton m_synchronicity = (ScriptedCommandSynchronicity) Args::StringToOptionEnum(option_arg, g_option_table[option_idx].enum_values, 0, error); 16060a305db7SEnrico Granata if (!error.Success()) 16070a305db7SEnrico Granata error.SetErrorStringWithFormat ("unrecognized value for synchronicity '%s'", option_arg); 16080a305db7SEnrico Granata break; 1609223383edSEnrico Granata default: 161086edbf41SGreg Clayton error.SetErrorStringWithFormat ("unrecognized option '%c'", short_option); 1611223383edSEnrico Granata break; 1612223383edSEnrico Granata } 1613223383edSEnrico Granata 1614223383edSEnrico Granata return error; 1615223383edSEnrico Granata } 1616223383edSEnrico Granata 1617223383edSEnrico Granata void 1618223383edSEnrico Granata OptionParsingStarting () 1619223383edSEnrico Granata { 1620223383edSEnrico Granata m_funct_name = ""; 1621*44d93782SGreg Clayton m_synchronicity = eScriptedCommandSynchronicitySynchronous; 1622223383edSEnrico Granata } 1623223383edSEnrico Granata 1624223383edSEnrico Granata const OptionDefinition* 1625223383edSEnrico Granata GetDefinitions () 1626223383edSEnrico Granata { 1627223383edSEnrico Granata return g_option_table; 1628223383edSEnrico Granata } 1629223383edSEnrico Granata 1630223383edSEnrico Granata // Options table: Required for subclasses of Options. 1631223383edSEnrico Granata 1632223383edSEnrico Granata static OptionDefinition g_option_table[]; 1633223383edSEnrico Granata 1634223383edSEnrico Granata // Instance variables to hold the values for command options. 1635223383edSEnrico Granata 1636223383edSEnrico Granata std::string m_funct_name; 1637*44d93782SGreg Clayton ScriptedCommandSynchronicity m_synchronicity; 1638223383edSEnrico Granata }; 1639223383edSEnrico Granata 1640*44d93782SGreg Clayton virtual void 1641*44d93782SGreg Clayton IOHandlerActivated (IOHandler &io_handler) 1642223383edSEnrico Granata { 1643*44d93782SGreg Clayton StreamFileSP output_sp(io_handler.GetOutputStreamFile()); 1644*44d93782SGreg Clayton if (output_sp) 1645223383edSEnrico Granata { 1646*44d93782SGreg Clayton output_sp->PutCString(g_python_command_instructions); 1647*44d93782SGreg Clayton output_sp->Flush(); 1648223383edSEnrico Granata } 1649223383edSEnrico Granata } 1650223383edSEnrico Granata 1651223383edSEnrico Granata 1652*44d93782SGreg Clayton virtual void 1653*44d93782SGreg Clayton IOHandlerInputComplete (IOHandler &io_handler, std::string &data) 1654223383edSEnrico Granata { 1655*44d93782SGreg Clayton StreamFileSP error_sp = io_handler.GetErrorStreamFile(); 1656*44d93782SGreg Clayton 1657*44d93782SGreg Clayton ScriptInterpreter *interpreter = m_interpreter.GetScriptInterpreter(); 1658*44d93782SGreg Clayton if (interpreter) 1659*44d93782SGreg Clayton { 1660*44d93782SGreg Clayton 1661*44d93782SGreg Clayton StringList lines; 1662*44d93782SGreg Clayton lines.SplitIntoLines(data); 1663*44d93782SGreg Clayton if (lines.GetSize() > 0) 1664*44d93782SGreg Clayton { 1665a73b7df7SEnrico Granata std::string funct_name_str; 1666*44d93782SGreg Clayton if (interpreter->GenerateScriptAliasFunction (lines, funct_name_str)) 1667223383edSEnrico Granata { 1668a73b7df7SEnrico Granata if (funct_name_str.empty()) 1669223383edSEnrico Granata { 1670*44d93782SGreg Clayton error_sp->Printf ("error: unable to obtain a function name, didn't add python command.\n"); 1671*44d93782SGreg Clayton error_sp->Flush(); 1672223383edSEnrico Granata } 1673*44d93782SGreg Clayton else 1674*44d93782SGreg Clayton { 1675223383edSEnrico Granata // everything should be fine now, let's add this alias 1676223383edSEnrico Granata 1677223383edSEnrico Granata CommandObjectSP command_obj_sp(new CommandObjectPythonFunction (m_interpreter, 1678223383edSEnrico Granata m_cmd_name, 1679a73b7df7SEnrico Granata funct_name_str.c_str(), 1680*44d93782SGreg Clayton m_synchronicity)); 1681223383edSEnrico Granata 16820a305db7SEnrico Granata if (!m_interpreter.AddUserCommand(m_cmd_name, command_obj_sp, true)) 1683223383edSEnrico Granata { 1684*44d93782SGreg Clayton error_sp->Printf ("error: unable to add selected command, didn't add python command.\n"); 1685*44d93782SGreg Clayton error_sp->Flush(); 1686223383edSEnrico Granata } 1687223383edSEnrico Granata } 1688*44d93782SGreg Clayton } 1689*44d93782SGreg Clayton else 1690*44d93782SGreg Clayton { 1691*44d93782SGreg Clayton error_sp->Printf ("error: unable to create function, didn't add python command.\n"); 1692*44d93782SGreg Clayton error_sp->Flush(); 1693*44d93782SGreg Clayton } 1694*44d93782SGreg Clayton } 1695*44d93782SGreg Clayton else 1696*44d93782SGreg Clayton { 1697*44d93782SGreg Clayton error_sp->Printf ("error: empty function, didn't add python command.\n"); 1698*44d93782SGreg Clayton error_sp->Flush(); 1699*44d93782SGreg Clayton } 1700*44d93782SGreg Clayton } 1701*44d93782SGreg Clayton else 1702*44d93782SGreg Clayton { 1703*44d93782SGreg Clayton error_sp->Printf ("error: script interpreter missing, didn't add python command.\n"); 1704*44d93782SGreg Clayton error_sp->Flush(); 1705*44d93782SGreg Clayton } 1706*44d93782SGreg Clayton 1707*44d93782SGreg Clayton io_handler.SetIsDone(true); 1708*44d93782SGreg Clayton 1709*44d93782SGreg Clayton 1710*44d93782SGreg Clayton } 1711223383edSEnrico Granata 17125a988416SJim Ingham protected: 1713223383edSEnrico Granata bool 17145a988416SJim Ingham DoExecute (Args& command, CommandReturnObject &result) 1715223383edSEnrico Granata { 171699f0b8f9SEnrico Granata 171799f0b8f9SEnrico Granata if (m_interpreter.GetDebugger().GetScriptLanguage() != lldb::eScriptLanguagePython) 171899f0b8f9SEnrico Granata { 171999f0b8f9SEnrico Granata result.AppendError ("only scripting language supported for scripted commands is currently Python"); 172099f0b8f9SEnrico Granata result.SetStatus (eReturnStatusFailed); 172199f0b8f9SEnrico Granata return false; 172299f0b8f9SEnrico Granata } 172399f0b8f9SEnrico Granata 17245a988416SJim Ingham size_t argc = command.GetArgumentCount(); 1725223383edSEnrico Granata 1726223383edSEnrico Granata if (argc != 1) 1727223383edSEnrico Granata { 1728223383edSEnrico Granata result.AppendError ("'command script add' requires one argument"); 1729223383edSEnrico Granata result.SetStatus (eReturnStatusFailed); 1730223383edSEnrico Granata return false; 1731223383edSEnrico Granata } 1732223383edSEnrico Granata 1733*44d93782SGreg Clayton // Store the command name and synchronicity in case we get multi-line input 1734*44d93782SGreg Clayton m_cmd_name = command.GetArgumentAtIndex(0); 1735*44d93782SGreg Clayton m_synchronicity = m_options.m_synchronicity; 1736223383edSEnrico Granata 1737223383edSEnrico Granata if (m_options.m_funct_name.empty()) 1738223383edSEnrico Granata { 1739*44d93782SGreg Clayton m_interpreter.GetPythonCommandsFromIOHandler (" ", // Prompt 1740*44d93782SGreg Clayton *this, // IOHandlerDelegate 1741*44d93782SGreg Clayton true, // Run IOHandler in async mode 1742*44d93782SGreg Clayton NULL); // Baton for the "io_handler" that will be passed back into our IOHandlerDelegate functions 1743223383edSEnrico Granata } 1744223383edSEnrico Granata else 1745223383edSEnrico Granata { 17460a305db7SEnrico Granata CommandObjectSP new_cmd(new CommandObjectPythonFunction(m_interpreter, 1747*44d93782SGreg Clayton m_cmd_name, 17480a305db7SEnrico Granata m_options.m_funct_name, 1749*44d93782SGreg Clayton m_synchronicity)); 1750*44d93782SGreg Clayton if (m_interpreter.AddUserCommand(m_cmd_name, new_cmd, true)) 1751223383edSEnrico Granata { 1752223383edSEnrico Granata result.SetStatus (eReturnStatusSuccessFinishNoResult); 1753223383edSEnrico Granata } 1754223383edSEnrico Granata else 1755223383edSEnrico Granata { 1756223383edSEnrico Granata result.AppendError("cannot add command"); 1757223383edSEnrico Granata result.SetStatus (eReturnStatusFailed); 1758223383edSEnrico Granata } 1759223383edSEnrico Granata } 1760223383edSEnrico Granata 1761223383edSEnrico Granata return result.Succeeded(); 1762223383edSEnrico Granata 1763223383edSEnrico Granata } 17645a988416SJim Ingham 17655a988416SJim Ingham CommandOptions m_options; 1766*44d93782SGreg Clayton std::string m_cmd_name; 1767*44d93782SGreg Clayton ScriptedCommandSynchronicity m_synchronicity; 1768223383edSEnrico Granata }; 1769223383edSEnrico Granata 17700a305db7SEnrico Granata static OptionEnumValueElement g_script_synchro_type[] = 17710a305db7SEnrico Granata { 17720a305db7SEnrico Granata { eScriptedCommandSynchronicitySynchronous, "synchronous", "Run synchronous"}, 17730a305db7SEnrico Granata { eScriptedCommandSynchronicityAsynchronous, "asynchronous", "Run asynchronous"}, 17740a305db7SEnrico Granata { eScriptedCommandSynchronicityCurrentValue, "current", "Do not alter current setting"}, 17750a305db7SEnrico Granata { 0, NULL, NULL } 17760a305db7SEnrico Granata }; 17770a305db7SEnrico Granata 1778223383edSEnrico Granata OptionDefinition 1779223383edSEnrico Granata CommandObjectCommandsScriptAdd::CommandOptions::g_option_table[] = 1780223383edSEnrico Granata { 1781e2607b50SVirgile Bello { LLDB_OPT_SET_1, false, "function", 'f', OptionParser::eRequiredArgument, NULL, 0, eArgTypePythonFunction, "Name of the Python function to bind to this command name."}, 1782e2607b50SVirgile Bello { LLDB_OPT_SET_1, false, "synchronicity", 's', OptionParser::eRequiredArgument, g_script_synchro_type, 0, eArgTypeScriptedCommandSynchronicity, "Set the synchronicity of this command's executions with regard to LLDB event system."}, 1783223383edSEnrico Granata { 0, false, NULL, 0, 0, NULL, 0, eArgTypeNone, NULL } 1784223383edSEnrico Granata }; 1785223383edSEnrico Granata 1786223383edSEnrico Granata //------------------------------------------------------------------------- 1787223383edSEnrico Granata // CommandObjectCommandsScriptList 1788223383edSEnrico Granata //------------------------------------------------------------------------- 1789223383edSEnrico Granata 17905a988416SJim Ingham class CommandObjectCommandsScriptList : public CommandObjectParsed 1791223383edSEnrico Granata { 1792223383edSEnrico Granata private: 1793223383edSEnrico Granata 1794223383edSEnrico Granata public: 1795223383edSEnrico Granata CommandObjectCommandsScriptList(CommandInterpreter &interpreter) : 17965a988416SJim Ingham CommandObjectParsed (interpreter, 1797223383edSEnrico Granata "command script list", 1798223383edSEnrico Granata "List defined scripted commands.", 1799223383edSEnrico Granata NULL) 1800223383edSEnrico Granata { 1801223383edSEnrico Granata } 1802223383edSEnrico Granata 1803223383edSEnrico Granata ~CommandObjectCommandsScriptList () 1804223383edSEnrico Granata { 1805223383edSEnrico Granata } 1806223383edSEnrico Granata 1807223383edSEnrico Granata bool 18085a988416SJim Ingham DoExecute (Args& command, CommandReturnObject &result) 1809223383edSEnrico Granata { 1810223383edSEnrico Granata 1811223383edSEnrico Granata m_interpreter.GetHelp(result, 1812223383edSEnrico Granata CommandInterpreter::eCommandTypesUserDef); 1813223383edSEnrico Granata 1814223383edSEnrico Granata result.SetStatus (eReturnStatusSuccessFinishResult); 1815223383edSEnrico Granata 1816223383edSEnrico Granata return true; 1817223383edSEnrico Granata 1818223383edSEnrico Granata 1819223383edSEnrico Granata } 1820223383edSEnrico Granata }; 1821223383edSEnrico Granata 1822223383edSEnrico Granata //------------------------------------------------------------------------- 1823223383edSEnrico Granata // CommandObjectCommandsScriptClear 1824223383edSEnrico Granata //------------------------------------------------------------------------- 1825223383edSEnrico Granata 18265a988416SJim Ingham class CommandObjectCommandsScriptClear : public CommandObjectParsed 1827223383edSEnrico Granata { 1828223383edSEnrico Granata private: 1829223383edSEnrico Granata 1830223383edSEnrico Granata public: 1831223383edSEnrico Granata CommandObjectCommandsScriptClear(CommandInterpreter &interpreter) : 18325a988416SJim Ingham CommandObjectParsed (interpreter, 1833223383edSEnrico Granata "command script clear", 1834223383edSEnrico Granata "Delete all scripted commands.", 1835223383edSEnrico Granata NULL) 1836223383edSEnrico Granata { 1837223383edSEnrico Granata } 1838223383edSEnrico Granata 1839223383edSEnrico Granata ~CommandObjectCommandsScriptClear () 1840223383edSEnrico Granata { 1841223383edSEnrico Granata } 1842223383edSEnrico Granata 18435a988416SJim Ingham protected: 1844223383edSEnrico Granata bool 18455a988416SJim Ingham DoExecute (Args& command, CommandReturnObject &result) 1846223383edSEnrico Granata { 1847223383edSEnrico Granata 1848223383edSEnrico Granata m_interpreter.RemoveAllUser(); 1849223383edSEnrico Granata 1850223383edSEnrico Granata result.SetStatus (eReturnStatusSuccessFinishResult); 1851223383edSEnrico Granata 1852223383edSEnrico Granata return true; 1853223383edSEnrico Granata } 1854223383edSEnrico Granata }; 1855223383edSEnrico Granata 1856223383edSEnrico Granata //------------------------------------------------------------------------- 1857223383edSEnrico Granata // CommandObjectCommandsScriptDelete 1858223383edSEnrico Granata //------------------------------------------------------------------------- 1859223383edSEnrico Granata 18605a988416SJim Ingham class CommandObjectCommandsScriptDelete : public CommandObjectParsed 1861223383edSEnrico Granata { 1862223383edSEnrico Granata public: 1863223383edSEnrico Granata CommandObjectCommandsScriptDelete(CommandInterpreter &interpreter) : 18645a988416SJim Ingham CommandObjectParsed (interpreter, 1865223383edSEnrico Granata "command script delete", 1866223383edSEnrico Granata "Delete a scripted command.", 1867223383edSEnrico Granata NULL) 1868223383edSEnrico Granata { 1869223383edSEnrico Granata CommandArgumentEntry arg1; 1870223383edSEnrico Granata CommandArgumentData cmd_arg; 1871223383edSEnrico Granata 1872223383edSEnrico Granata // Define the first (and only) variant of this arg. 1873223383edSEnrico Granata cmd_arg.arg_type = eArgTypeCommandName; 1874223383edSEnrico Granata cmd_arg.arg_repetition = eArgRepeatPlain; 1875223383edSEnrico Granata 1876223383edSEnrico Granata // There is only one variant this argument could be; put it into the argument entry. 1877223383edSEnrico Granata arg1.push_back (cmd_arg); 1878223383edSEnrico Granata 1879223383edSEnrico Granata // Push the data for the first argument into the m_arguments vector. 1880223383edSEnrico Granata m_arguments.push_back (arg1); 1881223383edSEnrico Granata } 1882223383edSEnrico Granata 1883223383edSEnrico Granata ~CommandObjectCommandsScriptDelete () 1884223383edSEnrico Granata { 1885223383edSEnrico Granata } 1886223383edSEnrico Granata 18875a988416SJim Ingham protected: 1888223383edSEnrico Granata bool 18895a988416SJim Ingham DoExecute (Args& command, CommandReturnObject &result) 1890223383edSEnrico Granata { 1891223383edSEnrico Granata 18925a988416SJim Ingham size_t argc = command.GetArgumentCount(); 1893223383edSEnrico Granata 1894223383edSEnrico Granata if (argc != 1) 1895223383edSEnrico Granata { 1896223383edSEnrico Granata result.AppendError ("'command script delete' requires one argument"); 1897223383edSEnrico Granata result.SetStatus (eReturnStatusFailed); 1898223383edSEnrico Granata return false; 1899223383edSEnrico Granata } 1900223383edSEnrico Granata 19015a988416SJim Ingham const char* cmd_name = command.GetArgumentAtIndex(0); 1902223383edSEnrico Granata 1903223383edSEnrico Granata if (cmd_name && *cmd_name && m_interpreter.HasUserCommands() && m_interpreter.UserCommandExists(cmd_name)) 1904223383edSEnrico Granata { 1905223383edSEnrico Granata m_interpreter.RemoveUser(cmd_name); 1906223383edSEnrico Granata result.SetStatus (eReturnStatusSuccessFinishResult); 1907223383edSEnrico Granata } 1908223383edSEnrico Granata else 1909223383edSEnrico Granata { 1910223383edSEnrico Granata result.AppendErrorWithFormat ("command %s not found", cmd_name); 1911223383edSEnrico Granata result.SetStatus (eReturnStatusFailed); 1912223383edSEnrico Granata } 1913223383edSEnrico Granata 1914223383edSEnrico Granata return result.Succeeded(); 1915223383edSEnrico Granata 1916223383edSEnrico Granata } 1917223383edSEnrico Granata }; 1918223383edSEnrico Granata 1919223383edSEnrico Granata #pragma mark CommandObjectMultiwordCommandsScript 1920223383edSEnrico Granata 1921223383edSEnrico Granata //------------------------------------------------------------------------- 1922223383edSEnrico Granata // CommandObjectMultiwordCommandsScript 1923223383edSEnrico Granata //------------------------------------------------------------------------- 1924223383edSEnrico Granata 1925223383edSEnrico Granata class CommandObjectMultiwordCommandsScript : public CommandObjectMultiword 1926223383edSEnrico Granata { 1927223383edSEnrico Granata public: 1928223383edSEnrico Granata CommandObjectMultiwordCommandsScript (CommandInterpreter &interpreter) : 1929223383edSEnrico Granata CommandObjectMultiword (interpreter, 1930223383edSEnrico Granata "command script", 1931223383edSEnrico Granata "A set of commands for managing or customizing script commands.", 1932223383edSEnrico Granata "command script <subcommand> [<subcommand-options>]") 1933223383edSEnrico Granata { 1934223383edSEnrico Granata LoadSubCommand ("add", CommandObjectSP (new CommandObjectCommandsScriptAdd (interpreter))); 1935223383edSEnrico Granata LoadSubCommand ("delete", CommandObjectSP (new CommandObjectCommandsScriptDelete (interpreter))); 1936223383edSEnrico Granata LoadSubCommand ("clear", CommandObjectSP (new CommandObjectCommandsScriptClear (interpreter))); 1937223383edSEnrico Granata LoadSubCommand ("list", CommandObjectSP (new CommandObjectCommandsScriptList (interpreter))); 1938a9dbf432SEnrico Granata LoadSubCommand ("import", CommandObjectSP (new CommandObjectCommandsScriptImport (interpreter))); 1939223383edSEnrico Granata } 1940223383edSEnrico Granata 1941223383edSEnrico Granata ~CommandObjectMultiwordCommandsScript () 1942223383edSEnrico Granata { 1943223383edSEnrico Granata } 1944223383edSEnrico Granata 1945223383edSEnrico Granata }; 1946223383edSEnrico Granata 1947223383edSEnrico Granata 1948ebc09c36SJim Ingham #pragma mark CommandObjectMultiwordCommands 1949ebc09c36SJim Ingham 1950ebc09c36SJim Ingham //------------------------------------------------------------------------- 1951ebc09c36SJim Ingham // CommandObjectMultiwordCommands 1952ebc09c36SJim Ingham //------------------------------------------------------------------------- 1953ebc09c36SJim Ingham 1954ebc09c36SJim Ingham CommandObjectMultiwordCommands::CommandObjectMultiwordCommands (CommandInterpreter &interpreter) : 1955a7015092SGreg Clayton CommandObjectMultiword (interpreter, 19560e5e5a79SGreg Clayton "command", 19573f4c09c1SCaroline Tice "A set of commands for managing or customizing the debugger commands.", 19580e5e5a79SGreg Clayton "command <subcommand> [<subcommand-options>]") 1959ebc09c36SJim Ingham { 1960a7015092SGreg Clayton LoadSubCommand ("source", CommandObjectSP (new CommandObjectCommandsSource (interpreter))); 1961a7015092SGreg Clayton LoadSubCommand ("alias", CommandObjectSP (new CommandObjectCommandsAlias (interpreter))); 1962a7015092SGreg Clayton LoadSubCommand ("unalias", CommandObjectSP (new CommandObjectCommandsUnalias (interpreter))); 1963de164aaaSGreg Clayton LoadSubCommand ("regex", CommandObjectSP (new CommandObjectCommandsAddRegex (interpreter))); 1964a5a97ebeSJim Ingham LoadSubCommand ("history", CommandObjectSP (new CommandObjectCommandsHistory (interpreter))); 1965223383edSEnrico Granata LoadSubCommand ("script", CommandObjectSP (new CommandObjectMultiwordCommandsScript (interpreter))); 1966ebc09c36SJim Ingham } 1967ebc09c36SJim Ingham 1968ebc09c36SJim Ingham CommandObjectMultiwordCommands::~CommandObjectMultiwordCommands () 1969ebc09c36SJim Ingham { 1970ebc09c36SJim Ingham } 1971ebc09c36SJim Ingham 1972