[sanitizer] Improve FreeBSD ASLR detection
[llvm-project.git] / lldb / tools / debugserver / source / debugserver.cpp
blob87198717d691becf062c640d8bf9246f887f017b
1 //===-- debugserver.cpp -----------------------------------------*- C++ -*-===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
9 #include <arpa/inet.h>
10 #include <asl.h>
11 #include <cerrno>
12 #include <crt_externs.h>
13 #include <getopt.h>
14 #include <netdb.h>
15 #include <netinet/in.h>
16 #include <netinet/tcp.h>
17 #include <string>
18 #include <sys/select.h>
19 #include <sys/socket.h>
20 #include <sys/sysctl.h>
21 #include <sys/types.h>
22 #include <sys/un.h>
24 #include <memory>
25 #include <vector>
27 #if defined(__APPLE__)
28 #include <sched.h>
29 extern "C" int proc_set_wakemon_params(pid_t, int,
30 int); // <libproc_internal.h> SPI
31 #endif
33 #include "CFString.h"
34 #include "DNB.h"
35 #include "DNBLog.h"
36 #include "DNBTimer.h"
37 #include "OsLogger.h"
38 #include "PseudoTerminal.h"
39 #include "RNBContext.h"
40 #include "RNBRemote.h"
41 #include "RNBServices.h"
42 #include "RNBSocket.h"
43 #include "SysSignal.h"
45 // Global PID in case we get a signal and need to stop the process...
46 nub_process_t g_pid = INVALID_NUB_PROCESS;
48 // Run loop modes which determine which run loop function will be called
49 enum RNBRunLoopMode {
50 eRNBRunLoopModeInvalid = 0,
51 eRNBRunLoopModeGetStartModeFromRemoteProtocol,
52 eRNBRunLoopModeInferiorAttaching,
53 eRNBRunLoopModeInferiorLaunching,
54 eRNBRunLoopModeInferiorExecuting,
55 eRNBRunLoopModePlatformMode,
56 eRNBRunLoopModeExit
59 // Global Variables
60 RNBRemoteSP g_remoteSP;
61 static int g_lockdown_opt = 0;
62 static int g_applist_opt = 0;
63 static nub_launch_flavor_t g_launch_flavor = eLaunchFlavorDefault;
64 int g_disable_aslr = 0;
66 int g_isatty = 0;
67 bool g_detach_on_error = true;
69 #define RNBLogSTDOUT(fmt, ...) \
70 do { \
71 if (g_isatty) { \
72 fprintf(stdout, fmt, ##__VA_ARGS__); \
73 } else { \
74 _DNBLog(0, fmt, ##__VA_ARGS__); \
75 } \
76 } while (0)
77 #define RNBLogSTDERR(fmt, ...) \
78 do { \
79 if (g_isatty) { \
80 fprintf(stderr, fmt, ##__VA_ARGS__); \
81 } else { \
82 _DNBLog(0, fmt, ##__VA_ARGS__); \
83 } \
84 } while (0)
86 // Get our program path and arguments from the remote connection.
87 // We will need to start up the remote connection without a PID, get the
88 // arguments, wait for the new process to finish launching and hit its
89 // entry point, and then return the run loop mode that should come next.
90 RNBRunLoopMode RNBRunLoopGetStartModeFromRemote(RNBRemote *remote) {
91 std::string packet;
93 if (remote) {
94 RNBContext &ctx = remote->Context();
95 uint32_t event_mask = RNBContext::event_read_packet_available |
96 RNBContext::event_read_thread_exiting;
98 // Spin waiting to get the A packet.
99 while (true) {
100 DNBLogThreadedIf(LOG_RNB_MAX,
101 "%s ctx.Events().WaitForSetEvents( 0x%08x ) ...",
102 __FUNCTION__, event_mask);
103 nub_event_t set_events = ctx.Events().WaitForSetEvents(event_mask);
104 DNBLogThreadedIf(LOG_RNB_MAX,
105 "%s ctx.Events().WaitForSetEvents( 0x%08x ) => 0x%08x",
106 __FUNCTION__, event_mask, set_events);
108 if (set_events & RNBContext::event_read_thread_exiting) {
109 RNBLogSTDERR("error: packet read thread exited.\n");
110 return eRNBRunLoopModeExit;
113 if (set_events & RNBContext::event_read_packet_available) {
114 rnb_err_t err = rnb_err;
115 RNBRemote::PacketEnum type;
117 err = remote->HandleReceivedPacket(&type);
119 // check if we tried to attach to a process
120 if (type == RNBRemote::vattach || type == RNBRemote::vattachwait ||
121 type == RNBRemote::vattachorwait) {
122 if (err == rnb_success) {
123 RNBLogSTDOUT("Attach succeeded, ready to debug.\n");
124 return eRNBRunLoopModeInferiorExecuting;
125 } else {
126 RNBLogSTDERR("error: attach failed.\n");
127 return eRNBRunLoopModeExit;
131 if (err == rnb_success) {
132 // If we got our arguments we are ready to launch using the arguments
133 // and any environment variables we received.
134 if (type == RNBRemote::set_argv) {
135 return eRNBRunLoopModeInferiorLaunching;
137 } else if (err == rnb_not_connected) {
138 RNBLogSTDERR("error: connection lost.\n");
139 return eRNBRunLoopModeExit;
140 } else {
141 // a catch all for any other gdb remote packets that failed
142 DNBLogThreadedIf(LOG_RNB_MINIMAL, "%s Error getting packet.",
143 __FUNCTION__);
144 continue;
147 DNBLogThreadedIf(LOG_RNB_MINIMAL, "#### %s", __FUNCTION__);
148 } else {
149 DNBLogThreadedIf(LOG_RNB_MINIMAL,
150 "%s Connection closed before getting \"A\" packet.",
151 __FUNCTION__);
152 return eRNBRunLoopModeExit;
156 return eRNBRunLoopModeExit;
159 static nub_launch_flavor_t default_launch_flavor(const char *app_name) {
160 #if defined(WITH_FBS) || defined(WITH_BKS) || defined(WITH_SPRINGBOARD)
161 // Check the name to see if it ends with .app
162 auto is_dot_app = [](const char *app_name) {
163 size_t len = strlen(app_name);
164 if (len < 4)
165 return false;
167 if (app_name[len - 4] == '.' && app_name[len - 3] == 'a' &&
168 app_name[len - 2] == 'p' && app_name[len - 1] == 'p')
169 return true;
170 return false;
173 if (is_dot_app(app_name)) {
174 #if defined WITH_FBS
175 // Check if we have an app bundle, if so launch using FrontBoard Services.
176 return eLaunchFlavorFBS;
177 #elif defined WITH_BKS
178 // Check if we have an app bundle, if so launch using BackBoard Services.
179 return eLaunchFlavorBKS;
180 #elif defined WITH_SPRINGBOARD
181 // Check if we have an app bundle, if so launch using SpringBoard.
182 return eLaunchFlavorSpringBoard;
183 #endif
185 #endif
187 // Our default launch method is posix spawn
188 return eLaunchFlavorPosixSpawn;
191 // This run loop mode will wait for the process to launch and hit its
192 // entry point. It will currently ignore all events except for the
193 // process state changed event, where it watches for the process stopped
194 // or crash process state.
195 RNBRunLoopMode RNBRunLoopLaunchInferior(RNBRemote *remote,
196 const char *stdin_path,
197 const char *stdout_path,
198 const char *stderr_path,
199 bool no_stdio) {
200 RNBContext &ctx = remote->Context();
202 // The Process stuff takes a c array, the RNBContext has a vector...
203 // So make up a c array.
205 DNBLogThreadedIf(LOG_RNB_MINIMAL, "%s Launching '%s'...", __FUNCTION__,
206 ctx.ArgumentAtIndex(0));
208 size_t inferior_argc = ctx.ArgumentCount();
209 // Initialize inferior_argv with inferior_argc + 1 NULLs
210 std::vector<const char *> inferior_argv(inferior_argc + 1, NULL);
212 size_t i;
213 for (i = 0; i < inferior_argc; i++)
214 inferior_argv[i] = ctx.ArgumentAtIndex(i);
216 // Pass the environment array the same way:
218 size_t inferior_envc = ctx.EnvironmentCount();
219 // Initialize inferior_argv with inferior_argc + 1 NULLs
220 std::vector<const char *> inferior_envp(inferior_envc + 1, NULL);
222 for (i = 0; i < inferior_envc; i++)
223 inferior_envp[i] = ctx.EnvironmentAtIndex(i);
225 // Our launch type hasn't been set to anything concrete, so we need to
226 // figure our how we are going to launch automatically.
228 nub_launch_flavor_t launch_flavor = g_launch_flavor;
229 if (launch_flavor == eLaunchFlavorDefault)
230 launch_flavor = default_launch_flavor(inferior_argv[0]);
232 ctx.SetLaunchFlavor(launch_flavor);
233 char resolved_path[PATH_MAX];
235 // If we fail to resolve the path to our executable, then just use what we
236 // were given and hope for the best
237 if (!DNBResolveExecutablePath(inferior_argv[0], resolved_path,
238 sizeof(resolved_path)))
239 ::strlcpy(resolved_path, inferior_argv[0], sizeof(resolved_path));
241 char launch_err_str[PATH_MAX];
242 launch_err_str[0] = '\0';
243 const char *cwd =
244 (ctx.GetWorkingDirPath() != NULL ? ctx.GetWorkingDirPath()
245 : ctx.GetWorkingDirectory());
246 const char *process_event = ctx.GetProcessEvent();
247 nub_process_t pid = DNBProcessLaunch(
248 &ctx, resolved_path, &inferior_argv[0], &inferior_envp[0], cwd,
249 stdin_path, stdout_path, stderr_path, no_stdio, g_disable_aslr,
250 process_event, launch_err_str, sizeof(launch_err_str));
252 g_pid = pid;
254 if (pid == INVALID_NUB_PROCESS && strlen(launch_err_str) > 0) {
255 DNBLogThreaded("%s DNBProcessLaunch() returned error: '%s'", __FUNCTION__,
256 launch_err_str);
257 ctx.LaunchStatus().SetError(-1, DNBError::Generic);
258 ctx.LaunchStatus().SetErrorString(launch_err_str);
259 } else if (pid == INVALID_NUB_PROCESS) {
260 DNBLogThreaded(
261 "%s DNBProcessLaunch() failed to launch process, unknown failure",
262 __FUNCTION__);
263 ctx.LaunchStatus().SetError(-1, DNBError::Generic);
264 ctx.LaunchStatus().SetErrorString("<unknown failure>");
265 } else {
266 ctx.LaunchStatus().Clear();
269 if (remote->Comm().IsConnected()) {
270 // It we are connected already, the next thing gdb will do is ask
271 // whether the launch succeeded, and if not, whether there is an
272 // error code. So we need to fetch one packet from gdb before we wait
273 // on the stop from the target.
275 uint32_t event_mask = RNBContext::event_read_packet_available;
276 nub_event_t set_events = ctx.Events().WaitForSetEvents(event_mask);
278 if (set_events & RNBContext::event_read_packet_available) {
279 rnb_err_t err = rnb_err;
280 RNBRemote::PacketEnum type;
282 err = remote->HandleReceivedPacket(&type);
284 if (err != rnb_success) {
285 DNBLogThreadedIf(LOG_RNB_MINIMAL, "%s Error getting packet.",
286 __FUNCTION__);
287 return eRNBRunLoopModeExit;
289 if (type != RNBRemote::query_launch_success) {
290 DNBLogThreadedIf(LOG_RNB_MINIMAL,
291 "%s Didn't get the expected qLaunchSuccess packet.",
292 __FUNCTION__);
297 while (pid != INVALID_NUB_PROCESS) {
298 // Wait for process to start up and hit entry point
299 DNBLogThreadedIf(LOG_RNB_EVENTS, "%s DNBProcessWaitForEvent (%4.4x, "
300 "eEventProcessRunningStateChanged | "
301 "eEventProcessStoppedStateChanged, true, "
302 "INFINITE)...",
303 __FUNCTION__, pid);
304 nub_event_t set_events =
305 DNBProcessWaitForEvents(pid, eEventProcessRunningStateChanged |
306 eEventProcessStoppedStateChanged,
307 true, NULL);
308 DNBLogThreadedIf(LOG_RNB_EVENTS, "%s DNBProcessWaitForEvent (%4.4x, "
309 "eEventProcessRunningStateChanged | "
310 "eEventProcessStoppedStateChanged, true, "
311 "INFINITE) => 0x%8.8x",
312 __FUNCTION__, pid, set_events);
314 if (set_events == 0) {
315 pid = INVALID_NUB_PROCESS;
316 g_pid = pid;
317 } else {
318 if (set_events & (eEventProcessRunningStateChanged |
319 eEventProcessStoppedStateChanged)) {
320 nub_state_t pid_state = DNBProcessGetState(pid);
321 DNBLogThreadedIf(
322 LOG_RNB_EVENTS,
323 "%s process %4.4x state changed (eEventProcessStateChanged): %s",
324 __FUNCTION__, pid, DNBStateAsString(pid_state));
326 switch (pid_state) {
327 case eStateInvalid:
328 case eStateUnloaded:
329 case eStateAttaching:
330 case eStateLaunching:
331 case eStateSuspended:
332 break; // Ignore
334 case eStateRunning:
335 case eStateStepping:
336 // Still waiting to stop at entry point...
337 break;
339 case eStateStopped:
340 case eStateCrashed:
341 ctx.SetProcessID(pid);
342 return eRNBRunLoopModeInferiorExecuting;
344 case eStateDetached:
345 case eStateExited:
346 pid = INVALID_NUB_PROCESS;
347 g_pid = pid;
348 return eRNBRunLoopModeExit;
352 DNBProcessResetEvents(pid, set_events);
356 return eRNBRunLoopModeExit;
359 // This run loop mode will wait for the process to launch and hit its
360 // entry point. It will currently ignore all events except for the
361 // process state changed event, where it watches for the process stopped
362 // or crash process state.
363 RNBRunLoopMode RNBRunLoopLaunchAttaching(RNBRemote *remote,
364 nub_process_t attach_pid,
365 nub_process_t &pid) {
366 RNBContext &ctx = remote->Context();
368 DNBLogThreadedIf(LOG_RNB_MINIMAL, "%s Attaching to pid %i...", __FUNCTION__,
369 attach_pid);
370 char err_str[1024];
371 pid = DNBProcessAttach(attach_pid, NULL, ctx.GetUnmaskSignals(), err_str,
372 sizeof(err_str));
373 g_pid = pid;
375 if (pid == INVALID_NUB_PROCESS) {
376 ctx.LaunchStatus().SetError(-1, DNBError::Generic);
377 if (err_str[0])
378 ctx.LaunchStatus().SetErrorString(err_str);
379 return eRNBRunLoopModeExit;
380 } else {
381 ctx.SetProcessID(pid);
382 return eRNBRunLoopModeInferiorExecuting;
386 // Watch for signals:
387 // SIGINT: so we can halt our inferior. (disabled for now)
388 // SIGPIPE: in case our child process dies
389 int g_sigint_received = 0;
390 int g_sigpipe_received = 0;
391 void signal_handler(int signo) {
392 DNBLogThreadedIf(LOG_RNB_MINIMAL, "%s (%s)", __FUNCTION__,
393 SysSignal::Name(signo));
395 switch (signo) {
396 case SIGINT:
397 g_sigint_received++;
398 if (g_pid != INVALID_NUB_PROCESS) {
399 // Only send a SIGINT once...
400 if (g_sigint_received == 1) {
401 switch (DNBProcessGetState(g_pid)) {
402 case eStateRunning:
403 case eStateStepping:
404 DNBProcessSignal(g_pid, SIGSTOP);
405 return;
406 default:
407 break;
411 exit(SIGINT);
412 break;
414 case SIGPIPE:
415 g_sigpipe_received = 1;
416 break;
420 // Return the new run loop mode based off of the current process state
421 RNBRunLoopMode HandleProcessStateChange(RNBRemote *remote, bool initialize) {
422 RNBContext &ctx = remote->Context();
423 nub_process_t pid = ctx.ProcessID();
425 if (pid == INVALID_NUB_PROCESS) {
426 DNBLogThreadedIf(LOG_RNB_MINIMAL, "#### %s error: pid invalid, exiting...",
427 __FUNCTION__);
428 return eRNBRunLoopModeExit;
430 nub_state_t pid_state = DNBProcessGetState(pid);
432 DNBLogThreadedIf(LOG_RNB_MINIMAL,
433 "%s (&remote, initialize=%i) pid_state = %s", __FUNCTION__,
434 (int)initialize, DNBStateAsString(pid_state));
436 switch (pid_state) {
437 case eStateInvalid:
438 case eStateUnloaded:
439 // Something bad happened
440 return eRNBRunLoopModeExit;
441 break;
443 case eStateAttaching:
444 case eStateLaunching:
445 return eRNBRunLoopModeInferiorExecuting;
447 case eStateSuspended:
448 case eStateCrashed:
449 case eStateStopped:
450 // If we stop due to a signal, so clear the fact that we got a SIGINT
451 // so we can stop ourselves again (but only while our inferior
452 // process is running..)
453 g_sigint_received = 0;
454 if (initialize == false) {
455 // Compare the last stop count to our current notion of a stop count
456 // to make sure we don't notify more than once for a given stop.
457 nub_size_t prev_pid_stop_count = ctx.GetProcessStopCount();
458 bool pid_stop_count_changed =
459 ctx.SetProcessStopCount(DNBProcessGetStopCount(pid));
460 if (pid_stop_count_changed) {
461 remote->FlushSTDIO();
463 if (ctx.GetProcessStopCount() == 1) {
464 DNBLogThreadedIf(
465 LOG_RNB_MINIMAL, "%s (&remote, initialize=%i) pid_state = %s "
466 "pid_stop_count %llu (old %llu)) Notify??? no, "
467 "first stop...",
468 __FUNCTION__, (int)initialize, DNBStateAsString(pid_state),
469 (uint64_t)ctx.GetProcessStopCount(),
470 (uint64_t)prev_pid_stop_count);
471 } else {
473 DNBLogThreadedIf(LOG_RNB_MINIMAL, "%s (&remote, initialize=%i) "
474 "pid_state = %s pid_stop_count "
475 "%llu (old %llu)) Notify??? YES!!!",
476 __FUNCTION__, (int)initialize,
477 DNBStateAsString(pid_state),
478 (uint64_t)ctx.GetProcessStopCount(),
479 (uint64_t)prev_pid_stop_count);
480 remote->NotifyThatProcessStopped();
482 } else {
483 DNBLogThreadedIf(
484 LOG_RNB_MINIMAL, "%s (&remote, initialize=%i) pid_state = %s "
485 "pid_stop_count %llu (old %llu)) Notify??? "
486 "skipping...",
487 __FUNCTION__, (int)initialize, DNBStateAsString(pid_state),
488 (uint64_t)ctx.GetProcessStopCount(), (uint64_t)prev_pid_stop_count);
491 return eRNBRunLoopModeInferiorExecuting;
493 case eStateStepping:
494 case eStateRunning:
495 return eRNBRunLoopModeInferiorExecuting;
497 case eStateExited:
498 remote->HandlePacket_last_signal(NULL);
499 return eRNBRunLoopModeExit;
500 case eStateDetached:
501 return eRNBRunLoopModeExit;
504 // Catch all...
505 return eRNBRunLoopModeExit;
508 // This function handles the case where our inferior program is stopped and
509 // we are waiting for gdb remote protocol packets. When a packet occurs that
510 // makes the inferior run, we need to leave this function with a new state
511 // as the return code.
512 RNBRunLoopMode RNBRunLoopInferiorExecuting(RNBRemote *remote) {
513 DNBLogThreadedIf(LOG_RNB_MINIMAL, "#### %s", __FUNCTION__);
514 RNBContext &ctx = remote->Context();
516 // Init our mode and set 'is_running' based on the current process state
517 RNBRunLoopMode mode = HandleProcessStateChange(remote, true);
519 while (ctx.ProcessID() != INVALID_NUB_PROCESS) {
521 std::string set_events_str;
522 uint32_t event_mask = ctx.NormalEventBits();
524 if (!ctx.ProcessStateRunning()) {
525 // Clear some bits if we are not running so we don't send any async
526 // packets
527 event_mask &= ~RNBContext::event_proc_stdio_available;
528 event_mask &= ~RNBContext::event_proc_profile_data;
531 // We want to make sure we consume all process state changes and have
532 // whomever is notifying us to wait for us to reset the event bit before
533 // continuing.
534 // ctx.Events().SetResetAckMask (RNBContext::event_proc_state_changed);
536 DNBLogThreadedIf(LOG_RNB_EVENTS,
537 "%s ctx.Events().WaitForSetEvents(0x%08x) ...",
538 __FUNCTION__, event_mask);
539 nub_event_t set_events = ctx.Events().WaitForSetEvents(event_mask);
540 DNBLogThreadedIf(LOG_RNB_EVENTS,
541 "%s ctx.Events().WaitForSetEvents(0x%08x) => 0x%08x (%s)",
542 __FUNCTION__, event_mask, set_events,
543 ctx.EventsAsString(set_events, set_events_str));
545 if (set_events) {
546 if ((set_events & RNBContext::event_proc_thread_exiting) ||
547 (set_events & RNBContext::event_proc_stdio_available)) {
548 remote->FlushSTDIO();
551 if (set_events & RNBContext::event_proc_profile_data) {
552 remote->SendAsyncProfileData();
555 if (set_events & RNBContext::event_read_packet_available) {
556 // handleReceivedPacket will take care of resetting the
557 // event_read_packet_available events when there are no more...
558 set_events ^= RNBContext::event_read_packet_available;
560 if (ctx.ProcessStateRunning()) {
561 if (remote->HandleAsyncPacket() == rnb_not_connected) {
562 // TODO: connect again? Exit?
564 } else {
565 if (remote->HandleReceivedPacket() == rnb_not_connected) {
566 // TODO: connect again? Exit?
571 if (set_events & RNBContext::event_proc_state_changed) {
572 mode = HandleProcessStateChange(remote, false);
573 ctx.Events().ResetEvents(RNBContext::event_proc_state_changed);
574 set_events ^= RNBContext::event_proc_state_changed;
577 if (set_events & RNBContext::event_proc_thread_exiting) {
578 DNBLog("debugserver's process monitoring thread has exited.");
579 mode = eRNBRunLoopModeExit;
582 if (set_events & RNBContext::event_read_thread_exiting) {
583 // Out remote packet receiving thread exited, exit for now.
584 DNBLog(
585 "debugserver's packet communication to lldb has been shut down.");
586 if (ctx.HasValidProcessID()) {
587 nub_process_t pid = ctx.ProcessID();
588 // TODO: We should add code that will leave the current process
589 // in its current state and listen for another connection...
590 if (ctx.ProcessStateRunning()) {
591 if (ctx.GetDetachOnError()) {
592 DNBLog("debugserver has a valid PID %d, it is still running. "
593 "detaching from the inferior process.",
594 pid);
595 DNBProcessDetach(pid);
596 } else {
597 DNBLog("debugserver killing the inferior process, pid %d.", pid);
598 DNBProcessKill(pid);
600 } else {
601 if (ctx.GetDetachOnError()) {
602 DNBLog("debugserver has a valid PID %d but it may no longer "
603 "be running, detaching from the inferior process.",
604 pid);
605 DNBProcessDetach(pid);
609 mode = eRNBRunLoopModeExit;
613 // Reset all event bits that weren't reset for now...
614 if (set_events != 0)
615 ctx.Events().ResetEvents(set_events);
617 if (mode != eRNBRunLoopModeInferiorExecuting)
618 break;
621 return mode;
624 RNBRunLoopMode RNBRunLoopPlatform(RNBRemote *remote) {
625 RNBRunLoopMode mode = eRNBRunLoopModePlatformMode;
626 RNBContext &ctx = remote->Context();
628 while (mode == eRNBRunLoopModePlatformMode) {
629 std::string set_events_str;
630 const uint32_t event_mask = RNBContext::event_read_packet_available |
631 RNBContext::event_read_thread_exiting;
633 DNBLogThreadedIf(LOG_RNB_EVENTS,
634 "%s ctx.Events().WaitForSetEvents(0x%08x) ...",
635 __FUNCTION__, event_mask);
636 nub_event_t set_events = ctx.Events().WaitForSetEvents(event_mask);
637 DNBLogThreadedIf(LOG_RNB_EVENTS,
638 "%s ctx.Events().WaitForSetEvents(0x%08x) => 0x%08x (%s)",
639 __FUNCTION__, event_mask, set_events,
640 ctx.EventsAsString(set_events, set_events_str));
642 if (set_events) {
643 if (set_events & RNBContext::event_read_packet_available) {
644 if (remote->HandleReceivedPacket() == rnb_not_connected)
645 mode = eRNBRunLoopModeExit;
648 if (set_events & RNBContext::event_read_thread_exiting) {
649 mode = eRNBRunLoopModeExit;
651 ctx.Events().ResetEvents(set_events);
654 return eRNBRunLoopModeExit;
657 // Convenience function to set up the remote listening port
658 // Returns 1 for success 0 for failure.
660 static void PortWasBoundCallbackUnixSocket(const void *baton, in_port_t port) {
661 //::printf ("PortWasBoundCallbackUnixSocket (baton = %p, port = %u)\n", baton,
662 //port);
664 const char *unix_socket_name = (const char *)baton;
666 if (unix_socket_name && unix_socket_name[0]) {
667 // We were given a unix socket name to use to communicate the port
668 // that we ended up binding to back to our parent process
669 struct sockaddr_un saddr_un;
670 int s = ::socket(AF_UNIX, SOCK_STREAM, 0);
671 if (s < 0) {
672 perror("error: socket (AF_UNIX, SOCK_STREAM, 0)");
673 exit(1);
676 saddr_un.sun_family = AF_UNIX;
677 ::strlcpy(saddr_un.sun_path, unix_socket_name,
678 sizeof(saddr_un.sun_path) - 1);
679 saddr_un.sun_path[sizeof(saddr_un.sun_path) - 1] = '\0';
680 saddr_un.sun_len = SUN_LEN(&saddr_un);
682 if (::connect(s, (struct sockaddr *)&saddr_un,
683 static_cast<socklen_t>(SUN_LEN(&saddr_un))) < 0) {
684 perror("error: connect (socket, &saddr_un, saddr_un_len)");
685 exit(1);
688 //::printf ("connect () sucess!!\n");
690 // We were able to connect to the socket, now write our PID so whomever
691 // launched us will know this process's ID
692 RNBLogSTDOUT("Listening to port %i...\n", port);
694 char pid_str[64];
695 const int pid_str_len = ::snprintf(pid_str, sizeof(pid_str), "%u", port);
696 const ssize_t bytes_sent = ::send(s, pid_str, pid_str_len, 0);
698 if (pid_str_len != bytes_sent) {
699 perror("error: send (s, pid_str, pid_str_len, 0)");
700 exit(1);
703 //::printf ("send () sucess!!\n");
705 // We are done with the socket
706 close(s);
710 static void PortWasBoundCallbackNamedPipe(const void *baton, uint16_t port) {
711 const char *named_pipe = (const char *)baton;
712 if (named_pipe && named_pipe[0]) {
713 int fd = ::open(named_pipe, O_WRONLY);
714 if (fd > -1) {
715 char port_str[64];
716 const ssize_t port_str_len =
717 ::snprintf(port_str, sizeof(port_str), "%u", port);
718 // Write the port number as a C string with the NULL terminator
719 ::write(fd, port_str, port_str_len + 1);
720 close(fd);
725 static int ConnectRemote(RNBRemote *remote, const char *host, int port,
726 bool reverse_connect, const char *named_pipe_path,
727 const char *unix_socket_name) {
728 if (!remote->Comm().IsConnected()) {
729 if (reverse_connect) {
730 if (port == 0) {
731 DNBLogThreaded(
732 "error: invalid port supplied for reverse connection: %i.\n", port);
733 return 0;
735 if (remote->Comm().Connect(host, port) != rnb_success) {
736 DNBLogThreaded("Failed to reverse connect to %s:%i.\n", host, port);
737 return 0;
739 } else {
740 if (port != 0)
741 RNBLogSTDOUT("Listening to port %i for a connection from %s...\n", port,
742 host ? host : "127.0.0.1");
743 if (unix_socket_name && unix_socket_name[0]) {
744 if (remote->Comm().Listen(host, port, PortWasBoundCallbackUnixSocket,
745 unix_socket_name) != rnb_success) {
746 RNBLogSTDERR("Failed to get connection from a remote gdb process.\n");
747 return 0;
749 } else {
750 if (remote->Comm().Listen(host, port, PortWasBoundCallbackNamedPipe,
751 named_pipe_path) != rnb_success) {
752 RNBLogSTDERR("Failed to get connection from a remote gdb process.\n");
753 return 0;
757 remote->StartReadRemoteDataThread();
759 return 1;
762 // ASL Logging callback that can be registered with DNBLogSetLogCallback
763 void ASLLogCallback(void *baton, uint32_t flags, const char *format,
764 va_list args) {
765 if (format == NULL)
766 return;
767 static aslmsg g_aslmsg = NULL;
768 if (g_aslmsg == NULL) {
769 g_aslmsg = ::asl_new(ASL_TYPE_MSG);
770 char asl_key_sender[PATH_MAX];
771 snprintf(asl_key_sender, sizeof(asl_key_sender), "com.apple.%s-%s",
772 DEBUGSERVER_PROGRAM_NAME, DEBUGSERVER_VERSION_STR);
773 ::asl_set(g_aslmsg, ASL_KEY_SENDER, asl_key_sender);
776 int asl_level;
777 if (flags & DNBLOG_FLAG_FATAL)
778 asl_level = ASL_LEVEL_CRIT;
779 else if (flags & DNBLOG_FLAG_ERROR)
780 asl_level = ASL_LEVEL_ERR;
781 else if (flags & DNBLOG_FLAG_WARNING)
782 asl_level = ASL_LEVEL_WARNING;
783 else if (flags & DNBLOG_FLAG_VERBOSE)
784 asl_level = ASL_LEVEL_WARNING; // ASL_LEVEL_INFO;
785 else
786 asl_level = ASL_LEVEL_WARNING; // ASL_LEVEL_DEBUG;
788 ::asl_vlog(NULL, g_aslmsg, asl_level, format, args);
791 // FILE based Logging callback that can be registered with
792 // DNBLogSetLogCallback
793 void FileLogCallback(void *baton, uint32_t flags, const char *format,
794 va_list args) {
795 if (baton == NULL || format == NULL)
796 return;
798 ::vfprintf((FILE *)baton, format, args);
799 ::fprintf((FILE *)baton, "\n");
800 ::fflush((FILE *)baton);
803 void show_version_and_exit(int exit_code) {
804 const char *in_translation = "";
805 if (DNBDebugserverIsTranslated())
806 in_translation = " (running under translation)";
807 printf("%s-%s for %s%s.\n", DEBUGSERVER_PROGRAM_NAME, DEBUGSERVER_VERSION_STR,
808 RNB_ARCH, in_translation);
809 exit(exit_code);
812 void show_usage_and_exit(int exit_code) {
813 RNBLogSTDERR(
814 "Usage:\n %s host:port [program-name program-arg1 program-arg2 ...]\n",
815 DEBUGSERVER_PROGRAM_NAME);
816 RNBLogSTDERR(" %s /path/file [program-name program-arg1 program-arg2 ...]\n",
817 DEBUGSERVER_PROGRAM_NAME);
818 RNBLogSTDERR(" %s host:port --attach=<pid>\n", DEBUGSERVER_PROGRAM_NAME);
819 RNBLogSTDERR(" %s /path/file --attach=<pid>\n", DEBUGSERVER_PROGRAM_NAME);
820 RNBLogSTDERR(" %s host:port --attach=<process_name>\n",
821 DEBUGSERVER_PROGRAM_NAME);
822 RNBLogSTDERR(" %s /path/file --attach=<process_name>\n",
823 DEBUGSERVER_PROGRAM_NAME);
824 exit(exit_code);
827 // option descriptors for getopt_long_only()
828 static struct option g_long_options[] = {
829 {"attach", required_argument, NULL, 'a'},
830 {"arch", required_argument, NULL, 'A'},
831 {"debug", no_argument, NULL, 'g'},
832 {"kill-on-error", no_argument, NULL, 'K'},
833 {"verbose", no_argument, NULL, 'v'},
834 {"version", no_argument, NULL, 'V'},
835 {"lockdown", no_argument, &g_lockdown_opt, 1}, // short option "-k"
836 {"applist", no_argument, &g_applist_opt, 1}, // short option "-t"
837 {"log-file", required_argument, NULL, 'l'},
838 {"log-flags", required_argument, NULL, 'f'},
839 {"launch", required_argument, NULL, 'x'}, // Valid values are "auto",
840 // "posix-spawn", "fork-exec",
841 // "springboard" (arm only)
842 {"waitfor", required_argument, NULL,
843 'w'}, // Wait for a process whose name starts with ARG
844 {"waitfor-interval", required_argument, NULL,
845 'i'}, // Time in usecs to wait between sampling the pid list when waiting
846 // for a process by name
847 {"waitfor-duration", required_argument, NULL,
848 'd'}, // The time in seconds to wait for a process to show up by name
849 {"native-regs", no_argument, NULL, 'r'}, // Specify to use the native
850 // registers instead of the gdb
851 // defaults for the architecture.
852 {"stdio-path", required_argument, NULL,
853 's'}, // Set the STDIO path to be used when launching applications (STDIN,
854 // STDOUT and STDERR) (only if debugserver launches the process)
855 {"stdin-path", required_argument, NULL,
856 'I'}, // Set the STDIN path to be used when launching applications (only if
857 // debugserver launches the process)
858 {"stdout-path", required_argument, NULL,
859 'O'}, // Set the STDOUT path to be used when launching applications (only
860 // if debugserver launches the process)
861 {"stderr-path", required_argument, NULL,
862 'E'}, // Set the STDERR path to be used when launching applications (only
863 // if debugserver launches the process)
864 {"no-stdio", no_argument, NULL,
865 'n'}, // Do not set up any stdio (perhaps the program is a GUI program)
866 // (only if debugserver launches the process)
867 {"setsid", no_argument, NULL,
868 'S'}, // call setsid() to make debugserver run in its own session
869 {"disable-aslr", no_argument, NULL, 'D'}, // Use _POSIX_SPAWN_DISABLE_ASLR
870 // to avoid shared library
871 // randomization
872 {"working-dir", required_argument, NULL,
873 'W'}, // The working directory that the inferior process should have (only
874 // if debugserver launches the process)
875 {"platform", required_argument, NULL,
876 'p'}, // Put this executable into a remote platform mode
877 {"unix-socket", required_argument, NULL,
878 'u'}, // If we need to handshake with our parent process, an option will be
879 // passed down that specifies a unix socket name to use
880 {"fd", required_argument, NULL,
881 '2'}, // A file descriptor was passed to this process when spawned that
882 // is already open and ready for communication
883 {"named-pipe", required_argument, NULL, 'P'},
884 {"reverse-connect", no_argument, NULL, 'R'},
885 {"env", required_argument, NULL,
886 'e'}, // When debugserver launches the process, set a single environment
887 // entry as specified by the option value ("./debugserver -e FOO=1 -e
888 // BAR=2 localhost:1234 -- /bin/ls")
889 {"forward-env", no_argument, NULL,
890 'F'}, // When debugserver launches the process, forward debugserver's
891 // current environment variables to the child process ("./debugserver
892 // -F localhost:1234 -- /bin/ls"
893 {"unmask-signals", no_argument, NULL,
894 'U'}, // debugserver will ignore EXC_MASK_BAD_ACCESS,
895 // EXC_MASK_BAD_INSTRUCTION and EXC_MASK_ARITHMETIC, which results in
896 // SIGSEGV, SIGILL and SIGFPE being propagated to the target process.
897 {NULL, 0, NULL, 0}};
899 int communication_fd = -1;
901 // main
902 int main(int argc, char *argv[]) {
903 // If debugserver is launched with DYLD_INSERT_LIBRARIES, unset it so we
904 // don't spawn child processes with this enabled.
905 unsetenv("DYLD_INSERT_LIBRARIES");
907 const char *argv_sub_zero =
908 argv[0]; // save a copy of argv[0] for error reporting post-launch
910 #if defined(__APPLE__)
911 pthread_setname_np("main thread");
912 #if defined(__arm__) || defined(__arm64__) || defined(__aarch64__)
913 struct sched_param thread_param;
914 int thread_sched_policy;
915 if (pthread_getschedparam(pthread_self(), &thread_sched_policy,
916 &thread_param) == 0) {
917 thread_param.sched_priority = 47;
918 pthread_setschedparam(pthread_self(), thread_sched_policy, &thread_param);
921 ::proc_set_wakemon_params(
922 getpid(), 500,
923 0); // Allow up to 500 wakeups/sec to avoid EXC_RESOURCE for normal use.
924 #endif
925 #endif
927 g_isatty = ::isatty(STDIN_FILENO);
929 // ::printf ("uid=%u euid=%u gid=%u egid=%u\n",
930 // getuid(),
931 // geteuid(),
932 // getgid(),
933 // getegid());
935 // signal (SIGINT, signal_handler);
936 signal(SIGPIPE, signal_handler);
937 signal(SIGHUP, signal_handler);
939 // We're always sitting in waitpid or kevent waiting on our target process'
940 // death,
941 // we don't need no stinking SIGCHLD's...
943 sigset_t sigset;
944 sigemptyset(&sigset);
945 sigaddset(&sigset, SIGCHLD);
946 sigprocmask(SIG_BLOCK, &sigset, NULL);
948 g_remoteSP = std::make_shared<RNBRemote>();
950 RNBRemote *remote = g_remoteSP.get();
951 if (remote == NULL) {
952 RNBLogSTDERR("error: failed to create a remote connection class\n");
953 return -1;
956 RNBContext &ctx = remote->Context();
958 int i;
959 int attach_pid = INVALID_NUB_PROCESS;
961 FILE *log_file = NULL;
962 uint32_t log_flags = 0;
963 // Parse our options
964 int ch;
965 int long_option_index = 0;
966 int debug = 0;
967 std::string compile_options;
968 std::string waitfor_pid_name; // Wait for a process that starts with this name
969 std::string attach_pid_name;
970 std::string arch_name;
971 std::string working_dir; // The new working directory to use for the inferior
972 std::string unix_socket_name; // If we need to handshake with our parent
973 // process, an option will be passed down that
974 // specifies a unix socket name to use
975 std::string named_pipe_path; // If we need to handshake with our parent
976 // process, an option will be passed down that
977 // specifies a named pipe to use
978 useconds_t waitfor_interval = 1000; // Time in usecs between process lists
979 // polls when waiting for a process by
980 // name, default 1 msec.
981 useconds_t waitfor_duration =
982 0; // Time in seconds to wait for a process by name, 0 means wait forever.
983 bool no_stdio = false;
984 bool reverse_connect = false; // Set to true by an option to indicate we
985 // should reverse connect to the host:port
986 // supplied as the first debugserver argument
988 #if !defined(DNBLOG_ENABLED)
989 compile_options += "(no-logging) ";
990 #endif
992 RNBRunLoopMode start_mode = eRNBRunLoopModeExit;
994 char short_options[512];
995 uint32_t short_options_idx = 0;
997 // Handle the two case that don't have short options in g_long_options
998 short_options[short_options_idx++] = 'k';
999 short_options[short_options_idx++] = 't';
1001 for (i = 0; g_long_options[i].name != NULL; ++i) {
1002 if (isalpha(g_long_options[i].val)) {
1003 short_options[short_options_idx++] = g_long_options[i].val;
1004 switch (g_long_options[i].has_arg) {
1005 default:
1006 case no_argument:
1007 break;
1009 case optional_argument:
1010 short_options[short_options_idx++] = ':';
1011 short_options[short_options_idx++] = ':';
1012 break;
1013 case required_argument:
1014 short_options[short_options_idx++] = ':';
1015 break;
1019 // NULL terminate the short option string.
1020 short_options[short_options_idx++] = '\0';
1022 #if __GLIBC__
1023 optind = 0;
1024 #else
1025 optreset = 1;
1026 optind = 1;
1027 #endif
1029 bool forward_env = false;
1030 while ((ch = getopt_long_only(argc, argv, short_options, g_long_options,
1031 &long_option_index)) != -1) {
1032 DNBLogDebug("option: ch == %c (0x%2.2x) --%s%c%s\n", ch, (uint8_t)ch,
1033 g_long_options[long_option_index].name,
1034 g_long_options[long_option_index].has_arg ? '=' : ' ',
1035 optarg ? optarg : "");
1036 switch (ch) {
1037 case 0: // Any optional that auto set themselves will return 0
1038 break;
1040 case 'A':
1041 if (optarg && optarg[0])
1042 arch_name.assign(optarg);
1043 break;
1045 case 'a':
1046 if (optarg && optarg[0]) {
1047 if (isdigit(optarg[0])) {
1048 char *end = NULL;
1049 attach_pid = static_cast<int>(strtoul(optarg, &end, 0));
1050 if (end == NULL || *end != '\0') {
1051 RNBLogSTDERR("error: invalid pid option '%s'\n", optarg);
1052 exit(4);
1054 } else {
1055 attach_pid_name = optarg;
1057 start_mode = eRNBRunLoopModeInferiorAttaching;
1059 break;
1061 // --waitfor=NAME
1062 case 'w':
1063 if (optarg && optarg[0]) {
1064 waitfor_pid_name = optarg;
1065 start_mode = eRNBRunLoopModeInferiorAttaching;
1067 break;
1069 // --waitfor-interval=USEC
1070 case 'i':
1071 if (optarg && optarg[0]) {
1072 char *end = NULL;
1073 waitfor_interval = static_cast<useconds_t>(strtoul(optarg, &end, 0));
1074 if (end == NULL || *end != '\0') {
1075 RNBLogSTDERR("error: invalid waitfor-interval option value '%s'.\n",
1076 optarg);
1077 exit(6);
1080 break;
1082 // --waitfor-duration=SEC
1083 case 'd':
1084 if (optarg && optarg[0]) {
1085 char *end = NULL;
1086 waitfor_duration = static_cast<useconds_t>(strtoul(optarg, &end, 0));
1087 if (end == NULL || *end != '\0') {
1088 RNBLogSTDERR("error: invalid waitfor-duration option value '%s'.\n",
1089 optarg);
1090 exit(7);
1093 break;
1095 case 'K':
1096 g_detach_on_error = false;
1097 break;
1098 case 'W':
1099 if (optarg && optarg[0])
1100 working_dir.assign(optarg);
1101 break;
1103 case 'x':
1104 if (optarg && optarg[0]) {
1105 if (strcasecmp(optarg, "auto") == 0)
1106 g_launch_flavor = eLaunchFlavorDefault;
1107 else if (strcasestr(optarg, "posix") == optarg) {
1108 DNBLog(
1109 "[LaunchAttach] launch flavor is posix_spawn via cmdline option");
1110 g_launch_flavor = eLaunchFlavorPosixSpawn;
1111 } else if (strcasestr(optarg, "fork") == optarg)
1112 g_launch_flavor = eLaunchFlavorForkExec;
1113 #ifdef WITH_SPRINGBOARD
1114 else if (strcasestr(optarg, "spring") == optarg) {
1115 DNBLog(
1116 "[LaunchAttach] launch flavor is SpringBoard via cmdline option");
1117 g_launch_flavor = eLaunchFlavorSpringBoard;
1119 #endif
1120 #ifdef WITH_BKS
1121 else if (strcasestr(optarg, "backboard") == optarg) {
1122 DNBLog("[LaunchAttach] launch flavor is BKS via cmdline option");
1123 g_launch_flavor = eLaunchFlavorBKS;
1125 #endif
1126 #ifdef WITH_FBS
1127 else if (strcasestr(optarg, "frontboard") == optarg) {
1128 DNBLog("[LaunchAttach] launch flavor is FBS via cmdline option");
1129 g_launch_flavor = eLaunchFlavorFBS;
1131 #endif
1133 else {
1134 RNBLogSTDERR("error: invalid TYPE for the --launch=TYPE (-x TYPE) "
1135 "option: '%s'\n",
1136 optarg);
1137 RNBLogSTDERR("Valid values TYPE are:\n");
1138 RNBLogSTDERR(
1139 " auto Auto-detect the best launch method to use.\n");
1140 RNBLogSTDERR(
1141 " posix Launch the executable using posix_spawn.\n");
1142 RNBLogSTDERR(
1143 " fork Launch the executable using fork and exec.\n");
1144 #ifdef WITH_SPRINGBOARD
1145 RNBLogSTDERR(
1146 " spring Launch the executable through Springboard.\n");
1147 #endif
1148 #ifdef WITH_BKS
1149 RNBLogSTDERR(" backboard Launch the executable through BackBoard "
1150 "Services.\n");
1151 #endif
1152 #ifdef WITH_FBS
1153 RNBLogSTDERR(" frontboard Launch the executable through FrontBoard "
1154 "Services.\n");
1155 #endif
1156 exit(5);
1159 break;
1161 case 'l': // Set Log File
1162 if (optarg && optarg[0]) {
1163 if (strcasecmp(optarg, "stdout") == 0)
1164 log_file = stdout;
1165 else if (strcasecmp(optarg, "stderr") == 0)
1166 log_file = stderr;
1167 else {
1168 log_file = fopen(optarg, "w");
1169 if (log_file != NULL)
1170 setlinebuf(log_file);
1173 if (log_file == NULL) {
1174 const char *errno_str = strerror(errno);
1175 RNBLogSTDERR(
1176 "Failed to open log file '%s' for writing: errno = %i (%s)",
1177 optarg, errno, errno_str ? errno_str : "unknown error");
1180 break;
1182 case 'f': // Log Flags
1183 if (optarg && optarg[0])
1184 log_flags = static_cast<uint32_t>(strtoul(optarg, NULL, 0));
1185 break;
1187 case 'g':
1188 debug = 1;
1189 DNBLogSetDebug(debug);
1190 break;
1192 case 't':
1193 g_applist_opt = 1;
1194 break;
1196 case 'k':
1197 g_lockdown_opt = 1;
1198 break;
1200 case 'r':
1201 // Do nothing, native regs is the default these days
1202 break;
1204 case 'R':
1205 reverse_connect = true;
1206 break;
1207 case 'v':
1208 DNBLogSetVerbose(1);
1209 break;
1211 case 'V':
1212 show_version_and_exit(0);
1213 break;
1215 case 's':
1216 ctx.GetSTDIN().assign(optarg);
1217 ctx.GetSTDOUT().assign(optarg);
1218 ctx.GetSTDERR().assign(optarg);
1219 break;
1221 case 'I':
1222 ctx.GetSTDIN().assign(optarg);
1223 break;
1225 case 'O':
1226 ctx.GetSTDOUT().assign(optarg);
1227 break;
1229 case 'E':
1230 ctx.GetSTDERR().assign(optarg);
1231 break;
1233 case 'n':
1234 no_stdio = true;
1235 break;
1237 case 'S':
1238 // Put debugserver into a new session. Terminals group processes
1239 // into sessions and when a special terminal key sequences
1240 // (like control+c) are typed they can cause signals to go out to
1241 // all processes in a session. Using this --setsid (-S) option
1242 // will cause debugserver to run in its own sessions and be free
1243 // from such issues.
1245 // This is useful when debugserver is spawned from a command
1246 // line application that uses debugserver to do the debugging,
1247 // yet that application doesn't want debugserver receiving the
1248 // signals sent to the session (i.e. dying when anyone hits ^C).
1249 setsid();
1250 break;
1251 case 'D':
1252 g_disable_aslr = 1;
1253 break;
1255 case 'p':
1256 start_mode = eRNBRunLoopModePlatformMode;
1257 break;
1259 case 'u':
1260 unix_socket_name.assign(optarg);
1261 break;
1263 case 'P':
1264 named_pipe_path.assign(optarg);
1265 break;
1267 case 'e':
1268 // Pass a single specified environment variable down to the process that
1269 // gets launched
1270 remote->Context().PushEnvironment(optarg);
1271 break;
1273 case 'F':
1274 forward_env = true;
1275 break;
1277 case 'U':
1278 ctx.SetUnmaskSignals(true);
1279 break;
1281 case '2':
1282 // File descriptor passed to this process during fork/exec and is already
1283 // open and ready for communication.
1284 communication_fd = atoi(optarg);
1285 break;
1289 if (arch_name.empty()) {
1290 #if defined(__arm__)
1291 arch_name.assign("arm");
1292 #endif
1293 } else {
1294 DNBSetArchitecture(arch_name.c_str());
1297 // if (arch_name.empty())
1298 // {
1299 // fprintf(stderr, "error: no architecture was specified\n");
1300 // exit (8);
1301 // }
1302 // Skip any options we consumed with getopt_long_only
1303 argc -= optind;
1304 argv += optind;
1306 if (!working_dir.empty()) {
1307 if (remote->Context().SetWorkingDirectory(working_dir.c_str()) == false) {
1308 RNBLogSTDERR("error: working directory doesn't exist '%s'.\n",
1309 working_dir.c_str());
1310 exit(8);
1314 remote->Context().SetDetachOnError(g_detach_on_error);
1316 remote->Initialize();
1318 // It is ok for us to set NULL as the logfile (this will disable any logging)
1320 if (log_file != NULL) {
1321 DNBLogSetLogCallback(FileLogCallback, log_file);
1322 // If our log file was set, yet we have no log flags, log everything!
1323 if (log_flags == 0)
1324 log_flags = LOG_ALL | LOG_RNB_ALL;
1326 DNBLogSetLogMask(log_flags);
1327 } else {
1328 // Enable DNB logging
1330 // if os_log() support is available, log through that.
1331 auto log_callback = OsLogger::GetLogFunction();
1332 if (log_callback) {
1333 DNBLogSetLogCallback(log_callback, nullptr);
1334 DNBLog("debugserver will use os_log for internal logging.");
1335 } else {
1336 // Fall back to ASL support.
1337 DNBLogSetLogCallback(ASLLogCallback, NULL);
1338 DNBLog("debugserver will use ASL for internal logging.");
1340 DNBLogSetLogMask(log_flags);
1343 if (DNBLogEnabled()) {
1344 for (i = 0; i < argc; i++)
1345 DNBLogDebug("argv[%i] = %s", i, argv[i]);
1348 // as long as we're dropping remotenub in as a replacement for gdbserver,
1349 // explicitly note that this is not gdbserver.
1351 const char *in_translation = "";
1352 if (DNBDebugserverIsTranslated())
1353 in_translation = " (running under translation)";
1354 RNBLogSTDOUT("%s-%s %sfor %s%s.\n", DEBUGSERVER_PROGRAM_NAME,
1355 DEBUGSERVER_VERSION_STR, compile_options.c_str(), RNB_ARCH,
1356 in_translation);
1358 std::string host;
1359 int port = INT32_MAX;
1360 char str[PATH_MAX];
1361 str[0] = '\0';
1363 if (g_lockdown_opt == 0 && g_applist_opt == 0 && communication_fd == -1) {
1364 // Make sure we at least have port
1365 if (argc < 1) {
1366 show_usage_and_exit(1);
1368 // accept 'localhost:' prefix on port number
1369 std::string host_specifier = argv[0];
1370 auto colon_location = host_specifier.rfind(':');
1371 if (colon_location != std::string::npos) {
1372 host = host_specifier.substr(0, colon_location);
1373 std::string port_str =
1374 host_specifier.substr(colon_location + 1, std::string::npos);
1375 char *end_ptr;
1376 port = strtoul(port_str.c_str(), &end_ptr, 0);
1377 if (end_ptr < port_str.c_str() + port_str.size())
1378 show_usage_and_exit(2);
1379 if (host.front() == '[' && host.back() == ']')
1380 host = host.substr(1, host.size() - 2);
1381 DNBLogDebug("host = '%s' port = %i", host.c_str(), port);
1382 } else {
1383 // No hostname means "localhost"
1384 int items_scanned = ::sscanf(argv[0], "%i", &port);
1385 if (items_scanned == 1) {
1386 host = "127.0.0.1";
1387 DNBLogDebug("host = '%s' port = %i", host.c_str(), port);
1388 } else if (argv[0][0] == '/') {
1389 port = INT32_MAX;
1390 strlcpy(str, argv[0], sizeof(str));
1391 } else {
1392 show_usage_and_exit(2);
1396 // We just used the 'host:port' or the '/path/file' arg...
1397 argc--;
1398 argv++;
1401 // If we know we're waiting to attach, we don't need any of this other info.
1402 if (start_mode != eRNBRunLoopModeInferiorAttaching &&
1403 start_mode != eRNBRunLoopModePlatformMode) {
1404 if (argc == 0 || g_lockdown_opt) {
1405 if (g_lockdown_opt != 0) {
1406 // Work around for SIGPIPE crashes due to posix_spawn issue.
1407 // We have to close STDOUT and STDERR, else the first time we
1408 // try and do any, we get SIGPIPE and die as posix_spawn is
1409 // doing bad things with our file descriptors at the moment.
1410 int null = open("/dev/null", O_RDWR);
1411 dup2(null, STDOUT_FILENO);
1412 dup2(null, STDERR_FILENO);
1413 } else if (g_applist_opt != 0) {
1414 DNBLog("debugserver running in --applist mode");
1415 // List all applications we are able to see
1416 std::string applist_plist;
1417 int err = ListApplications(applist_plist, false, false);
1418 if (err == 0) {
1419 fputs(applist_plist.c_str(), stdout);
1420 } else {
1421 RNBLogSTDERR("error: ListApplications returned error %i\n", err);
1423 // Exit with appropriate error if we were asked to list the applications
1424 // with no other args were given (and we weren't trying to do this over
1425 // lockdown)
1426 return err;
1429 DNBLogDebug("Get args from remote protocol...");
1430 start_mode = eRNBRunLoopModeGetStartModeFromRemoteProtocol;
1431 } else {
1432 start_mode = eRNBRunLoopModeInferiorLaunching;
1433 // Fill in the argv array in the context from the rest of our args.
1434 // Skip the name of this executable and the port number
1435 for (int i = 0; i < argc; i++) {
1436 DNBLogDebug("inferior_argv[%i] = '%s'", i, argv[i]);
1437 ctx.PushArgument(argv[i]);
1442 if (start_mode == eRNBRunLoopModeExit)
1443 return -1;
1445 if (forward_env || start_mode == eRNBRunLoopModeInferiorLaunching) {
1446 // Pass the current environment down to the process that gets launched
1447 // This happens automatically in the "launching" mode. For the rest, we
1448 // only do that if the user explicitly requested this via --forward-env
1449 // argument.
1450 char **host_env = *_NSGetEnviron();
1451 char *env_entry;
1452 size_t i;
1453 for (i = 0; (env_entry = host_env[i]) != NULL; ++i)
1454 remote->Context().PushEnvironmentIfNeeded(env_entry);
1457 RNBRunLoopMode mode = start_mode;
1458 char err_str[1024] = {'\0'};
1460 while (mode != eRNBRunLoopModeExit) {
1461 switch (mode) {
1462 case eRNBRunLoopModeGetStartModeFromRemoteProtocol:
1463 #ifdef WITH_LOCKDOWN
1464 if (g_lockdown_opt) {
1465 if (!remote->Comm().IsConnected()) {
1466 if (remote->Comm().ConnectToService() != rnb_success) {
1467 RNBLogSTDERR(
1468 "Failed to get connection from a remote gdb process.\n");
1469 mode = eRNBRunLoopModeExit;
1470 } else if (g_applist_opt != 0) {
1471 // List all applications we are able to see
1472 DNBLog("debugserver running in applist mode under lockdown");
1473 std::string applist_plist;
1474 if (ListApplications(applist_plist, false, false) == 0) {
1475 DNBLogDebug("Task list: %s", applist_plist.c_str());
1477 remote->Comm().Write(applist_plist.c_str(), applist_plist.size());
1478 // Issue a read that will never yield any data until the other
1479 // side
1480 // closes the socket so this process doesn't just exit and cause
1481 // the
1482 // socket to close prematurely on the other end and cause data
1483 // loss.
1484 std::string buf;
1485 remote->Comm().Read(buf);
1487 remote->Comm().Disconnect(false);
1488 mode = eRNBRunLoopModeExit;
1489 break;
1490 } else {
1491 // Start watching for remote packets
1492 remote->StartReadRemoteDataThread();
1495 } else
1496 #endif
1497 if (port != INT32_MAX) {
1498 if (!ConnectRemote(remote, host.c_str(), port, reverse_connect,
1499 named_pipe_path.c_str(), unix_socket_name.c_str()))
1500 mode = eRNBRunLoopModeExit;
1501 } else if (str[0] == '/') {
1502 if (remote->Comm().OpenFile(str))
1503 mode = eRNBRunLoopModeExit;
1504 } else if (communication_fd >= 0) {
1505 // We were passed a file descriptor to use during fork/exec that is
1506 // already open
1507 // in our process, so lets just use it!
1508 if (remote->Comm().useFD(communication_fd))
1509 mode = eRNBRunLoopModeExit;
1510 else
1511 remote->StartReadRemoteDataThread();
1514 if (mode != eRNBRunLoopModeExit) {
1515 RNBLogSTDOUT("Got a connection, waiting for process information for "
1516 "launching or attaching.\n");
1518 mode = RNBRunLoopGetStartModeFromRemote(remote);
1520 break;
1522 case eRNBRunLoopModeInferiorAttaching:
1523 if (!waitfor_pid_name.empty()) {
1524 // Set our end wait time if we are using a waitfor-duration
1525 // option that may have been specified
1526 struct timespec attach_timeout_abstime, *timeout_ptr = NULL;
1527 if (waitfor_duration != 0) {
1528 DNBTimer::OffsetTimeOfDay(&attach_timeout_abstime, waitfor_duration,
1530 timeout_ptr = &attach_timeout_abstime;
1532 nub_launch_flavor_t launch_flavor = g_launch_flavor;
1533 if (launch_flavor == eLaunchFlavorDefault)
1534 launch_flavor = default_launch_flavor(waitfor_pid_name.c_str());
1536 ctx.SetLaunchFlavor(launch_flavor);
1537 bool ignore_existing = false;
1538 RNBLogSTDOUT("Waiting to attach to process %s...\n",
1539 waitfor_pid_name.c_str());
1540 nub_process_t pid = DNBProcessAttachWait(
1541 &ctx, waitfor_pid_name.c_str(), ignore_existing, timeout_ptr,
1542 waitfor_interval, err_str, sizeof(err_str));
1543 g_pid = pid;
1545 if (pid == INVALID_NUB_PROCESS) {
1546 ctx.LaunchStatus().SetError(-1, DNBError::Generic);
1547 if (err_str[0])
1548 ctx.LaunchStatus().SetErrorString(err_str);
1549 RNBLogSTDERR("error: failed to attach to process named: \"%s\" %s\n",
1550 waitfor_pid_name.c_str(), err_str);
1551 mode = eRNBRunLoopModeExit;
1552 } else {
1553 ctx.SetProcessID(pid);
1554 mode = eRNBRunLoopModeInferiorExecuting;
1556 } else if (attach_pid != INVALID_NUB_PROCESS) {
1558 RNBLogSTDOUT("Attaching to process %i...\n", attach_pid);
1559 nub_process_t attached_pid;
1560 mode = RNBRunLoopLaunchAttaching(remote, attach_pid, attached_pid);
1561 if (mode != eRNBRunLoopModeInferiorExecuting) {
1562 const char *error_str = remote->Context().LaunchStatus().AsString();
1563 RNBLogSTDERR("error: failed to attach process %i: %s\n", attach_pid,
1564 error_str ? error_str : "unknown error.");
1565 mode = eRNBRunLoopModeExit;
1567 } else if (!attach_pid_name.empty()) {
1568 struct timespec attach_timeout_abstime, *timeout_ptr = NULL;
1569 if (waitfor_duration != 0) {
1570 DNBTimer::OffsetTimeOfDay(&attach_timeout_abstime, waitfor_duration,
1572 timeout_ptr = &attach_timeout_abstime;
1575 RNBLogSTDOUT("Attaching to process %s...\n", attach_pid_name.c_str());
1576 nub_process_t pid = DNBProcessAttachByName(
1577 attach_pid_name.c_str(), timeout_ptr, ctx.GetUnmaskSignals(),
1578 err_str, sizeof(err_str));
1579 g_pid = pid;
1580 if (pid == INVALID_NUB_PROCESS) {
1581 ctx.LaunchStatus().SetError(-1, DNBError::Generic);
1582 if (err_str[0])
1583 ctx.LaunchStatus().SetErrorString(err_str);
1584 RNBLogSTDERR("error: failed to attach to process named: \"%s\" %s\n",
1585 waitfor_pid_name.c_str(), err_str);
1586 mode = eRNBRunLoopModeExit;
1587 } else {
1588 ctx.SetProcessID(pid);
1589 mode = eRNBRunLoopModeInferiorExecuting;
1592 } else {
1593 RNBLogSTDERR(
1594 "error: asked to attach with empty name and invalid PID.\n");
1595 mode = eRNBRunLoopModeExit;
1598 if (mode != eRNBRunLoopModeExit) {
1599 if (port != INT32_MAX) {
1600 if (!ConnectRemote(remote, host.c_str(), port, reverse_connect,
1601 named_pipe_path.c_str(), unix_socket_name.c_str()))
1602 mode = eRNBRunLoopModeExit;
1603 } else if (str[0] == '/') {
1604 if (remote->Comm().OpenFile(str))
1605 mode = eRNBRunLoopModeExit;
1606 } else if (communication_fd >= 0) {
1607 // We were passed a file descriptor to use during fork/exec that is
1608 // already open
1609 // in our process, so lets just use it!
1610 if (remote->Comm().useFD(communication_fd))
1611 mode = eRNBRunLoopModeExit;
1612 else
1613 remote->StartReadRemoteDataThread();
1616 if (mode != eRNBRunLoopModeExit)
1617 RNBLogSTDOUT("Waiting for debugger instructions for process %d.\n",
1618 attach_pid);
1620 break;
1622 case eRNBRunLoopModeInferiorLaunching: {
1623 mode = RNBRunLoopLaunchInferior(remote, ctx.GetSTDINPath(),
1624 ctx.GetSTDOUTPath(), ctx.GetSTDERRPath(),
1625 no_stdio);
1627 if (mode == eRNBRunLoopModeInferiorExecuting) {
1628 if (port != INT32_MAX) {
1629 if (!ConnectRemote(remote, host.c_str(), port, reverse_connect,
1630 named_pipe_path.c_str(), unix_socket_name.c_str()))
1631 mode = eRNBRunLoopModeExit;
1632 } else if (str[0] == '/') {
1633 if (remote->Comm().OpenFile(str))
1634 mode = eRNBRunLoopModeExit;
1635 } else if (communication_fd >= 0) {
1636 // We were passed a file descriptor to use during fork/exec that is
1637 // already open
1638 // in our process, so lets just use it!
1639 if (remote->Comm().useFD(communication_fd))
1640 mode = eRNBRunLoopModeExit;
1641 else
1642 remote->StartReadRemoteDataThread();
1645 if (mode != eRNBRunLoopModeExit) {
1646 const char *proc_name = "<unknown>";
1647 if (ctx.ArgumentCount() > 0)
1648 proc_name = ctx.ArgumentAtIndex(0);
1649 DNBLog("[LaunchAttach] Successfully launched %s (pid = %d).\n",
1650 proc_name, ctx.ProcessID());
1651 RNBLogSTDOUT("Got a connection, launched process %s (pid = %d).\n",
1652 proc_name, ctx.ProcessID());
1654 } else {
1655 const char *error_str = remote->Context().LaunchStatus().AsString();
1656 RNBLogSTDERR("error: failed to launch process %s: %s\n", argv_sub_zero,
1657 error_str ? error_str : "unknown error.");
1659 } break;
1661 case eRNBRunLoopModeInferiorExecuting:
1662 mode = RNBRunLoopInferiorExecuting(remote);
1663 break;
1665 case eRNBRunLoopModePlatformMode:
1666 if (port != INT32_MAX) {
1667 if (!ConnectRemote(remote, host.c_str(), port, reverse_connect,
1668 named_pipe_path.c_str(), unix_socket_name.c_str()))
1669 mode = eRNBRunLoopModeExit;
1670 } else if (str[0] == '/') {
1671 if (remote->Comm().OpenFile(str))
1672 mode = eRNBRunLoopModeExit;
1673 } else if (communication_fd >= 0) {
1674 // We were passed a file descriptor to use during fork/exec that is
1675 // already open
1676 // in our process, so lets just use it!
1677 if (remote->Comm().useFD(communication_fd))
1678 mode = eRNBRunLoopModeExit;
1679 else
1680 remote->StartReadRemoteDataThread();
1683 if (mode != eRNBRunLoopModeExit)
1684 mode = RNBRunLoopPlatform(remote);
1685 break;
1687 default:
1688 mode = eRNBRunLoopModeExit;
1689 break;
1690 case eRNBRunLoopModeExit:
1691 break;
1695 remote->StopReadRemoteDataThread();
1696 remote->Context().SetProcessID(INVALID_NUB_PROCESS);
1697 RNBLogSTDOUT("Exiting.\n");
1699 return 0;