1
0

tun_linux.go 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977
  1. //go:build !android && !e2e_testing
  2. // +build !android,!e2e_testing
  3. package overlay
  4. import (
  5. "encoding/binary"
  6. "errors"
  7. "fmt"
  8. "io"
  9. "log/slog"
  10. "net"
  11. "net/netip"
  12. "os"
  13. "strings"
  14. "sync"
  15. "sync/atomic"
  16. "time"
  17. "unsafe"
  18. "github.com/gaissmai/bart"
  19. "github.com/slackhq/nebula/config"
  20. "github.com/slackhq/nebula/routing"
  21. "github.com/slackhq/nebula/util"
  22. "github.com/vishvananda/netlink"
  23. "golang.org/x/sys/unix"
  24. )
  25. // tunFile wraps a TUN file descriptor with poll-based reads. The FD provided will be changed to non-blocking.
  26. // A shared eventfd allows Close to wake all readers blocked in poll.
  27. type tunFile struct {
  28. fd int
  29. shutdownFd int
  30. lastOne bool
  31. readPoll [2]unix.PollFd
  32. writePoll [2]unix.PollFd
  33. closed bool
  34. }
  35. // newFriend makes a tunFile for a MultiQueueReader that copies the shutdown eventfd from the parent tun
  36. func (r *tunFile) newFriend(fd int) (*tunFile, error) {
  37. if err := unix.SetNonblock(fd, true); err != nil {
  38. return nil, fmt.Errorf("failed to set tun fd non-blocking: %w", err)
  39. }
  40. return &tunFile{
  41. fd: fd,
  42. shutdownFd: r.shutdownFd,
  43. readPoll: [2]unix.PollFd{
  44. {Fd: int32(fd), Events: unix.POLLIN},
  45. {Fd: int32(r.shutdownFd), Events: unix.POLLIN},
  46. },
  47. writePoll: [2]unix.PollFd{
  48. {Fd: int32(fd), Events: unix.POLLOUT},
  49. {Fd: int32(r.shutdownFd), Events: unix.POLLIN},
  50. },
  51. }, nil
  52. }
  53. func newTunFd(fd int) (*tunFile, error) {
  54. if err := unix.SetNonblock(fd, true); err != nil {
  55. return nil, fmt.Errorf("failed to set tun fd non-blocking: %w", err)
  56. }
  57. shutdownFd, err := unix.Eventfd(0, unix.EFD_NONBLOCK|unix.EFD_CLOEXEC)
  58. if err != nil {
  59. return nil, fmt.Errorf("failed to create eventfd: %w", err)
  60. }
  61. out := &tunFile{
  62. fd: fd,
  63. shutdownFd: shutdownFd,
  64. lastOne: true,
  65. readPoll: [2]unix.PollFd{
  66. {Fd: int32(fd), Events: unix.POLLIN},
  67. {Fd: int32(shutdownFd), Events: unix.POLLIN},
  68. },
  69. writePoll: [2]unix.PollFd{
  70. {Fd: int32(fd), Events: unix.POLLOUT},
  71. {Fd: int32(shutdownFd), Events: unix.POLLIN},
  72. },
  73. }
  74. return out, nil
  75. }
  76. func (r *tunFile) blockOnRead() error {
  77. const problemFlags = unix.POLLHUP | unix.POLLNVAL | unix.POLLERR
  78. var err error
  79. for {
  80. _, err = unix.Poll(r.readPoll[:], -1)
  81. if err != unix.EINTR {
  82. break
  83. }
  84. }
  85. //always reset these!
  86. tunEvents := r.readPoll[0].Revents
  87. shutdownEvents := r.readPoll[1].Revents
  88. r.readPoll[0].Revents = 0
  89. r.readPoll[1].Revents = 0
  90. //do the err check before trusting the potentially bogus bits we just got
  91. if err != nil {
  92. return err
  93. }
  94. if shutdownEvents&(unix.POLLIN|problemFlags) != 0 {
  95. return os.ErrClosed
  96. } else if tunEvents&problemFlags != 0 {
  97. return os.ErrClosed
  98. }
  99. return nil
  100. }
  101. func (r *tunFile) blockOnWrite() error {
  102. const problemFlags = unix.POLLHUP | unix.POLLNVAL | unix.POLLERR
  103. var err error
  104. for {
  105. _, err = unix.Poll(r.writePoll[:], -1)
  106. if err != unix.EINTR {
  107. break
  108. }
  109. }
  110. //always reset these!
  111. tunEvents := r.writePoll[0].Revents
  112. shutdownEvents := r.writePoll[1].Revents
  113. r.writePoll[0].Revents = 0
  114. r.writePoll[1].Revents = 0
  115. //do the err check before trusting the potentially bogus bits we just got
  116. if err != nil {
  117. return err
  118. }
  119. if shutdownEvents&(unix.POLLIN|problemFlags) != 0 {
  120. return os.ErrClosed
  121. } else if tunEvents&problemFlags != 0 {
  122. return os.ErrClosed
  123. }
  124. return nil
  125. }
  126. func (r *tunFile) Read(buf []byte) (int, error) {
  127. for {
  128. if n, err := unix.Read(r.fd, buf); err == nil {
  129. return n, nil
  130. } else if err == unix.EAGAIN {
  131. if err = r.blockOnRead(); err != nil {
  132. return 0, err
  133. }
  134. continue
  135. } else if err == unix.EINTR {
  136. continue
  137. } else if err == unix.EBADF {
  138. return 0, os.ErrClosed
  139. } else {
  140. return 0, err
  141. }
  142. }
  143. }
  144. func (r *tunFile) Write(buf []byte) (int, error) {
  145. for {
  146. if n, err := unix.Write(r.fd, buf); err == nil {
  147. return n, nil
  148. } else if err == unix.EAGAIN {
  149. if err = r.blockOnWrite(); err != nil {
  150. return 0, err
  151. }
  152. continue
  153. } else if err == unix.EINTR {
  154. continue
  155. } else if err == unix.EBADF {
  156. return 0, os.ErrClosed
  157. } else {
  158. return 0, err
  159. }
  160. }
  161. }
  162. func (r *tunFile) wakeForShutdown() error {
  163. var buf [8]byte
  164. binary.NativeEndian.PutUint64(buf[:], 1)
  165. _, err := unix.Write(int(r.readPoll[1].Fd), buf[:])
  166. return err
  167. }
  168. func (r *tunFile) Close() error {
  169. if r.closed { // avoid closing more than once. Technically a fd could get re-used, which would be a problem
  170. return nil
  171. }
  172. r.closed = true
  173. if r.lastOne {
  174. _ = unix.Close(r.shutdownFd)
  175. }
  176. return unix.Close(r.fd)
  177. }
  178. type tun struct {
  179. *tunFile
  180. readers []*tunFile
  181. closeLock sync.Mutex
  182. Device string
  183. vpnNetworks []netip.Prefix
  184. MaxMTU int
  185. DefaultMTU int
  186. TXQueueLen int
  187. deviceIndex int
  188. ioctlFd uintptr
  189. Routes atomic.Pointer[[]Route]
  190. routeTree atomic.Pointer[bart.Table[routing.Gateways]]
  191. routeChan chan struct{}
  192. useSystemRoutes bool
  193. useSystemRoutesBufferSize int
  194. // These are routes learned from `tun.use_system_route_table`
  195. // stored here to make it easier to restore them after a reload
  196. routesFromSystem map[netip.Prefix]routing.Gateways
  197. routesFromSystemLock sync.Mutex
  198. l *slog.Logger
  199. }
  200. func (t *tun) Networks() []netip.Prefix {
  201. return t.vpnNetworks
  202. }
  203. type ifReq struct {
  204. Name [16]byte
  205. Flags uint16
  206. pad [8]byte
  207. }
  208. type ifreqMTU struct {
  209. Name [16]byte
  210. MTU int32
  211. pad [8]byte
  212. }
  213. type ifreqQLEN struct {
  214. Name [16]byte
  215. Value int32
  216. pad [8]byte
  217. }
  218. func newTunFromFd(c *config.C, l *slog.Logger, deviceFd int, vpnNetworks []netip.Prefix) (*tun, error) {
  219. t, err := newTunGeneric(c, l, deviceFd, vpnNetworks)
  220. if err != nil {
  221. return nil, err
  222. }
  223. t.Device = "tun0"
  224. return t, nil
  225. }
  226. func newTun(c *config.C, l *slog.Logger, vpnNetworks []netip.Prefix, multiqueue bool) (*tun, error) {
  227. // Resolve (and validate) the device name up front so a bad tun.dev fails
  228. // fast, before we open /dev/net/tun or leak a file descriptor.
  229. tunName, err := findNextTunName(c.GetString("tun.dev", "nebula%d"))
  230. if err != nil {
  231. return nil, err
  232. }
  233. fd, err := unix.Open("/dev/net/tun", os.O_RDWR, 0)
  234. if err != nil {
  235. // If /dev/net/tun doesn't exist, try to create it (will happen in docker)
  236. if os.IsNotExist(err) {
  237. err = os.MkdirAll("/dev/net", 0755)
  238. if err != nil {
  239. return nil, fmt.Errorf("/dev/net/tun doesn't exist, failed to mkdir -p /dev/net: %w", err)
  240. }
  241. err = unix.Mknod("/dev/net/tun", unix.S_IFCHR|0600, int(unix.Mkdev(10, 200)))
  242. if err != nil {
  243. return nil, fmt.Errorf("failed to create /dev/net/tun: %w", err)
  244. }
  245. fd, err = unix.Open("/dev/net/tun", os.O_RDWR, 0)
  246. if err != nil {
  247. return nil, fmt.Errorf("created /dev/net/tun, but still failed: %w", err)
  248. }
  249. } else {
  250. return nil, err
  251. }
  252. }
  253. var req ifReq
  254. req.Flags = uint16(unix.IFF_TUN | unix.IFF_NO_PI)
  255. if multiqueue {
  256. req.Flags |= unix.IFF_MULTI_QUEUE
  257. }
  258. copy(req.Name[:], tunName)
  259. if err = ioctl(uintptr(fd), uintptr(unix.TUNSETIFF), uintptr(unsafe.Pointer(&req))); err != nil {
  260. _ = unix.Close(fd)
  261. return nil, &NameError{
  262. Name: tunName,
  263. Underlying: err,
  264. }
  265. }
  266. name := strings.Trim(string(req.Name[:]), "\x00")
  267. t, err := newTunGeneric(c, l, fd, vpnNetworks)
  268. if err != nil {
  269. return nil, err
  270. }
  271. t.Device = name
  272. return t, nil
  273. }
  274. func validateTunName(tunName string) error {
  275. if !strings.Contains(tunName, "%d") {
  276. if len(tunName) >= unix.IFNAMSIZ {
  277. return fmt.Errorf("tun.dev %q is not shorter than the maximum device name length of %d", tunName, unix.IFNAMSIZ)
  278. }
  279. return nil
  280. }
  281. if strings.Count(tunName, "%d") > 1 {
  282. return fmt.Errorf("tun.dev template %q may only contain a single %%d", tunName)
  283. }
  284. if tunName == "%d" {
  285. return errors.New("please don't name your tun device '%d'")
  286. }
  287. // The shortest name a template can produce replaces %d with a single digit;
  288. // if even that is not shorter than IFNAMSIZ the template can never yield a
  289. // usable name.
  290. if len(tunName)-len("%d")+len("0") >= unix.IFNAMSIZ {
  291. return fmt.Errorf("tun.dev template %q would result in a name that is not shorter than the maximum device name length of %d", tunName, unix.IFNAMSIZ)
  292. }
  293. return nil
  294. }
  295. // findNextTunName resolves a tun.dev value into a concrete device name. A value
  296. // without a "%d" is returned unchanged; a "%d" placeholder (anywhere in the
  297. // name) has the lowest unused integer substituted in based on the devices
  298. // currently present.
  299. func findNextTunName(tunName string) (string, error) {
  300. if err := validateTunName(tunName); err != nil {
  301. return "", err
  302. }
  303. if !strings.Contains(tunName, "%d") {
  304. return tunName, nil
  305. }
  306. links, err := netlink.LinkList()
  307. if err != nil {
  308. return "", err
  309. }
  310. used := make(map[string]struct{}, len(links))
  311. for _, link := range links {
  312. used[link.Attrs().Name] = struct{}{}
  313. }
  314. return nextTunName(tunName, used)
  315. }
  316. // nextTunName substitutes the lowest unused integer into a template's "%d"
  317. // placeholder, skipping any name present in used. tunName is assumed to have
  318. // already passed validateTunName (exactly one "%d", room for a digit). It errors
  319. // only if every candidate that is shorter than IFNAMSIZ is already taken.
  320. func nextTunName(tunName string, used map[string]struct{}) (string, error) {
  321. prefix, suffix, _ := strings.Cut(tunName, "%d")
  322. for i := 0; ; i++ {
  323. candidateName := fmt.Sprintf("%s%d%s", prefix, i, suffix)
  324. if len(candidateName) >= unix.IFNAMSIZ {
  325. return "", fmt.Errorf("all device names matching template %q shorter than the maximum length of %d are already in use", tunName, unix.IFNAMSIZ)
  326. }
  327. if _, taken := used[candidateName]; !taken {
  328. return candidateName, nil
  329. }
  330. }
  331. }
  332. // newTunGeneric does all the stuff common to different tun initialization paths. It will close your files on error.
  333. func newTunGeneric(c *config.C, l *slog.Logger, fd int, vpnNetworks []netip.Prefix) (*tun, error) {
  334. tfd, err := newTunFd(fd)
  335. if err != nil {
  336. _ = unix.Close(fd)
  337. return nil, err
  338. }
  339. t := &tun{
  340. tunFile: tfd,
  341. readers: []*tunFile{tfd},
  342. closeLock: sync.Mutex{},
  343. vpnNetworks: vpnNetworks,
  344. TXQueueLen: c.GetInt("tun.tx_queue", 500),
  345. useSystemRoutes: c.GetBool("tun.use_system_route_table", false),
  346. useSystemRoutesBufferSize: c.GetInt("tun.use_system_route_table_buffer_size", 0),
  347. routesFromSystem: map[netip.Prefix]routing.Gateways{},
  348. l: l,
  349. }
  350. if err = t.reload(c, true); err != nil {
  351. _ = t.Close()
  352. return nil, err
  353. }
  354. c.RegisterReloadCallback(func(c *config.C) {
  355. err := t.reload(c, false)
  356. if err != nil {
  357. util.LogWithContextIfNeeded("failed to reload tun device", err, t.l)
  358. }
  359. })
  360. return t, nil
  361. }
  362. func (t *tun) reload(c *config.C, initial bool) error {
  363. routeChange, routes, err := getAllRoutesFromConfig(c, t.vpnNetworks, initial)
  364. if err != nil {
  365. return err
  366. }
  367. if !initial && !routeChange && !c.HasChanged("tun.mtu") {
  368. return nil
  369. }
  370. routeTree, err := makeRouteTree(t.l, routes, true)
  371. if err != nil {
  372. return err
  373. }
  374. // Bring along any routes learned from the system route table on reload
  375. t.routesFromSystemLock.Lock()
  376. for dst, gw := range t.routesFromSystem {
  377. routeTree.Insert(dst, gw)
  378. }
  379. t.routesFromSystemLock.Unlock()
  380. oldDefaultMTU := t.DefaultMTU
  381. oldMaxMTU := t.MaxMTU
  382. newDefaultMTU := c.GetInt("tun.mtu", DefaultMTU)
  383. newMaxMTU := newDefaultMTU
  384. for i, r := range routes {
  385. if r.MTU == 0 {
  386. routes[i].MTU = newDefaultMTU
  387. }
  388. if r.MTU > t.MaxMTU {
  389. newMaxMTU = r.MTU
  390. }
  391. }
  392. t.MaxMTU = newMaxMTU
  393. t.DefaultMTU = newDefaultMTU
  394. // Teach nebula how to handle the routes before establishing them in the system table
  395. oldRoutes := t.Routes.Swap(&routes)
  396. t.routeTree.Store(routeTree)
  397. if !initial {
  398. if oldMaxMTU != newMaxMTU {
  399. t.setMTU()
  400. t.l.Info("Set max MTU", "mtu", t.MaxMTU, "oldMTU", oldMaxMTU)
  401. }
  402. if oldDefaultMTU != newDefaultMTU {
  403. for i := range t.vpnNetworks {
  404. err := t.setDefaultRoute(t.vpnNetworks[i])
  405. if err != nil {
  406. t.l.Warn(err.Error())
  407. } else {
  408. t.l.Info("Set default MTU", "mtu", t.DefaultMTU, "oldMTU", oldDefaultMTU)
  409. }
  410. }
  411. }
  412. // Remove first, if the system removes a wanted route hopefully it will be re-added next
  413. t.removeRoutes(findRemovedRoutes(routes, *oldRoutes))
  414. // Ensure any routes we actually want are installed
  415. err = t.addRoutes(true)
  416. if err != nil {
  417. // This should never be called since addRoutes should log its own errors in a reload condition
  418. util.LogWithContextIfNeeded("Failed to refresh routes", err, t.l)
  419. }
  420. }
  421. return nil
  422. }
  423. func (t *tun) SupportsMultiqueue() bool {
  424. return true
  425. }
  426. func (t *tun) NewMultiQueueReader() (io.ReadWriteCloser, error) {
  427. t.closeLock.Lock()
  428. defer t.closeLock.Unlock()
  429. fd, err := unix.Open("/dev/net/tun", os.O_RDWR, 0)
  430. if err != nil {
  431. return nil, err
  432. }
  433. var req ifReq
  434. req.Flags = uint16(unix.IFF_TUN | unix.IFF_NO_PI | unix.IFF_MULTI_QUEUE)
  435. copy(req.Name[:], t.Device)
  436. if err = ioctl(uintptr(fd), uintptr(unix.TUNSETIFF), uintptr(unsafe.Pointer(&req))); err != nil {
  437. _ = unix.Close(fd)
  438. return nil, err
  439. }
  440. out, err := t.tunFile.newFriend(fd)
  441. if err != nil {
  442. _ = unix.Close(fd)
  443. return nil, err
  444. }
  445. t.readers = append(t.readers, out)
  446. return out, nil
  447. }
  448. func (t *tun) RoutesFor(ip netip.Addr) routing.Gateways {
  449. r, _ := t.routeTree.Load().Lookup(ip)
  450. return r
  451. }
  452. func (t *tun) deviceBytes() (o [16]byte) {
  453. for i, c := range t.Device {
  454. o[i] = byte(c)
  455. }
  456. return
  457. }
  458. func hasNetlinkAddr(al []*netlink.Addr, x netlink.Addr) bool {
  459. for i := range al {
  460. if al[i].Equal(x) {
  461. return true
  462. }
  463. }
  464. return false
  465. }
  466. // addIPs uses netlink to add all addresses that don't exist, then it removes ones that should not be there
  467. func (t *tun) addIPs(link netlink.Link) error {
  468. newAddrs := make([]*netlink.Addr, len(t.vpnNetworks))
  469. for i := range t.vpnNetworks {
  470. newAddrs[i] = &netlink.Addr{
  471. IPNet: &net.IPNet{
  472. IP: t.vpnNetworks[i].Addr().AsSlice(),
  473. Mask: net.CIDRMask(t.vpnNetworks[i].Bits(), t.vpnNetworks[i].Addr().BitLen()),
  474. },
  475. Label: t.vpnNetworks[i].Addr().Zone(),
  476. }
  477. }
  478. //add all new addresses
  479. for i := range newAddrs {
  480. //AddrReplace still adds new IPs, but if their properties change it will change them as well
  481. if err := netlink.AddrReplace(link, newAddrs[i]); err != nil {
  482. return err
  483. }
  484. }
  485. //iterate over remainder, remove whoever shouldn't be there
  486. al, err := netlink.AddrList(link, netlink.FAMILY_ALL)
  487. if err != nil {
  488. return fmt.Errorf("failed to get tun address list: %s", err)
  489. }
  490. for i := range al {
  491. if hasNetlinkAddr(newAddrs, al[i]) {
  492. continue
  493. }
  494. err = netlink.AddrDel(link, &al[i])
  495. if err != nil {
  496. t.l.Error("failed to remove address from tun address list", "error", err)
  497. } else {
  498. t.l.Info("removed address not listed in cert(s)", "removed", al[i].String())
  499. }
  500. }
  501. return nil
  502. }
  503. func (t *tun) Activate() error {
  504. devName := t.deviceBytes()
  505. if t.useSystemRoutes {
  506. t.watchRoutes()
  507. }
  508. s, err := unix.Socket(
  509. unix.AF_INET, //because everything we use t.ioctlFd for is address family independent, this is fine
  510. unix.SOCK_DGRAM,
  511. unix.IPPROTO_IP,
  512. )
  513. if err != nil {
  514. return err
  515. }
  516. t.ioctlFd = uintptr(s)
  517. // Set the device name
  518. ifrf := ifReq{Name: devName}
  519. if err = ioctl(t.ioctlFd, unix.SIOCGIFFLAGS, uintptr(unsafe.Pointer(&ifrf))); err != nil {
  520. return fmt.Errorf("failed to set tun device name: %s", err)
  521. }
  522. link, err := netlink.LinkByName(t.Device)
  523. if err != nil {
  524. return fmt.Errorf("failed to get tun device link: %s", err)
  525. }
  526. t.deviceIndex = link.Attrs().Index
  527. // Setup our default MTU
  528. t.setMTU()
  529. // Set the transmit queue length
  530. ifrq := ifreqQLEN{Name: devName, Value: int32(t.TXQueueLen)}
  531. if err = ioctl(t.ioctlFd, unix.SIOCSIFTXQLEN, uintptr(unsafe.Pointer(&ifrq))); err != nil {
  532. // If we can't set the queue length nebula will still work but it may lead to packet loss
  533. t.l.Error("Failed to set tun tx queue length", "error", err)
  534. }
  535. const modeNone = 1
  536. if err = netlink.LinkSetIP6AddrGenMode(link, modeNone); err != nil {
  537. t.l.Warn("Failed to disable link local address generation", "error", err)
  538. }
  539. if err = t.addIPs(link); err != nil {
  540. return err
  541. }
  542. // Bring up the interface
  543. ifrf.Flags = ifrf.Flags | unix.IFF_UP
  544. if err = ioctl(t.ioctlFd, unix.SIOCSIFFLAGS, uintptr(unsafe.Pointer(&ifrf))); err != nil {
  545. return fmt.Errorf("failed to bring the tun device up: %s", err)
  546. }
  547. //set route MTU
  548. for i := range t.vpnNetworks {
  549. if err = t.setDefaultRoute(t.vpnNetworks[i]); err != nil {
  550. return fmt.Errorf("failed to set default route MTU: %w", err)
  551. }
  552. }
  553. // Set the routes
  554. if err = t.addRoutes(false); err != nil {
  555. return err
  556. }
  557. // Run the interface
  558. ifrf.Flags = ifrf.Flags | unix.IFF_UP | unix.IFF_RUNNING
  559. if err = ioctl(t.ioctlFd, unix.SIOCSIFFLAGS, uintptr(unsafe.Pointer(&ifrf))); err != nil {
  560. return fmt.Errorf("failed to run tun device: %s", err)
  561. }
  562. return nil
  563. }
  564. func (t *tun) setMTU() {
  565. // Set the MTU on the device
  566. ifm := ifreqMTU{Name: t.deviceBytes(), MTU: int32(t.MaxMTU)}
  567. if err := ioctl(t.ioctlFd, unix.SIOCSIFMTU, uintptr(unsafe.Pointer(&ifm))); err != nil {
  568. // This is currently a non fatal condition because the route table must have the MTU set appropriately as well
  569. t.l.Error("Failed to set tun mtu", "error", err)
  570. }
  571. }
  572. func (t *tun) setDefaultRoute(cidr netip.Prefix) error {
  573. dr := &net.IPNet{
  574. IP: cidr.Masked().Addr().AsSlice(),
  575. Mask: net.CIDRMask(cidr.Bits(), cidr.Addr().BitLen()),
  576. }
  577. nr := netlink.Route{
  578. LinkIndex: t.deviceIndex,
  579. Dst: dr,
  580. MTU: t.DefaultMTU,
  581. AdvMSS: t.advMSS(Route{}),
  582. Scope: unix.RT_SCOPE_LINK,
  583. Src: net.IP(cidr.Addr().AsSlice()),
  584. Protocol: unix.RTPROT_KERNEL,
  585. Table: unix.RT_TABLE_MAIN,
  586. Type: unix.RTN_UNICAST,
  587. }
  588. err := netlink.RouteReplace(&nr)
  589. if err != nil {
  590. t.l.Warn("Failed to set default route MTU, retrying", "error", err, "cidr", cidr)
  591. //retry twice more -- on some systems there appears to be a race condition where if we set routes too soon, netlink says `invalid argument`
  592. for i := 0; i < 2; i++ {
  593. time.Sleep(100 * time.Millisecond)
  594. err = netlink.RouteReplace(&nr)
  595. if err == nil {
  596. break
  597. } else {
  598. t.l.Warn("Failed to set default route MTU, retrying",
  599. "error", err,
  600. "cidr", cidr,
  601. "mtu", t.DefaultMTU,
  602. )
  603. }
  604. }
  605. if err != nil {
  606. return fmt.Errorf("failed to set mtu %v on the default route %v; %v", t.DefaultMTU, dr, err)
  607. }
  608. }
  609. return nil
  610. }
  611. func (t *tun) addRoutes(logErrors bool) error {
  612. // Path routes
  613. routes := *t.Routes.Load()
  614. for _, r := range routes {
  615. if !r.Install {
  616. continue
  617. }
  618. dr := &net.IPNet{
  619. IP: r.Cidr.Masked().Addr().AsSlice(),
  620. Mask: net.CIDRMask(r.Cidr.Bits(), r.Cidr.Addr().BitLen()),
  621. }
  622. nr := netlink.Route{
  623. LinkIndex: t.deviceIndex,
  624. Dst: dr,
  625. MTU: r.MTU,
  626. AdvMSS: t.advMSS(r),
  627. Scope: unix.RT_SCOPE_LINK,
  628. }
  629. if r.Metric > 0 {
  630. nr.Priority = r.Metric
  631. }
  632. err := netlink.RouteReplace(&nr)
  633. if err != nil {
  634. retErr := util.NewContextualError("Failed to add route", map[string]any{"route": r}, err)
  635. if logErrors {
  636. retErr.Log(t.l)
  637. } else {
  638. return retErr
  639. }
  640. } else {
  641. t.l.Info("Added route", "route", r)
  642. }
  643. }
  644. return nil
  645. }
  646. func (t *tun) removeRoutes(routes []Route) {
  647. for _, r := range routes {
  648. if !r.Install {
  649. continue
  650. }
  651. dr := &net.IPNet{
  652. IP: r.Cidr.Masked().Addr().AsSlice(),
  653. Mask: net.CIDRMask(r.Cidr.Bits(), r.Cidr.Addr().BitLen()),
  654. }
  655. nr := netlink.Route{
  656. LinkIndex: t.deviceIndex,
  657. Dst: dr,
  658. MTU: r.MTU,
  659. AdvMSS: t.advMSS(r),
  660. Scope: unix.RT_SCOPE_LINK,
  661. }
  662. if r.Metric > 0 {
  663. nr.Priority = r.Metric
  664. }
  665. err := netlink.RouteDel(&nr)
  666. if err != nil {
  667. t.l.Error("Failed to remove route", "error", err, "route", r)
  668. } else {
  669. t.l.Info("Removed route", "route", r)
  670. }
  671. }
  672. }
  673. func (t *tun) Name() string {
  674. return t.Device
  675. }
  676. func (t *tun) advMSS(r Route) int {
  677. mtu := r.MTU
  678. if r.MTU == 0 {
  679. mtu = t.DefaultMTU
  680. }
  681. // We only need to set advmss if the route MTU does not match the device MTU
  682. if mtu != t.MaxMTU {
  683. return mtu - 40
  684. }
  685. return 0
  686. }
  687. func (t *tun) watchRoutes() {
  688. rch := make(chan netlink.RouteUpdate)
  689. doneChan := make(chan struct{})
  690. netlinkOptions := netlink.RouteSubscribeOptions{
  691. ReceiveBufferSize: t.useSystemRoutesBufferSize,
  692. ReceiveBufferForceSize: t.useSystemRoutesBufferSize != 0,
  693. ErrorCallback: func(e error) { t.l.Error("netlink error", "error", e) },
  694. }
  695. if err := netlink.RouteSubscribeWithOptions(rch, doneChan, netlinkOptions); err != nil {
  696. t.l.Error("failed to subscribe to system route changes", "error", err)
  697. return
  698. }
  699. t.routeChan = doneChan
  700. go func() {
  701. for {
  702. select {
  703. case r, ok := <-rch:
  704. if ok {
  705. t.updateRoutes(r)
  706. } else {
  707. // may be should do something here as
  708. // netlink stops sending updates
  709. return
  710. }
  711. case <-doneChan:
  712. // netlink.RouteSubscriber will close the rch for us
  713. return
  714. }
  715. }
  716. }()
  717. }
  718. func (t *tun) isGatewayInVpnNetworks(gwAddr netip.Addr) bool {
  719. withinNetworks := false
  720. for i := range t.vpnNetworks {
  721. if t.vpnNetworks[i].Contains(gwAddr) {
  722. withinNetworks = true
  723. break
  724. }
  725. }
  726. return withinNetworks
  727. }
  728. func (t *tun) getGatewaysFromRoute(r *netlink.Route) routing.Gateways {
  729. var gateways routing.Gateways
  730. link, err := netlink.LinkByName(t.Device)
  731. if err != nil {
  732. t.l.Error("Ignoring route update: failed to get link by name", "deviceName", t.Device)
  733. return gateways
  734. }
  735. // If this route is relevant to our interface and there is a gateway then add it
  736. if r.LinkIndex == link.Attrs().Index {
  737. gwAddr, ok := getGatewayAddr(r.Gw, r.Via)
  738. if ok {
  739. if t.isGatewayInVpnNetworks(gwAddr) {
  740. gateways = append(gateways, routing.NewGateway(gwAddr, 1))
  741. } else {
  742. // Gateway isn't in our overlay network, ignore
  743. t.l.Debug("Ignoring route update, gateway is not in our network", "route", r)
  744. }
  745. } else {
  746. t.l.Debug("Ignoring route update, invalid gateway or via address", "route", r)
  747. }
  748. }
  749. for _, p := range r.MultiPath {
  750. // If this route is relevant to our interface and there is a gateway then add it
  751. if p.LinkIndex == link.Attrs().Index {
  752. gwAddr, ok := getGatewayAddr(p.Gw, p.Via)
  753. if ok {
  754. if t.isGatewayInVpnNetworks(gwAddr) {
  755. gateways = append(gateways, routing.NewGateway(gwAddr, p.Hops+1))
  756. } else {
  757. // Gateway isn't in our overlay network, ignore
  758. t.l.Debug("Ignoring route update, gateway is not in our network", "route", r)
  759. }
  760. } else {
  761. t.l.Debug("Ignoring route update, invalid gateway or via address", "route", r)
  762. }
  763. }
  764. }
  765. routing.CalculateBucketsForGateways(gateways)
  766. return gateways
  767. }
  768. func getGatewayAddr(gw net.IP, via netlink.Destination) (netip.Addr, bool) {
  769. // Try to use the old RTA_GATEWAY first
  770. gwAddr, ok := netip.AddrFromSlice(gw)
  771. if !ok {
  772. // Fallback to the new RTA_VIA
  773. rVia, ok := via.(*netlink.Via)
  774. if ok {
  775. gwAddr, ok = netip.AddrFromSlice(rVia.Addr)
  776. }
  777. }
  778. if gwAddr.IsValid() {
  779. gwAddr = gwAddr.Unmap()
  780. return gwAddr, true
  781. }
  782. return netip.Addr{}, false
  783. }
  784. func (t *tun) updateRoutes(r netlink.RouteUpdate) {
  785. gateways := t.getGatewaysFromRoute(&r.Route)
  786. if len(gateways) == 0 {
  787. // No gateways relevant to our network, no routing changes required.
  788. t.l.Debug("Ignoring route update, no gateways", "route", r)
  789. return
  790. }
  791. if r.Dst == nil {
  792. t.l.Debug("Ignoring route update, no destination address", "route", r)
  793. return
  794. }
  795. dstAddr, ok := netip.AddrFromSlice(r.Dst.IP)
  796. if !ok {
  797. t.l.Debug("Ignoring route update, invalid destination address", "route", r)
  798. return
  799. }
  800. ones, _ := r.Dst.Mask.Size()
  801. dst := netip.PrefixFrom(dstAddr, ones)
  802. newTree := t.routeTree.Load().Clone()
  803. t.routesFromSystemLock.Lock()
  804. if r.Type == unix.RTM_NEWROUTE {
  805. t.l.Info("Adding route", "destination", dst, "via", gateways)
  806. t.routesFromSystem[dst] = gateways
  807. newTree.Insert(dst, gateways)
  808. } else {
  809. t.l.Info("Removing route", "destination", dst, "via", gateways)
  810. delete(t.routesFromSystem, dst)
  811. newTree.Delete(dst)
  812. }
  813. t.routesFromSystemLock.Unlock()
  814. t.routeTree.Store(newTree)
  815. }
  816. func (t *tun) Close() error {
  817. t.closeLock.Lock()
  818. defer t.closeLock.Unlock()
  819. if t.routeChan != nil {
  820. close(t.routeChan)
  821. t.routeChan = nil
  822. }
  823. // Signal all readers blocked in poll to wake up and exit
  824. _ = t.tunFile.wakeForShutdown()
  825. if t.ioctlFd > 0 {
  826. _ = unix.Close(int(t.ioctlFd))
  827. t.ioctlFd = 0
  828. }
  829. for i := range t.readers {
  830. if i == 0 {
  831. continue //we want to close the zeroth reader last
  832. }
  833. err := t.readers[i].Close()
  834. if err != nil {
  835. t.l.Error("error closing tun reader", "reader", i, "error", err)
  836. } else {
  837. t.l.Info("closed tun reader", "reader", i)
  838. }
  839. }
  840. //this is t.readers[0] too
  841. err := t.tunFile.Close()
  842. if err != nil {
  843. t.l.Error("error closing tun reader", "reader", 0, "error", err)
  844. } else {
  845. t.l.Info("closed tun reader", "reader", 0)
  846. }
  847. return err
  848. }