Lockup bug found. Fixed.

belowaverage-org · Mar 9, 2024 · e52120f · e52120f
1 parent 43916aa
commit e52120f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/Tikhole.Engine/Forwarder.cs b/Tikhole.Engine/Forwarder.cs
@@ -23,7 +23,7 @@ private void Listener_RecievedRequestData(object? sender, RecievedRequestDataEve
         {
             if (Logger.VerboseMode) Logger.Verbose("Recieved request from " + e.IPEndPoint.ToString() + ", forwarding to " + DNSServer.ToString() + "...");
             byte[]? response = Director.Forward(e.Data);
-            if (response == null)
+            if (response == null || response.Length == 0)
             {
                 Logger.Info("Request from " + e.IPEndPoint.ToString() + " timed out or experienced an error when sent to " + DNSServer.ToString() + ".");
                 return;
@@ -98,6 +98,7 @@ private ushort GetID(Memory<byte> DNSPacket)
         }
         private void SetID(Memory<byte> DNSPacket, ushort ID)
         {
+            if (DNSPacket.Length < 2) return;
             byte[] bytes = BitConverter.GetBytes(ID);
             DNSPacket.Span[0] = bytes[1];
             DNSPacket.Span[1] = bytes[0];