reactos/rosapps/tests/moztest/moztest.c

252 lines
6.3 KiB
C
Raw Normal View History

/*
* 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;
}