github.com/dolotech/hongbao@v0.0.0-20191130105438-fd59d7a5dda5/src/golang.org/x/sys/windows/syscall_windows_test.go (about)

     1  // Copyright 2012 The Go Authors. All rights reserved.
     2  // Use of this source code is governed by a BSD-style
     3  // license that can be found in the LICENSE file.
     4  
     5  package windows_test
     6  
     7  import (
     8  	"io/ioutil"
     9  	"os"
    10  	"path/filepath"
    11  	"syscall"
    12  	"testing"
    13  	"unsafe"
    14  
    15  	"golang.org/x/sys/windows"
    16  )
    17  
    18  func TestWin32finddata(t *testing.T) {
    19  	dir, err := ioutil.TempDir("", "go-build")
    20  	if err != nil {
    21  		t.Fatalf("failed to create temp directory: %v", err)
    22  	}
    23  	defer os.RemoveAll(dir)
    24  
    25  	path := filepath.Join(dir, "long_name.and_extension")
    26  	f, err := os.Create(path)
    27  	if err != nil {
    28  		t.Fatalf("failed to create %v: %v", path, err)
    29  	}
    30  	f.Close()
    31  
    32  	type X struct {
    33  		fd  windows.Win32finddata
    34  		got byte
    35  		pad [10]byte // to protect ourselves
    36  
    37  	}
    38  	var want byte = 2 // it is unlikely to have this character in the filename
    39  	x := X{got: want}
    40  
    41  	pathp, _ := windows.UTF16PtrFromString(path)
    42  	h, err := windows.FindFirstFile(pathp, &(x.fd))
    43  	if err != nil {
    44  		t.Fatalf("FindFirstFile failed: %v", err)
    45  	}
    46  	err = windows.FindClose(h)
    47  	if err != nil {
    48  		t.Fatalf("FindClose failed: %v", err)
    49  	}
    50  
    51  	if x.got != want {
    52  		t.Fatalf("memory corruption: want=%d got=%d", want, x.got)
    53  	}
    54  }
    55  
    56  func TestFormatMessage(t *testing.T) {
    57  	dll := windows.MustLoadDLL("pdh.dll")
    58  
    59  	pdhOpenQuery := func(datasrc *uint16, userdata uint32, query *windows.Handle) (errno uintptr) {
    60  		r0, _, _ := syscall.Syscall(dll.MustFindProc("PdhOpenQueryW").Addr(), 3, uintptr(unsafe.Pointer(datasrc)), uintptr(userdata), uintptr(unsafe.Pointer(query)))
    61  		return r0
    62  	}
    63  
    64  	pdhCloseQuery := func(query windows.Handle) (errno uintptr) {
    65  		r0, _, _ := syscall.Syscall(dll.MustFindProc("PdhCloseQuery").Addr(), 1, uintptr(query), 0, 0)
    66  		return r0
    67  	}
    68  
    69  	var q windows.Handle
    70  	name, err := windows.UTF16PtrFromString("no_such_source")
    71  	if err != nil {
    72  		t.Fatal(err)
    73  	}
    74  	errno := pdhOpenQuery(name, 0, &q)
    75  	if errno == 0 {
    76  		pdhCloseQuery(q)
    77  		t.Fatal("PdhOpenQuery succeeded, but expected to fail.")
    78  	}
    79  
    80  	const flags uint32 = syscall.FORMAT_MESSAGE_FROM_HMODULE | syscall.FORMAT_MESSAGE_ARGUMENT_ARRAY | syscall.FORMAT_MESSAGE_IGNORE_INSERTS
    81  	buf := make([]uint16, 300)
    82  	_, err = windows.FormatMessage(flags, uintptr(dll.Handle), uint32(errno), 0, buf, nil)
    83  	if err != nil {
    84  		t.Fatal("FormatMessage for handle=%x and errno=%x failed: %v", dll.Handle, errno, err)
    85  	}
    86  }
    87  
    88  func abort(funcname string, err error) {
    89  	panic(funcname + " failed: " + err.Error())
    90  }
    91  
    92  func ExampleLoadLibrary() {
    93  	h, err := windows.LoadLibrary("kernel32.dll")
    94  	if err != nil {
    95  		abort("LoadLibrary", err)
    96  	}
    97  	defer windows.FreeLibrary(h)
    98  	proc, err := windows.GetProcAddress(h, "GetVersion")
    99  	if err != nil {
   100  		abort("GetProcAddress", err)
   101  	}
   102  	r, _, _ := syscall.Syscall(uintptr(proc), 0, 0, 0, 0)
   103  	major := byte(r)
   104  	minor := uint8(r >> 8)
   105  	build := uint16(r >> 16)
   106  	print("windows version ", major, ".", minor, " (Build ", build, ")\n")
   107  }