endpoint.go 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869
  1. //go:build with_gvisor
  2. package tailscale
  3. import (
  4. "context"
  5. "fmt"
  6. "net"
  7. "net/http"
  8. "net/netip"
  9. "net/url"
  10. "os"
  11. "path/filepath"
  12. "reflect"
  13. "runtime"
  14. "strings"
  15. "sync/atomic"
  16. "syscall"
  17. "time"
  18. "github.com/sagernet/gvisor/pkg/tcpip"
  19. "github.com/sagernet/gvisor/pkg/tcpip/adapters/gonet"
  20. "github.com/sagernet/gvisor/pkg/tcpip/header"
  21. "github.com/sagernet/gvisor/pkg/tcpip/stack"
  22. "github.com/sagernet/gvisor/pkg/tcpip/transport/icmp"
  23. "github.com/sagernet/sing-box/adapter"
  24. "github.com/sagernet/sing-box/adapter/endpoint"
  25. "github.com/sagernet/sing-box/common/dialer"
  26. C "github.com/sagernet/sing-box/constant"
  27. "github.com/sagernet/sing-box/experimental/deprecated"
  28. "github.com/sagernet/sing-box/log"
  29. "github.com/sagernet/sing-box/option"
  30. "github.com/sagernet/sing-box/route/rule"
  31. "github.com/sagernet/sing-tun"
  32. "github.com/sagernet/sing-tun/ping"
  33. "github.com/sagernet/sing/common"
  34. "github.com/sagernet/sing/common/bufio"
  35. "github.com/sagernet/sing/common/control"
  36. E "github.com/sagernet/sing/common/exceptions"
  37. F "github.com/sagernet/sing/common/format"
  38. "github.com/sagernet/sing/common/logger"
  39. M "github.com/sagernet/sing/common/metadata"
  40. N "github.com/sagernet/sing/common/network"
  41. "github.com/sagernet/sing/service"
  42. "github.com/sagernet/sing/service/filemanager"
  43. _ "github.com/sagernet/tailscale/feature/relayserver"
  44. "github.com/sagernet/tailscale/ipn"
  45. tsDNS "github.com/sagernet/tailscale/net/dns"
  46. "github.com/sagernet/tailscale/net/netmon"
  47. "github.com/sagernet/tailscale/net/netns"
  48. "github.com/sagernet/tailscale/net/tsaddr"
  49. tsTUN "github.com/sagernet/tailscale/net/tstun"
  50. "github.com/sagernet/tailscale/tsnet"
  51. "github.com/sagernet/tailscale/types/ipproto"
  52. "github.com/sagernet/tailscale/types/nettype"
  53. "github.com/sagernet/tailscale/version"
  54. "github.com/sagernet/tailscale/wgengine"
  55. "github.com/sagernet/tailscale/wgengine/filter"
  56. "github.com/sagernet/tailscale/wgengine/router"
  57. "github.com/sagernet/tailscale/wgengine/wgcfg"
  58. "go4.org/netipx"
  59. )
  60. var (
  61. _ adapter.OutboundWithPreferredRoutes = (*Endpoint)(nil)
  62. _ adapter.DirectRouteOutbound = (*Endpoint)(nil)
  63. _ dialer.PacketDialerWithDestination = (*Endpoint)(nil)
  64. )
  65. func init() {
  66. version.SetVersion("sing-box " + C.Version)
  67. }
  68. func RegisterEndpoint(registry *endpoint.Registry) {
  69. endpoint.Register[option.TailscaleEndpointOptions](registry, C.TypeTailscale, NewEndpoint)
  70. }
  71. type Endpoint struct {
  72. endpoint.Adapter
  73. ctx context.Context
  74. router adapter.Router
  75. logger logger.ContextLogger
  76. dnsRouter adapter.DNSRouter
  77. network adapter.NetworkManager
  78. platformInterface adapter.PlatformInterface
  79. server *tsnet.Server
  80. controlTransport adapter.HTTPTransport
  81. stack *stack.Stack
  82. icmpForwarder *tun.ICMPForwarder
  83. filter *atomic.Pointer[filter.Filter]
  84. onReconfigHook wgengine.ReconfigListener
  85. cfg *wgcfg.Config
  86. dnsCfg *tsDNS.Config
  87. routeDomains common.TypedValue[map[string]bool]
  88. routePrefixes atomic.Pointer[netipx.IPSet]
  89. acceptRoutes bool
  90. exitNode string
  91. exitNodeAllowLANAccess bool
  92. advertiseRoutes []netip.Prefix
  93. advertiseExitNode bool
  94. advertiseTags []string
  95. relayServerPort *uint16
  96. relayServerStaticEndpoints []netip.AddrPort
  97. udpTimeout time.Duration
  98. systemInterface bool
  99. systemInterfaceName string
  100. systemInterfaceMTU uint32
  101. systemTun tun.Tun
  102. systemDialer *dialer.DefaultDialer
  103. fallbackTCPCloser func()
  104. }
  105. func (t *Endpoint) registerNetstackHandlers() {
  106. netstack := t.server.ExportNetstack()
  107. if netstack == nil {
  108. return
  109. }
  110. previousTCP := netstack.GetTCPHandlerForFlow
  111. netstack.GetTCPHandlerForFlow = func(src, dst netip.AddrPort) (handler func(net.Conn), intercept bool) {
  112. if previousTCP != nil {
  113. handler, intercept = previousTCP(src, dst)
  114. if handler != nil || !intercept {
  115. return handler, intercept
  116. }
  117. }
  118. return func(conn net.Conn) {
  119. ctx := log.ContextWithNewID(t.ctx)
  120. source := M.SocksaddrFrom(src.Addr(), src.Port())
  121. destination := M.SocksaddrFrom(dst.Addr(), dst.Port())
  122. t.NewConnectionEx(ctx, conn, source, destination, nil)
  123. }, true
  124. }
  125. previousUDP := netstack.GetUDPHandlerForFlow
  126. netstack.GetUDPHandlerForFlow = func(src, dst netip.AddrPort) (handler func(nettype.ConnPacketConn), intercept bool) {
  127. if previousUDP != nil {
  128. handler, intercept = previousUDP(src, dst)
  129. if handler != nil || !intercept {
  130. return handler, intercept
  131. }
  132. }
  133. return func(conn nettype.ConnPacketConn) {
  134. ctx := log.ContextWithNewID(t.ctx)
  135. source := M.SocksaddrFrom(src.Addr(), src.Port())
  136. destination := M.SocksaddrFrom(dst.Addr(), dst.Port())
  137. packetConn := bufio.NewUnbindPacketConnWithAddr(conn, destination)
  138. t.NewPacketConnectionEx(ctx, packetConn, source, destination, nil)
  139. }, true
  140. }
  141. }
  142. func NewEndpoint(ctx context.Context, router adapter.Router, logger log.ContextLogger, tag string, options option.TailscaleEndpointOptions) (adapter.Endpoint, error) {
  143. stateDirectory := options.StateDirectory
  144. if stateDirectory == "" {
  145. stateDirectory = "tailscale"
  146. }
  147. hostname := options.Hostname
  148. if hostname == "" {
  149. osHostname, _ := os.Hostname()
  150. osHostname = strings.TrimSpace(osHostname)
  151. hostname = osHostname
  152. }
  153. if hostname == "" {
  154. hostname = "sing-box"
  155. }
  156. stateDirectory = filemanager.BasePath(ctx, os.ExpandEnv(stateDirectory))
  157. stateDirectory, _ = filepath.Abs(stateDirectory)
  158. for _, advertiseRoute := range options.AdvertiseRoutes {
  159. if advertiseRoute.Addr().IsUnspecified() && advertiseRoute.Bits() == 0 {
  160. return nil, E.New("`advertise_routes` cannot be default, use `advertise_exit_node` instead.")
  161. }
  162. }
  163. if options.AdvertiseExitNode && options.ExitNode != "" {
  164. return nil, E.New("cannot advertise an exit node and use an exit node at the same time.")
  165. }
  166. var udpTimeout time.Duration
  167. if options.UDPTimeout != 0 {
  168. udpTimeout = time.Duration(options.UDPTimeout)
  169. } else {
  170. udpTimeout = C.UDPTimeout
  171. }
  172. var remoteIsDomain bool
  173. if options.ControlURL != "" {
  174. controlURL, err := url.Parse(options.ControlURL)
  175. if err != nil {
  176. return nil, E.Cause(err, "parse control URL")
  177. }
  178. remoteIsDomain = M.ParseSocksaddr(controlURL.Hostname()).IsDomain()
  179. } else {
  180. // controlplane.tailscale.com
  181. remoteIsDomain = true
  182. }
  183. hasLegacyDialer := !reflect.DeepEqual(options.DialerOptions, option.DialerOptions{})
  184. hasControlHTTPClient := options.ControlHTTPClient != nil && !options.ControlHTTPClient.IsEmpty()
  185. if hasLegacyDialer && hasControlHTTPClient {
  186. return nil, E.New("control_http_client is conflict with deprecated dialer options")
  187. }
  188. controlHTTPClientOptions := common.PtrValueOrDefault(options.ControlHTTPClient)
  189. if hasLegacyDialer {
  190. deprecated.Report(ctx, deprecated.OptionLegacyTailscaleEndpointDialer)
  191. controlHTTPClientOptions.DialerOptions = options.DialerOptions
  192. }
  193. if remoteIsDomain {
  194. controlHTTPClientOptions.ResolveOnDetour = true
  195. }
  196. outboundDialer, err := dialer.NewWithOptions(dialer.Options{
  197. Context: ctx,
  198. Options: options.DialerOptions,
  199. RemoteIsDomain: remoteIsDomain,
  200. ResolverOnDetour: true,
  201. NewDialer: true,
  202. })
  203. if err != nil {
  204. return nil, err
  205. }
  206. dnsRouter := service.FromContext[adapter.DNSRouter](ctx)
  207. httpClientManager := service.FromContext[adapter.HTTPClientManager](ctx)
  208. controlTransport, err := httpClientManager.ResolveTransport(ctx, logger, controlHTTPClientOptions)
  209. if err != nil {
  210. return nil, E.Cause(err, "create control HTTP client")
  211. }
  212. controlHTTPClient := &http.Client{Transport: controlTransport}
  213. server := &tsnet.Server{
  214. Dir: stateDirectory,
  215. Hostname: hostname,
  216. Logf: func(format string, args ...any) {
  217. logger.Trace(fmt.Sprintf(format, args...))
  218. },
  219. UserLogf: func(format string, args ...any) {
  220. logger.Debug(fmt.Sprintf(format, args...))
  221. },
  222. Ephemeral: options.Ephemeral,
  223. AuthKey: options.AuthKey,
  224. ControlURL: options.ControlURL,
  225. AdvertiseTags: options.AdvertiseTags,
  226. Dialer: &endpointDialer{Dialer: outboundDialer, logger: logger},
  227. LookupHook: func(ctx context.Context, host string) ([]netip.Addr, error) {
  228. return dnsRouter.Lookup(ctx, host, outboundDialer.(dialer.ResolveDialer).QueryOptions())
  229. },
  230. DNS: &dnsConfigurtor{},
  231. HTTPClient: controlHTTPClient,
  232. }
  233. return &Endpoint{
  234. Adapter: endpoint.NewAdapter(C.TypeTailscale, tag, []string{N.NetworkTCP, N.NetworkUDP, N.NetworkICMP}, nil),
  235. ctx: ctx,
  236. router: router,
  237. logger: logger,
  238. dnsRouter: dnsRouter,
  239. network: service.FromContext[adapter.NetworkManager](ctx),
  240. platformInterface: service.FromContext[adapter.PlatformInterface](ctx),
  241. server: server,
  242. controlTransport: controlTransport,
  243. acceptRoutes: options.AcceptRoutes,
  244. exitNode: options.ExitNode,
  245. exitNodeAllowLANAccess: options.ExitNodeAllowLANAccess,
  246. advertiseRoutes: options.AdvertiseRoutes,
  247. advertiseExitNode: options.AdvertiseExitNode,
  248. advertiseTags: options.AdvertiseTags,
  249. relayServerPort: options.RelayServerPort,
  250. relayServerStaticEndpoints: options.RelayServerStaticEndpoints,
  251. udpTimeout: udpTimeout,
  252. systemInterface: options.SystemInterface,
  253. systemInterfaceName: options.SystemInterfaceName,
  254. systemInterfaceMTU: options.SystemInterfaceMTU,
  255. }, nil
  256. }
  257. func (t *Endpoint) Start(stage adapter.StartStage) error {
  258. switch stage {
  259. case adapter.StartStateStart:
  260. return t.start()
  261. case adapter.StartStatePostStart:
  262. return t.postStart()
  263. }
  264. return nil
  265. }
  266. func (t *Endpoint) start() error {
  267. if t.platformInterface != nil {
  268. err := t.network.UpdateInterfaces()
  269. if err != nil {
  270. return err
  271. }
  272. netmon.RegisterInterfaceGetter(func() ([]netmon.Interface, error) {
  273. return common.Map(t.network.InterfaceFinder().Interfaces(), func(it control.Interface) netmon.Interface {
  274. return netmon.Interface{
  275. Interface: &net.Interface{
  276. Index: it.Index,
  277. MTU: it.MTU,
  278. Name: it.Name,
  279. HardwareAddr: it.HardwareAddr,
  280. Flags: it.Flags,
  281. },
  282. AltAddrs: common.Map(it.Addresses, func(it netip.Prefix) net.Addr {
  283. return &net.IPNet{
  284. IP: it.Addr().AsSlice(),
  285. Mask: net.CIDRMask(it.Bits(), it.Addr().BitLen()),
  286. }
  287. }),
  288. }
  289. }), nil
  290. })
  291. }
  292. if t.systemInterface {
  293. mtu := t.systemInterfaceMTU
  294. if mtu == 0 {
  295. mtu = uint32(tsTUN.DefaultTUNMTU())
  296. }
  297. tunName := t.systemInterfaceName
  298. if tunName == "" {
  299. tunName = tun.CalculateInterfaceName("tailscale")
  300. }
  301. tunOptions := tun.Options{
  302. Name: tunName,
  303. MTU: mtu,
  304. GSO: true,
  305. InterfaceScope: true,
  306. InterfaceMonitor: t.network.InterfaceMonitor(),
  307. InterfaceFinder: t.network.InterfaceFinder(),
  308. Logger: t.logger,
  309. EXP_ExternalConfiguration: true,
  310. }
  311. systemTun, err := tun.New(tunOptions)
  312. if err != nil {
  313. return err
  314. }
  315. err = systemTun.Start()
  316. if err != nil {
  317. _ = systemTun.Close()
  318. return err
  319. }
  320. wgTunDevice, err := newTunDeviceAdapter(systemTun, int(mtu), t.logger)
  321. if err != nil {
  322. _ = systemTun.Close()
  323. return err
  324. }
  325. systemDialer, err := dialer.NewDefault(t.ctx, option.DialerOptions{
  326. BindInterface: tunName,
  327. })
  328. if err != nil {
  329. _ = systemTun.Close()
  330. return err
  331. }
  332. t.systemTun = systemTun
  333. t.systemDialer = systemDialer
  334. t.server.TunDevice = wgTunDevice
  335. }
  336. if mark := t.network.AutoRedirectOutputMark(); mark > 0 {
  337. controlFunc := t.network.AutoRedirectOutputMarkFunc()
  338. if bindFunc := t.network.AutoDetectInterfaceFunc(); bindFunc != nil {
  339. controlFunc = control.Append(controlFunc, bindFunc)
  340. }
  341. netns.SetControlFunc(controlFunc)
  342. } else if runtime.GOOS == "android" && t.platformInterface != nil {
  343. netns.SetControlFunc(func(network, address string, c syscall.RawConn) error {
  344. return control.Raw(c, func(fd uintptr) error {
  345. return t.platformInterface.AutoDetectInterfaceControl(int(fd))
  346. })
  347. })
  348. }
  349. return nil
  350. }
  351. func (t *Endpoint) postStart() error {
  352. err := t.server.Start()
  353. if err != nil {
  354. if t.systemTun != nil {
  355. _ = t.systemTun.Close()
  356. }
  357. return err
  358. }
  359. if t.fallbackTCPCloser == nil {
  360. t.fallbackTCPCloser = t.server.RegisterFallbackTCPHandler(func(src, dst netip.AddrPort) (handler func(net.Conn), intercept bool) {
  361. return func(conn net.Conn) {
  362. ctx := log.ContextWithNewID(t.ctx)
  363. source := M.SocksaddrFrom(src.Addr(), src.Port())
  364. destination := M.SocksaddrFrom(dst.Addr(), dst.Port())
  365. t.NewConnectionEx(ctx, conn, source, destination, nil)
  366. }, true
  367. })
  368. }
  369. t.server.ExportLocalBackend().ExportEngine().(wgengine.ExportedUserspaceEngine).SetOnReconfigListener(t.onReconfig)
  370. ipStack := t.server.ExportNetstack().ExportIPStack()
  371. gErr := ipStack.SetSpoofing(tun.DefaultNIC, true)
  372. if gErr != nil {
  373. return gonet.TranslateNetstackError(gErr)
  374. }
  375. gErr = ipStack.SetPromiscuousMode(tun.DefaultNIC, true)
  376. if gErr != nil {
  377. return gonet.TranslateNetstackError(gErr)
  378. }
  379. icmpForwarder := tun.NewICMPForwarder(t.ctx, ipStack, t, t.udpTimeout)
  380. ipStack.SetTransportProtocolHandler(icmp.ProtocolNumber4, icmpForwarder.HandlePacket)
  381. ipStack.SetTransportProtocolHandler(icmp.ProtocolNumber6, icmpForwarder.HandlePacket)
  382. t.stack = ipStack
  383. t.icmpForwarder = icmpForwarder
  384. t.registerNetstackHandlers()
  385. localBackend := t.server.ExportLocalBackend()
  386. perfs := &ipn.MaskedPrefs{
  387. Prefs: ipn.Prefs{
  388. RouteAll: t.acceptRoutes,
  389. AdvertiseRoutes: t.advertiseRoutes,
  390. },
  391. RouteAllSet: true,
  392. ExitNodeIPSet: true,
  393. AdvertiseRoutesSet: true,
  394. RelayServerPortSet: true,
  395. RelayServerStaticEndpointsSet: true,
  396. }
  397. if t.advertiseExitNode {
  398. perfs.AdvertiseRoutes = append(perfs.AdvertiseRoutes, tsaddr.ExitRoutes()...)
  399. }
  400. if t.relayServerPort != nil {
  401. perfs.RelayServerPort = t.relayServerPort
  402. }
  403. if len(t.relayServerStaticEndpoints) > 0 {
  404. perfs.RelayServerStaticEndpoints = t.relayServerStaticEndpoints
  405. }
  406. _, err = localBackend.EditPrefs(perfs)
  407. if err != nil {
  408. return E.Cause(err, "update prefs")
  409. }
  410. t.filter = localBackend.ExportFilter()
  411. go t.watchState()
  412. return nil
  413. }
  414. func (t *Endpoint) watchState() {
  415. localBackend := t.server.ExportLocalBackend()
  416. localBackend.WatchNotifications(t.ctx, ipn.NotifyInitialState, nil, func(roNotify *ipn.Notify) (keepGoing bool) {
  417. if roNotify.State != nil && *roNotify.State != ipn.NeedsLogin && *roNotify.State != ipn.NoState {
  418. return false
  419. }
  420. authURL := localBackend.StatusWithoutPeers().AuthURL
  421. if authURL != "" {
  422. t.logger.Info("Waiting for authentication: ", authURL)
  423. if t.platformInterface != nil {
  424. err := t.platformInterface.SendNotification(&adapter.Notification{
  425. Identifier: "tailscale-authentication",
  426. TypeName: "Tailscale Authentication Notifications",
  427. TypeID: 10,
  428. Title: "Tailscale Authentication",
  429. Body: F.ToString("Tailscale outbound[", t.Tag(), "] is waiting for authentication."),
  430. OpenURL: authURL,
  431. })
  432. if err != nil {
  433. t.logger.Error("send authentication notification: ", err)
  434. }
  435. }
  436. return false
  437. }
  438. return true
  439. })
  440. if t.exitNode != "" {
  441. localBackend.WatchNotifications(t.ctx, ipn.NotifyInitialState, nil, func(roNotify *ipn.Notify) (keepGoing bool) {
  442. if roNotify.State == nil || *roNotify.State != ipn.Running {
  443. return true
  444. }
  445. status, err := common.Must1(t.server.LocalClient()).Status(t.ctx)
  446. if err != nil {
  447. t.logger.Error("set exit node: ", err)
  448. return
  449. }
  450. perfs := &ipn.MaskedPrefs{
  451. Prefs: ipn.Prefs{
  452. ExitNodeAllowLANAccess: t.exitNodeAllowLANAccess,
  453. },
  454. ExitNodeIPSet: true,
  455. ExitNodeAllowLANAccessSet: true,
  456. }
  457. err = perfs.SetExitNodeIP(t.exitNode, status)
  458. if err != nil {
  459. t.logger.Error("set exit node: ", err)
  460. return true
  461. }
  462. _, err = localBackend.EditPrefs(perfs)
  463. if err != nil {
  464. t.logger.Error("set exit node: ", err)
  465. return true
  466. }
  467. return false
  468. })
  469. }
  470. }
  471. func (t *Endpoint) Close() error {
  472. err := common.Close(common.PtrOrNil(t.server), t.controlTransport)
  473. netmon.RegisterInterfaceGetter(nil)
  474. netns.SetControlFunc(nil)
  475. if t.fallbackTCPCloser != nil {
  476. t.fallbackTCPCloser()
  477. t.fallbackTCPCloser = nil
  478. }
  479. if t.systemTun != nil {
  480. t.systemTun.Close()
  481. t.systemTun = nil
  482. }
  483. return err
  484. }
  485. func (t *Endpoint) DialContext(ctx context.Context, network string, destination M.Socksaddr) (net.Conn, error) {
  486. switch network {
  487. case N.NetworkTCP:
  488. t.logger.InfoContext(ctx, "outbound connection to ", destination)
  489. case N.NetworkUDP:
  490. t.logger.InfoContext(ctx, "outbound packet connection to ", destination)
  491. }
  492. if destination.IsDomain() {
  493. destinationAddresses, err := t.dnsRouter.Lookup(ctx, destination.Fqdn, adapter.DNSQueryOptions{})
  494. if err != nil {
  495. return nil, err
  496. }
  497. return N.DialSerial(ctx, t, network, destination, destinationAddresses)
  498. }
  499. if t.systemDialer != nil {
  500. return t.systemDialer.DialContext(ctx, network, destination)
  501. }
  502. addr4, addr6 := t.server.TailscaleIPs()
  503. remoteAddr := tcpip.FullAddress{
  504. NIC: 1,
  505. Port: destination.Port,
  506. Addr: addressFromAddr(destination.Addr),
  507. }
  508. var localAddr tcpip.FullAddress
  509. var networkProtocol tcpip.NetworkProtocolNumber
  510. if destination.IsIPv4() {
  511. if !addr4.IsValid() {
  512. return nil, E.New("missing Tailscale IPv4 address")
  513. }
  514. networkProtocol = header.IPv4ProtocolNumber
  515. localAddr = tcpip.FullAddress{
  516. NIC: 1,
  517. Addr: addressFromAddr(addr4),
  518. }
  519. } else {
  520. if !addr6.IsValid() {
  521. return nil, E.New("missing Tailscale IPv6 address")
  522. }
  523. networkProtocol = header.IPv6ProtocolNumber
  524. localAddr = tcpip.FullAddress{
  525. NIC: 1,
  526. Addr: addressFromAddr(addr6),
  527. }
  528. }
  529. switch N.NetworkName(network) {
  530. case N.NetworkTCP:
  531. tcpConn, err := gonet.DialTCPWithBind(ctx, t.stack, localAddr, remoteAddr, networkProtocol)
  532. if err != nil {
  533. return nil, err
  534. }
  535. return tcpConn, nil
  536. case N.NetworkUDP:
  537. udpConn, err := gonet.DialUDP(t.stack, &localAddr, &remoteAddr, networkProtocol)
  538. if err != nil {
  539. return nil, err
  540. }
  541. return udpConn, nil
  542. default:
  543. return nil, E.Extend(N.ErrUnknownNetwork, network)
  544. }
  545. }
  546. func (t *Endpoint) listenPacketWithAddress(ctx context.Context, destination M.Socksaddr) (net.PacketConn, error) {
  547. if t.systemDialer != nil {
  548. return t.systemDialer.ListenPacket(ctx, destination)
  549. }
  550. addr4, addr6 := t.server.TailscaleIPs()
  551. bind := tcpip.FullAddress{
  552. NIC: 1,
  553. }
  554. var networkProtocol tcpip.NetworkProtocolNumber
  555. if destination.IsIPv4() {
  556. if !addr4.IsValid() {
  557. return nil, E.New("missing Tailscale IPv4 address")
  558. }
  559. networkProtocol = header.IPv4ProtocolNumber
  560. bind.Addr = addressFromAddr(addr4)
  561. } else {
  562. if !addr6.IsValid() {
  563. return nil, E.New("missing Tailscale IPv6 address")
  564. }
  565. networkProtocol = header.IPv6ProtocolNumber
  566. bind.Addr = addressFromAddr(addr6)
  567. }
  568. udpConn, err := gonet.DialUDP(t.stack, &bind, nil, networkProtocol)
  569. if err != nil {
  570. return nil, err
  571. }
  572. return udpConn, nil
  573. }
  574. func (t *Endpoint) ListenPacketWithDestination(ctx context.Context, destination M.Socksaddr) (net.PacketConn, netip.Addr, error) {
  575. t.logger.InfoContext(ctx, "outbound packet connection to ", destination)
  576. if destination.IsDomain() {
  577. destinationAddresses, err := t.dnsRouter.Lookup(ctx, destination.Fqdn, adapter.DNSQueryOptions{})
  578. if err != nil {
  579. return nil, netip.Addr{}, err
  580. }
  581. var errors []error
  582. for _, address := range destinationAddresses {
  583. packetConn, packetErr := t.listenPacketWithAddress(ctx, M.SocksaddrFrom(address, destination.Port))
  584. if packetErr == nil {
  585. return packetConn, address, nil
  586. }
  587. errors = append(errors, packetErr)
  588. }
  589. return nil, netip.Addr{}, E.Errors(errors...)
  590. }
  591. packetConn, err := t.listenPacketWithAddress(ctx, destination)
  592. if err != nil {
  593. return nil, netip.Addr{}, err
  594. }
  595. if destination.IsIP() {
  596. return packetConn, destination.Addr, nil
  597. }
  598. return packetConn, netip.Addr{}, nil
  599. }
  600. func (t *Endpoint) ListenPacket(ctx context.Context, destination M.Socksaddr) (net.PacketConn, error) {
  601. packetConn, destinationAddress, err := t.ListenPacketWithDestination(ctx, destination)
  602. if err != nil {
  603. return nil, err
  604. }
  605. if destinationAddress.IsValid() && destination != M.SocksaddrFrom(destinationAddress, destination.Port) {
  606. return bufio.NewNATPacketConn(bufio.NewPacketConn(packetConn), M.SocksaddrFrom(destinationAddress, destination.Port), destination), nil
  607. }
  608. return packetConn, nil
  609. }
  610. func (t *Endpoint) PrepareConnection(network string, source M.Socksaddr, destination M.Socksaddr, routeContext tun.DirectRouteContext, timeout time.Duration) (tun.DirectRouteDestination, error) {
  611. tsFilter := t.filter.Load()
  612. if tsFilter != nil {
  613. var ipProto ipproto.Proto
  614. switch N.NetworkName(network) {
  615. case N.NetworkTCP:
  616. ipProto = ipproto.TCP
  617. case N.NetworkUDP:
  618. ipProto = ipproto.UDP
  619. case N.NetworkICMP:
  620. if !destination.IsIPv6() {
  621. ipProto = ipproto.ICMPv4
  622. } else {
  623. ipProto = ipproto.ICMPv6
  624. }
  625. }
  626. response := tsFilter.Check(source.Addr, destination.Addr, destination.Port, ipProto)
  627. switch response {
  628. case filter.Drop:
  629. return nil, syscall.ECONNREFUSED
  630. case filter.DropSilently:
  631. return nil, tun.ErrDrop
  632. }
  633. }
  634. var ipVersion uint8
  635. if !destination.IsIPv6() {
  636. ipVersion = 4
  637. } else {
  638. ipVersion = 6
  639. }
  640. routeDestination, err := t.router.PreMatch(adapter.InboundContext{
  641. Inbound: t.Tag(),
  642. InboundType: t.Type(),
  643. IPVersion: ipVersion,
  644. Network: network,
  645. Source: source,
  646. Destination: destination,
  647. }, routeContext, timeout, false)
  648. if err != nil {
  649. switch {
  650. case rule.IsBypassed(err):
  651. err = nil
  652. case rule.IsRejected(err):
  653. t.logger.Trace("reject ", network, " connection from ", source.AddrString(), " to ", destination.AddrString())
  654. default:
  655. if network == N.NetworkICMP {
  656. t.logger.Warn(E.Cause(err, "link ", network, " connection from ", source.AddrString(), " to ", destination.AddrString()))
  657. }
  658. }
  659. }
  660. return routeDestination, err
  661. }
  662. func (t *Endpoint) NewConnectionEx(ctx context.Context, conn net.Conn, source M.Socksaddr, destination M.Socksaddr, onClose N.CloseHandlerFunc) {
  663. var metadata adapter.InboundContext
  664. metadata.Inbound = t.Tag()
  665. metadata.InboundType = t.Type()
  666. metadata.Source = source
  667. addr4, addr6 := t.server.TailscaleIPs()
  668. switch destination.Addr {
  669. case addr4:
  670. destination.Addr = netip.AddrFrom4([4]uint8{127, 0, 0, 1})
  671. case addr6:
  672. destination.Addr = netip.IPv6Loopback()
  673. }
  674. metadata.Destination = destination
  675. t.logger.InfoContext(ctx, "inbound connection from ", source)
  676. t.logger.InfoContext(ctx, "inbound connection to ", metadata.Destination)
  677. t.router.RouteConnectionEx(ctx, conn, metadata, onClose)
  678. }
  679. func (t *Endpoint) NewPacketConnectionEx(ctx context.Context, conn N.PacketConn, source M.Socksaddr, destination M.Socksaddr, onClose N.CloseHandlerFunc) {
  680. var metadata adapter.InboundContext
  681. metadata.Inbound = t.Tag()
  682. metadata.InboundType = t.Type()
  683. metadata.Source = source
  684. addr4, addr6 := t.server.TailscaleIPs()
  685. switch destination.Addr {
  686. case addr4:
  687. metadata.OriginDestination = destination
  688. destination.Addr = netip.AddrFrom4([4]uint8{127, 0, 0, 1})
  689. conn = bufio.NewNATPacketConn(bufio.NewNetPacketConn(conn), metadata.OriginDestination, destination)
  690. case addr6:
  691. metadata.OriginDestination = destination
  692. destination.Addr = netip.IPv6Loopback()
  693. conn = bufio.NewNATPacketConn(bufio.NewNetPacketConn(conn), metadata.OriginDestination, destination)
  694. }
  695. metadata.Destination = destination
  696. t.logger.InfoContext(ctx, "inbound packet connection from ", source)
  697. t.logger.InfoContext(ctx, "inbound packet connection to ", metadata.Destination)
  698. t.router.RoutePacketConnectionEx(ctx, conn, metadata, onClose)
  699. }
  700. func (t *Endpoint) NewDirectRouteConnection(metadata adapter.InboundContext, routeContext tun.DirectRouteContext, timeout time.Duration) (tun.DirectRouteDestination, error) {
  701. ctx := log.ContextWithNewID(t.ctx)
  702. var destination tun.DirectRouteDestination
  703. var err error
  704. if t.systemDialer != nil {
  705. destination, err = ping.ConnectDestination(
  706. ctx, t.logger,
  707. t.systemDialer.DialerForICMPDestination(metadata.Destination.Addr).Control,
  708. metadata.Destination.Addr, routeContext, timeout,
  709. )
  710. } else {
  711. inet4Address, inet6Address := t.server.TailscaleIPs()
  712. if metadata.Destination.Addr.Is4() && !inet4Address.IsValid() || metadata.Destination.Addr.Is6() && !inet6Address.IsValid() {
  713. return nil, E.New("Tailscale is not ready yet")
  714. }
  715. destination, err = ping.ConnectGVisor(
  716. ctx, t.logger,
  717. metadata.Source.Addr, metadata.Destination.Addr,
  718. routeContext,
  719. t.stack,
  720. inet4Address, inet6Address,
  721. timeout,
  722. )
  723. }
  724. if err != nil {
  725. return nil, err
  726. }
  727. t.logger.InfoContext(ctx, "linked ", metadata.Network, " connection from ", metadata.Source.AddrString(), " to ", metadata.Destination.AddrString())
  728. return destination, nil
  729. }
  730. func (t *Endpoint) PreferredDomain(domain string) bool {
  731. routeDomains := t.routeDomains.Load()
  732. if routeDomains == nil {
  733. return false
  734. }
  735. return routeDomains[strings.ToLower(domain)]
  736. }
  737. func (t *Endpoint) PreferredAddress(address netip.Addr) bool {
  738. routePrefixes := t.routePrefixes.Load()
  739. if routePrefixes == nil {
  740. return false
  741. }
  742. return routePrefixes.Contains(address)
  743. }
  744. func (t *Endpoint) Server() *tsnet.Server {
  745. return t.server
  746. }
  747. func (t *Endpoint) onReconfig(cfg *wgcfg.Config, routerCfg *router.Config, dnsCfg *tsDNS.Config) {
  748. if cfg == nil || dnsCfg == nil {
  749. return
  750. }
  751. if (t.cfg != nil && reflect.DeepEqual(t.cfg, cfg)) && (t.dnsCfg != nil && reflect.DeepEqual(t.dnsCfg, dnsCfg)) {
  752. return
  753. }
  754. var inet4Address, inet6Address netip.Addr
  755. for _, address := range cfg.Addresses {
  756. if address.Addr().Is4() {
  757. inet4Address = address.Addr()
  758. } else if address.Addr().Is6() {
  759. inet6Address = address.Addr()
  760. }
  761. }
  762. t.icmpForwarder.SetLocalAddresses(inet4Address, inet6Address)
  763. t.cfg = cfg
  764. t.dnsCfg = dnsCfg
  765. routeDomains := make(map[string]bool)
  766. for fqdn := range dnsCfg.Routes {
  767. routeDomains[fqdn.WithoutTrailingDot()] = true
  768. }
  769. for _, fqdn := range dnsCfg.SearchDomains {
  770. routeDomains[fqdn.WithoutTrailingDot()] = true
  771. }
  772. t.routeDomains.Store(routeDomains)
  773. var builder netipx.IPSetBuilder
  774. for _, peer := range cfg.Peers {
  775. for _, allowedIP := range peer.AllowedIPs {
  776. builder.AddPrefix(allowedIP)
  777. }
  778. }
  779. t.routePrefixes.Store(common.Must1(builder.IPSet()))
  780. if t.onReconfigHook != nil {
  781. t.onReconfigHook(cfg, routerCfg, dnsCfg)
  782. }
  783. }
  784. func addressFromAddr(destination netip.Addr) tcpip.Address {
  785. if destination.Is6() {
  786. return tcpip.AddrFrom16(destination.As16())
  787. } else {
  788. return tcpip.AddrFrom4(destination.As4())
  789. }
  790. }
  791. type endpointDialer struct {
  792. N.Dialer
  793. logger logger.ContextLogger
  794. }
  795. func (d *endpointDialer) DialContext(ctx context.Context, network string, destination M.Socksaddr) (net.Conn, error) {
  796. switch N.NetworkName(network) {
  797. case N.NetworkTCP:
  798. d.logger.InfoContext(ctx, "output connection to ", destination)
  799. case N.NetworkUDP:
  800. d.logger.InfoContext(ctx, "output packet connection to ", destination)
  801. }
  802. return d.Dialer.DialContext(ctx, network, destination)
  803. }
  804. func (d *endpointDialer) ListenPacket(ctx context.Context, destination M.Socksaddr) (net.PacketConn, error) {
  805. d.logger.InfoContext(ctx, "output packet connection")
  806. return d.Dialer.ListenPacket(ctx, destination)
  807. }
  808. type dnsConfigurtor struct {
  809. baseConfig tsDNS.OSConfig
  810. }
  811. func (c *dnsConfigurtor) SetDNS(cfg tsDNS.OSConfig) error {
  812. c.baseConfig = cfg
  813. return nil
  814. }
  815. func (c *dnsConfigurtor) SupportsSplitDNS() bool {
  816. return true
  817. }
  818. func (c *dnsConfigurtor) GetBaseConfig() (tsDNS.OSConfig, error) {
  819. return c.baseConfig, nil
  820. }
  821. func (c *dnsConfigurtor) Close() error {
  822. return nil
  823. }