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 #include "CommandObjectProcess.h"
1130fdc8d8SChris Lattner 
1230fdc8d8SChris Lattner // C Includes
1330fdc8d8SChris Lattner // C++ Includes
1430fdc8d8SChris Lattner // Other libraries and framework includes
1530fdc8d8SChris Lattner // Project includes
1640af72e1SJim Ingham #include "lldb/Interpreter/Args.h"
1740af72e1SJim Ingham #include "lldb/Interpreter/Options.h"
1830fdc8d8SChris Lattner #include "lldb/Core/State.h"
1930fdc8d8SChris Lattner #include "lldb/Interpreter/CommandInterpreter.h"
2030fdc8d8SChris Lattner #include "lldb/Interpreter/CommandReturnObject.h"
214b9bea87SJim Ingham #include "./CommandObjectThread.h"
2230fdc8d8SChris Lattner #include "lldb/Target/Process.h"
2330fdc8d8SChris Lattner #include "lldb/Target/Target.h"
2430fdc8d8SChris Lattner #include "lldb/Target/Thread.h"
2530fdc8d8SChris Lattner 
2630fdc8d8SChris Lattner using namespace lldb;
2730fdc8d8SChris Lattner using namespace lldb_private;
2830fdc8d8SChris Lattner 
2930fdc8d8SChris Lattner //-------------------------------------------------------------------------
3030fdc8d8SChris Lattner // CommandObjectProcessLaunch
3130fdc8d8SChris Lattner //-------------------------------------------------------------------------
3230fdc8d8SChris Lattner 
3330fdc8d8SChris Lattner class CommandObjectProcessLaunch : public CommandObject
3430fdc8d8SChris Lattner {
3530fdc8d8SChris Lattner public:
3630fdc8d8SChris Lattner 
3730fdc8d8SChris Lattner     class CommandOptions : public Options
3830fdc8d8SChris Lattner     {
3930fdc8d8SChris Lattner     public:
4030fdc8d8SChris Lattner 
4130fdc8d8SChris Lattner         CommandOptions () :
4230fdc8d8SChris Lattner             Options()
4330fdc8d8SChris Lattner         {
4430fdc8d8SChris Lattner             // Keep default values of all options in one place: ResetOptionValues ()
4530fdc8d8SChris Lattner             ResetOptionValues ();
4630fdc8d8SChris Lattner         }
4730fdc8d8SChris Lattner 
4830fdc8d8SChris Lattner         ~CommandOptions ()
4930fdc8d8SChris Lattner         {
5030fdc8d8SChris Lattner         }
5130fdc8d8SChris Lattner 
5230fdc8d8SChris Lattner         Error
5330fdc8d8SChris Lattner         SetOptionValue (int option_idx, const char *option_arg)
5430fdc8d8SChris Lattner         {
5530fdc8d8SChris Lattner             Error error;
5630fdc8d8SChris Lattner             char short_option = (char) m_getopt_table[option_idx].val;
5730fdc8d8SChris Lattner 
5830fdc8d8SChris Lattner             switch (short_option)
5930fdc8d8SChris Lattner             {
6030fdc8d8SChris Lattner                 case 's':   stop_at_entry = true;       break;
6130fdc8d8SChris Lattner                 case 'e':   stderr_path = option_arg;   break;
6230fdc8d8SChris Lattner                 case 'i':   stdin_path  = option_arg;   break;
6330fdc8d8SChris Lattner                 case 'o':   stdout_path = option_arg;   break;
6430fdc8d8SChris Lattner                 case 'p':   plugin_name = option_arg;   break;
6530fdc8d8SChris Lattner                 default:
6630fdc8d8SChris Lattner                     error.SetErrorStringWithFormat("Invalid short option character '%c'.\n", short_option);
6730fdc8d8SChris Lattner                     break;
6830fdc8d8SChris Lattner 
6930fdc8d8SChris Lattner             }
7030fdc8d8SChris Lattner             return error;
7130fdc8d8SChris Lattner         }
7230fdc8d8SChris Lattner 
7330fdc8d8SChris Lattner         void
7430fdc8d8SChris Lattner         ResetOptionValues ()
7530fdc8d8SChris Lattner         {
7630fdc8d8SChris Lattner             Options::ResetOptionValues();
7730fdc8d8SChris Lattner             stop_at_entry = false;
7830fdc8d8SChris Lattner             stdin_path.clear();
7930fdc8d8SChris Lattner             stdout_path.clear();
8030fdc8d8SChris Lattner             stderr_path.clear();
8130fdc8d8SChris Lattner             plugin_name.clear();
8230fdc8d8SChris Lattner         }
8330fdc8d8SChris Lattner 
8430fdc8d8SChris Lattner         const lldb::OptionDefinition*
8530fdc8d8SChris Lattner         GetDefinitions ()
8630fdc8d8SChris Lattner         {
8730fdc8d8SChris Lattner             return g_option_table;
8830fdc8d8SChris Lattner         }
8930fdc8d8SChris Lattner 
9030fdc8d8SChris Lattner         // Options table: Required for subclasses of Options.
9130fdc8d8SChris Lattner 
9230fdc8d8SChris Lattner         static lldb::OptionDefinition g_option_table[];
9330fdc8d8SChris Lattner 
9430fdc8d8SChris Lattner         // Instance variables to hold the values for command options.
9530fdc8d8SChris Lattner 
9630fdc8d8SChris Lattner         bool stop_at_entry;
9730fdc8d8SChris Lattner         std::string stderr_path;
9830fdc8d8SChris Lattner         std::string stdin_path;
9930fdc8d8SChris Lattner         std::string stdout_path;
10030fdc8d8SChris Lattner         std::string plugin_name;
10130fdc8d8SChris Lattner 
10230fdc8d8SChris Lattner     };
10330fdc8d8SChris Lattner 
10430fdc8d8SChris Lattner     CommandObjectProcessLaunch () :
10530fdc8d8SChris Lattner         CommandObject ("process launch",
10630fdc8d8SChris Lattner                        "Launches the executable in the debugger.",
10730fdc8d8SChris Lattner                        "process launch [<cmd-options>] [<arguments-for-running-the-program>]")
10830fdc8d8SChris Lattner     {
10930fdc8d8SChris Lattner     }
11030fdc8d8SChris Lattner 
11130fdc8d8SChris Lattner 
11230fdc8d8SChris Lattner     ~CommandObjectProcessLaunch ()
11330fdc8d8SChris Lattner     {
11430fdc8d8SChris Lattner     }
11530fdc8d8SChris Lattner 
11630fdc8d8SChris Lattner     Options *
11730fdc8d8SChris Lattner     GetOptions ()
11830fdc8d8SChris Lattner     {
11930fdc8d8SChris Lattner         return &m_options;
12030fdc8d8SChris Lattner     }
12130fdc8d8SChris Lattner 
12230fdc8d8SChris Lattner     bool
1236611103cSGreg Clayton     Execute (CommandInterpreter &interpreter,
1246611103cSGreg Clayton              Args& launch_args,
12530fdc8d8SChris Lattner              CommandReturnObject &result)
12630fdc8d8SChris Lattner     {
1272976d00aSJim Ingham         Target *target = interpreter.GetDebugger().GetSelectedTarget().get();
1286611103cSGreg Clayton         bool synchronous_execution = interpreter.GetSynchronous ();
12930fdc8d8SChris Lattner     //    bool launched = false;
13030fdc8d8SChris Lattner     //    bool stopped_after_launch = false;
13130fdc8d8SChris Lattner 
13230fdc8d8SChris Lattner         if (target == NULL)
13330fdc8d8SChris Lattner         {
13430fdc8d8SChris Lattner             result.AppendError ("invalid target, set executable file using 'file' command");
13530fdc8d8SChris Lattner             result.SetStatus (eReturnStatusFailed);
13630fdc8d8SChris Lattner             return false;
13730fdc8d8SChris Lattner         }
13830fdc8d8SChris Lattner 
13930fdc8d8SChris Lattner         // If our listener is NULL, users aren't allows to launch
14030fdc8d8SChris Lattner         char filename[PATH_MAX];
14130fdc8d8SChris Lattner         Module *exe_module = target->GetExecutableModule().get();
14230fdc8d8SChris Lattner         exe_module->GetFileSpec().GetPath(filename, sizeof(filename));
14330fdc8d8SChris Lattner 
1446611103cSGreg Clayton         Process *process = interpreter.GetDebugger().GetExecutionContext().process;
14530fdc8d8SChris Lattner         if (process)
14630fdc8d8SChris Lattner         {
14730fdc8d8SChris Lattner             if (process->IsAlive())
14830fdc8d8SChris Lattner             {
14930fdc8d8SChris Lattner                result.AppendErrorWithFormat ("Process %u is currently being debugged, kill the process before running again.\n",
15030fdc8d8SChris Lattner                                             process->GetID());
15130fdc8d8SChris Lattner                 result.SetStatus (eReturnStatusFailed);
15230fdc8d8SChris Lattner                 return false;
15330fdc8d8SChris Lattner             }
15430fdc8d8SChris Lattner         }
15530fdc8d8SChris Lattner 
15630fdc8d8SChris Lattner         const char *plugin_name;
15730fdc8d8SChris Lattner         if (!m_options.plugin_name.empty())
15830fdc8d8SChris Lattner             plugin_name = m_options.plugin_name.c_str();
15930fdc8d8SChris Lattner         else
16030fdc8d8SChris Lattner             plugin_name = NULL;
16130fdc8d8SChris Lattner 
1626611103cSGreg Clayton         process = target->CreateProcess (interpreter.GetDebugger().GetListener(), plugin_name).get();
16330fdc8d8SChris Lattner 
164*3df9a8dfSCaroline Tice         const char *process_name = process->GetInstanceName().AsCString();
165*3df9a8dfSCaroline Tice         StreamString run_args_var_name;
166*3df9a8dfSCaroline Tice         StreamString env_vars_var_name;
167*3df9a8dfSCaroline Tice         StreamString disable_aslr_var_name;
168*3df9a8dfSCaroline Tice         lldb::SettableVariableType var_type;
169*3df9a8dfSCaroline Tice 
170*3df9a8dfSCaroline Tice         Args *run_args = NULL;
171*3df9a8dfSCaroline Tice         run_args_var_name.Printf ("process.[%s].run-args", process_name);
172*3df9a8dfSCaroline Tice         StringList run_args_value = Debugger::GetSettingsController()->GetVariable (run_args_var_name.GetData(),
173*3df9a8dfSCaroline Tice                                                                                     var_type);
174*3df9a8dfSCaroline Tice         if (run_args_value.GetSize() > 0)
175*3df9a8dfSCaroline Tice         {
176*3df9a8dfSCaroline Tice             run_args = new Args;
177*3df9a8dfSCaroline Tice             for (int i = 0; i < run_args_value.GetSize(); ++i)
178*3df9a8dfSCaroline Tice                 run_args->AppendArgument (run_args_value.GetStringAtIndex (i));
179*3df9a8dfSCaroline Tice         }
180*3df9a8dfSCaroline Tice 
181*3df9a8dfSCaroline Tice         Args *environment = NULL;
182*3df9a8dfSCaroline Tice         env_vars_var_name.Printf ("process.[%s].env-vars", process_name);
183*3df9a8dfSCaroline Tice         StringList env_vars_value = Debugger::GetSettingsController()->GetVariable (env_vars_var_name.GetData(),
184*3df9a8dfSCaroline Tice                                                                                     var_type);
185*3df9a8dfSCaroline Tice         if (env_vars_value.GetSize() > 0)
186*3df9a8dfSCaroline Tice         {
187*3df9a8dfSCaroline Tice             environment = new Args;
188*3df9a8dfSCaroline Tice             for (int i = 0; i < env_vars_value.GetSize(); ++i)
189*3df9a8dfSCaroline Tice                 environment->AppendArgument (env_vars_value.GetStringAtIndex (i));
190*3df9a8dfSCaroline Tice         }
191*3df9a8dfSCaroline Tice 
192f681b94fSGreg Clayton         uint32_t launch_flags = eLaunchFlagNone;
193*3df9a8dfSCaroline Tice         disable_aslr_var_name.Printf ("process.[%s].disable-aslr", process_name);
194*3df9a8dfSCaroline Tice         StringList disable_aslr_value = Debugger::GetSettingsController()->GetVariable(disable_aslr_var_name.GetData(),
195*3df9a8dfSCaroline Tice                                                                                        var_type);
196*3df9a8dfSCaroline Tice         if (disable_aslr_value.GetSize() > 0)
197*3df9a8dfSCaroline Tice         {
198*3df9a8dfSCaroline Tice             if (strcmp (disable_aslr_value.GetStringAtIndex(0), "true") == 0)
199f681b94fSGreg Clayton                 launch_flags |= eLaunchFlagDisableASLR;
20030fdc8d8SChris Lattner 
201*3df9a8dfSCaroline Tice         }
202*3df9a8dfSCaroline Tice 
20330fdc8d8SChris Lattner         // There are two possible sources of args to be passed to the process upon launching:  Those the user
20430fdc8d8SChris Lattner         // typed at the run command (launch_args); or those the user pre-set in the run-args variable (run_args).
20530fdc8d8SChris Lattner 
20630fdc8d8SChris Lattner         // If launch_args is empty, use run_args.
20730fdc8d8SChris Lattner         if (launch_args.GetArgumentCount() == 0)
20830fdc8d8SChris Lattner         {
20930fdc8d8SChris Lattner             if (run_args != NULL)
21030fdc8d8SChris Lattner                 launch_args.AppendArguments (*run_args);
21130fdc8d8SChris Lattner         }
21230fdc8d8SChris Lattner         else
21330fdc8d8SChris Lattner         {
21430fdc8d8SChris Lattner             // launch-args was not empty; use that, AND re-set run-args to contains launch-args values.
215*3df9a8dfSCaroline Tice             std::string new_run_args;
216*3df9a8dfSCaroline Tice             launch_args.GetCommandString (new_run_args);
217*3df9a8dfSCaroline Tice             Debugger::GetSettingsController()->SetVariable (run_args_var_name.GetData(), new_run_args.c_str(), lldb::eVarSetOperationAssign, false);
21830fdc8d8SChris Lattner         }
21930fdc8d8SChris Lattner 
22030fdc8d8SChris Lattner 
22130fdc8d8SChris Lattner         if (process)
22230fdc8d8SChris Lattner         {
22330fdc8d8SChris Lattner             const char *archname = exe_module->GetArchitecture().AsCString();
22430fdc8d8SChris Lattner 
22530fdc8d8SChris Lattner             const char * stdin_path = NULL;
22630fdc8d8SChris Lattner             const char * stdout_path = NULL;
22730fdc8d8SChris Lattner             const char * stderr_path = NULL;
22830fdc8d8SChris Lattner 
22930fdc8d8SChris Lattner             if (!(m_options.stdin_path.empty() &&
23030fdc8d8SChris Lattner                 m_options.stdout_path.empty() &&
23130fdc8d8SChris Lattner                 m_options.stderr_path.empty()))
23230fdc8d8SChris Lattner             {
23330fdc8d8SChris Lattner                 stdin_path =    m_options.stdin_path.empty()  ? "/dev/null" : m_options.stdin_path.c_str();
23430fdc8d8SChris Lattner                 stdout_path =   m_options.stdout_path.empty() ? "/dev/null" : m_options.stdout_path.c_str();
23530fdc8d8SChris Lattner                 stderr_path =   m_options.stderr_path.empty() ? "/dev/null" : m_options.stderr_path.c_str();
23630fdc8d8SChris Lattner             }
23730fdc8d8SChris Lattner 
23830fdc8d8SChris Lattner             Error error (process->Launch (launch_args.GetConstArgumentVector(),
23930fdc8d8SChris Lattner                                           environment ? environment->GetConstArgumentVector() : NULL,
240f681b94fSGreg Clayton                                           launch_flags,
24130fdc8d8SChris Lattner                                           stdin_path,
24230fdc8d8SChris Lattner                                           stdout_path,
24330fdc8d8SChris Lattner                                           stderr_path));
24430fdc8d8SChris Lattner 
24530fdc8d8SChris Lattner             if (error.Success())
24630fdc8d8SChris Lattner             {
24730fdc8d8SChris Lattner                 result.AppendMessageWithFormat ("Launching '%s'  (%s)\n", filename, archname);
24830fdc8d8SChris Lattner                 result.SetStatus (eReturnStatusSuccessContinuingNoResult);
24930fdc8d8SChris Lattner                 if (m_options.stop_at_entry == false)
25030fdc8d8SChris Lattner                 {
25130fdc8d8SChris Lattner                     StateType state = process->WaitForProcessToStop (NULL);
25230fdc8d8SChris Lattner 
25330fdc8d8SChris Lattner                     if (state == eStateStopped)
25430fdc8d8SChris Lattner                     {
25530fdc8d8SChris Lattner                         // Call continue_command.
25630fdc8d8SChris Lattner                         CommandReturnObject continue_result;
2576611103cSGreg Clayton                         interpreter.HandleCommand("process continue", false, continue_result);
25830fdc8d8SChris Lattner                     }
25930fdc8d8SChris Lattner 
26030fdc8d8SChris Lattner                     if (synchronous_execution)
26130fdc8d8SChris Lattner                     {
26230fdc8d8SChris Lattner                         result.SetDidChangeProcessState (true);
26330fdc8d8SChris Lattner                         result.SetStatus (eReturnStatusSuccessFinishNoResult);
26430fdc8d8SChris Lattner                     }
26530fdc8d8SChris Lattner                 }
26630fdc8d8SChris Lattner             }
26730fdc8d8SChris Lattner             else
26830fdc8d8SChris Lattner             {
26930fdc8d8SChris Lattner                 result.AppendErrorWithFormat ("Process launch failed: %s",
27030fdc8d8SChris Lattner                                               error.AsCString());
27130fdc8d8SChris Lattner                 result.SetStatus (eReturnStatusFailed);
27230fdc8d8SChris Lattner             }
27330fdc8d8SChris Lattner         }
27430fdc8d8SChris Lattner         else
27530fdc8d8SChris Lattner         {
27630fdc8d8SChris Lattner             result.AppendErrorWithFormat ("Process launch failed: unable to create a process object.\n");
27730fdc8d8SChris Lattner             result.SetStatus (eReturnStatusFailed);
27830fdc8d8SChris Lattner             return false;
27930fdc8d8SChris Lattner         }
28030fdc8d8SChris Lattner 
28130fdc8d8SChris Lattner         return result.Succeeded();
28230fdc8d8SChris Lattner     }
28330fdc8d8SChris Lattner 
284ebc09c36SJim Ingham     virtual const char *GetRepeatCommand (Args &current_command_args, uint32_t index)
285ebc09c36SJim Ingham     {
286ebc09c36SJim Ingham         // No repeat for "process launch"...
287ebc09c36SJim Ingham         return "";
288ebc09c36SJim Ingham     }
289ebc09c36SJim Ingham 
29030fdc8d8SChris Lattner protected:
29130fdc8d8SChris Lattner 
29230fdc8d8SChris Lattner     CommandOptions m_options;
29330fdc8d8SChris Lattner };
29430fdc8d8SChris Lattner 
29530fdc8d8SChris Lattner 
29630fdc8d8SChris Lattner lldb::OptionDefinition
29730fdc8d8SChris Lattner CommandObjectProcessLaunch::CommandOptions::g_option_table[] =
29830fdc8d8SChris Lattner {
2998651121cSJim Ingham { LLDB_OPT_SET_1, false, "stop-at-entry", 's', no_argument,       NULL, 0, NULL,        "Stop at the entry point of the program when launching a process."},
3008651121cSJim Ingham { LLDB_OPT_SET_1, false, "stdin",         'i', required_argument, NULL, 0, "<path>",    "Redirect stdin for the process to <path>."},
3018651121cSJim Ingham { LLDB_OPT_SET_1, false, "stdout",        'o', required_argument, NULL, 0, "<path>",    "Redirect stdout for the process to <path>."},
3028651121cSJim Ingham { LLDB_OPT_SET_1, false, "stderr",        'e', required_argument, NULL, 0, "<path>",    "Redirect stderr for the process to <path>."},
3038651121cSJim Ingham { LLDB_OPT_SET_1, false, "plugin",        'p', required_argument, NULL, 0, "<plugin>",  "Name of the process plugin you want to use."},
30430fdc8d8SChris Lattner { 0, false, NULL, 0, 0, NULL, 0, NULL, NULL }
30530fdc8d8SChris Lattner };
30630fdc8d8SChris Lattner 
30730fdc8d8SChris Lattner 
30830fdc8d8SChris Lattner //-------------------------------------------------------------------------
30930fdc8d8SChris Lattner // CommandObjectProcessAttach
31030fdc8d8SChris Lattner //-------------------------------------------------------------------------
31130fdc8d8SChris Lattner 
31230fdc8d8SChris Lattner class CommandObjectProcessAttach : public CommandObject
31330fdc8d8SChris Lattner {
31430fdc8d8SChris Lattner public:
31530fdc8d8SChris Lattner 
31630fdc8d8SChris Lattner     class CommandOptions : public Options
31730fdc8d8SChris Lattner     {
31830fdc8d8SChris Lattner     public:
31930fdc8d8SChris Lattner 
32030fdc8d8SChris Lattner         CommandOptions () :
32130fdc8d8SChris Lattner             Options()
32230fdc8d8SChris Lattner         {
32330fdc8d8SChris Lattner             // Keep default values of all options in one place: ResetOptionValues ()
32430fdc8d8SChris Lattner             ResetOptionValues ();
32530fdc8d8SChris Lattner         }
32630fdc8d8SChris Lattner 
32730fdc8d8SChris Lattner         ~CommandOptions ()
32830fdc8d8SChris Lattner         {
32930fdc8d8SChris Lattner         }
33030fdc8d8SChris Lattner 
33130fdc8d8SChris Lattner         Error
33230fdc8d8SChris Lattner         SetOptionValue (int option_idx, const char *option_arg)
33330fdc8d8SChris Lattner         {
33430fdc8d8SChris Lattner             Error error;
33530fdc8d8SChris Lattner             char short_option = (char) m_getopt_table[option_idx].val;
33630fdc8d8SChris Lattner             bool success = false;
33730fdc8d8SChris Lattner             switch (short_option)
33830fdc8d8SChris Lattner             {
33930fdc8d8SChris Lattner                 case 'p':
34030fdc8d8SChris Lattner                     pid = Args::StringToUInt32 (option_arg, LLDB_INVALID_PROCESS_ID, 0, &success);
34130fdc8d8SChris Lattner                     if (!success || pid == LLDB_INVALID_PROCESS_ID)
34230fdc8d8SChris Lattner                     {
34330fdc8d8SChris Lattner                         error.SetErrorStringWithFormat("Invalid process ID '%s'.\n", option_arg);
34430fdc8d8SChris Lattner                     }
34530fdc8d8SChris Lattner                     break;
34630fdc8d8SChris Lattner 
34730fdc8d8SChris Lattner                 case 'P':
34830fdc8d8SChris Lattner                     plugin_name = option_arg;
34930fdc8d8SChris Lattner                     break;
35030fdc8d8SChris Lattner 
35130fdc8d8SChris Lattner                 case 'n':
35230fdc8d8SChris Lattner                     name.assign(option_arg);
35330fdc8d8SChris Lattner                     break;
35430fdc8d8SChris Lattner 
35530fdc8d8SChris Lattner                 case 'w':
35630fdc8d8SChris Lattner                     waitfor = true;
35730fdc8d8SChris Lattner                     break;
35830fdc8d8SChris Lattner 
35930fdc8d8SChris Lattner                 default:
36030fdc8d8SChris Lattner                     error.SetErrorStringWithFormat("Invalid short option character '%c'.\n", short_option);
36130fdc8d8SChris Lattner                     break;
36230fdc8d8SChris Lattner             }
36330fdc8d8SChris Lattner             return error;
36430fdc8d8SChris Lattner         }
36530fdc8d8SChris Lattner 
36630fdc8d8SChris Lattner         void
36730fdc8d8SChris Lattner         ResetOptionValues ()
36830fdc8d8SChris Lattner         {
36930fdc8d8SChris Lattner             Options::ResetOptionValues();
37030fdc8d8SChris Lattner             pid = LLDB_INVALID_PROCESS_ID;
37130fdc8d8SChris Lattner             name.clear();
37230fdc8d8SChris Lattner             waitfor = false;
37330fdc8d8SChris Lattner         }
37430fdc8d8SChris Lattner 
37530fdc8d8SChris Lattner         const lldb::OptionDefinition*
37630fdc8d8SChris Lattner         GetDefinitions ()
37730fdc8d8SChris Lattner         {
37830fdc8d8SChris Lattner             return g_option_table;
37930fdc8d8SChris Lattner         }
38030fdc8d8SChris Lattner 
3815aee162fSJim Ingham         virtual bool
3825aee162fSJim Ingham         HandleOptionArgumentCompletion (CommandInterpreter &interpreter,
3835aee162fSJim Ingham                                         Args &input,
3845aee162fSJim Ingham                                         int cursor_index,
3855aee162fSJim Ingham                                         int char_pos,
3865aee162fSJim Ingham                                         OptionElementVector &opt_element_vector,
3875aee162fSJim Ingham                                         int opt_element_index,
3885aee162fSJim Ingham                                         int match_start_point,
3895aee162fSJim Ingham                                         int max_return_elements,
3905aee162fSJim Ingham                                         bool &word_complete,
3915aee162fSJim Ingham                                         StringList &matches)
3925aee162fSJim Ingham         {
3935aee162fSJim Ingham             int opt_arg_pos = opt_element_vector[opt_element_index].opt_arg_pos;
3945aee162fSJim Ingham             int opt_defs_index = opt_element_vector[opt_element_index].opt_defs_index;
3955aee162fSJim Ingham 
3965aee162fSJim Ingham             // We are only completing the name option for now...
3975aee162fSJim Ingham 
3985aee162fSJim Ingham             const lldb::OptionDefinition *opt_defs = GetDefinitions();
3995aee162fSJim Ingham             if (opt_defs[opt_defs_index].short_option == 'n')
4005aee162fSJim Ingham             {
4015aee162fSJim Ingham                 // Are we in the name?
4025aee162fSJim Ingham 
4035aee162fSJim Ingham                 // Look to see if there is a -P argument provided, and if so use that plugin, otherwise
4045aee162fSJim Ingham                 // use the default plugin.
4055aee162fSJim Ingham                 Process *process = interpreter.GetDebugger().GetExecutionContext().process;
4065aee162fSJim Ingham                 bool need_to_delete_process = false;
4075aee162fSJim Ingham 
4085aee162fSJim Ingham                 const char *partial_name = NULL;
4095aee162fSJim Ingham                 partial_name = input.GetArgumentAtIndex(opt_arg_pos);
4105aee162fSJim Ingham 
4115aee162fSJim Ingham                 if (process && process->IsAlive())
4125aee162fSJim Ingham                     return true;
4135aee162fSJim Ingham 
4142976d00aSJim Ingham                 Target *target = interpreter.GetDebugger().GetSelectedTarget().get();
4155aee162fSJim Ingham                 if (target == NULL)
4165aee162fSJim Ingham                 {
4175aee162fSJim Ingham                     // No target has been set yet, for now do host completion.  Otherwise I don't know how we would
4185aee162fSJim Ingham                     // figure out what the right target to use is...
4195aee162fSJim Ingham                     std::vector<lldb::pid_t> pids;
4205aee162fSJim Ingham                     Host::ListProcessesMatchingName (partial_name, matches, pids);
4215aee162fSJim Ingham                     return true;
4225aee162fSJim Ingham                 }
4235aee162fSJim Ingham                 if (!process)
4245aee162fSJim Ingham                 {
4255aee162fSJim Ingham                     process = target->CreateProcess (interpreter.GetDebugger().GetListener(), partial_name).get();
4265aee162fSJim Ingham                     need_to_delete_process = true;
4275aee162fSJim Ingham                 }
4285aee162fSJim Ingham 
4295aee162fSJim Ingham                 if (process)
4305aee162fSJim Ingham                 {
4315aee162fSJim Ingham                     matches.Clear();
4325aee162fSJim Ingham                     std::vector<lldb::pid_t> pids;
4335aee162fSJim Ingham                     process->ListProcessesMatchingName (NULL, matches, pids);
4345aee162fSJim Ingham                     if (need_to_delete_process)
4355aee162fSJim Ingham                         target->DeleteCurrentProcess();
4365aee162fSJim Ingham                     return true;
4375aee162fSJim Ingham                 }
4385aee162fSJim Ingham             }
4395aee162fSJim Ingham 
4405aee162fSJim Ingham             return false;
4415aee162fSJim Ingham         }
4425aee162fSJim Ingham 
44330fdc8d8SChris Lattner         // Options table: Required for subclasses of Options.
44430fdc8d8SChris Lattner 
44530fdc8d8SChris Lattner         static lldb::OptionDefinition g_option_table[];
44630fdc8d8SChris Lattner 
44730fdc8d8SChris Lattner         // Instance variables to hold the values for command options.
44830fdc8d8SChris Lattner 
44930fdc8d8SChris Lattner         lldb::pid_t pid;
45030fdc8d8SChris Lattner         std::string plugin_name;
45130fdc8d8SChris Lattner         std::string name;
45230fdc8d8SChris Lattner         bool waitfor;
45330fdc8d8SChris Lattner     };
45430fdc8d8SChris Lattner 
4555aee162fSJim Ingham     CommandObjectProcessAttach () :
4565aee162fSJim Ingham         CommandObject ("process attach",
4575aee162fSJim Ingham                        "Attaches to a process.",
4585aee162fSJim Ingham                        "process attach <cmd-options>")
4595aee162fSJim Ingham     {
4605aee162fSJim Ingham         SetHelpLong("Currently, you must set the executable file before you can attach "
4615aee162fSJim Ingham                     "to a process.\n");
4625aee162fSJim Ingham     }
4635aee162fSJim Ingham 
4645aee162fSJim Ingham     ~CommandObjectProcessAttach ()
4655aee162fSJim Ingham     {
4665aee162fSJim Ingham     }
4675aee162fSJim Ingham 
4685aee162fSJim Ingham     bool
4695aee162fSJim Ingham     Execute (CommandInterpreter &interpreter,
4705aee162fSJim Ingham              Args& command,
4715aee162fSJim Ingham              CommandReturnObject &result)
4725aee162fSJim Ingham     {
4732976d00aSJim Ingham         Target *target = interpreter.GetDebugger().GetSelectedTarget().get();
4745aee162fSJim Ingham 
4755aee162fSJim Ingham         Process *process = interpreter.GetDebugger().GetExecutionContext().process;
4765aee162fSJim Ingham         if (process)
4775aee162fSJim Ingham         {
4785aee162fSJim Ingham             if (process->IsAlive())
4795aee162fSJim Ingham             {
4805aee162fSJim Ingham                 result.AppendErrorWithFormat ("Process %u is currently being debugged, kill the process before attaching.\n",
4815aee162fSJim Ingham                                               process->GetID());
4825aee162fSJim Ingham                 result.SetStatus (eReturnStatusFailed);
4835aee162fSJim Ingham                 return false;
4845aee162fSJim Ingham             }
4855aee162fSJim Ingham         }
4865aee162fSJim Ingham 
4875aee162fSJim Ingham         if (target == NULL)
4885aee162fSJim Ingham         {
4895aee162fSJim Ingham             // If there isn't a current target create one.
4905aee162fSJim Ingham             TargetSP new_target_sp;
4915aee162fSJim Ingham             FileSpec emptyFileSpec;
4925aee162fSJim Ingham             ArchSpec emptyArchSpec;
4935aee162fSJim Ingham             Error error;
4945aee162fSJim Ingham 
4955aee162fSJim Ingham             error = interpreter.GetDebugger().GetTargetList().CreateTarget(interpreter.GetDebugger(),
4965aee162fSJim Ingham                                                                            emptyFileSpec,
4975aee162fSJim Ingham                                                                            emptyArchSpec,
4985aee162fSJim Ingham                                                                            NULL,
4995aee162fSJim Ingham                                                                            false,
5005aee162fSJim Ingham                                                                            new_target_sp);
5015aee162fSJim Ingham             target = new_target_sp.get();
5025aee162fSJim Ingham             if (target == NULL || error.Fail())
5035aee162fSJim Ingham             {
5045aee162fSJim Ingham                 result.AppendError(error.AsCString("Error creating empty target"));
5055aee162fSJim Ingham                 return false;
5065aee162fSJim Ingham             }
5072976d00aSJim Ingham             interpreter.GetDebugger().GetTargetList().SetSelectedTarget(target);
5085aee162fSJim Ingham         }
5095aee162fSJim Ingham 
5105aee162fSJim Ingham         // Record the old executable module, we want to issue a warning if the process of attaching changed the
5115aee162fSJim Ingham         // current executable (like somebody said "file foo" then attached to a PID whose executable was bar.)
5125aee162fSJim Ingham 
5135aee162fSJim Ingham         ModuleSP old_exec_module_sp = target->GetExecutableModule();
5145aee162fSJim Ingham         ArchSpec old_arch_spec = target->GetArchitecture();
5155aee162fSJim Ingham 
5165aee162fSJim Ingham         if (command.GetArgumentCount())
5175aee162fSJim Ingham         {
5185aee162fSJim Ingham             result.AppendErrorWithFormat("Invalid arguments for '%s'.\nUsage: \n", m_cmd_name.c_str(), m_cmd_syntax.c_str());
5195aee162fSJim Ingham             result.SetStatus (eReturnStatusFailed);
5205aee162fSJim Ingham         }
5215aee162fSJim Ingham         else
5225aee162fSJim Ingham         {
5235aee162fSJim Ingham             const char *plugin_name = NULL;
5245aee162fSJim Ingham 
5255aee162fSJim Ingham             if (!m_options.plugin_name.empty())
5265aee162fSJim Ingham                 plugin_name = m_options.plugin_name.c_str();
5275aee162fSJim Ingham 
5285aee162fSJim Ingham             process = target->CreateProcess (interpreter.GetDebugger().GetListener(), plugin_name).get();
5295aee162fSJim Ingham 
5305aee162fSJim Ingham             if (process)
5315aee162fSJim Ingham             {
5325aee162fSJim Ingham                 Error error;
5335aee162fSJim Ingham                 int attach_pid = m_options.pid;
5345aee162fSJim Ingham 
5355aee162fSJim Ingham                 // If we are waiting for a process with this name to show up, do that first.
5365aee162fSJim Ingham                 if (m_options.waitfor)
5375aee162fSJim Ingham                 {
5385aee162fSJim Ingham                     if (m_options.name.empty())
5395aee162fSJim Ingham                     {
5405aee162fSJim Ingham                         result.AppendError("Invalid arguments: must supply a process name with the waitfor option.\n");
5415aee162fSJim Ingham                         result.SetStatus (eReturnStatusFailed);
5425aee162fSJim Ingham                         return false;
5435aee162fSJim Ingham                     }
5445aee162fSJim Ingham                     else
5455aee162fSJim Ingham                     {
5465aee162fSJim Ingham                         error = process->Attach (m_options.name.c_str(), m_options.waitfor);
5475aee162fSJim Ingham                         if (error.Success())
5485aee162fSJim Ingham                         {
5495aee162fSJim Ingham                             result.SetStatus (eReturnStatusSuccessContinuingNoResult);
5505aee162fSJim Ingham                         }
5515aee162fSJim Ingham                         else
5525aee162fSJim Ingham                         {
5535aee162fSJim Ingham                             result.AppendErrorWithFormat ("Waiting for a process to launch named '%s': %s\n",
5545aee162fSJim Ingham                                                              m_options.name.c_str(),
5555aee162fSJim Ingham                                                              error.AsCString());
5565aee162fSJim Ingham                             result.SetStatus (eReturnStatusFailed);
5575aee162fSJim Ingham                             return false;
5585aee162fSJim Ingham                         }
5595aee162fSJim Ingham                     }
5605aee162fSJim Ingham                 }
5615aee162fSJim Ingham                 else
5625aee162fSJim Ingham                 {
5635aee162fSJim Ingham                     // If the process was specified by name look it up, so we can warn if there are multiple
5645aee162fSJim Ingham                     // processes with this pid.
5655aee162fSJim Ingham 
5665aee162fSJim Ingham                     if (attach_pid == LLDB_INVALID_PROCESS_ID && !m_options.name.empty())
5675aee162fSJim Ingham                     {
5685aee162fSJim Ingham                         std::vector<lldb::pid_t> pids;
5695aee162fSJim Ingham                         StringList matches;
5705aee162fSJim Ingham 
5715aee162fSJim Ingham                         process->ListProcessesMatchingName(m_options.name.c_str(), matches, pids);
5725aee162fSJim Ingham                         if (matches.GetSize() > 1)
5735aee162fSJim Ingham                         {
5745aee162fSJim Ingham                             result.AppendErrorWithFormat("More than one process named %s\n", m_options.name.c_str());
5755aee162fSJim Ingham                             result.SetStatus (eReturnStatusFailed);
5765aee162fSJim Ingham                             return false;
5775aee162fSJim Ingham                         }
5785aee162fSJim Ingham                         else if (matches.GetSize() == 0)
5795aee162fSJim Ingham                         {
5805aee162fSJim Ingham                             result.AppendErrorWithFormat("Could not find a process named %s\n", m_options.name.c_str());
5815aee162fSJim Ingham                             result.SetStatus (eReturnStatusFailed);
5825aee162fSJim Ingham                             return false;
5835aee162fSJim Ingham                         }
5845aee162fSJim Ingham                         else
5855aee162fSJim Ingham                         {
5865aee162fSJim Ingham                             attach_pid = pids[0];
5875aee162fSJim Ingham                         }
5885aee162fSJim Ingham 
5895aee162fSJim Ingham                     }
5905aee162fSJim Ingham 
5915aee162fSJim Ingham                     if (attach_pid != LLDB_INVALID_PROCESS_ID)
5925aee162fSJim Ingham                     {
5935aee162fSJim Ingham                         error = process->Attach (attach_pid);
5945aee162fSJim Ingham                         if (error.Success())
5955aee162fSJim Ingham                         {
5965aee162fSJim Ingham                             result.SetStatus (eReturnStatusSuccessContinuingNoResult);
5975aee162fSJim Ingham                         }
5985aee162fSJim Ingham                         else
5995aee162fSJim Ingham                         {
6005aee162fSJim Ingham                             result.AppendErrorWithFormat ("Attaching to process %i failed: %s.\n",
6015aee162fSJim Ingham                                                          attach_pid,
6025aee162fSJim Ingham                                                          error.AsCString());
6035aee162fSJim Ingham                             result.SetStatus (eReturnStatusFailed);
6045aee162fSJim Ingham                         }
6055aee162fSJim Ingham                     }
6065aee162fSJim Ingham                     else
6075aee162fSJim Ingham                     {
6085aee162fSJim Ingham                         result.AppendErrorWithFormat ("No PID specified for attach\n",
6095aee162fSJim Ingham                                                          attach_pid,
6105aee162fSJim Ingham                                                          error.AsCString());
6115aee162fSJim Ingham                         result.SetStatus (eReturnStatusFailed);
6125aee162fSJim Ingham 
6135aee162fSJim Ingham                     }
6145aee162fSJim Ingham                 }
6155aee162fSJim Ingham             }
6165aee162fSJim Ingham         }
6175aee162fSJim Ingham 
6185aee162fSJim Ingham         if (result.Succeeded())
6195aee162fSJim Ingham         {
6205aee162fSJim Ingham             // Okay, we're done.  Last step is to warn if the executable module has changed:
6215aee162fSJim Ingham             if (!old_exec_module_sp)
6225aee162fSJim Ingham             {
6235aee162fSJim Ingham                 char new_path[PATH_MAX + 1];
6245aee162fSJim Ingham                 target->GetExecutableModule()->GetFileSpec().GetPath(new_path, PATH_MAX);
6255aee162fSJim Ingham 
6265aee162fSJim Ingham                 result.AppendMessageWithFormat("Executable module set to \"%s\".\n",
6275aee162fSJim Ingham                     new_path);
6285aee162fSJim Ingham             }
6295aee162fSJim Ingham             else if (old_exec_module_sp->GetFileSpec() != target->GetExecutableModule()->GetFileSpec())
6305aee162fSJim Ingham             {
6315aee162fSJim Ingham                 char old_path[PATH_MAX + 1];
6325aee162fSJim Ingham                 char new_path[PATH_MAX + 1];
6335aee162fSJim Ingham 
6345aee162fSJim Ingham                 old_exec_module_sp->GetFileSpec().GetPath(old_path, PATH_MAX);
6355aee162fSJim Ingham                 target->GetExecutableModule()->GetFileSpec().GetPath (new_path, PATH_MAX);
6365aee162fSJim Ingham 
6375aee162fSJim Ingham                 result.AppendWarningWithFormat("Executable module changed from \"%s\" to \"%s\".\n",
6385aee162fSJim Ingham                                                     old_path, new_path);
6395aee162fSJim Ingham             }
6405aee162fSJim Ingham 
6415aee162fSJim Ingham             if (!old_arch_spec.IsValid())
6425aee162fSJim Ingham             {
6435aee162fSJim Ingham                 result.AppendMessageWithFormat ("Architecture set to: %s.\n", target->GetArchitecture().AsCString());
6445aee162fSJim Ingham             }
6455aee162fSJim Ingham             else if (old_arch_spec != target->GetArchitecture())
6465aee162fSJim Ingham             {
6475aee162fSJim Ingham                 result.AppendWarningWithFormat("Architecture changed from %s to %s.\n",
6485aee162fSJim Ingham                                                 old_arch_spec.AsCString(), target->GetArchitecture().AsCString());
6495aee162fSJim Ingham             }
6505aee162fSJim Ingham         }
6515aee162fSJim Ingham         return result.Succeeded();
6525aee162fSJim Ingham     }
6535aee162fSJim Ingham 
6545aee162fSJim Ingham     Options *
6555aee162fSJim Ingham     GetOptions ()
6565aee162fSJim Ingham     {
6575aee162fSJim Ingham         return &m_options;
6585aee162fSJim Ingham     }
6595aee162fSJim Ingham 
66030fdc8d8SChris Lattner protected:
66130fdc8d8SChris Lattner 
66230fdc8d8SChris Lattner     CommandOptions m_options;
66330fdc8d8SChris Lattner };
66430fdc8d8SChris Lattner 
66530fdc8d8SChris Lattner 
66630fdc8d8SChris Lattner lldb::OptionDefinition
66730fdc8d8SChris Lattner CommandObjectProcessAttach::CommandOptions::g_option_table[] =
66830fdc8d8SChris Lattner {
6698651121cSJim Ingham { LLDB_OPT_SET_ALL, false, "plugin",       'P', required_argument, NULL, 0, "<plugin>",        "Name of the process plugin you want to use."},
6708651121cSJim Ingham { LLDB_OPT_SET_1, false, "pid",          'p', required_argument, NULL, 0, "<pid>",           "The process ID of an existing process to attach to."},
6718651121cSJim Ingham { LLDB_OPT_SET_2, true,  "name",         'n', required_argument, NULL, 0, "<process-name>",  "The name of the process to attach to."},
6728651121cSJim Ingham { LLDB_OPT_SET_2, false, "waitfor",      'w', no_argument,       NULL, 0, NULL,              "Wait for the the process with <process-name> to launch."},
67330fdc8d8SChris Lattner { 0, false, NULL, 0, 0, NULL, 0, NULL, NULL }
67430fdc8d8SChris Lattner };
67530fdc8d8SChris Lattner 
67630fdc8d8SChris Lattner //-------------------------------------------------------------------------
67730fdc8d8SChris Lattner // CommandObjectProcessContinue
67830fdc8d8SChris Lattner //-------------------------------------------------------------------------
67930fdc8d8SChris Lattner 
68030fdc8d8SChris Lattner class CommandObjectProcessContinue : public CommandObject
68130fdc8d8SChris Lattner {
68230fdc8d8SChris Lattner public:
68330fdc8d8SChris Lattner 
68430fdc8d8SChris Lattner     CommandObjectProcessContinue () :
68530fdc8d8SChris Lattner         CommandObject ("process continue",
68630fdc8d8SChris Lattner                        "Continues execution all threads in the current process.",
68730fdc8d8SChris Lattner                        "process continue",
68830fdc8d8SChris Lattner                        eFlagProcessMustBeLaunched | eFlagProcessMustBePaused)
68930fdc8d8SChris Lattner     {
69030fdc8d8SChris Lattner     }
69130fdc8d8SChris Lattner 
69230fdc8d8SChris Lattner 
69330fdc8d8SChris Lattner     ~CommandObjectProcessContinue ()
69430fdc8d8SChris Lattner     {
69530fdc8d8SChris Lattner     }
69630fdc8d8SChris Lattner 
69730fdc8d8SChris Lattner     bool
6986611103cSGreg Clayton     Execute (CommandInterpreter &interpreter,
6996611103cSGreg Clayton              Args& command,
70030fdc8d8SChris Lattner              CommandReturnObject &result)
70130fdc8d8SChris Lattner     {
7026611103cSGreg Clayton         Process *process = interpreter.GetDebugger().GetExecutionContext().process;
7036611103cSGreg Clayton         bool synchronous_execution = interpreter.GetSynchronous ();
70430fdc8d8SChris Lattner 
70530fdc8d8SChris Lattner         if (process == NULL)
70630fdc8d8SChris Lattner         {
70730fdc8d8SChris Lattner             result.AppendError ("no process to continue");
70830fdc8d8SChris Lattner             result.SetStatus (eReturnStatusFailed);
70930fdc8d8SChris Lattner             return false;
71030fdc8d8SChris Lattner          }
71130fdc8d8SChris Lattner 
71230fdc8d8SChris Lattner         StateType state = process->GetState();
71330fdc8d8SChris Lattner         if (state == eStateStopped)
71430fdc8d8SChris Lattner         {
71530fdc8d8SChris Lattner             if (command.GetArgumentCount() != 0)
71630fdc8d8SChris Lattner             {
71730fdc8d8SChris Lattner                 result.AppendErrorWithFormat ("The '%s' command does not take any arguments.\n", m_cmd_name.c_str());
71830fdc8d8SChris Lattner                 result.SetStatus (eReturnStatusFailed);
71930fdc8d8SChris Lattner                 return false;
72030fdc8d8SChris Lattner             }
72130fdc8d8SChris Lattner 
72230fdc8d8SChris Lattner             const uint32_t num_threads = process->GetThreadList().GetSize();
72330fdc8d8SChris Lattner 
72430fdc8d8SChris Lattner             // Set the actions that the threads should each take when resuming
72530fdc8d8SChris Lattner             for (uint32_t idx=0; idx<num_threads; ++idx)
72630fdc8d8SChris Lattner             {
72730fdc8d8SChris Lattner                 process->GetThreadList().GetThreadAtIndex(idx)->SetResumeState (eStateRunning);
72830fdc8d8SChris Lattner             }
72930fdc8d8SChris Lattner 
73030fdc8d8SChris Lattner             Error error(process->Resume());
73130fdc8d8SChris Lattner             if (error.Success())
73230fdc8d8SChris Lattner             {
73330fdc8d8SChris Lattner                 result.AppendMessageWithFormat ("Resuming process %i\n", process->GetID());
73430fdc8d8SChris Lattner                 if (synchronous_execution)
73530fdc8d8SChris Lattner                 {
736b132097bSGreg Clayton                     state = process->WaitForProcessToStop (NULL);
73730fdc8d8SChris Lattner 
73830fdc8d8SChris Lattner                     result.SetDidChangeProcessState (true);
73930fdc8d8SChris Lattner                     result.AppendMessageWithFormat ("Process %i %s\n", process->GetID(), StateAsCString (state));
74030fdc8d8SChris Lattner                     result.SetStatus (eReturnStatusSuccessFinishNoResult);
74130fdc8d8SChris Lattner                 }
74230fdc8d8SChris Lattner                 else
74330fdc8d8SChris Lattner                 {
74430fdc8d8SChris Lattner                     result.SetStatus (eReturnStatusSuccessContinuingNoResult);
74530fdc8d8SChris Lattner                 }
74630fdc8d8SChris Lattner             }
74730fdc8d8SChris Lattner             else
74830fdc8d8SChris Lattner             {
74930fdc8d8SChris Lattner                 result.AppendErrorWithFormat("Failed to resume process: %s.\n", error.AsCString());
75030fdc8d8SChris Lattner                 result.SetStatus (eReturnStatusFailed);
75130fdc8d8SChris Lattner             }
75230fdc8d8SChris Lattner         }
75330fdc8d8SChris Lattner         else
75430fdc8d8SChris Lattner         {
75530fdc8d8SChris Lattner             result.AppendErrorWithFormat ("Process cannot be continued from its current state (%s).\n",
75630fdc8d8SChris Lattner                                          StateAsCString(state));
75730fdc8d8SChris Lattner             result.SetStatus (eReturnStatusFailed);
75830fdc8d8SChris Lattner         }
75930fdc8d8SChris Lattner         return result.Succeeded();
76030fdc8d8SChris Lattner     }
76130fdc8d8SChris Lattner };
76230fdc8d8SChris Lattner 
76330fdc8d8SChris Lattner //-------------------------------------------------------------------------
76430fdc8d8SChris Lattner // CommandObjectProcessDetach
76530fdc8d8SChris Lattner //-------------------------------------------------------------------------
76630fdc8d8SChris Lattner 
76730fdc8d8SChris Lattner class CommandObjectProcessDetach : public CommandObject
76830fdc8d8SChris Lattner {
76930fdc8d8SChris Lattner public:
77030fdc8d8SChris Lattner 
77130fdc8d8SChris Lattner     CommandObjectProcessDetach () :
77230fdc8d8SChris Lattner         CommandObject ("process detach",
77330fdc8d8SChris Lattner                        "Detaches from the current process being debugged.",
77430fdc8d8SChris Lattner                        "process detach",
77530fdc8d8SChris Lattner                        eFlagProcessMustBeLaunched)
77630fdc8d8SChris Lattner     {
77730fdc8d8SChris Lattner     }
77830fdc8d8SChris Lattner 
77930fdc8d8SChris Lattner     ~CommandObjectProcessDetach ()
78030fdc8d8SChris Lattner     {
78130fdc8d8SChris Lattner     }
78230fdc8d8SChris Lattner 
78330fdc8d8SChris Lattner     bool
7846611103cSGreg Clayton     Execute (CommandInterpreter &interpreter,
7856611103cSGreg Clayton              Args& command,
78630fdc8d8SChris Lattner              CommandReturnObject &result)
78730fdc8d8SChris Lattner     {
7886611103cSGreg Clayton         Process *process = interpreter.GetDebugger().GetExecutionContext().process;
78930fdc8d8SChris Lattner         if (process == NULL)
79030fdc8d8SChris Lattner         {
79130fdc8d8SChris Lattner             result.AppendError ("must have a valid process in order to detach");
79230fdc8d8SChris Lattner             result.SetStatus (eReturnStatusFailed);
79330fdc8d8SChris Lattner             return false;
79430fdc8d8SChris Lattner         }
79530fdc8d8SChris Lattner 
79630fdc8d8SChris Lattner         Error error (process->Detach());
79730fdc8d8SChris Lattner         if (error.Success())
79830fdc8d8SChris Lattner         {
79930fdc8d8SChris Lattner             result.SetStatus (eReturnStatusSuccessFinishResult);
80030fdc8d8SChris Lattner         }
80130fdc8d8SChris Lattner         else
80230fdc8d8SChris Lattner         {
80330fdc8d8SChris Lattner             result.AppendErrorWithFormat ("Detach failed: %s\n", error.AsCString());
80430fdc8d8SChris Lattner             result.SetStatus (eReturnStatusFailed);
80530fdc8d8SChris Lattner             return false;
80630fdc8d8SChris Lattner         }
80730fdc8d8SChris Lattner         return result.Succeeded();
80830fdc8d8SChris Lattner     }
80930fdc8d8SChris Lattner };
81030fdc8d8SChris Lattner 
81130fdc8d8SChris Lattner //-------------------------------------------------------------------------
81230fdc8d8SChris Lattner // CommandObjectProcessSignal
81330fdc8d8SChris Lattner //-------------------------------------------------------------------------
81430fdc8d8SChris Lattner 
81530fdc8d8SChris Lattner class CommandObjectProcessSignal : public CommandObject
81630fdc8d8SChris Lattner {
81730fdc8d8SChris Lattner public:
81830fdc8d8SChris Lattner 
81930fdc8d8SChris Lattner     CommandObjectProcessSignal () :
82030fdc8d8SChris Lattner         CommandObject ("process signal",
82130fdc8d8SChris Lattner                        "Sends a UNIX signal to the current process being debugged.",
82230fdc8d8SChris Lattner                        "process signal <unix-signal-number>")
82330fdc8d8SChris Lattner     {
82430fdc8d8SChris Lattner     }
82530fdc8d8SChris Lattner 
82630fdc8d8SChris Lattner     ~CommandObjectProcessSignal ()
82730fdc8d8SChris Lattner     {
82830fdc8d8SChris Lattner     }
82930fdc8d8SChris Lattner 
83030fdc8d8SChris Lattner     bool
8316611103cSGreg Clayton     Execute (CommandInterpreter &interpreter,
8326611103cSGreg Clayton              Args& command,
83330fdc8d8SChris Lattner              CommandReturnObject &result)
83430fdc8d8SChris Lattner     {
8356611103cSGreg Clayton         Process *process = interpreter.GetDebugger().GetExecutionContext().process;
83630fdc8d8SChris Lattner         if (process == NULL)
83730fdc8d8SChris Lattner         {
83830fdc8d8SChris Lattner             result.AppendError ("no process to signal");
83930fdc8d8SChris Lattner             result.SetStatus (eReturnStatusFailed);
84030fdc8d8SChris Lattner             return false;
84130fdc8d8SChris Lattner         }
84230fdc8d8SChris Lattner 
84330fdc8d8SChris Lattner         if (command.GetArgumentCount() == 1)
84430fdc8d8SChris Lattner         {
84530fdc8d8SChris Lattner             int signo = Args::StringToSInt32(command.GetArgumentAtIndex(0), -1, 0);
84630fdc8d8SChris Lattner             if (signo == -1)
84730fdc8d8SChris Lattner             {
84830fdc8d8SChris Lattner                 result.AppendErrorWithFormat ("Invalid signal argument '%s'.\n", command.GetArgumentAtIndex(0));
84930fdc8d8SChris Lattner                 result.SetStatus (eReturnStatusFailed);
85030fdc8d8SChris Lattner             }
85130fdc8d8SChris Lattner             else
85230fdc8d8SChris Lattner             {
85330fdc8d8SChris Lattner                 Error error (process->Signal (signo));
85430fdc8d8SChris Lattner                 if (error.Success())
85530fdc8d8SChris Lattner                 {
85630fdc8d8SChris Lattner                     result.SetStatus (eReturnStatusSuccessFinishResult);
85730fdc8d8SChris Lattner                 }
85830fdc8d8SChris Lattner                 else
85930fdc8d8SChris Lattner                 {
86030fdc8d8SChris Lattner                     result.AppendErrorWithFormat ("Failed to send signal %i: %s\n", signo, error.AsCString());
86130fdc8d8SChris Lattner                     result.SetStatus (eReturnStatusFailed);
86230fdc8d8SChris Lattner                 }
86330fdc8d8SChris Lattner             }
86430fdc8d8SChris Lattner         }
86530fdc8d8SChris Lattner         else
86630fdc8d8SChris Lattner         {
86730fdc8d8SChris Lattner             result.AppendErrorWithFormat("'%s' takes exactly one signal number argument:\nUsage: \n", m_cmd_name.c_str(),
86830fdc8d8SChris Lattner                                         m_cmd_syntax.c_str());
86930fdc8d8SChris Lattner             result.SetStatus (eReturnStatusFailed);
87030fdc8d8SChris Lattner         }
87130fdc8d8SChris Lattner         return result.Succeeded();
87230fdc8d8SChris Lattner     }
87330fdc8d8SChris Lattner };
87430fdc8d8SChris Lattner 
87530fdc8d8SChris Lattner 
87630fdc8d8SChris Lattner //-------------------------------------------------------------------------
87730fdc8d8SChris Lattner // CommandObjectProcessInterrupt
87830fdc8d8SChris Lattner //-------------------------------------------------------------------------
87930fdc8d8SChris Lattner 
88030fdc8d8SChris Lattner class CommandObjectProcessInterrupt : public CommandObject
88130fdc8d8SChris Lattner {
88230fdc8d8SChris Lattner public:
88330fdc8d8SChris Lattner 
88430fdc8d8SChris Lattner 
88530fdc8d8SChris Lattner     CommandObjectProcessInterrupt () :
88630fdc8d8SChris Lattner     CommandObject ("process interrupt",
88730fdc8d8SChris Lattner                    "Interrupts the current process being debugged.",
88830fdc8d8SChris Lattner                    "process interrupt",
88930fdc8d8SChris Lattner                    eFlagProcessMustBeLaunched)
89030fdc8d8SChris Lattner     {
89130fdc8d8SChris Lattner     }
89230fdc8d8SChris Lattner 
89330fdc8d8SChris Lattner     ~CommandObjectProcessInterrupt ()
89430fdc8d8SChris Lattner     {
89530fdc8d8SChris Lattner     }
89630fdc8d8SChris Lattner 
89730fdc8d8SChris Lattner     bool
8986611103cSGreg Clayton     Execute (CommandInterpreter &interpreter,
8996611103cSGreg Clayton              Args& command,
90030fdc8d8SChris Lattner              CommandReturnObject &result)
90130fdc8d8SChris Lattner     {
9026611103cSGreg Clayton         Process *process = interpreter.GetDebugger().GetExecutionContext().process;
90330fdc8d8SChris Lattner         if (process == NULL)
90430fdc8d8SChris Lattner         {
90530fdc8d8SChris Lattner             result.AppendError ("no process to halt");
90630fdc8d8SChris Lattner             result.SetStatus (eReturnStatusFailed);
90730fdc8d8SChris Lattner             return false;
90830fdc8d8SChris Lattner         }
90930fdc8d8SChris Lattner 
91030fdc8d8SChris Lattner         if (command.GetArgumentCount() == 0)
91130fdc8d8SChris Lattner         {
91230fdc8d8SChris Lattner             Error error(process->Halt ());
91330fdc8d8SChris Lattner             if (error.Success())
91430fdc8d8SChris Lattner             {
91530fdc8d8SChris Lattner                 result.SetStatus (eReturnStatusSuccessFinishResult);
91630fdc8d8SChris Lattner 
91730fdc8d8SChris Lattner                 // Maybe we should add a "SuspendThreadPlans so we
91830fdc8d8SChris Lattner                 // can halt, and keep in place all the current thread plans.
91930fdc8d8SChris Lattner                 process->GetThreadList().DiscardThreadPlans();
92030fdc8d8SChris Lattner             }
92130fdc8d8SChris Lattner             else
92230fdc8d8SChris Lattner             {
92330fdc8d8SChris Lattner                 result.AppendErrorWithFormat ("Failed to halt process: %s\n", error.AsCString());
92430fdc8d8SChris Lattner                 result.SetStatus (eReturnStatusFailed);
92530fdc8d8SChris Lattner             }
92630fdc8d8SChris Lattner         }
92730fdc8d8SChris Lattner         else
92830fdc8d8SChris Lattner         {
92930fdc8d8SChris Lattner             result.AppendErrorWithFormat("'%s' takes no arguments:\nUsage: \n",
93030fdc8d8SChris Lattner                                         m_cmd_name.c_str(),
93130fdc8d8SChris Lattner                                         m_cmd_syntax.c_str());
93230fdc8d8SChris Lattner             result.SetStatus (eReturnStatusFailed);
93330fdc8d8SChris Lattner         }
93430fdc8d8SChris Lattner         return result.Succeeded();
93530fdc8d8SChris Lattner     }
93630fdc8d8SChris Lattner };
93730fdc8d8SChris Lattner 
93830fdc8d8SChris Lattner //-------------------------------------------------------------------------
93930fdc8d8SChris Lattner // CommandObjectProcessKill
94030fdc8d8SChris Lattner //-------------------------------------------------------------------------
94130fdc8d8SChris Lattner 
94230fdc8d8SChris Lattner class CommandObjectProcessKill : public CommandObject
94330fdc8d8SChris Lattner {
94430fdc8d8SChris Lattner public:
94530fdc8d8SChris Lattner 
94630fdc8d8SChris Lattner     CommandObjectProcessKill () :
94730fdc8d8SChris Lattner     CommandObject ("process kill",
94830fdc8d8SChris Lattner                    "Terminates the current process being debugged.",
94930fdc8d8SChris Lattner                    "process kill",
95030fdc8d8SChris Lattner                    eFlagProcessMustBeLaunched)
95130fdc8d8SChris Lattner     {
95230fdc8d8SChris Lattner     }
95330fdc8d8SChris Lattner 
95430fdc8d8SChris Lattner     ~CommandObjectProcessKill ()
95530fdc8d8SChris Lattner     {
95630fdc8d8SChris Lattner     }
95730fdc8d8SChris Lattner 
95830fdc8d8SChris Lattner     bool
9596611103cSGreg Clayton     Execute (CommandInterpreter &interpreter,
9606611103cSGreg Clayton              Args& command,
96130fdc8d8SChris Lattner              CommandReturnObject &result)
96230fdc8d8SChris Lattner     {
9636611103cSGreg Clayton         Process *process = interpreter.GetDebugger().GetExecutionContext().process;
96430fdc8d8SChris Lattner         if (process == NULL)
96530fdc8d8SChris Lattner         {
96630fdc8d8SChris Lattner             result.AppendError ("no process to kill");
96730fdc8d8SChris Lattner             result.SetStatus (eReturnStatusFailed);
96830fdc8d8SChris Lattner             return false;
96930fdc8d8SChris Lattner         }
97030fdc8d8SChris Lattner 
97130fdc8d8SChris Lattner         if (command.GetArgumentCount() == 0)
97230fdc8d8SChris Lattner         {
97330fdc8d8SChris Lattner             Error error (process->Destroy());
97430fdc8d8SChris Lattner             if (error.Success())
97530fdc8d8SChris Lattner             {
97630fdc8d8SChris Lattner                 result.SetStatus (eReturnStatusSuccessFinishResult);
97730fdc8d8SChris Lattner             }
97830fdc8d8SChris Lattner             else
97930fdc8d8SChris Lattner             {
98030fdc8d8SChris Lattner                 result.AppendErrorWithFormat ("Failed to kill process: %s\n", error.AsCString());
98130fdc8d8SChris Lattner                 result.SetStatus (eReturnStatusFailed);
98230fdc8d8SChris Lattner             }
98330fdc8d8SChris Lattner         }
98430fdc8d8SChris Lattner         else
98530fdc8d8SChris Lattner         {
98630fdc8d8SChris Lattner             result.AppendErrorWithFormat("'%s' takes no arguments:\nUsage: \n",
98730fdc8d8SChris Lattner                                         m_cmd_name.c_str(),
98830fdc8d8SChris Lattner                                         m_cmd_syntax.c_str());
98930fdc8d8SChris Lattner             result.SetStatus (eReturnStatusFailed);
99030fdc8d8SChris Lattner         }
99130fdc8d8SChris Lattner         return result.Succeeded();
99230fdc8d8SChris Lattner     }
99330fdc8d8SChris Lattner };
99430fdc8d8SChris Lattner 
99530fdc8d8SChris Lattner //-------------------------------------------------------------------------
9964b9bea87SJim Ingham // CommandObjectProcessStatus
9974b9bea87SJim Ingham //-------------------------------------------------------------------------
9984b9bea87SJim Ingham class CommandObjectProcessStatus : public CommandObject
9994b9bea87SJim Ingham {
10004b9bea87SJim Ingham public:
10014b9bea87SJim Ingham     CommandObjectProcessStatus () :
10024b9bea87SJim Ingham     CommandObject ("status",
10034b9bea87SJim Ingham                    "Shows the current status and location of executing process.",
10044b9bea87SJim Ingham                    "status",
10054b9bea87SJim Ingham                    0)
10064b9bea87SJim Ingham     {
10074b9bea87SJim Ingham     }
10084b9bea87SJim Ingham 
10094b9bea87SJim Ingham     ~CommandObjectProcessStatus()
10104b9bea87SJim Ingham     {
10114b9bea87SJim Ingham     }
10124b9bea87SJim Ingham 
10134b9bea87SJim Ingham 
10144b9bea87SJim Ingham     bool
10154b9bea87SJim Ingham     Execute
10164b9bea87SJim Ingham     (
10176611103cSGreg Clayton         CommandInterpreter &interpreter,
10184b9bea87SJim Ingham         Args& command,
10194b9bea87SJim Ingham         CommandReturnObject &result
10204b9bea87SJim Ingham     )
10214b9bea87SJim Ingham     {
10224b9bea87SJim Ingham         StreamString &output_stream = result.GetOutputStream();
10234b9bea87SJim Ingham         result.SetStatus (eReturnStatusSuccessFinishNoResult);
10246611103cSGreg Clayton         ExecutionContext exe_ctx(interpreter.GetDebugger().GetExecutionContext());
10254b9bea87SJim Ingham         if (exe_ctx.process)
10264b9bea87SJim Ingham         {
10274b9bea87SJim Ingham             const StateType state = exe_ctx.process->GetState();
10284b9bea87SJim Ingham             if (StateIsStoppedState(state))
10294b9bea87SJim Ingham             {
10304b9bea87SJim Ingham                 if (state == eStateExited)
10314b9bea87SJim Ingham                 {
10324b9bea87SJim Ingham                     int exit_status = exe_ctx.process->GetExitStatus();
10334b9bea87SJim Ingham                     const char *exit_description = exe_ctx.process->GetExitDescription();
10344b9bea87SJim Ingham                     output_stream.Printf ("Process %d exited with status = %i (0x%8.8x) %s\n",
10354b9bea87SJim Ingham                                           exe_ctx.process->GetID(),
10364b9bea87SJim Ingham                                           exit_status,
10374b9bea87SJim Ingham                                           exit_status,
10384b9bea87SJim Ingham                                           exit_description ? exit_description : "");
10394b9bea87SJim Ingham                 }
10404b9bea87SJim Ingham                 else
10414b9bea87SJim Ingham                 {
10424b9bea87SJim Ingham                     output_stream.Printf ("Process %d %s\n", exe_ctx.process->GetID(), StateAsCString (state));
10434b9bea87SJim Ingham                     if (exe_ctx.thread == NULL)
10444b9bea87SJim Ingham                         exe_ctx.thread = exe_ctx.process->GetThreadList().GetThreadAtIndex(0).get();
10454b9bea87SJim Ingham                     if (exe_ctx.thread != NULL)
10464b9bea87SJim Ingham                     {
10474b9bea87SJim Ingham                         DisplayThreadsInfo (interpreter, &exe_ctx, result, true, true);
10484b9bea87SJim Ingham                     }
10494b9bea87SJim Ingham                     else
10504b9bea87SJim Ingham                     {
10514b9bea87SJim Ingham                         result.AppendError ("No valid thread found in current process.");
10524b9bea87SJim Ingham                         result.SetStatus (eReturnStatusFailed);
10534b9bea87SJim Ingham                     }
10544b9bea87SJim Ingham                 }
10554b9bea87SJim Ingham             }
10564b9bea87SJim Ingham             else
10574b9bea87SJim Ingham             {
10584b9bea87SJim Ingham                 output_stream.Printf ("Process %d is running.\n",
10594b9bea87SJim Ingham                                           exe_ctx.process->GetID());
10604b9bea87SJim Ingham             }
10614b9bea87SJim Ingham         }
10624b9bea87SJim Ingham         else
10634b9bea87SJim Ingham         {
10644b9bea87SJim Ingham             result.AppendError ("No current location or status available.");
10654b9bea87SJim Ingham             result.SetStatus (eReturnStatusFailed);
10664b9bea87SJim Ingham         }
10674b9bea87SJim Ingham         return result.Succeeded();
10684b9bea87SJim Ingham     }
10694b9bea87SJim Ingham };
10704b9bea87SJim Ingham 
10714b9bea87SJim Ingham //-------------------------------------------------------------------------
107230fdc8d8SChris Lattner // CommandObjectMultiwordProcess
107330fdc8d8SChris Lattner //-------------------------------------------------------------------------
107430fdc8d8SChris Lattner 
10756611103cSGreg Clayton CommandObjectMultiwordProcess::CommandObjectMultiwordProcess (CommandInterpreter &interpreter) :
107630fdc8d8SChris Lattner     CommandObjectMultiword ("process",
107730fdc8d8SChris Lattner                               "A set of commands for operating on a process.",
107830fdc8d8SChris Lattner                               "process <subcommand> [<subcommand-options>]")
107930fdc8d8SChris Lattner {
10806611103cSGreg Clayton     LoadSubCommand (interpreter, "attach",      CommandObjectSP (new CommandObjectProcessAttach ()));
10816611103cSGreg Clayton     LoadSubCommand (interpreter, "launch",      CommandObjectSP (new CommandObjectProcessLaunch ()));
10826611103cSGreg Clayton     LoadSubCommand (interpreter, "continue",    CommandObjectSP (new CommandObjectProcessContinue ()));
10836611103cSGreg Clayton     LoadSubCommand (interpreter, "detach",      CommandObjectSP (new CommandObjectProcessDetach ()));
10846611103cSGreg Clayton     LoadSubCommand (interpreter, "signal",      CommandObjectSP (new CommandObjectProcessSignal ()));
10856611103cSGreg Clayton     LoadSubCommand (interpreter, "status",      CommandObjectSP (new CommandObjectProcessStatus ()));
10866611103cSGreg Clayton     LoadSubCommand (interpreter, "interrupt",   CommandObjectSP (new CommandObjectProcessInterrupt ()));
10876611103cSGreg Clayton     LoadSubCommand (interpreter, "kill",        CommandObjectSP (new CommandObjectProcessKill ()));
108830fdc8d8SChris Lattner }
108930fdc8d8SChris Lattner 
109030fdc8d8SChris Lattner CommandObjectMultiwordProcess::~CommandObjectMultiwordProcess ()
109130fdc8d8SChris Lattner {
109230fdc8d8SChris Lattner }
109330fdc8d8SChris Lattner 
1094