const char *
vp_pipe_read(char *args)
{
vp_stack_t stack;
int fd;
int nr;
int timeout;
DWORD n;
char buf[VP_READ_BUFSIZE];
VP_RETURN_IF_FAIL(vp_stack_from_args(&stack, args));
VP_RETURN_IF_FAIL(vp_stack_pop_num(&stack, "%d", &fd));
VP_RETURN_IF_FAIL(vp_stack_pop_num(&stack, "%d", &nr));
VP_RETURN_IF_FAIL(vp_stack_pop_num(&stack, "%d", &timeout));
vp_stack_push_str(&_result, ""); /* initialize */
while (nr != 0) {
n = 0;
if (!PeekNamedPipe((HANDLE)_get_osfhandle(fd), buf,
(nr < 0) ? VP_READ_BUFSIZE : (VP_READ_BUFSIZE < nr) ? VP_READ_BUFSIZE : nr,
&n, NULL, NULL))
{
/* can be ERROR_HANDLE_EOF? */
if (GetLastError() == 0 || GetLastError() == ERROR_BROKEN_PIPE) {
/* error or eof */
if (n != 0) {
/* decrease stack top for concatenate. */
_result.top--;
vp_stack_push_bin(&_result, buf, n);
}
vp_stack_push_num(&_result, "%d", 1);
return vp_stack_return(&_result);
}
return vp_stack_return_error(&_result, "PeekNamedPipe() error: %08X %s",
GetLastError(), lasterror());
}
if (n == 0) {
break;
}
if (read(fd, buf, n) == -1)
return vp_stack_return_error(&_result, "read() error: %s",
strerror(errno));
/* decrease stack top for concatenate. */
_result.top--;
vp_stack_push_bin(&_result, buf, n);
if (nr > 0)
nr -= n;
/* try read more bytes without waiting */
timeout = 0;
}
vp_stack_push_num(&_result, "%d", 0);
return vp_stack_return(&_result);
}
const char *
vp_socket_read(char *args)
{
vp_stack_t stack;
int sock;
int nr;
int timeout;
struct timeval tv;
int n;
char buf[VP_READ_BUFSIZE];
fd_set fdset;
VP_RETURN_IF_FAIL(vp_stack_from_args(&stack, args));
VP_RETURN_IF_FAIL(vp_stack_pop_num(&stack, "%d", &sock));
VP_RETURN_IF_FAIL(vp_stack_pop_num(&stack, "%d", &nr));
VP_RETURN_IF_FAIL(vp_stack_pop_num(&stack, "%d", &timeout));
tv.tv_sec = timeout / 1000;
tv.tv_usec = (timeout - tv.tv_sec * 1000) * 1000;
FD_ZERO(&fdset);
FD_SET((unsigned)sock, &fdset);
vp_stack_push_str(&_result, ""); /* initialize */
while (nr != 0) {
n = select(0, &fdset, NULL, NULL, (timeout == -1) ? NULL : &tv);
if (n == SOCKET_ERROR) {
return vp_stack_return_error(&_result, "select() error: %d",
WSAGetLastError());
} else if (n == 0) {
/* timeout */
break;
}
if (nr > 0)
n = recv(sock, buf,
(VP_READ_BUFSIZE < nr) ? VP_READ_BUFSIZE : nr, 0);
else
n = recv(sock, buf, VP_READ_BUFSIZE, 0);
if (n == -1) {
return vp_stack_return_error(&_result, "recv() error: %s",
strerror(errno));
} else if (n == 0) {
/* eof */
vp_stack_push_num(&_result, "%d", 1);
return vp_stack_return(&_result);
}
/* decrease stack top for concatenate. */
_result.top--;
vp_stack_push_bin(&_result, buf, n);
if (nr > 0)
nr -= n;
/* try read more bytes without waiting */
timeout = 0;
}
vp_stack_push_num(&_result, "%d", 0);
return vp_stack_return(&_result);
}
const char *
vp_file_read(char *args)
{
vp_stack_t stack;
int fd;
int nr;
int timeout;
DWORD ret;
int n;
char buf[VP_READ_BUFSIZE];
VP_RETURN_IF_FAIL(vp_stack_from_args(&stack, args));
VP_RETURN_IF_FAIL(vp_stack_pop_num(&stack, "%d", &fd));
VP_RETURN_IF_FAIL(vp_stack_pop_num(&stack, "%d", &nr));
VP_RETURN_IF_FAIL(vp_stack_pop_num(&stack, "%d", &timeout));
vp_stack_push_str(&_result, ""); /* initialize */
while (nr != 0) {
ret = WaitForSingleObject((HANDLE)_get_osfhandle(fd), timeout);
if (ret == WAIT_FAILED) {
return vp_stack_return_error(&_result, "WaitForSingleObject() error: %s",
lasterror());
} else if (ret == WAIT_TIMEOUT) {
/* timeout */
break;
}
if (nr > 0)
n = read(fd, buf, (VP_READ_BUFSIZE < nr) ? VP_READ_BUFSIZE : nr);
else
n = read(fd, buf, VP_READ_BUFSIZE);
if (n == -1) {
return vp_stack_return_error(&_result, "read() error: %s",
strerror(errno));
} else if (n == 0) {
/* eof */
vp_stack_push_num(&_result, "%d", 1);
return vp_stack_return(&_result);
}
/* decrease stack top for concatenate. */
_result.top--;
vp_stack_push_bin(&_result, buf, n);
if (nr > 0)
nr -= n;
/* try read more bytes without waiting */
timeout = 0;
}
vp_stack_push_num(&_result, "%d", 0);
return vp_stack_return(&_result);
}
const char *
vp_file_read(char *args)
{
vp_stack_t stack;
int fd;
int nr;
int timeout;
int n;
char buf[VP_READ_BUFSIZE];
struct pollfd pfd = {0, POLLIN, 0};
VP_RETURN_IF_FAIL(vp_stack_from_args(&stack, args));
VP_RETURN_IF_FAIL(vp_stack_pop_num(&stack, "%d", &fd));
VP_RETURN_IF_FAIL(vp_stack_pop_num(&stack, "%d", &nr));
VP_RETURN_IF_FAIL(vp_stack_pop_num(&stack, "%d", &timeout));
pfd.fd = fd;
vp_stack_push_str(&_result, ""); /* initialize */
while (nr != 0) {
n = poll(&pfd, 1, timeout);
if (n == -1) {
/* eof or error */
vp_stack_push_num(&_result, "%d", 1);
return vp_stack_return(&_result);
} else if (n == 0) {
/* timeout */
break;
}
if (pfd.revents & POLLIN) {
if (nr > 0)
n = read(fd, buf,
(VP_READ_BUFSIZE < nr) ? VP_READ_BUFSIZE : nr);
else
n = read(fd, buf, VP_READ_BUFSIZE);
if (n == -1) {
return vp_stack_return_error(&_result, "read() error: %s",
strerror(errno));
} else if (n == 0) {
/* eof */
vp_stack_push_num(&_result, "%d", 1);
return vp_stack_return(&_result);
}
/* decrease stack top for concatenate. */
_result.top--;
vp_stack_push_bin(&_result, buf, n);
if (nr > 0)
nr -= n;
/* try read more bytes without waiting */
timeout = 0;
continue;
} else if (pfd.revents & (POLLERR | POLLHUP)) {
/* eof or error */
vp_stack_push_num(&_result, "%d", 1);
return vp_stack_return(&_result);
} else if (pfd.revents & POLLNVAL) {
return vp_stack_return_error(&_result, "poll() POLLNVAL: %d",
pfd.revents);
}
/* DO NOT REACH HERE */
return vp_stack_return_error(&_result, "poll() unknown status: %d",
pfd.revents);
}
vp_stack_push_num(&_result, "%d", 0);
return vp_stack_return(&_result);
}
