1
2
3
4
5
6
7
8 package execenv
9
10 import (
11 "internal/syscall/windows"
12 "syscall"
13 "unicode/utf16"
14 "unsafe"
15 )
16
17
18
19
20
21
22
23
24
25 func Default(sys *syscall.SysProcAttr) (env []string, err error) {
26 if sys == nil || sys.Token == 0 {
27 return syscall.Environ(), nil
28 }
29 var block *uint16
30 err = windows.CreateEnvironmentBlock(&block, sys.Token, false)
31 if err != nil {
32 return nil, err
33 }
34 defer windows.DestroyEnvironmentBlock(block)
35 blockp := uintptr(unsafe.Pointer(block))
36 for {
37
38
39 end := unsafe.Pointer(blockp)
40 for *(*uint16)(end) != 0 {
41 end = unsafe.Pointer(uintptr(end) + 2)
42 }
43
44 n := (uintptr(end) - uintptr(unsafe.Pointer(blockp))) / 2
45 if n == 0 {
46
47 break
48 }
49
50 entry := (*[(1 << 30) - 1]uint16)(unsafe.Pointer(blockp))[:n:n]
51 env = append(env, string(utf16.Decode(entry)))
52 blockp += 2 * (uintptr(len(entry)) + 1)
53 }
54 return
55 }
56
View as plain text