init
commit
965dae2110
@ -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()
|
||||
}
|
||||
@ -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
|
||||
}
|
||||
@ -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
|
||||
}
|
||||
@ -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
|
||||
}
|
||||
@ -0,0 +1,10 @@
|
||||
package staros
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func Test_Process(t *testing.T) {
|
||||
fmt.Println(FindProcessByPid(16652))
|
||||
}
|
||||
@ -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
|
||||
}
|
||||
@ -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
|
||||
}
|
||||
@ -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
|
||||
}
|
||||
Loading…
Reference in New Issue