1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
|
extern "C" {
#include "berkeley.h"
#include "ksocket.h"
#include "wsk.h"
DRIVER_INITIALIZE DriverEntry;
DRIVER_UNLOAD DriverUnload;
#define DebuggerPrint(...) \
DbgPrintEx(DPFLTR_IHVDRIVER_ID, DPFLTR_ERROR_LEVEL, __VA_ARGS__);
NTSTATUS
NTAPI
DriverEntry(_In_ PDRIVER_OBJECT DriverObject,
_In_ PUNICODE_STRING RegistryPath) {
UNREFERENCED_PARAMETER(DriverObject);
UNREFERENCED_PARAMETER(RegistryPath);
NTSTATUS Status;
//
// Initialize KSOCKET.
//
Status = KsInitialize();
if (!NT_SUCCESS(Status)) {
return Status;
}
//
// Client.
// Perform HTTP request to http://httpbin.org/uuid
//
{
int result;
UNREFERENCED_PARAMETER(result);
char send_buffer[] = "GET /uuid HTTP/1.1\r\n"
"Host: httpbin.org\r\n"
"Connection: close\r\n"
"\r\n";
char recv_buffer[1024] = {};
struct addrinfo hints = {};
hints.ai_flags |= AI_CANONNAME;
hints.ai_family = AF_UNSPEC;
hints.ai_socktype = SOCK_STREAM;
struct addrinfo *res;
result = getaddrinfo("httpbin.org", "80", &hints, &res);
int sockfd;
sockfd = socket_connection(AF_INET, SOCK_STREAM, IPPROTO_TCP);
if (sockfd < 0) {
DebuggerPrint("TCP client socket_connection failed\n");
return STATUS_FAILED_DRIVER_ENTRY;
}
result = connect(sockfd, res->ai_addr, (int)res->ai_addrlen);
if (result != 0) {
DebuggerPrint("TCP client connect failed\n");
return STATUS_FAILED_DRIVER_ENTRY;
}
result = send(sockfd, send_buffer, sizeof(send_buffer), 0);
if (result <= 0) {
DebuggerPrint("TCP client send failed\n");
return STATUS_FAILED_DRIVER_ENTRY;
}
result = recv(sockfd, recv_buffer, sizeof(recv_buffer), 0);
if (result <= 0) {
DebuggerPrint("TCP client recv failed\n");
return STATUS_FAILED_DRIVER_ENTRY;
} else {
DebuggerPrint("TCP client:\n%.*s\n", result, recv_buffer);
}
closesocket(sockfd);
}
//
// TCP server.
// Listen on port 9095, wait for some message,
// then send our buffer and close connection.
//
{
int result;
char send_buffer[] = "Hello from WSK!";
char recv_buffer[1024] = {0};
int server_sockfd = socket_listen(AF_INET, SOCK_STREAM, 0);
struct sockaddr_in addr;
addr.sin_family = AF_INET;
addr.sin_addr.s_addr = INADDR_ANY;
addr.sin_port = htons(9095);
result = bind(server_sockfd, (struct sockaddr *)&addr, sizeof(addr));
if (result != 0) {
DebuggerPrint("TCP server bind failed\n");
return STATUS_FAILED_DRIVER_ENTRY;
}
result = listen(server_sockfd, 1);
if (result != 0) {
DebuggerPrint("TCP server listen failed\n");
return STATUS_FAILED_DRIVER_ENTRY;
}
socklen_t addrlen = sizeof(addr);
int client_sockfd =
accept(server_sockfd, (struct sockaddr *)&addr, &addrlen);
result = recv(client_sockfd, recv_buffer, sizeof(recv_buffer) - 1, 0);
if (result > 0) {
DebuggerPrint("TCP server:\n%.*s\n", result, recv_buffer);
} else {
DebuggerPrint("TCP server recv failed\n");
}
result = send(client_sockfd, send_buffer, sizeof(send_buffer), 0);
// Wait for the client to terminate the connection.
do {
} while (recv(client_sockfd, recv_buffer, sizeof(recv_buffer) - 1, 0) > 0);
closesocket(client_sockfd);
closesocket(server_sockfd);
}
KsDestroy();
return STATUS_SUCCESS;
}
VOID DriverUnload(_In_ struct _DRIVER_OBJECT *DriverObject) {
UNREFERENCED_PARAMETER(DriverObject);
DebuggerPrint("Bye.");
}
}
|