Init

Author: 0x5bfa

src/Files.App.CsWin32/NativeMethods.txt

@@ -269,3 +269,8 @@ HCERTSTORE
 HCRYPTMSG
 CERT_QUERY_ENCODING_TYPE
 CertGetNameString
+CreateEvent
+SetEvent
+CoWaitForMultipleObjects
+CWMO_FLAGS
+INFINITE

src/Files.App.Storage/Windows/Managers/STATask.cs

@@ -3,6 +3,9 @@
 
 using Microsoft.Extensions.Logging;
 using Windows.Win32;
+using Windows.Win32.Foundation;
+using Windows.Win32.System.Com;
+using Windows.Win32.Security;
 
 namespace Files.App.Storage
 {
@@ -140,5 +143,46 @@ public static Task Run(Func<Task> func, ILogger? logger = null)
 
 			return tcs.Task;
 		}
+
+		public unsafe static Task RunAsSync(Action action)
+		{
+			Debug.Assert(Thread.CurrentThread.GetApartmentState() is ApartmentState.STA);
+
+			HANDLE hEventHandle = PInvoke.CreateEvent((SECURITY_ATTRIBUTES*)null, true, false, default);
+
+			var tcs = new TaskCompletionSource();
+
+			Task.Run(() =>
+			{
+				try
+				{
+					action();
+					tcs.SetResult();
+				}
+				catch (Exception ex)
+				{
+					tcs.SetException(ex);
+				}
+				finally
+				{
+					PInvoke.SetEvent(hEventHandle);
+				}
+			});
+
+			HANDLE* pEventHandles = stackalloc HANDLE[1];
+			pEventHandles[0] = hEventHandle;
+			uint dwIndex = 0u;
+
+			PInvoke.CoWaitForMultipleObjects(
+				(uint)CWMO_FLAGS.CWMO_DEFAULT,
+				PInvoke.INFINITE,
+				1u,
+				pEventHandles,
+				&dwIndex);
+
+			PInvoke.CloseHandle(hEventHandle);
+
+			return tcs.Task;
+		}
 	}
 }

src/Files.App/Helpers/Win32/Win32PInvoke.Methods.cs

@@ -68,15 +68,6 @@ public static extern bool SetEvent(
 			IntPtr hEvent
 		);
 
-		[DllImport("ole32.dll")]
-		public static extern uint CoWaitForMultipleObjects(
-			uint dwFlags,
-			uint dwMilliseconds,
-			ulong nHandles,
-			IntPtr[] pHandles,
-			out uint dwIndex
-		);
-
 		[DllImport("shell32.dll")]
 		public static extern IntPtr SHBrowseForFolder(
 			ref BROWSEINFO lpbi

src/Files.App/Program.cs

@@ -9,7 +9,6 @@
 using System.Text;
 using Windows.ApplicationModel.Activation;
 using Windows.Storage;
-using static Files.App.Helpers.Win32PInvoke;
 
 namespace Files.App
 {
@@ -21,9 +20,6 @@ namespace Files.App
 	/// </remarks>
 	internal sealed class Program
 	{
-		private const uint CWMO_DEFAULT = 0;
-		private const uint INFINITE = 0xFFFFFFFF;
-
 		public static Semaphore? Pool { get; set; }
 
 		static Program()
@@ -250,20 +246,10 @@ private static async void OnActivated(object? sender, AppActivationArguments arg
 		/// </remarks>
 		public static void RedirectActivationTo(AppInstance keyInstance, AppActivationArguments args)
 		{
-			IntPtr eventHandle = CreateEvent(IntPtr.Zero, true, false, null);
-
-			Task.Run(() =>
+			STATask.RunAsSync(() =>
 			{
 				keyInstance.RedirectActivationToAsync(args).AsTask().Wait();
-				SetEvent(eventHandle);
 			});
-
-			_ = CoWaitForMultipleObjects(
-				CWMO_DEFAULT,
-				INFINITE,
-				1,
-				[eventHandle],
-				out uint handleIndex);
 		}
 
 		public static void OpenShellCommandInExplorer(string shellCommand, int pid)
@@ -273,20 +259,10 @@ public static void OpenShellCommandInExplorer(string shellCommand, int pid)
 
 		public static void OpenFileFromTile(string filePath)
 		{
-			IntPtr eventHandle = CreateEvent(IntPtr.Zero, true, false, null);
-
-			Task.Run(() =>
+			STATask.RunAsSync(() =>
 			{
 				LaunchHelper.LaunchAppAsync(filePath, null, null).Wait();
-				SetEvent(eventHandle);
 			});
-
-			_ = CoWaitForMultipleObjects(
-				CWMO_DEFAULT,
-				INFINITE,
-				1,
-				[eventHandle],
-				out uint handleIndex);
 		}
 	}
 }