From 5552d78562f4b9d08dcfec057f1b282abb6e9e98 Mon Sep 17 00:00:00 2001 From: neru Date: Wed, 13 May 2026 11:28:07 -0300 Subject: [PATCH] fix: add enabled arg to setNonBlocking --- src/proxy/tinymitm/proxy.cpp | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/src/proxy/tinymitm/proxy.cpp b/src/proxy/tinymitm/proxy.cpp index 18c9a6f..2016b68 100644 --- a/src/proxy/tinymitm/proxy.cpp +++ b/src/proxy/tinymitm/proxy.cpp @@ -103,13 +103,18 @@ struct HttpStream /* platform specific stuff */ -void setNonBlocking(SOCKET s) +void setNonBlocking(SOCKET s, bool enabled) { #ifdef _WIN32 - unsigned long mode = 1; + unsigned long mode = enabled ? 1 : 0; ioctlsocket(s, FIONBIO, &mode); #else - fcntl(s, F_SETFL, fcntl(s, F_GETFL, 0) | O_NONBLOCK); + int flags = fcntl(s, F_GETFL, 0); + if (enabled) + flags |= O_NONBLOCK; + else + flags &= ~O_NONBLOCK; + fcntl(s, F_SETFL, flags); #endif }