star/gdu/pkg/device/dev_bsd.go

76 lines
1.6 KiB
Go
Raw Normal View History

2025-04-26 19:33:14 +08:00
//go:build netbsd || openbsd
// +build netbsd openbsd
package device
import (
"bufio"
"bytes"
"errors"
"io"
"os/exec"
"regexp"
"strings"
)
// BSDDevicesInfoGetter returns info for Darwin devices
type BSDDevicesInfoGetter struct {
MountCmd string
}
// Getter is current instance of DevicesInfoGetter
var Getter DevicesInfoGetter = BSDDevicesInfoGetter{MountCmd: "/sbin/mount"}
// GetMounts returns all mounted filesystems from output of /sbin/mount
func (t BSDDevicesInfoGetter) GetMounts() (Devices, error) {
out, err := exec.Command(t.MountCmd).Output()
if err != nil {
return nil, err
}
rdr := bytes.NewReader(out)
return readMountOutput(rdr)
}
// GetDevicesInfo returns result of GetMounts with usage info about mounted devices (by calling Statfs syscall)
func (t BSDDevicesInfoGetter) GetDevicesInfo() (Devices, error) {
mounts, err := t.GetMounts()
if err != nil {
return nil, err
}
return processMounts(mounts, false)
}
func readMountOutput(rdr io.Reader) (Devices, error) {
mounts := Devices{}
scanner := bufio.NewScanner(rdr)
for scanner.Scan() {
line := scanner.Text()
re := regexp.MustCompile("^(.*) on (/.*) type (.*) \\(([^)]+)\\)$")
parts := re.FindAllStringSubmatch(line, -1)
if len(parts) < 1 {
return nil, errors.New("Cannot parse mount output")
}
fstype := strings.TrimSpace(strings.Split(parts[0][3], ",")[0])
device := &Device{
Name: parts[0][1],
MountPoint: parts[0][2],
Fstype: fstype,
}
mounts = append(mounts, device)
}
if err := scanner.Err(); err != nil {
return nil, err
}
return mounts, nil
}