Catch another couple sources of netencryption exceptions. (space-wizards#6498)

* Catch another couple sources of netencryption exceptions.

* Clean up tests a bit.

* prevent nonsense.

Author: moonheart08

Robust.Shared.Tests/Networking/NetEncryptionDoSTest.cs

@@ -22,9 +22,8 @@ public void ConnectionWorks()
 
         var packet = Receive(server);
 
-        Assert.That(packet, Is.Not.Null);
-
         Assert.That(packet.ReadVariableUInt64(), Is.EqualTo(Magic));
+        server.Shutdown(null);
     }
 
     [Test]
@@ -44,9 +43,8 @@ public void EncryptionWorks()
 
         var packet = Receive(server);
 
-        Assert.That(packet, Is.Not.Null);
-
         Assert.That(serverEnc.TryDecrypt(packet), Is.True);
+        server.Shutdown(null);
     }
 
     [Test]
@@ -64,25 +62,28 @@ public void WrongKeyFailureDoesNotThrow()
 
         client.SendMessage(message, NetDeliveryMethod.ReliableOrdered);
 
-        var packet = server.WaitMessage(1000);
-
-        Assert.That(packet, Is.Not.Null);
+        var packet = Receive(server);
 
         Assert.That(serverEnc.TryDecrypt(packet), Is.False);
+        server.Shutdown(null);
     }
 
-    private static byte[][] _badMessages =
+    private static int[] _badMessages =
     [
-        [1, 1, 1, 1, 1],
-        [1, 2],
-        [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,]
+        5,
+        1,
+        4,
+        16,
+        1024,
     ];
 
     [Test]
     [Description("Attempt to decrypt a packet that is bogus, ensuring it doesn't throw.")]
     [TestCaseSource(nameof(_badMessages))]
-    public void BadMessageDoesNotThrow(byte[] badMessage)
+    public void BadMessageDoesNotThrow(int badMessageLength)
     {
+        var badMessage = new byte[badMessageLength];
+        System.Random.Shared.NextBytes(badMessage);
         var (_, serverEnc) = MakeEncryptionPair(disjointKey: true);
         var (client, server) = MakeConnectionPair();
 
@@ -94,11 +95,13 @@ public void BadMessageDoesNotThrow(byte[] badMessage)
 
         client.SendMessage(message, NetDeliveryMethod.ReliableOrdered);
 
-        var packet = server.WaitMessage(1000);
+        var packet = Receive(server);
 
-        Assert.That(packet, Is.Not.Null);
+        Assert.That(packet.LengthBytes, Is.EqualTo(badMessageLength));
 
         Assert.That(serverEnc.TryDecrypt(packet), Is.False);
+
+        server.Shutdown(null);
     }
 
 

Robust.Shared/Network/NetEncryption.cs

@@ -91,6 +91,11 @@ public unsafe void Encrypt(NetOutgoingMessage message)
     /// <returns>Whether the operation was successful. If this fails, you likely want to drop the connection.</returns>
     public unsafe bool TryDecrypt(NetIncomingMessage message)
     {
+        // Minimum possible size a message can be is the nonce + 16 bytes of message.
+        // So we immediately bail on anything smaller.
+        if (message.LengthBytes < sizeof(ulong) + CryptoAeadXChaCha20Poly1305Ietf.AddBytes)
+            return false;
+
         var nonce = message.ReadUInt64();
         var cipherText = message.Data.AsSpan(sizeof(ulong), message.LengthBytes - sizeof(ulong));