router.go 30 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979
  1. package route
  2. import (
  3. "context"
  4. "io"
  5. "net"
  6. "net/http"
  7. "net/netip"
  8. "net/url"
  9. "os"
  10. "os/user"
  11. "path/filepath"
  12. "strings"
  13. "time"
  14. "github.com/sagernet/sing-box/adapter"
  15. "github.com/sagernet/sing-box/common/dialer"
  16. "github.com/sagernet/sing-box/common/geoip"
  17. "github.com/sagernet/sing-box/common/geosite"
  18. "github.com/sagernet/sing-box/common/mux"
  19. "github.com/sagernet/sing-box/common/process"
  20. "github.com/sagernet/sing-box/common/sniff"
  21. "github.com/sagernet/sing-box/common/warning"
  22. C "github.com/sagernet/sing-box/constant"
  23. "github.com/sagernet/sing-box/log"
  24. "github.com/sagernet/sing-box/option"
  25. "github.com/sagernet/sing-dns"
  26. "github.com/sagernet/sing-tun"
  27. "github.com/sagernet/sing/common"
  28. "github.com/sagernet/sing/common/buf"
  29. "github.com/sagernet/sing/common/bufio"
  30. "github.com/sagernet/sing/common/control"
  31. E "github.com/sagernet/sing/common/exceptions"
  32. F "github.com/sagernet/sing/common/format"
  33. M "github.com/sagernet/sing/common/metadata"
  34. N "github.com/sagernet/sing/common/network"
  35. "github.com/sagernet/sing/common/rw"
  36. "github.com/sagernet/sing/common/uot"
  37. )
  38. var warnDefaultInterfaceOnUnsupportedPlatform = warning.New(
  39. func() bool {
  40. return !(C.IsLinux || C.IsWindows || C.IsDarwin)
  41. },
  42. "route option `default_mark` is only supported on Linux and Windows",
  43. )
  44. var warnDefaultMarkOnNonLinux = warning.New(
  45. func() bool {
  46. return !C.IsLinux
  47. },
  48. "route option `default_mark` is only supported on Linux",
  49. )
  50. var warnFindProcessOnUnsupportedPlatform = warning.New(
  51. func() bool {
  52. return !(C.IsLinux || C.IsWindows || C.IsDarwin)
  53. },
  54. "route option `find_process` is only supported on Linux, Windows, and macOS",
  55. )
  56. var _ adapter.Router = (*Router)(nil)
  57. type Router struct {
  58. ctx context.Context
  59. logger log.ContextLogger
  60. dnsLogger log.ContextLogger
  61. inboundByTag map[string]adapter.Inbound
  62. outbounds []adapter.Outbound
  63. outboundByTag map[string]adapter.Outbound
  64. rules []adapter.Rule
  65. defaultDetour string
  66. defaultOutboundForConnection adapter.Outbound
  67. defaultOutboundForPacketConnection adapter.Outbound
  68. needGeoIPDatabase bool
  69. needGeositeDatabase bool
  70. geoIPOptions option.GeoIPOptions
  71. geositeOptions option.GeositeOptions
  72. geoIPReader *geoip.Reader
  73. geositeReader *geosite.Reader
  74. geositeCache map[string]adapter.Rule
  75. dnsClient *dns.Client
  76. defaultDomainStrategy dns.DomainStrategy
  77. dnsRules []adapter.DNSRule
  78. defaultTransport dns.Transport
  79. transports []dns.Transport
  80. transportMap map[string]dns.Transport
  81. transportDomainStrategy map[dns.Transport]dns.DomainStrategy
  82. interfaceBindManager control.BindManager
  83. autoDetectInterface bool
  84. defaultInterface string
  85. defaultMark int
  86. networkMonitor tun.NetworkUpdateMonitor
  87. interfaceMonitor tun.DefaultInterfaceMonitor
  88. packageManager tun.PackageManager
  89. trafficController adapter.TrafficController
  90. processSearcher process.Searcher
  91. }
  92. func NewRouter(ctx context.Context, logger log.ContextLogger, dnsLogger log.ContextLogger, options option.RouteOptions, dnsOptions option.DNSOptions, inbounds []option.Inbound) (*Router, error) {
  93. if options.DefaultInterface != "" {
  94. warnDefaultInterfaceOnUnsupportedPlatform.Check()
  95. }
  96. if options.DefaultMark != 0 {
  97. warnDefaultMarkOnNonLinux.Check()
  98. }
  99. if options.FindProcess {
  100. warnFindProcessOnUnsupportedPlatform.Check()
  101. }
  102. router := &Router{
  103. ctx: ctx,
  104. logger: logger,
  105. dnsLogger: dnsLogger,
  106. outboundByTag: make(map[string]adapter.Outbound),
  107. rules: make([]adapter.Rule, 0, len(options.Rules)),
  108. dnsRules: make([]adapter.DNSRule, 0, len(dnsOptions.Rules)),
  109. needGeoIPDatabase: hasRule(options.Rules, isGeoIPRule) || hasDNSRule(dnsOptions.Rules, isGeoIPDNSRule),
  110. needGeositeDatabase: hasRule(options.Rules, isGeositeRule) || hasDNSRule(dnsOptions.Rules, isGeositeDNSRule),
  111. geoIPOptions: common.PtrValueOrDefault(options.GeoIP),
  112. geositeOptions: common.PtrValueOrDefault(options.Geosite),
  113. geositeCache: make(map[string]adapter.Rule),
  114. defaultDetour: options.Final,
  115. dnsClient: dns.NewClient(dnsOptions.DNSClientOptions.DisableCache, dnsOptions.DNSClientOptions.DisableExpire),
  116. defaultDomainStrategy: dns.DomainStrategy(dnsOptions.Strategy),
  117. interfaceBindManager: control.NewBindManager(),
  118. autoDetectInterface: options.AutoDetectInterface,
  119. defaultInterface: options.DefaultInterface,
  120. defaultMark: options.DefaultMark,
  121. }
  122. for i, ruleOptions := range options.Rules {
  123. routeRule, err := NewRule(router, logger, ruleOptions)
  124. if err != nil {
  125. return nil, E.Cause(err, "parse rule[", i, "]")
  126. }
  127. router.rules = append(router.rules, routeRule)
  128. }
  129. for i, dnsRuleOptions := range dnsOptions.Rules {
  130. dnsRule, err := NewDNSRule(router, logger, dnsRuleOptions)
  131. if err != nil {
  132. return nil, E.Cause(err, "parse dns rule[", i, "]")
  133. }
  134. router.dnsRules = append(router.dnsRules, dnsRule)
  135. }
  136. transports := make([]dns.Transport, len(dnsOptions.Servers))
  137. dummyTransportMap := make(map[string]dns.Transport)
  138. transportMap := make(map[string]dns.Transport)
  139. transportTags := make([]string, len(dnsOptions.Servers))
  140. transportTagMap := make(map[string]bool)
  141. transportDomainStrategy := make(map[dns.Transport]dns.DomainStrategy)
  142. for i, server := range dnsOptions.Servers {
  143. var tag string
  144. if server.Tag != "" {
  145. tag = server.Tag
  146. } else {
  147. tag = F.ToString(i)
  148. }
  149. transportTags[i] = tag
  150. transportTagMap[tag] = true
  151. }
  152. for {
  153. lastLen := len(dummyTransportMap)
  154. for i, server := range dnsOptions.Servers {
  155. tag := transportTags[i]
  156. if _, exists := dummyTransportMap[tag]; exists {
  157. continue
  158. }
  159. var detour N.Dialer
  160. if server.Detour == "" {
  161. detour = dialer.NewRouter(router)
  162. } else {
  163. detour = dialer.NewDetour(router, server.Detour)
  164. }
  165. switch server.Address {
  166. case "local", "rcode":
  167. default:
  168. serverURL, err := url.Parse(server.Address)
  169. if err != nil {
  170. return nil, err
  171. }
  172. serverAddress := serverURL.Hostname()
  173. if serverAddress == "" {
  174. serverAddress = server.Address
  175. }
  176. _, notIpAddress := netip.ParseAddr(serverAddress)
  177. if server.AddressResolver != "" {
  178. if !transportTagMap[server.AddressResolver] {
  179. return nil, E.New("parse dns server[", tag, "]: address resolver not found: ", server.AddressResolver)
  180. }
  181. if upstream, exists := dummyTransportMap[server.AddressResolver]; exists {
  182. detour = dns.NewDialerWrapper(detour, router.dnsClient, upstream, dns.DomainStrategy(server.AddressStrategy), time.Duration(server.AddressFallbackDelay))
  183. } else {
  184. continue
  185. }
  186. } else if notIpAddress != nil && (serverURL == nil || serverURL.Scheme != "rcode") {
  187. return nil, E.New("parse dns server[", tag, "]: missing address_resolver")
  188. }
  189. }
  190. transport, err := dns.NewTransport(ctx, detour, server.Address)
  191. if err != nil {
  192. return nil, E.Cause(err, "parse dns server[", tag, "]")
  193. }
  194. transports[i] = transport
  195. dummyTransportMap[tag] = transport
  196. if server.Tag != "" {
  197. transportMap[server.Tag] = transport
  198. }
  199. strategy := dns.DomainStrategy(server.Strategy)
  200. if strategy != dns.DomainStrategyAsIS {
  201. transportDomainStrategy[transport] = strategy
  202. }
  203. }
  204. if len(transports) == len(dummyTransportMap) {
  205. break
  206. }
  207. if lastLen != len(dummyTransportMap) {
  208. continue
  209. }
  210. unresolvedTags := common.MapIndexed(common.FilterIndexed(dnsOptions.Servers, func(index int, server option.DNSServerOptions) bool {
  211. _, exists := dummyTransportMap[transportTags[index]]
  212. return !exists
  213. }), func(index int, server option.DNSServerOptions) string {
  214. return transportTags[index]
  215. })
  216. return nil, E.New("found circular reference in dns servers: ", strings.Join(unresolvedTags, " "))
  217. }
  218. var defaultTransport dns.Transport
  219. if dnsOptions.Final != "" {
  220. defaultTransport = dummyTransportMap[dnsOptions.Final]
  221. if defaultTransport == nil {
  222. return nil, E.New("default dns server not found: ", dnsOptions.Final)
  223. }
  224. }
  225. if defaultTransport == nil {
  226. if len(transports) == 0 {
  227. transports = append(transports, dns.NewLocalTransport())
  228. }
  229. defaultTransport = transports[0]
  230. }
  231. router.defaultTransport = defaultTransport
  232. router.transports = transports
  233. router.transportMap = transportMap
  234. router.transportDomainStrategy = transportDomainStrategy
  235. needInterfaceMonitor := options.AutoDetectInterface ||
  236. C.IsDarwin && common.Any(inbounds, func(inbound option.Inbound) bool {
  237. return inbound.HTTPOptions.SetSystemProxy || inbound.MixedOptions.SetSystemProxy
  238. })
  239. if router.interfaceBindManager != nil || needInterfaceMonitor {
  240. networkMonitor, err := tun.NewNetworkUpdateMonitor(router)
  241. if err == nil {
  242. router.networkMonitor = networkMonitor
  243. if router.interfaceBindManager != nil {
  244. networkMonitor.RegisterCallback(router.interfaceBindManager.Update)
  245. }
  246. }
  247. }
  248. if router.networkMonitor != nil && needInterfaceMonitor {
  249. interfaceMonitor, err := tun.NewDefaultInterfaceMonitor(router.networkMonitor)
  250. if err != nil {
  251. return nil, E.New("auto_detect_interface unsupported on current platform")
  252. }
  253. interfaceMonitor.RegisterCallback(func() error {
  254. router.logger.Info("updated default interface ", router.interfaceMonitor.DefaultInterfaceName(netip.IPv4Unspecified()), ", index ", router.interfaceMonitor.DefaultInterfaceIndex(netip.IPv4Unspecified()))
  255. return nil
  256. })
  257. router.interfaceMonitor = interfaceMonitor
  258. }
  259. needFindProcess := hasRule(options.Rules, isProcessRule) || hasDNSRule(dnsOptions.Rules, isProcessDNSRule) || options.FindProcess
  260. needPackageManager := C.IsAndroid && (needFindProcess || common.Any(inbounds, func(inbound option.Inbound) bool {
  261. return len(inbound.TunOptions.IncludePackage) > 0 || len(inbound.TunOptions.ExcludePackage) > 0
  262. }))
  263. if needPackageManager {
  264. packageManager, err := tun.NewPackageManager(router)
  265. if err != nil {
  266. return nil, E.Cause(err, "create package manager")
  267. }
  268. router.packageManager = packageManager
  269. }
  270. if needFindProcess {
  271. searcher, err := process.NewSearcher(process.Config{
  272. Logger: logger,
  273. PackageManager: router.packageManager,
  274. })
  275. if err != nil {
  276. if err != os.ErrInvalid {
  277. logger.Warn(E.Cause(err, "create process searcher"))
  278. }
  279. } else {
  280. router.processSearcher = searcher
  281. }
  282. }
  283. return router, nil
  284. }
  285. func (r *Router) Initialize(inbounds []adapter.Inbound, outbounds []adapter.Outbound, defaultOutbound func() adapter.Outbound) error {
  286. inboundByTag := make(map[string]adapter.Inbound)
  287. for _, inbound := range inbounds {
  288. inboundByTag[inbound.Tag()] = inbound
  289. }
  290. outboundByTag := make(map[string]adapter.Outbound)
  291. for _, detour := range outbounds {
  292. outboundByTag[detour.Tag()] = detour
  293. }
  294. var defaultOutboundForConnection adapter.Outbound
  295. var defaultOutboundForPacketConnection adapter.Outbound
  296. if r.defaultDetour != "" {
  297. detour, loaded := outboundByTag[r.defaultDetour]
  298. if !loaded {
  299. return E.New("default detour not found: ", r.defaultDetour)
  300. }
  301. if common.Contains(detour.Network(), N.NetworkTCP) {
  302. defaultOutboundForConnection = detour
  303. }
  304. if common.Contains(detour.Network(), N.NetworkUDP) {
  305. defaultOutboundForPacketConnection = detour
  306. }
  307. }
  308. var index, packetIndex int
  309. if defaultOutboundForConnection == nil {
  310. for i, detour := range outbounds {
  311. if common.Contains(detour.Network(), N.NetworkTCP) {
  312. index = i
  313. defaultOutboundForConnection = detour
  314. break
  315. }
  316. }
  317. }
  318. if defaultOutboundForPacketConnection == nil {
  319. for i, detour := range outbounds {
  320. if common.Contains(detour.Network(), N.NetworkUDP) {
  321. packetIndex = i
  322. defaultOutboundForPacketConnection = detour
  323. break
  324. }
  325. }
  326. }
  327. if defaultOutboundForConnection == nil || defaultOutboundForPacketConnection == nil {
  328. detour := defaultOutbound()
  329. if defaultOutboundForConnection == nil {
  330. defaultOutboundForConnection = detour
  331. }
  332. if defaultOutboundForPacketConnection == nil {
  333. defaultOutboundForPacketConnection = detour
  334. }
  335. outbounds = append(outbounds, detour)
  336. outboundByTag[detour.Tag()] = detour
  337. }
  338. if defaultOutboundForConnection != defaultOutboundForPacketConnection {
  339. var description string
  340. if defaultOutboundForConnection.Tag() != "" {
  341. description = defaultOutboundForConnection.Tag()
  342. } else {
  343. description = F.ToString(index)
  344. }
  345. var packetDescription string
  346. if defaultOutboundForPacketConnection.Tag() != "" {
  347. packetDescription = defaultOutboundForPacketConnection.Tag()
  348. } else {
  349. packetDescription = F.ToString(packetIndex)
  350. }
  351. r.logger.Info("using ", defaultOutboundForConnection.Type(), "[", description, "] as default outbound for connection")
  352. r.logger.Info("using ", defaultOutboundForPacketConnection.Type(), "[", packetDescription, "] as default outbound for packet connection")
  353. }
  354. r.inboundByTag = inboundByTag
  355. r.outbounds = outbounds
  356. r.defaultOutboundForConnection = defaultOutboundForConnection
  357. r.defaultOutboundForPacketConnection = defaultOutboundForPacketConnection
  358. r.outboundByTag = outboundByTag
  359. for i, rule := range r.rules {
  360. if _, loaded := outboundByTag[rule.Outbound()]; !loaded {
  361. return E.New("outbound not found for rule[", i, "]: ", rule.Outbound())
  362. }
  363. }
  364. return nil
  365. }
  366. func (r *Router) Outbounds() []adapter.Outbound {
  367. return r.outbounds
  368. }
  369. func (r *Router) Start() error {
  370. if r.needGeoIPDatabase {
  371. err := r.prepareGeoIPDatabase()
  372. if err != nil {
  373. return err
  374. }
  375. }
  376. if r.needGeositeDatabase {
  377. err := r.prepareGeositeDatabase()
  378. if err != nil {
  379. return err
  380. }
  381. }
  382. for _, rule := range r.rules {
  383. err := rule.Start()
  384. if err != nil {
  385. return err
  386. }
  387. }
  388. for _, rule := range r.dnsRules {
  389. err := rule.Start()
  390. if err != nil {
  391. return err
  392. }
  393. }
  394. if r.needGeositeDatabase {
  395. for _, rule := range r.rules {
  396. err := rule.UpdateGeosite()
  397. if err != nil {
  398. r.logger.Error("failed to initialize geosite: ", err)
  399. }
  400. }
  401. for _, rule := range r.dnsRules {
  402. err := rule.UpdateGeosite()
  403. if err != nil {
  404. r.logger.Error("failed to initialize geosite: ", err)
  405. }
  406. }
  407. err := common.Close(r.geositeReader)
  408. if err != nil {
  409. return err
  410. }
  411. r.geositeCache = nil
  412. r.geositeReader = nil
  413. }
  414. if r.interfaceMonitor != nil {
  415. err := r.interfaceMonitor.Start()
  416. if err != nil {
  417. return err
  418. }
  419. }
  420. if r.networkMonitor != nil {
  421. err := r.networkMonitor.Start()
  422. if err != nil {
  423. return err
  424. }
  425. }
  426. if r.packageManager != nil {
  427. err := r.packageManager.Start()
  428. if err != nil {
  429. return err
  430. }
  431. }
  432. return nil
  433. }
  434. func (r *Router) Close() error {
  435. for _, rule := range r.rules {
  436. err := rule.Close()
  437. if err != nil {
  438. return err
  439. }
  440. }
  441. for _, rule := range r.dnsRules {
  442. err := rule.Close()
  443. if err != nil {
  444. return err
  445. }
  446. }
  447. return common.Close(
  448. common.PtrOrNil(r.geoIPReader),
  449. r.interfaceMonitor,
  450. r.networkMonitor,
  451. r.packageManager,
  452. )
  453. }
  454. func (r *Router) GeoIPReader() *geoip.Reader {
  455. return r.geoIPReader
  456. }
  457. func (r *Router) LoadGeosite(code string) (adapter.Rule, error) {
  458. rule, cached := r.geositeCache[code]
  459. if cached {
  460. return rule, nil
  461. }
  462. items, err := r.geositeReader.Read(code)
  463. if err != nil {
  464. return nil, err
  465. }
  466. rule, err = NewDefaultRule(r, nil, geosite.Compile(items))
  467. if err != nil {
  468. return nil, err
  469. }
  470. r.geositeCache[code] = rule
  471. return rule, nil
  472. }
  473. func (r *Router) Outbound(tag string) (adapter.Outbound, bool) {
  474. outbound, loaded := r.outboundByTag[tag]
  475. return outbound, loaded
  476. }
  477. func (r *Router) DefaultOutbound(network string) adapter.Outbound {
  478. if network == N.NetworkTCP {
  479. return r.defaultOutboundForConnection
  480. } else {
  481. return r.defaultOutboundForPacketConnection
  482. }
  483. }
  484. func (r *Router) RouteConnection(ctx context.Context, conn net.Conn, metadata adapter.InboundContext) error {
  485. if metadata.InboundDetour != "" {
  486. if metadata.LastInbound == metadata.InboundDetour {
  487. return E.New("routing loop on detour: ", metadata.InboundDetour)
  488. }
  489. detour := r.inboundByTag[metadata.InboundDetour]
  490. if detour == nil {
  491. return E.New("inbound detour not found: ", metadata.InboundDetour)
  492. }
  493. injectable, isInjectable := detour.(adapter.InjectableInbound)
  494. if !isInjectable {
  495. return E.New("inbound detour is not injectable: ", metadata.InboundDetour)
  496. }
  497. if !common.Contains(injectable.Network(), N.NetworkTCP) {
  498. return E.New("inject: TCP unsupported")
  499. }
  500. metadata.InboundDetour = ""
  501. metadata.LastInbound = metadata.Inbound
  502. err := injectable.NewConnection(ctx, conn, metadata)
  503. if err != nil {
  504. return E.Cause(err, "inject ", detour.Tag())
  505. }
  506. return nil
  507. }
  508. metadata.Network = N.NetworkTCP
  509. switch metadata.Destination.Fqdn {
  510. case mux.Destination.Fqdn:
  511. r.logger.InfoContext(ctx, "inbound multiplex connection")
  512. return mux.NewConnection(ctx, r, r, r.logger, conn, metadata)
  513. case uot.UOTMagicAddress:
  514. r.logger.InfoContext(ctx, "inbound UoT connection")
  515. metadata.Destination = M.Socksaddr{}
  516. return r.RoutePacketConnection(ctx, uot.NewClientConn(conn), metadata)
  517. }
  518. if metadata.SniffEnabled {
  519. buffer := buf.NewPacket()
  520. buffer.FullReset()
  521. sniffMetadata, err := sniff.PeekStream(ctx, conn, buffer, sniff.StreamDomainNameQuery, sniff.TLSClientHello, sniff.HTTPHost)
  522. if err == nil {
  523. metadata.Protocol = sniffMetadata.Protocol
  524. metadata.Domain = sniffMetadata.Domain
  525. if metadata.SniffOverrideDestination && sniff.IsDomainName(metadata.Domain) {
  526. metadata.Destination = M.Socksaddr{
  527. Fqdn: metadata.Domain,
  528. Port: metadata.Destination.Port,
  529. }
  530. }
  531. if metadata.Domain != "" {
  532. r.logger.DebugContext(ctx, "sniffed protocol: ", metadata.Protocol, ", domain: ", metadata.Domain)
  533. } else {
  534. r.logger.DebugContext(ctx, "sniffed protocol: ", metadata.Protocol)
  535. }
  536. }
  537. if !buffer.IsEmpty() {
  538. conn = bufio.NewCachedConn(conn, buffer)
  539. } else {
  540. buffer.Release()
  541. }
  542. }
  543. if metadata.Destination.IsFqdn() && metadata.DomainStrategy != dns.DomainStrategyAsIS {
  544. addresses, err := r.Lookup(adapter.WithContext(ctx, &metadata), metadata.Destination.Fqdn, metadata.DomainStrategy)
  545. if err != nil {
  546. return err
  547. }
  548. metadata.DestinationAddresses = addresses
  549. r.dnsLogger.DebugContext(ctx, "resolved [", strings.Join(F.MapToString(metadata.DestinationAddresses), " "), "]")
  550. }
  551. matchedRule, detour := r.match(ctx, &metadata, r.defaultOutboundForConnection)
  552. if !common.Contains(detour.Network(), N.NetworkTCP) {
  553. conn.Close()
  554. return E.New("missing supported outbound, closing connection")
  555. }
  556. if r.trafficController != nil {
  557. trackerConn, tracker := r.trafficController.RoutedConnection(ctx, conn, metadata, matchedRule)
  558. defer tracker.Leave()
  559. conn = trackerConn
  560. }
  561. return detour.NewConnection(ctx, conn, metadata)
  562. }
  563. func (r *Router) RoutePacketConnection(ctx context.Context, conn N.PacketConn, metadata adapter.InboundContext) error {
  564. if metadata.InboundDetour != "" {
  565. if metadata.LastInbound == metadata.InboundDetour {
  566. return E.New("routing loop on detour: ", metadata.InboundDetour)
  567. }
  568. detour := r.inboundByTag[metadata.InboundDetour]
  569. if detour == nil {
  570. return E.New("inbound detour not found: ", metadata.InboundDetour)
  571. }
  572. injectable, isInjectable := detour.(adapter.InjectableInbound)
  573. if !isInjectable {
  574. return E.New("inbound detour is not injectable: ", metadata.InboundDetour)
  575. }
  576. if !common.Contains(injectable.Network(), N.NetworkUDP) {
  577. return E.New("inject: UDP unsupported")
  578. }
  579. metadata.InboundDetour = ""
  580. metadata.LastInbound = metadata.Inbound
  581. err := injectable.NewPacketConnection(ctx, conn, metadata)
  582. if err != nil {
  583. return E.Cause(err, "inject ", detour.Tag())
  584. }
  585. return nil
  586. }
  587. metadata.Network = N.NetworkUDP
  588. if metadata.SniffEnabled {
  589. buffer := buf.NewPacket()
  590. buffer.FullReset()
  591. destination, err := conn.ReadPacket(buffer)
  592. if err != nil {
  593. buffer.Release()
  594. return err
  595. }
  596. sniffMetadata, err := sniff.PeekPacket(ctx, buffer.Bytes(), sniff.DomainNameQuery, sniff.QUICClientHello, sniff.STUNMessage)
  597. if err == nil {
  598. metadata.Protocol = sniffMetadata.Protocol
  599. metadata.Domain = sniffMetadata.Domain
  600. if metadata.SniffOverrideDestination && sniff.IsDomainName(metadata.Domain) {
  601. metadata.Destination = M.Socksaddr{
  602. Fqdn: metadata.Domain,
  603. Port: metadata.Destination.Port,
  604. }
  605. }
  606. if metadata.Domain != "" {
  607. r.logger.DebugContext(ctx, "sniffed packet protocol: ", metadata.Protocol, ", domain: ", metadata.Domain)
  608. } else {
  609. r.logger.DebugContext(ctx, "sniffed packet protocol: ", metadata.Protocol)
  610. }
  611. }
  612. conn = bufio.NewCachedPacketConn(conn, buffer, destination)
  613. }
  614. if metadata.Destination.IsFqdn() && metadata.Destination.Fqdn != uot.UOTMagicAddress && metadata.DomainStrategy != dns.DomainStrategyAsIS {
  615. addresses, err := r.Lookup(adapter.WithContext(ctx, &metadata), metadata.Destination.Fqdn, metadata.DomainStrategy)
  616. if err != nil {
  617. return err
  618. }
  619. metadata.DestinationAddresses = addresses
  620. r.dnsLogger.DebugContext(ctx, "resolved [", strings.Join(F.MapToString(metadata.DestinationAddresses), " "), "]")
  621. }
  622. matchedRule, detour := r.match(ctx, &metadata, r.defaultOutboundForPacketConnection)
  623. if !common.Contains(detour.Network(), N.NetworkUDP) {
  624. conn.Close()
  625. return E.New("missing supported outbound, closing packet connection")
  626. }
  627. if r.trafficController != nil {
  628. trackerConn, tracker := r.trafficController.RoutedPacketConnection(ctx, conn, metadata, matchedRule)
  629. defer tracker.Leave()
  630. conn = trackerConn
  631. }
  632. return detour.NewPacketConnection(ctx, conn, metadata)
  633. }
  634. func (r *Router) match(ctx context.Context, metadata *adapter.InboundContext, defaultOutbound adapter.Outbound) (adapter.Rule, adapter.Outbound) {
  635. if r.processSearcher != nil {
  636. var originDestination netip.AddrPort
  637. if metadata.OriginDestination.IsValid() {
  638. originDestination = metadata.OriginDestination.AddrPort()
  639. } else if metadata.Destination.IsIP() {
  640. originDestination = metadata.Destination.AddrPort()
  641. }
  642. processInfo, err := process.FindProcessInfo(r.processSearcher, ctx, metadata.Network, metadata.Source.AddrPort(), originDestination)
  643. if err != nil {
  644. r.logger.DebugContext(ctx, "failed to search process: ", err)
  645. } else {
  646. if processInfo.ProcessPath != "" {
  647. r.logger.DebugContext(ctx, "found process path: ", processInfo.ProcessPath)
  648. } else if processInfo.PackageName != "" {
  649. r.logger.DebugContext(ctx, "found package name: ", processInfo.PackageName)
  650. } else if processInfo.UserId != -1 {
  651. if /*needUserName &&*/ true {
  652. osUser, _ := user.LookupId(F.ToString(processInfo.UserId))
  653. if osUser != nil {
  654. processInfo.User = osUser.Username
  655. }
  656. }
  657. if processInfo.User != "" {
  658. r.logger.DebugContext(ctx, "found user: ", processInfo.User)
  659. } else {
  660. r.logger.DebugContext(ctx, "found user id: ", processInfo.UserId)
  661. }
  662. }
  663. metadata.ProcessInfo = processInfo
  664. }
  665. }
  666. for i, rule := range r.rules {
  667. if rule.Match(metadata) {
  668. detour := rule.Outbound()
  669. r.logger.DebugContext(ctx, "match[", i, "] ", rule.String(), " => ", detour)
  670. if outbound, loaded := r.Outbound(detour); loaded {
  671. return rule, outbound
  672. }
  673. r.logger.ErrorContext(ctx, "outbound not found: ", detour)
  674. }
  675. }
  676. return nil, defaultOutbound
  677. }
  678. func (r *Router) InterfaceBindManager() control.BindManager {
  679. return r.interfaceBindManager
  680. }
  681. func (r *Router) AutoDetectInterface() bool {
  682. return r.autoDetectInterface
  683. }
  684. func (r *Router) DefaultInterface() string {
  685. return r.defaultInterface
  686. }
  687. func (r *Router) DefaultMark() int {
  688. return r.defaultMark
  689. }
  690. func (r *Router) Rules() []adapter.Rule {
  691. return r.rules
  692. }
  693. func (r *Router) NetworkMonitor() tun.NetworkUpdateMonitor {
  694. return r.networkMonitor
  695. }
  696. func (r *Router) InterfaceMonitor() tun.DefaultInterfaceMonitor {
  697. return r.interfaceMonitor
  698. }
  699. func (r *Router) PackageManager() tun.PackageManager {
  700. return r.packageManager
  701. }
  702. func (r *Router) SetTrafficController(controller adapter.TrafficController) {
  703. r.trafficController = controller
  704. }
  705. func hasRule(rules []option.Rule, cond func(rule option.DefaultRule) bool) bool {
  706. for _, rule := range rules {
  707. switch rule.Type {
  708. case C.RuleTypeDefault:
  709. if cond(rule.DefaultOptions) {
  710. return true
  711. }
  712. case C.RuleTypeLogical:
  713. for _, subRule := range rule.LogicalOptions.Rules {
  714. if cond(subRule) {
  715. return true
  716. }
  717. }
  718. }
  719. }
  720. return false
  721. }
  722. func hasDNSRule(rules []option.DNSRule, cond func(rule option.DefaultDNSRule) bool) bool {
  723. for _, rule := range rules {
  724. switch rule.Type {
  725. case C.RuleTypeDefault:
  726. if cond(rule.DefaultOptions) {
  727. return true
  728. }
  729. case C.RuleTypeLogical:
  730. for _, subRule := range rule.LogicalOptions.Rules {
  731. if cond(subRule) {
  732. return true
  733. }
  734. }
  735. }
  736. }
  737. return false
  738. }
  739. func isGeoIPRule(rule option.DefaultRule) bool {
  740. return len(rule.SourceGeoIP) > 0 && common.Any(rule.SourceGeoIP, notPrivateNode) || len(rule.GeoIP) > 0 && common.Any(rule.GeoIP, notPrivateNode)
  741. }
  742. func isGeoIPDNSRule(rule option.DefaultDNSRule) bool {
  743. return len(rule.SourceGeoIP) > 0 && common.Any(rule.SourceGeoIP, notPrivateNode)
  744. }
  745. func isGeositeRule(rule option.DefaultRule) bool {
  746. return len(rule.Geosite) > 0
  747. }
  748. func isGeositeDNSRule(rule option.DefaultDNSRule) bool {
  749. return len(rule.Geosite) > 0
  750. }
  751. func isProcessRule(rule option.DefaultRule) bool {
  752. return len(rule.ProcessName) > 0 || len(rule.PackageName) > 0 || len(rule.User) > 0 || len(rule.UserID) > 0
  753. }
  754. func isProcessDNSRule(rule option.DefaultDNSRule) bool {
  755. return len(rule.ProcessName) > 0 || len(rule.PackageName) > 0 || len(rule.User) > 0 || len(rule.UserID) > 0
  756. }
  757. func notPrivateNode(code string) bool {
  758. return code != "private"
  759. }
  760. func (r *Router) prepareGeoIPDatabase() error {
  761. var geoPath string
  762. if r.geoIPOptions.Path != "" {
  763. geoPath = r.geoIPOptions.Path
  764. } else {
  765. geoPath = "geoip.db"
  766. if foundPath, loaded := C.FindPath(geoPath); loaded {
  767. geoPath = foundPath
  768. }
  769. }
  770. if !rw.FileExists(geoPath) {
  771. r.logger.Warn("geoip database not exists: ", geoPath)
  772. var err error
  773. for attempts := 0; attempts < 3; attempts++ {
  774. err = r.downloadGeoIPDatabase(geoPath)
  775. if err == nil {
  776. break
  777. }
  778. r.logger.Error("download geoip database: ", err)
  779. os.Remove(geoPath)
  780. time.Sleep(10 * time.Second)
  781. }
  782. if err != nil {
  783. return err
  784. }
  785. }
  786. geoReader, codes, err := geoip.Open(geoPath)
  787. if err != nil {
  788. return E.Cause(err, "open geoip database")
  789. }
  790. r.logger.Info("loaded geoip database: ", len(codes), " codes")
  791. r.geoIPReader = geoReader
  792. return nil
  793. }
  794. func (r *Router) prepareGeositeDatabase() error {
  795. var geoPath string
  796. if r.geositeOptions.Path != "" {
  797. geoPath = r.geositeOptions.Path
  798. } else {
  799. geoPath = "geosite.db"
  800. if foundPath, loaded := C.FindPath(geoPath); loaded {
  801. geoPath = foundPath
  802. }
  803. }
  804. if !rw.FileExists(geoPath) {
  805. r.logger.Warn("geosite database not exists: ", geoPath)
  806. var err error
  807. for attempts := 0; attempts < 3; attempts++ {
  808. err = r.downloadGeositeDatabase(geoPath)
  809. if err == nil {
  810. break
  811. }
  812. r.logger.Error("download geosite database: ", err)
  813. os.Remove(geoPath)
  814. time.Sleep(10 * time.Second)
  815. }
  816. if err != nil {
  817. return err
  818. }
  819. }
  820. geoReader, codes, err := geosite.Open(geoPath)
  821. if err == nil {
  822. r.logger.Info("loaded geosite database: ", len(codes), " codes")
  823. r.geositeReader = geoReader
  824. } else {
  825. return E.Cause(err, "open geosite database")
  826. }
  827. return nil
  828. }
  829. func (r *Router) downloadGeoIPDatabase(savePath string) error {
  830. var downloadURL string
  831. if r.geoIPOptions.DownloadURL != "" {
  832. downloadURL = r.geoIPOptions.DownloadURL
  833. } else {
  834. downloadURL = "https://github.com/SagerNet/sing-geoip/releases/latest/download/geoip.db"
  835. }
  836. r.logger.Info("downloading geoip database")
  837. var detour adapter.Outbound
  838. if r.geoIPOptions.DownloadDetour != "" {
  839. outbound, loaded := r.Outbound(r.geoIPOptions.DownloadDetour)
  840. if !loaded {
  841. return E.New("detour outbound not found: ", r.geoIPOptions.DownloadDetour)
  842. }
  843. detour = outbound
  844. } else {
  845. detour = r.defaultOutboundForConnection
  846. }
  847. if parentDir := filepath.Dir(savePath); parentDir != "" {
  848. os.MkdirAll(parentDir, 0o755)
  849. }
  850. saveFile, err := os.OpenFile(savePath, os.O_CREATE|os.O_WRONLY, 0o644)
  851. if err != nil {
  852. return E.Cause(err, "open output file: ", downloadURL)
  853. }
  854. defer saveFile.Close()
  855. httpClient := &http.Client{
  856. Transport: &http.Transport{
  857. ForceAttemptHTTP2: true,
  858. TLSHandshakeTimeout: 5 * time.Second,
  859. DialContext: func(ctx context.Context, network, addr string) (net.Conn, error) {
  860. return detour.DialContext(ctx, network, M.ParseSocksaddr(addr))
  861. },
  862. },
  863. }
  864. response, err := httpClient.Get(downloadURL)
  865. if err != nil {
  866. return err
  867. }
  868. defer response.Body.Close()
  869. _, err = io.Copy(saveFile, response.Body)
  870. return err
  871. }
  872. func (r *Router) downloadGeositeDatabase(savePath string) error {
  873. var downloadURL string
  874. if r.geositeOptions.DownloadURL != "" {
  875. downloadURL = r.geositeOptions.DownloadURL
  876. } else {
  877. downloadURL = "https://github.com/SagerNet/sing-geosite/releases/latest/download/geosite.db"
  878. }
  879. r.logger.Info("downloading geosite database")
  880. var detour adapter.Outbound
  881. if r.geositeOptions.DownloadDetour != "" {
  882. outbound, loaded := r.Outbound(r.geositeOptions.DownloadDetour)
  883. if !loaded {
  884. return E.New("detour outbound not found: ", r.geositeOptions.DownloadDetour)
  885. }
  886. detour = outbound
  887. } else {
  888. detour = r.defaultOutboundForConnection
  889. }
  890. if parentDir := filepath.Dir(savePath); parentDir != "" {
  891. os.MkdirAll(parentDir, 0o755)
  892. }
  893. saveFile, err := os.OpenFile(savePath, os.O_CREATE|os.O_WRONLY, 0o644)
  894. if err != nil {
  895. return E.Cause(err, "open output file: ", downloadURL)
  896. }
  897. defer saveFile.Close()
  898. httpClient := &http.Client{
  899. Transport: &http.Transport{
  900. ForceAttemptHTTP2: true,
  901. TLSHandshakeTimeout: 5 * time.Second,
  902. DialContext: func(ctx context.Context, network, addr string) (net.Conn, error) {
  903. return detour.DialContext(ctx, network, M.ParseSocksaddr(addr))
  904. },
  905. },
  906. }
  907. response, err := httpClient.Get(downloadURL)
  908. if err != nil {
  909. return err
  910. }
  911. defer response.Body.Close()
  912. _, err = io.Copy(saveFile, response.Body)
  913. return err
  914. }
  915. func (r *Router) OnPackagesUpdated(packages int, sharedUsers int) {
  916. r.logger.Info("updated packages list: ", packages, " packages, ", sharedUsers, " shared users")
  917. }
  918. func (r *Router) NewError(ctx context.Context, err error) {
  919. common.Close(err)
  920. if E.IsClosedOrCanceled(err) {
  921. r.logger.DebugContext(ctx, "connection closed: ", err)
  922. return
  923. }
  924. r.logger.ErrorContext(ctx, err)
  925. }