mirror of
https://github.com/radareorg/radare2.git
synced 2024-12-05 20:17:43 +00:00
118 lines
2.7 KiB
C
118 lines
2.7 KiB
C
// Copyright (c) 2014-2017, The Lemon Man, All rights reserved. LGPLv3
|
|
#include <fcntl.h>
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <r_types.h>
|
|
#include <r_util.h>
|
|
#include "transport.h"
|
|
|
|
#if __WINDOWS__ || __CYGWIN__ || MINGW32
|
|
#include <windows.h>
|
|
|
|
static void *iob_pipe_open(const char *path) {
|
|
HANDLE hPipe;
|
|
LPTSTR path_ = r_sys_conv_utf8_to_utf16 (path);
|
|
|
|
hPipe = CreateFile (path_, GENERIC_READ | GENERIC_WRITE, 0, NULL, OPEN_EXISTING, 0, NULL);
|
|
free (path_);
|
|
return hPipe != INVALID_HANDLE_VALUE? (void *)(HANDLE)hPipe : NULL;
|
|
}
|
|
|
|
static int iob_pipe_close(void *p) {
|
|
return CloseHandle (p);
|
|
}
|
|
|
|
static int iob_pipe_read(void *p, uint8_t *buf, const uint64_t count, const int timeout) {
|
|
DWORD c = 0;
|
|
if (!ReadFile (p, buf, count, &c, NULL)) {
|
|
return -1;
|
|
}
|
|
return c;
|
|
}
|
|
|
|
static int iob_pipe_write(void *p, const uint8_t *buf, const uint64_t count, const int timeout) {
|
|
DWORD cbWrited = 0;
|
|
if (!WriteFile (p, buf, count, &cbWrited, NULL)) {
|
|
return -1;
|
|
}
|
|
return cbWrited;
|
|
}
|
|
#else
|
|
#include <errno.h>
|
|
#include <sys/socket.h>
|
|
#include <sys/select.h>
|
|
#include <sys/un.h>
|
|
|
|
static void *iob_pipe_open(const char *path) {
|
|
int sock;
|
|
struct sockaddr_un sa;
|
|
|
|
sock = socket (AF_UNIX, SOCK_STREAM, 0);
|
|
if (sock == -1) {
|
|
perror ("socket");
|
|
return 0;
|
|
}
|
|
|
|
memset (&sa, 0, sizeof (struct sockaddr_un));
|
|
|
|
sa.sun_family = AF_UNIX;
|
|
strncpy (sa.sun_path, path, sizeof(sa.sun_path));
|
|
sa.sun_path[sizeof (sa.sun_path) - 1] = 0;
|
|
if (connect (sock, (struct sockaddr *) &sa, sizeof(struct sockaddr_un)) == -1) {
|
|
perror ("connect");
|
|
close (sock);
|
|
return 0;
|
|
}
|
|
//struct timeval tv;
|
|
//tv.tv_sec = 5;
|
|
//tv.tv_usec = 0;
|
|
//setsockopt (sock, SOL_SOCKET, SO_RCVTIMEO, &tv, sizeof(struct timeval));
|
|
return (void *) (size_t) sock;
|
|
}
|
|
|
|
static int iob_pipe_close(void *p) {
|
|
return close ((int) (size_t) p);
|
|
}
|
|
|
|
static int iob_pipe_read(void *p, uint8_t *buf, const uint64_t count, const int timeout) {
|
|
int result;
|
|
fd_set readset;
|
|
int fd = (int) (size_t) p;
|
|
for (;;) {
|
|
FD_ZERO (&readset);
|
|
FD_SET (fd, &readset);
|
|
result = select (fd + 1, &readset, NULL, NULL, NULL);
|
|
if (result < 1) {
|
|
if (errno == EINTR) continue;
|
|
return -1;
|
|
}
|
|
if (FD_ISSET (fd, &readset)) {
|
|
return recv ((int) (size_t) p, buf, count, 0);
|
|
}
|
|
}
|
|
return EINTR;
|
|
}
|
|
|
|
static int iob_pipe_write(void *p, const uint8_t *buf, const uint64_t count, const int timeout) {
|
|
int ret = send ((int) (size_t) p, buf, count, 0);
|
|
if (ret < 1) {
|
|
r_sys_perror ("iob_pipe_write, send");
|
|
if (errno == EPIPE) {
|
|
exit (1);
|
|
}
|
|
}
|
|
return ret;
|
|
}
|
|
#endif
|
|
|
|
io_backend_t iob_pipe = {
|
|
.name = "pipe",
|
|
.init = NULL,
|
|
.deinit = NULL,
|
|
.config = NULL,
|
|
.open = &iob_pipe_open,
|
|
.close = &iob_pipe_close,
|
|
.read = &iob_pipe_read,
|
|
.write = &iob_pipe_write,
|
|
};
|