command_client.go 3.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143
  1. package libbox
  2. import (
  3. "encoding/binary"
  4. "net"
  5. "os"
  6. "path/filepath"
  7. "time"
  8. "github.com/sagernet/sing/common"
  9. E "github.com/sagernet/sing/common/exceptions"
  10. )
  11. type CommandClient struct {
  12. handler CommandClientHandler
  13. conn net.Conn
  14. options CommandClientOptions
  15. }
  16. type CommandClientOptions struct {
  17. Command int32
  18. StatusInterval int64
  19. IsMainClient bool
  20. }
  21. type CommandClientHandler interface {
  22. Connected()
  23. Disconnected(message string)
  24. ClearLogs()
  25. WriteLogs(messageList StringIterator)
  26. WriteStatus(message *StatusMessage)
  27. OpenURL(url string)
  28. WriteGroups(message OutboundGroupIterator)
  29. InitializeClashMode(modeList StringIterator, currentMode string)
  30. UpdateClashMode(newMode string)
  31. WriteConnections(message *Connections)
  32. }
  33. func NewStandaloneCommandClient() *CommandClient {
  34. return new(CommandClient)
  35. }
  36. func NewCommandClient(handler CommandClientHandler, options *CommandClientOptions) *CommandClient {
  37. return &CommandClient{
  38. handler: handler,
  39. options: common.PtrValueOrDefault(options),
  40. }
  41. }
  42. func (c *CommandClient) directConnect() (net.Conn, error) {
  43. if !sTVOS {
  44. return net.DialUnix("unix", nil, &net.UnixAddr{
  45. Name: filepath.Join(sBasePath, "command.sock"),
  46. Net: "unix",
  47. })
  48. } else {
  49. return net.Dial("tcp", "127.0.0.1:8964")
  50. }
  51. }
  52. func (c *CommandClient) directConnectWithRetry() (net.Conn, error) {
  53. var (
  54. conn net.Conn
  55. err error
  56. )
  57. for i := 0; i < 10; i++ {
  58. conn, err = c.directConnect()
  59. if err == nil {
  60. return conn, nil
  61. }
  62. time.Sleep(time.Duration(100+i*50) * time.Millisecond)
  63. }
  64. return nil, err
  65. }
  66. func (c *CommandClient) Connect() error {
  67. common.Close(c.conn)
  68. conn, err := c.directConnectWithRetry()
  69. if err != nil {
  70. return err
  71. }
  72. c.conn = conn
  73. err = binary.Write(conn, binary.BigEndian, uint8(c.options.Command))
  74. if err != nil {
  75. return err
  76. }
  77. switch c.options.Command {
  78. case CommandLog:
  79. err = binary.Write(conn, binary.BigEndian, c.options.StatusInterval)
  80. if err != nil {
  81. return E.Cause(err, "write interval")
  82. }
  83. c.handler.Connected()
  84. go c.handleLogConn(conn)
  85. case CommandStatus:
  86. err = binary.Write(conn, binary.BigEndian, c.options.IsMainClient)
  87. if err != nil {
  88. return E.Cause(err, "write is main client")
  89. }
  90. err = binary.Write(conn, binary.BigEndian, c.options.StatusInterval)
  91. if err != nil {
  92. return E.Cause(err, "write header")
  93. }
  94. c.handler.Connected()
  95. go c.handleStatusConn(conn)
  96. case CommandGroup:
  97. err = binary.Write(conn, binary.BigEndian, c.options.StatusInterval)
  98. if err != nil {
  99. return E.Cause(err, "write interval")
  100. }
  101. c.handler.Connected()
  102. go c.handleGroupConn(conn)
  103. case CommandClashMode:
  104. var (
  105. modeList []string
  106. currentMode string
  107. )
  108. modeList, currentMode, err = readClashModeList(conn)
  109. if err != nil {
  110. return err
  111. }
  112. c.handler.Connected()
  113. c.handler.InitializeClashMode(newIterator(modeList), currentMode)
  114. if len(modeList) == 0 {
  115. conn.Close()
  116. c.handler.Disconnected(os.ErrInvalid.Error())
  117. return nil
  118. }
  119. go c.handleModeConn(conn)
  120. case CommandConnections:
  121. err = binary.Write(conn, binary.BigEndian, c.options.StatusInterval)
  122. if err != nil {
  123. return E.Cause(err, "write interval")
  124. }
  125. c.handler.Connected()
  126. go c.handleConnectionsConn(conn)
  127. }
  128. return nil
  129. }
  130. func (c *CommandClient) Disconnect() error {
  131. return common.Close(c.conn)
  132. }