CSharp Forms 2fa and UI update

Author: Its-Networking

Form/KeyAuth.cs

@@ -14,6 +14,7 @@
 using System.Threading;
 using Cryptographic;
 using System.Runtime.InteropServices;
+using System.Windows.Forms;
 
 namespace KeyAuth
 {
@@ -98,6 +99,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 +358,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 +372,8 @@ public void login(string username, string pass)
                 ["hwid"] = hwid,
                 ["sessionid"] = sessionid,
                 ["name"] = name,
-                ["ownerid"] = ownerid
+                ["ownerid"] = ownerid,
+                ["code"] = code ?? null
             };
 
             var response = req(values_to_upload);
@@ -599,7 +604,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 +617,8 @@ public void license(string key)
                 ["hwid"] = hwid,
                 ["sessionid"] = sessionid,
                 ["name"] = name,
-                ["ownerid"] = ownerid
+                ["ownerid"] = ownerid,
+                ["code"]  = code ?? null
             };
 
             var response = req(values_to_upload);
@@ -661,6 +667,68 @@ 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);
+        }
+        /// <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) // First time enabling 2FA, no code provided
+                {
+                    // Display the secret code to the user
+                    Clipboard.SetText(json.twoFactor.SecretCode);
+                    System.Windows.MessageBox.Show($"Your 2FA Secret Code has been copied to your clipboard! \n\n: {json.twoFactor.SecretCode}", "2FA Secret");
+
+                }
+                else // Code provided by the user
+                {
+                    System.Windows.MessageBox.Show("2FA has been successfully enabled!", "2FA Setup");
+                }
+            }
+            else
+            {
+                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>
@@ -1282,6 +1350,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();
 

Form/Login.Designer.cs

@@ -121,7 +121,7 @@ private void linkLabel2_LinkClicked(object sender, LinkLabelLinkClickedEventArgs
 
         private void loginBtn_Click_1(object sender, EventArgs e)
         {
-            KeyAuthApp.login(usernameField.Text, passwordField.Text);
+            KeyAuthApp.login(usernameField.Text, passwordField.Text, tfaField.Text);
             if (KeyAuthApp.response.success)
             {
                 Main main = new Main();
@@ -153,7 +153,7 @@ private void registerBtn_Click(object sender, EventArgs e)
 
         private void licenseBtn_Click(object sender, EventArgs e)
         {
-            KeyAuthApp.license(keyField.Text);
+            KeyAuthApp.license(keyField.Text, tfaField.Text);
             if (KeyAuthApp.response.success)
             {
                 Main main = new Main();