mirror of
https://github.com/wyx2685/V2bX.git
synced 2026-02-05 05:00:08 +00:00
Compare commits
4 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
9bbccf772b | ||
|
|
9483526f3c | ||
|
|
65f2de55ea | ||
|
|
4ed6bc6d87 |
@@ -102,13 +102,13 @@ func (r *cachedReader) Interrupt() {
|
||||
|
||||
// DefaultDispatcher is a default implementation of Dispatcher.
|
||||
type DefaultDispatcher struct {
|
||||
ohm outbound.Manager
|
||||
router routing.Router
|
||||
policy policy.Manager
|
||||
stats stats.Manager
|
||||
fdns dns.FakeDNSEngine
|
||||
Wm *WriterManager
|
||||
Counter sync.Map
|
||||
ohm outbound.Manager
|
||||
router routing.Router
|
||||
policy policy.Manager
|
||||
stats stats.Manager
|
||||
fdns dns.FakeDNSEngine
|
||||
Counter sync.Map
|
||||
LinkManagers sync.Map // map[string]*LinkManager
|
||||
}
|
||||
|
||||
func init() {
|
||||
@@ -132,9 +132,6 @@ func (d *DefaultDispatcher) Init(config *Config, om outbound.Manager, router rou
|
||||
d.router = router
|
||||
d.policy = pm
|
||||
d.stats = sm
|
||||
d.Wm = &WriterManager{
|
||||
writers: make(map[string]map[*ManagedWriter]struct{}),
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
@@ -197,14 +194,23 @@ func (d *DefaultDispatcher) getLink(ctx context.Context, network net.Network) (*
|
||||
common.Interrupt(inboundLink.Reader)
|
||||
return nil, nil, nil, errors.New("Limited ", user.Email, " by conn or ip")
|
||||
}
|
||||
var lm *LinkManager
|
||||
if lmloaded, ok := d.LinkManagers.Load(user.Email); !ok {
|
||||
lm = &LinkManager{
|
||||
links: make(map[*ManagedWriter]buf.Reader),
|
||||
}
|
||||
d.LinkManagers.Store(user.Email, lm)
|
||||
} else {
|
||||
lm = lmloaded.(*LinkManager)
|
||||
}
|
||||
managedWriter := &ManagedWriter{
|
||||
writer: uplinkWriter,
|
||||
email: user.Email,
|
||||
manager: d.Wm,
|
||||
manager: lm,
|
||||
}
|
||||
d.Wm.AddWriter(managedWriter)
|
||||
lm.AddLink(managedWriter, outboundLink.Reader)
|
||||
inboundLink.Writer = managedWriter
|
||||
if w != nil {
|
||||
sessionInbound.CanSpliceCopy = 3
|
||||
inboundLink.Writer = rate.NewRateLimitWriter(inboundLink.Writer, w)
|
||||
outboundLink.Writer = rate.NewRateLimitWriter(outboundLink.Writer, w)
|
||||
}
|
||||
@@ -382,14 +388,22 @@ func (d *DefaultDispatcher) DispatchLink(ctx context.Context, destination net.De
|
||||
common.Interrupt(outbound.Reader)
|
||||
return errors.New("Limited ", user.Email, " by conn or ip")
|
||||
}
|
||||
var lm *LinkManager
|
||||
if lmloaded, ok := d.LinkManagers.Load(user.Email); !ok {
|
||||
lm = &LinkManager{
|
||||
links: make(map[*ManagedWriter]buf.Reader),
|
||||
}
|
||||
d.LinkManagers.Store(user.Email, lm)
|
||||
} else {
|
||||
lm = lmloaded.(*LinkManager)
|
||||
}
|
||||
managedWriter := &ManagedWriter{
|
||||
writer: outbound.Writer,
|
||||
email: user.Email,
|
||||
manager: d.Wm,
|
||||
manager: lm,
|
||||
}
|
||||
d.Wm.AddWriter(managedWriter)
|
||||
outbound.Writer = managedWriter
|
||||
if w != nil {
|
||||
sessionInbound.CanSpliceCopy = 3
|
||||
outbound.Writer = rate.NewRateLimitWriter(outbound.Writer, w)
|
||||
}
|
||||
var t *counter.TrafficCounter
|
||||
@@ -406,6 +420,7 @@ func (d *DefaultDispatcher) DispatchLink(ctx context.Context, destination net.De
|
||||
Reader: &buf.TimeoutWrapperReader{Reader: outbound.Reader},
|
||||
Counter: &ts.UpCounter,
|
||||
}
|
||||
lm.AddLink(managedWriter, outbound.Reader)
|
||||
outbound.Writer = &dispatcher.SizeStatWriter{
|
||||
Counter: downcounter,
|
||||
Writer: outbound.Writer,
|
||||
|
||||
@@ -7,48 +7,9 @@ import (
|
||||
"github.com/xtls/xray-core/common/buf"
|
||||
)
|
||||
|
||||
type WriterManager struct {
|
||||
writers map[string]map[*ManagedWriter]struct{}
|
||||
mu sync.Mutex
|
||||
}
|
||||
|
||||
func (m *WriterManager) AddWriter(writer *ManagedWriter) {
|
||||
m.mu.Lock()
|
||||
defer m.mu.Unlock()
|
||||
if _, exists := m.writers[writer.email]; !exists {
|
||||
m.writers[writer.email] = make(map[*ManagedWriter]struct{})
|
||||
}
|
||||
m.writers[writer.email][writer] = struct{}{}
|
||||
}
|
||||
|
||||
func (m *WriterManager) RemoveWriter(writer *ManagedWriter) {
|
||||
m.mu.Lock()
|
||||
defer m.mu.Unlock()
|
||||
|
||||
if _, exists := m.writers[writer.email]; !exists {
|
||||
return
|
||||
}
|
||||
delete(m.writers[writer.email], writer)
|
||||
}
|
||||
|
||||
func (m *WriterManager) RemoveWritersForUser(email string) {
|
||||
m.mu.Lock()
|
||||
defer m.mu.Unlock()
|
||||
|
||||
if _, exists := m.writers[email]; !exists {
|
||||
return
|
||||
}
|
||||
for writer := range m.writers[email] {
|
||||
delete(m.writers[email], writer)
|
||||
common.Close(writer.writer)
|
||||
}
|
||||
delete(m.writers, email)
|
||||
}
|
||||
|
||||
type ManagedWriter struct {
|
||||
writer buf.Writer
|
||||
email string
|
||||
manager *WriterManager
|
||||
manager *LinkManager
|
||||
}
|
||||
|
||||
func (w *ManagedWriter) WriteMultiBuffer(mb buf.MultiBuffer) error {
|
||||
@@ -59,3 +20,27 @@ func (w *ManagedWriter) Close() error {
|
||||
w.manager.RemoveWriter(w)
|
||||
return common.Close(w.writer)
|
||||
}
|
||||
|
||||
type LinkManager struct {
|
||||
links map[*ManagedWriter]buf.Reader
|
||||
mu sync.Mutex
|
||||
}
|
||||
|
||||
func (m *LinkManager) AddLink(writer *ManagedWriter, reader buf.Reader) {
|
||||
m.mu.Lock()
|
||||
defer m.mu.Unlock()
|
||||
m.links[writer] = reader
|
||||
}
|
||||
|
||||
func (m *LinkManager) RemoveWriter(writer *ManagedWriter) {
|
||||
m.mu.Lock()
|
||||
defer m.mu.Unlock()
|
||||
delete(m.links, writer)
|
||||
}
|
||||
|
||||
func (m *LinkManager) CloseAll() {
|
||||
for w, r := range m.links {
|
||||
common.Close(w)
|
||||
common.Interrupt(r)
|
||||
}
|
||||
}
|
||||
|
||||
@@ -8,6 +8,7 @@ import (
|
||||
"github.com/InazumaV/V2bX/common/counter"
|
||||
"github.com/InazumaV/V2bX/common/format"
|
||||
vCore "github.com/InazumaV/V2bX/core"
|
||||
"github.com/InazumaV/V2bX/core/xray/app/dispatcher"
|
||||
"github.com/xtls/xray-core/common/protocol"
|
||||
"github.com/xtls/xray-core/proxy"
|
||||
)
|
||||
@@ -43,8 +44,15 @@ func (c *Xray) DelUsers(users []panel.UserInfo, tag string, _ *panel.NodeInfo) e
|
||||
return err
|
||||
}
|
||||
delete(c.users.uidMap, user)
|
||||
c.dispatcher.Counter.Delete(user)
|
||||
c.dispatcher.Wm.RemoveWritersForUser(user)
|
||||
if v, ok := c.dispatcher.Counter.Load(tag); ok {
|
||||
tc := v.(*counter.TrafficCounter)
|
||||
tc.Delete(user)
|
||||
}
|
||||
if v, ok := c.dispatcher.LinkManagers.Load(user); ok {
|
||||
lm := v.(*dispatcher.LinkManager)
|
||||
lm.CloseAll()
|
||||
c.dispatcher.LinkManagers.Delete(user)
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
4
go.mod
4
go.mod
@@ -19,7 +19,7 @@ require (
|
||||
github.com/spf13/cobra v1.9.1
|
||||
github.com/spf13/viper v1.19.0
|
||||
github.com/vmihailenco/msgpack/v5 v5.4.1
|
||||
github.com/xtls/xray-core v1.250905.0
|
||||
github.com/xtls/xray-core v1.250911.1-0.20250911134820-83c5370eec69
|
||||
go.uber.org/zap v1.27.0
|
||||
golang.org/x/crypto v0.42.0
|
||||
golang.org/x/sys v0.36.0
|
||||
@@ -333,3 +333,5 @@ require (
|
||||
)
|
||||
|
||||
replace github.com/sagernet/sing-box v1.13.0 => github.com/wyx2685/sing-box_mod v1.13.0-alpha.5
|
||||
|
||||
replace github.com/xtls/xray-core v1.250911.1-0.20250911134820-83c5370eec69 => github.com/wyx2685/xray-core v0.0.0-20250912060919-a74bf884128d
|
||||
|
||||
4
go.sum
4
go.sum
@@ -1132,6 +1132,8 @@ github.com/wyx2685/sing-box_mod v1.13.0-alpha.5 h1:9aG8iDQgqCict6zhMsJU9O6rKQr17
|
||||
github.com/wyx2685/sing-box_mod v1.13.0-alpha.5/go.mod h1:HJYV1YYy7ucy2R4AotcCCQDxhc1tAMvq8c0VUqcoGZg=
|
||||
github.com/wyx2685/sing-vmess v0.0.0-20250723121437-95d5ab59ff92 h1:A5/G1mY1M3h+j6D3BgfRlngbsfTUGwE8S3UuyR9VfbU=
|
||||
github.com/wyx2685/sing-vmess v0.0.0-20250723121437-95d5ab59ff92/go.mod h1:ZvICtYzgNgbRy2fgKTVmG4o+6pP92dzY5GhwSlfya8Q=
|
||||
github.com/wyx2685/xray-core v0.0.0-20250912060919-a74bf884128d h1:5OM2F962gkzHZe+yk2fM1yeokiAetC884mefA8rgrCI=
|
||||
github.com/wyx2685/xray-core v0.0.0-20250912060919-a74bf884128d/go.mod h1:pOFdaOfxViLRo2izXjXYn10akJZpbemMWm8yh0jQEgc=
|
||||
github.com/x448/float16 v0.8.4 h1:qLwI1I70+NjRFUR3zs1JPUCgaCXSh3SW62uAKT1mSBM=
|
||||
github.com/x448/float16 v0.8.4/go.mod h1:14CWIYCyZA/cWjXOioeEpHeN/83MdbZDRQHoFcYsOfg=
|
||||
github.com/xdg-go/pbkdf2 v1.0.0/go.mod h1:jrpuAogTd400dnrH08LKmI/xc1MbPOebTwRqcT5RDeI=
|
||||
@@ -1145,8 +1147,6 @@ github.com/xeipuuv/gojsonschema v1.2.0/go.mod h1:anYRn/JVcOK2ZgGU+IjEV4nwlhoK5sQ
|
||||
github.com/xiang90/probing v0.0.0-20190116061207-43a291ad63a2/go.mod h1:UETIi67q53MR2AWcXfiuqkDkRtnGDLqkBTpCHuJHxtU=
|
||||
github.com/xtls/reality v0.0.0-20250904214705-431b6ff8c67c h1:LHLhQY3mKXSpTcQAkjFR4/6ar3rXjQryNeM7khK3AHU=
|
||||
github.com/xtls/reality v0.0.0-20250904214705-431b6ff8c67c/go.mod h1:XxvnCCgBee4WWE0bc4E+a7wbk8gkJ/rS0vNVNtC5qp0=
|
||||
github.com/xtls/xray-core v1.250905.0 h1:VNL3l/6fcwyeYXJTRbf+TYqPfJYkk0Wmmz7qoQNkxY8=
|
||||
github.com/xtls/xray-core v1.250905.0/go.mod h1:WB/73DmN9Vs7lxtx4Xc/D0Ub1VUu06hAh1mMh8JN2uM=
|
||||
github.com/yandex-cloud/go-genproto v0.14.0 h1:yDqD260mICkjodXyAaDhESfrLr6gIGwwRc9MYE0jvW0=
|
||||
github.com/yandex-cloud/go-genproto v0.14.0/go.mod h1:0LDD/IZLIUIV4iPH+YcF+jysO3jkSvADFGm4dCAuwQo=
|
||||
github.com/yandex-cloud/go-sdk/services/dns v0.0.3 h1:erphTBXKSpm/tETa6FXrw4niSHjhySzAKHUc2/BZKx0=
|
||||
|
||||
Reference in New Issue
Block a user