Added two factor authentication

Author: Its-Networking

Console/KeyAuth.cs

@@ -98,6 +98,9 @@ private class response_structure
 
             [DataMember]
             public List<users> users { get; set; }
+
+            [DataMember(Name = "2fa", IsRequired = false, EmitDefaultValue = false)] // Ensure mapping to "2fa"
+            public TwoFactorData twoFactor { get; set; } // Add a property for the 2FA data
         }
 
         public class msg
@@ -354,7 +357,7 @@ public void forgot(string username, string email)
         /// </summary>
         /// <param name="username">Username</param>
         /// <param name="pass">Password</param>
-        public void login(string username, string pass)
+        public void login(string username, string pass, string code = null)
         {
             CheckInit();
 
@@ -368,7 +371,8 @@ public void login(string username, string pass)
                 ["hwid"] = hwid,
                 ["sessionid"] = sessionid,
                 ["name"] = name,
-                ["ownerid"] = ownerid
+                ["ownerid"] = ownerid,
+                ["code"] = code ?? string.Empty
             };
 
             var response = req(values_to_upload);
@@ -599,7 +603,7 @@ public void upgrade(string username, string key)
         /// Authenticate without using usernames and passwords
         /// </summary>
         /// <param name="key">Licence used to login with</param>
-        public void license(string key)
+        public void license(string key, string code = null)
         {
             CheckInit();
 
@@ -612,7 +616,8 @@ public void license(string key)
                 ["hwid"] = hwid,
                 ["sessionid"] = sessionid,
                 ["name"] = name,
-                ["ownerid"] = ownerid
+                ["ownerid"] = ownerid,
+                ["code"] = code ?? string.Empty
             };
 
             var response = req(values_to_upload);
@@ -661,6 +666,73 @@ public void check()
             }
         }
         /// <summary>
+        /// Disable two factor authentication (2fa)
+        /// </summary>
+        public void disable2fa(string code)
+        {
+            CheckInit();
+
+            var values_to_upload = new NameValueCollection
+            {
+                ["type"] = "2fadisable",
+                ["sessionid"] = sessionid,
+                ["name"] = name,
+                ["ownerid"] = ownerid,
+                ["code"] = code
+            };
+
+            var response = req(values_to_upload);
+
+            var json = response_decoder.string_to_generic<response_structure>(response);
+            load_response_struct(json);
+
+            Console.WriteLine(json.message);
+        }
+        /// <summary>
+        /// Enable two factor authentication (2fa)
+        /// </summary>
+        public void enable2fa(string code = null)
+        {
+            CheckInit();
+
+            var values_to_upload = new NameValueCollection
+            {
+                ["type"] = "2faenable",
+                ["sessionid"] = sessionid,
+                ["name"] = name,
+                ["ownerid"] = ownerid,
+                ["code"] = code
+            };
+
+            var response = req(values_to_upload);
+
+            var json = response_decoder.string_to_generic<response_structure>(response);
+            load_response_struct(json);
+
+            if (json.success)
+            {
+                if (code == null)
+                {
+                    Console.WriteLine($"Your 2FA Secret is: {json.twoFactor.SecretCode}");
+
+                    Console.Write("Enter the 6 digit authentication code from your authentication app: ");
+                    string code6Digit = Console.ReadLine();
+                    this.enable2fa(code6Digit);
+                }
+                else
+                {
+                    Console.WriteLine("2FA has been successfully enabled!");
+                    Thread.Sleep(3000);
+                }
+            }
+            else
+            {
+                Console.WriteLine($"Error: {json.message}");
+                Thread.Sleep(3000);
+                TerminateProcess(GetCurrentProcess(), 1);
+            }
+        }
+        /// <summary>
         /// Change the data of an existing user variable, *User must be logged in*
         /// </summary>
         /// <param name="var">User variable name</param>
