|
1 | 1 | using System; |
2 | 2 | using System.IO; |
| 3 | +using System.Reflection; |
3 | 4 | using System.Runtime.CompilerServices; |
4 | 5 | using System.Runtime.ConstrainedExecution; |
5 | 6 | using System.Runtime.InteropServices; |
@@ -29,32 +30,144 @@ static NativeMethods() |
29 | 30 | { |
30 | 31 | if (Platform.IsRunningOnNetFramework() || Platform.IsRunningOnNetCore()) |
31 | 32 | { |
32 | | - string nativeLibraryDir = GlobalSettings.GetAndLockNativeLibraryPath(); |
33 | | - if (nativeLibraryDir != null) |
| 33 | + // Use .NET Core 3.0+ NativeLibrary when available. |
| 34 | + if (!TryUseNativeLibrary()) |
34 | 35 | { |
35 | | - string nativeLibraryPath = Path.Combine(nativeLibraryDir, libgit2 + Platform.GetNativeLibraryExtension()); |
| 36 | + // NativeLibrary is not available, fall back. |
36 | 37 |
|
| 38 | + // Use GlobalSettings.NativeLibraryPath when set. |
37 | 39 | // Try to load the .dll from the path explicitly. |
38 | 40 | // If this call succeeds further DllImports will find the library loaded and not attempt to load it again. |
39 | 41 | // If it fails the next DllImport will load the library from safe directories. |
40 | | -#if NETFRAMEWORK |
41 | | - if (Platform.OperatingSystem == OperatingSystemType.Windows) |
42 | | -#else |
43 | | - if (RuntimeInformation.IsOSPlatform(OSPlatform.Windows)) |
44 | | -#endif |
| 42 | + string nativeLibraryPath = GetGlobalSettingsNativeLibraryPath(); |
| 43 | + if (nativeLibraryPath != null) |
45 | 44 | { |
46 | | - LoadWindowsLibrary(nativeLibraryPath); |
47 | | - } |
48 | | - else |
49 | | - { |
50 | | - LoadUnixLibrary(nativeLibraryPath, RTLD_NOW); |
| 45 | + if (RuntimeInformation.IsOSPlatform(OSPlatform.Windows)) |
| 46 | + { |
| 47 | + LoadWindowsLibrary(nativeLibraryPath); |
| 48 | + } |
| 49 | + else |
| 50 | + { |
| 51 | + LoadUnixLibrary(nativeLibraryPath, RTLD_NOW); |
| 52 | + } |
51 | 53 | } |
52 | 54 | } |
53 | 55 | } |
54 | 56 |
|
55 | 57 | InitializeNativeLibrary(); |
56 | 58 | } |
57 | 59 |
|
| 60 | + private static string GetGlobalSettingsNativeLibraryPath() |
| 61 | + { |
| 62 | + string nativeLibraryDir = GlobalSettings.GetAndLockNativeLibraryPath(); |
| 63 | + if (nativeLibraryDir == null) |
| 64 | + { |
| 65 | + return null; |
| 66 | + } |
| 67 | + return Path.Combine(nativeLibraryDir, libgit2 + Platform.GetNativeLibraryExtension()); |
| 68 | + } |
| 69 | + |
| 70 | + private delegate bool TryLoadLibraryByNameDelegate(string libraryName, Assembly assembly, DllImportSearchPath? searchPath, out IntPtr handle); |
| 71 | + private delegate bool TryLoadLibraryByPathDelegate(string libraryPath, out IntPtr handle); |
| 72 | + |
| 73 | + static TryLoadLibraryByNameDelegate _tryLoadLibraryByName; |
| 74 | + static TryLoadLibraryByPathDelegate _tryLoadLibraryByPath; |
| 75 | + |
| 76 | + static bool TryLoadLibrary(string libraryName, Assembly assembly, DllImportSearchPath? searchPath, out IntPtr handle) |
| 77 | + { |
| 78 | + if (_tryLoadLibraryByName == null) |
| 79 | + { |
| 80 | + throw new NotSupportedException(); |
| 81 | + } |
| 82 | + return _tryLoadLibraryByName(libraryName, assembly, searchPath, out handle); |
| 83 | + } |
| 84 | + |
| 85 | + static bool TryLoadLibrary(string libraryPath, out IntPtr handle) |
| 86 | + { |
| 87 | + if (_tryLoadLibraryByPath == null) |
| 88 | + { |
| 89 | + throw new NotSupportedException(); |
| 90 | + } |
| 91 | + return _tryLoadLibraryByPath(libraryPath, out handle); |
| 92 | + } |
| 93 | + |
| 94 | + private static bool TryUseNativeLibrary() |
| 95 | + { |
| 96 | + // NativeLibrary is available in .NET Core 3.0+. |
| 97 | + // We use reflection to use NativeLibrary so this library can target 'netstandard2.0'. |
| 98 | + |
| 99 | + Type dllImportResolverType = Type.GetType("System.Runtime.InteropServices.DllImportResolver, System.Runtime.InteropServices", throwOnError: false); |
| 100 | + Type nativeLibraryType = Type.GetType("System.Runtime.InteropServices.NativeLibrary, System.Runtime.InteropServices", throwOnError: false); |
| 101 | + var tryLoadLibraryByName = (TryLoadLibraryByNameDelegate)nativeLibraryType?.GetMethod("TryLoad", |
| 102 | + new Type[] { typeof(string), typeof(Assembly), typeof(DllImportSearchPath?), typeof(IntPtr).MakeByRefType() })?.CreateDelegate(typeof(TryLoadLibraryByNameDelegate)); |
| 103 | + var tryLoadLibraryByPath = (TryLoadLibraryByPathDelegate)nativeLibraryType?.GetMethod("TryLoad", |
| 104 | + new Type[] { typeof(string), typeof(IntPtr).MakeByRefType() })?.CreateDelegate(typeof(TryLoadLibraryByPathDelegate)); |
| 105 | + MethodInfo setDllImportResolver = nativeLibraryType?.GetMethod("SetDllImportResolver", new Type[] { typeof(Assembly), dllImportResolverType}); |
| 106 | + |
| 107 | + if (dllImportResolverType == null || |
| 108 | + nativeLibraryType == null || |
| 109 | + tryLoadLibraryByName == null || |
| 110 | + tryLoadLibraryByPath == null || |
| 111 | + setDllImportResolver == null) |
| 112 | + { |
| 113 | + return false; |
| 114 | + } |
| 115 | + |
| 116 | + _tryLoadLibraryByPath = tryLoadLibraryByPath; |
| 117 | + _tryLoadLibraryByName = tryLoadLibraryByName; |
| 118 | + |
| 119 | + // NativeMethods.SetDllImportResolver(typeof(NativeMethods).Assembly, ResolveDll); |
| 120 | + object resolveDelegate = typeof(NativeMethods).GetMethod(nameof(ResolveDll), BindingFlags.NonPublic | BindingFlags.Static).CreateDelegate(dllImportResolverType); |
| 121 | + setDllImportResolver.Invoke(null, new object[] { typeof(NativeMethods).Assembly, resolveDelegate }); |
| 122 | + |
| 123 | + return true; |
| 124 | + } |
| 125 | + |
| 126 | + private static IntPtr ResolveDll(string libraryName, Assembly assembly, DllImportSearchPath? searchPath) |
| 127 | + { |
| 128 | + IntPtr handle = IntPtr.Zero; |
| 129 | + if (libraryName == libgit2) |
| 130 | + { |
| 131 | + // Use GlobalSettings.NativeLibraryPath when set. |
| 132 | + string nativeLibraryPath = GetGlobalSettingsNativeLibraryPath(); |
| 133 | + if (nativeLibraryPath != null && |
| 134 | + TryLoadLibrary(nativeLibraryPath, out handle)) |
| 135 | + { |
| 136 | + return handle; |
| 137 | + } |
| 138 | + |
| 139 | + // Use Default DllImport resolution. |
| 140 | + if (TryLoadLibrary(libraryName, assembly, searchPath, out handle)) |
| 141 | + { |
| 142 | + return handle; |
| 143 | + } |
| 144 | + |
| 145 | + // We cary a number of .so files for Linux which are linked against various |
| 146 | + // libc/OpenSSL libraries. Try them out. |
| 147 | + if (RuntimeInformation.IsOSPlatform(OSPlatform.Linux)) |
| 148 | + { |
| 149 | + // The libraries are located at 'runtimes/<rid>/native/lib{libraryName}.so' |
| 150 | + // The <rid> ends with the processor architecture. e.g. fedora-x64. |
| 151 | + string assemblyDirectory = Path.GetDirectoryName(typeof(NativeMethods).Assembly.Location); |
| 152 | + string processorArchitecture = RuntimeInformation.ProcessArchitecture.ToString().ToLowerInvariant(); |
| 153 | + string runtimesDirectory = Path.Combine(assemblyDirectory, "runtimes"); |
| 154 | + |
| 155 | + if (Directory.Exists(runtimesDirectory)) |
| 156 | + { |
| 157 | + foreach (var runtimeFolder in Directory.GetDirectories(runtimesDirectory, $"*-{processorArchitecture}")) |
| 158 | + { |
| 159 | + string libPath = Path.Combine(runtimeFolder, "native", $"lib{libraryName}.so"); |
| 160 | + if (TryLoadLibrary(libPath, out handle)) |
| 161 | + { |
| 162 | + return handle; |
| 163 | + } |
| 164 | + } |
| 165 | + } |
| 166 | + } |
| 167 | + } |
| 168 | + return handle; |
| 169 | + } |
| 170 | + |
58 | 171 | public const int RTLD_NOW = 0x002; |
59 | 172 |
|
60 | 173 | [DllImport("libdl", EntryPoint = "dlopen")] |
|
0 commit comments