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
|
package w32
import (
"testing"
)
var testProcess = "notepad.exe"
var wantCode = uint32(42)
func TestCreateProcess(t *testing.T) {
pi, err := CreateProcessQuick(testProcess)
if err != nil {
t.Errorf("[!!] Failed to create %s: %s", testProcess, err)
} else {
t.Logf("[OK] Created process %s with handle 0x%x, PID %d", testProcess, pi.Process, pi.ProcessId)
}
err = TerminateProcess(pi.Process, wantCode)
if err != nil {
t.Errorf("[!!]Failed to terminate %s: %s", testProcess, err)
} else {
t.Logf("[OK] Called TerminateProcess on PID %d", pi.ProcessId)
}
err = WaitForSingleObject(pi.Process, 1000) // 1000ms
if err != nil {
t.Errorf("[!!] failed in WaitForSingleObject: %s", err)
} else {
t.Logf("[OK] WaitForSingleObject returned...")
}
// make sure we see the magic exit code we asked for
code, err := GetExitCodeProcess(pi.Process)
if err != nil {
t.Errorf("[!!] Failed to get exit code for PID %d: %s", pi.ProcessId, err)
} else {
t.Logf("[OK] PID %d Exited with code %d", pi.ProcessId, code)
}
if code != 42 {
t.Errorf("[!!] Unexpected exit code for PID %d - want %d, got %d", pi.ProcessId, wantCode, code)
}
CloseHandle(pi.Process)
CloseHandle(pi.Thread)
}
|
