mirror of
https://github.com/reactos/reactos.git
synced 2024-08-28 22:28:36 +00:00
252 lines
6.3 KiB
C
252 lines
6.3 KiB
C
/*
|
|
* Mozilla Test
|
|
* Copyright (C) 2004 Filip Navara
|
|
*/
|
|
|
|
#include <winsock.h>
|
|
#include <stdio.h>
|
|
|
|
ULONG DbgPrint(PCH Format,...);
|
|
|
|
#undef DBG
|
|
#define DBG(x) \
|
|
printf("%s:%i - %s", __FILE__, __LINE__, x); \
|
|
DbgPrint("%s:%i - %s", __FILE__, __LINE__, x);
|
|
|
|
int SocketTest()
|
|
{
|
|
/*
|
|
* A socket pair is often used for interprocess communication,
|
|
* so we need to make sure neither socket is associated with
|
|
* the I/O completion port; otherwise it can't be used by a
|
|
* child process.
|
|
*
|
|
* The default implementation below cannot be used for NT
|
|
* because PR_Accept would have associated the I/O completion
|
|
* port with the listening and accepted sockets.
|
|
*/
|
|
SOCKET listenSock;
|
|
SOCKET osfd[2];
|
|
struct sockaddr_in selfAddr, peerAddr;
|
|
int addrLen;
|
|
WORD wVersionRequested;
|
|
WSADATA wsaData;
|
|
int err;
|
|
|
|
/*
|
|
* Initialization.
|
|
*/
|
|
|
|
wVersionRequested = MAKEWORD( 2, 2 );
|
|
|
|
DBG("Calling WSAStartup\n");
|
|
err = WSAStartup( wVersionRequested, &wsaData );
|
|
if ( err != 0 ) {
|
|
/* Tell the user that we could not find a usable */
|
|
/* WinSock DLL. */
|
|
DBG("WSAStartup failed\n");
|
|
return 1;
|
|
}
|
|
|
|
/* Confirm that the WinSock DLL supports 2.2.*/
|
|
/* Note that if the DLL supports versions greater */
|
|
/* than 2.2 in addition to 2.2, it will still return */
|
|
/* 2.2 in wVersion since that is the version we */
|
|
/* requested. */
|
|
|
|
if ( LOBYTE( wsaData.wVersion ) != 2 ||
|
|
HIBYTE( wsaData.wVersion ) != 2 ) {
|
|
/* Tell the user that we could not find a usable */
|
|
/* WinSock DLL. */
|
|
DBG("WSAStartup version unacceptable\n");
|
|
WSACleanup( );
|
|
return 1;
|
|
}
|
|
|
|
/* The WinSock DLL is acceptable. Proceed. */
|
|
|
|
DBG("Calling socket\n");
|
|
osfd[0] = osfd[1] = INVALID_SOCKET;
|
|
listenSock = socket(AF_INET, SOCK_STREAM, 0);
|
|
if (listenSock == INVALID_SOCKET) {
|
|
DBG("socket failed\n");
|
|
goto failed;
|
|
}
|
|
|
|
selfAddr.sin_family = AF_INET;
|
|
selfAddr.sin_port = 0;
|
|
selfAddr.sin_addr.s_addr = htonl(INADDR_LOOPBACK); /* BugZilla: 35408 */
|
|
addrLen = sizeof(selfAddr);
|
|
DBG("Calling bind\n");
|
|
if (bind(listenSock, (struct sockaddr *) &selfAddr,
|
|
addrLen) == SOCKET_ERROR) {
|
|
DBG("bind failed\n");
|
|
goto failed;
|
|
}
|
|
|
|
DBG("Calling getsockname\n");
|
|
if (getsockname(listenSock, (struct sockaddr *) &selfAddr,
|
|
&addrLen) == SOCKET_ERROR) {
|
|
DBG("getsockname failed\n");
|
|
goto failed;
|
|
}
|
|
|
|
DBG("Calling listen\n");
|
|
if (listen(listenSock, 5) == SOCKET_ERROR) {
|
|
DBG("listen failed\n");
|
|
goto failed;
|
|
}
|
|
|
|
DBG("Calling socket\n");
|
|
osfd[0] = socket(AF_INET, SOCK_STREAM, 0);
|
|
if (osfd[0] == INVALID_SOCKET) {
|
|
DBG("socket failed\n");
|
|
goto failed;
|
|
}
|
|
selfAddr.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
|
|
|
|
/*
|
|
* Only a thread is used to do the connect and accept.
|
|
* I am relying on the fact that connect returns
|
|
* successfully as soon as the connect request is put
|
|
* into the listen queue (but before accept is called).
|
|
* This is the behavior of the BSD socket code. If
|
|
* connect does not return until accept is called, we
|
|
* will need to create another thread to call connect.
|
|
*/
|
|
DBG("Calling connect\n");
|
|
if (connect(osfd[0], (struct sockaddr *) &selfAddr,
|
|
addrLen) == SOCKET_ERROR) {
|
|
DBG("connect failed\n");
|
|
goto failed;
|
|
}
|
|
|
|
/*
|
|
* A malicious local process may connect to the listening
|
|
* socket, so we need to verify that the accepted connection
|
|
* is made from our own socket osfd[0].
|
|
*/
|
|
DBG("Calling getsockname\n");
|
|
if (getsockname(osfd[0], (struct sockaddr *) &selfAddr,
|
|
&addrLen) == SOCKET_ERROR) {
|
|
DBG("getsockname failed\n");
|
|
goto failed;
|
|
}
|
|
|
|
DBG("Calling accept\n");
|
|
osfd[1] = accept(listenSock, (struct sockaddr *) &peerAddr, &addrLen);
|
|
if (osfd[1] == INVALID_SOCKET) {
|
|
DBG("accept failed\n");
|
|
goto failed;
|
|
}
|
|
if (peerAddr.sin_port != selfAddr.sin_port) {
|
|
/* the connection we accepted is not from osfd[0] */
|
|
DBG("peerAddr.sin_port != selfAddr.sin_port\n");
|
|
goto failed;
|
|
}
|
|
|
|
DBG("Hurray!\n");
|
|
|
|
closesocket(listenSock);
|
|
|
|
closesocket(osfd[0]);
|
|
closesocket(osfd[1]);
|
|
|
|
WSACleanup();
|
|
|
|
return 0;
|
|
|
|
failed:
|
|
if (listenSock != INVALID_SOCKET) {
|
|
closesocket(listenSock);
|
|
}
|
|
if (osfd[0] != INVALID_SOCKET) {
|
|
closesocket(osfd[0]);
|
|
}
|
|
if (osfd[1] != INVALID_SOCKET) {
|
|
closesocket(osfd[1]);
|
|
}
|
|
|
|
WSACleanup();
|
|
|
|
return 1;
|
|
}
|
|
|
|
int VirtualTest()
|
|
{
|
|
DWORD dwErr;
|
|
SYSTEM_INFO si;
|
|
HANDLE hMap;
|
|
PBYTE pBufferStart;
|
|
PCHAR pszFileName = "test.txt";
|
|
ULONG dwMaxSize = strlen(pszFileName);
|
|
|
|
DBG("Calling CreateFileMappingA\n");
|
|
hMap = CreateFileMappingA(INVALID_HANDLE_VALUE, NULL,
|
|
PAGE_READWRITE | SEC_RESERVE, 0, dwMaxSize, pszFileName);
|
|
if (!hMap)
|
|
{
|
|
DBG("CreateFileMappingA failed\n");
|
|
return 1;
|
|
}
|
|
|
|
dwErr = GetLastError();
|
|
DBG("Calling MapViewOfFile\n");
|
|
pBufferStart = (BYTE *)MapViewOfFile(hMap, FILE_MAP_ALL_ACCESS, 0, 0, 0);
|
|
if (!pBufferStart)
|
|
{
|
|
DBG("MapViewOfFile failed\n");
|
|
return 1;
|
|
}
|
|
|
|
GetSystemInfo(&si);
|
|
|
|
if (dwErr == ERROR_ALREADY_EXISTS)
|
|
{
|
|
DBG("MapViewOfFile returned ERROR_ALREADY_EXISTS\n");
|
|
DBG("This really shouldn't happen, but it's not fatal.\n");
|
|
UnmapViewOfFile(pBufferStart);
|
|
CloseHandle(hMap);
|
|
return 1;
|
|
}
|
|
else
|
|
{
|
|
DBG("Calling VirtualAlloc\n");
|
|
if (!VirtualAlloc(pBufferStart, si.dwPageSize, MEM_COMMIT, PAGE_READWRITE))
|
|
{
|
|
DBG("VirtualAlloc failed\n");
|
|
UnmapViewOfFile(pBufferStart);
|
|
CloseHandle(hMap);
|
|
return 1;
|
|
}
|
|
}
|
|
|
|
DBG("Hurray!\n");
|
|
|
|
UnmapViewOfFile(pBufferStart);
|
|
CloseHandle(hMap);
|
|
|
|
return 0;
|
|
}
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
if (argc != 2)
|
|
{
|
|
printf("Usage: %s test_name\n\n", argv[0]);
|
|
printf("Valid test names:\n");
|
|
printf("\tsocket\n");
|
|
printf("\tvirtual\n");
|
|
return 0;
|
|
}
|
|
|
|
if (!_stricmp(argv[1], "socket"))
|
|
return SocketTest();
|
|
if (!_stricmp(argv[1], "virtual"))
|
|
return VirtualTest();
|
|
|
|
printf("Test '%s' doesn't exist\n", argv[1]);
|
|
|
|
return 0;
|
|
}
|