Add strong name signing option '--strong-name-key' #229

Author: sunnamed434

src/BitMono.CLI/Modules/Options.cs

@@ -28,4 +28,7 @@ internal class Options
 
     [Option("no-watermark", Required = false, HelpText = "Disable watermarking (overrides obfuscation.json setting).")]
     public bool NoWatermark { get; set; }
+
+    [Option("strong-name-key", Required = false, HelpText = "Path to strong name key (.snk) file for assembly signing.")]
+    public string? StrongNameKey { get; set; }
 }

src/BitMono.CLI/Modules/OptionsObfuscationNeedsFactory.cs

@@ -42,6 +42,11 @@ public OptionsObfuscationNeedsFactory(string[] args)
             {
                 obfuscationSettings.Watermark = false;
             }
+            
+            if (obfuscationSettings != null)
+            {
+                obfuscationSettings.StrongNameKeyFile = options.StrongNameKey;
+            }
         }
         catch (Exception ex)
         {

src/BitMono.Obfuscation/BitMonoObfuscator.cs

@@ -283,6 +283,15 @@ private async Task<bool> WriteModuleAsync()
             fileBuilder
                 .CreateFile(_imageBuild!.ConstructedImage!)
                 .Write(memoryStream);
+            
+            if (!string.IsNullOrEmpty(_obfuscationSettings.StrongNameKeyFile) && File.Exists(_obfuscationSettings.StrongNameKeyFile))
+            {
+                _logger.Information("Applying strong name signature to PE image");
+                var privateKey = StrongNamePrivateKey.FromFile(_obfuscationSettings.StrongNameKeyFile);
+                var signer = new StrongNameSigner(privateKey);
+                signer.SignImage(memoryStream, _context.Module.Assembly.HashAlgorithm);
+            }
+            
             await _dataWriter.WriteAsync(_context.BitMonoContext.OutputFile, memoryStream.ToArray());
             _logger.Information("The protected module was saved in {0}", _context.BitMonoContext.OutputDirectoryName);
         }

src/BitMono.Obfuscation/GlobalUsings.cs

@@ -14,10 +14,12 @@
 global using System.Text;
 global using System.Threading;
 global using System.Threading.Tasks;
+global using AsmResolver.DotNet.Signatures;
 global using AsmResolver.PE;
 global using AsmResolver.PE.Builder;
 global using AsmResolver.PE.DotNet;
 global using AsmResolver.PE.DotNet.ReadyToRun;
+global using AsmResolver.PE.DotNet.StrongName;
 global using AsmResolver.PE.File;
 global using BitMono.API;
 global using BitMono.API.Protections;
@@ -33,6 +35,7 @@
 global using BitMono.Obfuscation.Protections;
 global using BitMono.Obfuscation.Referencing;
 global using BitMono.Obfuscation.Stripping;
+global using BitMono.Runtime;
 global using BitMono.Utilities.AsmResolver;
 global using BitMono.Utilities.Runtime;
 global using BitMono.Utilities.Time;

src/BitMono.Obfuscation/Modules/ModuleFactory.cs

@@ -46,9 +46,24 @@ public ModuleFactoryResult Create()
             throw;
         }
 
-        var managedPEImageBuilder =
-            new ManagedPEImageBuilder(new DotNetDirectoryFactory(_metadataBuilderFlags), _errorListener);
-
+        var factory = new DotNetDirectoryFactory(_metadataBuilderFlags);
+        
+        if (!string.IsNullOrEmpty(_obfuscationSettings.StrongNameKeyFile))
+        {
+            if (File.Exists(_obfuscationSettings.StrongNameKeyFile))
+            {
+                _logger.Information("Loading strong name key from: {KeyFile}", _obfuscationSettings.StrongNameKeyFile);
+                var privateKey = StrongNamePrivateKey.FromFile(_obfuscationSettings.StrongNameKeyFile);
+                factory.StrongNamePrivateKey = privateKey;
+                _logger.Information("Configured DotNetDirectoryFactory with strong name private key");
+            }
+            else
+            {
+                _logger.Error("Strong name key file not found: {KeyFile}", _obfuscationSettings.StrongNameKeyFile);
+            }
+        }
+        
+        var managedPEImageBuilder = new ManagedPEImageBuilder(factory, _errorListener);
         return new ModuleFactoryResult
         {
             Module = module,

src/BitMono.Shared/Models/ObfuscationSettings.cs

@@ -22,4 +22,5 @@ public class ObfuscationSettings
     public bool SpecificNamespacesObfuscationOnly { get; set; }
     public string[]? SpecificNamespaces { get; set; }
     public string[]? RandomStrings { get; set; }
+    public string? StrongNameKeyFile { get; set; }
 }