hostmap.go 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797
  1. package nebula
  2. import (
  3. "errors"
  4. "net"
  5. "net/netip"
  6. "sync"
  7. "sync/atomic"
  8. "time"
  9. "github.com/gaissmai/bart"
  10. "github.com/rcrowley/go-metrics"
  11. "github.com/sirupsen/logrus"
  12. "github.com/slackhq/nebula/cert"
  13. "github.com/slackhq/nebula/config"
  14. "github.com/slackhq/nebula/header"
  15. )
  16. // const ProbeLen = 100
  17. const defaultPromoteEvery = 1000 // Count of packets sent before we try moving a tunnel to a preferred underlay ip address
  18. const defaultReQueryEvery = 5000 // Count of packets sent before re-querying a hostinfo to the lighthouse
  19. const defaultReQueryWait = time.Minute // Minimum amount of seconds to wait before re-querying a hostinfo the lighthouse. Evaluated every ReQueryEvery
  20. const MaxRemotes = 10
  21. const maxRecvError = 4
  22. // MaxHostInfosPerVpnIp is the max number of hostinfos we will track for a given vpn ip
  23. // 5 allows for an initial handshake and each host pair re-handshaking twice
  24. const MaxHostInfosPerVpnIp = 5
  25. // How long we should prevent roaming back to the previous IP.
  26. // This helps prevent flapping due to packets already in flight
  27. const RoamingSuppressSeconds = 2
  28. const (
  29. Requested = iota
  30. PeerRequested
  31. Established
  32. Disestablished
  33. )
  34. const (
  35. Unknowntype = iota
  36. ForwardingType
  37. TerminalType
  38. )
  39. type Relay struct {
  40. Type int
  41. State int
  42. LocalIndex uint32
  43. RemoteIndex uint32
  44. PeerAddr netip.Addr
  45. }
  46. type HostMap struct {
  47. sync.RWMutex //Because we concurrently read and write to our maps
  48. Indexes map[uint32]*HostInfo
  49. Relays map[uint32]*HostInfo // Maps a Relay IDX to a Relay HostInfo object
  50. RemoteIndexes map[uint32]*HostInfo
  51. Hosts map[netip.Addr]*HostInfo
  52. preferredRanges atomic.Pointer[[]netip.Prefix]
  53. l *logrus.Logger
  54. }
  55. // For synchronization, treat the pointed-to Relay struct as immutable. To edit the Relay
  56. // struct, make a copy of an existing value, edit the fileds in the copy, and
  57. // then store a pointer to the new copy in both realyForBy* maps.
  58. type RelayState struct {
  59. sync.RWMutex
  60. relays map[netip.Addr]struct{} // Set of vpnAddr's of Hosts to use as relays to access this peer
  61. // For data race avoidance, the contents of a *Relay are treated immutably. To update a *Relay, copy the existing data,
  62. // modify what needs to be updated, and store the new modified copy in the relayForByIp and relayForByIdx maps (with
  63. // the RelayState Lock held)
  64. relayForByAddr map[netip.Addr]*Relay // Maps vpnAddr of peers for which this HostInfo is a relay to some Relay info
  65. relayForByIdx map[uint32]*Relay // Maps a local index to some Relay info
  66. }
  67. func (rs *RelayState) DeleteRelay(ip netip.Addr) {
  68. rs.Lock()
  69. defer rs.Unlock()
  70. delete(rs.relays, ip)
  71. }
  72. func (rs *RelayState) UpdateRelayForByIpState(vpnIp netip.Addr, state int) {
  73. rs.Lock()
  74. defer rs.Unlock()
  75. if r, ok := rs.relayForByAddr[vpnIp]; ok {
  76. newRelay := *r
  77. newRelay.State = state
  78. rs.relayForByAddr[newRelay.PeerAddr] = &newRelay
  79. rs.relayForByIdx[newRelay.LocalIndex] = &newRelay
  80. }
  81. }
  82. func (rs *RelayState) UpdateRelayForByIdxState(idx uint32, state int) {
  83. rs.Lock()
  84. defer rs.Unlock()
  85. if r, ok := rs.relayForByIdx[idx]; ok {
  86. newRelay := *r
  87. newRelay.State = state
  88. rs.relayForByAddr[newRelay.PeerAddr] = &newRelay
  89. rs.relayForByIdx[newRelay.LocalIndex] = &newRelay
  90. }
  91. }
  92. func (rs *RelayState) CopyAllRelayFor() []*Relay {
  93. rs.RLock()
  94. defer rs.RUnlock()
  95. ret := make([]*Relay, 0, len(rs.relayForByIdx))
  96. for _, r := range rs.relayForByIdx {
  97. ret = append(ret, r)
  98. }
  99. return ret
  100. }
  101. func (rs *RelayState) GetRelayForByAddr(addr netip.Addr) (*Relay, bool) {
  102. rs.RLock()
  103. defer rs.RUnlock()
  104. r, ok := rs.relayForByAddr[addr]
  105. return r, ok
  106. }
  107. func (rs *RelayState) InsertRelayTo(ip netip.Addr) {
  108. rs.Lock()
  109. defer rs.Unlock()
  110. rs.relays[ip] = struct{}{}
  111. }
  112. func (rs *RelayState) CopyRelayIps() []netip.Addr {
  113. rs.RLock()
  114. defer rs.RUnlock()
  115. ret := make([]netip.Addr, 0, len(rs.relays))
  116. for ip := range rs.relays {
  117. ret = append(ret, ip)
  118. }
  119. return ret
  120. }
  121. func (rs *RelayState) CopyRelayForIps() []netip.Addr {
  122. rs.RLock()
  123. defer rs.RUnlock()
  124. currentRelays := make([]netip.Addr, 0, len(rs.relayForByAddr))
  125. for relayIp := range rs.relayForByAddr {
  126. currentRelays = append(currentRelays, relayIp)
  127. }
  128. return currentRelays
  129. }
  130. func (rs *RelayState) CopyRelayForIdxs() []uint32 {
  131. rs.RLock()
  132. defer rs.RUnlock()
  133. ret := make([]uint32, 0, len(rs.relayForByIdx))
  134. for i := range rs.relayForByIdx {
  135. ret = append(ret, i)
  136. }
  137. return ret
  138. }
  139. func (rs *RelayState) CompleteRelayByIP(vpnIp netip.Addr, remoteIdx uint32) bool {
  140. rs.Lock()
  141. defer rs.Unlock()
  142. r, ok := rs.relayForByAddr[vpnIp]
  143. if !ok {
  144. return false
  145. }
  146. newRelay := *r
  147. newRelay.State = Established
  148. newRelay.RemoteIndex = remoteIdx
  149. rs.relayForByIdx[r.LocalIndex] = &newRelay
  150. rs.relayForByAddr[r.PeerAddr] = &newRelay
  151. return true
  152. }
  153. func (rs *RelayState) CompleteRelayByIdx(localIdx uint32, remoteIdx uint32) (*Relay, bool) {
  154. rs.Lock()
  155. defer rs.Unlock()
  156. r, ok := rs.relayForByIdx[localIdx]
  157. if !ok {
  158. return nil, false
  159. }
  160. newRelay := *r
  161. newRelay.State = Established
  162. newRelay.RemoteIndex = remoteIdx
  163. rs.relayForByIdx[r.LocalIndex] = &newRelay
  164. rs.relayForByAddr[r.PeerAddr] = &newRelay
  165. return &newRelay, true
  166. }
  167. func (rs *RelayState) QueryRelayForByIp(vpnIp netip.Addr) (*Relay, bool) {
  168. rs.RLock()
  169. defer rs.RUnlock()
  170. r, ok := rs.relayForByAddr[vpnIp]
  171. return r, ok
  172. }
  173. func (rs *RelayState) QueryRelayForByIdx(idx uint32) (*Relay, bool) {
  174. rs.RLock()
  175. defer rs.RUnlock()
  176. r, ok := rs.relayForByIdx[idx]
  177. return r, ok
  178. }
  179. func (rs *RelayState) InsertRelay(ip netip.Addr, idx uint32, r *Relay) {
  180. rs.Lock()
  181. defer rs.Unlock()
  182. rs.relayForByAddr[ip] = r
  183. rs.relayForByIdx[idx] = r
  184. }
  185. type HostInfo struct {
  186. remote netip.AddrPort
  187. remotes *RemoteList
  188. promoteCounter atomic.Uint32
  189. ConnectionState *ConnectionState
  190. remoteIndexId uint32
  191. localIndexId uint32
  192. vpnAddrs []netip.Addr
  193. recvError atomic.Uint32
  194. // networks are both all vpn and unsafe networks assigned to this host
  195. networks *bart.Table[struct{}]
  196. relayState RelayState
  197. // HandshakePacket records the packets used to create this hostinfo
  198. // We need these to avoid replayed handshake packets creating new hostinfos which causes churn
  199. HandshakePacket map[uint8][]byte
  200. // nextLHQuery is the earliest we can ask the lighthouse for new information.
  201. // This is used to limit lighthouse re-queries in chatty clients
  202. nextLHQuery atomic.Int64
  203. // lastRebindCount is the other side of Interface.rebindCount, if these values don't match then we need to ask LH
  204. // for a punch from the remote end of this tunnel. The goal being to prime their conntrack for our traffic just like
  205. // with a handshake
  206. lastRebindCount int8
  207. // lastHandshakeTime records the time the remote side told us about at the stage when the handshake was completed locally
  208. // Stage 1 packet will contain it if I am a responder, stage 2 packet if I am an initiator
  209. // This is used to avoid an attack where a handshake packet is replayed after some time
  210. lastHandshakeTime uint64
  211. lastRoam time.Time
  212. lastRoamRemote netip.AddrPort
  213. // Used to track other hostinfos for this vpn ip since only 1 can be primary
  214. // Synchronised via hostmap lock and not the hostinfo lock.
  215. next, prev *HostInfo
  216. }
  217. type ViaSender struct {
  218. relayHI *HostInfo // relayHI is the host info object of the relay
  219. remoteIdx uint32 // remoteIdx is the index included in the header of the received packet
  220. relay *Relay // relay contains the rest of the relay information, including the PeerIP of the host trying to communicate with us.
  221. }
  222. type cachedPacket struct {
  223. messageType header.MessageType
  224. messageSubType header.MessageSubType
  225. callback packetCallback
  226. packet []byte
  227. }
  228. type packetCallback func(t header.MessageType, st header.MessageSubType, h *HostInfo, p, nb, out []byte)
  229. type cachedPacketMetrics struct {
  230. sent metrics.Counter
  231. dropped metrics.Counter
  232. }
  233. func NewHostMapFromConfig(l *logrus.Logger, c *config.C) *HostMap {
  234. hm := newHostMap(l)
  235. hm.reload(c, true)
  236. c.RegisterReloadCallback(func(c *config.C) {
  237. hm.reload(c, false)
  238. })
  239. l.WithField("preferredRanges", hm.GetPreferredRanges()).
  240. Info("Main HostMap created")
  241. return hm
  242. }
  243. func newHostMap(l *logrus.Logger) *HostMap {
  244. return &HostMap{
  245. Indexes: map[uint32]*HostInfo{},
  246. Relays: map[uint32]*HostInfo{},
  247. RemoteIndexes: map[uint32]*HostInfo{},
  248. Hosts: map[netip.Addr]*HostInfo{},
  249. l: l,
  250. }
  251. }
  252. func (hm *HostMap) reload(c *config.C, initial bool) {
  253. if initial || c.HasChanged("preferred_ranges") {
  254. var preferredRanges []netip.Prefix
  255. rawPreferredRanges := c.GetStringSlice("preferred_ranges", []string{})
  256. for _, rawPreferredRange := range rawPreferredRanges {
  257. preferredRange, err := netip.ParsePrefix(rawPreferredRange)
  258. if err != nil {
  259. hm.l.WithError(err).WithField("range", rawPreferredRanges).Warn("Failed to parse preferred ranges, ignoring")
  260. continue
  261. }
  262. preferredRanges = append(preferredRanges, preferredRange)
  263. }
  264. oldRanges := hm.preferredRanges.Swap(&preferredRanges)
  265. if !initial {
  266. hm.l.WithField("oldPreferredRanges", *oldRanges).WithField("newPreferredRanges", preferredRanges).Info("preferred_ranges changed")
  267. }
  268. }
  269. }
  270. // EmitStats reports host, index, and relay counts to the stats collection system
  271. func (hm *HostMap) EmitStats() {
  272. hm.RLock()
  273. hostLen := len(hm.Hosts)
  274. indexLen := len(hm.Indexes)
  275. remoteIndexLen := len(hm.RemoteIndexes)
  276. relaysLen := len(hm.Relays)
  277. hm.RUnlock()
  278. metrics.GetOrRegisterGauge("hostmap.main.hosts", nil).Update(int64(hostLen))
  279. metrics.GetOrRegisterGauge("hostmap.main.indexes", nil).Update(int64(indexLen))
  280. metrics.GetOrRegisterGauge("hostmap.main.remoteIndexes", nil).Update(int64(remoteIndexLen))
  281. metrics.GetOrRegisterGauge("hostmap.main.relayIndexes", nil).Update(int64(relaysLen))
  282. }
  283. // DeleteHostInfo will fully unlink the hostinfo and return true if it was the final hostinfo for this vpn ip
  284. func (hm *HostMap) DeleteHostInfo(hostinfo *HostInfo) bool {
  285. // Delete the host itself, ensuring it's not modified anymore
  286. hm.Lock()
  287. // If we have a previous or next hostinfo then we are not the last one for this vpn ip
  288. final := (hostinfo.next == nil && hostinfo.prev == nil)
  289. hm.unlockedDeleteHostInfo(hostinfo)
  290. hm.Unlock()
  291. return final
  292. }
  293. func (hm *HostMap) MakePrimary(hostinfo *HostInfo) {
  294. hm.Lock()
  295. defer hm.Unlock()
  296. hm.unlockedMakePrimary(hostinfo)
  297. }
  298. func (hm *HostMap) unlockedMakePrimary(hostinfo *HostInfo) {
  299. //TODO: we may need to promote follow on hostinfos from these vpnAddrs as well since their oldHostinfo might not be the same as this one
  300. // this really looks like an ideal spot for memory leaks
  301. oldHostinfo := hm.Hosts[hostinfo.vpnAddrs[0]]
  302. if oldHostinfo == hostinfo {
  303. return
  304. }
  305. if hostinfo.prev != nil {
  306. hostinfo.prev.next = hostinfo.next
  307. }
  308. if hostinfo.next != nil {
  309. hostinfo.next.prev = hostinfo.prev
  310. }
  311. hm.Hosts[hostinfo.vpnAddrs[0]] = hostinfo
  312. if oldHostinfo == nil {
  313. return
  314. }
  315. hostinfo.next = oldHostinfo
  316. oldHostinfo.prev = hostinfo
  317. hostinfo.prev = nil
  318. }
  319. func (hm *HostMap) unlockedDeleteHostInfo(hostinfo *HostInfo) {
  320. for _, addr := range hostinfo.vpnAddrs {
  321. h := hm.Hosts[addr]
  322. for h != nil {
  323. if h == hostinfo {
  324. hm.unlockedInnerDeleteHostInfo(h, addr)
  325. }
  326. h = h.next
  327. }
  328. }
  329. }
  330. func (hm *HostMap) unlockedInnerDeleteHostInfo(hostinfo *HostInfo, addr netip.Addr) {
  331. primary, ok := hm.Hosts[addr]
  332. isLastHostinfo := hostinfo.next == nil && hostinfo.prev == nil
  333. if ok && primary == hostinfo {
  334. // The vpn addr pointer points to the same hostinfo as the local index id, we can remove it
  335. delete(hm.Hosts, addr)
  336. if len(hm.Hosts) == 0 {
  337. hm.Hosts = map[netip.Addr]*HostInfo{}
  338. }
  339. if hostinfo.next != nil {
  340. // We had more than 1 hostinfo at this vpn addr, promote the next in the list to primary
  341. hm.Hosts[addr] = hostinfo.next
  342. // It is primary, there is no previous hostinfo now
  343. hostinfo.next.prev = nil
  344. }
  345. } else {
  346. // Relink if we were in the middle of multiple hostinfos for this vpn addr
  347. if hostinfo.prev != nil {
  348. hostinfo.prev.next = hostinfo.next
  349. }
  350. if hostinfo.next != nil {
  351. hostinfo.next.prev = hostinfo.prev
  352. }
  353. }
  354. hostinfo.next = nil
  355. hostinfo.prev = nil
  356. // The remote index uses index ids outside our control so lets make sure we are only removing
  357. // the remote index pointer here if it points to the hostinfo we are deleting
  358. hostinfo2, ok := hm.RemoteIndexes[hostinfo.remoteIndexId]
  359. if ok && hostinfo2 == hostinfo {
  360. delete(hm.RemoteIndexes, hostinfo.remoteIndexId)
  361. if len(hm.RemoteIndexes) == 0 {
  362. hm.RemoteIndexes = map[uint32]*HostInfo{}
  363. }
  364. }
  365. delete(hm.Indexes, hostinfo.localIndexId)
  366. if len(hm.Indexes) == 0 {
  367. hm.Indexes = map[uint32]*HostInfo{}
  368. }
  369. if hm.l.Level >= logrus.DebugLevel {
  370. hm.l.WithField("hostMap", m{"mapTotalSize": len(hm.Hosts),
  371. "vpnAddrs": hostinfo.vpnAddrs, "indexNumber": hostinfo.localIndexId, "remoteIndexNumber": hostinfo.remoteIndexId}).
  372. Debug("Hostmap hostInfo deleted")
  373. }
  374. if isLastHostinfo {
  375. // I have lost connectivity to my peers. My relay tunnel is likely broken. Mark the next
  376. // hops as 'Requested' so that new relay tunnels are created in the future.
  377. hm.unlockedDisestablishVpnAddrRelayFor(hostinfo)
  378. }
  379. // Clean up any local relay indexes for which I am acting as a relay hop
  380. for _, localRelayIdx := range hostinfo.relayState.CopyRelayForIdxs() {
  381. delete(hm.Relays, localRelayIdx)
  382. }
  383. }
  384. func (hm *HostMap) QueryIndex(index uint32) *HostInfo {
  385. hm.RLock()
  386. if h, ok := hm.Indexes[index]; ok {
  387. hm.RUnlock()
  388. return h
  389. } else {
  390. hm.RUnlock()
  391. return nil
  392. }
  393. }
  394. func (hm *HostMap) QueryRelayIndex(index uint32) *HostInfo {
  395. hm.RLock()
  396. if h, ok := hm.Relays[index]; ok {
  397. hm.RUnlock()
  398. return h
  399. } else {
  400. hm.RUnlock()
  401. return nil
  402. }
  403. }
  404. func (hm *HostMap) QueryReverseIndex(index uint32) *HostInfo {
  405. hm.RLock()
  406. if h, ok := hm.RemoteIndexes[index]; ok {
  407. hm.RUnlock()
  408. return h
  409. } else {
  410. hm.RUnlock()
  411. return nil
  412. }
  413. }
  414. func (hm *HostMap) QueryVpnAddr(vpnIp netip.Addr) *HostInfo {
  415. return hm.queryVpnAddr(vpnIp, nil)
  416. }
  417. func (hm *HostMap) QueryVpnAddrsRelayFor(targetIps []netip.Addr, relayHostIp netip.Addr) (*HostInfo, *Relay, error) {
  418. hm.RLock()
  419. defer hm.RUnlock()
  420. h, ok := hm.Hosts[relayHostIp]
  421. if !ok {
  422. return nil, nil, errors.New("unable to find host")
  423. }
  424. for h != nil {
  425. for _, targetIp := range targetIps {
  426. r, ok := h.relayState.QueryRelayForByIp(targetIp)
  427. if ok && r.State == Established {
  428. return h, r, nil
  429. }
  430. }
  431. h = h.next
  432. }
  433. return nil, nil, errors.New("unable to find host with relay")
  434. }
  435. func (hm *HostMap) unlockedDisestablishVpnAddrRelayFor(hi *HostInfo) {
  436. for _, relayHostIp := range hi.relayState.CopyRelayIps() {
  437. if h, ok := hm.Hosts[relayHostIp]; ok {
  438. for h != nil {
  439. h.relayState.UpdateRelayForByIpState(hi.vpnAddrs[0], Disestablished)
  440. h = h.next
  441. }
  442. }
  443. }
  444. for _, rs := range hi.relayState.CopyAllRelayFor() {
  445. if rs.Type == ForwardingType {
  446. if h, ok := hm.Hosts[rs.PeerAddr]; ok {
  447. for h != nil {
  448. h.relayState.UpdateRelayForByIpState(hi.vpnAddrs[0], Disestablished)
  449. h = h.next
  450. }
  451. }
  452. }
  453. }
  454. }
  455. func (hm *HostMap) queryVpnAddr(vpnIp netip.Addr, promoteIfce *Interface) *HostInfo {
  456. hm.RLock()
  457. if h, ok := hm.Hosts[vpnIp]; ok {
  458. hm.RUnlock()
  459. // Do not attempt promotion if you are a lighthouse
  460. if promoteIfce != nil && !promoteIfce.lightHouse.amLighthouse {
  461. h.TryPromoteBest(hm.GetPreferredRanges(), promoteIfce)
  462. }
  463. return h
  464. }
  465. hm.RUnlock()
  466. return nil
  467. }
  468. // unlockedAddHostInfo assumes you have a write-lock and will add a hostinfo object to the hostmap Indexes and RemoteIndexes maps.
  469. // If an entry exists for the Hosts table (vpnIp -> hostinfo) then the provided hostinfo will be made primary
  470. func (hm *HostMap) unlockedAddHostInfo(hostinfo *HostInfo, f *Interface) {
  471. if f.serveDns {
  472. remoteCert := hostinfo.ConnectionState.peerCert
  473. dnsR.Add(remoteCert.Certificate.Name()+".", hostinfo.vpnAddrs)
  474. }
  475. for _, addr := range hostinfo.vpnAddrs {
  476. hm.unlockedInnerAddHostInfo(addr, hostinfo, f)
  477. }
  478. hm.Indexes[hostinfo.localIndexId] = hostinfo
  479. hm.RemoteIndexes[hostinfo.remoteIndexId] = hostinfo
  480. if hm.l.Level >= logrus.DebugLevel {
  481. hm.l.WithField("hostMap", m{"vpnAddrs": hostinfo.vpnAddrs, "mapTotalSize": len(hm.Hosts),
  482. "hostinfo": m{"existing": true, "localIndexId": hostinfo.localIndexId, "vpnAddrs": hostinfo.vpnAddrs}}).
  483. Debug("Hostmap vpnIp added")
  484. }
  485. }
  486. func (hm *HostMap) unlockedInnerAddHostInfo(vpnAddr netip.Addr, hostinfo *HostInfo, f *Interface) {
  487. existing := hm.Hosts[vpnAddr]
  488. hm.Hosts[vpnAddr] = hostinfo
  489. if existing != nil && existing != hostinfo {
  490. hostinfo.next = existing
  491. existing.prev = hostinfo
  492. }
  493. i := 1
  494. check := hostinfo
  495. for check != nil {
  496. if i > MaxHostInfosPerVpnIp {
  497. hm.unlockedDeleteHostInfo(check)
  498. }
  499. check = check.next
  500. i++
  501. }
  502. }
  503. func (hm *HostMap) GetPreferredRanges() []netip.Prefix {
  504. //NOTE: if preferredRanges is ever not stored before a load this will fail to dereference a nil pointer
  505. return *hm.preferredRanges.Load()
  506. }
  507. func (hm *HostMap) ForEachVpnAddr(f controlEach) {
  508. hm.RLock()
  509. defer hm.RUnlock()
  510. for _, v := range hm.Hosts {
  511. f(v)
  512. }
  513. }
  514. func (hm *HostMap) ForEachIndex(f controlEach) {
  515. hm.RLock()
  516. defer hm.RUnlock()
  517. for _, v := range hm.Indexes {
  518. f(v)
  519. }
  520. }
  521. // TryPromoteBest handles re-querying lighthouses and probing for better paths
  522. // NOTE: It is an error to call this if you are a lighthouse since they should not roam clients!
  523. func (i *HostInfo) TryPromoteBest(preferredRanges []netip.Prefix, ifce *Interface) {
  524. c := i.promoteCounter.Add(1)
  525. if c%ifce.tryPromoteEvery.Load() == 0 {
  526. remote := i.remote
  527. // return early if we are already on a preferred remote
  528. if remote.IsValid() {
  529. rIP := remote.Addr()
  530. for _, l := range preferredRanges {
  531. if l.Contains(rIP) {
  532. return
  533. }
  534. }
  535. }
  536. i.remotes.ForEach(preferredRanges, func(addr netip.AddrPort, preferred bool) {
  537. if remote.IsValid() && (!addr.IsValid() || !preferred) {
  538. return
  539. }
  540. // Try to send a test packet to that host, this should
  541. // cause it to detect a roaming event and switch remotes
  542. ifce.sendTo(header.Test, header.TestRequest, i.ConnectionState, i, addr, []byte(""), make([]byte, 12, 12), make([]byte, mtu))
  543. })
  544. }
  545. // Re query our lighthouses for new remotes occasionally
  546. if c%ifce.reQueryEvery.Load() == 0 && ifce.lightHouse != nil {
  547. now := time.Now().UnixNano()
  548. if now < i.nextLHQuery.Load() {
  549. return
  550. }
  551. i.nextLHQuery.Store(now + ifce.reQueryWait.Load())
  552. ifce.lightHouse.QueryServer(i.vpnAddrs[0])
  553. }
  554. }
  555. func (i *HostInfo) GetCert() *cert.CachedCertificate {
  556. if i.ConnectionState != nil {
  557. return i.ConnectionState.peerCert
  558. }
  559. return nil
  560. }
  561. func (i *HostInfo) SetRemote(remote netip.AddrPort) {
  562. // We copy here because we likely got this remote from a source that reuses the object
  563. if i.remote != remote {
  564. i.remote = remote
  565. i.remotes.LearnRemote(i.vpnAddrs[0], remote)
  566. }
  567. }
  568. // SetRemoteIfPreferred returns true if the remote was changed. The lastRoam
  569. // time on the HostInfo will also be updated.
  570. func (i *HostInfo) SetRemoteIfPreferred(hm *HostMap, newRemote netip.AddrPort) bool {
  571. if !newRemote.IsValid() {
  572. // relays have nil udp Addrs
  573. return false
  574. }
  575. currentRemote := i.remote
  576. if !currentRemote.IsValid() {
  577. i.SetRemote(newRemote)
  578. return true
  579. }
  580. // NOTE: We do this loop here instead of calling `isPreferred` in
  581. // remote_list.go so that we only have to loop over preferredRanges once.
  582. newIsPreferred := false
  583. for _, l := range hm.GetPreferredRanges() {
  584. // return early if we are already on a preferred remote
  585. if l.Contains(currentRemote.Addr()) {
  586. return false
  587. }
  588. if l.Contains(newRemote.Addr()) {
  589. newIsPreferred = true
  590. }
  591. }
  592. if newIsPreferred {
  593. // Consider this a roaming event
  594. i.lastRoam = time.Now()
  595. i.lastRoamRemote = currentRemote
  596. i.SetRemote(newRemote)
  597. return true
  598. }
  599. return false
  600. }
  601. func (i *HostInfo) RecvErrorExceeded() bool {
  602. if i.recvError.Add(1) >= maxRecvError {
  603. return true
  604. }
  605. return true
  606. }
  607. func (i *HostInfo) buildNetworks(c cert.Certificate) {
  608. if len(c.Networks()) == 1 && len(c.UnsafeNetworks()) == 0 {
  609. // Simple case, no CIDRTree needed
  610. return
  611. }
  612. i.networks = new(bart.Table[struct{}])
  613. for _, network := range c.Networks() {
  614. i.networks.Insert(network, struct{}{})
  615. }
  616. for _, network := range c.UnsafeNetworks() {
  617. i.networks.Insert(network, struct{}{})
  618. }
  619. }
  620. func (i *HostInfo) logger(l *logrus.Logger) *logrus.Entry {
  621. if i == nil {
  622. return logrus.NewEntry(l)
  623. }
  624. li := l.WithField("vpnAddrs", i.vpnAddrs).
  625. WithField("localIndex", i.localIndexId).
  626. WithField("remoteIndex", i.remoteIndexId)
  627. if connState := i.ConnectionState; connState != nil {
  628. if peerCert := connState.peerCert; peerCert != nil {
  629. li = li.WithField("certName", peerCert.Certificate.Name())
  630. }
  631. }
  632. return li
  633. }
  634. // Utility functions
  635. func localAddrs(l *logrus.Logger, allowList *LocalAllowList) []netip.Addr {
  636. //FIXME: This function is pretty garbage
  637. var finalAddrs []netip.Addr
  638. ifaces, _ := net.Interfaces()
  639. for _, i := range ifaces {
  640. allow := allowList.AllowName(i.Name)
  641. if l.Level >= logrus.TraceLevel {
  642. l.WithField("interfaceName", i.Name).WithField("allow", allow).Trace("localAllowList.AllowName")
  643. }
  644. if !allow {
  645. continue
  646. }
  647. addrs, _ := i.Addrs()
  648. for _, rawAddr := range addrs {
  649. var addr netip.Addr
  650. switch v := rawAddr.(type) {
  651. case *net.IPNet:
  652. //continue
  653. addr, _ = netip.AddrFromSlice(v.IP)
  654. case *net.IPAddr:
  655. addr, _ = netip.AddrFromSlice(v.IP)
  656. }
  657. if !addr.IsValid() {
  658. if l.Level >= logrus.DebugLevel {
  659. l.WithField("localAddr", rawAddr).Debug("addr was invalid")
  660. }
  661. continue
  662. }
  663. addr = addr.Unmap()
  664. if addr.IsLoopback() == false && addr.IsLinkLocalUnicast() == false {
  665. isAllowed := allowList.Allow(addr)
  666. if l.Level >= logrus.TraceLevel {
  667. l.WithField("localAddr", addr).WithField("allowed", isAllowed).Trace("localAllowList.Allow")
  668. }
  669. if !isAllowed {
  670. continue
  671. }
  672. finalAddrs = append(finalAddrs, addr)
  673. }
  674. }
  675. }
  676. return finalAddrs
  677. }