From cacb4f453fe912e083055fe97da703d42b605306 Mon Sep 17 00:00:00 2001
From: AJ Martinez <aj@tyronesama.moe>
Date: Sat, 18 Mar 2023 13:24:55 -0700
Subject: [PATCH] Scramble keys and challenges when init/closing node

---
 src/d_net.c | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/src/d_net.c b/src/d_net.c
index 3799ecd13..171e5f483 100644
--- a/src/d_net.c
+++ b/src/d_net.c
@@ -29,6 +29,7 @@
 #include "z_zone.h"
 #include "i_tcp.h"
 #include "d_main.h" // srb2home
+#include "stun.h"
 
 //
 // NETWORKING
@@ -629,7 +630,12 @@ static void InitAck(void)
 		ackpak[i].acknum = 0;
 
 	for (i = 0; i < MAXNETNODES; i++)
+	{
 		InitNode(&nodes[i]);
+
+		csprng(lastSentChallenge[i], sizeof(lastSentChallenge[i]));
+		csprng(lastReceivedKey[i], sizeof(lastReceivedKey[i]));
+	}
 }
 
 /** Removes all acks of a given packet type
@@ -699,6 +705,9 @@ void Net_CloseConnection(INT32 node)
 	if (server)
 		SV_AbortLuaFileTransfer(node);
 	I_NetFreeNodenum(node);
+
+	csprng(lastSentChallenge[node], sizeof(lastSentChallenge[node]));
+	csprng(lastReceivedKey[node], sizeof(lastReceivedKey[node]));
 }
 
 //