blob: 6de893d0396e304b5d995dc7b80190314c7800bb [file] [log] [blame]
Chris Lattner30fdc8d2010-06-08 16:52:24 +00001//===-- DNB.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// Created by Greg Clayton on 3/23/07.
11//
12//===----------------------------------------------------------------------===//
13
14#include "DNB.h"
15#include <signal.h>
16#include <stdio.h>
17#include <stdlib.h>
18#include <sys/resource.h>
19#include <sys/stat.h>
20#include <sys/types.h>
21#include <sys/wait.h>
22#include <unistd.h>
23#include <sys/sysctl.h>
24#include <map>
25#include <vector>
26
27#include "MacOSX/MachProcess.h"
28#include "MacOSX/MachTask.h"
29#include "CFString.h"
30#include "DNBLog.h"
31#include "DNBDataRef.h"
32#include "DNBThreadResumeActions.h"
33#include "DNBTimer.h"
34
Greg Claytond64afba2012-03-14 03:07:05 +000035typedef STD_SHARED_PTR(MachProcess) MachProcessSP;
Chris Lattner30fdc8d2010-06-08 16:52:24 +000036typedef std::map<nub_process_t, MachProcessSP> ProcessMap;
37typedef ProcessMap::iterator ProcessMapIter;
38typedef ProcessMap::const_iterator ProcessMapConstIter;
39
40static size_t GetAllInfos (std::vector<struct kinfo_proc>& proc_infos);
41static size_t GetAllInfosMatchingName (const char *process_name, std::vector<struct kinfo_proc>& matching_proc_infos);
42
43//----------------------------------------------------------------------
44// A Thread safe singleton to get a process map pointer.
45//
46// Returns a pointer to the existing process map, or a pointer to a
47// newly created process map if CAN_CREATE is non-zero.
48//----------------------------------------------------------------------
49static ProcessMap*
50GetProcessMap(bool can_create)
51{
52 static ProcessMap* g_process_map_ptr = NULL;
53
54 if (can_create && g_process_map_ptr == NULL)
55 {
56 static pthread_mutex_t g_process_map_mutex = PTHREAD_MUTEX_INITIALIZER;
57 PTHREAD_MUTEX_LOCKER (locker, &g_process_map_mutex);
58 if (g_process_map_ptr == NULL)
59 g_process_map_ptr = new ProcessMap;
60 }
61 return g_process_map_ptr;
62}
63
64//----------------------------------------------------------------------
65// Add PID to the shared process pointer map.
66//
67// Return non-zero value if we succeed in adding the process to the map.
68// The only time this should fail is if we run out of memory and can't
69// allocate a ProcessMap.
70//----------------------------------------------------------------------
71static nub_bool_t
72AddProcessToMap (nub_process_t pid, MachProcessSP& procSP)
73{
74 ProcessMap* process_map = GetProcessMap(true);
75 if (process_map)
76 {
77 process_map->insert(std::make_pair(pid, procSP));
78 return true;
79 }
80 return false;
81}
82
83//----------------------------------------------------------------------
84// Remove the shared pointer for PID from the process map.
85//
86// Returns the number of items removed from the process map.
87//----------------------------------------------------------------------
88static size_t
89RemoveProcessFromMap (nub_process_t pid)
90{
91 ProcessMap* process_map = GetProcessMap(false);
92 if (process_map)
93 {
94 return process_map->erase(pid);
95 }
96 return 0;
97}
98
99//----------------------------------------------------------------------
100// Get the shared pointer for PID from the existing process map.
101//
102// Returns true if we successfully find a shared pointer to a
103// MachProcess object.
104//----------------------------------------------------------------------
105static nub_bool_t
106GetProcessSP (nub_process_t pid, MachProcessSP& procSP)
107{
108 ProcessMap* process_map = GetProcessMap(false);
109 if (process_map != NULL)
110 {
111 ProcessMapIter pos = process_map->find(pid);
112 if (pos != process_map->end())
113 {
114 procSP = pos->second;
115 return true;
116 }
117 }
118 procSP.reset();
119 return false;
120}
121
122
123static void *
124waitpid_thread (void *arg)
125{
126 const pid_t pid = (pid_t)(intptr_t)arg;
127 int status;
128 while (1)
129 {
130 pid_t child_pid = waitpid(pid, &status, 0);
131 DNBLogThreadedIf(LOG_PROCESS, "waitpid_process_thread (): waitpid (pid = %i, &status, 0) => %i, status = %i, errno = %i", pid, child_pid, status, errno);
132
133 if (child_pid < 0)
134 {
135 if (errno == EINTR)
136 continue;
137 break;
138 }
139 else
140 {
141 if (WIFSTOPPED(status))
142 {
143 continue;
144 }
145 else// if (WIFEXITED(status) || WIFSIGNALED(status))
146 {
147 DNBLogThreadedIf(LOG_PROCESS, "waitpid_process_thread (): setting exit status for pid = %i to %i", child_pid, status);
148 DNBProcessSetExitStatus (child_pid, status);
149 return NULL;
150 }
151 }
152 }
153
154 // We should never exit as long as our child process is alive, so if we
155 // do something else went wrong and we should exit...
156 DNBLogThreadedIf(LOG_PROCESS, "waitpid_process_thread (): main loop exited, setting exit status to an invalid value (-1) for pid %i", pid);
157 DNBProcessSetExitStatus (pid, -1);
158 return NULL;
159}
160
161static bool
162spawn_waitpid_thread (pid_t pid)
163{
164 pthread_t thread = THREAD_NULL;
165 ::pthread_create (&thread, NULL, waitpid_thread, (void *)(intptr_t)pid);
166 if (thread != THREAD_NULL)
167 {
168 ::pthread_detach (thread);
169 return true;
170 }
171 return false;
172}
173
174nub_process_t
175DNBProcessLaunch (const char *path,
176 char const *argv[],
177 const char *envp[],
Greg Clayton6779606a2011-01-22 23:43:18 +0000178 const char *working_directory, // NULL => dont' change, non-NULL => set working directory for inferior to this
179 const char *stdin_path,
180 const char *stdout_path,
181 const char *stderr_path,
Caroline Ticef8da8632010-12-03 18:46:09 +0000182 bool no_stdio,
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000183 nub_launch_flavor_t launch_flavor,
Greg Claytonf681b942010-08-31 18:35:14 +0000184 int disable_aslr,
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000185 char *err_str,
186 size_t err_len)
187{
Greg Claytonbd82a5d2011-01-23 05:56:20 +0000188 DNBLogThreadedIf(LOG_PROCESS, "%s ( path='%s', argv = %p, envp = %p, working_dir=%s, stdin=%s, stdout=%s, stderr=%s, no-stdio=%i, launch_flavor = %u, disable_aslr = %d, err = %p, err_len = %zu) called...",
189 __FUNCTION__,
190 path,
191 argv,
192 envp,
193 working_directory,
194 stdin_path,
195 stdout_path,
196 stderr_path,
197 no_stdio,
198 launch_flavor,
199 disable_aslr,
200 err_str,
201 err_len);
202
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000203 if (err_str && err_len > 0)
204 err_str[0] = '\0';
205 struct stat path_stat;
206 if (::stat(path, &path_stat) == -1)
207 {
208 char stat_error[256];
209 ::strerror_r (errno, stat_error, sizeof(stat_error));
210 snprintf(err_str, err_len, "%s (%s)", stat_error, path);
211 return INVALID_NUB_PROCESS;
212 }
213
214 MachProcessSP processSP (new MachProcess);
215 if (processSP.get())
216 {
217 DNBError launch_err;
Greg Clayton6779606a2011-01-22 23:43:18 +0000218 pid_t pid = processSP->LaunchForDebug (path,
219 argv,
220 envp,
221 working_directory,
222 stdin_path,
223 stdout_path,
224 stderr_path,
225 no_stdio,
226 launch_flavor,
227 disable_aslr,
228 launch_err);
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000229 if (err_str)
230 {
231 *err_str = '\0';
232 if (launch_err.Fail())
233 {
234 const char *launch_err_str = launch_err.AsString();
235 if (launch_err_str)
236 {
237 strncpy(err_str, launch_err_str, err_len-1);
238 err_str[err_len-1] = '\0'; // Make sure the error string is terminated
239 }
240 }
241 }
242
243 DNBLogThreadedIf(LOG_PROCESS, "(DebugNub) new pid is %d...", pid);
244
245 if (pid != INVALID_NUB_PROCESS)
246 {
247 // Spawn a thread to reap our child inferior process...
248 spawn_waitpid_thread (pid);
249
250 if (processSP->Task().TaskPortForProcessID (launch_err) == TASK_NULL)
251 {
252 // We failed to get the task for our process ID which is bad.
Greg Claytonfb640c22012-02-02 19:23:22 +0000253 // Kill our process otherwise it will be stopped at the entry
254 // point and get reparented to someone else and never go away.
255 kill (SIGKILL, pid);
256
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000257 if (err_str && err_len > 0)
258 {
259 if (launch_err.AsString())
260 {
261 ::snprintf (err_str, err_len, "failed to get the task for process %i (%s)", pid, launch_err.AsString());
262 }
263 else
264 {
265 ::snprintf (err_str, err_len, "failed to get the task for process %i", pid);
266 }
267 }
268 }
269 else
270 {
Charles Davisb786e7d2012-02-21 00:53:12 +0000271 bool res = AddProcessToMap(pid, processSP);
272 assert(res && "Couldn't add process to map!");
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000273 return pid;
274 }
275 }
276 }
277 return INVALID_NUB_PROCESS;
278}
279
280nub_process_t
281DNBProcessAttachByName (const char *name, struct timespec *timeout, char *err_str, size_t err_len)
282{
283 if (err_str && err_len > 0)
284 err_str[0] = '\0';
285 std::vector<struct kinfo_proc> matching_proc_infos;
286 size_t num_matching_proc_infos = GetAllInfosMatchingName(name, matching_proc_infos);
287 if (num_matching_proc_infos == 0)
288 {
289 DNBLogError ("error: no processes match '%s'\n", name);
290 return INVALID_NUB_PROCESS;
291 }
292 else if (num_matching_proc_infos > 1)
293 {
Greg Clayton490fbbe2011-10-28 22:59:14 +0000294 DNBLogError ("error: %zu processes match '%s':\n", num_matching_proc_infos, name);
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000295 size_t i;
296 for (i=0; i<num_matching_proc_infos; ++i)
297 DNBLogError ("%6u - %s\n", matching_proc_infos[i].kp_proc.p_pid, matching_proc_infos[i].kp_proc.p_comm);
298 return INVALID_NUB_PROCESS;
299 }
Greg Clayton3af9ea52010-11-18 05:57:03 +0000300
301 return DNBProcessAttach (matching_proc_infos[0].kp_proc.p_pid, timeout, err_str, err_len);
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000302}
303
304nub_process_t
305DNBProcessAttach (nub_process_t attach_pid, struct timespec *timeout, char *err_str, size_t err_len)
306{
307 if (err_str && err_len > 0)
308 err_str[0] = '\0';
309
Johnny Chen64503c82011-08-11 19:03:44 +0000310 pid_t pid = INVALID_NUB_PROCESS;
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000311 MachProcessSP processSP(new MachProcess);
312 if (processSP.get())
313 {
314 DNBLogThreadedIf(LOG_PROCESS, "(DebugNub) attaching to pid %d...", attach_pid);
315 pid = processSP->AttachForDebug (attach_pid, err_str, err_len);
316
317 if (pid != INVALID_NUB_PROCESS)
318 {
Charles Davisb786e7d2012-02-21 00:53:12 +0000319 bool res = AddProcessToMap(pid, processSP);
320 assert(res && "Couldn't add process to map!");
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000321 spawn_waitpid_thread(pid);
322 }
323 }
324
325 while (pid != INVALID_NUB_PROCESS)
326 {
327 // Wait for process to start up and hit entry point
Greg Clayton3af9ea52010-11-18 05:57:03 +0000328 DNBLogThreadedIf (LOG_PROCESS,
329 "%s DNBProcessWaitForEvent (%4.4x, eEventProcessRunningStateChanged | eEventProcessStoppedStateChanged, true, INFINITE)...",
330 __FUNCTION__,
331 pid);
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000332 nub_event_t set_events = DNBProcessWaitForEvents (pid,
Greg Clayton3af9ea52010-11-18 05:57:03 +0000333 eEventProcessRunningStateChanged | eEventProcessStoppedStateChanged,
334 true,
335 timeout);
336
337 DNBLogThreadedIf (LOG_PROCESS,
338 "%s DNBProcessWaitForEvent (%4.4x, eEventProcessRunningStateChanged | eEventProcessStoppedStateChanged, true, INFINITE) => 0x%8.8x",
339 __FUNCTION__,
340 pid,
341 set_events);
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000342
343 if (set_events == 0)
344 {
345 if (err_str && err_len > 0)
346 snprintf(err_str, err_len, "operation timed out");
347 pid = INVALID_NUB_PROCESS;
348 }
349 else
350 {
351 if (set_events & (eEventProcessRunningStateChanged | eEventProcessStoppedStateChanged))
352 {
353 nub_state_t pid_state = DNBProcessGetState (pid);
354 DNBLogThreadedIf (LOG_PROCESS, "%s process %4.4x state changed (eEventProcessStateChanged): %s",
355 __FUNCTION__, pid, DNBStateAsString(pid_state));
356
357 switch (pid_state)
358 {
Greg Clayton3af9ea52010-11-18 05:57:03 +0000359 default:
360 case eStateInvalid:
361 case eStateUnloaded:
362 case eStateAttaching:
363 case eStateLaunching:
364 case eStateSuspended:
365 break; // Ignore
366
367 case eStateRunning:
368 case eStateStepping:
369 // Still waiting to stop at entry point...
370 break;
371
372 case eStateStopped:
373 case eStateCrashed:
374 return pid;
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000375
Greg Clayton3af9ea52010-11-18 05:57:03 +0000376 case eStateDetached:
377 case eStateExited:
378 if (err_str && err_len > 0)
379 snprintf(err_str, err_len, "process exited");
380 return INVALID_NUB_PROCESS;
381 }
382 }
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000383
384 DNBProcessResetEvents(pid, set_events);
385 }
386 }
387
388 return INVALID_NUB_PROCESS;
389}
390
391static size_t
Greg Clayton3af9ea52010-11-18 05:57:03 +0000392GetAllInfos (std::vector<struct kinfo_proc>& proc_infos)
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000393{
394 size_t size;
395 int name[] = { CTL_KERN, KERN_PROC, KERN_PROC_ALL };
396 u_int namelen = sizeof(name)/sizeof(int);
397 int err;
398
399 // Try to find out how many processes are around so we can
400 // size the buffer appropriately. sysctl's man page specifically suggests
401 // this approach, and says it returns a bit larger size than needed to
402 // handle any new processes created between then and now.
403
404 err = ::sysctl (name, namelen, NULL, &size, NULL, 0);
405
406 if ((err < 0) && (err != ENOMEM))
407 {
408 proc_infos.clear();
409 perror("sysctl (mib, miblen, NULL, &num_processes, NULL, 0)");
410 return 0;
411 }
412
413
414 // Increase the size of the buffer by a few processes in case more have
415 // been spawned
416 proc_infos.resize (size / sizeof(struct kinfo_proc));
417 size = proc_infos.size() * sizeof(struct kinfo_proc); // Make sure we don't exceed our resize...
418 err = ::sysctl (name, namelen, &proc_infos[0], &size, NULL, 0);
419 if (err < 0)
420 {
421 proc_infos.clear();
422 return 0;
423 }
424
425 // Trim down our array to fit what we actually got back
426 proc_infos.resize(size / sizeof(struct kinfo_proc));
427 return proc_infos.size();
428}
429
430
431static size_t
432GetAllInfosMatchingName(const char *full_process_name, std::vector<struct kinfo_proc>& matching_proc_infos)
433{
434
435 matching_proc_infos.clear();
436 if (full_process_name && full_process_name[0])
437 {
438 // We only get the process name, not the full path, from the proc_info. So just take the
439 // base name of the process name...
440 const char *process_name;
441 process_name = strrchr (full_process_name, '/');
442 if (process_name == NULL)
Greg Clayton7dab2be2012-07-19 02:45:35 +0000443 process_name = full_process_name;
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000444 else
Greg Clayton7dab2be2012-07-19 02:45:35 +0000445 process_name++;
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000446
Greg Clayton7dab2be2012-07-19 02:45:35 +0000447 const int process_name_len = strlen(process_name);
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000448 std::vector<struct kinfo_proc> proc_infos;
449 const size_t num_proc_infos = GetAllInfos(proc_infos);
450 if (num_proc_infos > 0)
451 {
452 uint32_t i;
453 for (i=0; i<num_proc_infos; i++)
454 {
455 // Skip zombie processes and processes with unset status
456 if (proc_infos[i].kp_proc.p_stat == 0 || proc_infos[i].kp_proc.p_stat == SZOMB)
457 continue;
458
459 // Check for process by name. We only check the first MAXCOMLEN
460 // chars as that is all that kp_proc.p_comm holds.
Greg Clayton7dab2be2012-07-19 02:45:35 +0000461 if (::strncasecmp(process_name, proc_infos[i].kp_proc.p_comm, MAXCOMLEN) == 0)
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000462 {
Greg Clayton7dab2be2012-07-19 02:45:35 +0000463 if (process_name_len > MAXCOMLEN)
464 {
465 // We found a matching process name whose first MAXCOMLEN
466 // characters match, but there is more to the name than
467 // this. We need to get the full process name.
468
469 int proc_args_mib[3] = { CTL_KERN, KERN_PROCARGS2, proc_infos[i].kp_proc.p_pid };
470
471 // Get PATH_MAX for argv[0] plus 4 bytes for the argc
472 char arg_data[PATH_MAX+4];
473 size_t arg_data_size = sizeof(arg_data);
474 // Skip the 4 byte argc integer value to get to argv[0]
475 const char *argv0 = arg_data + 4;
476 if (::sysctl (proc_args_mib, 3, arg_data, &arg_data_size , NULL, 0) == 0)
477 {
478 const char *argv_basename = strrchr(argv0, '/');
479 if (argv_basename)
480 {
481 // Skip the '/'
482 ++argv_basename;
483 }
484 else
485 {
486 // We didn't find a directory delimiter in the process argv[0], just use what was in there
487 argv_basename = argv0;
488 }
489
490 if (argv_basename)
491 {
492 if (::strncasecmp(process_name, argv_basename, PATH_MAX) == 0)
493 {
494 matching_proc_infos.push_back(proc_infos[i]);
495 }
496 }
497 }
498 }
499 else
500 {
501 // We found a matching process, add it to our list
Greg Clayton7dab2be2012-07-19 02:45:35 +0000502 matching_proc_infos.push_back(proc_infos[i]);
503 }
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000504 }
505 }
506 }
507 }
508 // return the newly added matches.
509 return matching_proc_infos.size();
510}
511
512nub_process_t
Greg Clayton19388cf2010-10-18 01:45:30 +0000513DNBProcessAttachWait (const char *waitfor_process_name,
514 nub_launch_flavor_t launch_flavor,
Jim Inghamcd16df92012-07-20 21:37:13 +0000515 bool ignore_existing,
Greg Clayton19388cf2010-10-18 01:45:30 +0000516 struct timespec *timeout_abstime,
517 useconds_t waitfor_interval,
518 char *err_str,
519 size_t err_len,
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000520 DNBShouldCancelCallback should_cancel_callback,
521 void *callback_data)
522{
523 DNBError prepare_error;
524 std::vector<struct kinfo_proc> exclude_proc_infos;
525 size_t num_exclude_proc_infos;
526
527 // If the PrepareForAttach returns a valid token, use MachProcess to check
528 // for the process, otherwise scan the process table.
529
530 const void *attach_token = MachProcess::PrepareForAttach (waitfor_process_name, launch_flavor, true, prepare_error);
531
532 if (prepare_error.Fail())
533 {
534 DNBLogError ("Error in PrepareForAttach: %s", prepare_error.AsString());
535 return INVALID_NUB_PROCESS;
536 }
537
538 if (attach_token == NULL)
Jim Inghamcd16df92012-07-20 21:37:13 +0000539 {
540 if (ignore_existing)
541 num_exclude_proc_infos = GetAllInfosMatchingName (waitfor_process_name, exclude_proc_infos);
542 else
543 num_exclude_proc_infos = 0;
544 }
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000545
546 DNBLogThreadedIf (LOG_PROCESS, "Waiting for '%s' to appear...\n", waitfor_process_name);
547
548 // Loop and try to find the process by name
549 nub_process_t waitfor_pid = INVALID_NUB_PROCESS;
550
551 while (waitfor_pid == INVALID_NUB_PROCESS)
552 {
553 if (attach_token != NULL)
554 {
555 nub_process_t pid;
556 pid = MachProcess::CheckForProcess(attach_token);
557 if (pid != INVALID_NUB_PROCESS)
558 {
559 waitfor_pid = pid;
560 break;
561 }
562 }
563 else
564 {
565
566 // Get the current process list, and check for matches that
567 // aren't in our original list. If anyone wants to attach
568 // to an existing process by name, they should do it with
569 // --attach=PROCNAME. Else we will wait for the first matching
570 // process that wasn't in our exclusion list.
571 std::vector<struct kinfo_proc> proc_infos;
572 const size_t num_proc_infos = GetAllInfosMatchingName (waitfor_process_name, proc_infos);
573 for (size_t i=0; i<num_proc_infos; i++)
574 {
575 nub_process_t curr_pid = proc_infos[i].kp_proc.p_pid;
576 for (size_t j=0; j<num_exclude_proc_infos; j++)
577 {
578 if (curr_pid == exclude_proc_infos[j].kp_proc.p_pid)
579 {
580 // This process was in our exclusion list, don't use it.
581 curr_pid = INVALID_NUB_PROCESS;
582 break;
583 }
584 }
585
586 // If we didn't find CURR_PID in our exclusion list, then use it.
587 if (curr_pid != INVALID_NUB_PROCESS)
588 {
589 // We found our process!
590 waitfor_pid = curr_pid;
591 break;
592 }
593 }
594 }
595
596 // If we haven't found our process yet, check for a timeout
597 // and then sleep for a bit until we poll again.
598 if (waitfor_pid == INVALID_NUB_PROCESS)
599 {
600 if (timeout_abstime != NULL)
601 {
602 // Check to see if we have a waitfor-duration option that
603 // has timed out?
604 if (DNBTimer::TimeOfDayLaterThan(*timeout_abstime))
605 {
606 if (err_str && err_len > 0)
607 snprintf(err_str, err_len, "operation timed out");
608 DNBLogError ("error: waiting for process '%s' timed out.\n", waitfor_process_name);
609 return INVALID_NUB_PROCESS;
610 }
611 }
612
613 // Call the should cancel callback as well...
614
615 if (should_cancel_callback != NULL
616 && should_cancel_callback (callback_data))
617 {
618 DNBLogThreadedIf (LOG_PROCESS, "DNBProcessAttachWait cancelled by should_cancel callback.");
619 waitfor_pid = INVALID_NUB_PROCESS;
620 break;
621 }
622
623 ::usleep (waitfor_interval); // Sleep for WAITFOR_INTERVAL, then poll again
624 }
625 }
626
627 if (waitfor_pid != INVALID_NUB_PROCESS)
628 {
629 DNBLogThreadedIf (LOG_PROCESS, "Attaching to %s with pid %i...\n", waitfor_process_name, waitfor_pid);
630 waitfor_pid = DNBProcessAttach (waitfor_pid, timeout_abstime, err_str, err_len);
631 }
632
633 bool success = waitfor_pid != INVALID_NUB_PROCESS;
634 MachProcess::CleanupAfterAttach (attach_token, success, prepare_error);
635
636 return waitfor_pid;
637}
638
639nub_bool_t
640DNBProcessDetach (nub_process_t pid)
641{
642 MachProcessSP procSP;
643 if (GetProcessSP (pid, procSP))
644 {
645 return procSP->Detach();
646 }
647 return false;
648}
649
650nub_bool_t
651DNBProcessKill (nub_process_t pid)
652{
653 MachProcessSP procSP;
654 if (GetProcessSP (pid, procSP))
655 {
656 return procSP->Kill ();
657 }
658 return false;
659}
660
661nub_bool_t
662DNBProcessSignal (nub_process_t pid, int signal)
663{
664 MachProcessSP procSP;
665 if (GetProcessSP (pid, procSP))
666 {
667 return procSP->Signal (signal);
668 }
669 return false;
670}
671
672
673nub_bool_t
674DNBProcessIsAlive (nub_process_t pid)
675{
676 MachProcessSP procSP;
677 if (GetProcessSP (pid, procSP))
678 {
679 return MachTask::IsValid (procSP->Task().TaskPort());
680 }
681 return eStateInvalid;
682}
683
684//----------------------------------------------------------------------
685// Process and Thread state information
686//----------------------------------------------------------------------
687nub_state_t
688DNBProcessGetState (nub_process_t pid)
689{
690 MachProcessSP procSP;
691 if (GetProcessSP (pid, procSP))
692 {
693 return procSP->GetState();
694 }
695 return eStateInvalid;
696}
697
698//----------------------------------------------------------------------
699// Process and Thread state information
700//----------------------------------------------------------------------
701nub_bool_t
702DNBProcessGetExitStatus (nub_process_t pid, int* status)
703{
704 MachProcessSP procSP;
705 if (GetProcessSP (pid, procSP))
706 {
707 return procSP->GetExitStatus(status);
708 }
709 return false;
710}
711
712nub_bool_t
713DNBProcessSetExitStatus (nub_process_t pid, int status)
714{
715 MachProcessSP procSP;
716 if (GetProcessSP (pid, procSP))
717 {
718 procSP->SetExitStatus(status);
719 return true;
720 }
721 return false;
722}
723
724
725const char *
726DNBThreadGetName (nub_process_t pid, nub_thread_t tid)
727{
728 MachProcessSP procSP;
729 if (GetProcessSP (pid, procSP))
730 return procSP->ThreadGetName(tid);
731 return NULL;
732}
733
734
735nub_bool_t
736DNBThreadGetIdentifierInfo (nub_process_t pid, nub_thread_t tid, thread_identifier_info_data_t *ident_info)
737{
738 MachProcessSP procSP;
739 if (GetProcessSP (pid, procSP))
740 return procSP->GetThreadList().GetIdentifierInfo(tid, ident_info);
741 return false;
742}
743
744nub_state_t
745DNBThreadGetState (nub_process_t pid, nub_thread_t tid)
746{
747 MachProcessSP procSP;
748 if (GetProcessSP (pid, procSP))
749 {
750 return procSP->ThreadGetState(tid);
751 }
752 return eStateInvalid;
753}
754
755const char *
756DNBStateAsString(nub_state_t state)
757{
758 switch (state)
759 {
Greg Claytoneffe5c92011-05-03 22:09:39 +0000760 case eStateInvalid: return "Invalid";
Chris Lattner30fdc8d2010-06-08 16:52:24 +0000761 case eStateUnloaded: return "Unloaded";
762 case eStateAttaching: return "Attaching";
763 case eStateLaunching: return "Launching";
764 case eStateStopped: return "Stopped";
765 case eStateRunning: return "Running";
766 case eStateStepping: return "Stepping";
767 case eStateCrashed: return "Crashed";
768 case eStateDetached: return "Detached";
769 case eStateExited: return "Exited";
770 case eStateSuspended: return "Suspended";
771 }
772 return "nub_state_t ???";
773}
774
775const char *
776DNBProcessGetExecutablePath (nub_process_t pid)
777{
778 MachProcessSP procSP;
779 if (GetProcessSP (pid, procSP))
780 {
781 return procSP->Path();
782 }
783 return NULL;
784}
785
786nub_size_t
787DNBProcessGetArgumentCount (nub_process_t pid)
788{
789 MachProcessSP procSP;
790 if (GetProcessSP (pid, procSP))
791 {
792 return procSP->ArgumentCount();
793 }
794 return 0;
795}
796
797const char *
798DNBProcessGetArgumentAtIndex (nub_process_t pid, nub_size_t idx)
799{
800 MachProcessSP procSP;
801 if (GetProcessSP (pid, procSP))
802 {
803 return procSP->ArgumentAtIndex (idx);
804 }
805 return NULL;
806}
807
808
809//----------------------------------------------------------------------
810// Execution control
811//----------------------------------------------------------------------
812nub_bool_t
813DNBProcessResume (nub_process_t pid, const DNBThreadResumeAction *actions, size_t num_actions)
814{
815 DNBLogThreadedIf(LOG_PROCESS, "%s(pid = %4.4x)", __FUNCTION__, pid);
816 MachProcessSP procSP;
817 if (GetProcessSP (pid, procSP))
818 {
819 DNBThreadResumeActions thread_actions (actions, num_actions);
820
821 // Below we add a default thread plan just in case one wasn't
822 // provided so all threads always know what they were supposed to do
823 if (thread_actions.IsEmpty())
824 {
825 // No thread plans were given, so the default it to run all threads
826 thread_actions.SetDefaultThreadActionIfNeeded (eStateRunning, 0);
827 }
828 else
829 {
830 // Some thread plans were given which means anything that wasn't
831 // specified should remain stopped.
832 thread_actions.SetDefaultThreadActionIfNeeded (eStateStopped, 0);
833 }
834 return procSP->Resume (thread_actions);
835 }
836 return false;
837}
838
839nub_bool_t
840DNBProcessHalt (nub_process_t pid)
841{
842 DNBLogThreadedIf(LOG_PROCESS, "%s(pid = %4.4x)", __FUNCTION__, pid);
843 MachProcessSP procSP;
844 if (GetProcessSP (pid, procSP))
845 return procSP->Signal (SIGSTOP);
846 return false;
847}
848//
849//nub_bool_t
850//DNBThreadResume (nub_process_t pid, nub_thread_t tid, nub_bool_t step)
851//{
852// DNBLogThreadedIf(LOG_THREAD, "%s(pid = %4.4x, tid = %4.4x, step = %u)", __FUNCTION__, pid, tid, (uint32_t)step);
853// MachProcessSP procSP;
854// if (GetProcessSP (pid, procSP))
855// {
856// return procSP->Resume(tid, step, 0);
857// }
858// return false;
859//}
860//
861//nub_bool_t
862//DNBThreadResumeWithSignal (nub_process_t pid, nub_thread_t tid, nub_bool_t step, int signal)
863//{
864// DNBLogThreadedIf(LOG_THREAD, "%s(pid = %4.4x, tid = %4.4x, step = %u, signal = %i)", __FUNCTION__, pid, tid, (uint32_t)step, signal);
865// MachProcessSP procSP;
866// if (GetProcessSP (pid, procSP))
867// {
868// return procSP->Resume(tid, step, signal);
869// }
870// return false;
871//}
872
873nub_event_t
874DNBProcessWaitForEvents (nub_process_t pid, nub_event_t event_mask, bool wait_for_set, struct timespec* timeout)
875{
876 nub_event_t result = 0;
877 MachProcessSP procSP;
878 if (GetProcessSP (pid, procSP))
879 {
880 if (wait_for_set)
881 result = procSP->Events().WaitForSetEvents(event_mask, timeout);
882 else
883 result = procSP->Events().WaitForEventsToReset(event_mask, timeout);
884 }
885 return result;
886}
887
888void
889DNBProcessResetEvents (nub_process_t pid, nub_event_t event_mask)
890{
891 MachProcessSP procSP;
892 if (GetProcessSP (pid, procSP))
893 procSP->Events().ResetEvents(event_mask);
894}
895
896void
897DNBProcessInterruptEvents (nub_process_t pid)
898{
899 MachProcessSP procSP;
900 if (GetProcessSP (pid, procSP))
901 procSP->Events().SetEvents(eEventProcessAsyncInterrupt);
902}
903
904
905// Breakpoints
906nub_break_t
907DNBBreakpointSet (nub_process_t pid, nub_addr_t addr, nub_size_t size, nub_bool_t hardware)
908{
909 MachProcessSP procSP;
910 if (GetProcessSP (pid, procSP))
911 {
912 return procSP->CreateBreakpoint(addr, size, hardware, THREAD_NULL);
913 }
914 return INVALID_NUB_BREAK_ID;
915}
916
917nub_bool_t
918DNBBreakpointClear (nub_process_t pid, nub_break_t breakID)
919{
920 if (NUB_BREAK_ID_IS_VALID(breakID))
921 {
922 MachProcessSP procSP;
923 if (GetProcessSP (pid, procSP))
924 {
925 return procSP->DisableBreakpoint(breakID, true);
926 }
927 }
928 return false; // Failed
929}
930
931nub_ssize_t
932DNBBreakpointGetHitCount (nub_process_t pid, nub_break_t breakID)
933{
934 if (NUB_BREAK_ID_IS_VALID(breakID))
935 {
936 MachProcessSP procSP;
937 if (GetProcessSP (pid, procSP))
938 {
939 DNBBreakpoint *bp = procSP->Breakpoints().FindByID(breakID);
940 if (bp)
941 return bp->GetHitCount();
942 }
943 }
944 return 0;
945}
946
947nub_ssize_t
948DNBBreakpointGetIgnoreCount (nub_process_t pid, nub_break_t breakID)
949{
950 if (NUB_BREAK_ID_IS_VALID(breakID))
951 {
952 MachProcessSP procSP;
953 if (GetProcessSP (pid, procSP))
954 {
955 DNBBreakpoint *bp = procSP->Breakpoints().FindByID(breakID);
956 if (bp)
957 return bp->GetIgnoreCount();
958 }
959 }
960 return 0;
961}
962
963nub_bool_t
964DNBBreakpointSetIgnoreCount (nub_process_t pid, nub_break_t breakID, nub_size_t ignore_count)
965{
966 if (NUB_BREAK_ID_IS_VALID(breakID))
967 {
968 MachProcessSP procSP;
969 if (GetProcessSP (pid, procSP))
970 {
971 DNBBreakpoint *bp = procSP->Breakpoints().FindByID(breakID);
972 if (bp)
973 {
974 bp->SetIgnoreCount(ignore_count);
975 return true;
976 }
977 }
978 }
979 return false;
980}
981
982// Set the callback function for a given breakpoint. The callback function will
983// get called as soon as the breakpoint is hit. The function will be called
984// with the process ID, thread ID, breakpoint ID and the baton, and can return
985//
986nub_bool_t
987DNBBreakpointSetCallback (nub_process_t pid, nub_break_t breakID, DNBCallbackBreakpointHit callback, void *baton)
988{
989 if (NUB_BREAK_ID_IS_VALID(breakID))
990 {
991 MachProcessSP procSP;
992 if (GetProcessSP (pid, procSP))
993 {
994 DNBBreakpoint *bp = procSP->Breakpoints().FindByID(breakID);
995 if (bp)
996 {
997 bp->SetCallback(callback, baton);
998 return true;
999 }
1000 }
1001 }
1002 return false;
1003}
1004
1005//----------------------------------------------------------------------
1006// Dump the breakpoints stats for process PID for a breakpoint by ID.
1007//----------------------------------------------------------------------
1008void
1009DNBBreakpointPrint (nub_process_t pid, nub_break_t breakID)
1010{
1011 MachProcessSP procSP;
1012 if (GetProcessSP (pid, procSP))
1013 procSP->DumpBreakpoint(breakID);
1014}
1015
1016//----------------------------------------------------------------------
1017// Watchpoints
1018//----------------------------------------------------------------------
1019nub_watch_t
1020DNBWatchpointSet (nub_process_t pid, nub_addr_t addr, nub_size_t size, uint32_t watch_flags, nub_bool_t hardware)
1021{
1022 MachProcessSP procSP;
1023 if (GetProcessSP (pid, procSP))
1024 {
1025 return procSP->CreateWatchpoint(addr, size, watch_flags, hardware, THREAD_NULL);
1026 }
Johnny Chen86f97a42011-09-06 19:52:49 +00001027 return INVALID_NUB_WATCH_ID;
Chris Lattner30fdc8d2010-06-08 16:52:24 +00001028}
1029
1030nub_bool_t
1031DNBWatchpointClear (nub_process_t pid, nub_watch_t watchID)
1032{
Johnny Chen86f97a42011-09-06 19:52:49 +00001033 if (NUB_WATCH_ID_IS_VALID(watchID))
Chris Lattner30fdc8d2010-06-08 16:52:24 +00001034 {
1035 MachProcessSP procSP;
1036 if (GetProcessSP (pid, procSP))
1037 {
1038 return procSP->DisableWatchpoint(watchID, true);
1039 }
1040 }
1041 return false; // Failed
1042}
1043
1044nub_ssize_t
1045DNBWatchpointGetHitCount (nub_process_t pid, nub_watch_t watchID)
1046{
Johnny Chen86f97a42011-09-06 19:52:49 +00001047 if (NUB_WATCH_ID_IS_VALID(watchID))
Chris Lattner30fdc8d2010-06-08 16:52:24 +00001048 {
1049 MachProcessSP procSP;
1050 if (GetProcessSP (pid, procSP))
1051 {
1052 DNBBreakpoint *bp = procSP->Watchpoints().FindByID(watchID);
1053 if (bp)
1054 return bp->GetHitCount();
1055 }
1056 }
1057 return 0;
1058}
1059
1060nub_ssize_t
1061DNBWatchpointGetIgnoreCount (nub_process_t pid, nub_watch_t watchID)
1062{
Johnny Chen86f97a42011-09-06 19:52:49 +00001063 if (NUB_WATCH_ID_IS_VALID(watchID))
Chris Lattner30fdc8d2010-06-08 16:52:24 +00001064 {
1065 MachProcessSP procSP;
1066 if (GetProcessSP (pid, procSP))
1067 {
1068 DNBBreakpoint *bp = procSP->Watchpoints().FindByID(watchID);
1069 if (bp)
1070 return bp->GetIgnoreCount();
1071 }
1072 }
1073 return 0;
1074}
1075
1076nub_bool_t
1077DNBWatchpointSetIgnoreCount (nub_process_t pid, nub_watch_t watchID, nub_size_t ignore_count)
1078{
Johnny Chen86f97a42011-09-06 19:52:49 +00001079 if (NUB_WATCH_ID_IS_VALID(watchID))
Chris Lattner30fdc8d2010-06-08 16:52:24 +00001080 {
1081 MachProcessSP procSP;
1082 if (GetProcessSP (pid, procSP))
1083 {
1084 DNBBreakpoint *bp = procSP->Watchpoints().FindByID(watchID);
1085 if (bp)
1086 {
1087 bp->SetIgnoreCount(ignore_count);
1088 return true;
1089 }
1090 }
1091 }
1092 return false;
1093}
1094
1095// Set the callback function for a given watchpoint. The callback function will
1096// get called as soon as the watchpoint is hit. The function will be called
1097// with the process ID, thread ID, watchpoint ID and the baton, and can return
1098//
1099nub_bool_t
1100DNBWatchpointSetCallback (nub_process_t pid, nub_watch_t watchID, DNBCallbackBreakpointHit callback, void *baton)
1101{
Johnny Chen86f97a42011-09-06 19:52:49 +00001102 if (NUB_WATCH_ID_IS_VALID(watchID))
Chris Lattner30fdc8d2010-06-08 16:52:24 +00001103 {
1104 MachProcessSP procSP;
1105 if (GetProcessSP (pid, procSP))
1106 {
1107 DNBBreakpoint *bp = procSP->Watchpoints().FindByID(watchID);
1108 if (bp)
1109 {
1110 bp->SetCallback(callback, baton);
1111 return true;
1112 }
1113 }
1114 }
1115 return false;
1116}
1117
1118//----------------------------------------------------------------------
1119// Dump the watchpoints stats for process PID for a watchpoint by ID.
1120//----------------------------------------------------------------------
1121void
1122DNBWatchpointPrint (nub_process_t pid, nub_watch_t watchID)
1123{
1124 MachProcessSP procSP;
1125 if (GetProcessSP (pid, procSP))
1126 procSP->DumpWatchpoint(watchID);
1127}
1128
1129//----------------------------------------------------------------------
Johnny Chen64637202012-05-23 21:09:52 +00001130// Return the number of supported hardware watchpoints.
1131//----------------------------------------------------------------------
1132uint32_t
1133DNBWatchpointGetNumSupportedHWP (nub_process_t pid)
1134{
1135 MachProcessSP procSP;
1136 if (GetProcessSP (pid, procSP))
1137 return procSP->GetNumSupportedHardwareWatchpoints();
1138 return 0;
1139}
1140
1141//----------------------------------------------------------------------
Chris Lattner30fdc8d2010-06-08 16:52:24 +00001142// Read memory in the address space of process PID. This call will take
1143// care of setting and restoring permissions and breaking up the memory
1144// read into multiple chunks as required.
1145//
1146// RETURNS: number of bytes actually read
1147//----------------------------------------------------------------------
1148nub_size_t
1149DNBProcessMemoryRead (nub_process_t pid, nub_addr_t addr, nub_size_t size, void *buf)
1150{
1151 MachProcessSP procSP;
1152 if (GetProcessSP (pid, procSP))
1153 return procSP->ReadMemory(addr, size, buf);
1154 return 0;
1155}
1156
1157//----------------------------------------------------------------------
1158// Write memory to the address space of process PID. This call will take
1159// care of setting and restoring permissions and breaking up the memory
1160// write into multiple chunks as required.
1161//
1162// RETURNS: number of bytes actually written
1163//----------------------------------------------------------------------
1164nub_size_t
1165DNBProcessMemoryWrite (nub_process_t pid, nub_addr_t addr, nub_size_t size, const void *buf)
1166{
1167 MachProcessSP procSP;
1168 if (GetProcessSP (pid, procSP))
1169 return procSP->WriteMemory(addr, size, buf);
1170 return 0;
1171}
1172
1173nub_addr_t
1174DNBProcessMemoryAllocate (nub_process_t pid, nub_size_t size, uint32_t permissions)
1175{
1176 MachProcessSP procSP;
1177 if (GetProcessSP (pid, procSP))
1178 return procSP->Task().AllocateMemory (size, permissions);
1179 return 0;
1180}
1181
1182nub_bool_t
1183DNBProcessMemoryDeallocate (nub_process_t pid, nub_addr_t addr)
1184{
1185 MachProcessSP procSP;
1186 if (GetProcessSP (pid, procSP))
1187 return procSP->Task().DeallocateMemory (addr);
1188 return 0;
1189}
1190
Jason Molenda1f3966b2011-11-08 04:28:12 +00001191//----------------------------------------------------------------------
Jason Molenda3dc85832011-11-09 08:03:56 +00001192// Find attributes of the memory region that contains ADDR for process PID,
1193// if possible, and return a string describing those attributes.
Jason Molenda1f3966b2011-11-08 04:28:12 +00001194//
Jason Molenda3dc85832011-11-09 08:03:56 +00001195// Returns 1 if we could find attributes for this region and OUTBUF can
1196// be sent to the remote debugger.
Jason Molenda1f3966b2011-11-08 04:28:12 +00001197//
Jason Molenda3dc85832011-11-09 08:03:56 +00001198// Returns 0 if we couldn't find the attributes for a region of memory at
1199// that address and OUTBUF should not be sent.
1200//
1201// Returns -1 if this platform cannot look up information about memory regions
1202// or if we do not yet have a valid launched process.
1203//
Jason Molenda1f3966b2011-11-08 04:28:12 +00001204//----------------------------------------------------------------------
1205int
Greg Claytonfc5dd292011-12-12 18:51:14 +00001206DNBProcessMemoryRegionInfo (nub_process_t pid, nub_addr_t addr, DNBRegionInfo *region_info)
Jason Molenda1f3966b2011-11-08 04:28:12 +00001207{
1208 MachProcessSP procSP;
1209 if (GetProcessSP (pid, procSP))
Greg Clayton46fb5582011-11-18 07:03:08 +00001210 return procSP->Task().GetMemoryRegionInfo (addr, region_info);
1211
Jason Molenda1f3966b2011-11-08 04:28:12 +00001212 return -1;
1213}
1214
Chris Lattner30fdc8d2010-06-08 16:52:24 +00001215
1216//----------------------------------------------------------------------
1217// Formatted output that uses memory and registers from process and
1218// thread in place of arguments.
1219//----------------------------------------------------------------------
1220nub_size_t
1221DNBPrintf (nub_process_t pid, nub_thread_t tid, nub_addr_t base_addr, FILE *file, const char *format)
1222{
1223 if (file == NULL)
1224 return 0;
1225 enum printf_flags
1226 {
1227 alternate_form = (1 << 0),
1228 zero_padding = (1 << 1),
1229 negative_field_width = (1 << 2),
1230 blank_space = (1 << 3),
1231 show_sign = (1 << 4),
1232 show_thousands_separator= (1 << 5),
1233 };
1234
1235 enum printf_length_modifiers
1236 {
1237 length_mod_h = (1 << 0),
1238 length_mod_hh = (1 << 1),
1239 length_mod_l = (1 << 2),
1240 length_mod_ll = (1 << 3),
1241 length_mod_L = (1 << 4),
1242 length_mod_j = (1 << 5),
1243 length_mod_t = (1 << 6),
1244 length_mod_z = (1 << 7),
1245 length_mod_q = (1 << 8),
1246 };
1247
1248 nub_addr_t addr = base_addr;
1249 char *end_format = (char*)format + strlen(format);
1250 char *end = NULL; // For strtoXXXX calls;
1251 std::basic_string<uint8_t> buf;
1252 nub_size_t total_bytes_read = 0;
1253 DNBDataRef data;
1254 const char *f;
1255 for (f = format; *f != '\0' && f < end_format; f++)
1256 {
1257 char ch = *f;
1258 switch (ch)
1259 {
1260 case '%':
1261 {
1262 f++; // Skip the '%' character
Greg Clayton23f59502012-07-17 03:23:13 +00001263// int min_field_width = 0;
1264// int precision = 0;
1265 //uint32_t flags = 0;
Chris Lattner30fdc8d2010-06-08 16:52:24 +00001266 uint32_t length_modifiers = 0;
1267 uint32_t byte_size = 0;
1268 uint32_t actual_byte_size = 0;
1269 bool is_string = false;
1270 bool is_register = false;
1271 DNBRegisterValue register_value;
1272 int64_t register_offset = 0;
1273 nub_addr_t register_addr = INVALID_NUB_ADDRESS;
1274
1275 // Create the format string to use for this conversion specification
1276 // so we can remove and mprintf specific flags and formatters.
1277 std::string fprintf_format("%");
1278
1279 // Decode any flags
1280 switch (*f)
1281 {
Greg Clayton23f59502012-07-17 03:23:13 +00001282 case '#': fprintf_format += *f++; break; //flags |= alternate_form; break;
1283 case '0': fprintf_format += *f++; break; //flags |= zero_padding; break;
1284 case '-': fprintf_format += *f++; break; //flags |= negative_field_width; break;
1285 case ' ': fprintf_format += *f++; break; //flags |= blank_space; break;
1286 case '+': fprintf_format += *f++; break; //flags |= show_sign; break;
1287 case ',': fprintf_format += *f++; break; //flags |= show_thousands_separator;break;
Chris Lattner30fdc8d2010-06-08 16:52:24 +00001288 case '{':
1289 case '[':
1290 {
1291 // We have a register name specification that can take two forms:
1292 // ${regname} or ${regname+offset}
1293 // The action is to read the register value and add the signed offset
1294 // (if any) and use that as the value to format.
1295 // $[regname] or $[regname+offset]
1296 // The action is to read the register value and add the signed offset
1297 // (if any) and use the result as an address to dereference. The size
1298 // of what is dereferenced is specified by the actual byte size that
1299 // follows the minimum field width and precision (see comments below).
1300 switch (*f)
1301 {
1302 case '{':
1303 case '[':
1304 {
1305 char open_scope_ch = *f;
1306 f++;
1307 const char *reg_name = f;
1308 size_t reg_name_length = strcspn(f, "+-}]");
1309 if (reg_name_length > 0)
1310 {
1311 std::string register_name(reg_name, reg_name_length);
1312 f += reg_name_length;
1313 register_offset = strtoll(f, &end, 0);
1314 if (f < end)
1315 f = end;
1316 if ((open_scope_ch == '{' && *f != '}') || (open_scope_ch == '[' && *f != ']'))
1317 {
1318 fprintf(file, "error: Invalid register format string. Valid formats are %%{regname} or %%{regname+offset}, %%[regname] or %%[regname+offset]\n");
1319 return total_bytes_read;
1320 }
1321 else
1322 {
1323 f++;
1324 if (DNBThreadGetRegisterValueByName(pid, tid, REGISTER_SET_ALL, register_name.c_str(), &register_value))
1325 {
1326 // Set the address to dereference using the register value plus the offset
1327 switch (register_value.info.size)
1328 {
1329 default:
1330 case 0:
1331 fprintf (file, "error: unsupported register size of %u.\n", register_value.info.size);
1332 return total_bytes_read;
1333
1334 case 1: register_addr = register_value.value.uint8 + register_offset; break;
1335 case 2: register_addr = register_value.value.uint16 + register_offset; break;
1336 case 4: register_addr = register_value.value.uint32 + register_offset; break;
1337 case 8: register_addr = register_value.value.uint64 + register_offset; break;
1338 case 16:
1339 if (open_scope_ch == '[')
1340 {
1341 fprintf (file, "error: register size (%u) too large for address.\n", register_value.info.size);
1342 return total_bytes_read;
1343 }
1344 break;
1345 }
1346
1347 if (open_scope_ch == '{')
1348 {
1349 byte_size = register_value.info.size;
1350 is_register = true; // value is in a register
1351
1352 }
1353 else
1354 {
1355 addr = register_addr; // Use register value and offset as the address
1356 }
1357 }
1358 else
1359 {
1360 fprintf(file, "error: unable to read register '%s' for process %#.4x and thread %#.4x\n", register_name.c_str(), pid, tid);
1361 return total_bytes_read;
1362 }
1363 }
1364 }
1365 }
1366 break;
1367
1368 default:
1369 fprintf(file, "error: %%$ must be followed by (regname + n) or [regname + n]\n");
1370 return total_bytes_read;
1371 }
1372 }
1373 break;
1374 }
1375
1376 // Check for a minimum field width
1377 if (isdigit(*f))
1378 {
Greg Clayton23f59502012-07-17 03:23:13 +00001379 //min_field_width = strtoul(f, &end, 10);
1380 strtoul(f, &end, 10);
Chris Lattner30fdc8d2010-06-08 16:52:24 +00001381 if (end > f)
1382 {
1383 fprintf_format.append(f, end - f);
1384 f = end;
1385 }
1386 }
1387
1388
1389 // Check for a precision
1390 if (*f == '.')
1391 {
1392 f++;
1393 if (isdigit(*f))
1394 {
1395 fprintf_format += '.';
Greg Clayton23f59502012-07-17 03:23:13 +00001396 //precision = strtoul(f, &end, 10);
1397 strtoul(f, &end, 10);
Chris Lattner30fdc8d2010-06-08 16:52:24 +00001398 if (end > f)
1399 {
1400 fprintf_format.append(f, end - f);
1401 f = end;
1402 }
1403 }
1404 }
1405
1406
1407 // mprintf specific: read the optional actual byte size (abs)
1408 // after the standard minimum field width (mfw) and precision (prec).
1409 // Standard printf calls you can have "mfw.prec" or ".prec", but
1410 // mprintf can have "mfw.prec.abs", ".prec.abs" or "..abs". This is nice
1411 // for strings that may be in a fixed size buffer, but may not use all bytes
1412 // in that buffer for printable characters.
1413 if (*f == '.')
1414 {
1415 f++;
1416 actual_byte_size = strtoul(f, &end, 10);
1417 if (end > f)
1418 {
1419 byte_size = actual_byte_size;
1420 f = end;
1421 }
1422 }
1423
1424 // Decode the length modifiers
1425 switch (*f)
1426 {
1427 case 'h': // h and hh length modifiers
1428 fprintf_format += *f++;
1429 length_modifiers |= length_mod_h;
1430 if (*f == 'h')
1431 {
1432 fprintf_format += *f++;
1433 length_modifiers |= length_mod_hh;
1434 }
1435 break;
1436
1437 case 'l': // l and ll length modifiers
1438 fprintf_format += *f++;
1439 length_modifiers |= length_mod_l;
1440 if (*f == 'h')
1441 {
1442 fprintf_format += *f++;
1443 length_modifiers |= length_mod_ll;
1444 }
1445 break;
1446
1447 case 'L': fprintf_format += *f++; length_modifiers |= length_mod_L; break;
1448 case 'j': fprintf_format += *f++; length_modifiers |= length_mod_j; break;
1449 case 't': fprintf_format += *f++; length_modifiers |= length_mod_t; break;
1450 case 'z': fprintf_format += *f++; length_modifiers |= length_mod_z; break;
1451 case 'q': fprintf_format += *f++; length_modifiers |= length_mod_q; break;
1452 }
1453
1454 // Decode the conversion specifier
1455 switch (*f)
1456 {
1457 case '_':
1458 // mprintf specific format items
1459 {
1460 ++f; // Skip the '_' character
1461 switch (*f)
1462 {
1463 case 'a': // Print the current address
1464 ++f;
1465 fprintf_format += "ll";
1466 fprintf_format += *f; // actual format to show address with folows the 'a' ("%_ax")
1467 fprintf (file, fprintf_format.c_str(), addr);
1468 break;
1469 case 'o': // offset from base address
1470 ++f;
1471 fprintf_format += "ll";
1472 fprintf_format += *f; // actual format to show address with folows the 'a' ("%_ox")
1473 fprintf(file, fprintf_format.c_str(), addr - base_addr);
1474 break;
1475 default:
1476 fprintf (file, "error: unsupported mprintf specific format character '%c'.\n", *f);
1477 break;
1478 }
1479 continue;
1480 }
1481 break;
1482
1483 case 'D':
1484 case 'O':
1485 case 'U':
1486 fprintf_format += *f;
1487 if (byte_size == 0)
1488 byte_size = sizeof(long int);
1489 break;
1490
1491 case 'd':
1492 case 'i':
1493 case 'o':
1494 case 'u':
1495 case 'x':
1496 case 'X':
1497 fprintf_format += *f;
1498 if (byte_size == 0)
1499 {
1500 if (length_modifiers & length_mod_hh)
1501 byte_size = sizeof(char);
1502 else if (length_modifiers & length_mod_h)
1503 byte_size = sizeof(short);
Greg Clayton23f59502012-07-17 03:23:13 +00001504 else if (length_modifiers & length_mod_ll)
Chris Lattner30fdc8d2010-06-08 16:52:24 +00001505 byte_size = sizeof(long long);
1506 else if (length_modifiers & length_mod_l)
1507 byte_size = sizeof(long);
1508 else
1509 byte_size = sizeof(int);
1510 }
1511 break;
1512
1513 case 'a':
1514 case 'A':
1515 case 'f':
1516 case 'F':
1517 case 'e':
1518 case 'E':
1519 case 'g':
1520 case 'G':
1521 fprintf_format += *f;
1522 if (byte_size == 0)
1523 {
1524 if (length_modifiers & length_mod_L)
1525 byte_size = sizeof(long double);
1526 else
1527 byte_size = sizeof(double);
1528 }
1529 break;
1530
1531 case 'c':
1532 if ((length_modifiers & length_mod_l) == 0)
1533 {
1534 fprintf_format += *f;
1535 if (byte_size == 0)
1536 byte_size = sizeof(char);
1537 break;
1538 }
1539 // Fall through to 'C' modifier below...
1540
1541 case 'C':
1542 fprintf_format += *f;
1543 if (byte_size == 0)
1544 byte_size = sizeof(wchar_t);
1545 break;
1546
1547 case 's':
1548 fprintf_format += *f;
1549 if (is_register || byte_size == 0)
1550 is_string = 1;
1551 break;
1552
1553 case 'p':
1554 fprintf_format += *f;
1555 if (byte_size == 0)
1556 byte_size = sizeof(void*);
1557 break;
1558 }
1559
1560 if (is_string)
1561 {
1562 std::string mem_string;
1563 const size_t string_buf_len = 4;
1564 char string_buf[string_buf_len+1];
1565 char *string_buf_end = string_buf + string_buf_len;
1566 string_buf[string_buf_len] = '\0';
1567 nub_size_t bytes_read;
1568 nub_addr_t str_addr = is_register ? register_addr : addr;
1569 while ((bytes_read = DNBProcessMemoryRead(pid, str_addr, string_buf_len, &string_buf[0])) > 0)
1570 {
1571 // Did we get a NULL termination character yet?
1572 if (strchr(string_buf, '\0') == string_buf_end)
1573 {
1574 // no NULL terminator yet, append as a std::string
1575 mem_string.append(string_buf, string_buf_len);
1576 str_addr += string_buf_len;
1577 }
1578 else
1579 {
1580 // yep
1581 break;
1582 }
1583 }
1584 // Append as a C-string so we don't get the extra NULL
1585 // characters in the temp buffer (since it was resized)
1586 mem_string += string_buf;
1587 size_t mem_string_len = mem_string.size() + 1;
1588 fprintf(file, fprintf_format.c_str(), mem_string.c_str());
1589 if (mem_string_len > 0)
1590 {
1591 if (!is_register)
1592 {
1593 addr += mem_string_len;
1594 total_bytes_read += mem_string_len;
1595 }
1596 }
1597 else
1598 return total_bytes_read;
1599 }
1600 else
1601 if (byte_size > 0)
1602 {
1603 buf.resize(byte_size);
1604 nub_size_t bytes_read = 0;
1605 if (is_register)
1606 bytes_read = register_value.info.size;
1607 else
1608 bytes_read = DNBProcessMemoryRead(pid, addr, buf.size(), &buf[0]);
1609 if (bytes_read > 0)
1610 {
1611 if (!is_register)
1612 total_bytes_read += bytes_read;
1613
1614 if (bytes_read == byte_size)
1615 {
1616 switch (*f)
1617 {
1618 case 'd':
1619 case 'i':
1620 case 'o':
1621 case 'u':
1622 case 'X':
1623 case 'x':
1624 case 'a':
1625 case 'A':
1626 case 'f':
1627 case 'F':
1628 case 'e':
1629 case 'E':
1630 case 'g':
1631 case 'G':
1632 case 'p':
1633 case 'c':
1634 case 'C':
1635 {
1636 if (is_register)
1637 data.SetData(&register_value.value.v_uint8[0], register_value.info.size);
1638 else
1639 data.SetData(&buf[0], bytes_read);
1640 DNBDataRef::offset_t data_offset = 0;
1641 if (byte_size <= 4)
1642 {
1643 uint32_t u32 = data.GetMax32(&data_offset, byte_size);
1644 // Show the actual byte width when displaying hex
1645 fprintf(file, fprintf_format.c_str(), u32);
1646 }
1647 else if (byte_size <= 8)
1648 {
1649 uint64_t u64 = data.GetMax64(&data_offset, byte_size);
1650 // Show the actual byte width when displaying hex
1651 fprintf(file, fprintf_format.c_str(), u64);
1652 }
1653 else
1654 {
1655 fprintf(file, "error: integer size not supported, must be 8 bytes or less (%u bytes).\n", byte_size);
1656 }
1657 if (!is_register)
1658 addr += byte_size;
1659 }
1660 break;
1661
1662 case 's':
1663 fprintf(file, fprintf_format.c_str(), buf.c_str());
1664 addr += byte_size;
1665 break;
1666
1667 default:
1668 fprintf(file, "error: unsupported conversion specifier '%c'.\n", *f);
1669 break;
1670 }
1671 }
1672 }
1673 }
1674 else
1675 return total_bytes_read;
1676 }
1677 break;
1678
1679 case '\\':
1680 {
1681 f++;
1682 switch (*f)
1683 {
1684 case 'e': ch = '\e'; break;
1685 case 'a': ch = '\a'; break;
1686 case 'b': ch = '\b'; break;
1687 case 'f': ch = '\f'; break;
1688 case 'n': ch = '\n'; break;
1689 case 'r': ch = '\r'; break;
1690 case 't': ch = '\t'; break;
1691 case 'v': ch = '\v'; break;
1692 case '\'': ch = '\''; break;
1693 case '\\': ch = '\\'; break;
1694 case '0':
1695 case '1':
1696 case '2':
1697 case '3':
1698 case '4':
1699 case '5':
1700 case '6':
1701 case '7':
1702 ch = strtoul(f, &end, 8);
1703 f = end;
1704 break;
1705 default:
1706 ch = *f;
1707 break;
1708 }
1709 fputc(ch, file);
1710 }
1711 break;
1712
1713 default:
1714 fputc(ch, file);
1715 break;
1716 }
1717 }
1718 return total_bytes_read;
1719}
1720
1721
1722//----------------------------------------------------------------------
1723// Get the number of threads for the specified process.
1724//----------------------------------------------------------------------
1725nub_size_t
1726DNBProcessGetNumThreads (nub_process_t pid)
1727{
1728 MachProcessSP procSP;
1729 if (GetProcessSP (pid, procSP))
1730 return procSP->GetNumThreads();
1731 return 0;
1732}
1733
1734//----------------------------------------------------------------------
1735// Get the thread ID of the current thread.
1736//----------------------------------------------------------------------
1737nub_thread_t
1738DNBProcessGetCurrentThread (nub_process_t pid)
1739{
1740 MachProcessSP procSP;
1741 if (GetProcessSP (pid, procSP))
1742 return procSP->GetCurrentThread();
1743 return 0;
1744}
1745
1746//----------------------------------------------------------------------
1747// Change the current thread.
1748//----------------------------------------------------------------------
1749nub_thread_t
1750DNBProcessSetCurrentThread (nub_process_t pid, nub_thread_t tid)
1751{
1752 MachProcessSP procSP;
1753 if (GetProcessSP (pid, procSP))
1754 return procSP->SetCurrentThread (tid);
1755 return INVALID_NUB_THREAD;
1756}
1757
1758
1759//----------------------------------------------------------------------
1760// Dump a string describing a thread's stop reason to the specified file
1761// handle
1762//----------------------------------------------------------------------
1763nub_bool_t
1764DNBThreadGetStopReason (nub_process_t pid, nub_thread_t tid, struct DNBThreadStopInfo *stop_info)
1765{
1766 MachProcessSP procSP;
1767 if (GetProcessSP (pid, procSP))
1768 return procSP->GetThreadStoppedReason (tid, stop_info);
1769 return false;
1770}
1771
1772//----------------------------------------------------------------------
1773// Return string description for the specified thread.
1774//
1775// RETURNS: NULL if the thread isn't valid, else a NULL terminated C
1776// string from a static buffer that must be copied prior to subsequent
1777// calls.
1778//----------------------------------------------------------------------
1779const char *
1780DNBThreadGetInfo (nub_process_t pid, nub_thread_t tid)
1781{
1782 MachProcessSP procSP;
1783 if (GetProcessSP (pid, procSP))
1784 return procSP->GetThreadInfo (tid);
1785 return NULL;
1786}
1787
1788//----------------------------------------------------------------------
1789// Get the thread ID given a thread index.
1790//----------------------------------------------------------------------
1791nub_thread_t
1792DNBProcessGetThreadAtIndex (nub_process_t pid, size_t thread_idx)
1793{
1794 MachProcessSP procSP;
1795 if (GetProcessSP (pid, procSP))
1796 return procSP->GetThreadAtIndex (thread_idx);
1797 return INVALID_NUB_THREAD;
1798}
1799
1800nub_addr_t
1801DNBProcessGetSharedLibraryInfoAddress (nub_process_t pid)
1802{
1803 MachProcessSP procSP;
1804 DNBError err;
1805 if (GetProcessSP (pid, procSP))
1806 return procSP->Task().GetDYLDAllImageInfosAddress (err);
1807 return INVALID_NUB_ADDRESS;
1808}
1809
1810
1811nub_bool_t
1812DNBProcessSharedLibrariesUpdated(nub_process_t pid)
1813{
1814 MachProcessSP procSP;
1815 if (GetProcessSP (pid, procSP))
1816 {
1817 procSP->SharedLibrariesUpdated ();
1818 return true;
1819 }
1820 return false;
1821}
1822
1823//----------------------------------------------------------------------
1824// Get the current shared library information for a process. Only return
1825// the shared libraries that have changed since the last shared library
1826// state changed event if only_changed is non-zero.
1827//----------------------------------------------------------------------
1828nub_size_t
1829DNBProcessGetSharedLibraryInfo (nub_process_t pid, nub_bool_t only_changed, struct DNBExecutableImageInfo **image_infos)
1830{
1831 MachProcessSP procSP;
1832 if (GetProcessSP (pid, procSP))
1833 return procSP->CopyImageInfos (image_infos, only_changed);
1834
1835 // If we have no process, then return NULL for the shared library info
1836 // and zero for shared library count
1837 *image_infos = NULL;
1838 return 0;
1839}
1840
1841//----------------------------------------------------------------------
1842// Get the register set information for a specific thread.
1843//----------------------------------------------------------------------
1844const DNBRegisterSetInfo *
1845DNBGetRegisterSetInfo (nub_size_t *num_reg_sets)
1846{
Greg Clayton3af9ea52010-11-18 05:57:03 +00001847 return DNBArchProtocol::GetRegisterSetInfo (num_reg_sets);
Chris Lattner30fdc8d2010-06-08 16:52:24 +00001848}
1849
1850
1851//----------------------------------------------------------------------
1852// Read a register value by register set and register index.
1853//----------------------------------------------------------------------
1854nub_bool_t
1855DNBThreadGetRegisterValueByID (nub_process_t pid, nub_thread_t tid, uint32_t set, uint32_t reg, DNBRegisterValue *value)
1856{
1857 MachProcessSP procSP;
1858 ::bzero (value, sizeof(DNBRegisterValue));
1859 if (GetProcessSP (pid, procSP))
1860 {
1861 if (tid != INVALID_NUB_THREAD)
1862 return procSP->GetRegisterValue (tid, set, reg, value);
1863 }
1864 return false;
1865}
1866
1867nub_bool_t
1868DNBThreadSetRegisterValueByID (nub_process_t pid, nub_thread_t tid, uint32_t set, uint32_t reg, const DNBRegisterValue *value)
1869{
1870 if (tid != INVALID_NUB_THREAD)
1871 {
1872 MachProcessSP procSP;
1873 if (GetProcessSP (pid, procSP))
1874 return procSP->SetRegisterValue (tid, set, reg, value);
1875 }
1876 return false;
1877}
1878
1879nub_size_t
1880DNBThreadGetRegisterContext (nub_process_t pid, nub_thread_t tid, void *buf, size_t buf_len)
1881{
1882 MachProcessSP procSP;
1883 if (GetProcessSP (pid, procSP))
1884 {
1885 if (tid != INVALID_NUB_THREAD)
1886 return procSP->GetThreadList().GetRegisterContext (tid, buf, buf_len);
1887 }
1888 ::bzero (buf, buf_len);
1889 return 0;
1890
1891}
1892
1893nub_size_t
1894DNBThreadSetRegisterContext (nub_process_t pid, nub_thread_t tid, const void *buf, size_t buf_len)
1895{
1896 MachProcessSP procSP;
1897 if (GetProcessSP (pid, procSP))
1898 {
1899 if (tid != INVALID_NUB_THREAD)
1900 return procSP->GetThreadList().SetRegisterContext (tid, buf, buf_len);
1901 }
1902 return 0;
1903}
1904
1905//----------------------------------------------------------------------
1906// Read a register value by name.
1907//----------------------------------------------------------------------
1908nub_bool_t
1909DNBThreadGetRegisterValueByName (nub_process_t pid, nub_thread_t tid, uint32_t reg_set, const char *reg_name, DNBRegisterValue *value)
1910{
1911 MachProcessSP procSP;
1912 ::bzero (value, sizeof(DNBRegisterValue));
1913 if (GetProcessSP (pid, procSP))
1914 {
1915 const struct DNBRegisterSetInfo *set_info;
1916 nub_size_t num_reg_sets = 0;
1917 set_info = DNBGetRegisterSetInfo (&num_reg_sets);
1918 if (set_info)
1919 {
1920 uint32_t set = reg_set;
1921 uint32_t reg;
1922 if (set == REGISTER_SET_ALL)
1923 {
1924 for (set = 1; set < num_reg_sets; ++set)
1925 {
1926 for (reg = 0; reg < set_info[set].num_registers; ++reg)
1927 {
1928 if (strcasecmp(reg_name, set_info[set].registers[reg].name) == 0)
1929 return procSP->GetRegisterValue (tid, set, reg, value);
1930 }
1931 }
1932 }
1933 else
1934 {
1935 for (reg = 0; reg < set_info[set].num_registers; ++reg)
1936 {
1937 if (strcasecmp(reg_name, set_info[set].registers[reg].name) == 0)
1938 return procSP->GetRegisterValue (tid, set, reg, value);
1939 }
1940 }
1941 }
1942 }
1943 return false;
1944}
1945
1946
1947//----------------------------------------------------------------------
1948// Read a register set and register number from the register name.
1949//----------------------------------------------------------------------
1950nub_bool_t
1951DNBGetRegisterInfoByName (const char *reg_name, DNBRegisterInfo* info)
1952{
1953 const struct DNBRegisterSetInfo *set_info;
1954 nub_size_t num_reg_sets = 0;
1955 set_info = DNBGetRegisterSetInfo (&num_reg_sets);
1956 if (set_info)
1957 {
1958 uint32_t set, reg;
1959 for (set = 1; set < num_reg_sets; ++set)
1960 {
1961 for (reg = 0; reg < set_info[set].num_registers; ++reg)
1962 {
1963 if (strcasecmp(reg_name, set_info[set].registers[reg].name) == 0)
1964 {
1965 *info = set_info[set].registers[reg];
1966 return true;
1967 }
1968 }
1969 }
1970
1971 for (set = 1; set < num_reg_sets; ++set)
1972 {
1973 uint32_t reg;
1974 for (reg = 0; reg < set_info[set].num_registers; ++reg)
1975 {
1976 if (set_info[set].registers[reg].alt == NULL)
1977 continue;
1978
1979 if (strcasecmp(reg_name, set_info[set].registers[reg].alt) == 0)
1980 {
1981 *info = set_info[set].registers[reg];
1982 return true;
1983 }
1984 }
1985 }
1986 }
1987
1988 ::bzero (info, sizeof(DNBRegisterInfo));
1989 return false;
1990}
1991
1992
1993//----------------------------------------------------------------------
1994// Set the name to address callback function that this nub can use
1995// for any name to address lookups that are needed.
1996//----------------------------------------------------------------------
1997nub_bool_t
1998DNBProcessSetNameToAddressCallback (nub_process_t pid, DNBCallbackNameToAddress callback, void *baton)
1999{
2000 MachProcessSP procSP;
2001 if (GetProcessSP (pid, procSP))
2002 {
2003 procSP->SetNameToAddressCallback (callback, baton);
2004 return true;
2005 }
2006 return false;
2007}
2008
2009
2010//----------------------------------------------------------------------
2011// Set the name to address callback function that this nub can use
2012// for any name to address lookups that are needed.
2013//----------------------------------------------------------------------
2014nub_bool_t
2015DNBProcessSetSharedLibraryInfoCallback (nub_process_t pid, DNBCallbackCopyExecutableImageInfos callback, void *baton)
2016{
2017 MachProcessSP procSP;
2018 if (GetProcessSP (pid, procSP))
2019 {
2020 procSP->SetSharedLibraryInfoCallback (callback, baton);
2021 return true;
2022 }
2023 return false;
2024}
2025
2026nub_addr_t
2027DNBProcessLookupAddress (nub_process_t pid, const char *name, const char *shlib)
2028{
2029 MachProcessSP procSP;
2030 if (GetProcessSP (pid, procSP))
2031 {
2032 return procSP->LookupSymbol (name, shlib);
2033 }
2034 return INVALID_NUB_ADDRESS;
2035}
2036
2037
2038nub_size_t
2039DNBProcessGetAvailableSTDOUT (nub_process_t pid, char *buf, nub_size_t buf_size)
2040{
2041 MachProcessSP procSP;
2042 if (GetProcessSP (pid, procSP))
2043 return procSP->GetAvailableSTDOUT (buf, buf_size);
2044 return 0;
2045}
2046
2047nub_size_t
2048DNBProcessGetAvailableSTDERR (nub_process_t pid, char *buf, nub_size_t buf_size)
2049{
2050 MachProcessSP procSP;
2051 if (GetProcessSP (pid, procSP))
2052 return procSP->GetAvailableSTDERR (buf, buf_size);
2053 return 0;
2054}
2055
2056nub_size_t
2057DNBProcessGetStopCount (nub_process_t pid)
2058{
2059 MachProcessSP procSP;
2060 if (GetProcessSP (pid, procSP))
2061 return procSP->StopCount();
2062 return 0;
2063}
2064
Greg Clayton71337622011-02-24 22:24:29 +00002065uint32_t
2066DNBProcessGetCPUType (nub_process_t pid)
2067{
2068 MachProcessSP procSP;
2069 if (GetProcessSP (pid, procSP))
2070 return procSP->GetCPUType ();
2071 return 0;
2072
2073}
2074
Chris Lattner30fdc8d2010-06-08 16:52:24 +00002075nub_bool_t
2076DNBResolveExecutablePath (const char *path, char *resolved_path, size_t resolved_path_size)
2077{
2078 if (path == NULL || path[0] == '\0')
2079 return false;
2080
2081 char max_path[PATH_MAX];
2082 std::string result;
2083 CFString::GlobPath(path, result);
2084
2085 if (result.empty())
2086 result = path;
2087
2088 if (realpath(path, max_path))
2089 {
2090 // Found the path relatively...
2091 ::strncpy(resolved_path, max_path, resolved_path_size);
2092 return strlen(resolved_path) + 1 < resolved_path_size;
2093 }
2094 else
2095 {
2096 // Not a relative path, check the PATH environment variable if the
2097 const char *PATH = getenv("PATH");
2098 if (PATH)
2099 {
2100 const char *curr_path_start = PATH;
2101 const char *curr_path_end;
2102 while (curr_path_start && *curr_path_start)
2103 {
2104 curr_path_end = strchr(curr_path_start, ':');
2105 if (curr_path_end == NULL)
2106 {
2107 result.assign(curr_path_start);
2108 curr_path_start = NULL;
2109 }
2110 else if (curr_path_end > curr_path_start)
2111 {
2112 size_t len = curr_path_end - curr_path_start;
2113 result.assign(curr_path_start, len);
2114 curr_path_start += len + 1;
2115 }
2116 else
2117 break;
2118
2119 result += '/';
2120 result += path;
2121 struct stat s;
2122 if (stat(result.c_str(), &s) == 0)
2123 {
2124 ::strncpy(resolved_path, result.c_str(), resolved_path_size);
2125 return result.size() + 1 < resolved_path_size;
2126 }
2127 }
2128 }
2129 }
2130 return false;
2131}
2132
Greg Clayton3af9ea52010-11-18 05:57:03 +00002133
2134void
2135DNBInitialize()
2136{
2137 DNBLogThreadedIf (LOG_PROCESS, "DNBInitialize ()");
2138#if defined (__i386__) || defined (__x86_64__)
2139 DNBArchImplI386::Initialize();
2140 DNBArchImplX86_64::Initialize();
2141#elif defined (__arm__)
2142 DNBArchMachARM::Initialize();
2143#endif
2144}
2145
2146void
2147DNBTerminate()
2148{
2149}
Greg Clayton3c144382010-12-01 22:45:40 +00002150
2151nub_bool_t
2152DNBSetArchitecture (const char *arch)
2153{
2154 if (arch && arch[0])
2155 {
2156 if (strcasecmp (arch, "i386") == 0)
2157 return DNBArchProtocol::SetArchitecture (CPU_TYPE_I386);
2158 else if (strcasecmp (arch, "x86_64") == 0)
2159 return DNBArchProtocol::SetArchitecture (CPU_TYPE_X86_64);
2160 else if (strstr (arch, "arm") == arch)
2161 return DNBArchProtocol::SetArchitecture (CPU_TYPE_ARM);
2162 }
2163 return false;
2164}