mirror of
https://github.com/golang/sys.git
synced 2026-01-29 07:02:06 +03:00
c3fa2b8b41
Fixes golang/go#65055, the unexpected behavior of token.Environ in parsing entries containing runes larger than 2 bytes in size Change-Id: I753d2c605e3a2d7a1d90cd18601d6b918f0d3f7a Reviewed-on: https://go-review.googlesource.com/c/sys/+/556895 Auto-Submit: Bryan Mills <bcmills@google.com> LUCI-TryBot-Result: Go LUCI <golang-scoped@luci-project-accounts.iam.gserviceaccount.com> Reviewed-by: Cherry Mui <cherryyz@google.com> Reviewed-by: Bryan Mills <bcmills@google.com> Reviewed-by: Quim Muntal <quimmuntal@gmail.com>
58 lines
1.3 KiB
Go
58 lines
1.3 KiB
Go
// Copyright 2010 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.
|
|
|
|
// Windows environment variables.
|
|
|
|
package windows
|
|
|
|
import (
|
|
"syscall"
|
|
"unsafe"
|
|
)
|
|
|
|
func Getenv(key string) (value string, found bool) {
|
|
return syscall.Getenv(key)
|
|
}
|
|
|
|
func Setenv(key, value string) error {
|
|
return syscall.Setenv(key, value)
|
|
}
|
|
|
|
func Clearenv() {
|
|
syscall.Clearenv()
|
|
}
|
|
|
|
func Environ() []string {
|
|
return syscall.Environ()
|
|
}
|
|
|
|
// Returns a default environment associated with the token, rather than the current
|
|
// process. If inheritExisting is true, then this environment also inherits the
|
|
// environment of the current process.
|
|
func (token Token) Environ(inheritExisting bool) (env []string, err error) {
|
|
var block *uint16
|
|
err = CreateEnvironmentBlock(&block, token, inheritExisting)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer DestroyEnvironmentBlock(block)
|
|
size := unsafe.Sizeof(*block)
|
|
for *block != 0 {
|
|
// find NUL terminator
|
|
end := unsafe.Pointer(block)
|
|
for *(*uint16)(end) != 0 {
|
|
end = unsafe.Add(end, size)
|
|
}
|
|
|
|
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
|
|
}
|
|
|
|
func Unsetenv(key string) error {
|
|
return syscall.Unsetenv(key)
|
|
}
|