Ensure COM threading apartment for API calls

Author: laozc

pkg/cim/wmi.go

@@ -5,6 +5,7 @@ package cim
 
 import (
 	"fmt"
+	"runtime"
 
 	"github.com/go-ole/go-ole"
 	"github.com/go-ole/go-ole/oleutil"
@@ -248,3 +249,30 @@ func IgnoreNotFound(err error) error {
 	}
 	return err
 }
+
+// WithCOMThread runs the given function `fn` on a locked OS thread
+// with COM initialized using COINIT_MULTITHREADED.
+//
+// This is necessary for using COM/OLE APIs directly (e.g., via go-ole),
+// because COM requires that initialization and usage occur on the same thread.
+//
+// It performs the following steps:
+//   - Locks the current goroutine to its OS thread
+//   - Calls ole.CoInitializeEx with COINIT_MULTITHREADED
+//   - Executes the user-provided function
+//   - Uninitializes COM
+//   - Unlocks the thread
+//
+// If COM initialization fails, or if the user's function returns an error,
+// that error is returned by WithCOMThread.
+func WithCOMThread(fn func() error) error {
+	runtime.LockOSThread()
+	defer runtime.UnlockOSThread()
+
+	if err := ole.CoInitializeEx(0, ole.COINIT_MULTITHREADED); err != nil {
+		return err
+	}
+	defer ole.CoUninitialize()
+
+	return fn()
+}

pkg/os/smb/api.go

@@ -28,15 +28,17 @@ func New(requirePrivacy bool) *SmbAPI {
 }
 
 func (*SmbAPI) IsSmbMapped(remotePath string) (bool, error) {
-	inst, err := cim.QuerySmbGlobalMappingByRemotePath(remotePath)
-	if err != nil {
-		return false, cim.IgnoreNotFound(err)
-	}
-
-	status, err := inst.GetProperty("Status")
-	if err != nil {
-		return false, err
-	}
+	err := cim.WithCOMThread(func() error {
+		inst, err := cim.QuerySmbGlobalMappingByRemotePath(remotePath)
+		if err != nil {
+			return false, cim.IgnoreNotFound(err)
+		}
+
+		status, err := inst.GetProperty("Status")
+		if err != nil {
+			return false, err
+		}
+	})
 
 	return status.(int32) == cim.SmbMappingStatusOK, nil
 }