Fix usage of PowerUnregisterSuspendResumeNotification

SagerNet · Aug 8, 2024 · 5723a3e · 5723a3e
1 parent 7beca62
commit 5723a3e
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/common/winpowrprof/event_windows.go b/common/winpowrprof/event_windows.go
@@ -3,6 +3,7 @@ package winpowrprof
 // modify from https://github.com/golang/go/blob/b634f6fdcbebee23b7da709a243f3db217b64776/src/runtime/os_windows.go#L257
 
 import (
+	"runtime"
 	"syscall"
 	"unsafe"
 
@@ -83,9 +84,10 @@ func (l *powerEventListener) Start() error {
 }
 
 func (l *powerEventListener) Close() error {
-	_, _, errno := syscall.SyscallN(procPowerUnregisterSuspendResumeNotification.Addr(), uintptr(unsafe.Pointer(&l.handle)))
-	if errno != 0 {
-		return errno
+	r0, _, _ := syscall.SyscallN(procPowerUnregisterSuspendResumeNotification.Addr(), l.handle)
+	if r0 != windows.NO_ERROR {
+		return syscall.Errno(r0)
 	}
+	runtime.KeepAlive(l)
 	return nil
 }