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 
32cdc6f8d7SJason Molenda #include <bitset>
33cdc6f8d7SJason Molenda 
3430fdc8d8SChris Lattner using namespace lldb;
3530fdc8d8SChris Lattner using namespace lldb_private;
3630fdc8d8SChris Lattner 
37b9c1b51eSKate Stone class CommandObjectProcessLaunchOrAttach : public CommandObjectParsed {
38dcb1d856SJim Ingham public:
39dcb1d856SJim Ingham   CommandObjectProcessLaunchOrAttach(CommandInterpreter &interpreter,
40b9c1b51eSKate Stone                                      const char *name, const char *help,
41b9c1b51eSKate Stone                                      const char *syntax, uint32_t flags,
42b9c1b51eSKate Stone                                      const char *new_process_action)
43b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, name, help, syntax, flags),
44dcb1d856SJim Ingham         m_new_process_action(new_process_action) {}
45dcb1d856SJim Ingham 
4649bcfd80SEugene Zelenko   ~CommandObjectProcessLaunchOrAttach() override = default;
4749bcfd80SEugene Zelenko 
48dcb1d856SJim Ingham protected:
49b9c1b51eSKate Stone   bool StopProcessIfNecessary(Process *process, StateType &state,
50b9c1b51eSKate Stone                               CommandReturnObject &result) {
51dcb1d856SJim Ingham     state = eStateInvalid;
52b9c1b51eSKate Stone     if (process) {
53dcb1d856SJim Ingham       state = process->GetState();
54dcb1d856SJim Ingham 
55b9c1b51eSKate Stone       if (process->IsAlive() && state != eStateConnected) {
56db203e02SRaphael Isemann         std::string message;
57dcb1d856SJim Ingham         if (process->GetState() == eStateAttaching)
58db203e02SRaphael Isemann           message =
59db203e02SRaphael Isemann               llvm::formatv("There is a pending attach, abort it and {0}?",
60db203e02SRaphael Isemann                             m_new_process_action);
61dcb1d856SJim Ingham         else if (process->GetShouldDetach())
62db203e02SRaphael Isemann           message = llvm::formatv(
63db203e02SRaphael Isemann               "There is a running process, detach from it and {0}?",
64db203e02SRaphael Isemann               m_new_process_action);
65dcb1d856SJim Ingham         else
66db203e02SRaphael Isemann           message =
67db203e02SRaphael Isemann               llvm::formatv("There is a running process, kill it and {0}?",
68db203e02SRaphael Isemann                             m_new_process_action);
69dcb1d856SJim Ingham 
70b9c1b51eSKate Stone         if (!m_interpreter.Confirm(message, true)) {
71dcb1d856SJim Ingham           result.SetStatus(eReturnStatusFailed);
72dcb1d856SJim Ingham           return false;
73b9c1b51eSKate Stone         } else {
74b9c1b51eSKate Stone           if (process->GetShouldDetach()) {
75acff8950SJim Ingham             bool keep_stopped = false;
7697206d57SZachary Turner             Status detach_error(process->Detach(keep_stopped));
77b9c1b51eSKate Stone             if (detach_error.Success()) {
78dcb1d856SJim Ingham               result.SetStatus(eReturnStatusSuccessFinishResult);
7949bcfd80SEugene Zelenko               process = nullptr;
80b9c1b51eSKate Stone             } else {
81b9c1b51eSKate Stone               result.AppendErrorWithFormat(
82b9c1b51eSKate Stone                   "Failed to detach from process: %s\n",
83b9c1b51eSKate Stone                   detach_error.AsCString());
84dcb1d856SJim Ingham             }
85b9c1b51eSKate Stone           } else {
8697206d57SZachary Turner             Status destroy_error(process->Destroy(false));
87b9c1b51eSKate Stone             if (destroy_error.Success()) {
88dcb1d856SJim Ingham               result.SetStatus(eReturnStatusSuccessFinishResult);
8949bcfd80SEugene Zelenko               process = nullptr;
90b9c1b51eSKate Stone             } else {
91b9c1b51eSKate Stone               result.AppendErrorWithFormat("Failed to kill process: %s\n",
92b9c1b51eSKate Stone                                            destroy_error.AsCString());
93dcb1d856SJim Ingham             }
94dcb1d856SJim Ingham           }
95dcb1d856SJim Ingham         }
96dcb1d856SJim Ingham       }
97dcb1d856SJim Ingham     }
98dcb1d856SJim Ingham     return result.Succeeded();
99dcb1d856SJim Ingham   }
10049bcfd80SEugene Zelenko 
101dcb1d856SJim Ingham   std::string m_new_process_action;
102dcb1d856SJim Ingham };
10349bcfd80SEugene Zelenko 
10430fdc8d8SChris Lattner // CommandObjectProcessLaunch
1054bddaeb5SJim Ingham #pragma mark CommandObjectProcessLaunch
106b9c1b51eSKate Stone class CommandObjectProcessLaunch : public CommandObjectProcessLaunchOrAttach {
10730fdc8d8SChris Lattner public:
108b9c1b51eSKate Stone   CommandObjectProcessLaunch(CommandInterpreter &interpreter)
109b9c1b51eSKate Stone       : CommandObjectProcessLaunchOrAttach(
110b9c1b51eSKate Stone             interpreter, "process launch",
111b9c1b51eSKate Stone             "Launch the executable in the debugger.", nullptr,
112b9c1b51eSKate Stone             eCommandRequiresTarget, "restart"),
1133e0ad115SMed Ismail Bennani         m_options(),
1143e0ad115SMed Ismail Bennani         m_class_options("scripted process", true, 'C', 'k', 'v', 0),
1153e0ad115SMed Ismail Bennani         m_all_options() {
1163e0ad115SMed Ismail Bennani     m_all_options.Append(&m_options);
1173e0ad115SMed Ismail Bennani     m_all_options.Append(&m_class_options, LLDB_OPT_SET_1 | LLDB_OPT_SET_2,
1183e0ad115SMed Ismail Bennani                          LLDB_OPT_SET_ALL);
1193e0ad115SMed Ismail Bennani     m_all_options.Finalize();
1203e0ad115SMed Ismail Bennani 
121405fe67fSCaroline Tice     CommandArgumentEntry arg;
122405fe67fSCaroline Tice     CommandArgumentData run_args_arg;
123405fe67fSCaroline Tice 
124405fe67fSCaroline Tice     // Define the first (and only) variant of this arg.
125405fe67fSCaroline Tice     run_args_arg.arg_type = eArgTypeRunArgs;
126405fe67fSCaroline Tice     run_args_arg.arg_repetition = eArgRepeatOptional;
127405fe67fSCaroline Tice 
128b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
129b9c1b51eSKate Stone     // argument entry.
130405fe67fSCaroline Tice     arg.push_back(run_args_arg);
131405fe67fSCaroline Tice 
132405fe67fSCaroline Tice     // Push the data for the first argument into the m_arguments vector.
133405fe67fSCaroline Tice     m_arguments.push_back(arg);
13430fdc8d8SChris Lattner   }
13530fdc8d8SChris Lattner 
13649bcfd80SEugene Zelenko   ~CommandObjectProcessLaunch() override = default;
13730fdc8d8SChris Lattner 
138ae34ed2cSRaphael Isemann   void
139ae34ed2cSRaphael Isemann   HandleArgumentCompletion(CompletionRequest &request,
1402443bbd4SRaphael Isemann                            OptionElementVector &opt_element_vector) override {
141e9ce62b6SJim Ingham 
142b9c1b51eSKate Stone     CommandCompletions::InvokeCommonCompletionCallbacks(
143b9c1b51eSKate Stone         GetCommandInterpreter(), CommandCompletions::eDiskFileCompletion,
144a2e76c0bSRaphael Isemann         request, nullptr);
145e9ce62b6SJim Ingham   }
146e9ce62b6SJim Ingham 
1473e0ad115SMed Ismail Bennani   Options *GetOptions() override { return &m_all_options; }
14830fdc8d8SChris Lattner 
149b9c1b51eSKate Stone   const char *GetRepeatCommand(Args &current_command_args,
150b9c1b51eSKate Stone                                uint32_t index) override {
1515a988416SJim Ingham     // No repeat for "process launch"...
1525a988416SJim Ingham     return "";
1535a988416SJim Ingham   }
1545a988416SJim Ingham 
1555a988416SJim Ingham protected:
156b9c1b51eSKate Stone   bool DoExecute(Args &launch_args, CommandReturnObject &result) override {
15757179860SJonas Devlieghere     Debugger &debugger = GetDebugger();
1581d885966SGreg Clayton     Target *target = debugger.GetSelectedTarget().get();
15949bcfd80SEugene Zelenko     // If our listener is nullptr, users aren't allows to launch
160b09c5384SGreg Clayton     ModuleSP exe_module_sp = target->GetExecutableModule();
16171337622SGreg Clayton 
162b9c1b51eSKate Stone     if (exe_module_sp == nullptr) {
163b9c1b51eSKate Stone       result.AppendError("no file in target, create a debug target using the "
164b9c1b51eSKate Stone                          "'target create' command");
16571337622SGreg Clayton       return false;
16671337622SGreg Clayton     }
16771337622SGreg Clayton 
16871337622SGreg Clayton     StateType state = eStateInvalid;
16971337622SGreg Clayton 
170b09c5384SGreg Clayton     if (!StopProcessIfNecessary(m_exe_ctx.GetProcessPtr(), state, result))
17130fdc8d8SChris Lattner       return false;
17230fdc8d8SChris Lattner 
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 
217be556d51SMed Ismail Bennani     llvm::StringRef target_settings_argv0 = target->GetArg0();
218be556d51SMed Ismail Bennani 
21931d97a5cSZachary Turner     if (!target_settings_argv0.empty()) {
220b9c1b51eSKate Stone       m_options.launch_info.GetArguments().AppendArgument(
22131d97a5cSZachary Turner           target_settings_argv0);
222b9c1b51eSKate Stone       m_options.launch_info.SetExecutableFile(
223b9c1b51eSKate Stone           exe_module_sp->GetPlatformFileSpec(), false);
224b9c1b51eSKate Stone     } else {
225b9c1b51eSKate Stone       m_options.launch_info.SetExecutableFile(
226b9c1b51eSKate Stone           exe_module_sp->GetPlatformFileSpec(), true);
22745392553SGreg Clayton     }
22845392553SGreg Clayton 
229b9c1b51eSKate Stone     if (launch_args.GetArgumentCount() == 0) {
230b9c1b51eSKate Stone       m_options.launch_info.GetArguments().AppendArguments(
231b9c1b51eSKate Stone           target->GetProcessLaunchInfo().GetArguments());
232b9c1b51eSKate Stone     } else {
23345392553SGreg Clayton       m_options.launch_info.GetArguments().AppendArguments(launch_args);
234162b597cSGreg Clayton       // Save the arguments for subsequent runs in the current target.
235162b597cSGreg Clayton       target->SetRunArguments(launch_args);
236982c9762SGreg Clayton     }
2371d885966SGreg Clayton 
238dc6224e0SGreg Clayton     StreamString stream;
23997206d57SZachary Turner     Status error = target->Launch(m_options.launch_info, &stream);
24030fdc8d8SChris Lattner 
241b9c1b51eSKate Stone     if (error.Success()) {
242b09c5384SGreg Clayton       ProcessSP process_sp(target->GetProcessSP());
243b9c1b51eSKate Stone       if (process_sp) {
244b9c1b51eSKate Stone         // There is a race condition where this thread will return up the call
24505097246SAdrian Prantl         // stack to the main command handler and show an (lldb) prompt before
24605097246SAdrian Prantl         // HandlePrivateEvent (from PrivateStateThread) has a chance to call
24705097246SAdrian Prantl         // PushProcessIOHandler().
2483879fe00SPavel Labath         process_sp->SyncIOHandler(0, std::chrono::seconds(2));
2498f0db3e1SIlia K 
250c156427dSZachary Turner         llvm::StringRef data = stream.GetString();
251c156427dSZachary Turner         if (!data.empty())
252c156427dSZachary Turner           result.AppendMessage(data);
253b9c1b51eSKate Stone         const char *archname =
254b9c1b51eSKate Stone             exe_module_sp->GetArchitecture().GetArchitectureName();
255b9c1b51eSKate Stone         result.AppendMessageWithFormat(
256b9c1b51eSKate Stone             "Process %" PRIu64 " launched: '%s' (%s)\n", process_sp->GetID(),
257b9c1b51eSKate Stone             exe_module_sp->GetFileSpec().GetPath().c_str(), archname);
25805faeb71SGreg Clayton         result.SetStatus(eReturnStatusSuccessFinishResult);
259b09c5384SGreg Clayton         result.SetDidChangeProcessState(true);
260b9c1b51eSKate Stone       } else {
261b9c1b51eSKate Stone         result.AppendError(
262b9c1b51eSKate Stone             "no error returned from Target::Launch, and target has no process");
26330fdc8d8SChris Lattner       }
264b9c1b51eSKate Stone     } else {
265b09c5384SGreg Clayton       result.AppendError(error.AsCString());
266514487e8SGreg Clayton     }
26730fdc8d8SChris Lattner     return result.Succeeded();
26830fdc8d8SChris Lattner   }
26930fdc8d8SChris Lattner 
2707169d3a3SMed Ismail Bennani   CommandOptionsProcessLaunch m_options;
2713e0ad115SMed Ismail Bennani   OptionGroupPythonClassWithDict m_class_options;
2723e0ad115SMed Ismail Bennani   OptionGroupOptions m_all_options;
27330fdc8d8SChris Lattner };
27430fdc8d8SChris Lattner 
275438dfcffSRaphael Isemann #define LLDB_OPTIONS_process_attach
276438dfcffSRaphael Isemann #include "CommandOptions.inc"
2771f0f5b5bSZachary Turner 
278bb9caf73SJim Ingham #pragma mark CommandObjectProcessAttach
279b9c1b51eSKate Stone class CommandObjectProcessAttach : public CommandObjectProcessLaunchOrAttach {
28030fdc8d8SChris Lattner public:
281b9c1b51eSKate Stone   class CommandOptions : public Options {
28230fdc8d8SChris Lattner   public:
283b9c1b51eSKate Stone     CommandOptions() : Options() {
284b9c1b51eSKate Stone       // Keep default values of all options in one place: OptionParsingStarting
285b9c1b51eSKate Stone       // ()
286e1cfbc79STodd Fiala       OptionParsingStarting(nullptr);
28730fdc8d8SChris Lattner     }
28830fdc8d8SChris Lattner 
28949bcfd80SEugene Zelenko     ~CommandOptions() override = default;
29030fdc8d8SChris Lattner 
29197206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
292b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
29397206d57SZachary Turner       Status error;
2943bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
295b9c1b51eSKate Stone       switch (short_option) {
296a95ce623SJohnny Chen       case 'c':
297a95ce623SJohnny Chen         attach_info.SetContinueOnceAttached(true);
298a95ce623SJohnny Chen         break;
299a95ce623SJohnny Chen 
300b9c1b51eSKate Stone       case 'p': {
301fe11483bSZachary Turner         lldb::pid_t pid;
302fe11483bSZachary Turner         if (option_arg.getAsInteger(0, pid)) {
303fe11483bSZachary Turner           error.SetErrorStringWithFormat("invalid process ID '%s'",
304fe11483bSZachary Turner                                          option_arg.str().c_str());
305b9c1b51eSKate Stone         } else {
306144f3a9cSGreg Clayton           attach_info.SetProcessID(pid);
307144f3a9cSGreg Clayton         }
308b9c1b51eSKate Stone       } break;
30930fdc8d8SChris Lattner 
31030fdc8d8SChris Lattner       case 'P':
311144f3a9cSGreg Clayton         attach_info.SetProcessPluginName(option_arg);
31230fdc8d8SChris Lattner         break;
31330fdc8d8SChris Lattner 
31430fdc8d8SChris Lattner       case 'n':
3158f3be7a3SJonas Devlieghere         attach_info.GetExecutableFile().SetFile(option_arg,
316937348cdSJonas Devlieghere                                                 FileSpec::Style::native);
31730fdc8d8SChris Lattner         break;
31830fdc8d8SChris Lattner 
31930fdc8d8SChris Lattner       case 'w':
320144f3a9cSGreg Clayton         attach_info.SetWaitForLaunch(true);
32130fdc8d8SChris Lattner         break;
32230fdc8d8SChris Lattner 
323cd16df91SJim Ingham       case 'i':
324cd16df91SJim Ingham         attach_info.SetIgnoreExisting(false);
325cd16df91SJim Ingham         break;
326cd16df91SJim Ingham 
32730fdc8d8SChris Lattner       default:
32836162014SRaphael Isemann         llvm_unreachable("Unimplemented option");
32930fdc8d8SChris Lattner       }
33030fdc8d8SChris Lattner       return error;
33130fdc8d8SChris Lattner     }
33230fdc8d8SChris Lattner 
333b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
334144f3a9cSGreg Clayton       attach_info.Clear();
33530fdc8d8SChris Lattner     }
33630fdc8d8SChris Lattner 
3371f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
33870602439SZachary Turner       return llvm::makeArrayRef(g_process_attach_options);
3391f0f5b5bSZachary Turner     }
34030fdc8d8SChris Lattner 
341144f3a9cSGreg Clayton     ProcessAttachInfo attach_info;
34230fdc8d8SChris Lattner   };
34330fdc8d8SChris Lattner 
344b9c1b51eSKate Stone   CommandObjectProcessAttach(CommandInterpreter &interpreter)
345b9c1b51eSKate Stone       : CommandObjectProcessLaunchOrAttach(
346b9c1b51eSKate Stone             interpreter, "process attach", "Attach to a process.",
347b9c1b51eSKate Stone             "process attach <cmd-options>", 0, "attach"),
348b9c1b51eSKate Stone         m_options() {}
3495aee162fSJim Ingham 
35049bcfd80SEugene Zelenko   ~CommandObjectProcessAttach() override = default;
3515aee162fSJim Ingham 
352b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
3535a988416SJim Ingham 
3545a988416SJim Ingham protected:
355b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
356b9c1b51eSKate Stone     PlatformSP platform_sp(
35757179860SJonas Devlieghere         GetDebugger().GetPlatformList().GetSelectedPlatform());
358926af0cdSOleksiy Vyalov 
35957179860SJonas Devlieghere     Target *target = GetDebugger().GetSelectedTarget().get();
360b9c1b51eSKate Stone     // N.B. The attach should be synchronous.  It doesn't help much to get the
36105097246SAdrian Prantl     // prompt back between initiating the attach and the target actually
36205097246SAdrian Prantl     // stopping.  So even if the interpreter is set to be asynchronous, we wait
36305097246SAdrian Prantl     // for the stop ourselves here.
3645aee162fSJim Ingham 
36571337622SGreg Clayton     StateType state = eStateInvalid;
366dcb1d856SJim Ingham     Process *process = m_exe_ctx.GetProcessPtr();
367dcb1d856SJim Ingham 
368dcb1d856SJim Ingham     if (!StopProcessIfNecessary(process, state, result))
3695aee162fSJim Ingham       return false;
3705aee162fSJim Ingham 
371b9c1b51eSKate Stone     if (target == nullptr) {
3725aee162fSJim Ingham       // If there isn't a current target create one.
3735aee162fSJim Ingham       TargetSP new_target_sp;
37497206d57SZachary Turner       Status error;
3755aee162fSJim Ingham 
37657179860SJonas Devlieghere       error = GetDebugger().GetTargetList().CreateTarget(
37757179860SJonas Devlieghere           GetDebugger(), "", "", eLoadDependentsNo,
37849bcfd80SEugene Zelenko           nullptr, // No platform options
3795aee162fSJim Ingham           new_target_sp);
3805aee162fSJim Ingham       target = new_target_sp.get();
381b9c1b51eSKate Stone       if (target == nullptr || error.Fail()) {
382b766a73dSGreg Clayton         result.AppendError(error.AsCString("Error creating target"));
3835aee162fSJim Ingham         return false;
3845aee162fSJim Ingham       }
3855aee162fSJim Ingham     }
3865aee162fSJim Ingham 
387b9c1b51eSKate Stone     // Record the old executable module, we want to issue a warning if the
38805097246SAdrian Prantl     // process of attaching changed the current executable (like somebody said
38905097246SAdrian Prantl     // "file foo" then attached to a PID whose executable was bar.)
3905aee162fSJim Ingham 
3915aee162fSJim Ingham     ModuleSP old_exec_module_sp = target->GetExecutableModule();
3925aee162fSJim Ingham     ArchSpec old_arch_spec = target->GetArchitecture();
3935aee162fSJim Ingham 
394b9c1b51eSKate Stone     if (command.GetArgumentCount()) {
395b9c1b51eSKate Stone       result.AppendErrorWithFormat("Invalid arguments for '%s'.\nUsage: %s\n",
396b9c1b51eSKate Stone                                    m_cmd_name.c_str(), m_cmd_syntax.c_str());
397926af0cdSOleksiy Vyalov       return false;
39871337622SGreg Clayton     }
3995aee162fSJim Ingham 
40037386143SOleksiy Vyalov     StreamString stream;
40137386143SOleksiy Vyalov     const auto error = target->Attach(m_options.attach_info, &stream);
402b9c1b51eSKate Stone     if (error.Success()) {
40337386143SOleksiy Vyalov       ProcessSP process_sp(target->GetProcessSP());
404b9c1b51eSKate Stone       if (process_sp) {
405c156427dSZachary Turner         result.AppendMessage(stream.GetString());
406bb3a283bSJim Ingham         result.SetStatus(eReturnStatusSuccessFinishNoResult);
40737386143SOleksiy Vyalov         result.SetDidChangeProcessState(true);
408b9c1b51eSKate Stone       } else {
409b9c1b51eSKate Stone         result.AppendError(
410b9c1b51eSKate Stone             "no error returned from Target::Attach, and target has no process");
41144d93782SGreg Clayton       }
412b9c1b51eSKate Stone     } else {
41344d93782SGreg Clayton       result.AppendErrorWithFormat("attach failed: %s\n", error.AsCString());
414aa739093SJohnny Chen     }
4155aee162fSJim Ingham 
416926af0cdSOleksiy Vyalov     if (!result.Succeeded())
417926af0cdSOleksiy Vyalov       return false;
418926af0cdSOleksiy Vyalov 
419b9c1b51eSKate Stone     // Okay, we're done.  Last step is to warn if the executable module has
420b9c1b51eSKate Stone     // changed:
421513c26ceSGreg Clayton     char new_path[PATH_MAX];
422aa149cbdSGreg Clayton     ModuleSP new_exec_module_sp(target->GetExecutableModule());
423b9c1b51eSKate Stone     if (!old_exec_module_sp) {
424513c26ceSGreg Clayton       // We might not have a module if we attached to a raw pid...
425b9c1b51eSKate Stone       if (new_exec_module_sp) {
426aa149cbdSGreg Clayton         new_exec_module_sp->GetFileSpec().GetPath(new_path, PATH_MAX);
427b9c1b51eSKate Stone         result.AppendMessageWithFormat("Executable module set to \"%s\".\n",
428b9c1b51eSKate Stone                                        new_path);
429513c26ceSGreg Clayton       }
430b9c1b51eSKate Stone     } else if (old_exec_module_sp->GetFileSpec() !=
431b9c1b51eSKate Stone                new_exec_module_sp->GetFileSpec()) {
432513c26ceSGreg Clayton       char old_path[PATH_MAX];
4335aee162fSJim Ingham 
4345aee162fSJim Ingham       old_exec_module_sp->GetFileSpec().GetPath(old_path, PATH_MAX);
435aa149cbdSGreg Clayton       new_exec_module_sp->GetFileSpec().GetPath(new_path, PATH_MAX);
4365aee162fSJim Ingham 
437b9c1b51eSKate Stone       result.AppendWarningWithFormat(
438b9c1b51eSKate Stone           "Executable module changed from \"%s\" to \"%s\".\n", old_path,
439b9c1b51eSKate Stone           new_path);
4405aee162fSJim Ingham     }
4415aee162fSJim Ingham 
442b9c1b51eSKate Stone     if (!old_arch_spec.IsValid()) {
443b9c1b51eSKate Stone       result.AppendMessageWithFormat(
444b9c1b51eSKate Stone           "Architecture set to: %s.\n",
445b9c1b51eSKate Stone           target->GetArchitecture().GetTriple().getTriple().c_str());
446b9c1b51eSKate Stone     } else if (!old_arch_spec.IsExactMatch(target->GetArchitecture())) {
447b9c1b51eSKate Stone       result.AppendWarningWithFormat(
448b9c1b51eSKate Stone           "Architecture changed from %s to %s.\n",
449c1b1f1eaSGreg Clayton           old_arch_spec.GetTriple().getTriple().c_str(),
450c1b1f1eaSGreg Clayton           target->GetArchitecture().GetTriple().getTriple().c_str());
4515aee162fSJim Ingham     }
452a95ce623SJohnny Chen 
45305097246SAdrian Prantl     // This supports the use-case scenario of immediately continuing the
45405097246SAdrian Prantl     // process once attached.
455a95ce623SJohnny Chen     if (m_options.attach_info.GetContinueOnceAttached())
4565bcaf583SSean Callanan       m_interpreter.HandleCommand("process continue", eLazyBoolNo, result);
457926af0cdSOleksiy Vyalov 
4585aee162fSJim Ingham     return result.Succeeded();
4595aee162fSJim Ingham   }
4605aee162fSJim Ingham 
46130fdc8d8SChris Lattner   CommandOptions m_options;
46230fdc8d8SChris Lattner };
46330fdc8d8SChris Lattner 
46430fdc8d8SChris Lattner // CommandObjectProcessContinue
4651f0f5b5bSZachary Turner 
466438dfcffSRaphael Isemann #define LLDB_OPTIONS_process_continue
467438dfcffSRaphael Isemann #include "CommandOptions.inc"
4681f0f5b5bSZachary Turner 
469bb9caf73SJim Ingham #pragma mark CommandObjectProcessContinue
47030fdc8d8SChris Lattner 
471b9c1b51eSKate Stone class CommandObjectProcessContinue : public CommandObjectParsed {
47230fdc8d8SChris Lattner public:
473b9c1b51eSKate Stone   CommandObjectProcessContinue(CommandInterpreter &interpreter)
474b9c1b51eSKate Stone       : CommandObjectParsed(
475b9c1b51eSKate Stone             interpreter, "process continue",
476e3d26315SCaroline Tice             "Continue execution of all threads in the current process.",
47730fdc8d8SChris Lattner             "process continue",
478b9c1b51eSKate Stone             eCommandRequiresProcess | eCommandTryTargetAPILock |
479b9c1b51eSKate Stone                 eCommandProcessMustBeLaunched | eCommandProcessMustBePaused),
480b9c1b51eSKate Stone         m_options() {}
48130fdc8d8SChris Lattner 
48249bcfd80SEugene Zelenko   ~CommandObjectProcessContinue() override = default;
48330fdc8d8SChris Lattner 
4845a988416SJim Ingham protected:
485b9c1b51eSKate Stone   class CommandOptions : public Options {
4860e41084aSJim Ingham   public:
487b9c1b51eSKate Stone     CommandOptions() : Options() {
488b9c1b51eSKate Stone       // Keep default values of all options in one place: OptionParsingStarting
489b9c1b51eSKate Stone       // ()
490e1cfbc79STodd Fiala       OptionParsingStarting(nullptr);
4910e41084aSJim Ingham     }
4920e41084aSJim Ingham 
49349bcfd80SEugene Zelenko     ~CommandOptions() override = default;
4940e41084aSJim Ingham 
49597206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
496b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
49797206d57SZachary Turner       Status error;
4983bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
499b9c1b51eSKate Stone       switch (short_option) {
5000e41084aSJim Ingham       case 'i':
501fe11483bSZachary Turner         if (option_arg.getAsInteger(0, m_ignore))
502b9c1b51eSKate Stone           error.SetErrorStringWithFormat(
503b9c1b51eSKate Stone               "invalid value for ignore option: \"%s\", should be a number.",
504fe11483bSZachary Turner               option_arg.str().c_str());
5050e41084aSJim Ingham         break;
5060e41084aSJim Ingham 
5070e41084aSJim Ingham       default:
50836162014SRaphael Isemann         llvm_unreachable("Unimplemented option");
5090e41084aSJim Ingham       }
5100e41084aSJim Ingham       return error;
5110e41084aSJim Ingham     }
5120e41084aSJim Ingham 
513b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
5140e41084aSJim Ingham       m_ignore = 0;
5150e41084aSJim Ingham     }
5160e41084aSJim Ingham 
5171f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
51870602439SZachary Turner       return llvm::makeArrayRef(g_process_continue_options);
5191f0f5b5bSZachary Turner     }
5200e41084aSJim Ingham 
5210e41084aSJim Ingham     uint32_t m_ignore;
5220e41084aSJim Ingham   };
5230e41084aSJim Ingham 
524b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
525f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
526a7015092SGreg Clayton     bool synchronous_execution = m_interpreter.GetSynchronous();
52730fdc8d8SChris Lattner     StateType state = process->GetState();
528b9c1b51eSKate Stone     if (state == eStateStopped) {
529b9c1b51eSKate Stone       if (command.GetArgumentCount() != 0) {
530b9c1b51eSKate Stone         result.AppendErrorWithFormat(
531b9c1b51eSKate Stone             "The '%s' command does not take any arguments.\n",
532b9c1b51eSKate Stone             m_cmd_name.c_str());
53330fdc8d8SChris Lattner         return false;
53430fdc8d8SChris Lattner       }
53530fdc8d8SChris Lattner 
536b9c1b51eSKate Stone       if (m_options.m_ignore > 0) {
5378d94ba0fSJim Ingham         ThreadSP sel_thread_sp(GetDefaultThread()->shared_from_this());
538b9c1b51eSKate Stone         if (sel_thread_sp) {
5390e41084aSJim Ingham           StopInfoSP stop_info_sp = sel_thread_sp->GetStopInfo();
540b9c1b51eSKate Stone           if (stop_info_sp &&
541b9c1b51eSKate Stone               stop_info_sp->GetStopReason() == eStopReasonBreakpoint) {
542b9c1b51eSKate Stone             lldb::break_id_t bp_site_id =
543b9c1b51eSKate Stone                 (lldb::break_id_t)stop_info_sp->GetValue();
544b9c1b51eSKate Stone             BreakpointSiteSP bp_site_sp(
545b9c1b51eSKate Stone                 process->GetBreakpointSiteList().FindByID(bp_site_id));
546b9c1b51eSKate Stone             if (bp_site_sp) {
547c7bece56SGreg Clayton               const size_t num_owners = bp_site_sp->GetNumberOfOwners();
548b9c1b51eSKate Stone               for (size_t i = 0; i < num_owners; i++) {
549b9c1b51eSKate Stone                 Breakpoint &bp_ref =
550b9c1b51eSKate Stone                     bp_site_sp->GetOwnerAtIndex(i)->GetBreakpoint();
551b9c1b51eSKate Stone                 if (!bp_ref.IsInternal()) {
5520e41084aSJim Ingham                   bp_ref.SetIgnoreCount(m_options.m_ignore);
5530e41084aSJim Ingham                 }
5540e41084aSJim Ingham               }
5550e41084aSJim Ingham             }
5560e41084aSJim Ingham           }
5570e41084aSJim Ingham         }
5580e41084aSJim Ingham       }
5590e41084aSJim Ingham 
56041f2b940SJim Ingham       { // Scope for thread list mutex:
561b9c1b51eSKate Stone         std::lock_guard<std::recursive_mutex> guard(
562b9c1b51eSKate Stone             process->GetThreadList().GetMutex());
56330fdc8d8SChris Lattner         const uint32_t num_threads = process->GetThreadList().GetSize();
56430fdc8d8SChris Lattner 
56530fdc8d8SChris Lattner         // Set the actions that the threads should each take when resuming
566b9c1b51eSKate Stone         for (uint32_t idx = 0; idx < num_threads; ++idx) {
5676c9ed91cSJim Ingham           const bool override_suspend = false;
568b9c1b51eSKate Stone           process->GetThreadList().GetThreadAtIndex(idx)->SetResumeState(
569b9c1b51eSKate Stone               eStateRunning, override_suspend);
57030fdc8d8SChris Lattner         }
57141f2b940SJim Ingham       }
57230fdc8d8SChris Lattner 
5734446487dSPavel Labath       const uint32_t iohandler_id = process->GetIOHandlerID();
5744446487dSPavel Labath 
575dc6224e0SGreg Clayton       StreamString stream;
57697206d57SZachary Turner       Status error;
577dc6224e0SGreg Clayton       if (synchronous_execution)
578dc6224e0SGreg Clayton         error = process->ResumeSynchronous(&stream);
579dc6224e0SGreg Clayton       else
580dc6224e0SGreg Clayton         error = process->Resume();
581a3b89e27STodd Fiala 
582b9c1b51eSKate Stone       if (error.Success()) {
583b9c1b51eSKate Stone         // There is a race condition where this thread will return up the call
58405097246SAdrian Prantl         // stack to the main command handler and show an (lldb) prompt before
58505097246SAdrian Prantl         // HandlePrivateEvent (from PrivateStateThread) has a chance to call
58605097246SAdrian Prantl         // PushProcessIOHandler().
5873879fe00SPavel Labath         process->SyncIOHandler(iohandler_id, std::chrono::seconds(2));
588a3b89e27STodd Fiala 
589b9c1b51eSKate Stone         result.AppendMessageWithFormat("Process %" PRIu64 " resuming\n",
590b9c1b51eSKate Stone                                        process->GetID());
591b9c1b51eSKate Stone         if (synchronous_execution) {
592b9c1b51eSKate Stone           // If any state changed events had anything to say, add that to the
593b9c1b51eSKate Stone           // result
594c156427dSZachary Turner           result.AppendMessage(stream.GetString());
59530fdc8d8SChris Lattner 
59630fdc8d8SChris Lattner           result.SetDidChangeProcessState(true);
59730fdc8d8SChris Lattner           result.SetStatus(eReturnStatusSuccessFinishNoResult);
598b9c1b51eSKate Stone         } else {
59930fdc8d8SChris Lattner           result.SetStatus(eReturnStatusSuccessContinuingNoResult);
60030fdc8d8SChris Lattner         }
601b9c1b51eSKate Stone       } else {
602b9c1b51eSKate Stone         result.AppendErrorWithFormat("Failed to resume process: %s.\n",
603b9c1b51eSKate Stone                                      error.AsCString());
60430fdc8d8SChris Lattner       }
605b9c1b51eSKate Stone     } else {
606b9c1b51eSKate Stone       result.AppendErrorWithFormat(
607b9c1b51eSKate Stone           "Process cannot be continued from its current state (%s).\n",
60830fdc8d8SChris Lattner           StateAsCString(state));
60930fdc8d8SChris Lattner     }
61030fdc8d8SChris Lattner     return result.Succeeded();
61130fdc8d8SChris Lattner   }
6120e41084aSJim Ingham 
613b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
6140e41084aSJim Ingham 
6150e41084aSJim Ingham   CommandOptions m_options;
6160e41084aSJim Ingham };
6170e41084aSJim Ingham 
61830fdc8d8SChris Lattner // CommandObjectProcessDetach
619438dfcffSRaphael Isemann #define LLDB_OPTIONS_process_detach
620438dfcffSRaphael Isemann #include "CommandOptions.inc"
6211f0f5b5bSZachary Turner 
622bb9caf73SJim Ingham #pragma mark CommandObjectProcessDetach
62330fdc8d8SChris Lattner 
624b9c1b51eSKate Stone class CommandObjectProcessDetach : public CommandObjectParsed {
62530fdc8d8SChris Lattner public:
626b9c1b51eSKate Stone   class CommandOptions : public Options {
627acff8950SJim Ingham   public:
628b9c1b51eSKate Stone     CommandOptions() : Options() { OptionParsingStarting(nullptr); }
629acff8950SJim Ingham 
63049bcfd80SEugene Zelenko     ~CommandOptions() override = default;
631acff8950SJim Ingham 
63297206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
633b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
63497206d57SZachary Turner       Status error;
635acff8950SJim Ingham       const int short_option = m_getopt_table[option_idx].val;
636acff8950SJim Ingham 
637b9c1b51eSKate Stone       switch (short_option) {
638acff8950SJim Ingham       case 's':
639acff8950SJim Ingham         bool tmp_result;
640acff8950SJim Ingham         bool success;
64147cbf4a0SPavel Labath         tmp_result = OptionArgParser::ToBoolean(option_arg, false, &success);
642acff8950SJim Ingham         if (!success)
643b9c1b51eSKate Stone           error.SetErrorStringWithFormat("invalid boolean option: \"%s\"",
644fe11483bSZachary Turner                                          option_arg.str().c_str());
645b9c1b51eSKate Stone         else {
646acff8950SJim Ingham           if (tmp_result)
647acff8950SJim Ingham             m_keep_stopped = eLazyBoolYes;
648acff8950SJim Ingham           else
649acff8950SJim Ingham             m_keep_stopped = eLazyBoolNo;
650acff8950SJim Ingham         }
651acff8950SJim Ingham         break;
652acff8950SJim Ingham       default:
65336162014SRaphael Isemann         llvm_unreachable("Unimplemented option");
654acff8950SJim Ingham       }
655acff8950SJim Ingham       return error;
656acff8950SJim Ingham     }
657acff8950SJim Ingham 
658b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
659acff8950SJim Ingham       m_keep_stopped = eLazyBoolCalculate;
660acff8950SJim Ingham     }
661acff8950SJim Ingham 
6621f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
66370602439SZachary Turner       return llvm::makeArrayRef(g_process_detach_options);
6641f0f5b5bSZachary Turner     }
665acff8950SJim Ingham 
666acff8950SJim Ingham     // Instance variables to hold the values for command options.
667acff8950SJim Ingham     LazyBool m_keep_stopped;
668acff8950SJim Ingham   };
66930fdc8d8SChris Lattner 
6707428a18cSKate Stone   CommandObjectProcessDetach(CommandInterpreter &interpreter)
671b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process detach",
672b9c1b51eSKate Stone                             "Detach from the current target process.",
673a7015092SGreg Clayton                             "process detach",
674b9c1b51eSKate Stone                             eCommandRequiresProcess | eCommandTryTargetAPILock |
675b9c1b51eSKate Stone                                 eCommandProcessMustBeLaunched),
676b9c1b51eSKate Stone         m_options() {}
67730fdc8d8SChris Lattner 
67849bcfd80SEugene Zelenko   ~CommandObjectProcessDetach() override = default;
67930fdc8d8SChris Lattner 
680b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
681acff8950SJim Ingham 
6825a988416SJim Ingham protected:
683b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
684f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
685acff8950SJim Ingham     // FIXME: This will be a Command Option:
686acff8950SJim Ingham     bool keep_stopped;
687b9c1b51eSKate Stone     if (m_options.m_keep_stopped == eLazyBoolCalculate) {
688acff8950SJim Ingham       // Check the process default:
68949bcfd80SEugene Zelenko       keep_stopped = process->GetDetachKeepsStopped();
690b9c1b51eSKate Stone     } else if (m_options.m_keep_stopped == eLazyBoolYes)
691acff8950SJim Ingham       keep_stopped = true;
692acff8950SJim Ingham     else
693acff8950SJim Ingham       keep_stopped = false;
694acff8950SJim Ingham 
69597206d57SZachary Turner     Status error(process->Detach(keep_stopped));
696b9c1b51eSKate Stone     if (error.Success()) {
69730fdc8d8SChris Lattner       result.SetStatus(eReturnStatusSuccessFinishResult);
698b9c1b51eSKate Stone     } else {
69930fdc8d8SChris Lattner       result.AppendErrorWithFormat("Detach failed: %s\n", error.AsCString());
70030fdc8d8SChris Lattner       return false;
70130fdc8d8SChris Lattner     }
70230fdc8d8SChris Lattner     return result.Succeeded();
70330fdc8d8SChris Lattner   }
704acff8950SJim Ingham 
705acff8950SJim Ingham   CommandOptions m_options;
706acff8950SJim Ingham };
707acff8950SJim Ingham 
708b766a73dSGreg Clayton // CommandObjectProcessConnect
709438dfcffSRaphael Isemann #define LLDB_OPTIONS_process_connect
710438dfcffSRaphael Isemann #include "CommandOptions.inc"
7111f0f5b5bSZachary Turner 
712b766a73dSGreg Clayton #pragma mark CommandObjectProcessConnect
713b766a73dSGreg Clayton 
714b9c1b51eSKate Stone class CommandObjectProcessConnect : public CommandObjectParsed {
715b766a73dSGreg Clayton public:
716b9c1b51eSKate Stone   class CommandOptions : public Options {
717b766a73dSGreg Clayton   public:
718b9c1b51eSKate Stone     CommandOptions() : Options() {
719b9c1b51eSKate Stone       // Keep default values of all options in one place: OptionParsingStarting
720b9c1b51eSKate Stone       // ()
721e1cfbc79STodd Fiala       OptionParsingStarting(nullptr);
722b766a73dSGreg Clayton     }
723b766a73dSGreg Clayton 
72449bcfd80SEugene Zelenko     ~CommandOptions() override = default;
725b766a73dSGreg Clayton 
72697206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
727b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
72897206d57SZachary Turner       Status error;
7293bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
730b766a73dSGreg Clayton 
731b9c1b51eSKate Stone       switch (short_option) {
732b766a73dSGreg Clayton       case 'p':
733adcd0268SBenjamin Kramer         plugin_name.assign(std::string(option_arg));
734b766a73dSGreg Clayton         break;
735b766a73dSGreg Clayton 
736b766a73dSGreg Clayton       default:
73736162014SRaphael Isemann         llvm_unreachable("Unimplemented option");
738b766a73dSGreg Clayton       }
739b766a73dSGreg Clayton       return error;
740b766a73dSGreg Clayton     }
741b766a73dSGreg Clayton 
742b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
743b766a73dSGreg Clayton       plugin_name.clear();
744b766a73dSGreg Clayton     }
745b766a73dSGreg Clayton 
7461f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
74770602439SZachary Turner       return llvm::makeArrayRef(g_process_connect_options);
7481f0f5b5bSZachary Turner     }
749b766a73dSGreg Clayton 
750b766a73dSGreg Clayton     // Instance variables to hold the values for command options.
751b766a73dSGreg Clayton 
752b766a73dSGreg Clayton     std::string plugin_name;
753b766a73dSGreg Clayton   };
754b766a73dSGreg Clayton 
755b9c1b51eSKate Stone   CommandObjectProcessConnect(CommandInterpreter &interpreter)
756b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process connect",
757b766a73dSGreg Clayton                             "Connect to a remote debug service.",
758b9c1b51eSKate Stone                             "process connect <remote-url>", 0),
759b9c1b51eSKate Stone         m_options() {}
760b766a73dSGreg Clayton 
76149bcfd80SEugene Zelenko   ~CommandObjectProcessConnect() override = default;
762b766a73dSGreg Clayton 
763b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
7645a988416SJim Ingham 
7655a988416SJim Ingham protected:
766b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
767b9c1b51eSKate Stone     if (command.GetArgumentCount() != 1) {
768b9c1b51eSKate Stone       result.AppendErrorWithFormat(
769b9c1b51eSKate Stone           "'%s' takes exactly one argument:\nUsage: %s\n", m_cmd_name.c_str(),
770ccd6cffbSTamas Berghammer           m_cmd_syntax.c_str());
771ccd6cffbSTamas Berghammer       return false;
772ccd6cffbSTamas Berghammer     }
773ccd6cffbSTamas Berghammer 
774ccd6cffbSTamas Berghammer     Process *process = m_exe_ctx.GetProcessPtr();
775b9c1b51eSKate Stone     if (process && process->IsAlive()) {
776b9c1b51eSKate Stone       result.AppendErrorWithFormat(
777b9c1b51eSKate Stone           "Process %" PRIu64
778b9c1b51eSKate Stone           " is currently being debugged, kill the process before connecting.\n",
779b766a73dSGreg Clayton           process->GetID());
780b766a73dSGreg Clayton       return false;
781b766a73dSGreg Clayton     }
782b766a73dSGreg Clayton 
783ccd6cffbSTamas Berghammer     const char *plugin_name = nullptr;
784b766a73dSGreg Clayton     if (!m_options.plugin_name.empty())
785b766a73dSGreg Clayton       plugin_name = m_options.plugin_name.c_str();
786b766a73dSGreg Clayton 
78797206d57SZachary Turner     Status error;
78857179860SJonas Devlieghere     Debugger &debugger = GetDebugger();
789ccd6cffbSTamas Berghammer     PlatformSP platform_sp = m_interpreter.GetPlatform(true);
790706cccb8SJonas Devlieghere     ProcessSP process_sp =
791706cccb8SJonas Devlieghere         debugger.GetAsyncExecution()
792706cccb8SJonas Devlieghere             ? platform_sp->ConnectProcess(
793b9c1b51eSKate Stone                   command.GetArgumentAtIndex(0), plugin_name, debugger,
794706cccb8SJonas Devlieghere                   debugger.GetSelectedTarget().get(), error)
795706cccb8SJonas Devlieghere             : platform_sp->ConnectProcessSynchronous(
796706cccb8SJonas Devlieghere                   command.GetArgumentAtIndex(0), plugin_name, debugger,
797706cccb8SJonas Devlieghere                   result.GetOutputStream(), debugger.GetSelectedTarget().get(),
798706cccb8SJonas Devlieghere                   error);
799b9c1b51eSKate Stone     if (error.Fail() || process_sp == nullptr) {
800ccd6cffbSTamas Berghammer       result.AppendError(error.AsCString("Error connecting to the process"));
801b766a73dSGreg Clayton       return false;
802b766a73dSGreg Clayton     }
803ccd6cffbSTamas Berghammer     return true;
804b766a73dSGreg Clayton   }
805b766a73dSGreg Clayton 
806b766a73dSGreg Clayton   CommandOptions m_options;
807b766a73dSGreg Clayton };
808b766a73dSGreg Clayton 
809998255bfSGreg Clayton // CommandObjectProcessPlugin
810998255bfSGreg Clayton #pragma mark CommandObjectProcessPlugin
811998255bfSGreg Clayton 
812b9c1b51eSKate Stone class CommandObjectProcessPlugin : public CommandObjectProxy {
813998255bfSGreg Clayton public:
8147428a18cSKate Stone   CommandObjectProcessPlugin(CommandInterpreter &interpreter)
815b9c1b51eSKate Stone       : CommandObjectProxy(
816b9c1b51eSKate Stone             interpreter, "process plugin",
817b9c1b51eSKate Stone             "Send a custom command to the current target process plug-in.",
818b9c1b51eSKate Stone             "process plugin <args>", 0) {}
819998255bfSGreg Clayton 
82049bcfd80SEugene Zelenko   ~CommandObjectProcessPlugin() override = default;
821998255bfSGreg Clayton 
822b9c1b51eSKate Stone   CommandObject *GetProxyCommandObject() override {
823e05b2efeSGreg Clayton     Process *process = m_interpreter.GetExecutionContext().GetProcessPtr();
824998255bfSGreg Clayton     if (process)
825998255bfSGreg Clayton       return process->GetPluginCommandObject();
82649bcfd80SEugene Zelenko     return nullptr;
827998255bfSGreg Clayton   }
828998255bfSGreg Clayton };
829998255bfSGreg Clayton 
8308f343b09SGreg Clayton // CommandObjectProcessLoad
831438dfcffSRaphael Isemann #define LLDB_OPTIONS_process_load
832438dfcffSRaphael Isemann #include "CommandOptions.inc"
8331f0f5b5bSZachary Turner 
834bb9caf73SJim Ingham #pragma mark CommandObjectProcessLoad
8358f343b09SGreg Clayton 
836b9c1b51eSKate Stone class CommandObjectProcessLoad : public CommandObjectParsed {
8378f343b09SGreg Clayton public:
838b9c1b51eSKate Stone   class CommandOptions : public Options {
8394fbd67acSTamas Berghammer   public:
840b9c1b51eSKate Stone     CommandOptions() : Options() {
841b9c1b51eSKate Stone       // Keep default values of all options in one place: OptionParsingStarting
842b9c1b51eSKate Stone       // ()
843e1cfbc79STodd Fiala       OptionParsingStarting(nullptr);
8444fbd67acSTamas Berghammer     }
8454fbd67acSTamas Berghammer 
8464fbd67acSTamas Berghammer     ~CommandOptions() override = default;
8474fbd67acSTamas Berghammer 
84897206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
849b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
85097206d57SZachary Turner       Status error;
8514fbd67acSTamas Berghammer       const int short_option = m_getopt_table[option_idx].val;
852b9c1b51eSKate Stone       switch (short_option) {
8534fbd67acSTamas Berghammer       case 'i':
8544fbd67acSTamas Berghammer         do_install = true;
855fe11483bSZachary Turner         if (!option_arg.empty())
8568f3be7a3SJonas Devlieghere           install_path.SetFile(option_arg, FileSpec::Style::native);
8574fbd67acSTamas Berghammer         break;
8584fbd67acSTamas Berghammer       default:
85936162014SRaphael Isemann         llvm_unreachable("Unimplemented option");
8604fbd67acSTamas Berghammer       }
8614fbd67acSTamas Berghammer       return error;
8624fbd67acSTamas Berghammer     }
8634fbd67acSTamas Berghammer 
864b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
8654fbd67acSTamas Berghammer       do_install = false;
8664fbd67acSTamas Berghammer       install_path.Clear();
8674fbd67acSTamas Berghammer     }
8684fbd67acSTamas Berghammer 
8691f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
87070602439SZachary Turner       return llvm::makeArrayRef(g_process_load_options);
8711f0f5b5bSZachary Turner     }
8724fbd67acSTamas Berghammer 
8734fbd67acSTamas Berghammer     // Instance variables to hold the values for command options.
8744fbd67acSTamas Berghammer     bool do_install;
8754fbd67acSTamas Berghammer     FileSpec install_path;
8764fbd67acSTamas Berghammer   };
8778f343b09SGreg Clayton 
878b9c1b51eSKate Stone   CommandObjectProcessLoad(CommandInterpreter &interpreter)
879b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process load",
8808f343b09SGreg Clayton                             "Load a shared library into the current process.",
8818f343b09SGreg Clayton                             "process load <filename> [<filename> ...]",
882b9c1b51eSKate Stone                             eCommandRequiresProcess | eCommandTryTargetAPILock |
883e87764f2SEnrico Granata                                 eCommandProcessMustBeLaunched |
8844fbd67acSTamas Berghammer                                 eCommandProcessMustBePaused),
885b9c1b51eSKate Stone         m_options() {}
8868f343b09SGreg Clayton 
8874fbd67acSTamas Berghammer   ~CommandObjectProcessLoad() override = default;
8884fbd67acSTamas Berghammer 
889e1cd7cacSGongyu Deng   void
890e1cd7cacSGongyu Deng   HandleArgumentCompletion(CompletionRequest &request,
891e1cd7cacSGongyu Deng                            OptionElementVector &opt_element_vector) override {
892e1cd7cacSGongyu Deng     if (!m_exe_ctx.HasProcessScope())
893e1cd7cacSGongyu Deng       return;
894e1cd7cacSGongyu Deng 
895e1cd7cacSGongyu Deng     CommandCompletions::InvokeCommonCompletionCallbacks(
896e1cd7cacSGongyu Deng         GetCommandInterpreter(), CommandCompletions::eDiskFileCompletion,
897e1cd7cacSGongyu Deng         request, nullptr);
898e1cd7cacSGongyu Deng   }
899e1cd7cacSGongyu Deng 
900b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
9018f343b09SGreg Clayton 
9025a988416SJim Ingham protected:
903b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
904f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
9058f343b09SGreg Clayton 
90697d2c401SZachary Turner     for (auto &entry : command.entries()) {
90797206d57SZachary Turner       Status error;
9083cb132a0STamas Berghammer       PlatformSP platform = process->GetTarget().GetPlatform();
9090d9a201eSRaphael Isemann       llvm::StringRef image_path = entry.ref();
9104fbd67acSTamas Berghammer       uint32_t image_token = LLDB_INVALID_IMAGE_TOKEN;
9114fbd67acSTamas Berghammer 
912b9c1b51eSKate Stone       if (!m_options.do_install) {
9138f3be7a3SJonas Devlieghere         FileSpec image_spec(image_path);
9143cb132a0STamas Berghammer         platform->ResolveRemotePath(image_spec, image_spec);
915b9c1b51eSKate Stone         image_token =
916b9c1b51eSKate Stone             platform->LoadImage(process, FileSpec(), image_spec, error);
917b9c1b51eSKate Stone       } else if (m_options.install_path) {
9188f3be7a3SJonas Devlieghere         FileSpec image_spec(image_path);
9198f3be7a3SJonas Devlieghere         FileSystem::Instance().Resolve(image_spec);
920b9c1b51eSKate Stone         platform->ResolveRemotePath(m_options.install_path,
921b9c1b51eSKate Stone                                     m_options.install_path);
922b9c1b51eSKate Stone         image_token = platform->LoadImage(process, image_spec,
923b9c1b51eSKate Stone                                           m_options.install_path, error);
924b9c1b51eSKate Stone       } else {
9258f3be7a3SJonas Devlieghere         FileSpec image_spec(image_path);
9268f3be7a3SJonas Devlieghere         FileSystem::Instance().Resolve(image_spec);
927b9c1b51eSKate Stone         image_token =
928b9c1b51eSKate Stone             platform->LoadImage(process, image_spec, FileSpec(), error);
9294fbd67acSTamas Berghammer       }
9304fbd67acSTamas Berghammer 
931b9c1b51eSKate Stone       if (image_token != LLDB_INVALID_IMAGE_TOKEN) {
932b9c1b51eSKate Stone         result.AppendMessageWithFormat(
93397d2c401SZachary Turner             "Loading \"%s\"...ok\nImage %u loaded.\n", image_path.str().c_str(),
93497d2c401SZachary Turner             image_token);
9358f343b09SGreg Clayton         result.SetStatus(eReturnStatusSuccessFinishResult);
936b9c1b51eSKate Stone       } else {
93797d2c401SZachary Turner         result.AppendErrorWithFormat("failed to load '%s': %s",
93897d2c401SZachary Turner                                      image_path.str().c_str(),
939b9c1b51eSKate Stone                                      error.AsCString());
9408f343b09SGreg Clayton       }
9418f343b09SGreg Clayton     }
9428f343b09SGreg Clayton     return result.Succeeded();
9438f343b09SGreg Clayton   }
9444fbd67acSTamas Berghammer 
9454fbd67acSTamas Berghammer   CommandOptions m_options;
9468f343b09SGreg Clayton };
9478f343b09SGreg Clayton 
9488f343b09SGreg Clayton // CommandObjectProcessUnload
949bb9caf73SJim Ingham #pragma mark CommandObjectProcessUnload
9508f343b09SGreg Clayton 
951b9c1b51eSKate Stone class CommandObjectProcessUnload : public CommandObjectParsed {
9528f343b09SGreg Clayton public:
953b9c1b51eSKate Stone   CommandObjectProcessUnload(CommandInterpreter &interpreter)
954b9c1b51eSKate Stone       : CommandObjectParsed(
955b9c1b51eSKate Stone             interpreter, "process unload",
956b9c1b51eSKate Stone             "Unload a shared library from the current process using the index "
957b9c1b51eSKate Stone             "returned by a previous call to \"process load\".",
9588f343b09SGreg Clayton             "process unload <index>",
959b9c1b51eSKate Stone             eCommandRequiresProcess | eCommandTryTargetAPILock |
960b9c1b51eSKate Stone                 eCommandProcessMustBeLaunched | eCommandProcessMustBePaused) {}
9618f343b09SGreg Clayton 
96249bcfd80SEugene Zelenko   ~CommandObjectProcessUnload() override = default;
9638f343b09SGreg Clayton 
964e1cd7cacSGongyu Deng   void
965e1cd7cacSGongyu Deng   HandleArgumentCompletion(CompletionRequest &request,
966e1cd7cacSGongyu Deng                            OptionElementVector &opt_element_vector) override {
967e1cd7cacSGongyu Deng 
968e1cd7cacSGongyu Deng     if (request.GetCursorIndex() || !m_exe_ctx.HasProcessScope())
969e1cd7cacSGongyu Deng       return;
970e1cd7cacSGongyu Deng 
971e1cd7cacSGongyu Deng     Process *process = m_exe_ctx.GetProcessPtr();
972e1cd7cacSGongyu Deng 
973e1cd7cacSGongyu Deng     const std::vector<lldb::addr_t> &tokens = process->GetImageTokens();
974e1cd7cacSGongyu Deng     const size_t token_num = tokens.size();
975e1cd7cacSGongyu Deng     for (size_t i = 0; i < token_num; ++i) {
976e1cd7cacSGongyu Deng       if (tokens[i] == LLDB_INVALID_IMAGE_TOKEN)
977e1cd7cacSGongyu Deng         continue;
978e1cd7cacSGongyu Deng       request.TryCompleteCurrentArg(std::to_string(i));
979e1cd7cacSGongyu Deng     }
980e1cd7cacSGongyu Deng   }
981e1cd7cacSGongyu Deng 
9825a988416SJim Ingham protected:
983b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
984f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
9858f343b09SGreg Clayton 
98697d2c401SZachary Turner     for (auto &entry : command.entries()) {
98797d2c401SZachary Turner       uint32_t image_token;
9880d9a201eSRaphael Isemann       if (entry.ref().getAsInteger(0, image_token)) {
989b9c1b51eSKate Stone         result.AppendErrorWithFormat("invalid image index argument '%s'",
9900d9a201eSRaphael Isemann                                      entry.ref().str().c_str());
9918f343b09SGreg Clayton         break;
992b9c1b51eSKate Stone       } else {
99397206d57SZachary Turner         Status error(process->GetTarget().GetPlatform()->UnloadImage(
994b9c1b51eSKate Stone             process, image_token));
995b9c1b51eSKate Stone         if (error.Success()) {
996b9c1b51eSKate Stone           result.AppendMessageWithFormat(
997b9c1b51eSKate Stone               "Unloading shared library with index %u...ok\n", image_token);
9988f343b09SGreg Clayton           result.SetStatus(eReturnStatusSuccessFinishResult);
999b9c1b51eSKate Stone         } else {
1000b9c1b51eSKate Stone           result.AppendErrorWithFormat("failed to unload image: %s",
1001b9c1b51eSKate Stone                                        error.AsCString());
10028f343b09SGreg Clayton           break;
10038f343b09SGreg Clayton         }
10048f343b09SGreg Clayton       }
10058f343b09SGreg Clayton     }
10068f343b09SGreg Clayton     return result.Succeeded();
10078f343b09SGreg Clayton   }
10088f343b09SGreg Clayton };
10098f343b09SGreg Clayton 
101030fdc8d8SChris Lattner // CommandObjectProcessSignal
1011bb9caf73SJim Ingham #pragma mark CommandObjectProcessSignal
101230fdc8d8SChris Lattner 
1013b9c1b51eSKate Stone class CommandObjectProcessSignal : public CommandObjectParsed {
101430fdc8d8SChris Lattner public:
10157428a18cSKate Stone   CommandObjectProcessSignal(CommandInterpreter &interpreter)
1016a925974bSAdrian Prantl       : CommandObjectParsed(
1017a925974bSAdrian Prantl             interpreter, "process signal",
1018a925974bSAdrian Prantl             "Send a UNIX signal to the current target process.", nullptr,
1019a925974bSAdrian Prantl             eCommandRequiresProcess | eCommandTryTargetAPILock) {
1020405fe67fSCaroline Tice     CommandArgumentEntry arg;
1021405fe67fSCaroline Tice     CommandArgumentData signal_arg;
1022405fe67fSCaroline Tice 
1023405fe67fSCaroline Tice     // Define the first (and only) variant of this arg.
1024c0dbdfb6SCaroline Tice     signal_arg.arg_type = eArgTypeUnixSignal;
1025405fe67fSCaroline Tice     signal_arg.arg_repetition = eArgRepeatPlain;
1026405fe67fSCaroline Tice 
1027b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
1028b9c1b51eSKate Stone     // argument entry.
1029405fe67fSCaroline Tice     arg.push_back(signal_arg);
1030405fe67fSCaroline Tice 
1031405fe67fSCaroline Tice     // Push the data for the first argument into the m_arguments vector.
1032405fe67fSCaroline Tice     m_arguments.push_back(arg);
103330fdc8d8SChris Lattner   }
103430fdc8d8SChris Lattner 
103549bcfd80SEugene Zelenko   ~CommandObjectProcessSignal() override = default;
103630fdc8d8SChris Lattner 
10372bba1c22SRaphael Isemann   void
10382bba1c22SRaphael Isemann   HandleArgumentCompletion(CompletionRequest &request,
10392bba1c22SRaphael Isemann                            OptionElementVector &opt_element_vector) override {
10402bba1c22SRaphael Isemann     if (!m_exe_ctx.HasProcessScope() || request.GetCursorIndex() != 0)
10412bba1c22SRaphael Isemann       return;
10422bba1c22SRaphael Isemann 
10432bba1c22SRaphael Isemann     UnixSignalsSP signals = m_exe_ctx.GetProcessPtr()->GetUnixSignals();
10442bba1c22SRaphael Isemann     int signo = signals->GetFirstSignalNumber();
10452bba1c22SRaphael Isemann     while (signo != LLDB_INVALID_SIGNAL_NUMBER) {
104607800348SRaphael Isemann       request.TryCompleteCurrentArg(signals->GetSignalAsCString(signo));
10472bba1c22SRaphael Isemann       signo = signals->GetNextSignalNumber(signo);
10482bba1c22SRaphael Isemann     }
10492bba1c22SRaphael Isemann   }
10502bba1c22SRaphael Isemann 
10515a988416SJim Ingham protected:
1052b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1053f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
105430fdc8d8SChris Lattner 
1055b9c1b51eSKate Stone     if (command.GetArgumentCount() == 1) {
1056237cd906SGreg Clayton       int signo = LLDB_INVALID_SIGNAL_NUMBER;
1057237cd906SGreg Clayton 
1058237cd906SGreg Clayton       const char *signal_name = command.GetArgumentAtIndex(0);
10599010cef2SRaphael Isemann       if (::isxdigit(signal_name[0])) {
10609010cef2SRaphael Isemann         if (!llvm::to_integer(signal_name, signo))
10619010cef2SRaphael Isemann           signo = LLDB_INVALID_SIGNAL_NUMBER;
10629010cef2SRaphael Isemann       } else
106398d0a4b3SChaoren Lin         signo = process->GetUnixSignals()->GetSignalNumberFromName(signal_name);
1064237cd906SGreg Clayton 
1065b9c1b51eSKate Stone       if (signo == LLDB_INVALID_SIGNAL_NUMBER) {
1066b9c1b51eSKate Stone         result.AppendErrorWithFormat("Invalid signal argument '%s'.\n",
1067b9c1b51eSKate Stone                                      command.GetArgumentAtIndex(0));
1068b9c1b51eSKate Stone       } else {
106997206d57SZachary Turner         Status error(process->Signal(signo));
1070b9c1b51eSKate Stone         if (error.Success()) {
107130fdc8d8SChris Lattner           result.SetStatus(eReturnStatusSuccessFinishResult);
1072b9c1b51eSKate Stone         } else {
1073b9c1b51eSKate Stone           result.AppendErrorWithFormat("Failed to send signal %i: %s\n", signo,
1074b9c1b51eSKate Stone                                        error.AsCString());
107530fdc8d8SChris Lattner         }
107630fdc8d8SChris Lattner       }
1077b9c1b51eSKate Stone     } else {
1078b9c1b51eSKate Stone       result.AppendErrorWithFormat(
1079b9c1b51eSKate Stone           "'%s' takes exactly one signal number argument:\nUsage: %s\n",
1080b9c1b51eSKate Stone           m_cmd_name.c_str(), m_cmd_syntax.c_str());
108130fdc8d8SChris Lattner     }
108230fdc8d8SChris Lattner     return result.Succeeded();
108330fdc8d8SChris Lattner   }
108430fdc8d8SChris Lattner };
108530fdc8d8SChris Lattner 
108630fdc8d8SChris Lattner // CommandObjectProcessInterrupt
1087bb9caf73SJim Ingham #pragma mark CommandObjectProcessInterrupt
108830fdc8d8SChris Lattner 
1089b9c1b51eSKate Stone class CommandObjectProcessInterrupt : public CommandObjectParsed {
109030fdc8d8SChris Lattner public:
10917428a18cSKate Stone   CommandObjectProcessInterrupt(CommandInterpreter &interpreter)
1092b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process interrupt",
1093b9c1b51eSKate Stone                             "Interrupt the current target process.",
1094a7015092SGreg Clayton                             "process interrupt",
1095b9c1b51eSKate Stone                             eCommandRequiresProcess | eCommandTryTargetAPILock |
1096b9c1b51eSKate Stone                                 eCommandProcessMustBeLaunched) {}
109730fdc8d8SChris Lattner 
109849bcfd80SEugene Zelenko   ~CommandObjectProcessInterrupt() override = default;
109930fdc8d8SChris Lattner 
11005a988416SJim Ingham protected:
1101b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1102f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
1103b9c1b51eSKate Stone     if (process == nullptr) {
110430fdc8d8SChris Lattner       result.AppendError("no process to halt");
110530fdc8d8SChris Lattner       return false;
110630fdc8d8SChris Lattner     }
110730fdc8d8SChris Lattner 
1108b9c1b51eSKate Stone     if (command.GetArgumentCount() == 0) {
1109f9b57b9dSGreg Clayton       bool clear_thread_plans = true;
111097206d57SZachary Turner       Status error(process->Halt(clear_thread_plans));
1111b9c1b51eSKate Stone       if (error.Success()) {
111230fdc8d8SChris Lattner         result.SetStatus(eReturnStatusSuccessFinishResult);
1113b9c1b51eSKate Stone       } else {
1114b9c1b51eSKate Stone         result.AppendErrorWithFormat("Failed to halt process: %s\n",
1115b9c1b51eSKate Stone                                      error.AsCString());
111630fdc8d8SChris Lattner       }
1117b9c1b51eSKate Stone     } else {
1118fd54b368SJason Molenda       result.AppendErrorWithFormat("'%s' takes no arguments:\nUsage: %s\n",
1119b9c1b51eSKate Stone                                    m_cmd_name.c_str(), m_cmd_syntax.c_str());
112030fdc8d8SChris Lattner     }
112130fdc8d8SChris Lattner     return result.Succeeded();
112230fdc8d8SChris Lattner   }
112330fdc8d8SChris Lattner };
112430fdc8d8SChris Lattner 
112530fdc8d8SChris Lattner // CommandObjectProcessKill
1126bb9caf73SJim Ingham #pragma mark CommandObjectProcessKill
112730fdc8d8SChris Lattner 
1128b9c1b51eSKate Stone class CommandObjectProcessKill : public CommandObjectParsed {
112930fdc8d8SChris Lattner public:
11307428a18cSKate Stone   CommandObjectProcessKill(CommandInterpreter &interpreter)
1131b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process kill",
1132b9c1b51eSKate Stone                             "Terminate the current target process.",
1133b9c1b51eSKate Stone                             "process kill",
1134b9c1b51eSKate Stone                             eCommandRequiresProcess | eCommandTryTargetAPILock |
1135b9c1b51eSKate Stone                                 eCommandProcessMustBeLaunched) {}
113630fdc8d8SChris Lattner 
113749bcfd80SEugene Zelenko   ~CommandObjectProcessKill() override = default;
113830fdc8d8SChris Lattner 
11395a988416SJim Ingham protected:
1140b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1141f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
1142b9c1b51eSKate Stone     if (process == nullptr) {
114330fdc8d8SChris Lattner       result.AppendError("no process to kill");
114430fdc8d8SChris Lattner       return false;
114530fdc8d8SChris Lattner     }
114630fdc8d8SChris Lattner 
1147b9c1b51eSKate Stone     if (command.GetArgumentCount() == 0) {
114897206d57SZachary Turner       Status error(process->Destroy(true));
1149b9c1b51eSKate Stone       if (error.Success()) {
115030fdc8d8SChris Lattner         result.SetStatus(eReturnStatusSuccessFinishResult);
1151b9c1b51eSKate Stone       } else {
1152b9c1b51eSKate Stone         result.AppendErrorWithFormat("Failed to kill process: %s\n",
1153b9c1b51eSKate Stone                                      error.AsCString());
115430fdc8d8SChris Lattner       }
1155b9c1b51eSKate Stone     } else {
1156fd54b368SJason Molenda       result.AppendErrorWithFormat("'%s' takes no arguments:\nUsage: %s\n",
1157b9c1b51eSKate Stone                                    m_cmd_name.c_str(), m_cmd_syntax.c_str());
115830fdc8d8SChris Lattner     }
115930fdc8d8SChris Lattner     return result.Succeeded();
116030fdc8d8SChris Lattner   }
116130fdc8d8SChris Lattner };
116230fdc8d8SChris Lattner 
1163a2715cf1SGreg Clayton // CommandObjectProcessSaveCore
1164a2715cf1SGreg Clayton #pragma mark CommandObjectProcessSaveCore
1165a2715cf1SGreg Clayton 
11669ea6dd5cSJason Molenda static constexpr OptionEnumValueElement g_corefile_save_style[] = {
11679ea6dd5cSJason Molenda     {eSaveCoreFull, "full", "Create a core file with all memory saved"},
11689ea6dd5cSJason Molenda     {eSaveCoreDirtyOnly, "modified-memory",
1169*8c31efeeSJason Molenda      "Create a corefile with only modified memory saved"},
1170*8c31efeeSJason Molenda     {eSaveCoreStackOnly, "stack",
1171*8c31efeeSJason Molenda      "Create a corefile with only stack  memory saved"}};
11729ea6dd5cSJason Molenda 
11739ea6dd5cSJason Molenda static constexpr OptionEnumValues SaveCoreStyles() {
11749ea6dd5cSJason Molenda   return OptionEnumValues(g_corefile_save_style);
11759ea6dd5cSJason Molenda }
11769ea6dd5cSJason Molenda 
11779ea6dd5cSJason Molenda #define LLDB_OPTIONS_process_save_core
11789ea6dd5cSJason Molenda #include "CommandOptions.inc"
11799ea6dd5cSJason Molenda 
1180b9c1b51eSKate Stone class CommandObjectProcessSaveCore : public CommandObjectParsed {
1181a2715cf1SGreg Clayton public:
1182b9c1b51eSKate Stone   CommandObjectProcessSaveCore(CommandInterpreter &interpreter)
1183b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process save-core",
1184b9c1b51eSKate Stone                             "Save the current process as a core file using an "
1185b9c1b51eSKate Stone                             "appropriate file type.",
11869ea6dd5cSJason Molenda                             "process save-core [-s corefile-style] FILE",
1187b9c1b51eSKate Stone                             eCommandRequiresProcess | eCommandTryTargetAPILock |
1188b9c1b51eSKate Stone                                 eCommandProcessMustBeLaunched) {}
1189a2715cf1SGreg Clayton 
119049bcfd80SEugene Zelenko   ~CommandObjectProcessSaveCore() override = default;
1191a2715cf1SGreg Clayton 
11929ea6dd5cSJason Molenda   Options *GetOptions() override { return &m_options; }
11939ea6dd5cSJason Molenda 
11949ea6dd5cSJason Molenda   class CommandOptions : public Options {
11959ea6dd5cSJason Molenda   public:
11969ea6dd5cSJason Molenda     CommandOptions()
11979ea6dd5cSJason Molenda         : Options(), m_requested_save_core_style(eSaveCoreUnspecified) {}
11989ea6dd5cSJason Molenda 
11999ea6dd5cSJason Molenda     ~CommandOptions() override = default;
12009ea6dd5cSJason Molenda 
12019ea6dd5cSJason Molenda     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
12029ea6dd5cSJason Molenda       return llvm::makeArrayRef(g_process_save_core_options);
12039ea6dd5cSJason Molenda     }
12049ea6dd5cSJason Molenda 
12059ea6dd5cSJason Molenda     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
12069ea6dd5cSJason Molenda                           ExecutionContext *execution_context) override {
12079ea6dd5cSJason Molenda       const int short_option = m_getopt_table[option_idx].val;
12089ea6dd5cSJason Molenda       Status error;
12099ea6dd5cSJason Molenda 
12109ea6dd5cSJason Molenda       switch (short_option) {
12119ea6dd5cSJason Molenda       case 's':
12129ea6dd5cSJason Molenda         m_requested_save_core_style =
12139ea6dd5cSJason Molenda             (lldb::SaveCoreStyle)OptionArgParser::ToOptionEnum(
12149ea6dd5cSJason Molenda                 option_arg, GetDefinitions()[option_idx].enum_values,
12159ea6dd5cSJason Molenda                 eSaveCoreUnspecified, error);
12169ea6dd5cSJason Molenda         break;
12179ea6dd5cSJason Molenda       default:
12189ea6dd5cSJason Molenda         llvm_unreachable("Unimplemented option");
12199ea6dd5cSJason Molenda       }
12209ea6dd5cSJason Molenda 
12219ea6dd5cSJason Molenda       return {};
12229ea6dd5cSJason Molenda     }
12239ea6dd5cSJason Molenda 
12249ea6dd5cSJason Molenda     void OptionParsingStarting(ExecutionContext *execution_context) override {
12259ea6dd5cSJason Molenda       m_requested_save_core_style = eSaveCoreUnspecified;
12269ea6dd5cSJason Molenda     }
12279ea6dd5cSJason Molenda 
12289ea6dd5cSJason Molenda     // Instance variables to hold the values for command options.
12299ea6dd5cSJason Molenda     SaveCoreStyle m_requested_save_core_style;
12309ea6dd5cSJason Molenda   };
12319ea6dd5cSJason Molenda 
1232a2715cf1SGreg Clayton protected:
1233b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1234a2715cf1SGreg Clayton     ProcessSP process_sp = m_exe_ctx.GetProcessSP();
1235b9c1b51eSKate Stone     if (process_sp) {
1236b9c1b51eSKate Stone       if (command.GetArgumentCount() == 1) {
12378f3be7a3SJonas Devlieghere         FileSpec output_file(command.GetArgumentAtIndex(0));
12389ea6dd5cSJason Molenda         SaveCoreStyle corefile_style = m_options.m_requested_save_core_style;
12399ea6dd5cSJason Molenda         Status error =
12409ea6dd5cSJason Molenda             PluginManager::SaveCore(process_sp, output_file, corefile_style);
1241b9c1b51eSKate Stone         if (error.Success()) {
1242*8c31efeeSJason Molenda           if (corefile_style == SaveCoreStyle::eSaveCoreDirtyOnly ||
1243*8c31efeeSJason Molenda               corefile_style == SaveCoreStyle::eSaveCoreStackOnly) {
12449ea6dd5cSJason Molenda             result.AppendMessageWithFormat(
1245*8c31efeeSJason Molenda                 "\nModified-memory or stack-memory only corefile "
1246*8c31efeeSJason Molenda                 "created.  This corefile may \n"
1247*8c31efeeSJason Molenda                 "not show library/framework/app binaries "
12489ea6dd5cSJason Molenda                 "on a different system, or when \n"
12499ea6dd5cSJason Molenda                 "those binaries have "
12509ea6dd5cSJason Molenda                 "been updated/modified. Copies are not included\n"
12519ea6dd5cSJason Molenda                 "in this corefile.  Use --style full to include all "
12529ea6dd5cSJason Molenda                 "process memory.\n");
12539ea6dd5cSJason Molenda           }
1254a2715cf1SGreg Clayton           result.SetStatus(eReturnStatusSuccessFinishResult);
1255b9c1b51eSKate Stone         } else {
1256b9c1b51eSKate Stone           result.AppendErrorWithFormat(
1257b9c1b51eSKate Stone               "Failed to save core file for process: %s\n", error.AsCString());
1258a2715cf1SGreg Clayton         }
1259b9c1b51eSKate Stone       } else {
1260a2715cf1SGreg Clayton         result.AppendErrorWithFormat("'%s' takes one arguments:\nUsage: %s\n",
1261b9c1b51eSKate Stone                                      m_cmd_name.c_str(), m_cmd_syntax.c_str());
1262a2715cf1SGreg Clayton       }
1263b9c1b51eSKate Stone     } else {
1264a2715cf1SGreg Clayton       result.AppendError("invalid process");
1265a2715cf1SGreg Clayton       return false;
1266a2715cf1SGreg Clayton     }
1267a2715cf1SGreg Clayton 
1268a2715cf1SGreg Clayton     return result.Succeeded();
1269a2715cf1SGreg Clayton   }
12709ea6dd5cSJason Molenda 
12719ea6dd5cSJason Molenda   CommandOptions m_options;
1272a2715cf1SGreg Clayton };
1273a2715cf1SGreg Clayton 
12744b9bea87SJim Ingham // CommandObjectProcessStatus
1275bb9caf73SJim Ingham #pragma mark CommandObjectProcessStatus
1276d7c403e6SMed Ismail Bennani #define LLDB_OPTIONS_process_status
1277d7c403e6SMed Ismail Bennani #include "CommandOptions.inc"
1278bb9caf73SJim Ingham 
1279b9c1b51eSKate Stone class CommandObjectProcessStatus : public CommandObjectParsed {
12804b9bea87SJim Ingham public:
12817428a18cSKate Stone   CommandObjectProcessStatus(CommandInterpreter &interpreter)
1282b9c1b51eSKate Stone       : CommandObjectParsed(
1283b9c1b51eSKate Stone             interpreter, "process status",
1284b9c1b51eSKate Stone             "Show status and stop location for the current target process.",
1285b9c1b51eSKate Stone             "process status",
1286d7c403e6SMed Ismail Bennani             eCommandRequiresProcess | eCommandTryTargetAPILock),
1287d7c403e6SMed Ismail Bennani         m_options() {}
12884b9bea87SJim Ingham 
128949bcfd80SEugene Zelenko   ~CommandObjectProcessStatus() override = default;
12904b9bea87SJim Ingham 
1291d7c403e6SMed Ismail Bennani   Options *GetOptions() override { return &m_options; }
1292d7c403e6SMed Ismail Bennani 
1293d7c403e6SMed Ismail Bennani   class CommandOptions : public Options {
1294d7c403e6SMed Ismail Bennani   public:
12959494c510SJonas Devlieghere     CommandOptions() : Options() {}
1296d7c403e6SMed Ismail Bennani 
1297d7c403e6SMed Ismail Bennani     ~CommandOptions() override = default;
1298d7c403e6SMed Ismail Bennani 
1299d7c403e6SMed Ismail Bennani     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
1300d7c403e6SMed Ismail Bennani                           ExecutionContext *execution_context) override {
1301d7c403e6SMed Ismail Bennani       const int short_option = m_getopt_table[option_idx].val;
1302d7c403e6SMed Ismail Bennani 
1303d7c403e6SMed Ismail Bennani       switch (short_option) {
1304d7c403e6SMed Ismail Bennani       case 'v':
1305d7c403e6SMed Ismail Bennani         m_verbose = true;
1306d7c403e6SMed Ismail Bennani         break;
1307d7c403e6SMed Ismail Bennani       default:
1308d7c403e6SMed Ismail Bennani         llvm_unreachable("Unimplemented option");
1309d7c403e6SMed Ismail Bennani       }
1310d7c403e6SMed Ismail Bennani 
1311d7c403e6SMed Ismail Bennani       return {};
1312d7c403e6SMed Ismail Bennani     }
1313d7c403e6SMed Ismail Bennani 
1314d7c403e6SMed Ismail Bennani     void OptionParsingStarting(ExecutionContext *execution_context) override {
1315d7c403e6SMed Ismail Bennani       m_verbose = false;
1316d7c403e6SMed Ismail Bennani     }
1317d7c403e6SMed Ismail Bennani 
1318d7c403e6SMed Ismail Bennani     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
1319d7c403e6SMed Ismail Bennani       return llvm::makeArrayRef(g_process_status_options);
1320d7c403e6SMed Ismail Bennani     }
1321d7c403e6SMed Ismail Bennani 
1322d7c403e6SMed Ismail Bennani     // Instance variables to hold the values for command options.
13239494c510SJonas Devlieghere     bool m_verbose = false;
1324d7c403e6SMed Ismail Bennani   };
1325d7c403e6SMed Ismail Bennani 
1326d7c403e6SMed Ismail Bennani protected:
1327b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
13287260f620SGreg Clayton     Stream &strm = result.GetOutputStream();
13294b9bea87SJim Ingham     result.SetStatus(eReturnStatusSuccessFinishNoResult);
1330d7c403e6SMed Ismail Bennani 
1331d7c403e6SMed Ismail Bennani     if (command.GetArgumentCount()) {
1332d7c403e6SMed Ismail Bennani       result.AppendError("'process status' takes no arguments");
1333d7c403e6SMed Ismail Bennani       return result.Succeeded();
1334d7c403e6SMed Ismail Bennani     }
1335d7c403e6SMed Ismail Bennani 
1336b9c1b51eSKate Stone     // No need to check "process" for validity as eCommandRequiresProcess
1337b9c1b51eSKate Stone     // ensures it is valid
1338f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
13397260f620SGreg Clayton     const bool only_threads_with_stop_reason = true;
13407260f620SGreg Clayton     const uint32_t start_frame = 0;
13417260f620SGreg Clayton     const uint32_t num_frames = 1;
13427260f620SGreg Clayton     const uint32_t num_frames_with_source = 1;
13436a9767c7SJim Ingham     const bool stop_format = true;
1344c14ee32dSGreg Clayton     process->GetStatus(strm);
1345b9c1b51eSKate Stone     process->GetThreadStatus(strm, only_threads_with_stop_reason, start_frame,
13466a9767c7SJim Ingham                              num_frames, num_frames_with_source, stop_format);
1347d7c403e6SMed Ismail Bennani 
1348d7c403e6SMed Ismail Bennani     if (m_options.m_verbose) {
1349cdc6f8d7SJason Molenda       addr_t code_mask = process->GetCodeAddressMask();
1350cdc6f8d7SJason Molenda       addr_t data_mask = process->GetDataAddressMask();
1351cdc6f8d7SJason Molenda       if (code_mask != 0) {
1352cdc6f8d7SJason Molenda         int bits = std::bitset<64>(~code_mask).count();
1353cdc6f8d7SJason Molenda         result.AppendMessageWithFormat(
1354cdc6f8d7SJason Molenda             "Addressable code address mask: 0x%" PRIx64 "\n", code_mask);
1355cdc6f8d7SJason Molenda         result.AppendMessageWithFormat(
1356cdc6f8d7SJason Molenda             "Addressable data address mask: 0x%" PRIx64 "\n", data_mask);
1357cdc6f8d7SJason Molenda         result.AppendMessageWithFormat(
1358cdc6f8d7SJason Molenda             "Number of bits used in addressing (code): %d\n", bits);
1359cdc6f8d7SJason Molenda       }
1360cdc6f8d7SJason Molenda 
1361d7c403e6SMed Ismail Bennani       PlatformSP platform_sp = process->GetTarget().GetPlatform();
1362d7c403e6SMed Ismail Bennani       if (!platform_sp) {
1363d7c403e6SMed Ismail Bennani         result.AppendError("Couldn'retrieve the target's platform");
13644b9bea87SJim Ingham         return result.Succeeded();
13654b9bea87SJim Ingham       }
1366d7c403e6SMed Ismail Bennani 
1367d7c403e6SMed Ismail Bennani       auto expected_crash_info =
1368eefda182SMed Ismail Bennani           platform_sp->FetchExtendedCrashInformation(*process);
1369d7c403e6SMed Ismail Bennani 
1370d7c403e6SMed Ismail Bennani       if (!expected_crash_info) {
1371d7c403e6SMed Ismail Bennani         result.AppendError(llvm::toString(expected_crash_info.takeError()));
1372d7c403e6SMed Ismail Bennani         return result.Succeeded();
1373d7c403e6SMed Ismail Bennani       }
1374d7c403e6SMed Ismail Bennani 
1375d7c403e6SMed Ismail Bennani       StructuredData::DictionarySP crash_info_sp = *expected_crash_info;
1376d7c403e6SMed Ismail Bennani 
1377d7c403e6SMed Ismail Bennani       if (crash_info_sp) {
1378d7c403e6SMed Ismail Bennani         strm.PutCString("Extended Crash Information:\n");
1379d7c403e6SMed Ismail Bennani         crash_info_sp->Dump(strm);
1380d7c403e6SMed Ismail Bennani       }
1381d7c403e6SMed Ismail Bennani     }
1382d7c403e6SMed Ismail Bennani 
1383d7c403e6SMed Ismail Bennani     return result.Succeeded();
1384d7c403e6SMed Ismail Bennani   }
1385d7c403e6SMed Ismail Bennani 
1386d7c403e6SMed Ismail Bennani private:
1387d7c403e6SMed Ismail Bennani   CommandOptions m_options;
13884b9bea87SJim Ingham };
13894b9bea87SJim Ingham 
139035731357SCaroline Tice // CommandObjectProcessHandle
1391438dfcffSRaphael Isemann #define LLDB_OPTIONS_process_handle
1392438dfcffSRaphael Isemann #include "CommandOptions.inc"
13931f0f5b5bSZachary Turner 
1394bb9caf73SJim Ingham #pragma mark CommandObjectProcessHandle
139535731357SCaroline Tice 
1396b9c1b51eSKate Stone class CommandObjectProcessHandle : public CommandObjectParsed {
139735731357SCaroline Tice public:
1398b9c1b51eSKate Stone   class CommandOptions : public Options {
139935731357SCaroline Tice   public:
1400b9c1b51eSKate Stone     CommandOptions() : Options() { OptionParsingStarting(nullptr); }
140135731357SCaroline Tice 
140249bcfd80SEugene Zelenko     ~CommandOptions() override = default;
140335731357SCaroline Tice 
140497206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
1405b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
140697206d57SZachary Turner       Status error;
14073bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
140835731357SCaroline Tice 
1409b9c1b51eSKate Stone       switch (short_option) {
141035731357SCaroline Tice       case 's':
1411adcd0268SBenjamin Kramer         stop = std::string(option_arg);
141235731357SCaroline Tice         break;
141335731357SCaroline Tice       case 'n':
1414adcd0268SBenjamin Kramer         notify = std::string(option_arg);
141535731357SCaroline Tice         break;
141635731357SCaroline Tice       case 'p':
1417adcd0268SBenjamin Kramer         pass = std::string(option_arg);
141835731357SCaroline Tice         break;
141935731357SCaroline Tice       default:
142036162014SRaphael Isemann         llvm_unreachable("Unimplemented option");
142135731357SCaroline Tice       }
142235731357SCaroline Tice       return error;
142335731357SCaroline Tice     }
142435731357SCaroline Tice 
1425b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
142635731357SCaroline Tice       stop.clear();
142735731357SCaroline Tice       notify.clear();
142835731357SCaroline Tice       pass.clear();
142935731357SCaroline Tice     }
143035731357SCaroline Tice 
14311f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
143270602439SZachary Turner       return llvm::makeArrayRef(g_process_handle_options);
14331f0f5b5bSZachary Turner     }
143435731357SCaroline Tice 
143535731357SCaroline Tice     // Instance variables to hold the values for command options.
143635731357SCaroline Tice 
143735731357SCaroline Tice     std::string stop;
143835731357SCaroline Tice     std::string notify;
143935731357SCaroline Tice     std::string pass;
144035731357SCaroline Tice   };
144135731357SCaroline Tice 
14427428a18cSKate Stone   CommandObjectProcessHandle(CommandInterpreter &interpreter)
1443b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process handle",
1444b9c1b51eSKate Stone                             "Manage LLDB handling of OS signals for the "
1445b9c1b51eSKate Stone                             "current target process.  Defaults to showing "
1446b9c1b51eSKate Stone                             "current policy.",
144704a4c091SRaphael Isemann                             nullptr, eCommandRequiresTarget),
1448b9c1b51eSKate Stone         m_options() {
1449ea671fbdSKate Stone     SetHelpLong("\nIf no signals are specified, update them all.  If no update "
1450ea671fbdSKate Stone                 "option is specified, list the current values.");
145135731357SCaroline Tice     CommandArgumentEntry arg;
1452c0dbdfb6SCaroline Tice     CommandArgumentData signal_arg;
145335731357SCaroline Tice 
1454c0dbdfb6SCaroline Tice     signal_arg.arg_type = eArgTypeUnixSignal;
1455c0dbdfb6SCaroline Tice     signal_arg.arg_repetition = eArgRepeatStar;
145635731357SCaroline Tice 
1457c0dbdfb6SCaroline Tice     arg.push_back(signal_arg);
145835731357SCaroline Tice 
145935731357SCaroline Tice     m_arguments.push_back(arg);
146035731357SCaroline Tice   }
146135731357SCaroline Tice 
146249bcfd80SEugene Zelenko   ~CommandObjectProcessHandle() override = default;
146335731357SCaroline Tice 
1464b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
146535731357SCaroline Tice 
1466b9c1b51eSKate Stone   bool VerifyCommandOptionValue(const std::string &option, int &real_value) {
146735731357SCaroline Tice     bool okay = true;
146810ad7993SCaroline Tice     bool success = false;
146947cbf4a0SPavel Labath     bool tmp_value = OptionArgParser::ToBoolean(option, false, &success);
147010ad7993SCaroline Tice 
147110ad7993SCaroline Tice     if (success && tmp_value)
147210ad7993SCaroline Tice       real_value = 1;
147310ad7993SCaroline Tice     else if (success && !tmp_value)
147410ad7993SCaroline Tice       real_value = 0;
1475b9c1b51eSKate Stone     else {
147635731357SCaroline Tice       // If the value isn't 'true' or 'false', it had better be 0 or 1.
14779010cef2SRaphael Isemann       if (!llvm::to_integer(option, real_value))
14789010cef2SRaphael Isemann         real_value = 3;
147910ad7993SCaroline Tice       if (real_value != 0 && real_value != 1)
148035731357SCaroline Tice         okay = false;
148135731357SCaroline Tice     }
148235731357SCaroline Tice 
148335731357SCaroline Tice     return okay;
148435731357SCaroline Tice   }
148535731357SCaroline Tice 
1486b9c1b51eSKate Stone   void PrintSignalHeader(Stream &str) {
148710ad7993SCaroline Tice     str.Printf("NAME         PASS   STOP   NOTIFY\n");
1488b84141a6SPavel Labath     str.Printf("===========  =====  =====  ======\n");
148910ad7993SCaroline Tice   }
149010ad7993SCaroline Tice 
1491b9c1b51eSKate Stone   void PrintSignal(Stream &str, int32_t signo, const char *sig_name,
1492b9c1b51eSKate Stone                    const UnixSignalsSP &signals_sp) {
149310ad7993SCaroline Tice     bool stop;
149410ad7993SCaroline Tice     bool suppress;
149510ad7993SCaroline Tice     bool notify;
149610ad7993SCaroline Tice 
1497b84141a6SPavel Labath     str.Printf("%-11s  ", sig_name);
1498b9c1b51eSKate Stone     if (signals_sp->GetSignalInfo(signo, suppress, stop, notify)) {
149910ad7993SCaroline Tice       bool pass = !suppress;
1500b9c1b51eSKate Stone       str.Printf("%s  %s  %s", (pass ? "true " : "false"),
1501b9c1b51eSKate Stone                  (stop ? "true " : "false"), (notify ? "true " : "false"));
150210ad7993SCaroline Tice     }
150310ad7993SCaroline Tice     str.Printf("\n");
150410ad7993SCaroline Tice   }
150510ad7993SCaroline Tice 
1506b9c1b51eSKate Stone   void PrintSignalInformation(Stream &str, Args &signal_args,
1507b9c1b51eSKate Stone                               int num_valid_signals,
1508b9c1b51eSKate Stone                               const UnixSignalsSP &signals_sp) {
150910ad7993SCaroline Tice     PrintSignalHeader(str);
151010ad7993SCaroline Tice 
1511b9c1b51eSKate Stone     if (num_valid_signals > 0) {
151210ad7993SCaroline Tice       size_t num_args = signal_args.GetArgumentCount();
1513b9c1b51eSKate Stone       for (size_t i = 0; i < num_args; ++i) {
1514b9c1b51eSKate Stone         int32_t signo = signals_sp->GetSignalNumberFromName(
1515b9c1b51eSKate Stone             signal_args.GetArgumentAtIndex(i));
151610ad7993SCaroline Tice         if (signo != LLDB_INVALID_SIGNAL_NUMBER)
1517b9c1b51eSKate Stone           PrintSignal(str, signo, signal_args.GetArgumentAtIndex(i),
1518b9c1b51eSKate Stone                       signals_sp);
151910ad7993SCaroline Tice       }
1520b9c1b51eSKate Stone     } else // Print info for ALL signals
152110ad7993SCaroline Tice     {
152298d0a4b3SChaoren Lin       int32_t signo = signals_sp->GetFirstSignalNumber();
1523b9c1b51eSKate Stone       while (signo != LLDB_INVALID_SIGNAL_NUMBER) {
1524b9c1b51eSKate Stone         PrintSignal(str, signo, signals_sp->GetSignalAsCString(signo),
1525b9c1b51eSKate Stone                     signals_sp);
152698d0a4b3SChaoren Lin         signo = signals_sp->GetNextSignalNumber(signo);
152710ad7993SCaroline Tice       }
152810ad7993SCaroline Tice     }
152910ad7993SCaroline Tice   }
153010ad7993SCaroline Tice 
15315a988416SJim Ingham protected:
1532b9c1b51eSKate Stone   bool DoExecute(Args &signal_args, CommandReturnObject &result) override {
153304a4c091SRaphael Isemann     Target *target_sp = &GetSelectedTarget();
153435731357SCaroline Tice 
153535731357SCaroline Tice     ProcessSP process_sp = target_sp->GetProcessSP();
153635731357SCaroline Tice 
1537b9c1b51eSKate Stone     if (!process_sp) {
1538b9c1b51eSKate Stone       result.AppendError("No current process; cannot handle signals until you "
1539b9c1b51eSKate Stone                          "have a valid process.\n");
154035731357SCaroline Tice       return false;
154135731357SCaroline Tice     }
154235731357SCaroline Tice 
154335731357SCaroline Tice     int stop_action = -1;   // -1 means leave the current setting alone
154435731357SCaroline Tice     int pass_action = -1;   // -1 means leave the current setting alone
154535731357SCaroline Tice     int notify_action = -1; // -1 means leave the current setting alone
154635731357SCaroline Tice 
1547b9c1b51eSKate Stone     if (!m_options.stop.empty() &&
1548b9c1b51eSKate Stone         !VerifyCommandOptionValue(m_options.stop, stop_action)) {
1549b9c1b51eSKate Stone       result.AppendError("Invalid argument for command option --stop; must be "
1550b9c1b51eSKate Stone                          "true or false.\n");
155135731357SCaroline Tice       return false;
155235731357SCaroline Tice     }
155335731357SCaroline Tice 
1554b9c1b51eSKate Stone     if (!m_options.notify.empty() &&
1555b9c1b51eSKate Stone         !VerifyCommandOptionValue(m_options.notify, notify_action)) {
1556b9c1b51eSKate Stone       result.AppendError("Invalid argument for command option --notify; must "
1557b9c1b51eSKate Stone                          "be true or false.\n");
155835731357SCaroline Tice       return false;
155935731357SCaroline Tice     }
156035731357SCaroline Tice 
1561b9c1b51eSKate Stone     if (!m_options.pass.empty() &&
1562b9c1b51eSKate Stone         !VerifyCommandOptionValue(m_options.pass, pass_action)) {
1563b9c1b51eSKate Stone       result.AppendError("Invalid argument for command option --pass; must be "
1564b9c1b51eSKate Stone                          "true or false.\n");
156535731357SCaroline Tice       return false;
156635731357SCaroline Tice     }
156735731357SCaroline Tice 
156835731357SCaroline Tice     size_t num_args = signal_args.GetArgumentCount();
156998d0a4b3SChaoren Lin     UnixSignalsSP signals_sp = process_sp->GetUnixSignals();
157035731357SCaroline Tice     int num_signals_set = 0;
157135731357SCaroline Tice 
1572b9c1b51eSKate Stone     if (num_args > 0) {
1573d6a24757SZachary Turner       for (const auto &arg : signal_args) {
1574d6a24757SZachary Turner         int32_t signo = signals_sp->GetSignalNumberFromName(arg.c_str());
1575b9c1b51eSKate Stone         if (signo != LLDB_INVALID_SIGNAL_NUMBER) {
1576b9c1b51eSKate Stone           // Casting the actions as bools here should be okay, because
157705097246SAdrian Prantl           // VerifyCommandOptionValue guarantees the value is either 0 or 1.
157835731357SCaroline Tice           if (stop_action != -1)
157998d0a4b3SChaoren Lin             signals_sp->SetShouldStop(signo, stop_action);
1580b9c1b51eSKate Stone           if (pass_action != -1) {
158198d0a4b3SChaoren Lin             bool suppress = !pass_action;
158298d0a4b3SChaoren Lin             signals_sp->SetShouldSuppress(signo, suppress);
158335731357SCaroline Tice           }
158435731357SCaroline Tice           if (notify_action != -1)
158598d0a4b3SChaoren Lin             signals_sp->SetShouldNotify(signo, notify_action);
158635731357SCaroline Tice           ++num_signals_set;
1587b9c1b51eSKate Stone         } else {
1588b9c1b51eSKate Stone           result.AppendErrorWithFormat("Invalid signal name '%s'\n",
1589d6a24757SZachary Turner                                        arg.c_str());
159035731357SCaroline Tice         }
159135731357SCaroline Tice       }
1592b9c1b51eSKate Stone     } else {
1593b9c1b51eSKate Stone       // No signal specified, if any command options were specified, update ALL
1594b9c1b51eSKate Stone       // signals.
1595b9c1b51eSKate Stone       if ((notify_action != -1) || (stop_action != -1) || (pass_action != -1)) {
1596b9c1b51eSKate Stone         if (m_interpreter.Confirm(
1597b9c1b51eSKate Stone                 "Do you really want to update all the signals?", false)) {
159898d0a4b3SChaoren Lin           int32_t signo = signals_sp->GetFirstSignalNumber();
1599b9c1b51eSKate Stone           while (signo != LLDB_INVALID_SIGNAL_NUMBER) {
160010ad7993SCaroline Tice             if (notify_action != -1)
160198d0a4b3SChaoren Lin               signals_sp->SetShouldNotify(signo, notify_action);
160210ad7993SCaroline Tice             if (stop_action != -1)
160398d0a4b3SChaoren Lin               signals_sp->SetShouldStop(signo, stop_action);
1604b9c1b51eSKate Stone             if (pass_action != -1) {
160598d0a4b3SChaoren Lin               bool suppress = !pass_action;
160698d0a4b3SChaoren Lin               signals_sp->SetShouldSuppress(signo, suppress);
160710ad7993SCaroline Tice             }
160898d0a4b3SChaoren Lin             signo = signals_sp->GetNextSignalNumber(signo);
160910ad7993SCaroline Tice           }
161010ad7993SCaroline Tice         }
161110ad7993SCaroline Tice       }
161210ad7993SCaroline Tice     }
161310ad7993SCaroline Tice 
1614b9c1b51eSKate Stone     PrintSignalInformation(result.GetOutputStream(), signal_args,
1615b9c1b51eSKate Stone                            num_signals_set, signals_sp);
161635731357SCaroline Tice 
161735731357SCaroline Tice     if (num_signals_set > 0)
161835731357SCaroline Tice       result.SetStatus(eReturnStatusSuccessFinishNoResult);
161935731357SCaroline Tice     else
162035731357SCaroline Tice       result.SetStatus(eReturnStatusFailed);
162135731357SCaroline Tice 
162235731357SCaroline Tice     return result.Succeeded();
162335731357SCaroline Tice   }
162435731357SCaroline Tice 
162535731357SCaroline Tice   CommandOptions m_options;
162635731357SCaroline Tice };
162735731357SCaroline Tice 
16280b697561SWalter Erquinigo // Next are the subcommands of CommandObjectMultiwordProcessTrace
16290b697561SWalter Erquinigo 
16300b697561SWalter Erquinigo // CommandObjectProcessTraceStart
16310b697561SWalter Erquinigo class CommandObjectProcessTraceStart : public CommandObjectTraceProxy {
16320b697561SWalter Erquinigo public:
16330b697561SWalter Erquinigo   CommandObjectProcessTraceStart(CommandInterpreter &interpreter)
16340b697561SWalter Erquinigo       : CommandObjectTraceProxy(
16350b697561SWalter Erquinigo             /*live_debug_session_only*/ true, interpreter,
16360b697561SWalter Erquinigo             "process trace start",
16370b697561SWalter Erquinigo             "Start tracing this process with the corresponding trace "
16380b697561SWalter Erquinigo             "plug-in.",
16390b697561SWalter Erquinigo             "process trace start [<trace-options>]") {}
16400b697561SWalter Erquinigo 
16410b697561SWalter Erquinigo protected:
16420b697561SWalter Erquinigo   lldb::CommandObjectSP GetDelegateCommand(Trace &trace) override {
16430b697561SWalter Erquinigo     return trace.GetProcessTraceStartCommand(m_interpreter);
16440b697561SWalter Erquinigo   }
16450b697561SWalter Erquinigo };
16460b697561SWalter Erquinigo 
16470b697561SWalter Erquinigo // CommandObjectProcessTraceStop
16480b697561SWalter Erquinigo class CommandObjectProcessTraceStop : public CommandObjectParsed {
16490b697561SWalter Erquinigo public:
16500b697561SWalter Erquinigo   CommandObjectProcessTraceStop(CommandInterpreter &interpreter)
16510b697561SWalter Erquinigo       : CommandObjectParsed(interpreter, "process trace stop",
16520b697561SWalter Erquinigo                             "Stop tracing this process. This does not affect "
16530b697561SWalter Erquinigo                             "traces started with the "
16540b697561SWalter Erquinigo                             "\"thread trace start\" command.",
16550b697561SWalter Erquinigo                             "process trace stop",
16560b697561SWalter Erquinigo                             eCommandRequiresProcess | eCommandTryTargetAPILock |
16570b697561SWalter Erquinigo                                 eCommandProcessMustBeLaunched |
16580b697561SWalter Erquinigo                                 eCommandProcessMustBePaused |
16590b697561SWalter Erquinigo                                 eCommandProcessMustBeTraced) {}
16600b697561SWalter Erquinigo 
16610b697561SWalter Erquinigo   ~CommandObjectProcessTraceStop() override = default;
16620b697561SWalter Erquinigo 
16630b697561SWalter Erquinigo   bool DoExecute(Args &command, CommandReturnObject &result) override {
16640b697561SWalter Erquinigo     ProcessSP process_sp = m_exe_ctx.GetProcessSP();
16650b697561SWalter Erquinigo 
16660b697561SWalter Erquinigo     TraceSP trace_sp = process_sp->GetTarget().GetTrace();
16670b697561SWalter Erquinigo 
1668bf9f21a2SWalter Erquinigo     if (llvm::Error err = trace_sp->Stop())
16691b1c8e4aSDavid Spickett       result.AppendError(toString(std::move(err)));
16700b697561SWalter Erquinigo     else
16710b697561SWalter Erquinigo       result.SetStatus(eReturnStatusSuccessFinishResult);
16720b697561SWalter Erquinigo 
16730b697561SWalter Erquinigo     return result.Succeeded();
16740b697561SWalter Erquinigo   }
16750b697561SWalter Erquinigo };
16760b697561SWalter Erquinigo 
16770b697561SWalter Erquinigo // CommandObjectMultiwordProcessTrace
16780b697561SWalter Erquinigo class CommandObjectMultiwordProcessTrace : public CommandObjectMultiword {
16790b697561SWalter Erquinigo public:
16800b697561SWalter Erquinigo   CommandObjectMultiwordProcessTrace(CommandInterpreter &interpreter)
16810b697561SWalter Erquinigo       : CommandObjectMultiword(
16820b697561SWalter Erquinigo             interpreter, "trace", "Commands for tracing the current process.",
16830b697561SWalter Erquinigo             "process trace <subcommand> [<subcommand objects>]") {
16840b697561SWalter Erquinigo     LoadSubCommand("start", CommandObjectSP(new CommandObjectProcessTraceStart(
16850b697561SWalter Erquinigo                                 interpreter)));
16860b697561SWalter Erquinigo     LoadSubCommand("stop", CommandObjectSP(
16870b697561SWalter Erquinigo                                new CommandObjectProcessTraceStop(interpreter)));
16880b697561SWalter Erquinigo   }
16890b697561SWalter Erquinigo 
16900b697561SWalter Erquinigo   ~CommandObjectMultiwordProcessTrace() override = default;
16910b697561SWalter Erquinigo };
16920b697561SWalter Erquinigo 
169330fdc8d8SChris Lattner // CommandObjectMultiwordProcess
169430fdc8d8SChris Lattner 
1695b9c1b51eSKate Stone CommandObjectMultiwordProcess::CommandObjectMultiwordProcess(
1696b9c1b51eSKate Stone     CommandInterpreter &interpreter)
1697b9c1b51eSKate Stone     : CommandObjectMultiword(
1698b9c1b51eSKate Stone           interpreter, "process",
1699b9c1b51eSKate Stone           "Commands for interacting with processes on the current platform.",
1700b9c1b51eSKate Stone           "process <subcommand> [<subcommand-options>]") {
1701b9c1b51eSKate Stone   LoadSubCommand("attach",
1702b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessAttach(interpreter)));
1703b9c1b51eSKate Stone   LoadSubCommand("launch",
1704b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessLaunch(interpreter)));
1705b9c1b51eSKate Stone   LoadSubCommand("continue", CommandObjectSP(new CommandObjectProcessContinue(
1706b9c1b51eSKate Stone                                  interpreter)));
1707b9c1b51eSKate Stone   LoadSubCommand("connect",
1708b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessConnect(interpreter)));
1709b9c1b51eSKate Stone   LoadSubCommand("detach",
1710b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessDetach(interpreter)));
1711b9c1b51eSKate Stone   LoadSubCommand("load",
1712b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessLoad(interpreter)));
1713b9c1b51eSKate Stone   LoadSubCommand("unload",
1714b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessUnload(interpreter)));
1715b9c1b51eSKate Stone   LoadSubCommand("signal",
1716b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessSignal(interpreter)));
1717b9c1b51eSKate Stone   LoadSubCommand("handle",
1718b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessHandle(interpreter)));
1719b9c1b51eSKate Stone   LoadSubCommand("status",
1720b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessStatus(interpreter)));
1721b9c1b51eSKate Stone   LoadSubCommand("interrupt", CommandObjectSP(new CommandObjectProcessInterrupt(
1722b9c1b51eSKate Stone                                   interpreter)));
1723b9c1b51eSKate Stone   LoadSubCommand("kill",
1724b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessKill(interpreter)));
1725b9c1b51eSKate Stone   LoadSubCommand("plugin",
1726b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessPlugin(interpreter)));
1727b9c1b51eSKate Stone   LoadSubCommand("save-core", CommandObjectSP(new CommandObjectProcessSaveCore(
1728b9c1b51eSKate Stone                                   interpreter)));
17290b697561SWalter Erquinigo   LoadSubCommand(
17300b697561SWalter Erquinigo       "trace",
17310b697561SWalter Erquinigo       CommandObjectSP(new CommandObjectMultiwordProcessTrace(interpreter)));
173230fdc8d8SChris Lattner }
173330fdc8d8SChris Lattner 
173449bcfd80SEugene Zelenko CommandObjectMultiwordProcess::~CommandObjectMultiwordProcess() = default;
1735