sing-box/box.go

240 lines
5.9 KiB
Go
Raw Normal View History

2022-06-30 13:27:56 +00:00
package box
import (
"context"
2022-07-12 07:17:29 +00:00
"io"
"os"
2022-07-04 11:34:45 +00:00
"time"
2022-06-30 13:27:56 +00:00
"github.com/sagernet/sing-box/adapter"
2022-07-19 23:12:40 +00:00
"github.com/sagernet/sing-box/experimental"
"github.com/sagernet/sing-box/inbound"
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"
"github.com/sagernet/sing-box/outbound"
"github.com/sagernet/sing-box/route"
2022-07-08 15:03:57 +00:00
"github.com/sagernet/sing/common"
E "github.com/sagernet/sing/common/exceptions"
F "github.com/sagernet/sing/common/format"
2022-06-30 13:27:56 +00:00
)
2022-07-07 13:47:21 +00:00
var _ adapter.Service = (*Box)(nil)
2022-06-30 13:27:56 +00:00
2022-07-07 13:47:21 +00:00
type Box struct {
2022-07-19 14:16:49 +00:00
createdAt time.Time
router adapter.Router
inbounds []adapter.Inbound
outbounds []adapter.Outbound
logFactory log.Factory
logger log.ContextLogger
logFile *os.File
2022-07-19 23:12:40 +00:00
clashServer adapter.ClashServer
2022-08-12 04:13:57 +00:00
done chan struct{}
2022-06-30 13:27:56 +00:00
}
2022-07-07 13:47:21 +00:00
func New(ctx context.Context, options option.Options) (*Box, error) {
2022-07-04 11:34:45 +00:00
createdAt := time.Now()
2022-07-12 07:17:29 +00:00
logOptions := common.PtrValueOrDefault(options.Log)
2022-07-19 14:16:49 +00:00
var needClashAPI bool
if options.Experimental != nil && options.Experimental.ClashAPI != nil && options.Experimental.ClashAPI.ExternalController != "" {
needClashAPI = true
}
2022-07-12 07:17:29 +00:00
var logFactory log.Factory
2022-07-19 14:16:49 +00:00
var observableLogFactory log.ObservableFactory
2022-07-12 07:17:29 +00:00
var logFile *os.File
if logOptions.Disabled {
2022-07-20 01:41:44 +00:00
observableLogFactory = log.NewNOPFactory()
logFactory = observableLogFactory
2022-07-12 07:17:29 +00:00
} else {
var logWriter io.Writer
switch logOptions.Output {
case "", "stderr":
logWriter = os.Stderr
case "stdout":
logWriter = os.Stdout
default:
var err error
logFile, err = os.OpenFile(logOptions.Output, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0o644)
if err != nil {
return nil, err
}
2022-08-18 23:50:19 +00:00
logWriter = logFile
2022-07-12 07:17:29 +00:00
}
logFormatter := log.Formatter{
BaseTime: createdAt,
DisableColors: logOptions.DisableColor || logFile != nil,
DisableTimestamp: !logOptions.Timestamp && logFile != nil,
FullTimestamp: logOptions.Timestamp,
TimestampFormat: "-0700 2006-01-02 15:04:05",
}
2022-07-19 14:16:49 +00:00
if needClashAPI {
observableLogFactory = log.NewObservableFactory(logFormatter, logWriter)
logFactory = observableLogFactory
} else {
logFactory = log.NewFactory(logFormatter, logWriter)
}
2022-07-18 04:32:31 +00:00
if logOptions.Level != "" {
logLevel, err := log.ParseLevel(logOptions.Level)
if err != nil {
return nil, E.Cause(err, "parse log level")
}
logFactory.SetLevel(logLevel)
} else {
logFactory.SetLevel(log.LevelTrace)
}
2022-06-30 13:27:56 +00:00
}
2022-07-12 07:17:29 +00:00
router, err := route.NewRouter(
ctx,
logFactory.NewLogger("router"),
logFactory.NewLogger("dns"),
common.PtrValueOrDefault(options.Route),
common.PtrValueOrDefault(options.DNS),
2022-08-04 14:01:20 +00:00
options.Inbounds,
2022-07-12 07:17:29 +00:00
)
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 in adapter.Inbound
2022-07-12 07:17:29 +00:00
var tag string
if inboundOptions.Tag != "" {
tag = inboundOptions.Tag
} else {
tag = F.ToString(i)
}
in, err = inbound.New(
ctx,
router,
logFactory.NewLogger(F.ToString("inbound/", inboundOptions.Type, "[", tag, "]")),
inboundOptions,
)
2022-07-02 14:55:10 +00:00
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
}
inbounds = append(inbounds, in)
2022-06-30 13:27:56 +00:00
}
for i, outboundOptions := range options.Outbounds {
var out adapter.Outbound
2022-07-12 07:17:29 +00:00
var tag string
if outboundOptions.Tag != "" {
tag = outboundOptions.Tag
} else {
tag = F.ToString(i)
}
out, err = outbound.New(
2022-07-29 16:29:22 +00:00
ctx,
2022-07-12 07:17:29 +00:00
router,
logFactory.NewLogger(F.ToString("outbound/", outboundOptions.Type, "[", tag, "]")),
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
}
outbounds = append(outbounds, out)
2022-07-02 06:07:50 +00:00
}
err = router.Initialize(outbounds, func() adapter.Outbound {
2022-07-29 16:29:22 +00:00
out, oErr := outbound.New(ctx, router, logFactory.NewLogger("outbound/direct"), option.Outbound{Type: "direct", Tag: "default"})
common.Must(oErr)
outbounds = append(outbounds, out)
return out
})
if err != nil {
return nil, err
2022-06-30 13:27:56 +00:00
}
2022-07-19 14:16:49 +00:00
2022-07-19 23:12:40 +00:00
var clashServer adapter.ClashServer
2022-07-19 14:16:49 +00:00
if needClashAPI {
2022-07-19 23:12:40 +00:00
clashServer, err = experimental.NewClashServer(router, observableLogFactory, common.PtrValueOrDefault(options.Experimental.ClashAPI))
if err != nil {
return nil, E.Cause(err, "create clash api server")
}
2022-07-19 14:16:49 +00:00
router.SetTrafficController(clashServer)
}
2022-07-07 13:47:21 +00:00
return &Box{
2022-07-19 14:16:49 +00:00
router: router,
inbounds: inbounds,
outbounds: outbounds,
createdAt: createdAt,
logFactory: logFactory,
logger: logFactory.NewLogger(""),
logFile: logFile,
clashServer: clashServer,
2022-08-12 04:13:57 +00:00
done: make(chan struct{}),
2022-07-02 06:07:50 +00:00
}, nil
2022-06-30 13:27:56 +00:00
}
2022-07-07 13:47:21 +00:00
func (s *Box) Start() error {
2022-08-16 15:37:51 +00:00
for i, out := range s.outbounds {
if starter, isStarter := out.(common.Starter); isStarter {
2022-08-20 01:13:00 +00:00
err := starter.Start()
2022-08-16 15:37:51 +00:00
if err != nil {
for _, in := range s.inbounds {
common.Close(in)
}
for g := 0; g < i; g++ {
common.Close(s.outbounds[g])
}
var tag string
if out.Tag() == "" {
tag = F.ToString(i)
} else {
tag = out.Tag()
}
return E.Cause(err, "initialize outbound/", out.Type(), "[", tag, "]")
}
2022-06-30 13:27:56 +00:00
}
}
2022-08-20 01:13:00 +00:00
err := s.router.Start()
if err != nil {
return err
}
for i, in := range s.inbounds {
err = in.Start()
if err != nil {
for g := 0; g < i; g++ {
s.inbounds[g].Close()
}
var tag string
if in.Tag() == "" {
tag = F.ToString(i)
} else {
tag = in.Tag()
}
return E.Cause(err, "initialize inbound/", in.Type(), "[", tag, "]")
}
}
2022-07-19 14:16:49 +00:00
if s.clashServer != nil {
err = s.clashServer.Start()
if err != nil {
2022-07-19 23:12:40 +00:00
return E.Cause(err, "start clash api server")
2022-07-19 14:16:49 +00:00
}
}
2022-07-04 11:34:45 +00:00
s.logger.Info("sing-box started (", F.Seconds(time.Since(s.createdAt).Seconds()), "s)")
return nil
2022-06-30 13:27:56 +00:00
}
2022-07-07 13:47:21 +00:00
func (s *Box) Close() error {
2022-08-12 04:13:57 +00:00
select {
case <-s.done:
return os.ErrClosed
default:
close(s.done)
}
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,
2022-07-30 06:50:33 +00:00
s.logFactory,
2022-07-19 23:12:40 +00:00
s.clashServer,
2022-07-12 07:17:29 +00:00
common.PtrOrNil(s.logFile),
2022-07-02 17:57:04 +00:00
)
2022-06-30 13:27:56 +00:00
}