212 lines
4.0 KiB
Go
212 lines
4.0 KiB
Go
// +build freebsd
|
|
|
|
package host
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/binary"
|
|
"io/ioutil"
|
|
"os"
|
|
"os/exec"
|
|
"runtime"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
"unsafe"
|
|
|
|
"github.com/shirou/gopsutil/internal/common"
|
|
"github.com/shirou/gopsutil/process"
|
|
)
|
|
|
|
const (
|
|
UTNameSize = 16 /* see MAXLOGNAME in <sys/param.h> */
|
|
UTLineSize = 8
|
|
UTHostSize = 16
|
|
)
|
|
|
|
func Info() (*InfoStat, error) {
|
|
ret := &InfoStat{
|
|
OS: runtime.GOOS,
|
|
PlatformFamily: "freebsd",
|
|
}
|
|
|
|
hostname, err := os.Hostname()
|
|
if err == nil {
|
|
ret.Hostname = hostname
|
|
}
|
|
|
|
platform, family, version, err := PlatformInformation()
|
|
if err == nil {
|
|
ret.Platform = platform
|
|
ret.PlatformFamily = family
|
|
ret.PlatformVersion = version
|
|
ret.KernelVersion = version
|
|
}
|
|
|
|
system, role, err := Virtualization()
|
|
if err == nil {
|
|
ret.VirtualizationSystem = system
|
|
ret.VirtualizationRole = role
|
|
}
|
|
|
|
boot, err := BootTime()
|
|
if err == nil {
|
|
ret.BootTime = boot
|
|
ret.Uptime = uptime(boot)
|
|
}
|
|
|
|
procs, err := process.Pids()
|
|
if err == nil {
|
|
ret.Procs = uint64(len(procs))
|
|
}
|
|
|
|
values, err := common.DoSysctrl("kern.hostuuid")
|
|
if err == nil && len(values) == 1 && values[0] != "" {
|
|
ret.HostID = values[0]
|
|
}
|
|
|
|
return ret, nil
|
|
}
|
|
|
|
func BootTime() (uint64, error) {
|
|
if cachedBootTime != 0 {
|
|
return cachedBootTime, nil
|
|
}
|
|
values, err := common.DoSysctrl("kern.boottime")
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
// ex: { sec = 1392261637, usec = 627534 } Thu Feb 13 12:20:37 2014
|
|
v := strings.Replace(values[2], ",", "", 1)
|
|
|
|
boottime, err := strconv.ParseUint(v, 10, 64)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
cachedBootTime = boottime
|
|
|
|
return boottime, nil
|
|
}
|
|
|
|
func uptime(boot uint64) uint64 {
|
|
return uint64(time.Now().Unix()) - boot
|
|
}
|
|
|
|
func Uptime() (uint64, error) {
|
|
boot, err := BootTime()
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
return uptime(boot), nil
|
|
}
|
|
|
|
func Users() ([]UserStat, error) {
|
|
utmpfile := "/var/run/utx.active"
|
|
if !common.PathExists(utmpfile) {
|
|
utmpfile = "/var/run/utmp" // before 9.0
|
|
return getUsersFromUtmp(utmpfile)
|
|
}
|
|
|
|
var ret []UserStat
|
|
file, err := os.Open(utmpfile)
|
|
if err != nil {
|
|
return ret, err
|
|
}
|
|
|
|
buf, err := ioutil.ReadAll(file)
|
|
if err != nil {
|
|
return ret, err
|
|
}
|
|
|
|
entrySize := sizeOfUtmpx
|
|
count := len(buf) / entrySize
|
|
|
|
for i := 0; i < count; i++ {
|
|
b := buf[i*sizeOfUtmpx : (i+1)*sizeOfUtmpx]
|
|
var u Utmpx
|
|
br := bytes.NewReader(b)
|
|
err := binary.Read(br, binary.LittleEndian, &u)
|
|
if err != nil || u.Type != 4 {
|
|
continue
|
|
}
|
|
sec := (binary.LittleEndian.Uint32(u.Tv.Sec[:])) / 2 // TODO:
|
|
user := UserStat{
|
|
User: common.IntToString(u.User[:]),
|
|
Terminal: common.IntToString(u.Line[:]),
|
|
Host: common.IntToString(u.Host[:]),
|
|
Started: int(sec),
|
|
}
|
|
|
|
ret = append(ret, user)
|
|
}
|
|
|
|
return ret, nil
|
|
|
|
}
|
|
|
|
func PlatformInformation() (string, string, string, error) {
|
|
platform := ""
|
|
family := ""
|
|
version := ""
|
|
uname, err := exec.LookPath("uname")
|
|
if err != nil {
|
|
return "", "", "", err
|
|
}
|
|
|
|
out, err := invoke.Command(uname, "-s")
|
|
if err == nil {
|
|
platform = strings.ToLower(strings.TrimSpace(string(out)))
|
|
}
|
|
|
|
out, err = invoke.Command(uname, "-r")
|
|
if err == nil {
|
|
version = strings.ToLower(strings.TrimSpace(string(out)))
|
|
}
|
|
|
|
return platform, family, version, nil
|
|
}
|
|
|
|
func Virtualization() (string, string, error) {
|
|
system := ""
|
|
role := ""
|
|
|
|
return system, role, nil
|
|
}
|
|
|
|
// before 9.0
|
|
func getUsersFromUtmp(utmpfile string) ([]UserStat, error) {
|
|
var ret []UserStat
|
|
file, err := os.Open(utmpfile)
|
|
if err != nil {
|
|
return ret, err
|
|
}
|
|
buf, err := ioutil.ReadAll(file)
|
|
if err != nil {
|
|
return ret, err
|
|
}
|
|
|
|
u := Utmp{}
|
|
entrySize := int(unsafe.Sizeof(u))
|
|
count := len(buf) / entrySize
|
|
|
|
for i := 0; i < count; i++ {
|
|
b := buf[i*entrySize : i*entrySize+entrySize]
|
|
var u Utmp
|
|
br := bytes.NewReader(b)
|
|
err := binary.Read(br, binary.LittleEndian, &u)
|
|
if err != nil || u.Time == 0 {
|
|
continue
|
|
}
|
|
user := UserStat{
|
|
User: common.IntToString(u.Name[:]),
|
|
Terminal: common.IntToString(u.Line[:]),
|
|
Host: common.IntToString(u.Host[:]),
|
|
Started: int(u.Time),
|
|
}
|
|
|
|
ret = append(ret, user)
|
|
}
|
|
|
|
return ret, nil
|
|
}
|