liuxiaolong
2022-06-28 37714b1093c04061e636e5b1d27179652e671c0a
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
// 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"
    "unicode/utf16"
    "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)
    blockp := uintptr(unsafe.Pointer(block))
    for {
        entry := (*[(1 << 30) - 1]uint16)(unsafe.Pointer(blockp))[:]
        for i, v := range entry {
            if v == 0 {
                entry = entry[:i]
                break
            }
        }
        if len(entry) == 0 {
            break
        }
        env = append(env, string(utf16.Decode(entry)))
        blockp += 2 * (uintptr(len(entry)) + 1)
    }
    return env, nil
}
 
func Unsetenv(key string) error {
    return syscall.Unsetenv(key)
}