sing-box/service.go

92 lines
2.3 KiB
Go
Raw Normal View History

2022-06-30 13:27:56 +00:00
package box
import (
"context"
"github.com/sagernet/sing-box/adapter"
2022-07-01 11:34:02 +00:00
"github.com/sagernet/sing-box/adapter/inbound"
"github.com/sagernet/sing-box/adapter/outbound"
"github.com/sagernet/sing-box/adapter/route"
2022-06-30 13:27:56 +00:00
"github.com/sagernet/sing-box/log"
2022-07-02 06:07:50 +00:00
"github.com/sagernet/sing-box/option"
2022-06-30 13:27:56 +00:00
"github.com/sagernet/sing/common"
2022-07-02 17:57:04 +00:00
E "github.com/sagernet/sing/common/exceptions"
2022-06-30 13:27:56 +00:00
)
var _ adapter.Service = (*Service)(nil)
type Service struct {
2022-07-02 06:07:50 +00:00
router adapter.Router
logger log.Logger
2022-06-30 13:27:56 +00:00
inbounds []adapter.Inbound
outbounds []adapter.Outbound
}
2022-07-02 14:55:10 +00:00
func NewService(ctx context.Context, options option.Options) (*Service, error) {
logger, err := log.NewLogger(common.PtrValueOrDefault(options.Log))
if err != nil {
2022-07-02 17:57:04 +00:00
return nil, E.Cause(err, "parse log options")
2022-06-30 13:27:56 +00:00
}
2022-07-02 14:55:10 +00:00
router, err := route.NewRouter(ctx, logger, common.PtrValueOrDefault(options.Route))
2022-07-02 06:07:50 +00:00
if err != nil {
2022-07-02 17:57:04 +00:00
return nil, E.Cause(err, "parse route options")
2022-06-30 13:27:56 +00:00
}
2022-07-02 14:55:10 +00:00
inbounds := make([]adapter.Inbound, 0, len(options.Inbounds))
outbounds := make([]adapter.Outbound, 0, len(options.Outbounds))
for i, inboundOptions := range options.Inbounds {
var inboundService adapter.Inbound
inboundService, err = inbound.New(ctx, router, logger, i, inboundOptions)
if err != nil {
2022-07-02 17:57:04 +00:00
return nil, E.Cause(err, "parse inbound[", i, "]")
2022-06-30 13:27:56 +00:00
}
2022-07-02 14:55:10 +00:00
inbounds = append(inbounds, inboundService)
2022-06-30 13:27:56 +00:00
}
for i, outboundOptions := range options.Outbounds {
2022-07-02 06:07:50 +00:00
var outboundService adapter.Outbound
outboundService, err = outbound.New(router, logger, i, outboundOptions)
2022-06-30 13:27:56 +00:00
if err != nil {
2022-07-02 17:57:04 +00:00
return nil, E.Cause(err, "parse outbound[", i, "]")
2022-06-30 13:27:56 +00:00
}
2022-07-02 06:07:50 +00:00
outbounds = append(outbounds, outboundService)
}
if len(outbounds) == 0 {
2022-07-02 14:55:10 +00:00
outbounds = append(outbounds, outbound.NewDirect(nil, logger, "direct", option.DirectOutboundOptions{}))
2022-06-30 13:27:56 +00:00
}
2022-07-02 06:07:50 +00:00
router.UpdateOutbounds(outbounds)
return &Service{
router: router,
logger: logger,
inbounds: inbounds,
outbounds: outbounds,
}, nil
2022-06-30 13:27:56 +00:00
}
func (s *Service) Start() error {
2022-07-02 17:57:04 +00:00
err := s.logger.Start()
if err != nil {
return err
}
2022-07-02 06:07:50 +00:00
for _, in := range s.inbounds {
2022-07-02 17:57:04 +00:00
err = in.Start()
2022-06-30 13:27:56 +00:00
if err != nil {
return err
}
}
2022-07-02 17:57:04 +00:00
return common.AnyError(
s.router.Start(),
)
2022-06-30 13:27:56 +00:00
}
func (s *Service) Close() error {
2022-07-02 06:07:50 +00:00
for _, in := range s.inbounds {
in.Close()
2022-06-30 13:27:56 +00:00
}
2022-07-02 06:07:50 +00:00
for _, out := range s.outbounds {
common.Close(out)
2022-06-30 13:27:56 +00:00
}
2022-07-02 17:57:04 +00:00
return common.Close(
s.router,
s.logger,
)
2022-06-30 13:27:56 +00:00
}