180814287SRaphael Isemann //===-- CommandObjectProcess.cpp ------------------------------------------===//
230fdc8d8SChris Lattner //
32946cd70SChandler Carruth // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
42946cd70SChandler Carruth // See https://llvm.org/LICENSE.txt for license information.
52946cd70SChandler 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"
153eb2b44dSZachary Turner #include "lldb/Host/OptionParser.h"
165275aaa0SVince Harron #include "lldb/Host/StringConvert.h"
1730fdc8d8SChris Lattner #include "lldb/Interpreter/CommandInterpreter.h"
1830fdc8d8SChris Lattner #include "lldb/Interpreter/CommandReturnObject.h"
1947cbf4a0SPavel Labath #include "lldb/Interpreter/OptionArgParser.h"
20b9c1b51eSKate Stone #include "lldb/Interpreter/Options.h"
21e996fd30SGreg Clayton #include "lldb/Target/Platform.h"
2230fdc8d8SChris Lattner #include "lldb/Target/Process.h"
230e41084aSJim Ingham #include "lldb/Target/StopInfo.h"
2430fdc8d8SChris Lattner #include "lldb/Target/Target.h"
2530fdc8d8SChris Lattner #include "lldb/Target/Thread.h"
2693749ab3SZachary Turner #include "lldb/Target/UnixSignals.h"
27145d95c9SPavel Labath #include "lldb/Utility/Args.h"
28d821c997SPavel Labath #include "lldb/Utility/State.h"
2930fdc8d8SChris Lattner 
3030fdc8d8SChris Lattner using namespace lldb;
3130fdc8d8SChris Lattner using namespace lldb_private;
3230fdc8d8SChris Lattner 
33b9c1b51eSKate Stone class CommandObjectProcessLaunchOrAttach : public CommandObjectParsed {
34dcb1d856SJim Ingham public:
35dcb1d856SJim Ingham   CommandObjectProcessLaunchOrAttach(CommandInterpreter &interpreter,
36b9c1b51eSKate Stone                                      const char *name, const char *help,
37b9c1b51eSKate Stone                                      const char *syntax, uint32_t flags,
38b9c1b51eSKate Stone                                      const char *new_process_action)
39b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, name, help, syntax, flags),
40dcb1d856SJim Ingham         m_new_process_action(new_process_action) {}
41dcb1d856SJim Ingham 
4249bcfd80SEugene Zelenko   ~CommandObjectProcessLaunchOrAttach() override = default;
4349bcfd80SEugene Zelenko 
44dcb1d856SJim Ingham protected:
45b9c1b51eSKate Stone   bool StopProcessIfNecessary(Process *process, StateType &state,
46b9c1b51eSKate Stone                               CommandReturnObject &result) {
47dcb1d856SJim Ingham     state = eStateInvalid;
48b9c1b51eSKate Stone     if (process) {
49dcb1d856SJim Ingham       state = process->GetState();
50dcb1d856SJim Ingham 
51b9c1b51eSKate Stone       if (process->IsAlive() && state != eStateConnected) {
52dcb1d856SJim Ingham         char message[1024];
53dcb1d856SJim Ingham         if (process->GetState() == eStateAttaching)
54b9c1b51eSKate Stone           ::snprintf(message, sizeof(message),
55b9c1b51eSKate Stone                      "There is a pending attach, abort it and %s?",
56b9c1b51eSKate Stone                      m_new_process_action.c_str());
57dcb1d856SJim Ingham         else if (process->GetShouldDetach())
58b9c1b51eSKate Stone           ::snprintf(message, sizeof(message),
59b9c1b51eSKate Stone                      "There is a running process, detach from it and %s?",
60b9c1b51eSKate Stone                      m_new_process_action.c_str());
61dcb1d856SJim Ingham         else
62b9c1b51eSKate Stone           ::snprintf(message, sizeof(message),
63b9c1b51eSKate Stone                      "There is a running process, kill it and %s?",
64b9c1b51eSKate Stone                      m_new_process_action.c_str());
65dcb1d856SJim Ingham 
66b9c1b51eSKate Stone         if (!m_interpreter.Confirm(message, true)) {
67dcb1d856SJim Ingham           result.SetStatus(eReturnStatusFailed);
68dcb1d856SJim Ingham           return false;
69b9c1b51eSKate Stone         } else {
70b9c1b51eSKate Stone           if (process->GetShouldDetach()) {
71acff8950SJim Ingham             bool keep_stopped = false;
7297206d57SZachary Turner             Status detach_error(process->Detach(keep_stopped));
73b9c1b51eSKate Stone             if (detach_error.Success()) {
74dcb1d856SJim Ingham               result.SetStatus(eReturnStatusSuccessFinishResult);
7549bcfd80SEugene Zelenko               process = nullptr;
76b9c1b51eSKate Stone             } else {
77b9c1b51eSKate Stone               result.AppendErrorWithFormat(
78b9c1b51eSKate Stone                   "Failed to detach from process: %s\n",
79b9c1b51eSKate Stone                   detach_error.AsCString());
80dcb1d856SJim Ingham               result.SetStatus(eReturnStatusFailed);
81dcb1d856SJim Ingham             }
82b9c1b51eSKate Stone           } else {
8397206d57SZachary Turner             Status destroy_error(process->Destroy(false));
84b9c1b51eSKate Stone             if (destroy_error.Success()) {
85dcb1d856SJim Ingham               result.SetStatus(eReturnStatusSuccessFinishResult);
8649bcfd80SEugene Zelenko               process = nullptr;
87b9c1b51eSKate Stone             } else {
88b9c1b51eSKate Stone               result.AppendErrorWithFormat("Failed to kill process: %s\n",
89b9c1b51eSKate Stone                                            destroy_error.AsCString());
90dcb1d856SJim Ingham               result.SetStatus(eReturnStatusFailed);
91dcb1d856SJim Ingham             }
92dcb1d856SJim Ingham           }
93dcb1d856SJim Ingham         }
94dcb1d856SJim Ingham       }
95dcb1d856SJim Ingham     }
96dcb1d856SJim Ingham     return result.Succeeded();
97dcb1d856SJim Ingham   }
9849bcfd80SEugene Zelenko 
99dcb1d856SJim Ingham   std::string m_new_process_action;
100dcb1d856SJim Ingham };
10149bcfd80SEugene Zelenko 
10230fdc8d8SChris Lattner // CommandObjectProcessLaunch
1034bddaeb5SJim Ingham #pragma mark CommandObjectProcessLaunch
104b9c1b51eSKate Stone class CommandObjectProcessLaunch : public CommandObjectProcessLaunchOrAttach {
10530fdc8d8SChris Lattner public:
106b9c1b51eSKate Stone   CommandObjectProcessLaunch(CommandInterpreter &interpreter)
107b9c1b51eSKate Stone       : CommandObjectProcessLaunchOrAttach(
108b9c1b51eSKate Stone             interpreter, "process launch",
109b9c1b51eSKate Stone             "Launch the executable in the debugger.", nullptr,
110b9c1b51eSKate Stone             eCommandRequiresTarget, "restart"),
111b9c1b51eSKate Stone         m_options() {
112405fe67fSCaroline Tice     CommandArgumentEntry arg;
113405fe67fSCaroline Tice     CommandArgumentData run_args_arg;
114405fe67fSCaroline Tice 
115405fe67fSCaroline Tice     // Define the first (and only) variant of this arg.
116405fe67fSCaroline Tice     run_args_arg.arg_type = eArgTypeRunArgs;
117405fe67fSCaroline Tice     run_args_arg.arg_repetition = eArgRepeatOptional;
118405fe67fSCaroline Tice 
119b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
120b9c1b51eSKate Stone     // argument entry.
121405fe67fSCaroline Tice     arg.push_back(run_args_arg);
122405fe67fSCaroline Tice 
123405fe67fSCaroline Tice     // Push the data for the first argument into the m_arguments vector.
124405fe67fSCaroline Tice     m_arguments.push_back(arg);
12530fdc8d8SChris Lattner   }
12630fdc8d8SChris Lattner 
12749bcfd80SEugene Zelenko   ~CommandObjectProcessLaunch() override = default;
12830fdc8d8SChris Lattner 
129ae34ed2cSRaphael Isemann   void
130ae34ed2cSRaphael Isemann   HandleArgumentCompletion(CompletionRequest &request,
1312443bbd4SRaphael Isemann                            OptionElementVector &opt_element_vector) override {
132e9ce62b6SJim Ingham 
133b9c1b51eSKate Stone     CommandCompletions::InvokeCommonCompletionCallbacks(
134b9c1b51eSKate Stone         GetCommandInterpreter(), CommandCompletions::eDiskFileCompletion,
135a2e76c0bSRaphael Isemann         request, nullptr);
136e9ce62b6SJim Ingham   }
137e9ce62b6SJim Ingham 
138b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
13930fdc8d8SChris Lattner 
140b9c1b51eSKate Stone   const char *GetRepeatCommand(Args &current_command_args,
141b9c1b51eSKate Stone                                uint32_t index) override {
1425a988416SJim Ingham     // No repeat for "process launch"...
1435a988416SJim Ingham     return "";
1445a988416SJim Ingham   }
1455a988416SJim Ingham 
1465a988416SJim Ingham protected:
147b9c1b51eSKate Stone   bool DoExecute(Args &launch_args, CommandReturnObject &result) override {
14857179860SJonas Devlieghere     Debugger &debugger = GetDebugger();
1491d885966SGreg Clayton     Target *target = debugger.GetSelectedTarget().get();
15049bcfd80SEugene Zelenko     // If our listener is nullptr, users aren't allows to launch
151b09c5384SGreg Clayton     ModuleSP exe_module_sp = target->GetExecutableModule();
15271337622SGreg Clayton 
153b9c1b51eSKate Stone     if (exe_module_sp == nullptr) {
154b9c1b51eSKate Stone       result.AppendError("no file in target, create a debug target using the "
155b9c1b51eSKate Stone                          "'target create' command");
15671337622SGreg Clayton       result.SetStatus(eReturnStatusFailed);
15771337622SGreg Clayton       return false;
15871337622SGreg Clayton     }
15971337622SGreg Clayton 
16071337622SGreg Clayton     StateType state = eStateInvalid;
16171337622SGreg Clayton 
162b09c5384SGreg Clayton     if (!StopProcessIfNecessary(m_exe_ctx.GetProcessPtr(), state, result))
16330fdc8d8SChris Lattner       return false;
16430fdc8d8SChris Lattner 
16531d97a5cSZachary Turner     llvm::StringRef target_settings_argv0 = target->GetArg0();
16645392553SGreg Clayton 
167b9c1b51eSKate Stone     // Determine whether we will disable ASLR or leave it in the default state
16805097246SAdrian Prantl     // (i.e. enabled if the platform supports it). First check if the process
16905097246SAdrian Prantl     // launch options explicitly turn on/off
170b9c1b51eSKate Stone     // disabling ASLR.  If so, use that setting;
1715163792bSTodd Fiala     // otherwise, use the 'settings target.disable-aslr' setting.
1725163792bSTodd Fiala     bool disable_aslr = false;
173b9c1b51eSKate Stone     if (m_options.disable_aslr != eLazyBoolCalculate) {
17405097246SAdrian Prantl       // The user specified an explicit setting on the process launch line.
17505097246SAdrian Prantl       // Use it.
1765163792bSTodd Fiala       disable_aslr = (m_options.disable_aslr == eLazyBoolYes);
177b9c1b51eSKate Stone     } else {
17805097246SAdrian Prantl       // The user did not explicitly specify whether to disable ASLR.  Fall
17905097246SAdrian Prantl       // back to the target.disable-aslr setting.
1805163792bSTodd Fiala       disable_aslr = target->GetDisableASLR();
1815163792bSTodd Fiala     }
1825163792bSTodd Fiala 
1835163792bSTodd Fiala     if (disable_aslr)
184b09c5384SGreg Clayton       m_options.launch_info.GetFlags().Set(eLaunchFlagDisableASLR);
1855163792bSTodd Fiala     else
1865163792bSTodd Fiala       m_options.launch_info.GetFlags().Clear(eLaunchFlagDisableASLR);
187b09c5384SGreg Clayton 
188106d0286SJim Ingham     if (target->GetDetachOnError())
189106d0286SJim Ingham       m_options.launch_info.GetFlags().Set(eLaunchFlagDetachOnError);
190106d0286SJim Ingham 
191b09c5384SGreg Clayton     if (target->GetDisableSTDIO())
192b09c5384SGreg Clayton       m_options.launch_info.GetFlags().Set(eLaunchFlagDisableSTDIO);
193b09c5384SGreg Clayton 
194ae54fc9fSJonas Devlieghere     // Merge the launch info environment with the target environment.
195ae54fc9fSJonas Devlieghere     Environment target_env = target->GetEnvironment();
196ae54fc9fSJonas Devlieghere     m_options.launch_info.GetEnvironment().insert(target_env.begin(),
197ae54fc9fSJonas Devlieghere                                                   target_env.end());
19845392553SGreg Clayton 
19931d97a5cSZachary Turner     if (!target_settings_argv0.empty()) {
200b9c1b51eSKate Stone       m_options.launch_info.GetArguments().AppendArgument(
20131d97a5cSZachary Turner           target_settings_argv0);
202b9c1b51eSKate Stone       m_options.launch_info.SetExecutableFile(
203b9c1b51eSKate Stone           exe_module_sp->GetPlatformFileSpec(), false);
204b9c1b51eSKate Stone     } else {
205b9c1b51eSKate Stone       m_options.launch_info.SetExecutableFile(
206b9c1b51eSKate Stone           exe_module_sp->GetPlatformFileSpec(), true);
20745392553SGreg Clayton     }
20845392553SGreg Clayton 
209b9c1b51eSKate Stone     if (launch_args.GetArgumentCount() == 0) {
210b9c1b51eSKate Stone       m_options.launch_info.GetArguments().AppendArguments(
211b9c1b51eSKate Stone           target->GetProcessLaunchInfo().GetArguments());
212b9c1b51eSKate Stone     } else {
21345392553SGreg Clayton       m_options.launch_info.GetArguments().AppendArguments(launch_args);
214162b597cSGreg Clayton       // Save the arguments for subsequent runs in the current target.
215162b597cSGreg Clayton       target->SetRunArguments(launch_args);
216982c9762SGreg Clayton     }
2171d885966SGreg Clayton 
218dc6224e0SGreg Clayton     StreamString stream;
21997206d57SZachary Turner     Status error = target->Launch(m_options.launch_info, &stream);
22030fdc8d8SChris Lattner 
221b9c1b51eSKate Stone     if (error.Success()) {
222b09c5384SGreg Clayton       ProcessSP process_sp(target->GetProcessSP());
223b9c1b51eSKate Stone       if (process_sp) {
224b9c1b51eSKate Stone         // There is a race condition where this thread will return up the call
22505097246SAdrian Prantl         // stack to the main command handler and show an (lldb) prompt before
22605097246SAdrian Prantl         // HandlePrivateEvent (from PrivateStateThread) has a chance to call
22705097246SAdrian Prantl         // PushProcessIOHandler().
2283879fe00SPavel Labath         process_sp->SyncIOHandler(0, std::chrono::seconds(2));
2298f0db3e1SIlia K 
230c156427dSZachary Turner         llvm::StringRef data = stream.GetString();
231c156427dSZachary Turner         if (!data.empty())
232c156427dSZachary Turner           result.AppendMessage(data);
233b9c1b51eSKate Stone         const char *archname =
234b9c1b51eSKate Stone             exe_module_sp->GetArchitecture().GetArchitectureName();
235b9c1b51eSKate Stone         result.AppendMessageWithFormat(
236b9c1b51eSKate Stone             "Process %" PRIu64 " launched: '%s' (%s)\n", process_sp->GetID(),
237b9c1b51eSKate Stone             exe_module_sp->GetFileSpec().GetPath().c_str(), archname);
23805faeb71SGreg Clayton         result.SetStatus(eReturnStatusSuccessFinishResult);
239b09c5384SGreg Clayton         result.SetDidChangeProcessState(true);
240b9c1b51eSKate Stone       } else {
241b9c1b51eSKate Stone         result.AppendError(
242b9c1b51eSKate Stone             "no error returned from Target::Launch, and target has no process");
243514487e8SGreg Clayton         result.SetStatus(eReturnStatusFailed);
24430fdc8d8SChris Lattner       }
245b9c1b51eSKate Stone     } else {
246b09c5384SGreg Clayton       result.AppendError(error.AsCString());
247514487e8SGreg Clayton       result.SetStatus(eReturnStatusFailed);
248514487e8SGreg Clayton     }
24930fdc8d8SChris Lattner     return result.Succeeded();
25030fdc8d8SChris Lattner   }
25130fdc8d8SChris Lattner 
25230fdc8d8SChris Lattner protected:
253982c9762SGreg Clayton   ProcessLaunchCommandOptions m_options;
25430fdc8d8SChris Lattner };
25530fdc8d8SChris Lattner 
256438dfcffSRaphael Isemann #define LLDB_OPTIONS_process_attach
257438dfcffSRaphael Isemann #include "CommandOptions.inc"
2581f0f5b5bSZachary Turner 
259bb9caf73SJim Ingham #pragma mark CommandObjectProcessAttach
260b9c1b51eSKate Stone class CommandObjectProcessAttach : public CommandObjectProcessLaunchOrAttach {
26130fdc8d8SChris Lattner public:
262b9c1b51eSKate Stone   class CommandOptions : public Options {
26330fdc8d8SChris Lattner   public:
264b9c1b51eSKate Stone     CommandOptions() : Options() {
265b9c1b51eSKate Stone       // Keep default values of all options in one place: OptionParsingStarting
266b9c1b51eSKate Stone       // ()
267e1cfbc79STodd Fiala       OptionParsingStarting(nullptr);
26830fdc8d8SChris Lattner     }
26930fdc8d8SChris Lattner 
27049bcfd80SEugene Zelenko     ~CommandOptions() override = default;
27130fdc8d8SChris Lattner 
27297206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
273b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
27497206d57SZachary Turner       Status error;
2753bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
276b9c1b51eSKate Stone       switch (short_option) {
277a95ce623SJohnny Chen       case 'c':
278a95ce623SJohnny Chen         attach_info.SetContinueOnceAttached(true);
279a95ce623SJohnny Chen         break;
280a95ce623SJohnny Chen 
281b9c1b51eSKate Stone       case 'p': {
282fe11483bSZachary Turner         lldb::pid_t pid;
283fe11483bSZachary Turner         if (option_arg.getAsInteger(0, pid)) {
284fe11483bSZachary Turner           error.SetErrorStringWithFormat("invalid process ID '%s'",
285fe11483bSZachary Turner                                          option_arg.str().c_str());
286b9c1b51eSKate Stone         } else {
287144f3a9cSGreg Clayton           attach_info.SetProcessID(pid);
288144f3a9cSGreg Clayton         }
289b9c1b51eSKate Stone       } break;
29030fdc8d8SChris Lattner 
29130fdc8d8SChris Lattner       case 'P':
292144f3a9cSGreg Clayton         attach_info.SetProcessPluginName(option_arg);
29330fdc8d8SChris Lattner         break;
29430fdc8d8SChris Lattner 
29530fdc8d8SChris Lattner       case 'n':
2968f3be7a3SJonas Devlieghere         attach_info.GetExecutableFile().SetFile(option_arg,
297937348cdSJonas Devlieghere                                                 FileSpec::Style::native);
29830fdc8d8SChris Lattner         break;
29930fdc8d8SChris Lattner 
30030fdc8d8SChris Lattner       case 'w':
301144f3a9cSGreg Clayton         attach_info.SetWaitForLaunch(true);
30230fdc8d8SChris Lattner         break;
30330fdc8d8SChris Lattner 
304cd16df91SJim Ingham       case 'i':
305cd16df91SJim Ingham         attach_info.SetIgnoreExisting(false);
306cd16df91SJim Ingham         break;
307cd16df91SJim Ingham 
30830fdc8d8SChris Lattner       default:
30936162014SRaphael Isemann         llvm_unreachable("Unimplemented option");
31030fdc8d8SChris Lattner       }
31130fdc8d8SChris Lattner       return error;
31230fdc8d8SChris Lattner     }
31330fdc8d8SChris Lattner 
314b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
315144f3a9cSGreg Clayton       attach_info.Clear();
31630fdc8d8SChris Lattner     }
31730fdc8d8SChris Lattner 
3181f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
31970602439SZachary Turner       return llvm::makeArrayRef(g_process_attach_options);
3201f0f5b5bSZachary Turner     }
32130fdc8d8SChris Lattner 
322494370c1SRaphael Isemann     void HandleOptionArgumentCompletion(
323a2e76c0bSRaphael Isemann         CompletionRequest &request, OptionElementVector &opt_element_vector,
324a2e76c0bSRaphael Isemann         int opt_element_index, CommandInterpreter &interpreter) override {
3255aee162fSJim Ingham       int opt_arg_pos = opt_element_vector[opt_element_index].opt_arg_pos;
3265aee162fSJim Ingham       int opt_defs_index = opt_element_vector[opt_element_index].opt_defs_index;
3275aee162fSJim Ingham 
3285aee162fSJim Ingham       // We are only completing the name option for now...
3295aee162fSJim Ingham 
3305aee162fSJim Ingham       // Are we in the name?
3311153dc96SRaphael Isemann       if (GetDefinitions()[opt_defs_index].short_option != 'n')
332494370c1SRaphael Isemann         return;
3335aee162fSJim Ingham 
334b9c1b51eSKate Stone       // Look to see if there is a -P argument provided, and if so use that
33505097246SAdrian Prantl       // plugin, otherwise use the default plugin.
3365aee162fSJim Ingham 
33749bcfd80SEugene Zelenko       const char *partial_name = nullptr;
338a2e76c0bSRaphael Isemann       partial_name = request.GetParsedLine().GetArgumentAtIndex(opt_arg_pos);
3395aee162fSJim Ingham 
340e1cfbc79STodd Fiala       PlatformSP platform_sp(interpreter.GetPlatform(true));
3411153dc96SRaphael Isemann       if (!platform_sp)
342494370c1SRaphael Isemann         return;
3438b82f087SGreg Clayton       ProcessInstanceInfoList process_infos;
3448b82f087SGreg Clayton       ProcessInstanceInfoMatch match_info;
345b9c1b51eSKate Stone       if (partial_name) {
346b9c1b51eSKate Stone         match_info.GetProcessInfo().GetExecutableFile().SetFile(
3478f3be7a3SJonas Devlieghere             partial_name, FileSpec::Style::native);
348c4a33951SPavel Labath         match_info.SetNameMatchType(NameMatch::StartsWith);
34932e0a750SGreg Clayton       }
35032e0a750SGreg Clayton       platform_sp->FindProcesses(match_info, process_infos);
351c7bece56SGreg Clayton       const size_t num_matches = process_infos.GetSize();
3521153dc96SRaphael Isemann       if (num_matches == 0)
353494370c1SRaphael Isemann         return;
354b9c1b51eSKate Stone       for (size_t i = 0; i < num_matches; ++i) {
35572ca5f36SRaphael Isemann         request.AddCompletion(process_infos.GetProcessNameAtIndex(i));
3565aee162fSJim Ingham       }
3575aee162fSJim Ingham     }
3585aee162fSJim Ingham 
35930fdc8d8SChris Lattner     // Instance variables to hold the values for command options.
36030fdc8d8SChris Lattner 
361144f3a9cSGreg Clayton     ProcessAttachInfo attach_info;
36230fdc8d8SChris Lattner   };
36330fdc8d8SChris Lattner 
364b9c1b51eSKate Stone   CommandObjectProcessAttach(CommandInterpreter &interpreter)
365b9c1b51eSKate Stone       : CommandObjectProcessLaunchOrAttach(
366b9c1b51eSKate Stone             interpreter, "process attach", "Attach to a process.",
367b9c1b51eSKate Stone             "process attach <cmd-options>", 0, "attach"),
368b9c1b51eSKate Stone         m_options() {}
3695aee162fSJim Ingham 
37049bcfd80SEugene Zelenko   ~CommandObjectProcessAttach() override = default;
3715aee162fSJim Ingham 
372b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
3735a988416SJim Ingham 
3745a988416SJim Ingham protected:
375b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
376b9c1b51eSKate Stone     PlatformSP platform_sp(
37757179860SJonas Devlieghere         GetDebugger().GetPlatformList().GetSelectedPlatform());
378926af0cdSOleksiy Vyalov 
37957179860SJonas Devlieghere     Target *target = GetDebugger().GetSelectedTarget().get();
380b9c1b51eSKate Stone     // N.B. The attach should be synchronous.  It doesn't help much to get the
38105097246SAdrian Prantl     // prompt back between initiating the attach and the target actually
38205097246SAdrian Prantl     // stopping.  So even if the interpreter is set to be asynchronous, we wait
38305097246SAdrian Prantl     // for the stop ourselves here.
3845aee162fSJim Ingham 
38571337622SGreg Clayton     StateType state = eStateInvalid;
386dcb1d856SJim Ingham     Process *process = m_exe_ctx.GetProcessPtr();
387dcb1d856SJim Ingham 
388dcb1d856SJim Ingham     if (!StopProcessIfNecessary(process, state, result))
3895aee162fSJim Ingham       return false;
3905aee162fSJim Ingham 
391b9c1b51eSKate Stone     if (target == nullptr) {
3925aee162fSJim Ingham       // If there isn't a current target create one.
3935aee162fSJim Ingham       TargetSP new_target_sp;
39497206d57SZachary Turner       Status error;
3955aee162fSJim Ingham 
39657179860SJonas Devlieghere       error = GetDebugger().GetTargetList().CreateTarget(
39757179860SJonas Devlieghere           GetDebugger(), "", "", eLoadDependentsNo,
39849bcfd80SEugene Zelenko           nullptr, // No platform options
3995aee162fSJim Ingham           new_target_sp);
4005aee162fSJim Ingham       target = new_target_sp.get();
401b9c1b51eSKate Stone       if (target == nullptr || error.Fail()) {
402b766a73dSGreg Clayton         result.AppendError(error.AsCString("Error creating target"));
4035aee162fSJim Ingham         return false;
4045aee162fSJim Ingham       }
40557179860SJonas Devlieghere       GetDebugger().GetTargetList().SetSelectedTarget(target);
4065aee162fSJim Ingham     }
4075aee162fSJim Ingham 
408b9c1b51eSKate Stone     // Record the old executable module, we want to issue a warning if the
40905097246SAdrian Prantl     // process of attaching changed the current executable (like somebody said
41005097246SAdrian Prantl     // "file foo" then attached to a PID whose executable was bar.)
4115aee162fSJim Ingham 
4125aee162fSJim Ingham     ModuleSP old_exec_module_sp = target->GetExecutableModule();
4135aee162fSJim Ingham     ArchSpec old_arch_spec = target->GetArchitecture();
4145aee162fSJim Ingham 
415b9c1b51eSKate Stone     if (command.GetArgumentCount()) {
416b9c1b51eSKate Stone       result.AppendErrorWithFormat("Invalid arguments for '%s'.\nUsage: %s\n",
417b9c1b51eSKate Stone                                    m_cmd_name.c_str(), m_cmd_syntax.c_str());
4185aee162fSJim Ingham       result.SetStatus(eReturnStatusFailed);
419926af0cdSOleksiy Vyalov       return false;
42071337622SGreg Clayton     }
4215aee162fSJim Ingham 
422926af0cdSOleksiy Vyalov     m_interpreter.UpdateExecutionContext(nullptr);
42337386143SOleksiy Vyalov     StreamString stream;
42437386143SOleksiy Vyalov     const auto error = target->Attach(m_options.attach_info, &stream);
425b9c1b51eSKate Stone     if (error.Success()) {
42637386143SOleksiy Vyalov       ProcessSP process_sp(target->GetProcessSP());
427b9c1b51eSKate Stone       if (process_sp) {
428c156427dSZachary Turner         result.AppendMessage(stream.GetString());
429bb3a283bSJim Ingham         result.SetStatus(eReturnStatusSuccessFinishNoResult);
43037386143SOleksiy Vyalov         result.SetDidChangeProcessState(true);
431b9c1b51eSKate Stone       } else {
432b9c1b51eSKate Stone         result.AppendError(
433b9c1b51eSKate Stone             "no error returned from Target::Attach, and target has no process");
434aa739093SJohnny Chen         result.SetStatus(eReturnStatusFailed);
43544d93782SGreg Clayton       }
436b9c1b51eSKate Stone     } else {
43744d93782SGreg Clayton       result.AppendErrorWithFormat("attach failed: %s\n", error.AsCString());
43844d93782SGreg Clayton       result.SetStatus(eReturnStatusFailed);
439aa739093SJohnny Chen     }
4405aee162fSJim Ingham 
441926af0cdSOleksiy Vyalov     if (!result.Succeeded())
442926af0cdSOleksiy Vyalov       return false;
443926af0cdSOleksiy Vyalov 
444b9c1b51eSKate Stone     // Okay, we're done.  Last step is to warn if the executable module has
445b9c1b51eSKate Stone     // changed:
446513c26ceSGreg Clayton     char new_path[PATH_MAX];
447aa149cbdSGreg Clayton     ModuleSP new_exec_module_sp(target->GetExecutableModule());
448b9c1b51eSKate Stone     if (!old_exec_module_sp) {
449513c26ceSGreg Clayton       // We might not have a module if we attached to a raw pid...
450b9c1b51eSKate Stone       if (new_exec_module_sp) {
451aa149cbdSGreg Clayton         new_exec_module_sp->GetFileSpec().GetPath(new_path, PATH_MAX);
452b9c1b51eSKate Stone         result.AppendMessageWithFormat("Executable module set to \"%s\".\n",
453b9c1b51eSKate Stone                                        new_path);
454513c26ceSGreg Clayton       }
455b9c1b51eSKate Stone     } else if (old_exec_module_sp->GetFileSpec() !=
456b9c1b51eSKate Stone                new_exec_module_sp->GetFileSpec()) {
457513c26ceSGreg Clayton       char old_path[PATH_MAX];
4585aee162fSJim Ingham 
4595aee162fSJim Ingham       old_exec_module_sp->GetFileSpec().GetPath(old_path, PATH_MAX);
460aa149cbdSGreg Clayton       new_exec_module_sp->GetFileSpec().GetPath(new_path, PATH_MAX);
4615aee162fSJim Ingham 
462b9c1b51eSKate Stone       result.AppendWarningWithFormat(
463b9c1b51eSKate Stone           "Executable module changed from \"%s\" to \"%s\".\n", old_path,
464b9c1b51eSKate Stone           new_path);
4655aee162fSJim Ingham     }
4665aee162fSJim Ingham 
467b9c1b51eSKate Stone     if (!old_arch_spec.IsValid()) {
468b9c1b51eSKate Stone       result.AppendMessageWithFormat(
469b9c1b51eSKate Stone           "Architecture set to: %s.\n",
470b9c1b51eSKate Stone           target->GetArchitecture().GetTriple().getTriple().c_str());
471b9c1b51eSKate Stone     } else if (!old_arch_spec.IsExactMatch(target->GetArchitecture())) {
472b9c1b51eSKate Stone       result.AppendWarningWithFormat(
473b9c1b51eSKate Stone           "Architecture changed from %s to %s.\n",
474c1b1f1eaSGreg Clayton           old_arch_spec.GetTriple().getTriple().c_str(),
475c1b1f1eaSGreg Clayton           target->GetArchitecture().GetTriple().getTriple().c_str());
4765aee162fSJim Ingham     }
477a95ce623SJohnny Chen 
47805097246SAdrian Prantl     // This supports the use-case scenario of immediately continuing the
47905097246SAdrian Prantl     // process once attached.
480a95ce623SJohnny Chen     if (m_options.attach_info.GetContinueOnceAttached())
4815bcaf583SSean Callanan       m_interpreter.HandleCommand("process continue", eLazyBoolNo, result);
482926af0cdSOleksiy Vyalov 
4835aee162fSJim Ingham     return result.Succeeded();
4845aee162fSJim Ingham   }
4855aee162fSJim Ingham 
48630fdc8d8SChris Lattner   CommandOptions m_options;
48730fdc8d8SChris Lattner };
48830fdc8d8SChris Lattner 
48930fdc8d8SChris Lattner // CommandObjectProcessContinue
4901f0f5b5bSZachary Turner 
491438dfcffSRaphael Isemann #define LLDB_OPTIONS_process_continue
492438dfcffSRaphael Isemann #include "CommandOptions.inc"
4931f0f5b5bSZachary Turner 
494bb9caf73SJim Ingham #pragma mark CommandObjectProcessContinue
49530fdc8d8SChris Lattner 
496b9c1b51eSKate Stone class CommandObjectProcessContinue : public CommandObjectParsed {
49730fdc8d8SChris Lattner public:
498b9c1b51eSKate Stone   CommandObjectProcessContinue(CommandInterpreter &interpreter)
499b9c1b51eSKate Stone       : CommandObjectParsed(
500b9c1b51eSKate Stone             interpreter, "process continue",
501e3d26315SCaroline Tice             "Continue execution of all threads in the current process.",
50230fdc8d8SChris Lattner             "process continue",
503b9c1b51eSKate Stone             eCommandRequiresProcess | eCommandTryTargetAPILock |
504b9c1b51eSKate Stone                 eCommandProcessMustBeLaunched | eCommandProcessMustBePaused),
505b9c1b51eSKate Stone         m_options() {}
50630fdc8d8SChris Lattner 
50749bcfd80SEugene Zelenko   ~CommandObjectProcessContinue() override = default;
50830fdc8d8SChris Lattner 
5095a988416SJim Ingham protected:
510b9c1b51eSKate Stone   class CommandOptions : public Options {
5110e41084aSJim Ingham   public:
512b9c1b51eSKate Stone     CommandOptions() : Options() {
513b9c1b51eSKate Stone       // Keep default values of all options in one place: OptionParsingStarting
514b9c1b51eSKate Stone       // ()
515e1cfbc79STodd Fiala       OptionParsingStarting(nullptr);
5160e41084aSJim Ingham     }
5170e41084aSJim Ingham 
51849bcfd80SEugene Zelenko     ~CommandOptions() override = default;
5190e41084aSJim Ingham 
52097206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
521b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
52297206d57SZachary Turner       Status error;
5233bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
524b9c1b51eSKate Stone       switch (short_option) {
5250e41084aSJim Ingham       case 'i':
526fe11483bSZachary Turner         if (option_arg.getAsInteger(0, m_ignore))
527b9c1b51eSKate Stone           error.SetErrorStringWithFormat(
528b9c1b51eSKate Stone               "invalid value for ignore option: \"%s\", should be a number.",
529fe11483bSZachary Turner               option_arg.str().c_str());
5300e41084aSJim Ingham         break;
5310e41084aSJim Ingham 
5320e41084aSJim Ingham       default:
53336162014SRaphael Isemann         llvm_unreachable("Unimplemented option");
5340e41084aSJim Ingham       }
5350e41084aSJim Ingham       return error;
5360e41084aSJim Ingham     }
5370e41084aSJim Ingham 
538b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
5390e41084aSJim Ingham       m_ignore = 0;
5400e41084aSJim Ingham     }
5410e41084aSJim Ingham 
5421f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
54370602439SZachary Turner       return llvm::makeArrayRef(g_process_continue_options);
5441f0f5b5bSZachary Turner     }
5450e41084aSJim Ingham 
5460e41084aSJim Ingham     uint32_t m_ignore;
5470e41084aSJim Ingham   };
5480e41084aSJim Ingham 
549b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
550f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
551a7015092SGreg Clayton     bool synchronous_execution = m_interpreter.GetSynchronous();
55230fdc8d8SChris Lattner     StateType state = process->GetState();
553b9c1b51eSKate Stone     if (state == eStateStopped) {
554b9c1b51eSKate Stone       if (command.GetArgumentCount() != 0) {
555b9c1b51eSKate Stone         result.AppendErrorWithFormat(
556b9c1b51eSKate Stone             "The '%s' command does not take any arguments.\n",
557b9c1b51eSKate Stone             m_cmd_name.c_str());
55830fdc8d8SChris Lattner         result.SetStatus(eReturnStatusFailed);
55930fdc8d8SChris Lattner         return false;
56030fdc8d8SChris Lattner       }
56130fdc8d8SChris Lattner 
562b9c1b51eSKate Stone       if (m_options.m_ignore > 0) {
5638d94ba0fSJim Ingham         ThreadSP sel_thread_sp(GetDefaultThread()->shared_from_this());
564b9c1b51eSKate Stone         if (sel_thread_sp) {
5650e41084aSJim Ingham           StopInfoSP stop_info_sp = sel_thread_sp->GetStopInfo();
566b9c1b51eSKate Stone           if (stop_info_sp &&
567b9c1b51eSKate Stone               stop_info_sp->GetStopReason() == eStopReasonBreakpoint) {
568b9c1b51eSKate Stone             lldb::break_id_t bp_site_id =
569b9c1b51eSKate Stone                 (lldb::break_id_t)stop_info_sp->GetValue();
570b9c1b51eSKate Stone             BreakpointSiteSP bp_site_sp(
571b9c1b51eSKate Stone                 process->GetBreakpointSiteList().FindByID(bp_site_id));
572b9c1b51eSKate Stone             if (bp_site_sp) {
573c7bece56SGreg Clayton               const size_t num_owners = bp_site_sp->GetNumberOfOwners();
574b9c1b51eSKate Stone               for (size_t i = 0; i < num_owners; i++) {
575b9c1b51eSKate Stone                 Breakpoint &bp_ref =
576b9c1b51eSKate Stone                     bp_site_sp->GetOwnerAtIndex(i)->GetBreakpoint();
577b9c1b51eSKate Stone                 if (!bp_ref.IsInternal()) {
5780e41084aSJim Ingham                   bp_ref.SetIgnoreCount(m_options.m_ignore);
5790e41084aSJim Ingham                 }
5800e41084aSJim Ingham               }
5810e41084aSJim Ingham             }
5820e41084aSJim Ingham           }
5830e41084aSJim Ingham         }
5840e41084aSJim Ingham       }
5850e41084aSJim Ingham 
58641f2b940SJim Ingham       { // Scope for thread list mutex:
587b9c1b51eSKate Stone         std::lock_guard<std::recursive_mutex> guard(
588b9c1b51eSKate Stone             process->GetThreadList().GetMutex());
58930fdc8d8SChris Lattner         const uint32_t num_threads = process->GetThreadList().GetSize();
59030fdc8d8SChris Lattner 
59130fdc8d8SChris Lattner         // Set the actions that the threads should each take when resuming
592b9c1b51eSKate Stone         for (uint32_t idx = 0; idx < num_threads; ++idx) {
5936c9ed91cSJim Ingham           const bool override_suspend = false;
594b9c1b51eSKate Stone           process->GetThreadList().GetThreadAtIndex(idx)->SetResumeState(
595b9c1b51eSKate Stone               eStateRunning, override_suspend);
59630fdc8d8SChris Lattner         }
59741f2b940SJim Ingham       }
59830fdc8d8SChris Lattner 
5994446487dSPavel Labath       const uint32_t iohandler_id = process->GetIOHandlerID();
6004446487dSPavel Labath 
601dc6224e0SGreg Clayton       StreamString stream;
60297206d57SZachary Turner       Status error;
603dc6224e0SGreg Clayton       if (synchronous_execution)
604dc6224e0SGreg Clayton         error = process->ResumeSynchronous(&stream);
605dc6224e0SGreg Clayton       else
606dc6224e0SGreg Clayton         error = process->Resume();
607a3b89e27STodd Fiala 
608b9c1b51eSKate Stone       if (error.Success()) {
609b9c1b51eSKate Stone         // There is a race condition where this thread will return up the call
61005097246SAdrian Prantl         // stack to the main command handler and show an (lldb) prompt before
61105097246SAdrian Prantl         // HandlePrivateEvent (from PrivateStateThread) has a chance to call
61205097246SAdrian Prantl         // PushProcessIOHandler().
6133879fe00SPavel Labath         process->SyncIOHandler(iohandler_id, std::chrono::seconds(2));
614a3b89e27STodd Fiala 
615b9c1b51eSKate Stone         result.AppendMessageWithFormat("Process %" PRIu64 " resuming\n",
616b9c1b51eSKate Stone                                        process->GetID());
617b9c1b51eSKate Stone         if (synchronous_execution) {
618b9c1b51eSKate Stone           // If any state changed events had anything to say, add that to the
619b9c1b51eSKate Stone           // result
620c156427dSZachary Turner           result.AppendMessage(stream.GetString());
62130fdc8d8SChris Lattner 
62230fdc8d8SChris Lattner           result.SetDidChangeProcessState(true);
62330fdc8d8SChris Lattner           result.SetStatus(eReturnStatusSuccessFinishNoResult);
624b9c1b51eSKate Stone         } else {
62530fdc8d8SChris Lattner           result.SetStatus(eReturnStatusSuccessContinuingNoResult);
62630fdc8d8SChris Lattner         }
627b9c1b51eSKate Stone       } else {
628b9c1b51eSKate Stone         result.AppendErrorWithFormat("Failed to resume process: %s.\n",
629b9c1b51eSKate Stone                                      error.AsCString());
63030fdc8d8SChris Lattner         result.SetStatus(eReturnStatusFailed);
63130fdc8d8SChris Lattner       }
632b9c1b51eSKate Stone     } else {
633b9c1b51eSKate Stone       result.AppendErrorWithFormat(
634b9c1b51eSKate Stone           "Process cannot be continued from its current state (%s).\n",
63530fdc8d8SChris Lattner           StateAsCString(state));
63630fdc8d8SChris Lattner       result.SetStatus(eReturnStatusFailed);
63730fdc8d8SChris Lattner     }
63830fdc8d8SChris Lattner     return result.Succeeded();
63930fdc8d8SChris Lattner   }
6400e41084aSJim Ingham 
641b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
6420e41084aSJim Ingham 
6430e41084aSJim Ingham   CommandOptions m_options;
6440e41084aSJim Ingham };
6450e41084aSJim Ingham 
64630fdc8d8SChris Lattner // CommandObjectProcessDetach
647438dfcffSRaphael Isemann #define LLDB_OPTIONS_process_detach
648438dfcffSRaphael Isemann #include "CommandOptions.inc"
6491f0f5b5bSZachary Turner 
650bb9caf73SJim Ingham #pragma mark CommandObjectProcessDetach
65130fdc8d8SChris Lattner 
652b9c1b51eSKate Stone class CommandObjectProcessDetach : public CommandObjectParsed {
65330fdc8d8SChris Lattner public:
654b9c1b51eSKate Stone   class CommandOptions : public Options {
655acff8950SJim Ingham   public:
656b9c1b51eSKate Stone     CommandOptions() : Options() { OptionParsingStarting(nullptr); }
657acff8950SJim Ingham 
65849bcfd80SEugene Zelenko     ~CommandOptions() override = default;
659acff8950SJim Ingham 
66097206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
661b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
66297206d57SZachary Turner       Status error;
663acff8950SJim Ingham       const int short_option = m_getopt_table[option_idx].val;
664acff8950SJim Ingham 
665b9c1b51eSKate Stone       switch (short_option) {
666acff8950SJim Ingham       case 's':
667acff8950SJim Ingham         bool tmp_result;
668acff8950SJim Ingham         bool success;
66947cbf4a0SPavel Labath         tmp_result = OptionArgParser::ToBoolean(option_arg, false, &success);
670acff8950SJim Ingham         if (!success)
671b9c1b51eSKate Stone           error.SetErrorStringWithFormat("invalid boolean option: \"%s\"",
672fe11483bSZachary Turner                                          option_arg.str().c_str());
673b9c1b51eSKate Stone         else {
674acff8950SJim Ingham           if (tmp_result)
675acff8950SJim Ingham             m_keep_stopped = eLazyBoolYes;
676acff8950SJim Ingham           else
677acff8950SJim Ingham             m_keep_stopped = eLazyBoolNo;
678acff8950SJim Ingham         }
679acff8950SJim Ingham         break;
680acff8950SJim Ingham       default:
68136162014SRaphael Isemann         llvm_unreachable("Unimplemented option");
682acff8950SJim Ingham       }
683acff8950SJim Ingham       return error;
684acff8950SJim Ingham     }
685acff8950SJim Ingham 
686b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
687acff8950SJim Ingham       m_keep_stopped = eLazyBoolCalculate;
688acff8950SJim Ingham     }
689acff8950SJim Ingham 
6901f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
69170602439SZachary Turner       return llvm::makeArrayRef(g_process_detach_options);
6921f0f5b5bSZachary Turner     }
693acff8950SJim Ingham 
694acff8950SJim Ingham     // Instance variables to hold the values for command options.
695acff8950SJim Ingham     LazyBool m_keep_stopped;
696acff8950SJim Ingham   };
69730fdc8d8SChris Lattner 
6987428a18cSKate Stone   CommandObjectProcessDetach(CommandInterpreter &interpreter)
699b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process detach",
700b9c1b51eSKate Stone                             "Detach from the current target process.",
701a7015092SGreg Clayton                             "process detach",
702b9c1b51eSKate Stone                             eCommandRequiresProcess | eCommandTryTargetAPILock |
703b9c1b51eSKate Stone                                 eCommandProcessMustBeLaunched),
704b9c1b51eSKate Stone         m_options() {}
70530fdc8d8SChris Lattner 
70649bcfd80SEugene Zelenko   ~CommandObjectProcessDetach() override = default;
70730fdc8d8SChris Lattner 
708b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
709acff8950SJim Ingham 
7105a988416SJim Ingham protected:
711b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
712f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
713acff8950SJim Ingham     // FIXME: This will be a Command Option:
714acff8950SJim Ingham     bool keep_stopped;
715b9c1b51eSKate Stone     if (m_options.m_keep_stopped == eLazyBoolCalculate) {
716acff8950SJim Ingham       // Check the process default:
71749bcfd80SEugene Zelenko       keep_stopped = process->GetDetachKeepsStopped();
718b9c1b51eSKate Stone     } else if (m_options.m_keep_stopped == eLazyBoolYes)
719acff8950SJim Ingham       keep_stopped = true;
720acff8950SJim Ingham     else
721acff8950SJim Ingham       keep_stopped = false;
722acff8950SJim Ingham 
72397206d57SZachary Turner     Status error(process->Detach(keep_stopped));
724b9c1b51eSKate Stone     if (error.Success()) {
72530fdc8d8SChris Lattner       result.SetStatus(eReturnStatusSuccessFinishResult);
726b9c1b51eSKate Stone     } else {
72730fdc8d8SChris Lattner       result.AppendErrorWithFormat("Detach failed: %s\n", error.AsCString());
72830fdc8d8SChris Lattner       result.SetStatus(eReturnStatusFailed);
72930fdc8d8SChris Lattner       return false;
73030fdc8d8SChris Lattner     }
73130fdc8d8SChris Lattner     return result.Succeeded();
73230fdc8d8SChris Lattner   }
733acff8950SJim Ingham 
734acff8950SJim Ingham   CommandOptions m_options;
735acff8950SJim Ingham };
736acff8950SJim Ingham 
737b766a73dSGreg Clayton // CommandObjectProcessConnect
738438dfcffSRaphael Isemann #define LLDB_OPTIONS_process_connect
739438dfcffSRaphael Isemann #include "CommandOptions.inc"
7401f0f5b5bSZachary Turner 
741b766a73dSGreg Clayton #pragma mark CommandObjectProcessConnect
742b766a73dSGreg Clayton 
743b9c1b51eSKate Stone class CommandObjectProcessConnect : public CommandObjectParsed {
744b766a73dSGreg Clayton public:
745b9c1b51eSKate Stone   class CommandOptions : public Options {
746b766a73dSGreg Clayton   public:
747b9c1b51eSKate Stone     CommandOptions() : Options() {
748b9c1b51eSKate Stone       // Keep default values of all options in one place: OptionParsingStarting
749b9c1b51eSKate Stone       // ()
750e1cfbc79STodd Fiala       OptionParsingStarting(nullptr);
751b766a73dSGreg Clayton     }
752b766a73dSGreg Clayton 
75349bcfd80SEugene Zelenko     ~CommandOptions() override = default;
754b766a73dSGreg Clayton 
75597206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
756b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
75797206d57SZachary Turner       Status error;
7583bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
759b766a73dSGreg Clayton 
760b9c1b51eSKate Stone       switch (short_option) {
761b766a73dSGreg Clayton       case 'p':
762adcd0268SBenjamin Kramer         plugin_name.assign(std::string(option_arg));
763b766a73dSGreg Clayton         break;
764b766a73dSGreg Clayton 
765b766a73dSGreg Clayton       default:
76636162014SRaphael Isemann         llvm_unreachable("Unimplemented option");
767b766a73dSGreg Clayton       }
768b766a73dSGreg Clayton       return error;
769b766a73dSGreg Clayton     }
770b766a73dSGreg Clayton 
771b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
772b766a73dSGreg Clayton       plugin_name.clear();
773b766a73dSGreg Clayton     }
774b766a73dSGreg Clayton 
7751f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
77670602439SZachary Turner       return llvm::makeArrayRef(g_process_connect_options);
7771f0f5b5bSZachary Turner     }
778b766a73dSGreg Clayton 
779b766a73dSGreg Clayton     // Instance variables to hold the values for command options.
780b766a73dSGreg Clayton 
781b766a73dSGreg Clayton     std::string plugin_name;
782b766a73dSGreg Clayton   };
783b766a73dSGreg Clayton 
784b9c1b51eSKate Stone   CommandObjectProcessConnect(CommandInterpreter &interpreter)
785b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process connect",
786b766a73dSGreg Clayton                             "Connect to a remote debug service.",
787b9c1b51eSKate Stone                             "process connect <remote-url>", 0),
788b9c1b51eSKate Stone         m_options() {}
789b766a73dSGreg Clayton 
79049bcfd80SEugene Zelenko   ~CommandObjectProcessConnect() override = default;
791b766a73dSGreg Clayton 
792b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
7935a988416SJim Ingham 
7945a988416SJim Ingham protected:
795b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
796b9c1b51eSKate Stone     if (command.GetArgumentCount() != 1) {
797b9c1b51eSKate Stone       result.AppendErrorWithFormat(
798b9c1b51eSKate Stone           "'%s' takes exactly one argument:\nUsage: %s\n", m_cmd_name.c_str(),
799ccd6cffbSTamas Berghammer           m_cmd_syntax.c_str());
800ccd6cffbSTamas Berghammer       result.SetStatus(eReturnStatusFailed);
801ccd6cffbSTamas Berghammer       return false;
802ccd6cffbSTamas Berghammer     }
803ccd6cffbSTamas Berghammer 
804ccd6cffbSTamas Berghammer     Process *process = m_exe_ctx.GetProcessPtr();
805b9c1b51eSKate Stone     if (process && process->IsAlive()) {
806b9c1b51eSKate Stone       result.AppendErrorWithFormat(
807b9c1b51eSKate Stone           "Process %" PRIu64
808b9c1b51eSKate Stone           " is currently being debugged, kill the process before connecting.\n",
809b766a73dSGreg Clayton           process->GetID());
810b766a73dSGreg Clayton       result.SetStatus(eReturnStatusFailed);
811b766a73dSGreg Clayton       return false;
812b766a73dSGreg Clayton     }
813b766a73dSGreg Clayton 
814ccd6cffbSTamas Berghammer     const char *plugin_name = nullptr;
815b766a73dSGreg Clayton     if (!m_options.plugin_name.empty())
816b766a73dSGreg Clayton       plugin_name = m_options.plugin_name.c_str();
817b766a73dSGreg Clayton 
81897206d57SZachary Turner     Status error;
81957179860SJonas Devlieghere     Debugger &debugger = GetDebugger();
820ccd6cffbSTamas Berghammer     PlatformSP platform_sp = m_interpreter.GetPlatform(true);
821b9c1b51eSKate Stone     ProcessSP process_sp = platform_sp->ConnectProcess(
822b9c1b51eSKate Stone         command.GetArgumentAtIndex(0), plugin_name, debugger,
823b9c1b51eSKate Stone         debugger.GetSelectedTarget().get(), error);
824b9c1b51eSKate Stone     if (error.Fail() || process_sp == nullptr) {
825ccd6cffbSTamas Berghammer       result.AppendError(error.AsCString("Error connecting to the process"));
826b766a73dSGreg Clayton       result.SetStatus(eReturnStatusFailed);
827b766a73dSGreg Clayton       return false;
828b766a73dSGreg Clayton     }
829ccd6cffbSTamas Berghammer     return true;
830b766a73dSGreg Clayton   }
831b766a73dSGreg Clayton 
832b766a73dSGreg Clayton   CommandOptions m_options;
833b766a73dSGreg Clayton };
834b766a73dSGreg Clayton 
835998255bfSGreg Clayton // CommandObjectProcessPlugin
836998255bfSGreg Clayton #pragma mark CommandObjectProcessPlugin
837998255bfSGreg Clayton 
838b9c1b51eSKate Stone class CommandObjectProcessPlugin : public CommandObjectProxy {
839998255bfSGreg Clayton public:
8407428a18cSKate Stone   CommandObjectProcessPlugin(CommandInterpreter &interpreter)
841b9c1b51eSKate Stone       : CommandObjectProxy(
842b9c1b51eSKate Stone             interpreter, "process plugin",
843b9c1b51eSKate Stone             "Send a custom command to the current target process plug-in.",
844b9c1b51eSKate Stone             "process plugin <args>", 0) {}
845998255bfSGreg Clayton 
84649bcfd80SEugene Zelenko   ~CommandObjectProcessPlugin() override = default;
847998255bfSGreg Clayton 
848b9c1b51eSKate Stone   CommandObject *GetProxyCommandObject() override {
849e05b2efeSGreg Clayton     Process *process = m_interpreter.GetExecutionContext().GetProcessPtr();
850998255bfSGreg Clayton     if (process)
851998255bfSGreg Clayton       return process->GetPluginCommandObject();
85249bcfd80SEugene Zelenko     return nullptr;
853998255bfSGreg Clayton   }
854998255bfSGreg Clayton };
855998255bfSGreg Clayton 
8568f343b09SGreg Clayton // CommandObjectProcessLoad
857438dfcffSRaphael Isemann #define LLDB_OPTIONS_process_load
858438dfcffSRaphael Isemann #include "CommandOptions.inc"
8591f0f5b5bSZachary Turner 
860bb9caf73SJim Ingham #pragma mark CommandObjectProcessLoad
8618f343b09SGreg Clayton 
862b9c1b51eSKate Stone class CommandObjectProcessLoad : public CommandObjectParsed {
8638f343b09SGreg Clayton public:
864b9c1b51eSKate Stone   class CommandOptions : public Options {
8654fbd67acSTamas Berghammer   public:
866b9c1b51eSKate Stone     CommandOptions() : Options() {
867b9c1b51eSKate Stone       // Keep default values of all options in one place: OptionParsingStarting
868b9c1b51eSKate Stone       // ()
869e1cfbc79STodd Fiala       OptionParsingStarting(nullptr);
8704fbd67acSTamas Berghammer     }
8714fbd67acSTamas Berghammer 
8724fbd67acSTamas Berghammer     ~CommandOptions() override = default;
8734fbd67acSTamas Berghammer 
87497206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
875b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
87697206d57SZachary Turner       Status error;
8774fbd67acSTamas Berghammer       const int short_option = m_getopt_table[option_idx].val;
878b9c1b51eSKate Stone       switch (short_option) {
8794fbd67acSTamas Berghammer       case 'i':
8804fbd67acSTamas Berghammer         do_install = true;
881fe11483bSZachary Turner         if (!option_arg.empty())
8828f3be7a3SJonas Devlieghere           install_path.SetFile(option_arg, FileSpec::Style::native);
8834fbd67acSTamas Berghammer         break;
8844fbd67acSTamas Berghammer       default:
88536162014SRaphael Isemann         llvm_unreachable("Unimplemented option");
8864fbd67acSTamas Berghammer       }
8874fbd67acSTamas Berghammer       return error;
8884fbd67acSTamas Berghammer     }
8894fbd67acSTamas Berghammer 
890b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
8914fbd67acSTamas Berghammer       do_install = false;
8924fbd67acSTamas Berghammer       install_path.Clear();
8934fbd67acSTamas Berghammer     }
8944fbd67acSTamas Berghammer 
8951f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
89670602439SZachary Turner       return llvm::makeArrayRef(g_process_load_options);
8971f0f5b5bSZachary Turner     }
8984fbd67acSTamas Berghammer 
8994fbd67acSTamas Berghammer     // Instance variables to hold the values for command options.
9004fbd67acSTamas Berghammer     bool do_install;
9014fbd67acSTamas Berghammer     FileSpec install_path;
9024fbd67acSTamas Berghammer   };
9038f343b09SGreg Clayton 
904b9c1b51eSKate Stone   CommandObjectProcessLoad(CommandInterpreter &interpreter)
905b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process load",
9068f343b09SGreg Clayton                             "Load a shared library into the current process.",
9078f343b09SGreg Clayton                             "process load <filename> [<filename> ...]",
908b9c1b51eSKate Stone                             eCommandRequiresProcess | eCommandTryTargetAPILock |
909e87764f2SEnrico Granata                                 eCommandProcessMustBeLaunched |
9104fbd67acSTamas Berghammer                                 eCommandProcessMustBePaused),
911b9c1b51eSKate Stone         m_options() {}
9128f343b09SGreg Clayton 
9134fbd67acSTamas Berghammer   ~CommandObjectProcessLoad() override = default;
9144fbd67acSTamas Berghammer 
915b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
9168f343b09SGreg Clayton 
9175a988416SJim Ingham protected:
918b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
919f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
9208f343b09SGreg Clayton 
92197d2c401SZachary Turner     for (auto &entry : command.entries()) {
92297206d57SZachary Turner       Status error;
9233cb132a0STamas Berghammer       PlatformSP platform = process->GetTarget().GetPlatform();
9240d9a201eSRaphael Isemann       llvm::StringRef image_path = entry.ref();
9254fbd67acSTamas Berghammer       uint32_t image_token = LLDB_INVALID_IMAGE_TOKEN;
9264fbd67acSTamas Berghammer 
927b9c1b51eSKate Stone       if (!m_options.do_install) {
9288f3be7a3SJonas Devlieghere         FileSpec image_spec(image_path);
9293cb132a0STamas Berghammer         platform->ResolveRemotePath(image_spec, image_spec);
930b9c1b51eSKate Stone         image_token =
931b9c1b51eSKate Stone             platform->LoadImage(process, FileSpec(), image_spec, error);
932b9c1b51eSKate Stone       } else if (m_options.install_path) {
9338f3be7a3SJonas Devlieghere         FileSpec image_spec(image_path);
9348f3be7a3SJonas Devlieghere         FileSystem::Instance().Resolve(image_spec);
935b9c1b51eSKate Stone         platform->ResolveRemotePath(m_options.install_path,
936b9c1b51eSKate Stone                                     m_options.install_path);
937b9c1b51eSKate Stone         image_token = platform->LoadImage(process, image_spec,
938b9c1b51eSKate Stone                                           m_options.install_path, error);
939b9c1b51eSKate Stone       } else {
9408f3be7a3SJonas Devlieghere         FileSpec image_spec(image_path);
9418f3be7a3SJonas Devlieghere         FileSystem::Instance().Resolve(image_spec);
942b9c1b51eSKate Stone         image_token =
943b9c1b51eSKate Stone             platform->LoadImage(process, image_spec, FileSpec(), error);
9444fbd67acSTamas Berghammer       }
9454fbd67acSTamas Berghammer 
946b9c1b51eSKate Stone       if (image_token != LLDB_INVALID_IMAGE_TOKEN) {
947b9c1b51eSKate Stone         result.AppendMessageWithFormat(
94897d2c401SZachary Turner             "Loading \"%s\"...ok\nImage %u loaded.\n", image_path.str().c_str(),
94997d2c401SZachary Turner             image_token);
9508f343b09SGreg Clayton         result.SetStatus(eReturnStatusSuccessFinishResult);
951b9c1b51eSKate Stone       } else {
95297d2c401SZachary Turner         result.AppendErrorWithFormat("failed to load '%s': %s",
95397d2c401SZachary Turner                                      image_path.str().c_str(),
954b9c1b51eSKate Stone                                      error.AsCString());
9558f343b09SGreg Clayton         result.SetStatus(eReturnStatusFailed);
9568f343b09SGreg Clayton       }
9578f343b09SGreg Clayton     }
9588f343b09SGreg Clayton     return result.Succeeded();
9598f343b09SGreg Clayton   }
9604fbd67acSTamas Berghammer 
9614fbd67acSTamas Berghammer   CommandOptions m_options;
9628f343b09SGreg Clayton };
9638f343b09SGreg Clayton 
9648f343b09SGreg Clayton // CommandObjectProcessUnload
965bb9caf73SJim Ingham #pragma mark CommandObjectProcessUnload
9668f343b09SGreg Clayton 
967b9c1b51eSKate Stone class CommandObjectProcessUnload : public CommandObjectParsed {
9688f343b09SGreg Clayton public:
969b9c1b51eSKate Stone   CommandObjectProcessUnload(CommandInterpreter &interpreter)
970b9c1b51eSKate Stone       : CommandObjectParsed(
971b9c1b51eSKate Stone             interpreter, "process unload",
972b9c1b51eSKate Stone             "Unload a shared library from the current process using the index "
973b9c1b51eSKate Stone             "returned by a previous call to \"process load\".",
9748f343b09SGreg Clayton             "process unload <index>",
975b9c1b51eSKate Stone             eCommandRequiresProcess | eCommandTryTargetAPILock |
976b9c1b51eSKate Stone                 eCommandProcessMustBeLaunched | eCommandProcessMustBePaused) {}
9778f343b09SGreg Clayton 
97849bcfd80SEugene Zelenko   ~CommandObjectProcessUnload() override = default;
9798f343b09SGreg Clayton 
9805a988416SJim Ingham protected:
981b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
982f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
9838f343b09SGreg Clayton 
98497d2c401SZachary Turner     for (auto &entry : command.entries()) {
98597d2c401SZachary Turner       uint32_t image_token;
9860d9a201eSRaphael Isemann       if (entry.ref().getAsInteger(0, image_token)) {
987b9c1b51eSKate Stone         result.AppendErrorWithFormat("invalid image index argument '%s'",
9880d9a201eSRaphael Isemann                                      entry.ref().str().c_str());
9898f343b09SGreg Clayton         result.SetStatus(eReturnStatusFailed);
9908f343b09SGreg Clayton         break;
991b9c1b51eSKate Stone       } else {
99297206d57SZachary Turner         Status error(process->GetTarget().GetPlatform()->UnloadImage(
993b9c1b51eSKate Stone             process, image_token));
994b9c1b51eSKate Stone         if (error.Success()) {
995b9c1b51eSKate Stone           result.AppendMessageWithFormat(
996b9c1b51eSKate Stone               "Unloading shared library with index %u...ok\n", image_token);
9978f343b09SGreg Clayton           result.SetStatus(eReturnStatusSuccessFinishResult);
998b9c1b51eSKate Stone         } else {
999b9c1b51eSKate Stone           result.AppendErrorWithFormat("failed to unload image: %s",
1000b9c1b51eSKate Stone                                        error.AsCString());
10018f343b09SGreg Clayton           result.SetStatus(eReturnStatusFailed);
10028f343b09SGreg Clayton           break;
10038f343b09SGreg Clayton         }
10048f343b09SGreg Clayton       }
10058f343b09SGreg Clayton     }
10068f343b09SGreg Clayton     return result.Succeeded();
10078f343b09SGreg Clayton   }
10088f343b09SGreg Clayton };
10098f343b09SGreg Clayton 
101030fdc8d8SChris Lattner // CommandObjectProcessSignal
1011bb9caf73SJim Ingham #pragma mark CommandObjectProcessSignal
101230fdc8d8SChris Lattner 
1013b9c1b51eSKate Stone class CommandObjectProcessSignal : public CommandObjectParsed {
101430fdc8d8SChris Lattner public:
10157428a18cSKate Stone   CommandObjectProcessSignal(CommandInterpreter &interpreter)
1016a925974bSAdrian Prantl       : CommandObjectParsed(
1017a925974bSAdrian Prantl             interpreter, "process signal",
1018a925974bSAdrian Prantl             "Send a UNIX signal to the current target process.", nullptr,
1019a925974bSAdrian Prantl             eCommandRequiresProcess | eCommandTryTargetAPILock) {
1020405fe67fSCaroline Tice     CommandArgumentEntry arg;
1021405fe67fSCaroline Tice     CommandArgumentData signal_arg;
1022405fe67fSCaroline Tice 
1023405fe67fSCaroline Tice     // Define the first (and only) variant of this arg.
1024c0dbdfb6SCaroline Tice     signal_arg.arg_type = eArgTypeUnixSignal;
1025405fe67fSCaroline Tice     signal_arg.arg_repetition = eArgRepeatPlain;
1026405fe67fSCaroline Tice 
1027b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
1028b9c1b51eSKate Stone     // argument entry.
1029405fe67fSCaroline Tice     arg.push_back(signal_arg);
1030405fe67fSCaroline Tice 
1031405fe67fSCaroline Tice     // Push the data for the first argument into the m_arguments vector.
1032405fe67fSCaroline Tice     m_arguments.push_back(arg);
103330fdc8d8SChris Lattner   }
103430fdc8d8SChris Lattner 
103549bcfd80SEugene Zelenko   ~CommandObjectProcessSignal() override = default;
103630fdc8d8SChris Lattner 
10375a988416SJim Ingham protected:
1038b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1039f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
104030fdc8d8SChris Lattner 
1041b9c1b51eSKate Stone     if (command.GetArgumentCount() == 1) {
1042237cd906SGreg Clayton       int signo = LLDB_INVALID_SIGNAL_NUMBER;
1043237cd906SGreg Clayton 
1044237cd906SGreg Clayton       const char *signal_name = command.GetArgumentAtIndex(0);
1045237cd906SGreg Clayton       if (::isxdigit(signal_name[0]))
1046b9c1b51eSKate Stone         signo =
1047b9c1b51eSKate Stone             StringConvert::ToSInt32(signal_name, LLDB_INVALID_SIGNAL_NUMBER, 0);
1048237cd906SGreg Clayton       else
104998d0a4b3SChaoren Lin         signo = process->GetUnixSignals()->GetSignalNumberFromName(signal_name);
1050237cd906SGreg Clayton 
1051b9c1b51eSKate Stone       if (signo == LLDB_INVALID_SIGNAL_NUMBER) {
1052b9c1b51eSKate Stone         result.AppendErrorWithFormat("Invalid signal argument '%s'.\n",
1053b9c1b51eSKate Stone                                      command.GetArgumentAtIndex(0));
105430fdc8d8SChris Lattner         result.SetStatus(eReturnStatusFailed);
1055b9c1b51eSKate Stone       } else {
105697206d57SZachary Turner         Status error(process->Signal(signo));
1057b9c1b51eSKate Stone         if (error.Success()) {
105830fdc8d8SChris Lattner           result.SetStatus(eReturnStatusSuccessFinishResult);
1059b9c1b51eSKate Stone         } else {
1060b9c1b51eSKate Stone           result.AppendErrorWithFormat("Failed to send signal %i: %s\n", signo,
1061b9c1b51eSKate Stone                                        error.AsCString());
106230fdc8d8SChris Lattner           result.SetStatus(eReturnStatusFailed);
106330fdc8d8SChris Lattner         }
106430fdc8d8SChris Lattner       }
1065b9c1b51eSKate Stone     } else {
1066b9c1b51eSKate Stone       result.AppendErrorWithFormat(
1067b9c1b51eSKate Stone           "'%s' takes exactly one signal number argument:\nUsage: %s\n",
1068b9c1b51eSKate Stone           m_cmd_name.c_str(), m_cmd_syntax.c_str());
106930fdc8d8SChris Lattner       result.SetStatus(eReturnStatusFailed);
107030fdc8d8SChris Lattner     }
107130fdc8d8SChris Lattner     return result.Succeeded();
107230fdc8d8SChris Lattner   }
107330fdc8d8SChris Lattner };
107430fdc8d8SChris Lattner 
107530fdc8d8SChris Lattner // CommandObjectProcessInterrupt
1076bb9caf73SJim Ingham #pragma mark CommandObjectProcessInterrupt
107730fdc8d8SChris Lattner 
1078b9c1b51eSKate Stone class CommandObjectProcessInterrupt : public CommandObjectParsed {
107930fdc8d8SChris Lattner public:
10807428a18cSKate Stone   CommandObjectProcessInterrupt(CommandInterpreter &interpreter)
1081b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process interrupt",
1082b9c1b51eSKate Stone                             "Interrupt the current target process.",
1083a7015092SGreg Clayton                             "process interrupt",
1084b9c1b51eSKate Stone                             eCommandRequiresProcess | eCommandTryTargetAPILock |
1085b9c1b51eSKate Stone                                 eCommandProcessMustBeLaunched) {}
108630fdc8d8SChris Lattner 
108749bcfd80SEugene Zelenko   ~CommandObjectProcessInterrupt() override = default;
108830fdc8d8SChris Lattner 
10895a988416SJim Ingham protected:
1090b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1091f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
1092b9c1b51eSKate Stone     if (process == nullptr) {
109330fdc8d8SChris Lattner       result.AppendError("no process to halt");
109430fdc8d8SChris Lattner       result.SetStatus(eReturnStatusFailed);
109530fdc8d8SChris Lattner       return false;
109630fdc8d8SChris Lattner     }
109730fdc8d8SChris Lattner 
1098b9c1b51eSKate Stone     if (command.GetArgumentCount() == 0) {
1099f9b57b9dSGreg Clayton       bool clear_thread_plans = true;
110097206d57SZachary Turner       Status error(process->Halt(clear_thread_plans));
1101b9c1b51eSKate Stone       if (error.Success()) {
110230fdc8d8SChris Lattner         result.SetStatus(eReturnStatusSuccessFinishResult);
1103b9c1b51eSKate Stone       } else {
1104b9c1b51eSKate Stone         result.AppendErrorWithFormat("Failed to halt process: %s\n",
1105b9c1b51eSKate Stone                                      error.AsCString());
110630fdc8d8SChris Lattner         result.SetStatus(eReturnStatusFailed);
110730fdc8d8SChris Lattner       }
1108b9c1b51eSKate Stone     } else {
1109fd54b368SJason Molenda       result.AppendErrorWithFormat("'%s' takes no arguments:\nUsage: %s\n",
1110b9c1b51eSKate Stone                                    m_cmd_name.c_str(), m_cmd_syntax.c_str());
111130fdc8d8SChris Lattner       result.SetStatus(eReturnStatusFailed);
111230fdc8d8SChris Lattner     }
111330fdc8d8SChris Lattner     return result.Succeeded();
111430fdc8d8SChris Lattner   }
111530fdc8d8SChris Lattner };
111630fdc8d8SChris Lattner 
111730fdc8d8SChris Lattner // CommandObjectProcessKill
1118bb9caf73SJim Ingham #pragma mark CommandObjectProcessKill
111930fdc8d8SChris Lattner 
1120b9c1b51eSKate Stone class CommandObjectProcessKill : public CommandObjectParsed {
112130fdc8d8SChris Lattner public:
11227428a18cSKate Stone   CommandObjectProcessKill(CommandInterpreter &interpreter)
1123b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process kill",
1124b9c1b51eSKate Stone                             "Terminate the current target process.",
1125b9c1b51eSKate Stone                             "process kill",
1126b9c1b51eSKate Stone                             eCommandRequiresProcess | eCommandTryTargetAPILock |
1127b9c1b51eSKate Stone                                 eCommandProcessMustBeLaunched) {}
112830fdc8d8SChris Lattner 
112949bcfd80SEugene Zelenko   ~CommandObjectProcessKill() override = default;
113030fdc8d8SChris Lattner 
11315a988416SJim Ingham protected:
1132b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1133f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
1134b9c1b51eSKate Stone     if (process == nullptr) {
113530fdc8d8SChris Lattner       result.AppendError("no process to kill");
113630fdc8d8SChris Lattner       result.SetStatus(eReturnStatusFailed);
113730fdc8d8SChris Lattner       return false;
113830fdc8d8SChris Lattner     }
113930fdc8d8SChris Lattner 
1140b9c1b51eSKate Stone     if (command.GetArgumentCount() == 0) {
114197206d57SZachary Turner       Status error(process->Destroy(true));
1142b9c1b51eSKate Stone       if (error.Success()) {
114330fdc8d8SChris Lattner         result.SetStatus(eReturnStatusSuccessFinishResult);
1144b9c1b51eSKate Stone       } else {
1145b9c1b51eSKate Stone         result.AppendErrorWithFormat("Failed to kill process: %s\n",
1146b9c1b51eSKate Stone                                      error.AsCString());
114730fdc8d8SChris Lattner         result.SetStatus(eReturnStatusFailed);
114830fdc8d8SChris Lattner       }
1149b9c1b51eSKate Stone     } else {
1150fd54b368SJason Molenda       result.AppendErrorWithFormat("'%s' takes no arguments:\nUsage: %s\n",
1151b9c1b51eSKate Stone                                    m_cmd_name.c_str(), m_cmd_syntax.c_str());
115230fdc8d8SChris Lattner       result.SetStatus(eReturnStatusFailed);
115330fdc8d8SChris Lattner     }
115430fdc8d8SChris Lattner     return result.Succeeded();
115530fdc8d8SChris Lattner   }
115630fdc8d8SChris Lattner };
115730fdc8d8SChris Lattner 
1158a2715cf1SGreg Clayton // CommandObjectProcessSaveCore
1159a2715cf1SGreg Clayton #pragma mark CommandObjectProcessSaveCore
1160a2715cf1SGreg Clayton 
1161b9c1b51eSKate Stone class CommandObjectProcessSaveCore : public CommandObjectParsed {
1162a2715cf1SGreg Clayton public:
1163b9c1b51eSKate Stone   CommandObjectProcessSaveCore(CommandInterpreter &interpreter)
1164b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process save-core",
1165b9c1b51eSKate Stone                             "Save the current process as a core file using an "
1166b9c1b51eSKate Stone                             "appropriate file type.",
1167a2715cf1SGreg Clayton                             "process save-core FILE",
1168b9c1b51eSKate Stone                             eCommandRequiresProcess | eCommandTryTargetAPILock |
1169b9c1b51eSKate Stone                                 eCommandProcessMustBeLaunched) {}
1170a2715cf1SGreg Clayton 
117149bcfd80SEugene Zelenko   ~CommandObjectProcessSaveCore() override = default;
1172a2715cf1SGreg Clayton 
1173a2715cf1SGreg Clayton protected:
1174b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1175a2715cf1SGreg Clayton     ProcessSP process_sp = m_exe_ctx.GetProcessSP();
1176b9c1b51eSKate Stone     if (process_sp) {
1177b9c1b51eSKate Stone       if (command.GetArgumentCount() == 1) {
11788f3be7a3SJonas Devlieghere         FileSpec output_file(command.GetArgumentAtIndex(0));
117997206d57SZachary Turner         Status error = PluginManager::SaveCore(process_sp, output_file);
1180b9c1b51eSKate Stone         if (error.Success()) {
1181a2715cf1SGreg Clayton           result.SetStatus(eReturnStatusSuccessFinishResult);
1182b9c1b51eSKate Stone         } else {
1183b9c1b51eSKate Stone           result.AppendErrorWithFormat(
1184b9c1b51eSKate Stone               "Failed to save core file for process: %s\n", error.AsCString());
1185a2715cf1SGreg Clayton           result.SetStatus(eReturnStatusFailed);
1186a2715cf1SGreg Clayton         }
1187b9c1b51eSKate Stone       } else {
1188a2715cf1SGreg Clayton         result.AppendErrorWithFormat("'%s' takes one arguments:\nUsage: %s\n",
1189b9c1b51eSKate Stone                                      m_cmd_name.c_str(), m_cmd_syntax.c_str());
1190a2715cf1SGreg Clayton         result.SetStatus(eReturnStatusFailed);
1191a2715cf1SGreg Clayton       }
1192b9c1b51eSKate Stone     } else {
1193a2715cf1SGreg Clayton       result.AppendError("invalid process");
1194a2715cf1SGreg Clayton       result.SetStatus(eReturnStatusFailed);
1195a2715cf1SGreg Clayton       return false;
1196a2715cf1SGreg Clayton     }
1197a2715cf1SGreg Clayton 
1198a2715cf1SGreg Clayton     return result.Succeeded();
1199a2715cf1SGreg Clayton   }
1200a2715cf1SGreg Clayton };
1201a2715cf1SGreg Clayton 
12024b9bea87SJim Ingham // CommandObjectProcessStatus
1203bb9caf73SJim Ingham #pragma mark CommandObjectProcessStatus
1204d7c403e6SMed Ismail Bennani #define LLDB_OPTIONS_process_status
1205d7c403e6SMed Ismail Bennani #include "CommandOptions.inc"
1206bb9caf73SJim Ingham 
1207b9c1b51eSKate Stone class CommandObjectProcessStatus : public CommandObjectParsed {
12084b9bea87SJim Ingham public:
12097428a18cSKate Stone   CommandObjectProcessStatus(CommandInterpreter &interpreter)
1210b9c1b51eSKate Stone       : CommandObjectParsed(
1211b9c1b51eSKate Stone             interpreter, "process status",
1212b9c1b51eSKate Stone             "Show status and stop location for the current target process.",
1213b9c1b51eSKate Stone             "process status",
1214d7c403e6SMed Ismail Bennani             eCommandRequiresProcess | eCommandTryTargetAPILock),
1215d7c403e6SMed Ismail Bennani         m_options() {}
12164b9bea87SJim Ingham 
121749bcfd80SEugene Zelenko   ~CommandObjectProcessStatus() override = default;
12184b9bea87SJim Ingham 
1219d7c403e6SMed Ismail Bennani   Options *GetOptions() override { return &m_options; }
1220d7c403e6SMed Ismail Bennani 
1221d7c403e6SMed Ismail Bennani   class CommandOptions : public Options {
1222d7c403e6SMed Ismail Bennani   public:
1223d7c403e6SMed Ismail Bennani     CommandOptions() : Options(), m_verbose(false) {}
1224d7c403e6SMed Ismail Bennani 
1225d7c403e6SMed Ismail Bennani     ~CommandOptions() override = default;
1226d7c403e6SMed Ismail Bennani 
1227d7c403e6SMed Ismail Bennani     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
1228d7c403e6SMed Ismail Bennani                           ExecutionContext *execution_context) override {
1229d7c403e6SMed Ismail Bennani       const int short_option = m_getopt_table[option_idx].val;
1230d7c403e6SMed Ismail Bennani 
1231d7c403e6SMed Ismail Bennani       switch (short_option) {
1232d7c403e6SMed Ismail Bennani       case 'v':
1233d7c403e6SMed Ismail Bennani         m_verbose = true;
1234d7c403e6SMed Ismail Bennani         break;
1235d7c403e6SMed Ismail Bennani       default:
1236d7c403e6SMed Ismail Bennani         llvm_unreachable("Unimplemented option");
1237d7c403e6SMed Ismail Bennani       }
1238d7c403e6SMed Ismail Bennani 
1239d7c403e6SMed Ismail Bennani       return {};
1240d7c403e6SMed Ismail Bennani     }
1241d7c403e6SMed Ismail Bennani 
1242d7c403e6SMed Ismail Bennani     void OptionParsingStarting(ExecutionContext *execution_context) override {
1243d7c403e6SMed Ismail Bennani       m_verbose = false;
1244d7c403e6SMed Ismail Bennani     }
1245d7c403e6SMed Ismail Bennani 
1246d7c403e6SMed Ismail Bennani     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
1247d7c403e6SMed Ismail Bennani       return llvm::makeArrayRef(g_process_status_options);
1248d7c403e6SMed Ismail Bennani     }
1249d7c403e6SMed Ismail Bennani 
1250d7c403e6SMed Ismail Bennani     // Instance variables to hold the values for command options.
1251d7c403e6SMed Ismail Bennani     bool m_verbose;
1252d7c403e6SMed Ismail Bennani   };
1253d7c403e6SMed Ismail Bennani 
1254d7c403e6SMed Ismail Bennani protected:
1255b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
12567260f620SGreg Clayton     Stream &strm = result.GetOutputStream();
12574b9bea87SJim Ingham     result.SetStatus(eReturnStatusSuccessFinishNoResult);
1258d7c403e6SMed Ismail Bennani 
1259d7c403e6SMed Ismail Bennani     if (command.GetArgumentCount()) {
1260d7c403e6SMed Ismail Bennani       result.AppendError("'process status' takes no arguments");
1261d7c403e6SMed Ismail Bennani       result.SetStatus(eReturnStatusFailed);
1262d7c403e6SMed Ismail Bennani       return result.Succeeded();
1263d7c403e6SMed Ismail Bennani     }
1264d7c403e6SMed Ismail Bennani 
1265b9c1b51eSKate Stone     // No need to check "process" for validity as eCommandRequiresProcess
1266b9c1b51eSKate Stone     // ensures it is valid
1267f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
12687260f620SGreg Clayton     const bool only_threads_with_stop_reason = true;
12697260f620SGreg Clayton     const uint32_t start_frame = 0;
12707260f620SGreg Clayton     const uint32_t num_frames = 1;
12717260f620SGreg Clayton     const uint32_t num_frames_with_source = 1;
12726a9767c7SJim Ingham     const bool stop_format = true;
1273c14ee32dSGreg Clayton     process->GetStatus(strm);
1274b9c1b51eSKate Stone     process->GetThreadStatus(strm, only_threads_with_stop_reason, start_frame,
12756a9767c7SJim Ingham                              num_frames, num_frames_with_source, stop_format);
1276d7c403e6SMed Ismail Bennani 
1277d7c403e6SMed Ismail Bennani     if (m_options.m_verbose) {
1278d7c403e6SMed Ismail Bennani       PlatformSP platform_sp = process->GetTarget().GetPlatform();
1279d7c403e6SMed Ismail Bennani       if (!platform_sp) {
1280d7c403e6SMed Ismail Bennani         result.AppendError("Couldn'retrieve the target's platform");
1281d7c403e6SMed Ismail Bennani         result.SetStatus(eReturnStatusFailed);
12824b9bea87SJim Ingham         return result.Succeeded();
12834b9bea87SJim Ingham       }
1284d7c403e6SMed Ismail Bennani 
1285d7c403e6SMed Ismail Bennani       auto expected_crash_info =
1286*eefda182SMed Ismail Bennani           platform_sp->FetchExtendedCrashInformation(*process);
1287d7c403e6SMed Ismail Bennani 
1288d7c403e6SMed Ismail Bennani       if (!expected_crash_info) {
1289d7c403e6SMed Ismail Bennani         result.AppendError(llvm::toString(expected_crash_info.takeError()));
1290d7c403e6SMed Ismail Bennani         result.SetStatus(eReturnStatusFailed);
1291d7c403e6SMed Ismail Bennani         return result.Succeeded();
1292d7c403e6SMed Ismail Bennani       }
1293d7c403e6SMed Ismail Bennani 
1294d7c403e6SMed Ismail Bennani       StructuredData::DictionarySP crash_info_sp = *expected_crash_info;
1295d7c403e6SMed Ismail Bennani 
1296d7c403e6SMed Ismail Bennani       if (crash_info_sp) {
1297d7c403e6SMed Ismail Bennani         strm.PutCString("Extended Crash Information:\n");
1298d7c403e6SMed Ismail Bennani         crash_info_sp->Dump(strm);
1299d7c403e6SMed Ismail Bennani       }
1300d7c403e6SMed Ismail Bennani     }
1301d7c403e6SMed Ismail Bennani 
1302d7c403e6SMed Ismail Bennani     return result.Succeeded();
1303d7c403e6SMed Ismail Bennani   }
1304d7c403e6SMed Ismail Bennani 
1305d7c403e6SMed Ismail Bennani private:
1306d7c403e6SMed Ismail Bennani   CommandOptions m_options;
13074b9bea87SJim Ingham };
13084b9bea87SJim Ingham 
130935731357SCaroline Tice // CommandObjectProcessHandle
1310438dfcffSRaphael Isemann #define LLDB_OPTIONS_process_handle
1311438dfcffSRaphael Isemann #include "CommandOptions.inc"
13121f0f5b5bSZachary Turner 
1313bb9caf73SJim Ingham #pragma mark CommandObjectProcessHandle
131435731357SCaroline Tice 
1315b9c1b51eSKate Stone class CommandObjectProcessHandle : public CommandObjectParsed {
131635731357SCaroline Tice public:
1317b9c1b51eSKate Stone   class CommandOptions : public Options {
131835731357SCaroline Tice   public:
1319b9c1b51eSKate Stone     CommandOptions() : Options() { OptionParsingStarting(nullptr); }
132035731357SCaroline Tice 
132149bcfd80SEugene Zelenko     ~CommandOptions() override = default;
132235731357SCaroline Tice 
132397206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
1324b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
132597206d57SZachary Turner       Status error;
13263bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
132735731357SCaroline Tice 
1328b9c1b51eSKate Stone       switch (short_option) {
132935731357SCaroline Tice       case 's':
1330adcd0268SBenjamin Kramer         stop = std::string(option_arg);
133135731357SCaroline Tice         break;
133235731357SCaroline Tice       case 'n':
1333adcd0268SBenjamin Kramer         notify = std::string(option_arg);
133435731357SCaroline Tice         break;
133535731357SCaroline Tice       case 'p':
1336adcd0268SBenjamin Kramer         pass = std::string(option_arg);
133735731357SCaroline Tice         break;
133835731357SCaroline Tice       default:
133936162014SRaphael Isemann         llvm_unreachable("Unimplemented option");
134035731357SCaroline Tice       }
134135731357SCaroline Tice       return error;
134235731357SCaroline Tice     }
134335731357SCaroline Tice 
1344b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
134535731357SCaroline Tice       stop.clear();
134635731357SCaroline Tice       notify.clear();
134735731357SCaroline Tice       pass.clear();
134835731357SCaroline Tice     }
134935731357SCaroline Tice 
13501f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
135170602439SZachary Turner       return llvm::makeArrayRef(g_process_handle_options);
13521f0f5b5bSZachary Turner     }
135335731357SCaroline Tice 
135435731357SCaroline Tice     // Instance variables to hold the values for command options.
135535731357SCaroline Tice 
135635731357SCaroline Tice     std::string stop;
135735731357SCaroline Tice     std::string notify;
135835731357SCaroline Tice     std::string pass;
135935731357SCaroline Tice   };
136035731357SCaroline Tice 
13617428a18cSKate Stone   CommandObjectProcessHandle(CommandInterpreter &interpreter)
1362b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process handle",
1363b9c1b51eSKate Stone                             "Manage LLDB handling of OS signals for the "
1364b9c1b51eSKate Stone                             "current target process.  Defaults to showing "
1365b9c1b51eSKate Stone                             "current policy.",
136604a4c091SRaphael Isemann                             nullptr, eCommandRequiresTarget),
1367b9c1b51eSKate Stone         m_options() {
1368ea671fbdSKate Stone     SetHelpLong("\nIf no signals are specified, update them all.  If no update "
1369ea671fbdSKate Stone                 "option is specified, list the current values.");
137035731357SCaroline Tice     CommandArgumentEntry arg;
1371c0dbdfb6SCaroline Tice     CommandArgumentData signal_arg;
137235731357SCaroline Tice 
1373c0dbdfb6SCaroline Tice     signal_arg.arg_type = eArgTypeUnixSignal;
1374c0dbdfb6SCaroline Tice     signal_arg.arg_repetition = eArgRepeatStar;
137535731357SCaroline Tice 
1376c0dbdfb6SCaroline Tice     arg.push_back(signal_arg);
137735731357SCaroline Tice 
137835731357SCaroline Tice     m_arguments.push_back(arg);
137935731357SCaroline Tice   }
138035731357SCaroline Tice 
138149bcfd80SEugene Zelenko   ~CommandObjectProcessHandle() override = default;
138235731357SCaroline Tice 
1383b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
138435731357SCaroline Tice 
1385b9c1b51eSKate Stone   bool VerifyCommandOptionValue(const std::string &option, int &real_value) {
138635731357SCaroline Tice     bool okay = true;
138710ad7993SCaroline Tice     bool success = false;
138847cbf4a0SPavel Labath     bool tmp_value = OptionArgParser::ToBoolean(option, false, &success);
138910ad7993SCaroline Tice 
139010ad7993SCaroline Tice     if (success && tmp_value)
139110ad7993SCaroline Tice       real_value = 1;
139210ad7993SCaroline Tice     else if (success && !tmp_value)
139310ad7993SCaroline Tice       real_value = 0;
1394b9c1b51eSKate Stone     else {
139535731357SCaroline Tice       // If the value isn't 'true' or 'false', it had better be 0 or 1.
13965275aaa0SVince Harron       real_value = StringConvert::ToUInt32(option.c_str(), 3);
139710ad7993SCaroline Tice       if (real_value != 0 && real_value != 1)
139835731357SCaroline Tice         okay = false;
139935731357SCaroline Tice     }
140035731357SCaroline Tice 
140135731357SCaroline Tice     return okay;
140235731357SCaroline Tice   }
140335731357SCaroline Tice 
1404b9c1b51eSKate Stone   void PrintSignalHeader(Stream &str) {
140510ad7993SCaroline Tice     str.Printf("NAME         PASS   STOP   NOTIFY\n");
1406b84141a6SPavel Labath     str.Printf("===========  =====  =====  ======\n");
140710ad7993SCaroline Tice   }
140810ad7993SCaroline Tice 
1409b9c1b51eSKate Stone   void PrintSignal(Stream &str, int32_t signo, const char *sig_name,
1410b9c1b51eSKate Stone                    const UnixSignalsSP &signals_sp) {
141110ad7993SCaroline Tice     bool stop;
141210ad7993SCaroline Tice     bool suppress;
141310ad7993SCaroline Tice     bool notify;
141410ad7993SCaroline Tice 
1415b84141a6SPavel Labath     str.Printf("%-11s  ", sig_name);
1416b9c1b51eSKate Stone     if (signals_sp->GetSignalInfo(signo, suppress, stop, notify)) {
141710ad7993SCaroline Tice       bool pass = !suppress;
1418b9c1b51eSKate Stone       str.Printf("%s  %s  %s", (pass ? "true " : "false"),
1419b9c1b51eSKate Stone                  (stop ? "true " : "false"), (notify ? "true " : "false"));
142010ad7993SCaroline Tice     }
142110ad7993SCaroline Tice     str.Printf("\n");
142210ad7993SCaroline Tice   }
142310ad7993SCaroline Tice 
1424b9c1b51eSKate Stone   void PrintSignalInformation(Stream &str, Args &signal_args,
1425b9c1b51eSKate Stone                               int num_valid_signals,
1426b9c1b51eSKate Stone                               const UnixSignalsSP &signals_sp) {
142710ad7993SCaroline Tice     PrintSignalHeader(str);
142810ad7993SCaroline Tice 
1429b9c1b51eSKate Stone     if (num_valid_signals > 0) {
143010ad7993SCaroline Tice       size_t num_args = signal_args.GetArgumentCount();
1431b9c1b51eSKate Stone       for (size_t i = 0; i < num_args; ++i) {
1432b9c1b51eSKate Stone         int32_t signo = signals_sp->GetSignalNumberFromName(
1433b9c1b51eSKate Stone             signal_args.GetArgumentAtIndex(i));
143410ad7993SCaroline Tice         if (signo != LLDB_INVALID_SIGNAL_NUMBER)
1435b9c1b51eSKate Stone           PrintSignal(str, signo, signal_args.GetArgumentAtIndex(i),
1436b9c1b51eSKate Stone                       signals_sp);
143710ad7993SCaroline Tice       }
1438b9c1b51eSKate Stone     } else // Print info for ALL signals
143910ad7993SCaroline Tice     {
144098d0a4b3SChaoren Lin       int32_t signo = signals_sp->GetFirstSignalNumber();
1441b9c1b51eSKate Stone       while (signo != LLDB_INVALID_SIGNAL_NUMBER) {
1442b9c1b51eSKate Stone         PrintSignal(str, signo, signals_sp->GetSignalAsCString(signo),
1443b9c1b51eSKate Stone                     signals_sp);
144498d0a4b3SChaoren Lin         signo = signals_sp->GetNextSignalNumber(signo);
144510ad7993SCaroline Tice       }
144610ad7993SCaroline Tice     }
144710ad7993SCaroline Tice   }
144810ad7993SCaroline Tice 
14495a988416SJim Ingham protected:
1450b9c1b51eSKate Stone   bool DoExecute(Args &signal_args, CommandReturnObject &result) override {
145104a4c091SRaphael Isemann     Target *target_sp = &GetSelectedTarget();
145235731357SCaroline Tice 
145335731357SCaroline Tice     ProcessSP process_sp = target_sp->GetProcessSP();
145435731357SCaroline Tice 
1455b9c1b51eSKate Stone     if (!process_sp) {
1456b9c1b51eSKate Stone       result.AppendError("No current process; cannot handle signals until you "
1457b9c1b51eSKate Stone                          "have a valid process.\n");
145835731357SCaroline Tice       result.SetStatus(eReturnStatusFailed);
145935731357SCaroline Tice       return false;
146035731357SCaroline Tice     }
146135731357SCaroline Tice 
146235731357SCaroline Tice     int stop_action = -1;   // -1 means leave the current setting alone
146335731357SCaroline Tice     int pass_action = -1;   // -1 means leave the current setting alone
146435731357SCaroline Tice     int notify_action = -1; // -1 means leave the current setting alone
146535731357SCaroline Tice 
1466b9c1b51eSKate Stone     if (!m_options.stop.empty() &&
1467b9c1b51eSKate Stone         !VerifyCommandOptionValue(m_options.stop, stop_action)) {
1468b9c1b51eSKate Stone       result.AppendError("Invalid argument for command option --stop; must be "
1469b9c1b51eSKate Stone                          "true or false.\n");
147035731357SCaroline Tice       result.SetStatus(eReturnStatusFailed);
147135731357SCaroline Tice       return false;
147235731357SCaroline Tice     }
147335731357SCaroline Tice 
1474b9c1b51eSKate Stone     if (!m_options.notify.empty() &&
1475b9c1b51eSKate Stone         !VerifyCommandOptionValue(m_options.notify, notify_action)) {
1476b9c1b51eSKate Stone       result.AppendError("Invalid argument for command option --notify; must "
1477b9c1b51eSKate Stone                          "be true or false.\n");
147835731357SCaroline Tice       result.SetStatus(eReturnStatusFailed);
147935731357SCaroline Tice       return false;
148035731357SCaroline Tice     }
148135731357SCaroline Tice 
1482b9c1b51eSKate Stone     if (!m_options.pass.empty() &&
1483b9c1b51eSKate Stone         !VerifyCommandOptionValue(m_options.pass, pass_action)) {
1484b9c1b51eSKate Stone       result.AppendError("Invalid argument for command option --pass; must be "
1485b9c1b51eSKate Stone                          "true or false.\n");
148635731357SCaroline Tice       result.SetStatus(eReturnStatusFailed);
148735731357SCaroline Tice       return false;
148835731357SCaroline Tice     }
148935731357SCaroline Tice 
149035731357SCaroline Tice     size_t num_args = signal_args.GetArgumentCount();
149198d0a4b3SChaoren Lin     UnixSignalsSP signals_sp = process_sp->GetUnixSignals();
149235731357SCaroline Tice     int num_signals_set = 0;
149335731357SCaroline Tice 
1494b9c1b51eSKate Stone     if (num_args > 0) {
1495d6a24757SZachary Turner       for (const auto &arg : signal_args) {
1496d6a24757SZachary Turner         int32_t signo = signals_sp->GetSignalNumberFromName(arg.c_str());
1497b9c1b51eSKate Stone         if (signo != LLDB_INVALID_SIGNAL_NUMBER) {
1498b9c1b51eSKate Stone           // Casting the actions as bools here should be okay, because
149905097246SAdrian Prantl           // VerifyCommandOptionValue guarantees the value is either 0 or 1.
150035731357SCaroline Tice           if (stop_action != -1)
150198d0a4b3SChaoren Lin             signals_sp->SetShouldStop(signo, stop_action);
1502b9c1b51eSKate Stone           if (pass_action != -1) {
150398d0a4b3SChaoren Lin             bool suppress = !pass_action;
150498d0a4b3SChaoren Lin             signals_sp->SetShouldSuppress(signo, suppress);
150535731357SCaroline Tice           }
150635731357SCaroline Tice           if (notify_action != -1)
150798d0a4b3SChaoren Lin             signals_sp->SetShouldNotify(signo, notify_action);
150835731357SCaroline Tice           ++num_signals_set;
1509b9c1b51eSKate Stone         } else {
1510b9c1b51eSKate Stone           result.AppendErrorWithFormat("Invalid signal name '%s'\n",
1511d6a24757SZachary Turner                                        arg.c_str());
151235731357SCaroline Tice         }
151335731357SCaroline Tice       }
1514b9c1b51eSKate Stone     } else {
1515b9c1b51eSKate Stone       // No signal specified, if any command options were specified, update ALL
1516b9c1b51eSKate Stone       // signals.
1517b9c1b51eSKate Stone       if ((notify_action != -1) || (stop_action != -1) || (pass_action != -1)) {
1518b9c1b51eSKate Stone         if (m_interpreter.Confirm(
1519b9c1b51eSKate Stone                 "Do you really want to update all the signals?", false)) {
152098d0a4b3SChaoren Lin           int32_t signo = signals_sp->GetFirstSignalNumber();
1521b9c1b51eSKate Stone           while (signo != LLDB_INVALID_SIGNAL_NUMBER) {
152210ad7993SCaroline Tice             if (notify_action != -1)
152398d0a4b3SChaoren Lin               signals_sp->SetShouldNotify(signo, notify_action);
152410ad7993SCaroline Tice             if (stop_action != -1)
152598d0a4b3SChaoren Lin               signals_sp->SetShouldStop(signo, stop_action);
1526b9c1b51eSKate Stone             if (pass_action != -1) {
152798d0a4b3SChaoren Lin               bool suppress = !pass_action;
152898d0a4b3SChaoren Lin               signals_sp->SetShouldSuppress(signo, suppress);
152910ad7993SCaroline Tice             }
153098d0a4b3SChaoren Lin             signo = signals_sp->GetNextSignalNumber(signo);
153110ad7993SCaroline Tice           }
153210ad7993SCaroline Tice         }
153310ad7993SCaroline Tice       }
153410ad7993SCaroline Tice     }
153510ad7993SCaroline Tice 
1536b9c1b51eSKate Stone     PrintSignalInformation(result.GetOutputStream(), signal_args,
1537b9c1b51eSKate Stone                            num_signals_set, signals_sp);
153835731357SCaroline Tice 
153935731357SCaroline Tice     if (num_signals_set > 0)
154035731357SCaroline Tice       result.SetStatus(eReturnStatusSuccessFinishNoResult);
154135731357SCaroline Tice     else
154235731357SCaroline Tice       result.SetStatus(eReturnStatusFailed);
154335731357SCaroline Tice 
154435731357SCaroline Tice     return result.Succeeded();
154535731357SCaroline Tice   }
154635731357SCaroline Tice 
154735731357SCaroline Tice   CommandOptions m_options;
154835731357SCaroline Tice };
154935731357SCaroline Tice 
155030fdc8d8SChris Lattner // CommandObjectMultiwordProcess
155130fdc8d8SChris Lattner 
1552b9c1b51eSKate Stone CommandObjectMultiwordProcess::CommandObjectMultiwordProcess(
1553b9c1b51eSKate Stone     CommandInterpreter &interpreter)
1554b9c1b51eSKate Stone     : CommandObjectMultiword(
1555b9c1b51eSKate Stone           interpreter, "process",
1556b9c1b51eSKate Stone           "Commands for interacting with processes on the current platform.",
1557b9c1b51eSKate Stone           "process <subcommand> [<subcommand-options>]") {
1558b9c1b51eSKate Stone   LoadSubCommand("attach",
1559b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessAttach(interpreter)));
1560b9c1b51eSKate Stone   LoadSubCommand("launch",
1561b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessLaunch(interpreter)));
1562b9c1b51eSKate Stone   LoadSubCommand("continue", CommandObjectSP(new CommandObjectProcessContinue(
1563b9c1b51eSKate Stone                                  interpreter)));
1564b9c1b51eSKate Stone   LoadSubCommand("connect",
1565b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessConnect(interpreter)));
1566b9c1b51eSKate Stone   LoadSubCommand("detach",
1567b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessDetach(interpreter)));
1568b9c1b51eSKate Stone   LoadSubCommand("load",
1569b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessLoad(interpreter)));
1570b9c1b51eSKate Stone   LoadSubCommand("unload",
1571b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessUnload(interpreter)));
1572b9c1b51eSKate Stone   LoadSubCommand("signal",
1573b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessSignal(interpreter)));
1574b9c1b51eSKate Stone   LoadSubCommand("handle",
1575b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessHandle(interpreter)));
1576b9c1b51eSKate Stone   LoadSubCommand("status",
1577b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessStatus(interpreter)));
1578b9c1b51eSKate Stone   LoadSubCommand("interrupt", CommandObjectSP(new CommandObjectProcessInterrupt(
1579b9c1b51eSKate Stone                                   interpreter)));
1580b9c1b51eSKate Stone   LoadSubCommand("kill",
1581b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessKill(interpreter)));
1582b9c1b51eSKate Stone   LoadSubCommand("plugin",
1583b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessPlugin(interpreter)));
1584b9c1b51eSKate Stone   LoadSubCommand("save-core", CommandObjectSP(new CommandObjectProcessSaveCore(
1585b9c1b51eSKate Stone                                   interpreter)));
158630fdc8d8SChris Lattner }
158730fdc8d8SChris Lattner 
158849bcfd80SEugene Zelenko CommandObjectMultiwordProcess::~CommandObjectMultiwordProcess() = default;
1589