static inline RList *__getioplugin_many(RIO *io, const char *_uri, int flags, int mode) {
RIOPlugin *plugin, *iop = NULL;
RList *list_fds = NULL;
RListIter *iter;
RIODesc *desc;
char *uri = strdup (_uri);
for (;;) {
// resolve
plugin = r_io_plugin_resolve (io, uri, 1);
if (plugin && plugin->open_many) {
// open
list_fds = plugin->open_many (io, uri, flags, mode);
if (io->redirect) {
free (uri);
uri = strdup (io->redirect);
r_io_redirect (io, NULL);
continue;
}
}
break;
}
if (!list_fds) {
free (uri);
return NULL;
}
r_list_foreach (list_fds, iter, desc) {
desc->uri = strdup (uri);
}
static inline RIODesc *__getioplugin(RIO *io, const char *uri, int flags, int mode) {
RIOPlugin *plugin, *iop = NULL;
RIODesc *desc = NULL;
for (;;) {
plugin = r_io_plugin_resolve (io, uri);
if (plugin && plugin->open) {
desc = plugin->open (io, uri, flags, mode);
if (io->redirect) {
uri = io->redirect;
r_io_redirect (io, NULL);
continue;
}
if (desc != NULL) {
r_io_desc_add (io, desc);
if (desc->fd != -1)
r_io_plugin_open (io, desc->fd, plugin);
if (desc != io->fd)
iop = plugin;
}
}
break;
}
io->plugin = iop;
return desc;
}
R_API RIODesc *r_io_open(struct r_io_t *io, const char *file, int flags, int mode) {
RIODesc *desc = NULL;
int fd = -2;
char *uri = strdup (file);
struct r_io_plugin_t *plugin;
if (!io) return NULL;
for (;;) {
plugin = r_io_plugin_resolve (io, uri);
if (plugin && plugin->open) {
desc = plugin->open (io, uri, flags, mode);
if (io->redirect) {
free ((void *)uri);
uri = strdup (io->redirect);
r_io_redirect (io, NULL);
continue;
}
if (desc != NULL) {
r_io_desc_add (io, desc);
fd = desc->fd;
if (fd != -1)
r_io_plugin_open (io, fd, plugin);
if (desc != io->fd)
io->plugin = plugin;
}
}
break;
}
if (fd == -2) {
#if __WINDOWS__
if (flags & R_IO_WRITE) {
fd = open (file, O_BINARY | 1);
if (fd == -1)
creat (file, O_BINARY);
fd = open (file, O_BINARY | 1);
} else fd = open (file, O_BINARY);
#else
fd = open (file, (flags&R_IO_WRITE)?O_RDWR:O_RDONLY, mode);
#endif
}
if (fd >= 0) {
if (desc == NULL)
desc = r_io_desc_new (io->plugin, fd, file, flags, mode, NULL);
r_io_desc_add (io, desc);
r_io_set_fd (io, desc);
}
free ((void *)uri);
return desc;
}
static inline RIODesc *__getioplugin(RIO *io, const char *_uri, int flags, int mode) {
RIOPlugin *plugin;
RIODesc *desc = NULL;
char *uri = strdup (_uri? _uri: "");
char *redir = NULL;
for (;;) {
plugin = r_io_plugin_resolve (io, uri, 0);
if (plugin && plugin->open) {
desc = plugin->open (io, uri, flags, mode);
if (io->redirect) {
redir = uri;
uri = strdup (io->redirect);
r_io_redirect (io, NULL);
continue;
}
if (desc) {
desc->uri = uri;
desc->referer = redir;
io->plugin = plugin;
}
}
break;
}
if (!desc) {
plugin = r_io_plugin_get_default (io, uri, 0);
desc = (plugin && plugin->open)
? plugin->open (io, uri, flags, mode)
: NULL;
if (desc) {
desc->uri = uri;
io->plugin = plugin;
}
}
if (!desc) {
free (uri);
io->plugin = NULL;
}
return desc;
}
static int fork_and_ptraceme(const char *cmd) {
char **argv;
int ret, status, pid = vfork ();
switch (pid) {
case -1:
perror ("fork_and_ptraceme");
break;
case 0:
#if __APPLE__
signal (SIGTRAP, SIG_IGN); //NEED BY STEP
#endif
#if __APPLE__ || __BSD__
/* we can probably remove this #if..as long as PT_TRACE_ME is redefined for OSX in r_debug.h */
signal (SIGABRT, inferior_abort_handler);
if (ptrace (PT_TRACE_ME, 0, 0, 0) != 0) {
#else
if (ptrace (PTRACE_TRACEME, 0, NULL, NULL) != 0) {
#endif
eprintf ("ptrace-traceme failed\n");
exit (MAGIC_EXIT);
}
// TODO: Add support to redirect filedescriptors
// TODO: Configure process environment
argv = r_str_argv (cmd, NULL);
execvp (argv[0], argv);
r_str_argv_free (argv);
perror ("fork_and_attach: execv");
//printf(stderr, "[%d] %s execv failed.\n", getpid(), ps.filename);
exit (MAGIC_EXIT); /* error */
return 0; // invalid pid // if exit is overriden.. :)
default:
/* XXX: clean this dirty code */
ret = wait (&status);
if (ret != pid)
eprintf ("Wait event received by different pid %d\n", ret);
if (WIFSTOPPED (status))
eprintf ("Process with PID %d started...\n", (int)pid);
if (WEXITSTATUS (status) == MAGIC_EXIT)
pid = -1;
// XXX kill (pid, SIGSTOP);
break;
}
eprintf ("PID = %d\n", pid);
return pid;
}
#endif
static int __plugin_open(struct r_io_t *io, const char *file) {
if (!memcmp (file, "dbg://", 6) && file[6])
return R_TRUE;
return R_FALSE;
}
static RIODesc *__open(RIO *io, const char *file, int rw, int mode) {
char uri[128];
if (__plugin_open (io, file)) {
int pid = atoi (file+6);
if (pid == 0) {
pid = fork_and_ptraceme (file+6);
if (pid==-1)
return NULL;
#if __WINDOWS__
sprintf (uri, "w32dbg://%d", pid);
#elif __APPLE__
sprintf (uri, "mach://%d", pid);
#else
// TODO: use io_procpid here? faster or what?
sprintf (uri, "ptrace://%d", pid);
#endif
eprintf ("io_redirect: %s\n", uri);
r_io_redirect (io, uri);
return NULL;
} else {
sprintf (uri, "attach://%d", pid);
r_io_redirect (io, uri);
return NULL;
}
}
r_io_redirect (io, NULL);
return NULL;
}
static int fork_and_ptraceme(int bits, const char *cmd) {
char **argv;
int ret, status, pid = fork ();
switch (pid) {
case -1:
perror ("fork_and_ptraceme");
break;
case 0:
#if __APPLE__
signal (SIGTRAP, SIG_IGN); //NEED BY STEP
#endif
#if __APPLE__ || __BSD__
/* we can probably remove this #if..as long as PT_TRACE_ME is redefined for OSX in r_debug.h */
signal (SIGABRT, inferior_abort_handler);
if (ptrace (PT_TRACE_ME, 0, 0, 0) != 0) {
#else
if (ptrace (PTRACE_TRACEME, 0, NULL, NULL) != 0) {
#endif
eprintf ("ptrace-traceme failed\n");
exit (MAGIC_EXIT);
}
// TODO: Add support to redirect filedescriptors
// TODO: Configure process environment
argv = r_str_argv (cmd, NULL);
#if __APPLE__
#include <spawn.h>
{
posix_spawnattr_t attr = {0};
size_t copied = 1;
cpu_type_t cpu;
int ret;
pid_t p = -1;
posix_spawnattr_init (&attr);
posix_spawnattr_setflags (&attr, POSIX_SPAWN_SETEXEC);
#if __i386__ || __x86_64__
cpu = CPU_TYPE_I386;
if (bits == 64)
cpu |= CPU_ARCH_ABI64;
#else
cpu = CPU_TYPE_ANY;
#endif
posix_spawnattr_setbinpref_np (&attr, 1, &cpu, &copied);
//ret = posix_spawnp (NULL, argv[0], NULL, &attr, argv, NULL);
ret = posix_spawnp (&p, argv[0], NULL, &attr, argv, NULL);
switch (ret) {
case 0:
eprintf ("Success\n");
break;
case 22:
eprintf ("posix_spawnp: Invalid argument\n");
break;
case 86:
eprintf ("Unsupported architecture\n");
break;
default:
eprintf ("posix_spawnp: unknown error %d\n", ret);
perror ("posix_spawnp");
break;
}
/* only required if no SETEXEC called
if (p != -1)
wait (p);
*/
exit (MAGIC_EXIT); /* error */
}
#else
execvp (argv[0], argv);
#endif
r_str_argv_free (argv);
perror ("fork_and_attach: execv");
//printf(stderr, "[%d] %s execv failed.\n", getpid(), ps.filename);
exit (MAGIC_EXIT); /* error */
return 0; // invalid pid // if exit is overriden.. :)
default:
/* XXX: clean this dirty code */
ret = wait (&status);
if (ret != pid)
eprintf ("Wait event received by different pid %d\n", ret);
if (WIFSTOPPED (status))
eprintf ("Process with PID %d started...\n", (int)pid);
if (WEXITSTATUS (status) == MAGIC_EXIT)
pid = -1;
// XXX kill (pid, SIGSTOP);
break;
}
eprintf ("PID = %d\n", pid);
return pid;
}
#endif
static int __plugin_open(RIO *io, const char *file) {
if (!memcmp (file, "dbg://", 6) && file[6])
return R_TRUE;
return R_FALSE;
}
static RIODesc *__open(RIO *io, const char *file, int rw, int mode) {
char uri[128];
if (__plugin_open (io, file)) {
int pid = atoi (file+6);
if (pid == 0) {
// TODO: get bits from ELF?
pid = fork_and_ptraceme (io->bits, file+6);
if (pid==-1)
return NULL;
#if __WINDOWS__
sprintf (uri, "w32dbg://%d", pid);
#elif __APPLE__
sprintf (uri, "mach://%d", pid);
#else
// TODO: use io_procpid here? faster or what?
sprintf (uri, "ptrace://%d", pid);
#endif
eprintf ("io_redirect: %s\n", uri);
r_io_redirect (io, uri);
return NULL;
} else {
sprintf (uri, "attach://%d", pid);
r_io_redirect (io, uri);
return NULL;
}
}
r_io_redirect (io, NULL);
return NULL;
}
