2019-03-07 11:39:54 +08:00
|
|
|
package win32api
|
|
|
|
|
|
|
|
|
|
import (
|
|
|
|
|
"syscall"
|
|
|
|
|
"unsafe"
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
/*
|
|
|
|
|
BOOL CreateEnvironmentBlock(
|
|
|
|
|
LPVOID *lpEnvironment,
|
|
|
|
|
HANDLE hToken,
|
|
|
|
|
BOOL bInherit
|
|
|
|
|
);
|
|
|
|
|
*/
|
|
|
|
|
|
|
|
|
|
func CreateEnvironmentBlock(lpEnvironment *HANDLE, hToken TOKEN, bInherit uintptr) error {
|
2026-06-06 17:46:02 +08:00
|
|
|
if lpEnvironment == nil {
|
|
|
|
|
return syscall.EINVAL
|
|
|
|
|
}
|
|
|
|
|
Dup, err := getProcAddr("userenv.dll", "CreateEnvironmentBlock")
|
2019-03-07 11:39:54 +08:00
|
|
|
if err != nil {
|
2026-06-06 17:46:02 +08:00
|
|
|
return err
|
|
|
|
|
}
|
|
|
|
|
r, _, errno := syscall.Syscall6(Dup, 3, uintptr(unsafe.Pointer(lpEnvironment)), uintptr(hToken), bInherit, 0, 0, 0)
|
|
|
|
|
if r == 0 {
|
|
|
|
|
if errno != 0 {
|
|
|
|
|
return error(errno)
|
|
|
|
|
}
|
|
|
|
|
return syscall.EINVAL
|
2019-03-07 11:39:54 +08:00
|
|
|
}
|
2026-06-06 17:46:02 +08:00
|
|
|
return nil
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func DestroyEnvironmentBlock(lpEnvironment HANDLE) error {
|
|
|
|
|
proc, err := getProcAddr("userenv.dll", "DestroyEnvironmentBlock")
|
2019-03-07 11:39:54 +08:00
|
|
|
if err != nil {
|
2026-06-06 17:46:02 +08:00
|
|
|
return err
|
2019-03-07 11:39:54 +08:00
|
|
|
}
|
2026-06-06 17:46:02 +08:00
|
|
|
r, _, errno := syscall.Syscall(proc, 1, uintptr(lpEnvironment), 0, 0)
|
2019-03-07 11:39:54 +08:00
|
|
|
if r == 0 {
|
2026-06-06 17:46:02 +08:00
|
|
|
if errno != 0 {
|
|
|
|
|
return error(errno)
|
|
|
|
|
}
|
|
|
|
|
return syscall.EINVAL
|
2019-03-07 11:39:54 +08:00
|
|
|
}
|
|
|
|
|
return nil
|
|
|
|
|
}
|