From c58d8f646a0a58c549e34509c9bbe34bb831961e Mon Sep 17 00:00:00 2001
From: Kawe Mazidjatari <48657826+Mauler125@users.noreply.github.com>
Date: Mon, 13 Mar 2023 20:39:49 +0100
Subject: [PATCH] CSocketCreator: add getter for authorized sockets

Add method for obtaining the amount of authorized sockets.
---
 r5dev/tier2/socketcreator.cpp | 19 +++++++++++++++++++
 r5dev/tier2/socketcreator.h   | 14 +++++++++++---
 2 files changed, 30 insertions(+), 3 deletions(-)

diff --git a/r5dev/tier2/socketcreator.cpp b/r5dev/tier2/socketcreator.cpp
index b3693de9..c9a813ba 100644
--- a/r5dev/tier2/socketcreator.cpp
+++ b/r5dev/tier2/socketcreator.cpp
@@ -315,6 +315,25 @@ bool CSocketCreator::IsSocketBlocking(void) const
 	return (WSAGetLastError() == WSAEWOULDBLOCK);
 }
 
+//-----------------------------------------------------------------------------
+// Purpose: returns authorized socket count
+// Output : int
+//-----------------------------------------------------------------------------
+int CSocketCreator::GetAuthorizedSocketCount(void) const
+{
+	int ret = 0;
+
+	for (size_t i = 0; i < m_hAcceptedSockets.size(); ++i)
+	{
+		if (m_hAcceptedSockets[i].m_pData->m_bAuthorized)
+		{
+			ret++;
+		}
+	}
+
+	return ret;
+}
+
 //-----------------------------------------------------------------------------
 // Purpose: returns accepted socket count
 // Output : int
diff --git a/r5dev/tier2/socketcreator.h b/r5dev/tier2/socketcreator.h
index a2149b1f..bea242a0 100644
--- a/r5dev/tier2/socketcreator.h
+++ b/r5dev/tier2/socketcreator.h
@@ -31,7 +31,9 @@ public:
 	bool IsListening(void) const;
 	bool IsSocketBlocking(void) const;
 
+	int GetAuthorizedSocketCount(void) const;
 	int GetAcceptedSocketCount(void) const;
+
 	SocketHandle_t GetAcceptedSocketHandle(int nIndex) const;
 	const netadr_t& GetAcceptedSocketAddress(int nIndex) const;
 	CConnectedNetConsoleData* GetAcceptedSocketData(int nIndex) const;
@@ -39,9 +41,15 @@ public:
 public:
 	struct AcceptedSocket_t
 	{
-		SocketHandle_t            m_hSocket{};
-		netadr_t                  m_Address{};
-		CConnectedNetConsoleData* m_pData = nullptr;
+		AcceptedSocket_t(void)
+		{
+			m_hSocket = NULL;
+			m_pData = nullptr;
+		}
+
+		SocketHandle_t            m_hSocket;
+		netadr_t                  m_Address;
+		CConnectedNetConsoleData* m_pData;
 	};
 
 	std::vector<AcceptedSocket_t> m_hAcceptedSockets;