From dd9804e0bc5b1a8e058457eb2baf3030027c33c1 Mon Sep 17 00:00:00 2001
From: Travis Nickles <ryochan7@protonmail.com>
Date: Mon, 6 Apr 2020 20:43:48 -0500
Subject: [PATCH] Added Semaphore to UDP server class

---
 DS4Windows/DS4Control/UdpServer.cs | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/DS4Windows/DS4Control/UdpServer.cs b/DS4Windows/DS4Control/UdpServer.cs
index 495712a..d2946fb 100644
--- a/DS4Windows/DS4Control/UdpServer.cs
+++ b/DS4Windows/DS4Control/UdpServer.cs
@@ -73,6 +73,7 @@ namespace DS4Windows
         private SocketAsyncEventArgs[] argsList;
         private int listInd = 0;
         private ReaderWriterLockSlim poolLock = new ReaderWriterLockSlim();
+        private SemaphoreSlim _pool;
         private const int ARG_BUFFER_LEN = 80;
 
         public delegate void GetPadDetail(int padIdx, ref DualShockPadMeta meta);
@@ -82,6 +83,7 @@ namespace DS4Windows
         public UdpServer(GetPadDetail getPadDetailDel)
         {
             portInfoGet = getPadDetailDel;
+            _pool = new SemaphoreSlim(ARG_BUFFER_LEN);
             argsList = new SocketAsyncEventArgs[ARG_BUFFER_LEN];
             for (int num = 0; num < ARG_BUFFER_LEN; num++)
             {
@@ -193,6 +195,7 @@ namespace DS4Windows
             SocketAsyncEventArgs args = argsList[temp];
             poolLock.ExitWriteLock();
 
+            _pool.Wait();
             args.RemoteEndPoint = clientEP;
             Array.Copy(packetData, args.Buffer, packetData.Length);
             //args.SetBuffer(packetData, 0, packetData.Length);
@@ -200,6 +203,7 @@ namespace DS4Windows
                 udpSock.SendToAsync(args);
             }
             catch (Exception e) { }
+            _pool.Release();
         }
 
         private void ProcessIncoming(byte[] localMsg, IPEndPoint clientEP)
@@ -687,12 +691,14 @@ namespace DS4Windows
                     SocketAsyncEventArgs args = argsList[temp];
                     poolLock.ExitWriteLock();
 
+                    _pool.Wait();
                     args.RemoteEndPoint = cl;
                     Array.Copy(outputData, args.Buffer, outputData.Length);
                     try {
                         udpSock.SendToAsync(args);
                     }
                     catch (SocketException ex) { }
+                    _pool.Release();
                 }
             }