Fixed DiffieHellman issues

Author: TwoTenPvP

MLAPI/MonoBehaviours/Core/NetworkingManager.cs

@@ -585,8 +585,8 @@ private void HandleIncomingData(int clientId, byte[] data, int channelId)
                                             byte[] aesKey = new byte[0];
                                             if(NetworkConfig.EnableEncryption)
                                             {
-                                                ushort diffiePublicSize = reader.ReadUInt16();
-                                                byte[] diffiePublic = reader.ReadBytes(diffiePublicSize);
+                                                ushort diffiePublicSize = messageReader.ReadUInt16();
+                                                byte[] diffiePublic = messageReader.ReadBytes(diffiePublicSize);
                                                 diffieHellmanPublicKeys.Add(clientId, diffiePublic);
                                                 /*
                                                 EllipticDiffieHellman diffieHellman = new EllipticDiffieHellman(EllipticDiffieHellman.DEFAULT_CURVE, EllipticDiffieHellman.DEFAULT_GENERATOR, EllipticDiffieHellman.DEFAULT_ORDER);
@@ -624,8 +624,8 @@ private void HandleIncomingData(int clientId, byte[] data, int channelId)
 
                                             if (NetworkConfig.EnableEncryption)
                                             {
-                                                ushort keyLength = reader.ReadUInt16();
-                                                clientAesKey = clientDiffieHellman.GetSharedSecret(reader.ReadBytes(keyLength));
+                                                ushort keyLength = messageReader.ReadUInt16();
+                                                clientAesKey = clientDiffieHellman.GetSharedSecret(messageReader.ReadBytes(keyLength));
                                             }
 
                                             float netTime = messageReader.ReadSingle();

MLAPI/NetworkingManagerComponents/DiffieHellman.cs

@@ -85,7 +85,7 @@ public static byte[] ToArray(this IntX v)
         {
             v.GetInternalState(out uint[] digits, out bool negative);
             byte[] b = DigitConverter.ToBytes(digits);
-            byte[] b1 = new byte[b.Length];
+            byte[] b1 = new byte[b.Length + 1];
             Array.Copy(b, b1, b.Length);
             b1[b.Length] = (byte)(negative ? 1 : 0);
             return b1;
@@ -98,7 +98,7 @@ public static IntX FromArray(byte[] b)
             uint[] u = DigitConverter.FromBytes(b1);
             return new IntX(u, b[b.Length - 1]==1);
         }
-        public static bool BitAt(this uint[] data, long index) => (data[index/8]&(1<<(int)(index%8)))!=0;
+        public static bool BitAt(this uint[] data, long index) => (data[index / 32] & (1 << (int)(index % 32))) != 0;
         public static IntX Abs(this IntX i) => i < 0 ? -i : i;
     }
 }