Handle self-contained application load

host/src/FunctionsNetHost/AppLoader/AppLoader.cs

@@ -1,10 +1,16 @@
 // Copyright (c) .NET Foundation. All rights reserved.
 // Licensed under the MIT License. See License.txt in the project root for license information.
 
+using System.Diagnostics;
 using System.Runtime.InteropServices;
 
 namespace FunctionsNetHost
 {
+    // If having problems with the managed host, enable the following:
+    // Environment.SetEnvironmentVariable("COREHOST_TRACE", "1");
+    // In Unix environment, you need to run the below command in the terminal to set the environment variable.
+    // export COREHOST_TRACE=1
+
     /// <summary>
     /// Manages loading hostfxr & worker assembly.
     /// </summary>
@@ -14,43 +20,36 @@ internal sealed class AppLoader : IDisposable
         private IntPtr _hostContextHandle = IntPtr.Zero;
         private bool _disposed;
 
-        internal AppLoader()
+        internal int RunApplication(string? assemblyPath)
         {
-            LoadHostfxrLibrary();
-        }
+            ArgumentNullException.ThrowIfNull(assemblyPath, nameof(assemblyPath));
 
-        private void LoadHostfxrLibrary()
-        {
-            // If having problems with the managed host, enable the following:
-            // Environment.SetEnvironmentVariable("COREHOST_TRACE", "1");
-            // In Unix environment, you need to run the below command in the terminal to set the environment variable.
-            // export COREHOST_TRACE=1
+            unsafe
+            {
+                var parameters = new NetHost.get_hostfxr_parameters
+                {
+                    size = sizeof(NetHost.get_hostfxr_parameters),
+                    assembly_path = GetCharArrayPointer(assemblyPath)
+                };
 
-            var hostfxrFullPath = NetHost.GetHostFxrPath();
-            Logger.LogTrace($"hostfxr path:{hostfxrFullPath}");
+                var hostfxrFullPath = NetHost.GetHostFxrPath(&parameters);
+                Logger.LogTrace($"hostfxr path:{hostfxrFullPath}");
 
-            _hostfxrHandle = NativeLibrary.Load(hostfxrFullPath);
-            if (_hostfxrHandle == IntPtr.Zero)
-            {
-                Logger.Log($"Failed to load hostfxr. hostfxr path:{hostfxrFullPath}");
-                return;
-            }
+                _hostfxrHandle = NativeLibrary.Load(hostfxrFullPath);
 
-            Logger.LogTrace($"hostfxr library loaded successfully.");
-        }
+                if (_hostfxrHandle == IntPtr.Zero)
+                {
+                    Logger.Log($"Failed to load hostfxr. hostfxrFullPath:{hostfxrFullPath}");
+                    return -1;
+                }
 
-        internal int RunApplication(string? assemblyPath)
-        {
-            ArgumentNullException.ThrowIfNull(assemblyPath, nameof(assemblyPath));
+                Logger.LogTrace($"hostfxr loaded.");
 
-            unsafe
-            {
                 var error = HostFxr.Initialize(1, new[] { assemblyPath }, IntPtr.Zero, out _hostContextHandle);
 
                 if (_hostContextHandle == IntPtr.Zero)
                 {
-                    Logger.Log(
-                        $"Failed to initialize the .NET Core runtime. Assembly path:{assemblyPath}");
+                    Logger.Log($"Failed to initialize the .NET Core runtime. Assembly path:{assemblyPath}");
                     return -1;
                 }
 
@@ -98,5 +97,14 @@ private void Dispose(bool disposing)
                 _disposed = true;
             }
         }
+
+        private static unsafe char* GetCharArrayPointer(string assemblyPath)
+        {
+#if OS_LINUX
+            return (char*)Marshal.StringToHGlobalAnsi(assemblyPath).ToPointer();
+#else
+            return (char*)Marshal.StringToHGlobalUni(assemblyPath).ToPointer();
+#endif
+        }
     }
 }

host/src/FunctionsNetHost/AppLoader/NetHost.cs

@@ -7,18 +7,28 @@ namespace FunctionsNetHost
 {
     internal class NetHost
     {
+        public unsafe struct get_hostfxr_parameters
+        {
+            public nint size;
+
+            // Optional.Path to the application assembly,
+            // If specified, hostfxr is located from this directory if present (For self-contained deployments)
+            public char* assembly_path;
+            public char* dotnet_root;
+        }
+
         [DllImport("nethost", CharSet = CharSet.Auto)]
-        private static extern int get_hostfxr_path(
+        private unsafe static extern int get_hostfxr_path(
         [Out] char[] buffer,
         [In] ref int buffer_size,
-        IntPtr reserved);
+        get_hostfxr_parameters* parameters);
 
-        internal static string GetHostFxrPath()
+        internal unsafe static string GetHostFxrPath(get_hostfxr_parameters* parameters)
         {
             char[] buffer = new char[200];
             int bufferSize = buffer.Length;
 
-            int rc = get_hostfxr_path(buffer, ref bufferSize, IntPtr.Zero);
+            int rc = get_hostfxr_path(buffer, ref bufferSize, parameters);
 
             if (rc != 0)
             {