mirror of
https://github.com/radareorg/radare2.git
synced 2024-12-09 14:04:41 +00:00
327 lines
6.8 KiB
C
327 lines
6.8 KiB
C
/* radare - LGPL - Copyright 2015-2016 - pancake */
|
|
/*
|
|
Usage Example:
|
|
|
|
#include <r_core.h>
|
|
int main() {
|
|
RCoreBind rcb;
|
|
RCore *core = r_core_new ();
|
|
r_core_bind (core, &rcb);
|
|
r2pipe_open_corebind (&rcb);
|
|
char *clippy = r2pipe_cmd ("?E hello");
|
|
eprintf ("%s\n", clippy);
|
|
free (clippy);
|
|
r2pipe_close (r2pipe);
|
|
r_core_free (core);
|
|
}
|
|
*/
|
|
|
|
#include <r_util.h>
|
|
#include <r_cons.h>
|
|
#include <r_socket.h>
|
|
|
|
#define R2P_PID(x) (((R2Pipe*)(x)->data)->pid)
|
|
#define R2P_INPUT(x) (((R2Pipe*)(x)->data)->input[0])
|
|
#define R2P_OUTPUT(x) (((R2Pipe*)(x)->data)->output[1])
|
|
|
|
#if !__WINDOWS__
|
|
static void env(const char *s, int f) {
|
|
char *a = r_str_newf ("%d", f);
|
|
r_sys_setenv (s, a);
|
|
free (a);
|
|
}
|
|
#endif
|
|
|
|
R_API int r2pipe_write(R2Pipe *r2pipe, const char *str) {
|
|
char *cmd;
|
|
int ret, len;
|
|
if (!r2pipe || !str) {
|
|
return -1;
|
|
}
|
|
len = strlen (str) + 2; /* include \n\x00 */
|
|
cmd = malloc (len + 2);
|
|
if (!cmd) {
|
|
return 0;
|
|
}
|
|
memcpy (cmd, str, len - 1);
|
|
strcpy (cmd + len - 2, "\n");
|
|
#if __WINDOWS__ && !defined(__CYGWIN__)
|
|
DWORD dwWritten = -1;
|
|
WriteFile (r2pipe->pipe, cmd, len, &dwWritten, NULL);
|
|
ret = (dwWritten == len);
|
|
#else
|
|
ret = (write (r2pipe->input[1], cmd, len) == len);
|
|
#endif
|
|
free (cmd);
|
|
return ret;
|
|
}
|
|
|
|
/* TODO: add timeout here ? */
|
|
R_API char *r2pipe_read(R2Pipe *r2pipe) {
|
|
int bufsz = 0;
|
|
char *buf = NULL;
|
|
if (!r2pipe) {
|
|
return NULL;
|
|
}
|
|
bufsz = 4096;
|
|
buf = calloc (1, bufsz);
|
|
if (!buf) {
|
|
return NULL;
|
|
}
|
|
#if __WINDOWS__ && !defined(__CYGWIN__)
|
|
BOOL bSuccess = FALSE;
|
|
DWORD dwRead = 0;
|
|
// TODO: handle > 4096 buffers here
|
|
bSuccess = ReadFile (r2pipe->pipe, buf, bufsz, &dwRead, NULL);
|
|
if (!bSuccess || !buf[0]) {
|
|
return NULL;
|
|
}
|
|
if (dwRead > 0) {
|
|
buf[dwRead] = 0;
|
|
}
|
|
buf[bufsz - 1] = 0;
|
|
#else
|
|
char *newbuf;
|
|
int i, rv;
|
|
for (i = 0; i < bufsz; i++) {
|
|
rv = read (r2pipe->output[0], buf + i, 1);
|
|
if (i + 2 >= bufsz) {
|
|
bufsz += 4096;
|
|
newbuf = realloc (buf, bufsz);
|
|
if (!newbuf) {
|
|
R_FREE (buf);
|
|
break;
|
|
}
|
|
buf = newbuf;
|
|
}
|
|
if (rv != 1 || !buf[i]) {
|
|
break;
|
|
}
|
|
}
|
|
if (buf) {
|
|
int zpos = (i < bufsz)? i: i - 1;
|
|
buf[zpos] = 0;
|
|
}
|
|
#endif
|
|
return buf;
|
|
}
|
|
|
|
R_API int r2pipe_close(R2Pipe *r2pipe) {
|
|
if (!r2pipe) {
|
|
return 0;
|
|
}
|
|
#if __WINDOWS__ && !defined(__CYGWIN__)
|
|
if (r2pipe->pipe) {
|
|
CloseHandle (r2pipe->pipe);
|
|
r2pipe->pipe = NULL;
|
|
}
|
|
#else
|
|
if (r2pipe->input[0] != -1) {
|
|
close (r2pipe->input[0]);
|
|
r2pipe->input[0] = -1;
|
|
}
|
|
if (r2pipe->input[1] != -1) {
|
|
close (r2pipe->input[1]);
|
|
r2pipe->input[1] = -1;
|
|
}
|
|
if (r2pipe->output[0] != -1) {
|
|
close (r2pipe->output[0]);
|
|
r2pipe->output[0] = -1;
|
|
}
|
|
if (r2pipe->output[1] != -1) {
|
|
close (r2pipe->output[1]);
|
|
r2pipe->output[1] = -1;
|
|
}
|
|
if (r2pipe->child != -1) {
|
|
kill (r2pipe->child, SIGTERM);
|
|
waitpid (r2pipe->child, NULL, 0);
|
|
r2pipe->child = -1;
|
|
}
|
|
#endif
|
|
free (r2pipe);
|
|
return 0;
|
|
}
|
|
|
|
#if __WINDOWS__ && !defined(__CYGWIN__)
|
|
static int w32_createPipe(R2Pipe *r2pipe, const char *cmd) {
|
|
CHAR buf[1024];
|
|
r2pipe->pipe = CreateNamedPipe (TEXT ("\\\\.\\pipe\\R2PIPE_IN"),
|
|
PIPE_ACCESS_DUPLEX,PIPE_TYPE_MESSAGE | \
|
|
PIPE_READMODE_MESSAGE | \
|
|
PIPE_WAIT, PIPE_UNLIMITED_INSTANCES,
|
|
sizeof (buf), sizeof (buf), 0, NULL);
|
|
if (r_sys_create_child_proc_w32 (cmd, NULL, NULL, NULL)) {
|
|
if (ConnectNamedPipe (r2pipe->pipe, NULL)) {
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
#endif
|
|
|
|
static R2Pipe* r2pipe_open_spawn(R2Pipe* r2pipe) {
|
|
#if __UNIX__ || defined(__CYGWIN__)
|
|
char *out = r_sys_getenv ("R2PIPE_IN");
|
|
char *in = r_sys_getenv ("R2PIPE_OUT");
|
|
int done = false;
|
|
if (in && out) {
|
|
int i_in = atoi (in);
|
|
int i_out = atoi (out);
|
|
if (i_in >= 0 && i_out >= 0) {
|
|
r2pipe->input[0] = r2pipe->input[1] = i_in;
|
|
r2pipe->output[0] = r2pipe->output[1] = i_out;
|
|
done = true;
|
|
}
|
|
}
|
|
if (!done) {
|
|
eprintf ("Cannot find R2PIPE_IN or R2PIPE_OUT environment\n");
|
|
R_FREE (r2pipe);
|
|
}
|
|
free (in);
|
|
free (out);
|
|
return r2pipe;
|
|
#else
|
|
eprintf ("r2pipe_open(NULL) not supported on windows\n");
|
|
return NULL;
|
|
#endif
|
|
}
|
|
|
|
static R2Pipe *r2pipe_new() {
|
|
R2Pipe *r2pipe = R_NEW0 (R2Pipe);
|
|
if (r2pipe) {
|
|
#if __UNIX__ || defined(__CYGWIN__)
|
|
r2pipe->input[0] = r2pipe->input[1] = -1;
|
|
r2pipe->output[0] = r2pipe->output[1] = -1;
|
|
#endif
|
|
r2pipe->child = -1;
|
|
}
|
|
return r2pipe;
|
|
}
|
|
|
|
R_API R2Pipe *r2pipe_open_corebind(RCoreBind *coreb) {
|
|
R2Pipe *r2pipe = r2pipe_new ();
|
|
if (r2pipe) {
|
|
memcpy (&r2pipe->coreb, coreb, sizeof (RCoreBind));
|
|
}
|
|
return r2pipe;
|
|
}
|
|
|
|
R_API R2Pipe *r2pipe_open(const char *cmd) {
|
|
R2Pipe *r2pipe = r2pipe_new ();
|
|
if (!r2pipe) {
|
|
return NULL;
|
|
}
|
|
if (!cmd) {
|
|
r2pipe->child = -1;
|
|
return r2pipe_open_spawn (r2pipe);
|
|
}
|
|
#if __WINDOWS__ && !defined(__CYGWIN__)
|
|
w32_createPipe (r2pipe, cmd);
|
|
r2pipe->child = (int)(r2pipe->pipe);
|
|
#else
|
|
int r = pipe (r2pipe->input);
|
|
if (r != 0) {
|
|
eprintf ("pipe failed on input\n");
|
|
r2pipe_close (r2pipe);
|
|
return NULL;
|
|
}
|
|
r = pipe (r2pipe->output);
|
|
if (r != 0) {
|
|
eprintf ("pipe failed on output\n");
|
|
r2pipe_close (r2pipe);
|
|
return NULL;
|
|
}
|
|
#if LIBC_HAVE_FORK
|
|
r2pipe->child = fork ();
|
|
#else
|
|
r2pipe->child = -1;
|
|
#endif
|
|
if (r2pipe->child == -1) {
|
|
r2pipe_close (r2pipe);
|
|
return NULL;
|
|
}
|
|
env ("R2PIPE_IN", r2pipe->input[0]);
|
|
env ("R2PIPE_OUT", r2pipe->output[1]);
|
|
|
|
if (r2pipe->child) {
|
|
char ch = 1;
|
|
// eprintf ("[+] r2pipeipe child is %d\n", r2pipe->child);
|
|
if (read (r2pipe->output[0], &ch, 1) != 1) {
|
|
eprintf ("Failed to read 1 byte\n");
|
|
r2pipe_close (r2pipe);
|
|
return NULL;
|
|
}
|
|
if (ch) {
|
|
eprintf ("[+] r2pipeipe-io link failed. Expected two null bytes.\n");
|
|
r2pipe_close (r2pipe);
|
|
return NULL;
|
|
}
|
|
// Close parent's end of pipes
|
|
close (r2pipe->input[0]);
|
|
close (r2pipe->output[1]);
|
|
r2pipe->input[0] = -1;
|
|
r2pipe->output[1] = -1;
|
|
} else {
|
|
int rc = 0;
|
|
if (cmd && *cmd) {
|
|
close (0);
|
|
close (1);
|
|
dup2 (r2pipe->input[0], 0);
|
|
dup2 (r2pipe->output[1], 1);
|
|
close (r2pipe->input[1]);
|
|
close (r2pipe->output[0]);
|
|
r2pipe->input[1] = -1;
|
|
r2pipe->output[0] = -1;
|
|
rc = r_sandbox_system (cmd, 0);
|
|
}
|
|
r2pipe_close (r2pipe);
|
|
exit (rc);
|
|
return NULL;
|
|
}
|
|
#endif
|
|
return r2pipe;
|
|
}
|
|
|
|
R_API char *r2pipe_cmd(R2Pipe *r2pipe, const char *str) {
|
|
if (r2pipe->coreb.core) {
|
|
return r2pipe->coreb.cmdstr (r2pipe->coreb.core, str);
|
|
}
|
|
if (!r2pipe_write (r2pipe, str)) {
|
|
perror ("r2pipe_write");
|
|
return NULL;
|
|
}
|
|
return r2pipe_read (r2pipe);
|
|
}
|
|
|
|
R_API char *r2pipe_cmdf(R2Pipe *r2pipe, const char *fmt, ...) {
|
|
int ret, ret2;
|
|
char *p, string[1024];
|
|
va_list ap, ap2;
|
|
va_start (ap, fmt);
|
|
va_start (ap2, fmt);
|
|
ret = vsnprintf (string, sizeof (string) - 1, fmt, ap);
|
|
if (ret < 1 || ret >= sizeof (string)) {
|
|
p = malloc (ret + 2);
|
|
if (!p) {
|
|
va_end (ap2);
|
|
va_end (ap);
|
|
return NULL;
|
|
}
|
|
ret2 = vsnprintf (p, ret + 1, fmt, ap2);
|
|
if (ret2 < 1 || ret2 > ret + 1) {
|
|
free (p);
|
|
va_end (ap2);
|
|
va_end (ap);
|
|
return NULL;
|
|
}
|
|
fmt = r2pipe_cmd (r2pipe, p);
|
|
free (p);
|
|
} else {
|
|
fmt = r2pipe_cmd (r2pipe, string);
|
|
}
|
|
va_end (ap2);
|
|
va_end (ap);
|
|
return (char*)fmt;
|
|
}
|
|
|