1 //===-- CommandObjectProcess.cpp --------------------------------*- C++ -*-===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 
10 #include "lldb/lldb-python.h"
11 
12 #include "CommandObjectProcess.h"
13 
14 // C Includes
15 // C++ Includes
16 // Other libraries and framework includes
17 // Project includes
18 #include "lldb/Breakpoint/Breakpoint.h"
19 #include "lldb/Breakpoint/BreakpointLocation.h"
20 #include "lldb/Breakpoint/BreakpointSite.h"
21 #include "lldb/Core/State.h"
22 #include "lldb/Core/Module.h"
23 #include "lldb/Core/PluginManager.h"
24 #include "lldb/Host/Host.h"
25 #include "lldb/Interpreter/Args.h"
26 #include "lldb/Interpreter/Options.h"
27 #include "lldb/Interpreter/CommandInterpreter.h"
28 #include "lldb/Interpreter/CommandReturnObject.h"
29 #include "lldb/Target/Platform.h"
30 #include "lldb/Target/Process.h"
31 #include "lldb/Target/StopInfo.h"
32 #include "lldb/Target/Target.h"
33 #include "lldb/Target/Thread.h"
34 
35 using namespace lldb;
36 using namespace lldb_private;
37 
38 class CommandObjectProcessLaunchOrAttach : public CommandObjectParsed
39 {
40 public:
41     CommandObjectProcessLaunchOrAttach (CommandInterpreter &interpreter,
42                                        const char *name,
43                                        const char *help,
44                                        const char *syntax,
45                                        uint32_t flags,
46                                        const char *new_process_action) :
47         CommandObjectParsed (interpreter, name, help, syntax, flags),
48         m_new_process_action (new_process_action) {}
49 
50     virtual ~CommandObjectProcessLaunchOrAttach () {}
51 protected:
52     bool
53     StopProcessIfNecessary (Process *process, StateType &state, CommandReturnObject &result)
54     {
55         state = eStateInvalid;
56         if (process)
57         {
58             state = process->GetState();
59 
60             if (process->IsAlive() && state != eStateConnected)
61             {
62                 char message[1024];
63                 if (process->GetState() == eStateAttaching)
64                     ::snprintf (message, sizeof(message), "There is a pending attach, abort it and %s?", m_new_process_action.c_str());
65                 else if (process->GetShouldDetach())
66                     ::snprintf (message, sizeof(message), "There is a running process, detach from it and %s?", m_new_process_action.c_str());
67                 else
68                     ::snprintf (message, sizeof(message), "There is a running process, kill it and %s?", m_new_process_action.c_str());
69 
70                 if (!m_interpreter.Confirm (message, true))
71                 {
72                     result.SetStatus (eReturnStatusFailed);
73                     return false;
74                 }
75                 else
76                 {
77                     if (process->GetShouldDetach())
78                     {
79                         bool keep_stopped = false;
80                         Error detach_error (process->Detach(keep_stopped));
81                         if (detach_error.Success())
82                         {
83                             result.SetStatus (eReturnStatusSuccessFinishResult);
84                             process = NULL;
85                         }
86                         else
87                         {
88                             result.AppendErrorWithFormat ("Failed to detach from process: %s\n", detach_error.AsCString());
89                             result.SetStatus (eReturnStatusFailed);
90                         }
91                     }
92                     else
93                     {
94                         Error destroy_error (process->Destroy());
95                         if (destroy_error.Success())
96                         {
97                             result.SetStatus (eReturnStatusSuccessFinishResult);
98                             process = NULL;
99                         }
100                         else
101                         {
102                             result.AppendErrorWithFormat ("Failed to kill process: %s\n", destroy_error.AsCString());
103                             result.SetStatus (eReturnStatusFailed);
104                         }
105                     }
106                 }
107             }
108         }
109         return result.Succeeded();
110     }
111     std::string m_new_process_action;
112 };
113 //-------------------------------------------------------------------------
114 // CommandObjectProcessLaunch
115 //-------------------------------------------------------------------------
116 #pragma mark CommandObjectProcessLaunch
117 class CommandObjectProcessLaunch : public CommandObjectProcessLaunchOrAttach
118 {
119 public:
120 
121     CommandObjectProcessLaunch (CommandInterpreter &interpreter) :
122         CommandObjectProcessLaunchOrAttach (interpreter,
123                                             "process launch",
124                                             "Launch the executable in the debugger.",
125                                             NULL,
126                                             eFlagRequiresTarget,
127                                             "restart"),
128         m_options (interpreter)
129     {
130         CommandArgumentEntry arg;
131         CommandArgumentData run_args_arg;
132 
133         // Define the first (and only) variant of this arg.
134         run_args_arg.arg_type = eArgTypeRunArgs;
135         run_args_arg.arg_repetition = eArgRepeatOptional;
136 
137         // There is only one variant this argument could be; put it into the argument entry.
138         arg.push_back (run_args_arg);
139 
140         // Push the data for the first argument into the m_arguments vector.
141         m_arguments.push_back (arg);
142     }
143 
144 
145     ~CommandObjectProcessLaunch ()
146     {
147     }
148 
149     virtual int
150     HandleArgumentCompletion (Args &input,
151                               int &cursor_index,
152                               int &cursor_char_position,
153                               OptionElementVector &opt_element_vector,
154                               int match_start_point,
155                               int max_return_elements,
156                               bool &word_complete,
157                               StringList &matches)
158     {
159         std::string completion_str (input.GetArgumentAtIndex(cursor_index));
160         completion_str.erase (cursor_char_position);
161 
162         CommandCompletions::InvokeCommonCompletionCallbacks (m_interpreter,
163                                                              CommandCompletions::eDiskFileCompletion,
164                                                              completion_str.c_str(),
165                                                              match_start_point,
166                                                              max_return_elements,
167                                                              NULL,
168                                                              word_complete,
169                                                              matches);
170         return matches.GetSize();
171     }
172 
173     Options *
174     GetOptions ()
175     {
176         return &m_options;
177     }
178 
179     virtual const char *GetRepeatCommand (Args &current_command_args, uint32_t index)
180     {
181         // No repeat for "process launch"...
182         return "";
183     }
184 
185 protected:
186     bool
187     DoExecute (Args& launch_args, CommandReturnObject &result)
188     {
189         Debugger &debugger = m_interpreter.GetDebugger();
190         Target *target = debugger.GetSelectedTarget().get();
191         // If our listener is NULL, users aren't allows to launch
192         ModuleSP exe_module_sp = target->GetExecutableModule();
193 
194         if (exe_module_sp == NULL)
195         {
196             result.AppendError ("no file in target, create a debug target using the 'target create' command");
197             result.SetStatus (eReturnStatusFailed);
198             return false;
199         }
200 
201         StateType state = eStateInvalid;
202 
203         if (!StopProcessIfNecessary(m_exe_ctx.GetProcessPtr(), state, result))
204             return false;
205 
206         const char *target_settings_argv0 = target->GetArg0();
207 
208         if (target->GetDisableASLR())
209             m_options.launch_info.GetFlags().Set (eLaunchFlagDisableASLR);
210 
211         if (target->GetDetachOnError())
212             m_options.launch_info.GetFlags().Set (eLaunchFlagDetachOnError);
213 
214         if (target->GetDisableSTDIO())
215             m_options.launch_info.GetFlags().Set (eLaunchFlagDisableSTDIO);
216 
217         Args environment;
218         target->GetEnvironmentAsArgs (environment);
219         if (environment.GetArgumentCount() > 0)
220             m_options.launch_info.GetEnvironmentEntries ().AppendArguments (environment);
221 
222         if (target_settings_argv0)
223         {
224             m_options.launch_info.GetArguments().AppendArgument (target_settings_argv0);
225             m_options.launch_info.SetExecutableFile(exe_module_sp->GetPlatformFileSpec(), false);
226         }
227         else
228         {
229             m_options.launch_info.SetExecutableFile(exe_module_sp->GetPlatformFileSpec(), true);
230         }
231 
232         if (launch_args.GetArgumentCount() == 0)
233         {
234             Args target_setting_args;
235             if (target->GetRunArguments(target_setting_args))
236                 m_options.launch_info.GetArguments().AppendArguments (target_setting_args);
237         }
238         else
239         {
240             m_options.launch_info.GetArguments().AppendArguments (launch_args);
241             // Save the arguments for subsequent runs in the current target.
242             target->SetRunArguments (launch_args);
243         }
244 
245         Error error = target->Launch(debugger.GetListener(), m_options.launch_info);
246 
247         if (error.Success())
248         {
249             const char *archname = exe_module_sp->GetArchitecture().GetArchitectureName();
250             ProcessSP process_sp (target->GetProcessSP());
251             if (process_sp)
252             {
253                 result.AppendMessageWithFormat ("Process %" PRIu64 " launched: '%s' (%s)\n", process_sp->GetID(), exe_module_sp->GetFileSpec().GetPath().c_str(), archname);
254                 result.SetStatus (eReturnStatusSuccessFinishResult);
255                 result.SetDidChangeProcessState (true);
256             }
257             else
258             {
259                 result.AppendError("no error returned from Target::Launch, and target has no process");
260                 result.SetStatus (eReturnStatusFailed);
261             }
262         }
263         else
264         {
265             result.AppendError(error.AsCString());
266             result.SetStatus (eReturnStatusFailed);
267         }
268         return result.Succeeded();
269     }
270 
271 protected:
272     ProcessLaunchCommandOptions m_options;
273 };
274 
275 
276 //#define SET1 LLDB_OPT_SET_1
277 //#define SET2 LLDB_OPT_SET_2
278 //#define SET3 LLDB_OPT_SET_3
279 //
280 //OptionDefinition
281 //CommandObjectProcessLaunch::CommandOptions::g_option_table[] =
282 //{
283 //{ SET1 | SET2 | SET3, false, "stop-at-entry", 's', OptionParser::eNoArgument,       NULL, 0, eArgTypeNone,    "Stop at the entry point of the program when launching a process."},
284 //{ SET1              , false, "stdin",         'i', OptionParser::eRequiredArgument, NULL, 0, eArgTypeDirectoryName,    "Redirect stdin for the process to <path>."},
285 //{ SET1              , false, "stdout",        'o', OptionParser::eRequiredArgument, NULL, 0, eArgTypeDirectoryName,    "Redirect stdout for the process to <path>."},
286 //{ SET1              , false, "stderr",        'e', OptionParser::eRequiredArgument, NULL, 0, eArgTypeDirectoryName,    "Redirect stderr for the process to <path>."},
287 //{ SET1 | SET2 | SET3, false, "plugin",        'p', OptionParser::eRequiredArgument, NULL, 0, eArgTypePlugin,  "Name of the process plugin you want to use."},
288 //{        SET2       , false, "tty",           't', OptionParser::eOptionalArgument, NULL, 0, eArgTypeDirectoryName,    "Start the process in a terminal. If <path> is specified, look for a terminal whose name contains <path>, else start the process in a new terminal."},
289 //{               SET3, false, "no-stdio",      'n', OptionParser::eNoArgument,       NULL, 0, eArgTypeNone,    "Do not set up for terminal I/O to go to running process."},
290 //{ SET1 | SET2 | SET3, false, "working-dir",   'w', OptionParser::eRequiredArgument, NULL, 0, eArgTypeDirectoryName,    "Set the current working directory to <path> when running the inferior."},
291 //{ 0,                  false, NULL,             0,  0,                 NULL, 0, eArgTypeNone,    NULL }
292 //};
293 //
294 //#undef SET1
295 //#undef SET2
296 //#undef SET3
297 
298 //-------------------------------------------------------------------------
299 // CommandObjectProcessAttach
300 //-------------------------------------------------------------------------
301 #pragma mark CommandObjectProcessAttach
302 class CommandObjectProcessAttach : public CommandObjectProcessLaunchOrAttach
303 {
304 public:
305 
306     class CommandOptions : public Options
307     {
308     public:
309 
310         CommandOptions (CommandInterpreter &interpreter) :
311             Options(interpreter)
312         {
313             // Keep default values of all options in one place: OptionParsingStarting ()
314             OptionParsingStarting ();
315         }
316 
317         ~CommandOptions ()
318         {
319         }
320 
321         Error
322         SetOptionValue (uint32_t option_idx, const char *option_arg)
323         {
324             Error error;
325             const int short_option = m_getopt_table[option_idx].val;
326             bool success = false;
327             switch (short_option)
328             {
329                 case 'c':
330                     attach_info.SetContinueOnceAttached(true);
331                     break;
332 
333                 case 'p':
334                     {
335                         lldb::pid_t pid = Args::StringToUInt32 (option_arg, LLDB_INVALID_PROCESS_ID, 0, &success);
336                         if (!success || pid == LLDB_INVALID_PROCESS_ID)
337                         {
338                             error.SetErrorStringWithFormat("invalid process ID '%s'", option_arg);
339                         }
340                         else
341                         {
342                             attach_info.SetProcessID (pid);
343                         }
344                     }
345                     break;
346 
347                 case 'P':
348                     attach_info.SetProcessPluginName (option_arg);
349                     break;
350 
351                 case 'n':
352                     attach_info.GetExecutableFile().SetFile(option_arg, false);
353                     break;
354 
355                 case 'w':
356                     attach_info.SetWaitForLaunch(true);
357                     break;
358 
359                 case 'i':
360                     attach_info.SetIgnoreExisting(false);
361                     break;
362 
363                 default:
364                     error.SetErrorStringWithFormat("invalid short option character '%c'", short_option);
365                     break;
366             }
367             return error;
368         }
369 
370         void
371         OptionParsingStarting ()
372         {
373             attach_info.Clear();
374         }
375 
376         const OptionDefinition*
377         GetDefinitions ()
378         {
379             return g_option_table;
380         }
381 
382         virtual bool
383         HandleOptionArgumentCompletion (Args &input,
384                                         int cursor_index,
385                                         int char_pos,
386                                         OptionElementVector &opt_element_vector,
387                                         int opt_element_index,
388                                         int match_start_point,
389                                         int max_return_elements,
390                                         bool &word_complete,
391                                         StringList &matches)
392         {
393             int opt_arg_pos = opt_element_vector[opt_element_index].opt_arg_pos;
394             int opt_defs_index = opt_element_vector[opt_element_index].opt_defs_index;
395 
396             // We are only completing the name option for now...
397 
398             const OptionDefinition *opt_defs = GetDefinitions();
399             if (opt_defs[opt_defs_index].short_option == 'n')
400             {
401                 // Are we in the name?
402 
403                 // Look to see if there is a -P argument provided, and if so use that plugin, otherwise
404                 // use the default plugin.
405 
406                 const char *partial_name = NULL;
407                 partial_name = input.GetArgumentAtIndex(opt_arg_pos);
408 
409                 PlatformSP platform_sp (m_interpreter.GetPlatform (true));
410                 if (platform_sp)
411                 {
412                     ProcessInstanceInfoList process_infos;
413                     ProcessInstanceInfoMatch match_info;
414                     if (partial_name)
415                     {
416                         match_info.GetProcessInfo().GetExecutableFile().SetFile(partial_name, false);
417                         match_info.SetNameMatchType(eNameMatchStartsWith);
418                     }
419                     platform_sp->FindProcesses (match_info, process_infos);
420                     const size_t num_matches = process_infos.GetSize();
421                     if (num_matches > 0)
422                     {
423                         for (size_t i=0; i<num_matches; ++i)
424                         {
425                             matches.AppendString (process_infos.GetProcessNameAtIndex(i),
426                                                   process_infos.GetProcessNameLengthAtIndex(i));
427                         }
428                     }
429                 }
430             }
431 
432             return false;
433         }
434 
435         // Options table: Required for subclasses of Options.
436 
437         static OptionDefinition g_option_table[];
438 
439         // Instance variables to hold the values for command options.
440 
441         ProcessAttachInfo attach_info;
442     };
443 
444     CommandObjectProcessAttach (CommandInterpreter &interpreter) :
445         CommandObjectProcessLaunchOrAttach (interpreter,
446                                             "process attach",
447                                             "Attach to a process.",
448                                             "process attach <cmd-options>",
449                                             0,
450                                             "attach"),
451         m_options (interpreter)
452     {
453     }
454 
455     ~CommandObjectProcessAttach ()
456     {
457     }
458 
459     Options *
460     GetOptions ()
461     {
462         return &m_options;
463     }
464 
465 protected:
466     bool
467     DoExecute (Args& command,
468              CommandReturnObject &result)
469     {
470         Target *target = m_interpreter.GetDebugger().GetSelectedTarget().get();
471         // N.B. The attach should be synchronous.  It doesn't help much to get the prompt back between initiating the attach
472         // and the target actually stopping.  So even if the interpreter is set to be asynchronous, we wait for the stop
473         // ourselves here.
474 
475         StateType state = eStateInvalid;
476         Process *process = m_exe_ctx.GetProcessPtr();
477 
478         if (!StopProcessIfNecessary (process, state, result))
479             return false;
480 
481         if (target == NULL)
482         {
483             // If there isn't a current target create one.
484             TargetSP new_target_sp;
485             Error error;
486 
487             error = m_interpreter.GetDebugger().GetTargetList().CreateTarget (m_interpreter.GetDebugger(),
488                                                                               NULL,
489                                                                               NULL,
490                                                                               false,
491                                                                               NULL, // No platform options
492                                                                               new_target_sp);
493             target = new_target_sp.get();
494             if (target == NULL || error.Fail())
495             {
496                 result.AppendError(error.AsCString("Error creating target"));
497                 return false;
498             }
499             m_interpreter.GetDebugger().GetTargetList().SetSelectedTarget(target);
500         }
501 
502         // Record the old executable module, we want to issue a warning if the process of attaching changed the
503         // current executable (like somebody said "file foo" then attached to a PID whose executable was bar.)
504 
505         ModuleSP old_exec_module_sp = target->GetExecutableModule();
506         ArchSpec old_arch_spec = target->GetArchitecture();
507 
508         if (command.GetArgumentCount())
509         {
510             result.AppendErrorWithFormat("Invalid arguments for '%s'.\nUsage: %s\n", m_cmd_name.c_str(), m_cmd_syntax.c_str());
511             result.SetStatus (eReturnStatusFailed);
512         }
513         else
514         {
515             if (state != eStateConnected)
516             {
517                 const char *plugin_name = m_options.attach_info.GetProcessPluginName();
518                 process = target->CreateProcess (m_interpreter.GetDebugger().GetListener(), plugin_name, NULL).get();
519             }
520 
521             if (process)
522             {
523                 Error error;
524                 // If no process info was specified, then use the target executable
525                 // name as the process to attach to by default
526                 if (!m_options.attach_info.ProcessInfoSpecified ())
527                 {
528                     if (old_exec_module_sp)
529                         m_options.attach_info.GetExecutableFile().GetFilename() = old_exec_module_sp->GetPlatformFileSpec().GetFilename();
530 
531                     if (!m_options.attach_info.ProcessInfoSpecified ())
532                     {
533                         error.SetErrorString ("no process specified, create a target with a file, or specify the --pid or --name command option");
534                     }
535                 }
536 
537                 if (error.Success())
538                 {
539                     // Update the execution context so the current target and process are now selected
540                     // in case we interrupt
541                     m_interpreter.UpdateExecutionContext(NULL);
542                     ListenerSP listener_sp (new Listener("lldb.CommandObjectProcessAttach.DoExecute.attach.hijack"));
543                     m_options.attach_info.SetHijackListener(listener_sp);
544                     process->HijackProcessEvents(listener_sp.get());
545                     error = process->Attach (m_options.attach_info);
546 
547                     if (error.Success())
548                     {
549                         result.SetStatus (eReturnStatusSuccessContinuingNoResult);
550                         StateType state = process->WaitForProcessToStop (NULL, NULL, false, listener_sp.get());
551 
552                         process->RestoreProcessEvents();
553 
554                         result.SetDidChangeProcessState (true);
555 
556                         if (state == eStateStopped)
557                         {
558                             result.AppendMessageWithFormat ("Process %" PRIu64 " %s\n", process->GetID(), StateAsCString (state));
559                             result.SetStatus (eReturnStatusSuccessFinishNoResult);
560                         }
561                         else
562                         {
563                             const char *exit_desc = process->GetExitDescription();
564                             if (exit_desc)
565                                 result.AppendErrorWithFormat ("attach failed: %s", exit_desc);
566                             else
567                                 result.AppendError ("attach failed: process did not stop (no such process or permission problem?)");
568                             process->Destroy();
569                             result.SetStatus (eReturnStatusFailed);
570                         }
571                     }
572                     else
573                     {
574                         result.AppendErrorWithFormat ("attach failed: %s\n", error.AsCString());
575                         result.SetStatus (eReturnStatusFailed);
576                     }
577                 }
578             }
579         }
580 
581         if (result.Succeeded())
582         {
583             // Okay, we're done.  Last step is to warn if the executable module has changed:
584             char new_path[PATH_MAX];
585             ModuleSP new_exec_module_sp (target->GetExecutableModule());
586             if (!old_exec_module_sp)
587             {
588                 // We might not have a module if we attached to a raw pid...
589                 if (new_exec_module_sp)
590                 {
591                     new_exec_module_sp->GetFileSpec().GetPath(new_path, PATH_MAX);
592                     result.AppendMessageWithFormat("Executable module set to \"%s\".\n", new_path);
593                 }
594             }
595             else if (old_exec_module_sp->GetFileSpec() != new_exec_module_sp->GetFileSpec())
596             {
597                 char old_path[PATH_MAX];
598 
599                 old_exec_module_sp->GetFileSpec().GetPath (old_path, PATH_MAX);
600                 new_exec_module_sp->GetFileSpec().GetPath (new_path, PATH_MAX);
601 
602                 result.AppendWarningWithFormat("Executable module changed from \"%s\" to \"%s\".\n",
603                                                     old_path, new_path);
604             }
605 
606             if (!old_arch_spec.IsValid())
607             {
608                 result.AppendMessageWithFormat ("Architecture set to: %s.\n", target->GetArchitecture().GetTriple().getTriple().c_str());
609             }
610             else if (!old_arch_spec.IsExactMatch(target->GetArchitecture()))
611             {
612                 result.AppendWarningWithFormat("Architecture changed from %s to %s.\n",
613                                                old_arch_spec.GetTriple().getTriple().c_str(),
614                                                target->GetArchitecture().GetTriple().getTriple().c_str());
615             }
616 
617             // This supports the use-case scenario of immediately continuing the process once attached.
618             if (m_options.attach_info.GetContinueOnceAttached())
619                 m_interpreter.HandleCommand("process continue", eLazyBoolNo, result);
620         }
621         return result.Succeeded();
622     }
623 
624     CommandOptions m_options;
625 };
626 
627 
628 OptionDefinition
629 CommandObjectProcessAttach::CommandOptions::g_option_table[] =
630 {
631 { LLDB_OPT_SET_ALL, false, "continue",'c', OptionParser::eNoArgument,         NULL, NULL, 0, eArgTypeNone,         "Immediately continue the process once attached."},
632 { LLDB_OPT_SET_ALL, false, "plugin",  'P', OptionParser::eRequiredArgument,   NULL, NULL, 0, eArgTypePlugin,       "Name of the process plugin you want to use."},
633 { LLDB_OPT_SET_1,   false, "pid",     'p', OptionParser::eRequiredArgument,   NULL, NULL, 0, eArgTypePid,          "The process ID of an existing process to attach to."},
634 { LLDB_OPT_SET_2,   false, "name",    'n', OptionParser::eRequiredArgument,   NULL, NULL, 0, eArgTypeProcessName,  "The name of the process to attach to."},
635 { LLDB_OPT_SET_2,   false, "include-existing", 'i', OptionParser::eNoArgument, NULL, NULL, 0, eArgTypeNone,         "Include existing processes when doing attach -w."},
636 { LLDB_OPT_SET_2,   false, "waitfor", 'w', OptionParser::eNoArgument,         NULL, NULL, 0, eArgTypeNone,         "Wait for the process with <process-name> to launch."},
637 { 0, false, NULL, 0, 0, NULL, NULL, 0, eArgTypeNone, NULL }
638 };
639 
640 //-------------------------------------------------------------------------
641 // CommandObjectProcessContinue
642 //-------------------------------------------------------------------------
643 #pragma mark CommandObjectProcessContinue
644 
645 class CommandObjectProcessContinue : public CommandObjectParsed
646 {
647 public:
648 
649     CommandObjectProcessContinue (CommandInterpreter &interpreter) :
650         CommandObjectParsed (interpreter,
651                              "process continue",
652                              "Continue execution of all threads in the current process.",
653                              "process continue",
654                              eFlagRequiresProcess       |
655                              eFlagTryTargetAPILock      |
656                              eFlagProcessMustBeLaunched |
657                              eFlagProcessMustBePaused   ),
658         m_options(interpreter)
659     {
660     }
661 
662 
663     ~CommandObjectProcessContinue ()
664     {
665     }
666 
667 protected:
668 
669     class CommandOptions : public Options
670     {
671     public:
672 
673         CommandOptions (CommandInterpreter &interpreter) :
674             Options(interpreter)
675         {
676             // Keep default values of all options in one place: OptionParsingStarting ()
677             OptionParsingStarting ();
678         }
679 
680         ~CommandOptions ()
681         {
682         }
683 
684         Error
685         SetOptionValue (uint32_t option_idx, const char *option_arg)
686         {
687             Error error;
688             const int short_option = m_getopt_table[option_idx].val;
689             bool success = false;
690             switch (short_option)
691             {
692                 case 'i':
693                     m_ignore = Args::StringToUInt32 (option_arg, 0, 0, &success);
694                     if (!success)
695                         error.SetErrorStringWithFormat ("invalid value for ignore option: \"%s\", should be a number.", option_arg);
696                     break;
697 
698                 default:
699                     error.SetErrorStringWithFormat("invalid short option character '%c'", short_option);
700                     break;
701             }
702             return error;
703         }
704 
705         void
706         OptionParsingStarting ()
707         {
708             m_ignore = 0;
709         }
710 
711         const OptionDefinition*
712         GetDefinitions ()
713         {
714             return g_option_table;
715         }
716 
717         // Options table: Required for subclasses of Options.
718 
719         static OptionDefinition g_option_table[];
720 
721         uint32_t m_ignore;
722     };
723 
724     bool
725     DoExecute (Args& command, CommandReturnObject &result)
726     {
727         Process *process = m_exe_ctx.GetProcessPtr();
728         bool synchronous_execution = m_interpreter.GetSynchronous ();
729         StateType state = process->GetState();
730         if (state == eStateStopped)
731         {
732             if (command.GetArgumentCount() != 0)
733             {
734                 result.AppendErrorWithFormat ("The '%s' command does not take any arguments.\n", m_cmd_name.c_str());
735                 result.SetStatus (eReturnStatusFailed);
736                 return false;
737             }
738 
739             if (m_options.m_ignore > 0)
740             {
741                 ThreadSP sel_thread_sp(process->GetThreadList().GetSelectedThread());
742                 if (sel_thread_sp)
743                 {
744                     StopInfoSP stop_info_sp = sel_thread_sp->GetStopInfo();
745                     if (stop_info_sp && stop_info_sp->GetStopReason() == eStopReasonBreakpoint)
746                     {
747                         lldb::break_id_t bp_site_id = (lldb::break_id_t)stop_info_sp->GetValue();
748                         BreakpointSiteSP bp_site_sp(process->GetBreakpointSiteList().FindByID(bp_site_id));
749                         if (bp_site_sp)
750                         {
751                             const size_t num_owners = bp_site_sp->GetNumberOfOwners();
752                             for (size_t i = 0; i < num_owners; i++)
753                             {
754                                 Breakpoint &bp_ref = bp_site_sp->GetOwnerAtIndex(i)->GetBreakpoint();
755                                 if (!bp_ref.IsInternal())
756                                 {
757                                     bp_ref.SetIgnoreCount(m_options.m_ignore);
758                                 }
759                             }
760                         }
761                     }
762                 }
763             }
764 
765             {  // Scope for thread list mutex:
766                 Mutex::Locker locker (process->GetThreadList().GetMutex());
767                 const uint32_t num_threads = process->GetThreadList().GetSize();
768 
769                 // Set the actions that the threads should each take when resuming
770                 for (uint32_t idx=0; idx<num_threads; ++idx)
771                 {
772                     const bool override_suspend = false;
773                     process->GetThreadList().GetThreadAtIndex(idx)->SetResumeState (eStateRunning, override_suspend);
774                 }
775             }
776 
777             Error error(process->Resume());
778 
779             if (error.Success())
780             {
781                 // There is a race condition where this thread will return up the call stack to the main command
782                 // handler and show an (lldb) prompt before HandlePrivateEvent (from PrivateStateThread) has
783                 // a chance to call PushProcessIOHandler().
784                 process->SyncIOHandler(2000);
785 
786                 result.AppendMessageWithFormat ("Process %" PRIu64 " resuming\n", process->GetID());
787                 if (synchronous_execution)
788                 {
789                     state = process->WaitForProcessToStop (NULL);
790 
791                     result.SetDidChangeProcessState (true);
792                     result.AppendMessageWithFormat ("Process %" PRIu64 " %s\n", process->GetID(), StateAsCString (state));
793                     result.SetStatus (eReturnStatusSuccessFinishNoResult);
794                 }
795                 else
796                 {
797                     result.SetStatus (eReturnStatusSuccessContinuingNoResult);
798                 }
799             }
800             else
801             {
802                 result.AppendErrorWithFormat("Failed to resume process: %s.\n", error.AsCString());
803                 result.SetStatus (eReturnStatusFailed);
804             }
805         }
806         else
807         {
808             result.AppendErrorWithFormat ("Process cannot be continued from its current state (%s).\n",
809                                          StateAsCString(state));
810             result.SetStatus (eReturnStatusFailed);
811         }
812         return result.Succeeded();
813     }
814 
815     Options *
816     GetOptions ()
817     {
818         return &m_options;
819     }
820 
821     CommandOptions m_options;
822 
823 };
824 
825 OptionDefinition
826 CommandObjectProcessContinue::CommandOptions::g_option_table[] =
827 {
828 { LLDB_OPT_SET_ALL, false, "ignore-count",'i', OptionParser::eRequiredArgument,         NULL, NULL, 0, eArgTypeUnsignedInteger,
829                            "Ignore <N> crossings of the breakpoint (if it exists) for the currently selected thread."},
830 { 0, false, NULL, 0, 0, NULL, NULL, 0, eArgTypeNone, NULL }
831 };
832 
833 //-------------------------------------------------------------------------
834 // CommandObjectProcessDetach
835 //-------------------------------------------------------------------------
836 #pragma mark CommandObjectProcessDetach
837 
838 class CommandObjectProcessDetach : public CommandObjectParsed
839 {
840 public:
841     class CommandOptions : public Options
842     {
843     public:
844 
845         CommandOptions (CommandInterpreter &interpreter) :
846             Options (interpreter)
847         {
848             OptionParsingStarting ();
849         }
850 
851         ~CommandOptions ()
852         {
853         }
854 
855         Error
856         SetOptionValue (uint32_t option_idx, const char *option_arg)
857         {
858             Error error;
859             const int short_option = m_getopt_table[option_idx].val;
860 
861             switch (short_option)
862             {
863                 case 's':
864                     bool tmp_result;
865                     bool success;
866                     tmp_result = Args::StringToBoolean(option_arg, false, &success);
867                     if (!success)
868                         error.SetErrorStringWithFormat("invalid boolean option: \"%s\"", option_arg);
869                     else
870                     {
871                         if (tmp_result)
872                             m_keep_stopped = eLazyBoolYes;
873                         else
874                             m_keep_stopped = eLazyBoolNo;
875                     }
876                     break;
877                 default:
878                     error.SetErrorStringWithFormat("invalid short option character '%c'", short_option);
879                     break;
880             }
881             return error;
882         }
883 
884         void
885         OptionParsingStarting ()
886         {
887             m_keep_stopped = eLazyBoolCalculate;
888         }
889 
890         const OptionDefinition*
891         GetDefinitions ()
892         {
893             return g_option_table;
894         }
895 
896         // Options table: Required for subclasses of Options.
897 
898         static OptionDefinition g_option_table[];
899 
900         // Instance variables to hold the values for command options.
901         LazyBool m_keep_stopped;
902     };
903 
904     CommandObjectProcessDetach (CommandInterpreter &interpreter) :
905         CommandObjectParsed (interpreter,
906                              "process detach",
907                              "Detach from the current process being debugged.",
908                              "process detach",
909                              eFlagRequiresProcess      |
910                              eFlagTryTargetAPILock     |
911                              eFlagProcessMustBeLaunched),
912         m_options(interpreter)
913     {
914     }
915 
916     ~CommandObjectProcessDetach ()
917     {
918     }
919 
920     Options *
921     GetOptions ()
922     {
923         return &m_options;
924     }
925 
926 
927 protected:
928     bool
929     DoExecute (Args& command, CommandReturnObject &result)
930     {
931         Process *process = m_exe_ctx.GetProcessPtr();
932         // FIXME: This will be a Command Option:
933         bool keep_stopped;
934         if (m_options.m_keep_stopped == eLazyBoolCalculate)
935         {
936             // Check the process default:
937             if (process->GetDetachKeepsStopped())
938                 keep_stopped = true;
939             else
940                 keep_stopped = false;
941         }
942         else if (m_options.m_keep_stopped == eLazyBoolYes)
943             keep_stopped = true;
944         else
945             keep_stopped = false;
946 
947         Error error (process->Detach(keep_stopped));
948         if (error.Success())
949         {
950             result.SetStatus (eReturnStatusSuccessFinishResult);
951         }
952         else
953         {
954             result.AppendErrorWithFormat ("Detach failed: %s\n", error.AsCString());
955             result.SetStatus (eReturnStatusFailed);
956             return false;
957         }
958         return result.Succeeded();
959     }
960 
961     CommandOptions m_options;
962 };
963 
964 OptionDefinition
965 CommandObjectProcessDetach::CommandOptions::g_option_table[] =
966 {
967 { LLDB_OPT_SET_1, false, "keep-stopped",   's', OptionParser::eRequiredArgument, NULL, NULL, 0, eArgTypeBoolean, "Whether or not the process should be kept stopped on detach (if possible)." },
968 { 0, false, NULL, 0, 0, NULL, NULL, 0, eArgTypeNone, NULL }
969 };
970 
971 //-------------------------------------------------------------------------
972 // CommandObjectProcessConnect
973 //-------------------------------------------------------------------------
974 #pragma mark CommandObjectProcessConnect
975 
976 class CommandObjectProcessConnect : public CommandObjectParsed
977 {
978 public:
979 
980     class CommandOptions : public Options
981     {
982     public:
983 
984         CommandOptions (CommandInterpreter &interpreter) :
985             Options(interpreter)
986         {
987             // Keep default values of all options in one place: OptionParsingStarting ()
988             OptionParsingStarting ();
989         }
990 
991         ~CommandOptions ()
992         {
993         }
994 
995         Error
996         SetOptionValue (uint32_t option_idx, const char *option_arg)
997         {
998             Error error;
999             const int short_option = m_getopt_table[option_idx].val;
1000 
1001             switch (short_option)
1002             {
1003             case 'p':
1004                 plugin_name.assign (option_arg);
1005                 break;
1006 
1007             default:
1008                 error.SetErrorStringWithFormat("invalid short option character '%c'", short_option);
1009                 break;
1010             }
1011             return error;
1012         }
1013 
1014         void
1015         OptionParsingStarting ()
1016         {
1017             plugin_name.clear();
1018         }
1019 
1020         const OptionDefinition*
1021         GetDefinitions ()
1022         {
1023             return g_option_table;
1024         }
1025 
1026         // Options table: Required for subclasses of Options.
1027 
1028         static OptionDefinition g_option_table[];
1029 
1030         // Instance variables to hold the values for command options.
1031 
1032         std::string plugin_name;
1033     };
1034 
1035     CommandObjectProcessConnect (CommandInterpreter &interpreter) :
1036         CommandObjectParsed (interpreter,
1037                              "process connect",
1038                              "Connect to a remote debug service.",
1039                              "process connect <remote-url>",
1040                              0),
1041         m_options (interpreter)
1042     {
1043     }
1044 
1045     ~CommandObjectProcessConnect ()
1046     {
1047     }
1048 
1049 
1050     Options *
1051     GetOptions ()
1052     {
1053         return &m_options;
1054     }
1055 
1056 protected:
1057     bool
1058     DoExecute (Args& command,
1059              CommandReturnObject &result)
1060     {
1061 
1062         TargetSP target_sp (m_interpreter.GetDebugger().GetSelectedTarget());
1063         Error error;
1064         Process *process = m_exe_ctx.GetProcessPtr();
1065         if (process)
1066         {
1067             if (process->IsAlive())
1068             {
1069                 result.AppendErrorWithFormat ("Process %" PRIu64 " is currently being debugged, kill the process before connecting.\n",
1070                                               process->GetID());
1071                 result.SetStatus (eReturnStatusFailed);
1072                 return false;
1073             }
1074         }
1075 
1076         if (!target_sp)
1077         {
1078             // If there isn't a current target create one.
1079 
1080             error = m_interpreter.GetDebugger().GetTargetList().CreateTarget (m_interpreter.GetDebugger(),
1081                                                                               NULL,
1082                                                                               NULL,
1083                                                                               false,
1084                                                                               NULL, // No platform options
1085                                                                               target_sp);
1086             if (!target_sp || error.Fail())
1087             {
1088                 result.AppendError(error.AsCString("Error creating target"));
1089                 result.SetStatus (eReturnStatusFailed);
1090                 return false;
1091             }
1092             m_interpreter.GetDebugger().GetTargetList().SetSelectedTarget(target_sp.get());
1093         }
1094 
1095         if (command.GetArgumentCount() == 1)
1096         {
1097             const char *plugin_name = NULL;
1098             if (!m_options.plugin_name.empty())
1099                 plugin_name = m_options.plugin_name.c_str();
1100 
1101             const char *remote_url = command.GetArgumentAtIndex(0);
1102             process = target_sp->CreateProcess (m_interpreter.GetDebugger().GetListener(), plugin_name, NULL).get();
1103 
1104             if (process)
1105             {
1106                 error = process->ConnectRemote (process->GetTarget().GetDebugger().GetOutputFile().get(), remote_url);
1107 
1108                 if (error.Fail())
1109                 {
1110                     result.AppendError(error.AsCString("Remote connect failed"));
1111                     result.SetStatus (eReturnStatusFailed);
1112                     target_sp->DeleteCurrentProcess();
1113                     return false;
1114                 }
1115             }
1116             else
1117             {
1118                 result.AppendErrorWithFormat ("Unable to find process plug-in for remote URL '%s'.\nPlease specify a process plug-in name with the --plugin option, or specify an object file using the \"file\" command.\n",
1119                                               remote_url);
1120                 result.SetStatus (eReturnStatusFailed);
1121             }
1122         }
1123         else
1124         {
1125             result.AppendErrorWithFormat ("'%s' takes exactly one argument:\nUsage: %s\n",
1126                                           m_cmd_name.c_str(),
1127                                           m_cmd_syntax.c_str());
1128             result.SetStatus (eReturnStatusFailed);
1129         }
1130         return result.Succeeded();
1131     }
1132 
1133     CommandOptions m_options;
1134 };
1135 
1136 OptionDefinition
1137 CommandObjectProcessConnect::CommandOptions::g_option_table[] =
1138 {
1139     { LLDB_OPT_SET_ALL, false, "plugin", 'p', OptionParser::eRequiredArgument, NULL, NULL, 0, eArgTypePlugin, "Name of the process plugin you want to use."},
1140     { 0,                false, NULL,      0 , 0,                 NULL, NULL, 0, eArgTypeNone,   NULL }
1141 };
1142 
1143 //-------------------------------------------------------------------------
1144 // CommandObjectProcessPlugin
1145 //-------------------------------------------------------------------------
1146 #pragma mark CommandObjectProcessPlugin
1147 
1148 class CommandObjectProcessPlugin : public CommandObjectProxy
1149 {
1150 public:
1151 
1152     CommandObjectProcessPlugin (CommandInterpreter &interpreter) :
1153         CommandObjectProxy (interpreter,
1154                             "process plugin",
1155                             "Send a custom command to the current process plug-in.",
1156                             "process plugin <args>",
1157                             0)
1158     {
1159     }
1160 
1161     ~CommandObjectProcessPlugin ()
1162     {
1163     }
1164 
1165     virtual CommandObject *
1166     GetProxyCommandObject()
1167     {
1168         Process *process = m_interpreter.GetExecutionContext().GetProcessPtr();
1169         if (process)
1170             return process->GetPluginCommandObject();
1171         return NULL;
1172     }
1173 };
1174 
1175 
1176 //-------------------------------------------------------------------------
1177 // CommandObjectProcessLoad
1178 //-------------------------------------------------------------------------
1179 #pragma mark CommandObjectProcessLoad
1180 
1181 class CommandObjectProcessLoad : public CommandObjectParsed
1182 {
1183 public:
1184 
1185     CommandObjectProcessLoad (CommandInterpreter &interpreter) :
1186         CommandObjectParsed (interpreter,
1187                              "process load",
1188                              "Load a shared library into the current process.",
1189                              "process load <filename> [<filename> ...]",
1190                              eFlagRequiresProcess       |
1191                              eFlagTryTargetAPILock      |
1192                              eFlagProcessMustBeLaunched |
1193                              eFlagProcessMustBePaused   )
1194     {
1195     }
1196 
1197     ~CommandObjectProcessLoad ()
1198     {
1199     }
1200 
1201 protected:
1202     bool
1203     DoExecute (Args& command,
1204              CommandReturnObject &result)
1205     {
1206         Process *process = m_exe_ctx.GetProcessPtr();
1207 
1208         const size_t argc = command.GetArgumentCount();
1209 
1210         for (uint32_t i=0; i<argc; ++i)
1211         {
1212             Error error;
1213             const char *image_path = command.GetArgumentAtIndex(i);
1214             FileSpec image_spec (image_path, false);
1215             process->GetTarget().GetPlatform()->ResolveRemotePath(image_spec, image_spec);
1216             uint32_t image_token = process->LoadImage(image_spec, error);
1217             if (image_token != LLDB_INVALID_IMAGE_TOKEN)
1218             {
1219                 result.AppendMessageWithFormat ("Loading \"%s\"...ok\nImage %u loaded.\n", image_path, image_token);
1220                 result.SetStatus (eReturnStatusSuccessFinishResult);
1221             }
1222             else
1223             {
1224                 result.AppendErrorWithFormat ("failed to load '%s': %s", image_path, error.AsCString());
1225                 result.SetStatus (eReturnStatusFailed);
1226             }
1227         }
1228         return result.Succeeded();
1229     }
1230 };
1231 
1232 
1233 //-------------------------------------------------------------------------
1234 // CommandObjectProcessUnload
1235 //-------------------------------------------------------------------------
1236 #pragma mark CommandObjectProcessUnload
1237 
1238 class CommandObjectProcessUnload : public CommandObjectParsed
1239 {
1240 public:
1241 
1242     CommandObjectProcessUnload (CommandInterpreter &interpreter) :
1243         CommandObjectParsed (interpreter,
1244                              "process unload",
1245                              "Unload a shared library from the current process using the index returned by a previous call to \"process load\".",
1246                              "process unload <index>",
1247                              eFlagRequiresProcess       |
1248                              eFlagTryTargetAPILock      |
1249                              eFlagProcessMustBeLaunched |
1250                              eFlagProcessMustBePaused   )
1251     {
1252     }
1253 
1254     ~CommandObjectProcessUnload ()
1255     {
1256     }
1257 
1258 protected:
1259     bool
1260     DoExecute (Args& command,
1261              CommandReturnObject &result)
1262     {
1263         Process *process = m_exe_ctx.GetProcessPtr();
1264 
1265         const size_t argc = command.GetArgumentCount();
1266 
1267         for (uint32_t i=0; i<argc; ++i)
1268         {
1269             const char *image_token_cstr = command.GetArgumentAtIndex(i);
1270             uint32_t image_token = Args::StringToUInt32(image_token_cstr, LLDB_INVALID_IMAGE_TOKEN, 0);
1271             if (image_token == LLDB_INVALID_IMAGE_TOKEN)
1272             {
1273                 result.AppendErrorWithFormat ("invalid image index argument '%s'", image_token_cstr);
1274                 result.SetStatus (eReturnStatusFailed);
1275                 break;
1276             }
1277             else
1278             {
1279                 Error error (process->UnloadImage(image_token));
1280                 if (error.Success())
1281                 {
1282                     result.AppendMessageWithFormat ("Unloading shared library with index %u...ok\n", image_token);
1283                     result.SetStatus (eReturnStatusSuccessFinishResult);
1284                 }
1285                 else
1286                 {
1287                     result.AppendErrorWithFormat ("failed to unload image: %s", error.AsCString());
1288                     result.SetStatus (eReturnStatusFailed);
1289                     break;
1290                 }
1291             }
1292         }
1293         return result.Succeeded();
1294     }
1295 };
1296 
1297 //-------------------------------------------------------------------------
1298 // CommandObjectProcessSignal
1299 //-------------------------------------------------------------------------
1300 #pragma mark CommandObjectProcessSignal
1301 
1302 class CommandObjectProcessSignal : public CommandObjectParsed
1303 {
1304 public:
1305 
1306     CommandObjectProcessSignal (CommandInterpreter &interpreter) :
1307         CommandObjectParsed (interpreter,
1308                              "process signal",
1309                              "Send a UNIX signal to the current process being debugged.",
1310                              NULL,
1311                              eFlagRequiresProcess | eFlagTryTargetAPILock)
1312     {
1313         CommandArgumentEntry arg;
1314         CommandArgumentData signal_arg;
1315 
1316         // Define the first (and only) variant of this arg.
1317         signal_arg.arg_type = eArgTypeUnixSignal;
1318         signal_arg.arg_repetition = eArgRepeatPlain;
1319 
1320         // There is only one variant this argument could be; put it into the argument entry.
1321         arg.push_back (signal_arg);
1322 
1323         // Push the data for the first argument into the m_arguments vector.
1324         m_arguments.push_back (arg);
1325     }
1326 
1327     ~CommandObjectProcessSignal ()
1328     {
1329     }
1330 
1331 protected:
1332     bool
1333     DoExecute (Args& command,
1334              CommandReturnObject &result)
1335     {
1336         Process *process = m_exe_ctx.GetProcessPtr();
1337 
1338         if (command.GetArgumentCount() == 1)
1339         {
1340             int signo = LLDB_INVALID_SIGNAL_NUMBER;
1341 
1342             const char *signal_name = command.GetArgumentAtIndex(0);
1343             if (::isxdigit (signal_name[0]))
1344                 signo = Args::StringToSInt32(signal_name, LLDB_INVALID_SIGNAL_NUMBER, 0);
1345             else
1346                 signo = process->GetUnixSignals().GetSignalNumberFromName (signal_name);
1347 
1348             if (signo == LLDB_INVALID_SIGNAL_NUMBER)
1349             {
1350                 result.AppendErrorWithFormat ("Invalid signal argument '%s'.\n", command.GetArgumentAtIndex(0));
1351                 result.SetStatus (eReturnStatusFailed);
1352             }
1353             else
1354             {
1355                 Error error (process->Signal (signo));
1356                 if (error.Success())
1357                 {
1358                     result.SetStatus (eReturnStatusSuccessFinishResult);
1359                 }
1360                 else
1361                 {
1362                     result.AppendErrorWithFormat ("Failed to send signal %i: %s\n", signo, error.AsCString());
1363                     result.SetStatus (eReturnStatusFailed);
1364                 }
1365             }
1366         }
1367         else
1368         {
1369             result.AppendErrorWithFormat("'%s' takes exactly one signal number argument:\nUsage: %s\n", m_cmd_name.c_str(),
1370                                         m_cmd_syntax.c_str());
1371             result.SetStatus (eReturnStatusFailed);
1372         }
1373         return result.Succeeded();
1374     }
1375 };
1376 
1377 
1378 //-------------------------------------------------------------------------
1379 // CommandObjectProcessInterrupt
1380 //-------------------------------------------------------------------------
1381 #pragma mark CommandObjectProcessInterrupt
1382 
1383 class CommandObjectProcessInterrupt : public CommandObjectParsed
1384 {
1385 public:
1386 
1387 
1388     CommandObjectProcessInterrupt (CommandInterpreter &interpreter) :
1389         CommandObjectParsed (interpreter,
1390                              "process interrupt",
1391                              "Interrupt the current process being debugged.",
1392                              "process interrupt",
1393                              eFlagRequiresProcess      |
1394                              eFlagTryTargetAPILock     |
1395                              eFlagProcessMustBeLaunched)
1396     {
1397     }
1398 
1399     ~CommandObjectProcessInterrupt ()
1400     {
1401     }
1402 
1403 protected:
1404     bool
1405     DoExecute (Args& command,
1406                CommandReturnObject &result)
1407     {
1408         Process *process = m_exe_ctx.GetProcessPtr();
1409         if (process == NULL)
1410         {
1411             result.AppendError ("no process to halt");
1412             result.SetStatus (eReturnStatusFailed);
1413             return false;
1414         }
1415 
1416         if (command.GetArgumentCount() == 0)
1417         {
1418             bool clear_thread_plans = true;
1419             Error error(process->Halt (clear_thread_plans));
1420             if (error.Success())
1421             {
1422                 result.SetStatus (eReturnStatusSuccessFinishResult);
1423             }
1424             else
1425             {
1426                 result.AppendErrorWithFormat ("Failed to halt process: %s\n", error.AsCString());
1427                 result.SetStatus (eReturnStatusFailed);
1428             }
1429         }
1430         else
1431         {
1432             result.AppendErrorWithFormat("'%s' takes no arguments:\nUsage: %s\n",
1433                                         m_cmd_name.c_str(),
1434                                         m_cmd_syntax.c_str());
1435             result.SetStatus (eReturnStatusFailed);
1436         }
1437         return result.Succeeded();
1438     }
1439 };
1440 
1441 //-------------------------------------------------------------------------
1442 // CommandObjectProcessKill
1443 //-------------------------------------------------------------------------
1444 #pragma mark CommandObjectProcessKill
1445 
1446 class CommandObjectProcessKill : public CommandObjectParsed
1447 {
1448 public:
1449 
1450     CommandObjectProcessKill (CommandInterpreter &interpreter) :
1451         CommandObjectParsed (interpreter,
1452                              "process kill",
1453                              "Terminate the current process being debugged.",
1454                              "process kill",
1455                              eFlagRequiresProcess      |
1456                              eFlagTryTargetAPILock     |
1457                              eFlagProcessMustBeLaunched)
1458     {
1459     }
1460 
1461     ~CommandObjectProcessKill ()
1462     {
1463     }
1464 
1465 protected:
1466     bool
1467     DoExecute (Args& command,
1468              CommandReturnObject &result)
1469     {
1470         Process *process = m_exe_ctx.GetProcessPtr();
1471         if (process == NULL)
1472         {
1473             result.AppendError ("no process to kill");
1474             result.SetStatus (eReturnStatusFailed);
1475             return false;
1476         }
1477 
1478         if (command.GetArgumentCount() == 0)
1479         {
1480             Error error (process->Destroy());
1481             if (error.Success())
1482             {
1483                 result.SetStatus (eReturnStatusSuccessFinishResult);
1484             }
1485             else
1486             {
1487                 result.AppendErrorWithFormat ("Failed to kill process: %s\n", error.AsCString());
1488                 result.SetStatus (eReturnStatusFailed);
1489             }
1490         }
1491         else
1492         {
1493             result.AppendErrorWithFormat("'%s' takes no arguments:\nUsage: %s\n",
1494                                         m_cmd_name.c_str(),
1495                                         m_cmd_syntax.c_str());
1496             result.SetStatus (eReturnStatusFailed);
1497         }
1498         return result.Succeeded();
1499     }
1500 };
1501 
1502 //-------------------------------------------------------------------------
1503 // CommandObjectProcessSaveCore
1504 //-------------------------------------------------------------------------
1505 #pragma mark CommandObjectProcessSaveCore
1506 
1507 class CommandObjectProcessSaveCore : public CommandObjectParsed
1508 {
1509 public:
1510 
1511     CommandObjectProcessSaveCore (CommandInterpreter &interpreter) :
1512     CommandObjectParsed (interpreter,
1513                          "process save-core",
1514                          "Save the current process as a core file using an appropriate file type.",
1515                          "process save-core FILE",
1516                          eFlagRequiresProcess      |
1517                          eFlagTryTargetAPILock     |
1518                          eFlagProcessMustBeLaunched)
1519     {
1520     }
1521 
1522     ~CommandObjectProcessSaveCore ()
1523     {
1524     }
1525 
1526 protected:
1527     bool
1528     DoExecute (Args& command,
1529                CommandReturnObject &result)
1530     {
1531         ProcessSP process_sp = m_exe_ctx.GetProcessSP();
1532         if (process_sp)
1533         {
1534             if (command.GetArgumentCount() == 1)
1535             {
1536                 FileSpec output_file(command.GetArgumentAtIndex(0), false);
1537                 Error error = PluginManager::SaveCore(process_sp, output_file);
1538                 if (error.Success())
1539                 {
1540                     result.SetStatus (eReturnStatusSuccessFinishResult);
1541                 }
1542                 else
1543                 {
1544                     result.AppendErrorWithFormat ("Failed to save core file for process: %s\n", error.AsCString());
1545                     result.SetStatus (eReturnStatusFailed);
1546                 }
1547             }
1548             else
1549             {
1550                 result.AppendErrorWithFormat ("'%s' takes one arguments:\nUsage: %s\n",
1551                                               m_cmd_name.c_str(),
1552                                               m_cmd_syntax.c_str());
1553                 result.SetStatus (eReturnStatusFailed);
1554             }
1555         }
1556         else
1557         {
1558             result.AppendError ("invalid process");
1559             result.SetStatus (eReturnStatusFailed);
1560             return false;
1561         }
1562 
1563         return result.Succeeded();
1564     }
1565 };
1566 
1567 //-------------------------------------------------------------------------
1568 // CommandObjectProcessStatus
1569 //-------------------------------------------------------------------------
1570 #pragma mark CommandObjectProcessStatus
1571 
1572 class CommandObjectProcessStatus : public CommandObjectParsed
1573 {
1574 public:
1575     CommandObjectProcessStatus (CommandInterpreter &interpreter) :
1576         CommandObjectParsed (interpreter,
1577                              "process status",
1578                              "Show the current status and location of executing process.",
1579                              "process status",
1580                              eFlagRequiresProcess | eFlagTryTargetAPILock)
1581     {
1582     }
1583 
1584     ~CommandObjectProcessStatus()
1585     {
1586     }
1587 
1588 
1589     bool
1590     DoExecute (Args& command, CommandReturnObject &result)
1591     {
1592         Stream &strm = result.GetOutputStream();
1593         result.SetStatus (eReturnStatusSuccessFinishNoResult);
1594         // No need to check "process" for validity as eFlagRequiresProcess ensures it is valid
1595         Process *process = m_exe_ctx.GetProcessPtr();
1596         const bool only_threads_with_stop_reason = true;
1597         const uint32_t start_frame = 0;
1598         const uint32_t num_frames = 1;
1599         const uint32_t num_frames_with_source = 1;
1600         process->GetStatus(strm);
1601         process->GetThreadStatus (strm,
1602                                   only_threads_with_stop_reason,
1603                                   start_frame,
1604                                   num_frames,
1605                                   num_frames_with_source);
1606         return result.Succeeded();
1607     }
1608 };
1609 
1610 //-------------------------------------------------------------------------
1611 // CommandObjectProcessHandle
1612 //-------------------------------------------------------------------------
1613 #pragma mark CommandObjectProcessHandle
1614 
1615 class CommandObjectProcessHandle : public CommandObjectParsed
1616 {
1617 public:
1618 
1619     class CommandOptions : public Options
1620     {
1621     public:
1622 
1623         CommandOptions (CommandInterpreter &interpreter) :
1624             Options (interpreter)
1625         {
1626             OptionParsingStarting ();
1627         }
1628 
1629         ~CommandOptions ()
1630         {
1631         }
1632 
1633         Error
1634         SetOptionValue (uint32_t option_idx, const char *option_arg)
1635         {
1636             Error error;
1637             const int short_option = m_getopt_table[option_idx].val;
1638 
1639             switch (short_option)
1640             {
1641                 case 's':
1642                     stop = option_arg;
1643                     break;
1644                 case 'n':
1645                     notify = option_arg;
1646                     break;
1647                 case 'p':
1648                     pass = option_arg;
1649                     break;
1650                 default:
1651                     error.SetErrorStringWithFormat("invalid short option character '%c'", short_option);
1652                     break;
1653             }
1654             return error;
1655         }
1656 
1657         void
1658         OptionParsingStarting ()
1659         {
1660             stop.clear();
1661             notify.clear();
1662             pass.clear();
1663         }
1664 
1665         const OptionDefinition*
1666         GetDefinitions ()
1667         {
1668             return g_option_table;
1669         }
1670 
1671         // Options table: Required for subclasses of Options.
1672 
1673         static OptionDefinition g_option_table[];
1674 
1675         // Instance variables to hold the values for command options.
1676 
1677         std::string stop;
1678         std::string notify;
1679         std::string pass;
1680     };
1681 
1682 
1683     CommandObjectProcessHandle (CommandInterpreter &interpreter) :
1684         CommandObjectParsed (interpreter,
1685                              "process handle",
1686                              "Show or update what the process and debugger should do with various signals received from the OS.",
1687                              NULL),
1688         m_options (interpreter)
1689     {
1690         SetHelpLong ("If no signals are specified, update them all.  If no update option is specified, list the current values.\n");
1691         CommandArgumentEntry arg;
1692         CommandArgumentData signal_arg;
1693 
1694         signal_arg.arg_type = eArgTypeUnixSignal;
1695         signal_arg.arg_repetition = eArgRepeatStar;
1696 
1697         arg.push_back (signal_arg);
1698 
1699         m_arguments.push_back (arg);
1700     }
1701 
1702     ~CommandObjectProcessHandle ()
1703     {
1704     }
1705 
1706     Options *
1707     GetOptions ()
1708     {
1709         return &m_options;
1710     }
1711 
1712     bool
1713     VerifyCommandOptionValue (const std::string &option, int &real_value)
1714     {
1715         bool okay = true;
1716 
1717         bool success = false;
1718         bool tmp_value = Args::StringToBoolean (option.c_str(), false, &success);
1719 
1720         if (success && tmp_value)
1721             real_value = 1;
1722         else if (success && !tmp_value)
1723             real_value = 0;
1724         else
1725         {
1726             // If the value isn't 'true' or 'false', it had better be 0 or 1.
1727             real_value = Args::StringToUInt32 (option.c_str(), 3);
1728             if (real_value != 0 && real_value != 1)
1729                 okay = false;
1730         }
1731 
1732         return okay;
1733     }
1734 
1735     void
1736     PrintSignalHeader (Stream &str)
1737     {
1738         str.Printf ("NAME        PASS   STOP   NOTIFY\n");
1739         str.Printf ("==========  =====  =====  ======\n");
1740     }
1741 
1742     void
1743     PrintSignal (Stream &str, int32_t signo, const char *sig_name, UnixSignals &signals)
1744     {
1745         bool stop;
1746         bool suppress;
1747         bool notify;
1748 
1749         str.Printf ("%-10s  ", sig_name);
1750         if (signals.GetSignalInfo (signo, suppress, stop, notify))
1751         {
1752             bool pass = !suppress;
1753             str.Printf ("%s  %s  %s",
1754                         (pass ? "true " : "false"),
1755                         (stop ? "true " : "false"),
1756                         (notify ? "true " : "false"));
1757         }
1758         str.Printf ("\n");
1759     }
1760 
1761     void
1762     PrintSignalInformation (Stream &str, Args &signal_args, int num_valid_signals, UnixSignals &signals)
1763     {
1764         PrintSignalHeader (str);
1765 
1766         if (num_valid_signals > 0)
1767         {
1768             size_t num_args = signal_args.GetArgumentCount();
1769             for (size_t i = 0; i < num_args; ++i)
1770             {
1771                 int32_t signo = signals.GetSignalNumberFromName (signal_args.GetArgumentAtIndex (i));
1772                 if (signo != LLDB_INVALID_SIGNAL_NUMBER)
1773                     PrintSignal (str, signo, signal_args.GetArgumentAtIndex (i), signals);
1774             }
1775         }
1776         else // Print info for ALL signals
1777         {
1778             int32_t signo = signals.GetFirstSignalNumber();
1779             while (signo != LLDB_INVALID_SIGNAL_NUMBER)
1780             {
1781                 PrintSignal (str, signo, signals.GetSignalAsCString (signo), signals);
1782                 signo = signals.GetNextSignalNumber (signo);
1783             }
1784         }
1785     }
1786 
1787 protected:
1788     bool
1789     DoExecute (Args &signal_args, CommandReturnObject &result)
1790     {
1791         TargetSP target_sp = m_interpreter.GetDebugger().GetSelectedTarget();
1792 
1793         if (!target_sp)
1794         {
1795             result.AppendError ("No current target;"
1796                                 " cannot handle signals until you have a valid target and process.\n");
1797             result.SetStatus (eReturnStatusFailed);
1798             return false;
1799         }
1800 
1801         ProcessSP process_sp = target_sp->GetProcessSP();
1802 
1803         if (!process_sp)
1804         {
1805             result.AppendError ("No current process; cannot handle signals until you have a valid process.\n");
1806             result.SetStatus (eReturnStatusFailed);
1807             return false;
1808         }
1809 
1810         int stop_action = -1;   // -1 means leave the current setting alone
1811         int pass_action = -1;   // -1 means leave the current setting alone
1812         int notify_action = -1; // -1 means leave the current setting alone
1813 
1814         if (! m_options.stop.empty()
1815             && ! VerifyCommandOptionValue (m_options.stop, stop_action))
1816         {
1817             result.AppendError ("Invalid argument for command option --stop; must be true or false.\n");
1818             result.SetStatus (eReturnStatusFailed);
1819             return false;
1820         }
1821 
1822         if (! m_options.notify.empty()
1823             && ! VerifyCommandOptionValue (m_options.notify, notify_action))
1824         {
1825             result.AppendError ("Invalid argument for command option --notify; must be true or false.\n");
1826             result.SetStatus (eReturnStatusFailed);
1827             return false;
1828         }
1829 
1830         if (! m_options.pass.empty()
1831             && ! VerifyCommandOptionValue (m_options.pass, pass_action))
1832         {
1833             result.AppendError ("Invalid argument for command option --pass; must be true or false.\n");
1834             result.SetStatus (eReturnStatusFailed);
1835             return false;
1836         }
1837 
1838         size_t num_args = signal_args.GetArgumentCount();
1839         UnixSignals &signals = process_sp->GetUnixSignals();
1840         int num_signals_set = 0;
1841 
1842         if (num_args > 0)
1843         {
1844             for (size_t i = 0; i < num_args; ++i)
1845             {
1846                 int32_t signo = signals.GetSignalNumberFromName (signal_args.GetArgumentAtIndex (i));
1847                 if (signo != LLDB_INVALID_SIGNAL_NUMBER)
1848                 {
1849                     // Casting the actions as bools here should be okay, because VerifyCommandOptionValue guarantees
1850                     // the value is either 0 or 1.
1851                     if (stop_action != -1)
1852                         signals.SetShouldStop (signo, (bool) stop_action);
1853                     if (pass_action != -1)
1854                     {
1855                         bool suppress = ! ((bool) pass_action);
1856                         signals.SetShouldSuppress (signo, suppress);
1857                     }
1858                     if (notify_action != -1)
1859                         signals.SetShouldNotify (signo, (bool) notify_action);
1860                     ++num_signals_set;
1861                 }
1862                 else
1863                 {
1864                     result.AppendErrorWithFormat ("Invalid signal name '%s'\n", signal_args.GetArgumentAtIndex (i));
1865                 }
1866             }
1867         }
1868         else
1869         {
1870             // No signal specified, if any command options were specified, update ALL signals.
1871             if ((notify_action != -1) || (stop_action != -1) || (pass_action != -1))
1872             {
1873                 if (m_interpreter.Confirm ("Do you really want to update all the signals?", false))
1874                 {
1875                     int32_t signo = signals.GetFirstSignalNumber();
1876                     while (signo != LLDB_INVALID_SIGNAL_NUMBER)
1877                     {
1878                         if (notify_action != -1)
1879                             signals.SetShouldNotify (signo, (bool) notify_action);
1880                         if (stop_action != -1)
1881                             signals.SetShouldStop (signo, (bool) stop_action);
1882                         if (pass_action != -1)
1883                         {
1884                             bool suppress = ! ((bool) pass_action);
1885                             signals.SetShouldSuppress (signo, suppress);
1886                         }
1887                         signo = signals.GetNextSignalNumber (signo);
1888                     }
1889                 }
1890             }
1891         }
1892 
1893         PrintSignalInformation (result.GetOutputStream(), signal_args, num_signals_set, signals);
1894 
1895         if (num_signals_set > 0)
1896             result.SetStatus (eReturnStatusSuccessFinishNoResult);
1897         else
1898             result.SetStatus (eReturnStatusFailed);
1899 
1900         return result.Succeeded();
1901     }
1902 
1903     CommandOptions m_options;
1904 };
1905 
1906 OptionDefinition
1907 CommandObjectProcessHandle::CommandOptions::g_option_table[] =
1908 {
1909 { LLDB_OPT_SET_1, false, "stop",   's', OptionParser::eRequiredArgument, NULL, NULL, 0, eArgTypeBoolean, "Whether or not the process should be stopped if the signal is received." },
1910 { LLDB_OPT_SET_1, false, "notify", 'n', OptionParser::eRequiredArgument, NULL, NULL, 0, eArgTypeBoolean, "Whether or not the debugger should notify the user if the signal is received." },
1911 { LLDB_OPT_SET_1, false, "pass",  'p', OptionParser::eRequiredArgument, NULL, NULL, 0, eArgTypeBoolean, "Whether or not the signal should be passed to the process." },
1912 { 0, false, NULL, 0, 0, NULL, NULL, 0, eArgTypeNone, NULL }
1913 };
1914 
1915 //-------------------------------------------------------------------------
1916 // CommandObjectMultiwordProcess
1917 //-------------------------------------------------------------------------
1918 
1919 CommandObjectMultiwordProcess::CommandObjectMultiwordProcess (CommandInterpreter &interpreter) :
1920     CommandObjectMultiword (interpreter,
1921                             "process",
1922                             "A set of commands for operating on a process.",
1923                             "process <subcommand> [<subcommand-options>]")
1924 {
1925     LoadSubCommand ("attach",      CommandObjectSP (new CommandObjectProcessAttach    (interpreter)));
1926     LoadSubCommand ("launch",      CommandObjectSP (new CommandObjectProcessLaunch    (interpreter)));
1927     LoadSubCommand ("continue",    CommandObjectSP (new CommandObjectProcessContinue  (interpreter)));
1928     LoadSubCommand ("connect",     CommandObjectSP (new CommandObjectProcessConnect   (interpreter)));
1929     LoadSubCommand ("detach",      CommandObjectSP (new CommandObjectProcessDetach    (interpreter)));
1930     LoadSubCommand ("load",        CommandObjectSP (new CommandObjectProcessLoad      (interpreter)));
1931     LoadSubCommand ("unload",      CommandObjectSP (new CommandObjectProcessUnload    (interpreter)));
1932     LoadSubCommand ("signal",      CommandObjectSP (new CommandObjectProcessSignal    (interpreter)));
1933     LoadSubCommand ("handle",      CommandObjectSP (new CommandObjectProcessHandle    (interpreter)));
1934     LoadSubCommand ("status",      CommandObjectSP (new CommandObjectProcessStatus    (interpreter)));
1935     LoadSubCommand ("interrupt",   CommandObjectSP (new CommandObjectProcessInterrupt (interpreter)));
1936     LoadSubCommand ("kill",        CommandObjectSP (new CommandObjectProcessKill      (interpreter)));
1937     LoadSubCommand ("plugin",      CommandObjectSP (new CommandObjectProcessPlugin    (interpreter)));
1938     LoadSubCommand ("save-core",   CommandObjectSP (new CommandObjectProcessSaveCore  (interpreter)));
1939 }
1940 
1941 CommandObjectMultiwordProcess::~CommandObjectMultiwordProcess ()
1942 {
1943 }
1944 
1945