OrgGo/shirou_gopsutil
1
0
Fork 0
mirror of https://github.com/shirou/gopsutil.git synced 2025-04-29 13:49:21 +08:00
Code Issues Projects Releases Wiki Activity

Merge pull request #1045 from Lomanic/issue714

Browse Source 
[process][windows] Fix #714 call OpenProcess with PROCESS_QUERY_INFORMATION on WinXP
This commit is contained in:
shirou 2021-03-07 17:15:37 +09:00 committed by GitHub
commit c637487c2c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 22 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
process/process_windows.go
View File

@ -38,6 +38,8 @@ var (
processorArchitecture uint processorArchitecture uint
) )
const processQueryInformation = windows.PROCESS_QUERY_LIMITED_INFORMATION | windows.PROCESS_QUERY_INFORMATION // WinXP doesn't know PROCESS_QUERY_LIMITED_INFORMATION
type SystemProcessInformation struct { type SystemProcessInformation struct {
NextEntryOffset uint64 NextEntryOffset uint64
NumberOfThreads uint64 NumberOfThreads uint64
@ -219,7 +221,7 @@ func PidExistsWithContext(ctx context.Context, pid int32) (bool, error) {
return false, err return false, err
} }
const STILL_ACTIVE = 259 // https://docs.microsoft.com/en-us/windows/win32/api/processthreadsapi/nf-processthreadsapi-getexitcodeprocess const STILL_ACTIVE = 259 // https://docs.microsoft.com/en-us/windows/win32/api/processthreadsapi/nf-processthreadsapi-getexitcodeprocess
h, err := windows.OpenProcess(windows.PROCESS_QUERY_LIMITED_INFORMATION, false, uint32(pid)) h, err := windows.OpenProcess(processQueryInformation, false, uint32(pid))
if err == windows.ERROR_ACCESS_DENIED { if err == windows.ERROR_ACCESS_DENIED {
return true, nil return true, nil
} }
@ -273,7 +275,7 @@ func (p *Process) TgidWithContext(ctx context.Context) (int32, error) {
} }
func (p *Process) ExeWithContext(ctx context.Context) (string, error) { func (p *Process) ExeWithContext(ctx context.Context) (string, error) {
c, err := windows.OpenProcess(windows.PROCESS_QUERY_LIMITED_INFORMATION, false, uint32(p.Pid)) c, err := windows.OpenProcess(processQueryInformation, false, uint32(p.Pid))
if err != nil { if err != nil {
return "", err return "", err
} }
@ -347,7 +349,7 @@ func (p *Process) ForegroundWithContext(ctx context.Context) (bool, error) {
func (p *Process) UsernameWithContext(ctx context.Context) (string, error) { func (p *Process) UsernameWithContext(ctx context.Context) (string, error) {
pid := p.Pid pid := p.Pid
c, err := windows.OpenProcess(windows.PROCESS_QUERY_LIMITED_INFORMATION, false, uint32(pid)) c, err := windows.OpenProcess(processQueryInformation, false, uint32(pid))
if err != nil { if err != nil {
return "", err return "", err
} }
@ -397,7 +399,7 @@ var priorityClasses = map[int]int32{
} }
func (p *Process) NiceWithContext(ctx context.Context) (int32, error) { func (p *Process) NiceWithContext(ctx context.Context) (int32, error) {
c, err := windows.OpenProcess(windows.PROCESS_QUERY_LIMITED_INFORMATION, false, uint32(p.Pid)) c, err := windows.OpenProcess(processQueryInformation, false, uint32(p.Pid))
if err != nil { if err != nil {
return 0, err return 0, err
} }
@ -426,7 +428,7 @@ func (p *Process) RlimitUsageWithContext(ctx context.Context, gatherUsed bool) (
} }
func (p *Process) IOCountersWithContext(ctx context.Context) (*IOCountersStat, error) { func (p *Process) IOCountersWithContext(ctx context.Context) (*IOCountersStat, error) {
c, err := windows.OpenProcess(windows.PROCESS_QUERY_LIMITED_INFORMATION, false, uint32(p.Pid)) c, err := windows.OpenProcess(processQueryInformation, false, uint32(p.Pid))
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -680,7 +682,7 @@ func ProcessesWithContext(ctx context.Context) ([]*Process, error) {
func getRusage(pid int32) (*windows.Rusage, error) { func getRusage(pid int32) (*windows.Rusage, error) {
var CPU windows.Rusage var CPU windows.Rusage
c, err := windows.OpenProcess(windows.PROCESS_QUERY_LIMITED_INFORMATION, false, uint32(pid)) c, err := windows.OpenProcess(processQueryInformation, false, uint32(pid))
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -695,7 +697,7 @@ func getRusage(pid int32) (*windows.Rusage, error) {
func getMemoryInfo(pid int32) (PROCESS_MEMORY_COUNTERS, error) { func getMemoryInfo(pid int32) (PROCESS_MEMORY_COUNTERS, error) {
var mem PROCESS_MEMORY_COUNTERS var mem PROCESS_MEMORY_COUNTERS
c, err := windows.OpenProcess(windows.PROCESS_QUERY_LIMITED_INFORMATION, false, uint32(pid)) c, err := windows.OpenProcess(processQueryInformation, false, uint32(pid))
if err != nil { if err != nil {
return mem, err return mem, err
} }
@ -729,7 +731,7 @@ type SYSTEM_TIMES struct {
func getProcessCPUTimes(pid int32) (SYSTEM_TIMES, error) { func getProcessCPUTimes(pid int32) (SYSTEM_TIMES, error) {
var times SYSTEM_TIMES var times SYSTEM_TIMES
h, err := windows.OpenProcess(windows.PROCESS_QUERY_LIMITED_INFORMATION, false, uint32(pid)) h, err := windows.OpenProcess(processQueryInformation, false, uint32(pid))
if err != nil { if err != nil {
return times, err return times, err
} }
@ -770,7 +772,7 @@ func is32BitProcess(procHandle syscall.Handle) bool {
} }
func getProcessCommandLine(pid int32) (string, error) { func getProcessCommandLine(pid int32) (string, error) {
h, err := windows.OpenProcess(windows.PROCESS_QUERY_LIMITED_INFORMATION|windows.PROCESS_VM_READ, false, uint32(pid)) h, err := windows.OpenProcess(processQueryInformation|windows.PROCESS_VM_READ, false, uint32(pid))
if err == windows.ERROR_ACCESS_DENIED || err == windows.ERROR_INVALID_PARAMETER { if err == windows.ERROR_ACCESS_DENIED || err == windows.ERROR_INVALID_PARAMETER {
return "", nil return "", nil
} }

20
v3/process/process_windows.go
View File

@ -38,6 +38,8 @@ var (
processorArchitecture uint processorArchitecture uint
) )
const processQueryInformation = windows.PROCESS_QUERY_LIMITED_INFORMATION | windows.PROCESS_QUERY_INFORMATION // WinXP doesn't know PROCESS_QUERY_LIMITED_INFORMATION
type systemProcessorInformation struct { type systemProcessorInformation struct {
ProcessorArchitecture uint16 ProcessorArchitecture uint16
ProcessorLevel uint16 ProcessorLevel uint16
@ -204,7 +206,7 @@ func PidExistsWithContext(ctx context.Context, pid int32) (bool, error) {
return false, err return false, err
} }
const STILL_ACTIVE = 259 // https://docs.microsoft.com/en-us/windows/win32/api/processthreadsapi/nf-processthreadsapi-getexitcodeprocess const STILL_ACTIVE = 259 // https://docs.microsoft.com/en-us/windows/win32/api/processthreadsapi/nf-processthreadsapi-getexitcodeprocess
h, err := windows.OpenProcess(windows.PROCESS_QUERY_LIMITED_INFORMATION, false, uint32(pid)) h, err := windows.OpenProcess(processQueryInformation, false, uint32(pid))
if err == windows.ERROR_ACCESS_DENIED { if err == windows.ERROR_ACCESS_DENIED {
return true, nil return true, nil
} }
@ -258,7 +260,7 @@ func (p *Process) TgidWithContext(ctx context.Context) (int32, error) {
} }
func (p *Process) ExeWithContext(ctx context.Context) (string, error) { func (p *Process) ExeWithContext(ctx context.Context) (string, error) {
c, err := windows.OpenProcess(windows.PROCESS_QUERY_LIMITED_INFORMATION, false, uint32(p.Pid)) c, err := windows.OpenProcess(processQueryInformation, false, uint32(p.Pid))
if err != nil { if err != nil {
return "", err return "", err
} }
@ -332,7 +334,7 @@ func (p *Process) ForegroundWithContext(ctx context.Context) (bool, error) {
func (p *Process) UsernameWithContext(ctx context.Context) (string, error) { func (p *Process) UsernameWithContext(ctx context.Context) (string, error) {
pid := p.Pid pid := p.Pid
c, err := windows.OpenProcess(windows.PROCESS_QUERY_LIMITED_INFORMATION, false, uint32(pid)) c, err := windows.OpenProcess(processQueryInformation, false, uint32(pid))
if err != nil { if err != nil {
return "", err return "", err
} }
@ -382,7 +384,7 @@ var priorityClasses = map[int]int32{
} }
func (p *Process) NiceWithContext(ctx context.Context) (int32, error) { func (p *Process) NiceWithContext(ctx context.Context) (int32, error) {
c, err := windows.OpenProcess(windows.PROCESS_QUERY_LIMITED_INFORMATION, false, uint32(p.Pid)) c, err := windows.OpenProcess(processQueryInformation, false, uint32(p.Pid))
if err != nil { if err != nil {
return 0, err return 0, err
} }
@ -411,7 +413,7 @@ func (p *Process) RlimitUsageWithContext(ctx context.Context, gatherUsed bool) (
} }
func (p *Process) IOCountersWithContext(ctx context.Context) (*IOCountersStat, error) { func (p *Process) IOCountersWithContext(ctx context.Context) (*IOCountersStat, error) {
c, err := windows.OpenProcess(windows.PROCESS_QUERY_LIMITED_INFORMATION, false, uint32(p.Pid)) c, err := windows.OpenProcess(processQueryInformation, false, uint32(p.Pid))
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -661,7 +663,7 @@ func ProcessesWithContext(ctx context.Context) ([]*Process, error) {
func getRusage(pid int32) (*windows.Rusage, error) { func getRusage(pid int32) (*windows.Rusage, error) {
var CPU windows.Rusage var CPU windows.Rusage
c, err := windows.OpenProcess(windows.PROCESS_QUERY_LIMITED_INFORMATION, false, uint32(pid)) c, err := windows.OpenProcess(processQueryInformation, false, uint32(pid))
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -676,7 +678,7 @@ func getRusage(pid int32) (*windows.Rusage, error) {
func getMemoryInfo(pid int32) (PROCESS_MEMORY_COUNTERS, error) { func getMemoryInfo(pid int32) (PROCESS_MEMORY_COUNTERS, error) {
var mem PROCESS_MEMORY_COUNTERS var mem PROCESS_MEMORY_COUNTERS
c, err := windows.OpenProcess(windows.PROCESS_QUERY_LIMITED_INFORMATION, false, uint32(pid)) c, err := windows.OpenProcess(processQueryInformation, false, uint32(pid))
if err != nil { if err != nil {
return mem, err return mem, err
} }
@ -710,7 +712,7 @@ type SYSTEM_TIMES struct {
func getProcessCPUTimes(pid int32) (SYSTEM_TIMES, error) { func getProcessCPUTimes(pid int32) (SYSTEM_TIMES, error) {
var times SYSTEM_TIMES var times SYSTEM_TIMES
h, err := windows.OpenProcess(windows.PROCESS_QUERY_LIMITED_INFORMATION, false, uint32(pid)) h, err := windows.OpenProcess(processQueryInformation, false, uint32(pid))
if err != nil { if err != nil {
return times, err return times, err
} }
@ -751,7 +753,7 @@ func is32BitProcess(procHandle syscall.Handle) bool {
} }
func getProcessCommandLine(pid int32) (string, error) { func getProcessCommandLine(pid int32) (string, error) {
h, err := windows.OpenProcess(windows.PROCESS_QUERY_LIMITED_INFORMATION|windows.PROCESS_VM_READ, false, uint32(pid)) h, err := windows.OpenProcess(processQueryInformation|windows.PROCESS_VM_READ, false, uint32(pid))
if err == windows.ERROR_ACCESS_DENIED || err == windows.ERROR_INVALID_PARAMETER { if err == windows.ERROR_ACCESS_DENIED || err == windows.ERROR_INVALID_PARAMETER {
return "", nil return "", nil
} }