mirror of
https://github.com/rwinkhart/go-winio.git
synced 2026-08-28 04:46:50 -04:00
Hamza El-Saawy
9f0d5dc7d2
Where ever possible, use `golang.org/x/sys/windows` instead of `syscall` (which has been deprecated since go1.11). Using `windows.LocalFree` requires using `unsafe.Pointer`, which ensures that the Go garbage collector does not try to free memory pre-maturely if it was previously declared as a pointer. Since `syscall.Handle` is part of API for `vhd` package, it was left unchanged. For security descriptor functions, switch to using `windows.SECURITY_DESCRIPTOR` to avoid unnecessary byte manipulation and panics due to missing input validation and error checking. Signed-off-by: Hamza El-Saawy <hamzaelsaawy@microsoft.com>
97 lines
2.1 KiB
Go
97 lines
2.1 KiB
Go
//go:build windows
|
|
// +build windows
|
|
|
|
package winio
|
|
|
|
import (
|
|
"os"
|
|
"reflect"
|
|
"testing"
|
|
"unsafe"
|
|
|
|
"golang.org/x/sys/windows"
|
|
)
|
|
|
|
var (
|
|
testEas = []ExtendedAttribute{
|
|
{Name: "foo", Value: []byte("bar")},
|
|
{Name: "fizz", Value: []byte("buzz")},
|
|
}
|
|
|
|
testEasEncoded = []byte{16, 0, 0, 0, 0, 3, 3, 0, 102, 111, 111, 0, 98, 97, 114, 0, 0,
|
|
0, 0, 0, 0, 4, 4, 0, 102, 105, 122, 122, 0, 98, 117, 122, 122, 0, 0, 0}
|
|
testEasNotPadded = testEasEncoded[0 : len(testEasEncoded)-3]
|
|
testEasTruncated = testEasEncoded[0:20]
|
|
)
|
|
|
|
func Test_RoundTripEas(t *testing.T) {
|
|
b, err := EncodeExtendedAttributes(testEas)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !reflect.DeepEqual(testEasEncoded, b) {
|
|
t.Fatalf("encoded mismatch %v %v", testEasEncoded, b)
|
|
}
|
|
eas, err := DecodeExtendedAttributes(b)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !reflect.DeepEqual(testEas, eas) {
|
|
t.Fatalf("mismatch %+v %+v", testEas, eas)
|
|
}
|
|
}
|
|
|
|
func Test_EasDontNeedPaddingAtEnd(t *testing.T) {
|
|
eas, err := DecodeExtendedAttributes(testEasNotPadded)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !reflect.DeepEqual(testEas, eas) {
|
|
t.Fatalf("mismatch %+v %+v", testEas, eas)
|
|
}
|
|
}
|
|
|
|
func Test_TruncatedEasFailCorrectly(t *testing.T) {
|
|
_, err := DecodeExtendedAttributes(testEasTruncated)
|
|
if err == nil {
|
|
t.Fatal("expected error")
|
|
}
|
|
}
|
|
|
|
func Test_NilEasEncodeAndDecodeAsNil(t *testing.T) {
|
|
b, err := EncodeExtendedAttributes(nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(b) != 0 {
|
|
t.Fatal("expected empty")
|
|
}
|
|
eas, err := DecodeExtendedAttributes(nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(eas) != 0 {
|
|
t.Fatal("expected empty")
|
|
}
|
|
}
|
|
|
|
// Test_SetFileEa makes sure that the test buffer is actually parsable by NtSetEaFile.
|
|
func Test_SetFileEa(t *testing.T) {
|
|
f, err := os.CreateTemp("", "winio")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer os.Remove(f.Name())
|
|
defer f.Close()
|
|
ntdll := windows.MustLoadDLL("ntdll.dll")
|
|
ntSetEaFile := ntdll.MustFindProc("NtSetEaFile")
|
|
var iosb [2]uintptr
|
|
r, _, _ := ntSetEaFile.Call(f.Fd(),
|
|
uintptr(unsafe.Pointer(&iosb[0])),
|
|
uintptr(unsafe.Pointer(&testEasEncoded[0])),
|
|
uintptr(len(testEasEncoded)))
|
|
if r != 0 {
|
|
t.Fatalf("NtSetEaFile failed with %08x", r)
|
|
}
|
|
}
|