mirror of
https://github.com/reactos/reactos.git
synced 2024-08-19 01:48:11 +00:00
202 lines
7.9 KiB
C
202 lines
7.9 KiB
C
/*
|
|
* 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();
|
|
}
|
|
|