mirror of
https://github.com/SagerNet/sing-box.git
synced 2025-02-16 14:24:31 +00:00
platform: Add sleep support for NetworkExtension
This commit is contained in:
parent
8d629ef323
commit
1983f54907
43
common/sleep/manager.go
Normal file
43
common/sleep/manager.go
Normal file
|
@ -0,0 +1,43 @@
|
|||
package sleep
|
||||
|
||||
import (
|
||||
"sync"
|
||||
)
|
||||
|
||||
type Manager struct {
|
||||
access sync.Mutex
|
||||
done chan struct{}
|
||||
}
|
||||
|
||||
func NewManager() *Manager {
|
||||
closedChan := make(chan struct{})
|
||||
close(closedChan)
|
||||
return &Manager{
|
||||
done: closedChan,
|
||||
}
|
||||
}
|
||||
|
||||
func (m *Manager) Sleep() {
|
||||
m.access.Lock()
|
||||
defer m.access.Unlock()
|
||||
select {
|
||||
case <-m.done:
|
||||
default:
|
||||
return
|
||||
}
|
||||
m.done = make(chan struct{})
|
||||
}
|
||||
|
||||
func (m *Manager) Wake() {
|
||||
m.access.Lock()
|
||||
defer m.access.Unlock()
|
||||
select {
|
||||
case <-m.done:
|
||||
default:
|
||||
close(m.done)
|
||||
}
|
||||
}
|
||||
|
||||
func (m *Manager) Active() <-chan struct{} {
|
||||
return m.done
|
||||
}
|
|
@ -8,6 +8,7 @@ import (
|
|||
"github.com/sagernet/sing-box"
|
||||
"github.com/sagernet/sing-box/adapter"
|
||||
"github.com/sagernet/sing-box/common/process"
|
||||
"github.com/sagernet/sing-box/common/sleep"
|
||||
"github.com/sagernet/sing-box/common/urltest"
|
||||
"github.com/sagernet/sing-box/experimental/libbox/internal/procfs"
|
||||
"github.com/sagernet/sing-box/experimental/libbox/platform"
|
||||
|
@ -25,6 +26,7 @@ type BoxService struct {
|
|||
ctx context.Context
|
||||
cancel context.CancelFunc
|
||||
instance *box.Box
|
||||
sleepManager *sleep.Manager
|
||||
}
|
||||
|
||||
func NewService(configContent string, platformInterface PlatformInterface) (*BoxService, error) {
|
||||
|
@ -35,6 +37,8 @@ func NewService(configContent string, platformInterface PlatformInterface) (*Box
|
|||
ctx, cancel := context.WithCancel(context.Background())
|
||||
ctx = filemanager.WithDefault(ctx, sWorkingPath, sTempPath, sUserID, sGroupID)
|
||||
ctx = service.ContextWithPtr(ctx, urltest.NewHistoryStorage())
|
||||
sleepManager := sleep.NewManager()
|
||||
ctx = service.ContextWithPtr(ctx, sleepManager)
|
||||
instance, err := box.New(box.Options{
|
||||
Context: ctx,
|
||||
Options: options,
|
||||
|
@ -48,6 +52,7 @@ func NewService(configContent string, platformInterface PlatformInterface) (*Box
|
|||
ctx: ctx,
|
||||
cancel: cancel,
|
||||
instance: instance,
|
||||
sleepManager: sleepManager,
|
||||
}, nil
|
||||
}
|
||||
|
||||
|
@ -60,6 +65,15 @@ func (s *BoxService) Close() error {
|
|||
return s.instance.Close()
|
||||
}
|
||||
|
||||
func (s *BoxService) Sleep() {
|
||||
s.sleepManager.Sleep()
|
||||
_ = s.instance.Router().ResetNetwork()
|
||||
}
|
||||
|
||||
func (s *BoxService) Wake() {
|
||||
s.sleepManager.Wake()
|
||||
}
|
||||
|
||||
var _ platform.Interface = (*platformInterfaceWrapper)(nil)
|
||||
|
||||
type platformInterfaceWrapper struct {
|
||||
|
|
|
@ -8,6 +8,7 @@ import (
|
|||
"time"
|
||||
|
||||
"github.com/sagernet/sing-box/adapter"
|
||||
"github.com/sagernet/sing-box/common/sleep"
|
||||
"github.com/sagernet/sing-box/common/urltest"
|
||||
C "github.com/sagernet/sing-box/constant"
|
||||
"github.com/sagernet/sing-box/log"
|
||||
|
@ -152,6 +153,7 @@ type URLTestGroup struct {
|
|||
tolerance uint16
|
||||
history *urltest.HistoryStorage
|
||||
checking atomic.Bool
|
||||
sleepManager *sleep.Manager
|
||||
|
||||
access sync.Mutex
|
||||
ticker *time.Ticker
|
||||
|
@ -182,6 +184,7 @@ func NewURLTestGroup(ctx context.Context, router adapter.Router, logger log.Logg
|
|||
tolerance: tolerance,
|
||||
history: history,
|
||||
close: make(chan struct{}),
|
||||
sleepManager: service.PtrFromContext[sleep.Manager](ctx),
|
||||
}
|
||||
}
|
||||
|
||||
|
@ -263,6 +266,9 @@ func (g *URLTestGroup) Fallback(used adapter.Outbound) []adapter.Outbound {
|
|||
func (g *URLTestGroup) loopCheck() {
|
||||
go g.CheckOutbounds(true)
|
||||
for {
|
||||
if g.sleepManager != nil {
|
||||
<-g.sleepManager.Active()
|
||||
}
|
||||
select {
|
||||
case <-g.close:
|
||||
return
|
||||
|
|
Loading…
Reference in a new issue