Merge pull request #634 from yaakov-h/626-stitch

Fix websocket servers not being marked as bad by the server list
SteamRE · Feb 9, 2019 · 3fe0d9e · 3fe0d9e
2 parents 4502f4e + 2de520a
commit 3fe0d9e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/SteamKit2/SteamKit2/Networking/Steam3/WebSocketConnection.cs b/SteamKit2/SteamKit2/Networking/Steam3/WebSocketConnection.cs
@@ -14,8 +14,7 @@ partial class WebSocketConnection : IConnection
 
         public event EventHandler<DisconnectedEventArgs> Disconnected;
 
-        public EndPoint CurrentEndPoint => currentContext?.EndPoint;
-
+        public EndPoint CurrentEndPoint { get; set; }
         public ProtocolTypes ProtocolTypes => ProtocolTypes.WebSocket;
 
         public void Connect(EndPoint endPoint, int timeout = 5000)
@@ -29,6 +28,7 @@ public void Connect(EndPoint endPoint, int timeout = 5000)
                 Disconnected?.Invoke(this, new DisconnectedEventArgs(false));
             }
 
+            CurrentEndPoint = newContext.EndPoint;
             newContext.Start(TimeSpan.FromMilliseconds(timeout));
         }
 
@@ -58,6 +58,7 @@ void DisconnectCore(bool userInitiated, WebSocketContext specificContext)
                 oldContext.Dispose();
 
                 Disconnected?.Invoke(this, new DisconnectedEventArgs(userInitiated));
+                CurrentEndPoint = null;
             }
             else
             {