tyler.durden
/
Win32-OpenSSH
mirror of https://github.com/PowerShell/Win32-OpenSSH.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

2-29 5

This commit is contained in:
manojampalam 2016-02-29 20:59:25 -08:00
parent afbfa98759
commit a4163ba0a3
2 changed files with 23 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
contrib/win32/w32-posix-prototype/win32posix/UnitTests/socket_tests.c
View File

@ -216,6 +216,7 @@ void socket_syncio_tests()
TEST_START("basic recv s->c");
ret = recv(connect_fd, small_recv_buf, SMALL_RECV_BUF_SIZE, 0);
ASSERT_INT_EQ(ret, strlen(small_send_buf));
small_recv_buf[ret] = '\0';
ASSERT_STRING_EQ(small_send_buf, small_recv_buf);
memset(small_recv_buf, 0, sizeof(small_recv_buf));
TEST_DONE();
@ -225,6 +226,7 @@ void socket_syncio_tests()
ASSERT_INT_EQ(ret, strlen(small_send_buf));
ret = recv(accept_fd, small_recv_buf, SMALL_RECV_BUF_SIZE, 0);
ASSERT_INT_EQ(ret, strlen(small_send_buf));
small_recv_buf[ret] = '\0';
ASSERT_STRING_EQ(small_send_buf, small_recv_buf);
memset(small_recv_buf, 0, sizeof(small_recv_buf));
TEST_DONE();

37
contrib/win32/w32-posix-prototype/win32posix/win32posix/socketio.c
View File

@ -13,23 +13,18 @@
static int errno_from_WSAError(int wsaerrno)
{
if (wsaerrno == WSAEWOULDBLOCK)
{
switch (wsaerrno) {
case WSAEWOULDBLOCK:
return EAGAIN;
}
if (wsaerrno == WSAEFAULT)
{
case WSAEFAULT:
return EFAULT;
}
if (wsaerrno == WSAEINVAL)
{
case WSAEINVAL:
return EINVAL;
case WSAESHUTDOWN:
return ECONNRESET;
default:
return wsaerrno;
}
return wsaerrno;
}
int socketio_initialize() {
@ -271,9 +266,19 @@ int socketio_recv(struct w32_io* pio, void *buf, size_t len, int flags) {
//if io is already pending
if (pio->read_details.pending) {
errno = EAGAIN;
debug2("Read is already pending, io:%p", pio);
return -1;
//if recv is now in blocking mode, wait for data to be available
if (w32_io_is_blocking(pio)) {
debug2("socket was previously non-blocing but now in blocking mode, waiting for io");
while (socketio_is_io_available(pio, TRUE) == FALSE) {
if (0 != wait_for_any_event(NULL, 0, INFINITE))
return -1;
}
}
else {
errno = EAGAIN;
debug2("Read is already pending, io:%p", pio);
return -1;
}
}
//if we have some buffer copy it and return #bytes copied