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 
10ebc09c36SJim Ingham // C Includes
11ebc09c36SJim Ingham // C++ Includes
12ebc09c36SJim Ingham // Other libraries and framework includes
130e5e5a79SGreg Clayton #include "llvm/ADT/StringRef.h"
140e5e5a79SGreg Clayton 
15ebc09c36SJim Ingham // Project includes
166e3d8e7fSEugene Zelenko #include "CommandObjectCommands.h"
1746d4aa21SEnrico Granata #include "CommandObjectHelp.h"
18ebc09c36SJim Ingham #include "lldb/Core/Debugger.h"
1944d93782SGreg Clayton #include "lldb/Core/IOHandler.h"
20be93a35aSEnrico Granata #include "lldb/Core/StringList.h"
21de164aaaSGreg Clayton #include "lldb/Interpreter/Args.h"
227594f14fSEnrico Granata #include "lldb/Interpreter/CommandHistory.h"
23ebc09c36SJim Ingham #include "lldb/Interpreter/CommandInterpreter.h"
24de164aaaSGreg Clayton #include "lldb/Interpreter/CommandObjectRegexCommand.h"
25ebc09c36SJim Ingham #include "lldb/Interpreter/CommandReturnObject.h"
26012d4fcaSEnrico Granata #include "lldb/Interpreter/OptionValueBoolean.h"
277594f14fSEnrico Granata #include "lldb/Interpreter/OptionValueUInt64.h"
28ebc09c36SJim Ingham #include "lldb/Interpreter/Options.h"
2999f0b8f9SEnrico Granata #include "lldb/Interpreter/ScriptInterpreter.h"
30ebc09c36SJim Ingham 
31ebc09c36SJim Ingham using namespace lldb;
32ebc09c36SJim Ingham using namespace lldb_private;
33ebc09c36SJim Ingham 
34ebc09c36SJim Ingham //-------------------------------------------------------------------------
35ebc09c36SJim Ingham // CommandObjectCommandsSource
36ebc09c36SJim Ingham //-------------------------------------------------------------------------
37ebc09c36SJim Ingham 
385a988416SJim Ingham class CommandObjectCommandsHistory : public CommandObjectParsed
39a5a97ebeSJim Ingham {
405a988416SJim Ingham public:
415a988416SJim Ingham     CommandObjectCommandsHistory(CommandInterpreter &interpreter) :
425a988416SJim Ingham         CommandObjectParsed(interpreter,
435a988416SJim Ingham                             "command history",
445a988416SJim Ingham                             "Dump the history of commands in this session.",
456e3d8e7fSEugene Zelenko                             nullptr),
465a988416SJim Ingham         m_options (interpreter)
475a988416SJim Ingham     {
485a988416SJim Ingham     }
495a988416SJim Ingham 
506e3d8e7fSEugene Zelenko     ~CommandObjectCommandsHistory() override = default;
515a988416SJim Ingham 
5213d21e9aSBruce Mitchener     Options *
5313d21e9aSBruce Mitchener     GetOptions () override
545a988416SJim Ingham     {
555a988416SJim Ingham         return &m_options;
565a988416SJim Ingham     }
575a988416SJim Ingham 
585a988416SJim Ingham protected:
59a5a97ebeSJim Ingham     class CommandOptions : public Options
60a5a97ebeSJim Ingham     {
61a5a97ebeSJim Ingham     public:
62a5a97ebeSJim Ingham         CommandOptions (CommandInterpreter &interpreter) :
637594f14fSEnrico Granata             Options (interpreter),
647594f14fSEnrico Granata             m_start_idx(0),
657594f14fSEnrico Granata             m_stop_idx(0),
667594f14fSEnrico Granata             m_count(0),
6763123b64SEnrico Granata             m_clear(false)
68a5a97ebeSJim Ingham         {
69a5a97ebeSJim Ingham         }
70a5a97ebeSJim Ingham 
716e3d8e7fSEugene Zelenko         ~CommandOptions() override = default;
72a5a97ebeSJim Ingham 
7313d21e9aSBruce Mitchener         Error
7413d21e9aSBruce Mitchener         SetOptionValue (uint32_t option_idx, const char *option_arg) override
75a5a97ebeSJim Ingham         {
76a5a97ebeSJim Ingham             Error error;
773bcdfc0eSGreg Clayton             const int short_option = m_getopt_table[option_idx].val;
78a5a97ebeSJim Ingham 
79a5a97ebeSJim Ingham             switch (short_option)
80a5a97ebeSJim Ingham             {
81a5a97ebeSJim Ingham                 case 'c':
82c95f7e2aSPavel Labath                     error = m_count.SetValueFromString(option_arg,eVarSetOperationAssign);
83a5a97ebeSJim Ingham                     break;
84a5a97ebeSJim Ingham                 case 's':
857594f14fSEnrico Granata                     if (option_arg && strcmp("end", option_arg) == 0)
867594f14fSEnrico Granata                     {
877594f14fSEnrico Granata                         m_start_idx.SetCurrentValue(UINT64_MAX);
887594f14fSEnrico Granata                         m_start_idx.SetOptionWasSet();
897594f14fSEnrico Granata                     }
907594f14fSEnrico Granata                     else
91c95f7e2aSPavel Labath                         error = m_start_idx.SetValueFromString(option_arg,eVarSetOperationAssign);
927594f14fSEnrico Granata                     break;
937594f14fSEnrico Granata                 case 'e':
94c95f7e2aSPavel Labath                     error = m_stop_idx.SetValueFromString(option_arg,eVarSetOperationAssign);
957594f14fSEnrico Granata                     break;
9663123b64SEnrico Granata                 case 'C':
9763123b64SEnrico Granata                     m_clear.SetCurrentValue(true);
9863123b64SEnrico Granata                     m_clear.SetOptionWasSet();
99a5a97ebeSJim Ingham                     break;
100a5a97ebeSJim Ingham                 default:
10186edbf41SGreg Clayton                     error.SetErrorStringWithFormat ("unrecognized option '%c'", short_option);
102a5a97ebeSJim Ingham                     break;
103a5a97ebeSJim Ingham             }
104a5a97ebeSJim Ingham 
105a5a97ebeSJim Ingham             return error;
106a5a97ebeSJim Ingham         }
107a5a97ebeSJim Ingham 
108a5a97ebeSJim Ingham         void
10913d21e9aSBruce Mitchener         OptionParsingStarting () override
110a5a97ebeSJim Ingham         {
1117594f14fSEnrico Granata             m_start_idx.Clear();
1127594f14fSEnrico Granata             m_stop_idx.Clear();
1137594f14fSEnrico Granata             m_count.Clear();
11463123b64SEnrico Granata             m_clear.Clear();
115a5a97ebeSJim Ingham         }
116a5a97ebeSJim Ingham 
117a5a97ebeSJim Ingham         const OptionDefinition*
11813d21e9aSBruce Mitchener         GetDefinitions () override
119a5a97ebeSJim Ingham         {
120a5a97ebeSJim Ingham             return g_option_table;
121a5a97ebeSJim Ingham         }
122a5a97ebeSJim Ingham 
123a5a97ebeSJim Ingham         // Options table: Required for subclasses of Options.
124a5a97ebeSJim Ingham 
125a5a97ebeSJim Ingham         static OptionDefinition g_option_table[];
126a5a97ebeSJim Ingham 
127a5a97ebeSJim Ingham         // Instance variables to hold the values for command options.
128a5a97ebeSJim Ingham 
1297594f14fSEnrico Granata         OptionValueUInt64 m_start_idx;
1307594f14fSEnrico Granata         OptionValueUInt64 m_stop_idx;
1317594f14fSEnrico Granata         OptionValueUInt64 m_count;
13263123b64SEnrico Granata         OptionValueBoolean m_clear;
133a5a97ebeSJim Ingham     };
134a5a97ebeSJim Ingham 
135a5a97ebeSJim Ingham     bool
13613d21e9aSBruce Mitchener     DoExecute (Args& command, CommandReturnObject &result) override
137a5a97ebeSJim Ingham     {
13863123b64SEnrico Granata         if (m_options.m_clear.GetCurrentValue() && m_options.m_clear.OptionWasSet())
1397594f14fSEnrico Granata         {
1407594f14fSEnrico Granata             m_interpreter.GetCommandHistory().Clear();
1417594f14fSEnrico Granata             result.SetStatus(lldb::eReturnStatusSuccessFinishNoResult);
1427594f14fSEnrico Granata         }
1437594f14fSEnrico Granata         else
1447594f14fSEnrico Granata         {
1457594f14fSEnrico Granata             if (m_options.m_start_idx.OptionWasSet() && m_options.m_stop_idx.OptionWasSet() && m_options.m_count.OptionWasSet())
1467594f14fSEnrico Granata             {
1477594f14fSEnrico Granata                 result.AppendError("--count, --start-index and --end-index cannot be all specified in the same invocation");
1487594f14fSEnrico Granata                 result.SetStatus(lldb::eReturnStatusFailed);
1497594f14fSEnrico Granata             }
1507594f14fSEnrico Granata             else
1517594f14fSEnrico Granata             {
15284400ec7SVirgile Bello                 std::pair<bool,uint64_t> start_idx(m_options.m_start_idx.OptionWasSet(),m_options.m_start_idx.GetCurrentValue());
15384400ec7SVirgile Bello                 std::pair<bool,uint64_t> stop_idx(m_options.m_stop_idx.OptionWasSet(),m_options.m_stop_idx.GetCurrentValue());
15484400ec7SVirgile Bello                 std::pair<bool,uint64_t> count(m_options.m_count.OptionWasSet(),m_options.m_count.GetCurrentValue());
155a5a97ebeSJim Ingham 
1567594f14fSEnrico Granata                 const CommandHistory& history(m_interpreter.GetCommandHistory());
1577594f14fSEnrico Granata 
1587594f14fSEnrico Granata                 if (start_idx.first && start_idx.second == UINT64_MAX)
1597594f14fSEnrico Granata                 {
1607594f14fSEnrico Granata                     if (count.first)
1617594f14fSEnrico Granata                     {
1627594f14fSEnrico Granata                         start_idx.second = history.GetSize() - count.second;
1637594f14fSEnrico Granata                         stop_idx.second = history.GetSize() - 1;
1647594f14fSEnrico Granata                     }
1657594f14fSEnrico Granata                     else if (stop_idx.first)
1667594f14fSEnrico Granata                     {
1677594f14fSEnrico Granata                         start_idx.second = stop_idx.second;
1687594f14fSEnrico Granata                         stop_idx.second = history.GetSize() - 1;
1697594f14fSEnrico Granata                     }
1707594f14fSEnrico Granata                     else
1717594f14fSEnrico Granata                     {
1727594f14fSEnrico Granata                         start_idx.second = 0;
1737594f14fSEnrico Granata                         stop_idx.second = history.GetSize() - 1;
1747594f14fSEnrico Granata                     }
1757594f14fSEnrico Granata                 }
1767594f14fSEnrico Granata                 else
1777594f14fSEnrico Granata                 {
1787594f14fSEnrico Granata                     if (!start_idx.first && !stop_idx.first && !count.first)
1797594f14fSEnrico Granata                     {
1807594f14fSEnrico Granata                         start_idx.second = 0;
1817594f14fSEnrico Granata                         stop_idx.second = history.GetSize() - 1;
1827594f14fSEnrico Granata                     }
1837594f14fSEnrico Granata                     else if (start_idx.first)
1847594f14fSEnrico Granata                     {
1857594f14fSEnrico Granata                         if (count.first)
1867594f14fSEnrico Granata                         {
1877594f14fSEnrico Granata                             stop_idx.second = start_idx.second + count.second - 1;
1887594f14fSEnrico Granata                         }
1897594f14fSEnrico Granata                         else if (!stop_idx.first)
1907594f14fSEnrico Granata                         {
1917594f14fSEnrico Granata                             stop_idx.second = history.GetSize() - 1;
1927594f14fSEnrico Granata                         }
1937594f14fSEnrico Granata                     }
1947594f14fSEnrico Granata                     else if (stop_idx.first)
1957594f14fSEnrico Granata                     {
1967594f14fSEnrico Granata                         if (count.first)
1977594f14fSEnrico Granata                         {
1987594f14fSEnrico Granata                             if (stop_idx.second >= count.second)
1997594f14fSEnrico Granata                                 start_idx.second = stop_idx.second - count.second + 1;
2007594f14fSEnrico Granata                             else
2017594f14fSEnrico Granata                                 start_idx.second = 0;
2027594f14fSEnrico Granata                         }
2037594f14fSEnrico Granata                     }
2047594f14fSEnrico Granata                     else /* if (count.first) */
2057594f14fSEnrico Granata                     {
2067594f14fSEnrico Granata                         start_idx.second = 0;
2077594f14fSEnrico Granata                         stop_idx.second = count.second - 1;
2087594f14fSEnrico Granata                     }
2097594f14fSEnrico Granata                 }
2107594f14fSEnrico Granata                 history.Dump(result.GetOutputStream(), start_idx.second, stop_idx.second);
2117594f14fSEnrico Granata             }
2127594f14fSEnrico Granata         }
213a5a97ebeSJim Ingham         return result.Succeeded();
214a5a97ebeSJim Ingham 
215a5a97ebeSJim Ingham     }
2165a988416SJim Ingham 
2175a988416SJim Ingham     CommandOptions m_options;
218a5a97ebeSJim Ingham };
219a5a97ebeSJim Ingham 
220a5a97ebeSJim Ingham OptionDefinition
221a5a97ebeSJim Ingham CommandObjectCommandsHistory::CommandOptions::g_option_table[] =
222a5a97ebeSJim Ingham {
2236e3d8e7fSEugene Zelenko { LLDB_OPT_SET_1, false, "count", 'c', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeUnsignedInteger,        "How many history commands to print."},
2246e3d8e7fSEugene Zelenko { LLDB_OPT_SET_1, false, "start-index", 's', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeUnsignedInteger,  "Index at which to start printing history commands (or end to mean tail mode)."},
2256e3d8e7fSEugene Zelenko { LLDB_OPT_SET_1, false, "end-index", 'e', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeUnsignedInteger,    "Index at which to stop printing history commands."},
2266e3d8e7fSEugene Zelenko { LLDB_OPT_SET_2, false, "clear", 'C', OptionParser::eNoArgument, nullptr, nullptr, 0, eArgTypeBoolean,    "Clears the current command history."},
2276e3d8e7fSEugene Zelenko { 0, false, nullptr, 0, 0, nullptr, nullptr, 0, eArgTypeNone, nullptr }
228a5a97ebeSJim Ingham };
229a5a97ebeSJim Ingham 
230a5a97ebeSJim Ingham //-------------------------------------------------------------------------
231a5a97ebeSJim Ingham // CommandObjectCommandsSource
232a5a97ebeSJim Ingham //-------------------------------------------------------------------------
233a5a97ebeSJim Ingham 
2345a988416SJim Ingham class CommandObjectCommandsSource : public CommandObjectParsed
235ebc09c36SJim Ingham {
2365a988416SJim Ingham public:
2375a988416SJim Ingham     CommandObjectCommandsSource(CommandInterpreter &interpreter) :
2385a988416SJim Ingham         CommandObjectParsed(interpreter,
2395a988416SJim Ingham                             "command source",
2405a988416SJim Ingham                             "Read in debugger commands from the file <filename> and execute them.",
2416e3d8e7fSEugene Zelenko                             nullptr),
2425a988416SJim Ingham         m_options (interpreter)
2435a988416SJim Ingham     {
2445a988416SJim Ingham         CommandArgumentEntry arg;
2455a988416SJim Ingham         CommandArgumentData file_arg;
2465a988416SJim Ingham 
2475a988416SJim Ingham         // Define the first (and only) variant of this arg.
2485a988416SJim Ingham         file_arg.arg_type = eArgTypeFilename;
2495a988416SJim Ingham         file_arg.arg_repetition = eArgRepeatPlain;
2505a988416SJim Ingham 
2515a988416SJim Ingham         // There is only one variant this argument could be; put it into the argument entry.
2525a988416SJim Ingham         arg.push_back (file_arg);
2535a988416SJim Ingham 
2545a988416SJim Ingham         // Push the data for the first argument into the m_arguments vector.
2555a988416SJim Ingham         m_arguments.push_back (arg);
2565a988416SJim Ingham     }
2575a988416SJim Ingham 
2586e3d8e7fSEugene Zelenko     ~CommandObjectCommandsSource() override = default;
2595a988416SJim Ingham 
26013d21e9aSBruce Mitchener     const char*
26113d21e9aSBruce Mitchener     GetRepeatCommand (Args &current_command_args, uint32_t index) override
2625a988416SJim Ingham     {
2635a988416SJim Ingham         return "";
2645a988416SJim Ingham     }
2655a988416SJim Ingham 
26613d21e9aSBruce Mitchener     int
2675a988416SJim Ingham     HandleArgumentCompletion (Args &input,
2685a988416SJim Ingham                               int &cursor_index,
2695a988416SJim Ingham                               int &cursor_char_position,
2705a988416SJim Ingham                               OptionElementVector &opt_element_vector,
2715a988416SJim Ingham                               int match_start_point,
2725a988416SJim Ingham                               int max_return_elements,
2735a988416SJim Ingham                               bool &word_complete,
27413d21e9aSBruce Mitchener                               StringList &matches) override
2755a988416SJim Ingham     {
2765a988416SJim Ingham         std::string completion_str (input.GetArgumentAtIndex(cursor_index));
2775a988416SJim Ingham         completion_str.erase (cursor_char_position);
2785a988416SJim Ingham 
2795a988416SJim Ingham         CommandCompletions::InvokeCommonCompletionCallbacks(m_interpreter,
2805a988416SJim Ingham                                                             CommandCompletions::eDiskFileCompletion,
2815a988416SJim Ingham                                                             completion_str.c_str(),
2825a988416SJim Ingham                                                             match_start_point,
2835a988416SJim Ingham                                                             max_return_elements,
2846e3d8e7fSEugene Zelenko                                                             nullptr,
2855a988416SJim Ingham                                                             word_complete,
2865a988416SJim Ingham                                                             matches);
2875a988416SJim Ingham         return matches.GetSize();
2885a988416SJim Ingham     }
2895a988416SJim Ingham 
29013d21e9aSBruce Mitchener     Options *
29113d21e9aSBruce Mitchener     GetOptions () override
2925a988416SJim Ingham     {
2935a988416SJim Ingham         return &m_options;
2945a988416SJim Ingham     }
2955a988416SJim Ingham 
2965a988416SJim Ingham protected:
297e16c50a1SJim Ingham     class CommandOptions : public Options
298e16c50a1SJim Ingham     {
299e16c50a1SJim Ingham     public:
300eb0103f2SGreg Clayton         CommandOptions (CommandInterpreter &interpreter) :
301012d4fcaSEnrico Granata             Options (interpreter),
302340b0309SGreg Clayton             m_stop_on_error (true),
303340b0309SGreg Clayton             m_silent_run (false),
304340b0309SGreg Clayton             m_stop_on_continue (true)
305eb0103f2SGreg Clayton         {
306eb0103f2SGreg Clayton         }
307e16c50a1SJim Ingham 
3086e3d8e7fSEugene Zelenko         ~CommandOptions() override = default;
309e16c50a1SJim Ingham 
31013d21e9aSBruce Mitchener         Error
31113d21e9aSBruce Mitchener         SetOptionValue (uint32_t option_idx, const char *option_arg) override
312e16c50a1SJim Ingham         {
313e16c50a1SJim Ingham             Error error;
3143bcdfc0eSGreg Clayton             const int short_option = m_getopt_table[option_idx].val;
315e16c50a1SJim Ingham 
316e16c50a1SJim Ingham             switch (short_option)
317e16c50a1SJim Ingham             {
318e16c50a1SJim Ingham                 case 'e':
319c95f7e2aSPavel Labath                     error = m_stop_on_error.SetValueFromString(option_arg);
320e16c50a1SJim Ingham                     break;
321340b0309SGreg Clayton 
322e16c50a1SJim Ingham                 case 'c':
323c95f7e2aSPavel Labath                     error = m_stop_on_continue.SetValueFromString(option_arg);
324e16c50a1SJim Ingham                     break;
325340b0309SGreg Clayton 
32660986174SMichael Sartain                 case 's':
327c95f7e2aSPavel Labath                     error = m_silent_run.SetValueFromString(option_arg);
32860986174SMichael Sartain                     break;
329340b0309SGreg Clayton 
330e16c50a1SJim Ingham                 default:
33186edbf41SGreg Clayton                     error.SetErrorStringWithFormat ("unrecognized option '%c'", short_option);
332e16c50a1SJim Ingham                     break;
333e16c50a1SJim Ingham             }
334e16c50a1SJim Ingham 
335e16c50a1SJim Ingham             return error;
336e16c50a1SJim Ingham         }
337e16c50a1SJim Ingham 
338e16c50a1SJim Ingham         void
33913d21e9aSBruce Mitchener         OptionParsingStarting () override
340e16c50a1SJim Ingham         {
341012d4fcaSEnrico Granata             m_stop_on_error.Clear();
342340b0309SGreg Clayton             m_silent_run.Clear();
343340b0309SGreg Clayton             m_stop_on_continue.Clear();
344e16c50a1SJim Ingham         }
345e16c50a1SJim Ingham 
346e0d378b3SGreg Clayton         const OptionDefinition*
34713d21e9aSBruce Mitchener         GetDefinitions () override
348e16c50a1SJim Ingham         {
349e16c50a1SJim Ingham             return g_option_table;
350e16c50a1SJim Ingham         }
351e16c50a1SJim Ingham 
352e16c50a1SJim Ingham         // Options table: Required for subclasses of Options.
353e16c50a1SJim Ingham 
354e0d378b3SGreg Clayton         static OptionDefinition g_option_table[];
355e16c50a1SJim Ingham 
356e16c50a1SJim Ingham         // Instance variables to hold the values for command options.
357e16c50a1SJim Ingham 
358012d4fcaSEnrico Granata         OptionValueBoolean m_stop_on_error;
359340b0309SGreg Clayton         OptionValueBoolean m_silent_run;
360340b0309SGreg Clayton         OptionValueBoolean m_stop_on_continue;
361e16c50a1SJim Ingham     };
362e16c50a1SJim Ingham 
363ebc09c36SJim Ingham     bool
36413d21e9aSBruce Mitchener     DoExecute(Args& command, CommandReturnObject &result) override
365ebc09c36SJim Ingham     {
366c7bece56SGreg Clayton         const size_t argc = command.GetArgumentCount();
367ebc09c36SJim Ingham         if (argc == 1)
368ebc09c36SJim Ingham         {
3695a988416SJim Ingham             const char *filename = command.GetArgumentAtIndex(0);
370ebc09c36SJim Ingham 
3711ee3853fSJohnny Chen             FileSpec cmd_file (filename, true);
3726e3d8e7fSEugene Zelenko             ExecutionContext *exe_ctx = nullptr;  // Just use the default context.
373ebc09c36SJim Ingham 
374340b0309SGreg Clayton             // If any options were set, then use them
375340b0309SGreg Clayton             if (m_options.m_stop_on_error.OptionWasSet()    ||
376340b0309SGreg Clayton                 m_options.m_silent_run.OptionWasSet()       ||
377340b0309SGreg Clayton                 m_options.m_stop_on_continue.OptionWasSet())
378340b0309SGreg Clayton             {
379340b0309SGreg Clayton                 // Use user set settings
38026c7bf93SJim Ingham                 CommandInterpreterRunOptions options;
38126c7bf93SJim Ingham                 options.SetStopOnContinue(m_options.m_stop_on_continue.GetCurrentValue());
38226c7bf93SJim Ingham                 options.SetStopOnError (m_options.m_stop_on_error.GetCurrentValue());
3837d8555c4SJim Ingham                 options.SetEchoCommands (!m_options.m_silent_run.GetCurrentValue());
3847d8555c4SJim Ingham                 options.SetPrintResults (!m_options.m_silent_run.GetCurrentValue());
38526c7bf93SJim Ingham 
386e16c50a1SJim Ingham                 m_interpreter.HandleCommandsFromFile (cmd_file,
387e16c50a1SJim Ingham                                                       exe_ctx,
38826c7bf93SJim Ingham                                                       options,
389e16c50a1SJim Ingham                                                       result);
390340b0309SGreg Clayton             }
391340b0309SGreg Clayton             else
392340b0309SGreg Clayton             {
393340b0309SGreg Clayton                 // No options were set, inherit any settings from nested "command source" commands,
394340b0309SGreg Clayton                 // or set to sane default settings...
39526c7bf93SJim Ingham                 CommandInterpreterRunOptions options;
396340b0309SGreg Clayton                 m_interpreter.HandleCommandsFromFile (cmd_file,
397340b0309SGreg Clayton                                                       exe_ctx,
39826c7bf93SJim Ingham                                                       options,
399340b0309SGreg Clayton                                                       result);
400340b0309SGreg Clayton             }
401ebc09c36SJim Ingham         }
402ebc09c36SJim Ingham         else
403ebc09c36SJim Ingham         {
404ebc09c36SJim Ingham             result.AppendErrorWithFormat("'%s' takes exactly one executable filename argument.\n", GetCommandName());
405ebc09c36SJim Ingham             result.SetStatus (eReturnStatusFailed);
406ebc09c36SJim Ingham         }
407ebc09c36SJim Ingham         return result.Succeeded();
408ebc09c36SJim Ingham     }
4096e3d8e7fSEugene Zelenko 
4105a988416SJim Ingham     CommandOptions m_options;
411ebc09c36SJim Ingham };
412ebc09c36SJim Ingham 
413e0d378b3SGreg Clayton OptionDefinition
414e16c50a1SJim Ingham CommandObjectCommandsSource::CommandOptions::g_option_table[] =
415e16c50a1SJim Ingham {
4166e3d8e7fSEugene Zelenko { LLDB_OPT_SET_ALL, false, "stop-on-error", 'e', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeBoolean,    "If true, stop executing commands on error."},
4176e3d8e7fSEugene Zelenko { LLDB_OPT_SET_ALL, false, "stop-on-continue", 'c', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeBoolean, "If true, stop executing commands on continue."},
4186e3d8e7fSEugene Zelenko { LLDB_OPT_SET_ALL, false, "silent-run", 's', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeBoolean, "If true don't echo commands while executing."},
4196e3d8e7fSEugene Zelenko { 0, false, nullptr, 0, 0, nullptr, nullptr, 0, eArgTypeNone, nullptr }
420e16c50a1SJim Ingham };
421e16c50a1SJim Ingham 
422ebc09c36SJim Ingham #pragma mark CommandObjectCommandsAlias
423ebc09c36SJim Ingham //-------------------------------------------------------------------------
424ebc09c36SJim Ingham // CommandObjectCommandsAlias
425ebc09c36SJim Ingham //-------------------------------------------------------------------------
426ebc09c36SJim Ingham 
427be93a35aSEnrico Granata static const char *g_python_command_instructions =   "Enter your Python command(s). Type 'DONE' to end.\n"
428be93a35aSEnrico Granata                                                      "You must define a Python function with this signature:\n"
42944d93782SGreg Clayton                                                      "def my_command_impl(debugger, args, result, internal_dict):\n";
430be93a35aSEnrico Granata 
4315a988416SJim Ingham class CommandObjectCommandsAlias : public CommandObjectRaw
432ebc09c36SJim Ingham {
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.",
4386e3d8e7fSEugene Zelenko                          nullptr)
439ebc09c36SJim Ingham     {
440ebc09c36SJim Ingham         SetHelpLong(
441ea671fbdSKate Stone "'alias' allows the user to create a short-cut or abbreviation for long \
442ea671fbdSKate Stone commands, multi-word commands, and commands that take particular options.  \
443ea671fbdSKate Stone Below are some simple examples of how one might use the 'alias' command:" R"(
444ea671fbdSKate Stone 
445ea671fbdSKate Stone (lldb) command alias sc script
446ea671fbdSKate Stone 
447ea671fbdSKate Stone     Creates the abbreviation 'sc' for the 'script' command.
448ea671fbdSKate Stone 
449ea671fbdSKate Stone (lldb) command alias bp breakpoint
450ea671fbdSKate Stone 
451ea671fbdSKate Stone )" "    Creates the abbreviation 'bp' for the 'breakpoint' command.  Since \
452ea671fbdSKate Stone breakpoint commands are two-word commands, the user would still need to \
453ea671fbdSKate Stone enter the second word after 'bp', e.g. 'bp enable' or 'bp delete'." R"(
454ea671fbdSKate Stone 
455ea671fbdSKate Stone (lldb) command alias bpl breakpoint list
456ea671fbdSKate Stone 
457ea671fbdSKate Stone     Creates the abbreviation 'bpl' for the two-word command 'breakpoint list'.
458ea671fbdSKate Stone 
459ea671fbdSKate Stone )" "An alias can include some options for the command, with the values either \
460ea671fbdSKate Stone filled in at the time the alias is created, or specified as positional \
461ea671fbdSKate Stone arguments, to be filled in when the alias is invoked.  The following example \
462ea671fbdSKate Stone shows how to create aliases with options:" R"(
463ea671fbdSKate Stone 
464ea671fbdSKate Stone (lldb) command alias bfl breakpoint set -f %1 -l %2
465ea671fbdSKate Stone 
466ea671fbdSKate Stone )" "    Creates the abbreviation 'bfl' (for break-file-line), with the -f and -l \
467ea671fbdSKate Stone options already part of the alias.  So if the user wants to set a breakpoint \
468ea671fbdSKate Stone by file and line without explicitly having to use the -f and -l options, the \
469ea671fbdSKate Stone user can now use 'bfl' instead.  The '%1' and '%2' are positional placeholders \
470ea671fbdSKate Stone for the actual arguments that will be passed when the alias command is used.  \
471ea671fbdSKate Stone The number in the placeholder refers to the position/order the actual value \
472ea671fbdSKate Stone occupies when the alias is used.  All the occurrences of '%1' in the alias \
473ea671fbdSKate Stone will be replaced with the first argument, all the occurrences of '%2' in the \
474ea671fbdSKate Stone alias will be replaced with the second argument, and so on.  This also allows \
475ea671fbdSKate Stone actual arguments to be used multiple times within an alias (see 'process \
476ea671fbdSKate Stone launch' example below)." R"(
477ea671fbdSKate Stone 
478ea671fbdSKate Stone )" "Note: the positional arguments must substitute as whole words in the resultant \
479ea671fbdSKate Stone command, so you can't at present do something like this to append the file extension \
480ea671fbdSKate Stone \".cpp\":" R"(
481ea671fbdSKate Stone 
482ea671fbdSKate Stone (lldb) command alias bcppfl breakpoint set -f %1.cpp -l %2
483ea671fbdSKate Stone 
484ea671fbdSKate Stone )" "For more complex aliasing, use the \"command regex\" command instead.  In the \
485ea671fbdSKate Stone 'bfl' case above, the actual file value will be filled in with the first argument \
486ea671fbdSKate Stone following 'bfl' and the actual line number value will be filled in with the second \
487ea671fbdSKate Stone argument.  The user would use this alias as follows:" R"(
488ea671fbdSKate Stone 
489ea671fbdSKate Stone (lldb) command alias bfl breakpoint set -f %1 -l %2
490ea671fbdSKate Stone (lldb) bfl my-file.c 137
491ea671fbdSKate Stone 
492ea671fbdSKate Stone This would be the same as if the user had entered 'breakpoint set -f my-file.c -l 137'.
493ea671fbdSKate Stone 
494ea671fbdSKate Stone Another example:
495ea671fbdSKate Stone 
496ea671fbdSKate Stone (lldb) command alias pltty process launch -s -o %1 -e %1
497ea671fbdSKate Stone (lldb) pltty /dev/tty0
498ea671fbdSKate Stone 
499ea671fbdSKate Stone     Interpreted as 'process launch -s -o /dev/tty0 -e /dev/tty0'
500ea671fbdSKate Stone 
501ea671fbdSKate Stone )" "If the user always wanted to pass the same value to a particular option, the \
502ea671fbdSKate Stone alias could be defined with that value directly in the alias as a constant, \
503ea671fbdSKate Stone rather than using a positional placeholder:" R"(
504ea671fbdSKate Stone 
505ea671fbdSKate Stone (lldb) command alias bl3 breakpoint set -f %1 -l 3
506ea671fbdSKate Stone 
507ea671fbdSKate Stone     Always sets a breakpoint on line 3 of whatever file is indicated.)"
508ea671fbdSKate Stone         );
509ebc09c36SJim Ingham 
510405fe67fSCaroline Tice         CommandArgumentEntry arg1;
511405fe67fSCaroline Tice         CommandArgumentEntry arg2;
512405fe67fSCaroline Tice         CommandArgumentEntry arg3;
513405fe67fSCaroline Tice         CommandArgumentData alias_arg;
514405fe67fSCaroline Tice         CommandArgumentData cmd_arg;
515405fe67fSCaroline Tice         CommandArgumentData options_arg;
516405fe67fSCaroline Tice 
517405fe67fSCaroline Tice         // Define the first (and only) variant of this arg.
518405fe67fSCaroline Tice         alias_arg.arg_type = eArgTypeAliasName;
519405fe67fSCaroline Tice         alias_arg.arg_repetition = eArgRepeatPlain;
520405fe67fSCaroline Tice 
521405fe67fSCaroline Tice         // There is only one variant this argument could be; put it into the argument entry.
522405fe67fSCaroline Tice         arg1.push_back (alias_arg);
523405fe67fSCaroline Tice 
524405fe67fSCaroline Tice         // Define the first (and only) variant of this arg.
525405fe67fSCaroline Tice         cmd_arg.arg_type = eArgTypeCommandName;
526405fe67fSCaroline Tice         cmd_arg.arg_repetition = eArgRepeatPlain;
527405fe67fSCaroline Tice 
528405fe67fSCaroline Tice         // There is only one variant this argument could be; put it into the argument entry.
529405fe67fSCaroline Tice         arg2.push_back (cmd_arg);
530405fe67fSCaroline Tice 
531405fe67fSCaroline Tice         // Define the first (and only) variant of this arg.
532405fe67fSCaroline Tice         options_arg.arg_type = eArgTypeAliasOptions;
533405fe67fSCaroline Tice         options_arg.arg_repetition = eArgRepeatOptional;
534405fe67fSCaroline Tice 
535405fe67fSCaroline Tice         // There is only one variant this argument could be; put it into the argument entry.
536405fe67fSCaroline Tice         arg3.push_back (options_arg);
537405fe67fSCaroline Tice 
538405fe67fSCaroline Tice         // Push the data for the first argument into the m_arguments vector.
539405fe67fSCaroline Tice         m_arguments.push_back (arg1);
540405fe67fSCaroline Tice         m_arguments.push_back (arg2);
541405fe67fSCaroline Tice         m_arguments.push_back (arg3);
542ebc09c36SJim Ingham     }
543ebc09c36SJim Ingham 
5446e3d8e7fSEugene Zelenko     ~CommandObjectCommandsAlias() override = default;
545ebc09c36SJim Ingham 
5465a988416SJim Ingham protected:
54713d21e9aSBruce Mitchener     bool
54813d21e9aSBruce Mitchener     DoExecute (const char *raw_command_line, CommandReturnObject &result) override
549844d2303SCaroline Tice     {
550844d2303SCaroline Tice         Args args (raw_command_line);
551844d2303SCaroline Tice         std::string raw_command_string (raw_command_line);
552844d2303SCaroline Tice 
553844d2303SCaroline Tice         size_t argc = args.GetArgumentCount();
554844d2303SCaroline Tice 
555844d2303SCaroline Tice         if (argc < 2)
556844d2303SCaroline Tice         {
557844d2303SCaroline Tice             result.AppendError ("'alias' requires at least two arguments");
558844d2303SCaroline Tice             result.SetStatus (eReturnStatusFailed);
559844d2303SCaroline Tice             return false;
560844d2303SCaroline Tice         }
561844d2303SCaroline Tice 
562844d2303SCaroline Tice         // Get the alias command.
563844d2303SCaroline Tice 
564844d2303SCaroline Tice         const std::string alias_command = args.GetArgumentAtIndex (0);
565844d2303SCaroline Tice 
566844d2303SCaroline Tice         // Strip the new alias name off 'raw_command_string'  (leave it on args, which gets passed to 'Execute', which
567844d2303SCaroline Tice         // does the stripping itself.
568844d2303SCaroline Tice         size_t pos = raw_command_string.find (alias_command);
569844d2303SCaroline Tice         if (pos == 0)
570844d2303SCaroline Tice         {
571844d2303SCaroline Tice             raw_command_string = raw_command_string.substr (alias_command.size());
572844d2303SCaroline Tice             pos = raw_command_string.find_first_not_of (' ');
573844d2303SCaroline Tice             if ((pos != std::string::npos) && (pos > 0))
574844d2303SCaroline Tice                 raw_command_string = raw_command_string.substr (pos);
575844d2303SCaroline Tice         }
576844d2303SCaroline Tice         else
577844d2303SCaroline Tice         {
578844d2303SCaroline Tice             result.AppendError ("Error parsing command string.  No alias created.");
579844d2303SCaroline Tice             result.SetStatus (eReturnStatusFailed);
580844d2303SCaroline Tice             return false;
581844d2303SCaroline Tice         }
582844d2303SCaroline Tice 
583844d2303SCaroline Tice 
584844d2303SCaroline Tice         // Verify that the command is alias-able.
585844d2303SCaroline Tice         if (m_interpreter.CommandExists (alias_command.c_str()))
586844d2303SCaroline Tice         {
587844d2303SCaroline Tice             result.AppendErrorWithFormat ("'%s' is a permanent debugger command and cannot be redefined.\n",
588844d2303SCaroline Tice                                           alias_command.c_str());
589844d2303SCaroline Tice             result.SetStatus (eReturnStatusFailed);
590844d2303SCaroline Tice             return false;
591844d2303SCaroline Tice         }
592844d2303SCaroline Tice 
593844d2303SCaroline Tice         // Get CommandObject that is being aliased. The command name is read from the front of raw_command_string.
594844d2303SCaroline Tice         // raw_command_string is returned with the name of the command object stripped off the front.
595844d2303SCaroline Tice         CommandObject *cmd_obj = m_interpreter.GetCommandObjectForCommand (raw_command_string);
596844d2303SCaroline Tice 
597844d2303SCaroline Tice         if (!cmd_obj)
598844d2303SCaroline Tice         {
59986edbf41SGreg Clayton             result.AppendErrorWithFormat ("invalid command given to 'alias'. '%s' does not begin with a valid command."
600844d2303SCaroline Tice                                           "  No alias created.", raw_command_string.c_str());
601844d2303SCaroline Tice             result.SetStatus (eReturnStatusFailed);
602844d2303SCaroline Tice             return false;
603844d2303SCaroline Tice         }
604844d2303SCaroline Tice         else if (!cmd_obj->WantsRawCommandString ())
605844d2303SCaroline Tice         {
606844d2303SCaroline Tice             // Note that args was initialized with the original command, and has not been updated to this point.
607844d2303SCaroline Tice             // Therefore can we pass it to the version of Execute that does not need/expect raw input in the alias.
6085a988416SJim Ingham             return HandleAliasingNormalCommand (args, result);
609844d2303SCaroline Tice         }
610844d2303SCaroline Tice         else
611844d2303SCaroline Tice         {
6125a988416SJim Ingham             return HandleAliasingRawCommand (alias_command, raw_command_string, *cmd_obj, result);
6135a988416SJim Ingham         }
6145a988416SJim Ingham         return result.Succeeded();
6155a988416SJim Ingham     }
6165a988416SJim Ingham 
6175a988416SJim Ingham     bool
6185a988416SJim Ingham     HandleAliasingRawCommand (const std::string &alias_command, std::string &raw_command_string, CommandObject &cmd_obj, CommandReturnObject &result)
6195a988416SJim Ingham     {
620844d2303SCaroline Tice             // Verify & handle any options/arguments passed to the alias command
621844d2303SCaroline Tice 
622844d2303SCaroline Tice             OptionArgVectorSP option_arg_vector_sp = OptionArgVectorSP (new OptionArgVector);
623844d2303SCaroline Tice 
624212130acSEnrico Granata             if (CommandObjectSP cmd_obj_sp = m_interpreter.GetCommandSPExact (cmd_obj.GetCommandName(), false))
625844d2303SCaroline Tice             {
626844d2303SCaroline Tice                 if (m_interpreter.AliasExists (alias_command.c_str())
627844d2303SCaroline Tice                     || m_interpreter.UserCommandExists (alias_command.c_str()))
628844d2303SCaroline Tice                 {
629844d2303SCaroline Tice                     result.AppendWarningWithFormat ("Overwriting existing definition for '%s'.\n",
630844d2303SCaroline Tice                                                     alias_command.c_str());
631844d2303SCaroline Tice                 }
632212130acSEnrico Granata                 if (m_interpreter.AddAlias (alias_command.c_str(), cmd_obj_sp, raw_command_string.c_str()))
633472362e6SCaroline Tice                 {
634844d2303SCaroline Tice                     result.SetStatus (eReturnStatusSuccessFinishNoResult);
635844d2303SCaroline Tice                 }
636472362e6SCaroline Tice                 else
637472362e6SCaroline Tice                 {
638472362e6SCaroline Tice                     result.AppendError ("Unable to create requested alias.\n");
639472362e6SCaroline Tice                     result.SetStatus (eReturnStatusFailed);
640472362e6SCaroline Tice                 }
641212130acSEnrico Granata 
642212130acSEnrico Granata             }
643212130acSEnrico Granata             else
644212130acSEnrico Granata             {
645212130acSEnrico Granata                 result.AppendError ("Unable to create requested alias.\n");
646212130acSEnrico Granata                 result.SetStatus (eReturnStatusFailed);
647212130acSEnrico Granata             }
648212130acSEnrico Granata 
649844d2303SCaroline Tice             return result.Succeeded ();
650844d2303SCaroline Tice     }
651ebc09c36SJim Ingham 
652ebc09c36SJim Ingham     bool
6535a988416SJim Ingham     HandleAliasingNormalCommand (Args& args, CommandReturnObject &result)
654ebc09c36SJim Ingham     {
655867b185dSCaroline Tice         size_t argc = args.GetArgumentCount();
656ebc09c36SJim Ingham 
657ebc09c36SJim Ingham         if (argc < 2)
658ebc09c36SJim Ingham         {
659ebc09c36SJim Ingham             result.AppendError ("'alias' requires at least two arguments");
660ebc09c36SJim Ingham             result.SetStatus (eReturnStatusFailed);
661ebc09c36SJim Ingham             return false;
662ebc09c36SJim Ingham         }
663ebc09c36SJim Ingham 
664ebc09c36SJim Ingham         const std::string alias_command = args.GetArgumentAtIndex(0);
665ebc09c36SJim Ingham         const std::string actual_command = args.GetArgumentAtIndex(1);
666ebc09c36SJim Ingham 
667ebc09c36SJim Ingham         args.Shift();  // Shift the alias command word off the argument vector.
668ebc09c36SJim Ingham         args.Shift();  // Shift the old command word off the argument vector.
669ebc09c36SJim Ingham 
670ebc09c36SJim Ingham         // Verify that the command is alias'able, and get the appropriate command object.
671ebc09c36SJim Ingham 
672a7015092SGreg Clayton         if (m_interpreter.CommandExists (alias_command.c_str()))
673ebc09c36SJim Ingham         {
674ebc09c36SJim Ingham             result.AppendErrorWithFormat ("'%s' is a permanent debugger command and cannot be redefined.\n",
675ebc09c36SJim Ingham                                          alias_command.c_str());
676ebc09c36SJim Ingham             result.SetStatus (eReturnStatusFailed);
677ebc09c36SJim Ingham         }
678ebc09c36SJim Ingham         else
679ebc09c36SJim Ingham         {
680a7015092SGreg Clayton              CommandObjectSP command_obj_sp(m_interpreter.GetCommandSPExact (actual_command.c_str(), true));
681ebc09c36SJim Ingham              CommandObjectSP subcommand_obj_sp;
682ebc09c36SJim Ingham              bool use_subcommand = false;
6836e3d8e7fSEugene Zelenko              if (command_obj_sp)
684ebc09c36SJim Ingham              {
685ebc09c36SJim Ingham                  CommandObject *cmd_obj = command_obj_sp.get();
6866e3d8e7fSEugene Zelenko                  CommandObject *sub_cmd_obj = nullptr;
687ebc09c36SJim Ingham                  OptionArgVectorSP option_arg_vector_sp = OptionArgVectorSP (new OptionArgVector);
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());
6966e3d8e7fSEugene Zelenko                          if (subcommand_obj_sp)
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 
716212130acSEnrico Granata                  std::string args_string;
717212130acSEnrico Granata 
718ebc09c36SJim Ingham                  if (args.GetArgumentCount () > 0)
719ebc09c36SJim Ingham                  {
720ca90c47eSCaroline Tice                     CommandObjectSP tmp_sp = m_interpreter.GetCommandSPExact (cmd_obj->GetCommandName(), false);
721ebc09c36SJim Ingham                     if (use_subcommand)
722ca90c47eSCaroline Tice                         tmp_sp = m_interpreter.GetCommandSPExact (sub_cmd_obj->GetCommandName(), false);
723ca90c47eSCaroline Tice 
724ca90c47eSCaroline Tice                     args.GetCommandString (args_string);
725867b185dSCaroline Tice                  }
726ebc09c36SJim Ingham 
727a7015092SGreg Clayton                  if (m_interpreter.AliasExists (alias_command.c_str())
728a7015092SGreg Clayton                      || m_interpreter.UserCommandExists (alias_command.c_str()))
729ebc09c36SJim Ingham                  {
730ebc09c36SJim Ingham                      result.AppendWarningWithFormat ("Overwriting existing definition for '%s'.\n",
731ebc09c36SJim Ingham                                                      alias_command.c_str());
732ebc09c36SJim Ingham                  }
733ebc09c36SJim Ingham 
734212130acSEnrico Granata                  if (m_interpreter.AddAlias(alias_command.c_str(),
735212130acSEnrico Granata                                             use_subcommand ? subcommand_obj_sp : command_obj_sp,
736212130acSEnrico Granata                                             args_string.c_str()))
737212130acSEnrico Granata                  {
738ebc09c36SJim Ingham                      result.SetStatus (eReturnStatusSuccessFinishNoResult);
739ebc09c36SJim Ingham                  }
740ebc09c36SJim Ingham                  else
741ebc09c36SJim Ingham                  {
742212130acSEnrico Granata                      result.AppendError ("Unable to create requested alias.\n");
743212130acSEnrico Granata                      result.SetStatus (eReturnStatusFailed);
744212130acSEnrico Granata                      return false;
745212130acSEnrico Granata                  }
746212130acSEnrico Granata              }
747212130acSEnrico Granata              else
748212130acSEnrico Granata              {
749ebc09c36SJim Ingham                  result.AppendErrorWithFormat ("'%s' is not an existing command.\n", actual_command.c_str());
750ebc09c36SJim Ingham                  result.SetStatus (eReturnStatusFailed);
751e7941795SCaroline Tice                  return false;
752ebc09c36SJim Ingham              }
753ebc09c36SJim Ingham         }
754ebc09c36SJim Ingham 
755ebc09c36SJim Ingham         return result.Succeeded();
756ebc09c36SJim Ingham     }
757ebc09c36SJim Ingham };
758ebc09c36SJim Ingham 
759ebc09c36SJim Ingham #pragma mark CommandObjectCommandsUnalias
760ebc09c36SJim Ingham //-------------------------------------------------------------------------
761ebc09c36SJim Ingham // CommandObjectCommandsUnalias
762ebc09c36SJim Ingham //-------------------------------------------------------------------------
763ebc09c36SJim Ingham 
7645a988416SJim Ingham class CommandObjectCommandsUnalias : public CommandObjectParsed
765ebc09c36SJim Ingham {
766ebc09c36SJim Ingham public:
767a7015092SGreg Clayton     CommandObjectCommandsUnalias (CommandInterpreter &interpreter) :
7685a988416SJim Ingham         CommandObjectParsed(interpreter,
7690e5e5a79SGreg Clayton                             "command unalias",
77086ddae50SCaroline Tice                             "Allow the user to remove/delete a user-defined command abbreviation.",
7716e3d8e7fSEugene Zelenko                             nullptr)
772ebc09c36SJim Ingham     {
773405fe67fSCaroline Tice         CommandArgumentEntry arg;
774405fe67fSCaroline Tice         CommandArgumentData alias_arg;
775405fe67fSCaroline Tice 
776405fe67fSCaroline Tice         // Define the first (and only) variant of this arg.
777405fe67fSCaroline Tice         alias_arg.arg_type = eArgTypeAliasName;
778405fe67fSCaroline Tice         alias_arg.arg_repetition = eArgRepeatPlain;
779405fe67fSCaroline Tice 
780405fe67fSCaroline Tice         // There is only one variant this argument could be; put it into the argument entry.
781405fe67fSCaroline Tice         arg.push_back (alias_arg);
782405fe67fSCaroline Tice 
783405fe67fSCaroline Tice         // Push the data for the first argument into the m_arguments vector.
784405fe67fSCaroline Tice         m_arguments.push_back (arg);
785ebc09c36SJim Ingham     }
786ebc09c36SJim Ingham 
7876e3d8e7fSEugene Zelenko     ~CommandObjectCommandsUnalias() override = default;
788ebc09c36SJim Ingham 
7895a988416SJim Ingham protected:
790ebc09c36SJim Ingham     bool
79113d21e9aSBruce Mitchener     DoExecute (Args& args, CommandReturnObject &result) override
792ebc09c36SJim Ingham     {
793ebc09c36SJim Ingham         CommandObject::CommandMap::iterator pos;
794ebc09c36SJim Ingham         CommandObject *cmd_obj;
795ebc09c36SJim Ingham 
796ebc09c36SJim Ingham         if (args.GetArgumentCount() != 0)
797ebc09c36SJim Ingham         {
798ebc09c36SJim Ingham             const char *command_name = args.GetArgumentAtIndex(0);
799a7015092SGreg Clayton             cmd_obj = m_interpreter.GetCommandObject(command_name);
800ebc09c36SJim Ingham             if (cmd_obj)
801ebc09c36SJim Ingham             {
802a7015092SGreg Clayton                 if (m_interpreter.CommandExists (command_name))
803ebc09c36SJim Ingham                 {
804b547278cSGreg Clayton                     if (cmd_obj->IsRemovable())
805b547278cSGreg Clayton                     {
806b547278cSGreg Clayton                         result.AppendErrorWithFormat ("'%s' is not an alias, it is a debugger command which can be removed using the 'command delete' command.\n",
807b547278cSGreg Clayton                                                       command_name);
808b547278cSGreg Clayton                     }
809b547278cSGreg Clayton                     else
810b547278cSGreg Clayton                     {
811ebc09c36SJim Ingham                         result.AppendErrorWithFormat ("'%s' is a permanent debugger command and cannot be removed.\n",
812ebc09c36SJim Ingham                                                       command_name);
813b547278cSGreg Clayton                     }
814ebc09c36SJim Ingham                     result.SetStatus (eReturnStatusFailed);
815ebc09c36SJim Ingham                 }
816ebc09c36SJim Ingham                 else
817ebc09c36SJim Ingham                 {
8186e3d8e7fSEugene Zelenko                     if (!m_interpreter.RemoveAlias(command_name))
819ebc09c36SJim Ingham                     {
820a7015092SGreg Clayton                         if (m_interpreter.AliasExists (command_name))
821ebc09c36SJim Ingham                             result.AppendErrorWithFormat ("Error occurred while attempting to unalias '%s'.\n",
822ebc09c36SJim Ingham                                                           command_name);
823ebc09c36SJim Ingham                         else
824ebc09c36SJim Ingham                             result.AppendErrorWithFormat ("'%s' is not an existing alias.\n", command_name);
825ebc09c36SJim Ingham                         result.SetStatus (eReturnStatusFailed);
826ebc09c36SJim Ingham                     }
827ebc09c36SJim Ingham                     else
828ebc09c36SJim Ingham                         result.SetStatus (eReturnStatusSuccessFinishNoResult);
829ebc09c36SJim Ingham                 }
830ebc09c36SJim Ingham             }
831ebc09c36SJim Ingham             else
832ebc09c36SJim Ingham             {
833ebc09c36SJim Ingham                 result.AppendErrorWithFormat ("'%s' is not a known command.\nTry 'help' to see a "
834ebc09c36SJim Ingham                                               "current list of commands.\n",
835ebc09c36SJim Ingham                                              command_name);
836ebc09c36SJim Ingham                 result.SetStatus (eReturnStatusFailed);
837ebc09c36SJim Ingham             }
838ebc09c36SJim Ingham         }
839ebc09c36SJim Ingham         else
840ebc09c36SJim Ingham         {
841ebc09c36SJim Ingham             result.AppendError ("must call 'unalias' with a valid alias");
842ebc09c36SJim Ingham             result.SetStatus (eReturnStatusFailed);
843ebc09c36SJim Ingham         }
844ebc09c36SJim Ingham 
845ebc09c36SJim Ingham         return result.Succeeded();
846ebc09c36SJim Ingham     }
847ebc09c36SJim Ingham };
848ebc09c36SJim Ingham 
849b547278cSGreg Clayton #pragma mark CommandObjectCommandsDelete
850b547278cSGreg Clayton //-------------------------------------------------------------------------
851b547278cSGreg Clayton // CommandObjectCommandsDelete
852b547278cSGreg Clayton //-------------------------------------------------------------------------
853b547278cSGreg Clayton 
854b547278cSGreg Clayton class CommandObjectCommandsDelete : public CommandObjectParsed
855b547278cSGreg Clayton {
856b547278cSGreg Clayton public:
857b547278cSGreg Clayton     CommandObjectCommandsDelete (CommandInterpreter &interpreter) :
858b547278cSGreg Clayton         CommandObjectParsed(interpreter,
859b547278cSGreg Clayton                             "command delete",
860b547278cSGreg Clayton                             "Allow the user to delete user-defined regular expression, python or multi-word commands.",
8616e3d8e7fSEugene Zelenko                             nullptr)
862b547278cSGreg Clayton     {
863b547278cSGreg Clayton         CommandArgumentEntry arg;
864b547278cSGreg Clayton         CommandArgumentData alias_arg;
865b547278cSGreg Clayton 
866b547278cSGreg Clayton         // Define the first (and only) variant of this arg.
867b547278cSGreg Clayton         alias_arg.arg_type = eArgTypeCommandName;
868b547278cSGreg Clayton         alias_arg.arg_repetition = eArgRepeatPlain;
869b547278cSGreg Clayton 
870b547278cSGreg Clayton         // There is only one variant this argument could be; put it into the argument entry.
871b547278cSGreg Clayton         arg.push_back (alias_arg);
872b547278cSGreg Clayton 
873b547278cSGreg Clayton         // Push the data for the first argument into the m_arguments vector.
874b547278cSGreg Clayton         m_arguments.push_back (arg);
875b547278cSGreg Clayton     }
876b547278cSGreg Clayton 
8776e3d8e7fSEugene Zelenko     ~CommandObjectCommandsDelete() override = default;
878b547278cSGreg Clayton 
879b547278cSGreg Clayton protected:
880b547278cSGreg Clayton     bool
88113d21e9aSBruce Mitchener     DoExecute (Args& args, CommandReturnObject &result) override
882b547278cSGreg Clayton     {
883b547278cSGreg Clayton         CommandObject::CommandMap::iterator pos;
884b547278cSGreg Clayton 
885b547278cSGreg Clayton         if (args.GetArgumentCount() != 0)
886b547278cSGreg Clayton         {
887b547278cSGreg Clayton             const char *command_name = args.GetArgumentAtIndex(0);
888b547278cSGreg Clayton             if (m_interpreter.CommandExists (command_name))
889b547278cSGreg Clayton             {
890b547278cSGreg Clayton                 if (m_interpreter.RemoveCommand (command_name))
891b547278cSGreg Clayton                 {
892b547278cSGreg Clayton                     result.SetStatus (eReturnStatusSuccessFinishNoResult);
893b547278cSGreg Clayton                 }
894b547278cSGreg Clayton                 else
895b547278cSGreg Clayton                 {
896b547278cSGreg Clayton                     result.AppendErrorWithFormat ("'%s' is a permanent debugger command and cannot be removed.\n",
897b547278cSGreg Clayton                                                   command_name);
898b547278cSGreg Clayton                     result.SetStatus (eReturnStatusFailed);
899b547278cSGreg Clayton                 }
900b547278cSGreg Clayton             }
901b547278cSGreg Clayton             else
902b547278cSGreg Clayton             {
90346d4aa21SEnrico Granata                 StreamString error_msg_stream;
90446d4aa21SEnrico Granata                 const bool generate_apropos = true;
90546d4aa21SEnrico Granata                 const bool generate_type_lookup = false;
90646d4aa21SEnrico Granata                 CommandObjectHelp::GenerateAdditionalHelpAvenuesMessage(&error_msg_stream,
90746d4aa21SEnrico Granata                                                                         command_name,
90846d4aa21SEnrico Granata                                                                         nullptr,
90946d4aa21SEnrico Granata                                                                         nullptr,
91046d4aa21SEnrico Granata                                                                         generate_apropos,
91146d4aa21SEnrico Granata                                                                         generate_type_lookup);
91246d4aa21SEnrico Granata                 result.AppendErrorWithFormat ("%s", error_msg_stream.GetData());
913b547278cSGreg Clayton                 result.SetStatus (eReturnStatusFailed);
914b547278cSGreg Clayton             }
915b547278cSGreg Clayton         }
916b547278cSGreg Clayton         else
917b547278cSGreg Clayton         {
918b547278cSGreg Clayton             result.AppendErrorWithFormat ("must call '%s' with one or more valid user defined regular expression, python or multi-word command names", GetCommandName ());
919b547278cSGreg Clayton             result.SetStatus (eReturnStatusFailed);
920b547278cSGreg Clayton         }
921b547278cSGreg Clayton 
922b547278cSGreg Clayton         return result.Succeeded();
923b547278cSGreg Clayton     }
924b547278cSGreg Clayton };
925b547278cSGreg Clayton 
926de164aaaSGreg Clayton //-------------------------------------------------------------------------
927de164aaaSGreg Clayton // CommandObjectCommandsAddRegex
928de164aaaSGreg Clayton //-------------------------------------------------------------------------
9295a988416SJim Ingham #pragma mark CommandObjectCommandsAddRegex
930de164aaaSGreg Clayton 
93144d93782SGreg Clayton class CommandObjectCommandsAddRegex :
93244d93782SGreg Clayton     public CommandObjectParsed,
933ea508635SGreg Clayton     public IOHandlerDelegateMultiline
934de164aaaSGreg Clayton {
935de164aaaSGreg Clayton public:
936de164aaaSGreg Clayton     CommandObjectCommandsAddRegex (CommandInterpreter &interpreter) :
9375a988416SJim Ingham         CommandObjectParsed (interpreter,
9380e5e5a79SGreg Clayton                        "command regex",
939de164aaaSGreg Clayton                        "Allow the user to create a regular expression command.",
9400e5e5a79SGreg Clayton                        "command regex <cmd-name> [s/<regex>/<subst>/ ...]"),
941ea508635SGreg Clayton         IOHandlerDelegateMultiline ("", IOHandlerDelegate::Completion::LLDBCommand),
942de164aaaSGreg Clayton         m_options (interpreter)
943de164aaaSGreg Clayton     {
944ea671fbdSKate Stone         SetHelpLong(R"(
945ea671fbdSKate Stone )" "This command allows the user to create powerful regular expression commands \
946ea671fbdSKate Stone with substitutions. The regular expressions and substitutions are specified \
947ea671fbdSKate Stone using the regular expression substitution format of:" R"(
948ea671fbdSKate Stone 
949ea671fbdSKate Stone     s/<regex>/<subst>/
950ea671fbdSKate Stone 
951ea671fbdSKate Stone )" "<regex> is a regular expression that can use parenthesis to capture regular \
952ea671fbdSKate Stone expression input and substitute the captured matches in the output using %1 \
953ea671fbdSKate Stone for the first match, %2 for the second, and so on." R"(
954ea671fbdSKate Stone 
955ea671fbdSKate Stone )" "The regular expressions can all be specified on the command line if more than \
956ea671fbdSKate Stone one argument is provided. If just the command name is provided on the command \
957ea671fbdSKate Stone line, then the regular expressions and substitutions can be entered on separate \
958ea671fbdSKate Stone lines, followed by an empty line to terminate the command definition." R"(
959ea671fbdSKate Stone 
960ea671fbdSKate Stone EXAMPLES
961ea671fbdSKate Stone 
962ea671fbdSKate Stone )" "The following example will define a regular expression command named 'f' that \
963ea671fbdSKate Stone will call 'finish' if there are no arguments, or 'frame select <frame-idx>' if \
964ea671fbdSKate Stone a number follows 'f':" R"(
965ea671fbdSKate Stone 
966ea671fbdSKate Stone     (lldb) command regex f s/^$/finish/ 's/([0-9]+)/frame select %1/')"
9670e5e5a79SGreg Clayton         );
968de164aaaSGreg Clayton     }
969de164aaaSGreg Clayton 
9706e3d8e7fSEugene Zelenko     ~CommandObjectCommandsAddRegex() override = default;
971de164aaaSGreg Clayton 
9725a988416SJim Ingham protected:
973ea508635SGreg Clayton     void
974ea508635SGreg Clayton     IOHandlerActivated (IOHandler &io_handler) override
97544d93782SGreg Clayton     {
97644d93782SGreg Clayton         StreamFileSP output_sp(io_handler.GetOutputStreamFile());
97744d93782SGreg Clayton         if (output_sp)
97844d93782SGreg Clayton         {
97944d93782SGreg 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");
98044d93782SGreg Clayton             output_sp->Flush();
98144d93782SGreg Clayton         }
98244d93782SGreg Clayton     }
98344d93782SGreg Clayton 
984ea508635SGreg Clayton     void
985ea508635SGreg Clayton     IOHandlerInputComplete (IOHandler &io_handler, std::string &data) override
98644d93782SGreg Clayton     {
98744d93782SGreg Clayton         io_handler.SetIsDone(true);
9886e3d8e7fSEugene Zelenko         if (m_regex_cmd_ap)
98944d93782SGreg Clayton         {
99044d93782SGreg Clayton             StringList lines;
99144d93782SGreg Clayton             if (lines.SplitIntoLines (data))
99244d93782SGreg Clayton             {
99344d93782SGreg Clayton                 const size_t num_lines = lines.GetSize();
99444d93782SGreg Clayton                 bool check_only = false;
99544d93782SGreg Clayton                 for (size_t i=0; i<num_lines; ++i)
99644d93782SGreg Clayton                 {
99744d93782SGreg Clayton                     llvm::StringRef bytes_strref (lines[i]);
99844d93782SGreg Clayton                     Error error = AppendRegexSubstitution (bytes_strref, check_only);
99944d93782SGreg Clayton                     if (error.Fail())
100044d93782SGreg Clayton                     {
100144d93782SGreg Clayton                         if (!m_interpreter.GetDebugger().GetCommandInterpreter().GetBatchCommandMode())
100244d93782SGreg Clayton                         {
100344d93782SGreg Clayton                             StreamSP out_stream = m_interpreter.GetDebugger().GetAsyncOutputStream();
100444d93782SGreg Clayton                             out_stream->Printf("error: %s\n", error.AsCString());
100544d93782SGreg Clayton                         }
100644d93782SGreg Clayton                     }
100744d93782SGreg Clayton                 }
100844d93782SGreg Clayton             }
100944d93782SGreg Clayton             if (m_regex_cmd_ap->HasRegexEntries())
101044d93782SGreg Clayton             {
101144d93782SGreg Clayton                 CommandObjectSP cmd_sp (m_regex_cmd_ap.release());
101244d93782SGreg Clayton                 m_interpreter.AddCommand(cmd_sp->GetCommandName(), cmd_sp, true);
101344d93782SGreg Clayton             }
101444d93782SGreg Clayton         }
101544d93782SGreg Clayton     }
101644d93782SGreg Clayton 
1017de164aaaSGreg Clayton     bool
1018b0a1814fSEric Christopher     DoExecute (Args& command, CommandReturnObject &result) override
1019de164aaaSGreg Clayton     {
10205a988416SJim Ingham         const size_t argc = command.GetArgumentCount();
10210e5e5a79SGreg Clayton         if (argc == 0)
1022de164aaaSGreg Clayton         {
102369c12ccbSJason Molenda             result.AppendError ("usage: 'command regex <command-name> [s/<regex1>/<subst1>/ s/<regex2>/<subst2>/ ...]'\n");
10240e5e5a79SGreg Clayton             result.SetStatus (eReturnStatusFailed);
10250e5e5a79SGreg Clayton         }
10260e5e5a79SGreg Clayton         else
10270e5e5a79SGreg Clayton         {
10280e5e5a79SGreg Clayton             Error error;
10295a988416SJim Ingham             const char *name = command.GetArgumentAtIndex(0);
1030de164aaaSGreg Clayton             m_regex_cmd_ap.reset (new CommandObjectRegexCommand (m_interpreter,
1031de164aaaSGreg Clayton                                                                  name,
1032de164aaaSGreg Clayton                                                                  m_options.GetHelp (),
1033de164aaaSGreg Clayton                                                                  m_options.GetSyntax (),
1034b547278cSGreg Clayton                                                                  10,
1035b547278cSGreg Clayton                                                                  0,
1036b547278cSGreg Clayton                                                                  true));
10370e5e5a79SGreg Clayton 
10380e5e5a79SGreg Clayton             if (argc == 1)
10390e5e5a79SGreg Clayton             {
104044d93782SGreg Clayton                 Debugger &debugger = m_interpreter.GetDebugger();
1041e30f11d9SKate Stone                 bool color_prompt = debugger.GetUseColor();
104244d93782SGreg Clayton                 const bool multiple_lines = true; // Get multiple lines
104344d93782SGreg Clayton                 IOHandlerSP io_handler_sp(new IOHandlerEditline(debugger,
1044e30f11d9SKate Stone                                                                 IOHandler::Type::Other,
104573d80faaSGreg Clayton                                                                 "lldb-regex", // Name of input reader for history
1046ea508635SGreg Clayton                                                                 "> ",         // Prompt
10476e3d8e7fSEugene Zelenko                                                                 nullptr,      // Continuation prompt
104844d93782SGreg Clayton                                                                 multiple_lines,
1049e30f11d9SKate Stone                                                                 color_prompt,
1050f6913cd7SGreg Clayton                                                                 0,            // Don't show line numbers
105144d93782SGreg Clayton                                                                 *this));
105244d93782SGreg Clayton 
105344d93782SGreg Clayton                 if (io_handler_sp)
1054de164aaaSGreg Clayton                 {
105544d93782SGreg Clayton                     debugger.PushIOHandler(io_handler_sp);
1056de164aaaSGreg Clayton                     result.SetStatus (eReturnStatusSuccessFinishNoResult);
1057de164aaaSGreg Clayton                 }
1058de164aaaSGreg Clayton             }
1059de164aaaSGreg Clayton             else
1060de164aaaSGreg Clayton             {
10610e5e5a79SGreg Clayton                 for (size_t arg_idx = 1; arg_idx < argc; ++arg_idx)
10620e5e5a79SGreg Clayton                 {
10635a988416SJim Ingham                     llvm::StringRef arg_strref (command.GetArgumentAtIndex(arg_idx));
106444d93782SGreg Clayton                     bool check_only = false;
106544d93782SGreg Clayton                     error = AppendRegexSubstitution (arg_strref, check_only);
10660e5e5a79SGreg Clayton                     if (error.Fail())
10670e5e5a79SGreg Clayton                         break;
10680e5e5a79SGreg Clayton                 }
10690e5e5a79SGreg Clayton 
10700e5e5a79SGreg Clayton                 if (error.Success())
10710e5e5a79SGreg Clayton                 {
10720e5e5a79SGreg Clayton                     AddRegexCommandToInterpreter();
10730e5e5a79SGreg Clayton                 }
10740e5e5a79SGreg Clayton             }
10750e5e5a79SGreg Clayton             if (error.Fail())
10760e5e5a79SGreg Clayton             {
10770e5e5a79SGreg Clayton                 result.AppendError (error.AsCString());
1078de164aaaSGreg Clayton                 result.SetStatus (eReturnStatusFailed);
1079de164aaaSGreg Clayton             }
10800e5e5a79SGreg Clayton         }
10810e5e5a79SGreg Clayton 
1082de164aaaSGreg Clayton         return result.Succeeded();
1083de164aaaSGreg Clayton     }
1084de164aaaSGreg Clayton 
10850e5e5a79SGreg Clayton     Error
108644d93782SGreg Clayton     AppendRegexSubstitution (const llvm::StringRef &regex_sed, bool check_only)
1087de164aaaSGreg Clayton     {
10880e5e5a79SGreg Clayton         Error error;
10890e5e5a79SGreg Clayton 
10906e3d8e7fSEugene Zelenko         if (!m_regex_cmd_ap)
1091de164aaaSGreg Clayton         {
10920e5e5a79SGreg Clayton             error.SetErrorStringWithFormat("invalid regular expression command object for: '%.*s'",
10930e5e5a79SGreg Clayton                                            (int)regex_sed.size(),
10940e5e5a79SGreg Clayton                                            regex_sed.data());
10950e5e5a79SGreg Clayton             return error;
1096de164aaaSGreg Clayton         }
10970e5e5a79SGreg Clayton 
10980e5e5a79SGreg Clayton         size_t regex_sed_size = regex_sed.size();
10990e5e5a79SGreg Clayton 
11000e5e5a79SGreg Clayton         if (regex_sed_size <= 1)
11010e5e5a79SGreg Clayton         {
11020e5e5a79SGreg Clayton             error.SetErrorStringWithFormat("regular expression substitution string is too short: '%.*s'",
11030e5e5a79SGreg Clayton                                            (int)regex_sed.size(),
11040e5e5a79SGreg Clayton                                            regex_sed.data());
11050e5e5a79SGreg Clayton             return error;
11060e5e5a79SGreg Clayton         }
11070e5e5a79SGreg Clayton 
11080e5e5a79SGreg Clayton         if (regex_sed[0] != 's')
11090e5e5a79SGreg Clayton         {
11100e5e5a79SGreg Clayton             error.SetErrorStringWithFormat("regular expression substitution string doesn't start with 's': '%.*s'",
11110e5e5a79SGreg Clayton                                            (int)regex_sed.size(),
11120e5e5a79SGreg Clayton                                            regex_sed.data());
11130e5e5a79SGreg Clayton             return error;
11140e5e5a79SGreg Clayton         }
11150e5e5a79SGreg Clayton         const size_t first_separator_char_pos = 1;
11160e5e5a79SGreg Clayton         // use the char that follows 's' as the regex separator character
11170e5e5a79SGreg Clayton         // so we can have "s/<regex>/<subst>/" or "s|<regex>|<subst>|"
11180e5e5a79SGreg Clayton         const char separator_char = regex_sed[first_separator_char_pos];
11190e5e5a79SGreg Clayton         const size_t second_separator_char_pos = regex_sed.find (separator_char, first_separator_char_pos + 1);
11200e5e5a79SGreg Clayton 
11210e5e5a79SGreg Clayton         if (second_separator_char_pos == std::string::npos)
11220e5e5a79SGreg Clayton         {
1123ea508635SGreg Clayton             error.SetErrorStringWithFormat("missing second '%c' separator char after '%.*s' in '%.*s'",
11240e5e5a79SGreg Clayton                                            separator_char,
11250e5e5a79SGreg Clayton                                            (int)(regex_sed.size() - first_separator_char_pos - 1),
1126ea508635SGreg Clayton                                            regex_sed.data() + (first_separator_char_pos + 1),
1127ea508635SGreg Clayton                                            (int)regex_sed.size(),
1128ea508635SGreg Clayton                                            regex_sed.data());
11290e5e5a79SGreg Clayton             return error;
11300e5e5a79SGreg Clayton         }
11310e5e5a79SGreg Clayton 
11320e5e5a79SGreg Clayton         const size_t third_separator_char_pos = regex_sed.find (separator_char, second_separator_char_pos + 1);
11330e5e5a79SGreg Clayton 
11340e5e5a79SGreg Clayton         if (third_separator_char_pos == std::string::npos)
11350e5e5a79SGreg Clayton         {
1136ea508635SGreg Clayton             error.SetErrorStringWithFormat("missing third '%c' separator char after '%.*s' in '%.*s'",
11370e5e5a79SGreg Clayton                                            separator_char,
11380e5e5a79SGreg Clayton                                            (int)(regex_sed.size() - second_separator_char_pos - 1),
1139ea508635SGreg Clayton                                            regex_sed.data() + (second_separator_char_pos + 1),
1140ea508635SGreg Clayton                                            (int)regex_sed.size(),
1141ea508635SGreg Clayton                                            regex_sed.data());
11420e5e5a79SGreg Clayton             return error;
11430e5e5a79SGreg Clayton         }
11440e5e5a79SGreg Clayton 
11450e5e5a79SGreg Clayton         if (third_separator_char_pos != regex_sed_size - 1)
11460e5e5a79SGreg Clayton         {
11470e5e5a79SGreg Clayton             // Make sure that everything that follows the last regex
11480e5e5a79SGreg Clayton             // separator char
11490e5e5a79SGreg Clayton             if (regex_sed.find_first_not_of("\t\n\v\f\r ", third_separator_char_pos + 1) != std::string::npos)
11500e5e5a79SGreg Clayton             {
11510e5e5a79SGreg Clayton                 error.SetErrorStringWithFormat("extra data found after the '%.*s' regular expression substitution string: '%.*s'",
11520e5e5a79SGreg Clayton                                                (int)third_separator_char_pos + 1,
11530e5e5a79SGreg Clayton                                                regex_sed.data(),
11540e5e5a79SGreg Clayton                                                (int)(regex_sed.size() - third_separator_char_pos - 1),
11550e5e5a79SGreg Clayton                                                regex_sed.data() + (third_separator_char_pos + 1));
11560e5e5a79SGreg Clayton                 return error;
11570e5e5a79SGreg Clayton             }
11580e5e5a79SGreg Clayton         }
11590e5e5a79SGreg Clayton         else if (first_separator_char_pos + 1 == second_separator_char_pos)
11600e5e5a79SGreg Clayton         {
11610e5e5a79SGreg Clayton             error.SetErrorStringWithFormat("<regex> can't be empty in 's%c<regex>%c<subst>%c' string: '%.*s'",
11620e5e5a79SGreg Clayton                                            separator_char,
11630e5e5a79SGreg Clayton                                            separator_char,
11640e5e5a79SGreg Clayton                                            separator_char,
11650e5e5a79SGreg Clayton                                            (int)regex_sed.size(),
11660e5e5a79SGreg Clayton                                            regex_sed.data());
11670e5e5a79SGreg Clayton             return error;
11680e5e5a79SGreg Clayton         }
11690e5e5a79SGreg Clayton         else if (second_separator_char_pos + 1 == third_separator_char_pos)
11700e5e5a79SGreg Clayton         {
11710e5e5a79SGreg Clayton             error.SetErrorStringWithFormat("<subst> can't be empty in 's%c<regex>%c<subst>%c' string: '%.*s'",
11720e5e5a79SGreg Clayton                                            separator_char,
11730e5e5a79SGreg Clayton                                            separator_char,
11740e5e5a79SGreg Clayton                                            separator_char,
11750e5e5a79SGreg Clayton                                            (int)regex_sed.size(),
11760e5e5a79SGreg Clayton                                            regex_sed.data());
11770e5e5a79SGreg Clayton             return error;
11780e5e5a79SGreg Clayton         }
117944d93782SGreg Clayton 
11806e3d8e7fSEugene Zelenko         if (!check_only)
118144d93782SGreg Clayton         {
11820e5e5a79SGreg Clayton             std::string regex(regex_sed.substr(first_separator_char_pos + 1, second_separator_char_pos - first_separator_char_pos - 1));
11830e5e5a79SGreg Clayton             std::string subst(regex_sed.substr(second_separator_char_pos + 1, third_separator_char_pos - second_separator_char_pos - 1));
11840e5e5a79SGreg Clayton             m_regex_cmd_ap->AddRegexCommand (regex.c_str(),
11850e5e5a79SGreg Clayton                                              subst.c_str());
118644d93782SGreg Clayton         }
11870e5e5a79SGreg Clayton         return error;
1188de164aaaSGreg Clayton     }
1189de164aaaSGreg Clayton 
1190de164aaaSGreg Clayton     void
11910e5e5a79SGreg Clayton     AddRegexCommandToInterpreter()
1192de164aaaSGreg Clayton     {
11936e3d8e7fSEugene Zelenko         if (m_regex_cmd_ap)
1194de164aaaSGreg Clayton         {
1195de164aaaSGreg Clayton             if (m_regex_cmd_ap->HasRegexEntries())
1196de164aaaSGreg Clayton             {
1197de164aaaSGreg Clayton                 CommandObjectSP cmd_sp (m_regex_cmd_ap.release());
1198de164aaaSGreg Clayton                 m_interpreter.AddCommand(cmd_sp->GetCommandName(), cmd_sp, true);
1199de164aaaSGreg Clayton             }
1200de164aaaSGreg Clayton         }
1201de164aaaSGreg Clayton     }
1202de164aaaSGreg Clayton 
1203de164aaaSGreg Clayton private:
12047b0992d9SGreg Clayton     std::unique_ptr<CommandObjectRegexCommand> m_regex_cmd_ap;
1205de164aaaSGreg Clayton 
1206de164aaaSGreg Clayton      class CommandOptions : public Options
1207de164aaaSGreg Clayton      {
1208de164aaaSGreg Clayton      public:
1209de164aaaSGreg Clayton          CommandOptions (CommandInterpreter &interpreter) :
1210de164aaaSGreg Clayton             Options (interpreter)
1211de164aaaSGreg Clayton          {
1212de164aaaSGreg Clayton          }
1213de164aaaSGreg Clayton 
12146e3d8e7fSEugene Zelenko          ~CommandOptions() override = default;
1215de164aaaSGreg Clayton 
121613d21e9aSBruce Mitchener          Error
121713d21e9aSBruce Mitchener          SetOptionValue (uint32_t option_idx, const char *option_arg) override
1218de164aaaSGreg Clayton          {
1219de164aaaSGreg Clayton              Error error;
12203bcdfc0eSGreg Clayton              const int short_option = m_getopt_table[option_idx].val;
1221de164aaaSGreg Clayton 
1222de164aaaSGreg Clayton              switch (short_option)
1223de164aaaSGreg Clayton              {
1224de164aaaSGreg Clayton                  case 'h':
1225de164aaaSGreg Clayton                      m_help.assign (option_arg);
1226de164aaaSGreg Clayton                      break;
1227de164aaaSGreg Clayton                  case 's':
1228de164aaaSGreg Clayton                      m_syntax.assign (option_arg);
1229de164aaaSGreg Clayton                      break;
1230de164aaaSGreg Clayton                  default:
123186edbf41SGreg Clayton                      error.SetErrorStringWithFormat ("unrecognized option '%c'", short_option);
1232de164aaaSGreg Clayton                      break;
1233de164aaaSGreg Clayton              }
1234de164aaaSGreg Clayton 
1235de164aaaSGreg Clayton              return error;
1236de164aaaSGreg Clayton          }
1237de164aaaSGreg Clayton 
1238de164aaaSGreg Clayton          void
123913d21e9aSBruce Mitchener          OptionParsingStarting () override
1240de164aaaSGreg Clayton          {
1241de164aaaSGreg Clayton              m_help.clear();
1242de164aaaSGreg Clayton              m_syntax.clear();
1243de164aaaSGreg Clayton          }
1244de164aaaSGreg Clayton 
1245de164aaaSGreg Clayton          const OptionDefinition*
124613d21e9aSBruce Mitchener          GetDefinitions () override
1247de164aaaSGreg Clayton          {
1248de164aaaSGreg Clayton              return g_option_table;
1249de164aaaSGreg Clayton          }
1250de164aaaSGreg Clayton 
1251de164aaaSGreg Clayton          // Options table: Required for subclasses of Options.
1252de164aaaSGreg Clayton 
1253de164aaaSGreg Clayton          static OptionDefinition g_option_table[];
1254de164aaaSGreg Clayton 
1255de164aaaSGreg Clayton          const char *
1256de164aaaSGreg Clayton          GetHelp()
1257de164aaaSGreg Clayton          {
12586e3d8e7fSEugene Zelenko              return (m_help.empty() ? nullptr : m_help.c_str());
1259de164aaaSGreg Clayton          }
12606e3d8e7fSEugene Zelenko 
1261de164aaaSGreg Clayton          const char *
1262de164aaaSGreg Clayton          GetSyntax ()
1263de164aaaSGreg Clayton          {
12646e3d8e7fSEugene Zelenko              return (m_syntax.empty() ? nullptr : m_syntax.c_str());
1265de164aaaSGreg Clayton          }
12666e3d8e7fSEugene Zelenko 
1267de164aaaSGreg Clayton      protected:
12686e3d8e7fSEugene Zelenko          // Instance variables to hold the values for command options.
12696e3d8e7fSEugene Zelenko 
1270de164aaaSGreg Clayton          std::string m_help;
1271de164aaaSGreg Clayton          std::string m_syntax;
1272de164aaaSGreg Clayton      };
1273de164aaaSGreg Clayton 
1274b0a1814fSEric Christopher      Options *
1275b0a1814fSEric Christopher      GetOptions () override
1276de164aaaSGreg Clayton      {
1277de164aaaSGreg Clayton          return &m_options;
1278de164aaaSGreg Clayton      }
1279de164aaaSGreg Clayton 
12805a988416SJim Ingham      CommandOptions m_options;
1281de164aaaSGreg Clayton };
1282de164aaaSGreg Clayton 
1283de164aaaSGreg Clayton OptionDefinition
1284de164aaaSGreg Clayton CommandObjectCommandsAddRegex::CommandOptions::g_option_table[] =
1285de164aaaSGreg Clayton {
12866e3d8e7fSEugene Zelenko { LLDB_OPT_SET_1, false, "help"  , 'h', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeNone, "The help text to display for this command."},
12876e3d8e7fSEugene Zelenko { LLDB_OPT_SET_1, false, "syntax", 's', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeNone, "A syntax string showing the typical usage syntax."},
12886e3d8e7fSEugene Zelenko { 0             , false,  nullptr   , 0  , 0                , nullptr, nullptr, 0, eArgTypeNone, nullptr }
1289de164aaaSGreg Clayton };
1290de164aaaSGreg Clayton 
12915a988416SJim Ingham class CommandObjectPythonFunction : public CommandObjectRaw
1292223383edSEnrico Granata {
1293223383edSEnrico Granata public:
1294223383edSEnrico Granata     CommandObjectPythonFunction (CommandInterpreter &interpreter,
1295223383edSEnrico Granata                                  std::string name,
12960a305db7SEnrico Granata                                  std::string funct,
1297735152e3SEnrico Granata                                  std::string help,
12980a305db7SEnrico Granata                                  ScriptedCommandSynchronicity synch) :
12995a988416SJim Ingham         CommandObjectRaw(interpreter,
1300223383edSEnrico Granata                          name.c_str(),
13016e3d8e7fSEugene Zelenko                          nullptr,
13026e3d8e7fSEugene Zelenko                          nullptr),
13030a305db7SEnrico Granata         m_function_name(funct),
1304fac939e9SEnrico Granata         m_synchro(synch),
1305fac939e9SEnrico Granata         m_fetched_help_long(false)
1306223383edSEnrico Granata     {
1307735152e3SEnrico Granata         if (!help.empty())
1308735152e3SEnrico Granata             SetHelp(help.c_str());
1309735152e3SEnrico Granata         else
1310735152e3SEnrico Granata         {
1311735152e3SEnrico Granata             StreamString stream;
1312735152e3SEnrico Granata             stream.Printf("For more information run 'help %s'",name.c_str());
1313735152e3SEnrico Granata             SetHelp(stream.GetData());
1314735152e3SEnrico Granata         }
1315223383edSEnrico Granata     }
1316223383edSEnrico Granata 
13176e3d8e7fSEugene Zelenko     ~CommandObjectPythonFunction() override = default;
1318223383edSEnrico Granata 
131913d21e9aSBruce Mitchener     bool
132013d21e9aSBruce Mitchener     IsRemovable () const override
13215a988416SJim Ingham     {
13225a988416SJim Ingham         return true;
13235a988416SJim Ingham     }
13245a988416SJim Ingham 
13255a988416SJim Ingham     const std::string&
13265a988416SJim Ingham     GetFunctionName ()
13275a988416SJim Ingham     {
13285a988416SJim Ingham         return m_function_name;
13295a988416SJim Ingham     }
13305a988416SJim Ingham 
13315a988416SJim Ingham     ScriptedCommandSynchronicity
13325a988416SJim Ingham     GetSynchronicity ()
13335a988416SJim Ingham     {
13345a988416SJim Ingham         return m_synchro;
13355a988416SJim Ingham     }
13365a988416SJim Ingham 
133713d21e9aSBruce Mitchener     const char *
133813d21e9aSBruce Mitchener     GetHelpLong () override
1339fac939e9SEnrico Granata     {
1340fac939e9SEnrico Granata         if (!m_fetched_help_long)
1341fac939e9SEnrico Granata         {
1342fac939e9SEnrico Granata             ScriptInterpreter* scripter = m_interpreter.GetScriptInterpreter();
1343fac939e9SEnrico Granata             if (scripter)
1344fac939e9SEnrico Granata             {
1345fac939e9SEnrico Granata                 std::string docstring;
1346fac939e9SEnrico Granata                 m_fetched_help_long = scripter->GetDocumentationForItem(m_function_name.c_str(),docstring);
1347fac939e9SEnrico Granata                 if (!docstring.empty())
1348*bfb75e9bSEnrico Granata                     SetHelpLong(docstring.c_str());
1349fac939e9SEnrico Granata             }
1350fac939e9SEnrico Granata         }
1351fac939e9SEnrico Granata         return CommandObjectRaw::GetHelpLong();
1352fac939e9SEnrico Granata     }
1353fac939e9SEnrico Granata 
13545a988416SJim Ingham protected:
135513d21e9aSBruce Mitchener     bool
135613d21e9aSBruce Mitchener     DoExecute (const char *raw_command_line, CommandReturnObject &result) override
1357223383edSEnrico Granata     {
1358223383edSEnrico Granata         ScriptInterpreter* scripter = m_interpreter.GetScriptInterpreter();
1359223383edSEnrico Granata 
1360223383edSEnrico Granata         Error error;
1361223383edSEnrico Granata 
136270f11f88SJim Ingham         result.SetStatus(eReturnStatusInvalid);
136370f11f88SJim Ingham 
13646e3d8e7fSEugene Zelenko         if (!scripter || !scripter->RunScriptBasedCommand(m_function_name.c_str(),
1365223383edSEnrico Granata                                                           raw_command_line,
13660a305db7SEnrico Granata                                                           m_synchro,
1367223383edSEnrico Granata                                                           result,
136806be059aSEnrico Granata                                                           error,
13696e3d8e7fSEugene Zelenko                                                           m_exe_ctx))
1370223383edSEnrico Granata         {
1371223383edSEnrico Granata             result.AppendError(error.AsCString());
1372223383edSEnrico Granata             result.SetStatus(eReturnStatusFailed);
1373223383edSEnrico Granata         }
1374223383edSEnrico Granata         else
137570f11f88SJim Ingham         {
137670f11f88SJim Ingham             // Don't change the status if the command already set it...
137770f11f88SJim Ingham             if (result.GetStatus() == eReturnStatusInvalid)
137870f11f88SJim Ingham             {
13796e3d8e7fSEugene Zelenko                 if (result.GetOutputData() == nullptr || result.GetOutputData()[0] == '\0')
1380223383edSEnrico Granata                     result.SetStatus(eReturnStatusSuccessFinishNoResult);
138170f11f88SJim Ingham                 else
138270f11f88SJim Ingham                     result.SetStatus(eReturnStatusSuccessFinishResult);
138370f11f88SJim Ingham             }
138470f11f88SJim Ingham         }
1385223383edSEnrico Granata 
1386223383edSEnrico Granata         return result.Succeeded();
1387223383edSEnrico Granata     }
1388223383edSEnrico Granata 
13896e3d8e7fSEugene Zelenko private:
13906e3d8e7fSEugene Zelenko     std::string m_function_name;
13916e3d8e7fSEugene Zelenko     ScriptedCommandSynchronicity m_synchro;
13926e3d8e7fSEugene Zelenko     bool m_fetched_help_long;
1393223383edSEnrico Granata };
1394223383edSEnrico Granata 
13959fe00e52SEnrico Granata class CommandObjectScriptingObject : public CommandObjectRaw
13969fe00e52SEnrico Granata {
13979fe00e52SEnrico Granata public:
13989fe00e52SEnrico Granata     CommandObjectScriptingObject (CommandInterpreter &interpreter,
13999fe00e52SEnrico Granata                                   std::string name,
14000641ca1aSZachary Turner                                   StructuredData::GenericSP cmd_obj_sp,
14019fe00e52SEnrico Granata                                   ScriptedCommandSynchronicity synch) :
14029fe00e52SEnrico Granata         CommandObjectRaw(interpreter,
14039fe00e52SEnrico Granata                          name.c_str(),
14046e3d8e7fSEugene Zelenko                          nullptr,
14056e3d8e7fSEugene Zelenko                          nullptr),
14069fe00e52SEnrico Granata         m_cmd_obj_sp(cmd_obj_sp),
14076f79bb2dSEnrico Granata         m_synchro(synch),
14086f79bb2dSEnrico Granata         m_fetched_help_short(false),
14096f79bb2dSEnrico Granata         m_fetched_help_long(false)
14109fe00e52SEnrico Granata     {
14119fe00e52SEnrico Granata         StreamString stream;
14129fe00e52SEnrico Granata         stream.Printf("For more information run 'help %s'",name.c_str());
14139fe00e52SEnrico Granata         SetHelp(stream.GetData());
1414e87764f2SEnrico Granata         if (ScriptInterpreter* scripter = m_interpreter.GetScriptInterpreter())
1415e87764f2SEnrico Granata             GetFlags().Set(scripter->GetFlagsForCommandObject(cmd_obj_sp));
14169fe00e52SEnrico Granata     }
14179fe00e52SEnrico Granata 
14186e3d8e7fSEugene Zelenko     ~CommandObjectScriptingObject() override = default;
14199fe00e52SEnrico Granata 
142013d21e9aSBruce Mitchener     bool
142113d21e9aSBruce Mitchener     IsRemovable () const override
14229fe00e52SEnrico Granata     {
14239fe00e52SEnrico Granata         return true;
14249fe00e52SEnrico Granata     }
14259fe00e52SEnrico Granata 
14260641ca1aSZachary Turner     StructuredData::GenericSP
14279fe00e52SEnrico Granata     GetImplementingObject ()
14289fe00e52SEnrico Granata     {
14299fe00e52SEnrico Granata         return m_cmd_obj_sp;
14309fe00e52SEnrico Granata     }
14319fe00e52SEnrico Granata 
14329fe00e52SEnrico Granata     ScriptedCommandSynchronicity
14339fe00e52SEnrico Granata     GetSynchronicity ()
14349fe00e52SEnrico Granata     {
14359fe00e52SEnrico Granata         return m_synchro;
14369fe00e52SEnrico Granata     }
14379fe00e52SEnrico Granata 
143813d21e9aSBruce Mitchener     const char *
143913d21e9aSBruce Mitchener     GetHelp () override
14406f79bb2dSEnrico Granata     {
14416f79bb2dSEnrico Granata         if (!m_fetched_help_short)
14426f79bb2dSEnrico Granata         {
14436f79bb2dSEnrico Granata             ScriptInterpreter* scripter = m_interpreter.GetScriptInterpreter();
14446f79bb2dSEnrico Granata             if (scripter)
14456f79bb2dSEnrico Granata             {
14466f79bb2dSEnrico Granata                 std::string docstring;
14476f79bb2dSEnrico Granata                 m_fetched_help_short = scripter->GetShortHelpForCommandObject(m_cmd_obj_sp,docstring);
14486f79bb2dSEnrico Granata                 if (!docstring.empty())
1449*bfb75e9bSEnrico Granata                     SetHelp(docstring.c_str());
14506f79bb2dSEnrico Granata             }
14516f79bb2dSEnrico Granata         }
14526f79bb2dSEnrico Granata         return CommandObjectRaw::GetHelp();
14536f79bb2dSEnrico Granata     }
14546f79bb2dSEnrico Granata 
145513d21e9aSBruce Mitchener     const char *
145613d21e9aSBruce Mitchener     GetHelpLong () override
14579fe00e52SEnrico Granata     {
14586f79bb2dSEnrico Granata         if (!m_fetched_help_long)
14596f79bb2dSEnrico Granata         {
14606f79bb2dSEnrico Granata             ScriptInterpreter* scripter = m_interpreter.GetScriptInterpreter();
14616f79bb2dSEnrico Granata             if (scripter)
14626f79bb2dSEnrico Granata             {
14636f79bb2dSEnrico Granata                 std::string docstring;
14646f79bb2dSEnrico Granata                 m_fetched_help_long = scripter->GetLongHelpForCommandObject(m_cmd_obj_sp,docstring);
14656f79bb2dSEnrico Granata                 if (!docstring.empty())
1466*bfb75e9bSEnrico Granata                     SetHelpLong(docstring.c_str());
14676f79bb2dSEnrico Granata             }
14686f79bb2dSEnrico Granata         }
14699fe00e52SEnrico Granata         return CommandObjectRaw::GetHelpLong();
14709fe00e52SEnrico Granata     }
14719fe00e52SEnrico Granata 
14729fe00e52SEnrico Granata protected:
147313d21e9aSBruce Mitchener     bool
147413d21e9aSBruce Mitchener     DoExecute (const char *raw_command_line, CommandReturnObject &result) override
14759fe00e52SEnrico Granata     {
14769fe00e52SEnrico Granata         ScriptInterpreter* scripter = m_interpreter.GetScriptInterpreter();
14779fe00e52SEnrico Granata 
14789fe00e52SEnrico Granata         Error error;
14799fe00e52SEnrico Granata 
14809fe00e52SEnrico Granata         result.SetStatus(eReturnStatusInvalid);
14819fe00e52SEnrico Granata 
14826e3d8e7fSEugene Zelenko         if (!scripter || !scripter->RunScriptBasedCommand(m_cmd_obj_sp,
14839fe00e52SEnrico Granata                                                           raw_command_line,
14849fe00e52SEnrico Granata                                                           m_synchro,
14859fe00e52SEnrico Granata                                                           result,
14869fe00e52SEnrico Granata                                                           error,
14876e3d8e7fSEugene Zelenko                                                           m_exe_ctx))
14889fe00e52SEnrico Granata         {
14899fe00e52SEnrico Granata             result.AppendError(error.AsCString());
14909fe00e52SEnrico Granata             result.SetStatus(eReturnStatusFailed);
14919fe00e52SEnrico Granata         }
14929fe00e52SEnrico Granata         else
14939fe00e52SEnrico Granata         {
14949fe00e52SEnrico Granata             // Don't change the status if the command already set it...
14959fe00e52SEnrico Granata             if (result.GetStatus() == eReturnStatusInvalid)
14969fe00e52SEnrico Granata             {
14976e3d8e7fSEugene Zelenko                 if (result.GetOutputData() == nullptr || result.GetOutputData()[0] == '\0')
14989fe00e52SEnrico Granata                     result.SetStatus(eReturnStatusSuccessFinishNoResult);
14999fe00e52SEnrico Granata                 else
15009fe00e52SEnrico Granata                     result.SetStatus(eReturnStatusSuccessFinishResult);
15019fe00e52SEnrico Granata             }
15029fe00e52SEnrico Granata         }
15039fe00e52SEnrico Granata 
15049fe00e52SEnrico Granata         return result.Succeeded();
15059fe00e52SEnrico Granata     }
15069fe00e52SEnrico Granata 
15076e3d8e7fSEugene Zelenko private:
15086e3d8e7fSEugene Zelenko     StructuredData::GenericSP m_cmd_obj_sp;
15096e3d8e7fSEugene Zelenko     ScriptedCommandSynchronicity m_synchro;
15106e3d8e7fSEugene Zelenko     bool m_fetched_help_short: 1;
15116e3d8e7fSEugene Zelenko     bool m_fetched_help_long: 1;
15129fe00e52SEnrico Granata };
15139fe00e52SEnrico Granata 
1514a9dbf432SEnrico Granata //-------------------------------------------------------------------------
1515a9dbf432SEnrico Granata // CommandObjectCommandsScriptImport
1516a9dbf432SEnrico Granata //-------------------------------------------------------------------------
1517a9dbf432SEnrico Granata 
15185a988416SJim Ingham class CommandObjectCommandsScriptImport : public CommandObjectParsed
1519a9dbf432SEnrico Granata {
15205a988416SJim Ingham public:
15215a988416SJim Ingham     CommandObjectCommandsScriptImport (CommandInterpreter &interpreter) :
15225a988416SJim Ingham         CommandObjectParsed(interpreter,
15235a988416SJim Ingham                             "command script import",
15245a988416SJim Ingham                             "Import a scripting module in LLDB.",
15256e3d8e7fSEugene Zelenko                             nullptr),
15265a988416SJim Ingham         m_options(interpreter)
15275a988416SJim Ingham     {
15285a988416SJim Ingham         CommandArgumentEntry arg1;
15295a988416SJim Ingham         CommandArgumentData cmd_arg;
15305a988416SJim Ingham 
15315a988416SJim Ingham         // Define the first (and only) variant of this arg.
15325a988416SJim Ingham         cmd_arg.arg_type = eArgTypeFilename;
15333b00e35bSEnrico Granata         cmd_arg.arg_repetition = eArgRepeatPlus;
15345a988416SJim Ingham 
15355a988416SJim Ingham         // There is only one variant this argument could be; put it into the argument entry.
15365a988416SJim Ingham         arg1.push_back (cmd_arg);
15375a988416SJim Ingham 
15385a988416SJim Ingham         // Push the data for the first argument into the m_arguments vector.
15395a988416SJim Ingham         m_arguments.push_back (arg1);
15405a988416SJim Ingham     }
15415a988416SJim Ingham 
15426e3d8e7fSEugene Zelenko     ~CommandObjectCommandsScriptImport() override = default;
15435a988416SJim Ingham 
154413d21e9aSBruce Mitchener     int
15455a988416SJim Ingham     HandleArgumentCompletion (Args &input,
15465a988416SJim Ingham                               int &cursor_index,
15475a988416SJim Ingham                               int &cursor_char_position,
15485a988416SJim Ingham                               OptionElementVector &opt_element_vector,
15495a988416SJim Ingham                               int match_start_point,
15505a988416SJim Ingham                               int max_return_elements,
15515a988416SJim Ingham                               bool &word_complete,
155213d21e9aSBruce Mitchener                               StringList &matches) override
15535a988416SJim Ingham     {
15545a988416SJim Ingham         std::string completion_str (input.GetArgumentAtIndex(cursor_index));
15555a988416SJim Ingham         completion_str.erase (cursor_char_position);
15565a988416SJim Ingham 
15575a988416SJim Ingham         CommandCompletions::InvokeCommonCompletionCallbacks(m_interpreter,
15585a988416SJim Ingham                                                             CommandCompletions::eDiskFileCompletion,
15595a988416SJim Ingham                                                             completion_str.c_str(),
15605a988416SJim Ingham                                                             match_start_point,
15615a988416SJim Ingham                                                             max_return_elements,
15626e3d8e7fSEugene Zelenko                                                             nullptr,
15635a988416SJim Ingham                                                             word_complete,
15645a988416SJim Ingham                                                             matches);
15655a988416SJim Ingham         return matches.GetSize();
15665a988416SJim Ingham     }
15675a988416SJim Ingham 
156813d21e9aSBruce Mitchener     Options *
156913d21e9aSBruce Mitchener     GetOptions () override
15705a988416SJim Ingham     {
15715a988416SJim Ingham         return &m_options;
15725a988416SJim Ingham     }
15735a988416SJim Ingham 
15745a988416SJim Ingham protected:
15750a305db7SEnrico Granata     class CommandOptions : public Options
15760a305db7SEnrico Granata     {
15770a305db7SEnrico Granata     public:
15780a305db7SEnrico Granata         CommandOptions (CommandInterpreter &interpreter) :
15790a305db7SEnrico Granata             Options (interpreter)
15800a305db7SEnrico Granata         {
15810a305db7SEnrico Granata         }
15820a305db7SEnrico Granata 
15836e3d8e7fSEugene Zelenko         ~CommandOptions() override = default;
15840a305db7SEnrico Granata 
158513d21e9aSBruce Mitchener         Error
158613d21e9aSBruce Mitchener         SetOptionValue (uint32_t option_idx, const char *option_arg) override
15870a305db7SEnrico Granata         {
15880a305db7SEnrico Granata             Error error;
15893bcdfc0eSGreg Clayton             const int short_option = m_getopt_table[option_idx].val;
15900a305db7SEnrico Granata 
15910a305db7SEnrico Granata             switch (short_option)
15920a305db7SEnrico Granata             {
15930a305db7SEnrico Granata                 case 'r':
15940a305db7SEnrico Granata                     m_allow_reload = true;
15950a305db7SEnrico Granata                     break;
15960a305db7SEnrico Granata                 default:
15970a305db7SEnrico Granata                     error.SetErrorStringWithFormat ("unrecognized option '%c'", short_option);
15980a305db7SEnrico Granata                     break;
15990a305db7SEnrico Granata             }
16000a305db7SEnrico Granata 
16010a305db7SEnrico Granata             return error;
16020a305db7SEnrico Granata         }
16030a305db7SEnrico Granata 
16040a305db7SEnrico Granata         void
160513d21e9aSBruce Mitchener         OptionParsingStarting () override
16060a305db7SEnrico Granata         {
1607e0c70f1bSEnrico Granata             m_allow_reload = true;
16080a305db7SEnrico Granata         }
16090a305db7SEnrico Granata 
16100a305db7SEnrico Granata         const OptionDefinition*
161113d21e9aSBruce Mitchener         GetDefinitions () override
16120a305db7SEnrico Granata         {
16130a305db7SEnrico Granata             return g_option_table;
16140a305db7SEnrico Granata         }
16150a305db7SEnrico Granata 
16160a305db7SEnrico Granata         // Options table: Required for subclasses of Options.
16170a305db7SEnrico Granata 
16180a305db7SEnrico Granata         static OptionDefinition g_option_table[];
16190a305db7SEnrico Granata 
16200a305db7SEnrico Granata         // Instance variables to hold the values for command options.
16210a305db7SEnrico Granata 
16220a305db7SEnrico Granata         bool m_allow_reload;
16230a305db7SEnrico Granata     };
16240a305db7SEnrico Granata 
1625a9dbf432SEnrico Granata     bool
162613d21e9aSBruce Mitchener     DoExecute (Args& command, CommandReturnObject &result) override
1627a9dbf432SEnrico Granata     {
1628a9dbf432SEnrico Granata         if (m_interpreter.GetDebugger().GetScriptLanguage() != lldb::eScriptLanguagePython)
1629a9dbf432SEnrico Granata         {
1630a9dbf432SEnrico Granata             result.AppendError ("only scripting language supported for module importing is currently Python");
1631a9dbf432SEnrico Granata             result.SetStatus (eReturnStatusFailed);
1632a9dbf432SEnrico Granata             return false;
1633a9dbf432SEnrico Granata         }
1634a9dbf432SEnrico Granata 
16355a988416SJim Ingham         size_t argc = command.GetArgumentCount();
16363b00e35bSEnrico Granata         if (0 == argc)
1637a9dbf432SEnrico Granata         {
16383b00e35bSEnrico Granata             result.AppendError("command script import needs one or more arguments");
1639a9dbf432SEnrico Granata             result.SetStatus (eReturnStatusFailed);
1640a9dbf432SEnrico Granata             return false;
1641a9dbf432SEnrico Granata         }
1642a9dbf432SEnrico Granata 
16430e978481SEd Maste         for (size_t i = 0;
16443b00e35bSEnrico Granata              i < argc;
16453b00e35bSEnrico Granata              i++)
16463b00e35bSEnrico Granata         {
16473b00e35bSEnrico Granata             std::string path = command.GetArgumentAtIndex(i);
1648a9dbf432SEnrico Granata             Error error;
1649a9dbf432SEnrico Granata 
1650c9d645d3SGreg Clayton             const bool init_session = true;
1651078551c7SEnrico Granata             // FIXME: this is necessary because CommandObject::CheckRequirements() assumes that
1652078551c7SEnrico Granata             // commands won't ever be recursively invoked, but it's actually possible to craft
1653078551c7SEnrico Granata             // a Python script that does other "command script imports" in __lldb_init_module
1654078551c7SEnrico Granata             // the real fix is to have recursive commands possible with a CommandInvocation object
1655078551c7SEnrico Granata             // separate from the CommandObject itself, so that recursive command invocations
1656078551c7SEnrico Granata             // won't stomp on each other (wrt to execution contents, options, and more)
1657078551c7SEnrico Granata             m_exe_ctx.Clear();
1658a9dbf432SEnrico Granata             if (m_interpreter.GetScriptInterpreter()->LoadScriptingModule(path.c_str(),
16590a305db7SEnrico Granata                                                                           m_options.m_allow_reload,
1660c9d645d3SGreg Clayton                                                                           init_session,
1661a9dbf432SEnrico Granata                                                                           error))
1662a9dbf432SEnrico Granata             {
1663a9dbf432SEnrico Granata                 result.SetStatus (eReturnStatusSuccessFinishNoResult);
1664a9dbf432SEnrico Granata             }
1665a9dbf432SEnrico Granata             else
1666a9dbf432SEnrico Granata             {
1667a9dbf432SEnrico Granata                 result.AppendErrorWithFormat("module importing failed: %s", error.AsCString());
1668a9dbf432SEnrico Granata                 result.SetStatus (eReturnStatusFailed);
1669a9dbf432SEnrico Granata             }
16703b00e35bSEnrico Granata         }
1671a9dbf432SEnrico Granata 
1672a9dbf432SEnrico Granata         return result.Succeeded();
1673a9dbf432SEnrico Granata     }
16740a305db7SEnrico Granata 
16755a988416SJim Ingham     CommandOptions m_options;
1676a9dbf432SEnrico Granata };
1677223383edSEnrico Granata 
16780a305db7SEnrico Granata OptionDefinition
16790a305db7SEnrico Granata CommandObjectCommandsScriptImport::CommandOptions::g_option_table[] =
16800a305db7SEnrico Granata {
16816e3d8e7fSEugene Zelenko     { LLDB_OPT_SET_1, false, "allow-reload", 'r', OptionParser::eNoArgument, nullptr, nullptr, 0, eArgTypeNone,        "Allow the script to be loaded even if it was already loaded before. This argument exists for backwards compatibility, but reloading is always allowed, whether you specify it or not."},
16826e3d8e7fSEugene Zelenko     { 0, false, nullptr, 0, 0, nullptr, nullptr, 0, eArgTypeNone, nullptr }
16830a305db7SEnrico Granata };
16840a305db7SEnrico Granata 
1685223383edSEnrico Granata //-------------------------------------------------------------------------
1686223383edSEnrico Granata // CommandObjectCommandsScriptAdd
1687223383edSEnrico Granata //-------------------------------------------------------------------------
1688223383edSEnrico Granata 
168944d93782SGreg Clayton class CommandObjectCommandsScriptAdd :
169044d93782SGreg Clayton     public CommandObjectParsed,
169144d93782SGreg Clayton     public IOHandlerDelegateMultiline
1692223383edSEnrico Granata {
16935a988416SJim Ingham public:
16945a988416SJim Ingham     CommandObjectCommandsScriptAdd(CommandInterpreter &interpreter) :
16955a988416SJim Ingham         CommandObjectParsed(interpreter,
16965a988416SJim Ingham                             "command script add",
16975a988416SJim Ingham                             "Add a scripted function as an LLDB command.",
16986e3d8e7fSEugene Zelenko                             nullptr),
1699c3d874a5SGreg Clayton         IOHandlerDelegateMultiline ("DONE"),
17005a988416SJim Ingham         m_options (interpreter)
17015a988416SJim Ingham     {
17025a988416SJim Ingham         CommandArgumentEntry arg1;
17035a988416SJim Ingham         CommandArgumentData cmd_arg;
17045a988416SJim Ingham 
17055a988416SJim Ingham         // Define the first (and only) variant of this arg.
17065a988416SJim Ingham         cmd_arg.arg_type = eArgTypeCommandName;
17075a988416SJim Ingham         cmd_arg.arg_repetition = eArgRepeatPlain;
17085a988416SJim Ingham 
17095a988416SJim Ingham         // There is only one variant this argument could be; put it into the argument entry.
17105a988416SJim Ingham         arg1.push_back (cmd_arg);
17115a988416SJim Ingham 
17125a988416SJim Ingham         // Push the data for the first argument into the m_arguments vector.
17135a988416SJim Ingham         m_arguments.push_back (arg1);
17145a988416SJim Ingham     }
17155a988416SJim Ingham 
17166e3d8e7fSEugene Zelenko     ~CommandObjectCommandsScriptAdd() override = default;
17175a988416SJim Ingham 
171813d21e9aSBruce Mitchener     Options *
171913d21e9aSBruce Mitchener     GetOptions () override
17205a988416SJim Ingham     {
17215a988416SJim Ingham         return &m_options;
17225a988416SJim Ingham     }
17235a988416SJim Ingham 
17245a988416SJim Ingham protected:
1725223383edSEnrico Granata     class CommandOptions : public Options
1726223383edSEnrico Granata     {
1727223383edSEnrico Granata     public:
1728223383edSEnrico Granata         CommandOptions (CommandInterpreter &interpreter) :
17299fe00e52SEnrico Granata             Options (interpreter),
17309fe00e52SEnrico Granata             m_class_name(),
17319fe00e52SEnrico Granata             m_funct_name(),
17329fe00e52SEnrico Granata             m_short_help(),
17339fe00e52SEnrico Granata             m_synchronicity(eScriptedCommandSynchronicitySynchronous)
1734223383edSEnrico Granata         {
1735223383edSEnrico Granata         }
1736223383edSEnrico Granata 
17376e3d8e7fSEugene Zelenko         ~CommandOptions() override = default;
1738223383edSEnrico Granata 
173913d21e9aSBruce Mitchener         Error
174013d21e9aSBruce Mitchener         SetOptionValue (uint32_t option_idx, const char *option_arg) override
1741223383edSEnrico Granata         {
1742223383edSEnrico Granata             Error error;
17433bcdfc0eSGreg Clayton             const int short_option = m_getopt_table[option_idx].val;
1744223383edSEnrico Granata 
1745223383edSEnrico Granata             switch (short_option)
1746223383edSEnrico Granata             {
1747223383edSEnrico Granata                 case 'f':
1748735152e3SEnrico Granata                     if (option_arg)
1749735152e3SEnrico Granata                         m_funct_name.assign(option_arg);
1750735152e3SEnrico Granata                     break;
17519fe00e52SEnrico Granata                 case 'c':
17529fe00e52SEnrico Granata                     if (option_arg)
17539fe00e52SEnrico Granata                         m_class_name.assign(option_arg);
17549fe00e52SEnrico Granata                     break;
1755735152e3SEnrico Granata                 case 'h':
1756735152e3SEnrico Granata                     if (option_arg)
1757735152e3SEnrico Granata                         m_short_help.assign(option_arg);
1758223383edSEnrico Granata                     break;
17590a305db7SEnrico Granata                 case 's':
176044d93782SGreg Clayton                     m_synchronicity = (ScriptedCommandSynchronicity) Args::StringToOptionEnum(option_arg, g_option_table[option_idx].enum_values, 0, error);
17610a305db7SEnrico Granata                     if (!error.Success())
17620a305db7SEnrico Granata                         error.SetErrorStringWithFormat ("unrecognized value for synchronicity '%s'", option_arg);
17630a305db7SEnrico Granata                     break;
1764223383edSEnrico Granata                 default:
176586edbf41SGreg Clayton                     error.SetErrorStringWithFormat ("unrecognized option '%c'", short_option);
1766223383edSEnrico Granata                     break;
1767223383edSEnrico Granata             }
1768223383edSEnrico Granata 
1769223383edSEnrico Granata             return error;
1770223383edSEnrico Granata         }
1771223383edSEnrico Granata 
1772223383edSEnrico Granata         void
177313d21e9aSBruce Mitchener         OptionParsingStarting () override
1774223383edSEnrico Granata         {
17759fe00e52SEnrico Granata             m_class_name.clear();
1776735152e3SEnrico Granata             m_funct_name.clear();
1777735152e3SEnrico Granata             m_short_help.clear();
177844d93782SGreg Clayton             m_synchronicity = eScriptedCommandSynchronicitySynchronous;
1779223383edSEnrico Granata         }
1780223383edSEnrico Granata 
1781223383edSEnrico Granata         const OptionDefinition*
178213d21e9aSBruce Mitchener         GetDefinitions () override
1783223383edSEnrico Granata         {
1784223383edSEnrico Granata             return g_option_table;
1785223383edSEnrico Granata         }
1786223383edSEnrico Granata 
1787223383edSEnrico Granata         // Options table: Required for subclasses of Options.
1788223383edSEnrico Granata 
1789223383edSEnrico Granata         static OptionDefinition g_option_table[];
1790223383edSEnrico Granata 
1791223383edSEnrico Granata         // Instance variables to hold the values for command options.
1792223383edSEnrico Granata 
17939fe00e52SEnrico Granata         std::string m_class_name;
1794223383edSEnrico Granata         std::string m_funct_name;
1795735152e3SEnrico Granata         std::string m_short_help;
179644d93782SGreg Clayton         ScriptedCommandSynchronicity m_synchronicity;
1797223383edSEnrico Granata     };
1798223383edSEnrico Granata 
179913d21e9aSBruce Mitchener     void
180013d21e9aSBruce Mitchener     IOHandlerActivated (IOHandler &io_handler) override
1801223383edSEnrico Granata     {
180244d93782SGreg Clayton         StreamFileSP output_sp(io_handler.GetOutputStreamFile());
180344d93782SGreg Clayton         if (output_sp)
1804223383edSEnrico Granata         {
180544d93782SGreg Clayton             output_sp->PutCString(g_python_command_instructions);
180644d93782SGreg Clayton             output_sp->Flush();
1807223383edSEnrico Granata         }
1808223383edSEnrico Granata     }
1809223383edSEnrico Granata 
1810223383edSEnrico Granata 
181113d21e9aSBruce Mitchener     void
181213d21e9aSBruce Mitchener     IOHandlerInputComplete (IOHandler &io_handler, std::string &data) override
1813223383edSEnrico Granata     {
181444d93782SGreg Clayton         StreamFileSP error_sp = io_handler.GetErrorStreamFile();
181544d93782SGreg Clayton 
181644d93782SGreg Clayton         ScriptInterpreter *interpreter = m_interpreter.GetScriptInterpreter();
181744d93782SGreg Clayton         if (interpreter)
181844d93782SGreg Clayton         {
181944d93782SGreg Clayton 
182044d93782SGreg Clayton             StringList lines;
182144d93782SGreg Clayton             lines.SplitIntoLines(data);
182244d93782SGreg Clayton             if (lines.GetSize() > 0)
182344d93782SGreg Clayton             {
1824a73b7df7SEnrico Granata                 std::string funct_name_str;
182544d93782SGreg Clayton                 if (interpreter->GenerateScriptAliasFunction (lines, funct_name_str))
1826223383edSEnrico Granata                 {
1827a73b7df7SEnrico Granata                     if (funct_name_str.empty())
1828223383edSEnrico Granata                     {
182944d93782SGreg Clayton                         error_sp->Printf ("error: unable to obtain a function name, didn't add python command.\n");
183044d93782SGreg Clayton                         error_sp->Flush();
1831223383edSEnrico Granata                     }
183244d93782SGreg Clayton                     else
183344d93782SGreg Clayton                     {
1834223383edSEnrico Granata                         // everything should be fine now, let's add this alias
1835223383edSEnrico Granata 
1836223383edSEnrico Granata                         CommandObjectSP command_obj_sp(new CommandObjectPythonFunction (m_interpreter,
1837223383edSEnrico Granata                                                                                         m_cmd_name,
1838a73b7df7SEnrico Granata                                                                                         funct_name_str.c_str(),
1839735152e3SEnrico Granata                                                                                         m_short_help,
184044d93782SGreg Clayton                                                                                         m_synchronicity));
1841223383edSEnrico Granata 
18420a305db7SEnrico Granata                         if (!m_interpreter.AddUserCommand(m_cmd_name, command_obj_sp, true))
1843223383edSEnrico Granata                         {
184444d93782SGreg Clayton                             error_sp->Printf ("error: unable to add selected command, didn't add python command.\n");
184544d93782SGreg Clayton                             error_sp->Flush();
1846223383edSEnrico Granata                         }
1847223383edSEnrico Granata                     }
184844d93782SGreg Clayton                 }
184944d93782SGreg Clayton                 else
185044d93782SGreg Clayton                 {
185144d93782SGreg Clayton                     error_sp->Printf ("error: unable to create function, didn't add python command.\n");
185244d93782SGreg Clayton                     error_sp->Flush();
185344d93782SGreg Clayton                 }
185444d93782SGreg Clayton             }
185544d93782SGreg Clayton             else
185644d93782SGreg Clayton             {
185744d93782SGreg Clayton                 error_sp->Printf ("error: empty function, didn't add python command.\n");
185844d93782SGreg Clayton                 error_sp->Flush();
185944d93782SGreg Clayton             }
186044d93782SGreg Clayton         }
186144d93782SGreg Clayton         else
186244d93782SGreg Clayton         {
186344d93782SGreg Clayton             error_sp->Printf ("error: script interpreter missing, didn't add python command.\n");
186444d93782SGreg Clayton             error_sp->Flush();
186544d93782SGreg Clayton         }
186644d93782SGreg Clayton 
186744d93782SGreg Clayton         io_handler.SetIsDone(true);
186844d93782SGreg Clayton     }
1869223383edSEnrico Granata 
18705a988416SJim Ingham protected:
1871223383edSEnrico Granata     bool
187213d21e9aSBruce Mitchener     DoExecute (Args& command, CommandReturnObject &result) override
1873223383edSEnrico Granata     {
187499f0b8f9SEnrico Granata         if (m_interpreter.GetDebugger().GetScriptLanguage() != lldb::eScriptLanguagePython)
187599f0b8f9SEnrico Granata         {
187699f0b8f9SEnrico Granata             result.AppendError ("only scripting language supported for scripted commands is currently Python");
187799f0b8f9SEnrico Granata             result.SetStatus (eReturnStatusFailed);
187899f0b8f9SEnrico Granata             return false;
187999f0b8f9SEnrico Granata         }
188099f0b8f9SEnrico Granata 
18815a988416SJim Ingham         size_t argc = command.GetArgumentCount();
1882223383edSEnrico Granata 
1883223383edSEnrico Granata         if (argc != 1)
1884223383edSEnrico Granata         {
1885223383edSEnrico Granata             result.AppendError ("'command script add' requires one argument");
1886223383edSEnrico Granata             result.SetStatus (eReturnStatusFailed);
1887223383edSEnrico Granata             return false;
1888223383edSEnrico Granata         }
1889223383edSEnrico Granata 
1890735152e3SEnrico Granata         // Store the options in case we get multi-line input
189144d93782SGreg Clayton         m_cmd_name = command.GetArgumentAtIndex(0);
1892735152e3SEnrico Granata         m_short_help.assign(m_options.m_short_help);
189344d93782SGreg Clayton         m_synchronicity = m_options.m_synchronicity;
1894223383edSEnrico Granata 
18959fe00e52SEnrico Granata         if (m_options.m_class_name.empty())
18969fe00e52SEnrico Granata         {
1897223383edSEnrico Granata             if (m_options.m_funct_name.empty())
1898223383edSEnrico Granata             {
189944d93782SGreg Clayton                 m_interpreter.GetPythonCommandsFromIOHandler("     ",  // Prompt
190044d93782SGreg Clayton                                                              *this,    // IOHandlerDelegate
190144d93782SGreg Clayton                                                              true,     // Run IOHandler in async mode
19026e3d8e7fSEugene Zelenko                                                              nullptr); // Baton for the "io_handler" that will be passed back into our IOHandlerDelegate functions
1903223383edSEnrico Granata             }
1904223383edSEnrico Granata             else
1905223383edSEnrico Granata             {
19060a305db7SEnrico Granata                 CommandObjectSP new_cmd(new CommandObjectPythonFunction(m_interpreter,
190744d93782SGreg Clayton                                                                         m_cmd_name,
19080a305db7SEnrico Granata                                                                         m_options.m_funct_name,
1909735152e3SEnrico Granata                                                                         m_options.m_short_help,
191044d93782SGreg Clayton                                                                         m_synchronicity));
191144d93782SGreg Clayton                 if (m_interpreter.AddUserCommand(m_cmd_name, new_cmd, true))
1912223383edSEnrico Granata                 {
1913223383edSEnrico Granata                     result.SetStatus (eReturnStatusSuccessFinishNoResult);
1914223383edSEnrico Granata                 }
1915223383edSEnrico Granata                 else
1916223383edSEnrico Granata                 {
1917223383edSEnrico Granata                     result.AppendError("cannot add command");
1918223383edSEnrico Granata                     result.SetStatus (eReturnStatusFailed);
1919223383edSEnrico Granata                 }
1920223383edSEnrico Granata             }
19219fe00e52SEnrico Granata         }
19229fe00e52SEnrico Granata         else
19239fe00e52SEnrico Granata         {
19249fe00e52SEnrico Granata             ScriptInterpreter *interpreter = GetCommandInterpreter().GetScriptInterpreter();
19259fe00e52SEnrico Granata             if (!interpreter)
19269fe00e52SEnrico Granata             {
19279fe00e52SEnrico Granata                 result.AppendError("cannot find ScriptInterpreter");
19289fe00e52SEnrico Granata                 result.SetStatus(eReturnStatusFailed);
19299fe00e52SEnrico Granata                 return false;
19309fe00e52SEnrico Granata             }
19319fe00e52SEnrico Granata 
19329fe00e52SEnrico Granata             auto cmd_obj_sp = interpreter->CreateScriptCommandObject(m_options.m_class_name.c_str());
19339fe00e52SEnrico Granata             if (!cmd_obj_sp)
19349fe00e52SEnrico Granata             {
19359fe00e52SEnrico Granata                 result.AppendError("cannot create helper object");
19369fe00e52SEnrico Granata                 result.SetStatus(eReturnStatusFailed);
19379fe00e52SEnrico Granata                 return false;
19389fe00e52SEnrico Granata             }
19399fe00e52SEnrico Granata 
19409fe00e52SEnrico Granata             CommandObjectSP new_cmd(new CommandObjectScriptingObject(m_interpreter,
19419fe00e52SEnrico Granata                                                                      m_cmd_name,
19429fe00e52SEnrico Granata                                                                      cmd_obj_sp,
19439fe00e52SEnrico Granata                                                                      m_synchronicity));
19449fe00e52SEnrico Granata             if (m_interpreter.AddUserCommand(m_cmd_name, new_cmd, true))
19459fe00e52SEnrico Granata             {
19469fe00e52SEnrico Granata                 result.SetStatus (eReturnStatusSuccessFinishNoResult);
19479fe00e52SEnrico Granata             }
19489fe00e52SEnrico Granata             else
19499fe00e52SEnrico Granata             {
19509fe00e52SEnrico Granata                 result.AppendError("cannot add command");
19519fe00e52SEnrico Granata                 result.SetStatus (eReturnStatusFailed);
19529fe00e52SEnrico Granata             }
19539fe00e52SEnrico Granata         }
1954223383edSEnrico Granata 
1955223383edSEnrico Granata         return result.Succeeded();
1956223383edSEnrico Granata     }
19575a988416SJim Ingham 
19585a988416SJim Ingham     CommandOptions m_options;
195944d93782SGreg Clayton     std::string m_cmd_name;
1960735152e3SEnrico Granata     std::string m_short_help;
196144d93782SGreg Clayton     ScriptedCommandSynchronicity m_synchronicity;
1962223383edSEnrico Granata };
1963223383edSEnrico Granata 
19640a305db7SEnrico Granata static OptionEnumValueElement g_script_synchro_type[] =
19650a305db7SEnrico Granata {
19660a305db7SEnrico Granata     { eScriptedCommandSynchronicitySynchronous,      "synchronous",       "Run synchronous"},
19670a305db7SEnrico Granata     { eScriptedCommandSynchronicityAsynchronous,     "asynchronous",      "Run asynchronous"},
19680a305db7SEnrico Granata     { eScriptedCommandSynchronicityCurrentValue,     "current",           "Do not alter current setting"},
19696e3d8e7fSEugene Zelenko     { 0, nullptr, nullptr }
19700a305db7SEnrico Granata };
19710a305db7SEnrico Granata 
1972223383edSEnrico Granata OptionDefinition
1973223383edSEnrico Granata CommandObjectCommandsScriptAdd::CommandOptions::g_option_table[] =
1974223383edSEnrico Granata {
19756e3d8e7fSEugene Zelenko     { LLDB_OPT_SET_1, false, "function", 'f', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypePythonFunction,        "Name of the Python function to bind to this command name."},
19766e3d8e7fSEugene Zelenko     { LLDB_OPT_SET_2, false, "class", 'c', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypePythonClass,        "Name of the Python class to bind to this command name."},
19776e3d8e7fSEugene Zelenko     { LLDB_OPT_SET_1, false, "help"  , 'h', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeHelpText, "The help text to display for this command."},
19786e3d8e7fSEugene Zelenko     { LLDB_OPT_SET_ALL, false, "synchronicity", 's', OptionParser::eRequiredArgument, nullptr, g_script_synchro_type, 0, eArgTypeScriptedCommandSynchronicity,        "Set the synchronicity of this command's executions with regard to LLDB event system."},
19796e3d8e7fSEugene Zelenko     { 0, false, nullptr, 0, 0, nullptr, nullptr, 0, eArgTypeNone, nullptr }
1980223383edSEnrico Granata };
1981223383edSEnrico Granata 
1982223383edSEnrico Granata //-------------------------------------------------------------------------
1983223383edSEnrico Granata // CommandObjectCommandsScriptList
1984223383edSEnrico Granata //-------------------------------------------------------------------------
1985223383edSEnrico Granata 
19865a988416SJim Ingham class CommandObjectCommandsScriptList : public CommandObjectParsed
1987223383edSEnrico Granata {
1988223383edSEnrico Granata public:
1989223383edSEnrico Granata     CommandObjectCommandsScriptList(CommandInterpreter &interpreter) :
19905a988416SJim Ingham         CommandObjectParsed(interpreter,
1991223383edSEnrico Granata                             "command script list",
1992223383edSEnrico Granata                             "List defined scripted commands.",
19936e3d8e7fSEugene Zelenko                             nullptr)
1994223383edSEnrico Granata     {
1995223383edSEnrico Granata     }
1996223383edSEnrico Granata 
19976e3d8e7fSEugene Zelenko     ~CommandObjectCommandsScriptList() override = default;
1998223383edSEnrico Granata 
1999223383edSEnrico Granata     bool
200013d21e9aSBruce Mitchener     DoExecute (Args& command, CommandReturnObject &result) override
2001223383edSEnrico Granata     {
2002223383edSEnrico Granata         m_interpreter.GetHelp(result,
2003223383edSEnrico Granata                               CommandInterpreter::eCommandTypesUserDef);
2004223383edSEnrico Granata 
2005223383edSEnrico Granata         result.SetStatus (eReturnStatusSuccessFinishResult);
2006223383edSEnrico Granata 
2007223383edSEnrico Granata         return true;
2008223383edSEnrico Granata     }
2009223383edSEnrico Granata };
2010223383edSEnrico Granata 
2011223383edSEnrico Granata //-------------------------------------------------------------------------
2012223383edSEnrico Granata // CommandObjectCommandsScriptClear
2013223383edSEnrico Granata //-------------------------------------------------------------------------
2014223383edSEnrico Granata 
20155a988416SJim Ingham class CommandObjectCommandsScriptClear : public CommandObjectParsed
2016223383edSEnrico Granata {
2017223383edSEnrico Granata public:
2018223383edSEnrico Granata     CommandObjectCommandsScriptClear(CommandInterpreter &interpreter) :
20195a988416SJim Ingham         CommandObjectParsed(interpreter,
2020223383edSEnrico Granata                             "command script clear",
2021223383edSEnrico Granata                             "Delete all scripted commands.",
20226e3d8e7fSEugene Zelenko                             nullptr)
2023223383edSEnrico Granata     {
2024223383edSEnrico Granata     }
2025223383edSEnrico Granata 
20266e3d8e7fSEugene Zelenko     ~CommandObjectCommandsScriptClear() override = default;
2027223383edSEnrico Granata 
20285a988416SJim Ingham protected:
2029223383edSEnrico Granata     bool
203013d21e9aSBruce Mitchener     DoExecute (Args& command, CommandReturnObject &result) override
2031223383edSEnrico Granata     {
2032223383edSEnrico Granata         m_interpreter.RemoveAllUser();
2033223383edSEnrico Granata 
2034223383edSEnrico Granata         result.SetStatus (eReturnStatusSuccessFinishResult);
2035223383edSEnrico Granata 
2036223383edSEnrico Granata         return true;
2037223383edSEnrico Granata     }
2038223383edSEnrico Granata };
2039223383edSEnrico Granata 
2040223383edSEnrico Granata //-------------------------------------------------------------------------
2041223383edSEnrico Granata // CommandObjectCommandsScriptDelete
2042223383edSEnrico Granata //-------------------------------------------------------------------------
2043223383edSEnrico Granata 
20445a988416SJim Ingham class CommandObjectCommandsScriptDelete : public CommandObjectParsed
2045223383edSEnrico Granata {
2046223383edSEnrico Granata public:
2047223383edSEnrico Granata     CommandObjectCommandsScriptDelete(CommandInterpreter &interpreter) :
20485a988416SJim Ingham         CommandObjectParsed(interpreter,
2049223383edSEnrico Granata                             "command script delete",
2050223383edSEnrico Granata                             "Delete a scripted command.",
20516e3d8e7fSEugene Zelenko                             nullptr)
2052223383edSEnrico Granata     {
2053223383edSEnrico Granata         CommandArgumentEntry arg1;
2054223383edSEnrico Granata         CommandArgumentData cmd_arg;
2055223383edSEnrico Granata 
2056223383edSEnrico Granata         // Define the first (and only) variant of this arg.
2057223383edSEnrico Granata         cmd_arg.arg_type = eArgTypeCommandName;
2058223383edSEnrico Granata         cmd_arg.arg_repetition = eArgRepeatPlain;
2059223383edSEnrico Granata 
2060223383edSEnrico Granata         // There is only one variant this argument could be; put it into the argument entry.
2061223383edSEnrico Granata         arg1.push_back (cmd_arg);
2062223383edSEnrico Granata 
2063223383edSEnrico Granata         // Push the data for the first argument into the m_arguments vector.
2064223383edSEnrico Granata         m_arguments.push_back (arg1);
2065223383edSEnrico Granata     }
2066223383edSEnrico Granata 
20676e3d8e7fSEugene Zelenko     ~CommandObjectCommandsScriptDelete() override = default;
2068223383edSEnrico Granata 
20695a988416SJim Ingham protected:
2070223383edSEnrico Granata     bool
207113d21e9aSBruce Mitchener     DoExecute (Args& command, CommandReturnObject &result) override
2072223383edSEnrico Granata     {
2073223383edSEnrico Granata 
20745a988416SJim Ingham         size_t argc = command.GetArgumentCount();
2075223383edSEnrico Granata 
2076223383edSEnrico Granata         if (argc != 1)
2077223383edSEnrico Granata         {
2078223383edSEnrico Granata             result.AppendError ("'command script delete' requires one argument");
2079223383edSEnrico Granata             result.SetStatus (eReturnStatusFailed);
2080223383edSEnrico Granata             return false;
2081223383edSEnrico Granata         }
2082223383edSEnrico Granata 
20835a988416SJim Ingham         const char* cmd_name = command.GetArgumentAtIndex(0);
2084223383edSEnrico Granata 
2085223383edSEnrico Granata         if (cmd_name && *cmd_name && m_interpreter.HasUserCommands() && m_interpreter.UserCommandExists(cmd_name))
2086223383edSEnrico Granata         {
2087223383edSEnrico Granata             m_interpreter.RemoveUser(cmd_name);
2088223383edSEnrico Granata             result.SetStatus (eReturnStatusSuccessFinishResult);
2089223383edSEnrico Granata         }
2090223383edSEnrico Granata         else
2091223383edSEnrico Granata         {
2092223383edSEnrico Granata             result.AppendErrorWithFormat ("command %s not found", cmd_name);
2093223383edSEnrico Granata             result.SetStatus (eReturnStatusFailed);
2094223383edSEnrico Granata         }
2095223383edSEnrico Granata 
2096223383edSEnrico Granata         return result.Succeeded();
2097223383edSEnrico Granata     }
2098223383edSEnrico Granata };
2099223383edSEnrico Granata 
2100223383edSEnrico Granata #pragma mark CommandObjectMultiwordCommandsScript
2101223383edSEnrico Granata 
2102223383edSEnrico Granata //-------------------------------------------------------------------------
2103223383edSEnrico Granata // CommandObjectMultiwordCommandsScript
2104223383edSEnrico Granata //-------------------------------------------------------------------------
2105223383edSEnrico Granata 
2106223383edSEnrico Granata class CommandObjectMultiwordCommandsScript : public CommandObjectMultiword
2107223383edSEnrico Granata {
2108223383edSEnrico Granata public:
2109223383edSEnrico Granata     CommandObjectMultiwordCommandsScript (CommandInterpreter &interpreter) :
2110223383edSEnrico Granata     CommandObjectMultiword (interpreter,
2111223383edSEnrico Granata                             "command script",
2112223383edSEnrico Granata                             "A set of commands for managing or customizing script commands.",
2113223383edSEnrico Granata                             "command script <subcommand> [<subcommand-options>]")
2114223383edSEnrico Granata     {
2115223383edSEnrico Granata         LoadSubCommand ("add",    CommandObjectSP (new CommandObjectCommandsScriptAdd (interpreter)));
2116223383edSEnrico Granata         LoadSubCommand ("delete", CommandObjectSP (new CommandObjectCommandsScriptDelete (interpreter)));
2117223383edSEnrico Granata         LoadSubCommand ("clear",  CommandObjectSP (new CommandObjectCommandsScriptClear (interpreter)));
2118223383edSEnrico Granata         LoadSubCommand ("list",   CommandObjectSP (new CommandObjectCommandsScriptList (interpreter)));
2119a9dbf432SEnrico Granata         LoadSubCommand ("import", CommandObjectSP (new CommandObjectCommandsScriptImport (interpreter)));
2120223383edSEnrico Granata     }
2121223383edSEnrico Granata 
21226e3d8e7fSEugene Zelenko     ~CommandObjectMultiwordCommandsScript() override = default;
2123223383edSEnrico Granata };
2124223383edSEnrico Granata 
2125ebc09c36SJim Ingham #pragma mark CommandObjectMultiwordCommands
2126ebc09c36SJim Ingham 
2127ebc09c36SJim Ingham //-------------------------------------------------------------------------
2128ebc09c36SJim Ingham // CommandObjectMultiwordCommands
2129ebc09c36SJim Ingham //-------------------------------------------------------------------------
2130ebc09c36SJim Ingham 
2131ebc09c36SJim Ingham CommandObjectMultiwordCommands::CommandObjectMultiwordCommands (CommandInterpreter &interpreter) :
2132a7015092SGreg Clayton     CommandObjectMultiword (interpreter,
21330e5e5a79SGreg Clayton                             "command",
21343f4c09c1SCaroline Tice                             "A set of commands for managing or customizing the debugger commands.",
21350e5e5a79SGreg Clayton                             "command <subcommand> [<subcommand-options>]")
2136ebc09c36SJim Ingham {
2137a7015092SGreg Clayton     LoadSubCommand ("source",  CommandObjectSP (new CommandObjectCommandsSource (interpreter)));
2138a7015092SGreg Clayton     LoadSubCommand ("alias",   CommandObjectSP (new CommandObjectCommandsAlias (interpreter)));
2139a7015092SGreg Clayton     LoadSubCommand ("unalias", CommandObjectSP (new CommandObjectCommandsUnalias (interpreter)));
2140b547278cSGreg Clayton     LoadSubCommand ("delete",  CommandObjectSP (new CommandObjectCommandsDelete (interpreter)));
2141de164aaaSGreg Clayton     LoadSubCommand ("regex",   CommandObjectSP (new CommandObjectCommandsAddRegex (interpreter)));
2142a5a97ebeSJim Ingham     LoadSubCommand ("history", CommandObjectSP (new CommandObjectCommandsHistory (interpreter)));
2143223383edSEnrico Granata     LoadSubCommand ("script",  CommandObjectSP (new CommandObjectMultiwordCommandsScript (interpreter)));
2144ebc09c36SJim Ingham }
2145ebc09c36SJim Ingham 
21466e3d8e7fSEugene Zelenko CommandObjectMultiwordCommands::~CommandObjectMultiwordCommands() = default;
2147