1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
|
//go:build windows
package ewindows
import (
"errors"
"unsafe"
"golang.org/x/sys/windows"
)
const (
INFINITE = 0xFFFFFFFF
)
const (
WAIT_OBJECT_0 = 0
WAIT_ABANDONED_0 = 0x00000080
WAIT_TIMEOUT = 0x00000102
WAIT_FAILED = 0xFFFFFFFF
)
var (
waitForMultipleObjects = kernel32.NewProc("WaitForMultipleObjects")
errTimeout = errors.New("WaitForMultipleObjects timeout")
)
// WaitForMultipleObjects blocks until any of the objects is triggered or
// timeout.
//
// DWORD WINAPI WaitForMultipleObjects(
//
// _In_ DWORD nCount,
// _In_ const HANDLE *lpHandles,
// _In_ BOOL bWaitAll,
// _In_ DWORD dwMilliseconds
//
// );
func WaitForMultipleObjects(handles []windows.Handle, waitAll bool,
timeout uint32) (trigger int, abandoned bool, err error) {
count := uintptr(len(handles))
ret, _, err := waitForMultipleObjects.Call(count,
uintptr(unsafe.Pointer(&handles[0])), boolToUintptr(waitAll), uintptr(timeout))
switch {
case WAIT_OBJECT_0 <= ret && ret < WAIT_OBJECT_0+count:
return int(ret - WAIT_OBJECT_0), false, nil
case WAIT_ABANDONED_0 <= ret && ret < WAIT_ABANDONED_0+count:
return int(ret - WAIT_ABANDONED_0), true, nil
case ret == WAIT_TIMEOUT:
return -1, false, errTimeout
default:
return -1, false, err
}
}
func boolToUintptr(b bool) uintptr {
if b {
return 1
}
return 0
}
|
