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"
100b697561SWalter Erquinigo #include "CommandObjectTrace.h"
117169d3a3SMed Ismail Bennani #include "CommandOptionsProcessLaunch.h"
120e41084aSJim Ingham #include "lldb/Breakpoint/Breakpoint.h"
130e41084aSJim Ingham #include "lldb/Breakpoint/BreakpointLocation.h"
140e41084aSJim Ingham #include "lldb/Breakpoint/BreakpointSite.h"
151f746071SGreg Clayton #include "lldb/Core/Module.h"
16a2715cf1SGreg Clayton #include "lldb/Core/PluginManager.h"
173eb2b44dSZachary Turner #include "lldb/Host/OptionParser.h"
1830fdc8d8SChris Lattner #include "lldb/Interpreter/CommandInterpreter.h"
1930fdc8d8SChris Lattner #include "lldb/Interpreter/CommandReturnObject.h"
2047cbf4a0SPavel Labath #include "lldb/Interpreter/OptionArgParser.h"
213e0ad115SMed Ismail Bennani #include "lldb/Interpreter/OptionGroupPythonClassWithDict.h"
22b9c1b51eSKate Stone #include "lldb/Interpreter/Options.h"
23e996fd30SGreg Clayton #include "lldb/Target/Platform.h"
2430fdc8d8SChris Lattner #include "lldb/Target/Process.h"
250e41084aSJim Ingham #include "lldb/Target/StopInfo.h"
2630fdc8d8SChris Lattner #include "lldb/Target/Target.h"
2730fdc8d8SChris Lattner #include "lldb/Target/Thread.h"
2893749ab3SZachary Turner #include "lldb/Target/UnixSignals.h"
29145d95c9SPavel Labath #include "lldb/Utility/Args.h"
30d821c997SPavel Labath #include "lldb/Utility/State.h"
3130fdc8d8SChris Lattner 
3230fdc8d8SChris Lattner using namespace lldb;
3330fdc8d8SChris Lattner using namespace lldb_private;
3430fdc8d8SChris Lattner 
35b9c1b51eSKate Stone class CommandObjectProcessLaunchOrAttach : public CommandObjectParsed {
36dcb1d856SJim Ingham public:
37dcb1d856SJim Ingham   CommandObjectProcessLaunchOrAttach(CommandInterpreter &interpreter,
38b9c1b51eSKate Stone                                      const char *name, const char *help,
39b9c1b51eSKate Stone                                      const char *syntax, uint32_t flags,
40b9c1b51eSKate Stone                                      const char *new_process_action)
41b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, name, help, syntax, flags),
42dcb1d856SJim Ingham         m_new_process_action(new_process_action) {}
43dcb1d856SJim Ingham 
4449bcfd80SEugene Zelenko   ~CommandObjectProcessLaunchOrAttach() override = default;
4549bcfd80SEugene Zelenko 
46dcb1d856SJim Ingham protected:
47b9c1b51eSKate Stone   bool StopProcessIfNecessary(Process *process, StateType &state,
48b9c1b51eSKate Stone                               CommandReturnObject &result) {
49dcb1d856SJim Ingham     state = eStateInvalid;
50b9c1b51eSKate Stone     if (process) {
51dcb1d856SJim Ingham       state = process->GetState();
52dcb1d856SJim Ingham 
53b9c1b51eSKate Stone       if (process->IsAlive() && state != eStateConnected) {
54db203e02SRaphael Isemann         std::string message;
55dcb1d856SJim Ingham         if (process->GetState() == eStateAttaching)
56db203e02SRaphael Isemann           message =
57db203e02SRaphael Isemann               llvm::formatv("There is a pending attach, abort it and {0}?",
58db203e02SRaphael Isemann                             m_new_process_action);
59dcb1d856SJim Ingham         else if (process->GetShouldDetach())
60db203e02SRaphael Isemann           message = llvm::formatv(
61db203e02SRaphael Isemann               "There is a running process, detach from it and {0}?",
62db203e02SRaphael Isemann               m_new_process_action);
63dcb1d856SJim Ingham         else
64db203e02SRaphael Isemann           message =
65db203e02SRaphael Isemann               llvm::formatv("There is a running process, kill it and {0}?",
66db203e02SRaphael Isemann                             m_new_process_action);
67dcb1d856SJim Ingham 
68b9c1b51eSKate Stone         if (!m_interpreter.Confirm(message, true)) {
69dcb1d856SJim Ingham           result.SetStatus(eReturnStatusFailed);
70dcb1d856SJim Ingham           return false;
71b9c1b51eSKate Stone         } else {
72b9c1b51eSKate Stone           if (process->GetShouldDetach()) {
73acff8950SJim Ingham             bool keep_stopped = false;
7497206d57SZachary Turner             Status detach_error(process->Detach(keep_stopped));
75b9c1b51eSKate Stone             if (detach_error.Success()) {
76dcb1d856SJim Ingham               result.SetStatus(eReturnStatusSuccessFinishResult);
7749bcfd80SEugene Zelenko               process = nullptr;
78b9c1b51eSKate Stone             } else {
79b9c1b51eSKate Stone               result.AppendErrorWithFormat(
80b9c1b51eSKate Stone                   "Failed to detach from process: %s\n",
81b9c1b51eSKate Stone                   detach_error.AsCString());
82dcb1d856SJim Ingham             }
83b9c1b51eSKate Stone           } else {
8497206d57SZachary Turner             Status destroy_error(process->Destroy(false));
85b9c1b51eSKate Stone             if (destroy_error.Success()) {
86dcb1d856SJim Ingham               result.SetStatus(eReturnStatusSuccessFinishResult);
8749bcfd80SEugene Zelenko               process = nullptr;
88b9c1b51eSKate Stone             } else {
89b9c1b51eSKate Stone               result.AppendErrorWithFormat("Failed to kill process: %s\n",
90b9c1b51eSKate Stone                                            destroy_error.AsCString());
91dcb1d856SJim Ingham             }
92dcb1d856SJim Ingham           }
93dcb1d856SJim Ingham         }
94dcb1d856SJim Ingham       }
95dcb1d856SJim Ingham     }
96dcb1d856SJim Ingham     return result.Succeeded();
97dcb1d856SJim Ingham   }
9849bcfd80SEugene Zelenko 
99dcb1d856SJim Ingham   std::string m_new_process_action;
100dcb1d856SJim Ingham };
10149bcfd80SEugene Zelenko 
10230fdc8d8SChris Lattner // CommandObjectProcessLaunch
1034bddaeb5SJim Ingham #pragma mark CommandObjectProcessLaunch
104b9c1b51eSKate Stone class CommandObjectProcessLaunch : public CommandObjectProcessLaunchOrAttach {
10530fdc8d8SChris Lattner public:
106b9c1b51eSKate Stone   CommandObjectProcessLaunch(CommandInterpreter &interpreter)
107b9c1b51eSKate Stone       : CommandObjectProcessLaunchOrAttach(
108b9c1b51eSKate Stone             interpreter, "process launch",
109b9c1b51eSKate Stone             "Launch the executable in the debugger.", nullptr,
110b9c1b51eSKate Stone             eCommandRequiresTarget, "restart"),
1113e0ad115SMed Ismail Bennani         m_options(),
1123e0ad115SMed Ismail Bennani         m_class_options("scripted process", true, 'C', 'k', 'v', 0),
1133e0ad115SMed Ismail Bennani         m_all_options() {
1143e0ad115SMed Ismail Bennani     m_all_options.Append(&m_options);
1153e0ad115SMed Ismail Bennani     m_all_options.Append(&m_class_options, LLDB_OPT_SET_1 | LLDB_OPT_SET_2,
1163e0ad115SMed Ismail Bennani                          LLDB_OPT_SET_ALL);
1173e0ad115SMed Ismail Bennani     m_all_options.Finalize();
1183e0ad115SMed Ismail Bennani 
119405fe67fSCaroline Tice     CommandArgumentEntry arg;
120405fe67fSCaroline Tice     CommandArgumentData run_args_arg;
121405fe67fSCaroline Tice 
122405fe67fSCaroline Tice     // Define the first (and only) variant of this arg.
123405fe67fSCaroline Tice     run_args_arg.arg_type = eArgTypeRunArgs;
124405fe67fSCaroline Tice     run_args_arg.arg_repetition = eArgRepeatOptional;
125405fe67fSCaroline Tice 
126b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
127b9c1b51eSKate Stone     // argument entry.
128405fe67fSCaroline Tice     arg.push_back(run_args_arg);
129405fe67fSCaroline Tice 
130405fe67fSCaroline Tice     // Push the data for the first argument into the m_arguments vector.
131405fe67fSCaroline Tice     m_arguments.push_back(arg);
13230fdc8d8SChris Lattner   }
13330fdc8d8SChris Lattner 
13449bcfd80SEugene Zelenko   ~CommandObjectProcessLaunch() override = default;
13530fdc8d8SChris Lattner 
136ae34ed2cSRaphael Isemann   void
137ae34ed2cSRaphael Isemann   HandleArgumentCompletion(CompletionRequest &request,
1382443bbd4SRaphael Isemann                            OptionElementVector &opt_element_vector) override {
139e9ce62b6SJim Ingham 
140b9c1b51eSKate Stone     CommandCompletions::InvokeCommonCompletionCallbacks(
141b9c1b51eSKate Stone         GetCommandInterpreter(), CommandCompletions::eDiskFileCompletion,
142a2e76c0bSRaphael Isemann         request, nullptr);
143e9ce62b6SJim Ingham   }
144e9ce62b6SJim Ingham 
1453e0ad115SMed Ismail Bennani   Options *GetOptions() override { return &m_all_options; }
14630fdc8d8SChris Lattner 
147b9c1b51eSKate Stone   const char *GetRepeatCommand(Args &current_command_args,
148b9c1b51eSKate Stone                                uint32_t index) override {
1495a988416SJim Ingham     // No repeat for "process launch"...
1505a988416SJim Ingham     return "";
1515a988416SJim Ingham   }
1525a988416SJim Ingham 
1535a988416SJim Ingham protected:
154b9c1b51eSKate Stone   bool DoExecute(Args &launch_args, CommandReturnObject &result) override {
15557179860SJonas Devlieghere     Debugger &debugger = GetDebugger();
1561d885966SGreg Clayton     Target *target = debugger.GetSelectedTarget().get();
15749bcfd80SEugene Zelenko     // If our listener is nullptr, users aren't allows to launch
158b09c5384SGreg Clayton     ModuleSP exe_module_sp = target->GetExecutableModule();
15971337622SGreg Clayton 
160b9c1b51eSKate Stone     if (exe_module_sp == nullptr) {
161b9c1b51eSKate Stone       result.AppendError("no file in target, create a debug target using the "
162b9c1b51eSKate Stone                          "'target create' command");
16371337622SGreg Clayton       return false;
16471337622SGreg Clayton     }
16571337622SGreg Clayton 
16671337622SGreg Clayton     StateType state = eStateInvalid;
16771337622SGreg Clayton 
168b09c5384SGreg Clayton     if (!StopProcessIfNecessary(m_exe_ctx.GetProcessPtr(), state, result))
16930fdc8d8SChris Lattner       return false;
17030fdc8d8SChris Lattner 
17131d97a5cSZachary Turner     llvm::StringRef target_settings_argv0 = target->GetArg0();
17245392553SGreg Clayton 
173b9c1b51eSKate Stone     // Determine whether we will disable ASLR or leave it in the default state
17405097246SAdrian Prantl     // (i.e. enabled if the platform supports it). First check if the process
17505097246SAdrian Prantl     // launch options explicitly turn on/off
176b9c1b51eSKate Stone     // disabling ASLR.  If so, use that setting;
1775163792bSTodd Fiala     // otherwise, use the 'settings target.disable-aslr' setting.
1785163792bSTodd Fiala     bool disable_aslr = false;
179b9c1b51eSKate Stone     if (m_options.disable_aslr != eLazyBoolCalculate) {
18005097246SAdrian Prantl       // The user specified an explicit setting on the process launch line.
18105097246SAdrian Prantl       // Use it.
1825163792bSTodd Fiala       disable_aslr = (m_options.disable_aslr == eLazyBoolYes);
183b9c1b51eSKate Stone     } else {
18405097246SAdrian Prantl       // The user did not explicitly specify whether to disable ASLR.  Fall
18505097246SAdrian Prantl       // back to the target.disable-aslr setting.
1865163792bSTodd Fiala       disable_aslr = target->GetDisableASLR();
1875163792bSTodd Fiala     }
1885163792bSTodd Fiala 
1893e0ad115SMed Ismail Bennani     if (!m_class_options.GetName().empty()) {
1903e0ad115SMed Ismail Bennani       m_options.launch_info.SetProcessPluginName("ScriptedProcess");
1913e0ad115SMed Ismail Bennani       m_options.launch_info.SetScriptedProcessClassName(
1923e0ad115SMed Ismail Bennani           m_class_options.GetName());
1933e0ad115SMed Ismail Bennani       m_options.launch_info.SetScriptedProcessDictionarySP(
1943e0ad115SMed Ismail Bennani           m_class_options.GetStructuredData());
1953e0ad115SMed Ismail Bennani       target->SetProcessLaunchInfo(m_options.launch_info);
1963e0ad115SMed Ismail Bennani     }
1973e0ad115SMed Ismail Bennani 
1985163792bSTodd Fiala     if (disable_aslr)
199b09c5384SGreg Clayton       m_options.launch_info.GetFlags().Set(eLaunchFlagDisableASLR);
2005163792bSTodd Fiala     else
2015163792bSTodd Fiala       m_options.launch_info.GetFlags().Clear(eLaunchFlagDisableASLR);
202b09c5384SGreg Clayton 
203249a1d4fSJonas Devlieghere     if (target->GetInheritTCC())
204249a1d4fSJonas Devlieghere       m_options.launch_info.GetFlags().Set(eLaunchFlagInheritTCCFromParent);
205249a1d4fSJonas Devlieghere 
206106d0286SJim Ingham     if (target->GetDetachOnError())
207106d0286SJim Ingham       m_options.launch_info.GetFlags().Set(eLaunchFlagDetachOnError);
208106d0286SJim Ingham 
209b09c5384SGreg Clayton     if (target->GetDisableSTDIO())
210b09c5384SGreg Clayton       m_options.launch_info.GetFlags().Set(eLaunchFlagDisableSTDIO);
211b09c5384SGreg Clayton 
212ae54fc9fSJonas Devlieghere     // Merge the launch info environment with the target environment.
213ae54fc9fSJonas Devlieghere     Environment target_env = target->GetEnvironment();
214ae54fc9fSJonas Devlieghere     m_options.launch_info.GetEnvironment().insert(target_env.begin(),
215ae54fc9fSJonas Devlieghere                                                   target_env.end());
21645392553SGreg Clayton 
21731d97a5cSZachary Turner     if (!target_settings_argv0.empty()) {
218b9c1b51eSKate Stone       m_options.launch_info.GetArguments().AppendArgument(
21931d97a5cSZachary Turner           target_settings_argv0);
220b9c1b51eSKate Stone       m_options.launch_info.SetExecutableFile(
221b9c1b51eSKate Stone           exe_module_sp->GetPlatformFileSpec(), false);
222b9c1b51eSKate Stone     } else {
223b9c1b51eSKate Stone       m_options.launch_info.SetExecutableFile(
224b9c1b51eSKate Stone           exe_module_sp->GetPlatformFileSpec(), true);
22545392553SGreg Clayton     }
22645392553SGreg Clayton 
227b9c1b51eSKate Stone     if (launch_args.GetArgumentCount() == 0) {
228b9c1b51eSKate Stone       m_options.launch_info.GetArguments().AppendArguments(
229b9c1b51eSKate Stone           target->GetProcessLaunchInfo().GetArguments());
230b9c1b51eSKate Stone     } else {
23145392553SGreg Clayton       m_options.launch_info.GetArguments().AppendArguments(launch_args);
232162b597cSGreg Clayton       // Save the arguments for subsequent runs in the current target.
233162b597cSGreg Clayton       target->SetRunArguments(launch_args);
234982c9762SGreg Clayton     }
2351d885966SGreg Clayton 
236dc6224e0SGreg Clayton     StreamString stream;
23797206d57SZachary Turner     Status error = target->Launch(m_options.launch_info, &stream);
23830fdc8d8SChris Lattner 
239b9c1b51eSKate Stone     if (error.Success()) {
240b09c5384SGreg Clayton       ProcessSP process_sp(target->GetProcessSP());
241b9c1b51eSKate Stone       if (process_sp) {
242b9c1b51eSKate Stone         // There is a race condition where this thread will return up the call
24305097246SAdrian Prantl         // stack to the main command handler and show an (lldb) prompt before
24405097246SAdrian Prantl         // HandlePrivateEvent (from PrivateStateThread) has a chance to call
24505097246SAdrian Prantl         // PushProcessIOHandler().
2463879fe00SPavel Labath         process_sp->SyncIOHandler(0, std::chrono::seconds(2));
2478f0db3e1SIlia K 
248c156427dSZachary Turner         llvm::StringRef data = stream.GetString();
249c156427dSZachary Turner         if (!data.empty())
250c156427dSZachary Turner           result.AppendMessage(data);
251b9c1b51eSKate Stone         const char *archname =
252b9c1b51eSKate Stone             exe_module_sp->GetArchitecture().GetArchitectureName();
253b9c1b51eSKate Stone         result.AppendMessageWithFormat(
254b9c1b51eSKate Stone             "Process %" PRIu64 " launched: '%s' (%s)\n", process_sp->GetID(),
255b9c1b51eSKate Stone             exe_module_sp->GetFileSpec().GetPath().c_str(), archname);
25605faeb71SGreg Clayton         result.SetStatus(eReturnStatusSuccessFinishResult);
257b09c5384SGreg Clayton         result.SetDidChangeProcessState(true);
258b9c1b51eSKate Stone       } else {
259b9c1b51eSKate Stone         result.AppendError(
260b9c1b51eSKate Stone             "no error returned from Target::Launch, and target has no process");
26130fdc8d8SChris Lattner       }
262b9c1b51eSKate Stone     } else {
263b09c5384SGreg Clayton       result.AppendError(error.AsCString());
264514487e8SGreg Clayton     }
26530fdc8d8SChris Lattner     return result.Succeeded();
26630fdc8d8SChris Lattner   }
26730fdc8d8SChris Lattner 
2687169d3a3SMed Ismail Bennani   CommandOptionsProcessLaunch m_options;
2693e0ad115SMed Ismail Bennani   OptionGroupPythonClassWithDict m_class_options;
2703e0ad115SMed Ismail Bennani   OptionGroupOptions m_all_options;
27130fdc8d8SChris Lattner };
27230fdc8d8SChris Lattner 
273438dfcffSRaphael Isemann #define LLDB_OPTIONS_process_attach
274438dfcffSRaphael Isemann #include "CommandOptions.inc"
2751f0f5b5bSZachary Turner 
276bb9caf73SJim Ingham #pragma mark CommandObjectProcessAttach
277b9c1b51eSKate Stone class CommandObjectProcessAttach : public CommandObjectProcessLaunchOrAttach {
27830fdc8d8SChris Lattner public:
279b9c1b51eSKate Stone   class CommandOptions : public Options {
28030fdc8d8SChris Lattner   public:
281b9c1b51eSKate Stone     CommandOptions() : Options() {
282b9c1b51eSKate Stone       // Keep default values of all options in one place: OptionParsingStarting
283b9c1b51eSKate Stone       // ()
284e1cfbc79STodd Fiala       OptionParsingStarting(nullptr);
28530fdc8d8SChris Lattner     }
28630fdc8d8SChris Lattner 
28749bcfd80SEugene Zelenko     ~CommandOptions() override = default;
28830fdc8d8SChris Lattner 
28997206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
290b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
29197206d57SZachary Turner       Status error;
2923bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
293b9c1b51eSKate Stone       switch (short_option) {
294a95ce623SJohnny Chen       case 'c':
295a95ce623SJohnny Chen         attach_info.SetContinueOnceAttached(true);
296a95ce623SJohnny Chen         break;
297a95ce623SJohnny Chen 
298b9c1b51eSKate Stone       case 'p': {
299fe11483bSZachary Turner         lldb::pid_t pid;
300fe11483bSZachary Turner         if (option_arg.getAsInteger(0, pid)) {
301fe11483bSZachary Turner           error.SetErrorStringWithFormat("invalid process ID '%s'",
302fe11483bSZachary Turner                                          option_arg.str().c_str());
303b9c1b51eSKate Stone         } else {
304144f3a9cSGreg Clayton           attach_info.SetProcessID(pid);
305144f3a9cSGreg Clayton         }
306b9c1b51eSKate Stone       } break;
30730fdc8d8SChris Lattner 
30830fdc8d8SChris Lattner       case 'P':
309144f3a9cSGreg Clayton         attach_info.SetProcessPluginName(option_arg);
31030fdc8d8SChris Lattner         break;
31130fdc8d8SChris Lattner 
31230fdc8d8SChris Lattner       case 'n':
3138f3be7a3SJonas Devlieghere         attach_info.GetExecutableFile().SetFile(option_arg,
314937348cdSJonas Devlieghere                                                 FileSpec::Style::native);
31530fdc8d8SChris Lattner         break;
31630fdc8d8SChris Lattner 
31730fdc8d8SChris Lattner       case 'w':
318144f3a9cSGreg Clayton         attach_info.SetWaitForLaunch(true);
31930fdc8d8SChris Lattner         break;
32030fdc8d8SChris Lattner 
321cd16df91SJim Ingham       case 'i':
322cd16df91SJim Ingham         attach_info.SetIgnoreExisting(false);
323cd16df91SJim Ingham         break;
324cd16df91SJim Ingham 
32530fdc8d8SChris Lattner       default:
32636162014SRaphael Isemann         llvm_unreachable("Unimplemented option");
32730fdc8d8SChris Lattner       }
32830fdc8d8SChris Lattner       return error;
32930fdc8d8SChris Lattner     }
33030fdc8d8SChris Lattner 
331b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
332144f3a9cSGreg Clayton       attach_info.Clear();
33330fdc8d8SChris Lattner     }
33430fdc8d8SChris Lattner 
3351f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
33670602439SZachary Turner       return llvm::makeArrayRef(g_process_attach_options);
3371f0f5b5bSZachary Turner     }
33830fdc8d8SChris Lattner 
339144f3a9cSGreg Clayton     ProcessAttachInfo attach_info;
34030fdc8d8SChris Lattner   };
34130fdc8d8SChris Lattner 
342b9c1b51eSKate Stone   CommandObjectProcessAttach(CommandInterpreter &interpreter)
343b9c1b51eSKate Stone       : CommandObjectProcessLaunchOrAttach(
344b9c1b51eSKate Stone             interpreter, "process attach", "Attach to a process.",
345b9c1b51eSKate Stone             "process attach <cmd-options>", 0, "attach"),
346b9c1b51eSKate Stone         m_options() {}
3475aee162fSJim Ingham 
34849bcfd80SEugene Zelenko   ~CommandObjectProcessAttach() override = default;
3495aee162fSJim Ingham 
350b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
3515a988416SJim Ingham 
3525a988416SJim Ingham protected:
353b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
354b9c1b51eSKate Stone     PlatformSP platform_sp(
35557179860SJonas Devlieghere         GetDebugger().GetPlatformList().GetSelectedPlatform());
356926af0cdSOleksiy Vyalov 
35757179860SJonas Devlieghere     Target *target = GetDebugger().GetSelectedTarget().get();
358b9c1b51eSKate Stone     // N.B. The attach should be synchronous.  It doesn't help much to get the
35905097246SAdrian Prantl     // prompt back between initiating the attach and the target actually
36005097246SAdrian Prantl     // stopping.  So even if the interpreter is set to be asynchronous, we wait
36105097246SAdrian Prantl     // for the stop ourselves here.
3625aee162fSJim Ingham 
36371337622SGreg Clayton     StateType state = eStateInvalid;
364dcb1d856SJim Ingham     Process *process = m_exe_ctx.GetProcessPtr();
365dcb1d856SJim Ingham 
366dcb1d856SJim Ingham     if (!StopProcessIfNecessary(process, state, result))
3675aee162fSJim Ingham       return false;
3685aee162fSJim Ingham 
369b9c1b51eSKate Stone     if (target == nullptr) {
3705aee162fSJim Ingham       // If there isn't a current target create one.
3715aee162fSJim Ingham       TargetSP new_target_sp;
37297206d57SZachary Turner       Status error;
3735aee162fSJim Ingham 
37457179860SJonas Devlieghere       error = GetDebugger().GetTargetList().CreateTarget(
37557179860SJonas Devlieghere           GetDebugger(), "", "", eLoadDependentsNo,
37649bcfd80SEugene Zelenko           nullptr, // No platform options
3775aee162fSJim Ingham           new_target_sp);
3785aee162fSJim Ingham       target = new_target_sp.get();
379b9c1b51eSKate Stone       if (target == nullptr || error.Fail()) {
380b766a73dSGreg Clayton         result.AppendError(error.AsCString("Error creating target"));
3815aee162fSJim Ingham         return false;
3825aee162fSJim Ingham       }
3835aee162fSJim Ingham     }
3845aee162fSJim Ingham 
385b9c1b51eSKate Stone     // Record the old executable module, we want to issue a warning if the
38605097246SAdrian Prantl     // process of attaching changed the current executable (like somebody said
38705097246SAdrian Prantl     // "file foo" then attached to a PID whose executable was bar.)
3885aee162fSJim Ingham 
3895aee162fSJim Ingham     ModuleSP old_exec_module_sp = target->GetExecutableModule();
3905aee162fSJim Ingham     ArchSpec old_arch_spec = target->GetArchitecture();
3915aee162fSJim Ingham 
392b9c1b51eSKate Stone     if (command.GetArgumentCount()) {
393b9c1b51eSKate Stone       result.AppendErrorWithFormat("Invalid arguments for '%s'.\nUsage: %s\n",
394b9c1b51eSKate Stone                                    m_cmd_name.c_str(), m_cmd_syntax.c_str());
395926af0cdSOleksiy Vyalov       return false;
39671337622SGreg Clayton     }
3975aee162fSJim Ingham 
39837386143SOleksiy Vyalov     StreamString stream;
39937386143SOleksiy Vyalov     const auto error = target->Attach(m_options.attach_info, &stream);
400b9c1b51eSKate Stone     if (error.Success()) {
40137386143SOleksiy Vyalov       ProcessSP process_sp(target->GetProcessSP());
402b9c1b51eSKate Stone       if (process_sp) {
403c156427dSZachary Turner         result.AppendMessage(stream.GetString());
404bb3a283bSJim Ingham         result.SetStatus(eReturnStatusSuccessFinishNoResult);
40537386143SOleksiy Vyalov         result.SetDidChangeProcessState(true);
406b9c1b51eSKate Stone       } else {
407b9c1b51eSKate Stone         result.AppendError(
408b9c1b51eSKate Stone             "no error returned from Target::Attach, and target has no process");
40944d93782SGreg Clayton       }
410b9c1b51eSKate Stone     } else {
41144d93782SGreg Clayton       result.AppendErrorWithFormat("attach failed: %s\n", error.AsCString());
412aa739093SJohnny Chen     }
4135aee162fSJim Ingham 
414926af0cdSOleksiy Vyalov     if (!result.Succeeded())
415926af0cdSOleksiy Vyalov       return false;
416926af0cdSOleksiy Vyalov 
417b9c1b51eSKate Stone     // Okay, we're done.  Last step is to warn if the executable module has
418b9c1b51eSKate Stone     // changed:
419513c26ceSGreg Clayton     char new_path[PATH_MAX];
420aa149cbdSGreg Clayton     ModuleSP new_exec_module_sp(target->GetExecutableModule());
421b9c1b51eSKate Stone     if (!old_exec_module_sp) {
422513c26ceSGreg Clayton       // We might not have a module if we attached to a raw pid...
423b9c1b51eSKate Stone       if (new_exec_module_sp) {
424aa149cbdSGreg Clayton         new_exec_module_sp->GetFileSpec().GetPath(new_path, PATH_MAX);
425b9c1b51eSKate Stone         result.AppendMessageWithFormat("Executable module set to \"%s\".\n",
426b9c1b51eSKate Stone                                        new_path);
427513c26ceSGreg Clayton       }
428b9c1b51eSKate Stone     } else if (old_exec_module_sp->GetFileSpec() !=
429b9c1b51eSKate Stone                new_exec_module_sp->GetFileSpec()) {
430513c26ceSGreg Clayton       char old_path[PATH_MAX];
4315aee162fSJim Ingham 
4325aee162fSJim Ingham       old_exec_module_sp->GetFileSpec().GetPath(old_path, PATH_MAX);
433aa149cbdSGreg Clayton       new_exec_module_sp->GetFileSpec().GetPath(new_path, PATH_MAX);
4345aee162fSJim Ingham 
435b9c1b51eSKate Stone       result.AppendWarningWithFormat(
436b9c1b51eSKate Stone           "Executable module changed from \"%s\" to \"%s\".\n", old_path,
437b9c1b51eSKate Stone           new_path);
4385aee162fSJim Ingham     }
4395aee162fSJim Ingham 
440b9c1b51eSKate Stone     if (!old_arch_spec.IsValid()) {
441b9c1b51eSKate Stone       result.AppendMessageWithFormat(
442b9c1b51eSKate Stone           "Architecture set to: %s.\n",
443b9c1b51eSKate Stone           target->GetArchitecture().GetTriple().getTriple().c_str());
444b9c1b51eSKate Stone     } else if (!old_arch_spec.IsExactMatch(target->GetArchitecture())) {
445b9c1b51eSKate Stone       result.AppendWarningWithFormat(
446b9c1b51eSKate Stone           "Architecture changed from %s to %s.\n",
447c1b1f1eaSGreg Clayton           old_arch_spec.GetTriple().getTriple().c_str(),
448c1b1f1eaSGreg Clayton           target->GetArchitecture().GetTriple().getTriple().c_str());
4495aee162fSJim Ingham     }
450a95ce623SJohnny Chen 
45105097246SAdrian Prantl     // This supports the use-case scenario of immediately continuing the
45205097246SAdrian Prantl     // process once attached.
453a95ce623SJohnny Chen     if (m_options.attach_info.GetContinueOnceAttached())
4545bcaf583SSean Callanan       m_interpreter.HandleCommand("process continue", eLazyBoolNo, result);
455926af0cdSOleksiy Vyalov 
4565aee162fSJim Ingham     return result.Succeeded();
4575aee162fSJim Ingham   }
4585aee162fSJim Ingham 
45930fdc8d8SChris Lattner   CommandOptions m_options;
46030fdc8d8SChris Lattner };
46130fdc8d8SChris Lattner 
46230fdc8d8SChris Lattner // CommandObjectProcessContinue
4631f0f5b5bSZachary Turner 
464438dfcffSRaphael Isemann #define LLDB_OPTIONS_process_continue
465438dfcffSRaphael Isemann #include "CommandOptions.inc"
4661f0f5b5bSZachary Turner 
467bb9caf73SJim Ingham #pragma mark CommandObjectProcessContinue
46830fdc8d8SChris Lattner 
469b9c1b51eSKate Stone class CommandObjectProcessContinue : public CommandObjectParsed {
47030fdc8d8SChris Lattner public:
471b9c1b51eSKate Stone   CommandObjectProcessContinue(CommandInterpreter &interpreter)
472b9c1b51eSKate Stone       : CommandObjectParsed(
473b9c1b51eSKate Stone             interpreter, "process continue",
474e3d26315SCaroline Tice             "Continue execution of all threads in the current process.",
47530fdc8d8SChris Lattner             "process continue",
476b9c1b51eSKate Stone             eCommandRequiresProcess | eCommandTryTargetAPILock |
477b9c1b51eSKate Stone                 eCommandProcessMustBeLaunched | eCommandProcessMustBePaused),
478b9c1b51eSKate Stone         m_options() {}
47930fdc8d8SChris Lattner 
48049bcfd80SEugene Zelenko   ~CommandObjectProcessContinue() override = default;
48130fdc8d8SChris Lattner 
4825a988416SJim Ingham protected:
483b9c1b51eSKate Stone   class CommandOptions : public Options {
4840e41084aSJim Ingham   public:
485b9c1b51eSKate Stone     CommandOptions() : Options() {
486b9c1b51eSKate Stone       // Keep default values of all options in one place: OptionParsingStarting
487b9c1b51eSKate Stone       // ()
488e1cfbc79STodd Fiala       OptionParsingStarting(nullptr);
4890e41084aSJim Ingham     }
4900e41084aSJim Ingham 
49149bcfd80SEugene Zelenko     ~CommandOptions() override = default;
4920e41084aSJim Ingham 
49397206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
494b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
49597206d57SZachary Turner       Status error;
4963bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
497b9c1b51eSKate Stone       switch (short_option) {
4980e41084aSJim Ingham       case 'i':
499fe11483bSZachary Turner         if (option_arg.getAsInteger(0, m_ignore))
500b9c1b51eSKate Stone           error.SetErrorStringWithFormat(
501b9c1b51eSKate Stone               "invalid value for ignore option: \"%s\", should be a number.",
502fe11483bSZachary Turner               option_arg.str().c_str());
5030e41084aSJim Ingham         break;
5040e41084aSJim Ingham 
5050e41084aSJim Ingham       default:
50636162014SRaphael Isemann         llvm_unreachable("Unimplemented option");
5070e41084aSJim Ingham       }
5080e41084aSJim Ingham       return error;
5090e41084aSJim Ingham     }
5100e41084aSJim Ingham 
511b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
5120e41084aSJim Ingham       m_ignore = 0;
5130e41084aSJim Ingham     }
5140e41084aSJim Ingham 
5151f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
51670602439SZachary Turner       return llvm::makeArrayRef(g_process_continue_options);
5171f0f5b5bSZachary Turner     }
5180e41084aSJim Ingham 
5190e41084aSJim Ingham     uint32_t m_ignore;
5200e41084aSJim Ingham   };
5210e41084aSJim Ingham 
522b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
523f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
524a7015092SGreg Clayton     bool synchronous_execution = m_interpreter.GetSynchronous();
52530fdc8d8SChris Lattner     StateType state = process->GetState();
526b9c1b51eSKate Stone     if (state == eStateStopped) {
527b9c1b51eSKate Stone       if (command.GetArgumentCount() != 0) {
528b9c1b51eSKate Stone         result.AppendErrorWithFormat(
529b9c1b51eSKate Stone             "The '%s' command does not take any arguments.\n",
530b9c1b51eSKate Stone             m_cmd_name.c_str());
53130fdc8d8SChris Lattner         return false;
53230fdc8d8SChris Lattner       }
53330fdc8d8SChris Lattner 
534b9c1b51eSKate Stone       if (m_options.m_ignore > 0) {
5358d94ba0fSJim Ingham         ThreadSP sel_thread_sp(GetDefaultThread()->shared_from_this());
536b9c1b51eSKate Stone         if (sel_thread_sp) {
5370e41084aSJim Ingham           StopInfoSP stop_info_sp = sel_thread_sp->GetStopInfo();
538b9c1b51eSKate Stone           if (stop_info_sp &&
539b9c1b51eSKate Stone               stop_info_sp->GetStopReason() == eStopReasonBreakpoint) {
540b9c1b51eSKate Stone             lldb::break_id_t bp_site_id =
541b9c1b51eSKate Stone                 (lldb::break_id_t)stop_info_sp->GetValue();
542b9c1b51eSKate Stone             BreakpointSiteSP bp_site_sp(
543b9c1b51eSKate Stone                 process->GetBreakpointSiteList().FindByID(bp_site_id));
544b9c1b51eSKate Stone             if (bp_site_sp) {
545c7bece56SGreg Clayton               const size_t num_owners = bp_site_sp->GetNumberOfOwners();
546b9c1b51eSKate Stone               for (size_t i = 0; i < num_owners; i++) {
547b9c1b51eSKate Stone                 Breakpoint &bp_ref =
548b9c1b51eSKate Stone                     bp_site_sp->GetOwnerAtIndex(i)->GetBreakpoint();
549b9c1b51eSKate Stone                 if (!bp_ref.IsInternal()) {
5500e41084aSJim Ingham                   bp_ref.SetIgnoreCount(m_options.m_ignore);
5510e41084aSJim Ingham                 }
5520e41084aSJim Ingham               }
5530e41084aSJim Ingham             }
5540e41084aSJim Ingham           }
5550e41084aSJim Ingham         }
5560e41084aSJim Ingham       }
5570e41084aSJim Ingham 
55841f2b940SJim Ingham       { // Scope for thread list mutex:
559b9c1b51eSKate Stone         std::lock_guard<std::recursive_mutex> guard(
560b9c1b51eSKate Stone             process->GetThreadList().GetMutex());
56130fdc8d8SChris Lattner         const uint32_t num_threads = process->GetThreadList().GetSize();
56230fdc8d8SChris Lattner 
56330fdc8d8SChris Lattner         // Set the actions that the threads should each take when resuming
564b9c1b51eSKate Stone         for (uint32_t idx = 0; idx < num_threads; ++idx) {
5656c9ed91cSJim Ingham           const bool override_suspend = false;
566b9c1b51eSKate Stone           process->GetThreadList().GetThreadAtIndex(idx)->SetResumeState(
567b9c1b51eSKate Stone               eStateRunning, override_suspend);
56830fdc8d8SChris Lattner         }
56941f2b940SJim Ingham       }
57030fdc8d8SChris Lattner 
5714446487dSPavel Labath       const uint32_t iohandler_id = process->GetIOHandlerID();
5724446487dSPavel Labath 
573dc6224e0SGreg Clayton       StreamString stream;
57497206d57SZachary Turner       Status error;
575dc6224e0SGreg Clayton       if (synchronous_execution)
576dc6224e0SGreg Clayton         error = process->ResumeSynchronous(&stream);
577dc6224e0SGreg Clayton       else
578dc6224e0SGreg Clayton         error = process->Resume();
579a3b89e27STodd Fiala 
580b9c1b51eSKate Stone       if (error.Success()) {
581b9c1b51eSKate Stone         // There is a race condition where this thread will return up the call
58205097246SAdrian Prantl         // stack to the main command handler and show an (lldb) prompt before
58305097246SAdrian Prantl         // HandlePrivateEvent (from PrivateStateThread) has a chance to call
58405097246SAdrian Prantl         // PushProcessIOHandler().
5853879fe00SPavel Labath         process->SyncIOHandler(iohandler_id, std::chrono::seconds(2));
586a3b89e27STodd Fiala 
587b9c1b51eSKate Stone         result.AppendMessageWithFormat("Process %" PRIu64 " resuming\n",
588b9c1b51eSKate Stone                                        process->GetID());
589b9c1b51eSKate Stone         if (synchronous_execution) {
590b9c1b51eSKate Stone           // If any state changed events had anything to say, add that to the
591b9c1b51eSKate Stone           // result
592c156427dSZachary Turner           result.AppendMessage(stream.GetString());
59330fdc8d8SChris Lattner 
59430fdc8d8SChris Lattner           result.SetDidChangeProcessState(true);
59530fdc8d8SChris Lattner           result.SetStatus(eReturnStatusSuccessFinishNoResult);
596b9c1b51eSKate Stone         } else {
59730fdc8d8SChris Lattner           result.SetStatus(eReturnStatusSuccessContinuingNoResult);
59830fdc8d8SChris Lattner         }
599b9c1b51eSKate Stone       } else {
600b9c1b51eSKate Stone         result.AppendErrorWithFormat("Failed to resume process: %s.\n",
601b9c1b51eSKate Stone                                      error.AsCString());
60230fdc8d8SChris Lattner       }
603b9c1b51eSKate Stone     } else {
604b9c1b51eSKate Stone       result.AppendErrorWithFormat(
605b9c1b51eSKate Stone           "Process cannot be continued from its current state (%s).\n",
60630fdc8d8SChris Lattner           StateAsCString(state));
60730fdc8d8SChris Lattner     }
60830fdc8d8SChris Lattner     return result.Succeeded();
60930fdc8d8SChris Lattner   }
6100e41084aSJim Ingham 
611b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
6120e41084aSJim Ingham 
6130e41084aSJim Ingham   CommandOptions m_options;
6140e41084aSJim Ingham };
6150e41084aSJim Ingham 
61630fdc8d8SChris Lattner // CommandObjectProcessDetach
617438dfcffSRaphael Isemann #define LLDB_OPTIONS_process_detach
618438dfcffSRaphael Isemann #include "CommandOptions.inc"
6191f0f5b5bSZachary Turner 
620bb9caf73SJim Ingham #pragma mark CommandObjectProcessDetach
62130fdc8d8SChris Lattner 
622b9c1b51eSKate Stone class CommandObjectProcessDetach : public CommandObjectParsed {
62330fdc8d8SChris Lattner public:
624b9c1b51eSKate Stone   class CommandOptions : public Options {
625acff8950SJim Ingham   public:
626b9c1b51eSKate Stone     CommandOptions() : Options() { OptionParsingStarting(nullptr); }
627acff8950SJim Ingham 
62849bcfd80SEugene Zelenko     ~CommandOptions() override = default;
629acff8950SJim Ingham 
63097206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
631b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
63297206d57SZachary Turner       Status error;
633acff8950SJim Ingham       const int short_option = m_getopt_table[option_idx].val;
634acff8950SJim Ingham 
635b9c1b51eSKate Stone       switch (short_option) {
636acff8950SJim Ingham       case 's':
637acff8950SJim Ingham         bool tmp_result;
638acff8950SJim Ingham         bool success;
63947cbf4a0SPavel Labath         tmp_result = OptionArgParser::ToBoolean(option_arg, false, &success);
640acff8950SJim Ingham         if (!success)
641b9c1b51eSKate Stone           error.SetErrorStringWithFormat("invalid boolean option: \"%s\"",
642fe11483bSZachary Turner                                          option_arg.str().c_str());
643b9c1b51eSKate Stone         else {
644acff8950SJim Ingham           if (tmp_result)
645acff8950SJim Ingham             m_keep_stopped = eLazyBoolYes;
646acff8950SJim Ingham           else
647acff8950SJim Ingham             m_keep_stopped = eLazyBoolNo;
648acff8950SJim Ingham         }
649acff8950SJim Ingham         break;
650acff8950SJim Ingham       default:
65136162014SRaphael Isemann         llvm_unreachable("Unimplemented option");
652acff8950SJim Ingham       }
653acff8950SJim Ingham       return error;
654acff8950SJim Ingham     }
655acff8950SJim Ingham 
656b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
657acff8950SJim Ingham       m_keep_stopped = eLazyBoolCalculate;
658acff8950SJim Ingham     }
659acff8950SJim Ingham 
6601f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
66170602439SZachary Turner       return llvm::makeArrayRef(g_process_detach_options);
6621f0f5b5bSZachary Turner     }
663acff8950SJim Ingham 
664acff8950SJim Ingham     // Instance variables to hold the values for command options.
665acff8950SJim Ingham     LazyBool m_keep_stopped;
666acff8950SJim Ingham   };
66730fdc8d8SChris Lattner 
6687428a18cSKate Stone   CommandObjectProcessDetach(CommandInterpreter &interpreter)
669b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process detach",
670b9c1b51eSKate Stone                             "Detach from the current target process.",
671a7015092SGreg Clayton                             "process detach",
672b9c1b51eSKate Stone                             eCommandRequiresProcess | eCommandTryTargetAPILock |
673b9c1b51eSKate Stone                                 eCommandProcessMustBeLaunched),
674b9c1b51eSKate Stone         m_options() {}
67530fdc8d8SChris Lattner 
67649bcfd80SEugene Zelenko   ~CommandObjectProcessDetach() override = default;
67730fdc8d8SChris Lattner 
678b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
679acff8950SJim Ingham 
6805a988416SJim Ingham protected:
681b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
682f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
683acff8950SJim Ingham     // FIXME: This will be a Command Option:
684acff8950SJim Ingham     bool keep_stopped;
685b9c1b51eSKate Stone     if (m_options.m_keep_stopped == eLazyBoolCalculate) {
686acff8950SJim Ingham       // Check the process default:
68749bcfd80SEugene Zelenko       keep_stopped = process->GetDetachKeepsStopped();
688b9c1b51eSKate Stone     } else if (m_options.m_keep_stopped == eLazyBoolYes)
689acff8950SJim Ingham       keep_stopped = true;
690acff8950SJim Ingham     else
691acff8950SJim Ingham       keep_stopped = false;
692acff8950SJim Ingham 
69397206d57SZachary Turner     Status error(process->Detach(keep_stopped));
694b9c1b51eSKate Stone     if (error.Success()) {
69530fdc8d8SChris Lattner       result.SetStatus(eReturnStatusSuccessFinishResult);
696b9c1b51eSKate Stone     } else {
69730fdc8d8SChris Lattner       result.AppendErrorWithFormat("Detach failed: %s\n", error.AsCString());
69830fdc8d8SChris Lattner       return false;
69930fdc8d8SChris Lattner     }
70030fdc8d8SChris Lattner     return result.Succeeded();
70130fdc8d8SChris Lattner   }
702acff8950SJim Ingham 
703acff8950SJim Ingham   CommandOptions m_options;
704acff8950SJim Ingham };
705acff8950SJim Ingham 
706b766a73dSGreg Clayton // CommandObjectProcessConnect
707438dfcffSRaphael Isemann #define LLDB_OPTIONS_process_connect
708438dfcffSRaphael Isemann #include "CommandOptions.inc"
7091f0f5b5bSZachary Turner 
710b766a73dSGreg Clayton #pragma mark CommandObjectProcessConnect
711b766a73dSGreg Clayton 
712b9c1b51eSKate Stone class CommandObjectProcessConnect : public CommandObjectParsed {
713b766a73dSGreg Clayton public:
714b9c1b51eSKate Stone   class CommandOptions : public Options {
715b766a73dSGreg Clayton   public:
716b9c1b51eSKate Stone     CommandOptions() : Options() {
717b9c1b51eSKate Stone       // Keep default values of all options in one place: OptionParsingStarting
718b9c1b51eSKate Stone       // ()
719e1cfbc79STodd Fiala       OptionParsingStarting(nullptr);
720b766a73dSGreg Clayton     }
721b766a73dSGreg Clayton 
72249bcfd80SEugene Zelenko     ~CommandOptions() override = default;
723b766a73dSGreg Clayton 
72497206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
725b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
72697206d57SZachary Turner       Status error;
7273bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
728b766a73dSGreg Clayton 
729b9c1b51eSKate Stone       switch (short_option) {
730b766a73dSGreg Clayton       case 'p':
731adcd0268SBenjamin Kramer         plugin_name.assign(std::string(option_arg));
732b766a73dSGreg Clayton         break;
733b766a73dSGreg Clayton 
734b766a73dSGreg Clayton       default:
73536162014SRaphael Isemann         llvm_unreachable("Unimplemented option");
736b766a73dSGreg Clayton       }
737b766a73dSGreg Clayton       return error;
738b766a73dSGreg Clayton     }
739b766a73dSGreg Clayton 
740b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
741b766a73dSGreg Clayton       plugin_name.clear();
742b766a73dSGreg Clayton     }
743b766a73dSGreg Clayton 
7441f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
74570602439SZachary Turner       return llvm::makeArrayRef(g_process_connect_options);
7461f0f5b5bSZachary Turner     }
747b766a73dSGreg Clayton 
748b766a73dSGreg Clayton     // Instance variables to hold the values for command options.
749b766a73dSGreg Clayton 
750b766a73dSGreg Clayton     std::string plugin_name;
751b766a73dSGreg Clayton   };
752b766a73dSGreg Clayton 
753b9c1b51eSKate Stone   CommandObjectProcessConnect(CommandInterpreter &interpreter)
754b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process connect",
755b766a73dSGreg Clayton                             "Connect to a remote debug service.",
756b9c1b51eSKate Stone                             "process connect <remote-url>", 0),
757b9c1b51eSKate Stone         m_options() {}
758b766a73dSGreg Clayton 
75949bcfd80SEugene Zelenko   ~CommandObjectProcessConnect() override = default;
760b766a73dSGreg Clayton 
761b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
7625a988416SJim Ingham 
7635a988416SJim Ingham protected:
764b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
765b9c1b51eSKate Stone     if (command.GetArgumentCount() != 1) {
766b9c1b51eSKate Stone       result.AppendErrorWithFormat(
767b9c1b51eSKate Stone           "'%s' takes exactly one argument:\nUsage: %s\n", m_cmd_name.c_str(),
768ccd6cffbSTamas Berghammer           m_cmd_syntax.c_str());
769ccd6cffbSTamas Berghammer       return false;
770ccd6cffbSTamas Berghammer     }
771ccd6cffbSTamas Berghammer 
772ccd6cffbSTamas Berghammer     Process *process = m_exe_ctx.GetProcessPtr();
773b9c1b51eSKate Stone     if (process && process->IsAlive()) {
774b9c1b51eSKate Stone       result.AppendErrorWithFormat(
775b9c1b51eSKate Stone           "Process %" PRIu64
776b9c1b51eSKate Stone           " is currently being debugged, kill the process before connecting.\n",
777b766a73dSGreg Clayton           process->GetID());
778b766a73dSGreg Clayton       return false;
779b766a73dSGreg Clayton     }
780b766a73dSGreg Clayton 
781ccd6cffbSTamas Berghammer     const char *plugin_name = nullptr;
782b766a73dSGreg Clayton     if (!m_options.plugin_name.empty())
783b766a73dSGreg Clayton       plugin_name = m_options.plugin_name.c_str();
784b766a73dSGreg Clayton 
78597206d57SZachary Turner     Status error;
78657179860SJonas Devlieghere     Debugger &debugger = GetDebugger();
787ccd6cffbSTamas Berghammer     PlatformSP platform_sp = m_interpreter.GetPlatform(true);
788706cccb8SJonas Devlieghere     ProcessSP process_sp =
789706cccb8SJonas Devlieghere         debugger.GetAsyncExecution()
790706cccb8SJonas Devlieghere             ? platform_sp->ConnectProcess(
791b9c1b51eSKate Stone                   command.GetArgumentAtIndex(0), plugin_name, debugger,
792706cccb8SJonas Devlieghere                   debugger.GetSelectedTarget().get(), error)
793706cccb8SJonas Devlieghere             : platform_sp->ConnectProcessSynchronous(
794706cccb8SJonas Devlieghere                   command.GetArgumentAtIndex(0), plugin_name, debugger,
795706cccb8SJonas Devlieghere                   result.GetOutputStream(), debugger.GetSelectedTarget().get(),
796706cccb8SJonas Devlieghere                   error);
797b9c1b51eSKate Stone     if (error.Fail() || process_sp == nullptr) {
798ccd6cffbSTamas Berghammer       result.AppendError(error.AsCString("Error connecting to the process"));
799b766a73dSGreg Clayton       return false;
800b766a73dSGreg Clayton     }
801ccd6cffbSTamas Berghammer     return true;
802b766a73dSGreg Clayton   }
803b766a73dSGreg Clayton 
804b766a73dSGreg Clayton   CommandOptions m_options;
805b766a73dSGreg Clayton };
806b766a73dSGreg Clayton 
807998255bfSGreg Clayton // CommandObjectProcessPlugin
808998255bfSGreg Clayton #pragma mark CommandObjectProcessPlugin
809998255bfSGreg Clayton 
810b9c1b51eSKate Stone class CommandObjectProcessPlugin : public CommandObjectProxy {
811998255bfSGreg Clayton public:
8127428a18cSKate Stone   CommandObjectProcessPlugin(CommandInterpreter &interpreter)
813b9c1b51eSKate Stone       : CommandObjectProxy(
814b9c1b51eSKate Stone             interpreter, "process plugin",
815b9c1b51eSKate Stone             "Send a custom command to the current target process plug-in.",
816b9c1b51eSKate Stone             "process plugin <args>", 0) {}
817998255bfSGreg Clayton 
81849bcfd80SEugene Zelenko   ~CommandObjectProcessPlugin() override = default;
819998255bfSGreg Clayton 
820b9c1b51eSKate Stone   CommandObject *GetProxyCommandObject() override {
821e05b2efeSGreg Clayton     Process *process = m_interpreter.GetExecutionContext().GetProcessPtr();
822998255bfSGreg Clayton     if (process)
823998255bfSGreg Clayton       return process->GetPluginCommandObject();
82449bcfd80SEugene Zelenko     return nullptr;
825998255bfSGreg Clayton   }
826998255bfSGreg Clayton };
827998255bfSGreg Clayton 
8288f343b09SGreg Clayton // CommandObjectProcessLoad
829438dfcffSRaphael Isemann #define LLDB_OPTIONS_process_load
830438dfcffSRaphael Isemann #include "CommandOptions.inc"
8311f0f5b5bSZachary Turner 
832bb9caf73SJim Ingham #pragma mark CommandObjectProcessLoad
8338f343b09SGreg Clayton 
834b9c1b51eSKate Stone class CommandObjectProcessLoad : public CommandObjectParsed {
8358f343b09SGreg Clayton public:
836b9c1b51eSKate Stone   class CommandOptions : public Options {
8374fbd67acSTamas Berghammer   public:
838b9c1b51eSKate Stone     CommandOptions() : Options() {
839b9c1b51eSKate Stone       // Keep default values of all options in one place: OptionParsingStarting
840b9c1b51eSKate Stone       // ()
841e1cfbc79STodd Fiala       OptionParsingStarting(nullptr);
8424fbd67acSTamas Berghammer     }
8434fbd67acSTamas Berghammer 
8444fbd67acSTamas Berghammer     ~CommandOptions() override = default;
8454fbd67acSTamas Berghammer 
84697206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
847b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
84897206d57SZachary Turner       Status error;
8494fbd67acSTamas Berghammer       const int short_option = m_getopt_table[option_idx].val;
850b9c1b51eSKate Stone       switch (short_option) {
8514fbd67acSTamas Berghammer       case 'i':
8524fbd67acSTamas Berghammer         do_install = true;
853fe11483bSZachary Turner         if (!option_arg.empty())
8548f3be7a3SJonas Devlieghere           install_path.SetFile(option_arg, FileSpec::Style::native);
8554fbd67acSTamas Berghammer         break;
8564fbd67acSTamas Berghammer       default:
85736162014SRaphael Isemann         llvm_unreachable("Unimplemented option");
8584fbd67acSTamas Berghammer       }
8594fbd67acSTamas Berghammer       return error;
8604fbd67acSTamas Berghammer     }
8614fbd67acSTamas Berghammer 
862b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
8634fbd67acSTamas Berghammer       do_install = false;
8644fbd67acSTamas Berghammer       install_path.Clear();
8654fbd67acSTamas Berghammer     }
8664fbd67acSTamas Berghammer 
8671f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
86870602439SZachary Turner       return llvm::makeArrayRef(g_process_load_options);
8691f0f5b5bSZachary Turner     }
8704fbd67acSTamas Berghammer 
8714fbd67acSTamas Berghammer     // Instance variables to hold the values for command options.
8724fbd67acSTamas Berghammer     bool do_install;
8734fbd67acSTamas Berghammer     FileSpec install_path;
8744fbd67acSTamas Berghammer   };
8758f343b09SGreg Clayton 
876b9c1b51eSKate Stone   CommandObjectProcessLoad(CommandInterpreter &interpreter)
877b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process load",
8788f343b09SGreg Clayton                             "Load a shared library into the current process.",
8798f343b09SGreg Clayton                             "process load <filename> [<filename> ...]",
880b9c1b51eSKate Stone                             eCommandRequiresProcess | eCommandTryTargetAPILock |
881e87764f2SEnrico Granata                                 eCommandProcessMustBeLaunched |
8824fbd67acSTamas Berghammer                                 eCommandProcessMustBePaused),
883b9c1b51eSKate Stone         m_options() {}
8848f343b09SGreg Clayton 
8854fbd67acSTamas Berghammer   ~CommandObjectProcessLoad() override = default;
8864fbd67acSTamas Berghammer 
887e1cd7cacSGongyu Deng   void
888e1cd7cacSGongyu Deng   HandleArgumentCompletion(CompletionRequest &request,
889e1cd7cacSGongyu Deng                            OptionElementVector &opt_element_vector) override {
890e1cd7cacSGongyu Deng     if (!m_exe_ctx.HasProcessScope())
891e1cd7cacSGongyu Deng       return;
892e1cd7cacSGongyu Deng 
893e1cd7cacSGongyu Deng     CommandCompletions::InvokeCommonCompletionCallbacks(
894e1cd7cacSGongyu Deng         GetCommandInterpreter(), CommandCompletions::eDiskFileCompletion,
895e1cd7cacSGongyu Deng         request, nullptr);
896e1cd7cacSGongyu Deng   }
897e1cd7cacSGongyu Deng 
898b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
8998f343b09SGreg Clayton 
9005a988416SJim Ingham protected:
901b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
902f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
9038f343b09SGreg Clayton 
90497d2c401SZachary Turner     for (auto &entry : command.entries()) {
90597206d57SZachary Turner       Status error;
9063cb132a0STamas Berghammer       PlatformSP platform = process->GetTarget().GetPlatform();
9070d9a201eSRaphael Isemann       llvm::StringRef image_path = entry.ref();
9084fbd67acSTamas Berghammer       uint32_t image_token = LLDB_INVALID_IMAGE_TOKEN;
9094fbd67acSTamas Berghammer 
910b9c1b51eSKate Stone       if (!m_options.do_install) {
9118f3be7a3SJonas Devlieghere         FileSpec image_spec(image_path);
9123cb132a0STamas Berghammer         platform->ResolveRemotePath(image_spec, image_spec);
913b9c1b51eSKate Stone         image_token =
914b9c1b51eSKate Stone             platform->LoadImage(process, FileSpec(), image_spec, error);
915b9c1b51eSKate Stone       } else if (m_options.install_path) {
9168f3be7a3SJonas Devlieghere         FileSpec image_spec(image_path);
9178f3be7a3SJonas Devlieghere         FileSystem::Instance().Resolve(image_spec);
918b9c1b51eSKate Stone         platform->ResolveRemotePath(m_options.install_path,
919b9c1b51eSKate Stone                                     m_options.install_path);
920b9c1b51eSKate Stone         image_token = platform->LoadImage(process, image_spec,
921b9c1b51eSKate Stone                                           m_options.install_path, error);
922b9c1b51eSKate Stone       } else {
9238f3be7a3SJonas Devlieghere         FileSpec image_spec(image_path);
9248f3be7a3SJonas Devlieghere         FileSystem::Instance().Resolve(image_spec);
925b9c1b51eSKate Stone         image_token =
926b9c1b51eSKate Stone             platform->LoadImage(process, image_spec, FileSpec(), error);
9274fbd67acSTamas Berghammer       }
9284fbd67acSTamas Berghammer 
929b9c1b51eSKate Stone       if (image_token != LLDB_INVALID_IMAGE_TOKEN) {
930b9c1b51eSKate Stone         result.AppendMessageWithFormat(
93197d2c401SZachary Turner             "Loading \"%s\"...ok\nImage %u loaded.\n", image_path.str().c_str(),
93297d2c401SZachary Turner             image_token);
9338f343b09SGreg Clayton         result.SetStatus(eReturnStatusSuccessFinishResult);
934b9c1b51eSKate Stone       } else {
93597d2c401SZachary Turner         result.AppendErrorWithFormat("failed to load '%s': %s",
93697d2c401SZachary Turner                                      image_path.str().c_str(),
937b9c1b51eSKate Stone                                      error.AsCString());
9388f343b09SGreg Clayton       }
9398f343b09SGreg Clayton     }
9408f343b09SGreg Clayton     return result.Succeeded();
9418f343b09SGreg Clayton   }
9424fbd67acSTamas Berghammer 
9434fbd67acSTamas Berghammer   CommandOptions m_options;
9448f343b09SGreg Clayton };
9458f343b09SGreg Clayton 
9468f343b09SGreg Clayton // CommandObjectProcessUnload
947bb9caf73SJim Ingham #pragma mark CommandObjectProcessUnload
9488f343b09SGreg Clayton 
949b9c1b51eSKate Stone class CommandObjectProcessUnload : public CommandObjectParsed {
9508f343b09SGreg Clayton public:
951b9c1b51eSKate Stone   CommandObjectProcessUnload(CommandInterpreter &interpreter)
952b9c1b51eSKate Stone       : CommandObjectParsed(
953b9c1b51eSKate Stone             interpreter, "process unload",
954b9c1b51eSKate Stone             "Unload a shared library from the current process using the index "
955b9c1b51eSKate Stone             "returned by a previous call to \"process load\".",
9568f343b09SGreg Clayton             "process unload <index>",
957b9c1b51eSKate Stone             eCommandRequiresProcess | eCommandTryTargetAPILock |
958b9c1b51eSKate Stone                 eCommandProcessMustBeLaunched | eCommandProcessMustBePaused) {}
9598f343b09SGreg Clayton 
96049bcfd80SEugene Zelenko   ~CommandObjectProcessUnload() override = default;
9618f343b09SGreg Clayton 
962e1cd7cacSGongyu Deng   void
963e1cd7cacSGongyu Deng   HandleArgumentCompletion(CompletionRequest &request,
964e1cd7cacSGongyu Deng                            OptionElementVector &opt_element_vector) override {
965e1cd7cacSGongyu Deng 
966e1cd7cacSGongyu Deng     if (request.GetCursorIndex() || !m_exe_ctx.HasProcessScope())
967e1cd7cacSGongyu Deng       return;
968e1cd7cacSGongyu Deng 
969e1cd7cacSGongyu Deng     Process *process = m_exe_ctx.GetProcessPtr();
970e1cd7cacSGongyu Deng 
971e1cd7cacSGongyu Deng     const std::vector<lldb::addr_t> &tokens = process->GetImageTokens();
972e1cd7cacSGongyu Deng     const size_t token_num = tokens.size();
973e1cd7cacSGongyu Deng     for (size_t i = 0; i < token_num; ++i) {
974e1cd7cacSGongyu Deng       if (tokens[i] == LLDB_INVALID_IMAGE_TOKEN)
975e1cd7cacSGongyu Deng         continue;
976e1cd7cacSGongyu Deng       request.TryCompleteCurrentArg(std::to_string(i));
977e1cd7cacSGongyu Deng     }
978e1cd7cacSGongyu Deng   }
979e1cd7cacSGongyu Deng 
9805a988416SJim Ingham protected:
981b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
982f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
9838f343b09SGreg Clayton 
98497d2c401SZachary Turner     for (auto &entry : command.entries()) {
98597d2c401SZachary Turner       uint32_t image_token;
9860d9a201eSRaphael Isemann       if (entry.ref().getAsInteger(0, image_token)) {
987b9c1b51eSKate Stone         result.AppendErrorWithFormat("invalid image index argument '%s'",
9880d9a201eSRaphael Isemann                                      entry.ref().str().c_str());
9898f343b09SGreg Clayton         break;
990b9c1b51eSKate Stone       } else {
99197206d57SZachary Turner         Status error(process->GetTarget().GetPlatform()->UnloadImage(
992b9c1b51eSKate Stone             process, image_token));
993b9c1b51eSKate Stone         if (error.Success()) {
994b9c1b51eSKate Stone           result.AppendMessageWithFormat(
995b9c1b51eSKate Stone               "Unloading shared library with index %u...ok\n", image_token);
9968f343b09SGreg Clayton           result.SetStatus(eReturnStatusSuccessFinishResult);
997b9c1b51eSKate Stone         } else {
998b9c1b51eSKate Stone           result.AppendErrorWithFormat("failed to unload image: %s",
999b9c1b51eSKate Stone                                        error.AsCString());
10008f343b09SGreg Clayton           break;
10018f343b09SGreg Clayton         }
10028f343b09SGreg Clayton       }
10038f343b09SGreg Clayton     }
10048f343b09SGreg Clayton     return result.Succeeded();
10058f343b09SGreg Clayton   }
10068f343b09SGreg Clayton };
10078f343b09SGreg Clayton 
100830fdc8d8SChris Lattner // CommandObjectProcessSignal
1009bb9caf73SJim Ingham #pragma mark CommandObjectProcessSignal
101030fdc8d8SChris Lattner 
1011b9c1b51eSKate Stone class CommandObjectProcessSignal : public CommandObjectParsed {
101230fdc8d8SChris Lattner public:
10137428a18cSKate Stone   CommandObjectProcessSignal(CommandInterpreter &interpreter)
1014a925974bSAdrian Prantl       : CommandObjectParsed(
1015a925974bSAdrian Prantl             interpreter, "process signal",
1016a925974bSAdrian Prantl             "Send a UNIX signal to the current target process.", nullptr,
1017a925974bSAdrian Prantl             eCommandRequiresProcess | eCommandTryTargetAPILock) {
1018405fe67fSCaroline Tice     CommandArgumentEntry arg;
1019405fe67fSCaroline Tice     CommandArgumentData signal_arg;
1020405fe67fSCaroline Tice 
1021405fe67fSCaroline Tice     // Define the first (and only) variant of this arg.
1022c0dbdfb6SCaroline Tice     signal_arg.arg_type = eArgTypeUnixSignal;
1023405fe67fSCaroline Tice     signal_arg.arg_repetition = eArgRepeatPlain;
1024405fe67fSCaroline Tice 
1025b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
1026b9c1b51eSKate Stone     // argument entry.
1027405fe67fSCaroline Tice     arg.push_back(signal_arg);
1028405fe67fSCaroline Tice 
1029405fe67fSCaroline Tice     // Push the data for the first argument into the m_arguments vector.
1030405fe67fSCaroline Tice     m_arguments.push_back(arg);
103130fdc8d8SChris Lattner   }
103230fdc8d8SChris Lattner 
103349bcfd80SEugene Zelenko   ~CommandObjectProcessSignal() override = default;
103430fdc8d8SChris Lattner 
10352bba1c22SRaphael Isemann   void
10362bba1c22SRaphael Isemann   HandleArgumentCompletion(CompletionRequest &request,
10372bba1c22SRaphael Isemann                            OptionElementVector &opt_element_vector) override {
10382bba1c22SRaphael Isemann     if (!m_exe_ctx.HasProcessScope() || request.GetCursorIndex() != 0)
10392bba1c22SRaphael Isemann       return;
10402bba1c22SRaphael Isemann 
10412bba1c22SRaphael Isemann     UnixSignalsSP signals = m_exe_ctx.GetProcessPtr()->GetUnixSignals();
10422bba1c22SRaphael Isemann     int signo = signals->GetFirstSignalNumber();
10432bba1c22SRaphael Isemann     while (signo != LLDB_INVALID_SIGNAL_NUMBER) {
10442bba1c22SRaphael Isemann       request.AddCompletion(signals->GetSignalAsCString(signo), "");
10452bba1c22SRaphael Isemann       signo = signals->GetNextSignalNumber(signo);
10462bba1c22SRaphael Isemann     }
10472bba1c22SRaphael Isemann   }
10482bba1c22SRaphael Isemann 
10495a988416SJim Ingham protected:
1050b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1051f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
105230fdc8d8SChris Lattner 
1053b9c1b51eSKate Stone     if (command.GetArgumentCount() == 1) {
1054237cd906SGreg Clayton       int signo = LLDB_INVALID_SIGNAL_NUMBER;
1055237cd906SGreg Clayton 
1056237cd906SGreg Clayton       const char *signal_name = command.GetArgumentAtIndex(0);
10579010cef2SRaphael Isemann       if (::isxdigit(signal_name[0])) {
10589010cef2SRaphael Isemann         if (!llvm::to_integer(signal_name, signo))
10599010cef2SRaphael Isemann           signo = LLDB_INVALID_SIGNAL_NUMBER;
10609010cef2SRaphael Isemann       } else
106198d0a4b3SChaoren Lin         signo = process->GetUnixSignals()->GetSignalNumberFromName(signal_name);
1062237cd906SGreg Clayton 
1063b9c1b51eSKate Stone       if (signo == LLDB_INVALID_SIGNAL_NUMBER) {
1064b9c1b51eSKate Stone         result.AppendErrorWithFormat("Invalid signal argument '%s'.\n",
1065b9c1b51eSKate Stone                                      command.GetArgumentAtIndex(0));
1066b9c1b51eSKate Stone       } else {
106797206d57SZachary Turner         Status error(process->Signal(signo));
1068b9c1b51eSKate Stone         if (error.Success()) {
106930fdc8d8SChris Lattner           result.SetStatus(eReturnStatusSuccessFinishResult);
1070b9c1b51eSKate Stone         } else {
1071b9c1b51eSKate Stone           result.AppendErrorWithFormat("Failed to send signal %i: %s\n", signo,
1072b9c1b51eSKate Stone                                        error.AsCString());
107330fdc8d8SChris Lattner         }
107430fdc8d8SChris Lattner       }
1075b9c1b51eSKate Stone     } else {
1076b9c1b51eSKate Stone       result.AppendErrorWithFormat(
1077b9c1b51eSKate Stone           "'%s' takes exactly one signal number argument:\nUsage: %s\n",
1078b9c1b51eSKate Stone           m_cmd_name.c_str(), m_cmd_syntax.c_str());
107930fdc8d8SChris Lattner     }
108030fdc8d8SChris Lattner     return result.Succeeded();
108130fdc8d8SChris Lattner   }
108230fdc8d8SChris Lattner };
108330fdc8d8SChris Lattner 
108430fdc8d8SChris Lattner // CommandObjectProcessInterrupt
1085bb9caf73SJim Ingham #pragma mark CommandObjectProcessInterrupt
108630fdc8d8SChris Lattner 
1087b9c1b51eSKate Stone class CommandObjectProcessInterrupt : public CommandObjectParsed {
108830fdc8d8SChris Lattner public:
10897428a18cSKate Stone   CommandObjectProcessInterrupt(CommandInterpreter &interpreter)
1090b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process interrupt",
1091b9c1b51eSKate Stone                             "Interrupt the current target process.",
1092a7015092SGreg Clayton                             "process interrupt",
1093b9c1b51eSKate Stone                             eCommandRequiresProcess | eCommandTryTargetAPILock |
1094b9c1b51eSKate Stone                                 eCommandProcessMustBeLaunched) {}
109530fdc8d8SChris Lattner 
109649bcfd80SEugene Zelenko   ~CommandObjectProcessInterrupt() override = default;
109730fdc8d8SChris Lattner 
10985a988416SJim Ingham protected:
1099b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1100f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
1101b9c1b51eSKate Stone     if (process == nullptr) {
110230fdc8d8SChris Lattner       result.AppendError("no process to halt");
110330fdc8d8SChris Lattner       return false;
110430fdc8d8SChris Lattner     }
110530fdc8d8SChris Lattner 
1106b9c1b51eSKate Stone     if (command.GetArgumentCount() == 0) {
1107f9b57b9dSGreg Clayton       bool clear_thread_plans = true;
110897206d57SZachary Turner       Status error(process->Halt(clear_thread_plans));
1109b9c1b51eSKate Stone       if (error.Success()) {
111030fdc8d8SChris Lattner         result.SetStatus(eReturnStatusSuccessFinishResult);
1111b9c1b51eSKate Stone       } else {
1112b9c1b51eSKate Stone         result.AppendErrorWithFormat("Failed to halt process: %s\n",
1113b9c1b51eSKate Stone                                      error.AsCString());
111430fdc8d8SChris Lattner       }
1115b9c1b51eSKate Stone     } else {
1116fd54b368SJason Molenda       result.AppendErrorWithFormat("'%s' takes no arguments:\nUsage: %s\n",
1117b9c1b51eSKate Stone                                    m_cmd_name.c_str(), m_cmd_syntax.c_str());
111830fdc8d8SChris Lattner     }
111930fdc8d8SChris Lattner     return result.Succeeded();
112030fdc8d8SChris Lattner   }
112130fdc8d8SChris Lattner };
112230fdc8d8SChris Lattner 
112330fdc8d8SChris Lattner // CommandObjectProcessKill
1124bb9caf73SJim Ingham #pragma mark CommandObjectProcessKill
112530fdc8d8SChris Lattner 
1126b9c1b51eSKate Stone class CommandObjectProcessKill : public CommandObjectParsed {
112730fdc8d8SChris Lattner public:
11287428a18cSKate Stone   CommandObjectProcessKill(CommandInterpreter &interpreter)
1129b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process kill",
1130b9c1b51eSKate Stone                             "Terminate the current target process.",
1131b9c1b51eSKate Stone                             "process kill",
1132b9c1b51eSKate Stone                             eCommandRequiresProcess | eCommandTryTargetAPILock |
1133b9c1b51eSKate Stone                                 eCommandProcessMustBeLaunched) {}
113430fdc8d8SChris Lattner 
113549bcfd80SEugene Zelenko   ~CommandObjectProcessKill() override = default;
113630fdc8d8SChris Lattner 
11375a988416SJim Ingham protected:
1138b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1139f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
1140b9c1b51eSKate Stone     if (process == nullptr) {
114130fdc8d8SChris Lattner       result.AppendError("no process to kill");
114230fdc8d8SChris Lattner       return false;
114330fdc8d8SChris Lattner     }
114430fdc8d8SChris Lattner 
1145b9c1b51eSKate Stone     if (command.GetArgumentCount() == 0) {
114697206d57SZachary Turner       Status error(process->Destroy(true));
1147b9c1b51eSKate Stone       if (error.Success()) {
114830fdc8d8SChris Lattner         result.SetStatus(eReturnStatusSuccessFinishResult);
1149b9c1b51eSKate Stone       } else {
1150b9c1b51eSKate Stone         result.AppendErrorWithFormat("Failed to kill process: %s\n",
1151b9c1b51eSKate Stone                                      error.AsCString());
115230fdc8d8SChris Lattner       }
1153b9c1b51eSKate Stone     } else {
1154fd54b368SJason Molenda       result.AppendErrorWithFormat("'%s' takes no arguments:\nUsage: %s\n",
1155b9c1b51eSKate Stone                                    m_cmd_name.c_str(), m_cmd_syntax.c_str());
115630fdc8d8SChris Lattner     }
115730fdc8d8SChris Lattner     return result.Succeeded();
115830fdc8d8SChris Lattner   }
115930fdc8d8SChris Lattner };
116030fdc8d8SChris Lattner 
1161a2715cf1SGreg Clayton // CommandObjectProcessSaveCore
1162a2715cf1SGreg Clayton #pragma mark CommandObjectProcessSaveCore
1163a2715cf1SGreg Clayton 
1164*9ea6dd5cSJason Molenda static constexpr OptionEnumValueElement g_corefile_save_style[] = {
1165*9ea6dd5cSJason Molenda     {eSaveCoreFull, "full", "Create a core file with all memory saved"},
1166*9ea6dd5cSJason Molenda     {eSaveCoreDirtyOnly, "modified-memory",
1167*9ea6dd5cSJason Molenda      "Create a corefile with only modified memory saved"}};
1168*9ea6dd5cSJason Molenda 
1169*9ea6dd5cSJason Molenda static constexpr OptionEnumValues SaveCoreStyles() {
1170*9ea6dd5cSJason Molenda   return OptionEnumValues(g_corefile_save_style);
1171*9ea6dd5cSJason Molenda }
1172*9ea6dd5cSJason Molenda 
1173*9ea6dd5cSJason Molenda #define LLDB_OPTIONS_process_save_core
1174*9ea6dd5cSJason Molenda #include "CommandOptions.inc"
1175*9ea6dd5cSJason Molenda 
1176b9c1b51eSKate Stone class CommandObjectProcessSaveCore : public CommandObjectParsed {
1177a2715cf1SGreg Clayton public:
1178b9c1b51eSKate Stone   CommandObjectProcessSaveCore(CommandInterpreter &interpreter)
1179b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process save-core",
1180b9c1b51eSKate Stone                             "Save the current process as a core file using an "
1181b9c1b51eSKate Stone                             "appropriate file type.",
1182*9ea6dd5cSJason Molenda                             "process save-core [-s corefile-style] FILE",
1183b9c1b51eSKate Stone                             eCommandRequiresProcess | eCommandTryTargetAPILock |
1184b9c1b51eSKate Stone                                 eCommandProcessMustBeLaunched) {}
1185a2715cf1SGreg Clayton 
118649bcfd80SEugene Zelenko   ~CommandObjectProcessSaveCore() override = default;
1187a2715cf1SGreg Clayton 
1188*9ea6dd5cSJason Molenda   Options *GetOptions() override { return &m_options; }
1189*9ea6dd5cSJason Molenda 
1190*9ea6dd5cSJason Molenda   class CommandOptions : public Options {
1191*9ea6dd5cSJason Molenda   public:
1192*9ea6dd5cSJason Molenda     CommandOptions()
1193*9ea6dd5cSJason Molenda         : Options(), m_requested_save_core_style(eSaveCoreUnspecified) {}
1194*9ea6dd5cSJason Molenda 
1195*9ea6dd5cSJason Molenda     ~CommandOptions() override = default;
1196*9ea6dd5cSJason Molenda 
1197*9ea6dd5cSJason Molenda     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
1198*9ea6dd5cSJason Molenda       return llvm::makeArrayRef(g_process_save_core_options);
1199*9ea6dd5cSJason Molenda     }
1200*9ea6dd5cSJason Molenda 
1201*9ea6dd5cSJason Molenda     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
1202*9ea6dd5cSJason Molenda                           ExecutionContext *execution_context) override {
1203*9ea6dd5cSJason Molenda       const int short_option = m_getopt_table[option_idx].val;
1204*9ea6dd5cSJason Molenda       Status error;
1205*9ea6dd5cSJason Molenda 
1206*9ea6dd5cSJason Molenda       switch (short_option) {
1207*9ea6dd5cSJason Molenda       case 's':
1208*9ea6dd5cSJason Molenda         m_requested_save_core_style =
1209*9ea6dd5cSJason Molenda             (lldb::SaveCoreStyle)OptionArgParser::ToOptionEnum(
1210*9ea6dd5cSJason Molenda                 option_arg, GetDefinitions()[option_idx].enum_values,
1211*9ea6dd5cSJason Molenda                 eSaveCoreUnspecified, error);
1212*9ea6dd5cSJason Molenda         break;
1213*9ea6dd5cSJason Molenda       default:
1214*9ea6dd5cSJason Molenda         llvm_unreachable("Unimplemented option");
1215*9ea6dd5cSJason Molenda       }
1216*9ea6dd5cSJason Molenda 
1217*9ea6dd5cSJason Molenda       return {};
1218*9ea6dd5cSJason Molenda     }
1219*9ea6dd5cSJason Molenda 
1220*9ea6dd5cSJason Molenda     void OptionParsingStarting(ExecutionContext *execution_context) override {
1221*9ea6dd5cSJason Molenda       m_requested_save_core_style = eSaveCoreUnspecified;
1222*9ea6dd5cSJason Molenda     }
1223*9ea6dd5cSJason Molenda 
1224*9ea6dd5cSJason Molenda     // Instance variables to hold the values for command options.
1225*9ea6dd5cSJason Molenda     SaveCoreStyle m_requested_save_core_style;
1226*9ea6dd5cSJason Molenda   };
1227*9ea6dd5cSJason Molenda 
1228a2715cf1SGreg Clayton protected:
1229b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1230a2715cf1SGreg Clayton     ProcessSP process_sp = m_exe_ctx.GetProcessSP();
1231b9c1b51eSKate Stone     if (process_sp) {
1232b9c1b51eSKate Stone       if (command.GetArgumentCount() == 1) {
12338f3be7a3SJonas Devlieghere         FileSpec output_file(command.GetArgumentAtIndex(0));
1234*9ea6dd5cSJason Molenda         SaveCoreStyle corefile_style = m_options.m_requested_save_core_style;
1235*9ea6dd5cSJason Molenda         Status error =
1236*9ea6dd5cSJason Molenda             PluginManager::SaveCore(process_sp, output_file, corefile_style);
1237b9c1b51eSKate Stone         if (error.Success()) {
1238*9ea6dd5cSJason Molenda           if (corefile_style == SaveCoreStyle::eSaveCoreDirtyOnly) {
1239*9ea6dd5cSJason Molenda             result.AppendMessageWithFormat(
1240*9ea6dd5cSJason Molenda                 "\nModified-memory only corefile "
1241*9ea6dd5cSJason Molenda                 "created.  This corefile may not show \n"
1242*9ea6dd5cSJason Molenda                 "library/framework/app binaries "
1243*9ea6dd5cSJason Molenda                 "on a different system, or when \n"
1244*9ea6dd5cSJason Molenda                 "those binaries have "
1245*9ea6dd5cSJason Molenda                 "been updated/modified. Copies are not included\n"
1246*9ea6dd5cSJason Molenda                 "in this corefile.  Use --style full to include all "
1247*9ea6dd5cSJason Molenda                 "process memory.\n");
1248*9ea6dd5cSJason Molenda           }
1249a2715cf1SGreg Clayton           result.SetStatus(eReturnStatusSuccessFinishResult);
1250b9c1b51eSKate Stone         } else {
1251b9c1b51eSKate Stone           result.AppendErrorWithFormat(
1252b9c1b51eSKate Stone               "Failed to save core file for process: %s\n", error.AsCString());
1253a2715cf1SGreg Clayton         }
1254b9c1b51eSKate Stone       } else {
1255a2715cf1SGreg Clayton         result.AppendErrorWithFormat("'%s' takes one arguments:\nUsage: %s\n",
1256b9c1b51eSKate Stone                                      m_cmd_name.c_str(), m_cmd_syntax.c_str());
1257a2715cf1SGreg Clayton       }
1258b9c1b51eSKate Stone     } else {
1259a2715cf1SGreg Clayton       result.AppendError("invalid process");
1260a2715cf1SGreg Clayton       return false;
1261a2715cf1SGreg Clayton     }
1262a2715cf1SGreg Clayton 
1263a2715cf1SGreg Clayton     return result.Succeeded();
1264a2715cf1SGreg Clayton   }
1265*9ea6dd5cSJason Molenda 
1266*9ea6dd5cSJason Molenda   CommandOptions m_options;
1267a2715cf1SGreg Clayton };
1268a2715cf1SGreg Clayton 
12694b9bea87SJim Ingham // CommandObjectProcessStatus
1270bb9caf73SJim Ingham #pragma mark CommandObjectProcessStatus
1271d7c403e6SMed Ismail Bennani #define LLDB_OPTIONS_process_status
1272d7c403e6SMed Ismail Bennani #include "CommandOptions.inc"
1273bb9caf73SJim Ingham 
1274b9c1b51eSKate Stone class CommandObjectProcessStatus : public CommandObjectParsed {
12754b9bea87SJim Ingham public:
12767428a18cSKate Stone   CommandObjectProcessStatus(CommandInterpreter &interpreter)
1277b9c1b51eSKate Stone       : CommandObjectParsed(
1278b9c1b51eSKate Stone             interpreter, "process status",
1279b9c1b51eSKate Stone             "Show status and stop location for the current target process.",
1280b9c1b51eSKate Stone             "process status",
1281d7c403e6SMed Ismail Bennani             eCommandRequiresProcess | eCommandTryTargetAPILock),
1282d7c403e6SMed Ismail Bennani         m_options() {}
12834b9bea87SJim Ingham 
128449bcfd80SEugene Zelenko   ~CommandObjectProcessStatus() override = default;
12854b9bea87SJim Ingham 
1286d7c403e6SMed Ismail Bennani   Options *GetOptions() override { return &m_options; }
1287d7c403e6SMed Ismail Bennani 
1288d7c403e6SMed Ismail Bennani   class CommandOptions : public Options {
1289d7c403e6SMed Ismail Bennani   public:
12909494c510SJonas Devlieghere     CommandOptions() : Options() {}
1291d7c403e6SMed Ismail Bennani 
1292d7c403e6SMed Ismail Bennani     ~CommandOptions() override = default;
1293d7c403e6SMed Ismail Bennani 
1294d7c403e6SMed Ismail Bennani     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
1295d7c403e6SMed Ismail Bennani                           ExecutionContext *execution_context) override {
1296d7c403e6SMed Ismail Bennani       const int short_option = m_getopt_table[option_idx].val;
1297d7c403e6SMed Ismail Bennani 
1298d7c403e6SMed Ismail Bennani       switch (short_option) {
1299d7c403e6SMed Ismail Bennani       case 'v':
1300d7c403e6SMed Ismail Bennani         m_verbose = true;
1301d7c403e6SMed Ismail Bennani         break;
1302d7c403e6SMed Ismail Bennani       default:
1303d7c403e6SMed Ismail Bennani         llvm_unreachable("Unimplemented option");
1304d7c403e6SMed Ismail Bennani       }
1305d7c403e6SMed Ismail Bennani 
1306d7c403e6SMed Ismail Bennani       return {};
1307d7c403e6SMed Ismail Bennani     }
1308d7c403e6SMed Ismail Bennani 
1309d7c403e6SMed Ismail Bennani     void OptionParsingStarting(ExecutionContext *execution_context) override {
1310d7c403e6SMed Ismail Bennani       m_verbose = false;
1311d7c403e6SMed Ismail Bennani     }
1312d7c403e6SMed Ismail Bennani 
1313d7c403e6SMed Ismail Bennani     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
1314d7c403e6SMed Ismail Bennani       return llvm::makeArrayRef(g_process_status_options);
1315d7c403e6SMed Ismail Bennani     }
1316d7c403e6SMed Ismail Bennani 
1317d7c403e6SMed Ismail Bennani     // Instance variables to hold the values for command options.
13189494c510SJonas Devlieghere     bool m_verbose = false;
1319d7c403e6SMed Ismail Bennani   };
1320d7c403e6SMed Ismail Bennani 
1321d7c403e6SMed Ismail Bennani protected:
1322b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
13237260f620SGreg Clayton     Stream &strm = result.GetOutputStream();
13244b9bea87SJim Ingham     result.SetStatus(eReturnStatusSuccessFinishNoResult);
1325d7c403e6SMed Ismail Bennani 
1326d7c403e6SMed Ismail Bennani     if (command.GetArgumentCount()) {
1327d7c403e6SMed Ismail Bennani       result.AppendError("'process status' takes no arguments");
1328d7c403e6SMed Ismail Bennani       return result.Succeeded();
1329d7c403e6SMed Ismail Bennani     }
1330d7c403e6SMed Ismail Bennani 
1331b9c1b51eSKate Stone     // No need to check "process" for validity as eCommandRequiresProcess
1332b9c1b51eSKate Stone     // ensures it is valid
1333f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
13347260f620SGreg Clayton     const bool only_threads_with_stop_reason = true;
13357260f620SGreg Clayton     const uint32_t start_frame = 0;
13367260f620SGreg Clayton     const uint32_t num_frames = 1;
13377260f620SGreg Clayton     const uint32_t num_frames_with_source = 1;
13386a9767c7SJim Ingham     const bool stop_format = true;
1339c14ee32dSGreg Clayton     process->GetStatus(strm);
1340b9c1b51eSKate Stone     process->GetThreadStatus(strm, only_threads_with_stop_reason, start_frame,
13416a9767c7SJim Ingham                              num_frames, num_frames_with_source, stop_format);
1342d7c403e6SMed Ismail Bennani 
1343d7c403e6SMed Ismail Bennani     if (m_options.m_verbose) {
1344d7c403e6SMed Ismail Bennani       PlatformSP platform_sp = process->GetTarget().GetPlatform();
1345d7c403e6SMed Ismail Bennani       if (!platform_sp) {
1346d7c403e6SMed Ismail Bennani         result.AppendError("Couldn'retrieve the target's platform");
13474b9bea87SJim Ingham         return result.Succeeded();
13484b9bea87SJim Ingham       }
1349d7c403e6SMed Ismail Bennani 
1350d7c403e6SMed Ismail Bennani       auto expected_crash_info =
1351eefda182SMed Ismail Bennani           platform_sp->FetchExtendedCrashInformation(*process);
1352d7c403e6SMed Ismail Bennani 
1353d7c403e6SMed Ismail Bennani       if (!expected_crash_info) {
1354d7c403e6SMed Ismail Bennani         result.AppendError(llvm::toString(expected_crash_info.takeError()));
1355d7c403e6SMed Ismail Bennani         return result.Succeeded();
1356d7c403e6SMed Ismail Bennani       }
1357d7c403e6SMed Ismail Bennani 
1358d7c403e6SMed Ismail Bennani       StructuredData::DictionarySP crash_info_sp = *expected_crash_info;
1359d7c403e6SMed Ismail Bennani 
1360d7c403e6SMed Ismail Bennani       if (crash_info_sp) {
1361d7c403e6SMed Ismail Bennani         strm.PutCString("Extended Crash Information:\n");
1362d7c403e6SMed Ismail Bennani         crash_info_sp->Dump(strm);
1363d7c403e6SMed Ismail Bennani       }
1364d7c403e6SMed Ismail Bennani     }
1365d7c403e6SMed Ismail Bennani 
1366d7c403e6SMed Ismail Bennani     return result.Succeeded();
1367d7c403e6SMed Ismail Bennani   }
1368d7c403e6SMed Ismail Bennani 
1369d7c403e6SMed Ismail Bennani private:
1370d7c403e6SMed Ismail Bennani   CommandOptions m_options;
13714b9bea87SJim Ingham };
13724b9bea87SJim Ingham 
137335731357SCaroline Tice // CommandObjectProcessHandle
1374438dfcffSRaphael Isemann #define LLDB_OPTIONS_process_handle
1375438dfcffSRaphael Isemann #include "CommandOptions.inc"
13761f0f5b5bSZachary Turner 
1377bb9caf73SJim Ingham #pragma mark CommandObjectProcessHandle
137835731357SCaroline Tice 
1379b9c1b51eSKate Stone class CommandObjectProcessHandle : public CommandObjectParsed {
138035731357SCaroline Tice public:
1381b9c1b51eSKate Stone   class CommandOptions : public Options {
138235731357SCaroline Tice   public:
1383b9c1b51eSKate Stone     CommandOptions() : Options() { OptionParsingStarting(nullptr); }
138435731357SCaroline Tice 
138549bcfd80SEugene Zelenko     ~CommandOptions() override = default;
138635731357SCaroline Tice 
138797206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
1388b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
138997206d57SZachary Turner       Status error;
13903bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
139135731357SCaroline Tice 
1392b9c1b51eSKate Stone       switch (short_option) {
139335731357SCaroline Tice       case 's':
1394adcd0268SBenjamin Kramer         stop = std::string(option_arg);
139535731357SCaroline Tice         break;
139635731357SCaroline Tice       case 'n':
1397adcd0268SBenjamin Kramer         notify = std::string(option_arg);
139835731357SCaroline Tice         break;
139935731357SCaroline Tice       case 'p':
1400adcd0268SBenjamin Kramer         pass = std::string(option_arg);
140135731357SCaroline Tice         break;
140235731357SCaroline Tice       default:
140336162014SRaphael Isemann         llvm_unreachable("Unimplemented option");
140435731357SCaroline Tice       }
140535731357SCaroline Tice       return error;
140635731357SCaroline Tice     }
140735731357SCaroline Tice 
1408b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
140935731357SCaroline Tice       stop.clear();
141035731357SCaroline Tice       notify.clear();
141135731357SCaroline Tice       pass.clear();
141235731357SCaroline Tice     }
141335731357SCaroline Tice 
14141f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
141570602439SZachary Turner       return llvm::makeArrayRef(g_process_handle_options);
14161f0f5b5bSZachary Turner     }
141735731357SCaroline Tice 
141835731357SCaroline Tice     // Instance variables to hold the values for command options.
141935731357SCaroline Tice 
142035731357SCaroline Tice     std::string stop;
142135731357SCaroline Tice     std::string notify;
142235731357SCaroline Tice     std::string pass;
142335731357SCaroline Tice   };
142435731357SCaroline Tice 
14257428a18cSKate Stone   CommandObjectProcessHandle(CommandInterpreter &interpreter)
1426b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process handle",
1427b9c1b51eSKate Stone                             "Manage LLDB handling of OS signals for the "
1428b9c1b51eSKate Stone                             "current target process.  Defaults to showing "
1429b9c1b51eSKate Stone                             "current policy.",
143004a4c091SRaphael Isemann                             nullptr, eCommandRequiresTarget),
1431b9c1b51eSKate Stone         m_options() {
1432ea671fbdSKate Stone     SetHelpLong("\nIf no signals are specified, update them all.  If no update "
1433ea671fbdSKate Stone                 "option is specified, list the current values.");
143435731357SCaroline Tice     CommandArgumentEntry arg;
1435c0dbdfb6SCaroline Tice     CommandArgumentData signal_arg;
143635731357SCaroline Tice 
1437c0dbdfb6SCaroline Tice     signal_arg.arg_type = eArgTypeUnixSignal;
1438c0dbdfb6SCaroline Tice     signal_arg.arg_repetition = eArgRepeatStar;
143935731357SCaroline Tice 
1440c0dbdfb6SCaroline Tice     arg.push_back(signal_arg);
144135731357SCaroline Tice 
144235731357SCaroline Tice     m_arguments.push_back(arg);
144335731357SCaroline Tice   }
144435731357SCaroline Tice 
144549bcfd80SEugene Zelenko   ~CommandObjectProcessHandle() override = default;
144635731357SCaroline Tice 
1447b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
144835731357SCaroline Tice 
1449b9c1b51eSKate Stone   bool VerifyCommandOptionValue(const std::string &option, int &real_value) {
145035731357SCaroline Tice     bool okay = true;
145110ad7993SCaroline Tice     bool success = false;
145247cbf4a0SPavel Labath     bool tmp_value = OptionArgParser::ToBoolean(option, false, &success);
145310ad7993SCaroline Tice 
145410ad7993SCaroline Tice     if (success && tmp_value)
145510ad7993SCaroline Tice       real_value = 1;
145610ad7993SCaroline Tice     else if (success && !tmp_value)
145710ad7993SCaroline Tice       real_value = 0;
1458b9c1b51eSKate Stone     else {
145935731357SCaroline Tice       // If the value isn't 'true' or 'false', it had better be 0 or 1.
14609010cef2SRaphael Isemann       if (!llvm::to_integer(option, real_value))
14619010cef2SRaphael Isemann         real_value = 3;
146210ad7993SCaroline Tice       if (real_value != 0 && real_value != 1)
146335731357SCaroline Tice         okay = false;
146435731357SCaroline Tice     }
146535731357SCaroline Tice 
146635731357SCaroline Tice     return okay;
146735731357SCaroline Tice   }
146835731357SCaroline Tice 
1469b9c1b51eSKate Stone   void PrintSignalHeader(Stream &str) {
147010ad7993SCaroline Tice     str.Printf("NAME         PASS   STOP   NOTIFY\n");
1471b84141a6SPavel Labath     str.Printf("===========  =====  =====  ======\n");
147210ad7993SCaroline Tice   }
147310ad7993SCaroline Tice 
1474b9c1b51eSKate Stone   void PrintSignal(Stream &str, int32_t signo, const char *sig_name,
1475b9c1b51eSKate Stone                    const UnixSignalsSP &signals_sp) {
147610ad7993SCaroline Tice     bool stop;
147710ad7993SCaroline Tice     bool suppress;
147810ad7993SCaroline Tice     bool notify;
147910ad7993SCaroline Tice 
1480b84141a6SPavel Labath     str.Printf("%-11s  ", sig_name);
1481b9c1b51eSKate Stone     if (signals_sp->GetSignalInfo(signo, suppress, stop, notify)) {
148210ad7993SCaroline Tice       bool pass = !suppress;
1483b9c1b51eSKate Stone       str.Printf("%s  %s  %s", (pass ? "true " : "false"),
1484b9c1b51eSKate Stone                  (stop ? "true " : "false"), (notify ? "true " : "false"));
148510ad7993SCaroline Tice     }
148610ad7993SCaroline Tice     str.Printf("\n");
148710ad7993SCaroline Tice   }
148810ad7993SCaroline Tice 
1489b9c1b51eSKate Stone   void PrintSignalInformation(Stream &str, Args &signal_args,
1490b9c1b51eSKate Stone                               int num_valid_signals,
1491b9c1b51eSKate Stone                               const UnixSignalsSP &signals_sp) {
149210ad7993SCaroline Tice     PrintSignalHeader(str);
149310ad7993SCaroline Tice 
1494b9c1b51eSKate Stone     if (num_valid_signals > 0) {
149510ad7993SCaroline Tice       size_t num_args = signal_args.GetArgumentCount();
1496b9c1b51eSKate Stone       for (size_t i = 0; i < num_args; ++i) {
1497b9c1b51eSKate Stone         int32_t signo = signals_sp->GetSignalNumberFromName(
1498b9c1b51eSKate Stone             signal_args.GetArgumentAtIndex(i));
149910ad7993SCaroline Tice         if (signo != LLDB_INVALID_SIGNAL_NUMBER)
1500b9c1b51eSKate Stone           PrintSignal(str, signo, signal_args.GetArgumentAtIndex(i),
1501b9c1b51eSKate Stone                       signals_sp);
150210ad7993SCaroline Tice       }
1503b9c1b51eSKate Stone     } else // Print info for ALL signals
150410ad7993SCaroline Tice     {
150598d0a4b3SChaoren Lin       int32_t signo = signals_sp->GetFirstSignalNumber();
1506b9c1b51eSKate Stone       while (signo != LLDB_INVALID_SIGNAL_NUMBER) {
1507b9c1b51eSKate Stone         PrintSignal(str, signo, signals_sp->GetSignalAsCString(signo),
1508b9c1b51eSKate Stone                     signals_sp);
150998d0a4b3SChaoren Lin         signo = signals_sp->GetNextSignalNumber(signo);
151010ad7993SCaroline Tice       }
151110ad7993SCaroline Tice     }
151210ad7993SCaroline Tice   }
151310ad7993SCaroline Tice 
15145a988416SJim Ingham protected:
1515b9c1b51eSKate Stone   bool DoExecute(Args &signal_args, CommandReturnObject &result) override {
151604a4c091SRaphael Isemann     Target *target_sp = &GetSelectedTarget();
151735731357SCaroline Tice 
151835731357SCaroline Tice     ProcessSP process_sp = target_sp->GetProcessSP();
151935731357SCaroline Tice 
1520b9c1b51eSKate Stone     if (!process_sp) {
1521b9c1b51eSKate Stone       result.AppendError("No current process; cannot handle signals until you "
1522b9c1b51eSKate Stone                          "have a valid process.\n");
152335731357SCaroline Tice       return false;
152435731357SCaroline Tice     }
152535731357SCaroline Tice 
152635731357SCaroline Tice     int stop_action = -1;   // -1 means leave the current setting alone
152735731357SCaroline Tice     int pass_action = -1;   // -1 means leave the current setting alone
152835731357SCaroline Tice     int notify_action = -1; // -1 means leave the current setting alone
152935731357SCaroline Tice 
1530b9c1b51eSKate Stone     if (!m_options.stop.empty() &&
1531b9c1b51eSKate Stone         !VerifyCommandOptionValue(m_options.stop, stop_action)) {
1532b9c1b51eSKate Stone       result.AppendError("Invalid argument for command option --stop; must be "
1533b9c1b51eSKate Stone                          "true or false.\n");
153435731357SCaroline Tice       return false;
153535731357SCaroline Tice     }
153635731357SCaroline Tice 
1537b9c1b51eSKate Stone     if (!m_options.notify.empty() &&
1538b9c1b51eSKate Stone         !VerifyCommandOptionValue(m_options.notify, notify_action)) {
1539b9c1b51eSKate Stone       result.AppendError("Invalid argument for command option --notify; must "
1540b9c1b51eSKate Stone                          "be true or false.\n");
154135731357SCaroline Tice       return false;
154235731357SCaroline Tice     }
154335731357SCaroline Tice 
1544b9c1b51eSKate Stone     if (!m_options.pass.empty() &&
1545b9c1b51eSKate Stone         !VerifyCommandOptionValue(m_options.pass, pass_action)) {
1546b9c1b51eSKate Stone       result.AppendError("Invalid argument for command option --pass; must be "
1547b9c1b51eSKate Stone                          "true or false.\n");
154835731357SCaroline Tice       return false;
154935731357SCaroline Tice     }
155035731357SCaroline Tice 
155135731357SCaroline Tice     size_t num_args = signal_args.GetArgumentCount();
155298d0a4b3SChaoren Lin     UnixSignalsSP signals_sp = process_sp->GetUnixSignals();
155335731357SCaroline Tice     int num_signals_set = 0;
155435731357SCaroline Tice 
1555b9c1b51eSKate Stone     if (num_args > 0) {
1556d6a24757SZachary Turner       for (const auto &arg : signal_args) {
1557d6a24757SZachary Turner         int32_t signo = signals_sp->GetSignalNumberFromName(arg.c_str());
1558b9c1b51eSKate Stone         if (signo != LLDB_INVALID_SIGNAL_NUMBER) {
1559b9c1b51eSKate Stone           // Casting the actions as bools here should be okay, because
156005097246SAdrian Prantl           // VerifyCommandOptionValue guarantees the value is either 0 or 1.
156135731357SCaroline Tice           if (stop_action != -1)
156298d0a4b3SChaoren Lin             signals_sp->SetShouldStop(signo, stop_action);
1563b9c1b51eSKate Stone           if (pass_action != -1) {
156498d0a4b3SChaoren Lin             bool suppress = !pass_action;
156598d0a4b3SChaoren Lin             signals_sp->SetShouldSuppress(signo, suppress);
156635731357SCaroline Tice           }
156735731357SCaroline Tice           if (notify_action != -1)
156898d0a4b3SChaoren Lin             signals_sp->SetShouldNotify(signo, notify_action);
156935731357SCaroline Tice           ++num_signals_set;
1570b9c1b51eSKate Stone         } else {
1571b9c1b51eSKate Stone           result.AppendErrorWithFormat("Invalid signal name '%s'\n",
1572d6a24757SZachary Turner                                        arg.c_str());
157335731357SCaroline Tice         }
157435731357SCaroline Tice       }
1575b9c1b51eSKate Stone     } else {
1576b9c1b51eSKate Stone       // No signal specified, if any command options were specified, update ALL
1577b9c1b51eSKate Stone       // signals.
1578b9c1b51eSKate Stone       if ((notify_action != -1) || (stop_action != -1) || (pass_action != -1)) {
1579b9c1b51eSKate Stone         if (m_interpreter.Confirm(
1580b9c1b51eSKate Stone                 "Do you really want to update all the signals?", false)) {
158198d0a4b3SChaoren Lin           int32_t signo = signals_sp->GetFirstSignalNumber();
1582b9c1b51eSKate Stone           while (signo != LLDB_INVALID_SIGNAL_NUMBER) {
158310ad7993SCaroline Tice             if (notify_action != -1)
158498d0a4b3SChaoren Lin               signals_sp->SetShouldNotify(signo, notify_action);
158510ad7993SCaroline Tice             if (stop_action != -1)
158698d0a4b3SChaoren Lin               signals_sp->SetShouldStop(signo, stop_action);
1587b9c1b51eSKate Stone             if (pass_action != -1) {
158898d0a4b3SChaoren Lin               bool suppress = !pass_action;
158998d0a4b3SChaoren Lin               signals_sp->SetShouldSuppress(signo, suppress);
159010ad7993SCaroline Tice             }
159198d0a4b3SChaoren Lin             signo = signals_sp->GetNextSignalNumber(signo);
159210ad7993SCaroline Tice           }
159310ad7993SCaroline Tice         }
159410ad7993SCaroline Tice       }
159510ad7993SCaroline Tice     }
159610ad7993SCaroline Tice 
1597b9c1b51eSKate Stone     PrintSignalInformation(result.GetOutputStream(), signal_args,
1598b9c1b51eSKate Stone                            num_signals_set, signals_sp);
159935731357SCaroline Tice 
160035731357SCaroline Tice     if (num_signals_set > 0)
160135731357SCaroline Tice       result.SetStatus(eReturnStatusSuccessFinishNoResult);
160235731357SCaroline Tice     else
160335731357SCaroline Tice       result.SetStatus(eReturnStatusFailed);
160435731357SCaroline Tice 
160535731357SCaroline Tice     return result.Succeeded();
160635731357SCaroline Tice   }
160735731357SCaroline Tice 
160835731357SCaroline Tice   CommandOptions m_options;
160935731357SCaroline Tice };
161035731357SCaroline Tice 
16110b697561SWalter Erquinigo // Next are the subcommands of CommandObjectMultiwordProcessTrace
16120b697561SWalter Erquinigo 
16130b697561SWalter Erquinigo // CommandObjectProcessTraceStart
16140b697561SWalter Erquinigo class CommandObjectProcessTraceStart : public CommandObjectTraceProxy {
16150b697561SWalter Erquinigo public:
16160b697561SWalter Erquinigo   CommandObjectProcessTraceStart(CommandInterpreter &interpreter)
16170b697561SWalter Erquinigo       : CommandObjectTraceProxy(
16180b697561SWalter Erquinigo             /*live_debug_session_only*/ true, interpreter,
16190b697561SWalter Erquinigo             "process trace start",
16200b697561SWalter Erquinigo             "Start tracing this process with the corresponding trace "
16210b697561SWalter Erquinigo             "plug-in.",
16220b697561SWalter Erquinigo             "process trace start [<trace-options>]") {}
16230b697561SWalter Erquinigo 
16240b697561SWalter Erquinigo protected:
16250b697561SWalter Erquinigo   lldb::CommandObjectSP GetDelegateCommand(Trace &trace) override {
16260b697561SWalter Erquinigo     return trace.GetProcessTraceStartCommand(m_interpreter);
16270b697561SWalter Erquinigo   }
16280b697561SWalter Erquinigo };
16290b697561SWalter Erquinigo 
16300b697561SWalter Erquinigo // CommandObjectProcessTraceStop
16310b697561SWalter Erquinigo class CommandObjectProcessTraceStop : public CommandObjectParsed {
16320b697561SWalter Erquinigo public:
16330b697561SWalter Erquinigo   CommandObjectProcessTraceStop(CommandInterpreter &interpreter)
16340b697561SWalter Erquinigo       : CommandObjectParsed(interpreter, "process trace stop",
16350b697561SWalter Erquinigo                             "Stop tracing this process. This does not affect "
16360b697561SWalter Erquinigo                             "traces started with the "
16370b697561SWalter Erquinigo                             "\"thread trace start\" command.",
16380b697561SWalter Erquinigo                             "process trace stop",
16390b697561SWalter Erquinigo                             eCommandRequiresProcess | eCommandTryTargetAPILock |
16400b697561SWalter Erquinigo                                 eCommandProcessMustBeLaunched |
16410b697561SWalter Erquinigo                                 eCommandProcessMustBePaused |
16420b697561SWalter Erquinigo                                 eCommandProcessMustBeTraced) {}
16430b697561SWalter Erquinigo 
16440b697561SWalter Erquinigo   ~CommandObjectProcessTraceStop() override = default;
16450b697561SWalter Erquinigo 
16460b697561SWalter Erquinigo   bool DoExecute(Args &command, CommandReturnObject &result) override {
16470b697561SWalter Erquinigo     ProcessSP process_sp = m_exe_ctx.GetProcessSP();
16480b697561SWalter Erquinigo 
16490b697561SWalter Erquinigo     TraceSP trace_sp = process_sp->GetTarget().GetTrace();
16500b697561SWalter Erquinigo 
1651bf9f21a2SWalter Erquinigo     if (llvm::Error err = trace_sp->Stop())
16520b697561SWalter Erquinigo       result.SetError(toString(std::move(err)));
16530b697561SWalter Erquinigo     else
16540b697561SWalter Erquinigo       result.SetStatus(eReturnStatusSuccessFinishResult);
16550b697561SWalter Erquinigo 
16560b697561SWalter Erquinigo     return result.Succeeded();
16570b697561SWalter Erquinigo   }
16580b697561SWalter Erquinigo };
16590b697561SWalter Erquinigo 
16600b697561SWalter Erquinigo // CommandObjectMultiwordProcessTrace
16610b697561SWalter Erquinigo class CommandObjectMultiwordProcessTrace : public CommandObjectMultiword {
16620b697561SWalter Erquinigo public:
16630b697561SWalter Erquinigo   CommandObjectMultiwordProcessTrace(CommandInterpreter &interpreter)
16640b697561SWalter Erquinigo       : CommandObjectMultiword(
16650b697561SWalter Erquinigo             interpreter, "trace", "Commands for tracing the current process.",
16660b697561SWalter Erquinigo             "process trace <subcommand> [<subcommand objects>]") {
16670b697561SWalter Erquinigo     LoadSubCommand("start", CommandObjectSP(new CommandObjectProcessTraceStart(
16680b697561SWalter Erquinigo                                 interpreter)));
16690b697561SWalter Erquinigo     LoadSubCommand("stop", CommandObjectSP(
16700b697561SWalter Erquinigo                                new CommandObjectProcessTraceStop(interpreter)));
16710b697561SWalter Erquinigo   }
16720b697561SWalter Erquinigo 
16730b697561SWalter Erquinigo   ~CommandObjectMultiwordProcessTrace() override = default;
16740b697561SWalter Erquinigo };
16750b697561SWalter Erquinigo 
167630fdc8d8SChris Lattner // CommandObjectMultiwordProcess
167730fdc8d8SChris Lattner 
1678b9c1b51eSKate Stone CommandObjectMultiwordProcess::CommandObjectMultiwordProcess(
1679b9c1b51eSKate Stone     CommandInterpreter &interpreter)
1680b9c1b51eSKate Stone     : CommandObjectMultiword(
1681b9c1b51eSKate Stone           interpreter, "process",
1682b9c1b51eSKate Stone           "Commands for interacting with processes on the current platform.",
1683b9c1b51eSKate Stone           "process <subcommand> [<subcommand-options>]") {
1684b9c1b51eSKate Stone   LoadSubCommand("attach",
1685b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessAttach(interpreter)));
1686b9c1b51eSKate Stone   LoadSubCommand("launch",
1687b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessLaunch(interpreter)));
1688b9c1b51eSKate Stone   LoadSubCommand("continue", CommandObjectSP(new CommandObjectProcessContinue(
1689b9c1b51eSKate Stone                                  interpreter)));
1690b9c1b51eSKate Stone   LoadSubCommand("connect",
1691b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessConnect(interpreter)));
1692b9c1b51eSKate Stone   LoadSubCommand("detach",
1693b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessDetach(interpreter)));
1694b9c1b51eSKate Stone   LoadSubCommand("load",
1695b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessLoad(interpreter)));
1696b9c1b51eSKate Stone   LoadSubCommand("unload",
1697b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessUnload(interpreter)));
1698b9c1b51eSKate Stone   LoadSubCommand("signal",
1699b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessSignal(interpreter)));
1700b9c1b51eSKate Stone   LoadSubCommand("handle",
1701b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessHandle(interpreter)));
1702b9c1b51eSKate Stone   LoadSubCommand("status",
1703b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessStatus(interpreter)));
1704b9c1b51eSKate Stone   LoadSubCommand("interrupt", CommandObjectSP(new CommandObjectProcessInterrupt(
1705b9c1b51eSKate Stone                                   interpreter)));
1706b9c1b51eSKate Stone   LoadSubCommand("kill",
1707b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessKill(interpreter)));
1708b9c1b51eSKate Stone   LoadSubCommand("plugin",
1709b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessPlugin(interpreter)));
1710b9c1b51eSKate Stone   LoadSubCommand("save-core", CommandObjectSP(new CommandObjectProcessSaveCore(
1711b9c1b51eSKate Stone                                   interpreter)));
17120b697561SWalter Erquinigo   LoadSubCommand(
17130b697561SWalter Erquinigo       "trace",
17140b697561SWalter Erquinigo       CommandObjectSP(new CommandObjectMultiwordProcessTrace(interpreter)));
171530fdc8d8SChris Lattner }
171630fdc8d8SChris Lattner 
171749bcfd80SEugene Zelenko CommandObjectMultiwordProcess::~CommandObjectMultiwordProcess() = default;
1718