130fdc8d8SChris Lattner //===-- CommandObjectProcess.cpp --------------------------------*- C++ -*-===//
230fdc8d8SChris Lattner //
330fdc8d8SChris Lattner //                     The LLVM Compiler Infrastructure
430fdc8d8SChris Lattner //
530fdc8d8SChris Lattner // This file is distributed under the University of Illinois Open Source
630fdc8d8SChris Lattner // License. See LICENSE.TXT for details.
730fdc8d8SChris Lattner //
830fdc8d8SChris Lattner //===----------------------------------------------------------------------===//
930fdc8d8SChris Lattner 
1030fdc8d8SChris Lattner // C Includes
1130fdc8d8SChris Lattner // C++ Includes
1230fdc8d8SChris Lattner // Other libraries and framework includes
1330fdc8d8SChris Lattner // Project includes
1449bcfd80SEugene Zelenko #include "CommandObjectProcess.h"
150e41084aSJim Ingham #include "lldb/Breakpoint/Breakpoint.h"
160e41084aSJim Ingham #include "lldb/Breakpoint/BreakpointLocation.h"
170e41084aSJim Ingham #include "lldb/Breakpoint/BreakpointSite.h"
181f746071SGreg Clayton #include "lldb/Core/Module.h"
19a2715cf1SGreg Clayton #include "lldb/Core/PluginManager.h"
20b9c1b51eSKate Stone #include "lldb/Core/State.h"
217260f620SGreg Clayton #include "lldb/Host/Host.h"
225275aaa0SVince Harron #include "lldb/Host/StringConvert.h"
230e41084aSJim Ingham #include "lldb/Interpreter/Args.h"
2430fdc8d8SChris Lattner #include "lldb/Interpreter/CommandInterpreter.h"
2530fdc8d8SChris Lattner #include "lldb/Interpreter/CommandReturnObject.h"
26b9c1b51eSKate Stone #include "lldb/Interpreter/Options.h"
27e996fd30SGreg Clayton #include "lldb/Target/Platform.h"
2830fdc8d8SChris Lattner #include "lldb/Target/Process.h"
290e41084aSJim Ingham #include "lldb/Target/StopInfo.h"
3030fdc8d8SChris Lattner #include "lldb/Target/Target.h"
3130fdc8d8SChris Lattner #include "lldb/Target/Thread.h"
3293749ab3SZachary Turner #include "lldb/Target/UnixSignals.h"
3330fdc8d8SChris Lattner 
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) {
56dcb1d856SJim Ingham         char message[1024];
57dcb1d856SJim Ingham         if (process->GetState() == eStateAttaching)
58b9c1b51eSKate Stone           ::snprintf(message, sizeof(message),
59b9c1b51eSKate Stone                      "There is a pending attach, abort it and %s?",
60b9c1b51eSKate Stone                      m_new_process_action.c_str());
61dcb1d856SJim Ingham         else if (process->GetShouldDetach())
62b9c1b51eSKate Stone           ::snprintf(message, sizeof(message),
63b9c1b51eSKate Stone                      "There is a running process, detach from it and %s?",
64b9c1b51eSKate Stone                      m_new_process_action.c_str());
65dcb1d856SJim Ingham         else
66b9c1b51eSKate Stone           ::snprintf(message, sizeof(message),
67b9c1b51eSKate Stone                      "There is a running process, kill it and %s?",
68b9c1b51eSKate Stone                      m_new_process_action.c_str());
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;
76acff8950SJim Ingham             Error 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               result.SetStatus(eReturnStatusFailed);
85dcb1d856SJim Ingham             }
86b9c1b51eSKate Stone           } else {
87ede3193bSJason Molenda             Error destroy_error(process->Destroy(false));
88b9c1b51eSKate Stone             if (destroy_error.Success()) {
89dcb1d856SJim Ingham               result.SetStatus(eReturnStatusSuccessFinishResult);
9049bcfd80SEugene Zelenko               process = nullptr;
91b9c1b51eSKate Stone             } else {
92b9c1b51eSKate Stone               result.AppendErrorWithFormat("Failed to kill process: %s\n",
93b9c1b51eSKate Stone                                            destroy_error.AsCString());
94dcb1d856SJim Ingham               result.SetStatus(eReturnStatusFailed);
95dcb1d856SJim Ingham             }
96dcb1d856SJim Ingham           }
97dcb1d856SJim Ingham         }
98dcb1d856SJim Ingham       }
99dcb1d856SJim Ingham     }
100dcb1d856SJim Ingham     return result.Succeeded();
101dcb1d856SJim Ingham   }
10249bcfd80SEugene Zelenko 
103dcb1d856SJim Ingham   std::string m_new_process_action;
104dcb1d856SJim Ingham };
10549bcfd80SEugene Zelenko 
10630fdc8d8SChris Lattner //-------------------------------------------------------------------------
10730fdc8d8SChris Lattner // CommandObjectProcessLaunch
10830fdc8d8SChris Lattner //-------------------------------------------------------------------------
1094bddaeb5SJim Ingham #pragma mark CommandObjectProcessLaunch
110b9c1b51eSKate Stone class CommandObjectProcessLaunch : public CommandObjectProcessLaunchOrAttach {
11130fdc8d8SChris Lattner public:
112b9c1b51eSKate Stone   CommandObjectProcessLaunch(CommandInterpreter &interpreter)
113b9c1b51eSKate Stone       : CommandObjectProcessLaunchOrAttach(
114b9c1b51eSKate Stone             interpreter, "process launch",
115b9c1b51eSKate Stone             "Launch the executable in the debugger.", nullptr,
116b9c1b51eSKate Stone             eCommandRequiresTarget, "restart"),
117b9c1b51eSKate Stone         m_options() {
118405fe67fSCaroline Tice     CommandArgumentEntry arg;
119405fe67fSCaroline Tice     CommandArgumentData run_args_arg;
120405fe67fSCaroline Tice 
121405fe67fSCaroline Tice     // Define the first (and only) variant of this arg.
122405fe67fSCaroline Tice     run_args_arg.arg_type = eArgTypeRunArgs;
123405fe67fSCaroline Tice     run_args_arg.arg_repetition = eArgRepeatOptional;
124405fe67fSCaroline Tice 
125b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
126b9c1b51eSKate Stone     // argument entry.
127405fe67fSCaroline Tice     arg.push_back(run_args_arg);
128405fe67fSCaroline Tice 
129405fe67fSCaroline Tice     // Push the data for the first argument into the m_arguments vector.
130405fe67fSCaroline Tice     m_arguments.push_back(arg);
13130fdc8d8SChris Lattner   }
13230fdc8d8SChris Lattner 
13349bcfd80SEugene Zelenko   ~CommandObjectProcessLaunch() override = default;
13430fdc8d8SChris Lattner 
135b9c1b51eSKate Stone   int HandleArgumentCompletion(Args &input, int &cursor_index,
136e9ce62b6SJim Ingham                                int &cursor_char_position,
137e9ce62b6SJim Ingham                                OptionElementVector &opt_element_vector,
138b9c1b51eSKate Stone                                int match_start_point, int max_return_elements,
139e9ce62b6SJim Ingham                                bool &word_complete,
140b9c1b51eSKate Stone                                StringList &matches) override {
141e9ce62b6SJim Ingham     std::string completion_str(input.GetArgumentAtIndex(cursor_index));
142e9ce62b6SJim Ingham     completion_str.erase(cursor_char_position);
143e9ce62b6SJim Ingham 
144b9c1b51eSKate Stone     CommandCompletions::InvokeCommonCompletionCallbacks(
145b9c1b51eSKate Stone         GetCommandInterpreter(), CommandCompletions::eDiskFileCompletion,
146b9c1b51eSKate Stone         completion_str.c_str(), match_start_point, max_return_elements, nullptr,
147b9c1b51eSKate Stone         word_complete, matches);
148e9ce62b6SJim Ingham     return matches.GetSize();
149e9ce62b6SJim Ingham   }
150e9ce62b6SJim Ingham 
151b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
15230fdc8d8SChris Lattner 
153b9c1b51eSKate Stone   const char *GetRepeatCommand(Args &current_command_args,
154b9c1b51eSKate Stone                                uint32_t index) override {
1555a988416SJim Ingham     // No repeat for "process launch"...
1565a988416SJim Ingham     return "";
1575a988416SJim Ingham   }
1585a988416SJim Ingham 
1595a988416SJim Ingham protected:
160b9c1b51eSKate Stone   bool DoExecute(Args &launch_args, CommandReturnObject &result) override {
1611d885966SGreg Clayton     Debugger &debugger = m_interpreter.GetDebugger();
1621d885966SGreg Clayton     Target *target = debugger.GetSelectedTarget().get();
16349bcfd80SEugene Zelenko     // If our listener is nullptr, users aren't allows to launch
164b09c5384SGreg Clayton     ModuleSP exe_module_sp = target->GetExecutableModule();
16571337622SGreg Clayton 
166b9c1b51eSKate Stone     if (exe_module_sp == nullptr) {
167b9c1b51eSKate Stone       result.AppendError("no file in target, create a debug target using the "
168b9c1b51eSKate Stone                          "'target create' command");
16971337622SGreg Clayton       result.SetStatus(eReturnStatusFailed);
17071337622SGreg Clayton       return false;
17171337622SGreg Clayton     }
17271337622SGreg Clayton 
17371337622SGreg Clayton     StateType state = eStateInvalid;
17471337622SGreg Clayton 
175b09c5384SGreg Clayton     if (!StopProcessIfNecessary(m_exe_ctx.GetProcessPtr(), state, result))
17630fdc8d8SChris Lattner       return false;
17730fdc8d8SChris Lattner 
17845392553SGreg Clayton     const char *target_settings_argv0 = target->GetArg0();
17945392553SGreg Clayton 
180b9c1b51eSKate Stone     // Determine whether we will disable ASLR or leave it in the default state
181b9c1b51eSKate Stone     // (i.e. enabled if the platform supports it).
182b9c1b51eSKate Stone     // First check if the process launch options explicitly turn on/off
183b9c1b51eSKate Stone     // disabling ASLR.  If so, use that setting;
1845163792bSTodd Fiala     // otherwise, use the 'settings target.disable-aslr' setting.
1855163792bSTodd Fiala     bool disable_aslr = false;
186b9c1b51eSKate Stone     if (m_options.disable_aslr != eLazyBoolCalculate) {
187b9c1b51eSKate Stone       // The user specified an explicit setting on the process launch line.  Use
188b9c1b51eSKate Stone       // it.
1895163792bSTodd Fiala       disable_aslr = (m_options.disable_aslr == eLazyBoolYes);
190b9c1b51eSKate Stone     } else {
191b9c1b51eSKate Stone       // The user did not explicitly specify whether to disable ASLR.  Fall back
192b9c1b51eSKate Stone       // to the target.disable-aslr setting.
1935163792bSTodd Fiala       disable_aslr = target->GetDisableASLR();
1945163792bSTodd Fiala     }
1955163792bSTodd Fiala 
1965163792bSTodd Fiala     if (disable_aslr)
197b09c5384SGreg Clayton       m_options.launch_info.GetFlags().Set(eLaunchFlagDisableASLR);
1985163792bSTodd Fiala     else
1995163792bSTodd Fiala       m_options.launch_info.GetFlags().Clear(eLaunchFlagDisableASLR);
200b09c5384SGreg Clayton 
201106d0286SJim Ingham     if (target->GetDetachOnError())
202106d0286SJim Ingham       m_options.launch_info.GetFlags().Set(eLaunchFlagDetachOnError);
203106d0286SJim Ingham 
204b09c5384SGreg Clayton     if (target->GetDisableSTDIO())
205b09c5384SGreg Clayton       m_options.launch_info.GetFlags().Set(eLaunchFlagDisableSTDIO);
206b09c5384SGreg Clayton 
207b09c5384SGreg Clayton     Args environment;
208b09c5384SGreg Clayton     target->GetEnvironmentAsArgs(environment);
209b09c5384SGreg Clayton     if (environment.GetArgumentCount() > 0)
210b9c1b51eSKate Stone       m_options.launch_info.GetEnvironmentEntries().AppendArguments(
211b9c1b51eSKate Stone           environment);
21245392553SGreg Clayton 
213b9c1b51eSKate Stone     if (target_settings_argv0) {
214b9c1b51eSKate Stone       m_options.launch_info.GetArguments().AppendArgument(
215ecbb0bb1SZachary Turner           llvm::StringRef(target_settings_argv0));
216b9c1b51eSKate Stone       m_options.launch_info.SetExecutableFile(
217b9c1b51eSKate Stone           exe_module_sp->GetPlatformFileSpec(), false);
218b9c1b51eSKate Stone     } else {
219b9c1b51eSKate Stone       m_options.launch_info.SetExecutableFile(
220b9c1b51eSKate Stone           exe_module_sp->GetPlatformFileSpec(), true);
22145392553SGreg Clayton     }
22245392553SGreg Clayton 
223b9c1b51eSKate Stone     if (launch_args.GetArgumentCount() == 0) {
224b9c1b51eSKate Stone       m_options.launch_info.GetArguments().AppendArguments(
225b9c1b51eSKate Stone           target->GetProcessLaunchInfo().GetArguments());
226b9c1b51eSKate Stone     } else {
22745392553SGreg Clayton       m_options.launch_info.GetArguments().AppendArguments(launch_args);
228162b597cSGreg Clayton       // Save the arguments for subsequent runs in the current target.
229162b597cSGreg Clayton       target->SetRunArguments(launch_args);
230982c9762SGreg Clayton     }
2311d885966SGreg Clayton 
232dc6224e0SGreg Clayton     StreamString stream;
2338012cadbSGreg Clayton     Error error = target->Launch(m_options.launch_info, &stream);
23430fdc8d8SChris Lattner 
235b9c1b51eSKate Stone     if (error.Success()) {
236b09c5384SGreg Clayton       ProcessSP process_sp(target->GetProcessSP());
237b9c1b51eSKate Stone       if (process_sp) {
238b9c1b51eSKate Stone         // There is a race condition where this thread will return up the call
239b9c1b51eSKate Stone         // stack to the main command
240b9c1b51eSKate Stone         // handler and show an (lldb) prompt before HandlePrivateEvent (from
241b9c1b51eSKate Stone         // PrivateStateThread) has
2428f0db3e1SIlia K         // a chance to call PushProcessIOHandler().
2434446487dSPavel Labath         process_sp->SyncIOHandler(0, 2000);
2448f0db3e1SIlia K 
245f2ef94e7SStephane Sezer         const char *data = stream.GetData();
246f2ef94e7SStephane Sezer         if (data && strlen(data) > 0)
247dc6224e0SGreg Clayton           result.AppendMessage(stream.GetData());
248b9c1b51eSKate Stone         const char *archname =
249b9c1b51eSKate Stone             exe_module_sp->GetArchitecture().GetArchitectureName();
250b9c1b51eSKate Stone         result.AppendMessageWithFormat(
251b9c1b51eSKate Stone             "Process %" PRIu64 " launched: '%s' (%s)\n", process_sp->GetID(),
252b9c1b51eSKate Stone             exe_module_sp->GetFileSpec().GetPath().c_str(), archname);
25305faeb71SGreg Clayton         result.SetStatus(eReturnStatusSuccessFinishResult);
254b09c5384SGreg Clayton         result.SetDidChangeProcessState(true);
255b9c1b51eSKate Stone       } else {
256b9c1b51eSKate Stone         result.AppendError(
257b9c1b51eSKate Stone             "no error returned from Target::Launch, and target has no process");
258514487e8SGreg Clayton         result.SetStatus(eReturnStatusFailed);
25930fdc8d8SChris Lattner       }
260b9c1b51eSKate Stone     } else {
261b09c5384SGreg Clayton       result.AppendError(error.AsCString());
262514487e8SGreg Clayton       result.SetStatus(eReturnStatusFailed);
263514487e8SGreg Clayton     }
26430fdc8d8SChris Lattner     return result.Succeeded();
26530fdc8d8SChris Lattner   }
26630fdc8d8SChris Lattner 
26730fdc8d8SChris Lattner protected:
268982c9762SGreg Clayton   ProcessLaunchCommandOptions m_options;
26930fdc8d8SChris Lattner };
27030fdc8d8SChris Lattner 
271982c9762SGreg Clayton //#define SET1 LLDB_OPT_SET_1
272982c9762SGreg Clayton //#define SET2 LLDB_OPT_SET_2
273982c9762SGreg Clayton //#define SET3 LLDB_OPT_SET_3
274982c9762SGreg Clayton //
275982c9762SGreg Clayton // OptionDefinition
276982c9762SGreg Clayton // CommandObjectProcessLaunch::CommandOptions::g_option_table[] =
277982c9762SGreg Clayton //{
278ac9c3a62SKate Stone //  // clang-format off
279b9c1b51eSKate Stone //  {SET1 | SET2 | SET3, false, "stop-at-entry", 's', OptionParser::eNoArgument,
280b9c1b51eSKate Stone //  nullptr, 0, eArgTypeNone,          "Stop at the entry point of the program
281b9c1b51eSKate Stone //  when launching a process."},
282b9c1b51eSKate Stone //  {SET1,               false, "stdin",         'i',
283b9c1b51eSKate Stone //  OptionParser::eRequiredArgument, nullptr, 0, eArgTypeDirectoryName,
284b9c1b51eSKate Stone //  "Redirect stdin for the process to <path>."},
285b9c1b51eSKate Stone //  {SET1,               false, "stdout",        'o',
286b9c1b51eSKate Stone //  OptionParser::eRequiredArgument, nullptr, 0, eArgTypeDirectoryName,
287b9c1b51eSKate Stone //  "Redirect stdout for the process to <path>."},
288b9c1b51eSKate Stone //  {SET1,               false, "stderr",        'e',
289b9c1b51eSKate Stone //  OptionParser::eRequiredArgument, nullptr, 0, eArgTypeDirectoryName,
290b9c1b51eSKate Stone //  "Redirect stderr for the process to <path>."},
291b9c1b51eSKate Stone //  {SET1 | SET2 | SET3, false, "plugin",        'p',
292b9c1b51eSKate Stone //  OptionParser::eRequiredArgument, nullptr, 0, eArgTypePlugin,        "Name of
293b9c1b51eSKate Stone //  the process plugin you want to use."},
294b9c1b51eSKate Stone //  {       SET2,        false, "tty",           't',
295b9c1b51eSKate Stone //  OptionParser::eOptionalArgument, nullptr, 0, eArgTypeDirectoryName, "Start
296b9c1b51eSKate Stone //  the process in a terminal. If <path> is specified, look for a terminal whose
297b9c1b51eSKate Stone //  name contains <path>, else start the process in a new terminal."},
298b9c1b51eSKate Stone //  {              SET3, false, "no-stdio",      'n', OptionParser::eNoArgument,
299b9c1b51eSKate Stone //  nullptr, 0, eArgTypeNone,          "Do not set up for terminal I/O to go to
300b9c1b51eSKate Stone //  running process."},
301b9c1b51eSKate Stone //  {SET1 | SET2 | SET3, false, "working-dir",   'w',
302b9c1b51eSKate Stone //  OptionParser::eRequiredArgument, nullptr, 0, eArgTypeDirectoryName, "Set the
303b9c1b51eSKate Stone //  current working directory to <path> when running the inferior."},
30449bcfd80SEugene Zelenko //  {0, false, nullptr, 0, 0, nullptr, 0, eArgTypeNone, nullptr}
305ac9c3a62SKate Stone //  // clang-format on
306982c9762SGreg Clayton //};
307982c9762SGreg Clayton //
308982c9762SGreg Clayton //#undef SET1
309982c9762SGreg Clayton //#undef SET2
310982c9762SGreg Clayton //#undef SET3
31130fdc8d8SChris Lattner 
31230fdc8d8SChris Lattner //-------------------------------------------------------------------------
31330fdc8d8SChris Lattner // CommandObjectProcessAttach
31430fdc8d8SChris Lattner //-------------------------------------------------------------------------
3151f0f5b5bSZachary Turner 
3161f0f5b5bSZachary Turner static OptionDefinition g_process_attach_options[] = {
3171f0f5b5bSZachary Turner     // clang-format off
3181f0f5b5bSZachary Turner   { LLDB_OPT_SET_ALL, false, "continue",         'c', OptionParser::eNoArgument,       nullptr, nullptr, 0, eArgTypeNone,         "Immediately continue the process once attached." },
3191f0f5b5bSZachary Turner   { LLDB_OPT_SET_ALL, false, "plugin",           'P', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypePlugin,       "Name of the process plugin you want to use." },
3201f0f5b5bSZachary Turner   { LLDB_OPT_SET_1,   false, "pid",              'p', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypePid,          "The process ID of an existing process to attach to." },
3211f0f5b5bSZachary Turner   { LLDB_OPT_SET_2,   false, "name",             'n', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeProcessName,  "The name of the process to attach to." },
3221f0f5b5bSZachary Turner   { LLDB_OPT_SET_2,   false, "include-existing", 'i', OptionParser::eNoArgument,       nullptr, nullptr, 0, eArgTypeNone,         "Include existing processes when doing attach -w." },
3231f0f5b5bSZachary Turner   { LLDB_OPT_SET_2,   false, "waitfor",          'w', OptionParser::eNoArgument,       nullptr, nullptr, 0, eArgTypeNone,         "Wait for the process with <process-name> to launch." },
3241f0f5b5bSZachary Turner     // clang-format on
3251f0f5b5bSZachary Turner };
3261f0f5b5bSZachary Turner 
327bb9caf73SJim Ingham #pragma mark CommandObjectProcessAttach
328b9c1b51eSKate Stone class CommandObjectProcessAttach : public CommandObjectProcessLaunchOrAttach {
32930fdc8d8SChris Lattner public:
330b9c1b51eSKate Stone   class CommandOptions : public Options {
33130fdc8d8SChris Lattner   public:
332b9c1b51eSKate Stone     CommandOptions() : Options() {
333b9c1b51eSKate Stone       // Keep default values of all options in one place: OptionParsingStarting
334b9c1b51eSKate Stone       // ()
335e1cfbc79STodd Fiala       OptionParsingStarting(nullptr);
33630fdc8d8SChris Lattner     }
33730fdc8d8SChris Lattner 
33849bcfd80SEugene Zelenko     ~CommandOptions() override = default;
33930fdc8d8SChris Lattner 
340b9c1b51eSKate Stone     Error SetOptionValue(uint32_t option_idx, const char *option_arg,
341b9c1b51eSKate Stone                          ExecutionContext *execution_context) override {
34230fdc8d8SChris Lattner       Error error;
3433bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
34430fdc8d8SChris Lattner       bool success = false;
345b9c1b51eSKate Stone       switch (short_option) {
346a95ce623SJohnny Chen       case 'c':
347a95ce623SJohnny Chen         attach_info.SetContinueOnceAttached(true);
348a95ce623SJohnny Chen         break;
349a95ce623SJohnny Chen 
350b9c1b51eSKate Stone       case 'p': {
351b9c1b51eSKate Stone         lldb::pid_t pid = StringConvert::ToUInt32(
352b9c1b51eSKate Stone             option_arg, LLDB_INVALID_PROCESS_ID, 0, &success);
353b9c1b51eSKate Stone         if (!success || pid == LLDB_INVALID_PROCESS_ID) {
35486edbf41SGreg Clayton           error.SetErrorStringWithFormat("invalid process ID '%s'", option_arg);
355b9c1b51eSKate Stone         } else {
356144f3a9cSGreg Clayton           attach_info.SetProcessID(pid);
357144f3a9cSGreg Clayton         }
358b9c1b51eSKate Stone       } break;
35930fdc8d8SChris Lattner 
36030fdc8d8SChris Lattner       case 'P':
361144f3a9cSGreg Clayton         attach_info.SetProcessPluginName(option_arg);
36230fdc8d8SChris Lattner         break;
36330fdc8d8SChris Lattner 
36430fdc8d8SChris Lattner       case 'n':
365144f3a9cSGreg Clayton         attach_info.GetExecutableFile().SetFile(option_arg, false);
36630fdc8d8SChris Lattner         break;
36730fdc8d8SChris Lattner 
36830fdc8d8SChris Lattner       case 'w':
369144f3a9cSGreg Clayton         attach_info.SetWaitForLaunch(true);
37030fdc8d8SChris Lattner         break;
37130fdc8d8SChris Lattner 
372cd16df91SJim Ingham       case 'i':
373cd16df91SJim Ingham         attach_info.SetIgnoreExisting(false);
374cd16df91SJim Ingham         break;
375cd16df91SJim Ingham 
37630fdc8d8SChris Lattner       default:
377b9c1b51eSKate Stone         error.SetErrorStringWithFormat("invalid short option character '%c'",
378b9c1b51eSKate Stone                                        short_option);
37930fdc8d8SChris Lattner         break;
38030fdc8d8SChris Lattner       }
38130fdc8d8SChris Lattner       return error;
38230fdc8d8SChris Lattner     }
38330fdc8d8SChris Lattner 
384b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
385144f3a9cSGreg Clayton       attach_info.Clear();
38630fdc8d8SChris Lattner     }
38730fdc8d8SChris Lattner 
3881f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
38970602439SZachary Turner       return llvm::makeArrayRef(g_process_attach_options);
3901f0f5b5bSZachary Turner     }
39130fdc8d8SChris Lattner 
392b9c1b51eSKate Stone     bool HandleOptionArgumentCompletion(
393b9c1b51eSKate Stone         Args &input, int cursor_index, int char_pos,
394b9c1b51eSKate Stone         OptionElementVector &opt_element_vector, int opt_element_index,
395b9c1b51eSKate Stone         int match_start_point, int max_return_elements,
396b9c1b51eSKate Stone         CommandInterpreter &interpreter, bool &word_complete,
397b9c1b51eSKate Stone         StringList &matches) override {
3985aee162fSJim Ingham       int opt_arg_pos = opt_element_vector[opt_element_index].opt_arg_pos;
3995aee162fSJim Ingham       int opt_defs_index = opt_element_vector[opt_element_index].opt_defs_index;
4005aee162fSJim Ingham 
4015aee162fSJim Ingham       // We are only completing the name option for now...
4025aee162fSJim Ingham 
4031f0f5b5bSZachary Turner       if (GetDefinitions()[opt_defs_index].short_option == 'n') {
4045aee162fSJim Ingham         // Are we in the name?
4055aee162fSJim Ingham 
406b9c1b51eSKate Stone         // Look to see if there is a -P argument provided, and if so use that
407b9c1b51eSKate Stone         // plugin, otherwise
4085aee162fSJim Ingham         // use the default plugin.
4095aee162fSJim Ingham 
41049bcfd80SEugene Zelenko         const char *partial_name = nullptr;
4115aee162fSJim Ingham         partial_name = input.GetArgumentAtIndex(opt_arg_pos);
4125aee162fSJim Ingham 
413e1cfbc79STodd Fiala         PlatformSP platform_sp(interpreter.GetPlatform(true));
414b9c1b51eSKate Stone         if (platform_sp) {
4158b82f087SGreg Clayton           ProcessInstanceInfoList process_infos;
4168b82f087SGreg Clayton           ProcessInstanceInfoMatch match_info;
417b9c1b51eSKate Stone           if (partial_name) {
418b9c1b51eSKate Stone             match_info.GetProcessInfo().GetExecutableFile().SetFile(
419b9c1b51eSKate Stone                 partial_name, false);
42032e0a750SGreg Clayton             match_info.SetNameMatchType(eNameMatchStartsWith);
42132e0a750SGreg Clayton           }
42232e0a750SGreg Clayton           platform_sp->FindProcesses(match_info, process_infos);
423c7bece56SGreg Clayton           const size_t num_matches = process_infos.GetSize();
424b9c1b51eSKate Stone           if (num_matches > 0) {
425b9c1b51eSKate Stone             for (size_t i = 0; i < num_matches; ++i) {
426b9c1b51eSKate Stone               matches.AppendString(
427b9c1b51eSKate Stone                   process_infos.GetProcessNameAtIndex(i),
428e996fd30SGreg Clayton                   process_infos.GetProcessNameLengthAtIndex(i));
4295aee162fSJim Ingham             }
4305aee162fSJim Ingham           }
4315aee162fSJim Ingham         }
4325aee162fSJim Ingham       }
4335aee162fSJim Ingham 
4345aee162fSJim Ingham       return false;
4355aee162fSJim Ingham     }
4365aee162fSJim Ingham 
43730fdc8d8SChris Lattner     // Instance variables to hold the values for command options.
43830fdc8d8SChris Lattner 
439144f3a9cSGreg Clayton     ProcessAttachInfo attach_info;
44030fdc8d8SChris Lattner   };
44130fdc8d8SChris Lattner 
442b9c1b51eSKate Stone   CommandObjectProcessAttach(CommandInterpreter &interpreter)
443b9c1b51eSKate Stone       : CommandObjectProcessLaunchOrAttach(
444b9c1b51eSKate Stone             interpreter, "process attach", "Attach to a process.",
445b9c1b51eSKate Stone             "process attach <cmd-options>", 0, "attach"),
446b9c1b51eSKate Stone         m_options() {}
4475aee162fSJim Ingham 
44849bcfd80SEugene Zelenko   ~CommandObjectProcessAttach() override = default;
4495aee162fSJim Ingham 
450b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
4515a988416SJim Ingham 
4525a988416SJim Ingham protected:
453b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
454b9c1b51eSKate Stone     PlatformSP platform_sp(
455b9c1b51eSKate Stone         m_interpreter.GetDebugger().GetPlatformList().GetSelectedPlatform());
456926af0cdSOleksiy Vyalov 
457a7015092SGreg Clayton     Target *target = m_interpreter.GetDebugger().GetSelectedTarget().get();
458b9c1b51eSKate Stone     // N.B. The attach should be synchronous.  It doesn't help much to get the
459b9c1b51eSKate Stone     // prompt back between initiating the attach
460b9c1b51eSKate Stone     // and the target actually stopping.  So even if the interpreter is set to
461b9c1b51eSKate Stone     // be asynchronous, we wait for the stop
46231412642SJim Ingham     // ourselves here.
4635aee162fSJim Ingham 
46471337622SGreg Clayton     StateType state = eStateInvalid;
465dcb1d856SJim Ingham     Process *process = m_exe_ctx.GetProcessPtr();
466dcb1d856SJim Ingham 
467dcb1d856SJim Ingham     if (!StopProcessIfNecessary(process, state, result))
4685aee162fSJim Ingham       return false;
4695aee162fSJim Ingham 
470b9c1b51eSKate Stone     if (target == nullptr) {
4715aee162fSJim Ingham       // If there isn't a current target create one.
4725aee162fSJim Ingham       TargetSP new_target_sp;
4735aee162fSJim Ingham       Error error;
4745aee162fSJim Ingham 
475b9c1b51eSKate Stone       error = m_interpreter.GetDebugger().GetTargetList().CreateTarget(
476b9c1b51eSKate Stone           m_interpreter.GetDebugger(), nullptr, nullptr, false,
47749bcfd80SEugene Zelenko           nullptr, // No platform options
4785aee162fSJim Ingham           new_target_sp);
4795aee162fSJim Ingham       target = new_target_sp.get();
480b9c1b51eSKate Stone       if (target == nullptr || error.Fail()) {
481b766a73dSGreg Clayton         result.AppendError(error.AsCString("Error creating target"));
4825aee162fSJim Ingham         return false;
4835aee162fSJim Ingham       }
484a7015092SGreg Clayton       m_interpreter.GetDebugger().GetTargetList().SetSelectedTarget(target);
4855aee162fSJim Ingham     }
4865aee162fSJim Ingham 
487b9c1b51eSKate Stone     // Record the old executable module, we want to issue a warning if the
488b9c1b51eSKate Stone     // process of attaching changed the
489b9c1b51eSKate Stone     // current executable (like somebody said "file foo" then attached to a PID
490b9c1b51eSKate Stone     // whose executable was bar.)
4915aee162fSJim Ingham 
4925aee162fSJim Ingham     ModuleSP old_exec_module_sp = target->GetExecutableModule();
4935aee162fSJim Ingham     ArchSpec old_arch_spec = target->GetArchitecture();
4945aee162fSJim Ingham 
495b9c1b51eSKate Stone     if (command.GetArgumentCount()) {
496b9c1b51eSKate Stone       result.AppendErrorWithFormat("Invalid arguments for '%s'.\nUsage: %s\n",
497b9c1b51eSKate Stone                                    m_cmd_name.c_str(), m_cmd_syntax.c_str());
4985aee162fSJim Ingham       result.SetStatus(eReturnStatusFailed);
499926af0cdSOleksiy Vyalov       return false;
50071337622SGreg Clayton     }
5015aee162fSJim Ingham 
502926af0cdSOleksiy Vyalov     m_interpreter.UpdateExecutionContext(nullptr);
50337386143SOleksiy Vyalov     StreamString stream;
50437386143SOleksiy Vyalov     const auto error = target->Attach(m_options.attach_info, &stream);
505b9c1b51eSKate Stone     if (error.Success()) {
50637386143SOleksiy Vyalov       ProcessSP process_sp(target->GetProcessSP());
507b9c1b51eSKate Stone       if (process_sp) {
508dc6224e0SGreg Clayton         if (stream.GetData())
509dc6224e0SGreg Clayton           result.AppendMessage(stream.GetData());
510bb3a283bSJim Ingham         result.SetStatus(eReturnStatusSuccessFinishNoResult);
51137386143SOleksiy Vyalov         result.SetDidChangeProcessState(true);
512962260c8SJim Ingham         result.SetAbnormalStopWasExpected(true);
513b9c1b51eSKate Stone       } else {
514b9c1b51eSKate Stone         result.AppendError(
515b9c1b51eSKate Stone             "no error returned from Target::Attach, and target has no process");
516aa739093SJohnny Chen         result.SetStatus(eReturnStatusFailed);
51744d93782SGreg Clayton       }
518b9c1b51eSKate Stone     } else {
51944d93782SGreg Clayton       result.AppendErrorWithFormat("attach failed: %s\n", error.AsCString());
52044d93782SGreg Clayton       result.SetStatus(eReturnStatusFailed);
521aa739093SJohnny Chen     }
5225aee162fSJim Ingham 
523926af0cdSOleksiy Vyalov     if (!result.Succeeded())
524926af0cdSOleksiy Vyalov       return false;
525926af0cdSOleksiy Vyalov 
526b9c1b51eSKate Stone     // Okay, we're done.  Last step is to warn if the executable module has
527b9c1b51eSKate Stone     // changed:
528513c26ceSGreg Clayton     char new_path[PATH_MAX];
529aa149cbdSGreg Clayton     ModuleSP new_exec_module_sp(target->GetExecutableModule());
530b9c1b51eSKate Stone     if (!old_exec_module_sp) {
531513c26ceSGreg Clayton       // We might not have a module if we attached to a raw pid...
532b9c1b51eSKate Stone       if (new_exec_module_sp) {
533aa149cbdSGreg Clayton         new_exec_module_sp->GetFileSpec().GetPath(new_path, PATH_MAX);
534b9c1b51eSKate Stone         result.AppendMessageWithFormat("Executable module set to \"%s\".\n",
535b9c1b51eSKate Stone                                        new_path);
536513c26ceSGreg Clayton       }
537b9c1b51eSKate Stone     } else if (old_exec_module_sp->GetFileSpec() !=
538b9c1b51eSKate Stone                new_exec_module_sp->GetFileSpec()) {
539513c26ceSGreg Clayton       char old_path[PATH_MAX];
5405aee162fSJim Ingham 
5415aee162fSJim Ingham       old_exec_module_sp->GetFileSpec().GetPath(old_path, PATH_MAX);
542aa149cbdSGreg Clayton       new_exec_module_sp->GetFileSpec().GetPath(new_path, PATH_MAX);
5435aee162fSJim Ingham 
544b9c1b51eSKate Stone       result.AppendWarningWithFormat(
545b9c1b51eSKate Stone           "Executable module changed from \"%s\" to \"%s\".\n", old_path,
546b9c1b51eSKate Stone           new_path);
5475aee162fSJim Ingham     }
5485aee162fSJim Ingham 
549b9c1b51eSKate Stone     if (!old_arch_spec.IsValid()) {
550b9c1b51eSKate Stone       result.AppendMessageWithFormat(
551b9c1b51eSKate Stone           "Architecture set to: %s.\n",
552b9c1b51eSKate Stone           target->GetArchitecture().GetTriple().getTriple().c_str());
553b9c1b51eSKate Stone     } else if (!old_arch_spec.IsExactMatch(target->GetArchitecture())) {
554b9c1b51eSKate Stone       result.AppendWarningWithFormat(
555b9c1b51eSKate Stone           "Architecture changed from %s to %s.\n",
556c1b1f1eaSGreg Clayton           old_arch_spec.GetTriple().getTriple().c_str(),
557c1b1f1eaSGreg Clayton           target->GetArchitecture().GetTriple().getTriple().c_str());
5585aee162fSJim Ingham     }
559a95ce623SJohnny Chen 
560b9c1b51eSKate Stone     // This supports the use-case scenario of immediately continuing the process
561b9c1b51eSKate Stone     // once attached.
562a95ce623SJohnny Chen     if (m_options.attach_info.GetContinueOnceAttached())
5635bcaf583SSean Callanan       m_interpreter.HandleCommand("process continue", eLazyBoolNo, result);
564926af0cdSOleksiy Vyalov 
5655aee162fSJim Ingham     return result.Succeeded();
5665aee162fSJim Ingham   }
5675aee162fSJim Ingham 
56830fdc8d8SChris Lattner   CommandOptions m_options;
56930fdc8d8SChris Lattner };
57030fdc8d8SChris Lattner 
57130fdc8d8SChris Lattner //-------------------------------------------------------------------------
57230fdc8d8SChris Lattner // CommandObjectProcessContinue
57330fdc8d8SChris Lattner //-------------------------------------------------------------------------
5741f0f5b5bSZachary Turner 
5751f0f5b5bSZachary Turner static OptionDefinition g_process_continue_options[] = {
5761f0f5b5bSZachary Turner     // clang-format off
5771f0f5b5bSZachary Turner   { LLDB_OPT_SET_ALL, false, "ignore-count",'i', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeUnsignedInteger, "Ignore <N> crossings of the breakpoint (if it exists) for the currently selected thread." }
5781f0f5b5bSZachary Turner     // clang-format on
5791f0f5b5bSZachary Turner };
5801f0f5b5bSZachary Turner 
581bb9caf73SJim Ingham #pragma mark CommandObjectProcessContinue
58230fdc8d8SChris Lattner 
583b9c1b51eSKate Stone class CommandObjectProcessContinue : public CommandObjectParsed {
58430fdc8d8SChris Lattner public:
585b9c1b51eSKate Stone   CommandObjectProcessContinue(CommandInterpreter &interpreter)
586b9c1b51eSKate Stone       : CommandObjectParsed(
587b9c1b51eSKate Stone             interpreter, "process continue",
588e3d26315SCaroline Tice             "Continue execution of all threads in the current process.",
58930fdc8d8SChris Lattner             "process continue",
590b9c1b51eSKate Stone             eCommandRequiresProcess | eCommandTryTargetAPILock |
591b9c1b51eSKate Stone                 eCommandProcessMustBeLaunched | eCommandProcessMustBePaused),
592b9c1b51eSKate Stone         m_options() {}
59330fdc8d8SChris Lattner 
59449bcfd80SEugene Zelenko   ~CommandObjectProcessContinue() override = default;
59530fdc8d8SChris Lattner 
5965a988416SJim Ingham protected:
597b9c1b51eSKate Stone   class CommandOptions : public Options {
5980e41084aSJim Ingham   public:
599b9c1b51eSKate Stone     CommandOptions() : Options() {
600b9c1b51eSKate Stone       // Keep default values of all options in one place: OptionParsingStarting
601b9c1b51eSKate Stone       // ()
602e1cfbc79STodd Fiala       OptionParsingStarting(nullptr);
6030e41084aSJim Ingham     }
6040e41084aSJim Ingham 
60549bcfd80SEugene Zelenko     ~CommandOptions() override = default;
6060e41084aSJim Ingham 
607b9c1b51eSKate Stone     Error SetOptionValue(uint32_t option_idx, const char *option_arg,
608b9c1b51eSKate Stone                          ExecutionContext *execution_context) override {
6090e41084aSJim Ingham       Error error;
6103bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
6110e41084aSJim Ingham       bool success = false;
612b9c1b51eSKate Stone       switch (short_option) {
6130e41084aSJim Ingham       case 'i':
6145275aaa0SVince Harron         m_ignore = StringConvert::ToUInt32(option_arg, 0, 0, &success);
6150e41084aSJim Ingham         if (!success)
616b9c1b51eSKate Stone           error.SetErrorStringWithFormat(
617b9c1b51eSKate Stone               "invalid value for ignore option: \"%s\", should be a number.",
618b9c1b51eSKate Stone               option_arg);
6190e41084aSJim Ingham         break;
6200e41084aSJim Ingham 
6210e41084aSJim Ingham       default:
622b9c1b51eSKate Stone         error.SetErrorStringWithFormat("invalid short option character '%c'",
623b9c1b51eSKate Stone                                        short_option);
6240e41084aSJim Ingham         break;
6250e41084aSJim Ingham       }
6260e41084aSJim Ingham       return error;
6270e41084aSJim Ingham     }
6280e41084aSJim Ingham 
629b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
6300e41084aSJim Ingham       m_ignore = 0;
6310e41084aSJim Ingham     }
6320e41084aSJim Ingham 
6331f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
63470602439SZachary Turner       return llvm::makeArrayRef(g_process_continue_options);
6351f0f5b5bSZachary Turner     }
6360e41084aSJim Ingham 
6370e41084aSJim Ingham     uint32_t m_ignore;
6380e41084aSJim Ingham   };
6390e41084aSJim Ingham 
640b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
641f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
642a7015092SGreg Clayton     bool synchronous_execution = m_interpreter.GetSynchronous();
64330fdc8d8SChris Lattner     StateType state = process->GetState();
644b9c1b51eSKate Stone     if (state == eStateStopped) {
645b9c1b51eSKate Stone       if (command.GetArgumentCount() != 0) {
646b9c1b51eSKate Stone         result.AppendErrorWithFormat(
647b9c1b51eSKate Stone             "The '%s' command does not take any arguments.\n",
648b9c1b51eSKate Stone             m_cmd_name.c_str());
64930fdc8d8SChris Lattner         result.SetStatus(eReturnStatusFailed);
65030fdc8d8SChris Lattner         return false;
65130fdc8d8SChris Lattner       }
65230fdc8d8SChris Lattner 
653b9c1b51eSKate Stone       if (m_options.m_ignore > 0) {
6548d94ba0fSJim Ingham         ThreadSP sel_thread_sp(GetDefaultThread()->shared_from_this());
655b9c1b51eSKate Stone         if (sel_thread_sp) {
6560e41084aSJim Ingham           StopInfoSP stop_info_sp = sel_thread_sp->GetStopInfo();
657b9c1b51eSKate Stone           if (stop_info_sp &&
658b9c1b51eSKate Stone               stop_info_sp->GetStopReason() == eStopReasonBreakpoint) {
659b9c1b51eSKate Stone             lldb::break_id_t bp_site_id =
660b9c1b51eSKate Stone                 (lldb::break_id_t)stop_info_sp->GetValue();
661b9c1b51eSKate Stone             BreakpointSiteSP bp_site_sp(
662b9c1b51eSKate Stone                 process->GetBreakpointSiteList().FindByID(bp_site_id));
663b9c1b51eSKate Stone             if (bp_site_sp) {
664c7bece56SGreg Clayton               const size_t num_owners = bp_site_sp->GetNumberOfOwners();
665b9c1b51eSKate Stone               for (size_t i = 0; i < num_owners; i++) {
666b9c1b51eSKate Stone                 Breakpoint &bp_ref =
667b9c1b51eSKate Stone                     bp_site_sp->GetOwnerAtIndex(i)->GetBreakpoint();
668b9c1b51eSKate Stone                 if (!bp_ref.IsInternal()) {
6690e41084aSJim Ingham                   bp_ref.SetIgnoreCount(m_options.m_ignore);
6700e41084aSJim Ingham                 }
6710e41084aSJim Ingham               }
6720e41084aSJim Ingham             }
6730e41084aSJim Ingham           }
6740e41084aSJim Ingham         }
6750e41084aSJim Ingham       }
6760e41084aSJim Ingham 
67741f2b940SJim Ingham       { // Scope for thread list mutex:
678b9c1b51eSKate Stone         std::lock_guard<std::recursive_mutex> guard(
679b9c1b51eSKate Stone             process->GetThreadList().GetMutex());
68030fdc8d8SChris Lattner         const uint32_t num_threads = process->GetThreadList().GetSize();
68130fdc8d8SChris Lattner 
68230fdc8d8SChris Lattner         // Set the actions that the threads should each take when resuming
683b9c1b51eSKate Stone         for (uint32_t idx = 0; idx < num_threads; ++idx) {
6846c9ed91cSJim Ingham           const bool override_suspend = false;
685b9c1b51eSKate Stone           process->GetThreadList().GetThreadAtIndex(idx)->SetResumeState(
686b9c1b51eSKate Stone               eStateRunning, override_suspend);
68730fdc8d8SChris Lattner         }
68841f2b940SJim Ingham       }
68930fdc8d8SChris Lattner 
6904446487dSPavel Labath       const uint32_t iohandler_id = process->GetIOHandlerID();
6914446487dSPavel Labath 
692dc6224e0SGreg Clayton       StreamString stream;
693dc6224e0SGreg Clayton       Error error;
694dc6224e0SGreg Clayton       if (synchronous_execution)
695dc6224e0SGreg Clayton         error = process->ResumeSynchronous(&stream);
696dc6224e0SGreg Clayton       else
697dc6224e0SGreg Clayton         error = process->Resume();
698a3b89e27STodd Fiala 
699b9c1b51eSKate Stone       if (error.Success()) {
700b9c1b51eSKate Stone         // There is a race condition where this thread will return up the call
701b9c1b51eSKate Stone         // stack to the main command
702b9c1b51eSKate Stone         // handler and show an (lldb) prompt before HandlePrivateEvent (from
703b9c1b51eSKate Stone         // PrivateStateThread) has
704a3b89e27STodd Fiala         // a chance to call PushProcessIOHandler().
7054446487dSPavel Labath         process->SyncIOHandler(iohandler_id, 2000);
706a3b89e27STodd Fiala 
707b9c1b51eSKate Stone         result.AppendMessageWithFormat("Process %" PRIu64 " resuming\n",
708b9c1b51eSKate Stone                                        process->GetID());
709b9c1b51eSKate Stone         if (synchronous_execution) {
710b9c1b51eSKate Stone           // If any state changed events had anything to say, add that to the
711b9c1b51eSKate Stone           // result
712dc6224e0SGreg Clayton           if (stream.GetData())
713dc6224e0SGreg Clayton             result.AppendMessage(stream.GetData());
71430fdc8d8SChris Lattner 
71530fdc8d8SChris Lattner           result.SetDidChangeProcessState(true);
71630fdc8d8SChris Lattner           result.SetStatus(eReturnStatusSuccessFinishNoResult);
717b9c1b51eSKate Stone         } else {
71830fdc8d8SChris Lattner           result.SetStatus(eReturnStatusSuccessContinuingNoResult);
71930fdc8d8SChris Lattner         }
720b9c1b51eSKate Stone       } else {
721b9c1b51eSKate Stone         result.AppendErrorWithFormat("Failed to resume process: %s.\n",
722b9c1b51eSKate Stone                                      error.AsCString());
72330fdc8d8SChris Lattner         result.SetStatus(eReturnStatusFailed);
72430fdc8d8SChris Lattner       }
725b9c1b51eSKate Stone     } else {
726b9c1b51eSKate Stone       result.AppendErrorWithFormat(
727b9c1b51eSKate Stone           "Process cannot be continued from its current state (%s).\n",
72830fdc8d8SChris Lattner           StateAsCString(state));
72930fdc8d8SChris Lattner       result.SetStatus(eReturnStatusFailed);
73030fdc8d8SChris Lattner     }
73130fdc8d8SChris Lattner     return result.Succeeded();
73230fdc8d8SChris Lattner   }
7330e41084aSJim Ingham 
734b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
7350e41084aSJim Ingham 
7360e41084aSJim Ingham   CommandOptions m_options;
7370e41084aSJim Ingham };
7380e41084aSJim Ingham 
73930fdc8d8SChris Lattner //-------------------------------------------------------------------------
74030fdc8d8SChris Lattner // CommandObjectProcessDetach
74130fdc8d8SChris Lattner //-------------------------------------------------------------------------
7421f0f5b5bSZachary Turner static OptionDefinition g_process_detach_options[] = {
7431f0f5b5bSZachary Turner     // clang-format off
7441f0f5b5bSZachary Turner   { LLDB_OPT_SET_1, false, "keep-stopped", 's', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeBoolean, "Whether or not the process should be kept stopped on detach (if possible)." },
7451f0f5b5bSZachary Turner     // clang-format on
7461f0f5b5bSZachary Turner };
7471f0f5b5bSZachary Turner 
748bb9caf73SJim Ingham #pragma mark CommandObjectProcessDetach
74930fdc8d8SChris Lattner 
750b9c1b51eSKate Stone class CommandObjectProcessDetach : public CommandObjectParsed {
75130fdc8d8SChris Lattner public:
752b9c1b51eSKate Stone   class CommandOptions : public Options {
753acff8950SJim Ingham   public:
754b9c1b51eSKate Stone     CommandOptions() : Options() { OptionParsingStarting(nullptr); }
755acff8950SJim Ingham 
75649bcfd80SEugene Zelenko     ~CommandOptions() override = default;
757acff8950SJim Ingham 
758b9c1b51eSKate Stone     Error SetOptionValue(uint32_t option_idx, const char *option_arg,
759b9c1b51eSKate Stone                          ExecutionContext *execution_context) override {
760acff8950SJim Ingham       Error error;
761acff8950SJim Ingham       const int short_option = m_getopt_table[option_idx].val;
762ecbb0bb1SZachary Turner       auto option_strref = llvm::StringRef::withNullAsEmpty(option_arg);
763acff8950SJim Ingham 
764b9c1b51eSKate Stone       switch (short_option) {
765acff8950SJim Ingham       case 's':
766acff8950SJim Ingham         bool tmp_result;
767acff8950SJim Ingham         bool success;
768ecbb0bb1SZachary Turner         tmp_result = Args::StringToBoolean(option_strref, false, &success);
769acff8950SJim Ingham         if (!success)
770b9c1b51eSKate Stone           error.SetErrorStringWithFormat("invalid boolean option: \"%s\"",
771b9c1b51eSKate Stone                                          option_arg);
772b9c1b51eSKate Stone         else {
773acff8950SJim Ingham           if (tmp_result)
774acff8950SJim Ingham             m_keep_stopped = eLazyBoolYes;
775acff8950SJim Ingham           else
776acff8950SJim Ingham             m_keep_stopped = eLazyBoolNo;
777acff8950SJim Ingham         }
778acff8950SJim Ingham         break;
779acff8950SJim Ingham       default:
780b9c1b51eSKate Stone         error.SetErrorStringWithFormat("invalid short option character '%c'",
781b9c1b51eSKate Stone                                        short_option);
782acff8950SJim Ingham         break;
783acff8950SJim Ingham       }
784acff8950SJim Ingham       return error;
785acff8950SJim Ingham     }
786acff8950SJim Ingham 
787b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
788acff8950SJim Ingham       m_keep_stopped = eLazyBoolCalculate;
789acff8950SJim Ingham     }
790acff8950SJim Ingham 
7911f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
79270602439SZachary Turner       return llvm::makeArrayRef(g_process_detach_options);
7931f0f5b5bSZachary Turner     }
794acff8950SJim Ingham 
795acff8950SJim Ingham     // Instance variables to hold the values for command options.
796acff8950SJim Ingham     LazyBool m_keep_stopped;
797acff8950SJim Ingham   };
79830fdc8d8SChris Lattner 
7997428a18cSKate Stone   CommandObjectProcessDetach(CommandInterpreter &interpreter)
800b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process detach",
801b9c1b51eSKate Stone                             "Detach from the current target process.",
802a7015092SGreg Clayton                             "process detach",
803b9c1b51eSKate Stone                             eCommandRequiresProcess | eCommandTryTargetAPILock |
804b9c1b51eSKate Stone                                 eCommandProcessMustBeLaunched),
805b9c1b51eSKate Stone         m_options() {}
80630fdc8d8SChris Lattner 
80749bcfd80SEugene Zelenko   ~CommandObjectProcessDetach() override = default;
80830fdc8d8SChris Lattner 
809b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
810acff8950SJim Ingham 
8115a988416SJim Ingham protected:
812b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
813f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
814acff8950SJim Ingham     // FIXME: This will be a Command Option:
815acff8950SJim Ingham     bool keep_stopped;
816b9c1b51eSKate Stone     if (m_options.m_keep_stopped == eLazyBoolCalculate) {
817acff8950SJim Ingham       // Check the process default:
81849bcfd80SEugene Zelenko       keep_stopped = process->GetDetachKeepsStopped();
819b9c1b51eSKate Stone     } else if (m_options.m_keep_stopped == eLazyBoolYes)
820acff8950SJim Ingham       keep_stopped = true;
821acff8950SJim Ingham     else
822acff8950SJim Ingham       keep_stopped = false;
823acff8950SJim Ingham 
824acff8950SJim Ingham     Error error(process->Detach(keep_stopped));
825b9c1b51eSKate Stone     if (error.Success()) {
82630fdc8d8SChris Lattner       result.SetStatus(eReturnStatusSuccessFinishResult);
827b9c1b51eSKate Stone     } else {
82830fdc8d8SChris Lattner       result.AppendErrorWithFormat("Detach failed: %s\n", error.AsCString());
82930fdc8d8SChris Lattner       result.SetStatus(eReturnStatusFailed);
83030fdc8d8SChris Lattner       return false;
83130fdc8d8SChris Lattner     }
83230fdc8d8SChris Lattner     return result.Succeeded();
83330fdc8d8SChris Lattner   }
834acff8950SJim Ingham 
835acff8950SJim Ingham   CommandOptions m_options;
836acff8950SJim Ingham };
837acff8950SJim Ingham 
83830fdc8d8SChris Lattner //-------------------------------------------------------------------------
839b766a73dSGreg Clayton // CommandObjectProcessConnect
840b766a73dSGreg Clayton //-------------------------------------------------------------------------
8411f0f5b5bSZachary Turner 
8421f0f5b5bSZachary Turner static OptionDefinition g_process_connect_options[] = {
8431f0f5b5bSZachary Turner     // clang-format off
8441f0f5b5bSZachary Turner   { LLDB_OPT_SET_ALL, false, "plugin", 'p', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypePlugin, "Name of the process plugin you want to use." },
8451f0f5b5bSZachary Turner     // clang-format on
8461f0f5b5bSZachary Turner };
8471f0f5b5bSZachary Turner 
848b766a73dSGreg Clayton #pragma mark CommandObjectProcessConnect
849b766a73dSGreg Clayton 
850b9c1b51eSKate Stone class CommandObjectProcessConnect : public CommandObjectParsed {
851b766a73dSGreg Clayton public:
852b9c1b51eSKate Stone   class CommandOptions : public Options {
853b766a73dSGreg Clayton   public:
854b9c1b51eSKate Stone     CommandOptions() : Options() {
855b9c1b51eSKate Stone       // Keep default values of all options in one place: OptionParsingStarting
856b9c1b51eSKate Stone       // ()
857e1cfbc79STodd Fiala       OptionParsingStarting(nullptr);
858b766a73dSGreg Clayton     }
859b766a73dSGreg Clayton 
86049bcfd80SEugene Zelenko     ~CommandOptions() override = default;
861b766a73dSGreg Clayton 
862b9c1b51eSKate Stone     Error SetOptionValue(uint32_t option_idx, const char *option_arg,
863b9c1b51eSKate Stone                          ExecutionContext *execution_context) override {
864b766a73dSGreg Clayton       Error error;
8653bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
866b766a73dSGreg Clayton 
867b9c1b51eSKate Stone       switch (short_option) {
868b766a73dSGreg Clayton       case 'p':
869b766a73dSGreg Clayton         plugin_name.assign(option_arg);
870b766a73dSGreg Clayton         break;
871b766a73dSGreg Clayton 
872b766a73dSGreg Clayton       default:
873b9c1b51eSKate Stone         error.SetErrorStringWithFormat("invalid short option character '%c'",
874b9c1b51eSKate Stone                                        short_option);
875b766a73dSGreg Clayton         break;
876b766a73dSGreg Clayton       }
877b766a73dSGreg Clayton       return error;
878b766a73dSGreg Clayton     }
879b766a73dSGreg Clayton 
880b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
881b766a73dSGreg Clayton       plugin_name.clear();
882b766a73dSGreg Clayton     }
883b766a73dSGreg Clayton 
8841f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
88570602439SZachary Turner       return llvm::makeArrayRef(g_process_connect_options);
8861f0f5b5bSZachary Turner     }
887b766a73dSGreg Clayton 
888b766a73dSGreg Clayton     // Instance variables to hold the values for command options.
889b766a73dSGreg Clayton 
890b766a73dSGreg Clayton     std::string plugin_name;
891b766a73dSGreg Clayton   };
892b766a73dSGreg Clayton 
893b9c1b51eSKate Stone   CommandObjectProcessConnect(CommandInterpreter &interpreter)
894b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process connect",
895b766a73dSGreg Clayton                             "Connect to a remote debug service.",
896b9c1b51eSKate Stone                             "process connect <remote-url>", 0),
897b9c1b51eSKate Stone         m_options() {}
898b766a73dSGreg Clayton 
89949bcfd80SEugene Zelenko   ~CommandObjectProcessConnect() override = default;
900b766a73dSGreg Clayton 
901b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
9025a988416SJim Ingham 
9035a988416SJim Ingham protected:
904b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
905b9c1b51eSKate Stone     if (command.GetArgumentCount() != 1) {
906b9c1b51eSKate Stone       result.AppendErrorWithFormat(
907b9c1b51eSKate Stone           "'%s' takes exactly one argument:\nUsage: %s\n", m_cmd_name.c_str(),
908ccd6cffbSTamas Berghammer           m_cmd_syntax.c_str());
909ccd6cffbSTamas Berghammer       result.SetStatus(eReturnStatusFailed);
910ccd6cffbSTamas Berghammer       return false;
911ccd6cffbSTamas Berghammer     }
912ccd6cffbSTamas Berghammer 
913ccd6cffbSTamas Berghammer     Process *process = m_exe_ctx.GetProcessPtr();
914b9c1b51eSKate Stone     if (process && process->IsAlive()) {
915b9c1b51eSKate Stone       result.AppendErrorWithFormat(
916b9c1b51eSKate Stone           "Process %" PRIu64
917b9c1b51eSKate Stone           " is currently being debugged, kill the process before connecting.\n",
918b766a73dSGreg Clayton           process->GetID());
919b766a73dSGreg Clayton       result.SetStatus(eReturnStatusFailed);
920b766a73dSGreg Clayton       return false;
921b766a73dSGreg Clayton     }
922b766a73dSGreg Clayton 
923ccd6cffbSTamas Berghammer     const char *plugin_name = nullptr;
924b766a73dSGreg Clayton     if (!m_options.plugin_name.empty())
925b766a73dSGreg Clayton       plugin_name = m_options.plugin_name.c_str();
926b766a73dSGreg Clayton 
927ccd6cffbSTamas Berghammer     Error error;
928ccd6cffbSTamas Berghammer     Debugger &debugger = m_interpreter.GetDebugger();
929ccd6cffbSTamas Berghammer     PlatformSP platform_sp = m_interpreter.GetPlatform(true);
930b9c1b51eSKate Stone     ProcessSP process_sp = platform_sp->ConnectProcess(
931b9c1b51eSKate Stone         command.GetArgumentAtIndex(0), plugin_name, debugger,
932b9c1b51eSKate Stone         debugger.GetSelectedTarget().get(), error);
933b9c1b51eSKate Stone     if (error.Fail() || process_sp == nullptr) {
934ccd6cffbSTamas Berghammer       result.AppendError(error.AsCString("Error connecting to the process"));
935b766a73dSGreg Clayton       result.SetStatus(eReturnStatusFailed);
936b766a73dSGreg Clayton       return false;
937b766a73dSGreg Clayton     }
938ccd6cffbSTamas Berghammer     return true;
939b766a73dSGreg Clayton   }
940b766a73dSGreg Clayton 
941b766a73dSGreg Clayton   CommandOptions m_options;
942b766a73dSGreg Clayton };
943b766a73dSGreg Clayton 
944b766a73dSGreg Clayton //-------------------------------------------------------------------------
945998255bfSGreg Clayton // CommandObjectProcessPlugin
946998255bfSGreg Clayton //-------------------------------------------------------------------------
947998255bfSGreg Clayton #pragma mark CommandObjectProcessPlugin
948998255bfSGreg Clayton 
949b9c1b51eSKate Stone class CommandObjectProcessPlugin : public CommandObjectProxy {
950998255bfSGreg Clayton public:
9517428a18cSKate Stone   CommandObjectProcessPlugin(CommandInterpreter &interpreter)
952b9c1b51eSKate Stone       : CommandObjectProxy(
953b9c1b51eSKate Stone             interpreter, "process plugin",
954b9c1b51eSKate Stone             "Send a custom command to the current target process plug-in.",
955b9c1b51eSKate Stone             "process plugin <args>", 0) {}
956998255bfSGreg Clayton 
95749bcfd80SEugene Zelenko   ~CommandObjectProcessPlugin() override = default;
958998255bfSGreg Clayton 
959b9c1b51eSKate Stone   CommandObject *GetProxyCommandObject() override {
960e05b2efeSGreg Clayton     Process *process = m_interpreter.GetExecutionContext().GetProcessPtr();
961998255bfSGreg Clayton     if (process)
962998255bfSGreg Clayton       return process->GetPluginCommandObject();
96349bcfd80SEugene Zelenko     return nullptr;
964998255bfSGreg Clayton   }
965998255bfSGreg Clayton };
966998255bfSGreg Clayton 
967998255bfSGreg Clayton //-------------------------------------------------------------------------
9688f343b09SGreg Clayton // CommandObjectProcessLoad
9698f343b09SGreg Clayton //-------------------------------------------------------------------------
9701f0f5b5bSZachary Turner 
9711f0f5b5bSZachary Turner static OptionDefinition g_process_load_options[] = {
9721f0f5b5bSZachary Turner     // clang-format off
9731f0f5b5bSZachary Turner   { LLDB_OPT_SET_ALL, false, "install", 'i', OptionParser::eOptionalArgument, nullptr, nullptr, 0, eArgTypePath, "Install the shared library to the target. If specified without an argument then the library will installed in the current working directory." },
9741f0f5b5bSZachary Turner     // clang-format on
9751f0f5b5bSZachary Turner };
9761f0f5b5bSZachary Turner 
977bb9caf73SJim Ingham #pragma mark CommandObjectProcessLoad
9788f343b09SGreg Clayton 
979b9c1b51eSKate Stone class CommandObjectProcessLoad : public CommandObjectParsed {
9808f343b09SGreg Clayton public:
981b9c1b51eSKate Stone   class CommandOptions : public Options {
9824fbd67acSTamas Berghammer   public:
983b9c1b51eSKate Stone     CommandOptions() : Options() {
984b9c1b51eSKate Stone       // Keep default values of all options in one place: OptionParsingStarting
985b9c1b51eSKate Stone       // ()
986e1cfbc79STodd Fiala       OptionParsingStarting(nullptr);
9874fbd67acSTamas Berghammer     }
9884fbd67acSTamas Berghammer 
9894fbd67acSTamas Berghammer     ~CommandOptions() override = default;
9904fbd67acSTamas Berghammer 
991b9c1b51eSKate Stone     Error SetOptionValue(uint32_t option_idx, const char *option_arg,
992b9c1b51eSKate Stone                          ExecutionContext *execution_context) override {
9934fbd67acSTamas Berghammer       Error error;
9944fbd67acSTamas Berghammer       const int short_option = m_getopt_table[option_idx].val;
995b9c1b51eSKate Stone       switch (short_option) {
9964fbd67acSTamas Berghammer       case 'i':
9974fbd67acSTamas Berghammer         do_install = true;
9984fbd67acSTamas Berghammer         if (option_arg && option_arg[0])
9994fbd67acSTamas Berghammer           install_path.SetFile(option_arg, false);
10004fbd67acSTamas Berghammer         break;
10014fbd67acSTamas Berghammer       default:
1002b9c1b51eSKate Stone         error.SetErrorStringWithFormat("invalid short option character '%c'",
1003b9c1b51eSKate Stone                                        short_option);
10044fbd67acSTamas Berghammer         break;
10054fbd67acSTamas Berghammer       }
10064fbd67acSTamas Berghammer       return error;
10074fbd67acSTamas Berghammer     }
10084fbd67acSTamas Berghammer 
1009b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
10104fbd67acSTamas Berghammer       do_install = false;
10114fbd67acSTamas Berghammer       install_path.Clear();
10124fbd67acSTamas Berghammer     }
10134fbd67acSTamas Berghammer 
10141f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
101570602439SZachary Turner       return llvm::makeArrayRef(g_process_load_options);
10161f0f5b5bSZachary Turner     }
10174fbd67acSTamas Berghammer 
10184fbd67acSTamas Berghammer     // Instance variables to hold the values for command options.
10194fbd67acSTamas Berghammer     bool do_install;
10204fbd67acSTamas Berghammer     FileSpec install_path;
10214fbd67acSTamas Berghammer   };
10228f343b09SGreg Clayton 
1023b9c1b51eSKate Stone   CommandObjectProcessLoad(CommandInterpreter &interpreter)
1024b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process load",
10258f343b09SGreg Clayton                             "Load a shared library into the current process.",
10268f343b09SGreg Clayton                             "process load <filename> [<filename> ...]",
1027b9c1b51eSKate Stone                             eCommandRequiresProcess | eCommandTryTargetAPILock |
1028e87764f2SEnrico Granata                                 eCommandProcessMustBeLaunched |
10294fbd67acSTamas Berghammer                                 eCommandProcessMustBePaused),
1030b9c1b51eSKate Stone         m_options() {}
10318f343b09SGreg Clayton 
10324fbd67acSTamas Berghammer   ~CommandObjectProcessLoad() override = default;
10334fbd67acSTamas Berghammer 
1034b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
10358f343b09SGreg Clayton 
10365a988416SJim Ingham protected:
1037b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1038f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
10398f343b09SGreg Clayton 
1040*97d2c401SZachary Turner     for (auto &entry : command.entries()) {
10418f343b09SGreg Clayton       Error error;
10423cb132a0STamas Berghammer       PlatformSP platform = process->GetTarget().GetPlatform();
1043*97d2c401SZachary Turner       llvm::StringRef image_path = entry.ref;
10444fbd67acSTamas Berghammer       uint32_t image_token = LLDB_INVALID_IMAGE_TOKEN;
10454fbd67acSTamas Berghammer 
1046b9c1b51eSKate Stone       if (!m_options.do_install) {
10474fbd67acSTamas Berghammer         FileSpec image_spec(image_path, false);
10483cb132a0STamas Berghammer         platform->ResolveRemotePath(image_spec, image_spec);
1049b9c1b51eSKate Stone         image_token =
1050b9c1b51eSKate Stone             platform->LoadImage(process, FileSpec(), image_spec, error);
1051b9c1b51eSKate Stone       } else if (m_options.install_path) {
10524fbd67acSTamas Berghammer         FileSpec image_spec(image_path, true);
1053b9c1b51eSKate Stone         platform->ResolveRemotePath(m_options.install_path,
1054b9c1b51eSKate Stone                                     m_options.install_path);
1055b9c1b51eSKate Stone         image_token = platform->LoadImage(process, image_spec,
1056b9c1b51eSKate Stone                                           m_options.install_path, error);
1057b9c1b51eSKate Stone       } else {
10584fbd67acSTamas Berghammer         FileSpec image_spec(image_path, true);
1059b9c1b51eSKate Stone         image_token =
1060b9c1b51eSKate Stone             platform->LoadImage(process, image_spec, FileSpec(), error);
10614fbd67acSTamas Berghammer       }
10624fbd67acSTamas Berghammer 
1063b9c1b51eSKate Stone       if (image_token != LLDB_INVALID_IMAGE_TOKEN) {
1064b9c1b51eSKate Stone         result.AppendMessageWithFormat(
1065*97d2c401SZachary Turner             "Loading \"%s\"...ok\nImage %u loaded.\n", image_path.str().c_str(),
1066*97d2c401SZachary Turner             image_token);
10678f343b09SGreg Clayton         result.SetStatus(eReturnStatusSuccessFinishResult);
1068b9c1b51eSKate Stone       } else {
1069*97d2c401SZachary Turner         result.AppendErrorWithFormat("failed to load '%s': %s",
1070*97d2c401SZachary Turner                                      image_path.str().c_str(),
1071b9c1b51eSKate Stone                                      error.AsCString());
10728f343b09SGreg Clayton         result.SetStatus(eReturnStatusFailed);
10738f343b09SGreg Clayton       }
10748f343b09SGreg Clayton     }
10758f343b09SGreg Clayton     return result.Succeeded();
10768f343b09SGreg Clayton   }
10774fbd67acSTamas Berghammer 
10784fbd67acSTamas Berghammer   CommandOptions m_options;
10798f343b09SGreg Clayton };
10808f343b09SGreg Clayton 
10818f343b09SGreg Clayton //-------------------------------------------------------------------------
10828f343b09SGreg Clayton // CommandObjectProcessUnload
10838f343b09SGreg Clayton //-------------------------------------------------------------------------
1084bb9caf73SJim Ingham #pragma mark CommandObjectProcessUnload
10858f343b09SGreg Clayton 
1086b9c1b51eSKate Stone class CommandObjectProcessUnload : public CommandObjectParsed {
10878f343b09SGreg Clayton public:
1088b9c1b51eSKate Stone   CommandObjectProcessUnload(CommandInterpreter &interpreter)
1089b9c1b51eSKate Stone       : CommandObjectParsed(
1090b9c1b51eSKate Stone             interpreter, "process unload",
1091b9c1b51eSKate Stone             "Unload a shared library from the current process using the index "
1092b9c1b51eSKate Stone             "returned by a previous call to \"process load\".",
10938f343b09SGreg Clayton             "process unload <index>",
1094b9c1b51eSKate Stone             eCommandRequiresProcess | eCommandTryTargetAPILock |
1095b9c1b51eSKate Stone                 eCommandProcessMustBeLaunched | eCommandProcessMustBePaused) {}
10968f343b09SGreg Clayton 
109749bcfd80SEugene Zelenko   ~CommandObjectProcessUnload() override = default;
10988f343b09SGreg Clayton 
10995a988416SJim Ingham protected:
1100b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1101f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
11028f343b09SGreg Clayton 
1103*97d2c401SZachary Turner     for (auto &entry : command.entries()) {
1104*97d2c401SZachary Turner       uint32_t image_token;
1105*97d2c401SZachary Turner       if (entry.ref.getAsInteger(0, image_token)) {
1106b9c1b51eSKate Stone         result.AppendErrorWithFormat("invalid image index argument '%s'",
1107*97d2c401SZachary Turner                                      entry.ref.str().c_str());
11088f343b09SGreg Clayton         result.SetStatus(eReturnStatusFailed);
11098f343b09SGreg Clayton         break;
1110b9c1b51eSKate Stone       } else {
1111b9c1b51eSKate Stone         Error error(process->GetTarget().GetPlatform()->UnloadImage(
1112b9c1b51eSKate Stone             process, image_token));
1113b9c1b51eSKate Stone         if (error.Success()) {
1114b9c1b51eSKate Stone           result.AppendMessageWithFormat(
1115b9c1b51eSKate Stone               "Unloading shared library with index %u...ok\n", image_token);
11168f343b09SGreg Clayton           result.SetStatus(eReturnStatusSuccessFinishResult);
1117b9c1b51eSKate Stone         } else {
1118b9c1b51eSKate Stone           result.AppendErrorWithFormat("failed to unload image: %s",
1119b9c1b51eSKate Stone                                        error.AsCString());
11208f343b09SGreg Clayton           result.SetStatus(eReturnStatusFailed);
11218f343b09SGreg Clayton           break;
11228f343b09SGreg Clayton         }
11238f343b09SGreg Clayton       }
11248f343b09SGreg Clayton     }
11258f343b09SGreg Clayton     return result.Succeeded();
11268f343b09SGreg Clayton   }
11278f343b09SGreg Clayton };
11288f343b09SGreg Clayton 
11298f343b09SGreg Clayton //-------------------------------------------------------------------------
113030fdc8d8SChris Lattner // CommandObjectProcessSignal
113130fdc8d8SChris Lattner //-------------------------------------------------------------------------
1132bb9caf73SJim Ingham #pragma mark CommandObjectProcessSignal
113330fdc8d8SChris Lattner 
1134b9c1b51eSKate Stone class CommandObjectProcessSignal : public CommandObjectParsed {
113530fdc8d8SChris Lattner public:
11367428a18cSKate Stone   CommandObjectProcessSignal(CommandInterpreter &interpreter)
1137b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process signal",
1138b9c1b51eSKate Stone                             "Send a UNIX signal to the current target process.",
1139b9c1b51eSKate Stone                             nullptr, eCommandRequiresProcess |
1140b9c1b51eSKate Stone                                          eCommandTryTargetAPILock) {
1141405fe67fSCaroline Tice     CommandArgumentEntry arg;
1142405fe67fSCaroline Tice     CommandArgumentData signal_arg;
1143405fe67fSCaroline Tice 
1144405fe67fSCaroline Tice     // Define the first (and only) variant of this arg.
1145c0dbdfb6SCaroline Tice     signal_arg.arg_type = eArgTypeUnixSignal;
1146405fe67fSCaroline Tice     signal_arg.arg_repetition = eArgRepeatPlain;
1147405fe67fSCaroline Tice 
1148b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
1149b9c1b51eSKate Stone     // argument entry.
1150405fe67fSCaroline Tice     arg.push_back(signal_arg);
1151405fe67fSCaroline Tice 
1152405fe67fSCaroline Tice     // Push the data for the first argument into the m_arguments vector.
1153405fe67fSCaroline Tice     m_arguments.push_back(arg);
115430fdc8d8SChris Lattner   }
115530fdc8d8SChris Lattner 
115649bcfd80SEugene Zelenko   ~CommandObjectProcessSignal() override = default;
115730fdc8d8SChris Lattner 
11585a988416SJim Ingham protected:
1159b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1160f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
116130fdc8d8SChris Lattner 
1162b9c1b51eSKate Stone     if (command.GetArgumentCount() == 1) {
1163237cd906SGreg Clayton       int signo = LLDB_INVALID_SIGNAL_NUMBER;
1164237cd906SGreg Clayton 
1165237cd906SGreg Clayton       const char *signal_name = command.GetArgumentAtIndex(0);
1166237cd906SGreg Clayton       if (::isxdigit(signal_name[0]))
1167b9c1b51eSKate Stone         signo =
1168b9c1b51eSKate Stone             StringConvert::ToSInt32(signal_name, LLDB_INVALID_SIGNAL_NUMBER, 0);
1169237cd906SGreg Clayton       else
117098d0a4b3SChaoren Lin         signo = process->GetUnixSignals()->GetSignalNumberFromName(signal_name);
1171237cd906SGreg Clayton 
1172b9c1b51eSKate Stone       if (signo == LLDB_INVALID_SIGNAL_NUMBER) {
1173b9c1b51eSKate Stone         result.AppendErrorWithFormat("Invalid signal argument '%s'.\n",
1174b9c1b51eSKate Stone                                      command.GetArgumentAtIndex(0));
117530fdc8d8SChris Lattner         result.SetStatus(eReturnStatusFailed);
1176b9c1b51eSKate Stone       } else {
117730fdc8d8SChris Lattner         Error error(process->Signal(signo));
1178b9c1b51eSKate Stone         if (error.Success()) {
117930fdc8d8SChris Lattner           result.SetStatus(eReturnStatusSuccessFinishResult);
1180b9c1b51eSKate Stone         } else {
1181b9c1b51eSKate Stone           result.AppendErrorWithFormat("Failed to send signal %i: %s\n", signo,
1182b9c1b51eSKate Stone                                        error.AsCString());
118330fdc8d8SChris Lattner           result.SetStatus(eReturnStatusFailed);
118430fdc8d8SChris Lattner         }
118530fdc8d8SChris Lattner       }
1186b9c1b51eSKate Stone     } else {
1187b9c1b51eSKate Stone       result.AppendErrorWithFormat(
1188b9c1b51eSKate Stone           "'%s' takes exactly one signal number argument:\nUsage: %s\n",
1189b9c1b51eSKate Stone           m_cmd_name.c_str(), m_cmd_syntax.c_str());
119030fdc8d8SChris Lattner       result.SetStatus(eReturnStatusFailed);
119130fdc8d8SChris Lattner     }
119230fdc8d8SChris Lattner     return result.Succeeded();
119330fdc8d8SChris Lattner   }
119430fdc8d8SChris Lattner };
119530fdc8d8SChris Lattner 
119630fdc8d8SChris Lattner //-------------------------------------------------------------------------
119730fdc8d8SChris Lattner // CommandObjectProcessInterrupt
119830fdc8d8SChris Lattner //-------------------------------------------------------------------------
1199bb9caf73SJim Ingham #pragma mark CommandObjectProcessInterrupt
120030fdc8d8SChris Lattner 
1201b9c1b51eSKate Stone class CommandObjectProcessInterrupt : public CommandObjectParsed {
120230fdc8d8SChris Lattner public:
12037428a18cSKate Stone   CommandObjectProcessInterrupt(CommandInterpreter &interpreter)
1204b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process interrupt",
1205b9c1b51eSKate Stone                             "Interrupt the current target process.",
1206a7015092SGreg Clayton                             "process interrupt",
1207b9c1b51eSKate Stone                             eCommandRequiresProcess | eCommandTryTargetAPILock |
1208b9c1b51eSKate Stone                                 eCommandProcessMustBeLaunched) {}
120930fdc8d8SChris Lattner 
121049bcfd80SEugene Zelenko   ~CommandObjectProcessInterrupt() override = default;
121130fdc8d8SChris Lattner 
12125a988416SJim Ingham protected:
1213b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1214f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
1215b9c1b51eSKate Stone     if (process == nullptr) {
121630fdc8d8SChris Lattner       result.AppendError("no process to halt");
121730fdc8d8SChris Lattner       result.SetStatus(eReturnStatusFailed);
121830fdc8d8SChris Lattner       return false;
121930fdc8d8SChris Lattner     }
122030fdc8d8SChris Lattner 
1221b9c1b51eSKate Stone     if (command.GetArgumentCount() == 0) {
1222f9b57b9dSGreg Clayton       bool clear_thread_plans = true;
1223f9b57b9dSGreg Clayton       Error error(process->Halt(clear_thread_plans));
1224b9c1b51eSKate Stone       if (error.Success()) {
122530fdc8d8SChris Lattner         result.SetStatus(eReturnStatusSuccessFinishResult);
1226b9c1b51eSKate Stone       } else {
1227b9c1b51eSKate Stone         result.AppendErrorWithFormat("Failed to halt process: %s\n",
1228b9c1b51eSKate Stone                                      error.AsCString());
122930fdc8d8SChris Lattner         result.SetStatus(eReturnStatusFailed);
123030fdc8d8SChris Lattner       }
1231b9c1b51eSKate Stone     } else {
1232fd54b368SJason Molenda       result.AppendErrorWithFormat("'%s' takes no arguments:\nUsage: %s\n",
1233b9c1b51eSKate Stone                                    m_cmd_name.c_str(), m_cmd_syntax.c_str());
123430fdc8d8SChris Lattner       result.SetStatus(eReturnStatusFailed);
123530fdc8d8SChris Lattner     }
123630fdc8d8SChris Lattner     return result.Succeeded();
123730fdc8d8SChris Lattner   }
123830fdc8d8SChris Lattner };
123930fdc8d8SChris Lattner 
124030fdc8d8SChris Lattner //-------------------------------------------------------------------------
124130fdc8d8SChris Lattner // CommandObjectProcessKill
124230fdc8d8SChris Lattner //-------------------------------------------------------------------------
1243bb9caf73SJim Ingham #pragma mark CommandObjectProcessKill
124430fdc8d8SChris Lattner 
1245b9c1b51eSKate Stone class CommandObjectProcessKill : public CommandObjectParsed {
124630fdc8d8SChris Lattner public:
12477428a18cSKate Stone   CommandObjectProcessKill(CommandInterpreter &interpreter)
1248b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process kill",
1249b9c1b51eSKate Stone                             "Terminate the current target process.",
1250b9c1b51eSKate Stone                             "process kill",
1251b9c1b51eSKate Stone                             eCommandRequiresProcess | eCommandTryTargetAPILock |
1252b9c1b51eSKate Stone                                 eCommandProcessMustBeLaunched) {}
125330fdc8d8SChris Lattner 
125449bcfd80SEugene Zelenko   ~CommandObjectProcessKill() override = default;
125530fdc8d8SChris Lattner 
12565a988416SJim Ingham protected:
1257b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1258f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
1259b9c1b51eSKate Stone     if (process == nullptr) {
126030fdc8d8SChris Lattner       result.AppendError("no process to kill");
126130fdc8d8SChris Lattner       result.SetStatus(eReturnStatusFailed);
126230fdc8d8SChris Lattner       return false;
126330fdc8d8SChris Lattner     }
126430fdc8d8SChris Lattner 
1265b9c1b51eSKate Stone     if (command.GetArgumentCount() == 0) {
12668980e6bcSJason Molenda       Error error(process->Destroy(true));
1267b9c1b51eSKate Stone       if (error.Success()) {
126830fdc8d8SChris Lattner         result.SetStatus(eReturnStatusSuccessFinishResult);
1269b9c1b51eSKate Stone       } else {
1270b9c1b51eSKate Stone         result.AppendErrorWithFormat("Failed to kill process: %s\n",
1271b9c1b51eSKate Stone                                      error.AsCString());
127230fdc8d8SChris Lattner         result.SetStatus(eReturnStatusFailed);
127330fdc8d8SChris Lattner       }
1274b9c1b51eSKate Stone     } else {
1275fd54b368SJason Molenda       result.AppendErrorWithFormat("'%s' takes no arguments:\nUsage: %s\n",
1276b9c1b51eSKate Stone                                    m_cmd_name.c_str(), m_cmd_syntax.c_str());
127730fdc8d8SChris Lattner       result.SetStatus(eReturnStatusFailed);
127830fdc8d8SChris Lattner     }
127930fdc8d8SChris Lattner     return result.Succeeded();
128030fdc8d8SChris Lattner   }
128130fdc8d8SChris Lattner };
128230fdc8d8SChris Lattner 
128330fdc8d8SChris Lattner //-------------------------------------------------------------------------
1284a2715cf1SGreg Clayton // CommandObjectProcessSaveCore
1285a2715cf1SGreg Clayton //-------------------------------------------------------------------------
1286a2715cf1SGreg Clayton #pragma mark CommandObjectProcessSaveCore
1287a2715cf1SGreg Clayton 
1288b9c1b51eSKate Stone class CommandObjectProcessSaveCore : public CommandObjectParsed {
1289a2715cf1SGreg Clayton public:
1290b9c1b51eSKate Stone   CommandObjectProcessSaveCore(CommandInterpreter &interpreter)
1291b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process save-core",
1292b9c1b51eSKate Stone                             "Save the current process as a core file using an "
1293b9c1b51eSKate Stone                             "appropriate file type.",
1294a2715cf1SGreg Clayton                             "process save-core FILE",
1295b9c1b51eSKate Stone                             eCommandRequiresProcess | eCommandTryTargetAPILock |
1296b9c1b51eSKate Stone                                 eCommandProcessMustBeLaunched) {}
1297a2715cf1SGreg Clayton 
129849bcfd80SEugene Zelenko   ~CommandObjectProcessSaveCore() override = default;
1299a2715cf1SGreg Clayton 
1300a2715cf1SGreg Clayton protected:
1301b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1302a2715cf1SGreg Clayton     ProcessSP process_sp = m_exe_ctx.GetProcessSP();
1303b9c1b51eSKate Stone     if (process_sp) {
1304b9c1b51eSKate Stone       if (command.GetArgumentCount() == 1) {
1305a2715cf1SGreg Clayton         FileSpec output_file(command.GetArgumentAtIndex(0), false);
1306a2715cf1SGreg Clayton         Error error = PluginManager::SaveCore(process_sp, output_file);
1307b9c1b51eSKate Stone         if (error.Success()) {
1308a2715cf1SGreg Clayton           result.SetStatus(eReturnStatusSuccessFinishResult);
1309b9c1b51eSKate Stone         } else {
1310b9c1b51eSKate Stone           result.AppendErrorWithFormat(
1311b9c1b51eSKate Stone               "Failed to save core file for process: %s\n", error.AsCString());
1312a2715cf1SGreg Clayton           result.SetStatus(eReturnStatusFailed);
1313a2715cf1SGreg Clayton         }
1314b9c1b51eSKate Stone       } else {
1315a2715cf1SGreg Clayton         result.AppendErrorWithFormat("'%s' takes one arguments:\nUsage: %s\n",
1316b9c1b51eSKate Stone                                      m_cmd_name.c_str(), m_cmd_syntax.c_str());
1317a2715cf1SGreg Clayton         result.SetStatus(eReturnStatusFailed);
1318a2715cf1SGreg Clayton       }
1319b9c1b51eSKate Stone     } else {
1320a2715cf1SGreg Clayton       result.AppendError("invalid process");
1321a2715cf1SGreg Clayton       result.SetStatus(eReturnStatusFailed);
1322a2715cf1SGreg Clayton       return false;
1323a2715cf1SGreg Clayton     }
1324a2715cf1SGreg Clayton 
1325a2715cf1SGreg Clayton     return result.Succeeded();
1326a2715cf1SGreg Clayton   }
1327a2715cf1SGreg Clayton };
1328a2715cf1SGreg Clayton 
1329a2715cf1SGreg Clayton //-------------------------------------------------------------------------
13304b9bea87SJim Ingham // CommandObjectProcessStatus
13314b9bea87SJim Ingham //-------------------------------------------------------------------------
1332bb9caf73SJim Ingham #pragma mark CommandObjectProcessStatus
1333bb9caf73SJim Ingham 
1334b9c1b51eSKate Stone class CommandObjectProcessStatus : public CommandObjectParsed {
13354b9bea87SJim Ingham public:
13367428a18cSKate Stone   CommandObjectProcessStatus(CommandInterpreter &interpreter)
1337b9c1b51eSKate Stone       : CommandObjectParsed(
1338b9c1b51eSKate Stone             interpreter, "process status",
1339b9c1b51eSKate Stone             "Show status and stop location for the current target process.",
1340b9c1b51eSKate Stone             "process status",
1341b9c1b51eSKate Stone             eCommandRequiresProcess | eCommandTryTargetAPILock) {}
13424b9bea87SJim Ingham 
134349bcfd80SEugene Zelenko   ~CommandObjectProcessStatus() override = default;
13444b9bea87SJim Ingham 
1345b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
13467260f620SGreg Clayton     Stream &strm = result.GetOutputStream();
13474b9bea87SJim Ingham     result.SetStatus(eReturnStatusSuccessFinishNoResult);
1348b9c1b51eSKate Stone     // No need to check "process" for validity as eCommandRequiresProcess
1349b9c1b51eSKate Stone     // ensures it is valid
1350f9fc609fSGreg Clayton     Process *process = m_exe_ctx.GetProcessPtr();
13517260f620SGreg Clayton     const bool only_threads_with_stop_reason = true;
13527260f620SGreg Clayton     const uint32_t start_frame = 0;
13537260f620SGreg Clayton     const uint32_t num_frames = 1;
13547260f620SGreg Clayton     const uint32_t num_frames_with_source = 1;
1355c14ee32dSGreg Clayton     process->GetStatus(strm);
1356b9c1b51eSKate Stone     process->GetThreadStatus(strm, only_threads_with_stop_reason, start_frame,
1357b9c1b51eSKate Stone                              num_frames, num_frames_with_source);
13584b9bea87SJim Ingham     return result.Succeeded();
13594b9bea87SJim Ingham   }
13604b9bea87SJim Ingham };
13614b9bea87SJim Ingham 
13624b9bea87SJim Ingham //-------------------------------------------------------------------------
136335731357SCaroline Tice // CommandObjectProcessHandle
136435731357SCaroline Tice //-------------------------------------------------------------------------
13651f0f5b5bSZachary Turner 
13661f0f5b5bSZachary Turner static OptionDefinition g_process_handle_options[] = {
13671f0f5b5bSZachary Turner     // clang-format off
13681f0f5b5bSZachary Turner   { LLDB_OPT_SET_1, false, "stop",   's', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeBoolean, "Whether or not the process should be stopped if the signal is received." },
13691f0f5b5bSZachary Turner   { LLDB_OPT_SET_1, false, "notify", 'n', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeBoolean, "Whether or not the debugger should notify the user if the signal is received." },
13701f0f5b5bSZachary Turner   { LLDB_OPT_SET_1, false, "pass",   'p', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeBoolean, "Whether or not the signal should be passed to the process." }
13711f0f5b5bSZachary Turner     // clang-format on
13721f0f5b5bSZachary Turner };
13731f0f5b5bSZachary Turner 
1374bb9caf73SJim Ingham #pragma mark CommandObjectProcessHandle
137535731357SCaroline Tice 
1376b9c1b51eSKate Stone class CommandObjectProcessHandle : public CommandObjectParsed {
137735731357SCaroline Tice public:
1378b9c1b51eSKate Stone   class CommandOptions : public Options {
137935731357SCaroline Tice   public:
1380b9c1b51eSKate Stone     CommandOptions() : Options() { OptionParsingStarting(nullptr); }
138135731357SCaroline Tice 
138249bcfd80SEugene Zelenko     ~CommandOptions() override = default;
138335731357SCaroline Tice 
1384b9c1b51eSKate Stone     Error SetOptionValue(uint32_t option_idx, const char *option_arg,
1385b9c1b51eSKate Stone                          ExecutionContext *execution_context) override {
138635731357SCaroline Tice       Error error;
13873bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
138835731357SCaroline Tice 
1389b9c1b51eSKate Stone       switch (short_option) {
139035731357SCaroline Tice       case 's':
139135731357SCaroline Tice         stop = option_arg;
139235731357SCaroline Tice         break;
139335731357SCaroline Tice       case 'n':
139435731357SCaroline Tice         notify = option_arg;
139535731357SCaroline Tice         break;
139635731357SCaroline Tice       case 'p':
139735731357SCaroline Tice         pass = option_arg;
139835731357SCaroline Tice         break;
139935731357SCaroline Tice       default:
1400b9c1b51eSKate Stone         error.SetErrorStringWithFormat("invalid short option character '%c'",
1401b9c1b51eSKate Stone                                        short_option);
140235731357SCaroline Tice         break;
140335731357SCaroline Tice       }
140435731357SCaroline Tice       return error;
140535731357SCaroline Tice     }
140635731357SCaroline Tice 
1407b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
140835731357SCaroline Tice       stop.clear();
140935731357SCaroline Tice       notify.clear();
141035731357SCaroline Tice       pass.clear();
141135731357SCaroline Tice     }
141235731357SCaroline Tice 
14131f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
141470602439SZachary Turner       return llvm::makeArrayRef(g_process_handle_options);
14151f0f5b5bSZachary Turner     }
141635731357SCaroline Tice 
141735731357SCaroline Tice     // Instance variables to hold the values for command options.
141835731357SCaroline Tice 
141935731357SCaroline Tice     std::string stop;
142035731357SCaroline Tice     std::string notify;
142135731357SCaroline Tice     std::string pass;
142235731357SCaroline Tice   };
142335731357SCaroline Tice 
14247428a18cSKate Stone   CommandObjectProcessHandle(CommandInterpreter &interpreter)
1425b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "process handle",
1426b9c1b51eSKate Stone                             "Manage LLDB handling of OS signals for the "
1427b9c1b51eSKate Stone                             "current target process.  Defaults to showing "
1428b9c1b51eSKate Stone                             "current policy.",
142949bcfd80SEugene Zelenko                             nullptr),
1430b9c1b51eSKate Stone         m_options() {
1431ea671fbdSKate Stone     SetHelpLong("\nIf no signals are specified, update them all.  If no update "
1432ea671fbdSKate Stone                 "option is specified, list the current values.");
143335731357SCaroline Tice     CommandArgumentEntry arg;
1434c0dbdfb6SCaroline Tice     CommandArgumentData signal_arg;
143535731357SCaroline Tice 
1436c0dbdfb6SCaroline Tice     signal_arg.arg_type = eArgTypeUnixSignal;
1437c0dbdfb6SCaroline Tice     signal_arg.arg_repetition = eArgRepeatStar;
143835731357SCaroline Tice 
1439c0dbdfb6SCaroline Tice     arg.push_back(signal_arg);
144035731357SCaroline Tice 
144135731357SCaroline Tice     m_arguments.push_back(arg);
144235731357SCaroline Tice   }
144335731357SCaroline Tice 
144449bcfd80SEugene Zelenko   ~CommandObjectProcessHandle() override = default;
144535731357SCaroline Tice 
1446b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
144735731357SCaroline Tice 
1448b9c1b51eSKate Stone   bool VerifyCommandOptionValue(const std::string &option, int &real_value) {
144935731357SCaroline Tice     bool okay = true;
145010ad7993SCaroline Tice     bool success = false;
1451ecbb0bb1SZachary Turner     bool tmp_value = Args::StringToBoolean(option, false, &success);
145210ad7993SCaroline Tice 
145310ad7993SCaroline Tice     if (success && tmp_value)
145410ad7993SCaroline Tice       real_value = 1;
145510ad7993SCaroline Tice     else if (success && !tmp_value)
145610ad7993SCaroline Tice       real_value = 0;
1457b9c1b51eSKate Stone     else {
145835731357SCaroline Tice       // If the value isn't 'true' or 'false', it had better be 0 or 1.
14595275aaa0SVince Harron       real_value = StringConvert::ToUInt32(option.c_str(), 3);
146010ad7993SCaroline Tice       if (real_value != 0 && real_value != 1)
146135731357SCaroline Tice         okay = false;
146235731357SCaroline Tice     }
146335731357SCaroline Tice 
146435731357SCaroline Tice     return okay;
146535731357SCaroline Tice   }
146635731357SCaroline Tice 
1467b9c1b51eSKate Stone   void PrintSignalHeader(Stream &str) {
146810ad7993SCaroline Tice     str.Printf("NAME         PASS   STOP   NOTIFY\n");
1469b84141a6SPavel Labath     str.Printf("===========  =====  =====  ======\n");
147010ad7993SCaroline Tice   }
147110ad7993SCaroline Tice 
1472b9c1b51eSKate Stone   void PrintSignal(Stream &str, int32_t signo, const char *sig_name,
1473b9c1b51eSKate Stone                    const UnixSignalsSP &signals_sp) {
147410ad7993SCaroline Tice     bool stop;
147510ad7993SCaroline Tice     bool suppress;
147610ad7993SCaroline Tice     bool notify;
147710ad7993SCaroline Tice 
1478b84141a6SPavel Labath     str.Printf("%-11s  ", sig_name);
1479b9c1b51eSKate Stone     if (signals_sp->GetSignalInfo(signo, suppress, stop, notify)) {
148010ad7993SCaroline Tice       bool pass = !suppress;
1481b9c1b51eSKate Stone       str.Printf("%s  %s  %s", (pass ? "true " : "false"),
1482b9c1b51eSKate Stone                  (stop ? "true " : "false"), (notify ? "true " : "false"));
148310ad7993SCaroline Tice     }
148410ad7993SCaroline Tice     str.Printf("\n");
148510ad7993SCaroline Tice   }
148610ad7993SCaroline Tice 
1487b9c1b51eSKate Stone   void PrintSignalInformation(Stream &str, Args &signal_args,
1488b9c1b51eSKate Stone                               int num_valid_signals,
1489b9c1b51eSKate Stone                               const UnixSignalsSP &signals_sp) {
149010ad7993SCaroline Tice     PrintSignalHeader(str);
149110ad7993SCaroline Tice 
1492b9c1b51eSKate Stone     if (num_valid_signals > 0) {
149310ad7993SCaroline Tice       size_t num_args = signal_args.GetArgumentCount();
1494b9c1b51eSKate Stone       for (size_t i = 0; i < num_args; ++i) {
1495b9c1b51eSKate Stone         int32_t signo = signals_sp->GetSignalNumberFromName(
1496b9c1b51eSKate Stone             signal_args.GetArgumentAtIndex(i));
149710ad7993SCaroline Tice         if (signo != LLDB_INVALID_SIGNAL_NUMBER)
1498b9c1b51eSKate Stone           PrintSignal(str, signo, signal_args.GetArgumentAtIndex(i),
1499b9c1b51eSKate Stone                       signals_sp);
150010ad7993SCaroline Tice       }
1501b9c1b51eSKate Stone     } else // Print info for ALL signals
150210ad7993SCaroline Tice     {
150398d0a4b3SChaoren Lin       int32_t signo = signals_sp->GetFirstSignalNumber();
1504b9c1b51eSKate Stone       while (signo != LLDB_INVALID_SIGNAL_NUMBER) {
1505b9c1b51eSKate Stone         PrintSignal(str, signo, signals_sp->GetSignalAsCString(signo),
1506b9c1b51eSKate Stone                     signals_sp);
150798d0a4b3SChaoren Lin         signo = signals_sp->GetNextSignalNumber(signo);
150810ad7993SCaroline Tice       }
150910ad7993SCaroline Tice     }
151010ad7993SCaroline Tice   }
151110ad7993SCaroline Tice 
15125a988416SJim Ingham protected:
1513b9c1b51eSKate Stone   bool DoExecute(Args &signal_args, CommandReturnObject &result) override {
151435731357SCaroline Tice     TargetSP target_sp = m_interpreter.GetDebugger().GetSelectedTarget();
151535731357SCaroline Tice 
1516b9c1b51eSKate Stone     if (!target_sp) {
151735731357SCaroline Tice       result.AppendError("No current target;"
1518b9c1b51eSKate Stone                          " cannot handle signals until you have a valid target "
1519b9c1b51eSKate Stone                          "and process.\n");
152035731357SCaroline Tice       result.SetStatus(eReturnStatusFailed);
152135731357SCaroline Tice       return false;
152235731357SCaroline Tice     }
152335731357SCaroline Tice 
152435731357SCaroline Tice     ProcessSP process_sp = target_sp->GetProcessSP();
152535731357SCaroline Tice 
1526b9c1b51eSKate Stone     if (!process_sp) {
1527b9c1b51eSKate Stone       result.AppendError("No current process; cannot handle signals until you "
1528b9c1b51eSKate Stone                          "have a valid process.\n");
152935731357SCaroline Tice       result.SetStatus(eReturnStatusFailed);
153035731357SCaroline Tice       return false;
153135731357SCaroline Tice     }
153235731357SCaroline Tice 
153335731357SCaroline Tice     int stop_action = -1;   // -1 means leave the current setting alone
153435731357SCaroline Tice     int pass_action = -1;   // -1 means leave the current setting alone
153535731357SCaroline Tice     int notify_action = -1; // -1 means leave the current setting alone
153635731357SCaroline Tice 
1537b9c1b51eSKate Stone     if (!m_options.stop.empty() &&
1538b9c1b51eSKate Stone         !VerifyCommandOptionValue(m_options.stop, stop_action)) {
1539b9c1b51eSKate Stone       result.AppendError("Invalid argument for command option --stop; must be "
1540b9c1b51eSKate Stone                          "true or false.\n");
154135731357SCaroline Tice       result.SetStatus(eReturnStatusFailed);
154235731357SCaroline Tice       return false;
154335731357SCaroline Tice     }
154435731357SCaroline Tice 
1545b9c1b51eSKate Stone     if (!m_options.notify.empty() &&
1546b9c1b51eSKate Stone         !VerifyCommandOptionValue(m_options.notify, notify_action)) {
1547b9c1b51eSKate Stone       result.AppendError("Invalid argument for command option --notify; must "
1548b9c1b51eSKate Stone                          "be true or false.\n");
154935731357SCaroline Tice       result.SetStatus(eReturnStatusFailed);
155035731357SCaroline Tice       return false;
155135731357SCaroline Tice     }
155235731357SCaroline Tice 
1553b9c1b51eSKate Stone     if (!m_options.pass.empty() &&
1554b9c1b51eSKate Stone         !VerifyCommandOptionValue(m_options.pass, pass_action)) {
1555b9c1b51eSKate Stone       result.AppendError("Invalid argument for command option --pass; must be "
1556b9c1b51eSKate Stone                          "true or false.\n");
155735731357SCaroline Tice       result.SetStatus(eReturnStatusFailed);
155835731357SCaroline Tice       return false;
155935731357SCaroline Tice     }
156035731357SCaroline Tice 
156135731357SCaroline Tice     size_t num_args = signal_args.GetArgumentCount();
156298d0a4b3SChaoren Lin     UnixSignalsSP signals_sp = process_sp->GetUnixSignals();
156335731357SCaroline Tice     int num_signals_set = 0;
156435731357SCaroline Tice 
1565b9c1b51eSKate Stone     if (num_args > 0) {
1566b9c1b51eSKate Stone       for (size_t i = 0; i < num_args; ++i) {
1567b9c1b51eSKate Stone         int32_t signo = signals_sp->GetSignalNumberFromName(
1568b9c1b51eSKate Stone             signal_args.GetArgumentAtIndex(i));
1569b9c1b51eSKate Stone         if (signo != LLDB_INVALID_SIGNAL_NUMBER) {
1570b9c1b51eSKate Stone           // Casting the actions as bools here should be okay, because
1571b9c1b51eSKate Stone           // VerifyCommandOptionValue guarantees
157235731357SCaroline Tice           // the value is either 0 or 1.
157335731357SCaroline Tice           if (stop_action != -1)
157498d0a4b3SChaoren Lin             signals_sp->SetShouldStop(signo, stop_action);
1575b9c1b51eSKate Stone           if (pass_action != -1) {
157698d0a4b3SChaoren Lin             bool suppress = !pass_action;
157798d0a4b3SChaoren Lin             signals_sp->SetShouldSuppress(signo, suppress);
157835731357SCaroline Tice           }
157935731357SCaroline Tice           if (notify_action != -1)
158098d0a4b3SChaoren Lin             signals_sp->SetShouldNotify(signo, notify_action);
158135731357SCaroline Tice           ++num_signals_set;
1582b9c1b51eSKate Stone         } else {
1583b9c1b51eSKate Stone           result.AppendErrorWithFormat("Invalid signal name '%s'\n",
1584b9c1b51eSKate Stone                                        signal_args.GetArgumentAtIndex(i));
158535731357SCaroline Tice         }
158635731357SCaroline Tice       }
1587b9c1b51eSKate Stone     } else {
1588b9c1b51eSKate Stone       // No signal specified, if any command options were specified, update ALL
1589b9c1b51eSKate Stone       // signals.
1590b9c1b51eSKate Stone       if ((notify_action != -1) || (stop_action != -1) || (pass_action != -1)) {
1591b9c1b51eSKate Stone         if (m_interpreter.Confirm(
1592b9c1b51eSKate Stone                 "Do you really want to update all the signals?", false)) {
159398d0a4b3SChaoren Lin           int32_t signo = signals_sp->GetFirstSignalNumber();
1594b9c1b51eSKate Stone           while (signo != LLDB_INVALID_SIGNAL_NUMBER) {
159510ad7993SCaroline Tice             if (notify_action != -1)
159698d0a4b3SChaoren Lin               signals_sp->SetShouldNotify(signo, notify_action);
159710ad7993SCaroline Tice             if (stop_action != -1)
159898d0a4b3SChaoren Lin               signals_sp->SetShouldStop(signo, stop_action);
1599b9c1b51eSKate Stone             if (pass_action != -1) {
160098d0a4b3SChaoren Lin               bool suppress = !pass_action;
160198d0a4b3SChaoren Lin               signals_sp->SetShouldSuppress(signo, suppress);
160210ad7993SCaroline Tice             }
160398d0a4b3SChaoren Lin             signo = signals_sp->GetNextSignalNumber(signo);
160410ad7993SCaroline Tice           }
160510ad7993SCaroline Tice         }
160610ad7993SCaroline Tice       }
160710ad7993SCaroline Tice     }
160810ad7993SCaroline Tice 
1609b9c1b51eSKate Stone     PrintSignalInformation(result.GetOutputStream(), signal_args,
1610b9c1b51eSKate Stone                            num_signals_set, signals_sp);
161135731357SCaroline Tice 
161235731357SCaroline Tice     if (num_signals_set > 0)
161335731357SCaroline Tice       result.SetStatus(eReturnStatusSuccessFinishNoResult);
161435731357SCaroline Tice     else
161535731357SCaroline Tice       result.SetStatus(eReturnStatusFailed);
161635731357SCaroline Tice 
161735731357SCaroline Tice     return result.Succeeded();
161835731357SCaroline Tice   }
161935731357SCaroline Tice 
162035731357SCaroline Tice   CommandOptions m_options;
162135731357SCaroline Tice };
162235731357SCaroline Tice 
162335731357SCaroline Tice //-------------------------------------------------------------------------
162430fdc8d8SChris Lattner // CommandObjectMultiwordProcess
162530fdc8d8SChris Lattner //-------------------------------------------------------------------------
162630fdc8d8SChris Lattner 
1627b9c1b51eSKate Stone CommandObjectMultiwordProcess::CommandObjectMultiwordProcess(
1628b9c1b51eSKate Stone     CommandInterpreter &interpreter)
1629b9c1b51eSKate Stone     : CommandObjectMultiword(
1630b9c1b51eSKate Stone           interpreter, "process",
1631b9c1b51eSKate Stone           "Commands for interacting with processes on the current platform.",
1632b9c1b51eSKate Stone           "process <subcommand> [<subcommand-options>]") {
1633b9c1b51eSKate Stone   LoadSubCommand("attach",
1634b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessAttach(interpreter)));
1635b9c1b51eSKate Stone   LoadSubCommand("launch",
1636b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessLaunch(interpreter)));
1637b9c1b51eSKate Stone   LoadSubCommand("continue", CommandObjectSP(new CommandObjectProcessContinue(
1638b9c1b51eSKate Stone                                  interpreter)));
1639b9c1b51eSKate Stone   LoadSubCommand("connect",
1640b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessConnect(interpreter)));
1641b9c1b51eSKate Stone   LoadSubCommand("detach",
1642b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessDetach(interpreter)));
1643b9c1b51eSKate Stone   LoadSubCommand("load",
1644b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessLoad(interpreter)));
1645b9c1b51eSKate Stone   LoadSubCommand("unload",
1646b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessUnload(interpreter)));
1647b9c1b51eSKate Stone   LoadSubCommand("signal",
1648b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessSignal(interpreter)));
1649b9c1b51eSKate Stone   LoadSubCommand("handle",
1650b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessHandle(interpreter)));
1651b9c1b51eSKate Stone   LoadSubCommand("status",
1652b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessStatus(interpreter)));
1653b9c1b51eSKate Stone   LoadSubCommand("interrupt", CommandObjectSP(new CommandObjectProcessInterrupt(
1654b9c1b51eSKate Stone                                   interpreter)));
1655b9c1b51eSKate Stone   LoadSubCommand("kill",
1656b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessKill(interpreter)));
1657b9c1b51eSKate Stone   LoadSubCommand("plugin",
1658b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectProcessPlugin(interpreter)));
1659b9c1b51eSKate Stone   LoadSubCommand("save-core", CommandObjectSP(new CommandObjectProcessSaveCore(
1660b9c1b51eSKate Stone                                   interpreter)));
166130fdc8d8SChris Lattner }
166230fdc8d8SChris Lattner 
166349bcfd80SEugene Zelenko CommandObjectMultiwordProcess::~CommandObjectMultiwordProcess() = default;
1664