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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
// Copyright 2013 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
//go:build windows
// +build windows
package windows_test
import (
"strings"
"syscall"
"testing"
"golang.org/x/sys/windows"
)
func testSetGetenv(t *testing.T, key, value string) {
err := windows.Setenv(key, value)
if err != nil {
t.Fatalf("Setenv failed to set %q: %v", value, err)
}
newvalue, found := windows.Getenv(key)
if !found {
t.Fatalf("Getenv failed to find %v variable (want value %q)", key, value)
}
if newvalue != value {
t.Fatalf("Getenv(%v) = %q; want %q", key, newvalue, value)
}
}
func TestEnv(t *testing.T) {
testSetGetenv(t, "TESTENV", "AVALUE")
// make sure TESTENV gets set to "", not deleted
testSetGetenv(t, "TESTENV", "")
}
func TestGetProcAddressByOrdinal(t *testing.T) {
// Attempt calling shlwapi.dll:IsOS, resolving it by ordinal, as
// suggested in
// https://msdn.microsoft.com/en-us/library/windows/desktop/bb773795.aspx
h, err := windows.LoadLibrary("shlwapi.dll")
if err != nil {
t.Fatalf("Failed to load shlwapi.dll: %s", err)
}
procIsOS, err := windows.GetProcAddressByOrdinal(h, 437)
if err != nil {
t.Fatalf("Could not find shlwapi.dll:IsOS by ordinal: %s", err)
}
const OS_NT = 1
r, _, _ := syscall.Syscall(procIsOS, 1, OS_NT, 0, 0)
if r == 0 {
t.Error("shlwapi.dll:IsOS(OS_NT) returned 0, expected non-zero value")
}
}
func TestGetSystemDirectory(t *testing.T) {
d, err := windows.GetSystemDirectory()
if err != nil {
t.Fatalf("Failed to get system directory: %s", err)
}
if !strings.HasSuffix(strings.ToLower(d), "\\system32") {
t.Fatalf("System directory does not end in system32: %s", d)
}
}
func TestGetWindowsDirectory(t *testing.T) {
d1, err := windows.GetWindowsDirectory()
if err != nil {
t.Fatalf("Failed to get Windows directory: %s", err)
}
d2, err := windows.GetSystemWindowsDirectory()
if err != nil {
t.Fatalf("Failed to get system Windows directory: %s", err)
}
if !strings.HasSuffix(strings.ToLower(d1), `\windows`) {
t.Fatalf("Windows directory does not end in windows: %s", d1)
}
if !strings.HasSuffix(strings.ToLower(d2), `\windows`) {
t.Fatalf("System Windows directory does not end in windows: %s", d2)
}
}
func TestFindProcByOrdinal(t *testing.T) {
// Attempt calling shlwapi.dll:IsOS, resolving it by ordinal, as
// suggested in
// https://msdn.microsoft.com/en-us/library/windows/desktop/bb773795.aspx
dll, err := windows.LoadDLL("shlwapi.dll")
if err != nil {
t.Fatalf("Failed to load shlwapi.dll: %s", err)
}
procIsOS, err := dll.FindProcByOrdinal(437)
if err != nil {
t.Fatalf("Could not find shlwapi.dll:IsOS by ordinal: %s", err)
}
if procIsOS.Name != "#437" {
t.Fatalf("Proc's name is incorrect: %s,expected #437", procIsOS.Name)
}
const OS_NT = 1
r, _, _ := syscall.Syscall(procIsOS.Addr(), 1, OS_NT, 0, 0)
if r == 0 {
t.Error("shlwapi.dll:IsOS(OS_NT) returned 0, expected non-zero value")
}
}
|
