From 5c1df2c50295999a424dbeec39515358e463c3d6 Mon Sep 17 00:00:00 2001
From: Connection Refused <cutecr@pm.me>
Date: Fri, 15 Oct 2021 12:55:21 +0800
Subject: [PATCH] [Redirector] TCP Keep Alive

---
 Redirector/Based.h         | 1 +
 Redirector/SocksHelper.cpp | 7 +++++++
 2 files changed, 8 insertions(+)

diff --git a/Redirector/Based.h b/Redirector/Based.h
index 50db46a1..24db84af 100644
--- a/Redirector/Based.h
+++ b/Redirector/Based.h
@@ -18,6 +18,7 @@
 #include <ws2ipdef.h>
 #include <WS2tcpip.h>
 #include <tlhelp32.h>
+#include <mstcpip.h>
 #include <Windows.h>
 
 #include <nfapi.h>
diff --git a/Redirector/SocksHelper.cpp b/Redirector/SocksHelper.cpp
index 17c66666..1b4baf97 100644
--- a/Redirector/SocksHelper.cpp
+++ b/Redirector/SocksHelper.cpp
@@ -38,6 +38,13 @@ SOCKET SocksHelper::Connect()
 		return INVALID_SOCKET;
 	}
 
+	{
+		DWORD returned = 0;
+
+		tcp_keepalive data = { 1, 120, 10 };
+		WSAIoctl(client, SIO_KEEPALIVE_VALS, &data, sizeof(data), NULL, 0, &returned, NULL, NULL);
+	}
+
 	return client;
 }