mirror of
https://github.com/reactos/reactos.git
synced 2025-08-05 16:52:59 +00:00
[WS2_32_APITESTS] Add super simple WSARecv tests used to test WSAGetOverlappedResult. ROSTESTS-197
svn path=/trunk/; revision=72957
This commit is contained in:
Peter Hater
parent
5bcbde18ec
commit
a0257c8b37
3 changed files with 205 additions and 1 deletions
|
|
@ -9,6 +9,7 @@ list(APPEND SOURCE
|
||||||
nostartup.c
|
nostartup.c
|
||||||
recv.c
|
recv.c
|
||||||
send.c
|
send.c
|
||||||
|
WSARecv.c
|
||||||
WSAStartup.c
|
WSAStartup.c
|
||||||
testlist.c)
|
testlist.c)
|
||||||
|
|
||||||
|
|
|
||||||
201
rostests/apitests/ws2_32/WSARecv.c
Normal file
201
rostests/apitests/ws2_32/WSARecv.c
Normal file
|
|
@ -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();
|
||||||
|
}
|
||||||
|
|
||||||
|
|
@ -8,10 +8,11 @@ extern void func_getnameinfo(void);
|
||||||
extern void func_getservbyname(void);
|
extern void func_getservbyname(void);
|
||||||
extern void func_getservbyport(void);
|
extern void func_getservbyport(void);
|
||||||
extern void func_ioctlsocket(void);
|
extern void func_ioctlsocket(void);
|
||||||
|
extern void func_nostartup(void);
|
||||||
extern void func_recv(void);
|
extern void func_recv(void);
|
||||||
extern void func_send(void);
|
extern void func_send(void);
|
||||||
|
extern void func_WSARecv(void);
|
||||||
extern void func_WSAStartup(void);
|
extern void func_WSAStartup(void);
|
||||||
extern void func_nostartup(void);
|
|
||||||
|
|
||||||
const struct test winetest_testlist[] =
|
const struct test winetest_testlist[] =
|
||||||
{
|
{
|
||||||
|
|
@ -23,6 +24,7 @@ const struct test winetest_testlist[] =
|
||||||
{ "nostartup", func_nostartup },
|
{ "nostartup", func_nostartup },
|
||||||
{ "recv", func_recv },
|
{ "recv", func_recv },
|
||||||
{ "send", func_send },
|
{ "send", func_send },
|
||||||
|
{ "WSARecv", func_WSARecv },
|
||||||
{ "WSAStartup", func_WSAStartup },
|
{ "WSAStartup", func_WSAStartup },
|
||||||
{ 0, 0 }
|
{ 0, 0 }
|
||||||
};
|
};
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue