chunyu/sing-box
1
0
Fork 0
mirror of https://github.com/SagerNet/sing-box.git synced 2025-06-13 21:54:13 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix: wait too long time when tor startup

Browse Source
This commit is contained in:
yaotthaha 2023-04-16 17:11:32 +08:00
parent 07d0b71d2b
commit 0b1ed41896
1 changed files with 21 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
outbound/tor.go
View File

@ -6,6 +6,7 @@ import (
"os"
"path/filepath"
"strings"
"sync/atomic"
"github.com/sagernet/sing-box/adapter"
"github.com/sagernet/sing-box/common/dialer"
@ -35,6 +36,8 @@ type Tor struct {
events chan control.Event
instance *tor.Tor
socksClient *socks.Client
startup atomic.Bool
}
func NewTor(ctx context.Context, router adapter.Router, logger log.ContextLogger, tag string, options option.TorOutboundOptions) (*Tor, error) {
@ -142,21 +145,30 @@ func (t *Tor) start() error {
}
}
}
err = torInstance.EnableNetwork(t.ctx, true)
go t.waitStartup()
return nil
}
func (t *Tor) waitStartup() {
err := t.instance.EnableNetwork(t.ctx, true)
if err != nil {
return err
t.logger.Error("enable tor network fail: ", err.Error())
return
}
info, err := torInstance.Control.GetInfo("net/listeners/socks")
info, err := t.instance.Control.GetInfo("net/listeners/socks")
if err != nil {
return err
t.logger.Error("get tor socks proxy address fail: ", err.Error())
return
}
if len(info) != 1 || info[0].Key != "net/listeners/socks" {
return E.New("get socks proxy address")
t.logger.Error("get socks proxy address")
return
}
t.logger.Trace("obtained tor socks5 address ", info[0].Val)
// TODO: set password for tor socks5 server if supported
t.socksClient = socks.NewClient(N.SystemDialer, M.ParseSocksaddr(info[0].Val), socks.Version5, "", "")
return nil
t.startup.Store(true)
t.logger.Info("tor startup complete")
}
func (t *Tor) recvLoop() {
@ -196,6 +208,9 @@ func (t *Tor) Close() error {
func (t *Tor) DialContext(ctx context.Context, network string, destination M.Socksaddr) (net.Conn, error) {
t.logger.InfoContext(ctx, "outbound connection to ", destination)
if !t.startup.Load() {
return nil, E.New("tor startup not complete")
}
return t.socksClient.DialContext(ctx, network, destination)
}