init

files.go

@@ -0,0 +1,33 @@
+package staros
+
+import "os"
+
+// 检测文件/文件夹是否存在
+func Exists(path string) bool {
+	_, err := os.Stat(path)
+	if err != nil && os.IsNotExist(err) {
+		return false
+	}
+	return true
+}
+
+
+// IsFile 返回给定文件地址是否是一个文件，
+//True为是一个文件,False为不是文件或路径无效
+func IsFile(fpath string) bool {
+	s, err := os.Stat(fpath)
+	if err != nil {
+		return false
+	}
+	return !s.IsDir()
+}
+
+// IsFolder 返回给定文件地址是否是一个文件夹，
+//True为是一个文件夹,False为不是文件夹或路径无效
+func IsFolder(fpath string) bool {
+	s, err := os.Stat(fpath)
+	if err != nil {
+		return false
+	}
+	return s.IsDir()
+}

os_linux.go

@@ -0,0 +1,66 @@
+// +build linux darwin
+
+package staros
+
+import (
+	"os/user"
+	"strconv"
+	"strings"
+	"syscall"
+	"time"
+)
+
+// StartTime 开机时间
+func StartTime() time.Time {
+	tmp, _ := readAsString("/proc/stat")
+	data := splitBy(ReplaceByte9(tmp), " ")
+	btime, _ := strconv.ParseInt(strings.TrimSpace(data["btime"]), 10, 64)
+	return time.Unix(btime, 0)
+}
+
+// IsRoot 当前是否是管理员用户
+func IsRoot() bool {
+	uid, _ := user.Current()
+	if uid.Uid == "0" {
+		return true
+	}
+	return false
+}
+
+// Memory 系统内存信息
+func Memory() MemStatus {
+	var mem MemStatus
+	ram := new(syscall.Sysinfo_t)
+	if err := syscall.Sysinfo(ram); err != nil {
+		return mem
+	}
+	mem.All = ram.Totalram
+	mem.BuffCache = ram.Bufferram
+	mem.Free = ram.Freeram
+	mem.Shared = ram.Sharedram
+	mem.Available = ram.Freeram + ram.Sharedram + ram.Bufferram
+	mem.SwapAll = ram.Totalswap
+	mem.SwapFree = ram.Freeswap
+	mem.SwapUsed = mem.SwapAll - mem.SwapFree
+	mem.Used = mem.All - mem.Free
+	return mem
+}
+
+func Whoami() (uid, gid int, uname, gname, home string, err error) {
+	var me *user.User
+	var gup *user.Group
+	me, err = user.Current()
+	if err != nil {
+		return
+	}
+	uid, _ = strconv.Atoi(me.Uid)
+	gid, _ = strconv.Atoi(me.Uid)
+	home = me.HomeDir
+	uname = me.Username
+	gup, err = user.LookupGroupId(me.Gid)
+	if err != nil {
+		return
+	}
+	gname = gup.Name
+	return
+}

os_windows.go

@@ -0,0 +1,43 @@
+// +build windows
+
+package staros
+
+import (
+	"time"
+
+	"b612.me/wincmd"
+
+	"b612.me/win32api"
+)
+
+// StartTime 开机时间
+func StartTime() time.Time {
+	data, _ := win32api.GetTickCount()
+	date := float64(time.Now().Unix())
+	unix := date - float64(data)/1000
+	max := (unix - float64(int64(unix))) * 1000000000
+	return time.Unix(int64(unix), int64(max))
+}
+
+// IsRoot 当前是否是管理员用户
+func IsRoot() bool {
+	return wincmd.Isas()
+}
+
+// Memory 系统内存信息
+func Memory() MemStatus {
+	var mem MemStatus
+	ram := new(win32api.MEMORYSTATUSEX)
+	win32api.GlobalMemoryStatusEx(ram)
+	mem.All = uint64(ram.UllTotalPhys)
+	mem.Free = uint64(ram.UllAvailPhys)
+	mem.Available = uint64(ram.UllAvailPhys)
+	mem.Used = uint64(mem.All - mem.Free)
+	mem.SwapAll = uint64(ram.UllTotalPageFile)
+	mem.SwapFree = uint64(ram.UllAvailPageFile)
+	mem.SwapUsed = mem.SwapAll - mem.SwapFree
+	mem.VirtualAll = uint64(mem.VirtualAll)
+	mem.VirtualAvail = uint64(mem.VirtualAvail)
+	mem.VirtualUsed = mem.VirtualAll - mem.VirtualUsed
+	return mem
+}

process_linux.go

