mirror of
https://github.com/golang/sys.git
synced 2026-02-07 19:26:03 +03:00
8dbc5d05d6
The current GetProcAddress implementation only resolves functions in DLLs by name. Add GetProcAddressByOrdinal that allows resolving functions by ordinal number, using the same GetProcAddress call from kernel32.dll in the background. This is particularly useful for some functions (e.g. IsOS from shlwapi.dll in some older versions of Windows) that cannot be found by name. Fixes golang/go#16507 Change-Id: Ib5fba7568c365a0aa2491c1261876b3a3929ec3d Reviewed-on: https://go-review.googlesource.com/70690 Reviewed-by: Alex Brainman <alex.brainman@gmail.com> Run-TryBot: Alex Brainman <alex.brainman@gmail.com> TryBot-Result: Gobot Gobot <gobot@golang.org>
54 lines
1.4 KiB
Go
54 lines
1.4 KiB
Go
// 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.
|
|
|
|
// +build windows
|
|
|
|
package windows_test
|
|
|
|
import (
|
|
"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")
|
|
}
|
|
}
|