mirror of https://github.com/hashicorp/consul
86 lines
1.6 KiB
Go
86 lines
1.6 KiB
Go
// +build freebsd openbsd
|
|
|
|
package common
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"os/exec"
|
|
"strings"
|
|
"unsafe"
|
|
|
|
"golang.org/x/sys/unix"
|
|
)
|
|
|
|
func SysctlUint(mib string) (uint64, error) {
|
|
buf, err := unix.SysctlRaw(mib)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
if len(buf) == 8 { // 64 bit
|
|
return *(*uint64)(unsafe.Pointer(&buf[0])), nil
|
|
}
|
|
if len(buf) == 4 { // 32bit
|
|
t := *(*uint32)(unsafe.Pointer(&buf[0]))
|
|
return uint64(t), nil
|
|
}
|
|
return 0, fmt.Errorf("unexpected size: %s, %d", mib, len(buf))
|
|
}
|
|
|
|
func DoSysctrl(mib string) ([]string, error) {
|
|
sysctl, err := exec.LookPath("sysctl")
|
|
if err != nil {
|
|
return []string{}, err
|
|
}
|
|
cmd := exec.Command(sysctl, "-n", mib)
|
|
cmd.Env = getSysctrlEnv(os.Environ())
|
|
out, err := cmd.Output()
|
|
if err != nil {
|
|
return []string{}, err
|
|
}
|
|
v := strings.Replace(string(out), "{ ", "", 1)
|
|
v = strings.Replace(string(v), " }", "", 1)
|
|
values := strings.Fields(string(v))
|
|
|
|
return values, nil
|
|
}
|
|
|
|
func CallSyscall(mib []int32) ([]byte, uint64, error) {
|
|
mibptr := unsafe.Pointer(&mib[0])
|
|
miblen := uint64(len(mib))
|
|
|
|
// get required buffer size
|
|
length := uint64(0)
|
|
_, _, err := unix.Syscall6(
|
|
unix.SYS___SYSCTL,
|
|
uintptr(mibptr),
|
|
uintptr(miblen),
|
|
0,
|
|
uintptr(unsafe.Pointer(&length)),
|
|
0,
|
|
0)
|
|
if err != 0 {
|
|
var b []byte
|
|
return b, length, err
|
|
}
|
|
if length == 0 {
|
|
var b []byte
|
|
return b, length, err
|
|
}
|
|
// get proc info itself
|
|
buf := make([]byte, length)
|
|
_, _, err = unix.Syscall6(
|
|
unix.SYS___SYSCTL,
|
|
uintptr(mibptr),
|
|
uintptr(miblen),
|
|
uintptr(unsafe.Pointer(&buf[0])),
|
|
uintptr(unsafe.Pointer(&length)),
|
|
0,
|
|
0)
|
|
if err != 0 {
|
|
return buf, length, err
|
|
}
|
|
|
|
return buf, length, nil
|
|
}
|