Commit3

Author: NightRaven3142

src/Backend/Encryptor.cs

@@ -39,7 +39,7 @@ public void AES_Encrypt(string iF, string oF, byte[] passwordBytes)
 
         }
 
-        public void AES_Decrypt(string iF, string oF, byte[] passwordBytes)
+        public bool AES_Decrypt(string iF, string oF, byte[] passwordBytes)
         {
 
             byte[] saltBytes = new byte[] { 1, 2, 3, 4, 5, 6, 7, 8 };
@@ -60,17 +60,24 @@ public void AES_Decrypt(string iF, string oF, byte[] passwordBytes)
                 AES.Key = key.GetBytes(AES.KeySize / 8);
                 AES.IV = key.GetBytes(AES.BlockSize / 8);
 
-                using (var inFile = File.OpenRead(iF))
-                using (var cs = new CryptoStream(inFile, AES.CreateDecryptor(), CryptoStreamMode.Read))
-                using (var outFile = File.Create(oF))
+                try
                 {
+                    using (var inFile = File.OpenRead(iF))
+                    using (var cs = new CryptoStream(inFile, AES.CreateDecryptor(), CryptoStreamMode.Read))
+                    using (var outFile = File.Create(oF))
+                    {
 
-                    sbyte data;
-                    while ((data = (sbyte)cs.ReadByte()) != -1)
-                        outFile.WriteByte((byte)data);
+                        sbyte data;
+                        while ((data = (sbyte)cs.ReadByte()) != -1)
+                            outFile.WriteByte((byte)data);
 
+                    }
                 }
-
+                catch (CryptographicException)
+                {
+                    return false;
+                }
+                return true;
             }
 
         }

src/UI/MainWindow.xaml.cs

@@ -79,7 +79,8 @@ private void Encrypt_Click(object sender, RoutedEventArgs e)
             string pwd = InpTxtBox.Text;
             string ofilePath = FileTxtBox.Text;
             AESCryptoManager encryptor = new AESCryptoManager();
-            encryptor.AES_Encrypt(ofilePath, @"D:\johnk\Documents\pnp.txt", Encoding.UTF8.GetBytes(pwd));
+            encryptor.AES_Encrypt(ofilePath, System.IO.Path.GetTempPath() + "tempdata.ini", Encoding.UTF8.GetBytes(pwd));
+            File.Copy(System.IO.Path.GetTempPath() + "tempdata.ini", ofilePath, true);
         }
 
         private void Decrypt_Click(object sender, RoutedEventArgs e)
@@ -90,8 +91,17 @@ private void Decrypt_Click(object sender, RoutedEventArgs e)
             FileInfo f = new FileInfo(ofilePath);
 
             AESCryptoManager decryptor = new AESCryptoManager();
+            bool worked = decryptor.AES_Decrypt(ofilePath, System.IO.Path.GetTempPath() + "tempdata.ini", Encoding.UTF8.GetBytes(pwd)); ;
+            if (worked) { File.Copy(System.IO.Path.GetTempPath() + "tempdata.ini", ofilePath, true); }
 
-            decryptor.AES_Decrypt(ofilePath, @"D:\johnk\Documents\btec.txt", Encoding.UTF8.GetBytes(pwd));
+            if (!worked)
+            {
+                MessageBox.Show("Wrong Password");
+            }
+            else
+            {
+                MessageBox.Show("Successfully Decrypted");
+            }
         }
     }
 }