From a3c7f66b2f5e9ee901ffaa90c4003a6992bf143c Mon Sep 17 00:00:00 2001
From: visionmercer <62051836+visionmercer@users.noreply.github.com>
Date: Tue, 23 Jun 2026 12:47:05 +0200
Subject: [PATCH] losing faith.

---
 terminkey.h | 75 ++++++++++++++++++++++++++++++++++++++---------------
 1 file changed, 54 insertions(+), 21 deletions(-)

diff --git a/terminkey.h b/terminkey.h
index cacf060..aabf3cf 100644
--- a/terminkey.h
+++ b/terminkey.h
@@ -42,6 +42,8 @@ void get_absolute_path(const char* rel_path, uintptr_t abs_path_ptr, int max_len
 #ifdef _WIN32
 
 static HANDLE hServerPipe = INVALID_HANDLE_VALUE;
+static OVERLAPPED oOverlap;
+static BOOL fPendingIO = FALSE;
 
 int terminkey() {
     if (_kbhit()) {
@@ -86,10 +88,11 @@ int termwidth() {
 }
 
 int ipc_init() {
+    // Create a regular blocking pipe, but enable FILE_FLAG_OVERLAPPED for safe async I/O
     hServerPipe = CreateNamedPipe(
         "\\\\.\\pipe\\cimp_ipc",
-        PIPE_ACCESS_DUPLEX | FILE_FLAG_FIRST_PIPE_INSTANCE,
-        PIPE_TYPE_BYTE | PIPE_READMODE_BYTE | PIPE_NOWAIT, // Enable non-blocking wait mode
+        PIPE_ACCESS_DUPLEX | FILE_FLAG_OVERLAPPED,
+        PIPE_TYPE_BYTE | PIPE_READMODE_BYTE, 
         1,
         4096,
         4096,
@@ -103,6 +106,12 @@ int ipc_init() {
         }
         return -1; // Other error
     }
+
+    // Initialize the overlapped structure and create an event
+    memset(&oOverlap, 0, sizeof(OVERLAPPED));
+    oOverlap.hEvent = CreateEvent(NULL, TRUE, TRUE, NULL);
+    fPendingIO = FALSE;
+
     return 1; // Server mode successfully started
 }
 
@@ -111,27 +120,46 @@ int ipc_check_message(uintptr_t buffer_ptr, int max_len) {
     
     char* buffer = (char*)buffer_ptr;
     DWORD bytesRead = 0;
-    
-    // In non-blocking mode, ConnectNamedPipe returns immediately.
-    // If a client is connected, it returns non-zero, or zero with ERROR_PIPE_CONNECTED.
-    BOOL connected = ConnectNamedPipe(hServerPipe, NULL) ? 
-                     TRUE : (GetLastError() == ERROR_PIPE_CONNECTED);
-                     
-    if (connected) {
-        DWORD bytesAvail = 0;
-        // Verify that data has safely populated the buffer before reading
-        if (PeekNamedPipe(hServerPipe, NULL, 0, NULL, &bytesAvail, NULL)) {
-            if (bytesAvail > 0) {
-                BOOL success = ReadFile(hServerPipe, buffer, max_len - 1, &bytesRead, NULL);
-                if (success && bytesRead > 0) {
-                    buffer[bytesRead] = '\0';
-                    DisconnectNamedPipe(hServerPipe); // Reset pipe state for the next command
-                    return (int)bytesRead;
-                }
+
+    if (!fPendingIO) {
+        // Start an asynchronous listen operation
+        if (!ConnectNamedPipe(hServerPipe, &oOverlap)) {
+            DWORD err = GetLastError();
+            if (err == ERROR_IO_PENDING) {
+                fPendingIO = TRUE;
+            } else if (err == ERROR_PIPE_CONNECTED) {
+                // Client already connected before we listened!
+                SetEvent(oOverlap.hEvent);
+                fPendingIO = TRUE;
             }
-        } else {
-            if (GetLastError() == ERROR_BROKEN_PIPE) {
+        }
+    }
+
+    if (fPendingIO) {
+        // Check if the connection event has triggered without waiting (0ms timeout)
+        if (WaitForSingleObject(oOverlap.hEvent, 0) == WAIT_OBJECT_0) {
+            DWORD cbRet = 0;
+            // Connection is ready, let's see if there is data
+            if (GetOverlappedResult(hServerPipe, &oOverlap, &cbRet, FALSE)) {
+                DWORD bytesAvail = 0;
+                if (PeekNamedPipe(hServerPipe, NULL, 0, NULL, &bytesAvail, NULL) && bytesAvail > 0) {
+                    // Data has arrived, read it synchronously
+                    BOOL success = ReadFile(hServerPipe, buffer, max_len - 1, &bytesRead, NULL);
+                    if (success && bytesRead > 0) {
+                        buffer[bytesRead] = '\0';
+                        
+                        // Clean up and reset for the next client connection
+                        DisconnectNamedPipe(hServerPipe);
+                        ResetEvent(oOverlap.hEvent);
+                        fPendingIO = FALSE;
+                        return (int)bytesRead;
+                    }
+                }
+            } else {
+                // Client disconnected or pipe broke before sending
                 DisconnectNamedPipe(hServerPipe);
+                ResetEvent(oOverlap.hEvent);
+                fPendingIO = FALSE;
             }
         }
     }
@@ -157,9 +185,14 @@ void ipc_send_message(const char* message) {
 
 void ipc_cleanup() {
     if (hServerPipe != INVALID_HANDLE_VALUE) {
+        DisconnectNamedPipe(hServerPipe);
         CloseHandle(hServerPipe);
         hServerPipe = INVALID_HANDLE_VALUE;
     }
+    if (oOverlap.hEvent != NULL) {
+        CloseHandle(oOverlap.hEvent);
+        oOverlap.hEvent = NULL;
+    }
 }
 
 void get_absolute_path(const char* rel_path, uintptr_t abs_path_ptr, int max_len) {