@@ -0,0 +1,132 @@
+// +build linux darwin
+
+package staros
+
+import (
+	"bytes"
+	"errors"
+	"fmt"
+	"io/ioutil"
+	"os"
+	"path/filepath"
+	"strconv"
+	"strings"
+	"time"
+)
+
+// FindProcessByName 通过进程名来查询应用信息
+func FindProcessByName(pname string) (datas []Process, err error) {
+	var name, main string
+	var mainb []byte
+	paths, errs := ioutil.ReadDir("/proc")
+	if errs != nil {
+		err = errs
+		return
+	}
+	for _, v := range paths {
+		if v.IsDir() && Exists("/proc/"+v.Name()+"/comm") {
+			name, err = readAsString("/proc/" + v.Name() + "/comm")
+			if err != nil {
+				return
+			}
+			if strings.TrimSpace(name) == pname {
+				var tmp Process
+				main, err = readAsString("/proc/" + v.Name() + "/status")
+				if err != nil {
+					return
+				}
+				data := splitBy(main, ":")
+				tmp.Name = pname
+				tmp.Pid, _ = strconv.ParseInt(data["Pid"], 10, 64)
+				tmp.PPid, _ = strconv.ParseInt(data["PPid"], 10, 64)
+				tmp.TPid, _ = strconv.ParseInt(data["TracerPid"], 10, 64)
+				uids := splitBySpace(data["Uid"])
+				gids := splitBySpace(data["Gid"])
+				tmp.RUID, _ = strconv.Atoi(uids[0])
+				tmp.EUID, _ = strconv.Atoi(uids[1])
+				tmp.RGID, _ = strconv.Atoi(gids[0])
+				tmp.EGID, _ = strconv.Atoi(gids[1])
+				mainb, err = ioutil.ReadFile("/proc/" + v.Name() + "/cmdline")
+				if err != nil {
+					return
+				}
+				args := bytes.Split(mainb, []byte{0})
+				for _, v := range args {
+					tmp.Args = append(tmp.Args, string(v))
+				}
+				tmp.LocalPath, err = os.Readlink("/proc/" + v.Name() + "/exe")
+				if err != nil {
+					return
+				}
+				tmp.LocalPath = filepath.Dir(tmp.LocalPath)
+				tmp.ExecPath, err = os.Readlink("/proc/" + v.Name() + "/cwd")
+				if err != nil {
+					return
+				}
+				main, err = readAsString("/proc/" + v.Name() + "/stat")
+				if err != nil {
+					return
+				}
+				times := splitBySpace(main)
+				uptime, _ := strconv.ParseInt(strings.TrimSpace(times[21]), 10, 64)
+				tmp.Uptime = time.Unix(StartTime().Unix()+uptime/100, int64((float64(uptime)/100-float64(uptime/100))*1000000000))
+				datas = append(datas, tmp)
+			}
+		}
+	}
+	return
+}
+
+// FindProcessByPid 通过Pid来查询应用信息
+func FindProcessByPid(pid int64) (datas Process, err error) {
+	var name, main string
+	var mainb []byte
+	if !Exists("/proc/" + fmt.Sprint(pid) + "/comm") {
+		err = errors.New("Not Found")
+		return
+	}
+	name, err = readAsString("/proc/" + fmt.Sprint(pid) + "/comm")
+	if err != nil {
+		return
+	}
+	main, err = readAsString("/proc/" + fmt.Sprint(pid) + "/status")
+	if err != nil {
+		return
+	}
+	data := splitBy(main, ":")
+	datas.Name = strings.TrimSpace(name)
+	datas.Pid, _ = strconv.ParseInt(data["Pid"], 10, 64)
+	datas.PPid, _ = strconv.ParseInt(data["PPid"], 10, 64)
+	datas.TPid, _ = strconv.ParseInt(data["TracerPid"], 10, 64)
+	uids := splitBySpace(data["Uid"])
+	gids := splitBySpace(data["Gid"])
+	datas.RUID, _ = strconv.Atoi(uids[0])
+	datas.EUID, _ = strconv.Atoi(uids[1])
+	datas.RGID, _ = strconv.Atoi(gids[0])
+	datas.EGID, _ = strconv.Atoi(gids[1])
+	mainb, err = ioutil.ReadFile("/proc/" + fmt.Sprint(pid) + "/cmdline")
+	if err != nil {
+		return
+	}
+	args := bytes.Split(mainb, []byte{0})
+	for _, v := range args {
+		datas.Args = append(datas.Args, string(v))
+	}
+	datas.LocalPath, err = os.Readlink("/proc/" + fmt.Sprint(pid) + "/exe")
+	if err != nil {
+		return
+	}
+	datas.LocalPath = filepath.Dir(datas.LocalPath)
+	datas.ExecPath, err = os.Readlink("/proc/" + fmt.Sprint(pid) + "/cwd")
+	if err != nil {
+		return
+	}
+	main, err = readAsString("/proc/" + fmt.Sprint(pid) + "/stat")
+	if err != nil {
+		return
+	}
+	times := splitBySpace(main)
+	uptime, _ := strconv.ParseInt(strings.TrimSpace(times[21]), 10, 64)
+	datas.Uptime = time.Unix(StartTime().Unix()+uptime/100, int64((float64(uptime)/100-float64(uptime/100))*1000000000))
+	return
+}

