feat: RsaStream 重写自己的 Dispose

SaltWood-Studio · Aug 13, 2024 · 6e30275 · 6e30275
1 parent 0b1404d
commit 6e30275
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/Network/RsaStream.cs b/Network/RsaStream.cs
@@ -13,7 +13,7 @@
 
 namespace TeraIO.Network;
 
-public class RsaStream : Stream
+public class RsaStream : Stream, IDisposable
 {
     protected readonly Stream _stream;
     protected RSA _rsaPrivate;
@@ -225,4 +225,17 @@ public override void Close()
     }
 
     #endregion
+
+    #region IDisposable Overrides
+    protected override void Dispose(bool disposing)
+    {
+        base.Dispose(disposing);
+        if (disposing)
+        {
+            this.Close();
+            _stream.Dispose();
+            this.Status = RsaStreamStatus.Disposed;
+        }
+    }
+    #endregion
 }
diff --git a/Network/RsaStreamStatus.cs b/Network/RsaStreamStatus.cs
@@ -12,6 +12,7 @@ public enum RsaStreamStatus
         Handshaking,
         Established,
         Closed,
+        Disposed,
         Failed = 100,
         AuthenticationFailed = 101,
         ProtocolVersionMismatch = 102