mirror of
https://github.com/SagerNet/sing-box.git
synced 2024-11-14 12:53:17 +00:00
102 lines
2.2 KiB
Go
102 lines
2.2 KiB
Go
package libbox
|
|
|
|
import (
|
|
"encoding/binary"
|
|
"net"
|
|
"os"
|
|
"path/filepath"
|
|
"sync"
|
|
|
|
"github.com/sagernet/sing-box/log"
|
|
"github.com/sagernet/sing/common"
|
|
E "github.com/sagernet/sing/common/exceptions"
|
|
"github.com/sagernet/sing/common/observable"
|
|
"github.com/sagernet/sing/common/x/list"
|
|
)
|
|
|
|
type CommandServer struct {
|
|
sockPath string
|
|
listener net.Listener
|
|
handler CommandServerHandler
|
|
|
|
access sync.Mutex
|
|
savedLines *list.List[string]
|
|
subscriber *observable.Subscriber[string]
|
|
observer *observable.Observer[string]
|
|
}
|
|
|
|
type CommandServerHandler interface {
|
|
ServiceStop() error
|
|
ServiceReload() error
|
|
}
|
|
|
|
func NewCommandServer(sharedDirectory string, handler CommandServerHandler) *CommandServer {
|
|
server := &CommandServer{
|
|
sockPath: filepath.Join(sharedDirectory, "command.sock"),
|
|
handler: handler,
|
|
savedLines: new(list.List[string]),
|
|
subscriber: observable.NewSubscriber[string](128),
|
|
}
|
|
server.observer = observable.NewObserver[string](server.subscriber, 64)
|
|
return server
|
|
}
|
|
|
|
func (s *CommandServer) Start() error {
|
|
os.Remove(s.sockPath)
|
|
listener, err := net.ListenUnix("unix", &net.UnixAddr{
|
|
Name: s.sockPath,
|
|
Net: "unix",
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
s.listener = listener
|
|
go s.loopConnection(listener)
|
|
return nil
|
|
}
|
|
|
|
func (s *CommandServer) Close() error {
|
|
return common.Close(
|
|
s.listener,
|
|
s.observer,
|
|
)
|
|
}
|
|
|
|
func (s *CommandServer) loopConnection(listener net.Listener) {
|
|
for {
|
|
conn, err := listener.Accept()
|
|
if err != nil {
|
|
return
|
|
}
|
|
go func() {
|
|
hErr := s.handleConnection(conn)
|
|
if hErr != nil && !E.IsClosed(err) {
|
|
log.Warn("log-server: process connection: ", hErr)
|
|
}
|
|
}()
|
|
}
|
|
}
|
|
|
|
func (s *CommandServer) handleConnection(conn net.Conn) error {
|
|
defer conn.Close()
|
|
var command uint8
|
|
err := binary.Read(conn, binary.BigEndian, &command)
|
|
if err != nil {
|
|
return E.Cause(err, "read command")
|
|
}
|
|
switch int32(command) {
|
|
case CommandLog:
|
|
return s.handleLogConn(conn)
|
|
case CommandStatus:
|
|
return s.handleStatusConn(conn)
|
|
case CommandServiceStop:
|
|
return s.handleServiceStop(conn)
|
|
case CommandServiceReload:
|
|
return s.handleServiceReload(conn)
|
|
case CommandCloseConnections:
|
|
return s.handleCloseConnections(conn)
|
|
default:
|
|
return E.New("unknown command: ", command)
|
|
}
|
|
}
|