feat(amneziawg): swap the app's integration points to the embedded manager

Hard cutover, part 2: every real call site that used to drive
internal/amneziawg's kernel-module Manager now drives
internal/amneziawgnet's instead --

- internal/web/job/amneziawg_job.go: the reconcile cron job. Traffic/
  online-status accounting is dropped entirely (not ported) -- once a
  peer's traffic is relayed through Xray's own SOCKS5 inbound, it's an
  ordinary Xray user and XrayTrafficJob's existing generic stats polling
  already handles it, with zero AmneziaWG-specific code.
- internal/web/runtime/local.go: the immediate-apply CRUD path
  (AddInbound/DelInbound/updateAmneziaWGInbound).
- internal/web/web.go: panel shutdown's StopAll.

internal/amneziawgnet.Manager gains Remove(id) to match the kernel-module
Manager's shape at these call sites (Reconcile alone doesn't cover a
single-inbound removal outside a full reconcile pass).

internal/web/service/inbound_amneziawg.go's applyLocalAmneziaWG needed no
change: it already goes through runtime.Runtime.UpdateInbound, which now
resolves to the updated local.go path.

Co-Authored-By: Claude Sonnet 5 <noreply@anthropic.com>
This commit is contained in:
Kuzz007
2026-08-02 15:04:51 +03:00
parent efca370bfc
commit f78dfa6f67
4 changed files with 46 additions and 78 deletions
+17
View File
@@ -203,6 +203,23 @@ func (m *Manager) Reconcile(desired []Desired) {
} }
} }
// Remove tears down inbound id's embedded interface, if any -- mirrors
// internal/amneziawg.Manager.Remove, for a caller that needs to drop a
// single inbound outside a full Reconcile pass (e.g. the immediate-apply
// CRUD path in internal/web/runtime/local.go).
func (m *Manager) Remove(id int) {
m.mu.Lock()
defer m.mu.Unlock()
cur, exists := m.ifaces[id]
if !exists {
return
}
cur.udpRelay.Close()
cur.dev.Close()
delete(m.ifaces, id)
logger.Infof("amneziawgnet: stopped embedded interface for removed inbound %d", id)
}
// StopAll tears down every managed interface. Called on panel shutdown. // StopAll tears down every managed interface. Called on panel shutdown.
func (m *Manager) StopAll() { func (m *Manager) StopAll() {
m.mu.Lock() m.mu.Lock()
+17 -68
View File
@@ -1,33 +1,32 @@
package job package job
import ( import (
"github.com/mhsanaei/3x-ui/v3/internal/amneziawg" "github.com/mhsanaei/3x-ui/v3/internal/amneziawgnet"
"github.com/mhsanaei/3x-ui/v3/internal/database/model"
"github.com/mhsanaei/3x-ui/v3/internal/logger" "github.com/mhsanaei/3x-ui/v3/internal/logger"
"github.com/mhsanaei/3x-ui/v3/internal/web/service" "github.com/mhsanaei/3x-ui/v3/internal/web/service"
"github.com/mhsanaei/3x-ui/v3/internal/xray"
) )
// AmneziaWGJob reconciles the running AmneziaWG interfaces against the // AmneziaWGJob reconciles the running embedded AmneziaWG interfaces
// enabled AmneziaWG inbounds in the database, restarts/reloads any that // (internal/amneziawgnet -- amneziawg-go over a gVisor netstack, no kernel
// drifted, and folds the per-peer traffic scraped from `awg show dump` into // module) against the enabled AmneziaWG inbounds in the database,
// the usual client and inbound traffic accounting. Mirrors MtprotoJob. // rebuilding/reconfiguring any that drifted. Unlike the retired
// kernel-module Manager this job used to drive, there is no traffic/
// online-status accounting here at all: once a peer's decapsulated traffic
// is relayed into Xray's own SOCKS5 inbound (see
// internal/web/service/xray.go's injectAmneziawgnetSocks, and
// internal/amneziawgnet.Manager's automatic forwarder/relay wiring), it's
// an ordinary Xray user, and XrayTrafficJob's existing, protocol-blind
// stats/online-status polling already picks it up for free.
type AmneziaWGJob struct { type AmneziaWGJob struct {
inboundService service.InboundService inboundService service.InboundService
// warnedMissing tracks whether the "awg/awg-quick not found" warning has
// already been logged, so a host without the AmneziaWG kernel module
// (RHEL, Arch, a container, or a failed install.sh PPA step) logs it
// once instead of every @every-10s tick forever.
warnedMissing bool
} }
// NewAmneziaWGJob creates a new AmneziaWG reconcile/traffic job instance. // NewAmneziaWGJob creates a new AmneziaWG reconcile job instance.
func NewAmneziaWGJob() *AmneziaWGJob { func NewAmneziaWGJob() *AmneziaWGJob {
return new(AmneziaWGJob) return new(AmneziaWGJob)
} }
// Run reconciles desired AmneziaWG inbounds with running interfaces and // Run reconciles desired AmneziaWG inbounds with running embedded interfaces.
// records per-peer traffic deltas and online status.
func (j *AmneziaWGJob) Run() { func (j *AmneziaWGJob) Run() {
desired, err := j.inboundService.DesiredAmneziaWGInstances() desired, err := j.inboundService.DesiredAmneziaWGInstances()
if err != nil { if err != nil {
@@ -35,59 +34,9 @@ func (j *AmneziaWGJob) Run() {
return return
} }
// Only relevant once an admin actually has an AmneziaWG inbound: no wanted := make([]amneziawgnet.Desired, 0, len(desired))
// point warning about a missing binary the panel never needed to touch.
if len(desired) > 0 && !amneziawg.IsAwgInstalled() {
if !j.warnedMissing {
j.warnedMissing = true
logger.Warningf("amneziawg job: %d AmneziaWG inbound(s) configured but awg/awg-quick not found on PATH; skipping reconcile until installed", len(desired))
}
return
}
j.warnedMissing = false
activeTags := make([]string, 0, len(desired))
for _, inst := range desired { for _, inst := range desired {
activeTags = append(activeTags, inst.Tag) wanted = append(wanted, amneziawgnet.Desired{Instance: inst})
} }
amneziawgnet.GetManager().Reconcile(wanted)
mgr := amneziawg.GetManager()
mgr.Reconcile(desired)
deltas, onlineEmails := mgr.CollectTraffic()
clientTraffics := make([]*xray.ClientTraffic, 0, len(deltas))
inboundUp := make(map[string]int64)
inboundDown := make(map[string]int64)
for _, d := range deltas {
clientTraffics = append(clientTraffics, &xray.ClientTraffic{
Email: d.Email,
Up: d.Up,
Down: d.Down,
})
inboundUp[d.Tag] += d.Up
inboundDown[d.Tag] += d.Down
}
traffics := make([]*xray.Traffic, 0, len(inboundUp))
for tag, up := range inboundUp {
traffics = append(traffics, &xray.Traffic{
IsInbound: true,
Tag: tag,
Up: up,
Down: inboundDown[tag],
})
}
if len(traffics) > 0 || len(clientTraffics) > 0 {
if _, _, err := j.inboundService.AddTraffic(traffics, clientTraffics); err != nil {
logger.Warning("amneziawg job: add traffic failed:", err)
}
}
// Live speed: AmneziaWG never runs inside xray-core, so XrayTrafficJob's
// own 5s broadcast never mentions these tags. See sidecar_traffic.go.
broadcastSidecarTraffic(string(model.AmneziaWG), traffics, clientTraffics)
j.inboundService.RefreshLocalOnlineClients(onlineEmails, activeTags)
} }
+10 -8
View File
@@ -9,6 +9,7 @@ import (
"sync" "sync"
"github.com/mhsanaei/3x-ui/v3/internal/amneziawg" "github.com/mhsanaei/3x-ui/v3/internal/amneziawg"
"github.com/mhsanaei/3x-ui/v3/internal/amneziawgnet"
"github.com/mhsanaei/3x-ui/v3/internal/database/model" "github.com/mhsanaei/3x-ui/v3/internal/database/model"
"github.com/mhsanaei/3x-ui/v3/internal/mtproto" "github.com/mhsanaei/3x-ui/v3/internal/mtproto"
"github.com/mhsanaei/3x-ui/v3/internal/xray" "github.com/mhsanaei/3x-ui/v3/internal/xray"
@@ -59,7 +60,7 @@ func (l *Local) AddInbound(_ context.Context, ib *model.Inbound) error {
if !ok { if !ok {
return nil return nil
} }
return amneziawg.GetManager().Ensure(inst) return amneziawgnet.GetManager().Ensure(amneziawgnet.Desired{Instance: inst})
} }
body, err := json.MarshalIndent(ib.GenXrayInboundConfig(), "", " ") body, err := json.MarshalIndent(ib.GenXrayInboundConfig(), "", " ")
if err != nil { if err != nil {
@@ -76,7 +77,7 @@ func (l *Local) DelInbound(_ context.Context, ib *model.Inbound) error {
return nil return nil
} }
if ib.Protocol == model.AmneziaWG { if ib.Protocol == model.AmneziaWG {
amneziawg.GetManager().Remove(ib.Id) amneziawgnet.GetManager().Remove(ib.Id)
return nil return nil
} }
return l.withAPI(func(api *xray.XrayAPI) error { return l.withAPI(func(api *xray.XrayAPI) error {
@@ -130,11 +131,12 @@ func (l *Local) updateMtprotoInbound(ctx context.Context, oldIb, newIb *model.In
// updateAmneziaWGInbound mirrors updateMtprotoInbound: it skips the // updateAmneziaWGInbound mirrors updateMtprotoInbound: it skips the
// Remove+Ensure sequence a plain Del+Add would force so that, on an // Remove+Ensure sequence a plain Del+Add would force so that, on an
// AmneziaWG-to-AmneziaWG edit, Manager.Ensure's own fingerprint comparison // AmneziaWG-to-AmneziaWG edit, Manager.Ensure's own fingerprint comparison
// can pick a peers-only `syncconf` instead of always bouncing the interface // can reconfigure the running embedded Device in place via IpcSet instead
// (see internal/amneziawg.Manager.ensureLocked). // of always rebuilding it (see internal/amneziawgnet.Manager.ensureLocked --
// only an address/MTU change forces a rebuild there, not a peer edit).
func (l *Local) updateAmneziaWGInbound(ctx context.Context, oldIb, newIb *model.Inbound) error { func (l *Local) updateAmneziaWGInbound(ctx context.Context, oldIb, newIb *model.Inbound) error {
if oldIb.Protocol == model.AmneziaWG && newIb.Protocol != model.AmneziaWG { if oldIb.Protocol == model.AmneziaWG && newIb.Protocol != model.AmneziaWG {
amneziawg.GetManager().Remove(oldIb.Id) amneziawgnet.GetManager().Remove(oldIb.Id)
if !newIb.Enable { if !newIb.Enable {
return nil return nil
} }
@@ -144,15 +146,15 @@ func (l *Local) updateAmneziaWGInbound(ctx context.Context, oldIb, newIb *model.
_ = l.DelInbound(ctx, oldIb) _ = l.DelInbound(ctx, oldIb)
} }
if !newIb.Enable { if !newIb.Enable {
amneziawg.GetManager().Remove(newIb.Id) amneziawgnet.GetManager().Remove(newIb.Id)
return nil return nil
} }
inst, ok := amneziawg.InstanceFromInbound(newIb) inst, ok := amneziawg.InstanceFromInbound(newIb)
if !ok { if !ok {
amneziawg.GetManager().Remove(newIb.Id) amneziawgnet.GetManager().Remove(newIb.Id)
return nil return nil
} }
return amneziawg.GetManager().Ensure(inst) return amneziawgnet.GetManager().Ensure(amneziawgnet.Desired{Instance: inst})
} }
func (l *Local) AddUser(_ context.Context, ib *model.Inbound, userMap map[string]any) error { func (l *Local) AddUser(_ context.Context, ib *model.Inbound, userMap map[string]any) error {
+2 -2
View File
@@ -16,7 +16,7 @@ import (
"strings" "strings"
"time" "time"
"github.com/mhsanaei/3x-ui/v3/internal/amneziawg" "github.com/mhsanaei/3x-ui/v3/internal/amneziawgnet"
"github.com/mhsanaei/3x-ui/v3/internal/config" "github.com/mhsanaei/3x-ui/v3/internal/config"
"github.com/mhsanaei/3x-ui/v3/internal/eventbus" "github.com/mhsanaei/3x-ui/v3/internal/eventbus"
"github.com/mhsanaei/3x-ui/v3/internal/logger" "github.com/mhsanaei/3x-ui/v3/internal/logger"
@@ -689,7 +689,7 @@ func (s *Server) stop(stopXray bool, stopTgBot bool) error {
if stopXray { if stopXray {
_ = s.xrayService.StopXray() _ = s.xrayService.StopXray()
mtproto.GetManager().StopAll() mtproto.GetManager().StopAll()
amneziawg.GetManager().StopAll() amneziawgnet.GetManager().StopAll()
} }
if s.cron != nil { if s.cron != nil {
s.cron.Stop() s.cron.Stop()