tun_linux.go 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867
  1. //go:build !android && !e2e_testing
  2. // +build !android,!e2e_testing
  3. package overlay
  4. import (
  5. "fmt"
  6. "io"
  7. "net"
  8. "net/netip"
  9. "os"
  10. "strings"
  11. "sync/atomic"
  12. "time"
  13. "unsafe"
  14. "github.com/gaissmai/bart"
  15. "github.com/sirupsen/logrus"
  16. "github.com/slackhq/nebula/config"
  17. "github.com/slackhq/nebula/routing"
  18. "github.com/slackhq/nebula/util"
  19. "github.com/vishvananda/netlink"
  20. "golang.org/x/sys/unix"
  21. )
  22. type tun struct {
  23. io.ReadWriteCloser
  24. fd int
  25. Device string
  26. vpnNetworks []netip.Prefix
  27. MaxMTU int
  28. DefaultMTU int
  29. TXQueueLen int
  30. deviceIndex int
  31. ioctlFd uintptr
  32. enableVnetHdr bool
  33. vnetHdrLen int
  34. queues []*tunQueue
  35. Routes atomic.Pointer[[]Route]
  36. routeTree atomic.Pointer[bart.Table[routing.Gateways]]
  37. routeChan chan struct{}
  38. useSystemRoutes bool
  39. useSystemRoutesBufferSize int
  40. l *logrus.Logger
  41. }
  42. func (t *tun) Networks() []netip.Prefix {
  43. return t.vpnNetworks
  44. }
  45. type ifReq struct {
  46. Name [16]byte
  47. Flags uint16
  48. pad [8]byte
  49. }
  50. type ifreqMTU struct {
  51. Name [16]byte
  52. MTU int32
  53. pad [8]byte
  54. }
  55. type ifreqQLEN struct {
  56. Name [16]byte
  57. Value int32
  58. pad [8]byte
  59. }
  60. const (
  61. virtioNetHdrLen = 12
  62. tunDefaultMaxPacket = 65536
  63. )
  64. type tunQueue struct {
  65. file *os.File
  66. fd int
  67. enableVnetHdr bool
  68. vnetHdrLen int
  69. maxPacket int
  70. writeScratch []byte
  71. readScratch []byte
  72. l *logrus.Logger
  73. }
  74. func newTunQueue(file *os.File, enableVnetHdr bool, vnetHdrLen, maxPacket int, l *logrus.Logger) *tunQueue {
  75. if maxPacket <= 0 {
  76. maxPacket = tunDefaultMaxPacket
  77. }
  78. q := &tunQueue{
  79. file: file,
  80. fd: int(file.Fd()),
  81. enableVnetHdr: enableVnetHdr,
  82. vnetHdrLen: vnetHdrLen,
  83. maxPacket: maxPacket,
  84. l: l,
  85. }
  86. if enableVnetHdr {
  87. q.growReadScratch(maxPacket)
  88. }
  89. return q
  90. }
  91. func (q *tunQueue) growReadScratch(packetSize int) {
  92. needed := q.vnetHdrLen + packetSize
  93. if needed < q.vnetHdrLen+DefaultMTU {
  94. needed = q.vnetHdrLen + DefaultMTU
  95. }
  96. if q.readScratch == nil || cap(q.readScratch) < needed {
  97. q.readScratch = make([]byte, needed)
  98. } else {
  99. q.readScratch = q.readScratch[:needed]
  100. }
  101. }
  102. func (q *tunQueue) setMaxPacket(packet int) {
  103. if packet <= 0 {
  104. packet = DefaultMTU
  105. }
  106. q.maxPacket = packet
  107. if q.enableVnetHdr {
  108. q.growReadScratch(packet)
  109. }
  110. }
  111. func configureVnetHdr(fd int, hdrLen int, l *logrus.Logger) error {
  112. features, err := unix.IoctlGetInt(fd, unix.TUNGETFEATURES)
  113. if err == nil && features&unix.IFF_VNET_HDR == 0 {
  114. return fmt.Errorf("kernel does not support IFF_VNET_HDR")
  115. }
  116. if err := unix.IoctlSetInt(fd, unix.TUNSETVNETHDRSZ, hdrLen); err != nil {
  117. return err
  118. }
  119. offload := unix.TUN_F_CSUM | unix.TUN_F_UFO
  120. if err := unix.IoctlSetInt(fd, unix.TUNSETOFFLOAD, offload); err != nil {
  121. if l != nil {
  122. l.WithError(err).Warn("Failed to enable TUN offload features")
  123. }
  124. }
  125. return nil
  126. }
  127. func newTunFromFd(c *config.C, l *logrus.Logger, deviceFd int, vpnNetworks []netip.Prefix) (*tun, error) {
  128. file := os.NewFile(uintptr(deviceFd), "/dev/net/tun")
  129. enableVnetHdr := c.GetBool("tun.enable_vnet_hdr", false)
  130. if enableVnetHdr {
  131. if err := configureVnetHdr(deviceFd, virtioNetHdrLen, l); err != nil {
  132. l.WithError(err).Warn("Failed to configure VNET header support on provided tun fd; disabling")
  133. enableVnetHdr = false
  134. }
  135. }
  136. t, err := newTunGeneric(c, l, file, vpnNetworks, enableVnetHdr)
  137. if err != nil {
  138. return nil, err
  139. }
  140. t.Device = "tun0"
  141. return t, nil
  142. }
  143. func newTun(c *config.C, l *logrus.Logger, vpnNetworks []netip.Prefix, multiqueue bool) (*tun, error) {
  144. fd, err := unix.Open("/dev/net/tun", os.O_RDWR, 0)
  145. if err != nil {
  146. // If /dev/net/tun doesn't exist, try to create it (will happen in docker)
  147. if os.IsNotExist(err) {
  148. err = os.MkdirAll("/dev/net", 0755)
  149. if err != nil {
  150. return nil, fmt.Errorf("/dev/net/tun doesn't exist, failed to mkdir -p /dev/net: %w", err)
  151. }
  152. err = unix.Mknod("/dev/net/tun", unix.S_IFCHR|0600, int(unix.Mkdev(10, 200)))
  153. if err != nil {
  154. return nil, fmt.Errorf("failed to create /dev/net/tun: %w", err)
  155. }
  156. fd, err = unix.Open("/dev/net/tun", os.O_RDWR, 0)
  157. if err != nil {
  158. return nil, fmt.Errorf("created /dev/net/tun, but still failed: %w", err)
  159. }
  160. } else {
  161. return nil, err
  162. }
  163. }
  164. var req ifReq
  165. req.Flags = uint16(unix.IFF_TUN | unix.IFF_NO_PI)
  166. if multiqueue {
  167. req.Flags |= unix.IFF_MULTI_QUEUE
  168. }
  169. enableVnetHdr := c.GetBool("tun.enable_vnet_hdr", false)
  170. if enableVnetHdr {
  171. req.Flags |= unix.IFF_VNET_HDR
  172. }
  173. copy(req.Name[:], c.GetString("tun.dev", ""))
  174. if err = ioctl(uintptr(fd), uintptr(unix.TUNSETIFF), uintptr(unsafe.Pointer(&req))); err != nil {
  175. return nil, err
  176. }
  177. name := strings.Trim(string(req.Name[:]), "\x00")
  178. if enableVnetHdr {
  179. if err := configureVnetHdr(fd, virtioNetHdrLen, l); err != nil {
  180. l.WithError(err).Warn("Failed to configure VNET header support on tun device; disabling")
  181. enableVnetHdr = false
  182. }
  183. }
  184. file := os.NewFile(uintptr(fd), "/dev/net/tun")
  185. t, err := newTunGeneric(c, l, file, vpnNetworks, enableVnetHdr)
  186. if err != nil {
  187. return nil, err
  188. }
  189. t.Device = name
  190. return t, nil
  191. }
  192. func newTunGeneric(c *config.C, l *logrus.Logger, file *os.File, vpnNetworks []netip.Prefix, enableVnetHdr bool) (*tun, error) {
  193. queue := newTunQueue(file, enableVnetHdr, virtioNetHdrLen, tunDefaultMaxPacket, l)
  194. t := &tun{
  195. ReadWriteCloser: queue,
  196. fd: int(file.Fd()),
  197. vpnNetworks: vpnNetworks,
  198. TXQueueLen: c.GetInt("tun.tx_queue", 500),
  199. useSystemRoutes: c.GetBool("tun.use_system_route_table", false),
  200. useSystemRoutesBufferSize: c.GetInt("tun.use_system_route_table_buffer_size", 0),
  201. l: l,
  202. enableVnetHdr: enableVnetHdr,
  203. vnetHdrLen: virtioNetHdrLen,
  204. queues: []*tunQueue{queue},
  205. }
  206. err := t.reload(c, true)
  207. if err != nil {
  208. return nil, err
  209. }
  210. if enableVnetHdr {
  211. for _, q := range t.queues {
  212. q.setMaxPacket(t.MaxMTU)
  213. }
  214. }
  215. c.RegisterReloadCallback(func(c *config.C) {
  216. err := t.reload(c, false)
  217. if err != nil {
  218. util.LogWithContextIfNeeded("failed to reload tun device", err, t.l)
  219. }
  220. })
  221. return t, nil
  222. }
  223. func (t *tun) reload(c *config.C, initial bool) error {
  224. routeChange, routes, err := getAllRoutesFromConfig(c, t.vpnNetworks, initial)
  225. if err != nil {
  226. return err
  227. }
  228. if !initial && !routeChange && !c.HasChanged("tun.mtu") {
  229. return nil
  230. }
  231. routeTree, err := makeRouteTree(t.l, routes, true)
  232. if err != nil {
  233. return err
  234. }
  235. oldDefaultMTU := t.DefaultMTU
  236. oldMaxMTU := t.MaxMTU
  237. newDefaultMTU := c.GetInt("tun.mtu", DefaultMTU)
  238. newMaxMTU := newDefaultMTU
  239. for i, r := range routes {
  240. if r.MTU == 0 {
  241. routes[i].MTU = newDefaultMTU
  242. }
  243. if r.MTU > t.MaxMTU {
  244. newMaxMTU = r.MTU
  245. }
  246. }
  247. t.MaxMTU = newMaxMTU
  248. t.DefaultMTU = newDefaultMTU
  249. if t.enableVnetHdr {
  250. for _, q := range t.queues {
  251. q.setMaxPacket(t.MaxMTU)
  252. }
  253. }
  254. // Teach nebula how to handle the routes before establishing them in the system table
  255. oldRoutes := t.Routes.Swap(&routes)
  256. t.routeTree.Store(routeTree)
  257. if !initial {
  258. if oldMaxMTU != newMaxMTU {
  259. t.setMTU()
  260. t.l.Infof("Set max MTU to %v was %v", t.MaxMTU, oldMaxMTU)
  261. }
  262. if oldDefaultMTU != newDefaultMTU {
  263. for i := range t.vpnNetworks {
  264. err := t.setDefaultRoute(t.vpnNetworks[i])
  265. if err != nil {
  266. t.l.Warn(err)
  267. } else {
  268. t.l.Infof("Set default MTU to %v was %v", t.DefaultMTU, oldDefaultMTU)
  269. }
  270. }
  271. }
  272. // Remove first, if the system removes a wanted route hopefully it will be re-added next
  273. t.removeRoutes(findRemovedRoutes(routes, *oldRoutes))
  274. // Ensure any routes we actually want are installed
  275. err = t.addRoutes(true)
  276. if err != nil {
  277. // This should never be called since addRoutes should log its own errors in a reload condition
  278. util.LogWithContextIfNeeded("Failed to refresh routes", err, t.l)
  279. }
  280. }
  281. return nil
  282. }
  283. func (t *tun) NewMultiQueueReader() (io.ReadWriteCloser, error) {
  284. fd, err := unix.Open("/dev/net/tun", os.O_RDWR, 0)
  285. if err != nil {
  286. return nil, err
  287. }
  288. var req ifReq
  289. req.Flags = uint16(unix.IFF_TUN | unix.IFF_NO_PI | unix.IFF_MULTI_QUEUE)
  290. if t.enableVnetHdr {
  291. req.Flags |= unix.IFF_VNET_HDR
  292. }
  293. copy(req.Name[:], t.Device)
  294. if err = ioctl(uintptr(fd), uintptr(unix.TUNSETIFF), uintptr(unsafe.Pointer(&req))); err != nil {
  295. return nil, err
  296. }
  297. file := os.NewFile(uintptr(fd), "/dev/net/tun")
  298. queue := newTunQueue(file, t.enableVnetHdr, t.vnetHdrLen, t.MaxMTU, t.l)
  299. if t.enableVnetHdr {
  300. if err := configureVnetHdr(fd, t.vnetHdrLen, t.l); err != nil {
  301. queue.enableVnetHdr = false
  302. }
  303. }
  304. t.queues = append(t.queues, queue)
  305. return queue, nil
  306. }
  307. func (q *tunQueue) Read(p []byte) (int, error) {
  308. if !q.enableVnetHdr {
  309. return q.file.Read(p)
  310. }
  311. if len(p)+q.vnetHdrLen > cap(q.readScratch) {
  312. q.growReadScratch(len(p))
  313. }
  314. buf := q.readScratch[:cap(q.readScratch)]
  315. n, err := q.file.Read(buf)
  316. if n <= 0 {
  317. return n, err
  318. }
  319. if n < q.vnetHdrLen {
  320. if err == nil {
  321. err = io.ErrUnexpectedEOF
  322. }
  323. return 0, err
  324. }
  325. payload := buf[q.vnetHdrLen:n]
  326. if len(payload) > len(p) {
  327. copy(p, payload[:len(p)])
  328. if err == nil {
  329. err = io.ErrShortBuffer
  330. }
  331. return len(p), err
  332. }
  333. copy(p, payload)
  334. return len(payload), err
  335. }
  336. func (q *tunQueue) Write(b []byte) (int, error) {
  337. if !q.enableVnetHdr {
  338. return unix.Write(q.fd, b)
  339. }
  340. total := q.vnetHdrLen + len(b)
  341. if cap(q.writeScratch) < total {
  342. q.writeScratch = make([]byte, total)
  343. } else {
  344. q.writeScratch = q.writeScratch[:total]
  345. }
  346. for i := 0; i < q.vnetHdrLen; i++ {
  347. q.writeScratch[i] = 0
  348. }
  349. copy(q.writeScratch[q.vnetHdrLen:], b)
  350. n, err := unix.Write(q.fd, q.writeScratch)
  351. if n >= q.vnetHdrLen {
  352. n -= q.vnetHdrLen
  353. } else {
  354. n = 0
  355. }
  356. return n, err
  357. }
  358. func (q *tunQueue) Close() error {
  359. return q.file.Close()
  360. }
  361. func (t *tun) RoutesFor(ip netip.Addr) routing.Gateways {
  362. r, _ := t.routeTree.Load().Lookup(ip)
  363. return r
  364. }
  365. func (t *tun) Write(b []byte) (int, error) {
  366. var nn int
  367. maximum := len(b)
  368. for {
  369. n, err := unix.Write(t.fd, b[nn:maximum])
  370. if n > 0 {
  371. nn += n
  372. }
  373. if nn == len(b) {
  374. return nn, err
  375. }
  376. if err != nil {
  377. return nn, err
  378. }
  379. if n == 0 {
  380. return nn, io.ErrUnexpectedEOF
  381. }
  382. }
  383. }
  384. func (t *tun) deviceBytes() (o [16]byte) {
  385. for i, c := range t.Device {
  386. o[i] = byte(c)
  387. }
  388. return
  389. }
  390. func hasNetlinkAddr(al []*netlink.Addr, x netlink.Addr) bool {
  391. for i := range al {
  392. if al[i].Equal(x) {
  393. return true
  394. }
  395. }
  396. return false
  397. }
  398. // addIPs uses netlink to add all addresses that don't exist, then it removes ones that should not be there
  399. func (t *tun) addIPs(link netlink.Link) error {
  400. newAddrs := make([]*netlink.Addr, len(t.vpnNetworks))
  401. for i := range t.vpnNetworks {
  402. newAddrs[i] = &netlink.Addr{
  403. IPNet: &net.IPNet{
  404. IP: t.vpnNetworks[i].Addr().AsSlice(),
  405. Mask: net.CIDRMask(t.vpnNetworks[i].Bits(), t.vpnNetworks[i].Addr().BitLen()),
  406. },
  407. Label: t.vpnNetworks[i].Addr().Zone(),
  408. }
  409. }
  410. //add all new addresses
  411. for i := range newAddrs {
  412. //AddrReplace still adds new IPs, but if their properties change it will change them as well
  413. if err := netlink.AddrReplace(link, newAddrs[i]); err != nil {
  414. return err
  415. }
  416. }
  417. //iterate over remainder, remove whoever shouldn't be there
  418. al, err := netlink.AddrList(link, netlink.FAMILY_ALL)
  419. if err != nil {
  420. return fmt.Errorf("failed to get tun address list: %s", err)
  421. }
  422. for i := range al {
  423. if hasNetlinkAddr(newAddrs, al[i]) {
  424. continue
  425. }
  426. err = netlink.AddrDel(link, &al[i])
  427. if err != nil {
  428. t.l.WithError(err).Error("failed to remove address from tun address list")
  429. } else {
  430. t.l.WithField("removed", al[i].String()).Info("removed address not listed in cert(s)")
  431. }
  432. }
  433. return nil
  434. }
  435. func (t *tun) Activate() error {
  436. devName := t.deviceBytes()
  437. if t.useSystemRoutes {
  438. t.watchRoutes()
  439. }
  440. s, err := unix.Socket(
  441. unix.AF_INET, //because everything we use t.ioctlFd for is address family independent, this is fine
  442. unix.SOCK_DGRAM,
  443. unix.IPPROTO_IP,
  444. )
  445. if err != nil {
  446. return err
  447. }
  448. t.ioctlFd = uintptr(s)
  449. // Set the device name
  450. ifrf := ifReq{Name: devName}
  451. if err = ioctl(t.ioctlFd, unix.SIOCGIFFLAGS, uintptr(unsafe.Pointer(&ifrf))); err != nil {
  452. return fmt.Errorf("failed to set tun device name: %s", err)
  453. }
  454. link, err := netlink.LinkByName(t.Device)
  455. if err != nil {
  456. return fmt.Errorf("failed to get tun device link: %s", err)
  457. }
  458. t.deviceIndex = link.Attrs().Index
  459. // Setup our default MTU
  460. t.setMTU()
  461. // Set the transmit queue length
  462. ifrq := ifreqQLEN{Name: devName, Value: int32(t.TXQueueLen)}
  463. if err = ioctl(t.ioctlFd, unix.SIOCSIFTXQLEN, uintptr(unsafe.Pointer(&ifrq))); err != nil {
  464. // If we can't set the queue length nebula will still work but it may lead to packet loss
  465. t.l.WithError(err).Error("Failed to set tun tx queue length")
  466. }
  467. const modeNone = 1
  468. if err = netlink.LinkSetIP6AddrGenMode(link, modeNone); err != nil {
  469. t.l.WithError(err).Warn("Failed to disable link local address generation")
  470. }
  471. if err = t.addIPs(link); err != nil {
  472. return err
  473. }
  474. // Bring up the interface
  475. ifrf.Flags = ifrf.Flags | unix.IFF_UP
  476. if err = ioctl(t.ioctlFd, unix.SIOCSIFFLAGS, uintptr(unsafe.Pointer(&ifrf))); err != nil {
  477. return fmt.Errorf("failed to bring the tun device up: %s", err)
  478. }
  479. //set route MTU
  480. for i := range t.vpnNetworks {
  481. if err = t.setDefaultRoute(t.vpnNetworks[i]); err != nil {
  482. return fmt.Errorf("failed to set default route MTU: %w", err)
  483. }
  484. }
  485. // Set the routes
  486. if err = t.addRoutes(false); err != nil {
  487. return err
  488. }
  489. // Run the interface
  490. ifrf.Flags = ifrf.Flags | unix.IFF_UP | unix.IFF_RUNNING
  491. if err = ioctl(t.ioctlFd, unix.SIOCSIFFLAGS, uintptr(unsafe.Pointer(&ifrf))); err != nil {
  492. return fmt.Errorf("failed to run tun device: %s", err)
  493. }
  494. return nil
  495. }
  496. func (t *tun) setMTU() {
  497. // Set the MTU on the device
  498. ifm := ifreqMTU{Name: t.deviceBytes(), MTU: int32(t.MaxMTU)}
  499. if err := ioctl(t.ioctlFd, unix.SIOCSIFMTU, uintptr(unsafe.Pointer(&ifm))); err != nil {
  500. // This is currently a non fatal condition because the route table must have the MTU set appropriately as well
  501. t.l.WithError(err).Error("Failed to set tun mtu")
  502. }
  503. }
  504. func (t *tun) setDefaultRoute(cidr netip.Prefix) error {
  505. dr := &net.IPNet{
  506. IP: cidr.Masked().Addr().AsSlice(),
  507. Mask: net.CIDRMask(cidr.Bits(), cidr.Addr().BitLen()),
  508. }
  509. nr := netlink.Route{
  510. LinkIndex: t.deviceIndex,
  511. Dst: dr,
  512. MTU: t.DefaultMTU,
  513. AdvMSS: t.advMSS(Route{}),
  514. Scope: unix.RT_SCOPE_LINK,
  515. Src: net.IP(cidr.Addr().AsSlice()),
  516. Protocol: unix.RTPROT_KERNEL,
  517. Table: unix.RT_TABLE_MAIN,
  518. Type: unix.RTN_UNICAST,
  519. }
  520. err := netlink.RouteReplace(&nr)
  521. if err != nil {
  522. t.l.WithError(err).WithField("cidr", cidr).Warn("Failed to set default route MTU, retrying")
  523. //retry twice more -- on some systems there appears to be a race condition where if we set routes too soon, netlink says `invalid argument`
  524. for i := 0; i < 2; i++ {
  525. time.Sleep(100 * time.Millisecond)
  526. err = netlink.RouteReplace(&nr)
  527. if err == nil {
  528. break
  529. } else {
  530. t.l.WithError(err).WithField("cidr", cidr).WithField("mtu", t.DefaultMTU).Warn("Failed to set default route MTU, retrying")
  531. }
  532. }
  533. if err != nil {
  534. return fmt.Errorf("failed to set mtu %v on the default route %v; %v", t.DefaultMTU, dr, err)
  535. }
  536. }
  537. return nil
  538. }
  539. func (t *tun) addRoutes(logErrors bool) error {
  540. // Path routes
  541. routes := *t.Routes.Load()
  542. for _, r := range routes {
  543. if !r.Install {
  544. continue
  545. }
  546. dr := &net.IPNet{
  547. IP: r.Cidr.Masked().Addr().AsSlice(),
  548. Mask: net.CIDRMask(r.Cidr.Bits(), r.Cidr.Addr().BitLen()),
  549. }
  550. nr := netlink.Route{
  551. LinkIndex: t.deviceIndex,
  552. Dst: dr,
  553. MTU: r.MTU,
  554. AdvMSS: t.advMSS(r),
  555. Scope: unix.RT_SCOPE_LINK,
  556. }
  557. if r.Metric > 0 {
  558. nr.Priority = r.Metric
  559. }
  560. err := netlink.RouteReplace(&nr)
  561. if err != nil {
  562. retErr := util.NewContextualError("Failed to add route", map[string]any{"route": r}, err)
  563. if logErrors {
  564. retErr.Log(t.l)
  565. } else {
  566. return retErr
  567. }
  568. } else {
  569. t.l.WithField("route", r).Info("Added route")
  570. }
  571. }
  572. return nil
  573. }
  574. func (t *tun) removeRoutes(routes []Route) {
  575. for _, r := range routes {
  576. if !r.Install {
  577. continue
  578. }
  579. dr := &net.IPNet{
  580. IP: r.Cidr.Masked().Addr().AsSlice(),
  581. Mask: net.CIDRMask(r.Cidr.Bits(), r.Cidr.Addr().BitLen()),
  582. }
  583. nr := netlink.Route{
  584. LinkIndex: t.deviceIndex,
  585. Dst: dr,
  586. MTU: r.MTU,
  587. AdvMSS: t.advMSS(r),
  588. Scope: unix.RT_SCOPE_LINK,
  589. }
  590. if r.Metric > 0 {
  591. nr.Priority = r.Metric
  592. }
  593. err := netlink.RouteDel(&nr)
  594. if err != nil {
  595. t.l.WithError(err).WithField("route", r).Error("Failed to remove route")
  596. } else {
  597. t.l.WithField("route", r).Info("Removed route")
  598. }
  599. }
  600. }
  601. func (t *tun) Name() string {
  602. return t.Device
  603. }
  604. func (t *tun) advMSS(r Route) int {
  605. mtu := r.MTU
  606. if r.MTU == 0 {
  607. mtu = t.DefaultMTU
  608. }
  609. // We only need to set advmss if the route MTU does not match the device MTU
  610. if mtu != t.MaxMTU {
  611. return mtu - 40
  612. }
  613. return 0
  614. }
  615. func (t *tun) watchRoutes() {
  616. rch := make(chan netlink.RouteUpdate)
  617. doneChan := make(chan struct{})
  618. netlinkOptions := netlink.RouteSubscribeOptions{
  619. ReceiveBufferSize: t.useSystemRoutesBufferSize,
  620. ReceiveBufferForceSize: t.useSystemRoutesBufferSize != 0,
  621. ErrorCallback: func(e error) { t.l.WithError(e).Errorf("netlink error") },
  622. }
  623. if err := netlink.RouteSubscribeWithOptions(rch, doneChan, netlinkOptions); err != nil {
  624. t.l.WithError(err).Errorf("failed to subscribe to system route changes")
  625. return
  626. }
  627. t.routeChan = doneChan
  628. go func() {
  629. for {
  630. select {
  631. case r, ok := <-rch:
  632. if ok {
  633. t.updateRoutes(r)
  634. } else {
  635. // may be should do something here as
  636. // netlink stops sending updates
  637. return
  638. }
  639. case <-doneChan:
  640. // netlink.RouteSubscriber will close the rch for us
  641. return
  642. }
  643. }
  644. }()
  645. }
  646. func (t *tun) isGatewayInVpnNetworks(gwAddr netip.Addr) bool {
  647. withinNetworks := false
  648. for i := range t.vpnNetworks {
  649. if t.vpnNetworks[i].Contains(gwAddr) {
  650. withinNetworks = true
  651. break
  652. }
  653. }
  654. return withinNetworks
  655. }
  656. func (t *tun) getGatewaysFromRoute(r *netlink.Route) routing.Gateways {
  657. var gateways routing.Gateways
  658. link, err := netlink.LinkByName(t.Device)
  659. if err != nil {
  660. t.l.WithField("Devicename", t.Device).Error("Ignoring route update: failed to get link by name")
  661. return gateways
  662. }
  663. // If this route is relevant to our interface and there is a gateway then add it
  664. if r.LinkIndex == link.Attrs().Index && len(r.Gw) > 0 {
  665. gwAddr, ok := netip.AddrFromSlice(r.Gw)
  666. if !ok {
  667. t.l.WithField("route", r).Debug("Ignoring route update, invalid gateway address")
  668. } else {
  669. gwAddr = gwAddr.Unmap()
  670. if !t.isGatewayInVpnNetworks(gwAddr) {
  671. // Gateway isn't in our overlay network, ignore
  672. t.l.WithField("route", r).Debug("Ignoring route update, not in our network")
  673. } else {
  674. gateways = append(gateways, routing.NewGateway(gwAddr, 1))
  675. }
  676. }
  677. }
  678. for _, p := range r.MultiPath {
  679. // If this route is relevant to our interface and there is a gateway then add it
  680. if p.LinkIndex == link.Attrs().Index && len(p.Gw) > 0 {
  681. gwAddr, ok := netip.AddrFromSlice(p.Gw)
  682. if !ok {
  683. t.l.WithField("route", r).Debug("Ignoring multipath route update, invalid gateway address")
  684. } else {
  685. gwAddr = gwAddr.Unmap()
  686. if !t.isGatewayInVpnNetworks(gwAddr) {
  687. // Gateway isn't in our overlay network, ignore
  688. t.l.WithField("route", r).Debug("Ignoring route update, not in our network")
  689. } else {
  690. // p.Hops+1 = weight of the route
  691. gateways = append(gateways, routing.NewGateway(gwAddr, p.Hops+1))
  692. }
  693. }
  694. }
  695. }
  696. routing.CalculateBucketsForGateways(gateways)
  697. return gateways
  698. }
  699. func (t *tun) updateRoutes(r netlink.RouteUpdate) {
  700. gateways := t.getGatewaysFromRoute(&r.Route)
  701. if len(gateways) == 0 {
  702. // No gateways relevant to our network, no routing changes required.
  703. t.l.WithField("route", r).Debug("Ignoring route update, no gateways")
  704. return
  705. }
  706. if r.Dst == nil {
  707. t.l.WithField("route", r).Debug("Ignoring route update, no destination address")
  708. return
  709. }
  710. dstAddr, ok := netip.AddrFromSlice(r.Dst.IP)
  711. if !ok {
  712. t.l.WithField("route", r).Debug("Ignoring route update, invalid destination address")
  713. return
  714. }
  715. ones, _ := r.Dst.Mask.Size()
  716. dst := netip.PrefixFrom(dstAddr, ones)
  717. newTree := t.routeTree.Load().Clone()
  718. if r.Type == unix.RTM_NEWROUTE {
  719. t.l.WithField("destination", dst).WithField("via", gateways).Info("Adding route")
  720. newTree.Insert(dst, gateways)
  721. } else {
  722. t.l.WithField("destination", dst).WithField("via", gateways).Info("Removing route")
  723. newTree.Delete(dst)
  724. }
  725. t.routeTree.Store(newTree)
  726. }
  727. func (t *tun) Close() error {
  728. if t.routeChan != nil {
  729. close(t.routeChan)
  730. }
  731. if t.ReadWriteCloser != nil {
  732. _ = t.ReadWriteCloser.Close()
  733. }
  734. if t.ioctlFd > 0 {
  735. _ = os.NewFile(t.ioctlFd, "ioctlFd").Close()
  736. }
  737. return nil
  738. }