peers.go 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872
  1. package logic
  2. import (
  3. "errors"
  4. "fmt"
  5. "net"
  6. "strconv"
  7. "strings"
  8. "time"
  9. "github.com/c-robinson/iplib"
  10. "github.com/gravitl/netclient/nmproxy/manager"
  11. proxy_models "github.com/gravitl/netclient/nmproxy/models"
  12. "github.com/gravitl/netmaker/database"
  13. "github.com/gravitl/netmaker/logger"
  14. "github.com/gravitl/netmaker/logic/acls/nodeacls"
  15. "github.com/gravitl/netmaker/models"
  16. "github.com/gravitl/netmaker/netclient/ncutils"
  17. "github.com/gravitl/netmaker/servercfg"
  18. "golang.org/x/exp/slices"
  19. "golang.zx2c4.com/wireguard/wgctrl/wgtypes"
  20. )
  21. // GetPeersforProxy calculates the peers for a proxy
  22. // TODO ==========================
  23. // TODO ==========================
  24. // TODO ==========================
  25. // TODO ==========================
  26. // TODO ==========================
  27. // revisit this logic with new host/node models.
  28. func GetPeersForProxy(node *models.Node, onlyPeers bool) (manager.ProxyManagerPayload, error) {
  29. proxyPayload := manager.ProxyManagerPayload{}
  30. var peers []wgtypes.PeerConfig
  31. peerConfMap := make(map[string]proxy_models.PeerConf)
  32. var err error
  33. currentPeers, err := GetNetworkNodes(node.Network)
  34. if err != nil {
  35. return proxyPayload, err
  36. }
  37. if !onlyPeers {
  38. if node.IsRelayed {
  39. relayNode := FindRelay(node)
  40. relayHost, err := GetHost(relayNode.ID.String())
  41. if err != nil {
  42. return proxyPayload, err
  43. }
  44. if relayNode != nil {
  45. host, err := GetHost(relayNode.HostID.String())
  46. if err != nil {
  47. logger.Log(0, "error retrieving host for relay node", relayNode.HostID.String(), err.Error())
  48. }
  49. relayEndpoint, err := net.ResolveUDPAddr("udp", fmt.Sprintf("%s:%d", relayHost.EndpointIP, host.LocalListenPort))
  50. if err != nil {
  51. logger.Log(1, "failed to resolve relay node endpoint: ", err.Error())
  52. }
  53. proxyPayload.IsRelayed = true
  54. proxyPayload.RelayedTo = relayEndpoint
  55. } else {
  56. logger.Log(0, "couldn't find relay node for: ", node.ID.String())
  57. }
  58. }
  59. if node.IsRelay {
  60. host, err := GetHost(node.HostID.String())
  61. if err != nil {
  62. logger.Log(0, "error retrieving host for relay node", node.ID.String(), err.Error())
  63. }
  64. relayedNodes, err := GetRelayedNodes(node)
  65. if err != nil {
  66. logger.Log(1, "failed to relayed nodes: ", node.ID.String(), err.Error())
  67. proxyPayload.IsRelay = false
  68. } else {
  69. relayPeersMap := make(map[string]proxy_models.RelayedConf)
  70. for _, relayedNode := range relayedNodes {
  71. payload, err := GetPeersForProxy(&relayedNode, true)
  72. if err == nil {
  73. relayedHost, err := GetHost(relayedNode.HostID.String())
  74. if err != nil {
  75. logger.Log(0, "error retrieving host for relayNode", relayedNode.ID.String(), err.Error())
  76. }
  77. relayedEndpoint, udpErr := net.ResolveUDPAddr("udp", fmt.Sprintf("%s:%d", relayedHost.EndpointIP, host.LocalListenPort))
  78. if udpErr == nil {
  79. relayPeersMap[host.PublicKey.String()] = proxy_models.RelayedConf{
  80. RelayedPeerEndpoint: relayedEndpoint,
  81. RelayedPeerPubKey: relayedHost.PublicKey.String(),
  82. Peers: payload.Peers,
  83. }
  84. }
  85. }
  86. }
  87. proxyPayload.IsRelay = true
  88. proxyPayload.RelayedPeerConf = relayPeersMap
  89. }
  90. }
  91. }
  92. for _, peer := range currentPeers {
  93. if peer.ID == node.ID {
  94. //skip yourself
  95. continue
  96. }
  97. host, err := GetHost(peer.HostID.String())
  98. if err != nil {
  99. continue
  100. }
  101. proxyStatus := host.ProxyEnabled
  102. listenPort := host.LocalListenPort
  103. if proxyStatus {
  104. listenPort = host.ProxyListenPort
  105. if listenPort == 0 {
  106. listenPort = proxy_models.NmProxyPort
  107. }
  108. } else if listenPort == 0 {
  109. listenPort = host.ListenPort
  110. }
  111. endpoint, err := net.ResolveUDPAddr("udp", fmt.Sprintf("%s:%d", host.EndpointIP, listenPort))
  112. if err != nil {
  113. logger.Log(1, "failed to resolve udp addr for node: ", peer.ID.String(), host.EndpointIP.String(), err.Error())
  114. continue
  115. }
  116. allowedips := GetAllowedIPs(node, &peer, nil, false)
  117. var keepalive time.Duration
  118. if node.PersistentKeepalive != 0 {
  119. // set_keepalive
  120. keepalive, _ = time.ParseDuration(strconv.FormatInt(int64(node.PersistentKeepalive), 10) + "s")
  121. }
  122. peers = append(peers, wgtypes.PeerConfig{
  123. PublicKey: host.PublicKey,
  124. Endpoint: endpoint,
  125. AllowedIPs: allowedips,
  126. PersistentKeepaliveInterval: &keepalive,
  127. ReplaceAllowedIPs: true,
  128. })
  129. peerConfMap[host.PublicKey.String()] = proxy_models.PeerConf{
  130. Address: net.ParseIP(peer.PrimaryAddress()),
  131. Proxy: proxyStatus,
  132. PublicListenPort: int32(listenPort),
  133. }
  134. if !onlyPeers && peer.IsRelayed {
  135. relayNode := FindRelay(&peer)
  136. if relayNode != nil {
  137. relayHost, err := GetHost(relayNode.HostID.String())
  138. if err != nil {
  139. logger.Log(0, "error retrieving host for relayNode", relayNode.ID.String(), err.Error())
  140. continue
  141. }
  142. relayTo, err := net.ResolveUDPAddr("udp", fmt.Sprintf("%s:%d", relayHost.EndpointIP, relayHost.LocalListenPort))
  143. if err == nil {
  144. peerConfMap[host.PublicKey.String()] = proxy_models.PeerConf{
  145. IsRelayed: true,
  146. RelayedTo: relayTo,
  147. Address: net.ParseIP(peer.PrimaryAddress()),
  148. Proxy: proxyStatus,
  149. PublicListenPort: int32(listenPort),
  150. }
  151. }
  152. }
  153. }
  154. }
  155. if node.IsIngressGateway {
  156. var extPeers []wgtypes.PeerConfig
  157. extPeers, peerConfMap, err = getExtPeersForProxy(node, peerConfMap)
  158. if err == nil {
  159. peers = append(peers, extPeers...)
  160. } else if !database.IsEmptyRecord(err) {
  161. logger.Log(1, "error retrieving external clients:", err.Error())
  162. }
  163. }
  164. proxyPayload.IsIngress = node.IsIngressGateway
  165. addr := node.Address
  166. if addr.String() == "" {
  167. addr = node.Address6
  168. }
  169. proxyPayload.WgAddr = addr.String()
  170. proxyPayload.Peers = peers
  171. proxyPayload.PeerMap = peerConfMap
  172. proxyPayload.Network = node.Network
  173. //proxyPayload.InterfaceName = node.Interface
  174. //hardcode or read from host ??
  175. proxyPayload.InterfaceName = models.WIREGUARD_INTERFACE
  176. return proxyPayload, nil
  177. }
  178. // GetPeerUpdate - gets a wireguard peer config for each peer of a node
  179. func GetPeerUpdate(node *models.Node) (models.PeerUpdate, error) {
  180. var peerUpdate models.PeerUpdate
  181. var peers []wgtypes.PeerConfig
  182. var serverNodeAddresses = []models.ServerAddr{}
  183. var peerMap = make(models.PeerMap)
  184. var metrics *models.Metrics
  185. if servercfg.Is_EE {
  186. metrics, _ = GetMetrics(node.ID.String())
  187. }
  188. if metrics == nil {
  189. metrics = &models.Metrics{}
  190. }
  191. if metrics.FailoverPeers == nil {
  192. metrics.FailoverPeers = make(map[string]string)
  193. }
  194. // udppeers = the peers parsed from the local interface
  195. // gives us correct port to reach
  196. udppeers, errN := database.GetPeers(node.Network)
  197. if errN != nil {
  198. logger.Log(2, errN.Error())
  199. }
  200. currentPeers, err := GetNetworkNodes(node.Network)
  201. if err != nil {
  202. return models.PeerUpdate{}, err
  203. }
  204. host, err := GetHost(node.ID.String())
  205. if err != nil {
  206. return peerUpdate, err
  207. }
  208. if node.IsRelayed && !host.ProxyEnabled {
  209. return GetPeerUpdateForRelayedNode(node, udppeers)
  210. }
  211. // #1 Set Keepalive values: set_keepalive
  212. // #2 Set local address: set_local - could be a LOT BETTER and fix some bugs with additional logic
  213. // #3 Set allowedips: set_allowedips
  214. for _, peer := range currentPeers {
  215. peerHost, err := GetHost(peer.ID.String())
  216. if err != nil {
  217. logger.Log(0, "error retrieving host for peer", node.ID.String(), err.Error())
  218. return models.PeerUpdate{}, err
  219. }
  220. if peer.ID == node.ID {
  221. //skip yourself
  222. continue
  223. }
  224. if node.Connected {
  225. //skip unconnected nodes
  226. continue
  227. }
  228. // if the node is not a server, set the endpoint
  229. var setEndpoint = true
  230. if peer.IsRelayed {
  231. if !peerHost.ProxyEnabled && !(node.IsRelay && ncutils.StringSliceContains(node.RelayAddrs, peer.PrimaryAddress())) {
  232. //skip -- will be added to relay
  233. continue
  234. } else if node.IsRelay && ncutils.StringSliceContains(node.RelayAddrs, peer.PrimaryAddress()) {
  235. // dont set peer endpoint if it's relayed by node
  236. setEndpoint = false
  237. }
  238. }
  239. if !nodeacls.AreNodesAllowed(nodeacls.NetworkID(node.Network), nodeacls.NodeID(node.ID.String()), nodeacls.NodeID(peer.ID.String())) {
  240. //skip if not permitted by acl
  241. continue
  242. }
  243. if len(metrics.FailoverPeers[peer.ID.String()]) > 0 && IsFailoverPresent(node.Network) {
  244. logger.Log(2, "peer", peer.ID.String(), peer.PrimaryAddress(), "was found to be in failover peers list for node", node.ID.String(), node.PrimaryAddress())
  245. continue
  246. }
  247. if err != nil {
  248. return models.PeerUpdate{}, err
  249. }
  250. host, err := GetHost(node.HostID.String())
  251. if err != nil {
  252. logger.Log(0, "error retrieving host for node", node.ID.String(), err.Error())
  253. return models.PeerUpdate{}, err
  254. }
  255. if host.EndpointIP.String() == peerHost.EndpointIP.String() {
  256. //peer is on same network
  257. // set_local
  258. if host.LocalAddress.String() != peerHost.LocalAddress.String() && peerHost.LocalAddress.IP != nil {
  259. peerHost.EndpointIP = peerHost.LocalAddress.IP
  260. if peerHost.LocalListenPort != 0 {
  261. peerHost.ListenPort = peerHost.LocalListenPort
  262. }
  263. } else {
  264. continue
  265. }
  266. }
  267. // set address if setEndpoint is true
  268. // otherwise, will get inserted as empty value
  269. var address *net.UDPAddr
  270. // Sets ListenPort to UDP Hole Punching Port assuming:
  271. // - UDP Hole Punching is enabled
  272. // - udppeers retrieval did not return an error
  273. // - the endpoint is valid
  274. if setEndpoint {
  275. var setUDPPort = false
  276. if CheckEndpoint(udppeers[peerHost.PublicKey.String()]) {
  277. endpointstring := udppeers[peerHost.PublicKey.String()]
  278. endpointarr := strings.Split(endpointstring, ":")
  279. if len(endpointarr) == 2 {
  280. port, err := strconv.Atoi(endpointarr[1])
  281. if err == nil {
  282. setUDPPort = true
  283. peerHost.ListenPort = port
  284. }
  285. }
  286. }
  287. // if udp hole punching is on, but udp hole punching did not set it, use the LocalListenPort instead
  288. // or, if port is for some reason zero use the LocalListenPort
  289. // but only do this if LocalListenPort is not zero
  290. if ((!setUDPPort) || peerHost.ListenPort == 0) && peerHost.LocalListenPort != 0 {
  291. peerHost.ListenPort = peerHost.LocalListenPort
  292. }
  293. endpoint := peerHost.EndpointIP.String() + ":" + strconv.FormatInt(int64(peerHost.ListenPort), 10)
  294. address, err = net.ResolveUDPAddr("udp", endpoint)
  295. if err != nil {
  296. return models.PeerUpdate{}, err
  297. }
  298. }
  299. fetchRelayedIps := true
  300. if host.ProxyEnabled {
  301. fetchRelayedIps = false
  302. }
  303. allowedips := GetAllowedIPs(node, &peer, metrics, fetchRelayedIps)
  304. var keepalive time.Duration
  305. if node.PersistentKeepalive != 0 {
  306. // set_keepalive
  307. keepalive, _ = time.ParseDuration(strconv.FormatInt(int64(node.PersistentKeepalive), 10) + "s")
  308. }
  309. var peerData = wgtypes.PeerConfig{
  310. PublicKey: peerHost.PublicKey,
  311. Endpoint: address,
  312. ReplaceAllowedIPs: true,
  313. AllowedIPs: allowedips,
  314. PersistentKeepaliveInterval: &keepalive,
  315. }
  316. peers = append(peers, peerData)
  317. peerMap[peerHost.PublicKey.String()] = models.IDandAddr{
  318. Name: peerHost.Name,
  319. ID: peer.ID.String(),
  320. Address: peer.PrimaryAddress(),
  321. IsServer: "no",
  322. }
  323. }
  324. if node.IsIngressGateway {
  325. extPeers, idsAndAddr, err := getExtPeers(node, true)
  326. if err == nil {
  327. peers = append(peers, extPeers...)
  328. for i := range idsAndAddr {
  329. peerMap[idsAndAddr[i].ID] = idsAndAddr[i]
  330. }
  331. } else if !database.IsEmptyRecord(err) {
  332. logger.Log(1, "error retrieving external clients:", err.Error())
  333. }
  334. }
  335. peerUpdate.Network = node.Network
  336. peerUpdate.ServerVersion = servercfg.Version
  337. peerUpdate.Peers = peers
  338. peerUpdate.ServerAddrs = serverNodeAddresses
  339. peerUpdate.DNS = getPeerDNS(node.Network)
  340. peerUpdate.PeerIDs = peerMap
  341. return peerUpdate, nil
  342. }
  343. func getExtPeers(node *models.Node, forIngressNode bool) ([]wgtypes.PeerConfig, []models.IDandAddr, error) {
  344. var peers []wgtypes.PeerConfig
  345. var idsAndAddr []models.IDandAddr
  346. extPeers, err := GetNetworkExtClients(node.Network)
  347. if err != nil {
  348. return peers, idsAndAddr, err
  349. }
  350. host, err := GetHost(node.HostID.String())
  351. if err != nil {
  352. return peers, idsAndAddr, err
  353. }
  354. for _, extPeer := range extPeers {
  355. pubkey, err := wgtypes.ParseKey(extPeer.PublicKey)
  356. if err != nil {
  357. logger.Log(1, "error parsing ext pub key:", err.Error())
  358. continue
  359. }
  360. if host.PublicKey.String() == extPeer.PublicKey {
  361. continue
  362. }
  363. var allowedips []net.IPNet
  364. var peer wgtypes.PeerConfig
  365. if forIngressNode && extPeer.Address != "" {
  366. var peeraddr = net.IPNet{
  367. IP: net.ParseIP(extPeer.Address),
  368. Mask: net.CIDRMask(32, 32),
  369. }
  370. if peeraddr.IP != nil && peeraddr.Mask != nil {
  371. allowedips = append(allowedips, peeraddr)
  372. }
  373. }
  374. if forIngressNode && extPeer.Address6 != "" {
  375. var addr6 = net.IPNet{
  376. IP: net.ParseIP(extPeer.Address6),
  377. Mask: net.CIDRMask(128, 128),
  378. }
  379. if addr6.IP != nil && addr6.Mask != nil {
  380. allowedips = append(allowedips, addr6)
  381. }
  382. }
  383. if !forIngressNode {
  384. if extPeer.InternalIPAddr != "" {
  385. peerInternalAddr := net.IPNet{
  386. IP: net.ParseIP(extPeer.InternalIPAddr),
  387. Mask: net.CIDRMask(32, 32),
  388. }
  389. if peerInternalAddr.IP != nil && peerInternalAddr.Mask != nil {
  390. allowedips = append(allowedips, peerInternalAddr)
  391. }
  392. }
  393. if extPeer.InternalIPAddr6 != "" {
  394. peerInternalAddr6 := net.IPNet{
  395. IP: net.ParseIP(extPeer.InternalIPAddr6),
  396. Mask: net.CIDRMask(32, 32),
  397. }
  398. if peerInternalAddr6.IP != nil && peerInternalAddr6.Mask != nil {
  399. allowedips = append(allowedips, peerInternalAddr6)
  400. }
  401. }
  402. }
  403. primaryAddr := extPeer.Address
  404. if primaryAddr == "" {
  405. primaryAddr = extPeer.Address6
  406. }
  407. peer = wgtypes.PeerConfig{
  408. PublicKey: pubkey,
  409. ReplaceAllowedIPs: true,
  410. AllowedIPs: allowedips,
  411. }
  412. peers = append(peers, peer)
  413. idsAndAddr = append(idsAndAddr, models.IDandAddr{
  414. ID: peer.PublicKey.String(),
  415. Address: primaryAddr,
  416. })
  417. }
  418. return peers, idsAndAddr, nil
  419. }
  420. func getExtPeersForProxy(node *models.Node, proxyPeerConf map[string]proxy_models.PeerConf) ([]wgtypes.PeerConfig, map[string]proxy_models.PeerConf, error) {
  421. var peers []wgtypes.PeerConfig
  422. host, err := GetHost(node.HostID.String())
  423. if err != nil {
  424. logger.Log(0, "error retrieving host for node", node.ID.String(), err.Error())
  425. }
  426. extPeers, err := GetNetworkExtClients(node.Network)
  427. if err != nil {
  428. return peers, proxyPeerConf, err
  429. }
  430. for _, extPeer := range extPeers {
  431. pubkey, err := wgtypes.ParseKey(extPeer.PublicKey)
  432. if err != nil {
  433. logger.Log(1, "error parsing ext pub key:", err.Error())
  434. continue
  435. }
  436. if host.PublicKey.String() == extPeer.PublicKey {
  437. continue
  438. }
  439. var allowedips []net.IPNet
  440. var peer wgtypes.PeerConfig
  441. if extPeer.Address != "" {
  442. var peeraddr = net.IPNet{
  443. IP: net.ParseIP(extPeer.Address),
  444. Mask: net.CIDRMask(32, 32),
  445. }
  446. if peeraddr.IP != nil && peeraddr.Mask != nil {
  447. allowedips = append(allowedips, peeraddr)
  448. }
  449. }
  450. if extPeer.Address6 != "" {
  451. var addr6 = net.IPNet{
  452. IP: net.ParseIP(extPeer.Address6),
  453. Mask: net.CIDRMask(128, 128),
  454. }
  455. if addr6.IP != nil && addr6.Mask != nil {
  456. allowedips = append(allowedips, addr6)
  457. }
  458. }
  459. peer = wgtypes.PeerConfig{
  460. PublicKey: pubkey,
  461. ReplaceAllowedIPs: true,
  462. AllowedIPs: allowedips,
  463. }
  464. extInternalPrimaryAddr := extPeer.InternalIPAddr
  465. if extInternalPrimaryAddr == "" {
  466. extInternalPrimaryAddr = extPeer.InternalIPAddr6
  467. }
  468. extConf := proxy_models.PeerConf{
  469. IsExtClient: true,
  470. Address: net.ParseIP(extPeer.Address),
  471. ExtInternalIp: net.ParseIP(extInternalPrimaryAddr),
  472. }
  473. if extPeer.IngressGatewayID == node.ID.String() {
  474. extConf.IsAttachedExtClient = true
  475. }
  476. ingGatewayUdpAddr, err := net.ResolveUDPAddr("udp", extPeer.IngressGatewayEndpoint)
  477. if err == nil {
  478. extConf.IngressGatewayEndPoint = ingGatewayUdpAddr
  479. }
  480. proxyPeerConf[peer.PublicKey.String()] = extConf
  481. peers = append(peers, peer)
  482. }
  483. return peers, proxyPeerConf, nil
  484. }
  485. // GetAllowedIPs - calculates the wireguard allowedip field for a peer of a node based on the peer and node settings
  486. func GetAllowedIPs(node, peer *models.Node, metrics *models.Metrics, fetchRelayedIps bool) []net.IPNet {
  487. var allowedips []net.IPNet
  488. allowedips = getNodeAllowedIPs(peer, node)
  489. // handle ingress gateway peers
  490. if peer.IsIngressGateway {
  491. extPeers, _, err := getExtPeers(peer, false)
  492. if err != nil {
  493. logger.Log(2, "could not retrieve ext peers for ", peer.ID.String(), err.Error())
  494. }
  495. for _, extPeer := range extPeers {
  496. allowedips = append(allowedips, extPeer.AllowedIPs...)
  497. }
  498. // if node is a failover node, add allowed ips from nodes it is handling
  499. if metrics != nil && peer.Failover && metrics.FailoverPeers != nil {
  500. // traverse through nodes that need handling
  501. logger.Log(3, "peer", peer.ID.String(), "was found to be failover for", node.ID.String(), "checking failover peers...")
  502. for k := range metrics.FailoverPeers {
  503. // if FailoverNode is me for this node, add allowedips
  504. if metrics.FailoverPeers[k] == peer.ID.String() {
  505. // get original node so we can traverse the allowed ips
  506. nodeToFailover, err := GetNodeByID(k)
  507. if err == nil {
  508. failoverNodeMetrics, err := GetMetrics(nodeToFailover.ID.String())
  509. if err == nil && failoverNodeMetrics != nil {
  510. if len(failoverNodeMetrics.NodeName) > 0 {
  511. allowedips = append(allowedips, getNodeAllowedIPs(&nodeToFailover, peer)...)
  512. logger.Log(0, "failing over node", nodeToFailover.ID.String(), nodeToFailover.PrimaryAddress(), "to failover node", peer.ID.String())
  513. }
  514. }
  515. }
  516. }
  517. }
  518. }
  519. }
  520. // handle relay gateway peers
  521. if fetchRelayedIps && peer.IsRelay {
  522. for _, ip := range peer.RelayAddrs {
  523. //find node ID of relayed peer
  524. relayedPeer, err := findNode(ip)
  525. if err != nil {
  526. logger.Log(0, "failed to find node for ip ", ip, err.Error())
  527. continue
  528. }
  529. if relayedPeer == nil {
  530. continue
  531. }
  532. if relayedPeer.ID == node.ID {
  533. //skip self
  534. continue
  535. }
  536. //check if acl permits comms
  537. if !nodeacls.AreNodesAllowed(nodeacls.NetworkID(node.Network), nodeacls.NodeID(node.ID.String()), nodeacls.NodeID(relayedPeer.ID.String())) {
  538. continue
  539. }
  540. if iplib.Version(net.ParseIP(ip)) == 4 {
  541. relayAddr := net.IPNet{
  542. IP: net.ParseIP(ip),
  543. Mask: net.CIDRMask(32, 32),
  544. }
  545. allowedips = append(allowedips, relayAddr)
  546. }
  547. if iplib.Version(net.ParseIP(ip)) == 6 {
  548. relayAddr := net.IPNet{
  549. IP: net.ParseIP(ip),
  550. Mask: net.CIDRMask(128, 128),
  551. }
  552. allowedips = append(allowedips, relayAddr)
  553. }
  554. relayedNode, err := findNode(ip)
  555. if err != nil {
  556. logger.Log(1, "unable to find node for relayed address", ip, err.Error())
  557. continue
  558. }
  559. if relayedNode.IsEgressGateway {
  560. extAllowedIPs := getEgressIPs(node, relayedNode)
  561. allowedips = append(allowedips, extAllowedIPs...)
  562. }
  563. if relayedNode.IsIngressGateway {
  564. extPeers, _, err := getExtPeers(relayedNode, false)
  565. if err == nil {
  566. for _, extPeer := range extPeers {
  567. allowedips = append(allowedips, extPeer.AllowedIPs...)
  568. }
  569. } else {
  570. logger.Log(0, "failed to retrieve extclients from relayed ingress", err.Error())
  571. }
  572. }
  573. }
  574. }
  575. return allowedips
  576. }
  577. func getPeerDNS(network string) string {
  578. var dns string
  579. if nodes, err := GetNetworkNodes(network); err == nil {
  580. for i, node := range nodes {
  581. host, err := GetHost(node.HostID.String())
  582. if err != nil {
  583. logger.Log(0, "error retrieving host for node", node.ID.String(), err.Error())
  584. }
  585. dns = dns + fmt.Sprintf("%s %s.%s\n", nodes[i].Address, host.Name, nodes[i].Network)
  586. }
  587. }
  588. if customDNSEntries, err := GetCustomDNS(network); err == nil {
  589. for _, entry := range customDNSEntries {
  590. // TODO - filter entries based on ACLs / given peers vs nodes in network
  591. dns = dns + fmt.Sprintf("%s %s.%s\n", entry.Address, entry.Name, entry.Network)
  592. }
  593. }
  594. return dns
  595. }
  596. // GetPeerUpdateForRelayedNode - calculates peer update for a relayed node by getting the relay
  597. // copying the relay node's allowed ips and making appropriate substitutions
  598. func GetPeerUpdateForRelayedNode(node *models.Node, udppeers map[string]string) (models.PeerUpdate, error) {
  599. var peerUpdate models.PeerUpdate
  600. var peers []wgtypes.PeerConfig
  601. var serverNodeAddresses = []models.ServerAddr{}
  602. var allowedips []net.IPNet
  603. //find node that is relaying us
  604. relay := FindRelay(node)
  605. if relay == nil {
  606. return models.PeerUpdate{}, errors.New("not found")
  607. }
  608. //add relay to lists of allowed ip
  609. if relay.Address.IP != nil {
  610. relayIP := relay.Address
  611. allowedips = append(allowedips, relayIP)
  612. }
  613. if relay.Address6.IP != nil {
  614. relayIP6 := relay.Address6
  615. allowedips = append(allowedips, relayIP6)
  616. }
  617. //get PeerUpdate for relayed node
  618. relayPeerUpdate, err := GetPeerUpdate(relay)
  619. if err != nil {
  620. return models.PeerUpdate{}, err
  621. }
  622. //add the relays allowed ips from all of the relay's peers
  623. for _, peer := range relayPeerUpdate.Peers {
  624. allowedips = append(allowedips, peer.AllowedIPs...)
  625. }
  626. //delete any ips not permitted by acl
  627. for i := len(allowedips) - 1; i >= 0; i-- {
  628. target, err := findNode(allowedips[i].IP.String())
  629. if err != nil {
  630. logger.Log(0, "failed to find node for ip", allowedips[i].IP.String(), err.Error())
  631. continue
  632. }
  633. if target == nil {
  634. logger.Log(0, "failed to find node for ip", allowedips[i].IP.String())
  635. continue
  636. }
  637. if !nodeacls.AreNodesAllowed(nodeacls.NetworkID(node.Network), nodeacls.NodeID(node.ID.String()), nodeacls.NodeID(target.ID.String())) {
  638. logger.Log(0, "deleting node from relayednode per acl", node.ID.String(), target.ID.String())
  639. allowedips = append(allowedips[:i], allowedips[i+1:]...)
  640. }
  641. }
  642. //delete self from allowed ips
  643. for i := len(allowedips) - 1; i >= 0; i-- {
  644. if allowedips[i].IP.String() == node.Address.IP.String() || allowedips[i].IP.String() == node.Address6.IP.String() {
  645. allowedips = append(allowedips[:i], allowedips[i+1:]...)
  646. }
  647. }
  648. //delete egressrange from allowedip if we are egress gateway
  649. if node.IsEgressGateway {
  650. for i := len(allowedips) - 1; i >= 0; i-- {
  651. if StringSliceContains(node.EgressGatewayRanges, allowedips[i].String()) {
  652. allowedips = append(allowedips[:i], allowedips[i+1:]...)
  653. }
  654. }
  655. }
  656. //delete extclients from allowedip if we are ingress gateway
  657. if node.IsIngressGateway {
  658. for i := len(allowedips) - 1; i >= 0; i-- {
  659. if strings.Contains(node.IngressGatewayRange, allowedips[i].IP.String()) {
  660. allowedips = append(allowedips[:i], allowedips[i+1:]...)
  661. }
  662. }
  663. }
  664. //add egress range if relay is egress
  665. if relay.IsEgressGateway {
  666. var ip *net.IPNet
  667. for _, cidr := range relay.EgressGatewayRanges {
  668. _, ip, err = net.ParseCIDR(cidr)
  669. if err != nil {
  670. continue
  671. }
  672. }
  673. allowedips = append(allowedips, *ip)
  674. }
  675. relayHost, err := GetHost(relay.HostID.String())
  676. if err != nil {
  677. logger.Log(0, "error retrieving host for relay node", node.ID.String(), err.Error())
  678. }
  679. if err != nil {
  680. return models.PeerUpdate{}, err
  681. }
  682. var setUDPPort = false
  683. var listenPort int
  684. if CheckEndpoint(udppeers[relayHost.PublicKey.String()]) {
  685. endpointstring := udppeers[relayHost.PublicKey.String()]
  686. endpointarr := strings.Split(endpointstring, ":")
  687. if len(endpointarr) == 2 {
  688. port, err := strconv.Atoi(endpointarr[1])
  689. if err == nil {
  690. setUDPPort = true
  691. listenPort = port
  692. }
  693. }
  694. }
  695. // if udp hole punching is on, but udp hole punching did not set it, use the LocalListenPort instead
  696. // or, if port is for some reason zero use the LocalListenPort
  697. // but only do this if LocalListenPort is not zero
  698. if ((!setUDPPort) || relayHost.ListenPort == 0) && relayHost.LocalListenPort != 0 {
  699. listenPort = relayHost.LocalListenPort
  700. }
  701. endpoint := relayHost.EndpointIP.String() + ":" + strconv.FormatInt(int64(listenPort), 10)
  702. address, err := net.ResolveUDPAddr("udp", endpoint)
  703. if err != nil {
  704. return models.PeerUpdate{}, err
  705. }
  706. var keepalive time.Duration
  707. if node.PersistentKeepalive != 0 {
  708. // set_keepalive
  709. keepalive, _ = time.ParseDuration(strconv.FormatInt(int64(node.PersistentKeepalive), 10) + "s")
  710. }
  711. var peerData = wgtypes.PeerConfig{
  712. PublicKey: relayHost.PublicKey,
  713. Endpoint: address,
  714. ReplaceAllowedIPs: true,
  715. AllowedIPs: allowedips,
  716. PersistentKeepaliveInterval: &keepalive,
  717. }
  718. peers = append(peers, peerData)
  719. //if ingress add extclients
  720. if node.IsIngressGateway {
  721. extPeers, _, err := getExtPeers(node, true)
  722. if err == nil {
  723. peers = append(peers, extPeers...)
  724. } else {
  725. logger.Log(2, "could not retrieve ext peers for ", node.ID.String(), err.Error())
  726. }
  727. }
  728. peerUpdate.Network = node.Network
  729. peerUpdate.ServerVersion = servercfg.Version
  730. peerUpdate.Peers = peers
  731. peerUpdate.ServerAddrs = serverNodeAddresses
  732. peerUpdate.DNS = getPeerDNS(node.Network)
  733. return peerUpdate, nil
  734. }
  735. func getEgressIPs(node, peer *models.Node) []net.IPNet {
  736. host, err := GetHost(node.HostID.String())
  737. if err != nil {
  738. logger.Log(0, "error retrieving host for node", node.ID.String(), err.Error())
  739. }
  740. peerHost, err := GetHost(peer.ID.String())
  741. if err != nil {
  742. logger.Log(0, "error retrieving host for peer", peer.ID.String(), err.Error())
  743. }
  744. //check for internet gateway
  745. internetGateway := false
  746. if slices.Contains(peer.EgressGatewayRanges, "0.0.0.0/0") || slices.Contains(peer.EgressGatewayRanges, "::/0") {
  747. internetGateway = true
  748. }
  749. allowedips := []net.IPNet{}
  750. for _, iprange := range peer.EgressGatewayRanges { // go through each cidr for egress gateway
  751. _, ipnet, err := net.ParseCIDR(iprange) // confirming it's valid cidr
  752. if err != nil {
  753. logger.Log(1, "could not parse gateway IP range. Not adding ", iprange)
  754. continue // if can't parse CIDR
  755. }
  756. nodeEndpointArr := strings.Split(peerHost.EndpointIP.String(), ":") // getting the public ip of node
  757. if ipnet.Contains(net.ParseIP(nodeEndpointArr[0])) && !internetGateway { // ensuring egress gateway range does not contain endpoint of node
  758. logger.Log(2, "egress IP range of ", iprange, " overlaps with ", host.EndpointIP.String(), ", omitting")
  759. continue // skip adding egress range if overlaps with node's ip
  760. }
  761. // TODO: Could put in a lot of great logic to avoid conflicts / bad routes
  762. if ipnet.Contains(net.ParseIP(host.LocalAddress.String())) && !internetGateway { // ensuring egress gateway range does not contain public ip of node
  763. logger.Log(2, "egress IP range of ", iprange, " overlaps with ", host.LocalAddress.String(), ", omitting")
  764. continue // skip adding egress range if overlaps with node's local ip
  765. }
  766. if err != nil {
  767. logger.Log(1, "error encountered when setting egress range", err.Error())
  768. } else {
  769. allowedips = append(allowedips, *ipnet)
  770. }
  771. }
  772. return allowedips
  773. }
  774. func getNodeAllowedIPs(peer, node *models.Node) []net.IPNet {
  775. var allowedips = []net.IPNet{}
  776. host, err := GetHost(node.ID.String())
  777. if err != nil {
  778. logger.Log(0, "error retrieving host for node", node.ID.String(), err.Error())
  779. }
  780. if peer.Address.IP != nil {
  781. allowedips = append(allowedips, peer.Address)
  782. }
  783. if peer.Address6.IP != nil {
  784. allowedips = append(allowedips, peer.Address6)
  785. }
  786. // handle manually set peers
  787. for _, allowedIp := range peer.AllowedIPs {
  788. // parsing as a CIDR first. If valid CIDR, append
  789. if _, ipnet, err := net.ParseCIDR(allowedIp); err == nil {
  790. nodeEndpointArr := strings.Split(host.EndpointIP.String(), ":")
  791. if !ipnet.Contains(net.IP(nodeEndpointArr[0])) && ipnet.IP.String() != peer.Address.IP.String() { // don't need to add an allowed ip that already exists..
  792. allowedips = append(allowedips, *ipnet)
  793. }
  794. } else { // parsing as an IP second. If valid IP, check if ipv4 or ipv6, then append
  795. if iplib.Version(net.ParseIP(allowedIp)) == 4 && allowedIp != peer.Address.IP.String() {
  796. ipnet := net.IPNet{
  797. IP: net.ParseIP(allowedIp),
  798. Mask: net.CIDRMask(32, 32),
  799. }
  800. allowedips = append(allowedips, ipnet)
  801. } else if iplib.Version(net.ParseIP(allowedIp)) == 6 && allowedIp != peer.Address6.IP.String() {
  802. ipnet := net.IPNet{
  803. IP: net.ParseIP(allowedIp),
  804. Mask: net.CIDRMask(128, 128),
  805. }
  806. allowedips = append(allowedips, ipnet)
  807. }
  808. }
  809. }
  810. // handle egress gateway peers
  811. if peer.IsEgressGateway {
  812. //hasGateway = true
  813. egressIPs := getEgressIPs(node, peer)
  814. allowedips = append(allowedips, egressIPs...)
  815. }
  816. return allowedips
  817. }