First commit

2026-01-09 00:25:03 +01:00 · 2024-11-03 18:29:07 +03:00
commit db63ef6ff2
41 changed files with 1742 additions and 0 deletions
@@ -0,0 +1,89 @@
+using System.Diagnostics;
+using ivinject.Common.Models;
+using ivinject.Features.Injection.Models;
+using RegularExpressions = ivinject.Features.Injection.Models.RegularExpressions;
+
+namespace ivinject.Features.Injection;
+
+internal static class DependencyExtensions
+{
+    internal static async Task<string[]> GetSharedLibraries(this IviMachOBinary binary)
+    {
+        using var process = Process.Start(
+            new ProcessStartInfo
+            {
+                FileName = "otool",
+                Arguments = $"-L {binary.FullName}",
+                RedirectStandardOutput = true
+            }
+        );
+        
+        var output = await process!.StandardOutput.ReadToEndAsync();
+        var matches = RegularExpressions.OToolSharedLibrary().Matches(output);
+        
+        // the first result is actually LC_ID_DYLIB, not LC_LOAD_DYLIB
+        return matches.Select(match => match.Groups[1].Value).ToArray()[1..];
+    }
+    
+    internal static async Task ChangeDependency(
+        this IviMachOBinary binary,
+        string oldPath,
+        string newPath
+    )
+    {
+        using var process = Process.Start(
+            new ProcessStartInfo
+            {
+                FileName = "install_name_tool",
+                Arguments = $"-change {oldPath} {newPath} {binary.FullName}",
+                RedirectStandardError = true
+            }
+        );
+        
+        await process!.WaitForExitAsync();
+    }
+    
+    internal static async Task<bool> AddRunPath(
+        this IviMachOBinary binary,
+        string rPath
+    )
+    {
+        using var process = Process.Start(
+            new ProcessStartInfo
+            {
+                FileName = "install_name_tool",
+                Arguments = $"-add_rpath {rPath} {binary.FullName}",
+                RedirectStandardOutput = true
+            }
+        );
+            
+        await process!.WaitForExitAsync();
+        return process.ExitCode == 0;
+    }
+    
+    internal static async Task InsertDependency(
+        this IviMachOBinary binary,
+        string dependency
+    )
+    {
+        using var process = Process.Start(
+            new ProcessStartInfo
+            {
+                FileName = "insert-dylib",
+                Arguments = $"{dependency} {binary.FullName} --all-yes --inplace",
+                RedirectStandardOutput = true
+            }
+        );
+            
+        await process!.WaitForExitAsync();
+    }
+
+    internal static async Task<IEnumerable<string>> AllDependencies(this List<IviCopiedBinary> copiedBinaries)
+    {
+        return (await Task.WhenAll(
+            copiedBinaries.Select(async binary => await binary.Binary.GetSharedLibraries())
+        ))
+        .SelectMany(dependencies => dependencies)
+        .Distinct();
+    }
+}