Added kernel/user space protocol buffer communication via TCP.

Signed-off-by: Toni Uhlig <matzeton@googlemail.com>

1 files changed, 4 insertions, 0 deletions

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) {