Import x-panel source

util/sys/psutil.go

@@ -0,0 +1,8 @@
+package sys
+
+import (
+	_ "unsafe"
+)
+
+//go:linkname HostProc github.com/shirou/gopsutil/v4/internal/common.HostProc
+func HostProc(combineWith ...string) string

util/sys/sys_darwin.go

@@ -0,0 +1,24 @@
+//go:build darwin
+// +build darwin
+
+package sys
+
+import (
+	"github.com/shirou/gopsutil/v4/net"
+)
+
+func GetTCPCount() (int, error) {
+	stats, err := net.Connections("tcp")
+	if err != nil {
+		return 0, err
+	}
+	return len(stats), nil
+}
+
+func GetUDPCount() (int, error) {
+	stats, err := net.Connections("udp")
+	if err != nil {
+		return 0, err
+	}
+	return len(stats), nil
+}

util/sys/sys_linux.go

@@ -0,0 +1,81 @@
+//go:build linux
+// +build linux
+
+package sys
+
+import (
+	"bytes"
+	"fmt"
+	"io"
+	"os"
+)
+
+func getLinesNum(filename string) (int, error) {
+	file, err := os.Open(filename)
+	if err != nil {
+		return 0, err
+	}
+	defer file.Close()
+
+	sum := 0
+	buf := make([]byte, 8192)
+	for {
+		n, err := file.Read(buf)
+
+		var buffPosition int
+		for {
+			i := bytes.IndexByte(buf[buffPosition:n], '\n')
+			if i < 0 {
+				break
+			}
+			buffPosition += i + 1
+			sum++
+		}
+
+		if err == io.EOF {
+			break
+		} else if err != nil {
+			return 0, err
+		}
+	}
+	return sum, nil
+}
+
+func GetTCPCount() (int, error) {
+	root := HostProc()
+
+	tcp4, err := safeGetLinesNum(fmt.Sprintf("%v/net/tcp", root))
+	if err != nil {
+		return 0, err
+	}
+	tcp6, err := safeGetLinesNum(fmt.Sprintf("%v/net/tcp6", root))
+	if err != nil {
+		return 0, err
+	}
+
+	return tcp4 + tcp6, nil
+}
+
+func GetUDPCount() (int, error) {
+	root := HostProc()
+
+	udp4, err := safeGetLinesNum(fmt.Sprintf("%v/net/udp", root))
+	if err != nil {
+		return 0, err
+	}
+	udp6, err := safeGetLinesNum(fmt.Sprintf("%v/net/udp6", root))
+	if err != nil {
+		return 0, err
+	}
+
+	return udp4 + udp6, nil
+}
+
+func safeGetLinesNum(path string) (int, error) {
+	if _, err := os.Stat(path); os.IsNotExist(err) {
+		return 0, nil
+	} else if err != nil {
+		return 0, err
+	}
+	return getLinesNum(path)
+}

util/sys/sys_windows.go

@@ -0,0 +1,30 @@
+//go:build windows
+// +build windows
+
+package sys
+
+import (
+	"errors"
+
+	"github.com/shirou/gopsutil/v4/net"
+)
+
+func GetConnectionCount(proto string) (int, error) {
+	if proto != "tcp" && proto != "udp" {
+		return 0, errors.New("invalid protocol")
+	}
+
+	stats, err := net.Connections(proto)
+	if err != nil {
+		return 0, err
+	}
+	return len(stats), nil
+}
+
+func GetTCPCount() (int, error) {
+	return GetConnectionCount("tcp")
+}
+
+func GetUDPCount() (int, error) {
+	return GetConnectionCount("udp")
+}