138 lines
2.8 KiB
Go
138 lines
2.8 KiB
Go
//go:build openbsd
|
|
// +build openbsd
|
|
|
|
package cpu
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"runtime"
|
|
"unsafe"
|
|
|
|
"github.com/shirou/gopsutil/v3/internal/common"
|
|
"github.com/tklauser/go-sysconf"
|
|
"golang.org/x/sys/unix"
|
|
)
|
|
|
|
const (
|
|
// sys/sched.h
|
|
cpuOnline = 0x0001 // CPUSTATS_ONLINE
|
|
|
|
// sys/sysctl.h
|
|
ctlKern = 1 // "high kernel": proc, limits
|
|
ctlHw = 6 // CTL_HW
|
|
smt = 24 // HW_SMT
|
|
kernCpTime = 40 // KERN_CPTIME
|
|
kernCPUStats = 85 // KERN_CPUSTATS
|
|
)
|
|
|
|
var ClocksPerSec = float64(128)
|
|
|
|
type cpuStats struct {
|
|
// cs_time[CPUSTATES]
|
|
User uint64
|
|
Nice uint64
|
|
Sys uint64
|
|
Spin uint64
|
|
Intr uint64
|
|
Idle uint64
|
|
|
|
// cs_flags
|
|
Flags uint64
|
|
}
|
|
|
|
func init() {
|
|
clkTck, err := sysconf.Sysconf(sysconf.SC_CLK_TCK)
|
|
// ignore errors
|
|
if err == nil {
|
|
ClocksPerSec = float64(clkTck)
|
|
}
|
|
}
|
|
|
|
func Times(percpu bool) ([]TimesStat, error) {
|
|
return TimesWithContext(context.Background(), percpu)
|
|
}
|
|
|
|
func TimesWithContext(ctx context.Context, percpu bool) (ret []TimesStat, err error) {
|
|
if !percpu {
|
|
mib := []int32{ctlKern, kernCpTime}
|
|
buf, _, err := common.CallSyscall(mib)
|
|
if err != nil {
|
|
return ret, err
|
|
}
|
|
times := (*cpuTimes)(unsafe.Pointer(&buf[0]))
|
|
stat := TimesStat{
|
|
CPU: "cpu-total",
|
|
User: float64(times.User) / ClocksPerSec,
|
|
Nice: float64(times.Nice) / ClocksPerSec,
|
|
System: float64(times.Sys) / ClocksPerSec,
|
|
Idle: float64(times.Idle) / ClocksPerSec,
|
|
Irq: float64(times.Intr) / ClocksPerSec,
|
|
}
|
|
return []TimesStat{stat}, nil
|
|
}
|
|
|
|
ncpu, err := unix.SysctlUint32("hw.ncpu")
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
var i uint32
|
|
for i = 0; i < ncpu; i++ {
|
|
mib := []int32{ctlKern, kernCPUStats, int32(i)}
|
|
buf, _, err := common.CallSyscall(mib)
|
|
if err != nil {
|
|
return ret, err
|
|
}
|
|
|
|
stats := (*cpuStats)(unsafe.Pointer(&buf[0]))
|
|
if (stats.Flags & cpuOnline) == 0 {
|
|
continue
|
|
}
|
|
ret = append(ret, TimesStat{
|
|
CPU: fmt.Sprintf("cpu%d", i),
|
|
User: float64(stats.User) / ClocksPerSec,
|
|
Nice: float64(stats.Nice) / ClocksPerSec,
|
|
System: float64(stats.Sys) / ClocksPerSec,
|
|
Idle: float64(stats.Idle) / ClocksPerSec,
|
|
Irq: float64(stats.Intr) / ClocksPerSec,
|
|
})
|
|
}
|
|
|
|
return ret, nil
|
|
}
|
|
|
|
// Returns only one (minimal) CPUInfoStat on OpenBSD
|
|
func Info() ([]InfoStat, error) {
|
|
return InfoWithContext(context.Background())
|
|
}
|
|
|
|
func InfoWithContext(ctx context.Context) ([]InfoStat, error) {
|
|
var ret []InfoStat
|
|
var err error
|
|
|
|
c := InfoStat{}
|
|
|
|
mhz, err := unix.SysctlUint32("hw.cpuspeed")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
c.Mhz = float64(mhz)
|
|
|
|
ncpu, err := unix.SysctlUint32("hw.ncpuonline")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
c.Cores = int32(ncpu)
|
|
|
|
if c.ModelName, err = unix.Sysctl("hw.model"); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return append(ret, c), nil
|
|
}
|
|
|
|
func CountsWithContext(ctx context.Context, logical bool) (int, error) {
|
|
return runtime.NumCPU(), nil
|
|
}
|