From ccb717ac524bb8a5c49ae92710864d022eebb401 Mon Sep 17 00:00:00 2001
From: Toni Uhlig <matzeton@googlemail.com>
Date: Fri, 9 Sep 2022 15:53:43 +0200
Subject: Added kernel/user space protocol buffer communication via TCP.

Signed-off-by: Toni Uhlig <matzeton@googlemail.com>
---
 driver.cpp | 4 ++++
 1 file changed, 4 insertions(+)

(limited to 'driver.cpp')

diff --git a/driver.cpp b/driver.cpp
index b7ecbc1..ceae8ff 100644
--- a/driver.cpp
+++ b/driver.cpp
@@ -117,6 +117,10 @@ DriverEntry(_In_ PDRIVER_OBJECT DriverObject,
     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) {
-- 
cgit v1.2.3