diff --git a/windows/env_windows.go b/windows/env_windows.go index b8ad19250..d4577a423 100644 --- a/windows/env_windows.go +++ b/windows/env_windows.go @@ -37,14 +37,17 @@ func (token Token) Environ(inheritExisting bool) (env []string, err error) { return nil, err } defer DestroyEnvironmentBlock(block) - blockp := unsafe.Pointer(block) - for { - entry := UTF16PtrToString((*uint16)(blockp)) - if len(entry) == 0 { - break + size := unsafe.Sizeof(*block) + for *block != 0 { + // find NUL terminator + end := unsafe.Pointer(block) + for *(*uint16)(end) != 0 { + end = unsafe.Add(end, size) } - env = append(env, entry) - blockp = unsafe.Add(blockp, 2*(len(entry)+1)) + + entry := unsafe.Slice(block, (uintptr(end)-uintptr(unsafe.Pointer(block)))/size) + env = append(env, UTF16ToString(entry)) + block = (*uint16)(unsafe.Add(end, size)) } return env, nil } diff --git a/windows/env_windows_test.go b/windows/env_windows_test.go new file mode 100644 index 000000000..6824e6ed6 --- /dev/null +++ b/windows/env_windows_test.go @@ -0,0 +1,40 @@ +// Copyright 2024 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. + +package windows_test + +import ( + "fmt" + "slices" + "testing" + + "golang.org/x/sys/windows" +) + +func TestEnvironUTF8(t *testing.T) { + testEnvVariable1Key := "__GO_X_SYS_WINDOWS_ENV_WINDOWS_TEST_VAR_BEAVER" + testEnvVariable1Val := "🦫" + t.Setenv(testEnvVariable1Key, testEnvVariable1Val) + + testEnvVariable2Key := "__GO_X_SYS_WINDOWS_ENV_WINDOWS_TEST_VAR_WHALE" + testEnvVariable2Val := "🐳" + t.Setenv(testEnvVariable2Key, testEnvVariable2Val) + + var userToken windows.Token + + env, err := userToken.Environ(true) + if err != nil { + t.Error(err) + } + + testEnvVariable1 := fmt.Sprintf("%s=%s", testEnvVariable1Key, testEnvVariable1Val) + if !slices.Contains(env, testEnvVariable1) { + t.Fatalf("expected to find %s in env", testEnvVariable1) + } + + testEnvVariable2 := fmt.Sprintf("%s=%s", testEnvVariable2Key, testEnvVariable2Val) + if !slices.Contains(env, testEnvVariable2) { + t.Fatalf("expected to find %s in env", testEnvVariable2) + } +}