From f3c3bd215731f55cf042e0d1c9eae4aa880f6257 Mon Sep 17 00:00:00 2001
From: "Jason A. Donenfeld" <Jason@zx2c4.com>
Date: Thu, 28 Feb 2019 07:19:06 +0100
Subject: service: track tunnel service status

---
 service/zsyscall_windows.go | 42 ++++++++++++++++++++++++++++++++++++++----
 1 file changed, 38 insertions(+), 4 deletions(-)

(limited to 'service/zsyscall_windows.go')

diff --git a/service/zsyscall_windows.go b/service/zsyscall_windows.go
index 649e3581..faf6d780 100644
--- a/service/zsyscall_windows.go
+++ b/service/zsyscall_windows.go
@@ -39,11 +39,14 @@ func errnoErr(e syscall.Errno) error {
 var (
 	modwtsapi32 = windows.NewLazySystemDLL("wtsapi32.dll")
 	modadvapi32 = windows.NewLazySystemDLL("advapi32.dll")
+	modkernel32 = windows.NewLazySystemDLL("kernel32.dll")
 
-	procWTSQueryUserToken     = modwtsapi32.NewProc("WTSQueryUserToken")
-	procWTSEnumerateSessionsW = modwtsapi32.NewProc("WTSEnumerateSessionsW")
-	procWTSFreeMemory         = modwtsapi32.NewProc("WTSFreeMemory")
-	procCreateWellKnownSid    = modadvapi32.NewProc("CreateWellKnownSid")
+	procWTSQueryUserToken          = modwtsapi32.NewProc("WTSQueryUserToken")
+	procWTSEnumerateSessionsW      = modwtsapi32.NewProc("WTSEnumerateSessionsW")
+	procWTSFreeMemory              = modwtsapi32.NewProc("WTSFreeMemory")
+	procCreateWellKnownSid         = modadvapi32.NewProc("CreateWellKnownSid")
+	procNotifyServiceStatusChangeW = modadvapi32.NewProc("NotifyServiceStatusChangeW")
+	procSleepEx                    = modkernel32.NewProc("SleepEx")
 )
 
 func wtfQueryUserToken(session uint32, token *windows.Token) (err error) {
@@ -86,3 +89,34 @@ func createWellKnownSid(sidType wellKnownSidType, domainSid *windows.SID, sid *w
 	}
 	return
 }
+
+func notifyServiceStatusChange(service windows.Handle, notifyMask uint32, notifyBuffer uintptr) (err error) {
+	r1, _, e1 := syscall.Syscall(procNotifyServiceStatusChangeW.Addr(), 3, uintptr(service), uintptr(notifyMask), uintptr(notifyBuffer))
+	if r1 != 0 {
+		if e1 != 0 {
+			err = errnoErr(e1)
+		} else {
+			err = syscall.EINVAL
+		}
+	}
+	return
+}
+
+func sleepEx(milliseconds uint32, alertable bool) (ret uint32, err error) {
+	var _p0 uint32
+	if alertable {
+		_p0 = 1
+	} else {
+		_p0 = 0
+	}
+	r0, _, e1 := syscall.Syscall(procSleepEx.Addr(), 2, uintptr(milliseconds), uintptr(_p0), 0)
+	ret = uint32(r0)
+	if ret == 0 {
+		if e1 != 0 {
+			err = errnoErr(e1)
+		} else {
+			err = syscall.EINVAL
+		}
+	}
+	return
+}
-- 
cgit v1.2.3-59-g8ed1b