win32api/userenv.go

48 lines
941 B
Go
Raw Normal View History

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 {
if lpEnvironment == nil {
return syscall.EINVAL
}
Dup, err := getProcAddr("userenv.dll", "CreateEnvironmentBlock")
2019-03-07 11:39:54 +08:00
if err != nil {
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
}
return nil
}
func DestroyEnvironmentBlock(lpEnvironment HANDLE) error {
proc, err := getProcAddr("userenv.dll", "DestroyEnvironmentBlock")
2019-03-07 11:39:54 +08:00
if err != nil {
return err
2019-03-07 11:39:54 +08:00
}
r, _, errno := syscall.Syscall(proc, 1, uintptr(lpEnvironment), 0, 0)
2019-03-07 11:39:54 +08:00
if r == 0 {
if errno != 0 {
return error(errno)
}
return syscall.EINVAL
2019-03-07 11:39:54 +08:00
}
return nil
}