reactos/modules/rostests/apitests/ws2_32/ioctlsocket.c

96 lines
3 KiB
C
Raw Normal View History

/*
* PROJECT: ReactOS api tests
* LICENSE: GPL - See COPYING in the top level directory
* PURPOSE: Test for ioctlsocket
* PROGRAMMERS: Colin Finck
*/
#include "ws2_32.h"
int Test_ioctlsocket()
{
LPSTR pszBuf;
int iResult;
SOCKET sck;
ULONG BytesAvailable;
ULONG BytesToRead;
WSADATA wdata;
/* Start up Winsock */
iResult = WSAStartup(MAKEWORD(2, 2), &wdata);
ok(iResult == 0, "WSAStartup failed. iResult = %d\n", iResult);
/* If we call ioctlsocket without a socket, it should return with an error and do nothing. */
BytesAvailable = 0xdeadbeef;
iResult = ioctlsocket(0, FIONREAD, &BytesAvailable);
ok(iResult == SOCKET_ERROR, "iResult = %d\n", iResult);
ok(BytesAvailable == 0xdeadbeef, "BytesAvailable = %ld\n", BytesAvailable);
/* Create the socket */
if (!CreateSocket(&sck))
{
ok(0, "CreateSocket failed. Aborting test.\n");
return 0;
}
/* Now we can pass at least a socket, but we have no connection yet. The function should return 0. */
BytesAvailable = 0xdeadbeef;
iResult = ioctlsocket(sck, FIONREAD, &BytesAvailable);
ok(iResult == 0, "iResult = %d\n", iResult);
ok(BytesAvailable == 0, "BytesAvailable = %ld\n", BytesAvailable);
/* Connect to "www.reactos.org" */
if (!ConnectToReactOSWebsite(sck))
{
ok(0, "ConnectToReactOSWebsite failed. Aborting test.\n");
return 0;
}
/* Even with a connection, there shouldn't be any bytes available. */
iResult = ioctlsocket(sck, FIONREAD, &BytesAvailable);
ok(iResult == 0, "iResult = %d\n", iResult);
ok(BytesAvailable == 0, "BytesAvailable = %ld\n", BytesAvailable);
/* Send the GET request */
if (!GetRequestAndWait(sck))
{
ok(0, "GetRequestAndWait failed. Aborting test.\n");
return 0;
}
/* Try ioctlsocket with FIONREAD. There should be bytes available now. */
SCKTEST(ioctlsocket(sck, FIONREAD, &BytesAvailable));
ok(BytesAvailable != 0, "BytesAvailable = %ld\n", BytesAvailable);
/* Get half of the data */
BytesToRead = BytesAvailable / 2;
pszBuf = (LPSTR) HeapAlloc(GetProcessHeap(), 0, BytesToRead);
SCKTEST(recv(sck, pszBuf, BytesToRead, 0));
HeapFree(GetProcessHeap(), 0, pszBuf);
BytesToRead = BytesAvailable - BytesToRead;
/* Now try ioctlsocket again. BytesAvailable should be at the value saved in BytesToRead now. */
SCKTEST(ioctlsocket(sck, FIONREAD, &BytesAvailable));
ok(BytesAvailable == BytesToRead, "BytesAvailable = %ld\n", BytesAvailable);
/* Read those bytes */
pszBuf = (LPSTR) HeapAlloc(GetProcessHeap(), 0, BytesToRead);
SCKTEST(recv(sck, pszBuf, BytesToRead, 0));
HeapFree(GetProcessHeap(), 0, pszBuf);
/* Try it for the last time. BytesAvailable should be at 0 now. */
SCKTEST(ioctlsocket(sck, FIONREAD, &BytesAvailable));
ok(BytesAvailable == 0, "BytesAvailable = %ld\n", BytesAvailable);
closesocket(sck);
WSACleanup();
return 1;
}
START_TEST(ioctlsocket)
{
Test_ioctlsocket();
}