fix(EncryptionTransport): implement PortTransport

MirrorNetworking · Oct 11, 2024 · 4dd3f1e · 4dd3f1e
1 parent ed53e0a
commit 4dd3f1e
Showing 1 changed file with 24 additions and 1 deletion.
diff --git a/Assets/Mirror/Transports/Encryption/EncryptionTransport.cs b/Assets/Mirror/Transports/Encryption/EncryptionTransport.cs
@@ -7,12 +7,35 @@
 namespace Mirror.Transports.Encryption
 {
     [HelpURL("https://mirror-networking.gitbook.io/docs/manual/transports/encryption-transport")]
-    public class EncryptionTransport : Transport
+    public class EncryptionTransport : Transport, PortTransport
     {
         public override bool IsEncrypted => true;
         public override string EncryptionCipher => "AES256-GCM";
         public Transport inner;
 
+        public ushort Port
+        {
+            get
+            {
+                if (inner is PortTransport portTransport)
+                {
+                    return portTransport.Port;
+                }
+
+                Debug.LogError($"EncryptionTransport can't get Port because {inner} is not a PortTransport");
+                return 0;
+            }
+            set
+            {
+                if (inner is PortTransport portTransport)
+                {
+                    portTransport.Port = value;
+                    return;
+                }
+                Debug.LogError($"EncryptionTransport can't set Port because {inner} is not a PortTransport");
+            }
+        }
+
         public enum ValidationMode
         {
             Off,