Author: phater
Date: Tue Oct 11 10:48:47 2016
New Revision: 72957
URL: http://svn.reactos.org/svn/reactos?rev=72957&view=rev
Log:
[WS2_32_APITESTS] Add super simple WSARecv tests used to test
WSAGetOverlappedResult. ROSTESTS-197
Added:
trunk/rostests/apitests/ws2_32/WSARecv.c (with props)
Modified:
trunk/rostests/apitests/ws2_32/CMakeLists.txt
trunk/rostests/apitests/ws2_32/testlist.c
Modified: trunk/rostests/apitests/ws2_32/CMakeLists.txt
URL:
http://svn.reactos.org/svn/reactos/trunk/rostests/apitests/ws2_32/CMakeLists.txt?rev=72957&r1=72956&r2=72957&view=diff
==============================================================================
--- trunk/rostests/apitests/ws2_32/CMakeLists.txt [iso-8859-1] (original)
+++ trunk/rostests/apitests/ws2_32/CMakeLists.txt [iso-8859-1] Tue Oct 11
10:48:47 2016
@@ -9,6 +9,7 @@
nostartup.c
recv.c
send.c
+ WSARecv.c
WSAStartup.c
testlist.c)
Added: trunk/rostests/apitests/ws2_32/WSARecv.c
URL:
http://svn.reactos.org/svn/reactos/trunk/rostests/apitests/ws2_32/WSARecv.c?rev=72957
==============================================================================
--- trunk/rostests/apitests/ws2_32/WSARecv.c (added)
+++ trunk/rostests/apitests/ws2_32/WSARecv.c [iso-8859-1] Tue Oct 11
10:48:47 2016
@@ -0,0 +1,201 @@
+/*
+ * PROJECT: ReactOS api tests
+ * LICENSE: GPL - See COPYING in the top level directory
+ * PURPOSE: Test for WSARecv
+ * PROGRAMMERS: Peter Hater
+ */
+
+#include <apitest.h>
+
+#include <stdio.h>
+#include <ntstatus.h>
+#include <wine/winternl.h>
+#include "ws2_32.h"
+
+#define RECV_BUF 4
+#define WSARecv_TIMEOUT 2000
+
+static int count = 0;
+
+void
+CALLBACK completion(
+ DWORD dwError,
+ DWORD cbTransferred,
+ LPWSAOVERLAPPED lpOverlapped,
+ DWORD dwFlags)
+{
+ //trace("completion called dwFlags %ld cbTransferred %ld lpOverlapped %p
dwFlags %ld\n", dwError, cbTransferred, lpOverlapped, dwFlags);
+ count++;
+ ok(count == 1, "completion sould be called only once count = %d\n", count);
+ ok(dwError == 0, "dwError = %ld\n", dwError);
+ ok(cbTransferred == RECV_BUF, "cbTransferred %ld != %d\n", cbTransferred,
RECV_BUF);
+ ok(lpOverlapped != NULL, "lpOverlapped %p\n", lpOverlapped);
+ if (lpOverlapped)
+ {
+ ok(lpOverlapped->hEvent != INVALID_HANDLE_VALUE, "lpOverlapped->hEvent
%d\n", (int)lpOverlapped->hEvent);
+ if (lpOverlapped->hEvent != INVALID_HANDLE_VALUE)
+ WSASetEvent(lpOverlapped->hEvent);
+ }
+}
+
+void Test_WSARecv()
+{
+ const char szDummyBytes[RECV_BUF] = { 0xFF, 0x00, 0xFF, 0x00 };
+
+ char szBuf[RECV_BUF];
+ char szRecvBuf[RECV_BUF];
+ int iResult, err;
+ SOCKET sck;
+ WSADATA wdata;
+ WSABUF buffers;
+ DWORD dwRecv, dwSent, dwFlags;
+ WSAOVERLAPPED overlapped;
+ char szGetRequest[] = "GET / HTTP/1.0\r\n\r\n";
+ struct fd_set readable;
+
+ /* Start up Winsock */
+ iResult = WSAStartup(MAKEWORD(2, 2), &wdata);
+ ok(iResult == 0, "WSAStartup failed, iResult == %d\n", iResult);
+
+ /* If we call recv without a socket, it should return with an error and do
nothing. */
+ memcpy(szBuf, szDummyBytes, RECV_BUF);
+ buffers.buf = szBuf;
+ buffers.len = sizeof(szBuf);
+ dwFlags = 0;
+ dwRecv = 0;
+ iResult = WSARecv(0, &buffers, 1, &dwRecv, &dwFlags, NULL, NULL);
+ ok(iResult == SOCKET_ERROR, "iRseult = %d\n", iResult);
+ ok(!memcmp(szBuf, szDummyBytes, RECV_BUF), "not equal\n");
+
+ /* Create the socket */
+ sck = WSASocketW(AF_INET, SOCK_STREAM, IPPROTO_TCP, NULL, 0,
WSA_FLAG_OVERLAPPED);
+ if(sck == INVALID_SOCKET)
+ {
+ WSACleanup();
+ skip("CreateSocket failed. Aborting test.\n");
+ return;
+ }
+
+ /* Now we can pass at least a socket, but we have no connection yet.
Should return with an error and do nothing. */
+ memcpy(szBuf, szDummyBytes, RECV_BUF);
+ buffers.buf = szBuf;
+ buffers.len = sizeof(szBuf);
+ dwFlags = 0;
+ dwRecv = 0;
+ iResult = WSARecv(sck, &buffers, 1, &dwRecv, &dwFlags, NULL, NULL);
+ ok(iResult == SOCKET_ERROR, "iResult = %d\n", iResult);
+ ok(!memcmp(szBuf, szDummyBytes, RECV_BUF), "not equal\n");
+
+ /* Connect to "www.reactos.org" */
+ if (!ConnectToReactOSWebsite(sck))
+ {
+ WSACleanup();
+ skip("ConnectToReactOSWebsite failed. Aborting test.\n");
+ return;
+ }
+
+ /* prepare overlapped */
+ memset(&overlapped, 0, sizeof(overlapped));
+ overlapped.hEvent = WSACreateEvent();
+
+ /* Send the GET request */
+ buffers.buf = szGetRequest;
+ buffers.len = strlen(szGetRequest);
+ dwSent = 0;
+ iResult = WSASend(sck, &buffers, 1, &dwSent, 0, &overlapped, NULL);
+ err = WSAGetLastError();
+ ok(iResult == 0 || (iResult == SOCKET_ERROR && err == WSA_IO_PENDING),
"iResult = %d, %d\n", iResult, err);
+ if (err == WSA_IO_PENDING)
+ {
+ iResult = WSAWaitForMultipleEvents(1, &overlapped.hEvent, TRUE,
WSARecv_TIMEOUT, TRUE);
+ ok(iResult == WSA_WAIT_EVENT_0, "WSAWaitForMultipleEvents failed
%d\n", iResult);
+ ok(WSAGetOverlappedResult(sck, &overlapped, &dwSent, TRUE, &dwFlags),
"WSAGetOverlappedResult failed %d\n", WSAGetLastError());
+ }
+ ok(dwSent == strlen(szGetRequest), "dwSent %ld != %d\n", dwSent,
strlen(szGetRequest));
+#if 0 /* break windows too */
+ /* Shutdown the SEND connection */
+ iResult = shutdown(sck, SD_SEND);
+ ok(iResult != SOCKET_ERROR, "iResult = %d\n", iResult);
+#endif
+ /* Wait until we're ready to read */
+ FD_ZERO(&readable);
+ FD_SET(sck, &readable);
+
+ iResult = select(0, &readable, NULL, NULL, NULL);
+ ok(iResult != SOCKET_ERROR, "iResult = %d\n", iResult);
+
+ /* Receive the data. */
+ buffers.buf = szBuf;
+ buffers.len = sizeof(szBuf);
+ dwRecv = sizeof(szBuf);
+ iResult = WSARecv(sck, &buffers, 1, &dwRecv, &dwFlags, NULL, NULL);
+ ok(iResult != SOCKET_ERROR, "iResult = %d\n", iResult);
+ ok(dwRecv == sizeof(szBuf), "dwRecv %ld != %d\n", dwRecv, sizeof(szBuf));
+ /* MSG_PEEK is invalid for overlapped (MSDN), but passes??? */
+ buffers.buf = szRecvBuf;
+ buffers.len = sizeof(szRecvBuf);
+ dwFlags = MSG_PEEK;
+ dwRecv = sizeof(szRecvBuf);
+ ok(overlapped.hEvent != NULL, "WSACreateEvent failed %d\n",
WSAGetLastError());
+ iResult = WSARecv(sck, &buffers, 1, &dwRecv, &dwFlags, &overlapped, NULL);
+ err = WSAGetLastError();
+ ok(iResult == 0 || (iResult == SOCKET_ERROR && err == WSA_IO_PENDING),
"iResult = %d, %d\n", iResult, err);
+ if (err == WSA_IO_PENDING)
+ {
+ iResult = WSAWaitForMultipleEvents(1, &overlapped.hEvent, TRUE,
WSARecv_TIMEOUT, TRUE);
+ ok(iResult == WSA_WAIT_EVENT_0, "WSAWaitForMultipleEvents failed
%d\n", iResult);
+ ok(WSAGetOverlappedResult(sck, &overlapped, &dwRecv, TRUE, &dwFlags),
"WSAGetOverlappedResult failed %d\n", WSAGetLastError());
+ }
+ ok(dwRecv == sizeof(szRecvBuf), "dwRecv %ld != %d\n", dwRecv,
sizeof(szRecvBuf));
+ /* normal overlapped, no completion */
+ buffers.buf = szBuf;
+ buffers.len = sizeof(szBuf);
+ dwFlags = 0;
+ dwRecv = sizeof(szBuf);
+ WSAResetEvent(overlapped.hEvent);
+ iResult = WSARecv(sck, &buffers, 1, &dwRecv, &dwFlags, &overlapped, NULL);
+ err = WSAGetLastError();
+ ok(iResult == 0 || (iResult == SOCKET_ERROR && err == WSA_IO_PENDING),
"iResult = %d, %d\n", iResult, err);
+ if (err == WSA_IO_PENDING)
+ {
+ iResult = WSAWaitForMultipleEvents(1, &overlapped.hEvent, TRUE,
WSARecv_TIMEOUT, TRUE);
+ ok(iResult == WSA_WAIT_EVENT_0, "WSAWaitForMultipleEvents failed
%d\n", iResult);
+ ok(WSAGetOverlappedResult(sck, &overlapped, &dwRecv, TRUE, &dwFlags),
"WSAGetOverlappedResult failed %d\n", WSAGetLastError());
+ }
+ ok(dwRecv == sizeof(szBuf), "dwRecv %ld != %d\n", dwRecv, sizeof(szBuf));
+ ok(memcmp(szRecvBuf, szBuf, sizeof(szBuf)) == 0, "MSG_PEEK shouldn't have
moved the pointer\n");
+ /* overlapped with completion */
+ dwFlags = 0;
+ dwRecv = sizeof(szBuf);
+ WSAResetEvent(overlapped.hEvent);
+ iResult = WSARecv(sck, &buffers, 1, &dwRecv, &dwFlags, &overlapped,
&completion);
+ err = WSAGetLastError();
+ ok(iResult == 0 || (iResult == SOCKET_ERROR && err == WSA_IO_PENDING),
"iResult = %d, %d\n", iResult, err);
+ if (err == WSA_IO_PENDING)
+ {
+ iResult = WSAWaitForMultipleEvents(1, &overlapped.hEvent, TRUE,
WSARecv_TIMEOUT, TRUE);
+ ok(iResult == WSA_WAIT_EVENT_0, "WSAWaitForMultipleEvents failed
%d\n", iResult);
+ ok(WSAGetOverlappedResult(sck, &overlapped, &dwRecv, TRUE, &dwFlags),
"WSAGetOverlappedResult failed %d\n", WSAGetLastError());
+ }
+ ok(WSACloseEvent(overlapped.hEvent), "WSAGetOverlappedResult failed %d\n",
WSAGetLastError());
+ ok(dwRecv == sizeof(szBuf), "dwRecv %ld != %d\n", dwRecv, sizeof(szBuf));
+ /* no overlapped with completion */
+ dwFlags = 0;
+ dwRecv = sizeof(szBuf);
+ /* call doesn't fail, but completion is not called */
+ iResult = WSARecv(sck, &buffers, 1, &dwRecv, &dwFlags, NULL, &completion);
+ err = WSAGetLastError();
+ ok(iResult == 0 || (iResult == SOCKET_ERROR && err == WSA_IO_PENDING),
"iResult = %d, %d\n", iResult, err);
+ ok(WSAGetLastError() == 0, "WSAGetLastError failed %d\n",
WSAGetLastError());
+ ok(dwRecv == sizeof(szBuf), "dwRecv %ld != %d and 0\n", dwRecv,
sizeof(szBuf));
+
+ closesocket(sck);
+ WSACleanup();
+ return;
+}
+
+START_TEST(WSARecv)
+{
+ Test_WSARecv();
+}
+
Propchange: trunk/rostests/apitests/ws2_32/WSARecv.c
------------------------------------------------------------------------------
svn:eol-style = native
Modified: trunk/rostests/apitests/ws2_32/testlist.c
URL:
http://svn.reactos.org/svn/reactos/trunk/rostests/apitests/ws2_32/testlist.c?rev=72957&r1=72956&r2=72957&view=diff
==============================================================================
--- trunk/rostests/apitests/ws2_32/testlist.c [iso-8859-1] (original)
+++ trunk/rostests/apitests/ws2_32/testlist.c [iso-8859-1] Tue Oct 11
10:48:47 2016
@@ -8,10 +8,11 @@
extern void func_getservbyname(void);
extern void func_getservbyport(void);
extern void func_ioctlsocket(void);
+extern void func_nostartup(void);
extern void func_recv(void);
extern void func_send(void);
+extern void func_WSARecv(void);
extern void func_WSAStartup(void);
-extern void func_nostartup(void);
const struct test winetest_testlist[] =
{
@@ -23,6 +24,7 @@
{ "nostartup", func_nostartup },
{ "recv", func_recv },
{ "send", func_send },
+ { "WSARecv", func_WSARecv },
{ "WSAStartup", func_WSAStartup },
{ 0, 0 }
};