process_test.go

@@ -0,0 +1,10 @@
+package staros
+
+import (
+	"fmt"
+	"testing"
+)
+
+func Test_Process(t *testing.T) {
+	fmt.Println(FindProcessByPid(16652))
+}

process_win.go

@@ -0,0 +1,49 @@
+// +build windows
+
+package staros
+
+import (
+	"errors"
+	"fmt"
+	"strconv"
+
+	"b612.me/wincmd"
+)
+
+// FindProcessByName 通过进程名来查询应用信息
+func FindProcessByName(pname string) (data []Process, err error) {
+	var lists []map[string]string
+	lists, err = wincmd.GetRunningProcess()
+	if err != nil {
+		return
+	}
+	for _, v := range lists {
+		if v["name"] == pname {
+			var tmp Process
+			tmp.Name = pname
+			tmp.Pid, _ = strconv.ParseInt(v["pid"], 10, 64)
+			tmp.PPid, _ = strconv.ParseInt(v["ppid"], 10, 64)
+			data = append(data, tmp)
+		}
+	}
+	return
+}
+
+// FindProcessByPid 通过pid来查询应用信息
+func FindProcessByPid(pid int64) (data Process, err error) {
+	var lists []map[string]string
+	lists, err = wincmd.GetRunningProcess()
+	if err != nil {
+		return
+	}
+	for _, v := range lists {
+		if v["pid"] == fmt.Sprint(pid) {
+			data.Name = v["name"]
+			data.Pid = pid
+			data.PPid, _ = strconv.ParseInt(v["ppid"], 10, 64)
+			return
+		}
+	}
+	err = errors.New("Not Found")
+	return
+}

staros.go

@@ -0,0 +1 @@
+package staros

tools.go

@@ -0,0 +1,44 @@
+package staros
+
+import (
+	"bytes"
+	"io/ioutil"
+	"strings"
+)
+
+func splitBy(data, sep string) map[string]string {
+	res := make(map[string]string)
+	lists := strings.Split(data, "\n")
+	for _, v := range lists {
+		list := strings.SplitN(v, sep, 2)
+		if len(list) != 2 {
+			continue
+		}
+		res[strings.TrimSpace(list[0])] = strings.TrimSpace(list[1])
+	}
+	return res
+}
+
+// 横向替换ASCII<9>
+func ReplaceByte9(data string) string {
+	return string(bytes.ReplaceAll([]byte(data), []byte{9}, []byte(" ")))
+}
+
+func splitBySpace(data string) []string {
+	data = string(bytes.ReplaceAll([]byte(data), []byte{9}, []byte(" ")))
+	nomorespace := func(data string) string {
+		return strings.ReplaceAll(data, "  ", " ")
+	}
+	for strings.Index(data, "  ") >= 0 {
+		data = nomorespace(data)
+	}
+	return strings.Split(data, " ")
+}
+
+func readAsString(path string) (string, error) {
+	data, err := ioutil.ReadFile(path)
+	if err != nil {
+		return "", err
+	}
+	return string(data), nil
+}

typed.go

@@ -0,0 +1,37 @@
+package staros
+
+import (
+	"time"
+)
+
+// Process 定义一个进程的信息
+type Process struct {
+	PPid      int64
+	Pid       int64
+	Name      string
+	ExecPath  string
+	LocalPath string
+	Args      []string
+	RUID      int
+	EUID      int
+	RGID      int
+	EGID      int
+	TPid      int64
+	Uptime    time.Time
+}
+
+type MemStatus struct {
+	All           uint64
+	Used          uint64
+	Free          uint64
+	Shared        uint64
+	BuffCache     uint64
+	Available     uint64
+	SwapAll       uint64
+	SwapUsed      uint64
+	SwapFree      uint64
+	VirtualAll    uint64
+	VirtualUsed   uint64
+	VirtualAvail  uint64
+	AvailExtended uint64
+}