mirror of
https://github.com/SagerNet/sing-box.git
synced 2024-11-22 08:31:30 +00:00
Migrate ntp service to library
This commit is contained in:
parent
11c7b4a866
commit
b5dcd6bf59
112
ntp/service.go
112
ntp/service.go
|
@ -1,112 +0,0 @@
|
||||||
package ntp
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
"os"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"github.com/sagernet/sing-box/adapter"
|
|
||||||
"github.com/sagernet/sing-box/common/dialer"
|
|
||||||
"github.com/sagernet/sing-box/common/settings"
|
|
||||||
C "github.com/sagernet/sing-box/constant"
|
|
||||||
"github.com/sagernet/sing-box/option"
|
|
||||||
"github.com/sagernet/sing/common"
|
|
||||||
E "github.com/sagernet/sing/common/exceptions"
|
|
||||||
"github.com/sagernet/sing/common/logger"
|
|
||||||
M "github.com/sagernet/sing/common/metadata"
|
|
||||||
N "github.com/sagernet/sing/common/network"
|
|
||||||
"github.com/sagernet/sing/common/ntp"
|
|
||||||
)
|
|
||||||
|
|
||||||
var _ ntp.TimeService = (*Service)(nil)
|
|
||||||
|
|
||||||
type Service struct {
|
|
||||||
ctx context.Context
|
|
||||||
cancel common.ContextCancelCauseFunc
|
|
||||||
server M.Socksaddr
|
|
||||||
writeToSystem bool
|
|
||||||
dialer N.Dialer
|
|
||||||
logger logger.Logger
|
|
||||||
ticker *time.Ticker
|
|
||||||
clockOffset time.Duration
|
|
||||||
}
|
|
||||||
|
|
||||||
func NewService(ctx context.Context, router adapter.Router, logger logger.Logger, options option.NTPOptions) (*Service, error) {
|
|
||||||
ctx, cancel := common.ContextWithCancelCause(ctx)
|
|
||||||
server := M.ParseSocksaddrHostPort(options.Server, options.ServerPort)
|
|
||||||
if server.Port == 0 {
|
|
||||||
server.Port = 123
|
|
||||||
}
|
|
||||||
var interval time.Duration
|
|
||||||
if options.Interval > 0 {
|
|
||||||
interval = time.Duration(options.Interval)
|
|
||||||
} else {
|
|
||||||
interval = 30 * time.Minute
|
|
||||||
}
|
|
||||||
outboundDialer, err := dialer.New(router, options.DialerOptions)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
return &Service{
|
|
||||||
ctx: ctx,
|
|
||||||
cancel: cancel,
|
|
||||||
server: server,
|
|
||||||
writeToSystem: options.WriteToSystem,
|
|
||||||
dialer: outboundDialer,
|
|
||||||
logger: logger,
|
|
||||||
ticker: time.NewTicker(interval),
|
|
||||||
}, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *Service) Start() error {
|
|
||||||
err := s.update()
|
|
||||||
if err != nil {
|
|
||||||
return E.Cause(err, "initialize time")
|
|
||||||
}
|
|
||||||
s.logger.Info("updated time: ", s.TimeFunc()().Local().Format(C.TimeLayout))
|
|
||||||
go s.loopUpdate()
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *Service) Close() error {
|
|
||||||
s.ticker.Stop()
|
|
||||||
s.cancel(os.ErrClosed)
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *Service) TimeFunc() func() time.Time {
|
|
||||||
return func() time.Time {
|
|
||||||
return time.Now().Add(s.clockOffset)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *Service) loopUpdate() {
|
|
||||||
for {
|
|
||||||
select {
|
|
||||||
case <-s.ctx.Done():
|
|
||||||
return
|
|
||||||
case <-s.ticker.C:
|
|
||||||
}
|
|
||||||
err := s.update()
|
|
||||||
if err == nil {
|
|
||||||
s.logger.Debug("updated time: ", s.TimeFunc()().Local().Format(C.TimeLayout))
|
|
||||||
} else {
|
|
||||||
s.logger.Warn("update time: ", err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *Service) update() error {
|
|
||||||
response, err := ntp.Exchange(s.ctx, s.dialer, s.server)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
s.clockOffset = response.ClockOffset
|
|
||||||
if s.writeToSystem {
|
|
||||||
writeErr := settings.SetSystemTime(s.TimeFunc()())
|
|
||||||
if writeErr != nil {
|
|
||||||
s.logger.Warn("write time to system: ", writeErr)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
|
@ -2,9 +2,8 @@ package option
|
||||||
|
|
||||||
type NTPOptions struct {
|
type NTPOptions struct {
|
||||||
Enabled bool `json:"enabled,omitempty"`
|
Enabled bool `json:"enabled,omitempty"`
|
||||||
Server string `json:"server,omitempty"`
|
|
||||||
ServerPort uint16 `json:"server_port,omitempty"`
|
|
||||||
Interval Duration `json:"interval,omitempty"`
|
Interval Duration `json:"interval,omitempty"`
|
||||||
WriteToSystem bool `json:"write_to_system,omitempty"`
|
WriteToSystem bool `json:"write_to_system,omitempty"`
|
||||||
|
ServerOptions
|
||||||
DialerOptions
|
DialerOptions
|
||||||
}
|
}
|
||||||
|
|
|
@ -23,7 +23,6 @@ import (
|
||||||
C "github.com/sagernet/sing-box/constant"
|
C "github.com/sagernet/sing-box/constant"
|
||||||
"github.com/sagernet/sing-box/experimental/libbox/platform"
|
"github.com/sagernet/sing-box/experimental/libbox/platform"
|
||||||
"github.com/sagernet/sing-box/log"
|
"github.com/sagernet/sing-box/log"
|
||||||
"github.com/sagernet/sing-box/ntp"
|
|
||||||
"github.com/sagernet/sing-box/option"
|
"github.com/sagernet/sing-box/option"
|
||||||
"github.com/sagernet/sing-box/outbound"
|
"github.com/sagernet/sing-box/outbound"
|
||||||
"github.com/sagernet/sing-box/transport/fakeip"
|
"github.com/sagernet/sing-box/transport/fakeip"
|
||||||
|
@ -40,7 +39,7 @@ import (
|
||||||
F "github.com/sagernet/sing/common/format"
|
F "github.com/sagernet/sing/common/format"
|
||||||
M "github.com/sagernet/sing/common/metadata"
|
M "github.com/sagernet/sing/common/metadata"
|
||||||
N "github.com/sagernet/sing/common/network"
|
N "github.com/sagernet/sing/common/network"
|
||||||
serviceNTP "github.com/sagernet/sing/common/ntp"
|
"github.com/sagernet/sing/common/ntp"
|
||||||
"github.com/sagernet/sing/common/task"
|
"github.com/sagernet/sing/common/task"
|
||||||
"github.com/sagernet/sing/common/uot"
|
"github.com/sagernet/sing/common/uot"
|
||||||
"github.com/sagernet/sing/common/winpowrprof"
|
"github.com/sagernet/sing/common/winpowrprof"
|
||||||
|
@ -333,11 +332,19 @@ func NewRouter(
|
||||||
}
|
}
|
||||||
|
|
||||||
if ntpOptions.Enabled {
|
if ntpOptions.Enabled {
|
||||||
timeService, err := ntp.NewService(ctx, router, logFactory.NewLogger("ntp"), ntpOptions)
|
ntpDialer, err := dialer.New(router, ntpOptions.DialerOptions)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, E.Cause(err, "create NTP service")
|
||||||
}
|
}
|
||||||
service.ContextWith[serviceNTP.TimeService](ctx, timeService)
|
timeService := ntp.NewService(ntp.Options{
|
||||||
|
Context: ctx,
|
||||||
|
Dialer: ntpDialer,
|
||||||
|
Logger: logFactory.NewLogger("ntp"),
|
||||||
|
Server: ntpOptions.ServerOptions.Build(),
|
||||||
|
Interval: time.Duration(ntpOptions.Interval),
|
||||||
|
WriteToSystem: ntpOptions.WriteToSystem,
|
||||||
|
})
|
||||||
|
service.MustRegister[ntp.TimeService](ctx, timeService)
|
||||||
router.timeService = timeService
|
router.timeService = timeService
|
||||||
}
|
}
|
||||||
return router, nil
|
return router, nil
|
||||||
|
|
Loading…
Reference in a new issue