Merge remote-tracking branch 'origin/dev'

src/MaksIT.LTO.Backup/Application.cs

@@ -1,66 +1,148 @@
 using System.Text;
 using System.Text.Json;
-using System.Diagnostics.CodeAnalysis;
+using System.Net;
+
+using Microsoft.Extensions.Logging;
+using Microsoft.Extensions.Options;
 
 using MaksIT.LTO.Core;
 using MaksIT.LTO.Backup.Entities;
-using System.Net;
+using MaksIT.LTO.Core.MassStorage;
+using MaksIT.LTO.Core.Networking;
+using MaksIT.LTO.Core.Utilities;
+using MaksIT.LTO.Core.Helpers;
+
 
 namespace MaksIT.LTO.Backup;
+
 public class Application {
 
   private const string _descriptoFileName = "descriptor.json";
-  private const string _configurationFileName = "configuration.json";
+  private const string _secretFileName = "secret.txt";
 
   private readonly string appPath = AppDomain.CurrentDomain.BaseDirectory;
   private readonly string _tapePath;
   private readonly string _descriptorFilePath;
+  private readonly string _secretFilePath;
 
-  private Configuration _configuration;
+  private readonly ILogger<Application> _logger;
+  private readonly ILogger<TapeDeviceHandler> _tapeDeviceLogger;
+  private readonly ILogger<NetworkConnection> _networkConnectionLogger;
+
+  private readonly Configuration _configuration;
+  private readonly string _secret;
+
+  public Application(
+    ILogger<Application> logger,
+    ILoggerFactory loggerFactory,
+    IOptions<Configuration> configuration
+  ) {
+    _logger = logger;
+    _tapeDeviceLogger = loggerFactory.CreateLogger<TapeDeviceHandler>();
+    _networkConnectionLogger = loggerFactory.CreateLogger<NetworkConnection>();
 
-  public Application() {
     _descriptorFilePath = Path.Combine(appPath, _descriptoFileName);
-    LoadConfiguration();
+    _secretFilePath = Path.Combine(appPath, _secretFileName);
 
+    _configuration = configuration.Value;
     _tapePath = _configuration.TapePath;
+
+    var secret = Environment.GetEnvironmentVariable("LTO_BACKUP_SECRET")
+      ?? Environment.GetEnvironmentVariable("LTO_BACKUP_SECRET", EnvironmentVariableTarget.Machine);
+
+    if (!string.IsNullOrWhiteSpace(secret))
+      _secret = secret;
+    else if (!File.Exists(_secretFilePath)) {
+      _secret = AESGCMUtility.GenerateKeyBase64();
+      File.WriteAllText(_secretFilePath, _secret);
+    }
+    else
+      _secret = File.ReadAllText(_secretFilePath);
+
+    if (string.IsNullOrWhiteSpace(_secret)) {
+      throw new InvalidOperationException("Secret is required for encryption.");
+    }
   }
 
-  [MemberNotNull(nameof(_configuration))]
-  public void LoadConfiguration() {
-    var configFilePath = Path.Combine(appPath, _configurationFileName);
-    var configuration = JsonSerializer.Deserialize<Configuration>(File.ReadAllText(configFilePath));
-    if (configuration == null)
-      throw new InvalidOperationException("Failed to deserialize configuration.");
+  public void Run() {
+    Console.OutputEncoding = Encoding.UTF8;
 
-    _configuration = configuration;
+    while (true) {
+      Console.WriteLine("MaksIT.LTO.Backup v0.0.1");
+      Console.WriteLine("© Maksym Sadovnychyy (MAKS-IT) 2024");
+
+      Console.WriteLine("\nSelect an action:");
+      Console.WriteLine("1. Load tape");
+      Console.WriteLine("2. Backup");
+      Console.WriteLine("3. Restore");
+      Console.WriteLine("4. Eject tape");
+      Console.WriteLine("5. Get device status");
+      Console.WriteLine("6. Tape Erase (Short)");
+      Console.WriteLine("7. Exit");
+      Console.Write("Enter your choice: ");
+
+      var choice = Console.ReadLine();
+
+      try {
+        switch (choice) {
+          case "1":
+            LoadTape();
+            break;
+          case "2":
+            Backup();
+            break;
+          case "3":
+            Restore();
+            break;
+          case "4":
+            EjectTape();
+            break;
+          case "5":
+            GetDeviceStatus();
+            break;
+          case "6":
+            TapeErase();
+            break;
+          case "7":
+            Console.WriteLine("Exiting...");
+            return;
+          default:
+            Console.WriteLine("Invalid choice. Please try again.");
+            break;
+        }
+      }
+      catch (Exception ex) {
+        _logger.LogError(ex, $"An error occurred: {ex.Message}");
+      }
+    }
   }
 
-  public void LoadTape() {
-    using var handler = new TapeDeviceHandler(_tapePath);
+  private void LoadTape() {
+    using var handler = new TapeDeviceHandler(_tapeDeviceLogger, _tapePath);
     LoadTape(handler);
   }
 
-  public void LoadTape(TapeDeviceHandler handler) {
+  private void LoadTape(TapeDeviceHandler handler) {
     handler.Prepare(TapeDeviceHandler.TAPE_LOAD);
     Thread.Sleep(2000);
 
-    Console.WriteLine("Tape loaded.");
+    _logger.LogInformation("Tape loaded.");
   }
 
-  public void EjectTape() {
-    using var handler = new TapeDeviceHandler(_tapePath);
+  private void EjectTape() {
+    using var handler = new TapeDeviceHandler(_tapeDeviceLogger, _tapePath);
     EjectTape(handler);
   }
 
-  public void EjectTape(TapeDeviceHandler handler) {
+  private void EjectTape(TapeDeviceHandler handler) {
     handler.Prepare(TapeDeviceHandler.TAPE_UNLOAD);
     Thread.Sleep(2000);
 
-    Console.WriteLine("Tape ejected.");
+    _logger.LogInformation("Tape ejected.");
   }
 
-  public void TapeErase() {
-    using var handler = new TapeDeviceHandler(_tapePath);
+  private void TapeErase() {
+    using var handler = new TapeDeviceHandler(_tapeDeviceLogger, _tapePath);
     LoadTape(handler);
 
     handler.SetMediaParams(LTOBlockSizes.LTO5);
@@ -80,17 +162,15 @@ public class Application {
     handler.SetPosition(TapeDeviceHandler.TAPE_REWIND);
     Thread.Sleep(2000);
 
-    Console.WriteLine("Tape erased.");
+    _logger.LogInformation("Tape erased.");
   }
 
-
-  public void GetDeviceStatus() {
-    using var handler = new TapeDeviceHandler(_tapePath);
+  private void GetDeviceStatus() {
+    using var handler = new TapeDeviceHandler(_tapeDeviceLogger, _tapePath);
     handler.GetStatus();
   }
 
-
-  public void PathAccessWrapper(WorkingFolder workingFolder, Action<string> myAction) {
+  private void PathAccessWrapper(WorkingFolder workingFolder, Action<string> myAction) {
 
     if (workingFolder.LocalPath != null) {
       var localPath = workingFolder.LocalPath.Path;
@@ -117,14 +197,14 @@ public class Application {
           throw new InvalidOperationException("Network credentials are required for remote paths.");
         }
           
-        using (new NetworkConnection(smbPath, networkCredential)) {
+        using (new NetworkConnection(_networkConnectionLogger, smbPath, networkCredential)) {
            myAction(smbPath);
         }
       }
     }
   }
 
-  public void CreateDescriptor(WorkingFolder workingFolder, string descriptorFilePath, uint blockSize) {
+  private void CreateDescriptor(WorkingFolder workingFolder, string descriptorFilePath, uint blockSize) {
 
     PathAccessWrapper(workingFolder, (directoryPath) => {
       var files = Directory.GetFiles(directoryPath, "*.*", SearchOption.AllDirectories);
@@ -138,18 +218,8 @@ public class Application {
         var relativePath = Path.GetRelativePath(directoryPath, filePath);
         var numberOfBlocks = (uint)((fileInfo.Length + blockSize - 1) / blockSize);
 
-        // Optional: Calculate a simple hash for file integrity (e.g., MD5)
-        using var md5 = System.Security.Cryptography.MD5.Create();
-        using var fileStream = new FileStream(filePath, FileMode.Open, FileAccess.Read);
-        using var bufferedStream = new BufferedStream(fileStream, (int)blockSize);
-
-        byte[] buffer = new byte[blockSize];
-        int bytesRead;
-        while ((bytesRead = bufferedStream.Read(buffer, 0, buffer.Length)) > 0) {
-          md5.TransformBlock(buffer, 0, bytesRead, null, 0);
-        }
-        md5.TransformFinalBlock(Array.Empty<byte>(), 0, 0);
-        string fileHash = BitConverter.ToString(md5.Hash).Replace("-", "").ToLower();
+        // Calculate CRC32 checksum for file integrity
+        string fileHash = ChecksumUtility.CalculateCRC32ChecksumFromFileInChunks(filePath, (int)blockSize);
 
         descriptor.Add(new FileDescriptor {
           StartBlock = currentTapeBlock, // Position of the file on the tape
@@ -174,27 +244,14 @@ public class Application {
     });
   }
 
-  private void ZeroFillBlocks(TapeDeviceHandler handler, int blocks, uint blockSize) {
-    Console.WriteLine($"Writing {blocks} zero-filled blocks to tape.");
-    Console.WriteLine($"Block Size: {blockSize}.");
 
-    var writeError = 0;
 
-    for (int i = 0; i < blocks; i++) {
-      writeError = handler.WriteData(new byte[blockSize]);
-      if (writeError != 0)
-        return;
-
-      Thread.Sleep(_configuration.WriteDelay);
-    }
-  }
-
-  public void WriteFilesToTape(WorkingFolder workingFolder, string descriptorFilePath, uint blockSize) {
+  private void WriteFilesToTape(WorkingFolder workingFolder, string descriptorFilePath, uint blockSize) {
     PathAccessWrapper(workingFolder, (directoryPath) => {
-      Console.WriteLine($"Writing files to tape from: {directoryPath}.");
-      Console.WriteLine($"Block Size: {blockSize}.");
+      _logger.LogInformation($"Writing files to tape from: {directoryPath}.");
+      _logger.LogInformation($"Block Size: {blockSize}.");
 
-      using var handler = new TapeDeviceHandler(_tapePath);
+      using var handler = new TapeDeviceHandler(_tapeDeviceLogger, _tapePath);
 
       LoadTape(handler);
 
@@ -221,9 +278,6 @@ public class Application {
 
       var currentTapeBlock = (descriptorJson.Length + blockSize - 1) / blockSize;
 
-
-      int writeError = 0;
-
       foreach (var file in descriptor.Files) {
         var filePath = Path.Combine(directoryPath, file.FilePath);
         using var fileStream = new FileStream(filePath, FileMode.Open, FileAccess.Read);
@@ -238,9 +292,9 @@ public class Application {
             Array.Clear(buffer, bytesRead, buffer.Length - bytesRead);
           }
           
-          writeError = handler.WriteData(buffer);
+          var writeError = handler.WriteData(buffer);
           if (writeError != 0) {
-            Console.WriteLine($"Failed to write file: {filePath}");
+            _logger.LogInformation($"Failed to write file: {filePath}");
             return;
           }
 
@@ -249,21 +303,28 @@ public class Application {
         }
       }
 
-
       // write mark to indicate end of files
       handler.WriteMarks(TapeDeviceHandler.TAPE_FILEMARKS, 1);
       Thread.Sleep(_configuration.WriteDelay);
 
       // write descriptor to tape
       var descriptorData = Encoding.UTF8.GetBytes(descriptorJson);
-      var descriptorBlocks = (descriptorData.Length + blockSize - 1) / blockSize;
-      for (int i = 0; i < descriptorBlocks; i++) {
+
+      // encrypt the serialized descriptor
+      var encryptedDescriptorData = AESGCMUtility.EncryptData(descriptorData, _secret);
+
+      // add padding to the encrypted descriptor data
+      var paddedDescriptorData = PaddingUtility.AddPadding(encryptedDescriptorData, (int)blockSize);
+
+      // calculate the number of blocks needed
+      var descriptorBlocks = (paddedDescriptorData.Length + blockSize - 1) / blockSize;
+      for (var i = 0; i < descriptorBlocks; i++) {
         var startIndex = i * blockSize;
-        var length = Math.Min(blockSize, descriptorData.Length - startIndex);
-        byte[] block = new byte[blockSize]; // Initialized with zeros by default
-        Array.Copy(descriptorData, startIndex, block, 0, length);
-        
-        writeError = handler.WriteData(block);
+        var length = Math.Min(blockSize, paddedDescriptorData.Length - startIndex);
+        var block = new byte[blockSize]; // Initialized with zeros by default
+        Array.Copy(paddedDescriptorData, startIndex, block, 0, length);
+
+        var writeError = handler.WriteData(block);
         if (writeError != 0)
           return;
 
@@ -271,8 +332,9 @@ public class Application {
         Thread.Sleep(_configuration.WriteDelay); // Small delay between blocks
       }
 
-      // write 3 0 filled blocks to indicate end of backup
-      ZeroFillBlocks(handler, 3, blockSize);
+      // write mark to indicate end of files
+      handler.WriteMarks(TapeDeviceHandler.TAPE_FILEMARKS, 2);
+      Thread.Sleep(_configuration.WriteDelay);
 
       handler.Prepare(TapeDeviceHandler.TAPE_UNLOCK);
       Thread.Sleep(2000);
@@ -283,11 +345,11 @@ public class Application {
     });
   }
 
-  public BackupDescriptor? FindDescriptor(uint blockSize) {
-    Console.WriteLine("Searching for descriptor on tape...");
-    Console.WriteLine($"Block Size: {blockSize}.");
+  private BackupDescriptor? FindDescriptor(uint blockSize) {
+    _logger.LogInformation("Searching for descriptor on tape...");
+    _logger.LogInformation($"Block Size: {blockSize}.");
 
-    using var handler = new TapeDeviceHandler(_tapePath);
+    using var handler = new TapeDeviceHandler(_tapeDeviceLogger,_tapePath);
 
     LoadTape(handler);
 
@@ -299,47 +361,66 @@ public class Application {
     handler.SetPosition(TapeDeviceHandler.TAPE_SPACE_FILEMARKS, 0, 1);
     Thread.Sleep(2000);
 
-    handler.WaitForTapeReady();
+    var endOfBackupMarkerPosition = handler.GetPosition(TapeDeviceHandler.TAPE_ABSOLUTE_BLOCK);
+    if (endOfBackupMarkerPosition.Error != null || endOfBackupMarkerPosition.OffsetLow == null)
+      return null;
 
-    // Read data from tape until 3 zero-filled blocks are found
-    var buffer = new List<byte>();
-    byte[] data;
-    var zeroBlocks = 0;
-    do {
-      data = handler.ReadData(blockSize);
-      buffer.AddRange(data);
-      if (data.All(b => b == 0)) {
-        zeroBlocks++;
-      }
-      else {
-        zeroBlocks = 0;
-      }
-    } while (zeroBlocks < 3);
+    _logger.LogInformation($"End of backup marker position: {endOfBackupMarkerPosition.OffsetLow}");
 
-    // Remove the last 3 zero-filled blocks from the buffer
-    var totalZeroBlocksSize = (int)(3 * blockSize);
-    if (buffer.Count >= totalZeroBlocksSize) {
-      buffer.RemoveRange(buffer.Count - totalZeroBlocksSize, totalZeroBlocksSize);
+    var descriptorBlocks = endOfBackupMarkerPosition.OffsetLow;
+
+    handler.SetPosition(TapeDeviceHandler.TAPE_SPACE_FILEMARKS, 0, 2);
+    Thread.Sleep(2000);
+
+    var endOfDescriptorMarkerPosition = handler.GetPosition(TapeDeviceHandler.TAPE_ABSOLUTE_BLOCK);
+    if (endOfDescriptorMarkerPosition.Error != null || endOfDescriptorMarkerPosition.OffsetLow == null)
+      return null;
+
+    _logger.LogInformation($"End of descriptor marker position: {endOfDescriptorMarkerPosition.OffsetLow}");
+
+    descriptorBlocks = endOfDescriptorMarkerPosition.OffsetLow - descriptorBlocks;
+
+    if (descriptorBlocks == null)
+      return null;
+
+    _logger.LogInformation($"Descriptor blocks to read: {descriptorBlocks}");
+
+    handler.SetPosition(TapeDeviceHandler.TAPE_ABSOLUTE_BLOCK, 0, (long)(endOfDescriptorMarkerPosition.OffsetLow - descriptorBlocks));
+    Thread.Sleep(2000);
+
+    descriptorBlocks -= 2;
+
+    var paddedData = new byte[(descriptorBlocks.Value) * blockSize];
+    var buffer = new byte[blockSize];
+
+    for (var i = 0; i < descriptorBlocks; i++) {
+      var bytesRead = handler.ReadData(buffer, 0, buffer.Length);
+
+      // Copy the read data into the encryptedData array
+      Array.Copy(buffer, 0, paddedData, i * blockSize, buffer.Length);
     }
 
-    // Convert buffer to byte array
-    var byteArray = buffer.ToArray();
+    // i need to remove padding from the data
+    var encryptedData = PaddingUtility.RemovePadding(paddedData, (int)blockSize);
+
+    // decrypt the data
+    var decryptedData = AESGCMUtility.DecryptData(encryptedData, _secret);
 
     // Convert byte array to string and trim ending zeros
-    var json = Encoding.UTF8.GetString(byteArray).TrimEnd('\0');
+    var json = Encoding.UTF8.GetString(decryptedData);
 
     try {
       var descriptor = JsonSerializer.Deserialize<BackupDescriptor>(json);
       if (descriptor != null) {
-        Console.WriteLine("Descriptor read successfully.");
+        _logger.LogInformation("Descriptor read successfully.");
         return descriptor;
       }
     }
     catch (JsonException ex) {
-      Console.WriteLine($"Failed to parse descriptor JSON: {ex.Message}");
+      _logger.LogInformation($"Failed to parse descriptor JSON: {ex.Message}");
+      return null;
     }
 
-
     handler.Prepare(TapeDeviceHandler.TAPE_UNLOCK);
     Thread.Sleep(2000);
 
@@ -349,13 +430,13 @@ public class Application {
     return null;
   }
 
-  public void RestoreDirectory(BackupDescriptor descriptor, WorkingFolder workingFolder) {
+  private void RestoreDirectory(BackupDescriptor descriptor, WorkingFolder workingFolder) {
 
     PathAccessWrapper(workingFolder, (restoreDirectoryPath) => {
-      Console.WriteLine("Restoring files to directory: " + restoreDirectoryPath);
-      Console.WriteLine("Block Size: " + descriptor.BlockSize);
+      _logger.LogInformation("Restoring files to directory: " + restoreDirectoryPath);
+      _logger.LogInformation("Block Size: " + descriptor.BlockSize);
 
-      using var handler = new TapeDeviceHandler(_tapePath);
+      using var handler = new TapeDeviceHandler(_tapeDeviceLogger,_tapePath);
 
       LoadTape(handler);
 
@@ -390,20 +471,11 @@ public class Application {
           }
         }
 
-        // check md5 checksum of restored file with the one in descriptor
-        using (var md5 = System.Security.Cryptography.MD5.Create()) {
-          using (var fileStreamRead = new FileStream(filePath, FileMode.Open, FileAccess.Read)) {
-            var fileHash = md5.ComputeHash(fileStreamRead);
-            var fileHashString = BitConverter.ToString(fileHash).Replace("-", "").ToLower();
-
-            if (fileHashString != file.FileHash) {
-              Console.WriteLine($"Checksum mismatch for file: {filePath}");
-            }
-            else {
-              Console.WriteLine($"Restored file: {filePath}");
-            }
-          }
-        }
+        // check checksum of restored file with the one in descriptor
+        if (ChecksumUtility.VerifyCRC32ChecksumFromFileInChunks(filePath, file.FileHash, (int)descriptor.BlockSize))
+          _logger.LogInformation($"Restored file: {filePath}");
+        else
+          _logger.LogInformation($"Checksum mismatch for file: {filePath}");
       }
 
       handler.SetPosition(TapeDeviceHandler.TAPE_REWIND);
@@ -411,52 +483,53 @@ public class Application {
     });
   }
 
-  public int CheckMediaSize(string ltoGen) {
+  private int CheckMediaSize(string ltoGen) {
     var descriptor = JsonSerializer.Deserialize<BackupDescriptor>(File.ReadAllText(_descriptorFilePath));
     if (descriptor == null) {
-      Console.WriteLine("Failed to read descriptor.");
+      _logger.LogInformation("Failed to read descriptor.");
       return 1;
     }
 
-    var totalBlocks = (ulong)descriptor.Files.Sum(f => f.NumberOfBlocks);
+    var encryptedDescriptorData = AESGCMUtility.EncryptData(File.ReadAllBytes(_descriptorFilePath), _secret);
 
-    const ulong fileMarkBlocks = 1;
-    const ulong terminalBlocks = 3;
+    var paddedDescriptorData = PaddingUtility.AddPadding(encryptedDescriptorData, (int)descriptor.BlockSize);
 
-    var descriptorSize = new FileInfo(_descriptoFileName).Length;
-    ulong descriptorSizeBlocks = (ulong)Math.Ceiling((double)descriptorSize / descriptor.BlockSize);
+    const ulong fileMarkBlocks = 2;
+ 
+    var descriptorSize = paddedDescriptorData.Length;
+    var descriptorSizeBlocks = Math.Ceiling((double)descriptorSize / descriptor.BlockSize);
 
-    totalBlocks += fileMarkBlocks + descriptorSizeBlocks + terminalBlocks;
+    var totalBlocks = fileMarkBlocks + descriptorSizeBlocks;
 
     var maxBlocks = LTOBlockSizes.GetMaxBlocks(ltoGen);
     if (totalBlocks > maxBlocks) {
-      Console.WriteLine("Backup will not fit on tape. Please use a larger tape.");
+      _logger.LogInformation("Backup will not fit on tape. Please use a larger tape.");
       return 1;
     }
     else {
-      Console.WriteLine("Backup will fit on tape.");
+      _logger.LogInformation("Backup will fit on tape.");
     }
 
     return 0;
   }
 
-  public void Backup() {
+  private void Backup() {
     while (true) {
-      Console.WriteLine("\nSelect a backup to perform:");
+      _logger.LogInformation("\nSelect a backup to perform:");
       for (int i = 0; i < _configuration.Backups.Count; i++) {
         var backupInt = _configuration.Backups[i];
-        Console.WriteLine($"{i + 1}. Backup Name: {backupInt.Name}, Bar code {(string.IsNullOrEmpty(backupInt.Barcode) ? "None" : backupInt.Barcode)}, Source: {backupInt.Source}, Destination: {backupInt.Destination}");
+        _logger.LogInformation($"{i + 1}. Backup Name: {backupInt.Name}, Bar code {(string.IsNullOrEmpty(backupInt.Barcode) ? "None" : backupInt.Barcode)}, Source: {backupInt.Source}, Destination: {backupInt.Destination}");
       }
 
       Console.Write("Enter your choice (or '0' to go back): ");
       var choice = Console.ReadLine();
 
       if (choice == "0") {
-        return; // Go back to the main menu
+        return;
       }
 
       if (!int.TryParse(choice, out int index) || index < 1 || index > _configuration.Backups.Count) {
-        Console.WriteLine("Invalid choice. Please try again.");
+        _logger.LogInformation("Invalid choice. Please try again.");
         continue;
       }
 
@@ -476,28 +549,28 @@ public class Application {
       WriteFilesToTape(backup.Source, _descriptorFilePath, blockSize);
 
       File.Delete(_descriptorFilePath);
-      Console.WriteLine("Backup completed.");
-      return; // Go back to the main menu after completing the backup
+      _logger.LogInformation("Backup completed.");
+      return;
     }
   }
 
-  public void Restore() {
+  private void Restore() {
     while (true) {
-      Console.WriteLine("\nSelect a backup to restore:");
+      _logger.LogInformation("\nSelect a backup to restore:");
       for (int i = 0; i < _configuration.Backups.Count; i++) {
         var backupInt = _configuration.Backups[i];
-        Console.WriteLine($"{i + 1}. Backup Name: {backupInt.Name}, Bar code {(string.IsNullOrEmpty(backupInt.Barcode) ? "None" : backupInt.Barcode)}, Source: {backupInt.Source}, Destination: {backupInt.Destination}");
+        _logger.LogInformation($"{i + 1}. Backup Name: {backupInt.Name}, Bar code {(string.IsNullOrEmpty(backupInt.Barcode) ? "None" : backupInt.Barcode)}, Source: {backupInt.Source}, Destination: {backupInt.Destination}");
       }
 
       Console.Write("Enter your choice (or '0' to go back): ");
       var choice = Console.ReadLine();
 
       if (choice == "0") {
-        return; // Go back to the main menu
+        return;
       }
 
       if (!int.TryParse(choice, out int index) || index < 1 || index > _configuration.Backups.Count) {
-        Console.WriteLine("Invalid choice. Please try again.");
+        _logger.LogInformation("Invalid choice. Please try again.");
         continue;
       }
 
@@ -505,22 +578,22 @@ public class Application {
       
       uint blockSize = LTOBlockSizes.GetBlockSize(backup.LTOGen);
 
+      // Step 1: Find Descriptor on Tape
       var descriptor = FindDescriptor(blockSize);
       if (descriptor != null) {
         var json = JsonSerializer.Serialize(descriptor, new JsonSerializerOptions { WriteIndented = true });
-        Console.WriteLine(json);
+        _logger.LogInformation(json);
       }
 
       if (descriptor == null) {
-        Console.WriteLine("Descriptor not found on tape.");
+        _logger.LogInformation("Descriptor not found on tape.");
         return;
       }
 
-      // Step 3: Test restore from tape
+      // Step 2: Restore Files to Directory
       RestoreDirectory(descriptor, backup.Destination);
-      Console.WriteLine("Restore completed.");
-      return; // Go back to the main menu after completing the restore
+      _logger.LogInformation("Restore completed.");
+      return;
     }
   }
 }
-

src/MaksIT.LTO.Backup/MaksIT.LTO.Backup.csproj

@@ -5,8 +5,17 @@
     <TargetFramework>net8.0</TargetFramework>
     <ImplicitUsings>enable</ImplicitUsings>
     <Nullable>enable</Nullable>
+    <Version>0.0.2</Version>
   </PropertyGroup>
 
+  <ItemGroup>
+    <PackageReference Include="Microsoft.Extensions.Configuration" Version="8.0.0" />
+    <PackageReference Include="Microsoft.Extensions.Configuration.Json" Version="8.0.1" />
+    <PackageReference Include="Microsoft.Extensions.DependencyInjection" Version="8.0.1" />
+    <PackageReference Include="Microsoft.Extensions.Logging" Version="8.0.1" />
+    <PackageReference Include="Microsoft.Extensions.Logging.Console" Version="8.0.1" />
+  </ItemGroup>
+
   <ItemGroup>
     <ProjectReference Include="..\MaksIT.LTO.Core\MaksIT.LTO.Core.csproj" />
   </ItemGroup>
@@ -15,6 +24,9 @@
     <None Update="configuration.json">
       <CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory>
     </None>
+    <None Update="secret.txt">
+      <CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory>
+    </None>
   </ItemGroup>
 
 </Project>

src/MaksIT.LTO.Backup/Program.cs

@@ -1,65 +1,35 @@
-namespace MaksIT.LTO.Backup;
+using Microsoft.Extensions.Logging;
+using Microsoft.Extensions.Configuration;
+using Microsoft.Extensions.DependencyInjection;
+using MaksIT.LTO.Core.Logging;
+
+namespace MaksIT.LTO.Backup;
 
 class Program {
 
   public static void Main() {
 
-    var app = new Application();
+    // Set up configuration with reload support
+    var configuration = new ConfigurationBuilder()
+        .SetBasePath(Directory.GetCurrentDirectory())
+        .AddJsonFile("configuration.json", optional: false, reloadOnChange: true)  // Enable reload on change
+        .Build();
 
-    Console.OutputEncoding = System.Text.Encoding.UTF8;
+    var serviceProvider = new ServiceCollection()
+        .Configure<Configuration>(configuration.GetSection("Configuration")) // Bind AppConfig directly
+        .AddSingleton(configuration) // Make IConfiguration available if needed
+        .AddLogging(builder =>
+        {
+          builder.AddConfiguration(configuration.GetSection("Logging"));
+          builder.AddConsole();
+          builder.AddFile(Path.Combine(Directory.GetCurrentDirectory(), "log.txt"));
+        })
+        .AddTransient<Application>()
+        .BuildServiceProvider();
 
-    while (true) {
-      // Console.Clear();
-      Console.WriteLine("MaksIT.LTO.Backup v0.0.1");
-      Console.WriteLine("© Maksym Sadovnychyy (MAKS-IT) 2024");
 
-      Console.WriteLine("\nSelect an action:");
-      Console.WriteLine("1. Load tape");
-      Console.WriteLine("2. Backup");
-      Console.WriteLine("3. Restore");
-      Console.WriteLine("4. Eject tape");
-      Console.WriteLine("5. Get device status");
-      Console.WriteLine("6. Tape Erase (Short)");
-      Console.WriteLine("7. Reload configurations");
-      Console.WriteLine("8. Exit");
-      Console.Write("Enter your choice: ");
-
-      var choice = Console.ReadLine();
-
-      try {
-        switch (choice) {
-          case "1":
-            app.LoadTape();
-            break;
-          case "2":
-            app.Backup();
-            break;
-          case "3":
-            app.Restore();
-            break;
-          case "4":
-            app.EjectTape();
-            break;
-          case "5":
-            app.GetDeviceStatus();
-            break;
-          case "6":
-            app.TapeErase();
-            break;
-          case "7":
-            app.LoadConfiguration();
-            break;
-          case "8":
-            Console.WriteLine("Exiting...");
-            return;
-          default:
-            Console.WriteLine("Invalid choice. Please try again.");
-            break;
-        }
-      }
-      catch (Exception ex) {
-        Console.WriteLine($"An error occurred: {ex.Message}");
-      }
-    }
+    // Get the App service and run it
+    var app = serviceProvider.GetRequiredService<Application>();
+    app.Run();
   }
 }

src/MaksIT.LTO.Backup/configuration.json

@@ -1,41 +1,51 @@
 {
-  "TapePath": "\\\\.\\Tape0",
-  "WriteDelay": 100,
-  "Backups": [
-    {
-      "Name": "Normal test",
-      "Barcode": "",
-      "LTOGen": "LTO5",
-      "Source": {
-        "LocalPath": {
-          "Path": "F:\\LTO\\Backup"
-        }
-      },
-      "Destination": {
-        "LocalPath": {
-          "Path": "F:\\LTO\\Restore"
-        }
-      }
-    },
-    {
-      "Name": "Network test",
-      "Barcode": "",
-      "LTOGen": "LTO5",
-      "Source": {
-        "RemotePath": {
-          "Path": "\\\\nassrv0002.corp.maks-it.com\\data-1\\Users",
-          "PasswordCredentials": {
-            "Username": "",
-            "Password": ""
-          },
-          "Protocol": "SMB"
-        }
-      },
-      "Destination": {
-        "LocalPath": {
-          "Path": "F:\\LTO\\Restore"
-        }
-      }
+  "Logging": {
+    "LogLevel": {
+      "Default": "Information",
+      "Microsoft": "Warning",
+      "Microsoft.Hosting.Lifetime": "Information"
     }
-  ]
+  },
+
+  "Configuration": {
+    "TapePath": "\\\\.\\Tape0",
+    "WriteDelay": 100,
+    "Backups": [
+      {
+        "Name": "Normal test",
+        "Barcode": "",
+        "LTOGen": "LTO5",
+        "Source": {
+          "LocalPath": {
+            "Path": "F:\\LTO\\Backup"
+          }
+        },
+        "Destination": {
+          "LocalPath": {
+            "Path": "F:\\LTO\\Restore"
+          }
+        }
+      },
+      {
+        "Name": "Network test",
+        "Barcode": "",
+        "LTOGen": "LTO5",
+        "Source": {
+          "RemotePath": {
+            "Path": "\\\\nassrv0002.corp.maks-it.com\\data-1\\Users",
+            "PasswordCredentials": {
+              "Username": "",
+              "Password": ""
+            },
+            "Protocol": "SMB"
+          }
+        },
+        "Destination": {
+          "LocalPath": {
+            "Path": "F:\\LTO\\Restore"
+          }
+        }
+      }
+    ]
+  }
 }

src/MaksIT.LTO.Core/Crc32.cs

@@ -0,0 +1,87 @@
+using System.Security.Cryptography;
+
+
+namespace MaksIT.LTO.Core;
+
+public class Crc32 : HashAlgorithm {
+  public const uint DefaultPolynomial = 0xedb88320;
+  public const uint DefaultSeed = 0xffffffff;
+
+  private static uint[]? defaultTable;
+
+  private readonly uint seed;
+  private readonly uint[] table;
+  private uint hash;
+
+  public Crc32()
+      : this(DefaultPolynomial, DefaultSeed) {
+  }
+
+  public Crc32(uint polynomial, uint seed) {
+    table = InitializeTable(polynomial);
+    this.seed = hash = seed;
+  }
+
+  public override void Initialize() {
+    hash = seed;
+  }
+
+  protected override void HashCore(byte[] buffer, int start, int length) {
+    hash = CalculateHash(table, hash, buffer, start, length);
+  }
+
+  protected override byte[] HashFinal() {
+    var hashBuffer = UInt32ToBigEndianBytes(~hash);
+    HashValue = hashBuffer;
+    return hashBuffer;
+  }
+
+  public override int HashSize => 32;
+
+  public static uint Compute(byte[] buffer) {
+    return Compute(DefaultPolynomial, DefaultSeed, buffer);
+  }
+
+  public static uint Compute(uint seed, byte[] buffer) {
+    return Compute(DefaultPolynomial, seed, buffer);
+  }
+
+  public static uint Compute(uint polynomial, uint seed, byte[] buffer) {
+    return ~CalculateHash(InitializeTable(polynomial), seed, buffer, 0, buffer.Length);
+  }
+
+  private static uint[] InitializeTable(uint polynomial) {
+    if (polynomial == DefaultPolynomial && defaultTable != null)
+      return defaultTable;
+
+    var createTable = new uint[256];
+    for (var i = 0; i < 256; i++) {
+      var entry = (uint)i;
+      for (var j = 0; j < 8; j++)
+        if ((entry & 1) == 1)
+          entry = (entry >> 1) ^ polynomial;
+        else
+          entry >>= 1;
+      createTable[i] = entry;
+    }
+
+    if (polynomial == DefaultPolynomial)
+      defaultTable = createTable;
+
+    return createTable;
+  }
+
+  private static uint CalculateHash(uint[] table, uint seed, byte[] buffer, int start, int size) {
+    var crc = seed;
+    for (var i = start; i < size - start; i++)
+      crc = (crc >> 8) ^ table[buffer[i] ^ crc & 0xff];
+    return crc;
+  }
+
+  private static byte[] UInt32ToBigEndianBytes(uint x) => [
+    (byte)((x >> 24) & 0xff),
+    (byte)((x >> 16) & 0xff),
+    (byte)((x >> 8) & 0xff),
+    (byte)(x & 0xff)
+  ];
+}

src/MaksIT.LTO.Core/DriverManager.cs

@@ -1,37 +0,0 @@
-using System;
-using System.Collections.Generic;
-using System.Diagnostics;
-using System.Linq;
-using System.Text;
-using System.Threading.Tasks;
-
-namespace MaksIT.LTO.Core;
-public class DriverManager {
-  public static void RestartDriver(string deviceName) {
-    string script = $@"
-            $device = Get-PnpDevice -FriendlyName '{deviceName}'
-            Disable-PnpDevice -InstanceId $device.InstanceId -Confirm:$false
-            Start-Sleep -Seconds 5
-            Enable-PnpDevice -InstanceId $device.InstanceId -Confirm:$false
-        ";
-
-    ProcessStartInfo psi = new ProcessStartInfo {
-      FileName = "powershell.exe",
-      Arguments = $"-NoProfile -ExecutionPolicy Bypass -Command \"{script}\"",
-      UseShellExecute = false,
-      RedirectStandardOutput = true,
-      RedirectStandardError = true
-    };
-
-    using (Process process = Process.Start(psi)) {
-      string output = process.StandardOutput.ReadToEnd();
-      string error = process.StandardError.ReadToEnd();
-      process.WaitForExit();
-
-      Console.WriteLine(output);
-      if (!string.IsNullOrEmpty(error)) {
-        Console.WriteLine($"Error: {error}");
-      }
-    }
-  }
-}

src/MaksIT.LTO.Core/LTOBlockSizes.cs

@@ -1,60 +0,0 @@
-
-namespace MaksIT.LTO.Core;
-
-public static class LTOBlockSizes {
-  public const uint LTO1 = 65536;    // 64 KB
-  public const uint LTO2 = 65536;    // 64 KB
-  public const uint LTO3 = 131072;   // 128 KB
-  public const uint LTO4 = 131072;   // 128 KB
-  public const uint LTO5 = 262144;   // 256 KB
-  public const uint LTO6 = 262144;   // 256 KB
-  public const uint LTO7 = 524288;   // 512 KB
-  public const uint LTO8 = 524288;   // 512 KB
-  public const uint LTO9 = 1048576;  // 1 MB
-
-  // Dictionary to store the total capacity for each LTO generation (in bytes)
-  private static readonly Dictionary<string, ulong> TapeCapacities = new Dictionary<string, ulong>
-  {
-    { "LTO1", 100UL * 1024 * 1024 * 1024 },  // 100 GB
-    { "LTO2", 200UL * 1024 * 1024 * 1024 },  // 200 GB
-    { "LTO3", 400UL * 1024 * 1024 * 1024 },  // 400 GB
-    { "LTO4", 800UL * 1024 * 1024 * 1024 },  // 800 GB
-    { "LTO5", 1500UL * 1024 * 1024 * 1024 }, // 1.5 TB
-    { "LTO6", 2500UL * 1024 * 1024 * 1024 }, // 2.5 TB
-    { "LTO7", 6000UL * 1024 * 1024 * 1024 }, // 6 TB
-    { "LTO8", 12000UL * 1024 * 1024 * 1024 },// 12 TB
-    { "LTO9", 18000UL * 1024 * 1024 * 1024 } // 18 TB
-  };
-
-  // Method to get the block size for a given LTO generation
-  // Method to get the block size for a given LTO generation
-  public static uint GetBlockSize(string ltoGen) {
-    return ltoGen switch {
-      "LTO1" => LTO1,
-      "LTO2" => LTO2,
-      "LTO3" => LTO3,
-      "LTO4" => LTO4,
-      "LTO5" => LTO5,
-      "LTO6" => LTO6,
-      "LTO7" => LTO7,
-      "LTO8" => LTO8,
-      "LTO9" => LTO9,
-      _ => throw new ArgumentException("Invalid LTO generation")
-    };
-  }
-
-  // Method to get the total capacity for a given LTO generation
-  public static ulong GetTapeCapacity(string ltoGen) {
-    if (TapeCapacities.TryGetValue(ltoGen, out var capacity)) {
-      return capacity;
-    }
-    throw new ArgumentException("Invalid LTO generation");
-  }
-
-  // Method to calculate the maximum number of blocks that can be written on the tape
-  public static ulong GetMaxBlocks(string ltoGen) {
-    var blockSize = GetBlockSize(ltoGen);
-    var tapeCapacity = GetTapeCapacity(ltoGen);
-    return tapeCapacity / blockSize;
-  }
-}

src/MaksIT.LTO.Core/Logging/FileLogger.cs

@@ -0,0 +1,37 @@
+using Microsoft.Extensions.Logging;
+
+
+namespace MaksIT.LTO.Core.Logging;
+
+public class FileLogger : ILogger {
+  private readonly string _filePath;
+  private readonly object _lock = new object();
+
+  public FileLogger(string filePath) {
+    _filePath = filePath;
+  }
+
+  public IDisposable? BeginScope<TState>(TState state) where TState : notnull => null;
+
+  public bool IsEnabled(LogLevel logLevel) {
+    return logLevel != LogLevel.None;
+  }
+
+  public void Log<TState>(LogLevel logLevel, EventId eventId, TState state, Exception? exception, Func<TState, Exception?, string> formatter) {
+    if (!IsEnabled(logLevel))
+      return;
+
+    var message = formatter(state, exception);
+    if (string.IsNullOrEmpty(message))
+      return;
+
+    var logRecord = $"{DateTime.Now:yyyy-MM-dd HH:mm:ss} [{logLevel}] {message}";
+    if (exception != null) {
+      logRecord += Environment.NewLine + exception;
+    }
+
+    lock (_lock) {
+      File.AppendAllText(_filePath, logRecord + Environment.NewLine);
+    }
+  }
+}

src/MaksIT.LTO.Core/Logging/FileLoggerProvider.cs

@@ -0,0 +1,19 @@
+using Microsoft.Extensions.Logging;
+
+
+namespace MaksIT.LTO.Core.Logging;
+
+[ProviderAlias("FileLogger")]
+public class FileLoggerProvider : ILoggerProvider {
+  private readonly string _filePath;
+
+  public FileLoggerProvider(string filePath) {
+    _filePath = filePath ?? throw new ArgumentNullException(nameof(filePath));
+  }
+
+  public ILogger CreateLogger(string categoryName) {
+    return new FileLogger(_filePath);
+  }
+
+  public void Dispose() { }
+}

src/MaksIT.LTO.Core/Logging/LoggingBuilderExtensions.cs

@@ -0,0 +1,12 @@
+using Microsoft.Extensions.DependencyInjection;
+using Microsoft.Extensions.Logging;
+
+
+namespace MaksIT.LTO.Core.Logging;
+
+public static class LoggingBuilderExtensions {
+  public static ILoggingBuilder AddFile(this ILoggingBuilder builder, string filePath) {
+    builder.Services.AddSingleton<ILoggerProvider>(new FileLoggerProvider(filePath));
+    return builder;
+  }
+}

src/MaksIT.LTO.Core/MaksIT.LTO.Core.csproj

@@ -5,6 +5,12 @@
     <ImplicitUsings>enable</ImplicitUsings>
     <Nullable>enable</Nullable>
     <DefineConstants>NTDDI_VERSION_05010000;NTDDI_WINXP_05010000</DefineConstants>
+    <Version>0.0.2</Version>
   </PropertyGroup>
 
+  <ItemGroup>
+    <PackageReference Include="Microsoft.Extensions.Logging" Version="8.0.1" />
+    <PackageReference Include="Microsoft.Extensions.Logging.Abstractions" Version="8.0.2" />
+  </ItemGroup>
+
 </Project>

src/MaksIT.LTO.Core/MassStorage/LTOBlockSizes.cs

@@ -0,0 +1,65 @@
+namespace MaksIT.LTO.Core.MassStorage;
+
+public static class LTOBlockSizes
+{
+    public const uint LTO1 = 65536;    // 64 KB
+    public const uint LTO2 = 65536;    // 64 KB
+    public const uint LTO3 = 131072;   // 128 KB
+    public const uint LTO4 = 131072;   // 128 KB
+    public const uint LTO5 = 262144;   // 256 KB
+    public const uint LTO6 = 262144;   // 256 KB
+    public const uint LTO7 = 524288;   // 512 KB
+    public const uint LTO8 = 524288;   // 512 KB
+    public const uint LTO9 = 1048576;  // 1 MB
+
+    // Dictionary to store the total capacity for each LTO generation (in bytes)
+    private static readonly Dictionary<string, ulong> TapeCapacities = new Dictionary<string, ulong>
+  {
+    { "LTO1", 100UL * 1024 * 1024 * 1024 },  // 100 GB
+    { "LTO2", 200UL * 1024 * 1024 * 1024 },  // 200 GB
+    { "LTO3", 400UL * 1024 * 1024 * 1024 },  // 400 GB
+    { "LTO4", 800UL * 1024 * 1024 * 1024 },  // 800 GB
+    { "LTO5", 1500UL * 1024 * 1024 * 1024 }, // 1.5 TB
+    { "LTO6", 2500UL * 1024 * 1024 * 1024 }, // 2.5 TB
+    { "LTO7", 6000UL * 1024 * 1024 * 1024 }, // 6 TB
+    { "LTO8", 12000UL * 1024 * 1024 * 1024 },// 12 TB
+    { "LTO9", 18000UL * 1024 * 1024 * 1024 } // 18 TB
+  };
+
+    // Method to get the block size for a given LTO generation
+    // Method to get the block size for a given LTO generation
+    public static uint GetBlockSize(string ltoGen)
+    {
+        return ltoGen switch
+        {
+            "LTO1" => LTO1,
+            "LTO2" => LTO2,
+            "LTO3" => LTO3,
+            "LTO4" => LTO4,
+            "LTO5" => LTO5,
+            "LTO6" => LTO6,
+            "LTO7" => LTO7,
+            "LTO8" => LTO8,
+            "LTO9" => LTO9,
+            _ => throw new ArgumentException("Invalid LTO generation")
+        };
+    }
+
+    // Method to get the total capacity for a given LTO generation
+    public static ulong GetTapeCapacity(string ltoGen)
+    {
+        if (TapeCapacities.TryGetValue(ltoGen, out var capacity))
+        {
+            return capacity;
+        }
+        throw new ArgumentException("Invalid LTO generation");
+    }
+
+    // Method to calculate the maximum number of blocks that can be written on the tape
+    public static ulong GetMaxBlocks(string ltoGen)
+    {
+        var blockSize = GetBlockSize(ltoGen);
+        var tapeCapacity = GetTapeCapacity(ltoGen);
+        return tapeCapacity / blockSize;
+    }
+}

src/MaksIT.LTO.Core/MassStorage/TapeDeviceHandler.cs

@@ -1,6 +1,7 @@
 using System;
 using System.Diagnostics.CodeAnalysis;
 using System.Runtime.InteropServices;
+using Microsoft.Extensions.Logging;
 using Microsoft.Win32.SafeHandles;
 
 // https://github.com/tpn/winsdk-10
@@ -8,11 +9,11 @@ using Microsoft.Win32.SafeHandles;
 namespace MaksIT.LTO.Core;
 
 public partial class TapeDeviceHandler : IDisposable {
+
+  private readonly ILogger<TapeDeviceHandler> _logger;
   private string _tapeDevicePath;
   private SafeFileHandle _tapeHandle;
 
-
-
   private const uint GENERIC_READ = 0x80000000;
   private const uint GENERIC_WRITE = 0x40000000;
   private const uint OPEN_EXISTING = 3;
@@ -67,7 +68,11 @@ public partial class TapeDeviceHandler : IDisposable {
       out uint lpNumberOfBytesRead,
       IntPtr lpOverlapped);
 
-  public TapeDeviceHandler(string tapeDevicePath) {
+  public TapeDeviceHandler(
+    ILogger<TapeDeviceHandler> logger,
+    string tapeDevicePath
+  ) {
+    _logger = logger;
     _tapeDevicePath = tapeDevicePath;
     OpenTapeDevice(GENERIC_READ | GENERIC_WRITE);
   }

src/MaksIT.LTO.Core/Networking/NetworkConnection.cs

@@ -1,27 +1,23 @@
 using System.Net;
 using System.Runtime.InteropServices;
 
-namespace MaksIT.LTO.Core;
+using Microsoft.Extensions.Logging;
 
-//public void RestoreFilesFromSmbShare(string smbPath, string username, string password, string domain, string restoreDirectory) {
-//  var credentials = new NetworkCredential(username, password, domain);
-//  using (new NetworkConnection(smbPath, credentials)) {
-//    var files = Directory.GetFiles(smbPath, "*.*", SearchOption.AllDirectories);
-//    foreach (var file in files) {
-//      var relativePath = Path.GetRelativePath(smbPath, file);
-//      var destinationPath = Path.Combine(restoreDirectory, relativePath);
-//      Directory.CreateDirectory(Path.GetDirectoryName(destinationPath));
-//      File.Copy(file, destinationPath, overwrite: true);
-//      Console.WriteLine($"Restored file: {file} to {destinationPath}");
-//    }
-//  }
-//}
 
+namespace MaksIT.LTO.Core.Networking;
 
 public class NetworkConnection : IDisposable {
+
+  private readonly ILogger<NetworkConnection> _logger;
   private readonly string _networkName;
 
-  public NetworkConnection(string networkName, NetworkCredential credentials) {
+  public NetworkConnection(
+    ILogger<NetworkConnection> logger,
+    string networkName,
+    NetworkCredential credentials) {
+
+    _logger = logger;
+
     _networkName = networkName;
 
     var netResource = new NetResource {

src/MaksIT.LTO.Core/Utilities/AESGCMUtility.cs

@@ -0,0 +1,57 @@
+using System.Security.Cryptography;
+
+
+namespace MaksIT.LTO.Core.Utilities;
+
+public static class AESGCMUtility {
+  private const int IvLength = 12; // 12 bytes for AES-GCM IV
+  private const int TagLength = 16; // 16 bytes for AES-GCM Tag
+
+  public static byte[] EncryptData(byte[] data, string base64Key) {
+    var key = Convert.FromBase64String(base64Key);
+    using (AesGcm aesGcm = new AesGcm(key, AesGcm.TagByteSizes.MaxSize)) {
+      var iv = new byte[IvLength];
+      RandomNumberGenerator.Fill(iv);
+
+      var cipherText = new byte[data.Length];
+      var tag = new byte[TagLength];
+
+      aesGcm.Encrypt(iv, data, cipherText, tag);
+
+      // Concatenate cipherText, tag, and iv
+      var result = new byte[cipherText.Length + tag.Length + iv.Length];
+      Buffer.BlockCopy(cipherText, 0, result, 0, cipherText.Length);
+      Buffer.BlockCopy(tag, 0, result, cipherText.Length, tag.Length);
+      Buffer.BlockCopy(iv, 0, result, cipherText.Length + tag.Length, iv.Length);
+
+      return result;
+    }
+  }
+
+  public static byte[] DecryptData(byte[] data, string base64Key) {
+    var key = Convert.FromBase64String(base64Key);
+
+    // Extract cipherText, tag, and iv
+    var cipherTextLength = data.Length - IvLength - TagLength;
+
+    var cipherText = new byte[cipherTextLength];
+    var tag = new byte[TagLength];
+    var iv = new byte[IvLength];
+
+    Buffer.BlockCopy(data, 0, cipherText, 0, cipherTextLength);
+    Buffer.BlockCopy(data, cipherTextLength, tag, 0, TagLength);
+    Buffer.BlockCopy(data, cipherTextLength + TagLength, iv, 0, IvLength);
+
+    using (AesGcm aesGcm = new AesGcm(key, AesGcm.TagByteSizes.MaxSize)) {
+      var decryptedData = new byte[cipherText.Length];
+      aesGcm.Decrypt(iv, cipherText, tag, decryptedData);
+      return decryptedData;
+    }
+  }
+
+  public static string GenerateKeyBase64() {
+    var key = new byte[32]; // 256-bit key for AES-256
+    RandomNumberGenerator.Fill(key);
+    return Convert.ToBase64String(key);
+  }
+}

src/MaksIT.LTO.Core/Utilities/ChecksumUtility.cs

@@ -0,0 +1,45 @@
+namespace MaksIT.LTO.Core.Helpers;
+
+public static class ChecksumUtility {
+  public static string CalculateCRC32Checksum(byte[] data) {
+    using var crc32 = new Crc32();
+    var hashBytes = crc32.ComputeHash(data);
+    return BitConverter.ToString(hashBytes).Replace("-", "").ToLower();
+  }
+
+  public static string CalculateCRC32ChecksumFromFile(string filePath) {
+    using var crc32 = new Crc32();
+    using var stream = File.OpenRead(filePath);
+    var hashBytes = crc32.ComputeHash(stream);
+    return BitConverter.ToString(hashBytes).Replace("-", "").ToLower();
+  }
+
+  public static string CalculateCRC32ChecksumFromFileInChunks(string filePath, int chunkSize = 8192) {
+    using var crc32 = new Crc32();
+    using var stream = File.OpenRead(filePath);
+    var buffer = new byte[chunkSize];
+    int bytesRead;
+    while ((bytesRead = stream.Read(buffer, 0, buffer.Length)) > 0) {
+      crc32.TransformBlock(buffer, 0, bytesRead, null, 0);
+    }
+    crc32.TransformFinalBlock(buffer, 0, 0);
+    var hashBytes = crc32.Hash;
+    return BitConverter.ToString(hashBytes ?? Array.Empty<byte>()).Replace("-", "").ToLower();
+  }
+
+  public static bool VerifyCRC32Checksum(byte[] data, string expectedChecksum) {
+    var calculatedChecksum = CalculateCRC32Checksum(data);
+    return string.Equals(calculatedChecksum, expectedChecksum, StringComparison.OrdinalIgnoreCase);
+  }
+
+  public static bool VerifyCRC32ChecksumFromFile(string filePath, string expectedChecksum) {
+    var calculatedChecksum = CalculateCRC32ChecksumFromFile(filePath);
+    return string.Equals(calculatedChecksum, expectedChecksum, StringComparison.OrdinalIgnoreCase);
+  }
+
+  public static bool VerifyCRC32ChecksumFromFileInChunks(string filePath, string expectedChecksum, int chunkSize = 8192) {
+    var calculatedChecksum = CalculateCRC32ChecksumFromFileInChunks(filePath, chunkSize);
+    return string.Equals(calculatedChecksum, expectedChecksum, StringComparison.OrdinalIgnoreCase);
+  }
+}
+

src/MaksIT.LTO.Core/Utilities/PaddingUtility.cs

@@ -0,0 +1,38 @@
+namespace MaksIT.LTO.Core.Utilities;
+
+public static class PaddingUtility {
+
+  private const byte _specialByte = 0x80;
+
+  public static byte[] AddPadding(byte[] data, int blockSize) {
+    var paddingSize = blockSize - (data.Length % blockSize);
+    if (paddingSize == blockSize) {
+      paddingSize = 0;  // no padding needed if already aligned
+    }
+
+    var paddedData = new byte[data.Length + paddingSize];
+    Array.Copy(data, paddedData, data.Length);
+
+    // fill the padding bytes with specialBytes
+    for (var i = data.Length; i < paddedData.Length; i++) {
+      paddedData[i] = _specialByte;
+    }
+
+    return paddedData;
+  }
+
+  public static byte[] RemovePadding(byte[] paddedData, int blockSize) {
+    var originalLength = paddedData.Length;
+
+    // find the original length by checking for the padding byte 0x80
+    while (originalLength > 0 && paddedData[originalLength - 1] == _specialByte) {
+      originalLength--;
+    }
+
+    // create a new array to hold the unpadded data
+    var unpaddedData = new byte[originalLength];
+    Array.Copy(paddedData, unpaddedData, originalLength);
+
+    return unpaddedData;
+  }
+}