|
|
|
@ -30,8 +30,8 @@ func CloseHandle(hObject HANDLE) error {
|
|
|
|
|
if err != nil {
|
|
|
|
|
return errors.New("Can't Load CloseHandle API")
|
|
|
|
|
}
|
|
|
|
|
if r, _, err := syscall.Syscall(uintptr(CH), 1, uintptr(hObject), 0, 0); r == 0 {
|
|
|
|
|
return err
|
|
|
|
|
if r, _, errno := syscall.Syscall(uintptr(CH), 1, uintptr(hObject), 0, 0); r == 0 {
|
|
|
|
|
return error(errno)
|
|
|
|
|
}
|
|
|
|
|
return nil
|
|
|
|
|
}
|
|
|
|
@ -46,9 +46,9 @@ func CreateToolhelp32Snapshot(dwFlags, th32ProcessID DWORD) (HANDLE, error) {
|
|
|
|
|
if err != nil {
|
|
|
|
|
return 0, errors.New("Can't Load CreateToolhelp32Snapshot API")
|
|
|
|
|
}
|
|
|
|
|
r, _, err := syscall.Syscall(uintptr(CTS), 2, uintptr(dwFlags), uintptr(th32ProcessID), 0)
|
|
|
|
|
r, _, errno := syscall.Syscall(uintptr(CTS), 2, uintptr(dwFlags), uintptr(th32ProcessID), 0)
|
|
|
|
|
if int(r) == -1 {
|
|
|
|
|
return 0, err
|
|
|
|
|
return 0, error(errno)
|
|
|
|
|
}
|
|
|
|
|
return HANDLE(r), nil
|
|
|
|
|
}
|
|
|
|
@ -63,9 +63,12 @@ func Process32Next(hSnapshot HANDLE, lppe *PROCESSENTRY32) error {
|
|
|
|
|
if err != nil {
|
|
|
|
|
return errors.New("Can't Load Process32Next API")
|
|
|
|
|
}
|
|
|
|
|
r, _, err := syscall.Syscall(uintptr(PN), 2, uintptr(hSnapshot), uintptr(unsafe.Pointer(lppe)), 0)
|
|
|
|
|
if int(r) != 1 {
|
|
|
|
|
return err
|
|
|
|
|
r, _, errno := syscall.Syscall(uintptr(PN), 2, uintptr(hSnapshot), uintptr(unsafe.Pointer(lppe)), 0)
|
|
|
|
|
if r == 0 {
|
|
|
|
|
if errno != 0 {
|
|
|
|
|
return error(errno)
|
|
|
|
|
}
|
|
|
|
|
return syscall.EINVAL
|
|
|
|
|
}
|
|
|
|
|
return nil
|
|
|
|
|
}
|
|
|
|
@ -109,10 +112,63 @@ func GlobalMemoryStatusEx(data *MEMORYSTATUSEX) (bool, error) {
|
|
|
|
|
if err != nil {
|
|
|
|
|
return false, err
|
|
|
|
|
}
|
|
|
|
|
r, _, err := syscall.Syscall(uintptr(GMS), 1, uintptr(unsafe.Pointer(data)), 0, 0)
|
|
|
|
|
r, _, errno := syscall.Syscall(uintptr(GMS), 1, uintptr(unsafe.Pointer(data)), 0, 0)
|
|
|
|
|
if r == 0 {
|
|
|
|
|
if errno != 0 {
|
|
|
|
|
return false, error(errno)
|
|
|
|
|
}
|
|
|
|
|
return false, syscall.EINVAL
|
|
|
|
|
}
|
|
|
|
|
return true, nil
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func LockFileEx(hFile HANDLE, dwFlags DWORD, dwReserved DWORD, nNumberOfBytesToLockLow DWORD,
|
|
|
|
|
nNumberOfBytesToLockHigh DWORD, lpOverlapped *syscall.Overlapped) (bool, error) {
|
|
|
|
|
kernel32, err := syscall.LoadLibrary("kernel32.dll")
|
|
|
|
|
if err != nil {
|
|
|
|
|
return false, err
|
|
|
|
|
} else {
|
|
|
|
|
return true, nil
|
|
|
|
|
}
|
|
|
|
|
defer syscall.FreeLibrary(kernel32)
|
|
|
|
|
Lck, err := syscall.GetProcAddress(syscall.Handle(kernel32), "LockFileEx")
|
|
|
|
|
if err != nil {
|
|
|
|
|
return false, err
|
|
|
|
|
}
|
|
|
|
|
r, _, errno := syscall.Syscall6(uintptr(Lck), 6, uintptr(hFile), uintptr(dwFlags), uintptr(dwReserved),
|
|
|
|
|
uintptr(nNumberOfBytesToLockLow), uintptr(nNumberOfBytesToLockHigh), uintptr(unsafe.Pointer(lpOverlapped)))
|
|
|
|
|
if r == 0 {
|
|
|
|
|
if errno != 0 {
|
|
|
|
|
return false, error(errno)
|
|
|
|
|
}
|
|
|
|
|
return false, syscall.EINVAL
|
|
|
|
|
}
|
|
|
|
|
return true, nil
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func CreateEventW(lpEventAttributes *syscall.SecurityAttributes, bManualReset bool,
|
|
|
|
|
bInitialState bool, lpName LPCWSTR) (HANDLE, error) {
|
|
|
|
|
var intBManualReset, intBInitialState int
|
|
|
|
|
if bManualReset {
|
|
|
|
|
intBManualReset = 1
|
|
|
|
|
}
|
|
|
|
|
if bInitialState {
|
|
|
|
|
intBInitialState = 1
|
|
|
|
|
}
|
|
|
|
|
kernel32, err := syscall.LoadLibrary("kernel32.dll")
|
|
|
|
|
if err != nil {
|
|
|
|
|
return 0, err
|
|
|
|
|
}
|
|
|
|
|
defer syscall.FreeLibrary(kernel32)
|
|
|
|
|
Lck, err := syscall.GetProcAddress(syscall.Handle(kernel32), "CreateEventW")
|
|
|
|
|
if err != nil {
|
|
|
|
|
return 0, err
|
|
|
|
|
}
|
|
|
|
|
r, _, errno := syscall.Syscall6(uintptr(Lck), 4, uintptr(unsafe.Pointer(lpEventAttributes)),
|
|
|
|
|
uintptr(intBManualReset), uintptr(intBInitialState), uintptr(unsafe.Pointer(lpName)), 0, 0)
|
|
|
|
|
if HANDLE(r) == 0 {
|
|
|
|
|
if errno != 0 {
|
|
|
|
|
return HANDLE(r), error(errno)
|
|
|
|
|
}
|
|
|
|
|
return HANDLE(r), syscall.EINVAL
|
|
|
|
|
}
|
|
|
|
|
return HANDLE(r), nil
|
|
|
|
|
}
|
|
|
|
|