130fdc8d8SChris Lattner //===-- CommandObjectProcess.cpp --------------------------------*- C++ -*-===//
230fdc8d8SChris Lattner //
330fdc8d8SChris Lattner //                     The LLVM Compiler Infrastructure
430fdc8d8SChris Lattner //
530fdc8d8SChris Lattner // This file is distributed under the University of Illinois Open Source
630fdc8d8SChris Lattner // License. See LICENSE.TXT for details.
730fdc8d8SChris Lattner //
830fdc8d8SChris Lattner //===----------------------------------------------------------------------===//
930fdc8d8SChris Lattner 
1030fdc8d8SChris Lattner // C Includes
1130fdc8d8SChris Lattner // C++ Includes
1230fdc8d8SChris Lattner // Other libraries and framework includes
1330fdc8d8SChris Lattner // Project includes
1449bcfd80SEugene Zelenko #include "CommandObjectProcess.h"
150e41084aSJim Ingham #include "lldb/Breakpoint/Breakpoint.h"
160e41084aSJim Ingham #include "lldb/Breakpoint/BreakpointLocation.h"
170e41084aSJim Ingham #include "lldb/Breakpoint/BreakpointSite.h"
181f746071SGreg Clayton #include "lldb/Core/Module.h"
19a2715cf1SGreg Clayton #include "lldb/Core/PluginManager.h"
20b9c1b51eSKate Stone #include "lldb/Core/State.h"
217260f620SGreg Clayton #include "lldb/Host/Host.h"
223eb2b44dSZachary Turner #include "lldb/Host/OptionParser.h"
235275aaa0SVince Harron #include "lldb/Host/StringConvert.h"
2430fdc8d8SChris Lattner #include "lldb/Interpreter/CommandInterpreter.h"
2530fdc8d8SChris Lattner #include "lldb/Interpreter/CommandReturnObject.h"
2647cbf4a0SPavel Labath #include "lldb/Interpreter/OptionArgParser.h"
27b9c1b51eSKate Stone #include "lldb/Interpreter/Options.h"
28e996fd30SGreg Clayton #include "lldb/Target/Platform.h"
2930fdc8d8SChris Lattner #include "lldb/Target/Process.h"
300e41084aSJim Ingham #include "lldb/Target/StopInfo.h"
3130fdc8d8SChris Lattner #include "lldb/Target/Target.h"
3230fdc8d8SChris Lattner #include "lldb/Target/Thread.h"
3393749ab3SZachary Turner #include "lldb/Target/UnixSignals.h"
34145d95c9SPavel Labath #include "lldb/Utility/Args.h"
3530fdc8d8SChris Lattner 
3630fdc8d8SChris Lattner using namespace lldb;
3730fdc8d8SChris Lattner using namespace lldb_private;
3830fdc8d8SChris Lattner 
39b9c1b51eSKate Stone class CommandObjectProcessLaunchOrAttach : public CommandObjectParsed {
40dcb1d856SJim Ingham public:
41dcb1d856SJim Ingham   CommandObjectProcessLaunchOrAttach(CommandInterpreter &interpreter,
42b9c1b51eSKate Stone                                      const char *name, const char *help,
43b9c1b51eSKate Stone                                      const char *syntax, uint32_t flags,
44b9c1b51eSKate Stone                                      const char *new_process_action)
45b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, name, help, syntax, flags),
46dcb1d856SJim Ingham         m_new_process_action(new_process_action) {}
47dcb1d856SJim Ingham 
4849bcfd80SEugene Zelenko   ~CommandObjectProcessLaunchOrAttach() override = default;
4949bcfd80SEugene Zelenko 
50dcb1d856SJim Ingham protected:
51b9c1b51eSKate Stone   bool StopProcessIfNecessary(Process *process, StateType &state,
52b9c1b51eSKate Stone                               CommandReturnObject &result) {
53dcb1d856SJim Ingham     state = eStateInvalid;
54b9c1b51eSKate Stone     if (process) {
55dcb1d856SJim Ingham       state = process->GetState();
56dcb1d856SJim Ingham 
57b9c1b51eSKate Stone       if (process->IsAlive() && state != eStateConnected) {
58dcb1d856SJim Ingham         char message[1024];
59dcb1d856SJim Ingham         if (process->GetState() == eStateAttaching)
60b9c1b51eSKate Stone           ::snprintf(message, sizeof(message),
61b9c1b51eSKate Stone                      "There is a pending attach, abort it and %s?",
62b9c1b51eSKate Stone                      m_new_process_action.c_str());
63dcb1d856SJim Ingham         else if (process->GetShouldDetach())
64b9c1b51eSKate Stone           ::snprintf(message, sizeof(message),
65b9c1b51eSKate Stone                      "There is a running process, detach from it and %s?",
66b9c1b51eSKate Stone                      m_new_process_action.c_str());
67dcb1d856SJim Ingham         else
68b9c1b51eSKate Stone           ::snprintf(message, sizeof(message),
69b9c1b51eSKate Stone                      "There is a running process, kill it and %s?",
70b9c1b51eSKate Stone                      m_new_process_action.c_str());
71dcb1d856SJim Ingham 
72b9c1b51eSKate Stone         if (!m_interpreter.Confirm(message, true)) {
73dcb1d856SJim Ingham           result.SetStatus(eReturnStatusFailed);
74dcb1d856SJim Ingham           return false;
75b9c1b51eSKate Stone         } else {
76b9c1b51eSKate Stone           if (process->GetShouldDetach()) {
77acff8950SJim Ingham             bool keep_stopped = false;
7897206d57SZachary Turner             Status detach_error(process->Detach(keep_stopped));
79b9c1b51eSKate Stone             if (detach_error.Success()) {
80dcb1d856SJim Ingham               result.SetStatus(eReturnStatusSuccessFinishResult);
8149bcfd80SEugene Zelenko               process = nullptr;
82b9c1b51eSKate Stone             } else {
83b9c1b51eSKate Stone               result.AppendErrorWithFormat(
84b9c1b51eSKate Stone                   "Failed to detach from process: %s\n",
85b9c1b51eSKate Stone                   detach_error.AsCString());
86dcb1d856SJim Ingham               result.SetStatus(eReturnStatusFailed);
87dcb1d856SJim Ingham             }
88b9c1b51eSKate Stone           } else {
8997206d57SZachary Turner             Status destroy_error(process->Destroy(false));
90b9c1b51eSKate Stone             if (destroy_error.Success()) {
91dcb1d856SJim Ingham               result.SetStatus(eReturnStatusSuccessFinishResult);
9249bcfd80SEugene Zelenko               process = nullptr;
93b9c1b51eSKate Stone             } else {
94b9c1b51eSKate Stone               result.AppendErrorWithFormat("Failed to kill process: %s\n",
95b9c1b51eSKate Stone                                            destroy_error.AsCString());
96dcb1d856SJim Ingham               result.SetStatus(eReturnStatusFailed);
97dcb1d856SJim Ingham             }
98dcb1d856SJim Ingham           }
99dcb1d856SJim Ingham         }
100dcb1d856SJim Ingham       }
101dcb1d856SJim Ingham     }
102dcb1d856SJim Ingham     return result.Succeeded();
103dcb1d856SJim Ingham   }
10449bcfd80SEugene Zelenko 
105dcb1d856SJim Ingham   std::string m_new_process_action;
106dcb1d856SJim Ingham };
10749bcfd80SEugene Zelenko 
10830fdc8d8SChris Lattner //-------------------------------------------------------------------------
10930fdc8d8SChris Lattner // CommandObjectProcessLaunch
11030fdc8d8SChris Lattner //-------------------------------------------------------------------------
1114bddaeb5SJim Ingham #pragma mark CommandObjectProcessLaunch
112b9c1b51eSKate Stone class CommandObjectProcessLaunch : public CommandObjectProcessLaunchOrAttach {
11330fdc8d8SChris Lattner public:
114b9c1b51eSKate Stone   CommandObjectProcessLaunch(CommandInterpreter &interpreter)
115b9c1b51eSKate Stone       : CommandObjectProcessLaunchOrAttach(
116b9c1b51eSKate Stone             interpreter, "process launch",
117b9c1b51eSKate Stone             "Launch the executable in the debugger.", nullptr,
118b9c1b51eSKate Stone             eCommandRequiresTarget, "restart"),
119b9c1b51eSKate Stone         m_options() {
120405fe67fSCaroline Tice     CommandArgumentEntry arg;
121405fe67fSCaroline Tice     CommandArgumentData run_args_arg;
122405fe67fSCaroline Tice 
123405fe67fSCaroline Tice     // Define the first (and only) variant of this arg.
124405fe67fSCaroline Tice     run_args_arg.arg_type = eArgTypeRunArgs;
125405fe67fSCaroline Tice     run_args_arg.arg_repetition = eArgRepeatOptional;
126405fe67fSCaroline Tice 
127b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
128b9c1b51eSKate Stone     // argument entry.
129405fe67fSCaroline Tice     arg.push_back(run_args_arg);
130405fe67fSCaroline Tice 
131405fe67fSCaroline Tice     // Push the data for the first argument into the m_arguments vector.
132405fe67fSCaroline Tice     m_arguments.push_back(arg);
13330fdc8d8SChris Lattner   }
13430fdc8d8SChris Lattner 
13549bcfd80SEugene Zelenko   ~CommandObjectProcessLaunch() override = default;
13630fdc8d8SChris Lattner 
137b9c1b51eSKate Stone   int HandleArgumentCompletion(Args &input, int &cursor_index,
138e9ce62b6SJim Ingham                                int &cursor_char_position,
139e9ce62b6SJim Ingham                                OptionElementVector &opt_element_vector,
140b9c1b51eSKate Stone                                int match_start_point, int max_return_elements,
141e9ce62b6SJim Ingham                                bool &word_complete,
142b9c1b51eSKate Stone                                StringList &matches) override {
143e9ce62b6SJim Ingham     std::string completion_str(input.GetArgumentAtIndex(cursor_index));
144e9ce62b6SJim Ingham     completion_str.erase(cursor_char_position);
145e9ce62b6SJim Ingham 
146b9c1b51eSKate Stone     CommandCompletions::InvokeCommonCompletionCallbacks(
147b9c1b51eSKate Stone         GetCommandInterpreter(), CommandCompletions::eDiskFileCompletion,
148b9c1b51eSKate Stone         completion_str.c_str(), match_start_point, max_return_elements, nullptr,
149b9c1b51eSKate Stone         word_complete, matches);
150e9ce62b6SJim Ingham     return matches.GetSize();
151e9ce62b6SJim Ingham   }
152e9ce62b6SJim Ingham 
153b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
15430fdc8d8SChris Lattner 
155b9c1b51eSKate Stone   const char *GetRepeatCommand(Args &current_command_args,
156b9c1b51eSKate Stone                                uint32_t index) override {
1575a988416SJim Ingham     // No repeat for "process launch"...
1585a988416SJim Ingham     return "";
1595a988416SJim Ingham   }
1605a988416SJim Ingham 
1615a988416SJim Ingham protected:
162b9c1b51eSKate Stone   bool DoExecute(Args &launch_args, CommandReturnObject &result) override {
1631d885966SGreg Clayton     Debugger &debugger = m_interpreter.GetDebugger();
1641d885966SGreg Clayton     Target *target = debugger.GetSelectedTarget().get();
16549bcfd80SEugene Zelenko     // If our listener is nullptr, users aren't allows to launch
166b09c5384SGreg Clayton     ModuleSP exe_module_sp = target->GetExecutableModule();
16771337622SGreg Clayton 
168b9c1b51eSKate Stone     if (exe_module_sp == nullptr) {
169b9c1b51eSKate Stone       result.AppendError("no file in target, create a debug target using the "
170b9c1b51eSKate Stone                          "'target create' command");
17171337622SGreg Clayton       result.SetStatus(eReturnStatusFailed);
17271337622SGreg Clayton       return false;
17371337622SGreg Clayton     }
17471337622SGreg Clayton 
17571337622SGreg Clayton     StateType state = eStateInvalid;
17671337622SGreg Clayton 
177b09c5384SGreg Clayton     if (!StopProcessIfNecessary(m_exe_ctx.GetProcessPtr(), state, result))
17830fdc8d8SChris Lattner       return false;
17930fdc8d8SChris Lattner 
18031d97a5cSZachary Turner     llvm::StringRef target_settings_argv0 = target->GetArg0();
18145392553SGreg Clayton 
182b9c1b51eSKate Stone     // Determine whether we will disable ASLR or leave it in the default state
18305097246SAdrian Prantl     // (i.e. enabled if the platform supports it). First check if the process
18405097246SAdrian Prantl     // launch options explicitly turn on/off
185b9c1b51eSKate Stone     // disabling ASLR.  If so, use that setting;
1865163792bSTodd Fiala     // otherwise, use the 'settings target.disable-aslr' setting.
1875163792bSTodd Fiala     bool disable_aslr = false;
188b9c1b51eSKate Stone     if (m_options.disable_aslr != eLazyBoolCalculate) {
18905097246SAdrian Prantl       // The user specified an explicit setting on the process launch line.
19005097246SAdrian Prantl       // Use it.
1915163792bSTodd Fiala       disable_aslr = (m_options.disable_aslr == eLazyBoolYes);
192b9c1b51eSKate Stone     } else {
19305097246SAdrian Prantl       // The user did not explicitly specify whether to disable ASLR.  Fall
19405097246SAdrian Prantl       // back to the target.disable-aslr setting.
1955163792bSTodd Fiala       disable_aslr = target->GetDisableASLR();
1965163792bSTodd Fiala     }
1975163792bSTodd Fiala 
1985163792bSTodd Fiala     if (disable_aslr)
199b09c5384SGreg Clayton       m_options.launch_info.GetFlags().Set(eLaunchFlagDisableASLR);
2005163792bSTodd Fiala     else
2015163792bSTodd Fiala       m_options.launch_info.GetFlags().Clear(eLaunchFlagDisableASLR);
202b09c5384SGreg Clayton 
203106d0286SJim Ingham     if (target->GetDetachOnError())
204106d0286SJim Ingham       m_options.launch_info.GetFlags().Set(eLaunchFlagDetachOnError);
205106d0286SJim Ingham 
206b09c5384SGreg Clayton     if (target->GetDisableSTDIO())
207b09c5384SGreg Clayton       m_options.launch_info.GetFlags().Set(eLaunchFlagDisableSTDIO);
208b09c5384SGreg Clayton 
20962930e57SPavel Labath     m_options.launch_info.GetEnvironment() = target->GetEnvironment();
21045392553SGreg Clayton 
21131d97a5cSZachary Turner     if (!target_settings_argv0.empty()) {
212b9c1b51eSKate Stone       m_options.launch_info.GetArguments().AppendArgument(
21331d97a5cSZachary Turner           target_settings_argv0);
214b9c1b51eSKate Stone       m_options.launch_info.SetExecutableFile(
215b9c1b51eSKate Stone           exe_module_sp->GetPlatformFileSpec(), false);
216b9c1b51eSKate Stone     } else {
217b9c1b51eSKate Stone       m_options.launch_info.SetExecutableFile(
218b9c1b51eSKate Stone           exe_module_sp->GetPlatformFileSpec(), true);
21945392553SGreg Clayton     }
22045392553SGreg Clayton 
221b9c1b51eSKate Stone     if (launch_args.GetArgumentCount() == 0) {
222b9c1b51eSKate Stone       m_options.launch_info.GetArguments().AppendArguments(
223b9c1b51eSKate Stone           target->GetProcessLaunchInfo().GetArguments());
224b9c1b51eSKate Stone     } else {
22545392553SGreg Clayton       m_options.launch_info.GetArguments().AppendArguments(launch_args);
226162b597cSGreg Clayton       // Save the arguments for subsequent runs in the current target.
227162b597cSGreg Clayton       target->SetRunArguments(launch_args);
228982c9762SGreg Clayton     }
2291d885966SGreg Clayton 
230dc6224e0SGreg Clayton     StreamString stream;
23197206d57SZachary Turner     Status error = target->Launch(m_options.launch_info, &stream);
23230fdc8d8SChris Lattner 
233b9c1b51eSKate Stone     if (error.Success()) {
234b09c5384SGreg Clayton       ProcessSP process_sp(target->GetProcessSP());
235b9c1b51eSKate Stone       if (process_sp) {
236b9c1b51eSKate Stone         // There is a race condition where this thread will return up the call
23705097246SAdrian Prantl         // stack to the main command handler and show an (lldb) prompt before
23805097246SAdrian Prantl         // HandlePrivateEvent (from PrivateStateThread) has a chance to call
23905097246SAdrian Prantl         // PushProcessIOHandler().
240*3879fe00SPavel Labath         process_sp->SyncIOHandler(0, std::chrono::seconds(2));
2418f0db3e1SIlia K 
242c156427dSZachary Turner         llvm::StringRef data = stream.GetString();
243c156427dSZachary Turner         if (!data.empty())
244c156427dSZachary Turner           result.AppendMessage(data);
245b9c1b51eSKate Stone         const char *archname =
246b9c1b51eSKate Stone             exe_module_sp->GetArchitecture().GetArchitectureName();
247b9c1b51eSKate Stone         result.AppendMessageWithFormat(
248b9c1b51eSKate Stone             "Process %" PRIu64 " launched: '%s' (%s)\n", process_sp->GetID(),
249b9c1b51eSKate Stone             exe_module_sp->GetFileSpec().GetPath().c_str(), archname);
25005faeb71SGreg Clayton         result.SetStatus(eReturnStatusSuccessFinishResult);
251b09c5384SGreg Clayton         result.SetDidChangeProcessState(true);
252b9c1b51eSKate Stone       } else {
253b9c1b51eSKate Stone         result.AppendError(
254b9c1b51eSKate Stone             "no error returned from Target::Launch, and target has no process");
255514487e8SGreg Clayton         result.SetStatus(eReturnStatusFailed);
25630fdc8d8SChris Lattner       }
257b9c1b51eSKate Stone     } else {
258b09c5384SGreg Clayton       result.AppendError(error.AsCString());
259514487e8SGreg Clayton       result.SetStatus(eReturnStatusFailed);
260514487e8SGreg Clayton     }
26130fdc8d8SChris Lattner     return result.Succeeded();
26230fdc8d8SChris Lattner   }
26330fdc8d8SChris Lattner 
26430fdc8d8SChris Lattner protected:
265982c9762SGreg Clayton   ProcessLaunchCommandOptions m_options;
26630fdc8d8SChris Lattner };
26730fdc8d8SChris Lattner 
268982c9762SGreg Clayton //#define SET1 LLDB_OPT_SET_1
269982c9762SGreg Clayton //#define SET2 LLDB_OPT_SET_2
270982c9762SGreg Clayton //#define SET3 LLDB_OPT_SET_3
271982c9762SGreg Clayton //
272982c9762SGreg Clayton // OptionDefinition
273982c9762SGreg Clayton // CommandObjectProcessLaunch::CommandOptions::g_option_table[] =
274982c9762SGreg Clayton //{
275ac9c3a62SKate Stone //  // clang-format off
276b9c1b51eSKate Stone //  {SET1 | SET2 | SET3, false, "stop-at-entry", 's', OptionParser::eNoArgument,
277b9c1b51eSKate Stone //  nullptr, 0, eArgTypeNone,          "Stop at the entry point of the program
278b9c1b51eSKate Stone //  when launching a process."},
279b9c1b51eSKate Stone //  {SET1,               false, "stdin",         'i',
280b9c1b51eSKate Stone //  OptionParser::eRequiredArgument, nullptr, 0, eArgTypeDirectoryName,
281b9c1b51eSKate Stone //  "Redirect stdin for the process to <path>."},
282b9c1b51eSKate Stone //  {SET1,               false, "stdout",        'o',
283b9c1b51eSKate Stone //  OptionParser::eRequiredArgument, nullptr, 0, eArgTypeDirectoryName,
284b9c1b51eSKate Stone //  "Redirect stdout for the process to <path>."},
285b9c1b51eSKate Stone //  {SET1,               false, "stderr",        'e',
286b9c1b51eSKate Stone //  OptionParser::eRequiredArgument, nullptr, 0, eArgTypeDirectoryName,
287b9c1b51eSKate Stone //  "Redirect stderr for the process to <path>."},
288b9c1b51eSKate Stone //  {SET1 | SET2 | SET3, false, "plugin",        'p',
289b9c1b51eSKate Stone //  OptionParser::eRequiredArgument, nullptr, 0, eArgTypePlugin,        "Name of
290b9c1b51eSKate Stone //  the process plugin you want to use."},
291b9c1b51eSKate Stone //  {       SET2,        false, "tty",           't',
292b9c1b51eSKate Stone //  OptionParser::eOptionalArgument, nullptr, 0, eArgTypeDirectoryName, "Start
293b9c1b51eSKate Stone //  the process in a terminal. If <path> is specified, look for a terminal whose
294b9c1b51eSKate Stone //  name contains <path>, else start the process in a new terminal."},
295b9c1b51eSKate Stone //  {              SET3, false, "no-stdio",      'n', OptionParser::eNoArgument,
296b9c1b51eSKate Stone //  nullptr, 0, eArgTypeNone,          "Do not set up for terminal I/O to go to
297b9c1b51eSKate Stone //  running process."},
298b9c1b51eSKate Stone //  {SET1 | SET2 | SET3, false, "working-dir",   'w',
299b9c1b51eSKate Stone //  OptionParser::eRequiredArgument, nullptr, 0, eArgTypeDirectoryName, "Set the
300b9c1b51eSKate Stone //  current working directory to <path> when running the inferior."},
30149bcfd80SEugene Zelenko //  {0, false, nullptr, 0, 0, nullptr, 0, eArgTypeNone, nullptr}
302ac9c3a62SKate Stone //  // clang-format on
303982c9762SGreg Clayton //};
304982c9762SGreg Clayton //
305982c9762SGreg Clayton //#undef SET1
306982c9762SGreg Clayton //#undef SET2
307982c9762SGreg Clayton //#undef SET3
30830fdc8d8SChris Lattner 
30930fdc8d8SChris Lattner //-------------------------------------------------------------------------
31030fdc8d8SChris Lattner // CommandObjectProcessAttach
31130fdc8d8SChris Lattner //-------------------------------------------------------------------------
3121f0f5b5bSZachary Turner 
3131f0f5b5bSZachary Turner static OptionDefinition g_process_attach_options[] = {
3141f0f5b5bSZachary Turner     // clang-format off
3151f0f5b5bSZachary Turner   { LLDB_OPT_SET_ALL, false, "continue",         'c', OptionParser::eNoArgument,       nullptr, nullptr, 0, eArgTypeNone,         "Immediately continue the process once attached." },
3161f0f5b5bSZachary Turner   { LLDB_OPT_SET_ALL, false, "plugin",           'P', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypePlugin,       "Name of the process plugin you want to use." },
3171f0f5b5bSZachary Turner   { LLDB_OPT_SET_1,   false, "pid",              'p', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypePid,          "The process ID of an existing process to attach to." },
3181f0f5b5bSZachary Turner   { LLDB_OPT_SET_2,   false, "name",             'n', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeProcessName,  "The name of the process to attach to." },
3191f0f5b5bSZachary Turner   { LLDB_OPT_SET_2,   false, "include-existing", 'i', OptionParser::eNoArgument,       nullptr, nullptr, 0, eArgTypeNone,         "Include existing processes when doing attach -w." },
3201f0f5b5bSZachary Turner   { LLDB_OPT_SET_2,   false, "waitfor",          'w', OptionParser::eNoArgument,       nullptr, nullptr, 0, eArgTypeNone,         "Wait for the process with <process-name> to launch." },
3211f0f5b5bSZachary Turner     // clang-format on
3221f0f5b5bSZachary Turner };
3231f0f5b5bSZachary Turner 
324bb9caf73SJim Ingham #pragma mark CommandObjectProcessAttach
325b9c1b51eSKate Stone class CommandObjectProcessAttach : public CommandObjectProcessLaunchOrAttach {
32630fdc8d8SChris Lattner public:
327b9c1b51eSKate Stone   class CommandOptions : public Options {
32830fdc8d8SChris Lattner   public:
329b9c1b51eSKate Stone     CommandOptions() : Options() {
330b9c1b51eSKate Stone       // Keep default values of all options in one place: OptionParsingStarting
331b9c1b51eSKate Stone       // ()
332e1cfbc79STodd Fiala       OptionParsingStarting(nullptr);
33330fdc8d8SChris Lattner     }
33430fdc8d8SChris Lattner 
33549bcfd80SEugene Zelenko     ~CommandOptions() override = default;
33630fdc8d8SChris Lattner 
33797206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
338b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
33997206d57SZachary Turner       Status error;
3403bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
341b9c1b51eSKate Stone       switch (short_option) {
342a95ce623SJohnny Chen       case 'c':
343a95ce623SJohnny Chen         attach_info.SetContinueOnceAttached(true);
344a95ce623SJohnny Chen         break;
345a95ce623SJohnny Chen 
346b9c1b51eSKate Stone       case 'p': {
347fe11483bSZachary Turner         lldb::pid_t pid;
348fe11483bSZachary Turner         if (option_arg.getAsInteger(0, pid)) {
349fe11483bSZachary Turner           error.SetErrorStringWithFormat("invalid process ID '%s'",
350fe11483bSZachary Turner                                          option_arg.str().c_str());
351b9c1b51eSKate Stone         } else {
352144f3a9cSGreg Clayton           attach_info.SetProcessID(pid);
353144f3a9cSGreg Clayton         }
354b9c1b51eSKate Stone       } break;
35530fdc8d8SChris Lattner 
35630fdc8d8SChris Lattner       case 'P':
357144f3a9cSGreg Clayton         attach_info.SetProcessPluginName(option_arg);
35830fdc8d8SChris Lattner         break;
35930fdc8d8SChris Lattner 
36030fdc8d8SChris Lattner       case 'n':
361144f3a9cSGreg Clayton         attach_info.GetExecutableFile().SetFile(option_arg, false);
36230fdc8d8SChris Lattner         break;
36330fdc8d8SChris Lattner 
36430fdc8d8SChris Lattner       case 'w':
365144f3a9cSGreg Clayton         attach_info.SetWaitForLaunch(true);
36630fdc8d8SChris Lattner         break;
36730fdc8d8SChris Lattner 
368cd16df91SJim Ingham       case 'i':
369cd16df91SJim Ingham         attach_info.SetIgnoreExisting(false);
370cd16df91SJim Ingham         break;
371cd16df91SJim Ingham 
37230fdc8d8SChris Lattner       default:
373b9c1b51eSKate Stone         error.SetErrorStringWithFormat("invalid short option character '%c'",
374b9c1b51eSKate Stone                                        short_option);
37530fdc8d8SChris Lattner         break;
37630fdc8d8SChris Lattner       }
37730fdc8d8SChris Lattner       return error;
37830fdc8d8SChris Lattner     }
37930fdc8d8SChris Lattner 
380b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
381144f3a9cSGreg Clayton       attach_info.Clear();
38230fdc8d8SChris Lattner     }
38330fdc8d8SChris Lattner 
3841f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
38570602439SZachary Turner       return llvm::makeArrayRef(g_process_attach_options);
3861f0f5b5bSZachary Turner     }
38730fdc8d8SChris Lattner 
388b9c1b51eSKate Stone     bool HandleOptionArgumentCompletion(
389b9c1b51eSKate Stone         Args &input, int cursor_index, int char_pos,
390b9c1b51eSKate Stone         OptionElementVector &opt_element_vector, int opt_element_index,
391b9c1b51eSKate Stone         int match_start_point, int max_return_elements,
392b9c1b51eSKate Stone         CommandInterpreter &interpreter, bool &word_complete,
393b9c1b51eSKate Stone         StringList &matches) override {
3945aee162fSJim Ingham       int opt_arg_pos = opt_element_vector[opt_element_index].opt_arg_pos;
3955aee162fSJim Ingham       int opt_defs_index = opt_element_vector[opt_element_index].opt_defs_index;
3965aee162fSJim Ingham 
3975aee162fSJim Ingham       // We are only completing the name option for now...
3985aee162fSJim Ingham 
3991f0f5b5bSZachary Turner       if (GetDefinitions()[opt_defs_index].short_option == 'n') {
4005aee162fSJim Ingham         // Are we in the name?
4015aee162fSJim Ingham 
402b9c1b51eSKate Stone         // Look to see if there is a -P argument provided, and if so use that
40305097246SAdrian Prantl         // plugin, otherwise use the default plugin.
4045aee162fSJim Ingham 
40549bcfd80SEugene Zelenko         const char *partial_name = nullptr;
4065aee162fSJim Ingham         partial_name = input.GetArgumentAtIndex(opt_arg_pos);
4075aee162fSJim Ingham 
408e1cfbc79STodd Fiala         PlatformSP platform_sp(interpreter.GetPlatform(true));
409b9c1b51eSKate Stone         if (platform_sp) {
4108b82f087SGreg Clayton           ProcessInstanceInfoList process_infos;
4118b82f087SGreg Clayton           ProcessInstanceInfoMatch match_info;
412b9c1b51eSKate Stone           if (partial_name) {
413b9c1b51eSKate Stone             match_info.GetProcessInfo().GetExecutableFile().SetFile(
414b9c1b51eSKate Stone                 partial_name, false);
415c4a33951SPavel Labath             match_info.SetNameMatchType(NameMatch::StartsWith);
41632e0a750SGreg Clayton           }
41732e0a750SGreg Clayton           platform_sp->FindProcesses(match_info, process_infos);
418c7bece56SGreg Clayton           const size_t num_matches = process_infos.GetSize();
419b9c1b51eSKate Stone           if (num_matches > 0) {
420b9c1b51eSKate Stone             for (size_t i = 0; i < num_matches; ++i) {
421b9c1b51eSKate Stone               matches.AppendString(
422b9c1b51eSKate Stone                   process_infos.GetProcessNameAtIndex(i),
423e996fd30SGreg Clayton                   process_infos.GetProcessNameLengthAtIndex(i));
4245aee162fSJim Ingham             }
4255aee162fSJim Ingham           }
4265aee162fSJim Ingham         }
4275aee162fSJim Ingham       }
4285aee162fSJim Ingham 
4295aee162fSJim Ingham       return false;
4305aee162fSJim Ingham     }
4315aee162fSJim Ingham 
43230fdc8d8SChris Lattner     // Instance variables to hold the values for command options.
43330fdc8d8SChris Lattner 
434144f3a9cSGreg Clayton     ProcessAttachInfo attach_info;
43530fdc8d8SChris Lattner   };
43630fdc8d8SChris Lattner 
437b9c1b51eSKate Stone   CommandObjectProcessAttach(CommandInterpreter &interpreter)
438b9c1b51eSKate Stone       : CommandObjectProcessLaunchOrAttach(
439b9c1b51eSKate Stone             interpreter, "process attach", "Attach to a process.",
440b9c1b51eSKate Stone             "process attach <cmd-options>", 0, "attach"),
441b9c1b51eSKate Stone         m_options() {}
4425aee162fSJim Ingham 
44349bcfd80SEugene Zelenko   ~CommandObjectProcessAttach() override = default;
4445aee162fSJim Ingham 
445b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
4465a988416SJim Ingham 
4475a988416SJim Ingham protected:
448b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
449b9c1b51eSKate Stone     PlatformSP platform_sp(
450b9c1b51eSKate Stone         m_interpreter.GetDebugger().GetPlatformList().GetSelectedPlatform());
451926af0cdSOleksiy Vyalov 
452a7015092SGreg Clayton     Target *target = m_interpreter.GetDebugger().GetSelectedTarget().get();
453b9c1b51eSKate Stone     // N.B. The attach should be synchronous.  It doesn't help much to get the
45405097246SAdrian Prantl     // prompt back between initiating the attach and the target actually
45505097246SAdrian Prantl     // stopping.  So even if the interpreter is set to be asynchronous, we wait
45605097246SAdrian Prantl     // for the stop ourselves here.
4575aee162fSJim Ingham 
45871337622SGreg Clayton     StateType state = eStateInvalid;
459dcb1d856SJim Ingham     Process *process = m_exe_ctx.GetProcessPtr();
460dcb1d856SJim Ingham 
461dcb1d856SJim Ingham     if (!StopProcessIfNecessary(process, state, result))
4625aee162fSJim Ingham       return false;
4635aee162fSJim Ingham 
464b9c1b51eSKate Stone     if (target == nullptr) {
4655aee162fSJim Ingham       // If there isn't a current target create one.
4665aee162fSJim Ingham       TargetSP new_target_sp;
46797206d57SZachary Turner       Status error;
4685aee162fSJim Ingham 
469b9c1b51eSKate Stone       error = m_interpreter.GetDebugger().GetTargetList().CreateTarget(
470a47464b2SZachary Turner           m_interpreter.GetDebugger(), "", "", false,
47149bcfd80SEugene Zelenko           nullptr, // No platform options
4725aee162fSJim Ingham           new_target_sp);
4735aee162fSJim Ingham       target = new_target_sp.get();
474b9c1b51eSKate Stone       if (target == nullptr || error.Fail()) {
475b766a73dSGreg Clayton         result.AppendError(error.AsCString("Error creating target"));
4765aee162fSJim Ingham         return false;
4775aee162fSJim Ingham       }
478a7015092SGreg Clayton       m_interpreter.GetDebugger().GetTargetList().SetSelectedTarget(target);
4795aee162fSJim Ingham     }
4805aee162fSJim Ingham 
481b9c1b51eSKate Stone     // Record the old executable module, we want to issue a warning if the
48205097246SAdrian Prantl     // process of attaching changed the current executable (like somebody said
48305097246SAdrian Prantl     // "file foo" then attached to a PID whose executable was bar.)
4845aee162fSJim Ingham 
4855aee162fSJim Ingham     ModuleSP old_exec_module_sp = target->GetExecutableModule();
4865aee162fSJim Ingham     ArchSpec old_arch_spec = target->GetArchitecture();
4875aee162fSJim Ingham 
488b9c1b51eSKate Stone     if (command.GetArgumentCount()) {
489b9c1b51eSKate Stone       result.AppendErrorWithFormat("Invalid arguments for '%s'.\nUsage: %s\n",
490b9c1b51eSKate Stone                                    m_cmd_name.c_str(), m_cmd_syntax.c_str());
4915aee162fSJim Ingham       result.SetStatus(eReturnStatusFailed);
492926af0cdSOleksiy Vyalov       return false;
49371337622SGreg Clayton     }
4945aee162fSJim Ingham 
495926af0cdSOleksiy Vyalov     m_interpreter.UpdateExecutionContext(nullptr);
49637386143SOleksiy Vyalov     StreamString stream;
49737386143SOleksiy Vyalov     const auto error = target->Attach(m_options.attach_info, &stream);
498b9c1b51eSKate Stone     if (error.Success()) {
49937386143SOleksiy Vyalov       ProcessSP process_sp(target->GetProcessSP());
500b9c1b51eSKate Stone       if (process_sp) {
501c156427dSZachary Turner         result.AppendMessage(stream.GetString());
502bb3a283bSJim Ingham         result.SetStatus(eReturnStatusSuccessFinishNoResult);
50337386143SOleksiy Vyalov         result.SetDidChangeProcessState(true);
504962260c8SJim Ingham         result.SetAbnormalStopWasExpected(true);
505b9c1b51eSKate Stone       } else {
506b9c1b51eSKate Stone         result.AppendError(
507b9c1b51eSKate Stone             "no error returned from Target::Attach, and target has no process");
508aa739093SJohnny Chen         result.SetStatus(eReturnStatusFailed);
50944d93782SGreg Clayton       }
510b9c1b51eSKate Stone     } else {
51144d93782SGreg Clayton       result.AppendErrorWithFormat("attach failed: %s\n", error.AsCString());
51244d93782SGreg Clayton       result.SetStatus(eReturnStatusFailed);
513aa739093SJohnny Chen     }
5145aee162fSJim Ingham 
515926af0cdSOleksiy Vyalov     if (!result.Succeeded())
516926af0cdSOleksiy Vyalov       return false;
517926af0cdSOleksiy Vyalov 
518b9c1b51eSKate Stone     // Okay, we're done.  Last step is to warn if the executable module has
519b9c1b51eSKate Stone     // changed:
520513c26ceSGreg Clayton     char new_path[PATH_MAX];
521aa149cbdSGreg Clayton     ModuleSP new_exec_module_sp(target->GetExecutableModule());
522b9c1b51eSKate Stone     if (!old_exec_module_sp) {
523513c26ceSGreg Clayton       // We might not have a module if we attached to a raw pid...
524b9c1b51eSKate Stone       if (new_exec_module_sp) {
525aa149cbdSGreg Clayton         new_exec_module_sp->GetFileSpec().GetPath(new_path, PATH_MAX);
526b9c1b51eSKate Stone         result.AppendMessageWithFormat("Executable module set to \"%s\".\n",
527b9c1b51eSKate Stone                                        new_path);
528513c26ceSGreg Clayton       }
529b9c1b51eSKate Stone     } else if (old_exec_module_sp->GetFileSpec() !=
530b9c1b51eSKate Stone                new_exec_module_sp->GetFileSpec()) {
531513c26ceSGreg Clayton       char old_path[PATH_MAX];
5325aee162fSJim Ingham 
5335aee162fSJim Ingham       old_exec_module_sp->GetFileSpec().GetPath(old_path, PATH_MAX);
534aa149cbdSGreg Clayton       new_exec_module_sp->GetFileSpec().GetPath(new_path, PATH_MAX);
5355aee162fSJim Ingham 
536b9c1b51eSKate Stone       result.AppendWarningWithFormat(
537b9c1b51eSKate Stone           "Executable module changed from \"%s\" to \"%s\".\n", old_path,
538b9c1b51eSKate Stone           new_path);
5395aee162fSJim Ingham     }
5405aee162fSJim Ingham 
541b9c1b51eSKate Stone     if (!old_arch_spec.IsValid()) {
542b9c1b51eSKate Stone       result.AppendMessageWithFormat(
543b9c1b51eSKate Stone           "Architecture set to: %s.\n",
544b9c1b51eSKate Stone           target->GetArchitecture().GetTriple().getTriple().c_str());
545b9c1b51eSKate Stone     } else if (!old_arch_spec.IsExactMatch(target->GetArchitecture())) {
546b9c1b51eSKate Stone       result.AppendWarningWithFormat(
547b9c1b51eSKate Stone           "Architecture changed from %s to %s.\n",
548c1b1f1eaSGreg Clayton           old_arch_spec.GetTriple().getTriple().c_str(),
549c1b1f1eaSGreg Clayton           target->GetArchitecture().GetTriple().getTriple().c_str());
5505aee162fSJim Ingham     }
551a95ce623SJohnny Chen 
55205097246SAdrian Prantl     // This supports the use-case scenario of immediately continuing the
55305097246SAdrian Prantl     // process once attached.
554a95ce623SJohnny Chen     if (m_options.attach_info.GetContinueOnceAttached())
5555bcaf583SSean Callanan       m_interpreter.HandleCommand("process continue", eLazyBoolNo, result);
556926af0cdSOleksiy Vyalov 
5575aee162fSJim Ingham     return result.Succeeded();
5585aee162fSJim Ingham   }
5595aee162fSJim Ingham 
56030fdc8d8SChris Lattner   CommandOptions m_options;
56130fdc8d8SChris Lattner };
56230fdc8d8SChris Lattner 
56330fdc8d8SChris Lattner //-------------------------------------------------------------------------
56430fdc8d8SChris Lattner // CommandObjectProcessContinue
56530fdc8d8SChris Lattner //-------------------------------------------------------------------------
5661f0f5b5bSZachary Turner 
5671f0f5b5bSZachary Turner static OptionDefinition g_process_continue_options[] = {
5681f0f5b5bSZachary Turner     // clang-format off
5691f0f5b5bSZachary Turner   { LLDB_OPT_SET_ALL, false, "ignore-count",'i', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeUnsignedInteger, "Ignore <N> crossings of the breakpoint (if it exists) for the currently selected thread." }
5701f0f5b5bSZachary Turner     // clang-format on
5711f0f5b5bSZachary Turner };
5721f0f5b5bSZachary Turner 
573bb9caf73SJim Ingham #pragma mark CommandObjectProcessContinue
57430fdc8d8SChris Lattner 
575b9c1b51eSKate Stone class CommandObjectProcessContinue : public CommandObjectParsed {
57630fdc8d8SChris Lattner public:
577b9c1b51eSKate Stone   CommandObjectProcessContinue(CommandInterpreter &interpreter)
578b9c1b51eSKate Stone       : CommandObjectParsed(
579b9c1b51eSKate Stone             interpreter, "process continue",
580e3d26315SCaroline Tice             "Continue execution of all threads in the current process.",
58130fdc8d8SChris Lattner             "process continue",
582b9c1b51eSKate Stone             eCommandRequiresProcess | eCommandTryTargetAPILock |
583b9c1b51eSKate Stone                 eCommandProcessMustBeLaunched | eCommandProcessMustBePaused),
584b9c1b51eSKate Stone         m_options() {}
58530fdc8d8SChris Lattner 
58649bcfd80SEugene Zelenko   ~CommandObjectProcessContinue() override = default;
58730fdc8d8SChris Lattner 
5885a988416SJim Ingham protected:
589b9c1b51eSKate Stone   class CommandOptions : public Options {
5900e41084aSJim Ingham   public:
591b9c1b51eSKate Stone     CommandOptions() : Options() {
592b9c1b51eSKate Stone       // Keep default values of all options in one place: OptionParsingStarting
593b9c1b51eSKate Stone       // ()
594e1cfbc79STodd Fiala       OptionParsingStarting(nullptr);
5950e41084aSJim Ingham     }
5960e41084aSJim Ingham 
59749bcfd80SEugene Zelenko     ~CommandOptions() override = default;
5980e41084aSJim Ingham 
59997206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
600b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
60197206d57SZachary Turner       Status error;
6023bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
603b9c1b51eSKate Stone       switch (short_option) {
6040e41084aSJim Ingham       case 'i':
605fe11483bSZachary Turner         if (option_arg.getAsInteger(0, m_ignore))
606b9c1b51eSKate Stone           error.SetErrorStringWithFormat(
607b9c1b51eSKate Stone               "invalid value for ignore option: \"%s\", should be a number.",
608fe11483bSZachary Turner               option_arg.str().c_str());
6090e41084aSJim Ingham         break;
6100e41084aSJim Ingham 
6110e41084aSJim Ingham       default:
612b9c1b51eSKate Stone         error.SetErrorStringWithFormat("invalid short option character '%c'",
613b9c1b51eSKate Stone                                        short_option);
6140e41084aSJim Ingham         break;
6150e41084aSJim Ingham       }
6160e41084aSJim Ingham       return error;
6170e41084aSJim Ingham     }
6180e41084aSJim Ingham 
619b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
6200e41084aSJim Ingham       m_ignore = 0;
6210e41084aSJim Ingham     }
6220e41084aSJim Ingham 
6231f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
62470602439SZachary Turner       return llvm::makeArrayRef(g_process_continue_options);
6251f0f5b5bSZachary Turner     }
6260e41084aSJim Ingham 
6270e41084aSJim Ingham     uint32_t m_ignore;
6280e41084aSJim Ingham   };
6290e41084aSJim Ingham 
630b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
631f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
632a7015092SGreg Clayton     bool synchronous_execution = m_interpreter.GetSynchronous();
63330fdc8d8SChris Lattner     StateType state = process->GetState();
634b9c1b51eSKate Stone     if (state == eStateStopped) {
635b9c1b51eSKate Stone       if (command.GetArgumentCount() != 0) {
636b9c1b51eSKate Stone         result.AppendErrorWithFormat(
637b9c1b51eSKate Stone             "The '%s' command does not take any arguments.\n",
638b9c1b51eSKate Stone             m_cmd_name.c_str());
63930fdc8d8SChris Lattner         result.SetStatus(eReturnStatusFailed);
64030fdc8d8SChris Lattner         return false;
64130fdc8d8SChris Lattner       }
64230fdc8d8SChris Lattner 
643b9c1b51eSKate Stone       if (m_options.m_ignore > 0) {
6448d94ba0fSJim Ingham         ThreadSP sel_thread_sp(GetDefaultThread()->shared_from_this());
645b9c1b51eSKate Stone         if (sel_thread_sp) {
6460e41084aSJim Ingham           StopInfoSP stop_info_sp = sel_thread_sp->GetStopInfo();
647b9c1b51eSKate Stone           if (stop_info_sp &&
648b9c1b51eSKate Stone               stop_info_sp->GetStopReason() == eStopReasonBreakpoint) {
649b9c1b51eSKate Stone             lldb::break_id_t bp_site_id =
650b9c1b51eSKate Stone                 (lldb::break_id_t)stop_info_sp->GetValue();
651b9c1b51eSKate Stone             BreakpointSiteSP bp_site_sp(
652b9c1b51eSKate Stone                 process->GetBreakpointSiteList().FindByID(bp_site_id));
653b9c1b51eSKate Stone             if (bp_site_sp) {
654c7bece56SGreg Clayton               const size_t num_owners = bp_site_sp->GetNumberOfOwners();
655b9c1b51eSKate Stone               for (size_t i = 0; i < num_owners; i++) {
656b9c1b51eSKate Stone                 Breakpoint &bp_ref =
657b9c1b51eSKate Stone                     bp_site_sp->GetOwnerAtIndex(i)->GetBreakpoint();
658b9c1b51eSKate Stone                 if (!bp_ref.IsInternal()) {
6590e41084aSJim Ingham                   bp_ref.SetIgnoreCount(m_options.m_ignore);
6600e41084aSJim Ingham                 }
6610e41084aSJim Ingham               }
6620e41084aSJim Ingham             }
6630e41084aSJim Ingham           }
6640e41084aSJim Ingham         }
6650e41084aSJim Ingham       }
6660e41084aSJim Ingham 
66741f2b940SJim Ingham       { // Scope for thread list mutex:
668b9c1b51eSKate Stone         std::lock_guard<std::recursive_mutex> guard(
669b9c1b51eSKate Stone             process->GetThreadList().GetMutex());
67030fdc8d8SChris Lattner         const uint32_t num_threads = process->GetThreadList().GetSize();
67130fdc8d8SChris Lattner 
67230fdc8d8SChris Lattner         // Set the actions that the threads should each take when resuming
673b9c1b51eSKate Stone         for (uint32_t idx = 0; idx < num_threads; ++idx) {
6746c9ed91cSJim Ingham           const bool override_suspend = false;
675b9c1b51eSKate Stone           process->GetThreadList().GetThreadAtIndex(idx)->SetResumeState(
676b9c1b51eSKate Stone               eStateRunning, override_suspend);
67730fdc8d8SChris Lattner         }
67841f2b940SJim Ingham       }
67930fdc8d8SChris Lattner 
6804446487dSPavel Labath       const uint32_t iohandler_id = process->GetIOHandlerID();
6814446487dSPavel Labath 
682dc6224e0SGreg Clayton       StreamString stream;
68397206d57SZachary Turner       Status error;
684dc6224e0SGreg Clayton       if (synchronous_execution)
685dc6224e0SGreg Clayton         error = process->ResumeSynchronous(&stream);
686dc6224e0SGreg Clayton       else
687dc6224e0SGreg Clayton         error = process->Resume();
688a3b89e27STodd Fiala 
689b9c1b51eSKate Stone       if (error.Success()) {
690b9c1b51eSKate Stone         // There is a race condition where this thread will return up the call
69105097246SAdrian Prantl         // stack to the main command handler and show an (lldb) prompt before
69205097246SAdrian Prantl         // HandlePrivateEvent (from PrivateStateThread) has a chance to call
69305097246SAdrian Prantl         // PushProcessIOHandler().
694*3879fe00SPavel Labath         process->SyncIOHandler(iohandler_id, std::chrono::seconds(2));
695a3b89e27STodd Fiala 
696b9c1b51eSKate Stone         result.AppendMessageWithFormat("Process %" PRIu64 " resuming\n",
697b9c1b51eSKate Stone                                        process->GetID());
698b9c1b51eSKate Stone         if (synchronous_execution) {
699b9c1b51eSKate Stone           // If any state changed events had anything to say, add that to the
700b9c1b51eSKate Stone           // result
701c156427dSZachary Turner           result.AppendMessage(stream.GetString());
70230fdc8d8SChris Lattner 
70330fdc8d8SChris Lattner           result.SetDidChangeProcessState(true);
70430fdc8d8SChris Lattner           result.SetStatus(eReturnStatusSuccessFinishNoResult);
705b9c1b51eSKate Stone         } else {
70630fdc8d8SChris Lattner           result.SetStatus(eReturnStatusSuccessContinuingNoResult);
70730fdc8d8SChris Lattner         }
708b9c1b51eSKate Stone       } else {
709b9c1b51eSKate Stone         result.AppendErrorWithFormat("Failed to resume process: %s.\n",
710b9c1b51eSKate Stone                                      error.AsCString());
71130fdc8d8SChris Lattner         result.SetStatus(eReturnStatusFailed);
71230fdc8d8SChris Lattner       }
713b9c1b51eSKate Stone     } else {
714b9c1b51eSKate Stone       result.AppendErrorWithFormat(
715b9c1b51eSKate Stone           "Process cannot be continued from its current state (%s).\n",
71630fdc8d8SChris Lattner           StateAsCString(state));
71730fdc8d8SChris Lattner       result.SetStatus(eReturnStatusFailed);
71830fdc8d8SChris Lattner     }
71930fdc8d8SChris Lattner     return result.Succeeded();
72030fdc8d8SChris Lattner   }
7210e41084aSJim Ingham 
722b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
7230e41084aSJim Ingham 
7240e41084aSJim Ingham   CommandOptions m_options;
7250e41084aSJim Ingham };
7260e41084aSJim Ingham 
72730fdc8d8SChris Lattner //-------------------------------------------------------------------------
72830fdc8d8SChris Lattner // CommandObjectProcessDetach
72930fdc8d8SChris Lattner //-------------------------------------------------------------------------
7301f0f5b5bSZachary Turner static OptionDefinition g_process_detach_options[] = {
7311f0f5b5bSZachary Turner     // clang-format off
7321f0f5b5bSZachary Turner   { LLDB_OPT_SET_1, false, "keep-stopped", 's', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeBoolean, "Whether or not the process should be kept stopped on detach (if possible)." },
7331f0f5b5bSZachary Turner     // clang-format on
7341f0f5b5bSZachary Turner };
7351f0f5b5bSZachary Turner 
736bb9caf73SJim Ingham #pragma mark CommandObjectProcessDetach
73730fdc8d8SChris Lattner 
738b9c1b51eSKate Stone class CommandObjectProcessDetach : public CommandObjectParsed {
73930fdc8d8SChris Lattner public:
740b9c1b51eSKate Stone   class CommandOptions : public Options {
741acff8950SJim Ingham   public:
742b9c1b51eSKate Stone     CommandOptions() : Options() { OptionParsingStarting(nullptr); }
743acff8950SJim Ingham 
74449bcfd80SEugene Zelenko     ~CommandOptions() override = default;
745acff8950SJim Ingham 
74697206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
747b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
74897206d57SZachary Turner       Status error;
749acff8950SJim Ingham       const int short_option = m_getopt_table[option_idx].val;
750acff8950SJim Ingham 
751b9c1b51eSKate Stone       switch (short_option) {
752acff8950SJim Ingham       case 's':
753acff8950SJim Ingham         bool tmp_result;
754acff8950SJim Ingham         bool success;
75547cbf4a0SPavel Labath         tmp_result = OptionArgParser::ToBoolean(option_arg, false, &success);
756acff8950SJim Ingham         if (!success)
757b9c1b51eSKate Stone           error.SetErrorStringWithFormat("invalid boolean option: \"%s\"",
758fe11483bSZachary Turner                                          option_arg.str().c_str());
759b9c1b51eSKate Stone         else {
760acff8950SJim Ingham           if (tmp_result)
761acff8950SJim Ingham             m_keep_stopped = eLazyBoolYes;
762acff8950SJim Ingham           else
763acff8950SJim Ingham             m_keep_stopped = eLazyBoolNo;
764acff8950SJim Ingham         }
765acff8950SJim Ingham         break;
766acff8950SJim Ingham       default:
767b9c1b51eSKate Stone         error.SetErrorStringWithFormat("invalid short option character '%c'",
768b9c1b51eSKate Stone                                        short_option);
769acff8950SJim Ingham         break;
770acff8950SJim Ingham       }
771acff8950SJim Ingham       return error;
772acff8950SJim Ingham     }
773acff8950SJim Ingham 
774b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
775acff8950SJim Ingham       m_keep_stopped = eLazyBoolCalculate;
776acff8950SJim Ingham     }
777acff8950SJim Ingham 
7781f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
77970602439SZachary Turner       return llvm::makeArrayRef(g_process_detach_options);
7801f0f5b5bSZachary Turner     }
781acff8950SJim Ingham 
782acff8950SJim Ingham     // Instance variables to hold the values for command options.
783acff8950SJim Ingham     LazyBool m_keep_stopped;
784acff8950SJim Ingham   };
78530fdc8d8SChris Lattner 
7867428a18cSKate Stone   CommandObjectProcessDetach(CommandInterpreter &interpreter)
787b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process detach",
788b9c1b51eSKate Stone                             "Detach from the current target process.",
789a7015092SGreg Clayton                             "process detach",
790b9c1b51eSKate Stone                             eCommandRequiresProcess | eCommandTryTargetAPILock |
791b9c1b51eSKate Stone                                 eCommandProcessMustBeLaunched),
792b9c1b51eSKate Stone         m_options() {}
79330fdc8d8SChris Lattner 
79449bcfd80SEugene Zelenko   ~CommandObjectProcessDetach() override = default;
79530fdc8d8SChris Lattner 
796b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
797acff8950SJim Ingham 
7985a988416SJim Ingham protected:
799b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
800f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
801acff8950SJim Ingham     // FIXME: This will be a Command Option:
802acff8950SJim Ingham     bool keep_stopped;
803b9c1b51eSKate Stone     if (m_options.m_keep_stopped == eLazyBoolCalculate) {
804acff8950SJim Ingham       // Check the process default:
80549bcfd80SEugene Zelenko       keep_stopped = process->GetDetachKeepsStopped();
806b9c1b51eSKate Stone     } else if (m_options.m_keep_stopped == eLazyBoolYes)
807acff8950SJim Ingham       keep_stopped = true;
808acff8950SJim Ingham     else
809acff8950SJim Ingham       keep_stopped = false;
810acff8950SJim Ingham 
81197206d57SZachary Turner     Status error(process->Detach(keep_stopped));
812b9c1b51eSKate Stone     if (error.Success()) {
81330fdc8d8SChris Lattner       result.SetStatus(eReturnStatusSuccessFinishResult);
814b9c1b51eSKate Stone     } else {
81530fdc8d8SChris Lattner       result.AppendErrorWithFormat("Detach failed: %s\n", error.AsCString());
81630fdc8d8SChris Lattner       result.SetStatus(eReturnStatusFailed);
81730fdc8d8SChris Lattner       return false;
81830fdc8d8SChris Lattner     }
81930fdc8d8SChris Lattner     return result.Succeeded();
82030fdc8d8SChris Lattner   }
821acff8950SJim Ingham 
822acff8950SJim Ingham   CommandOptions m_options;
823acff8950SJim Ingham };
824acff8950SJim Ingham 
82530fdc8d8SChris Lattner //-------------------------------------------------------------------------
826b766a73dSGreg Clayton // CommandObjectProcessConnect
827b766a73dSGreg Clayton //-------------------------------------------------------------------------
8281f0f5b5bSZachary Turner 
8291f0f5b5bSZachary Turner static OptionDefinition g_process_connect_options[] = {
8301f0f5b5bSZachary Turner     // clang-format off
8311f0f5b5bSZachary Turner   { LLDB_OPT_SET_ALL, false, "plugin", 'p', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypePlugin, "Name of the process plugin you want to use." },
8321f0f5b5bSZachary Turner     // clang-format on
8331f0f5b5bSZachary Turner };
8341f0f5b5bSZachary Turner 
835b766a73dSGreg Clayton #pragma mark CommandObjectProcessConnect
836b766a73dSGreg Clayton 
837b9c1b51eSKate Stone class CommandObjectProcessConnect : public CommandObjectParsed {
838b766a73dSGreg Clayton public:
839b9c1b51eSKate Stone   class CommandOptions : public Options {
840b766a73dSGreg Clayton   public:
841b9c1b51eSKate Stone     CommandOptions() : Options() {
842b9c1b51eSKate Stone       // Keep default values of all options in one place: OptionParsingStarting
843b9c1b51eSKate Stone       // ()
844e1cfbc79STodd Fiala       OptionParsingStarting(nullptr);
845b766a73dSGreg Clayton     }
846b766a73dSGreg Clayton 
84749bcfd80SEugene Zelenko     ~CommandOptions() override = default;
848b766a73dSGreg Clayton 
84997206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
850b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
85197206d57SZachary Turner       Status error;
8523bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
853b766a73dSGreg Clayton 
854b9c1b51eSKate Stone       switch (short_option) {
855b766a73dSGreg Clayton       case 'p':
856b766a73dSGreg Clayton         plugin_name.assign(option_arg);
857b766a73dSGreg Clayton         break;
858b766a73dSGreg Clayton 
859b766a73dSGreg Clayton       default:
860b9c1b51eSKate Stone         error.SetErrorStringWithFormat("invalid short option character '%c'",
861b9c1b51eSKate Stone                                        short_option);
862b766a73dSGreg Clayton         break;
863b766a73dSGreg Clayton       }
864b766a73dSGreg Clayton       return error;
865b766a73dSGreg Clayton     }
866b766a73dSGreg Clayton 
867b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
868b766a73dSGreg Clayton       plugin_name.clear();
869b766a73dSGreg Clayton     }
870b766a73dSGreg Clayton 
8711f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
87270602439SZachary Turner       return llvm::makeArrayRef(g_process_connect_options);
8731f0f5b5bSZachary Turner     }
874b766a73dSGreg Clayton 
875b766a73dSGreg Clayton     // Instance variables to hold the values for command options.
876b766a73dSGreg Clayton 
877b766a73dSGreg Clayton     std::string plugin_name;
878b766a73dSGreg Clayton   };
879b766a73dSGreg Clayton 
880b9c1b51eSKate Stone   CommandObjectProcessConnect(CommandInterpreter &interpreter)
881b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process connect",
882b766a73dSGreg Clayton                             "Connect to a remote debug service.",
883b9c1b51eSKate Stone                             "process connect <remote-url>", 0),
884b9c1b51eSKate Stone         m_options() {}
885b766a73dSGreg Clayton 
88649bcfd80SEugene Zelenko   ~CommandObjectProcessConnect() override = default;
887b766a73dSGreg Clayton 
888b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
8895a988416SJim Ingham 
8905a988416SJim Ingham protected:
891b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
892b9c1b51eSKate Stone     if (command.GetArgumentCount() != 1) {
893b9c1b51eSKate Stone       result.AppendErrorWithFormat(
894b9c1b51eSKate Stone           "'%s' takes exactly one argument:\nUsage: %s\n", m_cmd_name.c_str(),
895ccd6cffbSTamas Berghammer           m_cmd_syntax.c_str());
896ccd6cffbSTamas Berghammer       result.SetStatus(eReturnStatusFailed);
897ccd6cffbSTamas Berghammer       return false;
898ccd6cffbSTamas Berghammer     }
899ccd6cffbSTamas Berghammer 
900ccd6cffbSTamas Berghammer     Process *process = m_exe_ctx.GetProcessPtr();
901b9c1b51eSKate Stone     if (process && process->IsAlive()) {
902b9c1b51eSKate Stone       result.AppendErrorWithFormat(
903b9c1b51eSKate Stone           "Process %" PRIu64
904b9c1b51eSKate Stone           " is currently being debugged, kill the process before connecting.\n",
905b766a73dSGreg Clayton           process->GetID());
906b766a73dSGreg Clayton       result.SetStatus(eReturnStatusFailed);
907b766a73dSGreg Clayton       return false;
908b766a73dSGreg Clayton     }
909b766a73dSGreg Clayton 
910ccd6cffbSTamas Berghammer     const char *plugin_name = nullptr;
911b766a73dSGreg Clayton     if (!m_options.plugin_name.empty())
912b766a73dSGreg Clayton       plugin_name = m_options.plugin_name.c_str();
913b766a73dSGreg Clayton 
91497206d57SZachary Turner     Status error;
915ccd6cffbSTamas Berghammer     Debugger &debugger = m_interpreter.GetDebugger();
916ccd6cffbSTamas Berghammer     PlatformSP platform_sp = m_interpreter.GetPlatform(true);
917b9c1b51eSKate Stone     ProcessSP process_sp = platform_sp->ConnectProcess(
918b9c1b51eSKate Stone         command.GetArgumentAtIndex(0), plugin_name, debugger,
919b9c1b51eSKate Stone         debugger.GetSelectedTarget().get(), error);
920b9c1b51eSKate Stone     if (error.Fail() || process_sp == nullptr) {
921ccd6cffbSTamas Berghammer       result.AppendError(error.AsCString("Error connecting to the process"));
922b766a73dSGreg Clayton       result.SetStatus(eReturnStatusFailed);
923b766a73dSGreg Clayton       return false;
924b766a73dSGreg Clayton     }
925ccd6cffbSTamas Berghammer     return true;
926b766a73dSGreg Clayton   }
927b766a73dSGreg Clayton 
928b766a73dSGreg Clayton   CommandOptions m_options;
929b766a73dSGreg Clayton };
930b766a73dSGreg Clayton 
931b766a73dSGreg Clayton //-------------------------------------------------------------------------
932998255bfSGreg Clayton // CommandObjectProcessPlugin
933998255bfSGreg Clayton //-------------------------------------------------------------------------
934998255bfSGreg Clayton #pragma mark CommandObjectProcessPlugin
935998255bfSGreg Clayton 
936b9c1b51eSKate Stone class CommandObjectProcessPlugin : public CommandObjectProxy {
937998255bfSGreg Clayton public:
9387428a18cSKate Stone   CommandObjectProcessPlugin(CommandInterpreter &interpreter)
939b9c1b51eSKate Stone       : CommandObjectProxy(
940b9c1b51eSKate Stone             interpreter, "process plugin",
941b9c1b51eSKate Stone             "Send a custom command to the current target process plug-in.",
942b9c1b51eSKate Stone             "process plugin <args>", 0) {}
943998255bfSGreg Clayton 
94449bcfd80SEugene Zelenko   ~CommandObjectProcessPlugin() override = default;
945998255bfSGreg Clayton 
946b9c1b51eSKate Stone   CommandObject *GetProxyCommandObject() override {
947e05b2efeSGreg Clayton     Process *process = m_interpreter.GetExecutionContext().GetProcessPtr();
948998255bfSGreg Clayton     if (process)
949998255bfSGreg Clayton       return process->GetPluginCommandObject();
95049bcfd80SEugene Zelenko     return nullptr;
951998255bfSGreg Clayton   }
952998255bfSGreg Clayton };
953998255bfSGreg Clayton 
954998255bfSGreg Clayton //-------------------------------------------------------------------------
9558f343b09SGreg Clayton // CommandObjectProcessLoad
9568f343b09SGreg Clayton //-------------------------------------------------------------------------
9571f0f5b5bSZachary Turner 
9581f0f5b5bSZachary Turner static OptionDefinition g_process_load_options[] = {
9591f0f5b5bSZachary Turner     // clang-format off
9601f0f5b5bSZachary Turner   { LLDB_OPT_SET_ALL, false, "install", 'i', OptionParser::eOptionalArgument, nullptr, nullptr, 0, eArgTypePath, "Install the shared library to the target. If specified without an argument then the library will installed in the current working directory." },
9611f0f5b5bSZachary Turner     // clang-format on
9621f0f5b5bSZachary Turner };
9631f0f5b5bSZachary Turner 
964bb9caf73SJim Ingham #pragma mark CommandObjectProcessLoad
9658f343b09SGreg Clayton 
966b9c1b51eSKate Stone class CommandObjectProcessLoad : public CommandObjectParsed {
9678f343b09SGreg Clayton public:
968b9c1b51eSKate Stone   class CommandOptions : public Options {
9694fbd67acSTamas Berghammer   public:
970b9c1b51eSKate Stone     CommandOptions() : Options() {
971b9c1b51eSKate Stone       // Keep default values of all options in one place: OptionParsingStarting
972b9c1b51eSKate Stone       // ()
973e1cfbc79STodd Fiala       OptionParsingStarting(nullptr);
9744fbd67acSTamas Berghammer     }
9754fbd67acSTamas Berghammer 
9764fbd67acSTamas Berghammer     ~CommandOptions() override = default;
9774fbd67acSTamas Berghammer 
97897206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
979b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
98097206d57SZachary Turner       Status error;
9814fbd67acSTamas Berghammer       const int short_option = m_getopt_table[option_idx].val;
982b9c1b51eSKate Stone       switch (short_option) {
9834fbd67acSTamas Berghammer       case 'i':
9844fbd67acSTamas Berghammer         do_install = true;
985fe11483bSZachary Turner         if (!option_arg.empty())
9864fbd67acSTamas Berghammer           install_path.SetFile(option_arg, false);
9874fbd67acSTamas Berghammer         break;
9884fbd67acSTamas Berghammer       default:
989b9c1b51eSKate Stone         error.SetErrorStringWithFormat("invalid short option character '%c'",
990b9c1b51eSKate Stone                                        short_option);
9914fbd67acSTamas Berghammer         break;
9924fbd67acSTamas Berghammer       }
9934fbd67acSTamas Berghammer       return error;
9944fbd67acSTamas Berghammer     }
9954fbd67acSTamas Berghammer 
996b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
9974fbd67acSTamas Berghammer       do_install = false;
9984fbd67acSTamas Berghammer       install_path.Clear();
9994fbd67acSTamas Berghammer     }
10004fbd67acSTamas Berghammer 
10011f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
100270602439SZachary Turner       return llvm::makeArrayRef(g_process_load_options);
10031f0f5b5bSZachary Turner     }
10044fbd67acSTamas Berghammer 
10054fbd67acSTamas Berghammer     // Instance variables to hold the values for command options.
10064fbd67acSTamas Berghammer     bool do_install;
10074fbd67acSTamas Berghammer     FileSpec install_path;
10084fbd67acSTamas Berghammer   };
10098f343b09SGreg Clayton 
1010b9c1b51eSKate Stone   CommandObjectProcessLoad(CommandInterpreter &interpreter)
1011b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process load",
10128f343b09SGreg Clayton                             "Load a shared library into the current process.",
10138f343b09SGreg Clayton                             "process load <filename> [<filename> ...]",
1014b9c1b51eSKate Stone                             eCommandRequiresProcess | eCommandTryTargetAPILock |
1015e87764f2SEnrico Granata                                 eCommandProcessMustBeLaunched |
10164fbd67acSTamas Berghammer                                 eCommandProcessMustBePaused),
1017b9c1b51eSKate Stone         m_options() {}
10188f343b09SGreg Clayton 
10194fbd67acSTamas Berghammer   ~CommandObjectProcessLoad() override = default;
10204fbd67acSTamas Berghammer 
1021b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
10228f343b09SGreg Clayton 
10235a988416SJim Ingham protected:
1024b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1025f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
10268f343b09SGreg Clayton 
102797d2c401SZachary Turner     for (auto &entry : command.entries()) {
102897206d57SZachary Turner       Status error;
10293cb132a0STamas Berghammer       PlatformSP platform = process->GetTarget().GetPlatform();
103097d2c401SZachary Turner       llvm::StringRef image_path = entry.ref;
10314fbd67acSTamas Berghammer       uint32_t image_token = LLDB_INVALID_IMAGE_TOKEN;
10324fbd67acSTamas Berghammer 
1033b9c1b51eSKate Stone       if (!m_options.do_install) {
10344fbd67acSTamas Berghammer         FileSpec image_spec(image_path, false);
10353cb132a0STamas Berghammer         platform->ResolveRemotePath(image_spec, image_spec);
1036b9c1b51eSKate Stone         image_token =
1037b9c1b51eSKate Stone             platform->LoadImage(process, FileSpec(), image_spec, error);
1038b9c1b51eSKate Stone       } else if (m_options.install_path) {
10394fbd67acSTamas Berghammer         FileSpec image_spec(image_path, true);
1040b9c1b51eSKate Stone         platform->ResolveRemotePath(m_options.install_path,
1041b9c1b51eSKate Stone                                     m_options.install_path);
1042b9c1b51eSKate Stone         image_token = platform->LoadImage(process, image_spec,
1043b9c1b51eSKate Stone                                           m_options.install_path, error);
1044b9c1b51eSKate Stone       } else {
10454fbd67acSTamas Berghammer         FileSpec image_spec(image_path, true);
1046b9c1b51eSKate Stone         image_token =
1047b9c1b51eSKate Stone             platform->LoadImage(process, image_spec, FileSpec(), error);
10484fbd67acSTamas Berghammer       }
10494fbd67acSTamas Berghammer 
1050b9c1b51eSKate Stone       if (image_token != LLDB_INVALID_IMAGE_TOKEN) {
1051b9c1b51eSKate Stone         result.AppendMessageWithFormat(
105297d2c401SZachary Turner             "Loading \"%s\"...ok\nImage %u loaded.\n", image_path.str().c_str(),
105397d2c401SZachary Turner             image_token);
10548f343b09SGreg Clayton         result.SetStatus(eReturnStatusSuccessFinishResult);
1055b9c1b51eSKate Stone       } else {
105697d2c401SZachary Turner         result.AppendErrorWithFormat("failed to load '%s': %s",
105797d2c401SZachary Turner                                      image_path.str().c_str(),
1058b9c1b51eSKate Stone                                      error.AsCString());
10598f343b09SGreg Clayton         result.SetStatus(eReturnStatusFailed);
10608f343b09SGreg Clayton       }
10618f343b09SGreg Clayton     }
10628f343b09SGreg Clayton     return result.Succeeded();
10638f343b09SGreg Clayton   }
10644fbd67acSTamas Berghammer 
10654fbd67acSTamas Berghammer   CommandOptions m_options;
10668f343b09SGreg Clayton };
10678f343b09SGreg Clayton 
10688f343b09SGreg Clayton //-------------------------------------------------------------------------
10698f343b09SGreg Clayton // CommandObjectProcessUnload
10708f343b09SGreg Clayton //-------------------------------------------------------------------------
1071bb9caf73SJim Ingham #pragma mark CommandObjectProcessUnload
10728f343b09SGreg Clayton 
1073b9c1b51eSKate Stone class CommandObjectProcessUnload : public CommandObjectParsed {
10748f343b09SGreg Clayton public:
1075b9c1b51eSKate Stone   CommandObjectProcessUnload(CommandInterpreter &interpreter)
1076b9c1b51eSKate Stone       : CommandObjectParsed(
1077b9c1b51eSKate Stone             interpreter, "process unload",
1078b9c1b51eSKate Stone             "Unload a shared library from the current process using the index "
1079b9c1b51eSKate Stone             "returned by a previous call to \"process load\".",
10808f343b09SGreg Clayton             "process unload <index>",
1081b9c1b51eSKate Stone             eCommandRequiresProcess | eCommandTryTargetAPILock |
1082b9c1b51eSKate Stone                 eCommandProcessMustBeLaunched | eCommandProcessMustBePaused) {}
10838f343b09SGreg Clayton 
108449bcfd80SEugene Zelenko   ~CommandObjectProcessUnload() override = default;
10858f343b09SGreg Clayton 
10865a988416SJim Ingham protected:
1087b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1088f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
10898f343b09SGreg Clayton 
109097d2c401SZachary Turner     for (auto &entry : command.entries()) {
109197d2c401SZachary Turner       uint32_t image_token;
109297d2c401SZachary Turner       if (entry.ref.getAsInteger(0, image_token)) {
1093b9c1b51eSKate Stone         result.AppendErrorWithFormat("invalid image index argument '%s'",
109497d2c401SZachary Turner                                      entry.ref.str().c_str());
10958f343b09SGreg Clayton         result.SetStatus(eReturnStatusFailed);
10968f343b09SGreg Clayton         break;
1097b9c1b51eSKate Stone       } else {
109897206d57SZachary Turner         Status error(process->GetTarget().GetPlatform()->UnloadImage(
1099b9c1b51eSKate Stone             process, image_token));
1100b9c1b51eSKate Stone         if (error.Success()) {
1101b9c1b51eSKate Stone           result.AppendMessageWithFormat(
1102b9c1b51eSKate Stone               "Unloading shared library with index %u...ok\n", image_token);
11038f343b09SGreg Clayton           result.SetStatus(eReturnStatusSuccessFinishResult);
1104b9c1b51eSKate Stone         } else {
1105b9c1b51eSKate Stone           result.AppendErrorWithFormat("failed to unload image: %s",
1106b9c1b51eSKate Stone                                        error.AsCString());
11078f343b09SGreg Clayton           result.SetStatus(eReturnStatusFailed);
11088f343b09SGreg Clayton           break;
11098f343b09SGreg Clayton         }
11108f343b09SGreg Clayton       }
11118f343b09SGreg Clayton     }
11128f343b09SGreg Clayton     return result.Succeeded();
11138f343b09SGreg Clayton   }
11148f343b09SGreg Clayton };
11158f343b09SGreg Clayton 
11168f343b09SGreg Clayton //-------------------------------------------------------------------------
111730fdc8d8SChris Lattner // CommandObjectProcessSignal
111830fdc8d8SChris Lattner //-------------------------------------------------------------------------
1119bb9caf73SJim Ingham #pragma mark CommandObjectProcessSignal
112030fdc8d8SChris Lattner 
1121b9c1b51eSKate Stone class CommandObjectProcessSignal : public CommandObjectParsed {
112230fdc8d8SChris Lattner public:
11237428a18cSKate Stone   CommandObjectProcessSignal(CommandInterpreter &interpreter)
1124b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process signal",
1125b9c1b51eSKate Stone                             "Send a UNIX signal to the current target process.",
1126b9c1b51eSKate Stone                             nullptr, eCommandRequiresProcess |
1127b9c1b51eSKate Stone                                          eCommandTryTargetAPILock) {
1128405fe67fSCaroline Tice     CommandArgumentEntry arg;
1129405fe67fSCaroline Tice     CommandArgumentData signal_arg;
1130405fe67fSCaroline Tice 
1131405fe67fSCaroline Tice     // Define the first (and only) variant of this arg.
1132c0dbdfb6SCaroline Tice     signal_arg.arg_type = eArgTypeUnixSignal;
1133405fe67fSCaroline Tice     signal_arg.arg_repetition = eArgRepeatPlain;
1134405fe67fSCaroline Tice 
1135b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
1136b9c1b51eSKate Stone     // argument entry.
1137405fe67fSCaroline Tice     arg.push_back(signal_arg);
1138405fe67fSCaroline Tice 
1139405fe67fSCaroline Tice     // Push the data for the first argument into the m_arguments vector.
1140405fe67fSCaroline Tice     m_arguments.push_back(arg);
114130fdc8d8SChris Lattner   }
114230fdc8d8SChris Lattner 
114349bcfd80SEugene Zelenko   ~CommandObjectProcessSignal() override = default;
114430fdc8d8SChris Lattner 
11455a988416SJim Ingham protected:
1146b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1147f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
114830fdc8d8SChris Lattner 
1149b9c1b51eSKate Stone     if (command.GetArgumentCount() == 1) {
1150237cd906SGreg Clayton       int signo = LLDB_INVALID_SIGNAL_NUMBER;
1151237cd906SGreg Clayton 
1152237cd906SGreg Clayton       const char *signal_name = command.GetArgumentAtIndex(0);
1153237cd906SGreg Clayton       if (::isxdigit(signal_name[0]))
1154b9c1b51eSKate Stone         signo =
1155b9c1b51eSKate Stone             StringConvert::ToSInt32(signal_name, LLDB_INVALID_SIGNAL_NUMBER, 0);
1156237cd906SGreg Clayton       else
115798d0a4b3SChaoren Lin         signo = process->GetUnixSignals()->GetSignalNumberFromName(signal_name);
1158237cd906SGreg Clayton 
1159b9c1b51eSKate Stone       if (signo == LLDB_INVALID_SIGNAL_NUMBER) {
1160b9c1b51eSKate Stone         result.AppendErrorWithFormat("Invalid signal argument '%s'.\n",
1161b9c1b51eSKate Stone                                      command.GetArgumentAtIndex(0));
116230fdc8d8SChris Lattner         result.SetStatus(eReturnStatusFailed);
1163b9c1b51eSKate Stone       } else {
116497206d57SZachary Turner         Status error(process->Signal(signo));
1165b9c1b51eSKate Stone         if (error.Success()) {
116630fdc8d8SChris Lattner           result.SetStatus(eReturnStatusSuccessFinishResult);
1167b9c1b51eSKate Stone         } else {
1168b9c1b51eSKate Stone           result.AppendErrorWithFormat("Failed to send signal %i: %s\n", signo,
1169b9c1b51eSKate Stone                                        error.AsCString());
117030fdc8d8SChris Lattner           result.SetStatus(eReturnStatusFailed);
117130fdc8d8SChris Lattner         }
117230fdc8d8SChris Lattner       }
1173b9c1b51eSKate Stone     } else {
1174b9c1b51eSKate Stone       result.AppendErrorWithFormat(
1175b9c1b51eSKate Stone           "'%s' takes exactly one signal number argument:\nUsage: %s\n",
1176b9c1b51eSKate Stone           m_cmd_name.c_str(), m_cmd_syntax.c_str());
117730fdc8d8SChris Lattner       result.SetStatus(eReturnStatusFailed);
117830fdc8d8SChris Lattner     }
117930fdc8d8SChris Lattner     return result.Succeeded();
118030fdc8d8SChris Lattner   }
118130fdc8d8SChris Lattner };
118230fdc8d8SChris Lattner 
118330fdc8d8SChris Lattner //-------------------------------------------------------------------------
118430fdc8d8SChris Lattner // CommandObjectProcessInterrupt
118530fdc8d8SChris Lattner //-------------------------------------------------------------------------
1186bb9caf73SJim Ingham #pragma mark CommandObjectProcessInterrupt
118730fdc8d8SChris Lattner 
1188b9c1b51eSKate Stone class CommandObjectProcessInterrupt : public CommandObjectParsed {
118930fdc8d8SChris Lattner public:
11907428a18cSKate Stone   CommandObjectProcessInterrupt(CommandInterpreter &interpreter)
1191b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process interrupt",
1192b9c1b51eSKate Stone                             "Interrupt the current target process.",
1193a7015092SGreg Clayton                             "process interrupt",
1194b9c1b51eSKate Stone                             eCommandRequiresProcess | eCommandTryTargetAPILock |
1195b9c1b51eSKate Stone                                 eCommandProcessMustBeLaunched) {}
119630fdc8d8SChris Lattner 
119749bcfd80SEugene Zelenko   ~CommandObjectProcessInterrupt() override = default;
119830fdc8d8SChris Lattner 
11995a988416SJim Ingham protected:
1200b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1201f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
1202b9c1b51eSKate Stone     if (process == nullptr) {
120330fdc8d8SChris Lattner       result.AppendError("no process to halt");
120430fdc8d8SChris Lattner       result.SetStatus(eReturnStatusFailed);
120530fdc8d8SChris Lattner       return false;
120630fdc8d8SChris Lattner     }
120730fdc8d8SChris Lattner 
1208b9c1b51eSKate Stone     if (command.GetArgumentCount() == 0) {
1209f9b57b9dSGreg Clayton       bool clear_thread_plans = true;
121097206d57SZachary Turner       Status error(process->Halt(clear_thread_plans));
1211b9c1b51eSKate Stone       if (error.Success()) {
121230fdc8d8SChris Lattner         result.SetStatus(eReturnStatusSuccessFinishResult);
1213b9c1b51eSKate Stone       } else {
1214b9c1b51eSKate Stone         result.AppendErrorWithFormat("Failed to halt process: %s\n",
1215b9c1b51eSKate Stone                                      error.AsCString());
121630fdc8d8SChris Lattner         result.SetStatus(eReturnStatusFailed);
121730fdc8d8SChris Lattner       }
1218b9c1b51eSKate Stone     } else {
1219fd54b368SJason Molenda       result.AppendErrorWithFormat("'%s' takes no arguments:\nUsage: %s\n",
1220b9c1b51eSKate Stone                                    m_cmd_name.c_str(), m_cmd_syntax.c_str());
122130fdc8d8SChris Lattner       result.SetStatus(eReturnStatusFailed);
122230fdc8d8SChris Lattner     }
122330fdc8d8SChris Lattner     return result.Succeeded();
122430fdc8d8SChris Lattner   }
122530fdc8d8SChris Lattner };
122630fdc8d8SChris Lattner 
122730fdc8d8SChris Lattner //-------------------------------------------------------------------------
122830fdc8d8SChris Lattner // CommandObjectProcessKill
122930fdc8d8SChris Lattner //-------------------------------------------------------------------------
1230bb9caf73SJim Ingham #pragma mark CommandObjectProcessKill
123130fdc8d8SChris Lattner 
1232b9c1b51eSKate Stone class CommandObjectProcessKill : public CommandObjectParsed {
123330fdc8d8SChris Lattner public:
12347428a18cSKate Stone   CommandObjectProcessKill(CommandInterpreter &interpreter)
1235b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process kill",
1236b9c1b51eSKate Stone                             "Terminate the current target process.",
1237b9c1b51eSKate Stone                             "process kill",
1238b9c1b51eSKate Stone                             eCommandRequiresProcess | eCommandTryTargetAPILock |
1239b9c1b51eSKate Stone                                 eCommandProcessMustBeLaunched) {}
124030fdc8d8SChris Lattner 
124149bcfd80SEugene Zelenko   ~CommandObjectProcessKill() override = default;
124230fdc8d8SChris Lattner 
12435a988416SJim Ingham protected:
1244b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1245f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
1246b9c1b51eSKate Stone     if (process == nullptr) {
124730fdc8d8SChris Lattner       result.AppendError("no process to kill");
124830fdc8d8SChris Lattner       result.SetStatus(eReturnStatusFailed);
124930fdc8d8SChris Lattner       return false;
125030fdc8d8SChris Lattner     }
125130fdc8d8SChris Lattner 
1252b9c1b51eSKate Stone     if (command.GetArgumentCount() == 0) {
125397206d57SZachary Turner       Status error(process->Destroy(true));
1254b9c1b51eSKate Stone       if (error.Success()) {
125530fdc8d8SChris Lattner         result.SetStatus(eReturnStatusSuccessFinishResult);
1256b9c1b51eSKate Stone       } else {
1257b9c1b51eSKate Stone         result.AppendErrorWithFormat("Failed to kill process: %s\n",
1258b9c1b51eSKate Stone                                      error.AsCString());
125930fdc8d8SChris Lattner         result.SetStatus(eReturnStatusFailed);
126030fdc8d8SChris Lattner       }
1261b9c1b51eSKate Stone     } else {
1262fd54b368SJason Molenda       result.AppendErrorWithFormat("'%s' takes no arguments:\nUsage: %s\n",
1263b9c1b51eSKate Stone                                    m_cmd_name.c_str(), m_cmd_syntax.c_str());
126430fdc8d8SChris Lattner       result.SetStatus(eReturnStatusFailed);
126530fdc8d8SChris Lattner     }
126630fdc8d8SChris Lattner     return result.Succeeded();
126730fdc8d8SChris Lattner   }
126830fdc8d8SChris Lattner };
126930fdc8d8SChris Lattner 
127030fdc8d8SChris Lattner //-------------------------------------------------------------------------
1271a2715cf1SGreg Clayton // CommandObjectProcessSaveCore
1272a2715cf1SGreg Clayton //-------------------------------------------------------------------------
1273a2715cf1SGreg Clayton #pragma mark CommandObjectProcessSaveCore
1274a2715cf1SGreg Clayton 
1275b9c1b51eSKate Stone class CommandObjectProcessSaveCore : public CommandObjectParsed {
1276a2715cf1SGreg Clayton public:
1277b9c1b51eSKate Stone   CommandObjectProcessSaveCore(CommandInterpreter &interpreter)
1278b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process save-core",
1279b9c1b51eSKate Stone                             "Save the current process as a core file using an "
1280b9c1b51eSKate Stone                             "appropriate file type.",
1281a2715cf1SGreg Clayton                             "process save-core FILE",
1282b9c1b51eSKate Stone                             eCommandRequiresProcess | eCommandTryTargetAPILock |
1283b9c1b51eSKate Stone                                 eCommandProcessMustBeLaunched) {}
1284a2715cf1SGreg Clayton 
128549bcfd80SEugene Zelenko   ~CommandObjectProcessSaveCore() override = default;
1286a2715cf1SGreg Clayton 
1287a2715cf1SGreg Clayton protected:
1288b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1289a2715cf1SGreg Clayton     ProcessSP process_sp = m_exe_ctx.GetProcessSP();
1290b9c1b51eSKate Stone     if (process_sp) {
1291b9c1b51eSKate Stone       if (command.GetArgumentCount() == 1) {
1292a2715cf1SGreg Clayton         FileSpec output_file(command.GetArgumentAtIndex(0), false);
129397206d57SZachary Turner         Status error = PluginManager::SaveCore(process_sp, output_file);
1294b9c1b51eSKate Stone         if (error.Success()) {
1295a2715cf1SGreg Clayton           result.SetStatus(eReturnStatusSuccessFinishResult);
1296b9c1b51eSKate Stone         } else {
1297b9c1b51eSKate Stone           result.AppendErrorWithFormat(
1298b9c1b51eSKate Stone               "Failed to save core file for process: %s\n", error.AsCString());
1299a2715cf1SGreg Clayton           result.SetStatus(eReturnStatusFailed);
1300a2715cf1SGreg Clayton         }
1301b9c1b51eSKate Stone       } else {
1302a2715cf1SGreg Clayton         result.AppendErrorWithFormat("'%s' takes one arguments:\nUsage: %s\n",
1303b9c1b51eSKate Stone                                      m_cmd_name.c_str(), m_cmd_syntax.c_str());
1304a2715cf1SGreg Clayton         result.SetStatus(eReturnStatusFailed);
1305a2715cf1SGreg Clayton       }
1306b9c1b51eSKate Stone     } else {
1307a2715cf1SGreg Clayton       result.AppendError("invalid process");
1308a2715cf1SGreg Clayton       result.SetStatus(eReturnStatusFailed);
1309a2715cf1SGreg Clayton       return false;
1310a2715cf1SGreg Clayton     }
1311a2715cf1SGreg Clayton 
1312a2715cf1SGreg Clayton     return result.Succeeded();
1313a2715cf1SGreg Clayton   }
1314a2715cf1SGreg Clayton };
1315a2715cf1SGreg Clayton 
1316a2715cf1SGreg Clayton //-------------------------------------------------------------------------
13174b9bea87SJim Ingham // CommandObjectProcessStatus
13184b9bea87SJim Ingham //-------------------------------------------------------------------------
1319bb9caf73SJim Ingham #pragma mark CommandObjectProcessStatus
1320bb9caf73SJim Ingham 
1321b9c1b51eSKate Stone class CommandObjectProcessStatus : public CommandObjectParsed {
13224b9bea87SJim Ingham public:
13237428a18cSKate Stone   CommandObjectProcessStatus(CommandInterpreter &interpreter)
1324b9c1b51eSKate Stone       : CommandObjectParsed(
1325b9c1b51eSKate Stone             interpreter, "process status",
1326b9c1b51eSKate Stone             "Show status and stop location for the current target process.",
1327b9c1b51eSKate Stone             "process status",
1328b9c1b51eSKate Stone             eCommandRequiresProcess | eCommandTryTargetAPILock) {}
13294b9bea87SJim Ingham 
133049bcfd80SEugene Zelenko   ~CommandObjectProcessStatus() override = default;
13314b9bea87SJim Ingham 
1332b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
13337260f620SGreg Clayton     Stream &strm = result.GetOutputStream();
13344b9bea87SJim Ingham     result.SetStatus(eReturnStatusSuccessFinishNoResult);
1335b9c1b51eSKate Stone     // No need to check "process" for validity as eCommandRequiresProcess
1336b9c1b51eSKate Stone     // ensures it is valid
1337f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
13387260f620SGreg Clayton     const bool only_threads_with_stop_reason = true;
13397260f620SGreg Clayton     const uint32_t start_frame = 0;
13407260f620SGreg Clayton     const uint32_t num_frames = 1;
13417260f620SGreg Clayton     const uint32_t num_frames_with_source = 1;
13426a9767c7SJim Ingham     const bool     stop_format = true;
1343c14ee32dSGreg Clayton     process->GetStatus(strm);
1344b9c1b51eSKate Stone     process->GetThreadStatus(strm, only_threads_with_stop_reason, start_frame,
13456a9767c7SJim Ingham                              num_frames, num_frames_with_source, stop_format);
13464b9bea87SJim Ingham     return result.Succeeded();
13474b9bea87SJim Ingham   }
13484b9bea87SJim Ingham };
13494b9bea87SJim Ingham 
13504b9bea87SJim Ingham //-------------------------------------------------------------------------
135135731357SCaroline Tice // CommandObjectProcessHandle
135235731357SCaroline Tice //-------------------------------------------------------------------------
13531f0f5b5bSZachary Turner 
13541f0f5b5bSZachary Turner static OptionDefinition g_process_handle_options[] = {
13551f0f5b5bSZachary Turner     // clang-format off
13561f0f5b5bSZachary Turner   { LLDB_OPT_SET_1, false, "stop",   's', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeBoolean, "Whether or not the process should be stopped if the signal is received." },
13571f0f5b5bSZachary Turner   { LLDB_OPT_SET_1, false, "notify", 'n', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeBoolean, "Whether or not the debugger should notify the user if the signal is received." },
13581f0f5b5bSZachary Turner   { LLDB_OPT_SET_1, false, "pass",   'p', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeBoolean, "Whether or not the signal should be passed to the process." }
13591f0f5b5bSZachary Turner     // clang-format on
13601f0f5b5bSZachary Turner };
13611f0f5b5bSZachary Turner 
1362bb9caf73SJim Ingham #pragma mark CommandObjectProcessHandle
136335731357SCaroline Tice 
1364b9c1b51eSKate Stone class CommandObjectProcessHandle : public CommandObjectParsed {
136535731357SCaroline Tice public:
1366b9c1b51eSKate Stone   class CommandOptions : public Options {
136735731357SCaroline Tice   public:
1368b9c1b51eSKate Stone     CommandOptions() : Options() { OptionParsingStarting(nullptr); }
136935731357SCaroline Tice 
137049bcfd80SEugene Zelenko     ~CommandOptions() override = default;
137135731357SCaroline Tice 
137297206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
1373b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
137497206d57SZachary Turner       Status error;
13753bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
137635731357SCaroline Tice 
1377b9c1b51eSKate Stone       switch (short_option) {
137835731357SCaroline Tice       case 's':
137935731357SCaroline Tice         stop = option_arg;
138035731357SCaroline Tice         break;
138135731357SCaroline Tice       case 'n':
138235731357SCaroline Tice         notify = option_arg;
138335731357SCaroline Tice         break;
138435731357SCaroline Tice       case 'p':
138535731357SCaroline Tice         pass = option_arg;
138635731357SCaroline Tice         break;
138735731357SCaroline Tice       default:
1388b9c1b51eSKate Stone         error.SetErrorStringWithFormat("invalid short option character '%c'",
1389b9c1b51eSKate Stone                                        short_option);
139035731357SCaroline Tice         break;
139135731357SCaroline Tice       }
139235731357SCaroline Tice       return error;
139335731357SCaroline Tice     }
139435731357SCaroline Tice 
1395b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
139635731357SCaroline Tice       stop.clear();
139735731357SCaroline Tice       notify.clear();
139835731357SCaroline Tice       pass.clear();
139935731357SCaroline Tice     }
140035731357SCaroline Tice 
14011f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
140270602439SZachary Turner       return llvm::makeArrayRef(g_process_handle_options);
14031f0f5b5bSZachary Turner     }
140435731357SCaroline Tice 
140535731357SCaroline Tice     // Instance variables to hold the values for command options.
140635731357SCaroline Tice 
140735731357SCaroline Tice     std::string stop;
140835731357SCaroline Tice     std::string notify;
140935731357SCaroline Tice     std::string pass;
141035731357SCaroline Tice   };
141135731357SCaroline Tice 
14127428a18cSKate Stone   CommandObjectProcessHandle(CommandInterpreter &interpreter)
1413b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process handle",
1414b9c1b51eSKate Stone                             "Manage LLDB handling of OS signals for the "
1415b9c1b51eSKate Stone                             "current target process.  Defaults to showing "
1416b9c1b51eSKate Stone                             "current policy.",
141749bcfd80SEugene Zelenko                             nullptr),
1418b9c1b51eSKate Stone         m_options() {
1419ea671fbdSKate Stone     SetHelpLong("\nIf no signals are specified, update them all.  If no update "
1420ea671fbdSKate Stone                 "option is specified, list the current values.");
142135731357SCaroline Tice     CommandArgumentEntry arg;
1422c0dbdfb6SCaroline Tice     CommandArgumentData signal_arg;
142335731357SCaroline Tice 
1424c0dbdfb6SCaroline Tice     signal_arg.arg_type = eArgTypeUnixSignal;
1425c0dbdfb6SCaroline Tice     signal_arg.arg_repetition = eArgRepeatStar;
142635731357SCaroline Tice 
1427c0dbdfb6SCaroline Tice     arg.push_back(signal_arg);
142835731357SCaroline Tice 
142935731357SCaroline Tice     m_arguments.push_back(arg);
143035731357SCaroline Tice   }
143135731357SCaroline Tice 
143249bcfd80SEugene Zelenko   ~CommandObjectProcessHandle() override = default;
143335731357SCaroline Tice 
1434b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
143535731357SCaroline Tice 
1436b9c1b51eSKate Stone   bool VerifyCommandOptionValue(const std::string &option, int &real_value) {
143735731357SCaroline Tice     bool okay = true;
143810ad7993SCaroline Tice     bool success = false;
143947cbf4a0SPavel Labath     bool tmp_value = OptionArgParser::ToBoolean(option, false, &success);
144010ad7993SCaroline Tice 
144110ad7993SCaroline Tice     if (success && tmp_value)
144210ad7993SCaroline Tice       real_value = 1;
144310ad7993SCaroline Tice     else if (success && !tmp_value)
144410ad7993SCaroline Tice       real_value = 0;
1445b9c1b51eSKate Stone     else {
144635731357SCaroline Tice       // If the value isn't 'true' or 'false', it had better be 0 or 1.
14475275aaa0SVince Harron       real_value = StringConvert::ToUInt32(option.c_str(), 3);
144810ad7993SCaroline Tice       if (real_value != 0 && real_value != 1)
144935731357SCaroline Tice         okay = false;
145035731357SCaroline Tice     }
145135731357SCaroline Tice 
145235731357SCaroline Tice     return okay;
145335731357SCaroline Tice   }
145435731357SCaroline Tice 
1455b9c1b51eSKate Stone   void PrintSignalHeader(Stream &str) {
145610ad7993SCaroline Tice     str.Printf("NAME         PASS   STOP   NOTIFY\n");
1457b84141a6SPavel Labath     str.Printf("===========  =====  =====  ======\n");
145810ad7993SCaroline Tice   }
145910ad7993SCaroline Tice 
1460b9c1b51eSKate Stone   void PrintSignal(Stream &str, int32_t signo, const char *sig_name,
1461b9c1b51eSKate Stone                    const UnixSignalsSP &signals_sp) {
146210ad7993SCaroline Tice     bool stop;
146310ad7993SCaroline Tice     bool suppress;
146410ad7993SCaroline Tice     bool notify;
146510ad7993SCaroline Tice 
1466b84141a6SPavel Labath     str.Printf("%-11s  ", sig_name);
1467b9c1b51eSKate Stone     if (signals_sp->GetSignalInfo(signo, suppress, stop, notify)) {
146810ad7993SCaroline Tice       bool pass = !suppress;
1469b9c1b51eSKate Stone       str.Printf("%s  %s  %s", (pass ? "true " : "false"),
1470b9c1b51eSKate Stone                  (stop ? "true " : "false"), (notify ? "true " : "false"));
147110ad7993SCaroline Tice     }
147210ad7993SCaroline Tice     str.Printf("\n");
147310ad7993SCaroline Tice   }
147410ad7993SCaroline Tice 
1475b9c1b51eSKate Stone   void PrintSignalInformation(Stream &str, Args &signal_args,
1476b9c1b51eSKate Stone                               int num_valid_signals,
1477b9c1b51eSKate Stone                               const UnixSignalsSP &signals_sp) {
147810ad7993SCaroline Tice     PrintSignalHeader(str);
147910ad7993SCaroline Tice 
1480b9c1b51eSKate Stone     if (num_valid_signals > 0) {
148110ad7993SCaroline Tice       size_t num_args = signal_args.GetArgumentCount();
1482b9c1b51eSKate Stone       for (size_t i = 0; i < num_args; ++i) {
1483b9c1b51eSKate Stone         int32_t signo = signals_sp->GetSignalNumberFromName(
1484b9c1b51eSKate Stone             signal_args.GetArgumentAtIndex(i));
148510ad7993SCaroline Tice         if (signo != LLDB_INVALID_SIGNAL_NUMBER)
1486b9c1b51eSKate Stone           PrintSignal(str, signo, signal_args.GetArgumentAtIndex(i),
1487b9c1b51eSKate Stone                       signals_sp);
148810ad7993SCaroline Tice       }
1489b9c1b51eSKate Stone     } else // Print info for ALL signals
149010ad7993SCaroline Tice     {
149198d0a4b3SChaoren Lin       int32_t signo = signals_sp->GetFirstSignalNumber();
1492b9c1b51eSKate Stone       while (signo != LLDB_INVALID_SIGNAL_NUMBER) {
1493b9c1b51eSKate Stone         PrintSignal(str, signo, signals_sp->GetSignalAsCString(signo),
1494b9c1b51eSKate Stone                     signals_sp);
149598d0a4b3SChaoren Lin         signo = signals_sp->GetNextSignalNumber(signo);
149610ad7993SCaroline Tice       }
149710ad7993SCaroline Tice     }
149810ad7993SCaroline Tice   }
149910ad7993SCaroline Tice 
15005a988416SJim Ingham protected:
1501b9c1b51eSKate Stone   bool DoExecute(Args &signal_args, CommandReturnObject &result) override {
150235731357SCaroline Tice     TargetSP target_sp = m_interpreter.GetDebugger().GetSelectedTarget();
150335731357SCaroline Tice 
1504b9c1b51eSKate Stone     if (!target_sp) {
150535731357SCaroline Tice       result.AppendError("No current target;"
1506b9c1b51eSKate Stone                          " cannot handle signals until you have a valid target "
1507b9c1b51eSKate Stone                          "and process.\n");
150835731357SCaroline Tice       result.SetStatus(eReturnStatusFailed);
150935731357SCaroline Tice       return false;
151035731357SCaroline Tice     }
151135731357SCaroline Tice 
151235731357SCaroline Tice     ProcessSP process_sp = target_sp->GetProcessSP();
151335731357SCaroline Tice 
1514b9c1b51eSKate Stone     if (!process_sp) {
1515b9c1b51eSKate Stone       result.AppendError("No current process; cannot handle signals until you "
1516b9c1b51eSKate Stone                          "have a valid process.\n");
151735731357SCaroline Tice       result.SetStatus(eReturnStatusFailed);
151835731357SCaroline Tice       return false;
151935731357SCaroline Tice     }
152035731357SCaroline Tice 
152135731357SCaroline Tice     int stop_action = -1;   // -1 means leave the current setting alone
152235731357SCaroline Tice     int pass_action = -1;   // -1 means leave the current setting alone
152335731357SCaroline Tice     int notify_action = -1; // -1 means leave the current setting alone
152435731357SCaroline Tice 
1525b9c1b51eSKate Stone     if (!m_options.stop.empty() &&
1526b9c1b51eSKate Stone         !VerifyCommandOptionValue(m_options.stop, stop_action)) {
1527b9c1b51eSKate Stone       result.AppendError("Invalid argument for command option --stop; must be "
1528b9c1b51eSKate Stone                          "true or false.\n");
152935731357SCaroline Tice       result.SetStatus(eReturnStatusFailed);
153035731357SCaroline Tice       return false;
153135731357SCaroline Tice     }
153235731357SCaroline Tice 
1533b9c1b51eSKate Stone     if (!m_options.notify.empty() &&
1534b9c1b51eSKate Stone         !VerifyCommandOptionValue(m_options.notify, notify_action)) {
1535b9c1b51eSKate Stone       result.AppendError("Invalid argument for command option --notify; must "
1536b9c1b51eSKate Stone                          "be true or false.\n");
153735731357SCaroline Tice       result.SetStatus(eReturnStatusFailed);
153835731357SCaroline Tice       return false;
153935731357SCaroline Tice     }
154035731357SCaroline Tice 
1541b9c1b51eSKate Stone     if (!m_options.pass.empty() &&
1542b9c1b51eSKate Stone         !VerifyCommandOptionValue(m_options.pass, pass_action)) {
1543b9c1b51eSKate Stone       result.AppendError("Invalid argument for command option --pass; must be "
1544b9c1b51eSKate Stone                          "true or false.\n");
154535731357SCaroline Tice       result.SetStatus(eReturnStatusFailed);
154635731357SCaroline Tice       return false;
154735731357SCaroline Tice     }
154835731357SCaroline Tice 
154935731357SCaroline Tice     size_t num_args = signal_args.GetArgumentCount();
155098d0a4b3SChaoren Lin     UnixSignalsSP signals_sp = process_sp->GetUnixSignals();
155135731357SCaroline Tice     int num_signals_set = 0;
155235731357SCaroline Tice 
1553b9c1b51eSKate Stone     if (num_args > 0) {
1554d6a24757SZachary Turner       for (const auto &arg : signal_args) {
1555d6a24757SZachary Turner         int32_t signo = signals_sp->GetSignalNumberFromName(arg.c_str());
1556b9c1b51eSKate Stone         if (signo != LLDB_INVALID_SIGNAL_NUMBER) {
1557b9c1b51eSKate Stone           // Casting the actions as bools here should be okay, because
155805097246SAdrian Prantl           // VerifyCommandOptionValue guarantees the value is either 0 or 1.
155935731357SCaroline Tice           if (stop_action != -1)
156098d0a4b3SChaoren Lin             signals_sp->SetShouldStop(signo, stop_action);
1561b9c1b51eSKate Stone           if (pass_action != -1) {
156298d0a4b3SChaoren Lin             bool suppress = !pass_action;
156398d0a4b3SChaoren Lin             signals_sp->SetShouldSuppress(signo, suppress);
156435731357SCaroline Tice           }
156535731357SCaroline Tice           if (notify_action != -1)
156698d0a4b3SChaoren Lin             signals_sp->SetShouldNotify(signo, notify_action);
156735731357SCaroline Tice           ++num_signals_set;
1568b9c1b51eSKate Stone         } else {
1569b9c1b51eSKate Stone           result.AppendErrorWithFormat("Invalid signal name '%s'\n",
1570d6a24757SZachary Turner                                        arg.c_str());
157135731357SCaroline Tice         }
157235731357SCaroline Tice       }
1573b9c1b51eSKate Stone     } else {
1574b9c1b51eSKate Stone       // No signal specified, if any command options were specified, update ALL
1575b9c1b51eSKate Stone       // signals.
1576b9c1b51eSKate Stone       if ((notify_action != -1) || (stop_action != -1) || (pass_action != -1)) {
1577b9c1b51eSKate Stone         if (m_interpreter.Confirm(
1578b9c1b51eSKate Stone                 "Do you really want to update all the signals?", false)) {
157998d0a4b3SChaoren Lin           int32_t signo = signals_sp->GetFirstSignalNumber();
1580b9c1b51eSKate Stone           while (signo != LLDB_INVALID_SIGNAL_NUMBER) {
158110ad7993SCaroline Tice             if (notify_action != -1)
158298d0a4b3SChaoren Lin               signals_sp->SetShouldNotify(signo, notify_action);
158310ad7993SCaroline Tice             if (stop_action != -1)
158498d0a4b3SChaoren Lin               signals_sp->SetShouldStop(signo, stop_action);
1585b9c1b51eSKate Stone             if (pass_action != -1) {
158698d0a4b3SChaoren Lin               bool suppress = !pass_action;
158798d0a4b3SChaoren Lin               signals_sp->SetShouldSuppress(signo, suppress);
158810ad7993SCaroline Tice             }
158998d0a4b3SChaoren Lin             signo = signals_sp->GetNextSignalNumber(signo);
159010ad7993SCaroline Tice           }
159110ad7993SCaroline Tice         }
159210ad7993SCaroline Tice       }
159310ad7993SCaroline Tice     }
159410ad7993SCaroline Tice 
1595b9c1b51eSKate Stone     PrintSignalInformation(result.GetOutputStream(), signal_args,
1596b9c1b51eSKate Stone                            num_signals_set, signals_sp);
159735731357SCaroline Tice 
159835731357SCaroline Tice     if (num_signals_set > 0)
159935731357SCaroline Tice       result.SetStatus(eReturnStatusSuccessFinishNoResult);
160035731357SCaroline Tice     else
160135731357SCaroline Tice       result.SetStatus(eReturnStatusFailed);
160235731357SCaroline Tice 
160335731357SCaroline Tice     return result.Succeeded();
160435731357SCaroline Tice   }
160535731357SCaroline Tice 
160635731357SCaroline Tice   CommandOptions m_options;
160735731357SCaroline Tice };
160835731357SCaroline Tice 
160935731357SCaroline Tice //-------------------------------------------------------------------------
161030fdc8d8SChris Lattner // CommandObjectMultiwordProcess
161130fdc8d8SChris Lattner //-------------------------------------------------------------------------
161230fdc8d8SChris Lattner 
1613b9c1b51eSKate Stone CommandObjectMultiwordProcess::CommandObjectMultiwordProcess(
1614b9c1b51eSKate Stone     CommandInterpreter &interpreter)
1615b9c1b51eSKate Stone     : CommandObjectMultiword(
1616b9c1b51eSKate Stone           interpreter, "process",
1617b9c1b51eSKate Stone           "Commands for interacting with processes on the current platform.",
1618b9c1b51eSKate Stone           "process <subcommand> [<subcommand-options>]") {
1619b9c1b51eSKate Stone   LoadSubCommand("attach",
1620b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessAttach(interpreter)));
1621b9c1b51eSKate Stone   LoadSubCommand("launch",
1622b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessLaunch(interpreter)));
1623b9c1b51eSKate Stone   LoadSubCommand("continue", CommandObjectSP(new CommandObjectProcessContinue(
1624b9c1b51eSKate Stone                                  interpreter)));
1625b9c1b51eSKate Stone   LoadSubCommand("connect",
1626b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessConnect(interpreter)));
1627b9c1b51eSKate Stone   LoadSubCommand("detach",
1628b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessDetach(interpreter)));
1629b9c1b51eSKate Stone   LoadSubCommand("load",
1630b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessLoad(interpreter)));
1631b9c1b51eSKate Stone   LoadSubCommand("unload",
1632b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessUnload(interpreter)));
1633b9c1b51eSKate Stone   LoadSubCommand("signal",
1634b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessSignal(interpreter)));
1635b9c1b51eSKate Stone   LoadSubCommand("handle",
1636b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessHandle(interpreter)));
1637b9c1b51eSKate Stone   LoadSubCommand("status",
1638b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessStatus(interpreter)));
1639b9c1b51eSKate Stone   LoadSubCommand("interrupt", CommandObjectSP(new CommandObjectProcessInterrupt(
1640b9c1b51eSKate Stone                                   interpreter)));
1641b9c1b51eSKate Stone   LoadSubCommand("kill",
1642b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessKill(interpreter)));
1643b9c1b51eSKate Stone   LoadSubCommand("plugin",
1644b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessPlugin(interpreter)));
1645b9c1b51eSKate Stone   LoadSubCommand("save-core", CommandObjectSP(new CommandObjectProcessSaveCore(
1646b9c1b51eSKate Stone                                   interpreter)));
164730fdc8d8SChris Lattner }
164830fdc8d8SChris Lattner 
164949bcfd80SEugene Zelenko CommandObjectMultiwordProcess::~CommandObjectMultiwordProcess() = default;
1650