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"
1630fdc8d8SChris Lattner #include "lldb/Interpreter/CommandInterpreter.h"
1730fdc8d8SChris Lattner #include "lldb/Interpreter/CommandReturnObject.h"
1847cbf4a0SPavel Labath #include "lldb/Interpreter/OptionArgParser.h"
19b9c1b51eSKate Stone #include "lldb/Interpreter/Options.h"
20e996fd30SGreg Clayton #include "lldb/Target/Platform.h"
2130fdc8d8SChris Lattner #include "lldb/Target/Process.h"
220e41084aSJim Ingham #include "lldb/Target/StopInfo.h"
2330fdc8d8SChris Lattner #include "lldb/Target/Target.h"
2430fdc8d8SChris Lattner #include "lldb/Target/Thread.h"
2593749ab3SZachary Turner #include "lldb/Target/UnixSignals.h"
26145d95c9SPavel Labath #include "lldb/Utility/Args.h"
27d821c997SPavel Labath #include "lldb/Utility/State.h"
2830fdc8d8SChris Lattner 
2930fdc8d8SChris Lattner using namespace lldb;
3030fdc8d8SChris Lattner using namespace lldb_private;
3130fdc8d8SChris Lattner 
32b9c1b51eSKate Stone class CommandObjectProcessLaunchOrAttach : public CommandObjectParsed {
33dcb1d856SJim Ingham public:
34dcb1d856SJim Ingham   CommandObjectProcessLaunchOrAttach(CommandInterpreter &interpreter,
35b9c1b51eSKate Stone                                      const char *name, const char *help,
36b9c1b51eSKate Stone                                      const char *syntax, uint32_t flags,
37b9c1b51eSKate Stone                                      const char *new_process_action)
38b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, name, help, syntax, flags),
39dcb1d856SJim Ingham         m_new_process_action(new_process_action) {}
40dcb1d856SJim Ingham 
4149bcfd80SEugene Zelenko   ~CommandObjectProcessLaunchOrAttach() override = default;
4249bcfd80SEugene Zelenko 
43dcb1d856SJim Ingham protected:
44b9c1b51eSKate Stone   bool StopProcessIfNecessary(Process *process, StateType &state,
45b9c1b51eSKate Stone                               CommandReturnObject &result) {
46dcb1d856SJim Ingham     state = eStateInvalid;
47b9c1b51eSKate Stone     if (process) {
48dcb1d856SJim Ingham       state = process->GetState();
49dcb1d856SJim Ingham 
50b9c1b51eSKate Stone       if (process->IsAlive() && state != eStateConnected) {
51db203e02SRaphael Isemann         std::string message;
52dcb1d856SJim Ingham         if (process->GetState() == eStateAttaching)
53db203e02SRaphael Isemann           message =
54db203e02SRaphael Isemann               llvm::formatv("There is a pending attach, abort it and {0}?",
55db203e02SRaphael Isemann                             m_new_process_action);
56dcb1d856SJim Ingham         else if (process->GetShouldDetach())
57db203e02SRaphael Isemann           message = llvm::formatv(
58db203e02SRaphael Isemann               "There is a running process, detach from it and {0}?",
59db203e02SRaphael Isemann               m_new_process_action);
60dcb1d856SJim Ingham         else
61db203e02SRaphael Isemann           message =
62db203e02SRaphael Isemann               llvm::formatv("There is a running process, kill it and {0}?",
63db203e02SRaphael Isemann                             m_new_process_action);
64dcb1d856SJim Ingham 
65b9c1b51eSKate Stone         if (!m_interpreter.Confirm(message, true)) {
66dcb1d856SJim Ingham           result.SetStatus(eReturnStatusFailed);
67dcb1d856SJim Ingham           return false;
68b9c1b51eSKate Stone         } else {
69b9c1b51eSKate Stone           if (process->GetShouldDetach()) {
70acff8950SJim Ingham             bool keep_stopped = false;
7197206d57SZachary Turner             Status detach_error(process->Detach(keep_stopped));
72b9c1b51eSKate Stone             if (detach_error.Success()) {
73dcb1d856SJim Ingham               result.SetStatus(eReturnStatusSuccessFinishResult);
7449bcfd80SEugene Zelenko               process = nullptr;
75b9c1b51eSKate Stone             } else {
76b9c1b51eSKate Stone               result.AppendErrorWithFormat(
77b9c1b51eSKate Stone                   "Failed to detach from process: %s\n",
78b9c1b51eSKate Stone                   detach_error.AsCString());
79dcb1d856SJim Ingham               result.SetStatus(eReturnStatusFailed);
80dcb1d856SJim Ingham             }
81b9c1b51eSKate Stone           } else {
8297206d57SZachary Turner             Status destroy_error(process->Destroy(false));
83b9c1b51eSKate Stone             if (destroy_error.Success()) {
84dcb1d856SJim Ingham               result.SetStatus(eReturnStatusSuccessFinishResult);
8549bcfd80SEugene Zelenko               process = nullptr;
86b9c1b51eSKate Stone             } else {
87b9c1b51eSKate Stone               result.AppendErrorWithFormat("Failed to kill process: %s\n",
88b9c1b51eSKate Stone                                            destroy_error.AsCString());
89dcb1d856SJim Ingham               result.SetStatus(eReturnStatusFailed);
90dcb1d856SJim Ingham             }
91dcb1d856SJim Ingham           }
92dcb1d856SJim Ingham         }
93dcb1d856SJim Ingham       }
94dcb1d856SJim Ingham     }
95dcb1d856SJim Ingham     return result.Succeeded();
96dcb1d856SJim Ingham   }
9749bcfd80SEugene Zelenko 
98dcb1d856SJim Ingham   std::string m_new_process_action;
99dcb1d856SJim Ingham };
10049bcfd80SEugene Zelenko 
10130fdc8d8SChris Lattner // CommandObjectProcessLaunch
1024bddaeb5SJim Ingham #pragma mark CommandObjectProcessLaunch
103b9c1b51eSKate Stone class CommandObjectProcessLaunch : public CommandObjectProcessLaunchOrAttach {
10430fdc8d8SChris Lattner public:
105b9c1b51eSKate Stone   CommandObjectProcessLaunch(CommandInterpreter &interpreter)
106b9c1b51eSKate Stone       : CommandObjectProcessLaunchOrAttach(
107b9c1b51eSKate Stone             interpreter, "process launch",
108b9c1b51eSKate Stone             "Launch the executable in the debugger.", nullptr,
109b9c1b51eSKate Stone             eCommandRequiresTarget, "restart"),
110b9c1b51eSKate Stone         m_options() {
111405fe67fSCaroline Tice     CommandArgumentEntry arg;
112405fe67fSCaroline Tice     CommandArgumentData run_args_arg;
113405fe67fSCaroline Tice 
114405fe67fSCaroline Tice     // Define the first (and only) variant of this arg.
115405fe67fSCaroline Tice     run_args_arg.arg_type = eArgTypeRunArgs;
116405fe67fSCaroline Tice     run_args_arg.arg_repetition = eArgRepeatOptional;
117405fe67fSCaroline Tice 
118b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
119b9c1b51eSKate Stone     // argument entry.
120405fe67fSCaroline Tice     arg.push_back(run_args_arg);
121405fe67fSCaroline Tice 
122405fe67fSCaroline Tice     // Push the data for the first argument into the m_arguments vector.
123405fe67fSCaroline Tice     m_arguments.push_back(arg);
12430fdc8d8SChris Lattner   }
12530fdc8d8SChris Lattner 
12649bcfd80SEugene Zelenko   ~CommandObjectProcessLaunch() override = default;
12730fdc8d8SChris Lattner 
128ae34ed2cSRaphael Isemann   void
129ae34ed2cSRaphael Isemann   HandleArgumentCompletion(CompletionRequest &request,
1302443bbd4SRaphael Isemann                            OptionElementVector &opt_element_vector) override {
131e9ce62b6SJim Ingham 
132b9c1b51eSKate Stone     CommandCompletions::InvokeCommonCompletionCallbacks(
133b9c1b51eSKate Stone         GetCommandInterpreter(), CommandCompletions::eDiskFileCompletion,
134a2e76c0bSRaphael Isemann         request, nullptr);
135e9ce62b6SJim Ingham   }
136e9ce62b6SJim Ingham 
137b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
13830fdc8d8SChris Lattner 
139b9c1b51eSKate Stone   const char *GetRepeatCommand(Args &current_command_args,
140b9c1b51eSKate Stone                                uint32_t index) override {
1415a988416SJim Ingham     // No repeat for "process launch"...
1425a988416SJim Ingham     return "";
1435a988416SJim Ingham   }
1445a988416SJim Ingham 
1455a988416SJim Ingham protected:
146b9c1b51eSKate Stone   bool DoExecute(Args &launch_args, CommandReturnObject &result) override {
14757179860SJonas Devlieghere     Debugger &debugger = GetDebugger();
1481d885966SGreg Clayton     Target *target = debugger.GetSelectedTarget().get();
14949bcfd80SEugene Zelenko     // If our listener is nullptr, users aren't allows to launch
150b09c5384SGreg Clayton     ModuleSP exe_module_sp = target->GetExecutableModule();
15171337622SGreg Clayton 
152b9c1b51eSKate Stone     if (exe_module_sp == nullptr) {
153b9c1b51eSKate Stone       result.AppendError("no file in target, create a debug target using the "
154b9c1b51eSKate Stone                          "'target create' command");
15571337622SGreg Clayton       result.SetStatus(eReturnStatusFailed);
15671337622SGreg Clayton       return false;
15771337622SGreg Clayton     }
15871337622SGreg Clayton 
15971337622SGreg Clayton     StateType state = eStateInvalid;
16071337622SGreg Clayton 
161b09c5384SGreg Clayton     if (!StopProcessIfNecessary(m_exe_ctx.GetProcessPtr(), state, result))
16230fdc8d8SChris Lattner       return false;
16330fdc8d8SChris Lattner 
16431d97a5cSZachary Turner     llvm::StringRef target_settings_argv0 = target->GetArg0();
16545392553SGreg Clayton 
166b9c1b51eSKate Stone     // Determine whether we will disable ASLR or leave it in the default state
16705097246SAdrian Prantl     // (i.e. enabled if the platform supports it). First check if the process
16805097246SAdrian Prantl     // launch options explicitly turn on/off
169b9c1b51eSKate Stone     // disabling ASLR.  If so, use that setting;
1705163792bSTodd Fiala     // otherwise, use the 'settings target.disable-aslr' setting.
1715163792bSTodd Fiala     bool disable_aslr = false;
172b9c1b51eSKate Stone     if (m_options.disable_aslr != eLazyBoolCalculate) {
17305097246SAdrian Prantl       // The user specified an explicit setting on the process launch line.
17405097246SAdrian Prantl       // Use it.
1755163792bSTodd Fiala       disable_aslr = (m_options.disable_aslr == eLazyBoolYes);
176b9c1b51eSKate Stone     } else {
17705097246SAdrian Prantl       // The user did not explicitly specify whether to disable ASLR.  Fall
17805097246SAdrian Prantl       // back to the target.disable-aslr setting.
1795163792bSTodd Fiala       disable_aslr = target->GetDisableASLR();
1805163792bSTodd Fiala     }
1815163792bSTodd Fiala 
1825163792bSTodd Fiala     if (disable_aslr)
183b09c5384SGreg Clayton       m_options.launch_info.GetFlags().Set(eLaunchFlagDisableASLR);
1845163792bSTodd Fiala     else
1855163792bSTodd Fiala       m_options.launch_info.GetFlags().Clear(eLaunchFlagDisableASLR);
186b09c5384SGreg Clayton 
187249a1d4fSJonas Devlieghere     if (target->GetInheritTCC())
188249a1d4fSJonas Devlieghere       m_options.launch_info.GetFlags().Set(eLaunchFlagInheritTCCFromParent);
189249a1d4fSJonas Devlieghere 
190106d0286SJim Ingham     if (target->GetDetachOnError())
191106d0286SJim Ingham       m_options.launch_info.GetFlags().Set(eLaunchFlagDetachOnError);
192106d0286SJim Ingham 
193b09c5384SGreg Clayton     if (target->GetDisableSTDIO())
194b09c5384SGreg Clayton       m_options.launch_info.GetFlags().Set(eLaunchFlagDisableSTDIO);
195b09c5384SGreg Clayton 
196ae54fc9fSJonas Devlieghere     // Merge the launch info environment with the target environment.
197ae54fc9fSJonas Devlieghere     Environment target_env = target->GetEnvironment();
198ae54fc9fSJonas Devlieghere     m_options.launch_info.GetEnvironment().insert(target_env.begin(),
199ae54fc9fSJonas Devlieghere                                                   target_env.end());
20045392553SGreg Clayton 
20131d97a5cSZachary Turner     if (!target_settings_argv0.empty()) {
202b9c1b51eSKate Stone       m_options.launch_info.GetArguments().AppendArgument(
20331d97a5cSZachary Turner           target_settings_argv0);
204b9c1b51eSKate Stone       m_options.launch_info.SetExecutableFile(
205b9c1b51eSKate Stone           exe_module_sp->GetPlatformFileSpec(), false);
206b9c1b51eSKate Stone     } else {
207b9c1b51eSKate Stone       m_options.launch_info.SetExecutableFile(
208b9c1b51eSKate Stone           exe_module_sp->GetPlatformFileSpec(), true);
20945392553SGreg Clayton     }
21045392553SGreg Clayton 
211b9c1b51eSKate Stone     if (launch_args.GetArgumentCount() == 0) {
212b9c1b51eSKate Stone       m_options.launch_info.GetArguments().AppendArguments(
213b9c1b51eSKate Stone           target->GetProcessLaunchInfo().GetArguments());
214b9c1b51eSKate Stone     } else {
21545392553SGreg Clayton       m_options.launch_info.GetArguments().AppendArguments(launch_args);
216162b597cSGreg Clayton       // Save the arguments for subsequent runs in the current target.
217162b597cSGreg Clayton       target->SetRunArguments(launch_args);
218982c9762SGreg Clayton     }
2191d885966SGreg Clayton 
220dc6224e0SGreg Clayton     StreamString stream;
22197206d57SZachary Turner     Status error = target->Launch(m_options.launch_info, &stream);
22230fdc8d8SChris Lattner 
223b9c1b51eSKate Stone     if (error.Success()) {
224b09c5384SGreg Clayton       ProcessSP process_sp(target->GetProcessSP());
225b9c1b51eSKate Stone       if (process_sp) {
226b9c1b51eSKate Stone         // There is a race condition where this thread will return up the call
22705097246SAdrian Prantl         // stack to the main command handler and show an (lldb) prompt before
22805097246SAdrian Prantl         // HandlePrivateEvent (from PrivateStateThread) has a chance to call
22905097246SAdrian Prantl         // PushProcessIOHandler().
2303879fe00SPavel Labath         process_sp->SyncIOHandler(0, std::chrono::seconds(2));
2318f0db3e1SIlia K 
232c156427dSZachary Turner         llvm::StringRef data = stream.GetString();
233c156427dSZachary Turner         if (!data.empty())
234c156427dSZachary Turner           result.AppendMessage(data);
235b9c1b51eSKate Stone         const char *archname =
236b9c1b51eSKate Stone             exe_module_sp->GetArchitecture().GetArchitectureName();
237b9c1b51eSKate Stone         result.AppendMessageWithFormat(
238b9c1b51eSKate Stone             "Process %" PRIu64 " launched: '%s' (%s)\n", process_sp->GetID(),
239b9c1b51eSKate Stone             exe_module_sp->GetFileSpec().GetPath().c_str(), archname);
24005faeb71SGreg Clayton         result.SetStatus(eReturnStatusSuccessFinishResult);
241b09c5384SGreg Clayton         result.SetDidChangeProcessState(true);
242b9c1b51eSKate Stone       } else {
243b9c1b51eSKate Stone         result.AppendError(
244b9c1b51eSKate Stone             "no error returned from Target::Launch, and target has no process");
245514487e8SGreg Clayton         result.SetStatus(eReturnStatusFailed);
24630fdc8d8SChris Lattner       }
247b9c1b51eSKate Stone     } else {
248b09c5384SGreg Clayton       result.AppendError(error.AsCString());
249514487e8SGreg Clayton       result.SetStatus(eReturnStatusFailed);
250514487e8SGreg Clayton     }
25130fdc8d8SChris Lattner     return result.Succeeded();
25230fdc8d8SChris Lattner   }
25330fdc8d8SChris Lattner 
254982c9762SGreg Clayton   ProcessLaunchCommandOptions m_options;
25530fdc8d8SChris Lattner };
25630fdc8d8SChris Lattner 
257438dfcffSRaphael Isemann #define LLDB_OPTIONS_process_attach
258438dfcffSRaphael Isemann #include "CommandOptions.inc"
2591f0f5b5bSZachary Turner 
260bb9caf73SJim Ingham #pragma mark CommandObjectProcessAttach
261b9c1b51eSKate Stone class CommandObjectProcessAttach : public CommandObjectProcessLaunchOrAttach {
26230fdc8d8SChris Lattner public:
263b9c1b51eSKate Stone   class CommandOptions : public Options {
26430fdc8d8SChris Lattner   public:
265b9c1b51eSKate Stone     CommandOptions() : Options() {
266b9c1b51eSKate Stone       // Keep default values of all options in one place: OptionParsingStarting
267b9c1b51eSKate Stone       // ()
268e1cfbc79STodd Fiala       OptionParsingStarting(nullptr);
26930fdc8d8SChris Lattner     }
27030fdc8d8SChris Lattner 
27149bcfd80SEugene Zelenko     ~CommandOptions() override = default;
27230fdc8d8SChris Lattner 
27397206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
274b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
27597206d57SZachary Turner       Status error;
2763bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
277b9c1b51eSKate Stone       switch (short_option) {
278a95ce623SJohnny Chen       case 'c':
279a95ce623SJohnny Chen         attach_info.SetContinueOnceAttached(true);
280a95ce623SJohnny Chen         break;
281a95ce623SJohnny Chen 
282b9c1b51eSKate Stone       case 'p': {
283fe11483bSZachary Turner         lldb::pid_t pid;
284fe11483bSZachary Turner         if (option_arg.getAsInteger(0, pid)) {
285fe11483bSZachary Turner           error.SetErrorStringWithFormat("invalid process ID '%s'",
286fe11483bSZachary Turner                                          option_arg.str().c_str());
287b9c1b51eSKate Stone         } else {
288144f3a9cSGreg Clayton           attach_info.SetProcessID(pid);
289144f3a9cSGreg Clayton         }
290b9c1b51eSKate Stone       } break;
29130fdc8d8SChris Lattner 
29230fdc8d8SChris Lattner       case 'P':
293144f3a9cSGreg Clayton         attach_info.SetProcessPluginName(option_arg);
29430fdc8d8SChris Lattner         break;
29530fdc8d8SChris Lattner 
29630fdc8d8SChris Lattner       case 'n':
2978f3be7a3SJonas Devlieghere         attach_info.GetExecutableFile().SetFile(option_arg,
298937348cdSJonas Devlieghere                                                 FileSpec::Style::native);
29930fdc8d8SChris Lattner         break;
30030fdc8d8SChris Lattner 
30130fdc8d8SChris Lattner       case 'w':
302144f3a9cSGreg Clayton         attach_info.SetWaitForLaunch(true);
30330fdc8d8SChris Lattner         break;
30430fdc8d8SChris Lattner 
305cd16df91SJim Ingham       case 'i':
306cd16df91SJim Ingham         attach_info.SetIgnoreExisting(false);
307cd16df91SJim Ingham         break;
308cd16df91SJim Ingham 
30930fdc8d8SChris Lattner       default:
31036162014SRaphael Isemann         llvm_unreachable("Unimplemented option");
31130fdc8d8SChris Lattner       }
31230fdc8d8SChris Lattner       return error;
31330fdc8d8SChris Lattner     }
31430fdc8d8SChris Lattner 
315b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
316144f3a9cSGreg Clayton       attach_info.Clear();
31730fdc8d8SChris Lattner     }
31830fdc8d8SChris Lattner 
3191f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
32070602439SZachary Turner       return llvm::makeArrayRef(g_process_attach_options);
3211f0f5b5bSZachary Turner     }
32230fdc8d8SChris Lattner 
323494370c1SRaphael Isemann     void HandleOptionArgumentCompletion(
324a2e76c0bSRaphael Isemann         CompletionRequest &request, OptionElementVector &opt_element_vector,
325a2e76c0bSRaphael Isemann         int opt_element_index, CommandInterpreter &interpreter) override {
3265aee162fSJim Ingham       int opt_arg_pos = opt_element_vector[opt_element_index].opt_arg_pos;
3275aee162fSJim Ingham       int opt_defs_index = opt_element_vector[opt_element_index].opt_defs_index;
3285aee162fSJim Ingham 
329763bc230SGongyu Deng       switch (GetDefinitions()[opt_defs_index].short_option) {
330763bc230SGongyu Deng       case 'n': {
331b9c1b51eSKate Stone         // Look to see if there is a -P argument provided, and if so use that
33205097246SAdrian Prantl         // plugin, otherwise use the default plugin.
3335aee162fSJim Ingham 
33449bcfd80SEugene Zelenko         const char *partial_name = nullptr;
335a2e76c0bSRaphael Isemann         partial_name = request.GetParsedLine().GetArgumentAtIndex(opt_arg_pos);
3365aee162fSJim Ingham 
337e1cfbc79STodd Fiala         PlatformSP platform_sp(interpreter.GetPlatform(true));
3381153dc96SRaphael Isemann         if (!platform_sp)
339494370c1SRaphael Isemann           return;
3408b82f087SGreg Clayton         ProcessInstanceInfoList process_infos;
3418b82f087SGreg Clayton         ProcessInstanceInfoMatch match_info;
342b9c1b51eSKate Stone         if (partial_name) {
343b9c1b51eSKate Stone           match_info.GetProcessInfo().GetExecutableFile().SetFile(
3448f3be7a3SJonas Devlieghere               partial_name, FileSpec::Style::native);
345c4a33951SPavel Labath           match_info.SetNameMatchType(NameMatch::StartsWith);
34632e0a750SGreg Clayton         }
34732e0a750SGreg Clayton         platform_sp->FindProcesses(match_info, process_infos);
348638b06cfSJonas Devlieghere         const size_t num_matches = process_infos.size();
3491153dc96SRaphael Isemann         if (num_matches == 0)
350494370c1SRaphael Isemann           return;
351b9c1b51eSKate Stone         for (size_t i = 0; i < num_matches; ++i) {
352638b06cfSJonas Devlieghere           request.AddCompletion(process_infos[i].GetNameAsStringRef());
3535aee162fSJim Ingham         }
354763bc230SGongyu Deng       } break;
355763bc230SGongyu Deng 
356763bc230SGongyu Deng       case 'P':
357763bc230SGongyu Deng         CommandCompletions::InvokeCommonCompletionCallbacks(
358763bc230SGongyu Deng             interpreter, CommandCompletions::eProcessPluginCompletion, request,
359763bc230SGongyu Deng             nullptr);
360763bc230SGongyu Deng         break;
361763bc230SGongyu Deng       }
3625aee162fSJim Ingham     }
3635aee162fSJim Ingham 
36430fdc8d8SChris Lattner     // Instance variables to hold the values for command options.
36530fdc8d8SChris Lattner 
366144f3a9cSGreg Clayton     ProcessAttachInfo attach_info;
36730fdc8d8SChris Lattner   };
36830fdc8d8SChris Lattner 
369b9c1b51eSKate Stone   CommandObjectProcessAttach(CommandInterpreter &interpreter)
370b9c1b51eSKate Stone       : CommandObjectProcessLaunchOrAttach(
371b9c1b51eSKate Stone             interpreter, "process attach", "Attach to a process.",
372b9c1b51eSKate Stone             "process attach <cmd-options>", 0, "attach"),
373b9c1b51eSKate Stone         m_options() {}
3745aee162fSJim Ingham 
37549bcfd80SEugene Zelenko   ~CommandObjectProcessAttach() override = default;
3765aee162fSJim Ingham 
377b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
3785a988416SJim Ingham 
3795a988416SJim Ingham protected:
380b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
381b9c1b51eSKate Stone     PlatformSP platform_sp(
38257179860SJonas Devlieghere         GetDebugger().GetPlatformList().GetSelectedPlatform());
383926af0cdSOleksiy Vyalov 
38457179860SJonas Devlieghere     Target *target = GetDebugger().GetSelectedTarget().get();
385b9c1b51eSKate Stone     // N.B. The attach should be synchronous.  It doesn't help much to get the
38605097246SAdrian Prantl     // prompt back between initiating the attach and the target actually
38705097246SAdrian Prantl     // stopping.  So even if the interpreter is set to be asynchronous, we wait
38805097246SAdrian Prantl     // for the stop ourselves here.
3895aee162fSJim Ingham 
39071337622SGreg Clayton     StateType state = eStateInvalid;
391dcb1d856SJim Ingham     Process *process = m_exe_ctx.GetProcessPtr();
392dcb1d856SJim Ingham 
393dcb1d856SJim Ingham     if (!StopProcessIfNecessary(process, state, result))
3945aee162fSJim Ingham       return false;
3955aee162fSJim Ingham 
396b9c1b51eSKate Stone     if (target == nullptr) {
3975aee162fSJim Ingham       // If there isn't a current target create one.
3985aee162fSJim Ingham       TargetSP new_target_sp;
39997206d57SZachary Turner       Status error;
4005aee162fSJim Ingham 
40157179860SJonas Devlieghere       error = GetDebugger().GetTargetList().CreateTarget(
40257179860SJonas Devlieghere           GetDebugger(), "", "", eLoadDependentsNo,
40349bcfd80SEugene Zelenko           nullptr, // No platform options
4045aee162fSJim Ingham           new_target_sp);
4055aee162fSJim Ingham       target = new_target_sp.get();
406b9c1b51eSKate Stone       if (target == nullptr || error.Fail()) {
407b766a73dSGreg Clayton         result.AppendError(error.AsCString("Error creating target"));
4085aee162fSJim Ingham         return false;
4095aee162fSJim Ingham       }
41057179860SJonas Devlieghere       GetDebugger().GetTargetList().SetSelectedTarget(target);
4115aee162fSJim Ingham     }
4125aee162fSJim Ingham 
413b9c1b51eSKate Stone     // Record the old executable module, we want to issue a warning if the
41405097246SAdrian Prantl     // process of attaching changed the current executable (like somebody said
41505097246SAdrian Prantl     // "file foo" then attached to a PID whose executable was bar.)
4165aee162fSJim Ingham 
4175aee162fSJim Ingham     ModuleSP old_exec_module_sp = target->GetExecutableModule();
4185aee162fSJim Ingham     ArchSpec old_arch_spec = target->GetArchitecture();
4195aee162fSJim Ingham 
420b9c1b51eSKate Stone     if (command.GetArgumentCount()) {
421b9c1b51eSKate Stone       result.AppendErrorWithFormat("Invalid arguments for '%s'.\nUsage: %s\n",
422b9c1b51eSKate Stone                                    m_cmd_name.c_str(), m_cmd_syntax.c_str());
4235aee162fSJim Ingham       result.SetStatus(eReturnStatusFailed);
424926af0cdSOleksiy Vyalov       return false;
42571337622SGreg Clayton     }
4265aee162fSJim Ingham 
427926af0cdSOleksiy Vyalov     m_interpreter.UpdateExecutionContext(nullptr);
42837386143SOleksiy Vyalov     StreamString stream;
42937386143SOleksiy Vyalov     const auto error = target->Attach(m_options.attach_info, &stream);
430b9c1b51eSKate Stone     if (error.Success()) {
43137386143SOleksiy Vyalov       ProcessSP process_sp(target->GetProcessSP());
432b9c1b51eSKate Stone       if (process_sp) {
433c156427dSZachary Turner         result.AppendMessage(stream.GetString());
434bb3a283bSJim Ingham         result.SetStatus(eReturnStatusSuccessFinishNoResult);
43537386143SOleksiy Vyalov         result.SetDidChangeProcessState(true);
436b9c1b51eSKate Stone       } else {
437b9c1b51eSKate Stone         result.AppendError(
438b9c1b51eSKate Stone             "no error returned from Target::Attach, and target has no process");
439aa739093SJohnny Chen         result.SetStatus(eReturnStatusFailed);
44044d93782SGreg Clayton       }
441b9c1b51eSKate Stone     } else {
44244d93782SGreg Clayton       result.AppendErrorWithFormat("attach failed: %s\n", error.AsCString());
44344d93782SGreg Clayton       result.SetStatus(eReturnStatusFailed);
444aa739093SJohnny Chen     }
4455aee162fSJim Ingham 
446926af0cdSOleksiy Vyalov     if (!result.Succeeded())
447926af0cdSOleksiy Vyalov       return false;
448926af0cdSOleksiy Vyalov 
449b9c1b51eSKate Stone     // Okay, we're done.  Last step is to warn if the executable module has
450b9c1b51eSKate Stone     // changed:
451513c26ceSGreg Clayton     char new_path[PATH_MAX];
452aa149cbdSGreg Clayton     ModuleSP new_exec_module_sp(target->GetExecutableModule());
453b9c1b51eSKate Stone     if (!old_exec_module_sp) {
454513c26ceSGreg Clayton       // We might not have a module if we attached to a raw pid...
455b9c1b51eSKate Stone       if (new_exec_module_sp) {
456aa149cbdSGreg Clayton         new_exec_module_sp->GetFileSpec().GetPath(new_path, PATH_MAX);
457b9c1b51eSKate Stone         result.AppendMessageWithFormat("Executable module set to \"%s\".\n",
458b9c1b51eSKate Stone                                        new_path);
459513c26ceSGreg Clayton       }
460b9c1b51eSKate Stone     } else if (old_exec_module_sp->GetFileSpec() !=
461b9c1b51eSKate Stone                new_exec_module_sp->GetFileSpec()) {
462513c26ceSGreg Clayton       char old_path[PATH_MAX];
4635aee162fSJim Ingham 
4645aee162fSJim Ingham       old_exec_module_sp->GetFileSpec().GetPath(old_path, PATH_MAX);
465aa149cbdSGreg Clayton       new_exec_module_sp->GetFileSpec().GetPath(new_path, PATH_MAX);
4665aee162fSJim Ingham 
467b9c1b51eSKate Stone       result.AppendWarningWithFormat(
468b9c1b51eSKate Stone           "Executable module changed from \"%s\" to \"%s\".\n", old_path,
469b9c1b51eSKate Stone           new_path);
4705aee162fSJim Ingham     }
4715aee162fSJim Ingham 
472b9c1b51eSKate Stone     if (!old_arch_spec.IsValid()) {
473b9c1b51eSKate Stone       result.AppendMessageWithFormat(
474b9c1b51eSKate Stone           "Architecture set to: %s.\n",
475b9c1b51eSKate Stone           target->GetArchitecture().GetTriple().getTriple().c_str());
476b9c1b51eSKate Stone     } else if (!old_arch_spec.IsExactMatch(target->GetArchitecture())) {
477b9c1b51eSKate Stone       result.AppendWarningWithFormat(
478b9c1b51eSKate Stone           "Architecture changed from %s to %s.\n",
479c1b1f1eaSGreg Clayton           old_arch_spec.GetTriple().getTriple().c_str(),
480c1b1f1eaSGreg Clayton           target->GetArchitecture().GetTriple().getTriple().c_str());
4815aee162fSJim Ingham     }
482a95ce623SJohnny Chen 
48305097246SAdrian Prantl     // This supports the use-case scenario of immediately continuing the
48405097246SAdrian Prantl     // process once attached.
485a95ce623SJohnny Chen     if (m_options.attach_info.GetContinueOnceAttached())
4865bcaf583SSean Callanan       m_interpreter.HandleCommand("process continue", eLazyBoolNo, result);
487926af0cdSOleksiy Vyalov 
4885aee162fSJim Ingham     return result.Succeeded();
4895aee162fSJim Ingham   }
4905aee162fSJim Ingham 
49130fdc8d8SChris Lattner   CommandOptions m_options;
49230fdc8d8SChris Lattner };
49330fdc8d8SChris Lattner 
49430fdc8d8SChris Lattner // CommandObjectProcessContinue
4951f0f5b5bSZachary Turner 
496438dfcffSRaphael Isemann #define LLDB_OPTIONS_process_continue
497438dfcffSRaphael Isemann #include "CommandOptions.inc"
4981f0f5b5bSZachary Turner 
499bb9caf73SJim Ingham #pragma mark CommandObjectProcessContinue
50030fdc8d8SChris Lattner 
501b9c1b51eSKate Stone class CommandObjectProcessContinue : public CommandObjectParsed {
50230fdc8d8SChris Lattner public:
503b9c1b51eSKate Stone   CommandObjectProcessContinue(CommandInterpreter &interpreter)
504b9c1b51eSKate Stone       : CommandObjectParsed(
505b9c1b51eSKate Stone             interpreter, "process continue",
506e3d26315SCaroline Tice             "Continue execution of all threads in the current process.",
50730fdc8d8SChris Lattner             "process continue",
508b9c1b51eSKate Stone             eCommandRequiresProcess | eCommandTryTargetAPILock |
509b9c1b51eSKate Stone                 eCommandProcessMustBeLaunched | eCommandProcessMustBePaused),
510b9c1b51eSKate Stone         m_options() {}
51130fdc8d8SChris Lattner 
51249bcfd80SEugene Zelenko   ~CommandObjectProcessContinue() override = default;
51330fdc8d8SChris Lattner 
5145a988416SJim Ingham protected:
515b9c1b51eSKate Stone   class CommandOptions : public Options {
5160e41084aSJim Ingham   public:
517b9c1b51eSKate Stone     CommandOptions() : Options() {
518b9c1b51eSKate Stone       // Keep default values of all options in one place: OptionParsingStarting
519b9c1b51eSKate Stone       // ()
520e1cfbc79STodd Fiala       OptionParsingStarting(nullptr);
5210e41084aSJim Ingham     }
5220e41084aSJim Ingham 
52349bcfd80SEugene Zelenko     ~CommandOptions() override = default;
5240e41084aSJim Ingham 
52597206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
526b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
52797206d57SZachary Turner       Status error;
5283bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
529b9c1b51eSKate Stone       switch (short_option) {
5300e41084aSJim Ingham       case 'i':
531fe11483bSZachary Turner         if (option_arg.getAsInteger(0, m_ignore))
532b9c1b51eSKate Stone           error.SetErrorStringWithFormat(
533b9c1b51eSKate Stone               "invalid value for ignore option: \"%s\", should be a number.",
534fe11483bSZachary Turner               option_arg.str().c_str());
5350e41084aSJim Ingham         break;
5360e41084aSJim Ingham 
5370e41084aSJim Ingham       default:
53836162014SRaphael Isemann         llvm_unreachable("Unimplemented option");
5390e41084aSJim Ingham       }
5400e41084aSJim Ingham       return error;
5410e41084aSJim Ingham     }
5420e41084aSJim Ingham 
543b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
5440e41084aSJim Ingham       m_ignore = 0;
5450e41084aSJim Ingham     }
5460e41084aSJim Ingham 
5471f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
54870602439SZachary Turner       return llvm::makeArrayRef(g_process_continue_options);
5491f0f5b5bSZachary Turner     }
5500e41084aSJim Ingham 
5510e41084aSJim Ingham     uint32_t m_ignore;
5520e41084aSJim Ingham   };
5530e41084aSJim Ingham 
554b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
555f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
556a7015092SGreg Clayton     bool synchronous_execution = m_interpreter.GetSynchronous();
55730fdc8d8SChris Lattner     StateType state = process->GetState();
558b9c1b51eSKate Stone     if (state == eStateStopped) {
559b9c1b51eSKate Stone       if (command.GetArgumentCount() != 0) {
560b9c1b51eSKate Stone         result.AppendErrorWithFormat(
561b9c1b51eSKate Stone             "The '%s' command does not take any arguments.\n",
562b9c1b51eSKate Stone             m_cmd_name.c_str());
56330fdc8d8SChris Lattner         result.SetStatus(eReturnStatusFailed);
56430fdc8d8SChris Lattner         return false;
56530fdc8d8SChris Lattner       }
56630fdc8d8SChris Lattner 
567b9c1b51eSKate Stone       if (m_options.m_ignore > 0) {
5688d94ba0fSJim Ingham         ThreadSP sel_thread_sp(GetDefaultThread()->shared_from_this());
569b9c1b51eSKate Stone         if (sel_thread_sp) {
5700e41084aSJim Ingham           StopInfoSP stop_info_sp = sel_thread_sp->GetStopInfo();
571b9c1b51eSKate Stone           if (stop_info_sp &&
572b9c1b51eSKate Stone               stop_info_sp->GetStopReason() == eStopReasonBreakpoint) {
573b9c1b51eSKate Stone             lldb::break_id_t bp_site_id =
574b9c1b51eSKate Stone                 (lldb::break_id_t)stop_info_sp->GetValue();
575b9c1b51eSKate Stone             BreakpointSiteSP bp_site_sp(
576b9c1b51eSKate Stone                 process->GetBreakpointSiteList().FindByID(bp_site_id));
577b9c1b51eSKate Stone             if (bp_site_sp) {
578c7bece56SGreg Clayton               const size_t num_owners = bp_site_sp->GetNumberOfOwners();
579b9c1b51eSKate Stone               for (size_t i = 0; i < num_owners; i++) {
580b9c1b51eSKate Stone                 Breakpoint &bp_ref =
581b9c1b51eSKate Stone                     bp_site_sp->GetOwnerAtIndex(i)->GetBreakpoint();
582b9c1b51eSKate Stone                 if (!bp_ref.IsInternal()) {
5830e41084aSJim Ingham                   bp_ref.SetIgnoreCount(m_options.m_ignore);
5840e41084aSJim Ingham                 }
5850e41084aSJim Ingham               }
5860e41084aSJim Ingham             }
5870e41084aSJim Ingham           }
5880e41084aSJim Ingham         }
5890e41084aSJim Ingham       }
5900e41084aSJim Ingham 
59141f2b940SJim Ingham       { // Scope for thread list mutex:
592b9c1b51eSKate Stone         std::lock_guard<std::recursive_mutex> guard(
593b9c1b51eSKate Stone             process->GetThreadList().GetMutex());
59430fdc8d8SChris Lattner         const uint32_t num_threads = process->GetThreadList().GetSize();
59530fdc8d8SChris Lattner 
59630fdc8d8SChris Lattner         // Set the actions that the threads should each take when resuming
597b9c1b51eSKate Stone         for (uint32_t idx = 0; idx < num_threads; ++idx) {
5986c9ed91cSJim Ingham           const bool override_suspend = false;
599b9c1b51eSKate Stone           process->GetThreadList().GetThreadAtIndex(idx)->SetResumeState(
600b9c1b51eSKate Stone               eStateRunning, override_suspend);
60130fdc8d8SChris Lattner         }
60241f2b940SJim Ingham       }
60330fdc8d8SChris Lattner 
6044446487dSPavel Labath       const uint32_t iohandler_id = process->GetIOHandlerID();
6054446487dSPavel Labath 
606dc6224e0SGreg Clayton       StreamString stream;
60797206d57SZachary Turner       Status error;
608dc6224e0SGreg Clayton       if (synchronous_execution)
609dc6224e0SGreg Clayton         error = process->ResumeSynchronous(&stream);
610dc6224e0SGreg Clayton       else
611dc6224e0SGreg Clayton         error = process->Resume();
612a3b89e27STodd Fiala 
613b9c1b51eSKate Stone       if (error.Success()) {
614b9c1b51eSKate Stone         // There is a race condition where this thread will return up the call
61505097246SAdrian Prantl         // stack to the main command handler and show an (lldb) prompt before
61605097246SAdrian Prantl         // HandlePrivateEvent (from PrivateStateThread) has a chance to call
61705097246SAdrian Prantl         // PushProcessIOHandler().
6183879fe00SPavel Labath         process->SyncIOHandler(iohandler_id, std::chrono::seconds(2));
619a3b89e27STodd Fiala 
620b9c1b51eSKate Stone         result.AppendMessageWithFormat("Process %" PRIu64 " resuming\n",
621b9c1b51eSKate Stone                                        process->GetID());
622b9c1b51eSKate Stone         if (synchronous_execution) {
623b9c1b51eSKate Stone           // If any state changed events had anything to say, add that to the
624b9c1b51eSKate Stone           // result
625c156427dSZachary Turner           result.AppendMessage(stream.GetString());
62630fdc8d8SChris Lattner 
62730fdc8d8SChris Lattner           result.SetDidChangeProcessState(true);
62830fdc8d8SChris Lattner           result.SetStatus(eReturnStatusSuccessFinishNoResult);
629b9c1b51eSKate Stone         } else {
63030fdc8d8SChris Lattner           result.SetStatus(eReturnStatusSuccessContinuingNoResult);
63130fdc8d8SChris Lattner         }
632b9c1b51eSKate Stone       } else {
633b9c1b51eSKate Stone         result.AppendErrorWithFormat("Failed to resume process: %s.\n",
634b9c1b51eSKate Stone                                      error.AsCString());
63530fdc8d8SChris Lattner         result.SetStatus(eReturnStatusFailed);
63630fdc8d8SChris Lattner       }
637b9c1b51eSKate Stone     } else {
638b9c1b51eSKate Stone       result.AppendErrorWithFormat(
639b9c1b51eSKate Stone           "Process cannot be continued from its current state (%s).\n",
64030fdc8d8SChris Lattner           StateAsCString(state));
64130fdc8d8SChris Lattner       result.SetStatus(eReturnStatusFailed);
64230fdc8d8SChris Lattner     }
64330fdc8d8SChris Lattner     return result.Succeeded();
64430fdc8d8SChris Lattner   }
6450e41084aSJim Ingham 
646b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
6470e41084aSJim Ingham 
6480e41084aSJim Ingham   CommandOptions m_options;
6490e41084aSJim Ingham };
6500e41084aSJim Ingham 
65130fdc8d8SChris Lattner // CommandObjectProcessDetach
652438dfcffSRaphael Isemann #define LLDB_OPTIONS_process_detach
653438dfcffSRaphael Isemann #include "CommandOptions.inc"
6541f0f5b5bSZachary Turner 
655bb9caf73SJim Ingham #pragma mark CommandObjectProcessDetach
65630fdc8d8SChris Lattner 
657b9c1b51eSKate Stone class CommandObjectProcessDetach : public CommandObjectParsed {
65830fdc8d8SChris Lattner public:
659b9c1b51eSKate Stone   class CommandOptions : public Options {
660acff8950SJim Ingham   public:
661b9c1b51eSKate Stone     CommandOptions() : Options() { OptionParsingStarting(nullptr); }
662acff8950SJim Ingham 
66349bcfd80SEugene Zelenko     ~CommandOptions() override = default;
664acff8950SJim Ingham 
66597206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
666b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
66797206d57SZachary Turner       Status error;
668acff8950SJim Ingham       const int short_option = m_getopt_table[option_idx].val;
669acff8950SJim Ingham 
670b9c1b51eSKate Stone       switch (short_option) {
671acff8950SJim Ingham       case 's':
672acff8950SJim Ingham         bool tmp_result;
673acff8950SJim Ingham         bool success;
67447cbf4a0SPavel Labath         tmp_result = OptionArgParser::ToBoolean(option_arg, false, &success);
675acff8950SJim Ingham         if (!success)
676b9c1b51eSKate Stone           error.SetErrorStringWithFormat("invalid boolean option: \"%s\"",
677fe11483bSZachary Turner                                          option_arg.str().c_str());
678b9c1b51eSKate Stone         else {
679acff8950SJim Ingham           if (tmp_result)
680acff8950SJim Ingham             m_keep_stopped = eLazyBoolYes;
681acff8950SJim Ingham           else
682acff8950SJim Ingham             m_keep_stopped = eLazyBoolNo;
683acff8950SJim Ingham         }
684acff8950SJim Ingham         break;
685acff8950SJim Ingham       default:
68636162014SRaphael Isemann         llvm_unreachable("Unimplemented option");
687acff8950SJim Ingham       }
688acff8950SJim Ingham       return error;
689acff8950SJim Ingham     }
690acff8950SJim Ingham 
691b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
692acff8950SJim Ingham       m_keep_stopped = eLazyBoolCalculate;
693acff8950SJim Ingham     }
694acff8950SJim Ingham 
6951f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
69670602439SZachary Turner       return llvm::makeArrayRef(g_process_detach_options);
6971f0f5b5bSZachary Turner     }
698acff8950SJim Ingham 
699acff8950SJim Ingham     // Instance variables to hold the values for command options.
700acff8950SJim Ingham     LazyBool m_keep_stopped;
701acff8950SJim Ingham   };
70230fdc8d8SChris Lattner 
7037428a18cSKate Stone   CommandObjectProcessDetach(CommandInterpreter &interpreter)
704b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process detach",
705b9c1b51eSKate Stone                             "Detach from the current target process.",
706a7015092SGreg Clayton                             "process detach",
707b9c1b51eSKate Stone                             eCommandRequiresProcess | eCommandTryTargetAPILock |
708b9c1b51eSKate Stone                                 eCommandProcessMustBeLaunched),
709b9c1b51eSKate Stone         m_options() {}
71030fdc8d8SChris Lattner 
71149bcfd80SEugene Zelenko   ~CommandObjectProcessDetach() override = default;
71230fdc8d8SChris Lattner 
713b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
714acff8950SJim Ingham 
7155a988416SJim Ingham protected:
716b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
717f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
718acff8950SJim Ingham     // FIXME: This will be a Command Option:
719acff8950SJim Ingham     bool keep_stopped;
720b9c1b51eSKate Stone     if (m_options.m_keep_stopped == eLazyBoolCalculate) {
721acff8950SJim Ingham       // Check the process default:
72249bcfd80SEugene Zelenko       keep_stopped = process->GetDetachKeepsStopped();
723b9c1b51eSKate Stone     } else if (m_options.m_keep_stopped == eLazyBoolYes)
724acff8950SJim Ingham       keep_stopped = true;
725acff8950SJim Ingham     else
726acff8950SJim Ingham       keep_stopped = false;
727acff8950SJim Ingham 
72897206d57SZachary Turner     Status error(process->Detach(keep_stopped));
729b9c1b51eSKate Stone     if (error.Success()) {
73030fdc8d8SChris Lattner       result.SetStatus(eReturnStatusSuccessFinishResult);
731b9c1b51eSKate Stone     } else {
73230fdc8d8SChris Lattner       result.AppendErrorWithFormat("Detach failed: %s\n", error.AsCString());
73330fdc8d8SChris Lattner       result.SetStatus(eReturnStatusFailed);
73430fdc8d8SChris Lattner       return false;
73530fdc8d8SChris Lattner     }
73630fdc8d8SChris Lattner     return result.Succeeded();
73730fdc8d8SChris Lattner   }
738acff8950SJim Ingham 
739acff8950SJim Ingham   CommandOptions m_options;
740acff8950SJim Ingham };
741acff8950SJim Ingham 
742b766a73dSGreg Clayton // CommandObjectProcessConnect
743438dfcffSRaphael Isemann #define LLDB_OPTIONS_process_connect
744438dfcffSRaphael Isemann #include "CommandOptions.inc"
7451f0f5b5bSZachary Turner 
746b766a73dSGreg Clayton #pragma mark CommandObjectProcessConnect
747b766a73dSGreg Clayton 
748b9c1b51eSKate Stone class CommandObjectProcessConnect : public CommandObjectParsed {
749b766a73dSGreg Clayton public:
750b9c1b51eSKate Stone   class CommandOptions : public Options {
751b766a73dSGreg Clayton   public:
752b9c1b51eSKate Stone     CommandOptions() : Options() {
753b9c1b51eSKate Stone       // Keep default values of all options in one place: OptionParsingStarting
754b9c1b51eSKate Stone       // ()
755e1cfbc79STodd Fiala       OptionParsingStarting(nullptr);
756b766a73dSGreg Clayton     }
757b766a73dSGreg Clayton 
75849bcfd80SEugene Zelenko     ~CommandOptions() override = default;
759b766a73dSGreg Clayton 
76097206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
761b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
76297206d57SZachary Turner       Status error;
7633bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
764b766a73dSGreg Clayton 
765b9c1b51eSKate Stone       switch (short_option) {
766b766a73dSGreg Clayton       case 'p':
767adcd0268SBenjamin Kramer         plugin_name.assign(std::string(option_arg));
768b766a73dSGreg Clayton         break;
769b766a73dSGreg Clayton 
770b766a73dSGreg Clayton       default:
77136162014SRaphael Isemann         llvm_unreachable("Unimplemented option");
772b766a73dSGreg Clayton       }
773b766a73dSGreg Clayton       return error;
774b766a73dSGreg Clayton     }
775b766a73dSGreg Clayton 
776b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
777b766a73dSGreg Clayton       plugin_name.clear();
778b766a73dSGreg Clayton     }
779b766a73dSGreg Clayton 
7801f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
78170602439SZachary Turner       return llvm::makeArrayRef(g_process_connect_options);
7821f0f5b5bSZachary Turner     }
783b766a73dSGreg Clayton 
784b766a73dSGreg Clayton     // Instance variables to hold the values for command options.
785b766a73dSGreg Clayton 
786b766a73dSGreg Clayton     std::string plugin_name;
787b766a73dSGreg Clayton   };
788b766a73dSGreg Clayton 
789b9c1b51eSKate Stone   CommandObjectProcessConnect(CommandInterpreter &interpreter)
790b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process connect",
791b766a73dSGreg Clayton                             "Connect to a remote debug service.",
792b9c1b51eSKate Stone                             "process connect <remote-url>", 0),
793b9c1b51eSKate Stone         m_options() {}
794b766a73dSGreg Clayton 
79549bcfd80SEugene Zelenko   ~CommandObjectProcessConnect() override = default;
796b766a73dSGreg Clayton 
797b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
7985a988416SJim Ingham 
7995a988416SJim Ingham protected:
800b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
801b9c1b51eSKate Stone     if (command.GetArgumentCount() != 1) {
802b9c1b51eSKate Stone       result.AppendErrorWithFormat(
803b9c1b51eSKate Stone           "'%s' takes exactly one argument:\nUsage: %s\n", m_cmd_name.c_str(),
804ccd6cffbSTamas Berghammer           m_cmd_syntax.c_str());
805ccd6cffbSTamas Berghammer       result.SetStatus(eReturnStatusFailed);
806ccd6cffbSTamas Berghammer       return false;
807ccd6cffbSTamas Berghammer     }
808ccd6cffbSTamas Berghammer 
809ccd6cffbSTamas Berghammer     Process *process = m_exe_ctx.GetProcessPtr();
810b9c1b51eSKate Stone     if (process && process->IsAlive()) {
811b9c1b51eSKate Stone       result.AppendErrorWithFormat(
812b9c1b51eSKate Stone           "Process %" PRIu64
813b9c1b51eSKate Stone           " is currently being debugged, kill the process before connecting.\n",
814b766a73dSGreg Clayton           process->GetID());
815b766a73dSGreg Clayton       result.SetStatus(eReturnStatusFailed);
816b766a73dSGreg Clayton       return false;
817b766a73dSGreg Clayton     }
818b766a73dSGreg Clayton 
819ccd6cffbSTamas Berghammer     const char *plugin_name = nullptr;
820b766a73dSGreg Clayton     if (!m_options.plugin_name.empty())
821b766a73dSGreg Clayton       plugin_name = m_options.plugin_name.c_str();
822b766a73dSGreg Clayton 
82397206d57SZachary Turner     Status error;
82457179860SJonas Devlieghere     Debugger &debugger = GetDebugger();
825ccd6cffbSTamas Berghammer     PlatformSP platform_sp = m_interpreter.GetPlatform(true);
826706cccb8SJonas Devlieghere     ProcessSP process_sp =
827706cccb8SJonas Devlieghere         debugger.GetAsyncExecution()
828706cccb8SJonas Devlieghere             ? platform_sp->ConnectProcess(
829b9c1b51eSKate Stone                   command.GetArgumentAtIndex(0), plugin_name, debugger,
830706cccb8SJonas Devlieghere                   debugger.GetSelectedTarget().get(), error)
831706cccb8SJonas Devlieghere             : platform_sp->ConnectProcessSynchronous(
832706cccb8SJonas Devlieghere                   command.GetArgumentAtIndex(0), plugin_name, debugger,
833706cccb8SJonas Devlieghere                   result.GetOutputStream(), debugger.GetSelectedTarget().get(),
834706cccb8SJonas Devlieghere                   error);
835b9c1b51eSKate Stone     if (error.Fail() || process_sp == nullptr) {
836ccd6cffbSTamas Berghammer       result.AppendError(error.AsCString("Error connecting to the process"));
837b766a73dSGreg Clayton       result.SetStatus(eReturnStatusFailed);
838b766a73dSGreg Clayton       return false;
839b766a73dSGreg Clayton     }
840ccd6cffbSTamas Berghammer     return true;
841b766a73dSGreg Clayton   }
842b766a73dSGreg Clayton 
843b766a73dSGreg Clayton   CommandOptions m_options;
844b766a73dSGreg Clayton };
845b766a73dSGreg Clayton 
846998255bfSGreg Clayton // CommandObjectProcessPlugin
847998255bfSGreg Clayton #pragma mark CommandObjectProcessPlugin
848998255bfSGreg Clayton 
849b9c1b51eSKate Stone class CommandObjectProcessPlugin : public CommandObjectProxy {
850998255bfSGreg Clayton public:
8517428a18cSKate Stone   CommandObjectProcessPlugin(CommandInterpreter &interpreter)
852b9c1b51eSKate Stone       : CommandObjectProxy(
853b9c1b51eSKate Stone             interpreter, "process plugin",
854b9c1b51eSKate Stone             "Send a custom command to the current target process plug-in.",
855b9c1b51eSKate Stone             "process plugin <args>", 0) {}
856998255bfSGreg Clayton 
85749bcfd80SEugene Zelenko   ~CommandObjectProcessPlugin() override = default;
858998255bfSGreg Clayton 
859b9c1b51eSKate Stone   CommandObject *GetProxyCommandObject() override {
860e05b2efeSGreg Clayton     Process *process = m_interpreter.GetExecutionContext().GetProcessPtr();
861998255bfSGreg Clayton     if (process)
862998255bfSGreg Clayton       return process->GetPluginCommandObject();
86349bcfd80SEugene Zelenko     return nullptr;
864998255bfSGreg Clayton   }
865998255bfSGreg Clayton };
866998255bfSGreg Clayton 
8678f343b09SGreg Clayton // CommandObjectProcessLoad
868438dfcffSRaphael Isemann #define LLDB_OPTIONS_process_load
869438dfcffSRaphael Isemann #include "CommandOptions.inc"
8701f0f5b5bSZachary Turner 
871bb9caf73SJim Ingham #pragma mark CommandObjectProcessLoad
8728f343b09SGreg Clayton 
873b9c1b51eSKate Stone class CommandObjectProcessLoad : public CommandObjectParsed {
8748f343b09SGreg Clayton public:
875b9c1b51eSKate Stone   class CommandOptions : public Options {
8764fbd67acSTamas Berghammer   public:
877b9c1b51eSKate Stone     CommandOptions() : Options() {
878b9c1b51eSKate Stone       // Keep default values of all options in one place: OptionParsingStarting
879b9c1b51eSKate Stone       // ()
880e1cfbc79STodd Fiala       OptionParsingStarting(nullptr);
8814fbd67acSTamas Berghammer     }
8824fbd67acSTamas Berghammer 
8834fbd67acSTamas Berghammer     ~CommandOptions() override = default;
8844fbd67acSTamas Berghammer 
88597206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
886b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
88797206d57SZachary Turner       Status error;
8884fbd67acSTamas Berghammer       const int short_option = m_getopt_table[option_idx].val;
889b9c1b51eSKate Stone       switch (short_option) {
8904fbd67acSTamas Berghammer       case 'i':
8914fbd67acSTamas Berghammer         do_install = true;
892fe11483bSZachary Turner         if (!option_arg.empty())
8938f3be7a3SJonas Devlieghere           install_path.SetFile(option_arg, FileSpec::Style::native);
8944fbd67acSTamas Berghammer         break;
8954fbd67acSTamas Berghammer       default:
89636162014SRaphael Isemann         llvm_unreachable("Unimplemented option");
8974fbd67acSTamas Berghammer       }
8984fbd67acSTamas Berghammer       return error;
8994fbd67acSTamas Berghammer     }
9004fbd67acSTamas Berghammer 
901b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
9024fbd67acSTamas Berghammer       do_install = false;
9034fbd67acSTamas Berghammer       install_path.Clear();
9044fbd67acSTamas Berghammer     }
9054fbd67acSTamas Berghammer 
9061f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
90770602439SZachary Turner       return llvm::makeArrayRef(g_process_load_options);
9081f0f5b5bSZachary Turner     }
9094fbd67acSTamas Berghammer 
9104fbd67acSTamas Berghammer     // Instance variables to hold the values for command options.
9114fbd67acSTamas Berghammer     bool do_install;
9124fbd67acSTamas Berghammer     FileSpec install_path;
9134fbd67acSTamas Berghammer   };
9148f343b09SGreg Clayton 
915b9c1b51eSKate Stone   CommandObjectProcessLoad(CommandInterpreter &interpreter)
916b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process load",
9178f343b09SGreg Clayton                             "Load a shared library into the current process.",
9188f343b09SGreg Clayton                             "process load <filename> [<filename> ...]",
919b9c1b51eSKate Stone                             eCommandRequiresProcess | eCommandTryTargetAPILock |
920e87764f2SEnrico Granata                                 eCommandProcessMustBeLaunched |
9214fbd67acSTamas Berghammer                                 eCommandProcessMustBePaused),
922b9c1b51eSKate Stone         m_options() {}
9238f343b09SGreg Clayton 
9244fbd67acSTamas Berghammer   ~CommandObjectProcessLoad() override = default;
9254fbd67acSTamas Berghammer 
926*e1cd7cacSGongyu Deng   void
927*e1cd7cacSGongyu Deng   HandleArgumentCompletion(CompletionRequest &request,
928*e1cd7cacSGongyu Deng                            OptionElementVector &opt_element_vector) override {
929*e1cd7cacSGongyu Deng     if (!m_exe_ctx.HasProcessScope())
930*e1cd7cacSGongyu Deng       return;
931*e1cd7cacSGongyu Deng 
932*e1cd7cacSGongyu Deng     CommandCompletions::InvokeCommonCompletionCallbacks(
933*e1cd7cacSGongyu Deng         GetCommandInterpreter(), CommandCompletions::eDiskFileCompletion,
934*e1cd7cacSGongyu Deng         request, nullptr);
935*e1cd7cacSGongyu Deng   }
936*e1cd7cacSGongyu Deng 
937b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
9388f343b09SGreg Clayton 
9395a988416SJim Ingham protected:
940b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
941f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
9428f343b09SGreg Clayton 
94397d2c401SZachary Turner     for (auto &entry : command.entries()) {
94497206d57SZachary Turner       Status error;
9453cb132a0STamas Berghammer       PlatformSP platform = process->GetTarget().GetPlatform();
9460d9a201eSRaphael Isemann       llvm::StringRef image_path = entry.ref();
9474fbd67acSTamas Berghammer       uint32_t image_token = LLDB_INVALID_IMAGE_TOKEN;
9484fbd67acSTamas Berghammer 
949b9c1b51eSKate Stone       if (!m_options.do_install) {
9508f3be7a3SJonas Devlieghere         FileSpec image_spec(image_path);
9513cb132a0STamas Berghammer         platform->ResolveRemotePath(image_spec, image_spec);
952b9c1b51eSKate Stone         image_token =
953b9c1b51eSKate Stone             platform->LoadImage(process, FileSpec(), image_spec, error);
954b9c1b51eSKate Stone       } else if (m_options.install_path) {
9558f3be7a3SJonas Devlieghere         FileSpec image_spec(image_path);
9568f3be7a3SJonas Devlieghere         FileSystem::Instance().Resolve(image_spec);
957b9c1b51eSKate Stone         platform->ResolveRemotePath(m_options.install_path,
958b9c1b51eSKate Stone                                     m_options.install_path);
959b9c1b51eSKate Stone         image_token = platform->LoadImage(process, image_spec,
960b9c1b51eSKate Stone                                           m_options.install_path, error);
961b9c1b51eSKate Stone       } else {
9628f3be7a3SJonas Devlieghere         FileSpec image_spec(image_path);
9638f3be7a3SJonas Devlieghere         FileSystem::Instance().Resolve(image_spec);
964b9c1b51eSKate Stone         image_token =
965b9c1b51eSKate Stone             platform->LoadImage(process, image_spec, FileSpec(), error);
9664fbd67acSTamas Berghammer       }
9674fbd67acSTamas Berghammer 
968b9c1b51eSKate Stone       if (image_token != LLDB_INVALID_IMAGE_TOKEN) {
969b9c1b51eSKate Stone         result.AppendMessageWithFormat(
97097d2c401SZachary Turner             "Loading \"%s\"...ok\nImage %u loaded.\n", image_path.str().c_str(),
97197d2c401SZachary Turner             image_token);
9728f343b09SGreg Clayton         result.SetStatus(eReturnStatusSuccessFinishResult);
973b9c1b51eSKate Stone       } else {
97497d2c401SZachary Turner         result.AppendErrorWithFormat("failed to load '%s': %s",
97597d2c401SZachary Turner                                      image_path.str().c_str(),
976b9c1b51eSKate Stone                                      error.AsCString());
9778f343b09SGreg Clayton         result.SetStatus(eReturnStatusFailed);
9788f343b09SGreg Clayton       }
9798f343b09SGreg Clayton     }
9808f343b09SGreg Clayton     return result.Succeeded();
9818f343b09SGreg Clayton   }
9824fbd67acSTamas Berghammer 
9834fbd67acSTamas Berghammer   CommandOptions m_options;
9848f343b09SGreg Clayton };
9858f343b09SGreg Clayton 
9868f343b09SGreg Clayton // CommandObjectProcessUnload
987bb9caf73SJim Ingham #pragma mark CommandObjectProcessUnload
9888f343b09SGreg Clayton 
989b9c1b51eSKate Stone class CommandObjectProcessUnload : public CommandObjectParsed {
9908f343b09SGreg Clayton public:
991b9c1b51eSKate Stone   CommandObjectProcessUnload(CommandInterpreter &interpreter)
992b9c1b51eSKate Stone       : CommandObjectParsed(
993b9c1b51eSKate Stone             interpreter, "process unload",
994b9c1b51eSKate Stone             "Unload a shared library from the current process using the index "
995b9c1b51eSKate Stone             "returned by a previous call to \"process load\".",
9968f343b09SGreg Clayton             "process unload <index>",
997b9c1b51eSKate Stone             eCommandRequiresProcess | eCommandTryTargetAPILock |
998b9c1b51eSKate Stone                 eCommandProcessMustBeLaunched | eCommandProcessMustBePaused) {}
9998f343b09SGreg Clayton 
100049bcfd80SEugene Zelenko   ~CommandObjectProcessUnload() override = default;
10018f343b09SGreg Clayton 
1002*e1cd7cacSGongyu Deng   void
1003*e1cd7cacSGongyu Deng   HandleArgumentCompletion(CompletionRequest &request,
1004*e1cd7cacSGongyu Deng                            OptionElementVector &opt_element_vector) override {
1005*e1cd7cacSGongyu Deng 
1006*e1cd7cacSGongyu Deng     if (request.GetCursorIndex() || !m_exe_ctx.HasProcessScope())
1007*e1cd7cacSGongyu Deng       return;
1008*e1cd7cacSGongyu Deng 
1009*e1cd7cacSGongyu Deng     Process *process = m_exe_ctx.GetProcessPtr();
1010*e1cd7cacSGongyu Deng 
1011*e1cd7cacSGongyu Deng     const std::vector<lldb::addr_t> &tokens = process->GetImageTokens();
1012*e1cd7cacSGongyu Deng     const size_t token_num = tokens.size();
1013*e1cd7cacSGongyu Deng     for (size_t i = 0; i < token_num; ++i) {
1014*e1cd7cacSGongyu Deng       if (tokens[i] == LLDB_INVALID_IMAGE_TOKEN)
1015*e1cd7cacSGongyu Deng         continue;
1016*e1cd7cacSGongyu Deng       request.TryCompleteCurrentArg(std::to_string(i));
1017*e1cd7cacSGongyu Deng     }
1018*e1cd7cacSGongyu Deng   }
1019*e1cd7cacSGongyu Deng 
10205a988416SJim Ingham protected:
1021b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1022f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
10238f343b09SGreg Clayton 
102497d2c401SZachary Turner     for (auto &entry : command.entries()) {
102597d2c401SZachary Turner       uint32_t image_token;
10260d9a201eSRaphael Isemann       if (entry.ref().getAsInteger(0, image_token)) {
1027b9c1b51eSKate Stone         result.AppendErrorWithFormat("invalid image index argument '%s'",
10280d9a201eSRaphael Isemann                                      entry.ref().str().c_str());
10298f343b09SGreg Clayton         result.SetStatus(eReturnStatusFailed);
10308f343b09SGreg Clayton         break;
1031b9c1b51eSKate Stone       } else {
103297206d57SZachary Turner         Status error(process->GetTarget().GetPlatform()->UnloadImage(
1033b9c1b51eSKate Stone             process, image_token));
1034b9c1b51eSKate Stone         if (error.Success()) {
1035b9c1b51eSKate Stone           result.AppendMessageWithFormat(
1036b9c1b51eSKate Stone               "Unloading shared library with index %u...ok\n", image_token);
10378f343b09SGreg Clayton           result.SetStatus(eReturnStatusSuccessFinishResult);
1038b9c1b51eSKate Stone         } else {
1039b9c1b51eSKate Stone           result.AppendErrorWithFormat("failed to unload image: %s",
1040b9c1b51eSKate Stone                                        error.AsCString());
10418f343b09SGreg Clayton           result.SetStatus(eReturnStatusFailed);
10428f343b09SGreg Clayton           break;
10438f343b09SGreg Clayton         }
10448f343b09SGreg Clayton       }
10458f343b09SGreg Clayton     }
10468f343b09SGreg Clayton     return result.Succeeded();
10478f343b09SGreg Clayton   }
10488f343b09SGreg Clayton };
10498f343b09SGreg Clayton 
105030fdc8d8SChris Lattner // CommandObjectProcessSignal
1051bb9caf73SJim Ingham #pragma mark CommandObjectProcessSignal
105230fdc8d8SChris Lattner 
1053b9c1b51eSKate Stone class CommandObjectProcessSignal : public CommandObjectParsed {
105430fdc8d8SChris Lattner public:
10557428a18cSKate Stone   CommandObjectProcessSignal(CommandInterpreter &interpreter)
1056a925974bSAdrian Prantl       : CommandObjectParsed(
1057a925974bSAdrian Prantl             interpreter, "process signal",
1058a925974bSAdrian Prantl             "Send a UNIX signal to the current target process.", nullptr,
1059a925974bSAdrian Prantl             eCommandRequiresProcess | eCommandTryTargetAPILock) {
1060405fe67fSCaroline Tice     CommandArgumentEntry arg;
1061405fe67fSCaroline Tice     CommandArgumentData signal_arg;
1062405fe67fSCaroline Tice 
1063405fe67fSCaroline Tice     // Define the first (and only) variant of this arg.
1064c0dbdfb6SCaroline Tice     signal_arg.arg_type = eArgTypeUnixSignal;
1065405fe67fSCaroline Tice     signal_arg.arg_repetition = eArgRepeatPlain;
1066405fe67fSCaroline Tice 
1067b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
1068b9c1b51eSKate Stone     // argument entry.
1069405fe67fSCaroline Tice     arg.push_back(signal_arg);
1070405fe67fSCaroline Tice 
1071405fe67fSCaroline Tice     // Push the data for the first argument into the m_arguments vector.
1072405fe67fSCaroline Tice     m_arguments.push_back(arg);
107330fdc8d8SChris Lattner   }
107430fdc8d8SChris Lattner 
107549bcfd80SEugene Zelenko   ~CommandObjectProcessSignal() override = default;
107630fdc8d8SChris Lattner 
10772bba1c22SRaphael Isemann   void
10782bba1c22SRaphael Isemann   HandleArgumentCompletion(CompletionRequest &request,
10792bba1c22SRaphael Isemann                            OptionElementVector &opt_element_vector) override {
10802bba1c22SRaphael Isemann     if (!m_exe_ctx.HasProcessScope() || request.GetCursorIndex() != 0)
10812bba1c22SRaphael Isemann       return;
10822bba1c22SRaphael Isemann 
10832bba1c22SRaphael Isemann     UnixSignalsSP signals = m_exe_ctx.GetProcessPtr()->GetUnixSignals();
10842bba1c22SRaphael Isemann     int signo = signals->GetFirstSignalNumber();
10852bba1c22SRaphael Isemann     while (signo != LLDB_INVALID_SIGNAL_NUMBER) {
10862bba1c22SRaphael Isemann       request.AddCompletion(signals->GetSignalAsCString(signo), "");
10872bba1c22SRaphael Isemann       signo = signals->GetNextSignalNumber(signo);
10882bba1c22SRaphael Isemann     }
10892bba1c22SRaphael Isemann   }
10902bba1c22SRaphael Isemann 
10915a988416SJim Ingham protected:
1092b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1093f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
109430fdc8d8SChris Lattner 
1095b9c1b51eSKate Stone     if (command.GetArgumentCount() == 1) {
1096237cd906SGreg Clayton       int signo = LLDB_INVALID_SIGNAL_NUMBER;
1097237cd906SGreg Clayton 
1098237cd906SGreg Clayton       const char *signal_name = command.GetArgumentAtIndex(0);
10999010cef2SRaphael Isemann       if (::isxdigit(signal_name[0])) {
11009010cef2SRaphael Isemann         if (!llvm::to_integer(signal_name, signo))
11019010cef2SRaphael Isemann           signo = LLDB_INVALID_SIGNAL_NUMBER;
11029010cef2SRaphael Isemann       } else
110398d0a4b3SChaoren Lin         signo = process->GetUnixSignals()->GetSignalNumberFromName(signal_name);
1104237cd906SGreg Clayton 
1105b9c1b51eSKate Stone       if (signo == LLDB_INVALID_SIGNAL_NUMBER) {
1106b9c1b51eSKate Stone         result.AppendErrorWithFormat("Invalid signal argument '%s'.\n",
1107b9c1b51eSKate Stone                                      command.GetArgumentAtIndex(0));
110830fdc8d8SChris Lattner         result.SetStatus(eReturnStatusFailed);
1109b9c1b51eSKate Stone       } else {
111097206d57SZachary Turner         Status error(process->Signal(signo));
1111b9c1b51eSKate Stone         if (error.Success()) {
111230fdc8d8SChris Lattner           result.SetStatus(eReturnStatusSuccessFinishResult);
1113b9c1b51eSKate Stone         } else {
1114b9c1b51eSKate Stone           result.AppendErrorWithFormat("Failed to send signal %i: %s\n", signo,
1115b9c1b51eSKate Stone                                        error.AsCString());
111630fdc8d8SChris Lattner           result.SetStatus(eReturnStatusFailed);
111730fdc8d8SChris Lattner         }
111830fdc8d8SChris Lattner       }
1119b9c1b51eSKate Stone     } else {
1120b9c1b51eSKate Stone       result.AppendErrorWithFormat(
1121b9c1b51eSKate Stone           "'%s' takes exactly one signal number argument:\nUsage: %s\n",
1122b9c1b51eSKate Stone           m_cmd_name.c_str(), m_cmd_syntax.c_str());
112330fdc8d8SChris Lattner       result.SetStatus(eReturnStatusFailed);
112430fdc8d8SChris Lattner     }
112530fdc8d8SChris Lattner     return result.Succeeded();
112630fdc8d8SChris Lattner   }
112730fdc8d8SChris Lattner };
112830fdc8d8SChris Lattner 
112930fdc8d8SChris Lattner // CommandObjectProcessInterrupt
1130bb9caf73SJim Ingham #pragma mark CommandObjectProcessInterrupt
113130fdc8d8SChris Lattner 
1132b9c1b51eSKate Stone class CommandObjectProcessInterrupt : public CommandObjectParsed {
113330fdc8d8SChris Lattner public:
11347428a18cSKate Stone   CommandObjectProcessInterrupt(CommandInterpreter &interpreter)
1135b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process interrupt",
1136b9c1b51eSKate Stone                             "Interrupt the current target process.",
1137a7015092SGreg Clayton                             "process interrupt",
1138b9c1b51eSKate Stone                             eCommandRequiresProcess | eCommandTryTargetAPILock |
1139b9c1b51eSKate Stone                                 eCommandProcessMustBeLaunched) {}
114030fdc8d8SChris Lattner 
114149bcfd80SEugene Zelenko   ~CommandObjectProcessInterrupt() override = default;
114230fdc8d8SChris Lattner 
11435a988416SJim Ingham protected:
1144b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1145f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
1146b9c1b51eSKate Stone     if (process == nullptr) {
114730fdc8d8SChris Lattner       result.AppendError("no process to halt");
114830fdc8d8SChris Lattner       result.SetStatus(eReturnStatusFailed);
114930fdc8d8SChris Lattner       return false;
115030fdc8d8SChris Lattner     }
115130fdc8d8SChris Lattner 
1152b9c1b51eSKate Stone     if (command.GetArgumentCount() == 0) {
1153f9b57b9dSGreg Clayton       bool clear_thread_plans = true;
115497206d57SZachary Turner       Status error(process->Halt(clear_thread_plans));
1155b9c1b51eSKate Stone       if (error.Success()) {
115630fdc8d8SChris Lattner         result.SetStatus(eReturnStatusSuccessFinishResult);
1157b9c1b51eSKate Stone       } else {
1158b9c1b51eSKate Stone         result.AppendErrorWithFormat("Failed to halt process: %s\n",
1159b9c1b51eSKate Stone                                      error.AsCString());
116030fdc8d8SChris Lattner         result.SetStatus(eReturnStatusFailed);
116130fdc8d8SChris Lattner       }
1162b9c1b51eSKate Stone     } else {
1163fd54b368SJason Molenda       result.AppendErrorWithFormat("'%s' takes no arguments:\nUsage: %s\n",
1164b9c1b51eSKate Stone                                    m_cmd_name.c_str(), m_cmd_syntax.c_str());
116530fdc8d8SChris Lattner       result.SetStatus(eReturnStatusFailed);
116630fdc8d8SChris Lattner     }
116730fdc8d8SChris Lattner     return result.Succeeded();
116830fdc8d8SChris Lattner   }
116930fdc8d8SChris Lattner };
117030fdc8d8SChris Lattner 
117130fdc8d8SChris Lattner // CommandObjectProcessKill
1172bb9caf73SJim Ingham #pragma mark CommandObjectProcessKill
117330fdc8d8SChris Lattner 
1174b9c1b51eSKate Stone class CommandObjectProcessKill : public CommandObjectParsed {
117530fdc8d8SChris Lattner public:
11767428a18cSKate Stone   CommandObjectProcessKill(CommandInterpreter &interpreter)
1177b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process kill",
1178b9c1b51eSKate Stone                             "Terminate the current target process.",
1179b9c1b51eSKate Stone                             "process kill",
1180b9c1b51eSKate Stone                             eCommandRequiresProcess | eCommandTryTargetAPILock |
1181b9c1b51eSKate Stone                                 eCommandProcessMustBeLaunched) {}
118230fdc8d8SChris Lattner 
118349bcfd80SEugene Zelenko   ~CommandObjectProcessKill() override = default;
118430fdc8d8SChris Lattner 
11855a988416SJim Ingham protected:
1186b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1187f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
1188b9c1b51eSKate Stone     if (process == nullptr) {
118930fdc8d8SChris Lattner       result.AppendError("no process to kill");
119030fdc8d8SChris Lattner       result.SetStatus(eReturnStatusFailed);
119130fdc8d8SChris Lattner       return false;
119230fdc8d8SChris Lattner     }
119330fdc8d8SChris Lattner 
1194b9c1b51eSKate Stone     if (command.GetArgumentCount() == 0) {
119597206d57SZachary Turner       Status error(process->Destroy(true));
1196b9c1b51eSKate Stone       if (error.Success()) {
119730fdc8d8SChris Lattner         result.SetStatus(eReturnStatusSuccessFinishResult);
1198b9c1b51eSKate Stone       } else {
1199b9c1b51eSKate Stone         result.AppendErrorWithFormat("Failed to kill process: %s\n",
1200b9c1b51eSKate Stone                                      error.AsCString());
120130fdc8d8SChris Lattner         result.SetStatus(eReturnStatusFailed);
120230fdc8d8SChris Lattner       }
1203b9c1b51eSKate Stone     } else {
1204fd54b368SJason Molenda       result.AppendErrorWithFormat("'%s' takes no arguments:\nUsage: %s\n",
1205b9c1b51eSKate Stone                                    m_cmd_name.c_str(), m_cmd_syntax.c_str());
120630fdc8d8SChris Lattner       result.SetStatus(eReturnStatusFailed);
120730fdc8d8SChris Lattner     }
120830fdc8d8SChris Lattner     return result.Succeeded();
120930fdc8d8SChris Lattner   }
121030fdc8d8SChris Lattner };
121130fdc8d8SChris Lattner 
1212a2715cf1SGreg Clayton // CommandObjectProcessSaveCore
1213a2715cf1SGreg Clayton #pragma mark CommandObjectProcessSaveCore
1214a2715cf1SGreg Clayton 
1215b9c1b51eSKate Stone class CommandObjectProcessSaveCore : public CommandObjectParsed {
1216a2715cf1SGreg Clayton public:
1217b9c1b51eSKate Stone   CommandObjectProcessSaveCore(CommandInterpreter &interpreter)
1218b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process save-core",
1219b9c1b51eSKate Stone                             "Save the current process as a core file using an "
1220b9c1b51eSKate Stone                             "appropriate file type.",
1221a2715cf1SGreg Clayton                             "process save-core FILE",
1222b9c1b51eSKate Stone                             eCommandRequiresProcess | eCommandTryTargetAPILock |
1223b9c1b51eSKate Stone                                 eCommandProcessMustBeLaunched) {}
1224a2715cf1SGreg Clayton 
122549bcfd80SEugene Zelenko   ~CommandObjectProcessSaveCore() override = default;
1226a2715cf1SGreg Clayton 
1227a2715cf1SGreg Clayton protected:
1228b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1229a2715cf1SGreg Clayton     ProcessSP process_sp = m_exe_ctx.GetProcessSP();
1230b9c1b51eSKate Stone     if (process_sp) {
1231b9c1b51eSKate Stone       if (command.GetArgumentCount() == 1) {
12328f3be7a3SJonas Devlieghere         FileSpec output_file(command.GetArgumentAtIndex(0));
123397206d57SZachary Turner         Status error = PluginManager::SaveCore(process_sp, output_file);
1234b9c1b51eSKate Stone         if (error.Success()) {
1235a2715cf1SGreg Clayton           result.SetStatus(eReturnStatusSuccessFinishResult);
1236b9c1b51eSKate Stone         } else {
1237b9c1b51eSKate Stone           result.AppendErrorWithFormat(
1238b9c1b51eSKate Stone               "Failed to save core file for process: %s\n", error.AsCString());
1239a2715cf1SGreg Clayton           result.SetStatus(eReturnStatusFailed);
1240a2715cf1SGreg Clayton         }
1241b9c1b51eSKate Stone       } else {
1242a2715cf1SGreg Clayton         result.AppendErrorWithFormat("'%s' takes one arguments:\nUsage: %s\n",
1243b9c1b51eSKate Stone                                      m_cmd_name.c_str(), m_cmd_syntax.c_str());
1244a2715cf1SGreg Clayton         result.SetStatus(eReturnStatusFailed);
1245a2715cf1SGreg Clayton       }
1246b9c1b51eSKate Stone     } else {
1247a2715cf1SGreg Clayton       result.AppendError("invalid process");
1248a2715cf1SGreg Clayton       result.SetStatus(eReturnStatusFailed);
1249a2715cf1SGreg Clayton       return false;
1250a2715cf1SGreg Clayton     }
1251a2715cf1SGreg Clayton 
1252a2715cf1SGreg Clayton     return result.Succeeded();
1253a2715cf1SGreg Clayton   }
1254a2715cf1SGreg Clayton };
1255a2715cf1SGreg Clayton 
12564b9bea87SJim Ingham // CommandObjectProcessStatus
1257bb9caf73SJim Ingham #pragma mark CommandObjectProcessStatus
1258d7c403e6SMed Ismail Bennani #define LLDB_OPTIONS_process_status
1259d7c403e6SMed Ismail Bennani #include "CommandOptions.inc"
1260bb9caf73SJim Ingham 
1261b9c1b51eSKate Stone class CommandObjectProcessStatus : public CommandObjectParsed {
12624b9bea87SJim Ingham public:
12637428a18cSKate Stone   CommandObjectProcessStatus(CommandInterpreter &interpreter)
1264b9c1b51eSKate Stone       : CommandObjectParsed(
1265b9c1b51eSKate Stone             interpreter, "process status",
1266b9c1b51eSKate Stone             "Show status and stop location for the current target process.",
1267b9c1b51eSKate Stone             "process status",
1268d7c403e6SMed Ismail Bennani             eCommandRequiresProcess | eCommandTryTargetAPILock),
1269d7c403e6SMed Ismail Bennani         m_options() {}
12704b9bea87SJim Ingham 
127149bcfd80SEugene Zelenko   ~CommandObjectProcessStatus() override = default;
12724b9bea87SJim Ingham 
1273d7c403e6SMed Ismail Bennani   Options *GetOptions() override { return &m_options; }
1274d7c403e6SMed Ismail Bennani 
1275d7c403e6SMed Ismail Bennani   class CommandOptions : public Options {
1276d7c403e6SMed Ismail Bennani   public:
1277d7c403e6SMed Ismail Bennani     CommandOptions() : Options(), m_verbose(false) {}
1278d7c403e6SMed Ismail Bennani 
1279d7c403e6SMed Ismail Bennani     ~CommandOptions() override = default;
1280d7c403e6SMed Ismail Bennani 
1281d7c403e6SMed Ismail Bennani     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
1282d7c403e6SMed Ismail Bennani                           ExecutionContext *execution_context) override {
1283d7c403e6SMed Ismail Bennani       const int short_option = m_getopt_table[option_idx].val;
1284d7c403e6SMed Ismail Bennani 
1285d7c403e6SMed Ismail Bennani       switch (short_option) {
1286d7c403e6SMed Ismail Bennani       case 'v':
1287d7c403e6SMed Ismail Bennani         m_verbose = true;
1288d7c403e6SMed Ismail Bennani         break;
1289d7c403e6SMed Ismail Bennani       default:
1290d7c403e6SMed Ismail Bennani         llvm_unreachable("Unimplemented option");
1291d7c403e6SMed Ismail Bennani       }
1292d7c403e6SMed Ismail Bennani 
1293d7c403e6SMed Ismail Bennani       return {};
1294d7c403e6SMed Ismail Bennani     }
1295d7c403e6SMed Ismail Bennani 
1296d7c403e6SMed Ismail Bennani     void OptionParsingStarting(ExecutionContext *execution_context) override {
1297d7c403e6SMed Ismail Bennani       m_verbose = false;
1298d7c403e6SMed Ismail Bennani     }
1299d7c403e6SMed Ismail Bennani 
1300d7c403e6SMed Ismail Bennani     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
1301d7c403e6SMed Ismail Bennani       return llvm::makeArrayRef(g_process_status_options);
1302d7c403e6SMed Ismail Bennani     }
1303d7c403e6SMed Ismail Bennani 
1304d7c403e6SMed Ismail Bennani     // Instance variables to hold the values for command options.
1305d7c403e6SMed Ismail Bennani     bool m_verbose;
1306d7c403e6SMed Ismail Bennani   };
1307d7c403e6SMed Ismail Bennani 
1308d7c403e6SMed Ismail Bennani protected:
1309b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
13107260f620SGreg Clayton     Stream &strm = result.GetOutputStream();
13114b9bea87SJim Ingham     result.SetStatus(eReturnStatusSuccessFinishNoResult);
1312d7c403e6SMed Ismail Bennani 
1313d7c403e6SMed Ismail Bennani     if (command.GetArgumentCount()) {
1314d7c403e6SMed Ismail Bennani       result.AppendError("'process status' takes no arguments");
1315d7c403e6SMed Ismail Bennani       result.SetStatus(eReturnStatusFailed);
1316d7c403e6SMed Ismail Bennani       return result.Succeeded();
1317d7c403e6SMed Ismail Bennani     }
1318d7c403e6SMed Ismail Bennani 
1319b9c1b51eSKate Stone     // No need to check "process" for validity as eCommandRequiresProcess
1320b9c1b51eSKate Stone     // ensures it is valid
1321f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
13227260f620SGreg Clayton     const bool only_threads_with_stop_reason = true;
13237260f620SGreg Clayton     const uint32_t start_frame = 0;
13247260f620SGreg Clayton     const uint32_t num_frames = 1;
13257260f620SGreg Clayton     const uint32_t num_frames_with_source = 1;
13266a9767c7SJim Ingham     const bool stop_format = true;
1327c14ee32dSGreg Clayton     process->GetStatus(strm);
1328b9c1b51eSKate Stone     process->GetThreadStatus(strm, only_threads_with_stop_reason, start_frame,
13296a9767c7SJim Ingham                              num_frames, num_frames_with_source, stop_format);
1330d7c403e6SMed Ismail Bennani 
1331d7c403e6SMed Ismail Bennani     if (m_options.m_verbose) {
1332d7c403e6SMed Ismail Bennani       PlatformSP platform_sp = process->GetTarget().GetPlatform();
1333d7c403e6SMed Ismail Bennani       if (!platform_sp) {
1334d7c403e6SMed Ismail Bennani         result.AppendError("Couldn'retrieve the target's platform");
1335d7c403e6SMed Ismail Bennani         result.SetStatus(eReturnStatusFailed);
13364b9bea87SJim Ingham         return result.Succeeded();
13374b9bea87SJim Ingham       }
1338d7c403e6SMed Ismail Bennani 
1339d7c403e6SMed Ismail Bennani       auto expected_crash_info =
1340eefda182SMed Ismail Bennani           platform_sp->FetchExtendedCrashInformation(*process);
1341d7c403e6SMed Ismail Bennani 
1342d7c403e6SMed Ismail Bennani       if (!expected_crash_info) {
1343d7c403e6SMed Ismail Bennani         result.AppendError(llvm::toString(expected_crash_info.takeError()));
1344d7c403e6SMed Ismail Bennani         result.SetStatus(eReturnStatusFailed);
1345d7c403e6SMed Ismail Bennani         return result.Succeeded();
1346d7c403e6SMed Ismail Bennani       }
1347d7c403e6SMed Ismail Bennani 
1348d7c403e6SMed Ismail Bennani       StructuredData::DictionarySP crash_info_sp = *expected_crash_info;
1349d7c403e6SMed Ismail Bennani 
1350d7c403e6SMed Ismail Bennani       if (crash_info_sp) {
1351d7c403e6SMed Ismail Bennani         strm.PutCString("Extended Crash Information:\n");
1352d7c403e6SMed Ismail Bennani         crash_info_sp->Dump(strm);
1353d7c403e6SMed Ismail Bennani       }
1354d7c403e6SMed Ismail Bennani     }
1355d7c403e6SMed Ismail Bennani 
1356d7c403e6SMed Ismail Bennani     return result.Succeeded();
1357d7c403e6SMed Ismail Bennani   }
1358d7c403e6SMed Ismail Bennani 
1359d7c403e6SMed Ismail Bennani private:
1360d7c403e6SMed Ismail Bennani   CommandOptions m_options;
13614b9bea87SJim Ingham };
13624b9bea87SJim Ingham 
136335731357SCaroline Tice // CommandObjectProcessHandle
1364438dfcffSRaphael Isemann #define LLDB_OPTIONS_process_handle
1365438dfcffSRaphael Isemann #include "CommandOptions.inc"
13661f0f5b5bSZachary Turner 
1367bb9caf73SJim Ingham #pragma mark CommandObjectProcessHandle
136835731357SCaroline Tice 
1369b9c1b51eSKate Stone class CommandObjectProcessHandle : public CommandObjectParsed {
137035731357SCaroline Tice public:
1371b9c1b51eSKate Stone   class CommandOptions : public Options {
137235731357SCaroline Tice   public:
1373b9c1b51eSKate Stone     CommandOptions() : Options() { OptionParsingStarting(nullptr); }
137435731357SCaroline Tice 
137549bcfd80SEugene Zelenko     ~CommandOptions() override = default;
137635731357SCaroline Tice 
137797206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
1378b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
137997206d57SZachary Turner       Status error;
13803bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
138135731357SCaroline Tice 
1382b9c1b51eSKate Stone       switch (short_option) {
138335731357SCaroline Tice       case 's':
1384adcd0268SBenjamin Kramer         stop = std::string(option_arg);
138535731357SCaroline Tice         break;
138635731357SCaroline Tice       case 'n':
1387adcd0268SBenjamin Kramer         notify = std::string(option_arg);
138835731357SCaroline Tice         break;
138935731357SCaroline Tice       case 'p':
1390adcd0268SBenjamin Kramer         pass = std::string(option_arg);
139135731357SCaroline Tice         break;
139235731357SCaroline Tice       default:
139336162014SRaphael Isemann         llvm_unreachable("Unimplemented option");
139435731357SCaroline Tice       }
139535731357SCaroline Tice       return error;
139635731357SCaroline Tice     }
139735731357SCaroline Tice 
1398b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
139935731357SCaroline Tice       stop.clear();
140035731357SCaroline Tice       notify.clear();
140135731357SCaroline Tice       pass.clear();
140235731357SCaroline Tice     }
140335731357SCaroline Tice 
14041f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
140570602439SZachary Turner       return llvm::makeArrayRef(g_process_handle_options);
14061f0f5b5bSZachary Turner     }
140735731357SCaroline Tice 
140835731357SCaroline Tice     // Instance variables to hold the values for command options.
140935731357SCaroline Tice 
141035731357SCaroline Tice     std::string stop;
141135731357SCaroline Tice     std::string notify;
141235731357SCaroline Tice     std::string pass;
141335731357SCaroline Tice   };
141435731357SCaroline Tice 
14157428a18cSKate Stone   CommandObjectProcessHandle(CommandInterpreter &interpreter)
1416b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process handle",
1417b9c1b51eSKate Stone                             "Manage LLDB handling of OS signals for the "
1418b9c1b51eSKate Stone                             "current target process.  Defaults to showing "
1419b9c1b51eSKate Stone                             "current policy.",
142004a4c091SRaphael Isemann                             nullptr, eCommandRequiresTarget),
1421b9c1b51eSKate Stone         m_options() {
1422ea671fbdSKate Stone     SetHelpLong("\nIf no signals are specified, update them all.  If no update "
1423ea671fbdSKate Stone                 "option is specified, list the current values.");
142435731357SCaroline Tice     CommandArgumentEntry arg;
1425c0dbdfb6SCaroline Tice     CommandArgumentData signal_arg;
142635731357SCaroline Tice 
1427c0dbdfb6SCaroline Tice     signal_arg.arg_type = eArgTypeUnixSignal;
1428c0dbdfb6SCaroline Tice     signal_arg.arg_repetition = eArgRepeatStar;
142935731357SCaroline Tice 
1430c0dbdfb6SCaroline Tice     arg.push_back(signal_arg);
143135731357SCaroline Tice 
143235731357SCaroline Tice     m_arguments.push_back(arg);
143335731357SCaroline Tice   }
143435731357SCaroline Tice 
143549bcfd80SEugene Zelenko   ~CommandObjectProcessHandle() override = default;
143635731357SCaroline Tice 
1437b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
143835731357SCaroline Tice 
1439b9c1b51eSKate Stone   bool VerifyCommandOptionValue(const std::string &option, int &real_value) {
144035731357SCaroline Tice     bool okay = true;
144110ad7993SCaroline Tice     bool success = false;
144247cbf4a0SPavel Labath     bool tmp_value = OptionArgParser::ToBoolean(option, false, &success);
144310ad7993SCaroline Tice 
144410ad7993SCaroline Tice     if (success && tmp_value)
144510ad7993SCaroline Tice       real_value = 1;
144610ad7993SCaroline Tice     else if (success && !tmp_value)
144710ad7993SCaroline Tice       real_value = 0;
1448b9c1b51eSKate Stone     else {
144935731357SCaroline Tice       // If the value isn't 'true' or 'false', it had better be 0 or 1.
14509010cef2SRaphael Isemann       if (!llvm::to_integer(option, real_value))
14519010cef2SRaphael Isemann         real_value = 3;
145210ad7993SCaroline Tice       if (real_value != 0 && real_value != 1)
145335731357SCaroline Tice         okay = false;
145435731357SCaroline Tice     }
145535731357SCaroline Tice 
145635731357SCaroline Tice     return okay;
145735731357SCaroline Tice   }
145835731357SCaroline Tice 
1459b9c1b51eSKate Stone   void PrintSignalHeader(Stream &str) {
146010ad7993SCaroline Tice     str.Printf("NAME         PASS   STOP   NOTIFY\n");
1461b84141a6SPavel Labath     str.Printf("===========  =====  =====  ======\n");
146210ad7993SCaroline Tice   }
146310ad7993SCaroline Tice 
1464b9c1b51eSKate Stone   void PrintSignal(Stream &str, int32_t signo, const char *sig_name,
1465b9c1b51eSKate Stone                    const UnixSignalsSP &signals_sp) {
146610ad7993SCaroline Tice     bool stop;
146710ad7993SCaroline Tice     bool suppress;
146810ad7993SCaroline Tice     bool notify;
146910ad7993SCaroline Tice 
1470b84141a6SPavel Labath     str.Printf("%-11s  ", sig_name);
1471b9c1b51eSKate Stone     if (signals_sp->GetSignalInfo(signo, suppress, stop, notify)) {
147210ad7993SCaroline Tice       bool pass = !suppress;
1473b9c1b51eSKate Stone       str.Printf("%s  %s  %s", (pass ? "true " : "false"),
1474b9c1b51eSKate Stone                  (stop ? "true " : "false"), (notify ? "true " : "false"));
147510ad7993SCaroline Tice     }
147610ad7993SCaroline Tice     str.Printf("\n");
147710ad7993SCaroline Tice   }
147810ad7993SCaroline Tice 
1479b9c1b51eSKate Stone   void PrintSignalInformation(Stream &str, Args &signal_args,
1480b9c1b51eSKate Stone                               int num_valid_signals,
1481b9c1b51eSKate Stone                               const UnixSignalsSP &signals_sp) {
148210ad7993SCaroline Tice     PrintSignalHeader(str);
148310ad7993SCaroline Tice 
1484b9c1b51eSKate Stone     if (num_valid_signals > 0) {
148510ad7993SCaroline Tice       size_t num_args = signal_args.GetArgumentCount();
1486b9c1b51eSKate Stone       for (size_t i = 0; i < num_args; ++i) {
1487b9c1b51eSKate Stone         int32_t signo = signals_sp->GetSignalNumberFromName(
1488b9c1b51eSKate Stone             signal_args.GetArgumentAtIndex(i));
148910ad7993SCaroline Tice         if (signo != LLDB_INVALID_SIGNAL_NUMBER)
1490b9c1b51eSKate Stone           PrintSignal(str, signo, signal_args.GetArgumentAtIndex(i),
1491b9c1b51eSKate Stone                       signals_sp);
149210ad7993SCaroline Tice       }
1493b9c1b51eSKate Stone     } else // Print info for ALL signals
149410ad7993SCaroline Tice     {
149598d0a4b3SChaoren Lin       int32_t signo = signals_sp->GetFirstSignalNumber();
1496b9c1b51eSKate Stone       while (signo != LLDB_INVALID_SIGNAL_NUMBER) {
1497b9c1b51eSKate Stone         PrintSignal(str, signo, signals_sp->GetSignalAsCString(signo),
1498b9c1b51eSKate Stone                     signals_sp);
149998d0a4b3SChaoren Lin         signo = signals_sp->GetNextSignalNumber(signo);
150010ad7993SCaroline Tice       }
150110ad7993SCaroline Tice     }
150210ad7993SCaroline Tice   }
150310ad7993SCaroline Tice 
15045a988416SJim Ingham protected:
1505b9c1b51eSKate Stone   bool DoExecute(Args &signal_args, CommandReturnObject &result) override {
150604a4c091SRaphael Isemann     Target *target_sp = &GetSelectedTarget();
150735731357SCaroline Tice 
150835731357SCaroline Tice     ProcessSP process_sp = target_sp->GetProcessSP();
150935731357SCaroline Tice 
1510b9c1b51eSKate Stone     if (!process_sp) {
1511b9c1b51eSKate Stone       result.AppendError("No current process; cannot handle signals until you "
1512b9c1b51eSKate Stone                          "have a valid process.\n");
151335731357SCaroline Tice       result.SetStatus(eReturnStatusFailed);
151435731357SCaroline Tice       return false;
151535731357SCaroline Tice     }
151635731357SCaroline Tice 
151735731357SCaroline Tice     int stop_action = -1;   // -1 means leave the current setting alone
151835731357SCaroline Tice     int pass_action = -1;   // -1 means leave the current setting alone
151935731357SCaroline Tice     int notify_action = -1; // -1 means leave the current setting alone
152035731357SCaroline Tice 
1521b9c1b51eSKate Stone     if (!m_options.stop.empty() &&
1522b9c1b51eSKate Stone         !VerifyCommandOptionValue(m_options.stop, stop_action)) {
1523b9c1b51eSKate Stone       result.AppendError("Invalid argument for command option --stop; must be "
1524b9c1b51eSKate Stone                          "true or false.\n");
152535731357SCaroline Tice       result.SetStatus(eReturnStatusFailed);
152635731357SCaroline Tice       return false;
152735731357SCaroline Tice     }
152835731357SCaroline Tice 
1529b9c1b51eSKate Stone     if (!m_options.notify.empty() &&
1530b9c1b51eSKate Stone         !VerifyCommandOptionValue(m_options.notify, notify_action)) {
1531b9c1b51eSKate Stone       result.AppendError("Invalid argument for command option --notify; must "
1532b9c1b51eSKate Stone                          "be true or false.\n");
153335731357SCaroline Tice       result.SetStatus(eReturnStatusFailed);
153435731357SCaroline Tice       return false;
153535731357SCaroline Tice     }
153635731357SCaroline Tice 
1537b9c1b51eSKate Stone     if (!m_options.pass.empty() &&
1538b9c1b51eSKate Stone         !VerifyCommandOptionValue(m_options.pass, pass_action)) {
1539b9c1b51eSKate Stone       result.AppendError("Invalid argument for command option --pass; must be "
1540b9c1b51eSKate Stone                          "true or false.\n");
154135731357SCaroline Tice       result.SetStatus(eReturnStatusFailed);
154235731357SCaroline Tice       return false;
154335731357SCaroline Tice     }
154435731357SCaroline Tice 
154535731357SCaroline Tice     size_t num_args = signal_args.GetArgumentCount();
154698d0a4b3SChaoren Lin     UnixSignalsSP signals_sp = process_sp->GetUnixSignals();
154735731357SCaroline Tice     int num_signals_set = 0;
154835731357SCaroline Tice 
1549b9c1b51eSKate Stone     if (num_args > 0) {
1550d6a24757SZachary Turner       for (const auto &arg : signal_args) {
1551d6a24757SZachary Turner         int32_t signo = signals_sp->GetSignalNumberFromName(arg.c_str());
1552b9c1b51eSKate Stone         if (signo != LLDB_INVALID_SIGNAL_NUMBER) {
1553b9c1b51eSKate Stone           // Casting the actions as bools here should be okay, because
155405097246SAdrian Prantl           // VerifyCommandOptionValue guarantees the value is either 0 or 1.
155535731357SCaroline Tice           if (stop_action != -1)
155698d0a4b3SChaoren Lin             signals_sp->SetShouldStop(signo, stop_action);
1557b9c1b51eSKate Stone           if (pass_action != -1) {
155898d0a4b3SChaoren Lin             bool suppress = !pass_action;
155998d0a4b3SChaoren Lin             signals_sp->SetShouldSuppress(signo, suppress);
156035731357SCaroline Tice           }
156135731357SCaroline Tice           if (notify_action != -1)
156298d0a4b3SChaoren Lin             signals_sp->SetShouldNotify(signo, notify_action);
156335731357SCaroline Tice           ++num_signals_set;
1564b9c1b51eSKate Stone         } else {
1565b9c1b51eSKate Stone           result.AppendErrorWithFormat("Invalid signal name '%s'\n",
1566d6a24757SZachary Turner                                        arg.c_str());
156735731357SCaroline Tice         }
156835731357SCaroline Tice       }
1569b9c1b51eSKate Stone     } else {
1570b9c1b51eSKate Stone       // No signal specified, if any command options were specified, update ALL
1571b9c1b51eSKate Stone       // signals.
1572b9c1b51eSKate Stone       if ((notify_action != -1) || (stop_action != -1) || (pass_action != -1)) {
1573b9c1b51eSKate Stone         if (m_interpreter.Confirm(
1574b9c1b51eSKate Stone                 "Do you really want to update all the signals?", false)) {
157598d0a4b3SChaoren Lin           int32_t signo = signals_sp->GetFirstSignalNumber();
1576b9c1b51eSKate Stone           while (signo != LLDB_INVALID_SIGNAL_NUMBER) {
157710ad7993SCaroline Tice             if (notify_action != -1)
157898d0a4b3SChaoren Lin               signals_sp->SetShouldNotify(signo, notify_action);
157910ad7993SCaroline Tice             if (stop_action != -1)
158098d0a4b3SChaoren Lin               signals_sp->SetShouldStop(signo, stop_action);
1581b9c1b51eSKate Stone             if (pass_action != -1) {
158298d0a4b3SChaoren Lin               bool suppress = !pass_action;
158398d0a4b3SChaoren Lin               signals_sp->SetShouldSuppress(signo, suppress);
158410ad7993SCaroline Tice             }
158598d0a4b3SChaoren Lin             signo = signals_sp->GetNextSignalNumber(signo);
158610ad7993SCaroline Tice           }
158710ad7993SCaroline Tice         }
158810ad7993SCaroline Tice       }
158910ad7993SCaroline Tice     }
159010ad7993SCaroline Tice 
1591b9c1b51eSKate Stone     PrintSignalInformation(result.GetOutputStream(), signal_args,
1592b9c1b51eSKate Stone                            num_signals_set, signals_sp);
159335731357SCaroline Tice 
159435731357SCaroline Tice     if (num_signals_set > 0)
159535731357SCaroline Tice       result.SetStatus(eReturnStatusSuccessFinishNoResult);
159635731357SCaroline Tice     else
159735731357SCaroline Tice       result.SetStatus(eReturnStatusFailed);
159835731357SCaroline Tice 
159935731357SCaroline Tice     return result.Succeeded();
160035731357SCaroline Tice   }
160135731357SCaroline Tice 
160235731357SCaroline Tice   CommandOptions m_options;
160335731357SCaroline Tice };
160435731357SCaroline Tice 
160530fdc8d8SChris Lattner // CommandObjectMultiwordProcess
160630fdc8d8SChris Lattner 
1607b9c1b51eSKate Stone CommandObjectMultiwordProcess::CommandObjectMultiwordProcess(
1608b9c1b51eSKate Stone     CommandInterpreter &interpreter)
1609b9c1b51eSKate Stone     : CommandObjectMultiword(
1610b9c1b51eSKate Stone           interpreter, "process",
1611b9c1b51eSKate Stone           "Commands for interacting with processes on the current platform.",
1612b9c1b51eSKate Stone           "process <subcommand> [<subcommand-options>]") {
1613b9c1b51eSKate Stone   LoadSubCommand("attach",
1614b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessAttach(interpreter)));
1615b9c1b51eSKate Stone   LoadSubCommand("launch",
1616b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessLaunch(interpreter)));
1617b9c1b51eSKate Stone   LoadSubCommand("continue", CommandObjectSP(new CommandObjectProcessContinue(
1618b9c1b51eSKate Stone                                  interpreter)));
1619b9c1b51eSKate Stone   LoadSubCommand("connect",
1620b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessConnect(interpreter)));
1621b9c1b51eSKate Stone   LoadSubCommand("detach",
1622b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessDetach(interpreter)));
1623b9c1b51eSKate Stone   LoadSubCommand("load",
1624b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessLoad(interpreter)));
1625b9c1b51eSKate Stone   LoadSubCommand("unload",
1626b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessUnload(interpreter)));
1627b9c1b51eSKate Stone   LoadSubCommand("signal",
1628b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessSignal(interpreter)));
1629b9c1b51eSKate Stone   LoadSubCommand("handle",
1630b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessHandle(interpreter)));
1631b9c1b51eSKate Stone   LoadSubCommand("status",
1632b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessStatus(interpreter)));
1633b9c1b51eSKate Stone   LoadSubCommand("interrupt", CommandObjectSP(new CommandObjectProcessInterrupt(
1634b9c1b51eSKate Stone                                   interpreter)));
1635b9c1b51eSKate Stone   LoadSubCommand("kill",
1636b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessKill(interpreter)));
1637b9c1b51eSKate Stone   LoadSubCommand("plugin",
1638b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessPlugin(interpreter)));
1639b9c1b51eSKate Stone   LoadSubCommand("save-core", CommandObjectSP(new CommandObjectProcessSaveCore(
1640b9c1b51eSKate Stone                                   interpreter)));
164130fdc8d8SChris Lattner }
164230fdc8d8SChris Lattner 
164349bcfd80SEugene Zelenko CommandObjectMultiwordProcess::~CommandObjectMultiwordProcess() = default;
1644