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
16*6e3d8e7fSEugene Zelenko #include "CommandObjectCommands.h"
17ebc09c36SJim Ingham #include "lldb/Core/Debugger.h"
1844d93782SGreg Clayton #include "lldb/Core/IOHandler.h"
19be93a35aSEnrico Granata #include "lldb/Core/StringList.h"
20de164aaaSGreg Clayton #include "lldb/Interpreter/Args.h"
217594f14fSEnrico Granata #include "lldb/Interpreter/CommandHistory.h"
22ebc09c36SJim Ingham #include "lldb/Interpreter/CommandInterpreter.h"
23de164aaaSGreg Clayton #include "lldb/Interpreter/CommandObjectRegexCommand.h"
24ebc09c36SJim Ingham #include "lldb/Interpreter/CommandReturnObject.h"
25012d4fcaSEnrico Granata #include "lldb/Interpreter/OptionValueBoolean.h"
267594f14fSEnrico Granata #include "lldb/Interpreter/OptionValueUInt64.h"
27ebc09c36SJim Ingham #include "lldb/Interpreter/Options.h"
2899f0b8f9SEnrico Granata #include "lldb/Interpreter/ScriptInterpreter.h"
29ebc09c36SJim Ingham 
30ebc09c36SJim Ingham using namespace lldb;
31ebc09c36SJim Ingham using namespace lldb_private;
32ebc09c36SJim Ingham 
33ebc09c36SJim Ingham //-------------------------------------------------------------------------
34ebc09c36SJim Ingham // CommandObjectCommandsSource
35ebc09c36SJim Ingham //-------------------------------------------------------------------------
36ebc09c36SJim Ingham 
375a988416SJim Ingham class CommandObjectCommandsHistory : public CommandObjectParsed
38a5a97ebeSJim Ingham {
395a988416SJim Ingham public:
405a988416SJim Ingham     CommandObjectCommandsHistory(CommandInterpreter &interpreter) :
415a988416SJim Ingham         CommandObjectParsed(interpreter,
425a988416SJim Ingham                             "command history",
435a988416SJim Ingham                             "Dump the history of commands in this session.",
44*6e3d8e7fSEugene Zelenko                             nullptr),
455a988416SJim Ingham         m_options (interpreter)
465a988416SJim Ingham     {
475a988416SJim Ingham     }
485a988416SJim Ingham 
49*6e3d8e7fSEugene Zelenko     ~CommandObjectCommandsHistory() override = default;
505a988416SJim Ingham 
5113d21e9aSBruce Mitchener     Options *
5213d21e9aSBruce Mitchener     GetOptions () override
535a988416SJim Ingham     {
545a988416SJim Ingham         return &m_options;
555a988416SJim Ingham     }
565a988416SJim Ingham 
575a988416SJim Ingham protected:
58a5a97ebeSJim Ingham     class CommandOptions : public Options
59a5a97ebeSJim Ingham     {
60a5a97ebeSJim Ingham     public:
61a5a97ebeSJim Ingham         CommandOptions (CommandInterpreter &interpreter) :
627594f14fSEnrico Granata             Options (interpreter),
637594f14fSEnrico Granata             m_start_idx(0),
647594f14fSEnrico Granata             m_stop_idx(0),
657594f14fSEnrico Granata             m_count(0),
6663123b64SEnrico Granata             m_clear(false)
67a5a97ebeSJim Ingham         {
68a5a97ebeSJim Ingham         }
69a5a97ebeSJim Ingham 
70*6e3d8e7fSEugene Zelenko         ~CommandOptions() override = default;
71a5a97ebeSJim Ingham 
7213d21e9aSBruce Mitchener         Error
7313d21e9aSBruce Mitchener         SetOptionValue (uint32_t option_idx, const char *option_arg) override
74a5a97ebeSJim Ingham         {
75a5a97ebeSJim Ingham             Error error;
763bcdfc0eSGreg Clayton             const int short_option = m_getopt_table[option_idx].val;
77a5a97ebeSJim Ingham 
78a5a97ebeSJim Ingham             switch (short_option)
79a5a97ebeSJim Ingham             {
80a5a97ebeSJim Ingham                 case 'c':
81c95f7e2aSPavel Labath                     error = m_count.SetValueFromString(option_arg,eVarSetOperationAssign);
82a5a97ebeSJim Ingham                     break;
83a5a97ebeSJim Ingham                 case 's':
847594f14fSEnrico Granata                     if (option_arg && strcmp("end", option_arg) == 0)
857594f14fSEnrico Granata                     {
867594f14fSEnrico Granata                         m_start_idx.SetCurrentValue(UINT64_MAX);
877594f14fSEnrico Granata                         m_start_idx.SetOptionWasSet();
887594f14fSEnrico Granata                     }
897594f14fSEnrico Granata                     else
90c95f7e2aSPavel Labath                         error = m_start_idx.SetValueFromString(option_arg,eVarSetOperationAssign);
917594f14fSEnrico Granata                     break;
927594f14fSEnrico Granata                 case 'e':
93c95f7e2aSPavel Labath                     error = m_stop_idx.SetValueFromString(option_arg,eVarSetOperationAssign);
947594f14fSEnrico Granata                     break;
9563123b64SEnrico Granata                 case 'C':
9663123b64SEnrico Granata                     m_clear.SetCurrentValue(true);
9763123b64SEnrico Granata                     m_clear.SetOptionWasSet();
98a5a97ebeSJim Ingham                     break;
99a5a97ebeSJim Ingham                 default:
10086edbf41SGreg Clayton                     error.SetErrorStringWithFormat ("unrecognized option '%c'", short_option);
101a5a97ebeSJim Ingham                     break;
102a5a97ebeSJim Ingham             }
103a5a97ebeSJim Ingham 
104a5a97ebeSJim Ingham             return error;
105a5a97ebeSJim Ingham         }
106a5a97ebeSJim Ingham 
107a5a97ebeSJim Ingham         void
10813d21e9aSBruce Mitchener         OptionParsingStarting () override
109a5a97ebeSJim Ingham         {
1107594f14fSEnrico Granata             m_start_idx.Clear();
1117594f14fSEnrico Granata             m_stop_idx.Clear();
1127594f14fSEnrico Granata             m_count.Clear();
11363123b64SEnrico Granata             m_clear.Clear();
114a5a97ebeSJim Ingham         }
115a5a97ebeSJim Ingham 
116a5a97ebeSJim Ingham         const OptionDefinition*
11713d21e9aSBruce Mitchener         GetDefinitions () override
118a5a97ebeSJim Ingham         {
119a5a97ebeSJim Ingham             return g_option_table;
120a5a97ebeSJim Ingham         }
121a5a97ebeSJim Ingham 
122a5a97ebeSJim Ingham         // Options table: Required for subclasses of Options.
123a5a97ebeSJim Ingham 
124a5a97ebeSJim Ingham         static OptionDefinition g_option_table[];
125a5a97ebeSJim Ingham 
126a5a97ebeSJim Ingham         // Instance variables to hold the values for command options.
127a5a97ebeSJim Ingham 
1287594f14fSEnrico Granata         OptionValueUInt64 m_start_idx;
1297594f14fSEnrico Granata         OptionValueUInt64 m_stop_idx;
1307594f14fSEnrico Granata         OptionValueUInt64 m_count;
13163123b64SEnrico Granata         OptionValueBoolean m_clear;
132a5a97ebeSJim Ingham     };
133a5a97ebeSJim Ingham 
134a5a97ebeSJim Ingham     bool
13513d21e9aSBruce Mitchener     DoExecute (Args& command, CommandReturnObject &result) override
136a5a97ebeSJim Ingham     {
13763123b64SEnrico Granata         if (m_options.m_clear.GetCurrentValue() && m_options.m_clear.OptionWasSet())
1387594f14fSEnrico Granata         {
1397594f14fSEnrico Granata             m_interpreter.GetCommandHistory().Clear();
1407594f14fSEnrico Granata             result.SetStatus(lldb::eReturnStatusSuccessFinishNoResult);
1417594f14fSEnrico Granata         }
1427594f14fSEnrico Granata         else
1437594f14fSEnrico Granata         {
1447594f14fSEnrico Granata             if (m_options.m_start_idx.OptionWasSet() && m_options.m_stop_idx.OptionWasSet() && m_options.m_count.OptionWasSet())
1457594f14fSEnrico Granata             {
1467594f14fSEnrico Granata                 result.AppendError("--count, --start-index and --end-index cannot be all specified in the same invocation");
1477594f14fSEnrico Granata                 result.SetStatus(lldb::eReturnStatusFailed);
1487594f14fSEnrico Granata             }
1497594f14fSEnrico Granata             else
1507594f14fSEnrico Granata             {
15184400ec7SVirgile Bello                 std::pair<bool,uint64_t> start_idx(m_options.m_start_idx.OptionWasSet(),m_options.m_start_idx.GetCurrentValue());
15284400ec7SVirgile Bello                 std::pair<bool,uint64_t> stop_idx(m_options.m_stop_idx.OptionWasSet(),m_options.m_stop_idx.GetCurrentValue());
15384400ec7SVirgile Bello                 std::pair<bool,uint64_t> count(m_options.m_count.OptionWasSet(),m_options.m_count.GetCurrentValue());
154a5a97ebeSJim Ingham 
1557594f14fSEnrico Granata                 const CommandHistory& history(m_interpreter.GetCommandHistory());
1567594f14fSEnrico Granata 
1577594f14fSEnrico Granata                 if (start_idx.first && start_idx.second == UINT64_MAX)
1587594f14fSEnrico Granata                 {
1597594f14fSEnrico Granata                     if (count.first)
1607594f14fSEnrico Granata                     {
1617594f14fSEnrico Granata                         start_idx.second = history.GetSize() - count.second;
1627594f14fSEnrico Granata                         stop_idx.second = history.GetSize() - 1;
1637594f14fSEnrico Granata                     }
1647594f14fSEnrico Granata                     else if (stop_idx.first)
1657594f14fSEnrico Granata                     {
1667594f14fSEnrico Granata                         start_idx.second = stop_idx.second;
1677594f14fSEnrico Granata                         stop_idx.second = history.GetSize() - 1;
1687594f14fSEnrico Granata                     }
1697594f14fSEnrico Granata                     else
1707594f14fSEnrico Granata                     {
1717594f14fSEnrico Granata                         start_idx.second = 0;
1727594f14fSEnrico Granata                         stop_idx.second = history.GetSize() - 1;
1737594f14fSEnrico Granata                     }
1747594f14fSEnrico Granata                 }
1757594f14fSEnrico Granata                 else
1767594f14fSEnrico Granata                 {
1777594f14fSEnrico Granata                     if (!start_idx.first && !stop_idx.first && !count.first)
1787594f14fSEnrico Granata                     {
1797594f14fSEnrico Granata                         start_idx.second = 0;
1807594f14fSEnrico Granata                         stop_idx.second = history.GetSize() - 1;
1817594f14fSEnrico Granata                     }
1827594f14fSEnrico Granata                     else if (start_idx.first)
1837594f14fSEnrico Granata                     {
1847594f14fSEnrico Granata                         if (count.first)
1857594f14fSEnrico Granata                         {
1867594f14fSEnrico Granata                             stop_idx.second = start_idx.second + count.second - 1;
1877594f14fSEnrico Granata                         }
1887594f14fSEnrico Granata                         else if (!stop_idx.first)
1897594f14fSEnrico Granata                         {
1907594f14fSEnrico Granata                             stop_idx.second = history.GetSize() - 1;
1917594f14fSEnrico Granata                         }
1927594f14fSEnrico Granata                     }
1937594f14fSEnrico Granata                     else if (stop_idx.first)
1947594f14fSEnrico Granata                     {
1957594f14fSEnrico Granata                         if (count.first)
1967594f14fSEnrico Granata                         {
1977594f14fSEnrico Granata                             if (stop_idx.second >= count.second)
1987594f14fSEnrico Granata                                 start_idx.second = stop_idx.second - count.second + 1;
1997594f14fSEnrico Granata                             else
2007594f14fSEnrico Granata                                 start_idx.second = 0;
2017594f14fSEnrico Granata                         }
2027594f14fSEnrico Granata                     }
2037594f14fSEnrico Granata                     else /* if (count.first) */
2047594f14fSEnrico Granata                     {
2057594f14fSEnrico Granata                         start_idx.second = 0;
2067594f14fSEnrico Granata                         stop_idx.second = count.second - 1;
2077594f14fSEnrico Granata                     }
2087594f14fSEnrico Granata                 }
2097594f14fSEnrico Granata                 history.Dump(result.GetOutputStream(), start_idx.second, stop_idx.second);
2107594f14fSEnrico Granata             }
2117594f14fSEnrico Granata         }
212a5a97ebeSJim Ingham         return result.Succeeded();
213a5a97ebeSJim Ingham 
214a5a97ebeSJim Ingham     }
2155a988416SJim Ingham 
2165a988416SJim Ingham     CommandOptions m_options;
217a5a97ebeSJim Ingham };
218a5a97ebeSJim Ingham 
219a5a97ebeSJim Ingham OptionDefinition
220a5a97ebeSJim Ingham CommandObjectCommandsHistory::CommandOptions::g_option_table[] =
221a5a97ebeSJim Ingham {
222*6e3d8e7fSEugene Zelenko { LLDB_OPT_SET_1, false, "count", 'c', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeUnsignedInteger,        "How many history commands to print."},
223*6e3d8e7fSEugene 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)."},
224*6e3d8e7fSEugene Zelenko { LLDB_OPT_SET_1, false, "end-index", 'e', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeUnsignedInteger,    "Index at which to stop printing history commands."},
225*6e3d8e7fSEugene Zelenko { LLDB_OPT_SET_2, false, "clear", 'C', OptionParser::eNoArgument, nullptr, nullptr, 0, eArgTypeBoolean,    "Clears the current command history."},
226*6e3d8e7fSEugene Zelenko { 0, false, nullptr, 0, 0, nullptr, nullptr, 0, eArgTypeNone, nullptr }
227a5a97ebeSJim Ingham };
228a5a97ebeSJim Ingham 
229a5a97ebeSJim Ingham //-------------------------------------------------------------------------
230a5a97ebeSJim Ingham // CommandObjectCommandsSource
231a5a97ebeSJim Ingham //-------------------------------------------------------------------------
232a5a97ebeSJim Ingham 
2335a988416SJim Ingham class CommandObjectCommandsSource : public CommandObjectParsed
234ebc09c36SJim Ingham {
2355a988416SJim Ingham public:
2365a988416SJim Ingham     CommandObjectCommandsSource(CommandInterpreter &interpreter) :
2375a988416SJim Ingham         CommandObjectParsed(interpreter,
2385a988416SJim Ingham                             "command source",
2395a988416SJim Ingham                             "Read in debugger commands from the file <filename> and execute them.",
240*6e3d8e7fSEugene Zelenko                             nullptr),
2415a988416SJim Ingham         m_options (interpreter)
2425a988416SJim Ingham     {
2435a988416SJim Ingham         CommandArgumentEntry arg;
2445a988416SJim Ingham         CommandArgumentData file_arg;
2455a988416SJim Ingham 
2465a988416SJim Ingham         // Define the first (and only) variant of this arg.
2475a988416SJim Ingham         file_arg.arg_type = eArgTypeFilename;
2485a988416SJim Ingham         file_arg.arg_repetition = eArgRepeatPlain;
2495a988416SJim Ingham 
2505a988416SJim Ingham         // There is only one variant this argument could be; put it into the argument entry.
2515a988416SJim Ingham         arg.push_back (file_arg);
2525a988416SJim Ingham 
2535a988416SJim Ingham         // Push the data for the first argument into the m_arguments vector.
2545a988416SJim Ingham         m_arguments.push_back (arg);
2555a988416SJim Ingham     }
2565a988416SJim Ingham 
257*6e3d8e7fSEugene Zelenko     ~CommandObjectCommandsSource() override = default;
2585a988416SJim Ingham 
25913d21e9aSBruce Mitchener     const char*
26013d21e9aSBruce Mitchener     GetRepeatCommand (Args &current_command_args, uint32_t index) override
2615a988416SJim Ingham     {
2625a988416SJim Ingham         return "";
2635a988416SJim Ingham     }
2645a988416SJim Ingham 
26513d21e9aSBruce Mitchener     int
2665a988416SJim Ingham     HandleArgumentCompletion (Args &input,
2675a988416SJim Ingham                               int &cursor_index,
2685a988416SJim Ingham                               int &cursor_char_position,
2695a988416SJim Ingham                               OptionElementVector &opt_element_vector,
2705a988416SJim Ingham                               int match_start_point,
2715a988416SJim Ingham                               int max_return_elements,
2725a988416SJim Ingham                               bool &word_complete,
27313d21e9aSBruce Mitchener                               StringList &matches) override
2745a988416SJim Ingham     {
2755a988416SJim Ingham         std::string completion_str (input.GetArgumentAtIndex(cursor_index));
2765a988416SJim Ingham         completion_str.erase (cursor_char_position);
2775a988416SJim Ingham 
2785a988416SJim Ingham         CommandCompletions::InvokeCommonCompletionCallbacks(m_interpreter,
2795a988416SJim Ingham                                                             CommandCompletions::eDiskFileCompletion,
2805a988416SJim Ingham                                                             completion_str.c_str(),
2815a988416SJim Ingham                                                             match_start_point,
2825a988416SJim Ingham                                                             max_return_elements,
283*6e3d8e7fSEugene Zelenko                                                             nullptr,
2845a988416SJim Ingham                                                             word_complete,
2855a988416SJim Ingham                                                             matches);
2865a988416SJim Ingham         return matches.GetSize();
2875a988416SJim Ingham     }
2885a988416SJim Ingham 
28913d21e9aSBruce Mitchener     Options *
29013d21e9aSBruce Mitchener     GetOptions () override
2915a988416SJim Ingham     {
2925a988416SJim Ingham         return &m_options;
2935a988416SJim Ingham     }
2945a988416SJim Ingham 
2955a988416SJim Ingham protected:
296e16c50a1SJim Ingham     class CommandOptions : public Options
297e16c50a1SJim Ingham     {
298e16c50a1SJim Ingham     public:
299eb0103f2SGreg Clayton         CommandOptions (CommandInterpreter &interpreter) :
300012d4fcaSEnrico Granata             Options (interpreter),
301340b0309SGreg Clayton             m_stop_on_error (true),
302340b0309SGreg Clayton             m_silent_run (false),
303340b0309SGreg Clayton             m_stop_on_continue (true)
304eb0103f2SGreg Clayton         {
305eb0103f2SGreg Clayton         }
306e16c50a1SJim Ingham 
307*6e3d8e7fSEugene Zelenko         ~CommandOptions() override = default;
308e16c50a1SJim Ingham 
30913d21e9aSBruce Mitchener         Error
31013d21e9aSBruce Mitchener         SetOptionValue (uint32_t option_idx, const char *option_arg) override
311e16c50a1SJim Ingham         {
312e16c50a1SJim Ingham             Error error;
3133bcdfc0eSGreg Clayton             const int short_option = m_getopt_table[option_idx].val;
314e16c50a1SJim Ingham 
315e16c50a1SJim Ingham             switch (short_option)
316e16c50a1SJim Ingham             {
317e16c50a1SJim Ingham                 case 'e':
318c95f7e2aSPavel Labath                     error = m_stop_on_error.SetValueFromString(option_arg);
319e16c50a1SJim Ingham                     break;
320340b0309SGreg Clayton 
321e16c50a1SJim Ingham                 case 'c':
322c95f7e2aSPavel Labath                     error = m_stop_on_continue.SetValueFromString(option_arg);
323e16c50a1SJim Ingham                     break;
324340b0309SGreg Clayton 
32560986174SMichael Sartain                 case 's':
326c95f7e2aSPavel Labath                     error = m_silent_run.SetValueFromString(option_arg);
32760986174SMichael Sartain                     break;
328340b0309SGreg Clayton 
329e16c50a1SJim Ingham                 default:
33086edbf41SGreg Clayton                     error.SetErrorStringWithFormat ("unrecognized option '%c'", short_option);
331e16c50a1SJim Ingham                     break;
332e16c50a1SJim Ingham             }
333e16c50a1SJim Ingham 
334e16c50a1SJim Ingham             return error;
335e16c50a1SJim Ingham         }
336e16c50a1SJim Ingham 
337e16c50a1SJim Ingham         void
33813d21e9aSBruce Mitchener         OptionParsingStarting () override
339e16c50a1SJim Ingham         {
340012d4fcaSEnrico Granata             m_stop_on_error.Clear();
341340b0309SGreg Clayton             m_silent_run.Clear();
342340b0309SGreg Clayton             m_stop_on_continue.Clear();
343e16c50a1SJim Ingham         }
344e16c50a1SJim Ingham 
345e0d378b3SGreg Clayton         const OptionDefinition*
34613d21e9aSBruce Mitchener         GetDefinitions () override
347e16c50a1SJim Ingham         {
348e16c50a1SJim Ingham             return g_option_table;
349e16c50a1SJim Ingham         }
350e16c50a1SJim Ingham 
351e16c50a1SJim Ingham         // Options table: Required for subclasses of Options.
352e16c50a1SJim Ingham 
353e0d378b3SGreg Clayton         static OptionDefinition g_option_table[];
354e16c50a1SJim Ingham 
355e16c50a1SJim Ingham         // Instance variables to hold the values for command options.
356e16c50a1SJim Ingham 
357012d4fcaSEnrico Granata         OptionValueBoolean m_stop_on_error;
358340b0309SGreg Clayton         OptionValueBoolean m_silent_run;
359340b0309SGreg Clayton         OptionValueBoolean m_stop_on_continue;
360e16c50a1SJim Ingham     };
361e16c50a1SJim Ingham 
362ebc09c36SJim Ingham     bool
36313d21e9aSBruce Mitchener     DoExecute(Args& command, CommandReturnObject &result) override
364ebc09c36SJim Ingham     {
365c7bece56SGreg Clayton         const size_t argc = command.GetArgumentCount();
366ebc09c36SJim Ingham         if (argc == 1)
367ebc09c36SJim Ingham         {
3685a988416SJim Ingham             const char *filename = command.GetArgumentAtIndex(0);
369ebc09c36SJim Ingham 
3701ee3853fSJohnny Chen             FileSpec cmd_file (filename, true);
371*6e3d8e7fSEugene Zelenko             ExecutionContext *exe_ctx = nullptr;  // Just use the default context.
372ebc09c36SJim Ingham 
373340b0309SGreg Clayton             // If any options were set, then use them
374340b0309SGreg Clayton             if (m_options.m_stop_on_error.OptionWasSet()    ||
375340b0309SGreg Clayton                 m_options.m_silent_run.OptionWasSet()       ||
376340b0309SGreg Clayton                 m_options.m_stop_on_continue.OptionWasSet())
377340b0309SGreg Clayton             {
378340b0309SGreg Clayton                 // Use user set settings
37926c7bf93SJim Ingham                 CommandInterpreterRunOptions options;
38026c7bf93SJim Ingham                 options.SetStopOnContinue(m_options.m_stop_on_continue.GetCurrentValue());
38126c7bf93SJim Ingham                 options.SetStopOnError (m_options.m_stop_on_error.GetCurrentValue());
3827d8555c4SJim Ingham                 options.SetEchoCommands (!m_options.m_silent_run.GetCurrentValue());
3837d8555c4SJim Ingham                 options.SetPrintResults (!m_options.m_silent_run.GetCurrentValue());
38426c7bf93SJim Ingham 
385e16c50a1SJim Ingham                 m_interpreter.HandleCommandsFromFile (cmd_file,
386e16c50a1SJim Ingham                                                       exe_ctx,
38726c7bf93SJim Ingham                                                       options,
388e16c50a1SJim Ingham                                                       result);
389340b0309SGreg Clayton             }
390340b0309SGreg Clayton             else
391340b0309SGreg Clayton             {
392340b0309SGreg Clayton                 // No options were set, inherit any settings from nested "command source" commands,
393340b0309SGreg Clayton                 // or set to sane default settings...
39426c7bf93SJim Ingham                 CommandInterpreterRunOptions options;
395340b0309SGreg Clayton                 m_interpreter.HandleCommandsFromFile (cmd_file,
396340b0309SGreg Clayton                                                       exe_ctx,
39726c7bf93SJim Ingham                                                       options,
398340b0309SGreg Clayton                                                       result);
399340b0309SGreg Clayton             }
400ebc09c36SJim Ingham         }
401ebc09c36SJim Ingham         else
402ebc09c36SJim Ingham         {
403ebc09c36SJim Ingham             result.AppendErrorWithFormat("'%s' takes exactly one executable filename argument.\n", GetCommandName());
404ebc09c36SJim Ingham             result.SetStatus (eReturnStatusFailed);
405ebc09c36SJim Ingham         }
406ebc09c36SJim Ingham         return result.Succeeded();
407ebc09c36SJim Ingham     }
408*6e3d8e7fSEugene Zelenko 
4095a988416SJim Ingham     CommandOptions m_options;
410ebc09c36SJim Ingham };
411ebc09c36SJim Ingham 
412e0d378b3SGreg Clayton OptionDefinition
413e16c50a1SJim Ingham CommandObjectCommandsSource::CommandOptions::g_option_table[] =
414e16c50a1SJim Ingham {
415*6e3d8e7fSEugene Zelenko { LLDB_OPT_SET_ALL, false, "stop-on-error", 'e', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeBoolean,    "If true, stop executing commands on error."},
416*6e3d8e7fSEugene Zelenko { LLDB_OPT_SET_ALL, false, "stop-on-continue", 'c', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeBoolean, "If true, stop executing commands on continue."},
417*6e3d8e7fSEugene Zelenko { LLDB_OPT_SET_ALL, false, "silent-run", 's', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeBoolean, "If true don't echo commands while executing."},
418*6e3d8e7fSEugene Zelenko { 0, false, nullptr, 0, 0, nullptr, nullptr, 0, eArgTypeNone, nullptr }
419e16c50a1SJim Ingham };
420e16c50a1SJim Ingham 
421ebc09c36SJim Ingham #pragma mark CommandObjectCommandsAlias
422ebc09c36SJim Ingham //-------------------------------------------------------------------------
423ebc09c36SJim Ingham // CommandObjectCommandsAlias
424ebc09c36SJim Ingham //-------------------------------------------------------------------------
425ebc09c36SJim Ingham 
426be93a35aSEnrico Granata static const char *g_python_command_instructions =   "Enter your Python command(s). Type 'DONE' to end.\n"
427be93a35aSEnrico Granata                                                      "You must define a Python function with this signature:\n"
42844d93782SGreg Clayton                                                      "def my_command_impl(debugger, args, result, internal_dict):\n";
429be93a35aSEnrico Granata 
4305a988416SJim Ingham class CommandObjectCommandsAlias : public CommandObjectRaw
431ebc09c36SJim Ingham {
432ebc09c36SJim Ingham public:
433a7015092SGreg Clayton     CommandObjectCommandsAlias (CommandInterpreter &interpreter) :
4345a988416SJim Ingham         CommandObjectRaw(interpreter,
4350e5e5a79SGreg Clayton                          "command alias",
436e3d26315SCaroline Tice                          "Allow users to define their own debugger command abbreviations.",
437*6e3d8e7fSEugene Zelenko                          nullptr)
438ebc09c36SJim Ingham     {
439ebc09c36SJim Ingham         SetHelpLong(
440ea671fbdSKate Stone "'alias' allows the user to create a short-cut or abbreviation for long \
441ea671fbdSKate Stone commands, multi-word commands, and commands that take particular options.  \
442ea671fbdSKate Stone Below are some simple examples of how one might use the 'alias' command:" R"(
443ea671fbdSKate Stone 
444ea671fbdSKate Stone (lldb) command alias sc script
445ea671fbdSKate Stone 
446ea671fbdSKate Stone     Creates the abbreviation 'sc' for the 'script' command.
447ea671fbdSKate Stone 
448ea671fbdSKate Stone (lldb) command alias bp breakpoint
449ea671fbdSKate Stone 
450ea671fbdSKate Stone )" "    Creates the abbreviation 'bp' for the 'breakpoint' command.  Since \
451ea671fbdSKate Stone breakpoint commands are two-word commands, the user would still need to \
452ea671fbdSKate Stone enter the second word after 'bp', e.g. 'bp enable' or 'bp delete'." R"(
453ea671fbdSKate Stone 
454ea671fbdSKate Stone (lldb) command alias bpl breakpoint list
455ea671fbdSKate Stone 
456ea671fbdSKate Stone     Creates the abbreviation 'bpl' for the two-word command 'breakpoint list'.
457ea671fbdSKate Stone 
458ea671fbdSKate Stone )" "An alias can include some options for the command, with the values either \
459ea671fbdSKate Stone filled in at the time the alias is created, or specified as positional \
460ea671fbdSKate Stone arguments, to be filled in when the alias is invoked.  The following example \
461ea671fbdSKate Stone shows how to create aliases with options:" R"(
462ea671fbdSKate Stone 
463ea671fbdSKate Stone (lldb) command alias bfl breakpoint set -f %1 -l %2
464ea671fbdSKate Stone 
465ea671fbdSKate Stone )" "    Creates the abbreviation 'bfl' (for break-file-line), with the -f and -l \
466ea671fbdSKate Stone options already part of the alias.  So if the user wants to set a breakpoint \
467ea671fbdSKate Stone by file and line without explicitly having to use the -f and -l options, the \
468ea671fbdSKate Stone user can now use 'bfl' instead.  The '%1' and '%2' are positional placeholders \
469ea671fbdSKate Stone for the actual arguments that will be passed when the alias command is used.  \
470ea671fbdSKate Stone The number in the placeholder refers to the position/order the actual value \
471ea671fbdSKate Stone occupies when the alias is used.  All the occurrences of '%1' in the alias \
472ea671fbdSKate Stone will be replaced with the first argument, all the occurrences of '%2' in the \
473ea671fbdSKate Stone alias will be replaced with the second argument, and so on.  This also allows \
474ea671fbdSKate Stone actual arguments to be used multiple times within an alias (see 'process \
475ea671fbdSKate Stone launch' example below)." R"(
476ea671fbdSKate Stone 
477ea671fbdSKate Stone )" "Note: the positional arguments must substitute as whole words in the resultant \
478ea671fbdSKate Stone command, so you can't at present do something like this to append the file extension \
479ea671fbdSKate Stone \".cpp\":" R"(
480ea671fbdSKate Stone 
481ea671fbdSKate Stone (lldb) command alias bcppfl breakpoint set -f %1.cpp -l %2
482ea671fbdSKate Stone 
483ea671fbdSKate Stone )" "For more complex aliasing, use the \"command regex\" command instead.  In the \
484ea671fbdSKate Stone 'bfl' case above, the actual file value will be filled in with the first argument \
485ea671fbdSKate Stone following 'bfl' and the actual line number value will be filled in with the second \
486ea671fbdSKate Stone argument.  The user would use this alias as follows:" R"(
487ea671fbdSKate Stone 
488ea671fbdSKate Stone (lldb) command alias bfl breakpoint set -f %1 -l %2
489ea671fbdSKate Stone (lldb) bfl my-file.c 137
490ea671fbdSKate Stone 
491ea671fbdSKate Stone This would be the same as if the user had entered 'breakpoint set -f my-file.c -l 137'.
492ea671fbdSKate Stone 
493ea671fbdSKate Stone Another example:
494ea671fbdSKate Stone 
495ea671fbdSKate Stone (lldb) command alias pltty process launch -s -o %1 -e %1
496ea671fbdSKate Stone (lldb) pltty /dev/tty0
497ea671fbdSKate Stone 
498ea671fbdSKate Stone     Interpreted as 'process launch -s -o /dev/tty0 -e /dev/tty0'
499ea671fbdSKate Stone 
500ea671fbdSKate Stone )" "If the user always wanted to pass the same value to a particular option, the \
501ea671fbdSKate Stone alias could be defined with that value directly in the alias as a constant, \
502ea671fbdSKate Stone rather than using a positional placeholder:" R"(
503ea671fbdSKate Stone 
504ea671fbdSKate Stone (lldb) command alias bl3 breakpoint set -f %1 -l 3
505ea671fbdSKate Stone 
506ea671fbdSKate Stone     Always sets a breakpoint on line 3 of whatever file is indicated.)"
507ea671fbdSKate Stone         );
508ebc09c36SJim Ingham 
509405fe67fSCaroline Tice         CommandArgumentEntry arg1;
510405fe67fSCaroline Tice         CommandArgumentEntry arg2;
511405fe67fSCaroline Tice         CommandArgumentEntry arg3;
512405fe67fSCaroline Tice         CommandArgumentData alias_arg;
513405fe67fSCaroline Tice         CommandArgumentData cmd_arg;
514405fe67fSCaroline Tice         CommandArgumentData options_arg;
515405fe67fSCaroline Tice 
516405fe67fSCaroline Tice         // Define the first (and only) variant of this arg.
517405fe67fSCaroline Tice         alias_arg.arg_type = eArgTypeAliasName;
518405fe67fSCaroline Tice         alias_arg.arg_repetition = eArgRepeatPlain;
519405fe67fSCaroline Tice 
520405fe67fSCaroline Tice         // There is only one variant this argument could be; put it into the argument entry.
521405fe67fSCaroline Tice         arg1.push_back (alias_arg);
522405fe67fSCaroline Tice 
523405fe67fSCaroline Tice         // Define the first (and only) variant of this arg.
524405fe67fSCaroline Tice         cmd_arg.arg_type = eArgTypeCommandName;
525405fe67fSCaroline Tice         cmd_arg.arg_repetition = eArgRepeatPlain;
526405fe67fSCaroline Tice 
527405fe67fSCaroline Tice         // There is only one variant this argument could be; put it into the argument entry.
528405fe67fSCaroline Tice         arg2.push_back (cmd_arg);
529405fe67fSCaroline Tice 
530405fe67fSCaroline Tice         // Define the first (and only) variant of this arg.
531405fe67fSCaroline Tice         options_arg.arg_type = eArgTypeAliasOptions;
532405fe67fSCaroline Tice         options_arg.arg_repetition = eArgRepeatOptional;
533405fe67fSCaroline Tice 
534405fe67fSCaroline Tice         // There is only one variant this argument could be; put it into the argument entry.
535405fe67fSCaroline Tice         arg3.push_back (options_arg);
536405fe67fSCaroline Tice 
537405fe67fSCaroline Tice         // Push the data for the first argument into the m_arguments vector.
538405fe67fSCaroline Tice         m_arguments.push_back (arg1);
539405fe67fSCaroline Tice         m_arguments.push_back (arg2);
540405fe67fSCaroline Tice         m_arguments.push_back (arg3);
541ebc09c36SJim Ingham     }
542ebc09c36SJim Ingham 
543*6e3d8e7fSEugene Zelenko     ~CommandObjectCommandsAlias() override = default;
544ebc09c36SJim Ingham 
5455a988416SJim Ingham protected:
54613d21e9aSBruce Mitchener     bool
54713d21e9aSBruce Mitchener     DoExecute (const char *raw_command_line, CommandReturnObject &result) override
548844d2303SCaroline Tice     {
549844d2303SCaroline Tice         Args args (raw_command_line);
550844d2303SCaroline Tice         std::string raw_command_string (raw_command_line);
551844d2303SCaroline Tice 
552844d2303SCaroline Tice         size_t argc = args.GetArgumentCount();
553844d2303SCaroline Tice 
554844d2303SCaroline Tice         if (argc < 2)
555844d2303SCaroline Tice         {
556844d2303SCaroline Tice             result.AppendError ("'alias' requires at least two arguments");
557844d2303SCaroline Tice             result.SetStatus (eReturnStatusFailed);
558844d2303SCaroline Tice             return false;
559844d2303SCaroline Tice         }
560844d2303SCaroline Tice 
561844d2303SCaroline Tice         // Get the alias command.
562844d2303SCaroline Tice 
563844d2303SCaroline Tice         const std::string alias_command = args.GetArgumentAtIndex (0);
564844d2303SCaroline Tice 
565844d2303SCaroline Tice         // Strip the new alias name off 'raw_command_string'  (leave it on args, which gets passed to 'Execute', which
566844d2303SCaroline Tice         // does the stripping itself.
567844d2303SCaroline Tice         size_t pos = raw_command_string.find (alias_command);
568844d2303SCaroline Tice         if (pos == 0)
569844d2303SCaroline Tice         {
570844d2303SCaroline Tice             raw_command_string = raw_command_string.substr (alias_command.size());
571844d2303SCaroline Tice             pos = raw_command_string.find_first_not_of (' ');
572844d2303SCaroline Tice             if ((pos != std::string::npos) && (pos > 0))
573844d2303SCaroline Tice                 raw_command_string = raw_command_string.substr (pos);
574844d2303SCaroline Tice         }
575844d2303SCaroline Tice         else
576844d2303SCaroline Tice         {
577844d2303SCaroline Tice             result.AppendError ("Error parsing command string.  No alias created.");
578844d2303SCaroline Tice             result.SetStatus (eReturnStatusFailed);
579844d2303SCaroline Tice             return false;
580844d2303SCaroline Tice         }
581844d2303SCaroline Tice 
582844d2303SCaroline Tice 
583844d2303SCaroline Tice         // Verify that the command is alias-able.
584844d2303SCaroline Tice         if (m_interpreter.CommandExists (alias_command.c_str()))
585844d2303SCaroline Tice         {
586844d2303SCaroline Tice             result.AppendErrorWithFormat ("'%s' is a permanent debugger command and cannot be redefined.\n",
587844d2303SCaroline Tice                                           alias_command.c_str());
588844d2303SCaroline Tice             result.SetStatus (eReturnStatusFailed);
589844d2303SCaroline Tice             return false;
590844d2303SCaroline Tice         }
591844d2303SCaroline Tice 
592844d2303SCaroline Tice         // Get CommandObject that is being aliased. The command name is read from the front of raw_command_string.
593844d2303SCaroline Tice         // raw_command_string is returned with the name of the command object stripped off the front.
594844d2303SCaroline Tice         CommandObject *cmd_obj = m_interpreter.GetCommandObjectForCommand (raw_command_string);
595844d2303SCaroline Tice 
596844d2303SCaroline Tice         if (!cmd_obj)
597844d2303SCaroline Tice         {
59886edbf41SGreg Clayton             result.AppendErrorWithFormat ("invalid command given to 'alias'. '%s' does not begin with a valid command."
599844d2303SCaroline Tice                                           "  No alias created.", raw_command_string.c_str());
600844d2303SCaroline Tice             result.SetStatus (eReturnStatusFailed);
601844d2303SCaroline Tice             return false;
602844d2303SCaroline Tice         }
603844d2303SCaroline Tice         else if (!cmd_obj->WantsRawCommandString ())
604844d2303SCaroline Tice         {
605844d2303SCaroline Tice             // Note that args was initialized with the original command, and has not been updated to this point.
606844d2303SCaroline Tice             // Therefore can we pass it to the version of Execute that does not need/expect raw input in the alias.
6075a988416SJim Ingham             return HandleAliasingNormalCommand (args, result);
608844d2303SCaroline Tice         }
609844d2303SCaroline Tice         else
610844d2303SCaroline Tice         {
6115a988416SJim Ingham             return HandleAliasingRawCommand (alias_command, raw_command_string, *cmd_obj, result);
6125a988416SJim Ingham         }
6135a988416SJim Ingham         return result.Succeeded();
6145a988416SJim Ingham     }
6155a988416SJim Ingham 
6165a988416SJim Ingham     bool
6175a988416SJim Ingham     HandleAliasingRawCommand (const std::string &alias_command, std::string &raw_command_string, CommandObject &cmd_obj, CommandReturnObject &result)
6185a988416SJim Ingham     {
619844d2303SCaroline Tice             // Verify & handle any options/arguments passed to the alias command
620844d2303SCaroline Tice 
621844d2303SCaroline Tice             OptionArgVectorSP option_arg_vector_sp = OptionArgVectorSP (new OptionArgVector);
622844d2303SCaroline Tice             OptionArgVector *option_arg_vector = option_arg_vector_sp.get();
623844d2303SCaroline Tice 
6245a988416SJim Ingham             CommandObjectSP cmd_obj_sp = m_interpreter.GetCommandSPExact (cmd_obj.GetCommandName(), false);
625844d2303SCaroline Tice 
626ca90c47eSCaroline Tice             if (!m_interpreter.ProcessAliasOptionsArgs (cmd_obj_sp, raw_command_string.c_str(), option_arg_vector_sp))
627844d2303SCaroline Tice             {
628844d2303SCaroline Tice                 result.AppendError ("Unable to create requested alias.\n");
629ca90c47eSCaroline Tice                 result.SetStatus (eReturnStatusFailed);
630844d2303SCaroline Tice                 return false;
631844d2303SCaroline Tice             }
632844d2303SCaroline Tice 
633844d2303SCaroline Tice             // Create the alias
634844d2303SCaroline Tice             if (m_interpreter.AliasExists (alias_command.c_str())
635844d2303SCaroline Tice                 || m_interpreter.UserCommandExists (alias_command.c_str()))
636844d2303SCaroline Tice             {
637844d2303SCaroline Tice                 OptionArgVectorSP temp_option_arg_sp (m_interpreter.GetAliasOptions (alias_command.c_str()));
638*6e3d8e7fSEugene Zelenko                 if (temp_option_arg_sp)
639844d2303SCaroline Tice                 {
640*6e3d8e7fSEugene Zelenko                     if (option_arg_vector->empty())
641844d2303SCaroline Tice                         m_interpreter.RemoveAliasOptions (alias_command.c_str());
642844d2303SCaroline Tice                 }
643844d2303SCaroline Tice                 result.AppendWarningWithFormat ("Overwriting existing definition for '%s'.\n",
644844d2303SCaroline Tice                                                 alias_command.c_str());
645844d2303SCaroline Tice             }
646844d2303SCaroline Tice 
647472362e6SCaroline Tice             if (cmd_obj_sp)
648472362e6SCaroline Tice             {
649844d2303SCaroline Tice                 m_interpreter.AddAlias (alias_command.c_str(), cmd_obj_sp);
650*6e3d8e7fSEugene Zelenko                 if (!option_arg_vector->empty())
651844d2303SCaroline Tice                     m_interpreter.AddOrReplaceAliasOptions (alias_command.c_str(), option_arg_vector_sp);
652844d2303SCaroline Tice                 result.SetStatus (eReturnStatusSuccessFinishNoResult);
653844d2303SCaroline Tice             }
654472362e6SCaroline Tice             else
655472362e6SCaroline Tice             {
656472362e6SCaroline Tice                 result.AppendError ("Unable to create requested alias.\n");
657472362e6SCaroline Tice                 result.SetStatus (eReturnStatusFailed);
658472362e6SCaroline Tice             }
659844d2303SCaroline Tice             return result.Succeeded ();
660844d2303SCaroline Tice     }
661ebc09c36SJim Ingham 
662ebc09c36SJim Ingham     bool
6635a988416SJim Ingham     HandleAliasingNormalCommand (Args& args, CommandReturnObject &result)
664ebc09c36SJim Ingham     {
665867b185dSCaroline Tice         size_t argc = args.GetArgumentCount();
666ebc09c36SJim Ingham 
667ebc09c36SJim Ingham         if (argc < 2)
668ebc09c36SJim Ingham         {
669ebc09c36SJim Ingham             result.AppendError ("'alias' requires at least two arguments");
670ebc09c36SJim Ingham             result.SetStatus (eReturnStatusFailed);
671ebc09c36SJim Ingham             return false;
672ebc09c36SJim Ingham         }
673ebc09c36SJim Ingham 
674ebc09c36SJim Ingham         const std::string alias_command = args.GetArgumentAtIndex(0);
675ebc09c36SJim Ingham         const std::string actual_command = args.GetArgumentAtIndex(1);
676ebc09c36SJim Ingham 
677ebc09c36SJim Ingham         args.Shift();  // Shift the alias command word off the argument vector.
678ebc09c36SJim Ingham         args.Shift();  // Shift the old command word off the argument vector.
679ebc09c36SJim Ingham 
680ebc09c36SJim Ingham         // Verify that the command is alias'able, and get the appropriate command object.
681ebc09c36SJim Ingham 
682a7015092SGreg Clayton         if (m_interpreter.CommandExists (alias_command.c_str()))
683ebc09c36SJim Ingham         {
684ebc09c36SJim Ingham             result.AppendErrorWithFormat ("'%s' is a permanent debugger command and cannot be redefined.\n",
685ebc09c36SJim Ingham                                          alias_command.c_str());
686ebc09c36SJim Ingham             result.SetStatus (eReturnStatusFailed);
687ebc09c36SJim Ingham         }
688ebc09c36SJim Ingham         else
689ebc09c36SJim Ingham         {
690a7015092SGreg Clayton              CommandObjectSP command_obj_sp(m_interpreter.GetCommandSPExact (actual_command.c_str(), true));
691ebc09c36SJim Ingham              CommandObjectSP subcommand_obj_sp;
692ebc09c36SJim Ingham              bool use_subcommand = false;
693*6e3d8e7fSEugene Zelenko              if (command_obj_sp)
694ebc09c36SJim Ingham              {
695ebc09c36SJim Ingham                  CommandObject *cmd_obj = command_obj_sp.get();
696*6e3d8e7fSEugene Zelenko                  CommandObject *sub_cmd_obj = nullptr;
697ebc09c36SJim Ingham                  OptionArgVectorSP option_arg_vector_sp = OptionArgVectorSP (new OptionArgVector);
698ebc09c36SJim Ingham                  OptionArgVector *option_arg_vector = option_arg_vector_sp.get();
699ebc09c36SJim Ingham 
700844d2303SCaroline Tice                  while (cmd_obj->IsMultiwordObject() && args.GetArgumentCount() > 0)
701ebc09c36SJim Ingham                  {
702ebc09c36SJim Ingham                      if (argc >= 3)
703ebc09c36SJim Ingham                      {
704ebc09c36SJim Ingham                          const std::string sub_command = args.GetArgumentAtIndex(0);
705ebc09c36SJim Ingham                          assert (sub_command.length() != 0);
706998255bfSGreg Clayton                          subcommand_obj_sp = cmd_obj->GetSubcommandSP (sub_command.c_str());
707*6e3d8e7fSEugene Zelenko                          if (subcommand_obj_sp)
708ebc09c36SJim Ingham                          {
709ebc09c36SJim Ingham                              sub_cmd_obj = subcommand_obj_sp.get();
710ebc09c36SJim Ingham                              use_subcommand = true;
711ebc09c36SJim Ingham                              args.Shift();  // Shift the sub_command word off the argument vector.
712844d2303SCaroline Tice                              cmd_obj = sub_cmd_obj;
713ebc09c36SJim Ingham                          }
714ebc09c36SJim Ingham                          else
715ebc09c36SJim Ingham                          {
716f415eeb4SCaroline Tice                              result.AppendErrorWithFormat("'%s' is not a valid sub-command of '%s'.  "
717f415eeb4SCaroline Tice                                                           "Unable to create alias.\n",
718f415eeb4SCaroline Tice                                                           sub_command.c_str(), actual_command.c_str());
719ebc09c36SJim Ingham                              result.SetStatus (eReturnStatusFailed);
720ebc09c36SJim Ingham                              return false;
721ebc09c36SJim Ingham                          }
722ebc09c36SJim Ingham                      }
723ebc09c36SJim Ingham                  }
724ebc09c36SJim Ingham 
725ebc09c36SJim Ingham                  // Verify & handle any options/arguments passed to the alias command
726ebc09c36SJim Ingham 
727ebc09c36SJim Ingham                  if (args.GetArgumentCount () > 0)
728ebc09c36SJim Ingham                  {
729ca90c47eSCaroline Tice                     CommandObjectSP tmp_sp = m_interpreter.GetCommandSPExact (cmd_obj->GetCommandName(), false);
730ebc09c36SJim Ingham                     if (use_subcommand)
731ca90c47eSCaroline Tice                         tmp_sp = m_interpreter.GetCommandSPExact (sub_cmd_obj->GetCommandName(), false);
732ca90c47eSCaroline Tice 
733ca90c47eSCaroline Tice                     std::string args_string;
734ca90c47eSCaroline Tice                     args.GetCommandString (args_string);
735ca90c47eSCaroline Tice 
736ca90c47eSCaroline Tice                     if (!m_interpreter.ProcessAliasOptionsArgs (tmp_sp, args_string.c_str(), option_arg_vector_sp))
737ebc09c36SJim Ingham                     {
738ca90c47eSCaroline Tice                         result.AppendError ("Unable to create requested alias.\n");
739ca90c47eSCaroline Tice                         result.SetStatus (eReturnStatusFailed);
740e7941795SCaroline Tice                         return false;
741867b185dSCaroline Tice                     }
742867b185dSCaroline Tice                  }
743867b185dSCaroline Tice 
744ebc09c36SJim Ingham                  // Create the alias.
745ebc09c36SJim Ingham 
746a7015092SGreg Clayton                  if (m_interpreter.AliasExists (alias_command.c_str())
747a7015092SGreg Clayton                      || m_interpreter.UserCommandExists (alias_command.c_str()))
748ebc09c36SJim Ingham                  {
749a7015092SGreg Clayton                      OptionArgVectorSP tmp_option_arg_sp (m_interpreter.GetAliasOptions (alias_command.c_str()));
750*6e3d8e7fSEugene Zelenko                      if (tmp_option_arg_sp)
751ebc09c36SJim Ingham                      {
752*6e3d8e7fSEugene Zelenko                          if (option_arg_vector->empty())
753a7015092SGreg Clayton                              m_interpreter.RemoveAliasOptions (alias_command.c_str());
754ebc09c36SJim Ingham                      }
755ebc09c36SJim Ingham                      result.AppendWarningWithFormat ("Overwriting existing definition for '%s'.\n",
756ebc09c36SJim Ingham                                                      alias_command.c_str());
757ebc09c36SJim Ingham                  }
758ebc09c36SJim Ingham 
759ebc09c36SJim Ingham                  if (use_subcommand)
760a7015092SGreg Clayton                      m_interpreter.AddAlias (alias_command.c_str(), subcommand_obj_sp);
761ebc09c36SJim Ingham                  else
762a7015092SGreg Clayton                      m_interpreter.AddAlias (alias_command.c_str(), command_obj_sp);
763*6e3d8e7fSEugene Zelenko                  if (!option_arg_vector->empty())
764a7015092SGreg Clayton                      m_interpreter.AddOrReplaceAliasOptions (alias_command.c_str(), option_arg_vector_sp);
765ebc09c36SJim Ingham                  result.SetStatus (eReturnStatusSuccessFinishNoResult);
766ebc09c36SJim Ingham              }
767ebc09c36SJim Ingham              else
768ebc09c36SJim Ingham              {
769ebc09c36SJim Ingham                  result.AppendErrorWithFormat ("'%s' is not an existing command.\n", actual_command.c_str());
770ebc09c36SJim Ingham                  result.SetStatus (eReturnStatusFailed);
771e7941795SCaroline Tice                  return false;
772ebc09c36SJim Ingham              }
773ebc09c36SJim Ingham         }
774ebc09c36SJim Ingham 
775ebc09c36SJim Ingham         return result.Succeeded();
776ebc09c36SJim Ingham     }
777ebc09c36SJim Ingham };
778ebc09c36SJim Ingham 
779ebc09c36SJim Ingham #pragma mark CommandObjectCommandsUnalias
780ebc09c36SJim Ingham //-------------------------------------------------------------------------
781ebc09c36SJim Ingham // CommandObjectCommandsUnalias
782ebc09c36SJim Ingham //-------------------------------------------------------------------------
783ebc09c36SJim Ingham 
7845a988416SJim Ingham class CommandObjectCommandsUnalias : public CommandObjectParsed
785ebc09c36SJim Ingham {
786ebc09c36SJim Ingham public:
787a7015092SGreg Clayton     CommandObjectCommandsUnalias (CommandInterpreter &interpreter) :
7885a988416SJim Ingham         CommandObjectParsed(interpreter,
7890e5e5a79SGreg Clayton                             "command unalias",
79086ddae50SCaroline Tice                             "Allow the user to remove/delete a user-defined command abbreviation.",
791*6e3d8e7fSEugene Zelenko                             nullptr)
792ebc09c36SJim Ingham     {
793405fe67fSCaroline Tice         CommandArgumentEntry arg;
794405fe67fSCaroline Tice         CommandArgumentData alias_arg;
795405fe67fSCaroline Tice 
796405fe67fSCaroline Tice         // Define the first (and only) variant of this arg.
797405fe67fSCaroline Tice         alias_arg.arg_type = eArgTypeAliasName;
798405fe67fSCaroline Tice         alias_arg.arg_repetition = eArgRepeatPlain;
799405fe67fSCaroline Tice 
800405fe67fSCaroline Tice         // There is only one variant this argument could be; put it into the argument entry.
801405fe67fSCaroline Tice         arg.push_back (alias_arg);
802405fe67fSCaroline Tice 
803405fe67fSCaroline Tice         // Push the data for the first argument into the m_arguments vector.
804405fe67fSCaroline Tice         m_arguments.push_back (arg);
805ebc09c36SJim Ingham     }
806ebc09c36SJim Ingham 
807*6e3d8e7fSEugene Zelenko     ~CommandObjectCommandsUnalias() override = default;
808ebc09c36SJim Ingham 
8095a988416SJim Ingham protected:
810ebc09c36SJim Ingham     bool
81113d21e9aSBruce Mitchener     DoExecute (Args& args, CommandReturnObject &result) override
812ebc09c36SJim Ingham     {
813ebc09c36SJim Ingham         CommandObject::CommandMap::iterator pos;
814ebc09c36SJim Ingham         CommandObject *cmd_obj;
815ebc09c36SJim Ingham 
816ebc09c36SJim Ingham         if (args.GetArgumentCount() != 0)
817ebc09c36SJim Ingham         {
818ebc09c36SJim Ingham             const char *command_name = args.GetArgumentAtIndex(0);
819a7015092SGreg Clayton             cmd_obj = m_interpreter.GetCommandObject(command_name);
820ebc09c36SJim Ingham             if (cmd_obj)
821ebc09c36SJim Ingham             {
822a7015092SGreg Clayton                 if (m_interpreter.CommandExists (command_name))
823ebc09c36SJim Ingham                 {
824b547278cSGreg Clayton                     if (cmd_obj->IsRemovable())
825b547278cSGreg Clayton                     {
826b547278cSGreg Clayton                         result.AppendErrorWithFormat ("'%s' is not an alias, it is a debugger command which can be removed using the 'command delete' command.\n",
827b547278cSGreg Clayton                                                       command_name);
828b547278cSGreg Clayton                     }
829b547278cSGreg Clayton                     else
830b547278cSGreg Clayton                     {
831ebc09c36SJim Ingham                         result.AppendErrorWithFormat ("'%s' is a permanent debugger command and cannot be removed.\n",
832ebc09c36SJim Ingham                                                       command_name);
833b547278cSGreg Clayton                     }
834ebc09c36SJim Ingham                     result.SetStatus (eReturnStatusFailed);
835ebc09c36SJim Ingham                 }
836ebc09c36SJim Ingham                 else
837ebc09c36SJim Ingham                 {
838*6e3d8e7fSEugene Zelenko                     if (!m_interpreter.RemoveAlias(command_name))
839ebc09c36SJim Ingham                     {
840a7015092SGreg Clayton                         if (m_interpreter.AliasExists (command_name))
841ebc09c36SJim Ingham                             result.AppendErrorWithFormat ("Error occurred while attempting to unalias '%s'.\n",
842ebc09c36SJim Ingham                                                           command_name);
843ebc09c36SJim Ingham                         else
844ebc09c36SJim Ingham                             result.AppendErrorWithFormat ("'%s' is not an existing alias.\n", command_name);
845ebc09c36SJim Ingham                         result.SetStatus (eReturnStatusFailed);
846ebc09c36SJim Ingham                     }
847ebc09c36SJim Ingham                     else
848ebc09c36SJim Ingham                         result.SetStatus (eReturnStatusSuccessFinishNoResult);
849ebc09c36SJim Ingham                 }
850ebc09c36SJim Ingham             }
851ebc09c36SJim Ingham             else
852ebc09c36SJim Ingham             {
853ebc09c36SJim Ingham                 result.AppendErrorWithFormat ("'%s' is not a known command.\nTry 'help' to see a "
854ebc09c36SJim Ingham                                               "current list of commands.\n",
855ebc09c36SJim Ingham                                              command_name);
856ebc09c36SJim Ingham                 result.SetStatus (eReturnStatusFailed);
857ebc09c36SJim Ingham             }
858ebc09c36SJim Ingham         }
859ebc09c36SJim Ingham         else
860ebc09c36SJim Ingham         {
861ebc09c36SJim Ingham             result.AppendError ("must call 'unalias' with a valid alias");
862ebc09c36SJim Ingham             result.SetStatus (eReturnStatusFailed);
863ebc09c36SJim Ingham         }
864ebc09c36SJim Ingham 
865ebc09c36SJim Ingham         return result.Succeeded();
866ebc09c36SJim Ingham     }
867ebc09c36SJim Ingham };
868ebc09c36SJim Ingham 
869b547278cSGreg Clayton #pragma mark CommandObjectCommandsDelete
870b547278cSGreg Clayton //-------------------------------------------------------------------------
871b547278cSGreg Clayton // CommandObjectCommandsDelete
872b547278cSGreg Clayton //-------------------------------------------------------------------------
873b547278cSGreg Clayton 
874b547278cSGreg Clayton class CommandObjectCommandsDelete : public CommandObjectParsed
875b547278cSGreg Clayton {
876b547278cSGreg Clayton public:
877b547278cSGreg Clayton     CommandObjectCommandsDelete (CommandInterpreter &interpreter) :
878b547278cSGreg Clayton         CommandObjectParsed(interpreter,
879b547278cSGreg Clayton                             "command delete",
880b547278cSGreg Clayton                             "Allow the user to delete user-defined regular expression, python or multi-word commands.",
881*6e3d8e7fSEugene Zelenko                             nullptr)
882b547278cSGreg Clayton     {
883b547278cSGreg Clayton         CommandArgumentEntry arg;
884b547278cSGreg Clayton         CommandArgumentData alias_arg;
885b547278cSGreg Clayton 
886b547278cSGreg Clayton         // Define the first (and only) variant of this arg.
887b547278cSGreg Clayton         alias_arg.arg_type = eArgTypeCommandName;
888b547278cSGreg Clayton         alias_arg.arg_repetition = eArgRepeatPlain;
889b547278cSGreg Clayton 
890b547278cSGreg Clayton         // There is only one variant this argument could be; put it into the argument entry.
891b547278cSGreg Clayton         arg.push_back (alias_arg);
892b547278cSGreg Clayton 
893b547278cSGreg Clayton         // Push the data for the first argument into the m_arguments vector.
894b547278cSGreg Clayton         m_arguments.push_back (arg);
895b547278cSGreg Clayton     }
896b547278cSGreg Clayton 
897*6e3d8e7fSEugene Zelenko     ~CommandObjectCommandsDelete() override = default;
898b547278cSGreg Clayton 
899b547278cSGreg Clayton protected:
900b547278cSGreg Clayton     bool
90113d21e9aSBruce Mitchener     DoExecute (Args& args, CommandReturnObject &result) override
902b547278cSGreg Clayton     {
903b547278cSGreg Clayton         CommandObject::CommandMap::iterator pos;
904b547278cSGreg Clayton 
905b547278cSGreg Clayton         if (args.GetArgumentCount() != 0)
906b547278cSGreg Clayton         {
907b547278cSGreg Clayton             const char *command_name = args.GetArgumentAtIndex(0);
908b547278cSGreg Clayton             if (m_interpreter.CommandExists (command_name))
909b547278cSGreg Clayton             {
910b547278cSGreg Clayton                 if (m_interpreter.RemoveCommand (command_name))
911b547278cSGreg Clayton                 {
912b547278cSGreg Clayton                     result.SetStatus (eReturnStatusSuccessFinishNoResult);
913b547278cSGreg Clayton                 }
914b547278cSGreg Clayton                 else
915b547278cSGreg Clayton                 {
916b547278cSGreg Clayton                     result.AppendErrorWithFormat ("'%s' is a permanent debugger command and cannot be removed.\n",
917b547278cSGreg Clayton                                                   command_name);
918b547278cSGreg Clayton                     result.SetStatus (eReturnStatusFailed);
919b547278cSGreg Clayton                 }
920b547278cSGreg Clayton             }
921b547278cSGreg Clayton             else
922b547278cSGreg Clayton             {
923b547278cSGreg Clayton                 result.AppendErrorWithFormat ("'%s' is not a known command.\nTry 'help' to see a current list of commands.\n",
924b547278cSGreg Clayton                                               command_name);
925b547278cSGreg Clayton                 result.SetStatus (eReturnStatusFailed);
926b547278cSGreg Clayton             }
927b547278cSGreg Clayton         }
928b547278cSGreg Clayton         else
929b547278cSGreg Clayton         {
930b547278cSGreg Clayton             result.AppendErrorWithFormat ("must call '%s' with one or more valid user defined regular expression, python or multi-word command names", GetCommandName ());
931b547278cSGreg Clayton             result.SetStatus (eReturnStatusFailed);
932b547278cSGreg Clayton         }
933b547278cSGreg Clayton 
934b547278cSGreg Clayton         return result.Succeeded();
935b547278cSGreg Clayton     }
936b547278cSGreg Clayton };
937b547278cSGreg Clayton 
938de164aaaSGreg Clayton //-------------------------------------------------------------------------
939de164aaaSGreg Clayton // CommandObjectCommandsAddRegex
940de164aaaSGreg Clayton //-------------------------------------------------------------------------
9415a988416SJim Ingham #pragma mark CommandObjectCommandsAddRegex
942de164aaaSGreg Clayton 
94344d93782SGreg Clayton class CommandObjectCommandsAddRegex :
94444d93782SGreg Clayton     public CommandObjectParsed,
945ea508635SGreg Clayton     public IOHandlerDelegateMultiline
946de164aaaSGreg Clayton {
947de164aaaSGreg Clayton public:
948de164aaaSGreg Clayton     CommandObjectCommandsAddRegex (CommandInterpreter &interpreter) :
9495a988416SJim Ingham         CommandObjectParsed (interpreter,
9500e5e5a79SGreg Clayton                        "command regex",
951de164aaaSGreg Clayton                        "Allow the user to create a regular expression command.",
9520e5e5a79SGreg Clayton                        "command regex <cmd-name> [s/<regex>/<subst>/ ...]"),
953ea508635SGreg Clayton         IOHandlerDelegateMultiline ("", IOHandlerDelegate::Completion::LLDBCommand),
954de164aaaSGreg Clayton         m_options (interpreter)
955de164aaaSGreg Clayton     {
956ea671fbdSKate Stone         SetHelpLong(R"(
957ea671fbdSKate Stone )" "This command allows the user to create powerful regular expression commands \
958ea671fbdSKate Stone with substitutions. The regular expressions and substitutions are specified \
959ea671fbdSKate Stone using the regular expression substitution format of:" R"(
960ea671fbdSKate Stone 
961ea671fbdSKate Stone     s/<regex>/<subst>/
962ea671fbdSKate Stone 
963ea671fbdSKate Stone )" "<regex> is a regular expression that can use parenthesis to capture regular \
964ea671fbdSKate Stone expression input and substitute the captured matches in the output using %1 \
965ea671fbdSKate Stone for the first match, %2 for the second, and so on." R"(
966ea671fbdSKate Stone 
967ea671fbdSKate Stone )" "The regular expressions can all be specified on the command line if more than \
968ea671fbdSKate Stone one argument is provided. If just the command name is provided on the command \
969ea671fbdSKate Stone line, then the regular expressions and substitutions can be entered on separate \
970ea671fbdSKate Stone lines, followed by an empty line to terminate the command definition." R"(
971ea671fbdSKate Stone 
972ea671fbdSKate Stone EXAMPLES
973ea671fbdSKate Stone 
974ea671fbdSKate Stone )" "The following example will define a regular expression command named 'f' that \
975ea671fbdSKate Stone will call 'finish' if there are no arguments, or 'frame select <frame-idx>' if \
976ea671fbdSKate Stone a number follows 'f':" R"(
977ea671fbdSKate Stone 
978ea671fbdSKate Stone     (lldb) command regex f s/^$/finish/ 's/([0-9]+)/frame select %1/')"
9790e5e5a79SGreg Clayton         );
980de164aaaSGreg Clayton     }
981de164aaaSGreg Clayton 
982*6e3d8e7fSEugene Zelenko     ~CommandObjectCommandsAddRegex() override = default;
983de164aaaSGreg Clayton 
9845a988416SJim Ingham protected:
985ea508635SGreg Clayton     void
986ea508635SGreg Clayton     IOHandlerActivated (IOHandler &io_handler) override
98744d93782SGreg Clayton     {
98844d93782SGreg Clayton         StreamFileSP output_sp(io_handler.GetOutputStreamFile());
98944d93782SGreg Clayton         if (output_sp)
99044d93782SGreg Clayton         {
99144d93782SGreg 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");
99244d93782SGreg Clayton             output_sp->Flush();
99344d93782SGreg Clayton         }
99444d93782SGreg Clayton     }
99544d93782SGreg Clayton 
996ea508635SGreg Clayton     void
997ea508635SGreg Clayton     IOHandlerInputComplete (IOHandler &io_handler, std::string &data) override
99844d93782SGreg Clayton     {
99944d93782SGreg Clayton         io_handler.SetIsDone(true);
1000*6e3d8e7fSEugene Zelenko         if (m_regex_cmd_ap)
100144d93782SGreg Clayton         {
100244d93782SGreg Clayton             StringList lines;
100344d93782SGreg Clayton             if (lines.SplitIntoLines (data))
100444d93782SGreg Clayton             {
100544d93782SGreg Clayton                 const size_t num_lines = lines.GetSize();
100644d93782SGreg Clayton                 bool check_only = false;
100744d93782SGreg Clayton                 for (size_t i=0; i<num_lines; ++i)
100844d93782SGreg Clayton                 {
100944d93782SGreg Clayton                     llvm::StringRef bytes_strref (lines[i]);
101044d93782SGreg Clayton                     Error error = AppendRegexSubstitution (bytes_strref, check_only);
101144d93782SGreg Clayton                     if (error.Fail())
101244d93782SGreg Clayton                     {
101344d93782SGreg Clayton                         if (!m_interpreter.GetDebugger().GetCommandInterpreter().GetBatchCommandMode())
101444d93782SGreg Clayton                         {
101544d93782SGreg Clayton                             StreamSP out_stream = m_interpreter.GetDebugger().GetAsyncOutputStream();
101644d93782SGreg Clayton                             out_stream->Printf("error: %s\n", error.AsCString());
101744d93782SGreg Clayton                         }
101844d93782SGreg Clayton                     }
101944d93782SGreg Clayton                 }
102044d93782SGreg Clayton             }
102144d93782SGreg Clayton             if (m_regex_cmd_ap->HasRegexEntries())
102244d93782SGreg Clayton             {
102344d93782SGreg Clayton                 CommandObjectSP cmd_sp (m_regex_cmd_ap.release());
102444d93782SGreg Clayton                 m_interpreter.AddCommand(cmd_sp->GetCommandName(), cmd_sp, true);
102544d93782SGreg Clayton             }
102644d93782SGreg Clayton         }
102744d93782SGreg Clayton     }
102844d93782SGreg Clayton 
1029de164aaaSGreg Clayton     bool
1030b0a1814fSEric Christopher     DoExecute (Args& command, CommandReturnObject &result) override
1031de164aaaSGreg Clayton     {
10325a988416SJim Ingham         const size_t argc = command.GetArgumentCount();
10330e5e5a79SGreg Clayton         if (argc == 0)
1034de164aaaSGreg Clayton         {
103569c12ccbSJason Molenda             result.AppendError ("usage: 'command regex <command-name> [s/<regex1>/<subst1>/ s/<regex2>/<subst2>/ ...]'\n");
10360e5e5a79SGreg Clayton             result.SetStatus (eReturnStatusFailed);
10370e5e5a79SGreg Clayton         }
10380e5e5a79SGreg Clayton         else
10390e5e5a79SGreg Clayton         {
10400e5e5a79SGreg Clayton             Error error;
10415a988416SJim Ingham             const char *name = command.GetArgumentAtIndex(0);
1042de164aaaSGreg Clayton             m_regex_cmd_ap.reset (new CommandObjectRegexCommand (m_interpreter,
1043de164aaaSGreg Clayton                                                                  name,
1044de164aaaSGreg Clayton                                                                  m_options.GetHelp (),
1045de164aaaSGreg Clayton                                                                  m_options.GetSyntax (),
1046b547278cSGreg Clayton                                                                  10,
1047b547278cSGreg Clayton                                                                  0,
1048b547278cSGreg Clayton                                                                  true));
10490e5e5a79SGreg Clayton 
10500e5e5a79SGreg Clayton             if (argc == 1)
10510e5e5a79SGreg Clayton             {
105244d93782SGreg Clayton                 Debugger &debugger = m_interpreter.GetDebugger();
1053e30f11d9SKate Stone                 bool color_prompt = debugger.GetUseColor();
105444d93782SGreg Clayton                 const bool multiple_lines = true; // Get multiple lines
105544d93782SGreg Clayton                 IOHandlerSP io_handler_sp(new IOHandlerEditline(debugger,
1056e30f11d9SKate Stone                                                                 IOHandler::Type::Other,
105773d80faaSGreg Clayton                                                                 "lldb-regex", // Name of input reader for history
1058ea508635SGreg Clayton                                                                 "> ",         // Prompt
1059*6e3d8e7fSEugene Zelenko                                                                 nullptr,      // Continuation prompt
106044d93782SGreg Clayton                                                                 multiple_lines,
1061e30f11d9SKate Stone                                                                 color_prompt,
1062f6913cd7SGreg Clayton                                                                 0,            // Don't show line numbers
106344d93782SGreg Clayton                                                                 *this));
106444d93782SGreg Clayton 
106544d93782SGreg Clayton                 if (io_handler_sp)
1066de164aaaSGreg Clayton                 {
106744d93782SGreg Clayton                     debugger.PushIOHandler(io_handler_sp);
1068de164aaaSGreg Clayton                     result.SetStatus (eReturnStatusSuccessFinishNoResult);
1069de164aaaSGreg Clayton                 }
1070de164aaaSGreg Clayton             }
1071de164aaaSGreg Clayton             else
1072de164aaaSGreg Clayton             {
10730e5e5a79SGreg Clayton                 for (size_t arg_idx = 1; arg_idx < argc; ++arg_idx)
10740e5e5a79SGreg Clayton                 {
10755a988416SJim Ingham                     llvm::StringRef arg_strref (command.GetArgumentAtIndex(arg_idx));
107644d93782SGreg Clayton                     bool check_only = false;
107744d93782SGreg Clayton                     error = AppendRegexSubstitution (arg_strref, check_only);
10780e5e5a79SGreg Clayton                     if (error.Fail())
10790e5e5a79SGreg Clayton                         break;
10800e5e5a79SGreg Clayton                 }
10810e5e5a79SGreg Clayton 
10820e5e5a79SGreg Clayton                 if (error.Success())
10830e5e5a79SGreg Clayton                 {
10840e5e5a79SGreg Clayton                     AddRegexCommandToInterpreter();
10850e5e5a79SGreg Clayton                 }
10860e5e5a79SGreg Clayton             }
10870e5e5a79SGreg Clayton             if (error.Fail())
10880e5e5a79SGreg Clayton             {
10890e5e5a79SGreg Clayton                 result.AppendError (error.AsCString());
1090de164aaaSGreg Clayton                 result.SetStatus (eReturnStatusFailed);
1091de164aaaSGreg Clayton             }
10920e5e5a79SGreg Clayton         }
10930e5e5a79SGreg Clayton 
1094de164aaaSGreg Clayton         return result.Succeeded();
1095de164aaaSGreg Clayton     }
1096de164aaaSGreg Clayton 
10970e5e5a79SGreg Clayton     Error
109844d93782SGreg Clayton     AppendRegexSubstitution (const llvm::StringRef &regex_sed, bool check_only)
1099de164aaaSGreg Clayton     {
11000e5e5a79SGreg Clayton         Error error;
11010e5e5a79SGreg Clayton 
1102*6e3d8e7fSEugene Zelenko         if (!m_regex_cmd_ap)
1103de164aaaSGreg Clayton         {
11040e5e5a79SGreg Clayton             error.SetErrorStringWithFormat("invalid regular expression command object for: '%.*s'",
11050e5e5a79SGreg Clayton                                            (int)regex_sed.size(),
11060e5e5a79SGreg Clayton                                            regex_sed.data());
11070e5e5a79SGreg Clayton             return error;
1108de164aaaSGreg Clayton         }
11090e5e5a79SGreg Clayton 
11100e5e5a79SGreg Clayton         size_t regex_sed_size = regex_sed.size();
11110e5e5a79SGreg Clayton 
11120e5e5a79SGreg Clayton         if (regex_sed_size <= 1)
11130e5e5a79SGreg Clayton         {
11140e5e5a79SGreg Clayton             error.SetErrorStringWithFormat("regular expression substitution string is too short: '%.*s'",
11150e5e5a79SGreg Clayton                                            (int)regex_sed.size(),
11160e5e5a79SGreg Clayton                                            regex_sed.data());
11170e5e5a79SGreg Clayton             return error;
11180e5e5a79SGreg Clayton         }
11190e5e5a79SGreg Clayton 
11200e5e5a79SGreg Clayton         if (regex_sed[0] != 's')
11210e5e5a79SGreg Clayton         {
11220e5e5a79SGreg Clayton             error.SetErrorStringWithFormat("regular expression substitution string doesn't start with 's': '%.*s'",
11230e5e5a79SGreg Clayton                                            (int)regex_sed.size(),
11240e5e5a79SGreg Clayton                                            regex_sed.data());
11250e5e5a79SGreg Clayton             return error;
11260e5e5a79SGreg Clayton         }
11270e5e5a79SGreg Clayton         const size_t first_separator_char_pos = 1;
11280e5e5a79SGreg Clayton         // use the char that follows 's' as the regex separator character
11290e5e5a79SGreg Clayton         // so we can have "s/<regex>/<subst>/" or "s|<regex>|<subst>|"
11300e5e5a79SGreg Clayton         const char separator_char = regex_sed[first_separator_char_pos];
11310e5e5a79SGreg Clayton         const size_t second_separator_char_pos = regex_sed.find (separator_char, first_separator_char_pos + 1);
11320e5e5a79SGreg Clayton 
11330e5e5a79SGreg Clayton         if (second_separator_char_pos == std::string::npos)
11340e5e5a79SGreg Clayton         {
1135ea508635SGreg Clayton             error.SetErrorStringWithFormat("missing second '%c' separator char after '%.*s' in '%.*s'",
11360e5e5a79SGreg Clayton                                            separator_char,
11370e5e5a79SGreg Clayton                                            (int)(regex_sed.size() - first_separator_char_pos - 1),
1138ea508635SGreg Clayton                                            regex_sed.data() + (first_separator_char_pos + 1),
1139ea508635SGreg Clayton                                            (int)regex_sed.size(),
1140ea508635SGreg Clayton                                            regex_sed.data());
11410e5e5a79SGreg Clayton             return error;
11420e5e5a79SGreg Clayton         }
11430e5e5a79SGreg Clayton 
11440e5e5a79SGreg Clayton         const size_t third_separator_char_pos = regex_sed.find (separator_char, second_separator_char_pos + 1);
11450e5e5a79SGreg Clayton 
11460e5e5a79SGreg Clayton         if (third_separator_char_pos == std::string::npos)
11470e5e5a79SGreg Clayton         {
1148ea508635SGreg Clayton             error.SetErrorStringWithFormat("missing third '%c' separator char after '%.*s' in '%.*s'",
11490e5e5a79SGreg Clayton                                            separator_char,
11500e5e5a79SGreg Clayton                                            (int)(regex_sed.size() - second_separator_char_pos - 1),
1151ea508635SGreg Clayton                                            regex_sed.data() + (second_separator_char_pos + 1),
1152ea508635SGreg Clayton                                            (int)regex_sed.size(),
1153ea508635SGreg Clayton                                            regex_sed.data());
11540e5e5a79SGreg Clayton             return error;
11550e5e5a79SGreg Clayton         }
11560e5e5a79SGreg Clayton 
11570e5e5a79SGreg Clayton         if (third_separator_char_pos != regex_sed_size - 1)
11580e5e5a79SGreg Clayton         {
11590e5e5a79SGreg Clayton             // Make sure that everything that follows the last regex
11600e5e5a79SGreg Clayton             // separator char
11610e5e5a79SGreg Clayton             if (regex_sed.find_first_not_of("\t\n\v\f\r ", third_separator_char_pos + 1) != std::string::npos)
11620e5e5a79SGreg Clayton             {
11630e5e5a79SGreg Clayton                 error.SetErrorStringWithFormat("extra data found after the '%.*s' regular expression substitution string: '%.*s'",
11640e5e5a79SGreg Clayton                                                (int)third_separator_char_pos + 1,
11650e5e5a79SGreg Clayton                                                regex_sed.data(),
11660e5e5a79SGreg Clayton                                                (int)(regex_sed.size() - third_separator_char_pos - 1),
11670e5e5a79SGreg Clayton                                                regex_sed.data() + (third_separator_char_pos + 1));
11680e5e5a79SGreg Clayton                 return error;
11690e5e5a79SGreg Clayton             }
11700e5e5a79SGreg Clayton         }
11710e5e5a79SGreg Clayton         else if (first_separator_char_pos + 1 == second_separator_char_pos)
11720e5e5a79SGreg Clayton         {
11730e5e5a79SGreg Clayton             error.SetErrorStringWithFormat("<regex> can't be empty in 's%c<regex>%c<subst>%c' string: '%.*s'",
11740e5e5a79SGreg Clayton                                            separator_char,
11750e5e5a79SGreg Clayton                                            separator_char,
11760e5e5a79SGreg Clayton                                            separator_char,
11770e5e5a79SGreg Clayton                                            (int)regex_sed.size(),
11780e5e5a79SGreg Clayton                                            regex_sed.data());
11790e5e5a79SGreg Clayton             return error;
11800e5e5a79SGreg Clayton         }
11810e5e5a79SGreg Clayton         else if (second_separator_char_pos + 1 == third_separator_char_pos)
11820e5e5a79SGreg Clayton         {
11830e5e5a79SGreg Clayton             error.SetErrorStringWithFormat("<subst> can't be empty in 's%c<regex>%c<subst>%c' string: '%.*s'",
11840e5e5a79SGreg Clayton                                            separator_char,
11850e5e5a79SGreg Clayton                                            separator_char,
11860e5e5a79SGreg Clayton                                            separator_char,
11870e5e5a79SGreg Clayton                                            (int)regex_sed.size(),
11880e5e5a79SGreg Clayton                                            regex_sed.data());
11890e5e5a79SGreg Clayton             return error;
11900e5e5a79SGreg Clayton         }
119144d93782SGreg Clayton 
1192*6e3d8e7fSEugene Zelenko         if (!check_only)
119344d93782SGreg Clayton         {
11940e5e5a79SGreg Clayton             std::string regex(regex_sed.substr(first_separator_char_pos + 1, second_separator_char_pos - first_separator_char_pos - 1));
11950e5e5a79SGreg Clayton             std::string subst(regex_sed.substr(second_separator_char_pos + 1, third_separator_char_pos - second_separator_char_pos - 1));
11960e5e5a79SGreg Clayton             m_regex_cmd_ap->AddRegexCommand (regex.c_str(),
11970e5e5a79SGreg Clayton                                              subst.c_str());
119844d93782SGreg Clayton         }
11990e5e5a79SGreg Clayton         return error;
1200de164aaaSGreg Clayton     }
1201de164aaaSGreg Clayton 
1202de164aaaSGreg Clayton     void
12030e5e5a79SGreg Clayton     AddRegexCommandToInterpreter()
1204de164aaaSGreg Clayton     {
1205*6e3d8e7fSEugene Zelenko         if (m_regex_cmd_ap)
1206de164aaaSGreg Clayton         {
1207de164aaaSGreg Clayton             if (m_regex_cmd_ap->HasRegexEntries())
1208de164aaaSGreg Clayton             {
1209de164aaaSGreg Clayton                 CommandObjectSP cmd_sp (m_regex_cmd_ap.release());
1210de164aaaSGreg Clayton                 m_interpreter.AddCommand(cmd_sp->GetCommandName(), cmd_sp, true);
1211de164aaaSGreg Clayton             }
1212de164aaaSGreg Clayton         }
1213de164aaaSGreg Clayton     }
1214de164aaaSGreg Clayton 
1215de164aaaSGreg Clayton private:
12167b0992d9SGreg Clayton     std::unique_ptr<CommandObjectRegexCommand> m_regex_cmd_ap;
1217de164aaaSGreg Clayton 
1218de164aaaSGreg Clayton      class CommandOptions : public Options
1219de164aaaSGreg Clayton      {
1220de164aaaSGreg Clayton      public:
1221de164aaaSGreg Clayton          CommandOptions (CommandInterpreter &interpreter) :
1222de164aaaSGreg Clayton             Options (interpreter)
1223de164aaaSGreg Clayton          {
1224de164aaaSGreg Clayton          }
1225de164aaaSGreg Clayton 
1226*6e3d8e7fSEugene Zelenko          ~CommandOptions() override = default;
1227de164aaaSGreg Clayton 
122813d21e9aSBruce Mitchener          Error
122913d21e9aSBruce Mitchener          SetOptionValue (uint32_t option_idx, const char *option_arg) override
1230de164aaaSGreg Clayton          {
1231de164aaaSGreg Clayton              Error error;
12323bcdfc0eSGreg Clayton              const int short_option = m_getopt_table[option_idx].val;
1233de164aaaSGreg Clayton 
1234de164aaaSGreg Clayton              switch (short_option)
1235de164aaaSGreg Clayton              {
1236de164aaaSGreg Clayton                  case 'h':
1237de164aaaSGreg Clayton                      m_help.assign (option_arg);
1238de164aaaSGreg Clayton                      break;
1239de164aaaSGreg Clayton                  case 's':
1240de164aaaSGreg Clayton                      m_syntax.assign (option_arg);
1241de164aaaSGreg Clayton                      break;
1242de164aaaSGreg Clayton                  default:
124386edbf41SGreg Clayton                      error.SetErrorStringWithFormat ("unrecognized option '%c'", short_option);
1244de164aaaSGreg Clayton                      break;
1245de164aaaSGreg Clayton              }
1246de164aaaSGreg Clayton 
1247de164aaaSGreg Clayton              return error;
1248de164aaaSGreg Clayton          }
1249de164aaaSGreg Clayton 
1250de164aaaSGreg Clayton          void
125113d21e9aSBruce Mitchener          OptionParsingStarting () override
1252de164aaaSGreg Clayton          {
1253de164aaaSGreg Clayton              m_help.clear();
1254de164aaaSGreg Clayton              m_syntax.clear();
1255de164aaaSGreg Clayton          }
1256de164aaaSGreg Clayton 
1257de164aaaSGreg Clayton          const OptionDefinition*
125813d21e9aSBruce Mitchener          GetDefinitions () override
1259de164aaaSGreg Clayton          {
1260de164aaaSGreg Clayton              return g_option_table;
1261de164aaaSGreg Clayton          }
1262de164aaaSGreg Clayton 
1263de164aaaSGreg Clayton          // Options table: Required for subclasses of Options.
1264de164aaaSGreg Clayton 
1265de164aaaSGreg Clayton          static OptionDefinition g_option_table[];
1266de164aaaSGreg Clayton 
1267de164aaaSGreg Clayton          const char *
1268de164aaaSGreg Clayton          GetHelp()
1269de164aaaSGreg Clayton          {
1270*6e3d8e7fSEugene Zelenko              return (m_help.empty() ? nullptr : m_help.c_str());
1271de164aaaSGreg Clayton          }
1272*6e3d8e7fSEugene Zelenko 
1273de164aaaSGreg Clayton          const char *
1274de164aaaSGreg Clayton          GetSyntax ()
1275de164aaaSGreg Clayton          {
1276*6e3d8e7fSEugene Zelenko              return (m_syntax.empty() ? nullptr : m_syntax.c_str());
1277de164aaaSGreg Clayton          }
1278*6e3d8e7fSEugene Zelenko 
1279de164aaaSGreg Clayton      protected:
1280*6e3d8e7fSEugene Zelenko          // Instance variables to hold the values for command options.
1281*6e3d8e7fSEugene Zelenko 
1282de164aaaSGreg Clayton          std::string m_help;
1283de164aaaSGreg Clayton          std::string m_syntax;
1284de164aaaSGreg Clayton      };
1285de164aaaSGreg Clayton 
1286b0a1814fSEric Christopher      Options *
1287b0a1814fSEric Christopher      GetOptions () override
1288de164aaaSGreg Clayton      {
1289de164aaaSGreg Clayton          return &m_options;
1290de164aaaSGreg Clayton      }
1291de164aaaSGreg Clayton 
12925a988416SJim Ingham      CommandOptions m_options;
1293de164aaaSGreg Clayton };
1294de164aaaSGreg Clayton 
1295de164aaaSGreg Clayton OptionDefinition
1296de164aaaSGreg Clayton CommandObjectCommandsAddRegex::CommandOptions::g_option_table[] =
1297de164aaaSGreg Clayton {
1298*6e3d8e7fSEugene Zelenko { LLDB_OPT_SET_1, false, "help"  , 'h', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeNone, "The help text to display for this command."},
1299*6e3d8e7fSEugene Zelenko { LLDB_OPT_SET_1, false, "syntax", 's', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeNone, "A syntax string showing the typical usage syntax."},
1300*6e3d8e7fSEugene Zelenko { 0             , false,  nullptr   , 0  , 0                , nullptr, nullptr, 0, eArgTypeNone, nullptr }
1301de164aaaSGreg Clayton };
1302de164aaaSGreg Clayton 
13035a988416SJim Ingham class CommandObjectPythonFunction : public CommandObjectRaw
1304223383edSEnrico Granata {
1305223383edSEnrico Granata public:
1306223383edSEnrico Granata     CommandObjectPythonFunction (CommandInterpreter &interpreter,
1307223383edSEnrico Granata                                  std::string name,
13080a305db7SEnrico Granata                                  std::string funct,
1309735152e3SEnrico Granata                                  std::string help,
13100a305db7SEnrico Granata                                  ScriptedCommandSynchronicity synch) :
13115a988416SJim Ingham         CommandObjectRaw(interpreter,
1312223383edSEnrico Granata                          name.c_str(),
1313*6e3d8e7fSEugene Zelenko                          nullptr,
1314*6e3d8e7fSEugene Zelenko                          nullptr),
13150a305db7SEnrico Granata         m_function_name(funct),
1316fac939e9SEnrico Granata         m_synchro(synch),
1317fac939e9SEnrico Granata         m_fetched_help_long(false)
1318223383edSEnrico Granata     {
1319735152e3SEnrico Granata         if (!help.empty())
1320735152e3SEnrico Granata             SetHelp(help.c_str());
1321735152e3SEnrico Granata         else
1322735152e3SEnrico Granata         {
1323735152e3SEnrico Granata             StreamString stream;
1324735152e3SEnrico Granata             stream.Printf("For more information run 'help %s'",name.c_str());
1325735152e3SEnrico Granata             SetHelp(stream.GetData());
1326735152e3SEnrico Granata         }
1327223383edSEnrico Granata     }
1328223383edSEnrico Granata 
1329*6e3d8e7fSEugene Zelenko     ~CommandObjectPythonFunction() override = default;
1330223383edSEnrico Granata 
133113d21e9aSBruce Mitchener     bool
133213d21e9aSBruce Mitchener     IsRemovable () const override
13335a988416SJim Ingham     {
13345a988416SJim Ingham         return true;
13355a988416SJim Ingham     }
13365a988416SJim Ingham 
13375a988416SJim Ingham     const std::string&
13385a988416SJim Ingham     GetFunctionName ()
13395a988416SJim Ingham     {
13405a988416SJim Ingham         return m_function_name;
13415a988416SJim Ingham     }
13425a988416SJim Ingham 
13435a988416SJim Ingham     ScriptedCommandSynchronicity
13445a988416SJim Ingham     GetSynchronicity ()
13455a988416SJim Ingham     {
13465a988416SJim Ingham         return m_synchro;
13475a988416SJim Ingham     }
13485a988416SJim Ingham 
134913d21e9aSBruce Mitchener     const char *
135013d21e9aSBruce Mitchener     GetHelpLong () override
1351fac939e9SEnrico Granata     {
1352fac939e9SEnrico Granata         if (!m_fetched_help_long)
1353fac939e9SEnrico Granata         {
1354fac939e9SEnrico Granata             ScriptInterpreter* scripter = m_interpreter.GetScriptInterpreter();
1355fac939e9SEnrico Granata             if (scripter)
1356fac939e9SEnrico Granata             {
1357fac939e9SEnrico Granata                 std::string docstring;
1358fac939e9SEnrico Granata                 m_fetched_help_long = scripter->GetDocumentationForItem(m_function_name.c_str(),docstring);
1359fac939e9SEnrico Granata                 if (!docstring.empty())
1360fac939e9SEnrico Granata                     SetHelpLong(docstring);
1361fac939e9SEnrico Granata             }
1362fac939e9SEnrico Granata         }
1363fac939e9SEnrico Granata         return CommandObjectRaw::GetHelpLong();
1364fac939e9SEnrico Granata     }
1365fac939e9SEnrico Granata 
13665a988416SJim Ingham protected:
136713d21e9aSBruce Mitchener     bool
136813d21e9aSBruce Mitchener     DoExecute (const char *raw_command_line, CommandReturnObject &result) override
1369223383edSEnrico Granata     {
1370223383edSEnrico Granata         ScriptInterpreter* scripter = m_interpreter.GetScriptInterpreter();
1371223383edSEnrico Granata 
1372223383edSEnrico Granata         Error error;
1373223383edSEnrico Granata 
137470f11f88SJim Ingham         result.SetStatus(eReturnStatusInvalid);
137570f11f88SJim Ingham 
1376*6e3d8e7fSEugene Zelenko         if (!scripter || !scripter->RunScriptBasedCommand(m_function_name.c_str(),
1377223383edSEnrico Granata                                                           raw_command_line,
13780a305db7SEnrico Granata                                                           m_synchro,
1379223383edSEnrico Granata                                                           result,
138006be059aSEnrico Granata                                                           error,
1381*6e3d8e7fSEugene Zelenko                                                           m_exe_ctx))
1382223383edSEnrico Granata         {
1383223383edSEnrico Granata             result.AppendError(error.AsCString());
1384223383edSEnrico Granata             result.SetStatus(eReturnStatusFailed);
1385223383edSEnrico Granata         }
1386223383edSEnrico Granata         else
138770f11f88SJim Ingham         {
138870f11f88SJim Ingham             // Don't change the status if the command already set it...
138970f11f88SJim Ingham             if (result.GetStatus() == eReturnStatusInvalid)
139070f11f88SJim Ingham             {
1391*6e3d8e7fSEugene Zelenko                 if (result.GetOutputData() == nullptr || result.GetOutputData()[0] == '\0')
1392223383edSEnrico Granata                     result.SetStatus(eReturnStatusSuccessFinishNoResult);
139370f11f88SJim Ingham                 else
139470f11f88SJim Ingham                     result.SetStatus(eReturnStatusSuccessFinishResult);
139570f11f88SJim Ingham             }
139670f11f88SJim Ingham         }
1397223383edSEnrico Granata 
1398223383edSEnrico Granata         return result.Succeeded();
1399223383edSEnrico Granata     }
1400223383edSEnrico Granata 
1401*6e3d8e7fSEugene Zelenko private:
1402*6e3d8e7fSEugene Zelenko     std::string m_function_name;
1403*6e3d8e7fSEugene Zelenko     ScriptedCommandSynchronicity m_synchro;
1404*6e3d8e7fSEugene Zelenko     bool m_fetched_help_long;
1405223383edSEnrico Granata };
1406223383edSEnrico Granata 
14079fe00e52SEnrico Granata class CommandObjectScriptingObject : public CommandObjectRaw
14089fe00e52SEnrico Granata {
14099fe00e52SEnrico Granata public:
14109fe00e52SEnrico Granata     CommandObjectScriptingObject (CommandInterpreter &interpreter,
14119fe00e52SEnrico Granata                                   std::string name,
14120641ca1aSZachary Turner                                   StructuredData::GenericSP cmd_obj_sp,
14139fe00e52SEnrico Granata                                   ScriptedCommandSynchronicity synch) :
14149fe00e52SEnrico Granata         CommandObjectRaw(interpreter,
14159fe00e52SEnrico Granata                          name.c_str(),
1416*6e3d8e7fSEugene Zelenko                          nullptr,
1417*6e3d8e7fSEugene Zelenko                          nullptr),
14189fe00e52SEnrico Granata         m_cmd_obj_sp(cmd_obj_sp),
14196f79bb2dSEnrico Granata         m_synchro(synch),
14206f79bb2dSEnrico Granata         m_fetched_help_short(false),
14216f79bb2dSEnrico Granata         m_fetched_help_long(false)
14229fe00e52SEnrico Granata     {
14239fe00e52SEnrico Granata         StreamString stream;
14249fe00e52SEnrico Granata         stream.Printf("For more information run 'help %s'",name.c_str());
14259fe00e52SEnrico Granata         SetHelp(stream.GetData());
1426e87764f2SEnrico Granata         if (ScriptInterpreter* scripter = m_interpreter.GetScriptInterpreter())
1427e87764f2SEnrico Granata             GetFlags().Set(scripter->GetFlagsForCommandObject(cmd_obj_sp));
14289fe00e52SEnrico Granata     }
14299fe00e52SEnrico Granata 
1430*6e3d8e7fSEugene Zelenko     ~CommandObjectScriptingObject() override = default;
14319fe00e52SEnrico Granata 
143213d21e9aSBruce Mitchener     bool
143313d21e9aSBruce Mitchener     IsRemovable () const override
14349fe00e52SEnrico Granata     {
14359fe00e52SEnrico Granata         return true;
14369fe00e52SEnrico Granata     }
14379fe00e52SEnrico Granata 
14380641ca1aSZachary Turner     StructuredData::GenericSP
14399fe00e52SEnrico Granata     GetImplementingObject ()
14409fe00e52SEnrico Granata     {
14419fe00e52SEnrico Granata         return m_cmd_obj_sp;
14429fe00e52SEnrico Granata     }
14439fe00e52SEnrico Granata 
14449fe00e52SEnrico Granata     ScriptedCommandSynchronicity
14459fe00e52SEnrico Granata     GetSynchronicity ()
14469fe00e52SEnrico Granata     {
14479fe00e52SEnrico Granata         return m_synchro;
14489fe00e52SEnrico Granata     }
14499fe00e52SEnrico Granata 
145013d21e9aSBruce Mitchener     const char *
145113d21e9aSBruce Mitchener     GetHelp () override
14526f79bb2dSEnrico Granata     {
14536f79bb2dSEnrico Granata         if (!m_fetched_help_short)
14546f79bb2dSEnrico Granata         {
14556f79bb2dSEnrico Granata             ScriptInterpreter* scripter = m_interpreter.GetScriptInterpreter();
14566f79bb2dSEnrico Granata             if (scripter)
14576f79bb2dSEnrico Granata             {
14586f79bb2dSEnrico Granata                 std::string docstring;
14596f79bb2dSEnrico Granata                 m_fetched_help_short = scripter->GetShortHelpForCommandObject(m_cmd_obj_sp,docstring);
14606f79bb2dSEnrico Granata                 if (!docstring.empty())
14616f79bb2dSEnrico Granata                     SetHelp(docstring);
14626f79bb2dSEnrico Granata             }
14636f79bb2dSEnrico Granata         }
14646f79bb2dSEnrico Granata         return CommandObjectRaw::GetHelp();
14656f79bb2dSEnrico Granata     }
14666f79bb2dSEnrico Granata 
146713d21e9aSBruce Mitchener     const char *
146813d21e9aSBruce Mitchener     GetHelpLong () override
14699fe00e52SEnrico Granata     {
14706f79bb2dSEnrico Granata         if (!m_fetched_help_long)
14716f79bb2dSEnrico Granata         {
14726f79bb2dSEnrico Granata             ScriptInterpreter* scripter = m_interpreter.GetScriptInterpreter();
14736f79bb2dSEnrico Granata             if (scripter)
14746f79bb2dSEnrico Granata             {
14756f79bb2dSEnrico Granata                 std::string docstring;
14766f79bb2dSEnrico Granata                 m_fetched_help_long = scripter->GetLongHelpForCommandObject(m_cmd_obj_sp,docstring);
14776f79bb2dSEnrico Granata                 if (!docstring.empty())
14786f79bb2dSEnrico Granata                     SetHelpLong(docstring);
14796f79bb2dSEnrico Granata             }
14806f79bb2dSEnrico Granata         }
14819fe00e52SEnrico Granata         return CommandObjectRaw::GetHelpLong();
14829fe00e52SEnrico Granata     }
14839fe00e52SEnrico Granata 
14849fe00e52SEnrico Granata protected:
148513d21e9aSBruce Mitchener     bool
148613d21e9aSBruce Mitchener     DoExecute (const char *raw_command_line, CommandReturnObject &result) override
14879fe00e52SEnrico Granata     {
14889fe00e52SEnrico Granata         ScriptInterpreter* scripter = m_interpreter.GetScriptInterpreter();
14899fe00e52SEnrico Granata 
14909fe00e52SEnrico Granata         Error error;
14919fe00e52SEnrico Granata 
14929fe00e52SEnrico Granata         result.SetStatus(eReturnStatusInvalid);
14939fe00e52SEnrico Granata 
1494*6e3d8e7fSEugene Zelenko         if (!scripter || !scripter->RunScriptBasedCommand(m_cmd_obj_sp,
14959fe00e52SEnrico Granata                                                           raw_command_line,
14969fe00e52SEnrico Granata                                                           m_synchro,
14979fe00e52SEnrico Granata                                                           result,
14989fe00e52SEnrico Granata                                                           error,
1499*6e3d8e7fSEugene Zelenko                                                           m_exe_ctx))
15009fe00e52SEnrico Granata         {
15019fe00e52SEnrico Granata             result.AppendError(error.AsCString());
15029fe00e52SEnrico Granata             result.SetStatus(eReturnStatusFailed);
15039fe00e52SEnrico Granata         }
15049fe00e52SEnrico Granata         else
15059fe00e52SEnrico Granata         {
15069fe00e52SEnrico Granata             // Don't change the status if the command already set it...
15079fe00e52SEnrico Granata             if (result.GetStatus() == eReturnStatusInvalid)
15089fe00e52SEnrico Granata             {
1509*6e3d8e7fSEugene Zelenko                 if (result.GetOutputData() == nullptr || result.GetOutputData()[0] == '\0')
15109fe00e52SEnrico Granata                     result.SetStatus(eReturnStatusSuccessFinishNoResult);
15119fe00e52SEnrico Granata                 else
15129fe00e52SEnrico Granata                     result.SetStatus(eReturnStatusSuccessFinishResult);
15139fe00e52SEnrico Granata             }
15149fe00e52SEnrico Granata         }
15159fe00e52SEnrico Granata 
15169fe00e52SEnrico Granata         return result.Succeeded();
15179fe00e52SEnrico Granata     }
15189fe00e52SEnrico Granata 
1519*6e3d8e7fSEugene Zelenko private:
1520*6e3d8e7fSEugene Zelenko     StructuredData::GenericSP m_cmd_obj_sp;
1521*6e3d8e7fSEugene Zelenko     ScriptedCommandSynchronicity m_synchro;
1522*6e3d8e7fSEugene Zelenko     bool m_fetched_help_short: 1;
1523*6e3d8e7fSEugene Zelenko     bool m_fetched_help_long: 1;
15249fe00e52SEnrico Granata };
15259fe00e52SEnrico Granata 
1526a9dbf432SEnrico Granata //-------------------------------------------------------------------------
1527a9dbf432SEnrico Granata // CommandObjectCommandsScriptImport
1528a9dbf432SEnrico Granata //-------------------------------------------------------------------------
1529a9dbf432SEnrico Granata 
15305a988416SJim Ingham class CommandObjectCommandsScriptImport : public CommandObjectParsed
1531a9dbf432SEnrico Granata {
15325a988416SJim Ingham public:
15335a988416SJim Ingham     CommandObjectCommandsScriptImport (CommandInterpreter &interpreter) :
15345a988416SJim Ingham         CommandObjectParsed(interpreter,
15355a988416SJim Ingham                             "command script import",
15365a988416SJim Ingham                             "Import a scripting module in LLDB.",
1537*6e3d8e7fSEugene Zelenko                             nullptr),
15385a988416SJim Ingham         m_options(interpreter)
15395a988416SJim Ingham     {
15405a988416SJim Ingham         CommandArgumentEntry arg1;
15415a988416SJim Ingham         CommandArgumentData cmd_arg;
15425a988416SJim Ingham 
15435a988416SJim Ingham         // Define the first (and only) variant of this arg.
15445a988416SJim Ingham         cmd_arg.arg_type = eArgTypeFilename;
15453b00e35bSEnrico Granata         cmd_arg.arg_repetition = eArgRepeatPlus;
15465a988416SJim Ingham 
15475a988416SJim Ingham         // There is only one variant this argument could be; put it into the argument entry.
15485a988416SJim Ingham         arg1.push_back (cmd_arg);
15495a988416SJim Ingham 
15505a988416SJim Ingham         // Push the data for the first argument into the m_arguments vector.
15515a988416SJim Ingham         m_arguments.push_back (arg1);
15525a988416SJim Ingham     }
15535a988416SJim Ingham 
1554*6e3d8e7fSEugene Zelenko     ~CommandObjectCommandsScriptImport() override = default;
15555a988416SJim Ingham 
155613d21e9aSBruce Mitchener     int
15575a988416SJim Ingham     HandleArgumentCompletion (Args &input,
15585a988416SJim Ingham                               int &cursor_index,
15595a988416SJim Ingham                               int &cursor_char_position,
15605a988416SJim Ingham                               OptionElementVector &opt_element_vector,
15615a988416SJim Ingham                               int match_start_point,
15625a988416SJim Ingham                               int max_return_elements,
15635a988416SJim Ingham                               bool &word_complete,
156413d21e9aSBruce Mitchener                               StringList &matches) override
15655a988416SJim Ingham     {
15665a988416SJim Ingham         std::string completion_str (input.GetArgumentAtIndex(cursor_index));
15675a988416SJim Ingham         completion_str.erase (cursor_char_position);
15685a988416SJim Ingham 
15695a988416SJim Ingham         CommandCompletions::InvokeCommonCompletionCallbacks(m_interpreter,
15705a988416SJim Ingham                                                             CommandCompletions::eDiskFileCompletion,
15715a988416SJim Ingham                                                             completion_str.c_str(),
15725a988416SJim Ingham                                                             match_start_point,
15735a988416SJim Ingham                                                             max_return_elements,
1574*6e3d8e7fSEugene Zelenko                                                             nullptr,
15755a988416SJim Ingham                                                             word_complete,
15765a988416SJim Ingham                                                             matches);
15775a988416SJim Ingham         return matches.GetSize();
15785a988416SJim Ingham     }
15795a988416SJim Ingham 
158013d21e9aSBruce Mitchener     Options *
158113d21e9aSBruce Mitchener     GetOptions () override
15825a988416SJim Ingham     {
15835a988416SJim Ingham         return &m_options;
15845a988416SJim Ingham     }
15855a988416SJim Ingham 
15865a988416SJim Ingham protected:
15870a305db7SEnrico Granata     class CommandOptions : public Options
15880a305db7SEnrico Granata     {
15890a305db7SEnrico Granata     public:
15900a305db7SEnrico Granata         CommandOptions (CommandInterpreter &interpreter) :
15910a305db7SEnrico Granata             Options (interpreter)
15920a305db7SEnrico Granata         {
15930a305db7SEnrico Granata         }
15940a305db7SEnrico Granata 
1595*6e3d8e7fSEugene Zelenko         ~CommandOptions() override = default;
15960a305db7SEnrico Granata 
159713d21e9aSBruce Mitchener         Error
159813d21e9aSBruce Mitchener         SetOptionValue (uint32_t option_idx, const char *option_arg) override
15990a305db7SEnrico Granata         {
16000a305db7SEnrico Granata             Error error;
16013bcdfc0eSGreg Clayton             const int short_option = m_getopt_table[option_idx].val;
16020a305db7SEnrico Granata 
16030a305db7SEnrico Granata             switch (short_option)
16040a305db7SEnrico Granata             {
16050a305db7SEnrico Granata                 case 'r':
16060a305db7SEnrico Granata                     m_allow_reload = true;
16070a305db7SEnrico Granata                     break;
16080a305db7SEnrico Granata                 default:
16090a305db7SEnrico Granata                     error.SetErrorStringWithFormat ("unrecognized option '%c'", short_option);
16100a305db7SEnrico Granata                     break;
16110a305db7SEnrico Granata             }
16120a305db7SEnrico Granata 
16130a305db7SEnrico Granata             return error;
16140a305db7SEnrico Granata         }
16150a305db7SEnrico Granata 
16160a305db7SEnrico Granata         void
161713d21e9aSBruce Mitchener         OptionParsingStarting () override
16180a305db7SEnrico Granata         {
1619e0c70f1bSEnrico Granata             m_allow_reload = true;
16200a305db7SEnrico Granata         }
16210a305db7SEnrico Granata 
16220a305db7SEnrico Granata         const OptionDefinition*
162313d21e9aSBruce Mitchener         GetDefinitions () override
16240a305db7SEnrico Granata         {
16250a305db7SEnrico Granata             return g_option_table;
16260a305db7SEnrico Granata         }
16270a305db7SEnrico Granata 
16280a305db7SEnrico Granata         // Options table: Required for subclasses of Options.
16290a305db7SEnrico Granata 
16300a305db7SEnrico Granata         static OptionDefinition g_option_table[];
16310a305db7SEnrico Granata 
16320a305db7SEnrico Granata         // Instance variables to hold the values for command options.
16330a305db7SEnrico Granata 
16340a305db7SEnrico Granata         bool m_allow_reload;
16350a305db7SEnrico Granata     };
16360a305db7SEnrico Granata 
1637a9dbf432SEnrico Granata     bool
163813d21e9aSBruce Mitchener     DoExecute (Args& command, CommandReturnObject &result) override
1639a9dbf432SEnrico Granata     {
1640a9dbf432SEnrico Granata         if (m_interpreter.GetDebugger().GetScriptLanguage() != lldb::eScriptLanguagePython)
1641a9dbf432SEnrico Granata         {
1642a9dbf432SEnrico Granata             result.AppendError ("only scripting language supported for module importing is currently Python");
1643a9dbf432SEnrico Granata             result.SetStatus (eReturnStatusFailed);
1644a9dbf432SEnrico Granata             return false;
1645a9dbf432SEnrico Granata         }
1646a9dbf432SEnrico Granata 
16475a988416SJim Ingham         size_t argc = command.GetArgumentCount();
16483b00e35bSEnrico Granata         if (0 == argc)
1649a9dbf432SEnrico Granata         {
16503b00e35bSEnrico Granata             result.AppendError("command script import needs one or more arguments");
1651a9dbf432SEnrico Granata             result.SetStatus (eReturnStatusFailed);
1652a9dbf432SEnrico Granata             return false;
1653a9dbf432SEnrico Granata         }
1654a9dbf432SEnrico Granata 
16550e978481SEd Maste         for (size_t i = 0;
16563b00e35bSEnrico Granata              i < argc;
16573b00e35bSEnrico Granata              i++)
16583b00e35bSEnrico Granata         {
16593b00e35bSEnrico Granata             std::string path = command.GetArgumentAtIndex(i);
1660a9dbf432SEnrico Granata             Error error;
1661a9dbf432SEnrico Granata 
1662c9d645d3SGreg Clayton             const bool init_session = true;
1663078551c7SEnrico Granata             // FIXME: this is necessary because CommandObject::CheckRequirements() assumes that
1664078551c7SEnrico Granata             // commands won't ever be recursively invoked, but it's actually possible to craft
1665078551c7SEnrico Granata             // a Python script that does other "command script imports" in __lldb_init_module
1666078551c7SEnrico Granata             // the real fix is to have recursive commands possible with a CommandInvocation object
1667078551c7SEnrico Granata             // separate from the CommandObject itself, so that recursive command invocations
1668078551c7SEnrico Granata             // won't stomp on each other (wrt to execution contents, options, and more)
1669078551c7SEnrico Granata             m_exe_ctx.Clear();
1670a9dbf432SEnrico Granata             if (m_interpreter.GetScriptInterpreter()->LoadScriptingModule(path.c_str(),
16710a305db7SEnrico Granata                                                                           m_options.m_allow_reload,
1672c9d645d3SGreg Clayton                                                                           init_session,
1673a9dbf432SEnrico Granata                                                                           error))
1674a9dbf432SEnrico Granata             {
1675a9dbf432SEnrico Granata                 result.SetStatus (eReturnStatusSuccessFinishNoResult);
1676a9dbf432SEnrico Granata             }
1677a9dbf432SEnrico Granata             else
1678a9dbf432SEnrico Granata             {
1679a9dbf432SEnrico Granata                 result.AppendErrorWithFormat("module importing failed: %s", error.AsCString());
1680a9dbf432SEnrico Granata                 result.SetStatus (eReturnStatusFailed);
1681a9dbf432SEnrico Granata             }
16823b00e35bSEnrico Granata         }
1683a9dbf432SEnrico Granata 
1684a9dbf432SEnrico Granata         return result.Succeeded();
1685a9dbf432SEnrico Granata     }
16860a305db7SEnrico Granata 
16875a988416SJim Ingham     CommandOptions m_options;
1688a9dbf432SEnrico Granata };
1689223383edSEnrico Granata 
16900a305db7SEnrico Granata OptionDefinition
16910a305db7SEnrico Granata CommandObjectCommandsScriptImport::CommandOptions::g_option_table[] =
16920a305db7SEnrico Granata {
1693*6e3d8e7fSEugene 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."},
1694*6e3d8e7fSEugene Zelenko     { 0, false, nullptr, 0, 0, nullptr, nullptr, 0, eArgTypeNone, nullptr }
16950a305db7SEnrico Granata };
16960a305db7SEnrico Granata 
1697223383edSEnrico Granata //-------------------------------------------------------------------------
1698223383edSEnrico Granata // CommandObjectCommandsScriptAdd
1699223383edSEnrico Granata //-------------------------------------------------------------------------
1700223383edSEnrico Granata 
170144d93782SGreg Clayton class CommandObjectCommandsScriptAdd :
170244d93782SGreg Clayton     public CommandObjectParsed,
170344d93782SGreg Clayton     public IOHandlerDelegateMultiline
1704223383edSEnrico Granata {
17055a988416SJim Ingham public:
17065a988416SJim Ingham     CommandObjectCommandsScriptAdd(CommandInterpreter &interpreter) :
17075a988416SJim Ingham         CommandObjectParsed(interpreter,
17085a988416SJim Ingham                             "command script add",
17095a988416SJim Ingham                             "Add a scripted function as an LLDB command.",
1710*6e3d8e7fSEugene Zelenko                             nullptr),
1711c3d874a5SGreg Clayton         IOHandlerDelegateMultiline ("DONE"),
17125a988416SJim Ingham         m_options (interpreter)
17135a988416SJim Ingham     {
17145a988416SJim Ingham         CommandArgumentEntry arg1;
17155a988416SJim Ingham         CommandArgumentData cmd_arg;
17165a988416SJim Ingham 
17175a988416SJim Ingham         // Define the first (and only) variant of this arg.
17185a988416SJim Ingham         cmd_arg.arg_type = eArgTypeCommandName;
17195a988416SJim Ingham         cmd_arg.arg_repetition = eArgRepeatPlain;
17205a988416SJim Ingham 
17215a988416SJim Ingham         // There is only one variant this argument could be; put it into the argument entry.
17225a988416SJim Ingham         arg1.push_back (cmd_arg);
17235a988416SJim Ingham 
17245a988416SJim Ingham         // Push the data for the first argument into the m_arguments vector.
17255a988416SJim Ingham         m_arguments.push_back (arg1);
17265a988416SJim Ingham     }
17275a988416SJim Ingham 
1728*6e3d8e7fSEugene Zelenko     ~CommandObjectCommandsScriptAdd() override = default;
17295a988416SJim Ingham 
173013d21e9aSBruce Mitchener     Options *
173113d21e9aSBruce Mitchener     GetOptions () override
17325a988416SJim Ingham     {
17335a988416SJim Ingham         return &m_options;
17345a988416SJim Ingham     }
17355a988416SJim Ingham 
17365a988416SJim Ingham protected:
1737223383edSEnrico Granata     class CommandOptions : public Options
1738223383edSEnrico Granata     {
1739223383edSEnrico Granata     public:
1740223383edSEnrico Granata         CommandOptions (CommandInterpreter &interpreter) :
17419fe00e52SEnrico Granata             Options (interpreter),
17429fe00e52SEnrico Granata             m_class_name(),
17439fe00e52SEnrico Granata             m_funct_name(),
17449fe00e52SEnrico Granata             m_short_help(),
17459fe00e52SEnrico Granata             m_synchronicity(eScriptedCommandSynchronicitySynchronous)
1746223383edSEnrico Granata         {
1747223383edSEnrico Granata         }
1748223383edSEnrico Granata 
1749*6e3d8e7fSEugene Zelenko         ~CommandOptions() override = default;
1750223383edSEnrico Granata 
175113d21e9aSBruce Mitchener         Error
175213d21e9aSBruce Mitchener         SetOptionValue (uint32_t option_idx, const char *option_arg) override
1753223383edSEnrico Granata         {
1754223383edSEnrico Granata             Error error;
17553bcdfc0eSGreg Clayton             const int short_option = m_getopt_table[option_idx].val;
1756223383edSEnrico Granata 
1757223383edSEnrico Granata             switch (short_option)
1758223383edSEnrico Granata             {
1759223383edSEnrico Granata                 case 'f':
1760735152e3SEnrico Granata                     if (option_arg)
1761735152e3SEnrico Granata                         m_funct_name.assign(option_arg);
1762735152e3SEnrico Granata                     break;
17639fe00e52SEnrico Granata                 case 'c':
17649fe00e52SEnrico Granata                     if (option_arg)
17659fe00e52SEnrico Granata                         m_class_name.assign(option_arg);
17669fe00e52SEnrico Granata                     break;
1767735152e3SEnrico Granata                 case 'h':
1768735152e3SEnrico Granata                     if (option_arg)
1769735152e3SEnrico Granata                         m_short_help.assign(option_arg);
1770223383edSEnrico Granata                     break;
17710a305db7SEnrico Granata                 case 's':
177244d93782SGreg Clayton                     m_synchronicity = (ScriptedCommandSynchronicity) Args::StringToOptionEnum(option_arg, g_option_table[option_idx].enum_values, 0, error);
17730a305db7SEnrico Granata                     if (!error.Success())
17740a305db7SEnrico Granata                         error.SetErrorStringWithFormat ("unrecognized value for synchronicity '%s'", option_arg);
17750a305db7SEnrico Granata                     break;
1776223383edSEnrico Granata                 default:
177786edbf41SGreg Clayton                     error.SetErrorStringWithFormat ("unrecognized option '%c'", short_option);
1778223383edSEnrico Granata                     break;
1779223383edSEnrico Granata             }
1780223383edSEnrico Granata 
1781223383edSEnrico Granata             return error;
1782223383edSEnrico Granata         }
1783223383edSEnrico Granata 
1784223383edSEnrico Granata         void
178513d21e9aSBruce Mitchener         OptionParsingStarting () override
1786223383edSEnrico Granata         {
17879fe00e52SEnrico Granata             m_class_name.clear();
1788735152e3SEnrico Granata             m_funct_name.clear();
1789735152e3SEnrico Granata             m_short_help.clear();
179044d93782SGreg Clayton             m_synchronicity = eScriptedCommandSynchronicitySynchronous;
1791223383edSEnrico Granata         }
1792223383edSEnrico Granata 
1793223383edSEnrico Granata         const OptionDefinition*
179413d21e9aSBruce Mitchener         GetDefinitions () override
1795223383edSEnrico Granata         {
1796223383edSEnrico Granata             return g_option_table;
1797223383edSEnrico Granata         }
1798223383edSEnrico Granata 
1799223383edSEnrico Granata         // Options table: Required for subclasses of Options.
1800223383edSEnrico Granata 
1801223383edSEnrico Granata         static OptionDefinition g_option_table[];
1802223383edSEnrico Granata 
1803223383edSEnrico Granata         // Instance variables to hold the values for command options.
1804223383edSEnrico Granata 
18059fe00e52SEnrico Granata         std::string m_class_name;
1806223383edSEnrico Granata         std::string m_funct_name;
1807735152e3SEnrico Granata         std::string m_short_help;
180844d93782SGreg Clayton         ScriptedCommandSynchronicity m_synchronicity;
1809223383edSEnrico Granata     };
1810223383edSEnrico Granata 
181113d21e9aSBruce Mitchener     void
181213d21e9aSBruce Mitchener     IOHandlerActivated (IOHandler &io_handler) override
1813223383edSEnrico Granata     {
181444d93782SGreg Clayton         StreamFileSP output_sp(io_handler.GetOutputStreamFile());
181544d93782SGreg Clayton         if (output_sp)
1816223383edSEnrico Granata         {
181744d93782SGreg Clayton             output_sp->PutCString(g_python_command_instructions);
181844d93782SGreg Clayton             output_sp->Flush();
1819223383edSEnrico Granata         }
1820223383edSEnrico Granata     }
1821223383edSEnrico Granata 
1822223383edSEnrico Granata 
182313d21e9aSBruce Mitchener     void
182413d21e9aSBruce Mitchener     IOHandlerInputComplete (IOHandler &io_handler, std::string &data) override
1825223383edSEnrico Granata     {
182644d93782SGreg Clayton         StreamFileSP error_sp = io_handler.GetErrorStreamFile();
182744d93782SGreg Clayton 
182844d93782SGreg Clayton         ScriptInterpreter *interpreter = m_interpreter.GetScriptInterpreter();
182944d93782SGreg Clayton         if (interpreter)
183044d93782SGreg Clayton         {
183144d93782SGreg Clayton 
183244d93782SGreg Clayton             StringList lines;
183344d93782SGreg Clayton             lines.SplitIntoLines(data);
183444d93782SGreg Clayton             if (lines.GetSize() > 0)
183544d93782SGreg Clayton             {
1836a73b7df7SEnrico Granata                 std::string funct_name_str;
183744d93782SGreg Clayton                 if (interpreter->GenerateScriptAliasFunction (lines, funct_name_str))
1838223383edSEnrico Granata                 {
1839a73b7df7SEnrico Granata                     if (funct_name_str.empty())
1840223383edSEnrico Granata                     {
184144d93782SGreg Clayton                         error_sp->Printf ("error: unable to obtain a function name, didn't add python command.\n");
184244d93782SGreg Clayton                         error_sp->Flush();
1843223383edSEnrico Granata                     }
184444d93782SGreg Clayton                     else
184544d93782SGreg Clayton                     {
1846223383edSEnrico Granata                         // everything should be fine now, let's add this alias
1847223383edSEnrico Granata 
1848223383edSEnrico Granata                         CommandObjectSP command_obj_sp(new CommandObjectPythonFunction (m_interpreter,
1849223383edSEnrico Granata                                                                                         m_cmd_name,
1850a73b7df7SEnrico Granata                                                                                         funct_name_str.c_str(),
1851735152e3SEnrico Granata                                                                                         m_short_help,
185244d93782SGreg Clayton                                                                                         m_synchronicity));
1853223383edSEnrico Granata 
18540a305db7SEnrico Granata                         if (!m_interpreter.AddUserCommand(m_cmd_name, command_obj_sp, true))
1855223383edSEnrico Granata                         {
185644d93782SGreg Clayton                             error_sp->Printf ("error: unable to add selected command, didn't add python command.\n");
185744d93782SGreg Clayton                             error_sp->Flush();
1858223383edSEnrico Granata                         }
1859223383edSEnrico Granata                     }
186044d93782SGreg Clayton                 }
186144d93782SGreg Clayton                 else
186244d93782SGreg Clayton                 {
186344d93782SGreg Clayton                     error_sp->Printf ("error: unable to create function, didn't add python command.\n");
186444d93782SGreg Clayton                     error_sp->Flush();
186544d93782SGreg Clayton                 }
186644d93782SGreg Clayton             }
186744d93782SGreg Clayton             else
186844d93782SGreg Clayton             {
186944d93782SGreg Clayton                 error_sp->Printf ("error: empty function, didn't add python command.\n");
187044d93782SGreg Clayton                 error_sp->Flush();
187144d93782SGreg Clayton             }
187244d93782SGreg Clayton         }
187344d93782SGreg Clayton         else
187444d93782SGreg Clayton         {
187544d93782SGreg Clayton             error_sp->Printf ("error: script interpreter missing, didn't add python command.\n");
187644d93782SGreg Clayton             error_sp->Flush();
187744d93782SGreg Clayton         }
187844d93782SGreg Clayton 
187944d93782SGreg Clayton         io_handler.SetIsDone(true);
188044d93782SGreg Clayton     }
1881223383edSEnrico Granata 
18825a988416SJim Ingham protected:
1883223383edSEnrico Granata     bool
188413d21e9aSBruce Mitchener     DoExecute (Args& command, CommandReturnObject &result) override
1885223383edSEnrico Granata     {
188699f0b8f9SEnrico Granata         if (m_interpreter.GetDebugger().GetScriptLanguage() != lldb::eScriptLanguagePython)
188799f0b8f9SEnrico Granata         {
188899f0b8f9SEnrico Granata             result.AppendError ("only scripting language supported for scripted commands is currently Python");
188999f0b8f9SEnrico Granata             result.SetStatus (eReturnStatusFailed);
189099f0b8f9SEnrico Granata             return false;
189199f0b8f9SEnrico Granata         }
189299f0b8f9SEnrico Granata 
18935a988416SJim Ingham         size_t argc = command.GetArgumentCount();
1894223383edSEnrico Granata 
1895223383edSEnrico Granata         if (argc != 1)
1896223383edSEnrico Granata         {
1897223383edSEnrico Granata             result.AppendError ("'command script add' requires one argument");
1898223383edSEnrico Granata             result.SetStatus (eReturnStatusFailed);
1899223383edSEnrico Granata             return false;
1900223383edSEnrico Granata         }
1901223383edSEnrico Granata 
1902735152e3SEnrico Granata         // Store the options in case we get multi-line input
190344d93782SGreg Clayton         m_cmd_name = command.GetArgumentAtIndex(0);
1904735152e3SEnrico Granata         m_short_help.assign(m_options.m_short_help);
190544d93782SGreg Clayton         m_synchronicity = m_options.m_synchronicity;
1906223383edSEnrico Granata 
19079fe00e52SEnrico Granata         if (m_options.m_class_name.empty())
19089fe00e52SEnrico Granata         {
1909223383edSEnrico Granata             if (m_options.m_funct_name.empty())
1910223383edSEnrico Granata             {
191144d93782SGreg Clayton                 m_interpreter.GetPythonCommandsFromIOHandler("     ",  // Prompt
191244d93782SGreg Clayton                                                              *this,    // IOHandlerDelegate
191344d93782SGreg Clayton                                                              true,     // Run IOHandler in async mode
1914*6e3d8e7fSEugene Zelenko                                                              nullptr); // Baton for the "io_handler" that will be passed back into our IOHandlerDelegate functions
1915223383edSEnrico Granata             }
1916223383edSEnrico Granata             else
1917223383edSEnrico Granata             {
19180a305db7SEnrico Granata                 CommandObjectSP new_cmd(new CommandObjectPythonFunction(m_interpreter,
191944d93782SGreg Clayton                                                                         m_cmd_name,
19200a305db7SEnrico Granata                                                                         m_options.m_funct_name,
1921735152e3SEnrico Granata                                                                         m_options.m_short_help,
192244d93782SGreg Clayton                                                                         m_synchronicity));
192344d93782SGreg Clayton                 if (m_interpreter.AddUserCommand(m_cmd_name, new_cmd, true))
1924223383edSEnrico Granata                 {
1925223383edSEnrico Granata                     result.SetStatus (eReturnStatusSuccessFinishNoResult);
1926223383edSEnrico Granata                 }
1927223383edSEnrico Granata                 else
1928223383edSEnrico Granata                 {
1929223383edSEnrico Granata                     result.AppendError("cannot add command");
1930223383edSEnrico Granata                     result.SetStatus (eReturnStatusFailed);
1931223383edSEnrico Granata                 }
1932223383edSEnrico Granata             }
19339fe00e52SEnrico Granata         }
19349fe00e52SEnrico Granata         else
19359fe00e52SEnrico Granata         {
19369fe00e52SEnrico Granata             ScriptInterpreter *interpreter = GetCommandInterpreter().GetScriptInterpreter();
19379fe00e52SEnrico Granata             if (!interpreter)
19389fe00e52SEnrico Granata             {
19399fe00e52SEnrico Granata                 result.AppendError("cannot find ScriptInterpreter");
19409fe00e52SEnrico Granata                 result.SetStatus(eReturnStatusFailed);
19419fe00e52SEnrico Granata                 return false;
19429fe00e52SEnrico Granata             }
19439fe00e52SEnrico Granata 
19449fe00e52SEnrico Granata             auto cmd_obj_sp = interpreter->CreateScriptCommandObject(m_options.m_class_name.c_str());
19459fe00e52SEnrico Granata             if (!cmd_obj_sp)
19469fe00e52SEnrico Granata             {
19479fe00e52SEnrico Granata                 result.AppendError("cannot create helper object");
19489fe00e52SEnrico Granata                 result.SetStatus(eReturnStatusFailed);
19499fe00e52SEnrico Granata                 return false;
19509fe00e52SEnrico Granata             }
19519fe00e52SEnrico Granata 
19529fe00e52SEnrico Granata             CommandObjectSP new_cmd(new CommandObjectScriptingObject(m_interpreter,
19539fe00e52SEnrico Granata                                                                      m_cmd_name,
19549fe00e52SEnrico Granata                                                                      cmd_obj_sp,
19559fe00e52SEnrico Granata                                                                      m_synchronicity));
19569fe00e52SEnrico Granata             if (m_interpreter.AddUserCommand(m_cmd_name, new_cmd, true))
19579fe00e52SEnrico Granata             {
19589fe00e52SEnrico Granata                 result.SetStatus (eReturnStatusSuccessFinishNoResult);
19599fe00e52SEnrico Granata             }
19609fe00e52SEnrico Granata             else
19619fe00e52SEnrico Granata             {
19629fe00e52SEnrico Granata                 result.AppendError("cannot add command");
19639fe00e52SEnrico Granata                 result.SetStatus (eReturnStatusFailed);
19649fe00e52SEnrico Granata             }
19659fe00e52SEnrico Granata         }
1966223383edSEnrico Granata 
1967223383edSEnrico Granata         return result.Succeeded();
1968223383edSEnrico Granata     }
19695a988416SJim Ingham 
19705a988416SJim Ingham     CommandOptions m_options;
197144d93782SGreg Clayton     std::string m_cmd_name;
1972735152e3SEnrico Granata     std::string m_short_help;
197344d93782SGreg Clayton     ScriptedCommandSynchronicity m_synchronicity;
1974223383edSEnrico Granata };
1975223383edSEnrico Granata 
19760a305db7SEnrico Granata static OptionEnumValueElement g_script_synchro_type[] =
19770a305db7SEnrico Granata {
19780a305db7SEnrico Granata     { eScriptedCommandSynchronicitySynchronous,      "synchronous",       "Run synchronous"},
19790a305db7SEnrico Granata     { eScriptedCommandSynchronicityAsynchronous,     "asynchronous",      "Run asynchronous"},
19800a305db7SEnrico Granata     { eScriptedCommandSynchronicityCurrentValue,     "current",           "Do not alter current setting"},
1981*6e3d8e7fSEugene Zelenko     { 0, nullptr, nullptr }
19820a305db7SEnrico Granata };
19830a305db7SEnrico Granata 
1984223383edSEnrico Granata OptionDefinition
1985223383edSEnrico Granata CommandObjectCommandsScriptAdd::CommandOptions::g_option_table[] =
1986223383edSEnrico Granata {
1987*6e3d8e7fSEugene 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."},
1988*6e3d8e7fSEugene 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."},
1989*6e3d8e7fSEugene Zelenko     { LLDB_OPT_SET_1, false, "help"  , 'h', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeHelpText, "The help text to display for this command."},
1990*6e3d8e7fSEugene 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."},
1991*6e3d8e7fSEugene Zelenko     { 0, false, nullptr, 0, 0, nullptr, nullptr, 0, eArgTypeNone, nullptr }
1992223383edSEnrico Granata };
1993223383edSEnrico Granata 
1994223383edSEnrico Granata //-------------------------------------------------------------------------
1995223383edSEnrico Granata // CommandObjectCommandsScriptList
1996223383edSEnrico Granata //-------------------------------------------------------------------------
1997223383edSEnrico Granata 
19985a988416SJim Ingham class CommandObjectCommandsScriptList : public CommandObjectParsed
1999223383edSEnrico Granata {
2000223383edSEnrico Granata public:
2001223383edSEnrico Granata     CommandObjectCommandsScriptList(CommandInterpreter &interpreter) :
20025a988416SJim Ingham         CommandObjectParsed(interpreter,
2003223383edSEnrico Granata                             "command script list",
2004223383edSEnrico Granata                             "List defined scripted commands.",
2005*6e3d8e7fSEugene Zelenko                             nullptr)
2006223383edSEnrico Granata     {
2007223383edSEnrico Granata     }
2008223383edSEnrico Granata 
2009*6e3d8e7fSEugene Zelenko     ~CommandObjectCommandsScriptList() override = default;
2010223383edSEnrico Granata 
2011223383edSEnrico Granata     bool
201213d21e9aSBruce Mitchener     DoExecute (Args& command, CommandReturnObject &result) override
2013223383edSEnrico Granata     {
2014223383edSEnrico Granata         m_interpreter.GetHelp(result,
2015223383edSEnrico Granata                               CommandInterpreter::eCommandTypesUserDef);
2016223383edSEnrico Granata 
2017223383edSEnrico Granata         result.SetStatus (eReturnStatusSuccessFinishResult);
2018223383edSEnrico Granata 
2019223383edSEnrico Granata         return true;
2020223383edSEnrico Granata     }
2021223383edSEnrico Granata };
2022223383edSEnrico Granata 
2023223383edSEnrico Granata //-------------------------------------------------------------------------
2024223383edSEnrico Granata // CommandObjectCommandsScriptClear
2025223383edSEnrico Granata //-------------------------------------------------------------------------
2026223383edSEnrico Granata 
20275a988416SJim Ingham class CommandObjectCommandsScriptClear : public CommandObjectParsed
2028223383edSEnrico Granata {
2029223383edSEnrico Granata public:
2030223383edSEnrico Granata     CommandObjectCommandsScriptClear(CommandInterpreter &interpreter) :
20315a988416SJim Ingham         CommandObjectParsed(interpreter,
2032223383edSEnrico Granata                             "command script clear",
2033223383edSEnrico Granata                             "Delete all scripted commands.",
2034*6e3d8e7fSEugene Zelenko                             nullptr)
2035223383edSEnrico Granata     {
2036223383edSEnrico Granata     }
2037223383edSEnrico Granata 
2038*6e3d8e7fSEugene Zelenko     ~CommandObjectCommandsScriptClear() override = default;
2039223383edSEnrico Granata 
20405a988416SJim Ingham protected:
2041223383edSEnrico Granata     bool
204213d21e9aSBruce Mitchener     DoExecute (Args& command, CommandReturnObject &result) override
2043223383edSEnrico Granata     {
2044223383edSEnrico Granata         m_interpreter.RemoveAllUser();
2045223383edSEnrico Granata 
2046223383edSEnrico Granata         result.SetStatus (eReturnStatusSuccessFinishResult);
2047223383edSEnrico Granata 
2048223383edSEnrico Granata         return true;
2049223383edSEnrico Granata     }
2050223383edSEnrico Granata };
2051223383edSEnrico Granata 
2052223383edSEnrico Granata //-------------------------------------------------------------------------
2053223383edSEnrico Granata // CommandObjectCommandsScriptDelete
2054223383edSEnrico Granata //-------------------------------------------------------------------------
2055223383edSEnrico Granata 
20565a988416SJim Ingham class CommandObjectCommandsScriptDelete : public CommandObjectParsed
2057223383edSEnrico Granata {
2058223383edSEnrico Granata public:
2059223383edSEnrico Granata     CommandObjectCommandsScriptDelete(CommandInterpreter &interpreter) :
20605a988416SJim Ingham         CommandObjectParsed(interpreter,
2061223383edSEnrico Granata                             "command script delete",
2062223383edSEnrico Granata                             "Delete a scripted command.",
2063*6e3d8e7fSEugene Zelenko                             nullptr)
2064223383edSEnrico Granata     {
2065223383edSEnrico Granata         CommandArgumentEntry arg1;
2066223383edSEnrico Granata         CommandArgumentData cmd_arg;
2067223383edSEnrico Granata 
2068223383edSEnrico Granata         // Define the first (and only) variant of this arg.
2069223383edSEnrico Granata         cmd_arg.arg_type = eArgTypeCommandName;
2070223383edSEnrico Granata         cmd_arg.arg_repetition = eArgRepeatPlain;
2071223383edSEnrico Granata 
2072223383edSEnrico Granata         // There is only one variant this argument could be; put it into the argument entry.
2073223383edSEnrico Granata         arg1.push_back (cmd_arg);
2074223383edSEnrico Granata 
2075223383edSEnrico Granata         // Push the data for the first argument into the m_arguments vector.
2076223383edSEnrico Granata         m_arguments.push_back (arg1);
2077223383edSEnrico Granata     }
2078223383edSEnrico Granata 
2079*6e3d8e7fSEugene Zelenko     ~CommandObjectCommandsScriptDelete() override = default;
2080223383edSEnrico Granata 
20815a988416SJim Ingham protected:
2082223383edSEnrico Granata     bool
208313d21e9aSBruce Mitchener     DoExecute (Args& command, CommandReturnObject &result) override
2084223383edSEnrico Granata     {
2085223383edSEnrico Granata 
20865a988416SJim Ingham         size_t argc = command.GetArgumentCount();
2087223383edSEnrico Granata 
2088223383edSEnrico Granata         if (argc != 1)
2089223383edSEnrico Granata         {
2090223383edSEnrico Granata             result.AppendError ("'command script delete' requires one argument");
2091223383edSEnrico Granata             result.SetStatus (eReturnStatusFailed);
2092223383edSEnrico Granata             return false;
2093223383edSEnrico Granata         }
2094223383edSEnrico Granata 
20955a988416SJim Ingham         const char* cmd_name = command.GetArgumentAtIndex(0);
2096223383edSEnrico Granata 
2097223383edSEnrico Granata         if (cmd_name && *cmd_name && m_interpreter.HasUserCommands() && m_interpreter.UserCommandExists(cmd_name))
2098223383edSEnrico Granata         {
2099223383edSEnrico Granata             m_interpreter.RemoveUser(cmd_name);
2100223383edSEnrico Granata             result.SetStatus (eReturnStatusSuccessFinishResult);
2101223383edSEnrico Granata         }
2102223383edSEnrico Granata         else
2103223383edSEnrico Granata         {
2104223383edSEnrico Granata             result.AppendErrorWithFormat ("command %s not found", cmd_name);
2105223383edSEnrico Granata             result.SetStatus (eReturnStatusFailed);
2106223383edSEnrico Granata         }
2107223383edSEnrico Granata 
2108223383edSEnrico Granata         return result.Succeeded();
2109223383edSEnrico Granata     }
2110223383edSEnrico Granata };
2111223383edSEnrico Granata 
2112223383edSEnrico Granata #pragma mark CommandObjectMultiwordCommandsScript
2113223383edSEnrico Granata 
2114223383edSEnrico Granata //-------------------------------------------------------------------------
2115223383edSEnrico Granata // CommandObjectMultiwordCommandsScript
2116223383edSEnrico Granata //-------------------------------------------------------------------------
2117223383edSEnrico Granata 
2118223383edSEnrico Granata class CommandObjectMultiwordCommandsScript : public CommandObjectMultiword
2119223383edSEnrico Granata {
2120223383edSEnrico Granata public:
2121223383edSEnrico Granata     CommandObjectMultiwordCommandsScript (CommandInterpreter &interpreter) :
2122223383edSEnrico Granata     CommandObjectMultiword (interpreter,
2123223383edSEnrico Granata                             "command script",
2124223383edSEnrico Granata                             "A set of commands for managing or customizing script commands.",
2125223383edSEnrico Granata                             "command script <subcommand> [<subcommand-options>]")
2126223383edSEnrico Granata     {
2127223383edSEnrico Granata         LoadSubCommand ("add",    CommandObjectSP (new CommandObjectCommandsScriptAdd (interpreter)));
2128223383edSEnrico Granata         LoadSubCommand ("delete", CommandObjectSP (new CommandObjectCommandsScriptDelete (interpreter)));
2129223383edSEnrico Granata         LoadSubCommand ("clear",  CommandObjectSP (new CommandObjectCommandsScriptClear (interpreter)));
2130223383edSEnrico Granata         LoadSubCommand ("list",   CommandObjectSP (new CommandObjectCommandsScriptList (interpreter)));
2131a9dbf432SEnrico Granata         LoadSubCommand ("import", CommandObjectSP (new CommandObjectCommandsScriptImport (interpreter)));
2132223383edSEnrico Granata     }
2133223383edSEnrico Granata 
2134*6e3d8e7fSEugene Zelenko     ~CommandObjectMultiwordCommandsScript() override = default;
2135223383edSEnrico Granata };
2136223383edSEnrico Granata 
2137ebc09c36SJim Ingham #pragma mark CommandObjectMultiwordCommands
2138ebc09c36SJim Ingham 
2139ebc09c36SJim Ingham //-------------------------------------------------------------------------
2140ebc09c36SJim Ingham // CommandObjectMultiwordCommands
2141ebc09c36SJim Ingham //-------------------------------------------------------------------------
2142ebc09c36SJim Ingham 
2143ebc09c36SJim Ingham CommandObjectMultiwordCommands::CommandObjectMultiwordCommands (CommandInterpreter &interpreter) :
2144a7015092SGreg Clayton     CommandObjectMultiword (interpreter,
21450e5e5a79SGreg Clayton                             "command",
21463f4c09c1SCaroline Tice                             "A set of commands for managing or customizing the debugger commands.",
21470e5e5a79SGreg Clayton                             "command <subcommand> [<subcommand-options>]")
2148ebc09c36SJim Ingham {
2149a7015092SGreg Clayton     LoadSubCommand ("source",  CommandObjectSP (new CommandObjectCommandsSource (interpreter)));
2150a7015092SGreg Clayton     LoadSubCommand ("alias",   CommandObjectSP (new CommandObjectCommandsAlias (interpreter)));
2151a7015092SGreg Clayton     LoadSubCommand ("unalias", CommandObjectSP (new CommandObjectCommandsUnalias (interpreter)));
2152b547278cSGreg Clayton     LoadSubCommand ("delete",  CommandObjectSP (new CommandObjectCommandsDelete (interpreter)));
2153de164aaaSGreg Clayton     LoadSubCommand ("regex",   CommandObjectSP (new CommandObjectCommandsAddRegex (interpreter)));
2154a5a97ebeSJim Ingham     LoadSubCommand ("history", CommandObjectSP (new CommandObjectCommandsHistory (interpreter)));
2155223383edSEnrico Granata     LoadSubCommand ("script",  CommandObjectSP (new CommandObjectMultiwordCommandsScript (interpreter)));
2156ebc09c36SJim Ingham }
2157ebc09c36SJim Ingham 
2158*6e3d8e7fSEugene Zelenko CommandObjectMultiwordCommands::~CommandObjectMultiwordCommands() = default;
2159