TcpIp_user.c

Go to the documentation of this file.

/*
 * PROJECT:         ReactOS kernel-mode tests
 * LICENSE:         GPLv2+ - See COPYING in the top level directory
 * PURPOSE:         User mode part of the TcpIp.sys test suite
 * PROGRAMMER:      Jérôme Gardou <jerome.gardou@reactos.org>
 */
 
#include <kmt_test.h>
#include <winsock2.h>
 
#include "tcpip.h"
 
static
DWORD
LoadTcpIpTestDriver(void)
{
    DWORD Error;
 
    /* Start the special-purpose driver */
    Error = KmtLoadAndOpenDriver(L"TcpIp", FALSE);
    ok_eq_int(Error, ERROR_SUCCESS);
    if (Error)
        return Error;
 
    return ERROR_SUCCESS;
}
 
static
void
UnloadTcpIpTestDriver(void)
{
    /* Stop the driver. */
    KmtCloseDriver();
    KmtUnloadDriver();
}
 
START_TEST(TcpIpTdi)
{
    DWORD Error;
 
    Error = LoadTcpIpTestDriver();
    ok_eq_int(Error, 0);
    if (Error)
        return;
 
    Error = KmtSendToDriver(IOCTL_TEST_TDI);
    ok_eq_ulong(Error, ERROR_SUCCESS);
 
    UnloadTcpIpTestDriver();
}
 
static
DWORD
WINAPI
AcceptProc(
    _In_ LPVOID Parameter)
{
    WORD WinsockVersion;
    WSADATA WsaData;
    int Error;
    SOCKET ListenSocket, AcceptSocket;
    struct sockaddr_in ListenAddress, AcceptAddress;
    int AcceptAddressLength;
    HANDLE ReadyToConnectEvent = (HANDLE)Parameter;
 
    /* Initialize winsock */
    WinsockVersion = MAKEWORD(2, 0);
    Error = WSAStartup(WinsockVersion, &WsaData);
    ok_eq_int(Error, 0);
 
    ListenSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
    ok(ListenSocket != INVALID_SOCKET, "socket failed\n");
 
    ZeroMemory(&ListenAddress, sizeof(ListenAddress));
    ListenAddress.sin_addr.S_un.S_addr = inet_addr("127.0.0.1");
    ListenAddress.sin_port = htons(TEST_CONNECT_SERVER_PORT);
    ListenAddress.sin_family = AF_INET;
 
    Error = bind(ListenSocket, (struct sockaddr*)&ListenAddress, sizeof(ListenAddress));
    ok_eq_int(Error, 0);
 
    Error = listen(ListenSocket, 1);
    ok_eq_int(Error, 0);
 
    SetEvent(ReadyToConnectEvent);
 
    AcceptAddressLength = sizeof(AcceptAddress);
    AcceptSocket = accept(ListenSocket, (struct sockaddr*)&AcceptAddress, &AcceptAddressLength);
    ok(AcceptSocket != INVALID_SOCKET, "accept failed\n");
    ok_eq_long(AcceptAddressLength, sizeof(AcceptAddress));
    ok_eq_hex(AcceptAddress.sin_addr.S_un.S_addr, inet_addr("127.0.0.1"));
    ok_eq_hex(AcceptAddress.sin_port, htons(TEST_CONNECT_CLIENT_PORT));
 
    return 0;
}
 
START_TEST(TcpIpConnect)
{
    HANDLE AcceptThread;
    HANDLE ReadyToConnectEvent;
    DWORD Error;
 
    ReadyToConnectEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
    ok(ReadyToConnectEvent != NULL, "CreateEvent failed\n");
 
    AcceptThread = CreateThread(NULL, 0, AcceptProc, (PVOID)ReadyToConnectEvent, 0, NULL);
    ok(AcceptThread != NULL, "CreateThread failed\n");
 
    WaitForSingleObject(ReadyToConnectEvent, INFINITE);
 
    LoadTcpIpTestDriver();
 
    Error = KmtSendToDriver(IOCTL_TEST_CONNECT);
    ok_eq_ulong(Error, ERROR_SUCCESS);
 
    WaitForSingleObject(AcceptThread, INFINITE);
 
    UnloadTcpIpTestDriver();
 
    WSACleanup();
}