130fdc8d8SChris Lattner //===-- CommandObjectProcess.cpp --------------------------------*- C++ -*-===//
230fdc8d8SChris Lattner //
3*2946cd70SChandler Carruth // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4*2946cd70SChandler Carruth // See https://llvm.org/LICENSE.txt for license information.
5*2946cd70SChandler Carruth // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
630fdc8d8SChris Lattner //
730fdc8d8SChris Lattner //===----------------------------------------------------------------------===//
830fdc8d8SChris Lattner 
949bcfd80SEugene Zelenko #include "CommandObjectProcess.h"
100e41084aSJim Ingham #include "lldb/Breakpoint/Breakpoint.h"
110e41084aSJim Ingham #include "lldb/Breakpoint/BreakpointLocation.h"
120e41084aSJim Ingham #include "lldb/Breakpoint/BreakpointSite.h"
131f746071SGreg Clayton #include "lldb/Core/Module.h"
14a2715cf1SGreg Clayton #include "lldb/Core/PluginManager.h"
157260f620SGreg Clayton #include "lldb/Host/Host.h"
163eb2b44dSZachary Turner #include "lldb/Host/OptionParser.h"
175275aaa0SVince Harron #include "lldb/Host/StringConvert.h"
1830fdc8d8SChris Lattner #include "lldb/Interpreter/CommandInterpreter.h"
1930fdc8d8SChris Lattner #include "lldb/Interpreter/CommandReturnObject.h"
2047cbf4a0SPavel Labath #include "lldb/Interpreter/OptionArgParser.h"
21b9c1b51eSKate Stone #include "lldb/Interpreter/Options.h"
22e996fd30SGreg Clayton #include "lldb/Target/Platform.h"
2330fdc8d8SChris Lattner #include "lldb/Target/Process.h"
240e41084aSJim Ingham #include "lldb/Target/StopInfo.h"
2530fdc8d8SChris Lattner #include "lldb/Target/Target.h"
2630fdc8d8SChris Lattner #include "lldb/Target/Thread.h"
2793749ab3SZachary Turner #include "lldb/Target/UnixSignals.h"
28145d95c9SPavel Labath #include "lldb/Utility/Args.h"
29d821c997SPavel Labath #include "lldb/Utility/State.h"
3030fdc8d8SChris Lattner 
3130fdc8d8SChris Lattner using namespace lldb;
3230fdc8d8SChris Lattner using namespace lldb_private;
3330fdc8d8SChris Lattner 
34b9c1b51eSKate Stone class CommandObjectProcessLaunchOrAttach : public CommandObjectParsed {
35dcb1d856SJim Ingham public:
36dcb1d856SJim Ingham   CommandObjectProcessLaunchOrAttach(CommandInterpreter &interpreter,
37b9c1b51eSKate Stone                                      const char *name, const char *help,
38b9c1b51eSKate Stone                                      const char *syntax, uint32_t flags,
39b9c1b51eSKate Stone                                      const char *new_process_action)
40b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, name, help, syntax, flags),
41dcb1d856SJim Ingham         m_new_process_action(new_process_action) {}
42dcb1d856SJim Ingham 
4349bcfd80SEugene Zelenko   ~CommandObjectProcessLaunchOrAttach() override = default;
4449bcfd80SEugene Zelenko 
45dcb1d856SJim Ingham protected:
46b9c1b51eSKate Stone   bool StopProcessIfNecessary(Process *process, StateType &state,
47b9c1b51eSKate Stone                               CommandReturnObject &result) {
48dcb1d856SJim Ingham     state = eStateInvalid;
49b9c1b51eSKate Stone     if (process) {
50dcb1d856SJim Ingham       state = process->GetState();
51dcb1d856SJim Ingham 
52b9c1b51eSKate Stone       if (process->IsAlive() && state != eStateConnected) {
53dcb1d856SJim Ingham         char message[1024];
54dcb1d856SJim Ingham         if (process->GetState() == eStateAttaching)
55b9c1b51eSKate Stone           ::snprintf(message, sizeof(message),
56b9c1b51eSKate Stone                      "There is a pending attach, abort it and %s?",
57b9c1b51eSKate Stone                      m_new_process_action.c_str());
58dcb1d856SJim Ingham         else if (process->GetShouldDetach())
59b9c1b51eSKate Stone           ::snprintf(message, sizeof(message),
60b9c1b51eSKate Stone                      "There is a running process, detach from it and %s?",
61b9c1b51eSKate Stone                      m_new_process_action.c_str());
62dcb1d856SJim Ingham         else
63b9c1b51eSKate Stone           ::snprintf(message, sizeof(message),
64b9c1b51eSKate Stone                      "There is a running process, kill it and %s?",
65b9c1b51eSKate Stone                      m_new_process_action.c_str());
66dcb1d856SJim Ingham 
67b9c1b51eSKate Stone         if (!m_interpreter.Confirm(message, true)) {
68dcb1d856SJim Ingham           result.SetStatus(eReturnStatusFailed);
69dcb1d856SJim Ingham           return false;
70b9c1b51eSKate Stone         } else {
71b9c1b51eSKate Stone           if (process->GetShouldDetach()) {
72acff8950SJim Ingham             bool keep_stopped = false;
7397206d57SZachary Turner             Status detach_error(process->Detach(keep_stopped));
74b9c1b51eSKate Stone             if (detach_error.Success()) {
75dcb1d856SJim Ingham               result.SetStatus(eReturnStatusSuccessFinishResult);
7649bcfd80SEugene Zelenko               process = nullptr;
77b9c1b51eSKate Stone             } else {
78b9c1b51eSKate Stone               result.AppendErrorWithFormat(
79b9c1b51eSKate Stone                   "Failed to detach from process: %s\n",
80b9c1b51eSKate Stone                   detach_error.AsCString());
81dcb1d856SJim Ingham               result.SetStatus(eReturnStatusFailed);
82dcb1d856SJim Ingham             }
83b9c1b51eSKate Stone           } else {
8497206d57SZachary Turner             Status destroy_error(process->Destroy(false));
85b9c1b51eSKate Stone             if (destroy_error.Success()) {
86dcb1d856SJim Ingham               result.SetStatus(eReturnStatusSuccessFinishResult);
8749bcfd80SEugene Zelenko               process = nullptr;
88b9c1b51eSKate Stone             } else {
89b9c1b51eSKate Stone               result.AppendErrorWithFormat("Failed to kill process: %s\n",
90b9c1b51eSKate Stone                                            destroy_error.AsCString());
91dcb1d856SJim Ingham               result.SetStatus(eReturnStatusFailed);
92dcb1d856SJim Ingham             }
93dcb1d856SJim Ingham           }
94dcb1d856SJim Ingham         }
95dcb1d856SJim Ingham       }
96dcb1d856SJim Ingham     }
97dcb1d856SJim Ingham     return result.Succeeded();
98dcb1d856SJim Ingham   }
9949bcfd80SEugene Zelenko 
100dcb1d856SJim Ingham   std::string m_new_process_action;
101dcb1d856SJim Ingham };
10249bcfd80SEugene Zelenko 
10330fdc8d8SChris Lattner //-------------------------------------------------------------------------
10430fdc8d8SChris Lattner // CommandObjectProcessLaunch
10530fdc8d8SChris Lattner //-------------------------------------------------------------------------
1064bddaeb5SJim Ingham #pragma mark CommandObjectProcessLaunch
107b9c1b51eSKate Stone class CommandObjectProcessLaunch : public CommandObjectProcessLaunchOrAttach {
10830fdc8d8SChris Lattner public:
109b9c1b51eSKate Stone   CommandObjectProcessLaunch(CommandInterpreter &interpreter)
110b9c1b51eSKate Stone       : CommandObjectProcessLaunchOrAttach(
111b9c1b51eSKate Stone             interpreter, "process launch",
112b9c1b51eSKate Stone             "Launch the executable in the debugger.", nullptr,
113b9c1b51eSKate Stone             eCommandRequiresTarget, "restart"),
114b9c1b51eSKate Stone         m_options() {
115405fe67fSCaroline Tice     CommandArgumentEntry arg;
116405fe67fSCaroline Tice     CommandArgumentData run_args_arg;
117405fe67fSCaroline Tice 
118405fe67fSCaroline Tice     // Define the first (and only) variant of this arg.
119405fe67fSCaroline Tice     run_args_arg.arg_type = eArgTypeRunArgs;
120405fe67fSCaroline Tice     run_args_arg.arg_repetition = eArgRepeatOptional;
121405fe67fSCaroline Tice 
122b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
123b9c1b51eSKate Stone     // argument entry.
124405fe67fSCaroline Tice     arg.push_back(run_args_arg);
125405fe67fSCaroline Tice 
126405fe67fSCaroline Tice     // Push the data for the first argument into the m_arguments vector.
127405fe67fSCaroline Tice     m_arguments.push_back(arg);
12830fdc8d8SChris Lattner   }
12930fdc8d8SChris Lattner 
13049bcfd80SEugene Zelenko   ~CommandObjectProcessLaunch() override = default;
13130fdc8d8SChris Lattner 
1322443bbd4SRaphael Isemann   int HandleArgumentCompletion(
1332443bbd4SRaphael Isemann       CompletionRequest &request,
1342443bbd4SRaphael Isemann       OptionElementVector &opt_element_vector) override {
135e9ce62b6SJim Ingham 
136b9c1b51eSKate Stone     CommandCompletions::InvokeCommonCompletionCallbacks(
137b9c1b51eSKate Stone         GetCommandInterpreter(), CommandCompletions::eDiskFileCompletion,
138a2e76c0bSRaphael Isemann         request, nullptr);
1391a6d7ab5SRaphael Isemann     return request.GetNumberOfMatches();
140e9ce62b6SJim Ingham   }
141e9ce62b6SJim Ingham 
142b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
14330fdc8d8SChris Lattner 
144b9c1b51eSKate Stone   const char *GetRepeatCommand(Args &current_command_args,
145b9c1b51eSKate Stone                                uint32_t index) override {
1465a988416SJim Ingham     // No repeat for "process launch"...
1475a988416SJim Ingham     return "";
1485a988416SJim Ingham   }
1495a988416SJim Ingham 
1505a988416SJim Ingham protected:
151b9c1b51eSKate Stone   bool DoExecute(Args &launch_args, CommandReturnObject &result) override {
1521d885966SGreg Clayton     Debugger &debugger = m_interpreter.GetDebugger();
1531d885966SGreg Clayton     Target *target = debugger.GetSelectedTarget().get();
15449bcfd80SEugene Zelenko     // If our listener is nullptr, users aren't allows to launch
155b09c5384SGreg Clayton     ModuleSP exe_module_sp = target->GetExecutableModule();
15671337622SGreg Clayton 
157b9c1b51eSKate Stone     if (exe_module_sp == nullptr) {
158b9c1b51eSKate Stone       result.AppendError("no file in target, create a debug target using the "
159b9c1b51eSKate Stone                          "'target create' command");
16071337622SGreg Clayton       result.SetStatus(eReturnStatusFailed);
16171337622SGreg Clayton       return false;
16271337622SGreg Clayton     }
16371337622SGreg Clayton 
16471337622SGreg Clayton     StateType state = eStateInvalid;
16571337622SGreg Clayton 
166b09c5384SGreg Clayton     if (!StopProcessIfNecessary(m_exe_ctx.GetProcessPtr(), state, result))
16730fdc8d8SChris Lattner       return false;
16830fdc8d8SChris Lattner 
16931d97a5cSZachary Turner     llvm::StringRef target_settings_argv0 = target->GetArg0();
17045392553SGreg Clayton 
171b9c1b51eSKate Stone     // Determine whether we will disable ASLR or leave it in the default state
17205097246SAdrian Prantl     // (i.e. enabled if the platform supports it). First check if the process
17305097246SAdrian Prantl     // launch options explicitly turn on/off
174b9c1b51eSKate Stone     // disabling ASLR.  If so, use that setting;
1755163792bSTodd Fiala     // otherwise, use the 'settings target.disable-aslr' setting.
1765163792bSTodd Fiala     bool disable_aslr = false;
177b9c1b51eSKate Stone     if (m_options.disable_aslr != eLazyBoolCalculate) {
17805097246SAdrian Prantl       // The user specified an explicit setting on the process launch line.
17905097246SAdrian Prantl       // Use it.
1805163792bSTodd Fiala       disable_aslr = (m_options.disable_aslr == eLazyBoolYes);
181b9c1b51eSKate Stone     } else {
18205097246SAdrian Prantl       // The user did not explicitly specify whether to disable ASLR.  Fall
18305097246SAdrian Prantl       // back to the target.disable-aslr setting.
1845163792bSTodd Fiala       disable_aslr = target->GetDisableASLR();
1855163792bSTodd Fiala     }
1865163792bSTodd Fiala 
1875163792bSTodd Fiala     if (disable_aslr)
188b09c5384SGreg Clayton       m_options.launch_info.GetFlags().Set(eLaunchFlagDisableASLR);
1895163792bSTodd Fiala     else
1905163792bSTodd Fiala       m_options.launch_info.GetFlags().Clear(eLaunchFlagDisableASLR);
191b09c5384SGreg Clayton 
192106d0286SJim Ingham     if (target->GetDetachOnError())
193106d0286SJim Ingham       m_options.launch_info.GetFlags().Set(eLaunchFlagDetachOnError);
194106d0286SJim Ingham 
195b09c5384SGreg Clayton     if (target->GetDisableSTDIO())
196b09c5384SGreg Clayton       m_options.launch_info.GetFlags().Set(eLaunchFlagDisableSTDIO);
197b09c5384SGreg Clayton 
19862930e57SPavel Labath     m_options.launch_info.GetEnvironment() = target->GetEnvironment();
19945392553SGreg Clayton 
20031d97a5cSZachary Turner     if (!target_settings_argv0.empty()) {
201b9c1b51eSKate Stone       m_options.launch_info.GetArguments().AppendArgument(
20231d97a5cSZachary Turner           target_settings_argv0);
203b9c1b51eSKate Stone       m_options.launch_info.SetExecutableFile(
204b9c1b51eSKate Stone           exe_module_sp->GetPlatformFileSpec(), false);
205b9c1b51eSKate Stone     } else {
206b9c1b51eSKate Stone       m_options.launch_info.SetExecutableFile(
207b9c1b51eSKate Stone           exe_module_sp->GetPlatformFileSpec(), true);
20845392553SGreg Clayton     }
20945392553SGreg Clayton 
210b9c1b51eSKate Stone     if (launch_args.GetArgumentCount() == 0) {
211b9c1b51eSKate Stone       m_options.launch_info.GetArguments().AppendArguments(
212b9c1b51eSKate Stone           target->GetProcessLaunchInfo().GetArguments());
213b9c1b51eSKate Stone     } else {
21445392553SGreg Clayton       m_options.launch_info.GetArguments().AppendArguments(launch_args);
215162b597cSGreg Clayton       // Save the arguments for subsequent runs in the current target.
216162b597cSGreg Clayton       target->SetRunArguments(launch_args);
217982c9762SGreg Clayton     }
2181d885966SGreg Clayton 
219dc6224e0SGreg Clayton     StreamString stream;
22097206d57SZachary Turner     Status error = target->Launch(m_options.launch_info, &stream);
22130fdc8d8SChris Lattner 
222b9c1b51eSKate Stone     if (error.Success()) {
223b09c5384SGreg Clayton       ProcessSP process_sp(target->GetProcessSP());
224b9c1b51eSKate Stone       if (process_sp) {
225b9c1b51eSKate Stone         // There is a race condition where this thread will return up the call
22605097246SAdrian Prantl         // stack to the main command handler and show an (lldb) prompt before
22705097246SAdrian Prantl         // HandlePrivateEvent (from PrivateStateThread) has a chance to call
22805097246SAdrian Prantl         // PushProcessIOHandler().
2293879fe00SPavel Labath         process_sp->SyncIOHandler(0, std::chrono::seconds(2));
2308f0db3e1SIlia K 
231c156427dSZachary Turner         llvm::StringRef data = stream.GetString();
232c156427dSZachary Turner         if (!data.empty())
233c156427dSZachary Turner           result.AppendMessage(data);
234b9c1b51eSKate Stone         const char *archname =
235b9c1b51eSKate Stone             exe_module_sp->GetArchitecture().GetArchitectureName();
236b9c1b51eSKate Stone         result.AppendMessageWithFormat(
237b9c1b51eSKate Stone             "Process %" PRIu64 " launched: '%s' (%s)\n", process_sp->GetID(),
238b9c1b51eSKate Stone             exe_module_sp->GetFileSpec().GetPath().c_str(), archname);
23905faeb71SGreg Clayton         result.SetStatus(eReturnStatusSuccessFinishResult);
240b09c5384SGreg Clayton         result.SetDidChangeProcessState(true);
241b9c1b51eSKate Stone       } else {
242b9c1b51eSKate Stone         result.AppendError(
243b9c1b51eSKate Stone             "no error returned from Target::Launch, and target has no process");
244514487e8SGreg Clayton         result.SetStatus(eReturnStatusFailed);
24530fdc8d8SChris Lattner       }
246b9c1b51eSKate Stone     } else {
247b09c5384SGreg Clayton       result.AppendError(error.AsCString());
248514487e8SGreg Clayton       result.SetStatus(eReturnStatusFailed);
249514487e8SGreg Clayton     }
25030fdc8d8SChris Lattner     return result.Succeeded();
25130fdc8d8SChris Lattner   }
25230fdc8d8SChris Lattner 
25330fdc8d8SChris Lattner protected:
254982c9762SGreg Clayton   ProcessLaunchCommandOptions m_options;
25530fdc8d8SChris Lattner };
25630fdc8d8SChris Lattner 
257982c9762SGreg Clayton //#define SET1 LLDB_OPT_SET_1
258982c9762SGreg Clayton //#define SET2 LLDB_OPT_SET_2
259982c9762SGreg Clayton //#define SET3 LLDB_OPT_SET_3
260982c9762SGreg Clayton //
261982c9762SGreg Clayton // OptionDefinition
262982c9762SGreg Clayton // CommandObjectProcessLaunch::CommandOptions::g_option_table[] =
263982c9762SGreg Clayton //{
264ac9c3a62SKate Stone //  // clang-format off
265b9c1b51eSKate Stone //  {SET1 | SET2 | SET3, false, "stop-at-entry", 's', OptionParser::eNoArgument,
266b9c1b51eSKate Stone //  nullptr, 0, eArgTypeNone,          "Stop at the entry point of the program
267b9c1b51eSKate Stone //  when launching a process."},
268b9c1b51eSKate Stone //  {SET1,               false, "stdin",         'i',
269b9c1b51eSKate Stone //  OptionParser::eRequiredArgument, nullptr, 0, eArgTypeDirectoryName,
270b9c1b51eSKate Stone //  "Redirect stdin for the process to <path>."},
271b9c1b51eSKate Stone //  {SET1,               false, "stdout",        'o',
272b9c1b51eSKate Stone //  OptionParser::eRequiredArgument, nullptr, 0, eArgTypeDirectoryName,
273b9c1b51eSKate Stone //  "Redirect stdout for the process to <path>."},
274b9c1b51eSKate Stone //  {SET1,               false, "stderr",        'e',
275b9c1b51eSKate Stone //  OptionParser::eRequiredArgument, nullptr, 0, eArgTypeDirectoryName,
276b9c1b51eSKate Stone //  "Redirect stderr for the process to <path>."},
277b9c1b51eSKate Stone //  {SET1 | SET2 | SET3, false, "plugin",        'p',
278b9c1b51eSKate Stone //  OptionParser::eRequiredArgument, nullptr, 0, eArgTypePlugin,        "Name of
279b9c1b51eSKate Stone //  the process plugin you want to use."},
280b9c1b51eSKate Stone //  {       SET2,        false, "tty",           't',
281b9c1b51eSKate Stone //  OptionParser::eOptionalArgument, nullptr, 0, eArgTypeDirectoryName, "Start
282b9c1b51eSKate Stone //  the process in a terminal. If <path> is specified, look for a terminal whose
283b9c1b51eSKate Stone //  name contains <path>, else start the process in a new terminal."},
284b9c1b51eSKate Stone //  {              SET3, false, "no-stdio",      'n', OptionParser::eNoArgument,
285b9c1b51eSKate Stone //  nullptr, 0, eArgTypeNone,          "Do not set up for terminal I/O to go to
286b9c1b51eSKate Stone //  running process."},
287b9c1b51eSKate Stone //  {SET1 | SET2 | SET3, false, "working-dir",   'w',
288b9c1b51eSKate Stone //  OptionParser::eRequiredArgument, nullptr, 0, eArgTypeDirectoryName, "Set the
289b9c1b51eSKate Stone //  current working directory to <path> when running the inferior."},
29049bcfd80SEugene Zelenko //  {0, false, nullptr, 0, 0, nullptr, 0, eArgTypeNone, nullptr}
291ac9c3a62SKate Stone //  // clang-format on
292982c9762SGreg Clayton //};
293982c9762SGreg Clayton //
294982c9762SGreg Clayton //#undef SET1
295982c9762SGreg Clayton //#undef SET2
296982c9762SGreg Clayton //#undef SET3
29730fdc8d8SChris Lattner 
29830fdc8d8SChris Lattner //-------------------------------------------------------------------------
29930fdc8d8SChris Lattner // CommandObjectProcessAttach
30030fdc8d8SChris Lattner //-------------------------------------------------------------------------
3011f0f5b5bSZachary Turner 
3028fe53c49STatyana Krasnukha static constexpr OptionDefinition g_process_attach_options[] = {
3031f0f5b5bSZachary Turner     // clang-format off
3048fe53c49STatyana Krasnukha   { LLDB_OPT_SET_ALL, false, "continue",         'c', OptionParser::eNoArgument,       nullptr, {}, 0, eArgTypeNone,         "Immediately continue the process once attached." },
3058fe53c49STatyana Krasnukha   { LLDB_OPT_SET_ALL, false, "plugin",           'P', OptionParser::eRequiredArgument, nullptr, {}, 0, eArgTypePlugin,       "Name of the process plugin you want to use." },
3068fe53c49STatyana Krasnukha   { LLDB_OPT_SET_1,   false, "pid",              'p', OptionParser::eRequiredArgument, nullptr, {}, 0, eArgTypePid,          "The process ID of an existing process to attach to." },
3078fe53c49STatyana Krasnukha   { LLDB_OPT_SET_2,   false, "name",             'n', OptionParser::eRequiredArgument, nullptr, {}, 0, eArgTypeProcessName,  "The name of the process to attach to." },
3088fe53c49STatyana Krasnukha   { LLDB_OPT_SET_2,   false, "include-existing", 'i', OptionParser::eNoArgument,       nullptr, {}, 0, eArgTypeNone,         "Include existing processes when doing attach -w." },
3098fe53c49STatyana Krasnukha   { LLDB_OPT_SET_2,   false, "waitfor",          'w', OptionParser::eNoArgument,       nullptr, {}, 0, eArgTypeNone,         "Wait for the process with <process-name> to launch." },
3101f0f5b5bSZachary Turner     // clang-format on
3111f0f5b5bSZachary Turner };
3121f0f5b5bSZachary Turner 
313bb9caf73SJim Ingham #pragma mark CommandObjectProcessAttach
314b9c1b51eSKate Stone class CommandObjectProcessAttach : public CommandObjectProcessLaunchOrAttach {
31530fdc8d8SChris Lattner public:
316b9c1b51eSKate Stone   class CommandOptions : public Options {
31730fdc8d8SChris Lattner   public:
318b9c1b51eSKate Stone     CommandOptions() : Options() {
319b9c1b51eSKate Stone       // Keep default values of all options in one place: OptionParsingStarting
320b9c1b51eSKate Stone       // ()
321e1cfbc79STodd Fiala       OptionParsingStarting(nullptr);
32230fdc8d8SChris Lattner     }
32330fdc8d8SChris Lattner 
32449bcfd80SEugene Zelenko     ~CommandOptions() override = default;
32530fdc8d8SChris Lattner 
32697206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
327b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
32897206d57SZachary Turner       Status error;
3293bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
330b9c1b51eSKate Stone       switch (short_option) {
331a95ce623SJohnny Chen       case 'c':
332a95ce623SJohnny Chen         attach_info.SetContinueOnceAttached(true);
333a95ce623SJohnny Chen         break;
334a95ce623SJohnny Chen 
335b9c1b51eSKate Stone       case 'p': {
336fe11483bSZachary Turner         lldb::pid_t pid;
337fe11483bSZachary Turner         if (option_arg.getAsInteger(0, pid)) {
338fe11483bSZachary Turner           error.SetErrorStringWithFormat("invalid process ID '%s'",
339fe11483bSZachary Turner                                          option_arg.str().c_str());
340b9c1b51eSKate Stone         } else {
341144f3a9cSGreg Clayton           attach_info.SetProcessID(pid);
342144f3a9cSGreg Clayton         }
343b9c1b51eSKate Stone       } break;
34430fdc8d8SChris Lattner 
34530fdc8d8SChris Lattner       case 'P':
346144f3a9cSGreg Clayton         attach_info.SetProcessPluginName(option_arg);
34730fdc8d8SChris Lattner         break;
34830fdc8d8SChris Lattner 
34930fdc8d8SChris Lattner       case 'n':
3508f3be7a3SJonas Devlieghere         attach_info.GetExecutableFile().SetFile(option_arg,
351937348cdSJonas Devlieghere                                                 FileSpec::Style::native);
35230fdc8d8SChris Lattner         break;
35330fdc8d8SChris Lattner 
35430fdc8d8SChris Lattner       case 'w':
355144f3a9cSGreg Clayton         attach_info.SetWaitForLaunch(true);
35630fdc8d8SChris Lattner         break;
35730fdc8d8SChris Lattner 
358cd16df91SJim Ingham       case 'i':
359cd16df91SJim Ingham         attach_info.SetIgnoreExisting(false);
360cd16df91SJim Ingham         break;
361cd16df91SJim Ingham 
36230fdc8d8SChris Lattner       default:
363b9c1b51eSKate Stone         error.SetErrorStringWithFormat("invalid short option character '%c'",
364b9c1b51eSKate Stone                                        short_option);
36530fdc8d8SChris Lattner         break;
36630fdc8d8SChris Lattner       }
36730fdc8d8SChris Lattner       return error;
36830fdc8d8SChris Lattner     }
36930fdc8d8SChris Lattner 
370b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
371144f3a9cSGreg Clayton       attach_info.Clear();
37230fdc8d8SChris Lattner     }
37330fdc8d8SChris Lattner 
3741f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
37570602439SZachary Turner       return llvm::makeArrayRef(g_process_attach_options);
3761f0f5b5bSZachary Turner     }
37730fdc8d8SChris Lattner 
378b9c1b51eSKate Stone     bool HandleOptionArgumentCompletion(
379a2e76c0bSRaphael Isemann         CompletionRequest &request, OptionElementVector &opt_element_vector,
380a2e76c0bSRaphael Isemann         int opt_element_index, CommandInterpreter &interpreter) override {
3815aee162fSJim Ingham       int opt_arg_pos = opt_element_vector[opt_element_index].opt_arg_pos;
3825aee162fSJim Ingham       int opt_defs_index = opt_element_vector[opt_element_index].opt_defs_index;
3835aee162fSJim Ingham 
3845aee162fSJim Ingham       // We are only completing the name option for now...
3855aee162fSJim Ingham 
3861f0f5b5bSZachary Turner       if (GetDefinitions()[opt_defs_index].short_option == 'n') {
3875aee162fSJim Ingham         // Are we in the name?
3885aee162fSJim Ingham 
389b9c1b51eSKate Stone         // Look to see if there is a -P argument provided, and if so use that
39005097246SAdrian Prantl         // plugin, otherwise use the default plugin.
3915aee162fSJim Ingham 
39249bcfd80SEugene Zelenko         const char *partial_name = nullptr;
393a2e76c0bSRaphael Isemann         partial_name = request.GetParsedLine().GetArgumentAtIndex(opt_arg_pos);
3945aee162fSJim Ingham 
395e1cfbc79STodd Fiala         PlatformSP platform_sp(interpreter.GetPlatform(true));
396b9c1b51eSKate Stone         if (platform_sp) {
3978b82f087SGreg Clayton           ProcessInstanceInfoList process_infos;
3988b82f087SGreg Clayton           ProcessInstanceInfoMatch match_info;
399b9c1b51eSKate Stone           if (partial_name) {
400b9c1b51eSKate Stone             match_info.GetProcessInfo().GetExecutableFile().SetFile(
4018f3be7a3SJonas Devlieghere                 partial_name, FileSpec::Style::native);
402c4a33951SPavel Labath             match_info.SetNameMatchType(NameMatch::StartsWith);
40332e0a750SGreg Clayton           }
40432e0a750SGreg Clayton           platform_sp->FindProcesses(match_info, process_infos);
405c7bece56SGreg Clayton           const size_t num_matches = process_infos.GetSize();
406b9c1b51eSKate Stone           if (num_matches > 0) {
407b9c1b51eSKate Stone             for (size_t i = 0; i < num_matches; ++i) {
4081a6d7ab5SRaphael Isemann               request.AddCompletion(llvm::StringRef(
409b9c1b51eSKate Stone                   process_infos.GetProcessNameAtIndex(i),
4101a6d7ab5SRaphael Isemann                   process_infos.GetProcessNameLengthAtIndex(i)));
4115aee162fSJim Ingham             }
4125aee162fSJim Ingham           }
4135aee162fSJim Ingham         }
4145aee162fSJim Ingham       }
4155aee162fSJim Ingham 
4165aee162fSJim Ingham       return false;
4175aee162fSJim Ingham     }
4185aee162fSJim Ingham 
41930fdc8d8SChris Lattner     // Instance variables to hold the values for command options.
42030fdc8d8SChris Lattner 
421144f3a9cSGreg Clayton     ProcessAttachInfo attach_info;
42230fdc8d8SChris Lattner   };
42330fdc8d8SChris Lattner 
424b9c1b51eSKate Stone   CommandObjectProcessAttach(CommandInterpreter &interpreter)
425b9c1b51eSKate Stone       : CommandObjectProcessLaunchOrAttach(
426b9c1b51eSKate Stone             interpreter, "process attach", "Attach to a process.",
427b9c1b51eSKate Stone             "process attach <cmd-options>", 0, "attach"),
428b9c1b51eSKate Stone         m_options() {}
4295aee162fSJim Ingham 
43049bcfd80SEugene Zelenko   ~CommandObjectProcessAttach() override = default;
4315aee162fSJim Ingham 
432b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
4335a988416SJim Ingham 
4345a988416SJim Ingham protected:
435b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
436b9c1b51eSKate Stone     PlatformSP platform_sp(
437b9c1b51eSKate Stone         m_interpreter.GetDebugger().GetPlatformList().GetSelectedPlatform());
438926af0cdSOleksiy Vyalov 
439a7015092SGreg Clayton     Target *target = m_interpreter.GetDebugger().GetSelectedTarget().get();
440b9c1b51eSKate Stone     // N.B. The attach should be synchronous.  It doesn't help much to get the
44105097246SAdrian Prantl     // prompt back between initiating the attach and the target actually
44205097246SAdrian Prantl     // stopping.  So even if the interpreter is set to be asynchronous, we wait
44305097246SAdrian Prantl     // for the stop ourselves here.
4445aee162fSJim Ingham 
44571337622SGreg Clayton     StateType state = eStateInvalid;
446dcb1d856SJim Ingham     Process *process = m_exe_ctx.GetProcessPtr();
447dcb1d856SJim Ingham 
448dcb1d856SJim Ingham     if (!StopProcessIfNecessary(process, state, result))
4495aee162fSJim Ingham       return false;
4505aee162fSJim Ingham 
451b9c1b51eSKate Stone     if (target == nullptr) {
4525aee162fSJim Ingham       // If there isn't a current target create one.
4535aee162fSJim Ingham       TargetSP new_target_sp;
45497206d57SZachary Turner       Status error;
4555aee162fSJim Ingham 
456b9c1b51eSKate Stone       error = m_interpreter.GetDebugger().GetTargetList().CreateTarget(
457f9a07e9fSJonas Devlieghere           m_interpreter.GetDebugger(), "", "", eLoadDependentsNo,
45849bcfd80SEugene Zelenko           nullptr, // No platform options
4595aee162fSJim Ingham           new_target_sp);
4605aee162fSJim Ingham       target = new_target_sp.get();
461b9c1b51eSKate Stone       if (target == nullptr || error.Fail()) {
462b766a73dSGreg Clayton         result.AppendError(error.AsCString("Error creating target"));
4635aee162fSJim Ingham         return false;
4645aee162fSJim Ingham       }
465a7015092SGreg Clayton       m_interpreter.GetDebugger().GetTargetList().SetSelectedTarget(target);
4665aee162fSJim Ingham     }
4675aee162fSJim Ingham 
468b9c1b51eSKate Stone     // Record the old executable module, we want to issue a warning if the
46905097246SAdrian Prantl     // process of attaching changed the current executable (like somebody said
47005097246SAdrian Prantl     // "file foo" then attached to a PID whose executable was bar.)
4715aee162fSJim Ingham 
4725aee162fSJim Ingham     ModuleSP old_exec_module_sp = target->GetExecutableModule();
4735aee162fSJim Ingham     ArchSpec old_arch_spec = target->GetArchitecture();
4745aee162fSJim Ingham 
475b9c1b51eSKate Stone     if (command.GetArgumentCount()) {
476b9c1b51eSKate Stone       result.AppendErrorWithFormat("Invalid arguments for '%s'.\nUsage: %s\n",
477b9c1b51eSKate Stone                                    m_cmd_name.c_str(), m_cmd_syntax.c_str());
4785aee162fSJim Ingham       result.SetStatus(eReturnStatusFailed);
479926af0cdSOleksiy Vyalov       return false;
48071337622SGreg Clayton     }
4815aee162fSJim Ingham 
482926af0cdSOleksiy Vyalov     m_interpreter.UpdateExecutionContext(nullptr);
48337386143SOleksiy Vyalov     StreamString stream;
48437386143SOleksiy Vyalov     const auto error = target->Attach(m_options.attach_info, &stream);
485b9c1b51eSKate Stone     if (error.Success()) {
48637386143SOleksiy Vyalov       ProcessSP process_sp(target->GetProcessSP());
487b9c1b51eSKate Stone       if (process_sp) {
488c156427dSZachary Turner         result.AppendMessage(stream.GetString());
489bb3a283bSJim Ingham         result.SetStatus(eReturnStatusSuccessFinishNoResult);
49037386143SOleksiy Vyalov         result.SetDidChangeProcessState(true);
491962260c8SJim Ingham         result.SetAbnormalStopWasExpected(true);
492b9c1b51eSKate Stone       } else {
493b9c1b51eSKate Stone         result.AppendError(
494b9c1b51eSKate Stone             "no error returned from Target::Attach, and target has no process");
495aa739093SJohnny Chen         result.SetStatus(eReturnStatusFailed);
49644d93782SGreg Clayton       }
497b9c1b51eSKate Stone     } else {
49844d93782SGreg Clayton       result.AppendErrorWithFormat("attach failed: %s\n", error.AsCString());
49944d93782SGreg Clayton       result.SetStatus(eReturnStatusFailed);
500aa739093SJohnny Chen     }
5015aee162fSJim Ingham 
502926af0cdSOleksiy Vyalov     if (!result.Succeeded())
503926af0cdSOleksiy Vyalov       return false;
504926af0cdSOleksiy Vyalov 
505b9c1b51eSKate Stone     // Okay, we're done.  Last step is to warn if the executable module has
506b9c1b51eSKate Stone     // changed:
507513c26ceSGreg Clayton     char new_path[PATH_MAX];
508aa149cbdSGreg Clayton     ModuleSP new_exec_module_sp(target->GetExecutableModule());
509b9c1b51eSKate Stone     if (!old_exec_module_sp) {
510513c26ceSGreg Clayton       // We might not have a module if we attached to a raw pid...
511b9c1b51eSKate Stone       if (new_exec_module_sp) {
512aa149cbdSGreg Clayton         new_exec_module_sp->GetFileSpec().GetPath(new_path, PATH_MAX);
513b9c1b51eSKate Stone         result.AppendMessageWithFormat("Executable module set to \"%s\".\n",
514b9c1b51eSKate Stone                                        new_path);
515513c26ceSGreg Clayton       }
516b9c1b51eSKate Stone     } else if (old_exec_module_sp->GetFileSpec() !=
517b9c1b51eSKate Stone                new_exec_module_sp->GetFileSpec()) {
518513c26ceSGreg Clayton       char old_path[PATH_MAX];
5195aee162fSJim Ingham 
5205aee162fSJim Ingham       old_exec_module_sp->GetFileSpec().GetPath(old_path, PATH_MAX);
521aa149cbdSGreg Clayton       new_exec_module_sp->GetFileSpec().GetPath(new_path, PATH_MAX);
5225aee162fSJim Ingham 
523b9c1b51eSKate Stone       result.AppendWarningWithFormat(
524b9c1b51eSKate Stone           "Executable module changed from \"%s\" to \"%s\".\n", old_path,
525b9c1b51eSKate Stone           new_path);
5265aee162fSJim Ingham     }
5275aee162fSJim Ingham 
528b9c1b51eSKate Stone     if (!old_arch_spec.IsValid()) {
529b9c1b51eSKate Stone       result.AppendMessageWithFormat(
530b9c1b51eSKate Stone           "Architecture set to: %s.\n",
531b9c1b51eSKate Stone           target->GetArchitecture().GetTriple().getTriple().c_str());
532b9c1b51eSKate Stone     } else if (!old_arch_spec.IsExactMatch(target->GetArchitecture())) {
533b9c1b51eSKate Stone       result.AppendWarningWithFormat(
534b9c1b51eSKate Stone           "Architecture changed from %s to %s.\n",
535c1b1f1eaSGreg Clayton           old_arch_spec.GetTriple().getTriple().c_str(),
536c1b1f1eaSGreg Clayton           target->GetArchitecture().GetTriple().getTriple().c_str());
5375aee162fSJim Ingham     }
538a95ce623SJohnny Chen 
53905097246SAdrian Prantl     // This supports the use-case scenario of immediately continuing the
54005097246SAdrian Prantl     // process once attached.
541a95ce623SJohnny Chen     if (m_options.attach_info.GetContinueOnceAttached())
5425bcaf583SSean Callanan       m_interpreter.HandleCommand("process continue", eLazyBoolNo, result);
543926af0cdSOleksiy Vyalov 
5445aee162fSJim Ingham     return result.Succeeded();
5455aee162fSJim Ingham   }
5465aee162fSJim Ingham 
54730fdc8d8SChris Lattner   CommandOptions m_options;
54830fdc8d8SChris Lattner };
54930fdc8d8SChris Lattner 
55030fdc8d8SChris Lattner //-------------------------------------------------------------------------
55130fdc8d8SChris Lattner // CommandObjectProcessContinue
55230fdc8d8SChris Lattner //-------------------------------------------------------------------------
5531f0f5b5bSZachary Turner 
5548fe53c49STatyana Krasnukha static constexpr OptionDefinition g_process_continue_options[] = {
5551f0f5b5bSZachary Turner     // clang-format off
5568fe53c49STatyana Krasnukha   { LLDB_OPT_SET_ALL, false, "ignore-count",'i', OptionParser::eRequiredArgument, nullptr, {}, 0, eArgTypeUnsignedInteger, "Ignore <N> crossings of the breakpoint (if it exists) for the currently selected thread." }
5571f0f5b5bSZachary Turner     // clang-format on
5581f0f5b5bSZachary Turner };
5591f0f5b5bSZachary Turner 
560bb9caf73SJim Ingham #pragma mark CommandObjectProcessContinue
56130fdc8d8SChris Lattner 
562b9c1b51eSKate Stone class CommandObjectProcessContinue : public CommandObjectParsed {
56330fdc8d8SChris Lattner public:
564b9c1b51eSKate Stone   CommandObjectProcessContinue(CommandInterpreter &interpreter)
565b9c1b51eSKate Stone       : CommandObjectParsed(
566b9c1b51eSKate Stone             interpreter, "process continue",
567e3d26315SCaroline Tice             "Continue execution of all threads in the current process.",
56830fdc8d8SChris Lattner             "process continue",
569b9c1b51eSKate Stone             eCommandRequiresProcess | eCommandTryTargetAPILock |
570b9c1b51eSKate Stone                 eCommandProcessMustBeLaunched | eCommandProcessMustBePaused),
571b9c1b51eSKate Stone         m_options() {}
57230fdc8d8SChris Lattner 
57349bcfd80SEugene Zelenko   ~CommandObjectProcessContinue() override = default;
57430fdc8d8SChris Lattner 
5755a988416SJim Ingham protected:
576b9c1b51eSKate Stone   class CommandOptions : public Options {
5770e41084aSJim Ingham   public:
578b9c1b51eSKate Stone     CommandOptions() : Options() {
579b9c1b51eSKate Stone       // Keep default values of all options in one place: OptionParsingStarting
580b9c1b51eSKate Stone       // ()
581e1cfbc79STodd Fiala       OptionParsingStarting(nullptr);
5820e41084aSJim Ingham     }
5830e41084aSJim Ingham 
58449bcfd80SEugene Zelenko     ~CommandOptions() override = default;
5850e41084aSJim Ingham 
58697206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
587b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
58897206d57SZachary Turner       Status error;
5893bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
590b9c1b51eSKate Stone       switch (short_option) {
5910e41084aSJim Ingham       case 'i':
592fe11483bSZachary Turner         if (option_arg.getAsInteger(0, m_ignore))
593b9c1b51eSKate Stone           error.SetErrorStringWithFormat(
594b9c1b51eSKate Stone               "invalid value for ignore option: \"%s\", should be a number.",
595fe11483bSZachary Turner               option_arg.str().c_str());
5960e41084aSJim Ingham         break;
5970e41084aSJim Ingham 
5980e41084aSJim Ingham       default:
599b9c1b51eSKate Stone         error.SetErrorStringWithFormat("invalid short option character '%c'",
600b9c1b51eSKate Stone                                        short_option);
6010e41084aSJim Ingham         break;
6020e41084aSJim Ingham       }
6030e41084aSJim Ingham       return error;
6040e41084aSJim Ingham     }
6050e41084aSJim Ingham 
606b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
6070e41084aSJim Ingham       m_ignore = 0;
6080e41084aSJim Ingham     }
6090e41084aSJim Ingham 
6101f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
61170602439SZachary Turner       return llvm::makeArrayRef(g_process_continue_options);
6121f0f5b5bSZachary Turner     }
6130e41084aSJim Ingham 
6140e41084aSJim Ingham     uint32_t m_ignore;
6150e41084aSJim Ingham   };
6160e41084aSJim Ingham 
617b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
618f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
619a7015092SGreg Clayton     bool synchronous_execution = m_interpreter.GetSynchronous();
62030fdc8d8SChris Lattner     StateType state = process->GetState();
621b9c1b51eSKate Stone     if (state == eStateStopped) {
622b9c1b51eSKate Stone       if (command.GetArgumentCount() != 0) {
623b9c1b51eSKate Stone         result.AppendErrorWithFormat(
624b9c1b51eSKate Stone             "The '%s' command does not take any arguments.\n",
625b9c1b51eSKate Stone             m_cmd_name.c_str());
62630fdc8d8SChris Lattner         result.SetStatus(eReturnStatusFailed);
62730fdc8d8SChris Lattner         return false;
62830fdc8d8SChris Lattner       }
62930fdc8d8SChris Lattner 
630b9c1b51eSKate Stone       if (m_options.m_ignore > 0) {
6318d94ba0fSJim Ingham         ThreadSP sel_thread_sp(GetDefaultThread()->shared_from_this());
632b9c1b51eSKate Stone         if (sel_thread_sp) {
6330e41084aSJim Ingham           StopInfoSP stop_info_sp = sel_thread_sp->GetStopInfo();
634b9c1b51eSKate Stone           if (stop_info_sp &&
635b9c1b51eSKate Stone               stop_info_sp->GetStopReason() == eStopReasonBreakpoint) {
636b9c1b51eSKate Stone             lldb::break_id_t bp_site_id =
637b9c1b51eSKate Stone                 (lldb::break_id_t)stop_info_sp->GetValue();
638b9c1b51eSKate Stone             BreakpointSiteSP bp_site_sp(
639b9c1b51eSKate Stone                 process->GetBreakpointSiteList().FindByID(bp_site_id));
640b9c1b51eSKate Stone             if (bp_site_sp) {
641c7bece56SGreg Clayton               const size_t num_owners = bp_site_sp->GetNumberOfOwners();
642b9c1b51eSKate Stone               for (size_t i = 0; i < num_owners; i++) {
643b9c1b51eSKate Stone                 Breakpoint &bp_ref =
644b9c1b51eSKate Stone                     bp_site_sp->GetOwnerAtIndex(i)->GetBreakpoint();
645b9c1b51eSKate Stone                 if (!bp_ref.IsInternal()) {
6460e41084aSJim Ingham                   bp_ref.SetIgnoreCount(m_options.m_ignore);
6470e41084aSJim Ingham                 }
6480e41084aSJim Ingham               }
6490e41084aSJim Ingham             }
6500e41084aSJim Ingham           }
6510e41084aSJim Ingham         }
6520e41084aSJim Ingham       }
6530e41084aSJim Ingham 
65441f2b940SJim Ingham       { // Scope for thread list mutex:
655b9c1b51eSKate Stone         std::lock_guard<std::recursive_mutex> guard(
656b9c1b51eSKate Stone             process->GetThreadList().GetMutex());
65730fdc8d8SChris Lattner         const uint32_t num_threads = process->GetThreadList().GetSize();
65830fdc8d8SChris Lattner 
65930fdc8d8SChris Lattner         // Set the actions that the threads should each take when resuming
660b9c1b51eSKate Stone         for (uint32_t idx = 0; idx < num_threads; ++idx) {
6616c9ed91cSJim Ingham           const bool override_suspend = false;
662b9c1b51eSKate Stone           process->GetThreadList().GetThreadAtIndex(idx)->SetResumeState(
663b9c1b51eSKate Stone               eStateRunning, override_suspend);
66430fdc8d8SChris Lattner         }
66541f2b940SJim Ingham       }
66630fdc8d8SChris Lattner 
6674446487dSPavel Labath       const uint32_t iohandler_id = process->GetIOHandlerID();
6684446487dSPavel Labath 
669dc6224e0SGreg Clayton       StreamString stream;
67097206d57SZachary Turner       Status error;
671dc6224e0SGreg Clayton       if (synchronous_execution)
672dc6224e0SGreg Clayton         error = process->ResumeSynchronous(&stream);
673dc6224e0SGreg Clayton       else
674dc6224e0SGreg Clayton         error = process->Resume();
675a3b89e27STodd Fiala 
676b9c1b51eSKate Stone       if (error.Success()) {
677b9c1b51eSKate Stone         // There is a race condition where this thread will return up the call
67805097246SAdrian Prantl         // stack to the main command handler and show an (lldb) prompt before
67905097246SAdrian Prantl         // HandlePrivateEvent (from PrivateStateThread) has a chance to call
68005097246SAdrian Prantl         // PushProcessIOHandler().
6813879fe00SPavel Labath         process->SyncIOHandler(iohandler_id, std::chrono::seconds(2));
682a3b89e27STodd Fiala 
683b9c1b51eSKate Stone         result.AppendMessageWithFormat("Process %" PRIu64 " resuming\n",
684b9c1b51eSKate Stone                                        process->GetID());
685b9c1b51eSKate Stone         if (synchronous_execution) {
686b9c1b51eSKate Stone           // If any state changed events had anything to say, add that to the
687b9c1b51eSKate Stone           // result
688c156427dSZachary Turner           result.AppendMessage(stream.GetString());
68930fdc8d8SChris Lattner 
69030fdc8d8SChris Lattner           result.SetDidChangeProcessState(true);
69130fdc8d8SChris Lattner           result.SetStatus(eReturnStatusSuccessFinishNoResult);
692b9c1b51eSKate Stone         } else {
69330fdc8d8SChris Lattner           result.SetStatus(eReturnStatusSuccessContinuingNoResult);
69430fdc8d8SChris Lattner         }
695b9c1b51eSKate Stone       } else {
696b9c1b51eSKate Stone         result.AppendErrorWithFormat("Failed to resume process: %s.\n",
697b9c1b51eSKate Stone                                      error.AsCString());
69830fdc8d8SChris Lattner         result.SetStatus(eReturnStatusFailed);
69930fdc8d8SChris Lattner       }
700b9c1b51eSKate Stone     } else {
701b9c1b51eSKate Stone       result.AppendErrorWithFormat(
702b9c1b51eSKate Stone           "Process cannot be continued from its current state (%s).\n",
70330fdc8d8SChris Lattner           StateAsCString(state));
70430fdc8d8SChris Lattner       result.SetStatus(eReturnStatusFailed);
70530fdc8d8SChris Lattner     }
70630fdc8d8SChris Lattner     return result.Succeeded();
70730fdc8d8SChris Lattner   }
7080e41084aSJim Ingham 
709b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
7100e41084aSJim Ingham 
7110e41084aSJim Ingham   CommandOptions m_options;
7120e41084aSJim Ingham };
7130e41084aSJim Ingham 
71430fdc8d8SChris Lattner //-------------------------------------------------------------------------
71530fdc8d8SChris Lattner // CommandObjectProcessDetach
71630fdc8d8SChris Lattner //-------------------------------------------------------------------------
7178fe53c49STatyana Krasnukha static constexpr OptionDefinition g_process_detach_options[] = {
7181f0f5b5bSZachary Turner     // clang-format off
7198fe53c49STatyana Krasnukha   { LLDB_OPT_SET_1, false, "keep-stopped", 's', OptionParser::eRequiredArgument, nullptr, {}, 0, eArgTypeBoolean, "Whether or not the process should be kept stopped on detach (if possible)." },
7201f0f5b5bSZachary Turner     // clang-format on
7211f0f5b5bSZachary Turner };
7221f0f5b5bSZachary Turner 
723bb9caf73SJim Ingham #pragma mark CommandObjectProcessDetach
72430fdc8d8SChris Lattner 
725b9c1b51eSKate Stone class CommandObjectProcessDetach : public CommandObjectParsed {
72630fdc8d8SChris Lattner public:
727b9c1b51eSKate Stone   class CommandOptions : public Options {
728acff8950SJim Ingham   public:
729b9c1b51eSKate Stone     CommandOptions() : Options() { OptionParsingStarting(nullptr); }
730acff8950SJim Ingham 
73149bcfd80SEugene Zelenko     ~CommandOptions() override = default;
732acff8950SJim Ingham 
73397206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
734b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
73597206d57SZachary Turner       Status error;
736acff8950SJim Ingham       const int short_option = m_getopt_table[option_idx].val;
737acff8950SJim Ingham 
738b9c1b51eSKate Stone       switch (short_option) {
739acff8950SJim Ingham       case 's':
740acff8950SJim Ingham         bool tmp_result;
741acff8950SJim Ingham         bool success;
74247cbf4a0SPavel Labath         tmp_result = OptionArgParser::ToBoolean(option_arg, false, &success);
743acff8950SJim Ingham         if (!success)
744b9c1b51eSKate Stone           error.SetErrorStringWithFormat("invalid boolean option: \"%s\"",
745fe11483bSZachary Turner                                          option_arg.str().c_str());
746b9c1b51eSKate Stone         else {
747acff8950SJim Ingham           if (tmp_result)
748acff8950SJim Ingham             m_keep_stopped = eLazyBoolYes;
749acff8950SJim Ingham           else
750acff8950SJim Ingham             m_keep_stopped = eLazyBoolNo;
751acff8950SJim Ingham         }
752acff8950SJim Ingham         break;
753acff8950SJim Ingham       default:
754b9c1b51eSKate Stone         error.SetErrorStringWithFormat("invalid short option character '%c'",
755b9c1b51eSKate Stone                                        short_option);
756acff8950SJim Ingham         break;
757acff8950SJim Ingham       }
758acff8950SJim Ingham       return error;
759acff8950SJim Ingham     }
760acff8950SJim Ingham 
761b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
762acff8950SJim Ingham       m_keep_stopped = eLazyBoolCalculate;
763acff8950SJim Ingham     }
764acff8950SJim Ingham 
7651f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
76670602439SZachary Turner       return llvm::makeArrayRef(g_process_detach_options);
7671f0f5b5bSZachary Turner     }
768acff8950SJim Ingham 
769acff8950SJim Ingham     // Instance variables to hold the values for command options.
770acff8950SJim Ingham     LazyBool m_keep_stopped;
771acff8950SJim Ingham   };
77230fdc8d8SChris Lattner 
7737428a18cSKate Stone   CommandObjectProcessDetach(CommandInterpreter &interpreter)
774b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process detach",
775b9c1b51eSKate Stone                             "Detach from the current target process.",
776a7015092SGreg Clayton                             "process detach",
777b9c1b51eSKate Stone                             eCommandRequiresProcess | eCommandTryTargetAPILock |
778b9c1b51eSKate Stone                                 eCommandProcessMustBeLaunched),
779b9c1b51eSKate Stone         m_options() {}
78030fdc8d8SChris Lattner 
78149bcfd80SEugene Zelenko   ~CommandObjectProcessDetach() override = default;
78230fdc8d8SChris Lattner 
783b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
784acff8950SJim Ingham 
7855a988416SJim Ingham protected:
786b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
787f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
788acff8950SJim Ingham     // FIXME: This will be a Command Option:
789acff8950SJim Ingham     bool keep_stopped;
790b9c1b51eSKate Stone     if (m_options.m_keep_stopped == eLazyBoolCalculate) {
791acff8950SJim Ingham       // Check the process default:
79249bcfd80SEugene Zelenko       keep_stopped = process->GetDetachKeepsStopped();
793b9c1b51eSKate Stone     } else if (m_options.m_keep_stopped == eLazyBoolYes)
794acff8950SJim Ingham       keep_stopped = true;
795acff8950SJim Ingham     else
796acff8950SJim Ingham       keep_stopped = false;
797acff8950SJim Ingham 
79897206d57SZachary Turner     Status error(process->Detach(keep_stopped));
799b9c1b51eSKate Stone     if (error.Success()) {
80030fdc8d8SChris Lattner       result.SetStatus(eReturnStatusSuccessFinishResult);
801b9c1b51eSKate Stone     } else {
80230fdc8d8SChris Lattner       result.AppendErrorWithFormat("Detach failed: %s\n", error.AsCString());
80330fdc8d8SChris Lattner       result.SetStatus(eReturnStatusFailed);
80430fdc8d8SChris Lattner       return false;
80530fdc8d8SChris Lattner     }
80630fdc8d8SChris Lattner     return result.Succeeded();
80730fdc8d8SChris Lattner   }
808acff8950SJim Ingham 
809acff8950SJim Ingham   CommandOptions m_options;
810acff8950SJim Ingham };
811acff8950SJim Ingham 
81230fdc8d8SChris Lattner //-------------------------------------------------------------------------
813b766a73dSGreg Clayton // CommandObjectProcessConnect
814b766a73dSGreg Clayton //-------------------------------------------------------------------------
8151f0f5b5bSZachary Turner 
8168fe53c49STatyana Krasnukha static constexpr OptionDefinition g_process_connect_options[] = {
8171f0f5b5bSZachary Turner     // clang-format off
8188fe53c49STatyana Krasnukha   { LLDB_OPT_SET_ALL, false, "plugin", 'p', OptionParser::eRequiredArgument, nullptr, {}, 0, eArgTypePlugin, "Name of the process plugin you want to use." },
8191f0f5b5bSZachary Turner     // clang-format on
8201f0f5b5bSZachary Turner };
8211f0f5b5bSZachary Turner 
822b766a73dSGreg Clayton #pragma mark CommandObjectProcessConnect
823b766a73dSGreg Clayton 
824b9c1b51eSKate Stone class CommandObjectProcessConnect : public CommandObjectParsed {
825b766a73dSGreg Clayton public:
826b9c1b51eSKate Stone   class CommandOptions : public Options {
827b766a73dSGreg Clayton   public:
828b9c1b51eSKate Stone     CommandOptions() : Options() {
829b9c1b51eSKate Stone       // Keep default values of all options in one place: OptionParsingStarting
830b9c1b51eSKate Stone       // ()
831e1cfbc79STodd Fiala       OptionParsingStarting(nullptr);
832b766a73dSGreg Clayton     }
833b766a73dSGreg Clayton 
83449bcfd80SEugene Zelenko     ~CommandOptions() override = default;
835b766a73dSGreg Clayton 
83697206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
837b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
83897206d57SZachary Turner       Status error;
8393bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
840b766a73dSGreg Clayton 
841b9c1b51eSKate Stone       switch (short_option) {
842b766a73dSGreg Clayton       case 'p':
843b766a73dSGreg Clayton         plugin_name.assign(option_arg);
844b766a73dSGreg Clayton         break;
845b766a73dSGreg Clayton 
846b766a73dSGreg Clayton       default:
847b9c1b51eSKate Stone         error.SetErrorStringWithFormat("invalid short option character '%c'",
848b9c1b51eSKate Stone                                        short_option);
849b766a73dSGreg Clayton         break;
850b766a73dSGreg Clayton       }
851b766a73dSGreg Clayton       return error;
852b766a73dSGreg Clayton     }
853b766a73dSGreg Clayton 
854b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
855b766a73dSGreg Clayton       plugin_name.clear();
856b766a73dSGreg Clayton     }
857b766a73dSGreg Clayton 
8581f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
85970602439SZachary Turner       return llvm::makeArrayRef(g_process_connect_options);
8601f0f5b5bSZachary Turner     }
861b766a73dSGreg Clayton 
862b766a73dSGreg Clayton     // Instance variables to hold the values for command options.
863b766a73dSGreg Clayton 
864b766a73dSGreg Clayton     std::string plugin_name;
865b766a73dSGreg Clayton   };
866b766a73dSGreg Clayton 
867b9c1b51eSKate Stone   CommandObjectProcessConnect(CommandInterpreter &interpreter)
868b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process connect",
869b766a73dSGreg Clayton                             "Connect to a remote debug service.",
870b9c1b51eSKate Stone                             "process connect <remote-url>", 0),
871b9c1b51eSKate Stone         m_options() {}
872b766a73dSGreg Clayton 
87349bcfd80SEugene Zelenko   ~CommandObjectProcessConnect() override = default;
874b766a73dSGreg Clayton 
875b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
8765a988416SJim Ingham 
8775a988416SJim Ingham protected:
878b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
879b9c1b51eSKate Stone     if (command.GetArgumentCount() != 1) {
880b9c1b51eSKate Stone       result.AppendErrorWithFormat(
881b9c1b51eSKate Stone           "'%s' takes exactly one argument:\nUsage: %s\n", m_cmd_name.c_str(),
882ccd6cffbSTamas Berghammer           m_cmd_syntax.c_str());
883ccd6cffbSTamas Berghammer       result.SetStatus(eReturnStatusFailed);
884ccd6cffbSTamas Berghammer       return false;
885ccd6cffbSTamas Berghammer     }
886ccd6cffbSTamas Berghammer 
887ccd6cffbSTamas Berghammer     Process *process = m_exe_ctx.GetProcessPtr();
888b9c1b51eSKate Stone     if (process && process->IsAlive()) {
889b9c1b51eSKate Stone       result.AppendErrorWithFormat(
890b9c1b51eSKate Stone           "Process %" PRIu64
891b9c1b51eSKate Stone           " is currently being debugged, kill the process before connecting.\n",
892b766a73dSGreg Clayton           process->GetID());
893b766a73dSGreg Clayton       result.SetStatus(eReturnStatusFailed);
894b766a73dSGreg Clayton       return false;
895b766a73dSGreg Clayton     }
896b766a73dSGreg Clayton 
897ccd6cffbSTamas Berghammer     const char *plugin_name = nullptr;
898b766a73dSGreg Clayton     if (!m_options.plugin_name.empty())
899b766a73dSGreg Clayton       plugin_name = m_options.plugin_name.c_str();
900b766a73dSGreg Clayton 
90197206d57SZachary Turner     Status error;
902ccd6cffbSTamas Berghammer     Debugger &debugger = m_interpreter.GetDebugger();
903ccd6cffbSTamas Berghammer     PlatformSP platform_sp = m_interpreter.GetPlatform(true);
904b9c1b51eSKate Stone     ProcessSP process_sp = platform_sp->ConnectProcess(
905b9c1b51eSKate Stone         command.GetArgumentAtIndex(0), plugin_name, debugger,
906b9c1b51eSKate Stone         debugger.GetSelectedTarget().get(), error);
907b9c1b51eSKate Stone     if (error.Fail() || process_sp == nullptr) {
908ccd6cffbSTamas Berghammer       result.AppendError(error.AsCString("Error connecting to the process"));
909b766a73dSGreg Clayton       result.SetStatus(eReturnStatusFailed);
910b766a73dSGreg Clayton       return false;
911b766a73dSGreg Clayton     }
912ccd6cffbSTamas Berghammer     return true;
913b766a73dSGreg Clayton   }
914b766a73dSGreg Clayton 
915b766a73dSGreg Clayton   CommandOptions m_options;
916b766a73dSGreg Clayton };
917b766a73dSGreg Clayton 
918b766a73dSGreg Clayton //-------------------------------------------------------------------------
919998255bfSGreg Clayton // CommandObjectProcessPlugin
920998255bfSGreg Clayton //-------------------------------------------------------------------------
921998255bfSGreg Clayton #pragma mark CommandObjectProcessPlugin
922998255bfSGreg Clayton 
923b9c1b51eSKate Stone class CommandObjectProcessPlugin : public CommandObjectProxy {
924998255bfSGreg Clayton public:
9257428a18cSKate Stone   CommandObjectProcessPlugin(CommandInterpreter &interpreter)
926b9c1b51eSKate Stone       : CommandObjectProxy(
927b9c1b51eSKate Stone             interpreter, "process plugin",
928b9c1b51eSKate Stone             "Send a custom command to the current target process plug-in.",
929b9c1b51eSKate Stone             "process plugin <args>", 0) {}
930998255bfSGreg Clayton 
93149bcfd80SEugene Zelenko   ~CommandObjectProcessPlugin() override = default;
932998255bfSGreg Clayton 
933b9c1b51eSKate Stone   CommandObject *GetProxyCommandObject() override {
934e05b2efeSGreg Clayton     Process *process = m_interpreter.GetExecutionContext().GetProcessPtr();
935998255bfSGreg Clayton     if (process)
936998255bfSGreg Clayton       return process->GetPluginCommandObject();
93749bcfd80SEugene Zelenko     return nullptr;
938998255bfSGreg Clayton   }
939998255bfSGreg Clayton };
940998255bfSGreg Clayton 
941998255bfSGreg Clayton //-------------------------------------------------------------------------
9428f343b09SGreg Clayton // CommandObjectProcessLoad
9438f343b09SGreg Clayton //-------------------------------------------------------------------------
9441f0f5b5bSZachary Turner 
9458fe53c49STatyana Krasnukha static constexpr OptionDefinition g_process_load_options[] = {
9461f0f5b5bSZachary Turner     // clang-format off
9478fe53c49STatyana Krasnukha   { LLDB_OPT_SET_ALL, false, "install", 'i', OptionParser::eOptionalArgument, 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." },
9481f0f5b5bSZachary Turner     // clang-format on
9491f0f5b5bSZachary Turner };
9501f0f5b5bSZachary Turner 
951bb9caf73SJim Ingham #pragma mark CommandObjectProcessLoad
9528f343b09SGreg Clayton 
953b9c1b51eSKate Stone class CommandObjectProcessLoad : public CommandObjectParsed {
9548f343b09SGreg Clayton public:
955b9c1b51eSKate Stone   class CommandOptions : public Options {
9564fbd67acSTamas Berghammer   public:
957b9c1b51eSKate Stone     CommandOptions() : Options() {
958b9c1b51eSKate Stone       // Keep default values of all options in one place: OptionParsingStarting
959b9c1b51eSKate Stone       // ()
960e1cfbc79STodd Fiala       OptionParsingStarting(nullptr);
9614fbd67acSTamas Berghammer     }
9624fbd67acSTamas Berghammer 
9634fbd67acSTamas Berghammer     ~CommandOptions() override = default;
9644fbd67acSTamas Berghammer 
96597206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
966b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
96797206d57SZachary Turner       Status error;
9684fbd67acSTamas Berghammer       const int short_option = m_getopt_table[option_idx].val;
969b9c1b51eSKate Stone       switch (short_option) {
9704fbd67acSTamas Berghammer       case 'i':
9714fbd67acSTamas Berghammer         do_install = true;
972fe11483bSZachary Turner         if (!option_arg.empty())
9738f3be7a3SJonas Devlieghere           install_path.SetFile(option_arg, FileSpec::Style::native);
9744fbd67acSTamas Berghammer         break;
9754fbd67acSTamas Berghammer       default:
976b9c1b51eSKate Stone         error.SetErrorStringWithFormat("invalid short option character '%c'",
977b9c1b51eSKate Stone                                        short_option);
9784fbd67acSTamas Berghammer         break;
9794fbd67acSTamas Berghammer       }
9804fbd67acSTamas Berghammer       return error;
9814fbd67acSTamas Berghammer     }
9824fbd67acSTamas Berghammer 
983b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
9844fbd67acSTamas Berghammer       do_install = false;
9854fbd67acSTamas Berghammer       install_path.Clear();
9864fbd67acSTamas Berghammer     }
9874fbd67acSTamas Berghammer 
9881f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
98970602439SZachary Turner       return llvm::makeArrayRef(g_process_load_options);
9901f0f5b5bSZachary Turner     }
9914fbd67acSTamas Berghammer 
9924fbd67acSTamas Berghammer     // Instance variables to hold the values for command options.
9934fbd67acSTamas Berghammer     bool do_install;
9944fbd67acSTamas Berghammer     FileSpec install_path;
9954fbd67acSTamas Berghammer   };
9968f343b09SGreg Clayton 
997b9c1b51eSKate Stone   CommandObjectProcessLoad(CommandInterpreter &interpreter)
998b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process load",
9998f343b09SGreg Clayton                             "Load a shared library into the current process.",
10008f343b09SGreg Clayton                             "process load <filename> [<filename> ...]",
1001b9c1b51eSKate Stone                             eCommandRequiresProcess | eCommandTryTargetAPILock |
1002e87764f2SEnrico Granata                                 eCommandProcessMustBeLaunched |
10034fbd67acSTamas Berghammer                                 eCommandProcessMustBePaused),
1004b9c1b51eSKate Stone         m_options() {}
10058f343b09SGreg Clayton 
10064fbd67acSTamas Berghammer   ~CommandObjectProcessLoad() override = default;
10074fbd67acSTamas Berghammer 
1008b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
10098f343b09SGreg Clayton 
10105a988416SJim Ingham protected:
1011b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1012f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
10138f343b09SGreg Clayton 
101497d2c401SZachary Turner     for (auto &entry : command.entries()) {
101597206d57SZachary Turner       Status error;
10163cb132a0STamas Berghammer       PlatformSP platform = process->GetTarget().GetPlatform();
101797d2c401SZachary Turner       llvm::StringRef image_path = entry.ref;
10184fbd67acSTamas Berghammer       uint32_t image_token = LLDB_INVALID_IMAGE_TOKEN;
10194fbd67acSTamas Berghammer 
1020b9c1b51eSKate Stone       if (!m_options.do_install) {
10218f3be7a3SJonas Devlieghere         FileSpec image_spec(image_path);
10223cb132a0STamas Berghammer         platform->ResolveRemotePath(image_spec, image_spec);
1023b9c1b51eSKate Stone         image_token =
1024b9c1b51eSKate Stone             platform->LoadImage(process, FileSpec(), image_spec, error);
1025b9c1b51eSKate Stone       } else if (m_options.install_path) {
10268f3be7a3SJonas Devlieghere         FileSpec image_spec(image_path);
10278f3be7a3SJonas Devlieghere         FileSystem::Instance().Resolve(image_spec);
1028b9c1b51eSKate Stone         platform->ResolveRemotePath(m_options.install_path,
1029b9c1b51eSKate Stone                                     m_options.install_path);
1030b9c1b51eSKate Stone         image_token = platform->LoadImage(process, image_spec,
1031b9c1b51eSKate Stone                                           m_options.install_path, error);
1032b9c1b51eSKate Stone       } else {
10338f3be7a3SJonas Devlieghere         FileSpec image_spec(image_path);
10348f3be7a3SJonas Devlieghere         FileSystem::Instance().Resolve(image_spec);
1035b9c1b51eSKate Stone         image_token =
1036b9c1b51eSKate Stone             platform->LoadImage(process, image_spec, FileSpec(), error);
10374fbd67acSTamas Berghammer       }
10384fbd67acSTamas Berghammer 
1039b9c1b51eSKate Stone       if (image_token != LLDB_INVALID_IMAGE_TOKEN) {
1040b9c1b51eSKate Stone         result.AppendMessageWithFormat(
104197d2c401SZachary Turner             "Loading \"%s\"...ok\nImage %u loaded.\n", image_path.str().c_str(),
104297d2c401SZachary Turner             image_token);
10438f343b09SGreg Clayton         result.SetStatus(eReturnStatusSuccessFinishResult);
1044b9c1b51eSKate Stone       } else {
104597d2c401SZachary Turner         result.AppendErrorWithFormat("failed to load '%s': %s",
104697d2c401SZachary Turner                                      image_path.str().c_str(),
1047b9c1b51eSKate Stone                                      error.AsCString());
10488f343b09SGreg Clayton         result.SetStatus(eReturnStatusFailed);
10498f343b09SGreg Clayton       }
10508f343b09SGreg Clayton     }
10518f343b09SGreg Clayton     return result.Succeeded();
10528f343b09SGreg Clayton   }
10534fbd67acSTamas Berghammer 
10544fbd67acSTamas Berghammer   CommandOptions m_options;
10558f343b09SGreg Clayton };
10568f343b09SGreg Clayton 
10578f343b09SGreg Clayton //-------------------------------------------------------------------------
10588f343b09SGreg Clayton // CommandObjectProcessUnload
10598f343b09SGreg Clayton //-------------------------------------------------------------------------
1060bb9caf73SJim Ingham #pragma mark CommandObjectProcessUnload
10618f343b09SGreg Clayton 
1062b9c1b51eSKate Stone class CommandObjectProcessUnload : public CommandObjectParsed {
10638f343b09SGreg Clayton public:
1064b9c1b51eSKate Stone   CommandObjectProcessUnload(CommandInterpreter &interpreter)
1065b9c1b51eSKate Stone       : CommandObjectParsed(
1066b9c1b51eSKate Stone             interpreter, "process unload",
1067b9c1b51eSKate Stone             "Unload a shared library from the current process using the index "
1068b9c1b51eSKate Stone             "returned by a previous call to \"process load\".",
10698f343b09SGreg Clayton             "process unload <index>",
1070b9c1b51eSKate Stone             eCommandRequiresProcess | eCommandTryTargetAPILock |
1071b9c1b51eSKate Stone                 eCommandProcessMustBeLaunched | eCommandProcessMustBePaused) {}
10728f343b09SGreg Clayton 
107349bcfd80SEugene Zelenko   ~CommandObjectProcessUnload() override = default;
10748f343b09SGreg Clayton 
10755a988416SJim Ingham protected:
1076b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1077f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
10788f343b09SGreg Clayton 
107997d2c401SZachary Turner     for (auto &entry : command.entries()) {
108097d2c401SZachary Turner       uint32_t image_token;
108197d2c401SZachary Turner       if (entry.ref.getAsInteger(0, image_token)) {
1082b9c1b51eSKate Stone         result.AppendErrorWithFormat("invalid image index argument '%s'",
108397d2c401SZachary Turner                                      entry.ref.str().c_str());
10848f343b09SGreg Clayton         result.SetStatus(eReturnStatusFailed);
10858f343b09SGreg Clayton         break;
1086b9c1b51eSKate Stone       } else {
108797206d57SZachary Turner         Status error(process->GetTarget().GetPlatform()->UnloadImage(
1088b9c1b51eSKate Stone             process, image_token));
1089b9c1b51eSKate Stone         if (error.Success()) {
1090b9c1b51eSKate Stone           result.AppendMessageWithFormat(
1091b9c1b51eSKate Stone               "Unloading shared library with index %u...ok\n", image_token);
10928f343b09SGreg Clayton           result.SetStatus(eReturnStatusSuccessFinishResult);
1093b9c1b51eSKate Stone         } else {
1094b9c1b51eSKate Stone           result.AppendErrorWithFormat("failed to unload image: %s",
1095b9c1b51eSKate Stone                                        error.AsCString());
10968f343b09SGreg Clayton           result.SetStatus(eReturnStatusFailed);
10978f343b09SGreg Clayton           break;
10988f343b09SGreg Clayton         }
10998f343b09SGreg Clayton       }
11008f343b09SGreg Clayton     }
11018f343b09SGreg Clayton     return result.Succeeded();
11028f343b09SGreg Clayton   }
11038f343b09SGreg Clayton };
11048f343b09SGreg Clayton 
11058f343b09SGreg Clayton //-------------------------------------------------------------------------
110630fdc8d8SChris Lattner // CommandObjectProcessSignal
110730fdc8d8SChris Lattner //-------------------------------------------------------------------------
1108bb9caf73SJim Ingham #pragma mark CommandObjectProcessSignal
110930fdc8d8SChris Lattner 
1110b9c1b51eSKate Stone class CommandObjectProcessSignal : public CommandObjectParsed {
111130fdc8d8SChris Lattner public:
11127428a18cSKate Stone   CommandObjectProcessSignal(CommandInterpreter &interpreter)
1113b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process signal",
1114b9c1b51eSKate Stone                             "Send a UNIX signal to the current target process.",
1115b9c1b51eSKate Stone                             nullptr, eCommandRequiresProcess |
1116b9c1b51eSKate Stone                                          eCommandTryTargetAPILock) {
1117405fe67fSCaroline Tice     CommandArgumentEntry arg;
1118405fe67fSCaroline Tice     CommandArgumentData signal_arg;
1119405fe67fSCaroline Tice 
1120405fe67fSCaroline Tice     // Define the first (and only) variant of this arg.
1121c0dbdfb6SCaroline Tice     signal_arg.arg_type = eArgTypeUnixSignal;
1122405fe67fSCaroline Tice     signal_arg.arg_repetition = eArgRepeatPlain;
1123405fe67fSCaroline Tice 
1124b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
1125b9c1b51eSKate Stone     // argument entry.
1126405fe67fSCaroline Tice     arg.push_back(signal_arg);
1127405fe67fSCaroline Tice 
1128405fe67fSCaroline Tice     // Push the data for the first argument into the m_arguments vector.
1129405fe67fSCaroline Tice     m_arguments.push_back(arg);
113030fdc8d8SChris Lattner   }
113130fdc8d8SChris Lattner 
113249bcfd80SEugene Zelenko   ~CommandObjectProcessSignal() override = default;
113330fdc8d8SChris Lattner 
11345a988416SJim Ingham protected:
1135b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1136f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
113730fdc8d8SChris Lattner 
1138b9c1b51eSKate Stone     if (command.GetArgumentCount() == 1) {
1139237cd906SGreg Clayton       int signo = LLDB_INVALID_SIGNAL_NUMBER;
1140237cd906SGreg Clayton 
1141237cd906SGreg Clayton       const char *signal_name = command.GetArgumentAtIndex(0);
1142237cd906SGreg Clayton       if (::isxdigit(signal_name[0]))
1143b9c1b51eSKate Stone         signo =
1144b9c1b51eSKate Stone             StringConvert::ToSInt32(signal_name, LLDB_INVALID_SIGNAL_NUMBER, 0);
1145237cd906SGreg Clayton       else
114698d0a4b3SChaoren Lin         signo = process->GetUnixSignals()->GetSignalNumberFromName(signal_name);
1147237cd906SGreg Clayton 
1148b9c1b51eSKate Stone       if (signo == LLDB_INVALID_SIGNAL_NUMBER) {
1149b9c1b51eSKate Stone         result.AppendErrorWithFormat("Invalid signal argument '%s'.\n",
1150b9c1b51eSKate Stone                                      command.GetArgumentAtIndex(0));
115130fdc8d8SChris Lattner         result.SetStatus(eReturnStatusFailed);
1152b9c1b51eSKate Stone       } else {
115397206d57SZachary Turner         Status error(process->Signal(signo));
1154b9c1b51eSKate Stone         if (error.Success()) {
115530fdc8d8SChris Lattner           result.SetStatus(eReturnStatusSuccessFinishResult);
1156b9c1b51eSKate Stone         } else {
1157b9c1b51eSKate Stone           result.AppendErrorWithFormat("Failed to send signal %i: %s\n", signo,
1158b9c1b51eSKate Stone                                        error.AsCString());
115930fdc8d8SChris Lattner           result.SetStatus(eReturnStatusFailed);
116030fdc8d8SChris Lattner         }
116130fdc8d8SChris Lattner       }
1162b9c1b51eSKate Stone     } else {
1163b9c1b51eSKate Stone       result.AppendErrorWithFormat(
1164b9c1b51eSKate Stone           "'%s' takes exactly one signal number argument:\nUsage: %s\n",
1165b9c1b51eSKate Stone           m_cmd_name.c_str(), m_cmd_syntax.c_str());
116630fdc8d8SChris Lattner       result.SetStatus(eReturnStatusFailed);
116730fdc8d8SChris Lattner     }
116830fdc8d8SChris Lattner     return result.Succeeded();
116930fdc8d8SChris Lattner   }
117030fdc8d8SChris Lattner };
117130fdc8d8SChris Lattner 
117230fdc8d8SChris Lattner //-------------------------------------------------------------------------
117330fdc8d8SChris Lattner // CommandObjectProcessInterrupt
117430fdc8d8SChris Lattner //-------------------------------------------------------------------------
1175bb9caf73SJim Ingham #pragma mark CommandObjectProcessInterrupt
117630fdc8d8SChris Lattner 
1177b9c1b51eSKate Stone class CommandObjectProcessInterrupt : public CommandObjectParsed {
117830fdc8d8SChris Lattner public:
11797428a18cSKate Stone   CommandObjectProcessInterrupt(CommandInterpreter &interpreter)
1180b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process interrupt",
1181b9c1b51eSKate Stone                             "Interrupt the current target process.",
1182a7015092SGreg Clayton                             "process interrupt",
1183b9c1b51eSKate Stone                             eCommandRequiresProcess | eCommandTryTargetAPILock |
1184b9c1b51eSKate Stone                                 eCommandProcessMustBeLaunched) {}
118530fdc8d8SChris Lattner 
118649bcfd80SEugene Zelenko   ~CommandObjectProcessInterrupt() override = default;
118730fdc8d8SChris Lattner 
11885a988416SJim Ingham protected:
1189b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1190f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
1191b9c1b51eSKate Stone     if (process == nullptr) {
119230fdc8d8SChris Lattner       result.AppendError("no process to halt");
119330fdc8d8SChris Lattner       result.SetStatus(eReturnStatusFailed);
119430fdc8d8SChris Lattner       return false;
119530fdc8d8SChris Lattner     }
119630fdc8d8SChris Lattner 
1197b9c1b51eSKate Stone     if (command.GetArgumentCount() == 0) {
1198f9b57b9dSGreg Clayton       bool clear_thread_plans = true;
119997206d57SZachary Turner       Status error(process->Halt(clear_thread_plans));
1200b9c1b51eSKate Stone       if (error.Success()) {
120130fdc8d8SChris Lattner         result.SetStatus(eReturnStatusSuccessFinishResult);
1202b9c1b51eSKate Stone       } else {
1203b9c1b51eSKate Stone         result.AppendErrorWithFormat("Failed to halt process: %s\n",
1204b9c1b51eSKate Stone                                      error.AsCString());
120530fdc8d8SChris Lattner         result.SetStatus(eReturnStatusFailed);
120630fdc8d8SChris Lattner       }
1207b9c1b51eSKate Stone     } else {
1208fd54b368SJason Molenda       result.AppendErrorWithFormat("'%s' takes no arguments:\nUsage: %s\n",
1209b9c1b51eSKate Stone                                    m_cmd_name.c_str(), m_cmd_syntax.c_str());
121030fdc8d8SChris Lattner       result.SetStatus(eReturnStatusFailed);
121130fdc8d8SChris Lattner     }
121230fdc8d8SChris Lattner     return result.Succeeded();
121330fdc8d8SChris Lattner   }
121430fdc8d8SChris Lattner };
121530fdc8d8SChris Lattner 
121630fdc8d8SChris Lattner //-------------------------------------------------------------------------
121730fdc8d8SChris Lattner // CommandObjectProcessKill
121830fdc8d8SChris Lattner //-------------------------------------------------------------------------
1219bb9caf73SJim Ingham #pragma mark CommandObjectProcessKill
122030fdc8d8SChris Lattner 
1221b9c1b51eSKate Stone class CommandObjectProcessKill : public CommandObjectParsed {
122230fdc8d8SChris Lattner public:
12237428a18cSKate Stone   CommandObjectProcessKill(CommandInterpreter &interpreter)
1224b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process kill",
1225b9c1b51eSKate Stone                             "Terminate the current target process.",
1226b9c1b51eSKate Stone                             "process kill",
1227b9c1b51eSKate Stone                             eCommandRequiresProcess | eCommandTryTargetAPILock |
1228b9c1b51eSKate Stone                                 eCommandProcessMustBeLaunched) {}
122930fdc8d8SChris Lattner 
123049bcfd80SEugene Zelenko   ~CommandObjectProcessKill() override = default;
123130fdc8d8SChris Lattner 
12325a988416SJim Ingham protected:
1233b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1234f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
1235b9c1b51eSKate Stone     if (process == nullptr) {
123630fdc8d8SChris Lattner       result.AppendError("no process to kill");
123730fdc8d8SChris Lattner       result.SetStatus(eReturnStatusFailed);
123830fdc8d8SChris Lattner       return false;
123930fdc8d8SChris Lattner     }
124030fdc8d8SChris Lattner 
1241b9c1b51eSKate Stone     if (command.GetArgumentCount() == 0) {
124297206d57SZachary Turner       Status error(process->Destroy(true));
1243b9c1b51eSKate Stone       if (error.Success()) {
124430fdc8d8SChris Lattner         result.SetStatus(eReturnStatusSuccessFinishResult);
1245b9c1b51eSKate Stone       } else {
1246b9c1b51eSKate Stone         result.AppendErrorWithFormat("Failed to kill process: %s\n",
1247b9c1b51eSKate Stone                                      error.AsCString());
124830fdc8d8SChris Lattner         result.SetStatus(eReturnStatusFailed);
124930fdc8d8SChris Lattner       }
1250b9c1b51eSKate Stone     } else {
1251fd54b368SJason Molenda       result.AppendErrorWithFormat("'%s' takes no arguments:\nUsage: %s\n",
1252b9c1b51eSKate Stone                                    m_cmd_name.c_str(), m_cmd_syntax.c_str());
125330fdc8d8SChris Lattner       result.SetStatus(eReturnStatusFailed);
125430fdc8d8SChris Lattner     }
125530fdc8d8SChris Lattner     return result.Succeeded();
125630fdc8d8SChris Lattner   }
125730fdc8d8SChris Lattner };
125830fdc8d8SChris Lattner 
125930fdc8d8SChris Lattner //-------------------------------------------------------------------------
1260a2715cf1SGreg Clayton // CommandObjectProcessSaveCore
1261a2715cf1SGreg Clayton //-------------------------------------------------------------------------
1262a2715cf1SGreg Clayton #pragma mark CommandObjectProcessSaveCore
1263a2715cf1SGreg Clayton 
1264b9c1b51eSKate Stone class CommandObjectProcessSaveCore : public CommandObjectParsed {
1265a2715cf1SGreg Clayton public:
1266b9c1b51eSKate Stone   CommandObjectProcessSaveCore(CommandInterpreter &interpreter)
1267b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process save-core",
1268b9c1b51eSKate Stone                             "Save the current process as a core file using an "
1269b9c1b51eSKate Stone                             "appropriate file type.",
1270a2715cf1SGreg Clayton                             "process save-core FILE",
1271b9c1b51eSKate Stone                             eCommandRequiresProcess | eCommandTryTargetAPILock |
1272b9c1b51eSKate Stone                                 eCommandProcessMustBeLaunched) {}
1273a2715cf1SGreg Clayton 
127449bcfd80SEugene Zelenko   ~CommandObjectProcessSaveCore() override = default;
1275a2715cf1SGreg Clayton 
1276a2715cf1SGreg Clayton protected:
1277b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1278a2715cf1SGreg Clayton     ProcessSP process_sp = m_exe_ctx.GetProcessSP();
1279b9c1b51eSKate Stone     if (process_sp) {
1280b9c1b51eSKate Stone       if (command.GetArgumentCount() == 1) {
12818f3be7a3SJonas Devlieghere         FileSpec output_file(command.GetArgumentAtIndex(0));
128297206d57SZachary Turner         Status error = PluginManager::SaveCore(process_sp, output_file);
1283b9c1b51eSKate Stone         if (error.Success()) {
1284a2715cf1SGreg Clayton           result.SetStatus(eReturnStatusSuccessFinishResult);
1285b9c1b51eSKate Stone         } else {
1286b9c1b51eSKate Stone           result.AppendErrorWithFormat(
1287b9c1b51eSKate Stone               "Failed to save core file for process: %s\n", error.AsCString());
1288a2715cf1SGreg Clayton           result.SetStatus(eReturnStatusFailed);
1289a2715cf1SGreg Clayton         }
1290b9c1b51eSKate Stone       } else {
1291a2715cf1SGreg Clayton         result.AppendErrorWithFormat("'%s' takes one arguments:\nUsage: %s\n",
1292b9c1b51eSKate Stone                                      m_cmd_name.c_str(), m_cmd_syntax.c_str());
1293a2715cf1SGreg Clayton         result.SetStatus(eReturnStatusFailed);
1294a2715cf1SGreg Clayton       }
1295b9c1b51eSKate Stone     } else {
1296a2715cf1SGreg Clayton       result.AppendError("invalid process");
1297a2715cf1SGreg Clayton       result.SetStatus(eReturnStatusFailed);
1298a2715cf1SGreg Clayton       return false;
1299a2715cf1SGreg Clayton     }
1300a2715cf1SGreg Clayton 
1301a2715cf1SGreg Clayton     return result.Succeeded();
1302a2715cf1SGreg Clayton   }
1303a2715cf1SGreg Clayton };
1304a2715cf1SGreg Clayton 
1305a2715cf1SGreg Clayton //-------------------------------------------------------------------------
13064b9bea87SJim Ingham // CommandObjectProcessStatus
13074b9bea87SJim Ingham //-------------------------------------------------------------------------
1308bb9caf73SJim Ingham #pragma mark CommandObjectProcessStatus
1309bb9caf73SJim Ingham 
1310b9c1b51eSKate Stone class CommandObjectProcessStatus : public CommandObjectParsed {
13114b9bea87SJim Ingham public:
13127428a18cSKate Stone   CommandObjectProcessStatus(CommandInterpreter &interpreter)
1313b9c1b51eSKate Stone       : CommandObjectParsed(
1314b9c1b51eSKate Stone             interpreter, "process status",
1315b9c1b51eSKate Stone             "Show status and stop location for the current target process.",
1316b9c1b51eSKate Stone             "process status",
1317b9c1b51eSKate Stone             eCommandRequiresProcess | eCommandTryTargetAPILock) {}
13184b9bea87SJim Ingham 
131949bcfd80SEugene Zelenko   ~CommandObjectProcessStatus() override = default;
13204b9bea87SJim Ingham 
1321b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
13227260f620SGreg Clayton     Stream &strm = result.GetOutputStream();
13234b9bea87SJim Ingham     result.SetStatus(eReturnStatusSuccessFinishNoResult);
1324b9c1b51eSKate Stone     // No need to check "process" for validity as eCommandRequiresProcess
1325b9c1b51eSKate Stone     // ensures it is valid
1326f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
13277260f620SGreg Clayton     const bool only_threads_with_stop_reason = true;
13287260f620SGreg Clayton     const uint32_t start_frame = 0;
13297260f620SGreg Clayton     const uint32_t num_frames = 1;
13307260f620SGreg Clayton     const uint32_t num_frames_with_source = 1;
13316a9767c7SJim Ingham     const bool     stop_format = true;
1332c14ee32dSGreg Clayton     process->GetStatus(strm);
1333b9c1b51eSKate Stone     process->GetThreadStatus(strm, only_threads_with_stop_reason, start_frame,
13346a9767c7SJim Ingham                              num_frames, num_frames_with_source, stop_format);
13354b9bea87SJim Ingham     return result.Succeeded();
13364b9bea87SJim Ingham   }
13374b9bea87SJim Ingham };
13384b9bea87SJim Ingham 
13394b9bea87SJim Ingham //-------------------------------------------------------------------------
134035731357SCaroline Tice // CommandObjectProcessHandle
134135731357SCaroline Tice //-------------------------------------------------------------------------
13421f0f5b5bSZachary Turner 
13438fe53c49STatyana Krasnukha static constexpr OptionDefinition g_process_handle_options[] = {
13441f0f5b5bSZachary Turner     // clang-format off
13458fe53c49STatyana Krasnukha   { LLDB_OPT_SET_1, false, "stop",   's', OptionParser::eRequiredArgument, nullptr, {}, 0, eArgTypeBoolean, "Whether or not the process should be stopped if the signal is received." },
13468fe53c49STatyana Krasnukha   { LLDB_OPT_SET_1, false, "notify", 'n', OptionParser::eRequiredArgument, nullptr, {}, 0, eArgTypeBoolean, "Whether or not the debugger should notify the user if the signal is received." },
13478fe53c49STatyana Krasnukha   { LLDB_OPT_SET_1, false, "pass",   'p', OptionParser::eRequiredArgument, nullptr, {}, 0, eArgTypeBoolean, "Whether or not the signal should be passed to the process." }
13481f0f5b5bSZachary Turner     // clang-format on
13491f0f5b5bSZachary Turner };
13501f0f5b5bSZachary Turner 
1351bb9caf73SJim Ingham #pragma mark CommandObjectProcessHandle
135235731357SCaroline Tice 
1353b9c1b51eSKate Stone class CommandObjectProcessHandle : public CommandObjectParsed {
135435731357SCaroline Tice public:
1355b9c1b51eSKate Stone   class CommandOptions : public Options {
135635731357SCaroline Tice   public:
1357b9c1b51eSKate Stone     CommandOptions() : Options() { OptionParsingStarting(nullptr); }
135835731357SCaroline Tice 
135949bcfd80SEugene Zelenko     ~CommandOptions() override = default;
136035731357SCaroline Tice 
136197206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
1362b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
136397206d57SZachary Turner       Status error;
13643bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
136535731357SCaroline Tice 
1366b9c1b51eSKate Stone       switch (short_option) {
136735731357SCaroline Tice       case 's':
136835731357SCaroline Tice         stop = option_arg;
136935731357SCaroline Tice         break;
137035731357SCaroline Tice       case 'n':
137135731357SCaroline Tice         notify = option_arg;
137235731357SCaroline Tice         break;
137335731357SCaroline Tice       case 'p':
137435731357SCaroline Tice         pass = option_arg;
137535731357SCaroline Tice         break;
137635731357SCaroline Tice       default:
1377b9c1b51eSKate Stone         error.SetErrorStringWithFormat("invalid short option character '%c'",
1378b9c1b51eSKate Stone                                        short_option);
137935731357SCaroline Tice         break;
138035731357SCaroline Tice       }
138135731357SCaroline Tice       return error;
138235731357SCaroline Tice     }
138335731357SCaroline Tice 
1384b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
138535731357SCaroline Tice       stop.clear();
138635731357SCaroline Tice       notify.clear();
138735731357SCaroline Tice       pass.clear();
138835731357SCaroline Tice     }
138935731357SCaroline Tice 
13901f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
139170602439SZachary Turner       return llvm::makeArrayRef(g_process_handle_options);
13921f0f5b5bSZachary Turner     }
139335731357SCaroline Tice 
139435731357SCaroline Tice     // Instance variables to hold the values for command options.
139535731357SCaroline Tice 
139635731357SCaroline Tice     std::string stop;
139735731357SCaroline Tice     std::string notify;
139835731357SCaroline Tice     std::string pass;
139935731357SCaroline Tice   };
140035731357SCaroline Tice 
14017428a18cSKate Stone   CommandObjectProcessHandle(CommandInterpreter &interpreter)
1402b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process handle",
1403b9c1b51eSKate Stone                             "Manage LLDB handling of OS signals for the "
1404b9c1b51eSKate Stone                             "current target process.  Defaults to showing "
1405b9c1b51eSKate Stone                             "current policy.",
140649bcfd80SEugene Zelenko                             nullptr),
1407b9c1b51eSKate Stone         m_options() {
1408ea671fbdSKate Stone     SetHelpLong("\nIf no signals are specified, update them all.  If no update "
1409ea671fbdSKate Stone                 "option is specified, list the current values.");
141035731357SCaroline Tice     CommandArgumentEntry arg;
1411c0dbdfb6SCaroline Tice     CommandArgumentData signal_arg;
141235731357SCaroline Tice 
1413c0dbdfb6SCaroline Tice     signal_arg.arg_type = eArgTypeUnixSignal;
1414c0dbdfb6SCaroline Tice     signal_arg.arg_repetition = eArgRepeatStar;
141535731357SCaroline Tice 
1416c0dbdfb6SCaroline Tice     arg.push_back(signal_arg);
141735731357SCaroline Tice 
141835731357SCaroline Tice     m_arguments.push_back(arg);
141935731357SCaroline Tice   }
142035731357SCaroline Tice 
142149bcfd80SEugene Zelenko   ~CommandObjectProcessHandle() override = default;
142235731357SCaroline Tice 
1423b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
142435731357SCaroline Tice 
1425b9c1b51eSKate Stone   bool VerifyCommandOptionValue(const std::string &option, int &real_value) {
142635731357SCaroline Tice     bool okay = true;
142710ad7993SCaroline Tice     bool success = false;
142847cbf4a0SPavel Labath     bool tmp_value = OptionArgParser::ToBoolean(option, false, &success);
142910ad7993SCaroline Tice 
143010ad7993SCaroline Tice     if (success && tmp_value)
143110ad7993SCaroline Tice       real_value = 1;
143210ad7993SCaroline Tice     else if (success && !tmp_value)
143310ad7993SCaroline Tice       real_value = 0;
1434b9c1b51eSKate Stone     else {
143535731357SCaroline Tice       // If the value isn't 'true' or 'false', it had better be 0 or 1.
14365275aaa0SVince Harron       real_value = StringConvert::ToUInt32(option.c_str(), 3);
143710ad7993SCaroline Tice       if (real_value != 0 && real_value != 1)
143835731357SCaroline Tice         okay = false;
143935731357SCaroline Tice     }
144035731357SCaroline Tice 
144135731357SCaroline Tice     return okay;
144235731357SCaroline Tice   }
144335731357SCaroline Tice 
1444b9c1b51eSKate Stone   void PrintSignalHeader(Stream &str) {
144510ad7993SCaroline Tice     str.Printf("NAME         PASS   STOP   NOTIFY\n");
1446b84141a6SPavel Labath     str.Printf("===========  =====  =====  ======\n");
144710ad7993SCaroline Tice   }
144810ad7993SCaroline Tice 
1449b9c1b51eSKate Stone   void PrintSignal(Stream &str, int32_t signo, const char *sig_name,
1450b9c1b51eSKate Stone                    const UnixSignalsSP &signals_sp) {
145110ad7993SCaroline Tice     bool stop;
145210ad7993SCaroline Tice     bool suppress;
145310ad7993SCaroline Tice     bool notify;
145410ad7993SCaroline Tice 
1455b84141a6SPavel Labath     str.Printf("%-11s  ", sig_name);
1456b9c1b51eSKate Stone     if (signals_sp->GetSignalInfo(signo, suppress, stop, notify)) {
145710ad7993SCaroline Tice       bool pass = !suppress;
1458b9c1b51eSKate Stone       str.Printf("%s  %s  %s", (pass ? "true " : "false"),
1459b9c1b51eSKate Stone                  (stop ? "true " : "false"), (notify ? "true " : "false"));
146010ad7993SCaroline Tice     }
146110ad7993SCaroline Tice     str.Printf("\n");
146210ad7993SCaroline Tice   }
146310ad7993SCaroline Tice 
1464b9c1b51eSKate Stone   void PrintSignalInformation(Stream &str, Args &signal_args,
1465b9c1b51eSKate Stone                               int num_valid_signals,
1466b9c1b51eSKate Stone                               const UnixSignalsSP &signals_sp) {
146710ad7993SCaroline Tice     PrintSignalHeader(str);
146810ad7993SCaroline Tice 
1469b9c1b51eSKate Stone     if (num_valid_signals > 0) {
147010ad7993SCaroline Tice       size_t num_args = signal_args.GetArgumentCount();
1471b9c1b51eSKate Stone       for (size_t i = 0; i < num_args; ++i) {
1472b9c1b51eSKate Stone         int32_t signo = signals_sp->GetSignalNumberFromName(
1473b9c1b51eSKate Stone             signal_args.GetArgumentAtIndex(i));
147410ad7993SCaroline Tice         if (signo != LLDB_INVALID_SIGNAL_NUMBER)
1475b9c1b51eSKate Stone           PrintSignal(str, signo, signal_args.GetArgumentAtIndex(i),
1476b9c1b51eSKate Stone                       signals_sp);
147710ad7993SCaroline Tice       }
1478b9c1b51eSKate Stone     } else // Print info for ALL signals
147910ad7993SCaroline Tice     {
148098d0a4b3SChaoren Lin       int32_t signo = signals_sp->GetFirstSignalNumber();
1481b9c1b51eSKate Stone       while (signo != LLDB_INVALID_SIGNAL_NUMBER) {
1482b9c1b51eSKate Stone         PrintSignal(str, signo, signals_sp->GetSignalAsCString(signo),
1483b9c1b51eSKate Stone                     signals_sp);
148498d0a4b3SChaoren Lin         signo = signals_sp->GetNextSignalNumber(signo);
148510ad7993SCaroline Tice       }
148610ad7993SCaroline Tice     }
148710ad7993SCaroline Tice   }
148810ad7993SCaroline Tice 
14895a988416SJim Ingham protected:
1490b9c1b51eSKate Stone   bool DoExecute(Args &signal_args, CommandReturnObject &result) override {
149135731357SCaroline Tice     TargetSP target_sp = m_interpreter.GetDebugger().GetSelectedTarget();
149235731357SCaroline Tice 
1493b9c1b51eSKate Stone     if (!target_sp) {
149435731357SCaroline Tice       result.AppendError("No current target;"
1495b9c1b51eSKate Stone                          " cannot handle signals until you have a valid target "
1496b9c1b51eSKate Stone                          "and process.\n");
149735731357SCaroline Tice       result.SetStatus(eReturnStatusFailed);
149835731357SCaroline Tice       return false;
149935731357SCaroline Tice     }
150035731357SCaroline Tice 
150135731357SCaroline Tice     ProcessSP process_sp = target_sp->GetProcessSP();
150235731357SCaroline Tice 
1503b9c1b51eSKate Stone     if (!process_sp) {
1504b9c1b51eSKate Stone       result.AppendError("No current process; cannot handle signals until you "
1505b9c1b51eSKate Stone                          "have a valid process.\n");
150635731357SCaroline Tice       result.SetStatus(eReturnStatusFailed);
150735731357SCaroline Tice       return false;
150835731357SCaroline Tice     }
150935731357SCaroline Tice 
151035731357SCaroline Tice     int stop_action = -1;   // -1 means leave the current setting alone
151135731357SCaroline Tice     int pass_action = -1;   // -1 means leave the current setting alone
151235731357SCaroline Tice     int notify_action = -1; // -1 means leave the current setting alone
151335731357SCaroline Tice 
1514b9c1b51eSKate Stone     if (!m_options.stop.empty() &&
1515b9c1b51eSKate Stone         !VerifyCommandOptionValue(m_options.stop, stop_action)) {
1516b9c1b51eSKate Stone       result.AppendError("Invalid argument for command option --stop; must be "
1517b9c1b51eSKate Stone                          "true or false.\n");
151835731357SCaroline Tice       result.SetStatus(eReturnStatusFailed);
151935731357SCaroline Tice       return false;
152035731357SCaroline Tice     }
152135731357SCaroline Tice 
1522b9c1b51eSKate Stone     if (!m_options.notify.empty() &&
1523b9c1b51eSKate Stone         !VerifyCommandOptionValue(m_options.notify, notify_action)) {
1524b9c1b51eSKate Stone       result.AppendError("Invalid argument for command option --notify; must "
1525b9c1b51eSKate Stone                          "be true or false.\n");
152635731357SCaroline Tice       result.SetStatus(eReturnStatusFailed);
152735731357SCaroline Tice       return false;
152835731357SCaroline Tice     }
152935731357SCaroline Tice 
1530b9c1b51eSKate Stone     if (!m_options.pass.empty() &&
1531b9c1b51eSKate Stone         !VerifyCommandOptionValue(m_options.pass, pass_action)) {
1532b9c1b51eSKate Stone       result.AppendError("Invalid argument for command option --pass; must be "
1533b9c1b51eSKate Stone                          "true or false.\n");
153435731357SCaroline Tice       result.SetStatus(eReturnStatusFailed);
153535731357SCaroline Tice       return false;
153635731357SCaroline Tice     }
153735731357SCaroline Tice 
153835731357SCaroline Tice     size_t num_args = signal_args.GetArgumentCount();
153998d0a4b3SChaoren Lin     UnixSignalsSP signals_sp = process_sp->GetUnixSignals();
154035731357SCaroline Tice     int num_signals_set = 0;
154135731357SCaroline Tice 
1542b9c1b51eSKate Stone     if (num_args > 0) {
1543d6a24757SZachary Turner       for (const auto &arg : signal_args) {
1544d6a24757SZachary Turner         int32_t signo = signals_sp->GetSignalNumberFromName(arg.c_str());
1545b9c1b51eSKate Stone         if (signo != LLDB_INVALID_SIGNAL_NUMBER) {
1546b9c1b51eSKate Stone           // Casting the actions as bools here should be okay, because
154705097246SAdrian Prantl           // VerifyCommandOptionValue guarantees the value is either 0 or 1.
154835731357SCaroline Tice           if (stop_action != -1)
154998d0a4b3SChaoren Lin             signals_sp->SetShouldStop(signo, stop_action);
1550b9c1b51eSKate Stone           if (pass_action != -1) {
155198d0a4b3SChaoren Lin             bool suppress = !pass_action;
155298d0a4b3SChaoren Lin             signals_sp->SetShouldSuppress(signo, suppress);
155335731357SCaroline Tice           }
155435731357SCaroline Tice           if (notify_action != -1)
155598d0a4b3SChaoren Lin             signals_sp->SetShouldNotify(signo, notify_action);
155635731357SCaroline Tice           ++num_signals_set;
1557b9c1b51eSKate Stone         } else {
1558b9c1b51eSKate Stone           result.AppendErrorWithFormat("Invalid signal name '%s'\n",
1559d6a24757SZachary Turner                                        arg.c_str());
156035731357SCaroline Tice         }
156135731357SCaroline Tice       }
1562b9c1b51eSKate Stone     } else {
1563b9c1b51eSKate Stone       // No signal specified, if any command options were specified, update ALL
1564b9c1b51eSKate Stone       // signals.
1565b9c1b51eSKate Stone       if ((notify_action != -1) || (stop_action != -1) || (pass_action != -1)) {
1566b9c1b51eSKate Stone         if (m_interpreter.Confirm(
1567b9c1b51eSKate Stone                 "Do you really want to update all the signals?", false)) {
156898d0a4b3SChaoren Lin           int32_t signo = signals_sp->GetFirstSignalNumber();
1569b9c1b51eSKate Stone           while (signo != LLDB_INVALID_SIGNAL_NUMBER) {
157010ad7993SCaroline Tice             if (notify_action != -1)
157198d0a4b3SChaoren Lin               signals_sp->SetShouldNotify(signo, notify_action);
157210ad7993SCaroline Tice             if (stop_action != -1)
157398d0a4b3SChaoren Lin               signals_sp->SetShouldStop(signo, stop_action);
1574b9c1b51eSKate Stone             if (pass_action != -1) {
157598d0a4b3SChaoren Lin               bool suppress = !pass_action;
157698d0a4b3SChaoren Lin               signals_sp->SetShouldSuppress(signo, suppress);
157710ad7993SCaroline Tice             }
157898d0a4b3SChaoren Lin             signo = signals_sp->GetNextSignalNumber(signo);
157910ad7993SCaroline Tice           }
158010ad7993SCaroline Tice         }
158110ad7993SCaroline Tice       }
158210ad7993SCaroline Tice     }
158310ad7993SCaroline Tice 
1584b9c1b51eSKate Stone     PrintSignalInformation(result.GetOutputStream(), signal_args,
1585b9c1b51eSKate Stone                            num_signals_set, signals_sp);
158635731357SCaroline Tice 
158735731357SCaroline Tice     if (num_signals_set > 0)
158835731357SCaroline Tice       result.SetStatus(eReturnStatusSuccessFinishNoResult);
158935731357SCaroline Tice     else
159035731357SCaroline Tice       result.SetStatus(eReturnStatusFailed);
159135731357SCaroline Tice 
159235731357SCaroline Tice     return result.Succeeded();
159335731357SCaroline Tice   }
159435731357SCaroline Tice 
159535731357SCaroline Tice   CommandOptions m_options;
159635731357SCaroline Tice };
159735731357SCaroline Tice 
159835731357SCaroline Tice //-------------------------------------------------------------------------
159930fdc8d8SChris Lattner // CommandObjectMultiwordProcess
160030fdc8d8SChris Lattner //-------------------------------------------------------------------------
160130fdc8d8SChris Lattner 
1602b9c1b51eSKate Stone CommandObjectMultiwordProcess::CommandObjectMultiwordProcess(
1603b9c1b51eSKate Stone     CommandInterpreter &interpreter)
1604b9c1b51eSKate Stone     : CommandObjectMultiword(
1605b9c1b51eSKate Stone           interpreter, "process",
1606b9c1b51eSKate Stone           "Commands for interacting with processes on the current platform.",
1607b9c1b51eSKate Stone           "process <subcommand> [<subcommand-options>]") {
1608b9c1b51eSKate Stone   LoadSubCommand("attach",
1609b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessAttach(interpreter)));
1610b9c1b51eSKate Stone   LoadSubCommand("launch",
1611b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessLaunch(interpreter)));
1612b9c1b51eSKate Stone   LoadSubCommand("continue", CommandObjectSP(new CommandObjectProcessContinue(
1613b9c1b51eSKate Stone                                  interpreter)));
1614b9c1b51eSKate Stone   LoadSubCommand("connect",
1615b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessConnect(interpreter)));
1616b9c1b51eSKate Stone   LoadSubCommand("detach",
1617b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessDetach(interpreter)));
1618b9c1b51eSKate Stone   LoadSubCommand("load",
1619b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessLoad(interpreter)));
1620b9c1b51eSKate Stone   LoadSubCommand("unload",
1621b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessUnload(interpreter)));
1622b9c1b51eSKate Stone   LoadSubCommand("signal",
1623b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessSignal(interpreter)));
1624b9c1b51eSKate Stone   LoadSubCommand("handle",
1625b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessHandle(interpreter)));
1626b9c1b51eSKate Stone   LoadSubCommand("status",
1627b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessStatus(interpreter)));
1628b9c1b51eSKate Stone   LoadSubCommand("interrupt", CommandObjectSP(new CommandObjectProcessInterrupt(
1629b9c1b51eSKate Stone                                   interpreter)));
1630b9c1b51eSKate Stone   LoadSubCommand("kill",
1631b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessKill(interpreter)));
1632b9c1b51eSKate Stone   LoadSubCommand("plugin",
1633b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessPlugin(interpreter)));
1634b9c1b51eSKate Stone   LoadSubCommand("save-core", CommandObjectSP(new CommandObjectProcessSaveCore(
1635b9c1b51eSKate Stone                                   interpreter)));
163630fdc8d8SChris Lattner }
163730fdc8d8SChris Lattner 
163849bcfd80SEugene Zelenko CommandObjectMultiwordProcess::~CommandObjectMultiwordProcess() = default;
1639