Fix RegisterWaitForSingleObject and SetFileCompletionNotificationModes usage

- Use .put() instead of & for wil::unique_any output parameters in
  RegisterWaitForSingleObject calls. The & operator returns the address
  of the wrapper, not a HANDLE*; .put() correctly resets any prior
  handle and returns a pointer to the internal storage.

- Check the return value of SetFileCompletionNotificationModes. If it
  fails, the IOCP state machine assumption that synchronous completions
  will not queue a packet is violated, causing double-processing.

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>

Author: Ben Hillis

src/windows/common/relay.cpp

@@ -1150,7 +1150,7 @@ void EventHandle::Schedule()
     if (Iocp != nullptr && !WaitHandle)
     {
         THROW_IF_WIN32_BOOL_FALSE(
-            RegisterWaitForSingleObject(&WaitHandle, Handle.Get(), &EventHandle::WaitCallback, this, INFINITE, WT_EXECUTEONLYONCE));
+            RegisterWaitForSingleObject(WaitHandle.put(), Handle.Get(), &EventHandle::WaitCallback, this, INFINITE, WT_EXECUTEONLYONCE));
     }
 
     State = IOHandleStatus::Pending;
@@ -1222,7 +1222,8 @@ void ReadHandle::Register(HANDLE iocp, OverlappedIOHandle* completionTarget)
         auto result = CreateIoCompletionPort(Handle.Get(), iocp, reinterpret_cast<ULONG_PTR>(completionTarget), 0);
         if (result != nullptr)
         {
-            SetFileCompletionNotificationModes(Handle.Get(), FILE_SKIP_COMPLETION_PORT_ON_SUCCESS);
+            THROW_IF_WIN32_BOOL_FALSE(
+                SetFileCompletionNotificationModes(Handle.Get(), FILE_SKIP_COMPLETION_PORT_ON_SUCCESS));
 
             // Clear the event from OVERLAPPED — completions now go to the IOCP.
             Overlapped.hEvent = nullptr;
@@ -1280,7 +1281,7 @@ void ReadHandle::Schedule()
         if (!RegisteredWithIocp && Iocp != nullptr && !WaitBridge)
         {
             THROW_IF_WIN32_BOOL_FALSE(RegisterWaitForSingleObject(
-                &WaitBridge, Event.get(), &ReadHandle::WaitBridgeCallback, this, INFINITE, WT_EXECUTEONLYONCE));
+                WaitBridge.put(), Event.get(), &ReadHandle::WaitBridgeCallback, this, INFINITE, WT_EXECUTEONLYONCE));
         }
     }
 }
@@ -1365,7 +1366,8 @@ void SingleAcceptHandle::Register(HANDLE iocp, OverlappedIOHandle* completionTar
         auto result = CreateIoCompletionPort(ListenSocket.Get(), iocp, reinterpret_cast<ULONG_PTR>(completionTarget), 0);
         if (result != nullptr)
         {
-            SetFileCompletionNotificationModes(ListenSocket.Get(), FILE_SKIP_COMPLETION_PORT_ON_SUCCESS);
+            THROW_IF_WIN32_BOOL_FALSE(
+                SetFileCompletionNotificationModes(ListenSocket.Get(), FILE_SKIP_COMPLETION_PORT_ON_SUCCESS));
             Overlapped.hEvent = nullptr;
             RegisteredWithIocp = true;
         }
@@ -1394,7 +1396,7 @@ void SingleAcceptHandle::Schedule()
         if (!RegisteredWithIocp && Iocp != nullptr && !WaitBridge)
         {
             THROW_IF_WIN32_BOOL_FALSE(RegisterWaitForSingleObject(
-                &WaitBridge, Event.get(), &SingleAcceptHandle::WaitBridgeCallback, this, INFINITE, WT_EXECUTEONLYONCE));
+                WaitBridge.put(), Event.get(), &SingleAcceptHandle::WaitBridgeCallback, this, INFINITE, WT_EXECUTEONLYONCE));
         }
     }
 }
@@ -1621,7 +1623,8 @@ void WriteHandle::Register(HANDLE iocp, OverlappedIOHandle* completionTarget)
         auto result = CreateIoCompletionPort(Handle.Get(), iocp, reinterpret_cast<ULONG_PTR>(completionTarget), 0);
         if (result != nullptr)
         {
-            SetFileCompletionNotificationModes(Handle.Get(), FILE_SKIP_COMPLETION_PORT_ON_SUCCESS);
+            THROW_IF_WIN32_BOOL_FALSE(
+                SetFileCompletionNotificationModes(Handle.Get(), FILE_SKIP_COMPLETION_PORT_ON_SUCCESS));
             Overlapped.hEvent = nullptr;
             RegisteredWithIocp = true;
         }
@@ -1660,7 +1663,7 @@ void WriteHandle::Schedule()
         if (!RegisteredWithIocp && Iocp != nullptr && !WaitBridge)
         {
             THROW_IF_WIN32_BOOL_FALSE(RegisterWaitForSingleObject(
-                &WaitBridge, Event.get(), &WriteHandle::WaitBridgeCallback, this, INFINITE, WT_EXECUTEONLYONCE));
+                WaitBridge.put(), Event.get(), &WriteHandle::WaitBridgeCallback, this, INFINITE, WT_EXECUTEONLYONCE));
         }
     }
 }

src/windows/wslrelay/localhost.cpp

@@ -676,7 +676,8 @@ void wsl::windows::wslrelay::localhost::RunWSLCPortRelay(const GUID& VmId, uint3
                     THROW_LAST_ERROR_IF_NULL(CreateIoCompletionPort(
                         reinterpret_cast<HANDLE>(e.second->ListenSocket.get()), iocp.get(), reinterpret_cast<ULONG_PTR>(e.second.get()), 0));
 
-                    SetFileCompletionNotificationModes(reinterpret_cast<HANDLE>(e.second->ListenSocket.get()), FILE_SKIP_COMPLETION_PORT_ON_SUCCESS);
+                    THROW_IF_WIN32_BOOL_FALSE(SetFileCompletionNotificationModes(
+                        reinterpret_cast<HANDLE>(e.second->ListenSocket.get()), FILE_SKIP_COMPLETION_PORT_ON_SUCCESS));
 
                     e.second->AssociatedWithIocp = true;
                 }