Add bad CM handling to logon (#773)

SteamRE · Nov 2, 2019 · 919af92 · 919af92
1 parent 611c1dd
commit 919af92
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/SteamKit2/SteamKit2/Steam/CMClient.cs b/SteamKit2/SteamKit2/Steam/CMClient.cs
@@ -520,8 +520,9 @@ void HandleLogOnResponse( IPacketMsg packetMsg )
             }
 
             var logonResp = new ClientMsgProtobuf<CMsgClientLogonResponse>( packetMsg );
+            var logonResult = ( EResult )logonResp.Body.eresult;
 
-            if ( logonResp.Body.eresult == ( int )EResult.OK )
+            if ( logonResult == EResult.OK )
             {
                 SessionID = logonResp.ProtoHeader.client_sessionid;
                 SteamID = logonResp.ProtoHeader.steamid;
@@ -537,6 +538,10 @@ void HandleLogOnResponse( IPacketMsg packetMsg )
                 heartBeatFunc.Delay = TimeSpan.FromSeconds( hbDelay );
                 heartBeatFunc.Start();
             }
+            else if ( logonResult == EResult.TryAnotherCM || logonResult == EResult.ServiceUnavailable )
+            {
+                Servers.TryMark( connection.CurrentEndPoint, connection.ProtocolTypes, ServerQuality.Bad );
+            }
         }
         void HandleLoggedOff( IPacketMsg packetMsg )
         {