mirror of
https://github.com/reactos/reactos.git
synced 2024-06-28 17:01:28 +00:00
96 lines
3 KiB
C
96 lines
3 KiB
C
/*
|
|
* 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();
|
|
}
|
|
|