TcpIp_user.c File Reference

#include <kmt_test.h>
#include <winsock2.h>
#include "tcpip.h"

Include dependency graph for TcpIp_user.c:

Go to the source code of this file.

Functions
static DWORD	LoadTcpIpTestDriver (void)
static void	UnloadTcpIpTestDriver (void)
	START_TEST (TcpIpTdi)
static DWORD WINAPI	AcceptProc (_In_ LPVOID Parameter)
	START_TEST (TcpIpConnect)

Function Documentation

AcceptProc()

static DWORD WINAPI AcceptProc ( _In_ LPVOID  Parameter )

static

Definition at line 55 of file TcpIp_user.c.

{
    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;
}

Referenced by START_TEST().

LoadTcpIpTestDriver()

static DWORD LoadTcpIpTestDriver ( void  )

static

Definition at line 15 of file TcpIp_user.c.

{
    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;
}

Referenced by START_TEST().

START_TEST() [1/2]

START_TEST

(

TcpIpConnect

)

Definition at line 97 of file TcpIp_user.c.

{
    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);
 
    Error = WaitForSingleObject(AcceptThread, 10 * 1000);
    ok(Error == WAIT_OBJECT_0, "AcceptThread timed out\n");
 
    UnloadTcpIpTestDriver();
 
    WSACleanup();
}

START_TEST() [2/2]

START_TEST

(

TcpIpTdi

)

Definition at line 37 of file TcpIp_user.c.

{
    DWORD Error;
 
    Error = LoadTcpIpTestDriver();
    ok_eq_int(Error, 0);
    if (Error)
        return;
 
    Error = KmtSendToDriver(IOCTL_TEST_TDI);
    ok_eq_ulong(Error, ERROR_SUCCESS);
 
    UnloadTcpIpTestDriver();
}

UnloadTcpIpTestDriver()

static void UnloadTcpIpTestDriver ( void  )

static

Definition at line 30 of file TcpIp_user.c.

{
    /* Stop the driver. */
    KmtCloseDriver();
    KmtUnloadDriver();
}

Referenced by START_TEST().