7d9fb562a9fa275f988f1f331062be1b65ad8b0f
[reactos.git] / modules / rostests / kmtests / tcpip / TcpIp_user.c
1 /*
2 * PROJECT: ReactOS kernel-mode tests
3 * LICENSE: GPLv2+ - See COPYING in the top level directory
4 * PURPOSE: User mode part of the TcpIp.sys test suite
5 * PROGRAMMER: Jérôme Gardou <jerome.gardou@reactos.org>
6 */
7
8 #include <kmt_test.h>
9 #include <winsock2.h>
10
11 #include "tcpip.h"
12
13 static
14 void
15 LoadTcpIpTestDriver(void)
16 {
17 /* Start the special-purpose driver */
18 KmtLoadDriver(L"TcpIp", FALSE);
19 KmtOpenDriver();
20 }
21
22 static
23 void
24 UnloadTcpIpTestDriver(void)
25 {
26 /* Stop the driver. */
27 KmtCloseDriver();
28 KmtUnloadDriver();
29 }
30
31 START_TEST(TcpIpTdi)
32 {
33 LoadTcpIpTestDriver();
34
35 ok(KmtSendToDriver(IOCTL_TEST_TDI) == ERROR_SUCCESS, "\n");
36
37 UnloadTcpIpTestDriver();
38 }
39
40 static
41 DWORD
42 WINAPI
43 AcceptProc(
44 _In_ LPVOID Parameter)
45 {
46 WORD WinsockVersion;
47 WSADATA WsaData;
48 int Error;
49 SOCKET ListenSocket, AcceptSocket;
50 struct sockaddr_in ListenAddress, AcceptAddress;
51 int AcceptAddressLength;
52 HANDLE ReadyToConnectEvent = (HANDLE)Parameter;
53
54 /* Initialize winsock */
55 WinsockVersion = MAKEWORD(2, 0);
56 Error = WSAStartup(WinsockVersion, &WsaData);
57 ok(Error == 0, "");
58
59 ListenSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
60 ok(ListenSocket != INVALID_SOCKET, "socket failed");
61
62 ZeroMemory(&ListenAddress, sizeof(ListenAddress));
63 ListenAddress.sin_addr.S_un.S_addr = inet_addr("127.0.0.1");
64 ListenAddress.sin_port = htons(TEST_CONNECT_SERVER_PORT);
65 ListenAddress.sin_family = AF_INET;
66
67 Error = bind(ListenSocket, (struct sockaddr*)&ListenAddress, sizeof(ListenAddress));
68 ok(Error == 0, "");
69
70 Error = listen(ListenSocket, 1);
71 ok(Error == 0, "");
72
73 SetEvent(ReadyToConnectEvent);
74
75 AcceptAddressLength = sizeof(AcceptAddress);
76 AcceptSocket = accept(ListenSocket, (struct sockaddr*)&AcceptAddress, &AcceptAddressLength);
77 ok(AcceptSocket != INVALID_SOCKET, "\n");
78 ok_eq_long(AcceptAddressLength, sizeof(AcceptAddress));
79 ok_eq_hex(AcceptAddress.sin_addr.S_un.S_addr, inet_addr("127.0.0.1"));
80 ok_eq_hex(AcceptAddress.sin_port, htons(TEST_CONNECT_CLIENT_PORT));
81
82 return 0;
83 }
84
85 START_TEST(TcpIpConnect)
86 {
87 HANDLE AcceptThread;
88 HANDLE ReadyToConnectEvent;
89
90 ReadyToConnectEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
91 ok(ReadyToConnectEvent != NULL, "\n");
92
93 AcceptThread = CreateThread(NULL, 0, AcceptProc, (PVOID)ReadyToConnectEvent, 0, NULL);
94 ok(AcceptThread != NULL, "");
95
96 WaitForSingleObject(ReadyToConnectEvent, INFINITE);
97
98 LoadTcpIpTestDriver();
99
100 ok(KmtSendToDriver(IOCTL_TEST_CONNECT) == ERROR_SUCCESS, "\n");
101
102 WaitForSingleObject(AcceptThread, INFINITE);
103
104 UnloadTcpIpTestDriver();
105
106 WSACleanup();
107 }