From e9f0d8e69f545f0d4b9d3128b33a24787f58eae0 Mon Sep 17 00:00:00 2001
From: Connection Refused <cutecr@pm.me>
Date: Tue, 19 Oct 2021 16:02:16 +0800
Subject: [PATCH] [Redirector] Fix udpContext not clear

---
 Redirector/EventHandler.cpp | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/Redirector/EventHandler.cpp b/Redirector/EventHandler.cpp
index 3e721ead..e26ed565 100644
--- a/Redirector/EventHandler.cpp
+++ b/Redirector/EventHandler.cpp
@@ -146,13 +146,22 @@ bool eh_init()
 	if (!DNSHandler::INIT())
 		return false;
 
-	return TCPHandler::INIT();
+	if (!TCPHandler::INIT())
+		return false;
+
+	return true;
 }
 
 void eh_free()
 {
+	lock_guard<mutex> lg(udpContextLock);
+
 	TCPHandler::FREE();
 
+	for (auto i : udpContext)
+		delete i.second;
+	udpContext.clear();
+
 	UP = 0;
 	DL = 0;
 }