@@ -1173,7 +1245,7 @@ private static bool assertSSL(object sender, X509Certificate certificate, X509Ch
 
         private static void sigCheck(string resp, WebHeaderCollection headers, string type)
         {
-            if(type == "log" || type == "file") // log doesn't return a response.
+            if(type == "log" || type == "file" || type == "2faenable" || type == "2fadisable") // log doesn't return a response.
             {
                 return;
             }
@@ -1282,6 +1354,16 @@ private void load_user_data(user_data_structure data)
         }
         #endregion
 
+        [DataContract]
+        private class TwoFactorData
+        {
+            [DataMember(Name = "secret_code")]
+            public string SecretCode { get; set; }
+
+            [DataMember(Name = "QRCode")]
+            public string QRCode { get; set; }
+        }
+
         #region response_struct
         public response_class response = new response_class();
 
@@ -1295,7 +1377,7 @@ private void load_response_struct(response_structure data)
         {
             response.success = data.success;
             response.message = data.message;
-        }
+        } 
         #endregion
 
         private json_wrapper response_decoder = new json_wrapper(new response_structure());

Console/Program.cs

@@ -24,8 +24,8 @@ class Program
 
         public static api KeyAuthApp = new api(
             name: "", // Application Name
-            ownerid: "", // Owner ID
-            version: "" // Application Version /*
+            ownerid: "", // Account ID
+            version: "" // Application version. Used for automatic downloads see video here https://www.youtube.com/watch?v=kW195PLCBKs
                            //path: @"Your_Path_Here" // (OPTIONAL) see tutorial here https://www.youtube.com/watch?v=I9rxt821gMk&t=1s
         );
 
@@ -58,7 +58,7 @@ static void Main(string[] args)
 
             Console.Write("\n [1] Login\n [2] Register\n [3] Upgrade\n [4] License key only\n [5] Forgot password\n\n Choose option: ");
 
-            string username, password, key, email;
+            string username, password, key, email, code;
 
             int option = int.Parse(Console.ReadLine());
             switch (option)
@@ -68,7 +68,9 @@ static void Main(string[] args)
                     username = Console.ReadLine();
                     Console.Write("\n\n Enter password: ");
                     password = Console.ReadLine();
-                    KeyAuthApp.login(username, password);
+                    Console.Write("\n\n Enter 2fa code: (not using 2fa? Just press enter) ");
+                    code = Console.ReadLine();
+                    KeyAuthApp.login(username, password, code);
                     break;
                 case 2:
                     Console.Write("\n\n Enter username: ");
@@ -95,7 +97,9 @@ static void Main(string[] args)
                 case 4:
                     Console.Write("\n\n Enter license: ");
                     key = Console.ReadLine();
-                    KeyAuthApp.license(key);
+                    Console.Write("\n\n Enter 2fa code: (not using 2fa? Just press enter");
+                    code = Console.ReadLine();
+                    KeyAuthApp.license(key, code);
                     break;
                 case 5:
                     Console.Write("\n\n Enter username: ");
@@ -140,6 +144,25 @@ static void Main(string[] args)
                 Console.WriteLine(" Subscription name: " + KeyAuthApp.user_data.subscriptions[i].subscription + " - Expires at: " + UnixTimeToDateTime(long.Parse(KeyAuthApp.user_data.subscriptions[i].expiry)) + " - Time left in seconds: " + KeyAuthApp.user_data.subscriptions[i].timeleft);
             }
 
+            Console.Write("\n [1] Enable 2FA\n [2] Disable 2FA\n Choose option: ");
+            int tfaOptions = int.Parse(Console.ReadLine());
+            switch (tfaOptions)
+            {
+                case 1:
+                    KeyAuthApp.enable2fa();
+                    break;
+                case 2:
+                    Console.Write("Enter your 6 digit authorization code: ");
+                    code = Console.ReadLine();
+                    KeyAuthApp.disable2fa(code);
+                    break;
+                default:
+                    Console.WriteLine("\n\n Invalid Selection");
+                    Thread.Sleep(2500);
+                    TerminateProcess(GetCurrentProcess(), 1);
+                    break; // no point in this other than to not get error from IDE
+            }
+
             Console.WriteLine("\n Closing in five seconds...");
             Thread.Sleep(-1);
             Environment.Exit(0);