Compare commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
a81eff8c0d | ||
|
|
1d7eb16784 | ||
|
|
a13179ed9b | ||
|
|
93c1620c15 | ||
|
|
ba57fb3c00 | ||
|
|
151b07b73f | ||
|
|
cec5905483 | ||
|
|
1df78dd874 | ||
|
|
92dfdc542c | ||
|
|
a25c51003e | ||
|
|
b27bcfcb5b | ||
|
|
75f10b27fa |
@@ -0,0 +1,270 @@
|
|||||||
|
//go:build windows
|
||||||
|
|
||||||
|
// Command xui-agent-windows is the standalone Windows build of the
|
||||||
|
// 3x-ui node agent. Windows nodes are probe-only by design: the agent
|
||||||
|
// reports host facts/metrics and runs latency probes, but no xray or
|
||||||
|
// sing-box core is ever fetched or started here (the agent core
|
||||||
|
// machinery stays the master's Linux fleet concern; the local
|
||||||
|
// agent.json carries "probeOnly": true, written by
|
||||||
|
// install-agent.ps1).
|
||||||
|
//
|
||||||
|
// Usage:
|
||||||
|
//
|
||||||
|
// xui-agent-windows.exe [-config C:\x-ui\agent.json] run (service or console)
|
||||||
|
// xui-agent-windows.exe install [-config <path>] register the service
|
||||||
|
// xui-agent-windows.exe uninstall remove the service
|
||||||
|
// xui-agent-windows.exe start | stop control the service
|
||||||
|
//
|
||||||
|
// The agent lifecycle wiring (config load, baseline derivation,
|
||||||
|
// logger init, graceful stop) mirrors main.go's agent branch — the
|
||||||
|
// same internal/agent package does the work either way.
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"flag"
|
||||||
|
"fmt"
|
||||||
|
"log"
|
||||||
|
"os"
|
||||||
|
"os/signal"
|
||||||
|
"path/filepath"
|
||||||
|
"syscall"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/mhsanaei/3x-ui/v3/internal/agent"
|
||||||
|
"github.com/mhsanaei/3x-ui/v3/internal/logger"
|
||||||
|
|
||||||
|
"github.com/op/go-logging"
|
||||||
|
"golang.org/x/sys/windows/svc"
|
||||||
|
"golang.org/x/sys/windows/svc/mgr"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
serviceName = "x-ui-agent"
|
||||||
|
defaultCfgPath = `C:\x-ui\agent.json`
|
||||||
|
)
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
log.SetPrefix("[x-ui-agent] ")
|
||||||
|
args := os.Args[1:]
|
||||||
|
|
||||||
|
// Management subcommands come first; anything else is a run.
|
||||||
|
if len(args) > 0 {
|
||||||
|
switch args[0] {
|
||||||
|
case "install", "uninstall", "start", "stop":
|
||||||
|
cfgPath := parseConfigFlag("xui-agent-windows "+args[0], args[1:])
|
||||||
|
if err := runControl(args[0], cfgPath); err != nil {
|
||||||
|
log.Fatalf("%s: %v", args[0], err)
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
cfgPath := parseConfigFlag("xui-agent-windows", args)
|
||||||
|
|
||||||
|
isService, err := svc.IsWindowsService()
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalf("detect service environment: %v", err)
|
||||||
|
}
|
||||||
|
if isService {
|
||||||
|
if err := svc.Run(serviceName, &agentHandler{cfgPath: cfgPath}); err != nil {
|
||||||
|
log.Fatalf("service run: %v", err)
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Interactive run: console foreground, Ctrl+C stops gracefully.
|
||||||
|
running, err := startAgent(cfgPath)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalf("start agent: %v", err)
|
||||||
|
}
|
||||||
|
sigCh := make(chan os.Signal, 1)
|
||||||
|
signal.Notify(sigCh, os.Interrupt, syscall.SIGTERM)
|
||||||
|
select {
|
||||||
|
case <-sigCh:
|
||||||
|
log.Println("Shutting down agent...")
|
||||||
|
running.Stop()
|
||||||
|
<-running.done
|
||||||
|
case err := <-running.done:
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalf("Agent error: %v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func parseConfigFlag(name string, args []string) string {
|
||||||
|
fs := flag.NewFlagSet(name, flag.ExitOnError)
|
||||||
|
cfg := fs.String("config", defaultCfgPath, "Path to agent config file")
|
||||||
|
_ = fs.Parse(args)
|
||||||
|
return *cfg
|
||||||
|
}
|
||||||
|
|
||||||
|
// runningAgent is one started agent plus its completion signal.
|
||||||
|
type runningAgent struct {
|
||||||
|
ag *agent.Agent
|
||||||
|
done chan error
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *runningAgent) Stop() { r.ag.Stop() }
|
||||||
|
|
||||||
|
// startAgent performs the same setup as main.go's agent branch and
|
||||||
|
// launches the agent loop in the background.
|
||||||
|
func startAgent(cfgPath string) (*runningAgent, error) {
|
||||||
|
// Agent mode uses the panel's directory layout for core binaries
|
||||||
|
// and configs; on a probe-only Windows node nothing consumes it,
|
||||||
|
// but keep the wiring identical to the Linux entry so behaviour
|
||||||
|
// cannot drift between the two.
|
||||||
|
if os.Getenv("XUI_BIN_FOLDER") == "" {
|
||||||
|
_ = os.Setenv("XUI_BIN_FOLDER", agent.DefaultBinFolder)
|
||||||
|
}
|
||||||
|
|
||||||
|
cfg, err := agent.LoadConfig(cfgPath)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("load agent config from %s: %w", cfgPath, err)
|
||||||
|
}
|
||||||
|
// The acknowledged traffic baseline persists next to the agent
|
||||||
|
// config (its state home), unless the config names a path.
|
||||||
|
if cfg.BaselinePath == "" {
|
||||||
|
cfg.BaselinePath = filepath.Join(filepath.Dir(cfgPath), "agent-traffic-baseline.json")
|
||||||
|
}
|
||||||
|
|
||||||
|
logger.InitLogger(logging.INFO)
|
||||||
|
log.Printf("Starting 3x-ui agent (node: %s)", cfg.NodeName)
|
||||||
|
|
||||||
|
ag := agent.New(cfg)
|
||||||
|
r := &runningAgent{ag: ag, done: make(chan error, 1)}
|
||||||
|
go func() { r.done <- ag.Run() }()
|
||||||
|
return r, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// agentHandler adapts the agent loop to the SCM protocol.
|
||||||
|
type agentHandler struct {
|
||||||
|
cfgPath string
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *agentHandler) Execute(_ []string, r <-chan svc.ChangeRequest, status chan<- svc.Status) (bool, uint32) {
|
||||||
|
const accepted = svc.AcceptStop | svc.AcceptShutdown
|
||||||
|
status <- svc.Status{State: svc.StartPending}
|
||||||
|
running, err := startAgent(h.cfgPath)
|
||||||
|
if err != nil {
|
||||||
|
log.Printf("start agent: %v", err)
|
||||||
|
return false, 1
|
||||||
|
}
|
||||||
|
status <- svc.Status{State: svc.Running, Accepts: accepted}
|
||||||
|
for {
|
||||||
|
select {
|
||||||
|
case err := <-running.done:
|
||||||
|
// The agent loop ended on its own (e.g. fatal run error).
|
||||||
|
if err != nil {
|
||||||
|
log.Printf("agent exited: %v", err)
|
||||||
|
return false, 1
|
||||||
|
}
|
||||||
|
return false, 0
|
||||||
|
case c := <-r:
|
||||||
|
switch c.Cmd {
|
||||||
|
case svc.Interrogate:
|
||||||
|
status <- c.CurrentStatus
|
||||||
|
case svc.Stop, svc.Shutdown:
|
||||||
|
status <- svc.Status{State: svc.StopPending}
|
||||||
|
running.Stop()
|
||||||
|
<-running.done
|
||||||
|
return false, 0
|
||||||
|
default:
|
||||||
|
log.Printf("unexpected service command: %v", c.Cmd)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// runControl implements the install/uninstall/start/stop subcommands
|
||||||
|
// through the service control manager.
|
||||||
|
func runControl(cmd, cfgPath string) error {
|
||||||
|
m, err := mgr.ConnectRemote("")
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("connect to service manager: %w", err)
|
||||||
|
}
|
||||||
|
defer m.Disconnect()
|
||||||
|
|
||||||
|
switch cmd {
|
||||||
|
case "install":
|
||||||
|
exePath, err := os.Executable()
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("locate own binary: %w", err)
|
||||||
|
}
|
||||||
|
absCfg, err := filepath.Abs(cfgPath)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("resolve config path: %w", err)
|
||||||
|
}
|
||||||
|
if s, err := m.OpenService(serviceName); err == nil {
|
||||||
|
s.Close()
|
||||||
|
return fmt.Errorf("service %s already exists (uninstall first)", serviceName)
|
||||||
|
}
|
||||||
|
s, err := m.CreateService(serviceName, exePath, mgr.Config{
|
||||||
|
DisplayName: "3x-ui Agent",
|
||||||
|
Description: "3x-ui probe-only node agent (facts, metrics and latency probes; runs no proxy core).",
|
||||||
|
StartType: mgr.StartAutomatic,
|
||||||
|
ErrorControl: mgr.ErrorNormal,
|
||||||
|
}, "-config", absCfg)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("create service: %w", err)
|
||||||
|
}
|
||||||
|
defer s.Close()
|
||||||
|
log.Printf("service %s installed (config: %s)", serviceName, absCfg)
|
||||||
|
return nil
|
||||||
|
|
||||||
|
case "uninstall":
|
||||||
|
s, err := openService(m)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer s.Close()
|
||||||
|
// Stop first so the binary is not locked when the operator
|
||||||
|
// deletes or replaces it; an already-stopped service is fine.
|
||||||
|
_, _ = s.Control(svc.Stop)
|
||||||
|
deadline := time.Now().Add(15 * time.Second)
|
||||||
|
for time.Now().Before(deadline) {
|
||||||
|
st, err := s.Query()
|
||||||
|
if err != nil || st.State == svc.Stopped {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
time.Sleep(300 * time.Millisecond)
|
||||||
|
}
|
||||||
|
if err := s.Delete(); err != nil {
|
||||||
|
return fmt.Errorf("delete service: %w", err)
|
||||||
|
}
|
||||||
|
log.Printf("service %s removed", serviceName)
|
||||||
|
return nil
|
||||||
|
|
||||||
|
case "start":
|
||||||
|
s, err := openService(m)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer s.Close()
|
||||||
|
if err := s.Start(); err != nil {
|
||||||
|
return fmt.Errorf("start service: %w", err)
|
||||||
|
}
|
||||||
|
log.Printf("service %s started", serviceName)
|
||||||
|
return nil
|
||||||
|
|
||||||
|
case "stop":
|
||||||
|
s, err := openService(m)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer s.Close()
|
||||||
|
if _, err := s.Control(svc.Stop); err != nil {
|
||||||
|
return fmt.Errorf("stop service: %w", err)
|
||||||
|
}
|
||||||
|
log.Printf("service %s stop requested", serviceName)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
return fmt.Errorf("unknown subcommand %q", cmd)
|
||||||
|
}
|
||||||
|
|
||||||
|
func openService(m *mgr.Mgr) (*mgr.Service, error) {
|
||||||
|
s, err := m.OpenService(serviceName)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("open service %s: %w", serviceName, err)
|
||||||
|
}
|
||||||
|
return s, nil
|
||||||
|
}
|
||||||
@@ -307,13 +307,13 @@ export default function NodeList({
|
|||||||
</Tooltip>
|
</Tooltip>
|
||||||
) : (
|
) : (
|
||||||
<Space>
|
<Space>
|
||||||
<Tooltip title={t('pages.nodes.probe')}>
|
<Tooltip title={t('pages.nodes.testConnection')}>
|
||||||
<Button
|
<Button
|
||||||
type="text"
|
type="text"
|
||||||
size="small"
|
size="small"
|
||||||
style={{ fontSize: 16 }}
|
style={{ fontSize: 16 }}
|
||||||
icon={<ThunderboltOutlined />}
|
icon={<ThunderboltOutlined />}
|
||||||
aria-label={t('pages.nodes.probe')}
|
aria-label={t('pages.nodes.testConnection')}
|
||||||
onClick={() => onProbe(record)}
|
onClick={() => onProbe(record)}
|
||||||
/>
|
/>
|
||||||
</Tooltip>
|
</Tooltip>
|
||||||
@@ -716,7 +716,7 @@ export default function NodeList({
|
|||||||
key: 'probe',
|
key: 'probe',
|
||||||
label: (
|
label: (
|
||||||
<>
|
<>
|
||||||
<ThunderboltOutlined /> {t('pages.nodes.probe')}
|
<ThunderboltOutlined /> {t('pages.nodes.testConnection')}
|
||||||
</>
|
</>
|
||||||
),
|
),
|
||||||
onClick: () => onProbe(record),
|
onClick: () => onProbe(record),
|
||||||
|
|||||||
@@ -24,12 +24,20 @@ interface ProbeMetaWire {
|
|||||||
ipv4Pin: string;
|
ipv4Pin: string;
|
||||||
ipv6Pin: string;
|
ipv6Pin: string;
|
||||||
// V2.0.1: ledger corrections in GB (either sign) and the counted
|
// V2.0.1: ledger corrections in GB (either sign) and the counted
|
||||||
// interface pin ('' = automatic selection on the agent).
|
// interface pin ('' = automatic selection on the agent). Since
|
||||||
|
// V2.2.1 the corrections are one-shot deltas: the inputs below are
|
||||||
|
// local state (never prefilled from the wire), a save submits them
|
||||||
|
// once, and they clear back to empty on success.
|
||||||
trafficCalUpGb: number;
|
trafficCalUpGb: number;
|
||||||
trafficCalDownGb: number;
|
trafficCalDownGb: number;
|
||||||
statIface: string;
|
statIface: string;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// GB entry converts on the binary convention every byte formatter in
|
||||||
|
// the panel and on the public page uses (V2.2.1: was decimal 1e9,
|
||||||
|
// which made a 3000 GB limit render as 2.73 TB instead of 2.93 TB).
|
||||||
|
const BYTES_PER_GB = 1024 ** 3;
|
||||||
|
|
||||||
const JSON_HEADERS = { headers: { 'Content-Type': 'application/json' } };
|
const JSON_HEADERS = { headers: { 'Content-Type': 'application/json' } };
|
||||||
|
|
||||||
const NAMED_CYCLES = [
|
const NAMED_CYCLES = [
|
||||||
@@ -59,6 +67,9 @@ export default function ProbeMetaSection({ nodeId }: { nodeId: number }) {
|
|||||||
// in NodeFormModal); visibility derives during render.
|
// in NodeFormModal); visibility derives during render.
|
||||||
const [loaded, setLoaded] = useState<{ nodeId: number; meta: ProbeMetaWire } | null>(null);
|
const [loaded, setLoaded] = useState<{ nodeId: number; meta: ProbeMetaWire } | null>(null);
|
||||||
const [trafficGb, setTrafficGb] = useState(0);
|
const [trafficGb, setTrafficGb] = useState(0);
|
||||||
|
// One-shot calibration deltas (GB, either sign): null = empty box.
|
||||||
|
const [calUpGb, setCalUpGb] = useState<number | null>(null);
|
||||||
|
const [calDownGb, setCalDownGb] = useState<number | null>(null);
|
||||||
const [saving, setSaving] = useState(false);
|
const [saving, setSaving] = useState(false);
|
||||||
|
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
@@ -68,7 +79,7 @@ export default function ProbeMetaSection({ nodeId }: { nodeId: number }) {
|
|||||||
}).then((msg) => {
|
}).then((msg) => {
|
||||||
if (!cancelled && msg?.success && msg.obj) {
|
if (!cancelled && msg?.success && msg.obj) {
|
||||||
setLoaded({ nodeId, meta: msg.obj });
|
setLoaded({ nodeId, meta: msg.obj });
|
||||||
setTrafficGb(msg.obj.trafficLimit / 1e9);
|
setTrafficGb(msg.obj.trafficLimit / BYTES_PER_GB);
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
return () => {
|
return () => {
|
||||||
@@ -116,14 +127,23 @@ export default function ProbeMetaSection({ nodeId }: { nodeId: number }) {
|
|||||||
try {
|
try {
|
||||||
const payload: ProbeMetaWire = {
|
const payload: ProbeMetaWire = {
|
||||||
...meta,
|
...meta,
|
||||||
trafficLimit: Math.round(trafficGb * 1e9),
|
trafficLimit: Math.round(trafficGb * BYTES_PER_GB),
|
||||||
|
trafficCalUpGb: calUpGb ?? 0,
|
||||||
|
trafficCalDownGb: calDownGb ?? 0,
|
||||||
};
|
};
|
||||||
const msg = await HttpUtil.put(`/panel/api/agent/probe-meta/${nodeId}`, payload, {
|
const msg = await HttpUtil.put(`/panel/api/agent/probe-meta/${nodeId}`, payload, {
|
||||||
silent: true,
|
silent: true,
|
||||||
...JSON_HEADERS,
|
...JSON_HEADERS,
|
||||||
});
|
});
|
||||||
if (msg?.success) {
|
if (msg?.success) {
|
||||||
setLoaded({ nodeId, meta: payload });
|
// One-shot: the submitted corrections are folded in server-
|
||||||
|
// side; the boxes clear so a later save resubmits nothing.
|
||||||
|
setLoaded({
|
||||||
|
nodeId,
|
||||||
|
meta: { ...payload, trafficCalUpGb: 0, trafficCalDownGb: 0 },
|
||||||
|
});
|
||||||
|
setCalUpGb(null);
|
||||||
|
setCalDownGb(null);
|
||||||
messageApi.success(t('pages.nodes.probe.saved'));
|
messageApi.success(t('pages.nodes.probe.saved'));
|
||||||
} else {
|
} else {
|
||||||
messageApi.error(msg?.msg || t('pages.nodes.probe.saveFailed'));
|
messageApi.error(msg?.msg || t('pages.nodes.probe.saveFailed'));
|
||||||
@@ -249,9 +269,10 @@ export default function ProbeMetaSection({ nodeId }: { nodeId: number }) {
|
|||||||
<Col xs={24} md={8}>
|
<Col xs={24} md={8}>
|
||||||
<Field label={t('pages.nodes.probe.calUp')}>
|
<Field label={t('pages.nodes.probe.calUp')}>
|
||||||
<InputNumber
|
<InputNumber
|
||||||
value={meta.trafficCalUpGb}
|
value={calUpGb}
|
||||||
step={0.1}
|
step={0.1}
|
||||||
onChange={(v) => set('trafficCalUpGb', v ?? 0)}
|
placeholder="0"
|
||||||
|
onChange={(v) => setCalUpGb(v)}
|
||||||
style={{ width: '100%' }}
|
style={{ width: '100%' }}
|
||||||
/>
|
/>
|
||||||
</Field>
|
</Field>
|
||||||
@@ -259,9 +280,10 @@ export default function ProbeMetaSection({ nodeId }: { nodeId: number }) {
|
|||||||
<Col xs={24} md={8}>
|
<Col xs={24} md={8}>
|
||||||
<Field label={t('pages.nodes.probe.calDown')}>
|
<Field label={t('pages.nodes.probe.calDown')}>
|
||||||
<InputNumber
|
<InputNumber
|
||||||
value={meta.trafficCalDownGb}
|
value={calDownGb}
|
||||||
step={0.1}
|
step={0.1}
|
||||||
onChange={(v) => set('trafficCalDownGb', v ?? 0)}
|
placeholder="0"
|
||||||
|
onChange={(v) => setCalDownGb(v)}
|
||||||
style={{ width: '100%' }}
|
style={{ width: '100%' }}
|
||||||
/>
|
/>
|
||||||
</Field>
|
</Field>
|
||||||
|
|||||||
@@ -0,0 +1,102 @@
|
|||||||
|
#Requires -RunAsAdministrator
|
||||||
|
<#
|
||||||
|
.SYNOPSIS
|
||||||
|
Installs the 3x-ui probe-only agent on Windows (64-bit).
|
||||||
|
|
||||||
|
.DESCRIPTION
|
||||||
|
Downloads x-ui-agent-windows-amd64.exe into C:\x-ui, writes
|
||||||
|
agent.json with "probeOnly": true (Windows nodes run no proxy
|
||||||
|
core: facts, metrics and latency probes only), registers the
|
||||||
|
x-ui-agent Windows service (auto-start) and starts it.
|
||||||
|
|
||||||
|
Re-running is safe: an existing service is stopped and removed
|
||||||
|
first, and any of -Master/-Token/-Name left empty is inherited
|
||||||
|
from the agent.json already on disk.
|
||||||
|
|
||||||
|
.EXAMPLE
|
||||||
|
.\install-agent.ps1 -Master "https://panel.example.com/rogue" -Token "<agent token>" -Name "win-node-1"
|
||||||
|
#>
|
||||||
|
[CmdletBinding()]
|
||||||
|
param(
|
||||||
|
[string]$Master,
|
||||||
|
[string]$Token,
|
||||||
|
[string]$Name,
|
||||||
|
[string]$InstallDir = "C:\x-ui",
|
||||||
|
[string]$AgentUrl = $env:XUI_URL
|
||||||
|
)
|
||||||
|
|
||||||
|
$ErrorActionPreference = "Stop"
|
||||||
|
|
||||||
|
# Release download base, same shape as install-agent.sh's
|
||||||
|
# RELEASE_BASE (the version segment is bumped by the release flow).
|
||||||
|
$ReleaseBase = "https://git.881218.xyz/rogue/3x-ui-agent/releases/download/V2.2.1"
|
||||||
|
if (-not $AgentUrl) {
|
||||||
|
$AgentUrl = "$ReleaseBase/x-ui-agent-windows-amd64.exe"
|
||||||
|
}
|
||||||
|
|
||||||
|
$ServiceName = "x-ui-agent"
|
||||||
|
$ExeName = "x-ui-agent-windows-amd64.exe"
|
||||||
|
$ExePath = Join-Path $InstallDir $ExeName
|
||||||
|
$ConfigPath = Join-Path $InstallDir "agent.json"
|
||||||
|
|
||||||
|
# Inherit missing parameters from an existing agent.json (reruns).
|
||||||
|
if (Test-Path $ConfigPath) {
|
||||||
|
$existing = Get-Content $ConfigPath -Raw | ConvertFrom-Json
|
||||||
|
if (-not $Master) { $Master = $existing.master_url }
|
||||||
|
if (-not $Token) { $Token = $existing.token }
|
||||||
|
if (-not $Name) { $Name = $existing.node_name }
|
||||||
|
}
|
||||||
|
if (-not $Master) { throw "missing -Master (and no existing agent.json to inherit from)" }
|
||||||
|
if (-not $Token) { throw "missing -Token (and no existing agent.json to inherit from)" }
|
||||||
|
if (-not $Name) { throw "missing -Name (and no existing agent.json to inherit from)" }
|
||||||
|
|
||||||
|
New-Item -ItemType Directory -Force -Path $InstallDir | Out-Null
|
||||||
|
|
||||||
|
# Stop and remove any previous install so the exe file is not locked
|
||||||
|
# by a running service when it is replaced below. Prefer the old
|
||||||
|
# binary's own subcommands; fall back to sc.exe when it is gone.
|
||||||
|
$svc = Get-Service -Name $ServiceName -ErrorAction SilentlyContinue
|
||||||
|
if ($svc) {
|
||||||
|
if (Test-Path $ExePath) {
|
||||||
|
& $ExePath stop | Out-Null
|
||||||
|
& $ExePath uninstall | Out-Null
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
Stop-Service -Name $ServiceName -Force -ErrorAction SilentlyContinue
|
||||||
|
& sc.exe delete $ServiceName | Out-Null
|
||||||
|
}
|
||||||
|
Start-Sleep -Seconds 1
|
||||||
|
}
|
||||||
|
|
||||||
|
# Download the agent binary (via a temp file, then swap into place).
|
||||||
|
[Net.ServicePointManager]::SecurityProtocol = [Net.SecurityProtocolType]::Tls12
|
||||||
|
$tmpExe = "$ExePath.new"
|
||||||
|
Write-Host "Downloading $AgentUrl"
|
||||||
|
Invoke-WebRequest -Uri $AgentUrl -OutFile $tmpExe -UseBasicParsing
|
||||||
|
if (-not (Test-Path $tmpExe) -or (Get-Item $tmpExe).Length -eq 0) {
|
||||||
|
throw "agent download produced an empty file: $AgentUrl"
|
||||||
|
}
|
||||||
|
Move-Item -Force $tmpExe $ExePath
|
||||||
|
|
||||||
|
# Write agent.json. Windows agents are probe-only: probeOnly is
|
||||||
|
# always set (the design contract, not an option).
|
||||||
|
$config = [ordered]@{
|
||||||
|
master_url = $Master
|
||||||
|
token = $Token
|
||||||
|
node_name = $Name
|
||||||
|
probeOnly = $true
|
||||||
|
}
|
||||||
|
$json = $config | ConvertTo-Json
|
||||||
|
# Go's encoding/json rejects a UTF-8 BOM, which Set-Content -Encoding
|
||||||
|
# UTF8 would add on Windows PowerShell 5.1: write BOM-less UTF-8.
|
||||||
|
[System.IO.File]::WriteAllText($ConfigPath, $json, (New-Object System.Text.UTF8Encoding($false)))
|
||||||
|
Write-Host "Wrote $ConfigPath (probeOnly)"
|
||||||
|
|
||||||
|
# Register and start the service.
|
||||||
|
& $ExePath install -config $ConfigPath
|
||||||
|
if ($LASTEXITCODE -ne 0) { throw "service install failed (exit $LASTEXITCODE)" }
|
||||||
|
& $ExePath start
|
||||||
|
if ($LASTEXITCODE -ne 0) { throw "service start failed (exit $LASTEXITCODE)" }
|
||||||
|
|
||||||
|
Write-Host "3x-ui agent installed and running as service '$ServiceName'."
|
||||||
|
Write-Host "Update later by re-running this script; the master's one-click agent update is intentionally not consumed on Windows."
|
||||||
+62
-5
@@ -11,7 +11,7 @@
|
|||||||
#
|
#
|
||||||
# 测试用环境变量(一般无需设置):
|
# 测试用环境变量(一般无需设置):
|
||||||
# XUI_ROOT 安装路径前缀(假根目录验证用),默认空 = 真实路径
|
# XUI_ROOT 安装路径前缀(假根目录验证用),默认空 = 真实路径
|
||||||
# XUI_INIT_SYSTEM 强制 init 系统: systemd | openrc | none,默认 auto 探测
|
# XUI_INIT_SYSTEM 强制 init 系统: systemd | openrc | procd | none,默认 auto 探测
|
||||||
# XUI_URL agent 二进制下载地址覆盖
|
# XUI_URL agent 二进制下载地址覆盖
|
||||||
#
|
#
|
||||||
# 二进制命名(P12 改名):agent 独立装为 /usr/local/x-ui/xui-agent,不再与主控共用
|
# 二进制命名(P12 改名):agent 独立装为 /usr/local/x-ui/xui-agent,不再与主控共用
|
||||||
@@ -81,7 +81,7 @@ esac
|
|||||||
# sing-box 核心在 NexusGate 仓的专用发行里(见下方 SINGBOX_BASE)。
|
# sing-box 核心在 NexusGate 仓的专用发行里(见下方 SINGBOX_BASE)。
|
||||||
# 注意:本脚本是 V2 版——只装 sing-box 运行时;主控未升 V2 前不要用它
|
# 注意:本脚本是 V2 版——只装 sing-box 运行时;主控未升 V2 前不要用它
|
||||||
# 重装节点(节点升级走面板一键更新,不需要本脚本)。
|
# 重装节点(节点升级走面板一键更新,不需要本脚本)。
|
||||||
RELEASE_BASE="https://git.881218.xyz/rogue/3x-ui-agent/releases/download/V2.1.0"
|
RELEASE_BASE="https://git.881218.xyz/rogue/3x-ui-agent/releases/download/V2.2.1"
|
||||||
XUI_URL="${XUI_URL:-${RELEASE_BASE}/x-ui-linux-${BIN_ARCH}}"
|
XUI_URL="${XUI_URL:-${RELEASE_BASE}/x-ui-linux-${BIN_ARCH}}"
|
||||||
|
|
||||||
ROOT="${XUI_ROOT:-}"
|
ROOT="${XUI_ROOT:-}"
|
||||||
@@ -220,7 +220,7 @@ stop_legacy_agent() {
|
|||||||
|
|
||||||
detect_init_system() {
|
detect_init_system() {
|
||||||
case "${XUI_INIT_SYSTEM:-auto}" in
|
case "${XUI_INIT_SYSTEM:-auto}" in
|
||||||
systemd|openrc|none) echo "${XUI_INIT_SYSTEM}"; return;;
|
systemd|openrc|procd|none) echo "${XUI_INIT_SYSTEM}"; return;;
|
||||||
esac
|
esac
|
||||||
if [ -d /run/systemd/system ] && command -v systemctl >/dev/null 2>&1; then
|
if [ -d /run/systemd/system ] && command -v systemctl >/dev/null 2>&1; then
|
||||||
echo systemd; return
|
echo systemd; return
|
||||||
@@ -228,6 +228,9 @@ detect_init_system() {
|
|||||||
if command -v rc-update >/dev/null 2>&1 && command -v rc-service >/dev/null 2>&1; then
|
if command -v rc-update >/dev/null 2>&1 && command -v rc-service >/dev/null 2>&1; then
|
||||||
echo openrc; return
|
echo openrc; return
|
||||||
fi
|
fi
|
||||||
|
if [ -f /etc/rc.common ] && command -v procd >/dev/null 2>&1; then
|
||||||
|
echo procd; return
|
||||||
|
fi
|
||||||
echo none
|
echo none
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -286,10 +289,14 @@ command_args="agent"
|
|||||||
# supervise-daemon 守护(第 23 条):进程崩溃后按 respawn 参数自动拉起,
|
# supervise-daemon 守护(第 23 条):进程崩溃后按 respawn 参数自动拉起,
|
||||||
# 对齐 systemd 支的 Restart=on-failure + RestartSec=5;旧版 command_background
|
# 对齐 systemd 支的 Restart=on-failure + RestartSec=5;旧版 command_background
|
||||||
# 只负责后台化、崩溃不重启。respawn_max=0 = 不限次(同 systemd 的持续重试)。
|
# 只负责后台化、崩溃不重启。respawn_max=0 = 不限次(同 systemd 的持续重试)。
|
||||||
|
# respawn_period 必须为正数:supervise-daemon 启动时校验该值,0 直接报
|
||||||
|
# "Invalid respawn-period value '0'" 拒绝拉起(V2.2.1 ⑤,Alpine 真机实证);
|
||||||
|
# 取 1800(真机上其他 supervise-daemon 服务同值)。max=0 不限次时 period
|
||||||
|
# 仅是计数窗口,不影响持续重试语义。
|
||||||
supervisor="supervise-daemon"
|
supervisor="supervise-daemon"
|
||||||
respawn_delay=5
|
respawn_delay=5
|
||||||
respawn_max=0
|
respawn_max=0
|
||||||
respawn_period=0
|
respawn_period=1800
|
||||||
pidfile="/run/x-ui-agent.pid"
|
pidfile="/run/x-ui-agent.pid"
|
||||||
output_log="/var/log/x-ui-agent.log"
|
output_log="/var/log/x-ui-agent.log"
|
||||||
error_log="/var/log/x-ui-agent.log"
|
error_log="/var/log/x-ui-agent.log"
|
||||||
@@ -322,9 +329,58 @@ EOF
|
|||||||
fi
|
fi
|
||||||
}
|
}
|
||||||
|
|
||||||
|
# 以 procd 服务方式安装并启动(开机自启,OpenWrt)
|
||||||
|
start_via_procd() {
|
||||||
|
local init_dir="${ROOT}/etc/init.d"
|
||||||
|
mkdir -p "$init_dir"
|
||||||
|
# 服务脚本与 2026-10-23 AX6600 真机手工验证版逐字一致(V2.2.1 ④
|
||||||
|
# 收编):路径按脚本变量展开,XUI_ROOT 为空时即验证版的
|
||||||
|
# /usr/local/x-ui/xui-agent 与 /etc/x-ui/agent.json。procd 托管
|
||||||
|
# + respawn 守护,stdout/stderr 进 syslog(logread 查看)。
|
||||||
|
cat > "${init_dir}/xui-agent" <<EOF
|
||||||
|
#!/bin/sh /etc/rc.common
|
||||||
|
START=99
|
||||||
|
STOP=10
|
||||||
|
USE_PROCD=1
|
||||||
|
start_service() {
|
||||||
|
procd_open_instance
|
||||||
|
procd_set_param command ${AGENT_BIN} agent -config ${CONF_DIR}/agent.json
|
||||||
|
procd_set_param respawn 3600 5 5
|
||||||
|
procd_set_param stdout 1
|
||||||
|
procd_set_param stderr 1
|
||||||
|
procd_close_instance
|
||||||
|
}
|
||||||
|
EOF
|
||||||
|
chmod +x "${init_dir}/xui-agent"
|
||||||
|
info "已生成 procd 服务文件: ${init_dir}/xui-agent"
|
||||||
|
if [ -n "$ROOT" ]; then
|
||||||
|
info "(XUI_ROOT 模式:跳过 /etc/init.d 实际操作)"
|
||||||
|
return 0
|
||||||
|
fi
|
||||||
|
/etc/init.d/xui-agent stop 2>/dev/null || true
|
||||||
|
stop_legacy_agent
|
||||||
|
pkill -f "${AGENT_BIN} agent" 2>/dev/null || true; sleep 1
|
||||||
|
/etc/init.d/xui-agent enable
|
||||||
|
/etc/init.d/xui-agent start
|
||||||
|
sleep 2
|
||||||
|
if pgrep -f "${AGENT_BIN} agent" >/dev/null 2>&1 || ps w 2>/dev/null | grep "[x]ui-agent agent" >/dev/null 2>&1; then
|
||||||
|
info "agent 已由 procd 托管启动(开机自启)"
|
||||||
|
echo ""
|
||||||
|
echo " 主控: $MASTER"
|
||||||
|
echo " 节点名: $NAME"
|
||||||
|
echo " 日志: logread -f"
|
||||||
|
echo " agent 会在 30 秒内从主控拉取配置"
|
||||||
|
else
|
||||||
|
error "agent 服务启动失败,查看: logread | tail -n 50"
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
# 无服务管理器时回落 nohup(无开机自启,明确警告)
|
# 无服务管理器时回落 nohup(无开机自启,明确警告)
|
||||||
start_via_nohup() {
|
start_via_nohup() {
|
||||||
warn "未检测到 systemd / OpenRC,退回 nohup 方式启动"
|
# BusyBox 精简系统可能连 nohup 都没有(AX6600 实证):先讲清再退,
|
||||||
|
# 不要假装启动、走到最后才报一个看不懂的失败。
|
||||||
|
command -v nohup >/dev/null 2>&1 || error "未检测到 systemd / OpenRC / procd,且本机无 nohup 命令,无法启动 agent"
|
||||||
|
warn "未检测到 systemd / OpenRC / procd,退回 nohup 方式启动"
|
||||||
warn "注意:nohup 方式没有开机自启,机器重启后 agent 不会自动恢复,需手动重跑本脚本!"
|
warn "注意:nohup 方式没有开机自启,机器重启后 agent 不会自动恢复,需手动重跑本脚本!"
|
||||||
stop_legacy_agent
|
stop_legacy_agent
|
||||||
pkill -f "${AGENT_BIN} agent" 2>/dev/null || true; sleep 1
|
pkill -f "${AGENT_BIN} agent" 2>/dev/null || true; sleep 1
|
||||||
@@ -348,5 +404,6 @@ info "启动 agent..."
|
|||||||
case "$(detect_init_system)" in
|
case "$(detect_init_system)" in
|
||||||
systemd) start_via_systemd;;
|
systemd) start_via_systemd;;
|
||||||
openrc) start_via_openrc;;
|
openrc) start_via_openrc;;
|
||||||
|
procd) start_via_procd;;
|
||||||
*) start_via_nohup;;
|
*) start_via_nohup;;
|
||||||
esac
|
esac
|
||||||
|
|||||||
+41
-786
@@ -1,34 +1,34 @@
|
|||||||
package agent
|
package agent
|
||||||
|
|
||||||
// probe.go ports monitor-probe's agent collector (collect.rs, MIT) to Go:
|
// probe.go is the platform-neutral core of the probe collector, ported
|
||||||
// host Facts (once per process) and per-heartbeat Metrics read directly from
|
// from monitor-probe's agent collector (collect.rs, MIT): host Facts
|
||||||
// /proc and /sys with the standard library only. The port keeps the source's
|
// (once per process) and per-heartbeat Metrics with the source's
|
||||||
// accounting discipline:
|
// accounting discipline:
|
||||||
// - memory "used" is free(1)'s column: MemTotal - MemAvailable;
|
// - memory "used" is free(1)'s column: total - available;
|
||||||
// - disk "used" is df(1)'s: sum over real mounts of (f_blocks-f_bfree);
|
// - disk "used" is df(1)'s: real storage only, total - free blocks
|
||||||
// - network totals sum only "counted" interfaces — loopback, container,
|
// (not the unprivileged-available figure);
|
||||||
// tunnel, bridge/bond/VLAN stacked devices are excluded by a three-layer
|
// - network totals sum only "counted" interfaces — loopback,
|
||||||
// filter so no byte is counted twice;
|
// container, tunnel and stacked devices are excluded by a filter so
|
||||||
// - boot epoch = kernel boot_id + FNV-1a fingerprint of the counted set,
|
// no byte is counted twice;
|
||||||
// so a set change re-baselines exactly like a reboot;
|
// - boot epoch = a stable boot identity + FNV-1a fingerprint of the
|
||||||
// - every source degrades independently: an unreadable file yields a zero
|
// counted set, so a set change re-baselines exactly like a reboot;
|
||||||
// field, never a failed collection round.
|
// - every source degrades independently: an unreadable source yields
|
||||||
|
// a zero field, never a failed collection round.
|
||||||
|
//
|
||||||
|
// The samplers themselves are per-OS: probe_linux.go reads /proc and
|
||||||
|
// /sys, probe_windows.go reads the Win32 performance/registry surface.
|
||||||
|
// Both feed the shared rate/epoch machinery below and must keep the
|
||||||
|
// same accounting semantics; where a source has no equivalent on the
|
||||||
|
// platform, the field stays zero (documented at the sampler).
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"hash/fnv"
|
"hash/fnv"
|
||||||
"net"
|
"net"
|
||||||
"os"
|
|
||||||
"path/filepath"
|
|
||||||
"runtime"
|
|
||||||
"sort"
|
"sort"
|
||||||
"strconv"
|
|
||||||
"strings"
|
"strings"
|
||||||
"sync"
|
"sync"
|
||||||
"sync/atomic"
|
"sync/atomic"
|
||||||
"syscall"
|
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/mhsanaei/3x-ui/v3/internal/config"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
// ProbeFacts mirrors monitor's Facts frame: static host identity, sampled
|
// ProbeFacts mirrors monitor's Facts frame: static host identity, sampled
|
||||||
@@ -72,7 +72,7 @@ type ProbeMetrics struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// probeCollector holds the cross-sample state the rates need (previous CPU
|
// probeCollector holds the cross-sample state the rates need (previous CPU
|
||||||
// jiffies and per-interface counters). Kept in memory only, like the source.
|
// counters and per-interface counters). Kept in memory only, like the source.
|
||||||
type probeCollector struct {
|
type probeCollector struct {
|
||||||
mu sync.Mutex
|
mu sync.Mutex
|
||||||
prevCPU *cpuSample
|
prevCPU *cpuSample
|
||||||
@@ -80,6 +80,9 @@ type probeCollector struct {
|
|||||||
prevNetT time.Time
|
prevNetT time.Time
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// cpuSample is one cumulative CPU reading in whatever tick unit the
|
||||||
|
// platform reports: total covers every state, idle only the idle state.
|
||||||
|
// The differencing in cpuPercent is unit-agnostic.
|
||||||
type cpuSample struct{ total, idle uint64 }
|
type cpuSample struct{ total, idle uint64 }
|
||||||
|
|
||||||
var (
|
var (
|
||||||
@@ -94,37 +97,15 @@ func collectFacts() *ProbeFacts {
|
|||||||
return factsCache
|
return factsCache
|
||||||
}
|
}
|
||||||
|
|
||||||
func sampleFacts() *ProbeFacts {
|
|
||||||
mem := parseMeminfo(readProcFile("meminfo"))
|
|
||||||
diskTotal, _ := diskUsage(realMountPoints(readProcFile("self/mounts")))
|
|
||||||
cpuName, cpuCores := cpuInfo(readProcFile("cpuinfo"))
|
|
||||||
v4, v6 := probeAddresses()
|
|
||||||
return &ProbeFacts{
|
|
||||||
Hostname: readTrimFile("/proc/sys/kernel/hostname", "unknown"),
|
|
||||||
OS: osPrettyName(),
|
|
||||||
Kernel: readTrimFile("/proc/sys/kernel/osrelease", "unknown"),
|
|
||||||
Arch: runtime.GOARCH,
|
|
||||||
Virt: detectVirt(readProcFile("self/mounts"), readProcFile("cpuinfo")),
|
|
||||||
CPUName: cpuName,
|
|
||||||
CPUCores: cpuCores,
|
|
||||||
MemTotal: mem["MemTotal"],
|
|
||||||
SwapTotal: mem["SwapTotal"],
|
|
||||||
DiskTotal: diskTotal,
|
|
||||||
AgentVersion: config.GetAgentDisplayVersion(),
|
|
||||||
IPv4: v4,
|
|
||||||
IPv6: v6,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// statIfacePin (V2.0.1) names the single interface to count when
|
// statIfacePin (V2.0.1) names the single interface to count when
|
||||||
// the operator pinned one on the master (its probe_meta.stat_iface,
|
// the operator pinned one on the master (its probe_meta.stat_iface,
|
||||||
// delivered on the config poll). "" keeps the automatic counted-set
|
// delivered on the config poll). "" keeps the automatic counted-set
|
||||||
// selection below. The pin bypasses the three-layer filter on
|
// selection. The pin bypasses the filter on purpose: the operator
|
||||||
// purpose: the operator named this exact device, including when it
|
// named this exact device, including when it is one the filter would
|
||||||
// is one the filter would skip (a tunnel the node actually bills
|
// skip (a tunnel the node actually bills on). A pinned name absent
|
||||||
// on). A pinned name absent from /proc/net/dev counts nothing —
|
// from the interface table counts nothing — the boot epoch changes
|
||||||
// the boot epoch changes with the counted set, so the master
|
// with the counted set, so the master re-baselines instead of booking
|
||||||
// re-baselines instead of booking a phantom delta.
|
// a phantom delta.
|
||||||
var statIfacePin atomic.Value // string
|
var statIfacePin atomic.Value // string
|
||||||
|
|
||||||
// SetStatIfacePin records the master's counted-interface pin.
|
// SetStatIfacePin records the master's counted-interface pin.
|
||||||
@@ -137,6 +118,12 @@ func statIfacePinValue() string {
|
|||||||
return ""
|
return ""
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// netCounters is one interface's cumulative byte totals.
|
||||||
|
type netCounters struct {
|
||||||
|
name string
|
||||||
|
rx, tx uint64
|
||||||
|
}
|
||||||
|
|
||||||
func pinnedInterfaces(all []netCounters, pin string) []netCounters {
|
func pinnedInterfaces(all []netCounters, pin string) []netCounters {
|
||||||
for _, c := range all {
|
for _, c := range all {
|
||||||
if c.name == pin {
|
if c.name == pin {
|
||||||
@@ -146,151 +133,9 @@ func pinnedInterfaces(all []netCounters, pin string) []netCounters {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// collectMetrics takes one full sample. Individual sources fail soft.
|
// cpuPercent diffs two cumulative samples into a busy percentage. A
|
||||||
func collectMetrics() *ProbeMetrics {
|
// missing or non-advancing previous sample reports 0 (the round only
|
||||||
probeColl.mu.Lock()
|
// re-baselines) — the source's discipline on both platforms.
|
||||||
defer probeColl.mu.Unlock()
|
|
||||||
mem := parseMeminfo(readProcFile("meminfo"))
|
|
||||||
memTotal, memUsed := memUsedFree(mem)
|
|
||||||
swapTotal, swapUsed := swapUsedFree(mem)
|
|
||||||
diskTotal, diskUsed := diskUsage(realMountPoints(readProcFile("self/mounts")))
|
|
||||||
allIfaces := parseNetDev(readProcFile("net/dev"))
|
|
||||||
counted := countedInterfaces(allIfaces, sysNetRoot)
|
|
||||||
if pin := statIfacePinValue(); pin != "" {
|
|
||||||
counted = pinnedInterfaces(allIfaces, pin)
|
|
||||||
}
|
|
||||||
var rxTotal, txTotal uint64
|
|
||||||
names := make([]string, 0, len(counted))
|
|
||||||
for _, c := range counted {
|
|
||||||
rxTotal += c.rx
|
|
||||||
txTotal += c.tx
|
|
||||||
names = append(names, c.name)
|
|
||||||
}
|
|
||||||
bootID := bootEpoch(readTrimFile("/proc/sys/kernel/random/boot_id", ""), names)
|
|
||||||
rxRate, txRate := probeColl.netRate(counted, time.Now())
|
|
||||||
tcp, udp := connCounts(readProcFile("net/sockstat"), readProcFile("net/sockstat6"))
|
|
||||||
return &ProbeMetrics{
|
|
||||||
BootID: bootID,
|
|
||||||
Uptime: parseUptime(readProcFile("uptime")),
|
|
||||||
CPU: probeColl.cpuPercent(parseCPUJiffies(readProcFile("stat"))),
|
|
||||||
Load: parseLoadavg(readProcFile("loadavg")),
|
|
||||||
MemTotal: memTotal,
|
|
||||||
MemUsed: memUsed,
|
|
||||||
SwapTotal: swapTotal,
|
|
||||||
SwapUsed: swapUsed,
|
|
||||||
DiskTotal: diskTotal,
|
|
||||||
DiskUsed: diskUsed,
|
|
||||||
NetRxTotal: rxTotal,
|
|
||||||
NetTxTotal: txTotal,
|
|
||||||
NetRx: rxRate,
|
|
||||||
NetTx: txRate,
|
|
||||||
TCP: tcp,
|
|
||||||
UDP: udp,
|
|
||||||
Procs: procCount(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// --- file helpers -----------------------------------------------------------
|
|
||||||
|
|
||||||
const sysNetRoot = "/sys/class/net"
|
|
||||||
|
|
||||||
func readProcFile(name string) string {
|
|
||||||
b, err := os.ReadFile(filepath.Join("/proc", name))
|
|
||||||
if err != nil {
|
|
||||||
return ""
|
|
||||||
}
|
|
||||||
return string(b)
|
|
||||||
}
|
|
||||||
|
|
||||||
func readTrimFile(path, fallback string) string {
|
|
||||||
b, err := os.ReadFile(path)
|
|
||||||
if err != nil {
|
|
||||||
return fallback
|
|
||||||
}
|
|
||||||
if s := strings.TrimSpace(string(b)); s != "" {
|
|
||||||
return s
|
|
||||||
}
|
|
||||||
return fallback
|
|
||||||
}
|
|
||||||
|
|
||||||
// --- memory (free口径) -------------------------------------------------------
|
|
||||||
|
|
||||||
func parseMeminfo(text string) map[string]uint64 {
|
|
||||||
out := make(map[string]uint64)
|
|
||||||
for _, line := range strings.Split(text, "\n") {
|
|
||||||
key, rest, ok := strings.Cut(line, ":")
|
|
||||||
if !ok {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
f := strings.Fields(rest)
|
|
||||||
if len(f) == 0 {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
kb, err := strconv.ParseUint(f[0], 10, 64)
|
|
||||||
if err != nil {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
out[key] = kb * 1024
|
|
||||||
}
|
|
||||||
return out
|
|
||||||
}
|
|
||||||
|
|
||||||
// memUsedFree is free(1)'s used column: total - MemAvailable, with the
|
|
||||||
// source's two fallbacks (no MemAvailable -> Free+Buffers+Cached; an
|
|
||||||
// available above the total, as LXCFS can report, -> total - MemFree).
|
|
||||||
func memUsedFree(m map[string]uint64) (total, used uint64) {
|
|
||||||
total = m["MemTotal"]
|
|
||||||
if total == 0 {
|
|
||||||
return 0, 0
|
|
||||||
}
|
|
||||||
avail, ok := m["MemAvailable"]
|
|
||||||
if !ok {
|
|
||||||
avail = m["MemFree"] + m["Buffers"] + m["Cached"]
|
|
||||||
}
|
|
||||||
if avail > total {
|
|
||||||
avail = m["MemFree"]
|
|
||||||
}
|
|
||||||
if avail > total {
|
|
||||||
avail = total
|
|
||||||
}
|
|
||||||
return total, total - avail
|
|
||||||
}
|
|
||||||
|
|
||||||
func swapUsedFree(m map[string]uint64) (total, used uint64) {
|
|
||||||
total = m["SwapTotal"]
|
|
||||||
free := m["SwapFree"]
|
|
||||||
if free > total {
|
|
||||||
free = total
|
|
||||||
}
|
|
||||||
return total, total - free
|
|
||||||
}
|
|
||||||
|
|
||||||
// --- cpu --------------------------------------------------------------------
|
|
||||||
|
|
||||||
func parseCPUJiffies(stat string) *cpuSample {
|
|
||||||
for _, line := range strings.Split(stat, "\n") {
|
|
||||||
if !strings.HasPrefix(line, "cpu ") {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
f := strings.Fields(line[4:])
|
|
||||||
if len(f) < 5 {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
var v [8]uint64
|
|
||||||
var total uint64
|
|
||||||
for i := 0; i < len(f) && i < 8; i++ {
|
|
||||||
n, err := strconv.ParseUint(f[i], 10, 64)
|
|
||||||
if err != nil {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
v[i] = n
|
|
||||||
total += n
|
|
||||||
}
|
|
||||||
return &cpuSample{total: total, idle: v[3] + v[4]}
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *probeCollector) cpuPercent(now *cpuSample) float32 {
|
func (c *probeCollector) cpuPercent(now *cpuSample) float32 {
|
||||||
if now == nil {
|
if now == nil {
|
||||||
return 0
|
return 0
|
||||||
@@ -315,344 +160,14 @@ func (c *probeCollector) cpuPercent(now *cpuSample) float32 {
|
|||||||
return pct
|
return pct
|
||||||
}
|
}
|
||||||
|
|
||||||
func parseLoadavg(text string) [3]float32 {
|
// bootEpoch = "{boot identity}/{FNV-1a-64 of the sorted counted names}".
|
||||||
var out [3]float32
|
|
||||||
f := strings.Fields(text)
|
|
||||||
for i := 0; i < 3 && i < len(f); i++ {
|
|
||||||
if v, err := strconv.ParseFloat(f[i], 32); err == nil {
|
|
||||||
out[i] = float32(v)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return out
|
|
||||||
}
|
|
||||||
|
|
||||||
func parseUptime(text string) uint64 {
|
|
||||||
f := strings.Fields(text)
|
|
||||||
if len(f) == 0 {
|
|
||||||
return 0
|
|
||||||
}
|
|
||||||
v, err := strconv.ParseFloat(f[0], 64)
|
|
||||||
if err != nil || v < 0 {
|
|
||||||
return 0
|
|
||||||
}
|
|
||||||
return uint64(v)
|
|
||||||
}
|
|
||||||
|
|
||||||
// cpuInfo returns the display name and logical core count from /proc/cpuinfo.
|
|
||||||
func cpuInfo(text string) (string, uint32) {
|
|
||||||
name := ""
|
|
||||||
cores := uint32(0)
|
|
||||||
for _, line := range strings.Split(text, "\n") {
|
|
||||||
key, val, ok := strings.Cut(line, ":")
|
|
||||||
if !ok {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
switch strings.TrimSpace(key) {
|
|
||||||
case "processor":
|
|
||||||
cores++
|
|
||||||
case "model name", "Model", "cpu model":
|
|
||||||
if name == "" {
|
|
||||||
name = strings.TrimSpace(val)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if cores == 0 {
|
|
||||||
cores = uint32(runtime.NumCPU())
|
|
||||||
if cores == 0 {
|
|
||||||
cores = 1
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if name == "" {
|
|
||||||
name = armCPUName(text)
|
|
||||||
}
|
|
||||||
if name == "" {
|
|
||||||
name = "unknown"
|
|
||||||
}
|
|
||||||
return name, cores
|
|
||||||
}
|
|
||||||
|
|
||||||
// armCPUName decodes AArch64 MIDR implementer/part pairs the way lscpu does,
|
|
||||||
// joining distinct core types with "/" (e.g. Cortex-A55/Cortex-A76).
|
|
||||||
func armCPUName(text string) string {
|
|
||||||
var vendor *armVendor
|
|
||||||
var names []string
|
|
||||||
seen := map[string]bool{}
|
|
||||||
for _, line := range strings.Split(text, "\n") {
|
|
||||||
key, val, ok := strings.Cut(line, ":")
|
|
||||||
if !ok {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
switch strings.TrimSpace(key) {
|
|
||||||
case "CPU implementer":
|
|
||||||
vendor = nil
|
|
||||||
if id, ok := parseHexID(val); ok {
|
|
||||||
for i := range armVendors {
|
|
||||||
if armVendors[i].id == id {
|
|
||||||
vendor = &armVendors[i]
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
case "CPU part":
|
|
||||||
if vendor == nil {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
name := vendor.name
|
|
||||||
if id, ok := parseHexID(val); ok {
|
|
||||||
for _, p := range vendor.parts {
|
|
||||||
if p.id == id {
|
|
||||||
name = p.name
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if !seen[name] {
|
|
||||||
seen[name] = true
|
|
||||||
names = append(names, name)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return strings.Join(names, "/")
|
|
||||||
}
|
|
||||||
|
|
||||||
func parseHexID(s string) (uint32, bool) {
|
|
||||||
s = strings.TrimSpace(strings.TrimPrefix(strings.TrimSpace(s), "0x"))
|
|
||||||
v, err := strconv.ParseUint(s, 16, 32)
|
|
||||||
if err != nil {
|
|
||||||
return 0, false
|
|
||||||
}
|
|
||||||
return uint32(v), true
|
|
||||||
}
|
|
||||||
|
|
||||||
type armPart struct {
|
|
||||||
id uint32
|
|
||||||
name string
|
|
||||||
}
|
|
||||||
type armVendor struct {
|
|
||||||
id uint32
|
|
||||||
name string
|
|
||||||
parts []armPart
|
|
||||||
}
|
|
||||||
|
|
||||||
// Subset of the source's MIDR table (util-linux lscpu-arm names): the parts
|
|
||||||
// plausibly met on VPS hardware. An unlisted part falls back to the vendor.
|
|
||||||
var armVendors = []armVendor{
|
|
||||||
{0x41, "ARM", []armPart{
|
|
||||||
{0xd03, "Cortex-A53"}, {0xd04, "Cortex-A35"}, {0xd05, "Cortex-A55"},
|
|
||||||
{0xd07, "Cortex-A57"}, {0xd08, "Cortex-A72"}, {0xd09, "Cortex-A73"},
|
|
||||||
{0xd0a, "Cortex-A75"}, {0xd0b, "Cortex-A76"}, {0xd0c, "Neoverse-N1"},
|
|
||||||
{0xd0d, "Cortex-A77"}, {0xd40, "Neoverse-V1"}, {0xd41, "Cortex-A78"},
|
|
||||||
{0xd44, "Cortex-X1"}, {0xd46, "Cortex-A510"}, {0xd47, "Cortex-A710"},
|
|
||||||
{0xd48, "Cortex-X2"}, {0xd49, "Neoverse-N2"}, {0xd4d, "Cortex-A715"},
|
|
||||||
{0xd4e, "Cortex-X3"}, {0xd4f, "Neoverse-V2"}, {0xd80, "Cortex-A520"},
|
|
||||||
{0xd81, "Cortex-A720"}, {0xd82, "Cortex-X4"}, {0xd84, "Neoverse-V3"},
|
|
||||||
{0xd85, "Cortex-X925"}, {0xd87, "Cortex-A725"}, {0xd8e, "Neoverse-N3"},
|
|
||||||
}},
|
|
||||||
{0x48, "HiSilicon", []armPart{
|
|
||||||
{0xd01, "Kunpeng-920"}, {0xd02, "Kunpeng-920"}, {0xd03, "Kunpeng-920"},
|
|
||||||
{0xd06, "Kunpeng-950"}, {0xd22, "Kunpeng-920"},
|
|
||||||
}},
|
|
||||||
{0x51, "Qualcomm", []armPart{{0x001, "Oryon"}, {0x002, "Oryon-2"}}},
|
|
||||||
{0x50, "APM", []armPart{{0x000, "X-Gene"}}},
|
|
||||||
{0xc0, "Ampere", []armPart{{0xac3, "Ampere-1"}, {0xac4, "Ampere-1a"}}},
|
|
||||||
{0x6d, "Microsoft", []armPart{{0xd49, "Azure-Cobalt-100"}}},
|
|
||||||
}
|
|
||||||
|
|
||||||
// --- os / virt ---------------------------------------------------------------
|
|
||||||
|
|
||||||
func osPrettyName() string {
|
|
||||||
for _, path := range []string{"/etc/os-release", "/usr/lib/os-release"} {
|
|
||||||
b, err := os.ReadFile(path)
|
|
||||||
if err != nil {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
for _, line := range strings.Split(string(b), "\n") {
|
|
||||||
if v, ok := strings.CutPrefix(line, "PRETTY_NAME="); ok {
|
|
||||||
return strings.Trim(v, "\"'")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return "Linux"
|
|
||||||
}
|
|
||||||
|
|
||||||
// detectVirt ports the source's chain: containers first, then hypervisor.
|
|
||||||
// CPUID is unreachable from Go's stdlib, so the x86 signature check is
|
|
||||||
// replaced by the DMI strings (the source's own non-x86 fallback) plus the
|
|
||||||
// /proc/cpuinfo hypervisor flag — a guest of a hypervisor that hides both
|
|
||||||
// reads as "none", same caveat the source documents for cleared CPUID bits.
|
|
||||||
func detectVirt(mounts, cpuinfo string) string {
|
|
||||||
if _, err := os.Stat("/proc/vz"); err == nil {
|
|
||||||
if _, err := os.Stat("/proc/bc"); err != nil {
|
|
||||||
return "openvz"
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if s := readTrimFile("/run/systemd/container", ""); s != "" {
|
|
||||||
return s
|
|
||||||
}
|
|
||||||
if _, err := os.Stat("/.dockerenv"); err == nil {
|
|
||||||
return "docker"
|
|
||||||
}
|
|
||||||
if _, err := os.Stat("/run/.containerenv"); err == nil {
|
|
||||||
return "podman"
|
|
||||||
}
|
|
||||||
for _, line := range strings.Split(mounts, "\n") {
|
|
||||||
f := strings.Fields(line)
|
|
||||||
if len(f) >= 3 && f[2] == "fuse.lxcfs" && strings.HasPrefix(f[1], "/proc/") {
|
|
||||||
return "lxc"
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if _, err := os.Stat("/proc/xen"); err == nil {
|
|
||||||
return "xen"
|
|
||||||
}
|
|
||||||
if _, err := os.Stat("/sys/hypervisor/type"); err == nil {
|
|
||||||
return "xen"
|
|
||||||
}
|
|
||||||
dmi := strings.ToLower(readTrimFile("/sys/class/dmi/id/product_name", "") + " " +
|
|
||||||
readTrimFile("/sys/class/dmi/id/sys_vendor", ""))
|
|
||||||
for _, m := range []struct{ word, name string }{
|
|
||||||
{"kvm", "kvm"}, {"qemu", "kvm"}, {"amazon ec2", "kvm"}, {"google", "kvm"},
|
|
||||||
{"microsoft", "hyper-v"}, {"vmware", "vmware"}, {"virtualbox", "virtualbox"},
|
|
||||||
{"bhyve", "bhyve"}, {"xen", "xen"},
|
|
||||||
} {
|
|
||||||
if strings.Contains(dmi, m.word) {
|
|
||||||
return m.name
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if _, err := os.Stat("/sys/bus/acpi/devices/QEMU0002:00"); err == nil {
|
|
||||||
return "kvm"
|
|
||||||
}
|
|
||||||
if strings.Contains(cpuinfo, "hypervisor") {
|
|
||||||
return "vm"
|
|
||||||
}
|
|
||||||
return "none"
|
|
||||||
}
|
|
||||||
|
|
||||||
// --- network: counted set, epoch, rates -------------------------------------
|
|
||||||
|
|
||||||
type netCounters struct {
|
|
||||||
name string
|
|
||||||
rx, tx uint64
|
|
||||||
}
|
|
||||||
|
|
||||||
func parseNetDev(text string) []netCounters {
|
|
||||||
var out []netCounters
|
|
||||||
lines := strings.Split(text, "\n")
|
|
||||||
for _, line := range lines {
|
|
||||||
name, rest, ok := strings.Cut(line, ":")
|
|
||||||
if !ok {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
f := strings.Fields(rest)
|
|
||||||
if len(f) < 16 {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
rx, err1 := strconv.ParseUint(f[0], 10, 64)
|
|
||||||
tx, err2 := strconv.ParseUint(f[8], 10, 64)
|
|
||||||
if err1 != nil || err2 != nil {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
out = append(out, netCounters{name: strings.TrimSpace(name), rx: rx, tx: tx})
|
|
||||||
}
|
|
||||||
return out
|
|
||||||
}
|
|
||||||
|
|
||||||
// skipIfacePrefixes is layer 1 of the filter (source SKIP_IFACES).
|
|
||||||
var skipIfacePrefixes = []string{
|
|
||||||
"lo", "docker", "veth", "br-", "virbr", "tap", "tun", "wg", "tailscale",
|
|
||||||
"cni", "flannel", "podman", "fwbr", "fwpr", "fwln", "ifb", "gretap",
|
|
||||||
"erspan", "kube", "cali", "nerdctl", "lxc", "cilium", "zt",
|
|
||||||
}
|
|
||||||
|
|
||||||
func skipIfaceName(name string) bool {
|
|
||||||
for _, p := range skipIfacePrefixes {
|
|
||||||
if strings.HasPrefix(name, p) {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
// isStackedName is layer 2: bonds, bridges, VLAN children, PPPoE overlays —
|
|
||||||
// the kernel books their bytes again on the device beneath.
|
|
||||||
func isStackedName(name string) bool {
|
|
||||||
if strings.Contains(name, ".") {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
for _, p := range []string{"bond", "br", "vlan", "vmbr", "pppoe-"} {
|
|
||||||
if strings.HasPrefix(name, p) {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
var tunnelLinkTypes = map[string]bool{
|
|
||||||
"65534": true, "768": true, "769": true, "776": true, "778": true, "823": true,
|
|
||||||
}
|
|
||||||
|
|
||||||
// countedElsewhere is layer 3 (sysfs): the kernel shows this device's bytes
|
|
||||||
// are already counted on another device. Hardware (a `device` link) exempts
|
|
||||||
// a device from the tunnel/port rules — a raw-IP LTE modem shares its link
|
|
||||||
// type with WireGuard but carries real traffic.
|
|
||||||
func countedElsewhere(sysRoot, name string) bool {
|
|
||||||
dev := filepath.Join(sysRoot, name)
|
|
||||||
uevent := readTrimFile(filepath.Join(dev, "uevent"), "")
|
|
||||||
devtypeIs := func(set ...string) bool {
|
|
||||||
for _, line := range strings.Split(uevent, "\n") {
|
|
||||||
for _, want := range set {
|
|
||||||
if line == want {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
if devtypeIs("DEVTYPE=bridge", "DEVTYPE=bond") {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
if entries, err := os.ReadDir(dev); err == nil {
|
|
||||||
for _, e := range entries {
|
|
||||||
if strings.HasPrefix(e.Name(), "lower_") {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if _, err := os.Stat(filepath.Join(dev, "device")); err == nil {
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
if _, err := os.Stat(filepath.Join(dev, "brport")); err == nil {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
if tunnelLinkTypes[readTrimFile(filepath.Join(dev, "type"), "")] {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
return devtypeIs("DEVTYPE=vxlan", "DEVTYPE=geneve")
|
|
||||||
}
|
|
||||||
|
|
||||||
// ifaceCounts applies the three layers. Exposed for tests with a fake sysfs.
|
|
||||||
func ifaceCounts(sysRoot, name string) bool {
|
|
||||||
if skipIfaceName(name) || isStackedName(name) {
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
return !countedElsewhere(sysRoot, name)
|
|
||||||
}
|
|
||||||
|
|
||||||
func countedInterfaces(all []netCounters, sysRoot string) []netCounters {
|
|
||||||
out := make([]netCounters, 0, len(all))
|
|
||||||
for _, c := range all {
|
|
||||||
if ifaceCounts(sysRoot, c.name) {
|
|
||||||
out = append(out, c)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return out
|
|
||||||
}
|
|
||||||
|
|
||||||
// bootEpoch = "{kernel boot_id}/{FNV-1a-64 of the sorted counted names}".
|
|
||||||
// The master only tests it for equality: any change re-baselines.
|
// The master only tests it for equality: any change re-baselines.
|
||||||
func bootEpoch(kernelBootID string, names []string) string {
|
func bootEpoch(bootID string, names []string) string {
|
||||||
sorted := append([]string(nil), names...)
|
sorted := append([]string(nil), names...)
|
||||||
sort.Strings(sorted)
|
sort.Strings(sorted)
|
||||||
h := fnv.New64a()
|
h := fnv.New64a()
|
||||||
_, _ = h.Write([]byte(strings.Join(sorted, "\n")))
|
_, _ = h.Write([]byte(strings.Join(sorted, "\n")))
|
||||||
return kernelBootID + "/" + hex16(h.Sum64())
|
return bootID + "/" + hex16(h.Sum64())
|
||||||
}
|
}
|
||||||
|
|
||||||
func hex16(v uint64) string {
|
func hex16(v uint64) string {
|
||||||
@@ -697,136 +212,6 @@ func (c *probeCollector) netRate(counted []netCounters, now time.Time) (uint64,
|
|||||||
return uint64(float64(rx) / secs), uint64(float64(tx) / secs)
|
return uint64(float64(rx) / secs), uint64(float64(tx) / secs)
|
||||||
}
|
}
|
||||||
|
|
||||||
// --- addresses (scope §1.5) ---------------------------------------------------
|
|
||||||
|
|
||||||
// probeAddresses picks one address per family: candidates come from
|
|
||||||
// non-skip interfaces that are up, are not link-local, prefer public over
|
|
||||||
// private, and for v6 avoid temporary/deprecated/tentative addresses.
|
|
||||||
func probeAddresses() (string, string) {
|
|
||||||
transient := transientV6(readProcFile("net/if_inet6"))
|
|
||||||
ifaces, err := net.Interfaces()
|
|
||||||
if err != nil {
|
|
||||||
return "", ""
|
|
||||||
}
|
|
||||||
type cand struct {
|
|
||||||
ip net.IP
|
|
||||||
transient bool
|
|
||||||
}
|
|
||||||
var held []cand
|
|
||||||
for _, ifc := range ifaces {
|
|
||||||
if ifc.Flags&net.FlagUp == 0 {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if readTrimFile(filepath.Join(sysNetRoot, ifc.Name, "operstate"), "up") != "up" {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
addrs, err := ifc.Addrs()
|
|
||||||
if err != nil {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
for _, addr := range addrs {
|
|
||||||
var ip net.IP
|
|
||||||
switch v := addr.(type) {
|
|
||||||
case *net.IPNet:
|
|
||||||
ip = v.IP
|
|
||||||
case *net.IPAddr:
|
|
||||||
ip = v.IP
|
|
||||||
}
|
|
||||||
if ip == nil || ip.IsLinkLocalUnicast() || ip.IsLinkLocalMulticast() {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if !lendsAddress(ifc.Name, ip) {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
c := cand{ip: ip}
|
|
||||||
if ip6 := ip.To16(); ip.To4() == nil && ip6 != nil {
|
|
||||||
for _, t := range transient {
|
|
||||||
if t.Equal(ip6) {
|
|
||||||
c.transient = true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
held = append(held, c)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
pick := func(v6 bool) string {
|
|
||||||
var best *cand
|
|
||||||
for i := range held {
|
|
||||||
c := &held[i]
|
|
||||||
if (c.ip.To4() == nil) != v6 {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if best == nil {
|
|
||||||
best = c
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
bp, cp := isPublicIP(best.ip), isPublicIP(c.ip)
|
|
||||||
if cp != bp {
|
|
||||||
if cp {
|
|
||||||
best = c
|
|
||||||
}
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if v6 && best.transient && !c.transient {
|
|
||||||
best = c
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if best == nil {
|
|
||||||
return ""
|
|
||||||
}
|
|
||||||
return best.ip.String()
|
|
||||||
}
|
|
||||||
return pick(false), pick(true)
|
|
||||||
}
|
|
||||||
|
|
||||||
// lendsAddress mirrors the source: a skipped interface lends no address,
|
|
||||||
// except a WireGuard interface lending a public v6 that is not in
|
|
||||||
// Cloudflare WARP's shared 2606:4700::/32.
|
|
||||||
func lendsAddress(name string, ip net.IP) bool {
|
|
||||||
if !skipIfaceName(name) {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
if strings.HasPrefix(name, "wg") && ip.To4() == nil && isPublicIP(ip) {
|
|
||||||
ip16 := ip.To16()
|
|
||||||
return !(ip16[0] == 0x26 && ip16[1] == 0x06 && ip16[2] == 0x47 && ip16[3] == 0x00)
|
|
||||||
}
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
// transientV6 parses /proc/net/if_inet6 for addresses carrying
|
|
||||||
// TEMPORARY|DADFAILED|DEPRECATED|TENTATIVE flags.
|
|
||||||
func transientV6(text string) []net.IP {
|
|
||||||
const transientFlags = 0x01 | 0x08 | 0x20 | 0x40
|
|
||||||
var out []net.IP
|
|
||||||
for _, line := range strings.Split(text, "\n") {
|
|
||||||
f := strings.Fields(line)
|
|
||||||
if len(f) < 6 {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
flags, err := strconv.ParseUint(f[4], 16, 32)
|
|
||||||
if err != nil || flags&transientFlags == 0 {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if len(f[0]) != 32 {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
ip := make(net.IP, 16)
|
|
||||||
ok := true
|
|
||||||
for i := 0; i < 16; i++ {
|
|
||||||
b, err := strconv.ParseUint(f[0][i*2:i*2+2], 16, 8)
|
|
||||||
if err != nil {
|
|
||||||
ok = false
|
|
||||||
break
|
|
||||||
}
|
|
||||||
ip[i] = byte(b)
|
|
||||||
}
|
|
||||||
if ok {
|
|
||||||
out = append(out, ip)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return out
|
|
||||||
}
|
|
||||||
|
|
||||||
// isPublicIP mirrors monitor's is_public: v4 excludes RFC1918, CGNAT
|
// isPublicIP mirrors monitor's is_public: v4 excludes RFC1918, CGNAT
|
||||||
// 100.64/10, loopback, link-local, 0/8, >=224, 192.0.0/24, 198.18/15;
|
// 100.64/10, loopback, link-local, 0/8, >=224, 192.0.0/24, 198.18/15;
|
||||||
// v6 is only 2000::/3.
|
// v6 is only 2000::/3.
|
||||||
@@ -853,133 +238,3 @@ func isPublicIP(ip net.IP) bool {
|
|||||||
}
|
}
|
||||||
return ip16[0]&0xe0 == 0x20
|
return ip16[0]&0xe0 == 0x20
|
||||||
}
|
}
|
||||||
|
|
||||||
// --- sockets / processes ------------------------------------------------------
|
|
||||||
|
|
||||||
func connCounts(v4, v6 string) (tcp, udp uint32) {
|
|
||||||
stat := func(text, prefix, key string) uint32 {
|
|
||||||
for _, line := range strings.Split(text, "\n") {
|
|
||||||
if !strings.HasPrefix(line, prefix) {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
f := strings.Fields(line[len(prefix):])
|
|
||||||
for i := 0; i+1 < len(f); i++ {
|
|
||||||
if f[i] == key {
|
|
||||||
if n, err := strconv.ParseUint(f[i+1], 10, 32); err == nil {
|
|
||||||
return uint32(n)
|
|
||||||
}
|
|
||||||
return 0
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return 0
|
|
||||||
}
|
|
||||||
tcp = stat(v4, "TCP:", "inuse") + stat(v4, "TCP:", "tw") + stat(v6, "TCP6:", "inuse")
|
|
||||||
udp = stat(v4, "UDP:", "inuse") + stat(v6, "UDP6:", "inuse")
|
|
||||||
return tcp, udp
|
|
||||||
}
|
|
||||||
|
|
||||||
func procCount() uint32 {
|
|
||||||
entries, err := os.ReadDir("/proc")
|
|
||||||
if err != nil {
|
|
||||||
return 0
|
|
||||||
}
|
|
||||||
var n uint32
|
|
||||||
for _, e := range entries {
|
|
||||||
name := e.Name()
|
|
||||||
digits := name != ""
|
|
||||||
for i := 0; i < len(name); i++ {
|
|
||||||
if name[i] < '0' || name[i] > '9' {
|
|
||||||
digits = false
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if digits {
|
|
||||||
n++
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return n
|
|
||||||
}
|
|
||||||
|
|
||||||
// --- disk (df口径) ---------------------------------------------------------------
|
|
||||||
|
|
||||||
var skipFSTypes = map[string]bool{
|
|
||||||
"tmpfs": true, "devtmpfs": true, "proc": true, "sysfs": true,
|
|
||||||
"cgroup": true, "cgroup2": true, "devpts": true, "mqueue": true,
|
|
||||||
"hugetlbfs": true, "debugfs": true, "tracefs": true, "securityfs": true,
|
|
||||||
"pstore": true, "bpf": true, "configfs": true, "fusectl": true,
|
|
||||||
"binfmt_misc": true, "autofs": true, "squashfs": true, "ramfs": true,
|
|
||||||
"efivarfs": true, "nsfs": true, "overlay": true, "ecryptfs": true,
|
|
||||||
"fuse": true, "rpc_pipefs": true,
|
|
||||||
"nfs": true, "nfs4": true, "cifs": true, "smb3": true, "ceph": true,
|
|
||||||
"glusterfs": true, "9p": true,
|
|
||||||
}
|
|
||||||
|
|
||||||
func skipFSType(fstype string) bool {
|
|
||||||
if skipFSTypes[fstype] {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
// Flavours: "fuse.lxcfs" is skipped by its "fuse." prefix, like the source.
|
|
||||||
if i := strings.IndexByte(fstype, '.'); i > 0 && skipFSTypes[fstype[:i]] {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
// realMountPoints ports parse_mounts: only the topmost mount per path,
|
|
||||||
// deduped by source device (zfs by pool), real storage only.
|
|
||||||
func realMountPoints(text string) []string {
|
|
||||||
type row struct{ dev, mount, fstype string }
|
|
||||||
var rows []row
|
|
||||||
top := map[string]int{}
|
|
||||||
for _, line := range strings.Split(text, "\n") {
|
|
||||||
f := strings.Fields(line)
|
|
||||||
if len(f) < 3 {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
rows = append(rows, row{f[0], f[1], f[2]})
|
|
||||||
top[f[1]] = len(rows) - 1
|
|
||||||
}
|
|
||||||
seen := map[string]bool{}
|
|
||||||
var out []string
|
|
||||||
for i, r := range rows {
|
|
||||||
if top[r.mount] != i || skipFSType(r.fstype) {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if !strings.HasPrefix(r.dev, "/") && r.fstype != "zfs" && r.fstype != "btrfs" {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
key := r.dev
|
|
||||||
if r.fstype == "zfs" {
|
|
||||||
if j := strings.IndexByte(key, '/'); j >= 0 {
|
|
||||||
key = key[:j]
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if seen[key] {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
seen[key] = true
|
|
||||||
out = append(out, strings.ReplaceAll(r.mount, `\040`, " "))
|
|
||||||
}
|
|
||||||
return out
|
|
||||||
}
|
|
||||||
|
|
||||||
// diskUsage sums statfs over the mounts: total = f_blocks, used =
|
|
||||||
// f_blocks - f_bfree (NOT f_bavail — df does not charge the ext reserve).
|
|
||||||
func diskUsage(mounts []string) (total, used uint64) {
|
|
||||||
for _, m := range mounts {
|
|
||||||
var st syscall.Statfs_t
|
|
||||||
if err := syscall.Statfs(m, &st); err != nil {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
bs := uint64(st.Frsize)
|
|
||||||
if bs == 0 {
|
|
||||||
bs = uint64(st.Bsize)
|
|
||||||
}
|
|
||||||
total += st.Blocks * bs
|
|
||||||
if st.Blocks > st.Bfree {
|
|
||||||
used += (st.Blocks - st.Bfree) * bs
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return total, used
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -0,0 +1,56 @@
|
|||||||
|
package agent
|
||||||
|
|
||||||
|
import (
|
||||||
|
"sync"
|
||||||
|
|
||||||
|
"github.com/mhsanaei/3x-ui/v3/internal/logger"
|
||||||
|
)
|
||||||
|
|
||||||
|
// This file holds the probe collector's degrade-don't-die plumbing:
|
||||||
|
// the per-source independence discipline as reusable mechanics,
|
||||||
|
// shared by every platform sampler but exercised today by the Windows
|
||||||
|
// sampler (the Linux sampler reads /proc files, whose failure mode is
|
||||||
|
// an error return, not a panic).
|
||||||
|
//
|
||||||
|
// Two layers:
|
||||||
|
//
|
||||||
|
// - source gates: an optional OS facility (a Win32 procedure, say)
|
||||||
|
// is probed once; when it is missing the source is skipped —
|
||||||
|
// zeroed — with a single warning, never re-probed, and never
|
||||||
|
// reached through a call path that panics on a missing export the
|
||||||
|
// way LazyProc.Call does when it must-finds a procedure;
|
||||||
|
// - guarded calls: a panicking source is recovered, zeroed and
|
||||||
|
// logged, so one bad source can never take the agent process (and
|
||||||
|
// every other source with it) down.
|
||||||
|
|
||||||
|
// sourceGate caches one availability probe. The zero value is ready.
|
||||||
|
type sourceGate struct {
|
||||||
|
once sync.Once
|
||||||
|
available bool
|
||||||
|
}
|
||||||
|
|
||||||
|
// enabled runs probe on first use and reports whether the source may
|
||||||
|
// be used. probe returns nil when the facility is present. A failed
|
||||||
|
// probe disables the source for the process lifetime and logs once.
|
||||||
|
func (g *sourceGate) enabled(name string, probe func() error) bool {
|
||||||
|
g.once.Do(func() {
|
||||||
|
if err := probe(); err != nil {
|
||||||
|
logger.Warningf("probe: %s unavailable, source disabled: %v", name, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
g.available = true
|
||||||
|
})
|
||||||
|
return g.available
|
||||||
|
}
|
||||||
|
|
||||||
|
// guardedCall runs fn and converts a panic into the zero value plus a
|
||||||
|
// warning, so one failing metrics source degrades alone.
|
||||||
|
func guardedCall[T any](name string, fn func() T) (out T) {
|
||||||
|
defer func() {
|
||||||
|
if r := recover(); r != nil {
|
||||||
|
logger.Warningf("probe: %s panicked, source zeroed: %v", name, r)
|
||||||
|
out = *new(T)
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
return fn()
|
||||||
|
}
|
||||||
@@ -0,0 +1,63 @@
|
|||||||
|
package agent
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
// The degrade-don't-die mechanics behind V2.2.1 ①: on the trimmed
|
||||||
|
// Tianyi cloud-PC image the Windows sampler died on its first
|
||||||
|
// collection because a missing iphlpapi export made LazyProc.Call
|
||||||
|
// panic. These tests pin the two layers that replace that failure
|
||||||
|
// mode — a failed availability probe disables the source (probed
|
||||||
|
// once, not per poll), and a panicking source is zeroed, never
|
||||||
|
// propagated.
|
||||||
|
|
||||||
|
func TestSourceGateDisablesFailedProbeOnce(t *testing.T) {
|
||||||
|
var g sourceGate
|
||||||
|
calls := 0
|
||||||
|
probe := func() error {
|
||||||
|
calls++
|
||||||
|
return errors.New("procedure not found")
|
||||||
|
}
|
||||||
|
for i := range 3 {
|
||||||
|
if g.enabled("test-source", probe) {
|
||||||
|
t.Fatalf("call %d: gate enabled despite failed probe", i)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if calls != 1 {
|
||||||
|
t.Fatalf("probe ran %d times, want exactly 1 (cached verdict)", calls)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSourceGateEnablesSuccessfulProbeOnce(t *testing.T) {
|
||||||
|
var g sourceGate
|
||||||
|
calls := 0
|
||||||
|
probe := func() error {
|
||||||
|
calls++
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
for i := range 3 {
|
||||||
|
if !g.enabled("test-source", probe) {
|
||||||
|
t.Fatalf("call %d: gate disabled despite successful probe", i)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if calls != 1 {
|
||||||
|
t.Fatalf("probe ran %d times, want exactly 1 (cached verdict)", calls)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestGuardedCallRecoversPanicToZero(t *testing.T) {
|
||||||
|
got := guardedCall("test-source", func() uint64 {
|
||||||
|
panic("Failed to find FreeMibIfTable2 procedure in iphlpapi.dll")
|
||||||
|
})
|
||||||
|
if got != 0 {
|
||||||
|
t.Fatalf("guardedCall after panic = %d, want zero value", got)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestGuardedCallPassesValueThrough(t *testing.T) {
|
||||||
|
if got := guardedCall("test-source", func() string { return "ok" }); got != "ok" {
|
||||||
|
t.Fatalf("guardedCall = %q, want %q", got, "ok")
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,778 @@
|
|||||||
|
//go:build linux
|
||||||
|
|
||||||
|
package agent
|
||||||
|
|
||||||
|
// probe_linux.go is the Linux sampler for the probe collector: host
|
||||||
|
// Facts and per-heartbeat Metrics read directly from /proc and /sys
|
||||||
|
// with the standard library only. The accounting discipline (see
|
||||||
|
// probe.go) is the source's:
|
||||||
|
// - memory "used" is free(1)'s column: MemTotal - MemAvailable;
|
||||||
|
// - disk "used" is df(1)'s: sum over real mounts of (f_blocks-f_bfree);
|
||||||
|
// - network totals sum only "counted" interfaces, selected by the
|
||||||
|
// three-layer filter below so no byte is counted twice;
|
||||||
|
// - boot epoch = kernel boot_id + fingerprint of the counted set.
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"runtime"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
"syscall"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/mhsanaei/3x-ui/v3/internal/config"
|
||||||
|
)
|
||||||
|
|
||||||
|
func sampleFacts() *ProbeFacts {
|
||||||
|
mem := parseMeminfo(readProcFile("meminfo"))
|
||||||
|
diskTotal, _ := diskUsage(realMountPoints(readProcFile("self/mounts")))
|
||||||
|
cpuName, cpuCores := cpuInfo(readProcFile("cpuinfo"))
|
||||||
|
v4, v6 := probeAddresses()
|
||||||
|
return &ProbeFacts{
|
||||||
|
Hostname: readTrimFile("/proc/sys/kernel/hostname", "unknown"),
|
||||||
|
OS: osPrettyName(),
|
||||||
|
Kernel: readTrimFile("/proc/sys/kernel/osrelease", "unknown"),
|
||||||
|
Arch: runtime.GOARCH,
|
||||||
|
Virt: detectVirt(readProcFile("self/mounts"), readProcFile("cpuinfo")),
|
||||||
|
CPUName: cpuName,
|
||||||
|
CPUCores: cpuCores,
|
||||||
|
MemTotal: mem["MemTotal"],
|
||||||
|
SwapTotal: mem["SwapTotal"],
|
||||||
|
DiskTotal: diskTotal,
|
||||||
|
AgentVersion: config.GetAgentDisplayVersion(),
|
||||||
|
IPv4: v4,
|
||||||
|
IPv6: v6,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// collectMetrics takes one full sample. Individual sources fail soft.
|
||||||
|
func collectMetrics() *ProbeMetrics {
|
||||||
|
probeColl.mu.Lock()
|
||||||
|
defer probeColl.mu.Unlock()
|
||||||
|
mem := parseMeminfo(readProcFile("meminfo"))
|
||||||
|
memTotal, memUsed := memUsedFree(mem)
|
||||||
|
swapTotal, swapUsed := swapUsedFree(mem)
|
||||||
|
diskTotal, diskUsed := diskUsage(realMountPoints(readProcFile("self/mounts")))
|
||||||
|
allIfaces := parseNetDev(readProcFile("net/dev"))
|
||||||
|
counted := countedInterfaces(allIfaces, sysNetRoot)
|
||||||
|
if pin := statIfacePinValue(); pin != "" {
|
||||||
|
counted = pinnedInterfaces(allIfaces, pin)
|
||||||
|
}
|
||||||
|
var rxTotal, txTotal uint64
|
||||||
|
names := make([]string, 0, len(counted))
|
||||||
|
for _, c := range counted {
|
||||||
|
rxTotal += c.rx
|
||||||
|
txTotal += c.tx
|
||||||
|
names = append(names, c.name)
|
||||||
|
}
|
||||||
|
bootID := bootEpoch(readTrimFile("/proc/sys/kernel/random/boot_id", ""), names)
|
||||||
|
rxRate, txRate := probeColl.netRate(counted, time.Now())
|
||||||
|
tcp, udp := connCounts(readProcFile("net/sockstat"), readProcFile("net/sockstat6"))
|
||||||
|
return &ProbeMetrics{
|
||||||
|
BootID: bootID,
|
||||||
|
Uptime: parseUptime(readProcFile("uptime")),
|
||||||
|
CPU: probeColl.cpuPercent(parseCPUJiffies(readProcFile("stat"))),
|
||||||
|
Load: parseLoadavg(readProcFile("loadavg")),
|
||||||
|
MemTotal: memTotal,
|
||||||
|
MemUsed: memUsed,
|
||||||
|
SwapTotal: swapTotal,
|
||||||
|
SwapUsed: swapUsed,
|
||||||
|
DiskTotal: diskTotal,
|
||||||
|
DiskUsed: diskUsed,
|
||||||
|
NetRxTotal: rxTotal,
|
||||||
|
NetTxTotal: txTotal,
|
||||||
|
NetRx: rxRate,
|
||||||
|
NetTx: txRate,
|
||||||
|
TCP: tcp,
|
||||||
|
UDP: udp,
|
||||||
|
Procs: procCount(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// --- file helpers -----------------------------------------------------------
|
||||||
|
|
||||||
|
const sysNetRoot = "/sys/class/net"
|
||||||
|
|
||||||
|
func readProcFile(name string) string {
|
||||||
|
b, err := os.ReadFile(filepath.Join("/proc", name))
|
||||||
|
if err != nil {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
return string(b)
|
||||||
|
}
|
||||||
|
|
||||||
|
func readTrimFile(path, fallback string) string {
|
||||||
|
b, err := os.ReadFile(path)
|
||||||
|
if err != nil {
|
||||||
|
return fallback
|
||||||
|
}
|
||||||
|
if s := strings.TrimSpace(string(b)); s != "" {
|
||||||
|
return s
|
||||||
|
}
|
||||||
|
return fallback
|
||||||
|
}
|
||||||
|
|
||||||
|
// --- memory (free口径) -------------------------------------------------------
|
||||||
|
|
||||||
|
func parseMeminfo(text string) map[string]uint64 {
|
||||||
|
out := make(map[string]uint64)
|
||||||
|
for _, line := range strings.Split(text, "\n") {
|
||||||
|
key, rest, ok := strings.Cut(line, ":")
|
||||||
|
if !ok {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
f := strings.Fields(rest)
|
||||||
|
if len(f) == 0 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
kb, err := strconv.ParseUint(f[0], 10, 64)
|
||||||
|
if err != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
out[key] = kb * 1024
|
||||||
|
}
|
||||||
|
return out
|
||||||
|
}
|
||||||
|
|
||||||
|
// memUsedFree is free(1)'s used column: total - MemAvailable, with the
|
||||||
|
// source's two fallbacks (no MemAvailable -> Free+Buffers+Cached; an
|
||||||
|
// available above the total, as LXCFS can report, -> total - MemFree).
|
||||||
|
func memUsedFree(m map[string]uint64) (total, used uint64) {
|
||||||
|
total = m["MemTotal"]
|
||||||
|
if total == 0 {
|
||||||
|
return 0, 0
|
||||||
|
}
|
||||||
|
avail, ok := m["MemAvailable"]
|
||||||
|
if !ok {
|
||||||
|
avail = m["MemFree"] + m["Buffers"] + m["Cached"]
|
||||||
|
}
|
||||||
|
if avail > total {
|
||||||
|
avail = m["MemFree"]
|
||||||
|
}
|
||||||
|
if avail > total {
|
||||||
|
avail = total
|
||||||
|
}
|
||||||
|
return total, total - avail
|
||||||
|
}
|
||||||
|
|
||||||
|
func swapUsedFree(m map[string]uint64) (total, used uint64) {
|
||||||
|
total = m["SwapTotal"]
|
||||||
|
free := m["SwapFree"]
|
||||||
|
if free > total {
|
||||||
|
free = total
|
||||||
|
}
|
||||||
|
return total, total - free
|
||||||
|
}
|
||||||
|
|
||||||
|
// --- cpu --------------------------------------------------------------------
|
||||||
|
|
||||||
|
func parseCPUJiffies(stat string) *cpuSample {
|
||||||
|
for _, line := range strings.Split(stat, "\n") {
|
||||||
|
if !strings.HasPrefix(line, "cpu ") {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
f := strings.Fields(line[4:])
|
||||||
|
if len(f) < 5 {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
var v [8]uint64
|
||||||
|
var total uint64
|
||||||
|
for i := 0; i < len(f) && i < 8; i++ {
|
||||||
|
n, err := strconv.ParseUint(f[i], 10, 64)
|
||||||
|
if err != nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
v[i] = n
|
||||||
|
total += n
|
||||||
|
}
|
||||||
|
return &cpuSample{total: total, idle: v[3] + v[4]}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func parseLoadavg(text string) [3]float32 {
|
||||||
|
var out [3]float32
|
||||||
|
f := strings.Fields(text)
|
||||||
|
for i := 0; i < 3 && i < len(f); i++ {
|
||||||
|
if v, err := strconv.ParseFloat(f[i], 32); err == nil {
|
||||||
|
out[i] = float32(v)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return out
|
||||||
|
}
|
||||||
|
|
||||||
|
func parseUptime(text string) uint64 {
|
||||||
|
f := strings.Fields(text)
|
||||||
|
if len(f) == 0 {
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
v, err := strconv.ParseFloat(f[0], 64)
|
||||||
|
if err != nil || v < 0 {
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
return uint64(v)
|
||||||
|
}
|
||||||
|
|
||||||
|
// cpuInfo returns the display name and logical core count from /proc/cpuinfo.
|
||||||
|
func cpuInfo(text string) (string, uint32) {
|
||||||
|
name := ""
|
||||||
|
cores := uint32(0)
|
||||||
|
for _, line := range strings.Split(text, "\n") {
|
||||||
|
key, val, ok := strings.Cut(line, ":")
|
||||||
|
if !ok {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
switch strings.TrimSpace(key) {
|
||||||
|
case "processor":
|
||||||
|
cores++
|
||||||
|
case "model name", "Model", "cpu model":
|
||||||
|
if name == "" {
|
||||||
|
name = strings.TrimSpace(val)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if cores == 0 {
|
||||||
|
cores = uint32(runtime.NumCPU())
|
||||||
|
if cores == 0 {
|
||||||
|
cores = 1
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if name == "" {
|
||||||
|
name = armCPUName(text)
|
||||||
|
}
|
||||||
|
if name == "" {
|
||||||
|
name = "unknown"
|
||||||
|
}
|
||||||
|
return name, cores
|
||||||
|
}
|
||||||
|
|
||||||
|
// armCPUName decodes AArch64 MIDR implementer/part pairs the way lscpu does,
|
||||||
|
// joining distinct core types with "/" (e.g. Cortex-A55/Cortex-A76).
|
||||||
|
func armCPUName(text string) string {
|
||||||
|
var vendor *armVendor
|
||||||
|
var names []string
|
||||||
|
seen := map[string]bool{}
|
||||||
|
for _, line := range strings.Split(text, "\n") {
|
||||||
|
key, val, ok := strings.Cut(line, ":")
|
||||||
|
if !ok {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
switch strings.TrimSpace(key) {
|
||||||
|
case "CPU implementer":
|
||||||
|
vendor = nil
|
||||||
|
if id, ok := parseHexID(val); ok {
|
||||||
|
for i := range armVendors {
|
||||||
|
if armVendors[i].id == id {
|
||||||
|
vendor = &armVendors[i]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
case "CPU part":
|
||||||
|
if vendor == nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
name := vendor.name
|
||||||
|
if id, ok := parseHexID(val); ok {
|
||||||
|
for _, p := range vendor.parts {
|
||||||
|
if p.id == id {
|
||||||
|
name = p.name
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if !seen[name] {
|
||||||
|
seen[name] = true
|
||||||
|
names = append(names, name)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return strings.Join(names, "/")
|
||||||
|
}
|
||||||
|
|
||||||
|
func parseHexID(s string) (uint32, bool) {
|
||||||
|
s = strings.TrimSpace(strings.TrimPrefix(strings.TrimSpace(s), "0x"))
|
||||||
|
v, err := strconv.ParseUint(s, 16, 32)
|
||||||
|
if err != nil {
|
||||||
|
return 0, false
|
||||||
|
}
|
||||||
|
return uint32(v), true
|
||||||
|
}
|
||||||
|
|
||||||
|
type armPart struct {
|
||||||
|
id uint32
|
||||||
|
name string
|
||||||
|
}
|
||||||
|
type armVendor struct {
|
||||||
|
id uint32
|
||||||
|
name string
|
||||||
|
parts []armPart
|
||||||
|
}
|
||||||
|
|
||||||
|
// Subset of the source's MIDR table (util-linux lscpu-arm names): the parts
|
||||||
|
// plausibly met on VPS hardware. An unlisted part falls back to the vendor.
|
||||||
|
var armVendors = []armVendor{
|
||||||
|
{0x41, "ARM", []armPart{
|
||||||
|
{0xd03, "Cortex-A53"}, {0xd04, "Cortex-A35"}, {0xd05, "Cortex-A55"},
|
||||||
|
{0xd07, "Cortex-A57"}, {0xd08, "Cortex-A72"}, {0xd09, "Cortex-A73"},
|
||||||
|
{0xd0a, "Cortex-A75"}, {0xd0b, "Cortex-A76"}, {0xd0c, "Neoverse-N1"},
|
||||||
|
{0xd0d, "Cortex-A77"}, {0xd40, "Neoverse-V1"}, {0xd41, "Cortex-A78"},
|
||||||
|
{0xd44, "Cortex-X1"}, {0xd46, "Cortex-A510"}, {0xd47, "Cortex-A710"},
|
||||||
|
{0xd48, "Cortex-X2"}, {0xd49, "Neoverse-N2"}, {0xd4d, "Cortex-A715"},
|
||||||
|
{0xd4e, "Cortex-X3"}, {0xd4f, "Neoverse-V2"}, {0xd80, "Cortex-A520"},
|
||||||
|
{0xd81, "Cortex-A720"}, {0xd82, "Cortex-X4"}, {0xd84, "Neoverse-V3"},
|
||||||
|
{0xd85, "Cortex-X925"}, {0xd87, "Cortex-A725"}, {0xd8e, "Neoverse-N3"},
|
||||||
|
}},
|
||||||
|
{0x48, "HiSilicon", []armPart{
|
||||||
|
{0xd01, "Kunpeng-920"}, {0xd02, "Kunpeng-920"}, {0xd03, "Kunpeng-920"},
|
||||||
|
{0xd06, "Kunpeng-950"}, {0xd22, "Kunpeng-920"},
|
||||||
|
}},
|
||||||
|
{0x51, "Qualcomm", []armPart{{0x001, "Oryon"}, {0x002, "Oryon-2"}}},
|
||||||
|
{0x50, "APM", []armPart{{0x000, "X-Gene"}}},
|
||||||
|
{0xc0, "Ampere", []armPart{{0xac3, "Ampere-1"}, {0xac4, "Ampere-1a"}}},
|
||||||
|
{0x6d, "Microsoft", []armPart{{0xd49, "Azure-Cobalt-100"}}},
|
||||||
|
}
|
||||||
|
|
||||||
|
// --- os / virt ---------------------------------------------------------------
|
||||||
|
|
||||||
|
func osPrettyName() string {
|
||||||
|
for _, path := range []string{"/etc/os-release", "/usr/lib/os-release"} {
|
||||||
|
b, err := os.ReadFile(path)
|
||||||
|
if err != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
for _, line := range strings.Split(string(b), "\n") {
|
||||||
|
if v, ok := strings.CutPrefix(line, "PRETTY_NAME="); ok {
|
||||||
|
return strings.Trim(v, "\"'")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return "Linux"
|
||||||
|
}
|
||||||
|
|
||||||
|
// detectVirt ports the source's chain: containers first, then hypervisor.
|
||||||
|
// CPUID is unreachable from Go's stdlib, so the x86 signature check is
|
||||||
|
// replaced by the DMI strings (the source's own non-x86 fallback) plus the
|
||||||
|
// /proc/cpuinfo hypervisor flag — a guest of a hypervisor that hides both
|
||||||
|
// reads as "none", same caveat the source documents for cleared CPUID bits.
|
||||||
|
func detectVirt(mounts, cpuinfo string) string {
|
||||||
|
if _, err := os.Stat("/proc/vz"); err == nil {
|
||||||
|
if _, err := os.Stat("/proc/bc"); err != nil {
|
||||||
|
return "openvz"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if s := readTrimFile("/run/systemd/container", ""); s != "" {
|
||||||
|
return s
|
||||||
|
}
|
||||||
|
if _, err := os.Stat("/.dockerenv"); err == nil {
|
||||||
|
return "docker"
|
||||||
|
}
|
||||||
|
if _, err := os.Stat("/run/.containerenv"); err == nil {
|
||||||
|
return "podman"
|
||||||
|
}
|
||||||
|
for _, line := range strings.Split(mounts, "\n") {
|
||||||
|
f := strings.Fields(line)
|
||||||
|
if len(f) >= 3 && f[2] == "fuse.lxcfs" && strings.HasPrefix(f[1], "/proc/") {
|
||||||
|
return "lxc"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if _, err := os.Stat("/proc/xen"); err == nil {
|
||||||
|
return "xen"
|
||||||
|
}
|
||||||
|
if _, err := os.Stat("/sys/hypervisor/type"); err == nil {
|
||||||
|
return "xen"
|
||||||
|
}
|
||||||
|
dmi := strings.ToLower(readTrimFile("/sys/class/dmi/id/product_name", "") + " " +
|
||||||
|
readTrimFile("/sys/class/dmi/id/sys_vendor", ""))
|
||||||
|
for _, m := range []struct{ word, name string }{
|
||||||
|
{"kvm", "kvm"}, {"qemu", "kvm"}, {"amazon ec2", "kvm"}, {"google", "kvm"},
|
||||||
|
{"microsoft", "hyper-v"}, {"vmware", "vmware"}, {"virtualbox", "virtualbox"},
|
||||||
|
{"bhyve", "bhyve"}, {"xen", "xen"},
|
||||||
|
} {
|
||||||
|
if strings.Contains(dmi, m.word) {
|
||||||
|
return m.name
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if _, err := os.Stat("/sys/bus/acpi/devices/QEMU0002:00"); err == nil {
|
||||||
|
return "kvm"
|
||||||
|
}
|
||||||
|
if strings.Contains(cpuinfo, "hypervisor") {
|
||||||
|
return "vm"
|
||||||
|
}
|
||||||
|
return "none"
|
||||||
|
}
|
||||||
|
|
||||||
|
// --- network: counted set, epoch, rates -------------------------------------
|
||||||
|
|
||||||
|
func parseNetDev(text string) []netCounters {
|
||||||
|
var out []netCounters
|
||||||
|
lines := strings.Split(text, "\n")
|
||||||
|
for _, line := range lines {
|
||||||
|
name, rest, ok := strings.Cut(line, ":")
|
||||||
|
if !ok {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
f := strings.Fields(rest)
|
||||||
|
if len(f) < 16 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
rx, err1 := strconv.ParseUint(f[0], 10, 64)
|
||||||
|
tx, err2 := strconv.ParseUint(f[8], 10, 64)
|
||||||
|
if err1 != nil || err2 != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
out = append(out, netCounters{name: strings.TrimSpace(name), rx: rx, tx: tx})
|
||||||
|
}
|
||||||
|
return out
|
||||||
|
}
|
||||||
|
|
||||||
|
// skipIfacePrefixes is layer 1 of the filter (source SKIP_IFACES).
|
||||||
|
var skipIfacePrefixes = []string{
|
||||||
|
"lo", "docker", "veth", "br-", "virbr", "tap", "tun", "wg", "tailscale",
|
||||||
|
"cni", "flannel", "podman", "fwbr", "fwpr", "fwln", "ifb", "gretap",
|
||||||
|
"erspan", "kube", "cali", "nerdctl", "lxc", "cilium", "zt",
|
||||||
|
}
|
||||||
|
|
||||||
|
func skipIfaceName(name string) bool {
|
||||||
|
for _, p := range skipIfacePrefixes {
|
||||||
|
if strings.HasPrefix(name, p) {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
// isStackedName is layer 2: bonds, bridges, VLAN children, PPPoE overlays —
|
||||||
|
// the kernel books their bytes again on the device beneath.
|
||||||
|
func isStackedName(name string) bool {
|
||||||
|
if strings.Contains(name, ".") {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
for _, p := range []string{"bond", "br", "vlan", "vmbr", "pppoe-"} {
|
||||||
|
if strings.HasPrefix(name, p) {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
var tunnelLinkTypes = map[string]bool{
|
||||||
|
"65534": true, "768": true, "769": true, "776": true, "778": true, "823": true,
|
||||||
|
}
|
||||||
|
|
||||||
|
// countedElsewhere is layer 3 (sysfs): the kernel shows this device's bytes
|
||||||
|
// are already counted on another device. Hardware (a `device` link) exempts
|
||||||
|
// a device from the tunnel/port rules — a raw-IP LTE modem shares its link
|
||||||
|
// type with WireGuard but carries real traffic.
|
||||||
|
func countedElsewhere(sysRoot, name string) bool {
|
||||||
|
dev := filepath.Join(sysRoot, name)
|
||||||
|
uevent := readTrimFile(filepath.Join(dev, "uevent"), "")
|
||||||
|
devtypeIs := func(set ...string) bool {
|
||||||
|
for _, line := range strings.Split(uevent, "\n") {
|
||||||
|
for _, want := range set {
|
||||||
|
if line == want {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
if devtypeIs("DEVTYPE=bridge", "DEVTYPE=bond") {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
if entries, err := os.ReadDir(dev); err == nil {
|
||||||
|
for _, e := range entries {
|
||||||
|
if strings.HasPrefix(e.Name(), "lower_") {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if _, err := os.Stat(filepath.Join(dev, "device")); err == nil {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
if _, err := os.Stat(filepath.Join(dev, "brport")); err == nil {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
if tunnelLinkTypes[readTrimFile(filepath.Join(dev, "type"), "")] {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
return devtypeIs("DEVTYPE=vxlan", "DEVTYPE=geneve")
|
||||||
|
}
|
||||||
|
|
||||||
|
// ifaceCounts applies the three layers. Exposed for tests with a fake sysfs.
|
||||||
|
func ifaceCounts(sysRoot, name string) bool {
|
||||||
|
if skipIfaceName(name) || isStackedName(name) {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
return !countedElsewhere(sysRoot, name)
|
||||||
|
}
|
||||||
|
|
||||||
|
func countedInterfaces(all []netCounters, sysRoot string) []netCounters {
|
||||||
|
out := make([]netCounters, 0, len(all))
|
||||||
|
for _, c := range all {
|
||||||
|
if ifaceCounts(sysRoot, c.name) {
|
||||||
|
out = append(out, c)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return out
|
||||||
|
}
|
||||||
|
|
||||||
|
// --- addresses (scope §1.5) ---------------------------------------------------
|
||||||
|
|
||||||
|
// probeAddresses picks one address per family: candidates come from
|
||||||
|
// non-skip interfaces that are up, are not link-local, prefer public over
|
||||||
|
// private, and for v6 avoid temporary/deprecated/tentative addresses.
|
||||||
|
func probeAddresses() (string, string) {
|
||||||
|
transient := transientV6(readProcFile("net/if_inet6"))
|
||||||
|
ifaces, err := net.Interfaces()
|
||||||
|
if err != nil {
|
||||||
|
return "", ""
|
||||||
|
}
|
||||||
|
type cand struct {
|
||||||
|
ip net.IP
|
||||||
|
transient bool
|
||||||
|
}
|
||||||
|
var held []cand
|
||||||
|
for _, ifc := range ifaces {
|
||||||
|
if ifc.Flags&net.FlagUp == 0 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if readTrimFile(filepath.Join(sysNetRoot, ifc.Name, "operstate"), "up") != "up" {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
addrs, err := ifc.Addrs()
|
||||||
|
if err != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
for _, addr := range addrs {
|
||||||
|
var ip net.IP
|
||||||
|
switch v := addr.(type) {
|
||||||
|
case *net.IPNet:
|
||||||
|
ip = v.IP
|
||||||
|
case *net.IPAddr:
|
||||||
|
ip = v.IP
|
||||||
|
}
|
||||||
|
if ip == nil || ip.IsLinkLocalUnicast() || ip.IsLinkLocalMulticast() {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if !lendsAddress(ifc.Name, ip) {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
c := cand{ip: ip}
|
||||||
|
if ip6 := ip.To16(); ip.To4() == nil && ip6 != nil {
|
||||||
|
for _, t := range transient {
|
||||||
|
if t.Equal(ip6) {
|
||||||
|
c.transient = true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
held = append(held, c)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
pick := func(v6 bool) string {
|
||||||
|
var best *cand
|
||||||
|
for i := range held {
|
||||||
|
c := &held[i]
|
||||||
|
if (c.ip.To4() == nil) != v6 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if best == nil {
|
||||||
|
best = c
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
bp, cp := isPublicIP(best.ip), isPublicIP(c.ip)
|
||||||
|
if cp != bp {
|
||||||
|
if cp {
|
||||||
|
best = c
|
||||||
|
}
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if v6 && best.transient && !c.transient {
|
||||||
|
best = c
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if best == nil {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
return best.ip.String()
|
||||||
|
}
|
||||||
|
return pick(false), pick(true)
|
||||||
|
}
|
||||||
|
|
||||||
|
// lendsAddress mirrors the source: a skipped interface lends no address,
|
||||||
|
// except a WireGuard interface lending a public v6 that is not in
|
||||||
|
// Cloudflare WARP's shared 2606:4700::/32.
|
||||||
|
func lendsAddress(name string, ip net.IP) bool {
|
||||||
|
if !skipIfaceName(name) {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
if strings.HasPrefix(name, "wg") && ip.To4() == nil && isPublicIP(ip) {
|
||||||
|
ip16 := ip.To16()
|
||||||
|
return !(ip16[0] == 0x26 && ip16[1] == 0x06 && ip16[2] == 0x47 && ip16[3] == 0x00)
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
// transientV6 parses /proc/net/if_inet6 for addresses carrying
|
||||||
|
// TEMPORARY|DADFAILED|DEPRECATED|TENTATIVE flags.
|
||||||
|
func transientV6(text string) []net.IP {
|
||||||
|
const transientFlags = 0x01 | 0x08 | 0x20 | 0x40
|
||||||
|
var out []net.IP
|
||||||
|
for _, line := range strings.Split(text, "\n") {
|
||||||
|
f := strings.Fields(line)
|
||||||
|
if len(f) < 6 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
flags, err := strconv.ParseUint(f[4], 16, 32)
|
||||||
|
if err != nil || flags&transientFlags == 0 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if len(f[0]) != 32 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
ip := make(net.IP, 16)
|
||||||
|
ok := true
|
||||||
|
for i := 0; i < 16; i++ {
|
||||||
|
b, err := strconv.ParseUint(f[0][i*2:i*2+2], 16, 8)
|
||||||
|
if err != nil {
|
||||||
|
ok = false
|
||||||
|
break
|
||||||
|
}
|
||||||
|
ip[i] = byte(b)
|
||||||
|
}
|
||||||
|
if ok {
|
||||||
|
out = append(out, ip)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return out
|
||||||
|
}
|
||||||
|
|
||||||
|
// --- sockets / processes ------------------------------------------------------
|
||||||
|
|
||||||
|
func connCounts(v4, v6 string) (tcp, udp uint32) {
|
||||||
|
stat := func(text, prefix, key string) uint32 {
|
||||||
|
for _, line := range strings.Split(text, "\n") {
|
||||||
|
if !strings.HasPrefix(line, prefix) {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
f := strings.Fields(line[len(prefix):])
|
||||||
|
for i := 0; i+1 < len(f); i++ {
|
||||||
|
if f[i] == key {
|
||||||
|
if n, err := strconv.ParseUint(f[i+1], 10, 32); err == nil {
|
||||||
|
return uint32(n)
|
||||||
|
}
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
tcp = stat(v4, "TCP:", "inuse") + stat(v4, "TCP:", "tw") + stat(v6, "TCP6:", "inuse")
|
||||||
|
udp = stat(v4, "UDP:", "inuse") + stat(v6, "UDP6:", "inuse")
|
||||||
|
return tcp, udp
|
||||||
|
}
|
||||||
|
|
||||||
|
func procCount() uint32 {
|
||||||
|
entries, err := os.ReadDir("/proc")
|
||||||
|
if err != nil {
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
var n uint32
|
||||||
|
for _, e := range entries {
|
||||||
|
name := e.Name()
|
||||||
|
digits := name != ""
|
||||||
|
for i := 0; i < len(name); i++ {
|
||||||
|
if name[i] < '0' || name[i] > '9' {
|
||||||
|
digits = false
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if digits {
|
||||||
|
n++
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return n
|
||||||
|
}
|
||||||
|
|
||||||
|
// --- disk (df口径) ---------------------------------------------------------------
|
||||||
|
|
||||||
|
var skipFSTypes = map[string]bool{
|
||||||
|
"tmpfs": true, "devtmpfs": true, "proc": true, "sysfs": true,
|
||||||
|
"cgroup": true, "cgroup2": true, "devpts": true, "mqueue": true,
|
||||||
|
"hugetlbfs": true, "debugfs": true, "tracefs": true, "securityfs": true,
|
||||||
|
"pstore": true, "bpf": true, "configfs": true, "fusectl": true,
|
||||||
|
"binfmt_misc": true, "autofs": true, "squashfs": true, "ramfs": true,
|
||||||
|
"efivarfs": true, "nsfs": true, "overlay": true, "ecryptfs": true,
|
||||||
|
"fuse": true, "rpc_pipefs": true,
|
||||||
|
"nfs": true, "nfs4": true, "cifs": true, "smb3": true, "ceph": true,
|
||||||
|
"glusterfs": true, "9p": true,
|
||||||
|
}
|
||||||
|
|
||||||
|
func skipFSType(fstype string) bool {
|
||||||
|
if skipFSTypes[fstype] {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
// Flavours: "fuse.lxcfs" is skipped by its "fuse." prefix, like the source.
|
||||||
|
if i := strings.IndexByte(fstype, '.'); i > 0 && skipFSTypes[fstype[:i]] {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
// realMountPoints ports parse_mounts: only the topmost mount per path,
|
||||||
|
// deduped by source device (zfs by pool), real storage only.
|
||||||
|
func realMountPoints(text string) []string {
|
||||||
|
type row struct{ dev, mount, fstype string }
|
||||||
|
var rows []row
|
||||||
|
top := map[string]int{}
|
||||||
|
for _, line := range strings.Split(text, "\n") {
|
||||||
|
f := strings.Fields(line)
|
||||||
|
if len(f) < 3 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
rows = append(rows, row{f[0], f[1], f[2]})
|
||||||
|
top[f[1]] = len(rows) - 1
|
||||||
|
}
|
||||||
|
seen := map[string]bool{}
|
||||||
|
var out []string
|
||||||
|
for i, r := range rows {
|
||||||
|
if top[r.mount] != i || skipFSType(r.fstype) {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if !strings.HasPrefix(r.dev, "/") && r.fstype != "zfs" && r.fstype != "btrfs" {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
key := r.dev
|
||||||
|
if r.fstype == "zfs" {
|
||||||
|
if j := strings.IndexByte(key, '/'); j >= 0 {
|
||||||
|
key = key[:j]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if seen[key] {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
seen[key] = true
|
||||||
|
out = append(out, strings.ReplaceAll(r.mount, `\040`, " "))
|
||||||
|
}
|
||||||
|
return out
|
||||||
|
}
|
||||||
|
|
||||||
|
// diskUsage sums statfs over the mounts: total = f_blocks, used =
|
||||||
|
// f_blocks - f_bfree (NOT f_bavail — df does not charge the ext reserve).
|
||||||
|
func diskUsage(mounts []string) (total, used uint64) {
|
||||||
|
for _, m := range mounts {
|
||||||
|
var st syscall.Statfs_t
|
||||||
|
if err := syscall.Statfs(m, &st); err != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
bs := uint64(st.Frsize)
|
||||||
|
if bs == 0 {
|
||||||
|
bs = uint64(st.Bsize)
|
||||||
|
}
|
||||||
|
total += st.Blocks * bs
|
||||||
|
if st.Blocks > st.Bfree {
|
||||||
|
used += (st.Blocks - st.Bfree) * bs
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return total, used
|
||||||
|
}
|
||||||
@@ -1,3 +1,9 @@
|
|||||||
|
//go:build linux
|
||||||
|
|
||||||
|
// This file tests the Linux sampler's /proc and /sys parsers
|
||||||
|
// (probe_linux.go); the shared machinery has its own coverage and the
|
||||||
|
// Windows sampler's pure helpers live in probe_windows_test.go.
|
||||||
|
|
||||||
package agent
|
package agent
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
|||||||
@@ -0,0 +1,530 @@
|
|||||||
|
//go:build windows
|
||||||
|
|
||||||
|
package agent
|
||||||
|
|
||||||
|
// probe_windows.go is the Windows sampler for the probe collector,
|
||||||
|
// feeding the same Facts/Metrics contract and accounting discipline
|
||||||
|
// as the Linux sampler (see probe.go) from the Win32 surface:
|
||||||
|
//
|
||||||
|
// - CPU%: GetSystemTimes differenced through the shared cpuPercent
|
||||||
|
// (kernel time includes idle, exactly like /proc/stat jiffies);
|
||||||
|
// - memory: GlobalMemoryStatusEx, used = TotalPhys - AvailPhys
|
||||||
|
// (the free(1) "available" column's spirit); swap is the pagefile
|
||||||
|
// pair (TotalPageFile/AvailPageFile);
|
||||||
|
// - disk: fixed drives only (GetDriveType == DRIVE_FIXED), summed
|
||||||
|
// like df: used = total - free bytes (GetDiskFreeSpaceEx's
|
||||||
|
// TotalFree, NOT the caller-available figure, which discounts
|
||||||
|
// quotas the way f_bavail discounts the ext reserve);
|
||||||
|
// - network: GetIfTable2 rows filtered to OperStatus Up, ifType
|
||||||
|
// Ethernet (6) or IEEE80211 (71), non-loopback. This approximates
|
||||||
|
// the Linux three-layer filter: Windows exposes no stacking
|
||||||
|
// topology in the MIB, but tunnel/virtual adapters report their
|
||||||
|
// own ifTypes (tunnel 131, loopback 24, ...), so an allowlist of
|
||||||
|
// physical media lands on the same counted set in practice. The
|
||||||
|
// cost of the difference: a physical-medium bond/virtual switch
|
||||||
|
// vNIC presenting as Ethernet IS counted here even where Linux
|
||||||
|
// layer 3 would have skipped it;
|
||||||
|
// - boot identity: the registry MachineGuid stands in for the
|
||||||
|
// kernel boot_id; the counted-set FNV fingerprint is shared, so a
|
||||||
|
// NIC change re-baselines exactly like on Linux;
|
||||||
|
// - uptime: GetTickCount64 (via x/sys' DurationSinceBoot wrapper);
|
||||||
|
// - load averages, TCP/UDP connection counts have no exported
|
||||||
|
// x/sys/windows (v0.47.0) source — GetSystemTimes-style raw
|
||||||
|
// counters exist for CPU/memory, but the package exports no
|
||||||
|
// GetTcpTable2/GetUdpTable2 wrapper, so per the independent-
|
||||||
|
// degradation discipline Load stays zero and TCP/UDP stay 0
|
||||||
|
// until a wrapper exists;
|
||||||
|
// - every call fails soft to zeros, like an unreadable /proc file.
|
||||||
|
// Optional procedures are resolved through source gates
|
||||||
|
// (probe_guard.go): LazyProc.Call must-finds and PANICS on a
|
||||||
|
// missing export, and trimmed images do ship DLLs without the
|
||||||
|
// full surface — the Tianyi cloud PC's iphlpapi.dll has no
|
||||||
|
// FreeMibIfTable2, which killed the V2.2.0 agent on its first
|
||||||
|
// collection (V2.2.1 ①). A missing procedure now disables just
|
||||||
|
// its source, and collectMetrics additionally guards each source
|
||||||
|
// so a panic anywhere degrades that source alone.
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net"
|
||||||
|
"os"
|
||||||
|
"runtime"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
"unsafe"
|
||||||
|
|
||||||
|
"github.com/mhsanaei/3x-ui/v3/internal/config"
|
||||||
|
|
||||||
|
"golang.org/x/sys/windows"
|
||||||
|
"golang.org/x/sys/windows/registry"
|
||||||
|
)
|
||||||
|
|
||||||
|
// kernel32 procs x/sys/windows v0.47.0 does not wrap, plus the
|
||||||
|
// iphlpapi pair this file calls directly (gated below, so x/sys'
|
||||||
|
// own GetIfTable2Ex wrapper — whose Addr() panics the same way on a
|
||||||
|
// missing export — is deliberately not used).
|
||||||
|
var (
|
||||||
|
kernel32DLL = windows.NewLazySystemDLL("kernel32.dll")
|
||||||
|
procGetSystemTimes = kernel32DLL.NewProc("GetSystemTimes")
|
||||||
|
procGlobalMemoryStatus = kernel32DLL.NewProc("GlobalMemoryStatusEx")
|
||||||
|
iphlpapiDLL = windows.NewLazySystemDLL("iphlpapi.dll")
|
||||||
|
procGetIfTable2Ex = iphlpapiDLL.NewProc("GetIfTable2Ex")
|
||||||
|
procFreeMibIfTable2 = iphlpapiDLL.NewProc("FreeMibIfTable2")
|
||||||
|
)
|
||||||
|
|
||||||
|
// Availability gates for the optional procedures above. A LazyProc
|
||||||
|
// must never be Call()ed before its gate has proven the export
|
||||||
|
// exists: Call must-finds, so a missing procedure is a process-killing
|
||||||
|
// panic, not an error.
|
||||||
|
var (
|
||||||
|
gateSystemTimes sourceGate
|
||||||
|
gateMemoryStatus sourceGate
|
||||||
|
gateIfTable sourceGate
|
||||||
|
)
|
||||||
|
|
||||||
|
// procProbe adapts a LazyProc to a source-gate probe: Find resolves
|
||||||
|
// the export WITHOUT the mustFind panic baked into Call/Addr.
|
||||||
|
func procProbe(p *windows.LazyProc) func() error {
|
||||||
|
return func() error { return p.Find() }
|
||||||
|
}
|
||||||
|
|
||||||
|
func sampleFacts() *ProbeFacts {
|
||||||
|
memTotal, _, swapTotal, _ := winMemory()
|
||||||
|
diskTotal, _ := winDiskUsage()
|
||||||
|
v4, v6 := winAddresses()
|
||||||
|
return &ProbeFacts{
|
||||||
|
Hostname: winHostname(),
|
||||||
|
OS: winOSName(),
|
||||||
|
Kernel: winKernelBuild(),
|
||||||
|
Arch: runtime.GOARCH,
|
||||||
|
Virt: winVirt(),
|
||||||
|
CPUName: winCPUName(),
|
||||||
|
CPUCores: uint32(runtime.NumCPU()),
|
||||||
|
MemTotal: memTotal,
|
||||||
|
SwapTotal: swapTotal,
|
||||||
|
DiskTotal: diskTotal,
|
||||||
|
AgentVersion: config.GetAgentDisplayVersion(),
|
||||||
|
IPv4: v4,
|
||||||
|
IPv6: v6,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// memSnapshot / diskSnapshot pack the multi-value samplers so each
|
||||||
|
// can run as one guarded source.
|
||||||
|
type memSnapshot struct{ total, used, swapTotal, swapUsed uint64 }
|
||||||
|
type diskSnapshot struct{ total, used uint64 }
|
||||||
|
|
||||||
|
func collectMetrics() *ProbeMetrics {
|
||||||
|
probeColl.mu.Lock()
|
||||||
|
defer probeColl.mu.Unlock()
|
||||||
|
// Every source runs guarded: a panic in one (a Win32 surface the
|
||||||
|
// gates did not foresee) zeroes that source alone instead of
|
||||||
|
// killing the agent on its first collection.
|
||||||
|
mem := guardedCall("memory", func() memSnapshot {
|
||||||
|
total, used, swapTotal, swapUsed := winMemory()
|
||||||
|
return memSnapshot{total, used, swapTotal, swapUsed}
|
||||||
|
})
|
||||||
|
disk := guardedCall("disk", func() diskSnapshot {
|
||||||
|
total, used := winDiskUsage()
|
||||||
|
return diskSnapshot{total, used}
|
||||||
|
})
|
||||||
|
rows := guardedCall("interfaces", winIfaceRows)
|
||||||
|
all := make([]netCounters, 0, len(rows))
|
||||||
|
counted := make([]netCounters, 0, len(rows))
|
||||||
|
for _, r := range rows {
|
||||||
|
c := netCounters{name: r.name, rx: r.rx, tx: r.tx}
|
||||||
|
all = append(all, c)
|
||||||
|
if winIfaceCounts(r) {
|
||||||
|
counted = append(counted, c)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if pin := statIfacePinValue(); pin != "" {
|
||||||
|
counted = pinnedInterfaces(all, pin)
|
||||||
|
}
|
||||||
|
var rxTotal, txTotal uint64
|
||||||
|
names := make([]string, 0, len(counted))
|
||||||
|
for _, c := range counted {
|
||||||
|
rxTotal += c.rx
|
||||||
|
txTotal += c.tx
|
||||||
|
names = append(names, c.name)
|
||||||
|
}
|
||||||
|
rxRate, txRate := probeColl.netRate(counted, time.Now())
|
||||||
|
return &ProbeMetrics{
|
||||||
|
BootID: bootEpoch(winMachineGUID(), names),
|
||||||
|
Uptime: guardedCall("uptime", func() uint64 {
|
||||||
|
return uint64(windows.DurationSinceBoot().Seconds())
|
||||||
|
}),
|
||||||
|
CPU: probeColl.cpuPercent(guardedCall("cpu", winCPUSample)),
|
||||||
|
Load: [3]float32{},
|
||||||
|
MemTotal: mem.total,
|
||||||
|
MemUsed: mem.used,
|
||||||
|
SwapTotal: mem.swapTotal,
|
||||||
|
SwapUsed: mem.swapUsed,
|
||||||
|
DiskTotal: disk.total,
|
||||||
|
DiskUsed: disk.used,
|
||||||
|
NetRxTotal: rxTotal,
|
||||||
|
NetTxTotal: txTotal,
|
||||||
|
NetRx: rxRate,
|
||||||
|
NetTx: txRate,
|
||||||
|
TCP: 0, // no exported x/sys wrapper (see header)
|
||||||
|
UDP: 0, // no exported x/sys wrapper (see header)
|
||||||
|
Procs: guardedCall("processes", winProcCount),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// --- registry helpers --------------------------------------------------------
|
||||||
|
|
||||||
|
func regGetString(root registry.Key, path, name string) string {
|
||||||
|
k, err := registry.OpenKey(root, path, registry.QUERY_VALUE)
|
||||||
|
if err != nil {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
defer k.Close()
|
||||||
|
v, _, err := k.GetStringValue(name)
|
||||||
|
if err != nil {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
return strings.TrimSpace(v)
|
||||||
|
}
|
||||||
|
|
||||||
|
func winMachineGUID() string {
|
||||||
|
return regGetString(registry.LOCAL_MACHINE, `SOFTWARE\Microsoft\Cryptography`, "MachineGuid")
|
||||||
|
}
|
||||||
|
|
||||||
|
func winOSName() string {
|
||||||
|
const key = `SOFTWARE\Microsoft\Windows NT\CurrentVersion`
|
||||||
|
name := regGetString(registry.LOCAL_MACHINE, key, "ProductName")
|
||||||
|
if name == "" {
|
||||||
|
return "Windows"
|
||||||
|
}
|
||||||
|
// DisplayVersion ("23H2") disambiguates: ProductName still says
|
||||||
|
// "Windows 10" on most Windows 11 installs.
|
||||||
|
if dv := regGetString(registry.LOCAL_MACHINE, key, "DisplayVersion"); dv != "" {
|
||||||
|
name += " " + dv
|
||||||
|
}
|
||||||
|
return name
|
||||||
|
}
|
||||||
|
|
||||||
|
func winKernelBuild() string {
|
||||||
|
const key = `SOFTWARE\Microsoft\Windows NT\CurrentVersion`
|
||||||
|
if b := regGetString(registry.LOCAL_MACHINE, key, "CurrentBuild"); b != "" {
|
||||||
|
return b
|
||||||
|
}
|
||||||
|
return "unknown"
|
||||||
|
}
|
||||||
|
|
||||||
|
func winCPUName() string {
|
||||||
|
if n := regGetString(registry.LOCAL_MACHINE,
|
||||||
|
`HARDWARE\DESCRIPTION\System\CentralProcessor\0`, "ProcessorNameString"); n != "" {
|
||||||
|
return n
|
||||||
|
}
|
||||||
|
return "unknown"
|
||||||
|
}
|
||||||
|
|
||||||
|
// winVirt is best-effort: the BIOS strings Windows itself surfaces.
|
||||||
|
// No match reads "unknown" (Windows guests of a hypervisor that hides
|
||||||
|
// them are indistinguishable from bare metal this way).
|
||||||
|
func winVirt() string {
|
||||||
|
const bios = `HARDWARE\DESCRIPTION\System\BIOS`
|
||||||
|
hay := strings.ToLower(regGetString(registry.LOCAL_MACHINE, bios, "SystemManufacturer") +
|
||||||
|
" " + regGetString(registry.LOCAL_MACHINE, bios, "SystemProductName"))
|
||||||
|
for _, m := range []struct{ word, name string }{
|
||||||
|
{"vmware", "vmware"}, {"virtualbox", "virtualbox"}, {"qemu", "kvm"},
|
||||||
|
{"kvm", "kvm"}, {"xen", "xen"}, {"virtual machine", "hyper-v"},
|
||||||
|
{"amazon ec2", "kvm"}, {"google", "kvm"}, {"bochs", "bochs"},
|
||||||
|
} {
|
||||||
|
if strings.Contains(hay, m.word) {
|
||||||
|
return m.name
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return "unknown"
|
||||||
|
}
|
||||||
|
|
||||||
|
func winHostname() string {
|
||||||
|
if h, err := os.Hostname(); err == nil && h != "" {
|
||||||
|
return h
|
||||||
|
}
|
||||||
|
return "unknown"
|
||||||
|
}
|
||||||
|
|
||||||
|
// --- cpu / memory --------------------------------------------------------------
|
||||||
|
|
||||||
|
// filetime100ns flattens a FILETIME (100ns ticks) for the shared differ.
|
||||||
|
func filetime100ns(ft windows.Filetime) uint64 {
|
||||||
|
return uint64(ft.HighDateTime)<<32 | uint64(ft.LowDateTime)
|
||||||
|
}
|
||||||
|
|
||||||
|
// winCPUSample reads the cumulative system times as one cpuSample:
|
||||||
|
// total = kernel + user (kernel already includes idle, like jiffies).
|
||||||
|
func winCPUSample() *cpuSample {
|
||||||
|
if !gateSystemTimes.enabled("GetSystemTimes", procProbe(procGetSystemTimes)) {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
var idle, kernel, user windows.Filetime
|
||||||
|
r, _, _ := procGetSystemTimes.Call(
|
||||||
|
uintptr(unsafe.Pointer(&idle)),
|
||||||
|
uintptr(unsafe.Pointer(&kernel)),
|
||||||
|
uintptr(unsafe.Pointer(&user)))
|
||||||
|
if r == 0 {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
return &cpuSample{
|
||||||
|
total: filetime100ns(kernel) + filetime100ns(user),
|
||||||
|
idle: filetime100ns(idle),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
type memoryStatusEx struct {
|
||||||
|
Length uint32
|
||||||
|
MemoryLoad uint32
|
||||||
|
TotalPhys uint64
|
||||||
|
AvailPhys uint64
|
||||||
|
TotalPageFile uint64
|
||||||
|
AvailPageFile uint64
|
||||||
|
TotalVirtual uint64
|
||||||
|
AvailVirtual uint64
|
||||||
|
AvailExtendedVirtual uint64
|
||||||
|
}
|
||||||
|
|
||||||
|
// usedFromTotalAvail is the used-column discipline as a pure
|
||||||
|
// function: used = total - avail, saturating at 0.
|
||||||
|
func usedFromTotalAvail(total, avail uint64) uint64 {
|
||||||
|
if avail > total {
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
return total - avail
|
||||||
|
}
|
||||||
|
|
||||||
|
func winMemory() (total, used, swapTotal, swapUsed uint64) {
|
||||||
|
if !gateMemoryStatus.enabled("GlobalMemoryStatusEx", procProbe(procGlobalMemoryStatus)) {
|
||||||
|
return 0, 0, 0, 0
|
||||||
|
}
|
||||||
|
var ms memoryStatusEx
|
||||||
|
ms.Length = uint32(unsafe.Sizeof(ms))
|
||||||
|
r, _, _ := procGlobalMemoryStatus.Call(uintptr(unsafe.Pointer(&ms)))
|
||||||
|
if r == 0 {
|
||||||
|
return 0, 0, 0, 0
|
||||||
|
}
|
||||||
|
return ms.TotalPhys, usedFromTotalAvail(ms.TotalPhys, ms.AvailPhys),
|
||||||
|
ms.TotalPageFile, usedFromTotalAvail(ms.TotalPageFile, ms.AvailPageFile)
|
||||||
|
}
|
||||||
|
|
||||||
|
// --- disk (df口径) -----------------------------------------------------------
|
||||||
|
|
||||||
|
// driveSpace is one fixed drive's GetDiskFreeSpaceEx result.
|
||||||
|
type driveSpace struct{ total, free uint64 }
|
||||||
|
|
||||||
|
// sumDriveSpace aggregates fixed drives the df way: used = total -
|
||||||
|
// free (the caller-available figure is deliberately not consulted).
|
||||||
|
func sumDriveSpace(ds []driveSpace) (total, used uint64) {
|
||||||
|
for _, d := range ds {
|
||||||
|
total += d.total
|
||||||
|
used += usedFromTotalAvail(d.total, d.free)
|
||||||
|
}
|
||||||
|
return total, used
|
||||||
|
}
|
||||||
|
|
||||||
|
// splitDriveStrings decodes the GetLogicalDriveStrings multi-string
|
||||||
|
// buffer (NUL-separated roots, double-NUL terminated). Pure, so the
|
||||||
|
// layout is unit-testable off-device.
|
||||||
|
func splitDriveStrings(buf []uint16) []string {
|
||||||
|
var out []string
|
||||||
|
start := 0
|
||||||
|
for i, c := range buf {
|
||||||
|
if c != 0 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if i == start {
|
||||||
|
break // double NUL: end of the list
|
||||||
|
}
|
||||||
|
out = append(out, windows.UTF16ToString(buf[start:i]))
|
||||||
|
start = i + 1
|
||||||
|
}
|
||||||
|
return out
|
||||||
|
}
|
||||||
|
|
||||||
|
// fixedDriveRoots lists roots ("C:\") of DRIVE_FIXED drives.
|
||||||
|
func fixedDriveRoots() []string {
|
||||||
|
n, err := windows.GetLogicalDriveStrings(0, nil)
|
||||||
|
if err != nil || n == 0 {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
buf := make([]uint16, n)
|
||||||
|
n, err = windows.GetLogicalDriveStrings(n, &buf[0])
|
||||||
|
if err != nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
var roots []string
|
||||||
|
for _, s := range splitDriveStrings(buf[:n]) {
|
||||||
|
p, err := windows.UTF16PtrFromString(s)
|
||||||
|
if err != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if windows.GetDriveType(p) == windows.DRIVE_FIXED {
|
||||||
|
roots = append(roots, s)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return roots
|
||||||
|
}
|
||||||
|
|
||||||
|
func winDiskUsage() (total, used uint64) {
|
||||||
|
var spaces []driveSpace
|
||||||
|
for _, root := range fixedDriveRoots() {
|
||||||
|
p, err := windows.UTF16PtrFromString(root)
|
||||||
|
if err != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
var freeToCaller, totalBytes, totalFree uint64
|
||||||
|
if err := windows.GetDiskFreeSpaceEx(p, &freeToCaller, &totalBytes, &totalFree); err != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
spaces = append(spaces, driveSpace{total: totalBytes, free: totalFree})
|
||||||
|
}
|
||||||
|
return sumDriveSpace(spaces)
|
||||||
|
}
|
||||||
|
|
||||||
|
// --- network: counted set ------------------------------------------------------
|
||||||
|
|
||||||
|
// winIfaceRow is the filter-relevant slice of one MIB_IF_ROW2.
|
||||||
|
type winIfaceRow struct {
|
||||||
|
name string
|
||||||
|
ifType uint32
|
||||||
|
operUp bool
|
||||||
|
rx, tx uint64
|
||||||
|
}
|
||||||
|
|
||||||
|
// winIfaceCounts applies the counted-set rule to one row (see the
|
||||||
|
// file header for how it approximates the Linux three-layer filter).
|
||||||
|
func winIfaceCounts(r winIfaceRow) bool {
|
||||||
|
if !r.operUp {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
if r.ifType == windows.IF_TYPE_SOFTWARE_LOOPBACK {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
return r.ifType == windows.IF_TYPE_ETHERNET_CSMACD ||
|
||||||
|
r.ifType == windows.IF_TYPE_IEEE80211
|
||||||
|
}
|
||||||
|
|
||||||
|
// winIfaceRows reads the full MIB_IF_TABLE2 once; collectMetrics
|
||||||
|
// derives both the all-interfaces list (for the operator pin) and the
|
||||||
|
// counted set from it.
|
||||||
|
func winIfaceRows() []winIfaceRow {
|
||||||
|
// The get/free pair is gated as ONE source: calling GetIfTable2Ex
|
||||||
|
// without FreeMibIfTable2 would leak the system-allocated table
|
||||||
|
// on every poll, and either export missing must degrade the
|
||||||
|
// source to zero rows instead of panicking — the trimmed Tianyi
|
||||||
|
// cloud-PC image ships an iphlpapi.dll without FreeMibIfTable2,
|
||||||
|
// and the V2.2.0 sampler's unguarded LazyProc.Call died on it.
|
||||||
|
if !gateIfTable.enabled("GetIfTable2Ex/FreeMibIfTable2 (iphlpapi)", func() error {
|
||||||
|
if err := procGetIfTable2Ex.Find(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return procFreeMibIfTable2.Find()
|
||||||
|
}) {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
var table *windows.MibIfTable2
|
||||||
|
r, _, _ := procGetIfTable2Ex.Call(
|
||||||
|
uintptr(windows.MibIfTableNormal),
|
||||||
|
uintptr(unsafe.Pointer(&table)))
|
||||||
|
if r != 0 || table == nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
defer procFreeMibIfTable2.Call(uintptr(unsafe.Pointer(table)))
|
||||||
|
n := int(table.NumEntries)
|
||||||
|
if n == 0 {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
rows := unsafe.Slice(&table.Table[0], n)
|
||||||
|
out := make([]winIfaceRow, 0, n)
|
||||||
|
for i := range rows {
|
||||||
|
r := &rows[i]
|
||||||
|
out = append(out, winIfaceRow{
|
||||||
|
name: windows.UTF16ToString(r.Alias[:]),
|
||||||
|
ifType: r.Type,
|
||||||
|
operUp: r.OperStatus == windows.IfOperStatusUp,
|
||||||
|
rx: r.InOctets,
|
||||||
|
tx: r.OutOctets,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
return out
|
||||||
|
}
|
||||||
|
|
||||||
|
// --- processes -----------------------------------------------------------------
|
||||||
|
|
||||||
|
func winProcCount() uint32 {
|
||||||
|
snap, err := windows.CreateToolhelp32Snapshot(windows.TH32CS_SNAPPROCESS, 0)
|
||||||
|
if err != nil {
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
defer windows.CloseHandle(snap)
|
||||||
|
var pe windows.ProcessEntry32
|
||||||
|
pe.Size = uint32(unsafe.Sizeof(pe))
|
||||||
|
if err := windows.Process32First(snap, &pe); err != nil {
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
var n uint32 = 1
|
||||||
|
for {
|
||||||
|
if err := windows.Process32Next(snap, &pe); err != nil {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
n++
|
||||||
|
}
|
||||||
|
return n
|
||||||
|
}
|
||||||
|
|
||||||
|
// --- addresses -------------------------------------------------------------------
|
||||||
|
|
||||||
|
// winAddresses picks one address per family from up, non-loopback
|
||||||
|
// interfaces: public preferred over private, like the Linux picker —
|
||||||
|
// only without its /proc inputs (no operstate file, no temporary-
|
||||||
|
// address flags; net.Interfaces' own flags do the loopback cut).
|
||||||
|
func winAddresses() (string, string) {
|
||||||
|
ifaces, err := net.Interfaces()
|
||||||
|
if err != nil {
|
||||||
|
return "", ""
|
||||||
|
}
|
||||||
|
var cands []net.IP
|
||||||
|
for _, ifc := range ifaces {
|
||||||
|
if ifc.Flags&net.FlagUp == 0 || ifc.Flags&net.FlagLoopback != 0 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
addrs, err := ifc.Addrs()
|
||||||
|
if err != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
for _, addr := range addrs {
|
||||||
|
var ip net.IP
|
||||||
|
switch v := addr.(type) {
|
||||||
|
case *net.IPNet:
|
||||||
|
ip = v.IP
|
||||||
|
case *net.IPAddr:
|
||||||
|
ip = v.IP
|
||||||
|
}
|
||||||
|
if ip == nil || ip.IsLinkLocalUnicast() || ip.IsLinkLocalMulticast() || ip.IsLoopback() {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
cands = append(cands, ip)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
pick := func(v6 bool) string {
|
||||||
|
var best net.IP
|
||||||
|
for _, ip := range cands {
|
||||||
|
if (ip.To4() == nil) != v6 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if best == nil {
|
||||||
|
best = ip
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if isPublicIP(ip) && !isPublicIP(best) {
|
||||||
|
best = ip
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if best == nil {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
return best.String()
|
||||||
|
}
|
||||||
|
return pick(false), pick(true)
|
||||||
|
}
|
||||||
@@ -0,0 +1,99 @@
|
|||||||
|
//go:build windows
|
||||||
|
|
||||||
|
package agent
|
||||||
|
|
||||||
|
import (
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"golang.org/x/sys/windows"
|
||||||
|
)
|
||||||
|
|
||||||
|
// The Windows sampler's pure helpers: the Win32 calls themselves need
|
||||||
|
// a real machine, but the accounting math layered on their results is
|
||||||
|
// pinned here.
|
||||||
|
|
||||||
|
func TestFiletime100nsAndCPUSampleDiff(t *testing.T) {
|
||||||
|
lo := windows.Filetime{LowDateTime: 0x00000005, HighDateTime: 0x00000002}
|
||||||
|
if got := filetime100ns(lo); got != 0x200000005 {
|
||||||
|
t.Fatalf("filetime100ns = %#x, want 0x200000005", got)
|
||||||
|
}
|
||||||
|
// The shared differ over Windows-style counters (kernel includes
|
||||||
|
// idle): idle advanced by the full window -> 0% busy; then a
|
||||||
|
// window where only a quarter of the time is idle -> 75%.
|
||||||
|
c := &probeCollector{}
|
||||||
|
first := &cpuSample{total: 1000, idle: 400}
|
||||||
|
if got := c.cpuPercent(first); got != 0 {
|
||||||
|
t.Fatalf("first sample = %v, want 0 (baseline only)", got)
|
||||||
|
}
|
||||||
|
idleAll := &cpuSample{total: 2000, idle: 1400}
|
||||||
|
if got := c.cpuPercent(idleAll); got != 0 {
|
||||||
|
t.Fatalf("all-idle window = %v, want 0", got)
|
||||||
|
}
|
||||||
|
busy := &cpuSample{total: 3000, idle: 1650}
|
||||||
|
if got := c.cpuPercent(busy); got != 75 {
|
||||||
|
t.Fatalf("busy window = %v, want 75", got)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestUsedFromTotalAvail(t *testing.T) {
|
||||||
|
if got := usedFromTotalAvail(100, 30); got != 70 {
|
||||||
|
t.Fatalf("used = %d, want 70", got)
|
||||||
|
}
|
||||||
|
if got := usedFromTotalAvail(100, 100); got != 0 {
|
||||||
|
t.Fatalf("used = %d, want 0", got)
|
||||||
|
}
|
||||||
|
if got := usedFromTotalAvail(100, 130); got != 0 {
|
||||||
|
t.Fatalf("used saturates, got %d", got)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSumDriveSpaceDfSemantics(t *testing.T) {
|
||||||
|
total, used := sumDriveSpace([]driveSpace{
|
||||||
|
{total: 1000, free: 250},
|
||||||
|
{total: 2000, free: 2000},
|
||||||
|
{total: 500, free: 900}, // inconsistent source clamps to full
|
||||||
|
})
|
||||||
|
if total != 3500 || used != 1250 {
|
||||||
|
t.Fatalf("sumDriveSpace = %d/%d, want 3500/1250", total, used)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSplitDriveStrings(t *testing.T) {
|
||||||
|
mk := func(ss ...string) []uint16 {
|
||||||
|
var out []uint16
|
||||||
|
for _, s := range ss {
|
||||||
|
for _, r := range s {
|
||||||
|
out = append(out, uint16(r))
|
||||||
|
}
|
||||||
|
out = append(out, 0)
|
||||||
|
}
|
||||||
|
return append(out, 0)
|
||||||
|
}
|
||||||
|
got := splitDriveStrings(mk(`C:\`, `D:\`))
|
||||||
|
if len(got) != 2 || got[0] != `C:\` || got[1] != `D:\` {
|
||||||
|
t.Fatalf("splitDriveStrings = %q, want [C:\\ D:\\]", got)
|
||||||
|
}
|
||||||
|
if got := splitDriveStrings(mk()); len(got) != 0 {
|
||||||
|
t.Fatalf("empty buffer = %q, want none", got)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestWinIfaceCountsRule(t *testing.T) {
|
||||||
|
cases := []struct {
|
||||||
|
name string
|
||||||
|
row winIfaceRow
|
||||||
|
want bool
|
||||||
|
}{
|
||||||
|
{"up ethernet", winIfaceRow{ifType: windows.IF_TYPE_ETHERNET_CSMACD, operUp: true}, true},
|
||||||
|
{"up wifi", winIfaceRow{ifType: windows.IF_TYPE_IEEE80211, operUp: true}, true},
|
||||||
|
{"down ethernet", winIfaceRow{ifType: windows.IF_TYPE_ETHERNET_CSMACD, operUp: false}, false},
|
||||||
|
{"loopback", winIfaceRow{ifType: windows.IF_TYPE_SOFTWARE_LOOPBACK, operUp: true}, false},
|
||||||
|
{"tunnel", winIfaceRow{ifType: 131, operUp: true}, false},
|
||||||
|
{"ppp", winIfaceRow{ifType: 23, operUp: true}, false},
|
||||||
|
}
|
||||||
|
for _, tc := range cases {
|
||||||
|
if got := winIfaceCounts(tc.row); got != tc.want {
|
||||||
|
t.Errorf("%s: winIfaceCounts = %v, want %v", tc.name, got, tc.want)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -52,6 +52,15 @@ var execSelf = func(path string) error {
|
|||||||
return syscall.Exec(path, append([]string{path}, os.Args[1:]...), os.Environ())
|
return syscall.Exec(path, append([]string{path}, os.Args[1:]...), os.Environ())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// selfUpdateSupported reports whether this platform can consume a
|
||||||
|
// master update intent. Windows agents are the standalone
|
||||||
|
// xui-agent-windows binary (see cmd/xui-agent-windows), serviced by
|
||||||
|
// re-running install-agent.ps1: the release assets an intent points
|
||||||
|
// at are the panel's Linux binaries, so an intent delivered to a
|
||||||
|
// Windows node is acknowledged and skipped, never downloaded.
|
||||||
|
// A package variable so tests can exercise the skip on any OS.
|
||||||
|
var selfUpdateSupported = runtime.GOOS != "windows"
|
||||||
|
|
||||||
func (a *Agent) setUpdateReport(st agentUpdateStatus) {
|
func (a *Agent) setUpdateReport(st agentUpdateStatus) {
|
||||||
a.updateMu.Lock()
|
a.updateMu.Lock()
|
||||||
cp := st
|
cp := st
|
||||||
@@ -82,6 +91,19 @@ func (a *Agent) maybeHandleUpdate(intent *updateIntent) {
|
|||||||
a.updateMu.Unlock()
|
a.updateMu.Unlock()
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
if !selfUpdateSupported {
|
||||||
|
// Windows (see selfUpdateSupported): log once per intent and
|
||||||
|
// mark it handled so the poll does not re-log it every cycle.
|
||||||
|
// No report is filed — the master-side intent simply ages out,
|
||||||
|
// and the node updates by re-running install-agent.ps1.
|
||||||
|
a.updateMu.Lock()
|
||||||
|
if intent.RequestedAt != a.updateHandled {
|
||||||
|
a.updateHandled = intent.RequestedAt
|
||||||
|
logger.Infof("agent update to %s offered but skipped on %s: Windows agents update by re-running install-agent.ps1", intent.Target, runtime.GOOS)
|
||||||
|
}
|
||||||
|
a.updateMu.Unlock()
|
||||||
|
return
|
||||||
|
}
|
||||||
a.updateMu.Lock()
|
a.updateMu.Lock()
|
||||||
if a.updateRunning || intent.RequestedAt == a.updateHandled ||
|
if a.updateRunning || intent.RequestedAt == a.updateHandled ||
|
||||||
(intent.RequestedAt == a.updateFailedReq && time.Since(a.updateFailedAt) < updateRetryBackoff) {
|
(intent.RequestedAt == a.updateFailedReq && time.Since(a.updateFailedAt) < updateRetryBackoff) {
|
||||||
|
|||||||
@@ -159,6 +159,39 @@ func TestAgentUpdateExecFailureRollsBack(t *testing.T) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestAgentUpdateSkippedOnUnsupportedOS(t *testing.T) {
|
||||||
|
orig := selfUpdateSupported
|
||||||
|
selfUpdateSupported = false
|
||||||
|
t.Cleanup(func() { selfUpdateSupported = orig })
|
||||||
|
|
||||||
|
var hits atomic.Int32
|
||||||
|
srv := fakeReleaseServer(t, []byte("new-binary"), true, &hits)
|
||||||
|
defer srv.Close()
|
||||||
|
a, exe := newUpdateTestAgent(t, srv.URL)
|
||||||
|
|
||||||
|
intent := &updateIntent{RequestedAt: 46, Target: "A2.0.1", Tag: "V2.0.1", BaseURL: srv.URL + "/"}
|
||||||
|
a.maybeHandleUpdate(intent)
|
||||||
|
// A repeated delivery of the same intent is also a no-op.
|
||||||
|
a.maybeHandleUpdate(intent)
|
||||||
|
|
||||||
|
if hits.Load() != 0 {
|
||||||
|
t.Fatalf("release server hit %d times on an unsupported OS, want 0", hits.Load())
|
||||||
|
}
|
||||||
|
got, _ := os.ReadFile(exe)
|
||||||
|
if string(got) != "old-binary" {
|
||||||
|
t.Fatalf("exe = %q, want untouched old-binary", got)
|
||||||
|
}
|
||||||
|
a.updateMu.Lock()
|
||||||
|
handled, running := a.updateHandled, a.updateRunning
|
||||||
|
a.updateMu.Unlock()
|
||||||
|
if handled != 46 || running {
|
||||||
|
t.Fatalf("handled=%d running=%v, want handled=46 running=false", handled, running)
|
||||||
|
}
|
||||||
|
if rep := a.currentUpdateReport(); rep != nil {
|
||||||
|
t.Fatalf("report = %+v, want none for a skipped intent", rep)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
var errExecBoom = &execBoomError{}
|
var errExecBoom = &execBoomError{}
|
||||||
|
|
||||||
type execBoomError struct{}
|
type execBoomError struct{}
|
||||||
|
|||||||
@@ -125,16 +125,30 @@ func GetDisplayVersion() string {
|
|||||||
return defaultDisplayVersion
|
return defaultDisplayVersion
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// bareDisplayVersion is GetDisplayVersion without a single leading
|
||||||
|
// V/v. Release builds inject the number either bare ("2.1.0") or in
|
||||||
|
// the master's V-form ("V2.1.0"); the letter-prefixed getters below
|
||||||
|
// must compose from the bare form so the letter never doubles
|
||||||
|
// ("VV2.1.0" / "AV2.1.0" — the latter is what agent rows showed when
|
||||||
|
// the V-form was injected).
|
||||||
|
func bareDisplayVersion() string {
|
||||||
|
v := GetDisplayVersion()
|
||||||
|
if len(v) > 1 && (v[0] == 'V' || v[0] == 'v') {
|
||||||
|
return v[1:]
|
||||||
|
}
|
||||||
|
return v
|
||||||
|
}
|
||||||
|
|
||||||
// GetMasterDisplayVersion is the master's display version ("V1.0.0"),
|
// GetMasterDisplayVersion is the master's display version ("V1.0.0"),
|
||||||
// shown in the panel's version slot.
|
// shown in the panel's version slot.
|
||||||
func GetMasterDisplayVersion() string {
|
func GetMasterDisplayVersion() string {
|
||||||
return "V" + GetDisplayVersion()
|
return "V" + bareDisplayVersion()
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetAgentDisplayVersion is the agent's display version ("A1.0.0"),
|
// GetAgentDisplayVersion is the agent's display version ("A1.0.0"),
|
||||||
// reported in agent telemetry and shown on agent node rows.
|
// reported in agent telemetry and shown on agent node rows.
|
||||||
func GetAgentDisplayVersion() string {
|
func GetAgentDisplayVersion() string {
|
||||||
return "A" + GetDisplayVersion()
|
return "A" + bareDisplayVersion()
|
||||||
}
|
}
|
||||||
|
|
||||||
// ParseDisplayVersion parses a display version into its three numeric
|
// ParseDisplayVersion parses a display version into its three numeric
|
||||||
|
|||||||
@@ -75,6 +75,20 @@ func TestDisplayVersionDefaultsAndOverride(t *testing.T) {
|
|||||||
if got := GetAgentDisplayVersion(); got != "A1.0.1" {
|
if got := GetAgentDisplayVersion(); got != "A1.0.1" {
|
||||||
t.Fatalf("injected agent display version = %q, want A1.0.1", got)
|
t.Fatalf("injected agent display version = %q, want A1.0.1", got)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// A release build may inject the number in the master's V-form:
|
||||||
|
// neither getter may double its letter ("VV1.0.2" / "AV1.0.2").
|
||||||
|
displayVersion = "V1.0.2"
|
||||||
|
if got := GetMasterDisplayVersion(); got != "V1.0.2" {
|
||||||
|
t.Fatalf("V-injected master display version = %q, want V1.0.2", got)
|
||||||
|
}
|
||||||
|
if got := GetAgentDisplayVersion(); got != "A1.0.2" {
|
||||||
|
t.Fatalf("V-injected agent display version = %q, want A1.0.2", got)
|
||||||
|
}
|
||||||
|
displayVersion = "v1.0.3"
|
||||||
|
if got := GetAgentDisplayVersion(); got != "A1.0.3" {
|
||||||
|
t.Fatalf("v-injected agent display version = %q, want A1.0.3", got)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestParseDisplayVersion(t *testing.T) {
|
func TestParseDisplayVersion(t *testing.T) {
|
||||||
|
|||||||
+35
-3
@@ -957,7 +957,7 @@ func (s *SubService) genVlessLink(inbound *model.Inbound, email string) string {
|
|||||||
case "tls":
|
case "tls":
|
||||||
applyShareTLSParams(stream, params)
|
applyShareTLSParams(stream, params)
|
||||||
case "reality":
|
case "reality":
|
||||||
applyShareRealityParams(stream, params)
|
applyShareRealityParams(stream, singboxFragmentServerName(inbound.SingboxConfig), params)
|
||||||
default:
|
default:
|
||||||
params["security"] = "none"
|
params["security"] = "none"
|
||||||
}
|
}
|
||||||
@@ -1010,7 +1010,7 @@ func (s *SubService) genTrojanLink(inbound *model.Inbound, email string) string
|
|||||||
case "tls":
|
case "tls":
|
||||||
applyShareTLSParams(stream, params)
|
applyShareTLSParams(stream, params)
|
||||||
case "reality":
|
case "reality":
|
||||||
applyShareRealityParams(stream, params)
|
applyShareRealityParams(stream, singboxFragmentServerName(inbound.SingboxConfig), params)
|
||||||
if streamNetwork == "tcp" && len(client.Flow) > 0 && !inbound.DisableFlow {
|
if streamNetwork == "tcp" && len(client.Flow) > 0 && !inbound.DisableFlow {
|
||||||
params["flow"] = client.Flow
|
params["flow"] = client.Flow
|
||||||
}
|
}
|
||||||
@@ -1700,7 +1700,30 @@ func hysteriaPinHex(pin string) string {
|
|||||||
return pin
|
return pin
|
||||||
}
|
}
|
||||||
|
|
||||||
func applyShareRealityParams(stream map[string]any, params map[string]string) {
|
// singboxFragmentServerName returns the tls.server_name of the
|
||||||
|
// inbound's stored sing-box fragment — the ONE name the sing-box
|
||||||
|
// REALITY server actually serves. The converter puts exactly one
|
||||||
|
// name there (realitySettings.serverNames[0], see internal/singbox)
|
||||||
|
// while the legacy settings may list dozens. "" when the inbound has
|
||||||
|
// no fragment, the fragment is unparseable, or it carries no server
|
||||||
|
// name; callers then keep the historical behaviour.
|
||||||
|
func singboxFragmentServerName(singboxConfig string) string {
|
||||||
|
if strings.TrimSpace(singboxConfig) == "" {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
var fragment map[string]any
|
||||||
|
if err := json.Unmarshal([]byte(singboxConfig), &fragment); err != nil {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
if tlsBlock, ok := fragment["tls"].(map[string]any); ok {
|
||||||
|
if name, ok := tlsBlock["server_name"].(string); ok {
|
||||||
|
return strings.TrimSpace(name)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
func applyShareRealityParams(stream map[string]any, fragmentSNI string, params map[string]string) {
|
||||||
params["security"] = "reality"
|
params["security"] = "reality"
|
||||||
realitySetting, _ := stream["realitySettings"].(map[string]any)
|
realitySetting, _ := stream["realitySettings"].(map[string]any)
|
||||||
realitySettings, _ := searchKey(realitySetting, "settings")
|
realitySettings, _ := searchKey(realitySetting, "settings")
|
||||||
@@ -1710,6 +1733,15 @@ func applyShareRealityParams(stream map[string]any, params map[string]string) {
|
|||||||
params["sni"], _ = sNames[random.Num(len(sNames))].(string)
|
params["sni"], _ = sNames[random.Num(len(sNames))].(string)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
// The fragment's server_name wins over the pick above: xray
|
||||||
|
// serves the whole serverNames list, so a random entry was
|
||||||
|
// safe there, but the sing-box core serves only the
|
||||||
|
// fragment's single name — any other entry is refused as an
|
||||||
|
// invalid REALITY connection (the PPF-L3 outage: 29 listed
|
||||||
|
// names, core serving only the first).
|
||||||
|
if fragmentSNI != "" {
|
||||||
|
params["sni"] = fragmentSNI
|
||||||
|
}
|
||||||
if pbkValue, ok := searchKey(realitySettings, "publicKey"); ok {
|
if pbkValue, ok := searchKey(realitySettings, "publicKey"); ok {
|
||||||
params["pbk"], _ = pbkValue.(string)
|
params["pbk"], _ = pbkValue.(string)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -89,6 +89,60 @@ func TestGenVlessLink_RealityParamsMapped(t *testing.T) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Scope D: with several serverNames, the generated link's sni must be
|
||||||
|
// the sing-box fragment's tls.server_name — the only name the sing-box
|
||||||
|
// REALITY server serves (the converter stores serverNames[0] there).
|
||||||
|
// The historical behaviour picked a uniformly random list entry per
|
||||||
|
// generation, which xray tolerated (it serves the whole list) but
|
||||||
|
// sing-box refuses as an invalid connection. Without a fragment the
|
||||||
|
// sni still comes from the list.
|
||||||
|
func TestGenVlessLink_RealitySniPrefersSingboxFragment(t *testing.T) {
|
||||||
|
stream := `{
|
||||||
|
"network":"tcp","security":"reality",
|
||||||
|
"tcpSettings":{"header":{"type":"none"}},
|
||||||
|
"realitySettings":{
|
||||||
|
"serverNames":["amd.example.com","connect-sit.amd.example.com","www.amd.example.com"],
|
||||||
|
"shortIds":["ab12cd"],
|
||||||
|
"settings":{"publicKey":"PBKvalue","fingerprint":"firefox"}
|
||||||
|
}
|
||||||
|
}`
|
||||||
|
s := &SubService{}
|
||||||
|
inbound := shareLinkInbound(stream)
|
||||||
|
inbound.SingboxConfig = `{"type":"vless","tag":"in","listen":"::","listen_port":443,` +
|
||||||
|
`"tls":{"enabled":true,"server_name":"amd.example.com",` +
|
||||||
|
`"reality":{"enabled":true,"short_id":["ab12cd"]}}}`
|
||||||
|
|
||||||
|
// The losing branch is a random pick: sample enough generations
|
||||||
|
// that a regression to it cannot survive by luck (2/3 wrong per
|
||||||
|
// draw; 30 draws make a false pass ~2e-6).
|
||||||
|
for i := 0; i < 30; i++ {
|
||||||
|
link := s.genVlessLink(inbound, "user")
|
||||||
|
u, err := url.Parse(link)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("parse link %q: %v", link, err)
|
||||||
|
}
|
||||||
|
if got := u.Query().Get("sni"); got != "amd.example.com" {
|
||||||
|
t.Fatalf("draw %d: sni = %q, want the fragment's amd.example.com\nlink: %s", i, got, link)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// No fragment (or an unusable one): historical behaviour — the
|
||||||
|
// sni is some entry of the serverNames list.
|
||||||
|
for _, frag := range []string{"", "{not json", `{"type":"vless"}`} {
|
||||||
|
inbound.SingboxConfig = frag
|
||||||
|
link := s.genVlessLink(inbound, "user")
|
||||||
|
u, err := url.Parse(link)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("parse link %q: %v", link, err)
|
||||||
|
}
|
||||||
|
switch got := u.Query().Get("sni"); got {
|
||||||
|
case "amd.example.com", "connect-sit.amd.example.com", "www.amd.example.com":
|
||||||
|
default:
|
||||||
|
t.Fatalf("fragment %q: sni = %q, want a serverNames entry", frag, got)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// realityTwoClientInbound builds a reality VLESS inbound carrying two clients
|
// realityTwoClientInbound builds a reality VLESS inbound carrying two clients
|
||||||
// with distinct subIds.
|
// with distinct subIds.
|
||||||
func realityTwoClientInbound() *model.Inbound {
|
func realityTwoClientInbound() *model.Inbound {
|
||||||
|
|||||||
@@ -52,7 +52,10 @@ type probeMetaForm struct {
|
|||||||
IPv4Pin string `json:"ipv4Pin"`
|
IPv4Pin string `json:"ipv4Pin"`
|
||||||
IPv6Pin string `json:"ipv6Pin"`
|
IPv6Pin string `json:"ipv6Pin"`
|
||||||
// V2.0.1: traffic calibration (GB, either sign) and the counted
|
// V2.0.1: traffic calibration (GB, either sign) and the counted
|
||||||
// interface pin ("" = automatic).
|
// interface pin ("" = automatic). Since V2.2.1 ⑥b the calibration
|
||||||
|
// fields are ONE-SHOT deltas: a non-zero value is folded into the
|
||||||
|
// stored standing correction on save (see probe.MergeCalDelta);
|
||||||
|
// zero/absent applies nothing.
|
||||||
TrafficCalUpGB float64 `json:"trafficCalUpGb"`
|
TrafficCalUpGB float64 `json:"trafficCalUpGb"`
|
||||||
TrafficCalDownGB float64 `json:"trafficCalDownGb"`
|
TrafficCalDownGB float64 `json:"trafficCalDownGb"`
|
||||||
StatIface string `json:"statIface"`
|
StatIface string `json:"statIface"`
|
||||||
@@ -75,8 +78,11 @@ func metaToForm(m probe.Meta) probeMetaForm {
|
|||||||
CountryPin: m.CountryPin,
|
CountryPin: m.CountryPin,
|
||||||
IPv4Pin: m.IPv4Pin,
|
IPv4Pin: m.IPv4Pin,
|
||||||
IPv6Pin: m.IPv6Pin,
|
IPv6Pin: m.IPv6Pin,
|
||||||
TrafficCalUpGB: m.TrafficCalUpGB,
|
// One-shot delta inputs (V2.2.1 ⑥b): the form's calibration
|
||||||
TrafficCalDownGB: m.TrafficCalDownGB,
|
// boxes always start empty; the standing correction folded
|
||||||
|
// into the stored row is never echoed back into them.
|
||||||
|
TrafficCalUpGB: 0,
|
||||||
|
TrafficCalDownGB: 0,
|
||||||
StatIface: m.StatIface,
|
StatIface: m.StatIface,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -150,7 +156,16 @@ func (pc *ProbeMetaController) putMeta(c *gin.Context) {
|
|||||||
jsonMsg(c, msg, errRequest)
|
jsonMsg(c, msg, errRequest)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
if err := probe.SaveMeta(form.toMeta(int64(agent.Id))); err != nil {
|
// Calibration is one-shot (V2.2.1 ⑥b): fold the submitted deltas
|
||||||
|
// into the stored standing correction; all other fields replace.
|
||||||
|
meta := form.toMeta(int64(agent.Id))
|
||||||
|
if existing, found, err := probe.MetaFor(int64(agent.Id)); err != nil {
|
||||||
|
jsonMsg(c, "save probe meta failed", err)
|
||||||
|
return
|
||||||
|
} else if found {
|
||||||
|
meta = probe.MergeCalDelta(existing, meta)
|
||||||
|
}
|
||||||
|
if err := probe.SaveMeta(meta); err != nil {
|
||||||
jsonMsg(c, "save probe meta failed", err)
|
jsonMsg(c, "save probe meta failed", err)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -323,7 +323,9 @@ func TestProbePublicContract(t *testing.T) {
|
|||||||
assertKeys(t, keySet(t, nodeBody), anonNodeKeys)
|
assertKeys(t, keySet(t, nodeBody), anonNodeKeys)
|
||||||
var node map[string]any
|
var node map[string]any
|
||||||
_ = json.Unmarshal(nodeBody, &node)
|
_ = json.Unmarshal(nodeBody, &node)
|
||||||
if node["id"] != float64(pubID) || node["name"] != "PubNode" || node["online"] != true ||
|
// V2.2.1 ⑦: the public view's name is the public remark when one
|
||||||
|
// is set (the seed sets "shown publicly"), not the node name.
|
||||||
|
if node["id"] != float64(pubID) || node["name"] != "shown publicly" || node["online"] != true ||
|
||||||
node["month_rx"] != float64(33) || node["month_used"] != float64(77) ||
|
node["month_rx"] != float64(33) || node["month_used"] != float64(77) ||
|
||||||
node["total_tx"] != float64(222) || node["price"] != float64(10) {
|
node["total_tx"] != float64(222) || node["price"] != float64(10) {
|
||||||
t.Fatalf("anon node values wrong: %v", node)
|
t.Fatalf("anon node values wrong: %v", node)
|
||||||
|
|||||||
@@ -1234,7 +1234,74 @@ func (s *NodeService) AggregateNodeMetric(id int, metric string, bucketSeconds i
|
|||||||
return nodeMetrics.aggregate(nodeMetricKey(id, metric), bucketSeconds, maxPoints)
|
return nodeMetrics.aggregate(nodeMetricKey(id, metric), bucketSeconds, maxPoints)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// agentOnlineWindow is how fresh an agent heartbeat must be for its
|
||||||
|
// node to read online — the same 180s the heartbeat job expires agent
|
||||||
|
// nodes by (agentHeartbeatStaleAfter in internal/web/job: max(3
|
||||||
|
// heartbeat intervals, 120s), so two lost heartbeats don't flap).
|
||||||
|
const agentOnlineWindow = 180 * time.Second
|
||||||
|
|
||||||
|
var errAgentOffline = errors.New("agent node offline: no fresh heartbeat")
|
||||||
|
|
||||||
|
// agentHeartbeatVerdict is the pure decision behind the agent-node
|
||||||
|
// probe: a fresh stamp reads online with the heartbeat AGE as the
|
||||||
|
// latency (there is no dial to time); a stale or never-seen stamp
|
||||||
|
// reads offline with LatencyMs -1. LastError stays empty either way —
|
||||||
|
// the answer comes from the heartbeat, not from a dial failure.
|
||||||
|
func agentHeartbeatVerdict(lastSeenMs int64, now time.Time) (HeartbeatPatch, bool) {
|
||||||
|
patch := HeartbeatPatch{LatencyMs: -1}
|
||||||
|
if lastSeenMs <= 0 {
|
||||||
|
return patch, false
|
||||||
|
}
|
||||||
|
patch.LastHeartbeat = lastSeenMs / 1000
|
||||||
|
if age := now.Sub(time.UnixMilli(lastSeenMs)); age <= agentOnlineWindow {
|
||||||
|
patch.LastHeartbeat = now.Unix()
|
||||||
|
patch.LatencyMs = int(age / time.Millisecond)
|
||||||
|
return patch, true
|
||||||
|
}
|
||||||
|
return patch, false
|
||||||
|
}
|
||||||
|
|
||||||
|
// agentNodeProbe answers a probe for an agent-managed node from the
|
||||||
|
// bound agent's last_seen stamp (Unix ms, the unit the agent
|
||||||
|
// controller stamps). ok=false means the node has no bound agent — or
|
||||||
|
// the lookup failed — and the caller takes the dial path. The stored
|
||||||
|
// telemetry fields carry through unchanged (same as the heartbeat
|
||||||
|
// job's agent-expiry patch); only the error is dropped, so a ⚡ click
|
||||||
|
// also heals the bogus dial errors recorded before this change.
|
||||||
|
func (s *NodeService) agentNodeProbe(n *model.Node) (patch HeartbeatPatch, err error, ok bool) {
|
||||||
|
var lastSeen *int64 // MAX() over zero rows is NULL
|
||||||
|
qerr := database.GetDB().Model(&model.AgentNode{}).
|
||||||
|
Where("node_id = ?", n.Id).
|
||||||
|
Select("MAX(last_seen)").
|
||||||
|
Scan(&lastSeen).Error
|
||||||
|
if qerr != nil || lastSeen == nil {
|
||||||
|
return HeartbeatPatch{}, nil, false
|
||||||
|
}
|
||||||
|
patch, online := agentHeartbeatVerdict(*lastSeen, time.Now())
|
||||||
|
patch.XrayVersion = n.XrayVersion
|
||||||
|
patch.PanelVersion = n.PanelVersion
|
||||||
|
patch.Guid = n.Guid
|
||||||
|
patch.CpuPct = n.CpuPct
|
||||||
|
patch.MemPct = n.MemPct
|
||||||
|
patch.UptimeSecs = n.UptimeSecs
|
||||||
|
patch.NetUp = n.NetUp
|
||||||
|
patch.NetDown = n.NetDown
|
||||||
|
patch.XrayState = n.XrayState
|
||||||
|
patch.XrayError = n.XrayError
|
||||||
|
if !online {
|
||||||
|
return patch, errAgentOffline, true
|
||||||
|
}
|
||||||
|
return patch, nil, true
|
||||||
|
}
|
||||||
|
|
||||||
func (s *NodeService) Probe(ctx context.Context, n *model.Node) (HeartbeatPatch, error) {
|
func (s *NodeService) Probe(ctx context.Context, n *model.Node) (HeartbeatPatch, error) {
|
||||||
|
// Agent-managed nodes have no panel to dial — their address row
|
||||||
|
// is a placeholder ("agent") — so a probe answers from the agent
|
||||||
|
// heartbeat instead of dialing a dead URL and persisting a
|
||||||
|
// LastError the agent heartbeat never clears (V2.2.1 ②).
|
||||||
|
if patch, err, ok := s.agentNodeProbe(n); ok {
|
||||||
|
return patch, err
|
||||||
|
}
|
||||||
proxyURL := ""
|
proxyURL := ""
|
||||||
if n.OutboundTag != "" {
|
if n.OutboundTag != "" {
|
||||||
if mgr := runtime.GetManager(); mgr != nil {
|
if mgr := runtime.GetManager(); mgr != nil {
|
||||||
|
|||||||
@@ -0,0 +1,70 @@
|
|||||||
|
package service
|
||||||
|
|
||||||
|
import (
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
// V2.2.1 ②: the ⚡ probe for an agent-managed node is synthesized
|
||||||
|
// from the agent's heartbeat stamp instead of dialing the placeholder
|
||||||
|
// address. These tests pin the pure verdict: fresh stamp → online
|
||||||
|
// with the heartbeat age as latency; stale / never-seen → offline
|
||||||
|
// with LatencyMs -1; LastError empty in every case (a dial error must
|
||||||
|
// never be recorded for an agent node again).
|
||||||
|
|
||||||
|
func TestAgentHeartbeatVerdictFreshIsOnlineWithAgeLatency(t *testing.T) {
|
||||||
|
now := time.Unix(1_700_000_000, 0)
|
||||||
|
lastSeenMs := now.Add(-30 * time.Second).UnixMilli()
|
||||||
|
patch, online := agentHeartbeatVerdict(lastSeenMs, now)
|
||||||
|
if !online {
|
||||||
|
t.Fatal("30s-old heartbeat must read online")
|
||||||
|
}
|
||||||
|
if patch.LatencyMs != 30_000 {
|
||||||
|
t.Fatalf("LatencyMs = %d, want heartbeat age 30000", patch.LatencyMs)
|
||||||
|
}
|
||||||
|
if patch.LastHeartbeat != now.Unix() {
|
||||||
|
t.Fatalf("LastHeartbeat = %d, want now %d", patch.LastHeartbeat, now.Unix())
|
||||||
|
}
|
||||||
|
if patch.LastError != "" {
|
||||||
|
t.Fatalf("LastError = %q, want empty", patch.LastError)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestAgentHeartbeatVerdictWindowEdge(t *testing.T) {
|
||||||
|
now := time.Unix(1_700_000_000, 0)
|
||||||
|
if _, online := agentHeartbeatVerdict(now.Add(-agentOnlineWindow).UnixMilli(), now); !online {
|
||||||
|
t.Fatal("heartbeat exactly at the window edge must read online")
|
||||||
|
}
|
||||||
|
if _, online := agentHeartbeatVerdict(now.Add(-agentOnlineWindow-time.Second).UnixMilli(), now); online {
|
||||||
|
t.Fatal("heartbeat past the window must read offline")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestAgentHeartbeatVerdictStaleIsOffline(t *testing.T) {
|
||||||
|
now := time.Unix(1_700_000_000, 0)
|
||||||
|
lastSeenMs := now.Add(-10 * time.Minute).UnixMilli()
|
||||||
|
patch, online := agentHeartbeatVerdict(lastSeenMs, now)
|
||||||
|
if online {
|
||||||
|
t.Fatal("10min-old heartbeat must read offline")
|
||||||
|
}
|
||||||
|
if patch.LatencyMs != -1 {
|
||||||
|
t.Fatalf("LatencyMs = %d, want -1", patch.LatencyMs)
|
||||||
|
}
|
||||||
|
if patch.LastHeartbeat != lastSeenMs/1000 {
|
||||||
|
t.Fatalf("LastHeartbeat = %d, want the truthful last-seen %d", patch.LastHeartbeat, lastSeenMs/1000)
|
||||||
|
}
|
||||||
|
if patch.LastError != "" {
|
||||||
|
t.Fatalf("LastError = %q, want empty", patch.LastError)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestAgentHeartbeatVerdictNeverSeenIsOffline(t *testing.T) {
|
||||||
|
now := time.Unix(1_700_000_000, 0)
|
||||||
|
patch, online := agentHeartbeatVerdict(0, now)
|
||||||
|
if online {
|
||||||
|
t.Fatal("never-seen agent must read offline")
|
||||||
|
}
|
||||||
|
if patch.LatencyMs != -1 || patch.LastHeartbeat != 0 || patch.LastError != "" {
|
||||||
|
t.Fatalf("patch = %+v, want zero stamp, latency -1, no error", patch)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -7,6 +7,28 @@ import (
|
|||||||
"github.com/mhsanaei/3x-ui/v3/internal/database/model"
|
"github.com/mhsanaei/3x-ui/v3/internal/database/model"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
// The panel-version cell must read as the bare A-form across the
|
||||||
|
// fleet: pre-fix agents report the doubled "AV2.1.0" and are repaired
|
||||||
|
// at view assembly; legacy build versions keep their "agent " prefix.
|
||||||
|
func TestAgentVersionCell(t *testing.T) {
|
||||||
|
cases := map[string]string{
|
||||||
|
"A2.1.0": "A2.1.0",
|
||||||
|
"A1.0.0": "A1.0.0",
|
||||||
|
"AV2.1.0": "A2.1.0",
|
||||||
|
"AV2.0.1": "A2.0.1",
|
||||||
|
"agent AV2.0.3": "A2.0.3",
|
||||||
|
"3.7.0-agent.1": "agent 3.7.0-agent.1",
|
||||||
|
"agent A2.1.0": "A2.1.0",
|
||||||
|
" A2.1.0 ": "A2.1.0",
|
||||||
|
"some-dev-build": "agent some-dev-build",
|
||||||
|
}
|
||||||
|
for in, want := range cases {
|
||||||
|
if got := agentVersionCell(in); got != want {
|
||||||
|
t.Errorf("agentVersionCell(%q) = %q, want %q", in, got, want)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// P13: annotateAgentTelemetry overlays the agent-reported heartbeat
|
// P13: annotateAgentTelemetry overlays the agent-reported heartbeat
|
||||||
// telemetry onto agent node views only — panel version becomes
|
// telemetry onto agent node views only — panel version becomes
|
||||||
// "agent <version>", xray version/uptime/latency/CPU/mem/IPs take the
|
// "agent <version>", xray version/uptime/latency/CPU/mem/IPs take the
|
||||||
|
|||||||
@@ -190,6 +190,30 @@ func annotateNodeTypes(views []*NodeView) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// agentVersionCell renders one agent's reported version for the
|
||||||
|
// node list's panel-version cell, uniformly across the fleet:
|
||||||
|
// - current agents report the A-series display version and are
|
||||||
|
// shown bare ("A2.1.0");
|
||||||
|
// - agents built while releases injected the V-form number report
|
||||||
|
// the doubled "AV2.1.0" (see config.GetAgentDisplayVersion) and
|
||||||
|
// are repaired to the bare A-form here, so those nodes read
|
||||||
|
// correctly without waiting for their next agent update;
|
||||||
|
// - pre-display-version agents report the internal build version
|
||||||
|
// and keep the historical "agent " prefix.
|
||||||
|
func agentVersionCell(reported string) string {
|
||||||
|
s := strings.TrimSpace(reported)
|
||||||
|
s = strings.TrimPrefix(s, "agent ")
|
||||||
|
if len(s) > 2 && (s[0] == 'A' || s[0] == 'a') && (s[1] == 'V' || s[1] == 'v') {
|
||||||
|
if _, ok := config.ParseDisplayVersion(s[1:]); ok {
|
||||||
|
return "A" + s[2:]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if _, ok := config.ParseDisplayVersion(s); ok && strings.HasPrefix(s, "A") {
|
||||||
|
return s
|
||||||
|
}
|
||||||
|
return "agent " + s
|
||||||
|
}
|
||||||
|
|
||||||
// annotateAgentTelemetry overlays the latest heartbeat telemetry reported
|
// annotateAgentTelemetry overlays the latest heartbeat telemetry reported
|
||||||
// by each node's bound agent onto its view (P13): xray version, the
|
// by each node's bound agent onto its view (P13): xray version, the
|
||||||
// "agent <version>" panel-version label, host uptime, node->master latency,
|
// "agent <version>" panel-version label, host uptime, node->master latency,
|
||||||
@@ -244,14 +268,7 @@ func annotateAgentTelemetry(views []*NodeView) {
|
|||||||
v.SingboxVersion = ag.SingboxVersion
|
v.SingboxVersion = ag.SingboxVersion
|
||||||
}
|
}
|
||||||
if ag.AgentVersion != "" {
|
if ag.AgentVersion != "" {
|
||||||
// New agents report the A-series display version, shown
|
v.PanelVersion = agentVersionCell(ag.AgentVersion)
|
||||||
// verbatim; older agents report the internal build version
|
|
||||||
// and keep the historical "agent " prefix.
|
|
||||||
if _, ok := config.ParseDisplayVersion(ag.AgentVersion); ok && strings.HasPrefix(ag.AgentVersion, "A") {
|
|
||||||
v.PanelVersion = ag.AgentVersion
|
|
||||||
} else {
|
|
||||||
v.PanelVersion = "agent " + ag.AgentVersion
|
|
||||||
}
|
|
||||||
v.AgentUpdateState = ag.UpdateState
|
v.AgentUpdateState = ag.UpdateState
|
||||||
v.AgentUpdateError = ag.UpdateError
|
v.AgentUpdateError = ag.UpdateError
|
||||||
v.AgentUpdateTarget = ag.UpdateTarget
|
v.AgentUpdateTarget = ag.UpdateTarget
|
||||||
|
|||||||
@@ -164,13 +164,38 @@ func (s *PanelService) GetUpdateInfo() (*PanelUpdateInfo, error) {
|
|||||||
// display version. An unparseable tag (not a V-series release) yields ""
|
// display version. An unparseable tag (not a V-series release) yields ""
|
||||||
// so the UI falls back to the raw fields.
|
// so the UI falls back to the raw fields.
|
||||||
func displayUpdateFields(latestTag string) (string, bool) {
|
func displayUpdateFields(latestTag string) (string, bool) {
|
||||||
|
return displayUpdateFieldsFor(latestTag, config.GetMasterDisplayVersion(), config.GetBaseVersion())
|
||||||
|
}
|
||||||
|
|
||||||
|
// displayUpdateFieldsFor is displayUpdateFields with the running
|
||||||
|
// build's versions as inputs. When the DISPLAY comparison cannot
|
||||||
|
// judge — the running build's display version is itself unparseable,
|
||||||
|
// the "VV2.1.0" double-letter builds — the verdict falls back to the
|
||||||
|
// raw version comparison instead of reporting "no update": the
|
||||||
|
// display path used to swallow a real update entirely (the panel
|
||||||
|
// showed no prompt while /releases/latest happily served the newer
|
||||||
|
// tag, and the native updater refused to run). A parseable display
|
||||||
|
// version keeps the segment rule exactly as before.
|
||||||
|
func displayUpdateFieldsFor(latestTag, currentDisplay, currentRaw string) (string, bool) {
|
||||||
parts, ok := config.ParseDisplayVersion(latestTag)
|
parts, ok := config.ParseDisplayVersion(latestTag)
|
||||||
if !ok {
|
if !ok {
|
||||||
return "", false
|
return "", false
|
||||||
}
|
}
|
||||||
disp := fmt.Sprintf("V%d.%d.%d", parts[0], parts[1], parts[2])
|
disp := fmt.Sprintf("V%d.%d.%d", parts[0], parts[1], parts[2])
|
||||||
cmp, ok := config.CompareDisplayVersion(disp, config.GetMasterDisplayVersion())
|
cmp, ok := config.CompareDisplayVersion(disp, currentDisplay)
|
||||||
return disp, ok && cmp > 0
|
if !ok {
|
||||||
|
// Raw fallback: judge the latest tag against the running
|
||||||
|
// build's raw base version, parsed with the same tolerant
|
||||||
|
// display parser (bare and V/A forms both parse). An
|
||||||
|
// unparseable raw base yields no verdict → false. This is
|
||||||
|
// deliberately NOT isNewerVersion: on this fork's uppercase-V
|
||||||
|
// tags that helper degrades to string inequality (its
|
||||||
|
// normalizer strips only a lowercase "v"), which would report
|
||||||
|
// a phantom update whenever latest merely differs in form.
|
||||||
|
rawCmp, rawOk := config.CompareDisplayVersion(disp, currentRaw)
|
||||||
|
return disp, rawOk && rawCmp > 0
|
||||||
|
}
|
||||||
|
return disp, cmp > 0
|
||||||
}
|
}
|
||||||
|
|
||||||
// devChannelActive reports whether self-update should track the rolling dev
|
// devChannelActive reports whether self-update should track the rolling dev
|
||||||
|
|||||||
@@ -25,6 +25,42 @@ func TestDisplayUpdateFields(t *testing.T) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// V2.2.1 ③: when the running build's display version is unparseable
|
||||||
|
// (the "VV2.1.0" double-letter builds), the display comparison cannot
|
||||||
|
// judge and the verdict must fall back to the raw version comparison
|
||||||
|
// instead of swallowing the update. A parseable display version keeps
|
||||||
|
// the segment rule — the raw string plays no part there.
|
||||||
|
func TestDisplayUpdateFieldsForFallback(t *testing.T) {
|
||||||
|
// Unparseable display "VV2.1.0", newer latest → available via raw.
|
||||||
|
if disp, avail := displayUpdateFieldsFor("V2.2.0", "VV2.1.0", "2.1.0"); disp != "V2.2.0" || !avail {
|
||||||
|
t.Fatalf("VV display, newer latest = %q, %v; want V2.2.0, true", disp, avail)
|
||||||
|
}
|
||||||
|
// Unparseable display, latest equal to the raw base → not
|
||||||
|
// available, in both raw forms (bare and V-prefixed).
|
||||||
|
if disp, avail := displayUpdateFieldsFor("V2.1.0", "VV2.1.0", "2.1.0"); disp != "V2.1.0" || avail {
|
||||||
|
t.Fatalf("VV display, same latest (bare raw) = %q, %v; want V2.1.0, false", disp, avail)
|
||||||
|
}
|
||||||
|
if _, avail := displayUpdateFieldsFor("V2.1.0", "VV2.1.0", "V2.1.0"); avail {
|
||||||
|
t.Fatal("VV display, same latest (V raw) must not report an update")
|
||||||
|
}
|
||||||
|
// Unparseable display AND unparseable raw base → no verdict, false.
|
||||||
|
if _, avail := displayUpdateFieldsFor("V2.2.0", "VV2.1.0", "3.7.0-agent.1"); avail {
|
||||||
|
t.Fatal("unparseable raw base must yield no update verdict")
|
||||||
|
}
|
||||||
|
// Parseable display: segment rule decides, raw base is ignored
|
||||||
|
// even when it disagrees.
|
||||||
|
if disp, avail := displayUpdateFieldsFor("V2.2.0", "V2.1.0", "9.9.9"); disp != "V2.2.0" || !avail {
|
||||||
|
t.Fatalf("parseable display, newer = %q, %v; want V2.2.0, true", disp, avail)
|
||||||
|
}
|
||||||
|
if _, avail := displayUpdateFieldsFor("V2.1.0", "V2.1.0", "1.0.0"); avail {
|
||||||
|
t.Fatal("parseable display, equal versions must not report an update (raw base must not leak in)")
|
||||||
|
}
|
||||||
|
// Unparseable latest tag still yields "" so the UI uses raw fields.
|
||||||
|
if disp, avail := displayUpdateFieldsFor("3.7.0-agent.1", "VV2.1.0", "2.1.0"); disp != "" || avail {
|
||||||
|
t.Fatalf("legacy tag = %q, %v; want empty, false", disp, avail)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func TestServiceUnitFromCgroup(t *testing.T) {
|
func TestServiceUnitFromCgroup(t *testing.T) {
|
||||||
content := "0::/system.slice/x-ui.service\n"
|
content := "0::/system.slice/x-ui.service\n"
|
||||||
if got := serviceUnitFromCgroup(content); got != "x-ui" {
|
if got := serviceUnitFromCgroup(content); got != "x-ui" {
|
||||||
|
|||||||
@@ -34,6 +34,9 @@ type Meta struct {
|
|||||||
// corrections in GB (V2.0.1), either sign: added to the counted
|
// corrections in GB (V2.0.1), either sign: added to the counted
|
||||||
// uplink (tx) / downlink (rx) figures wherever the ledger is
|
// uplink (tx) / downlink (rx) figures wherever the ledger is
|
||||||
// read — panel statistics, the public page and traffic alerts.
|
// read — panel statistics, the public page and traffic alerts.
|
||||||
|
// The stored values are the STANDING accumulated correction;
|
||||||
|
// the edit form submits one-shot deltas that MergeCalDelta folds
|
||||||
|
// into them (V2.2.1 ⑥b).
|
||||||
TrafficCalUpGB float64
|
TrafficCalUpGB float64
|
||||||
TrafficCalDownGB float64
|
TrafficCalDownGB float64
|
||||||
// StatIface pins the one interface whose counters the agent
|
// StatIface pins the one interface whose counters the agent
|
||||||
@@ -138,6 +141,30 @@ func (s *Store) UpsertMeta(m Meta) error {
|
|||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// MergeCalDelta folds a probe-meta form submission into the stored
|
||||||
|
// row under the one-shot calibration semantics (V2.2.1 ⑥b): the
|
||||||
|
// form's calibration fields are a correction TO APPLY NOW, not the
|
||||||
|
// new standing value. A non-zero submitted value adds onto the stored
|
||||||
|
// standing correction; a zero field (an absent field is the same
|
||||||
|
// zero — the wire type cannot distinguish them, and both mean
|
||||||
|
// "nothing to apply") keeps the stored value untouched. Every other
|
||||||
|
// field takes the submitted value, as UpsertMeta's full replace
|
||||||
|
// does. Paired with the form clearing its calibration inputs after a
|
||||||
|
// successful save, a later unrelated save can never re-apply an old
|
||||||
|
// correction.
|
||||||
|
func MergeCalDelta(existing, submitted Meta) Meta {
|
||||||
|
out := submitted
|
||||||
|
out.TrafficCalUpGB = existing.TrafficCalUpGB
|
||||||
|
if submitted.TrafficCalUpGB != 0 {
|
||||||
|
out.TrafficCalUpGB += submitted.TrafficCalUpGB
|
||||||
|
}
|
||||||
|
out.TrafficCalDownGB = existing.TrafficCalDownGB
|
||||||
|
if submitted.TrafficCalDownGB != 0 {
|
||||||
|
out.TrafficCalDownGB += submitted.TrafficCalDownGB
|
||||||
|
}
|
||||||
|
return out
|
||||||
|
}
|
||||||
|
|
||||||
func boolToInt(b bool) int64 {
|
func boolToInt(b bool) int64 {
|
||||||
if b {
|
if b {
|
||||||
return 1
|
return 1
|
||||||
|
|||||||
@@ -0,0 +1,143 @@
|
|||||||
|
package probe
|
||||||
|
|
||||||
|
import (
|
||||||
|
"path/filepath"
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
// V2.2.1 ⑥: GB entries convert on the binary convention (1024^3)
|
||||||
|
// every formatter uses, stored decimal-era limits are rescaled once,
|
||||||
|
// and calibration submissions are one-shot deltas folded into the
|
||||||
|
// standing correction — never re-applied by a later unrelated save.
|
||||||
|
|
||||||
|
func TestCalBytesBinaryConvention(t *testing.T) {
|
||||||
|
if got := calBytes(3000); got != 3000*(1<<30) {
|
||||||
|
t.Fatalf("calBytes(3000) = %d, want %d (3000 GiB)", got, 3000*(1<<30))
|
||||||
|
}
|
||||||
|
if got := calBytes(0); got != 0 {
|
||||||
|
t.Fatalf("calBytes(0) = %d, want 0", got)
|
||||||
|
}
|
||||||
|
if got := calBytes(-1.5); got != -1610612736 {
|
||||||
|
t.Fatalf("calBytes(-1.5) = %d, want -1610612736", got)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestGBConventionMigrationRescalesLimitsOnce(t *testing.T) {
|
||||||
|
path := filepath.Join(t.TempDir(), "probe.db")
|
||||||
|
st, err := Open(path)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("open: %v", err)
|
||||||
|
}
|
||||||
|
// Seed a legacy row as a pre-V2.2.1 database holds it — 3000 GB
|
||||||
|
// typed at the decimal convention = 3.0e12 stored bytes — and
|
||||||
|
// strike the migration mark so the next open sees an unmigrated
|
||||||
|
// database.
|
||||||
|
if _, err := st.db.Exec(`INSERT INTO probe_meta (agent_id, traffic_limit) VALUES (9, 3000000000000)`); err != nil {
|
||||||
|
t.Fatalf("seed legacy limit: %v", err)
|
||||||
|
}
|
||||||
|
if _, err := st.db.Exec(`DELETE FROM probe_setting WHERE key = ?`, gbConventionKey); err != nil {
|
||||||
|
t.Fatalf("strike mark: %v", err)
|
||||||
|
}
|
||||||
|
if err := st.Close(); err != nil {
|
||||||
|
t.Fatalf("close: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
st2, err := Open(path)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("reopen (migration run): %v", err)
|
||||||
|
}
|
||||||
|
m, found, err := st2.Meta(9)
|
||||||
|
if err != nil || !found {
|
||||||
|
t.Fatalf("meta after migration: found=%v err=%v", found, err)
|
||||||
|
}
|
||||||
|
// 3.0e12 x 1024^3/1e9 = 3000 x 2^30 exactly (~3.221e12).
|
||||||
|
if want := int64(3000) * (1 << 30); m.TrafficLimit != want {
|
||||||
|
t.Fatalf("migrated limit = %d, want %d", m.TrafficLimit, want)
|
||||||
|
}
|
||||||
|
if err := st2.Close(); err != nil {
|
||||||
|
t.Fatalf("close 2: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
st3, err := Open(path)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("reopen 2: %v", err)
|
||||||
|
}
|
||||||
|
defer st3.Close()
|
||||||
|
m, found, err = st3.Meta(9)
|
||||||
|
if err != nil || !found {
|
||||||
|
t.Fatalf("meta after second open: found=%v err=%v", found, err)
|
||||||
|
}
|
||||||
|
if want := int64(3000) * (1 << 30); m.TrafficLimit != want {
|
||||||
|
t.Fatalf("limit after second open = %d, want %d (no double rescale)", m.TrafficLimit, want)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestOneShotCalibrationSaveFlow(t *testing.T) {
|
||||||
|
st := newTestStore(t)
|
||||||
|
now := time.Now()
|
||||||
|
book(t, st, 5, "boot-a", 1000, 1000, now, nil, 0) // baseline: books 0
|
||||||
|
book(t, st, 5, "boot-a", 3000, 2000, now, nil, 0) // books rx 2000, tx 1000
|
||||||
|
|
||||||
|
// First save carries a +1 GB uplink correction delta.
|
||||||
|
submitted := DefaultMeta(5)
|
||||||
|
submitted.TrafficCalUpGB = 1
|
||||||
|
merged := MergeCalDelta(DefaultMeta(5), submitted)
|
||||||
|
if merged.TrafficCalUpGB != 1 {
|
||||||
|
t.Fatalf("standing after first fold = %v, want 1", merged.TrafficCalUpGB)
|
||||||
|
}
|
||||||
|
if err := st.UpsertMeta(merged); err != nil {
|
||||||
|
t.Fatalf("upsert 1: %v", err)
|
||||||
|
}
|
||||||
|
after1, ok, err := st.TrafficFor(5)
|
||||||
|
if err != nil || !ok {
|
||||||
|
t.Fatalf("traffic after save 1: ok=%v err=%v", ok, err)
|
||||||
|
}
|
||||||
|
if want := int64(1000) + (1 << 30); after1.TotalTx != want {
|
||||||
|
t.Fatalf("TotalTx after save 1 = %d, want %d", after1.TotalTx, want)
|
||||||
|
}
|
||||||
|
if after1.TotalRx != 2000 {
|
||||||
|
t.Fatalf("TotalRx after save 1 = %d, want 2000 (uncalibrated)", after1.TotalRx)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Second save changes another field; the (cleared) form submits
|
||||||
|
// zero calibration — the calibrated figure must not move again.
|
||||||
|
stored, found, err := st.Meta(5)
|
||||||
|
if err != nil || !found {
|
||||||
|
t.Fatalf("stored meta: found=%v err=%v", found, err)
|
||||||
|
}
|
||||||
|
sub2 := DefaultMeta(5)
|
||||||
|
sub2.GroupName = "g2"
|
||||||
|
merged2 := MergeCalDelta(stored, sub2)
|
||||||
|
if merged2.TrafficCalUpGB != 1 {
|
||||||
|
t.Fatalf("standing after zero-delta save = %v, want 1 (unchanged)", merged2.TrafficCalUpGB)
|
||||||
|
}
|
||||||
|
if err := st.UpsertMeta(merged2); err != nil {
|
||||||
|
t.Fatalf("upsert 2: %v", err)
|
||||||
|
}
|
||||||
|
after2, ok, err := st.TrafficFor(5)
|
||||||
|
if err != nil || !ok {
|
||||||
|
t.Fatalf("traffic after save 2: ok=%v err=%v", ok, err)
|
||||||
|
}
|
||||||
|
if after2.TotalTx != after1.TotalTx {
|
||||||
|
t.Fatalf("TotalTx moved on an unrelated save: %d -> %d (double-applied calibration)", after1.TotalTx, after2.TotalTx)
|
||||||
|
}
|
||||||
|
|
||||||
|
// A further non-zero delta accumulates exactly once more.
|
||||||
|
sub3 := DefaultMeta(5)
|
||||||
|
sub3.TrafficCalUpGB = 0.5
|
||||||
|
merged3 := MergeCalDelta(merged2, sub3)
|
||||||
|
if merged3.TrafficCalUpGB != 1.5 {
|
||||||
|
t.Fatalf("standing after second delta = %v, want 1.5", merged3.TrafficCalUpGB)
|
||||||
|
}
|
||||||
|
if err := st.UpsertMeta(merged3); err != nil {
|
||||||
|
t.Fatalf("upsert 3: %v", err)
|
||||||
|
}
|
||||||
|
after3, _, err := st.TrafficFor(5)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("traffic after save 3: %v", err)
|
||||||
|
}
|
||||||
|
if want := int64(1000) + calBytes(1.5); after3.TotalTx != want {
|
||||||
|
t.Fatalf("TotalTx after save 3 = %d, want %d", after3.TotalTx, want)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -91,8 +91,10 @@ func TestTrafficCalibrationAppliedAndFloored(t *testing.T) {
|
|||||||
t.Fatalf("AllTraffic: %v", err)
|
t.Fatalf("AllTraffic: %v", err)
|
||||||
}
|
}
|
||||||
tr := all[5]
|
tr := all[5]
|
||||||
if tr.TotalRx != 3_500_000_000 || tr.MonthRx != 3_500_000_000 || tr.DayRx != 3_500_000_000 {
|
// V2.2.1 ⑥: calibration converts on the binary convention —
|
||||||
t.Errorf("rx figures after +0.5 GB calibration = %d/%d/%d, want 3500000000 each", tr.TotalRx, tr.MonthRx, tr.DayRx)
|
// +0.5 GB = 0.5 x 2^30 = 536870912 B (was 5e8 at the old 1e9).
|
||||||
|
if tr.TotalRx != 3_536_870_912 || tr.MonthRx != 3_536_870_912 || tr.DayRx != 3_536_870_912 {
|
||||||
|
t.Errorf("rx figures after +0.5 GB calibration = %d/%d/%d, want 3536870912 each", tr.TotalRx, tr.MonthRx, tr.DayRx)
|
||||||
}
|
}
|
||||||
if tr.TotalTx != 0 || tr.MonthTx != 0 || tr.DayTx != 0 {
|
if tr.TotalTx != 0 || tr.MonthTx != 0 || tr.DayTx != 0 {
|
||||||
t.Errorf("tx figures after -2 GB calibration = %d/%d/%d, want 0 (floored)", tr.TotalTx, tr.MonthTx, tr.DayTx)
|
t.Errorf("tx figures after -2 GB calibration = %d/%d/%d, want 0 (floored)", tr.TotalTx, tr.MonthTx, tr.DayTx)
|
||||||
|
|||||||
@@ -213,9 +213,52 @@ func Open(path string) (*Store, error) {
|
|||||||
db.Close()
|
db.Close()
|
||||||
return nil, fmt.Errorf("probe schema migration: %w", err)
|
return nil, fmt.Errorf("probe schema migration: %w", err)
|
||||||
}
|
}
|
||||||
|
if err := migrateGBConvention(db); err != nil {
|
||||||
|
db.Close()
|
||||||
|
return nil, fmt.Errorf("probe gb migration: %w", err)
|
||||||
|
}
|
||||||
return &Store{db: db, nodes: make(map[int64]*nodeState)}, nil
|
return &Store{db: db, nodes: make(map[int64]*nodeState)}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// gbConventionKey is the probe_setting mark recording that the
|
||||||
|
// V2.2.1 ⑥ traffic-limit rescale has been applied to this database.
|
||||||
|
const gbConventionKey = "gb_1024_migrated_v1"
|
||||||
|
|
||||||
|
// migrateGBConvention rescales stored probe_meta traffic limits from
|
||||||
|
// the old decimal entry convention to the unified binary one, once.
|
||||||
|
// Limits are typed in GB and were stored as GB × 1e9 bytes, while
|
||||||
|
// every byte formatter (panel, public page) divides by 1024^3 — a
|
||||||
|
// 3000 GB limit displayed as 2.73 TB instead of 2.93 TB. The GB the
|
||||||
|
// operator typed is the intent, so stored bytes are scaled by
|
||||||
|
// 1024^3/1e9 = 1.073741824 and now read back correctly under the
|
||||||
|
// binary conversion. Only traffic_limit is byte-denominated:
|
||||||
|
// calibration values are stored in GB (their byte conversion,
|
||||||
|
// calBytes, carries the fix) and price/sort/reset-day are unitless —
|
||||||
|
// neither is touched here. The probe_setting mark makes the rescale
|
||||||
|
// run exactly once per database.
|
||||||
|
func migrateGBConvention(db *sql.DB) error {
|
||||||
|
var v string
|
||||||
|
err := db.QueryRow(`SELECT value FROM probe_setting WHERE key = ?`, gbConventionKey).Scan(&v)
|
||||||
|
if err == nil {
|
||||||
|
return nil // already applied
|
||||||
|
}
|
||||||
|
if err != sql.ErrNoRows {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
tx, err := db.Begin()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer tx.Rollback()
|
||||||
|
if _, err := tx.Exec(`UPDATE probe_meta SET traffic_limit = CAST(ROUND(traffic_limit * 1.073741824) AS INTEGER) WHERE traffic_limit > 0`); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if _, err := tx.Exec(`INSERT OR REPLACE INTO probe_setting (key, value) VALUES (?, '1')`, gbConventionKey); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return tx.Commit()
|
||||||
|
}
|
||||||
|
|
||||||
// metaColumnMigrations are the probe_meta columns added after the
|
// metaColumnMigrations are the probe_meta columns added after the
|
||||||
// table's first release; CREATE TABLE IF NOT EXISTS never retrofits
|
// table's first release; CREATE TABLE IF NOT EXISTS never retrofits
|
||||||
// them onto an existing database, so they are added here, once, by
|
// them onto an existing database, so they are added here, once, by
|
||||||
|
|||||||
@@ -332,9 +332,13 @@ func (s *Store) TrafficFor(agentID int64) (Traffic, bool, error) {
|
|||||||
// AllTraffic returns every agent's ledger in one query, gated as in
|
// AllTraffic returns every agent's ledger in one query, gated as in
|
||||||
// TrafficFor — the bulk read port for B3's node list.
|
// TrafficFor — the bulk read port for B3's node list.
|
||||||
// calBytes converts a GB calibration value to bytes on the panel's
|
// calBytes converts a GB calibration value to bytes on the panel's
|
||||||
// 1 GB = 1e9 B convention (the same one the traffic limit uses).
|
// binary convention, 1 GB = 1024^3 B — the convention every byte
|
||||||
|
// formatter in the panel and on the public page uses, and the one
|
||||||
|
// the traffic limit's GB entry converts with (V2.2.1 ⑥; the decimal
|
||||||
|
// 1e9 stood here before, which made a 3000 GB limit read as 2.73 TB
|
||||||
|
// instead of 2.93 TB wherever bytes were formatted).
|
||||||
func calBytes(gb float64) int64 {
|
func calBytes(gb float64) int64 {
|
||||||
return int64(math.Round(gb * 1e9))
|
return int64(math.Round(gb * (1 << 30)))
|
||||||
}
|
}
|
||||||
|
|
||||||
// applyCalibration shifts a ledger row by the operator's per-node
|
// applyCalibration shifts a ledger row by the operator's per-node
|
||||||
|
|||||||
@@ -268,6 +268,19 @@ func expiresInDays(expires *string, now time.Time) any {
|
|||||||
return days
|
return days
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// probeDisplayName is the name a node carries in the public views
|
||||||
|
// (V2.2.1 ⑦): the operator's public remark when one is set — it is
|
||||||
|
// the label written for public consumption — else the node name. The
|
||||||
|
// contract field stays "name"; themes and the public API keep their
|
||||||
|
// shape, only the value's source changes. The panel's internal node
|
||||||
|
// list is a different surface and keeps the plain node name.
|
||||||
|
func probeDisplayName(name, publicRemark string) string {
|
||||||
|
if r := strings.TrimSpace(publicRemark); r != "" {
|
||||||
|
return r
|
||||||
|
}
|
||||||
|
return name
|
||||||
|
}
|
||||||
|
|
||||||
// buildProbeNodeView renders one node for one audience. The public
|
// buildProbeNodeView renders one node for one audience. The public
|
||||||
// shape is built by whitelist (hub node_view): address, hostname and
|
// shape is built by whitelist (hub node_view): address, hostname and
|
||||||
// the private remark exist only in the admin (full) rendering.
|
// the private remark exist only in the admin (full) rendering.
|
||||||
@@ -298,7 +311,7 @@ func buildProbeNodeView(src probeNodeSource, full bool) map[string]any {
|
|||||||
|
|
||||||
view := map[string]any{
|
view := map[string]any{
|
||||||
"id": a.Id,
|
"id": a.Id,
|
||||||
"name": src.name,
|
"name": probeDisplayName(src.name, m.PublicRemark),
|
||||||
"country": m.CountryPin,
|
"country": m.CountryPin,
|
||||||
"group": m.GroupName,
|
"group": m.GroupName,
|
||||||
"public_remark": m.PublicRemark,
|
"public_remark": m.PublicRemark,
|
||||||
|
|||||||
@@ -0,0 +1,40 @@
|
|||||||
|
package service
|
||||||
|
|
||||||
|
import (
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/mhsanaei/3x-ui/v3/internal/web/service/probe"
|
||||||
|
)
|
||||||
|
|
||||||
|
// V2.2.1 ⑦: the public views' node name is the operator's public
|
||||||
|
// remark when one is set, and the node name otherwise — the panel's
|
||||||
|
// own list (a different surface) is untouched.
|
||||||
|
|
||||||
|
func TestProbeDisplayNamePrefersPublicRemark(t *testing.T) {
|
||||||
|
if got := probeDisplayName("node-name", "东京 01"); got != "东京 01" {
|
||||||
|
t.Fatalf("with remark = %q, want the remark", got)
|
||||||
|
}
|
||||||
|
if got := probeDisplayName("node-name", ""); got != "node-name" {
|
||||||
|
t.Fatalf("empty remark = %q, want node name", got)
|
||||||
|
}
|
||||||
|
if got := probeDisplayName("node-name", " "); got != "node-name" {
|
||||||
|
t.Fatalf("whitespace-only remark = %q, want node name", got)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestBuildProbeNodeViewNameSource(t *testing.T) {
|
||||||
|
withRemark := probeNodeSource{
|
||||||
|
name: "internal-name",
|
||||||
|
meta: probe.Meta{PublicRemark: "公开名", Public: true},
|
||||||
|
}
|
||||||
|
if got := buildProbeNodeView(withRemark, false)["name"]; got != "公开名" {
|
||||||
|
t.Fatalf("anonymous view name = %v, want the public remark", got)
|
||||||
|
}
|
||||||
|
if got := buildProbeNodeView(withRemark, true)["name"]; got != "公开名" {
|
||||||
|
t.Fatalf("admin view name = %v, want the public remark", got)
|
||||||
|
}
|
||||||
|
without := probeNodeSource{name: "internal-name", meta: probe.Meta{Public: true}}
|
||||||
|
if got := buildProbeNodeView(without, false)["name"]; got != "internal-name" {
|
||||||
|
t.Fatalf("view name without remark = %v, want the node name", got)
|
||||||
|
}
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user