123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119 |
- package os
- import (
- "errors"
- "runtime"
- "syscall"
- "time"
- "unsafe"
- )
- func (p *Process) wait() (ps *ProcessState, err error) {
- s, e := syscall.WaitForSingleObject(syscall.Handle(p.handle), syscall.INFINITE)
- switch s {
- case syscall.WAIT_OBJECT_0:
- break
- case syscall.WAIT_FAILED:
- return nil, NewSyscallError("WaitForSingleObject", e)
- default:
- return nil, errors.New("os: unexpected result from WaitForSingleObject")
- }
- var ec uint32
- e = syscall.GetExitCodeProcess(syscall.Handle(p.handle), &ec)
- if e != nil {
- return nil, NewSyscallError("GetExitCodeProcess", e)
- }
- var u syscall.Rusage
- e = syscall.GetProcessTimes(syscall.Handle(p.handle), &u.CreationTime, &u.ExitTime, &u.KernelTime, &u.UserTime)
- if e != nil {
- return nil, NewSyscallError("GetProcessTimes", e)
- }
- p.setDone()
-
-
-
-
- defer time.Sleep(5 * time.Millisecond)
- defer p.Release()
- return &ProcessState{p.Pid, syscall.WaitStatus{ExitCode: ec}, &u}, nil
- }
- func terminateProcess(pid, exitcode int) error {
- h, e := syscall.OpenProcess(syscall.PROCESS_TERMINATE, false, uint32(pid))
- if e != nil {
- return NewSyscallError("OpenProcess", e)
- }
- defer syscall.CloseHandle(h)
- e = syscall.TerminateProcess(h, uint32(exitcode))
- return NewSyscallError("TerminateProcess", e)
- }
- func (p *Process) signal(sig Signal) error {
- if p.handle == uintptr(syscall.InvalidHandle) {
- return syscall.EINVAL
- }
- if p.done() {
- return errors.New("os: process already finished")
- }
- if sig == Kill {
- return terminateProcess(p.Pid, 1)
- }
-
- return syscall.Errno(syscall.EWINDOWS)
- }
- func (p *Process) release() error {
- if p.handle == uintptr(syscall.InvalidHandle) {
- return syscall.EINVAL
- }
- e := syscall.CloseHandle(syscall.Handle(p.handle))
- if e != nil {
- return NewSyscallError("CloseHandle", e)
- }
- p.handle = uintptr(syscall.InvalidHandle)
-
- runtime.SetFinalizer(p, nil)
- return nil
- }
- func findProcess(pid int) (p *Process, err error) {
- const da = syscall.STANDARD_RIGHTS_READ |
- syscall.PROCESS_QUERY_INFORMATION | syscall.SYNCHRONIZE
- h, e := syscall.OpenProcess(da, false, uint32(pid))
- if e != nil {
- return nil, NewSyscallError("OpenProcess", e)
- }
- return newProcess(pid, uintptr(h)), nil
- }
- func init() {
- var argc int32
- cmd := syscall.GetCommandLine()
- argv, e := syscall.CommandLineToArgv(cmd, &argc)
- if e != nil {
- return
- }
- defer syscall.LocalFree(syscall.Handle(uintptr(unsafe.Pointer(argv))))
- Args = make([]string, argc)
- for i, v := range (*argv)[:argc] {
- Args[i] = string(syscall.UTF16ToString((*v)[:]))
- }
- }
- func ftToDuration(ft *syscall.Filetime) time.Duration {
- n := int64(ft.HighDateTime)<<32 + int64(ft.LowDateTime)
- return time.Duration(n*100) * time.Nanosecond
- }
- func (p *ProcessState) userTime() time.Duration {
- return ftToDuration(&p.rusage.UserTime)
- }
- func (p *ProcessState) systemTime() time.Duration {
- return ftToDuration(&p.rusage.KernelTime)
- }
|