aboutsummaryrefslogtreecommitdiff
path: root/driver.cpp
diff options
context:
space:
mode:
authorToni Uhlig <matzeton@googlemail.com>2022-09-09 16:52:28 +0200
committerToni Uhlig <matzeton@googlemail.com>2022-09-09 16:52:28 +0200
commit38495d423f23b145a4377f87a536db5d2721f814 (patch)
tree61087a4c5e4138873cc4e6e0f9eeb4ac22d617e2 /driver.cpp
parentccb717ac524bb8a5c49ae92710864d022eebb401 (diff)
Cleaned up repository, moved files where they belong to.
Signed-off-by: Toni Uhlig <matzeton@googlemail.com>
Diffstat (limited to 'driver.cpp')
-rw-r--r--driver.cpp152
1 files changed, 0 insertions, 152 deletions
diff --git a/driver.cpp b/driver.cpp
deleted file mode 100644
index ceae8ff..0000000
--- a/driver.cpp
+++ /dev/null
@@ -1,152 +0,0 @@
-
-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);
- if (client_sockfd < 0) {
- DebuggerPrint("TCP accept failed\n");
- return STATUS_FAILED_DRIVER_ENTRY;
- }
-
- 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.");
-}
-}