1 //===-- MachProcess.h -------------------------------------------*- 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 //  Created by Greg Clayton on 6/15/07.
11 //
12 //===----------------------------------------------------------------------===//
13 
14 #ifndef __MachProcess_h__
15 #define __MachProcess_h__
16 
17 #include "DNBDefs.h"
18 #include "DNBBreakpoint.h"
19 #include "DNBError.h"
20 #include "DNBThreadResumeActions.h"
21 //#include "MachDYLD.h"
22 #include "MachException.h"
23 #include "MachVMMemory.h"
24 #include "MachTask.h"
25 #include "MachThreadList.h"
26 #include "PThreadCondition.h"
27 #include "PThreadEvent.h"
28 #include "PThreadMutex.h"
29 
30 #include <mach/mach.h>
31 #include <sys/signal.h>
32 #include <pthread.h>
33 #include <vector>
34 
35 class DNBThreadResumeActions;
36 
37 class MachProcess
38 {
39 public:
40     //----------------------------------------------------------------------
41     // Constructors and Destructors
42     //----------------------------------------------------------------------
43     MachProcess ();
44     ~MachProcess ();
45 
46     //----------------------------------------------------------------------
47     // Child process control
48     //----------------------------------------------------------------------
49     pid_t                   AttachForDebug (pid_t pid, char *err_str, size_t err_len);
50     pid_t                   LaunchForDebug (const char *path,
51                                             char const *argv[],
52                                             char const *envp[],
53                                             const char *working_directory,
54                                             const char *stdin_path,
55                                             const char *stdout_path,
56                                             const char *stderr_path,
57                                             bool no_stdio,
58                                             nub_launch_flavor_t launch_flavor,
59                                             int disable_aslr,
60                                             DNBError &err);
61 
62     static uint32_t         GetCPUTypeForLocalProcess (pid_t pid);
63     static pid_t            ForkChildForPTraceDebugging (const char *path, char const *argv[], char const *envp[], MachProcess* process, DNBError &err);
64     static pid_t            PosixSpawnChildForPTraceDebugging (const char *path,
65                                                                cpu_type_t cpu_type,
66                                                                char const *argv[],
67                                                                char const *envp[],
68                                                                const char *working_directory,
69                                                                const char *stdin_path,
70                                                                const char *stdout_path,
71                                                                const char *stderr_path,
72                                                                bool no_stdio,
73                                                                MachProcess* process,
74                                                                int disable_aslr,
75                                                                DNBError& err);
76     nub_addr_t              GetDYLDAllImageInfosAddress ();
77     static const void *     PrepareForAttach (const char *path, nub_launch_flavor_t launch_flavor, bool waitfor, DNBError &err_str);
78     static void             CleanupAfterAttach (const void *attach_token, bool success, DNBError &err_str);
79     static nub_process_t    CheckForProcess (const void *attach_token);
80 #ifdef WITH_SPRINGBOARD
81     pid_t                   SBLaunchForDebug (const char *app_bundle_path, char const *argv[], char const *envp[], bool no_stdio, DNBError &launch_err);
82     static pid_t            SBForkChildForPTraceDebugging (const char *path, char const *argv[], char const *envp[], bool no_stdio, MachProcess* process, DNBError &launch_err);
83 #endif
84     nub_addr_t              LookupSymbol (const char *name, const char *shlib);
85     void                    SetNameToAddressCallback (DNBCallbackNameToAddress callback, void *baton)
86                             {
87                                 m_name_to_addr_callback = callback;
88                                 m_name_to_addr_baton    = baton;
89                             }
90     void                    SetSharedLibraryInfoCallback (DNBCallbackCopyExecutableImageInfos callback, void *baton)
91                             {
92                                 m_image_infos_callback    = callback;
93                                 m_image_infos_baton        = baton;
94                             }
95 
96     bool                    Resume (const DNBThreadResumeActions& thread_actions);
97     bool                    Signal  (int signal, const struct timespec *timeout_abstime = NULL);
98     bool                    Kill (const struct timespec *timeout_abstime = NULL);
99     bool                    Detach ();
100     nub_size_t              ReadMemory (nub_addr_t addr, nub_size_t size, void *buf);
101     nub_size_t              WriteMemory (nub_addr_t addr, nub_size_t size, const void *buf);
102 
103     //----------------------------------------------------------------------
104     // Path and arg accessors
105     //----------------------------------------------------------------------
106     const char *            Path () const { return m_path.c_str(); }
107     size_t                  ArgumentCount () const { return m_args.size(); }
108     const char *            ArgumentAtIndex (size_t arg_idx) const
109                             {
110                                 if (arg_idx < m_args.size())
111                                     return m_args[arg_idx].c_str();
112                                 return NULL;
113                             }
114 
115     //----------------------------------------------------------------------
116     // Breakpoint functions
117     //----------------------------------------------------------------------
118     nub_break_t             CreateBreakpoint (nub_addr_t addr, nub_size_t length, bool hardware, thread_t thread);
119     bool                    DisableBreakpoint (nub_break_t breakID, bool remove);
120     nub_size_t              DisableAllBreakpoints (bool remove);
121     bool                    EnableBreakpoint (nub_break_t breakID);
122     void                    DumpBreakpoint(nub_break_t breakID) const;
123     DNBBreakpointList&      Breakpoints() { return m_breakpoints; }
124     const DNBBreakpointList& Breakpoints() const { return m_breakpoints; }
125 
126     //----------------------------------------------------------------------
127     // Watchpoint functions
128     //----------------------------------------------------------------------
129     nub_watch_t             CreateWatchpoint (nub_addr_t addr, nub_size_t length, uint32_t watch_type, bool hardware, thread_t thread);
130     bool                    DisableWatchpoint (nub_watch_t watchID, bool remove);
131     nub_size_t              DisableAllWatchpoints (bool remove);
132     bool                    EnableWatchpoint (nub_watch_t watchID);
133     void                    DumpWatchpoint(nub_watch_t watchID) const;
134     uint32_t                GetNumSupportedHardwareWatchpoints () const;
135     DNBBreakpointList&      Watchpoints() { return m_watchpoints; }
136     const DNBBreakpointList& Watchpoints() const { return m_watchpoints; }
137 
138     //----------------------------------------------------------------------
139     // Exception thread functions
140     //----------------------------------------------------------------------
141     bool                    StartSTDIOThread ();
142     static void *           STDIOThread (void *arg);
143     void                    ExceptionMessageReceived (const MachException::Message& exceptionMessage);
144     void                    ExceptionMessageBundleComplete ();
145     void                    SharedLibrariesUpdated ();
146     nub_size_t              CopyImageInfos (struct DNBExecutableImageInfo **image_infos, bool only_changed);
147 
148     //----------------------------------------------------------------------
149     // Accessors
150     //----------------------------------------------------------------------
151     pid_t                   ProcessID () const { return m_pid; }
152     bool                    ProcessIDIsValid () const { return m_pid > 0; }
153     pid_t                   SetProcessID (pid_t pid);
154     MachTask&               Task() { return m_task; }
155     const MachTask&         Task() const { return m_task; }
156 
157     PThreadEvent&           Events() { return m_events; }
158     const DNBRegisterSetInfo *
159                             GetRegisterSetInfo (nub_thread_t tid, nub_size_t *num_reg_sets) const;
160     bool                    GetRegisterValue (nub_thread_t tid, uint32_t set, uint32_t reg, DNBRegisterValue *reg_value) const;
161     bool                    SetRegisterValue (nub_thread_t tid, uint32_t set, uint32_t reg, const DNBRegisterValue *value) const;
162     nub_bool_t              SyncThreadState (nub_thread_t tid);
163     const char *            ThreadGetName (nub_thread_t tid);
164     nub_state_t             ThreadGetState (nub_thread_t tid);
165     nub_size_t              GetNumThreads () const;
166     nub_thread_t            GetThreadAtIndex (nub_size_t thread_idx) const;
167     nub_thread_t            GetCurrentThread ();
168     nub_thread_t            SetCurrentThread (nub_thread_t tid);
169     MachThreadList &        GetThreadList() { return m_thread_list; }
170     bool                    GetThreadStoppedReason(nub_thread_t tid, struct DNBThreadStopInfo *stop_info) const;
171     void                    DumpThreadStoppedReason(nub_thread_t tid) const;
172     const char *            GetThreadInfo (nub_thread_t tid) const;
173 
174     uint32_t                GetCPUType ();
175     nub_state_t             GetState ();
176     void                    SetState (nub_state_t state);
177     bool                    IsRunning (nub_state_t state)
178                             {
179                                 return    state == eStateRunning || IsStepping(state);
180                             }
181     bool                    IsStepping (nub_state_t state)
182                             {
183                                 return    state == eStateStepping;
184                             }
185     bool                    CanResume (nub_state_t state)
186                             {
187                                 return state == eStateStopped;
188                             }
189 
190     bool                    GetExitStatus(int* status)
191                             {
192                                 if (GetState() == eStateExited)
193                                 {
194                                     if (status)
195                                         *status = m_exit_status;
196                                     return true;
197                                 }
198                                 return false;
199                             }
200     void                    SetExitStatus(int status)
201                             {
202                                 m_exit_status = status;
203                                 SetState(eStateExited);
204                             }
205 
206     uint32_t                StopCount() const { return m_stop_count; }
207     void                    SetChildFileDescriptors (int stdin_fileno, int stdout_fileno, int stderr_fileno)
208                             {
209                                 m_child_stdin   = stdin_fileno;
210                                 m_child_stdout  = stdout_fileno;
211                                 m_child_stderr  = stderr_fileno;
212                             }
213 
214     int                     GetStdinFileDescriptor () const { return m_child_stdin; }
215     int                     GetStdoutFileDescriptor () const { return m_child_stdout; }
216     int                     GetStderrFileDescriptor () const { return m_child_stderr; }
217     void                    AppendSTDOUT (char* s, size_t len);
218     size_t                  GetAvailableSTDOUT (char *buf, size_t buf_size);
219     size_t                  GetAvailableSTDERR (char *buf, size_t buf_size);
220     void                    CloseChildFileDescriptors ()
221                             {
222                                 if (m_child_stdin >= 0)
223                                 {
224                                     ::close (m_child_stdin);
225                                     m_child_stdin = -1;
226                                 }
227                                 if (m_child_stdout >= 0)
228                                 {
229                                     ::close (m_child_stdout);
230                                     m_child_stdout = -1;
231                                 }
232                                 if (m_child_stderr >= 0)
233                                 {
234                                     ::close (m_child_stderr);
235                                     m_child_stderr = -1;
236                                 }
237                             }
238 
239     bool                    ProcessUsingSpringBoard() const { return (m_flags & eMachProcessFlagsUsingSBS) != 0; }
240 private:
241     enum
242     {
243         eMachProcessFlagsNone = 0,
244         eMachProcessFlagsAttached = (1 << 0),
245         eMachProcessFlagsUsingSBS = (1 << 1)
246     };
247     void                    Clear ();
248     void                    ReplyToAllExceptions ();
249     void                    PrivateResume ();
250     nub_size_t              RemoveTrapsFromBuffer (nub_addr_t addr, nub_size_t size, uint8_t *buf) const;
251 
252     uint32_t                Flags () const { return m_flags; }
253     nub_state_t             DoSIGSTOP (bool clear_bps_and_wps, bool allow_running, uint32_t *thread_idx_ptr);
254 
255     pid_t                       m_pid;                      // Process ID of child process
256     cpu_type_t                  m_cpu_type;                 // The CPU type of this process
257     int                         m_child_stdin;
258     int                         m_child_stdout;
259     int                         m_child_stderr;
260     std::string                 m_path;                     // A path to the executable if we have one
261     std::vector<std::string>    m_args;                     // The arguments with which the process was lauched
262     int                         m_exit_status;              // The exit status for the process
263     MachTask                    m_task;                     // The mach task for this process
264     uint32_t                    m_flags;                    // Process specific flags (see eMachProcessFlags enums)
265     uint32_t                    m_stop_count;               // A count of many times have we stopped
266     pthread_t                   m_stdio_thread;             // Thread ID for the thread that watches for child process stdio
267     PThreadMutex                m_stdio_mutex;              // Multithreaded protection for stdio
268     std::string                 m_stdout_data;
269     DNBThreadResumeActions      m_thread_actions;           // The thread actions for the current MachProcess::Resume() call
270     MachException::Message::collection
271                                 m_exception_messages;       // A collection of exception messages caught when listening to the exception port
272     PThreadMutex                m_exception_messages_mutex; // Multithreaded protection for m_exception_messages
273 
274     MachThreadList              m_thread_list;               // A list of threads that is maintained/updated after each stop
275     nub_state_t                 m_state;                    // The state of our process
276     PThreadMutex                m_state_mutex;              // Multithreaded protection for m_state
277     PThreadEvent                m_events;                   // Process related events in the child processes lifetime can be waited upon
278     DNBBreakpointList           m_breakpoints;              // Breakpoint list for this process
279     DNBBreakpointList           m_watchpoints;              // Watchpoint list for this process
280     DNBCallbackNameToAddress    m_name_to_addr_callback;
281     void *                      m_name_to_addr_baton;
282     DNBCallbackCopyExecutableImageInfos
283                                 m_image_infos_callback;
284     void *                      m_image_infos_baton;
285 };
286 
287 
288 #endif // __MachProcess_h__
289