mirror of https://github.com/shirou/gopsutil
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
139 lines
2.5 KiB
Go
139 lines
2.5 KiB
Go
11 years ago
|
// +build freebsd
|
||
|
|
||
|
package gopsutil
|
||
|
|
||
|
import (
|
||
|
"bytes"
|
||
|
"encoding/binary"
|
||
|
"syscall"
|
||
|
"unsafe"
|
||
|
)
|
||
|
|
||
|
// Refresh reloads all the data associated with this process.
|
||
|
func (p *Process) Refresh() error {
|
||
|
|
||
|
mib := []int32{CTL_KERN, KERN_PROC, KERN_PROC_PID, p.Pid}
|
||
|
|
||
|
buf, length, err := call_syscall(mib)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
proc_k := Kinfo_proc{}
|
||
|
if length != uint64(unsafe.Sizeof(proc_k)) {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
k, err := parse_kinfo_proc(buf)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
copy_params(&k, p)
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
func copy_params(k *Kinfo_proc, p *Process) error {
|
||
|
p.Exe = byteToString(k.Ki_comm[:])
|
||
|
p.Ppid = k.Ki_ppid
|
||
|
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
func findProcess(pid int32) (*Process, error) {
|
||
|
mib := []int32{CTL_KERN, KERN_PROC, KERN_PROC_PATHNAME, pid}
|
||
|
|
||
|
_, _, err := call_syscall(mib)
|
||
|
if err != nil {
|
||
|
return &Process{}, err
|
||
|
}
|
||
|
|
||
|
return newProcess(pid)
|
||
|
}
|
||
|
|
||
|
func processes() ([]Process, error) {
|
||
|
results := make([]Process, 0, 50)
|
||
|
|
||
|
mib := []int32{CTL_KERN, KERN_PROC, KERN_PROC_PROC, 0}
|
||
|
buf, length, err := call_syscall(mib)
|
||
|
if err != nil {
|
||
|
return results, err
|
||
|
}
|
||
|
|
||
|
// get kinfo_proc size
|
||
|
k := Kinfo_proc{}
|
||
|
procinfo_len := int(unsafe.Sizeof(k))
|
||
|
count := int(length / uint64(procinfo_len))
|
||
|
|
||
|
// parse buf to procs
|
||
|
for i := 0; i < count; i++ {
|
||
|
b := buf[i*procinfo_len : i*procinfo_len+procinfo_len]
|
||
|
k, err := parse_kinfo_proc(b)
|
||
|
if err != nil {
|
||
|
continue
|
||
|
}
|
||
|
p, err := newProcess(int32(k.Ki_pid))
|
||
|
if err != nil {
|
||
|
continue
|
||
|
}
|
||
|
copy_params(&k, p)
|
||
|
|
||
|
results = append(results, *p)
|
||
|
}
|
||
|
|
||
|
return results, nil
|
||
|
}
|
||
|
|
||
|
func parse_kinfo_proc(buf []byte) (Kinfo_proc, error) {
|
||
|
var k Kinfo_proc
|
||
|
br := bytes.NewReader(buf)
|
||
|
err := binary.Read(br, binary.LittleEndian, &k)
|
||
|
if err != nil {
|
||
|
return k, err
|
||
|
}
|
||
|
|
||
|
return k, nil
|
||
|
}
|
||
|
|
||
|
func call_syscall(mib []int32) ([]byte, uint64, error) {
|
||
|
miblen := uint64(len(mib))
|
||
|
|
||
|
// get required buffer size
|
||
|
length := uint64(0)
|
||
|
_, _, err := syscall.Syscall6(
|
||
|
syscall.SYS___SYSCTL,
|
||
|
uintptr(unsafe.Pointer(&mib[0])),
|
||
|
uintptr(miblen),
|
||
|
0,
|
||
|
uintptr(unsafe.Pointer(&length)),
|
||
|
0,
|
||
|
0)
|
||
|
if err != 0 {
|
||
|
b := make([]byte, 0)
|
||
|
return b, length, err
|
||
|
}
|
||
|
if length == 0 {
|
||
|
b := make([]byte, 0)
|
||
|
return b, length, err
|
||
|
}
|
||
|
// get proc info itself
|
||
|
buf := make([]byte, length)
|
||
|
_, _, err = syscall.Syscall6(
|
||
|
syscall.SYS___SYSCTL,
|
||
|
uintptr(unsafe.Pointer(&mib[0])),
|
||
|
uintptr(miblen),
|
||
|
uintptr(unsafe.Pointer(&buf[0])),
|
||
|
uintptr(unsafe.Pointer(&length)),
|
||
|
0,
|
||
|
0)
|
||
|
if err != 0 {
|
||
|
return buf, length, err
|
||
|
}
|
||
|
|
||
|
return buf, length, nil
|
||
|
}
|
||
|
|
||
|
func newProcess(pid int32) (*Process, error) {
|
||
|
p := &Process{Pid: pid}
|
||
|
return p, p.Refresh()
|
||
|
}
|