File/Folder validation

mark-s · Feb 18, 2021 · f1812fa · f1812fa
1 parent 0761aec
commit f1812fa
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/QnapBackupDecryptor.Core/JobMaker.cs b/src/QnapBackupDecryptor.Core/JobMaker.cs
@@ -11,8 +11,14 @@ public static class JobMaker
 
         public static List<FileJob> GetDecryptJobs(string encryptedSource, string decryptedTarget, bool overwrite, bool includeSubFolders)
         {
+            if (Directory.Exists(encryptedSource) == false & File.Exists(encryptedSource) == false)
+                return new List<FileJob> { new FileJob(new DirectoryInfo(encryptedSource), new FileInfo(decryptedTarget), false, "Source does not exist") };
+
             var sourceIsFolder = File.GetAttributes(encryptedSource).HasFlag(FileAttributes.Directory);
-            var destIsFolder = File.GetAttributes(decryptedTarget).HasFlag(FileAttributes.Directory);
+
+            bool destIsFolder = false;
+            if (Directory.Exists(decryptedTarget))
+                destIsFolder = File.GetAttributes(decryptedTarget).HasFlag(FileAttributes.Directory);
 
             if (sourceIsFolder & destIsFolder == false)
                 return new List<FileJob> { new FileJob(new DirectoryInfo(encryptedSource), new FileInfo(decryptedTarget), false, "Cannot write an encrypted folder to a single file") };