chunyu/sing-box
1
0
Fork 0
mirror of https://github.com/SagerNet/sing-box.git synced 2025-06-13 05:44:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix may pick nil outbound if fallback not set

Browse Source
This commit is contained in:
jebbs 2022-10-10 16:40:11 +08:00
parent ebe5a80f19
commit 922c839c13
1 changed files with 11 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
outbound/balancer.go
View File

@ -90,6 +90,16 @@ func (s *Balancer) NewPacketConnection(ctx context.Context, conn N.PacketConn, m
// initialize inits the balancer
func (s *Balancer) initialize() error {
// the fallback is required, in case that all outbounds are not available,
// we can pick it instead of returning nil to avoid panic.
if s.fallbackTag == "" {
return E.New("fallback not set")
}
outbound, loaded := s.router.Outbound(s.fallbackTag)
if !loaded {
return E.New("fallback outbound not found: ", s.fallbackTag)
}
s.fallback = outbound
for i, tag := range s.tags {
outbound, loaded := s.router.Outbound(tag)
if !loaded {
@ -97,13 +107,6 @@ func (s *Balancer) initialize() error {
}
s.nodes = append(s.nodes, balancer.NewNode(outbound))
}
if s.fallbackTag != "" {
outbound, loaded := s.router.Outbound(s.fallbackTag)
if !loaded {
return E.New("fallback outbound not found: ", s.fallbackTag)
}
s.fallback = outbound
}
return nil
}
@ -129,7 +132,7 @@ func (s *Balancer) pick() adapter.Outbound {
// not started
count := len(s.nodes)
if count == 0 {
// goes to fallbackTag
// goes to fallback
return s.fallback
}
picked := s.nodes[rand.Intn(count)]