From f03a43d573884b2022965208cbbe2f42ea5ddbfa Mon Sep 17 00:00:00 2001
From: JustArchi <JustArchi@JustArchi.net>
Date: Fri, 15 Apr 2016 00:32:55 +0200
Subject: [PATCH] Misc

---
 ArchiSteamFarm/Bot.cs | 22 +++++++++++++++++-----
 1 file changed, 17 insertions(+), 5 deletions(-)

diff --git a/ArchiSteamFarm/Bot.cs b/ArchiSteamFarm/Bot.cs
index eb0a932bc..0d82f1072 100755
--- a/ArchiSteamFarm/Bot.cs
+++ b/ArchiSteamFarm/Bot.cs
@@ -1226,15 +1226,27 @@ namespace ArchiSteamFarm {
 
 			string sms = Program.GetUserInput(Program.EUserInputType.SMS, BotName);
 			if (string.IsNullOrEmpty(sms)) {
+				Logging.LogGenericWarning("Aborted!", BotName);
 				DelinkMobileAuthenticator();
 				return;
 			}
 
-			AuthenticatorLinker.FinalizeResult finalizeResult = authenticatorLinker.FinalizeAddAuthenticator(sms);
-			if (finalizeResult != AuthenticatorLinker.FinalizeResult.Success) {
-				Logging.LogGenericError("Unhandled situation: " + finalizeResult, BotName);
-				DelinkMobileAuthenticator();
-				return;
+			AuthenticatorLinker.FinalizeResult finalizeResult;
+			while ((finalizeResult = authenticatorLinker.FinalizeAddAuthenticator(sms)) != AuthenticatorLinker.FinalizeResult.Success) {
+				switch (finalizeResult) {
+					case AuthenticatorLinker.FinalizeResult.BadSMSCode:
+						sms = Program.GetUserInput(Program.EUserInputType.SMS, BotName);
+						if (string.IsNullOrEmpty(sms)) {
+							Logging.LogGenericWarning("Aborted!", BotName);
+							DelinkMobileAuthenticator();
+							return;
+						}
+						break;
+					default:
+						Logging.LogGenericError("Unhandled situation: " + finalizeResult, BotName);
+						DelinkMobileAuthenticator();
+						return;
+				}
 			}
 
 			// Ensure that we also save changes made by finalization step (if any)