peers.go 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680
  1. package logic
  2. import (
  3. "errors"
  4. "net"
  5. "net/netip"
  6. "github.com/gravitl/netmaker/database"
  7. "github.com/gravitl/netmaker/logger"
  8. "github.com/gravitl/netmaker/logic/acls/nodeacls"
  9. "github.com/gravitl/netmaker/models"
  10. "github.com/gravitl/netmaker/servercfg"
  11. "golang.org/x/exp/slices"
  12. "golang.zx2c4.com/wireguard/wgctrl/wgtypes"
  13. )
  14. // NodePeersInfo - fetches node's peers with their ids and addrs.
  15. func NodePeersInfo(client *models.Client) (models.NodePeersInfo, error) {
  16. nodePeersInfo := models.NodePeersInfo{
  17. PeerIDs: make(models.PeerMap),
  18. Peers: []wgtypes.PeerConfig{},
  19. }
  20. nodes, err := GetNetworkNodes(client.Node.Network)
  21. if err != nil {
  22. return models.NodePeersInfo{}, err
  23. }
  24. for _, peer := range nodes {
  25. if peer.ID == client.Node.ID {
  26. continue
  27. }
  28. if (client.Node.IsRelayed && client.Node.RelayedBy != peer.ID.String()) ||
  29. (peer.IsRelayed && peer.RelayedBy != client.Node.ID.String()) ||
  30. ShouldRemovePeer(client.Node, peer) {
  31. continue
  32. }
  33. peerHost, err := GetHost(peer.HostID.String())
  34. if err != nil {
  35. continue
  36. }
  37. var peerConfig wgtypes.PeerConfig
  38. peerConfig.PublicKey = peerHost.PublicKey
  39. peerConfig.PersistentKeepaliveInterval = &peer.PersistentKeepalive
  40. peerConfig.ReplaceAllowedIPs = true
  41. uselocal := false
  42. if client.Host.EndpointIP.String() == peerHost.EndpointIP.String() {
  43. // peer is on same network
  44. // set to localaddress
  45. uselocal = true
  46. if client.Node.LocalAddress.IP == nil {
  47. // use public endpint
  48. uselocal = false
  49. }
  50. if client.Node.LocalAddress.String() == peer.LocalAddress.String() {
  51. uselocal = false
  52. }
  53. }
  54. peerConfig.Endpoint = &net.UDPAddr{
  55. IP: peerHost.EndpointIP,
  56. Port: getPeerWgListenPort(peerHost),
  57. }
  58. if uselocal {
  59. peerConfig.Endpoint.IP = peer.LocalAddress.IP
  60. peerConfig.Endpoint.Port = peerHost.ListenPort
  61. }
  62. allowedips := GetNetworkAllowedIPs(*client, models.Client{Host: *peerHost, Node: peer})
  63. peerConfig.AllowedIPs = allowedips
  64. nodePeersInfo.Peers = append(nodePeersInfo.Peers, peerConfig)
  65. nodePeersInfo.PeerIDs[peerHost.PublicKey.String()] = models.IDandAddr{
  66. ID: peer.ID.String(),
  67. Address: peer.Address.IP.String(),
  68. Name: peerHost.Name,
  69. Network: peer.Network,
  70. ListenPort: GetPeerListenPort(peerHost),
  71. }
  72. }
  73. if client.Node.IsIngressGateway {
  74. extPeers, extPeerIDAndAddrs, err := GetExtPeers(&client.Node)
  75. if err == nil {
  76. nodePeersInfo.Peers = append(nodePeersInfo.Peers, extPeers...)
  77. for _, extPeerIdAndAddr := range extPeerIDAndAddrs {
  78. extPeerIdAndAddr := extPeerIdAndAddr
  79. nodePeersInfo.PeerIDs[extPeerIdAndAddr.ID] = extPeerIdAndAddr
  80. }
  81. } else if !database.IsEmptyRecord(err) {
  82. logger.Log(1, "error retrieving external clients:", err.Error())
  83. }
  84. }
  85. return nodePeersInfo, nil
  86. }
  87. // GetPeerUpdateForHost - gets the consolidated peer update for the host from all networks
  88. func GetPeerUpdateForHost(host *models.Host, allNodes []models.Node) (models.HostPeerUpdate, error) {
  89. if host == nil {
  90. return models.HostPeerUpdate{}, errors.New("host is nil")
  91. }
  92. // track which nodes are deleted
  93. // after peer calculation, if peer not in list, add delete config of peer
  94. hostPeerUpdate := models.HostPeerUpdate{
  95. Host: *host,
  96. Server: servercfg.GetServer(),
  97. ServerVersion: servercfg.GetVersion(),
  98. Peers: []wgtypes.PeerConfig{},
  99. HostNetworkInfo: models.HostInfoMap{},
  100. }
  101. logger.Log(1, "peer update for host", host.ID.String())
  102. peerIndexMap := make(map[string]int)
  103. for _, nodeID := range host.Nodes {
  104. nodeID := nodeID
  105. node, err := GetNodeByID(nodeID)
  106. if err != nil {
  107. continue
  108. }
  109. if !node.Connected || node.PendingDelete || node.Action == models.NODE_DELETE {
  110. continue
  111. }
  112. currentPeers := GetNetworkNodesMemory(allNodes, node.Network)
  113. for _, peer := range currentPeers {
  114. peer := peer
  115. if peer.ID.String() == node.ID.String() {
  116. logger.Log(2, "peer update, skipping self")
  117. //skip yourself
  118. continue
  119. }
  120. peerHost, err := GetHost(peer.HostID.String())
  121. if err != nil {
  122. logger.Log(1, "no peer host", peer.HostID.String(), err.Error())
  123. return models.HostPeerUpdate{}, err
  124. }
  125. peerConfig := wgtypes.PeerConfig{
  126. PublicKey: peerHost.PublicKey,
  127. PersistentKeepaliveInterval: &peer.PersistentKeepalive,
  128. ReplaceAllowedIPs: true,
  129. }
  130. if (node.IsRelayed && node.RelayedBy != peer.ID.String()) ||
  131. (peer.IsRelayed && peer.RelayedBy != node.ID.String()) ||
  132. ShouldRemovePeer(node, peer) {
  133. // if node is relayed and peer is not the relay, set remove to true
  134. peerConfig.Remove = true
  135. hostPeerUpdate.Peers = append(hostPeerUpdate.Peers, peerConfig)
  136. peerIndexMap[peerHost.PublicKey.String()] = len(hostPeerUpdate.Peers) - 1
  137. continue
  138. }
  139. uselocal := false
  140. if host.EndpointIP.String() == peerHost.EndpointIP.String() {
  141. // peer is on same network
  142. // set to localaddress
  143. uselocal = true
  144. if node.LocalAddress.IP == nil {
  145. // use public endpint
  146. uselocal = false
  147. }
  148. if node.LocalAddress.String() == peer.LocalAddress.String() {
  149. uselocal = false
  150. }
  151. }
  152. peerConfig.Endpoint = &net.UDPAddr{
  153. IP: peerHost.EndpointIP,
  154. Port: getPeerWgListenPort(peerHost),
  155. }
  156. if uselocal {
  157. peerConfig.Endpoint.IP = peer.LocalAddress.IP
  158. peerConfig.Endpoint.Port = peerHost.ListenPort
  159. }
  160. peerConfig.AllowedIPs = GetNetworkAllowedIPs(models.Client{Host: *host, Node: node}, models.Client{Host: *peerHost, Node: peer})
  161. if _, ok := peerIndexMap[peerHost.PublicKey.String()]; !ok {
  162. hostPeerUpdate.Peers = append(hostPeerUpdate.Peers, peerConfig)
  163. peerIndexMap[peerHost.PublicKey.String()] = len(hostPeerUpdate.Peers) - 1
  164. hostPeerUpdate.HostNetworkInfo[peerHost.PublicKey.String()] = models.HostNetworkInfo{
  165. Interfaces: peerHost.Interfaces,
  166. }
  167. } else {
  168. peerAllowedIPs := hostPeerUpdate.Peers[peerIndexMap[peerHost.PublicKey.String()]].AllowedIPs
  169. peerAllowedIPs = append(peerAllowedIPs, peerConfig.AllowedIPs...)
  170. hostPeerUpdate.Peers[peerIndexMap[peerHost.PublicKey.String()]].Remove = false
  171. hostPeerUpdate.Peers[peerIndexMap[peerHost.PublicKey.String()]].AllowedIPs = peerAllowedIPs
  172. }
  173. }
  174. if node.IsIngressGateway {
  175. extPeers, _, err := GetExtPeers(&node)
  176. if err == nil {
  177. hostPeerUpdate.Peers = append(hostPeerUpdate.Peers, extPeers...)
  178. } else if !database.IsEmptyRecord(err) {
  179. logger.Log(1, "error retrieving external clients:", err.Error())
  180. }
  181. }
  182. }
  183. return hostPeerUpdate, nil
  184. }
  185. func ShouldRemovePeer(node, peer models.Node) (remove bool) {
  186. if peer.Action == models.NODE_DELETE || peer.PendingDelete || !peer.Connected ||
  187. !nodeacls.AreNodesAllowed(nodeacls.NetworkID(node.Network), nodeacls.NodeID(node.ID.String()), nodeacls.NodeID(peer.ID.String())) {
  188. remove = true
  189. }
  190. return
  191. }
  192. // GetFwUpdate - fetches the firewall update for the gateway nodes on the host
  193. func GetFwUpdate(host *models.Host) (models.FwUpdate, error) {
  194. fwUpdate := models.FwUpdate{
  195. IngressInfo: models.IngressInfo{
  196. ExtPeers: make(map[string]models.ExtClientInfo),
  197. },
  198. EgressInfo: make(map[string]models.EgressInfo),
  199. }
  200. allNodes, err := GetAllNodes()
  201. if err != nil {
  202. return fwUpdate, err
  203. }
  204. for _, nodeID := range host.Nodes {
  205. nodeID := nodeID
  206. node, err := GetNodeByID(nodeID)
  207. if err != nil {
  208. continue
  209. }
  210. if !(node.IsEgressGateway || node.IsIngressGateway) {
  211. continue
  212. }
  213. if !node.Connected || node.PendingDelete || node.Action == models.NODE_DELETE {
  214. continue
  215. }
  216. currentPeers := GetNetworkNodesMemory(allNodes, node.Network)
  217. var nodePeerMap map[string]models.PeerRouteInfo
  218. if node.IsIngressGateway || node.IsEgressGateway {
  219. nodePeerMap = make(map[string]models.PeerRouteInfo)
  220. }
  221. for _, peer := range currentPeers {
  222. peer := peer
  223. if peer.ID.String() == node.ID.String() {
  224. logger.Log(2, "fw update, skipping self")
  225. //skip yourself
  226. continue
  227. }
  228. peerHost, err := GetHost(peer.HostID.String())
  229. if err != nil {
  230. logger.Log(1, "no peer host", peer.HostID.String(), err.Error())
  231. continue
  232. }
  233. if node.IsIngressGateway || node.IsEgressGateway {
  234. if peer.IsIngressGateway {
  235. _, extPeerIDAndAddrs, err := GetExtPeers(&peer)
  236. if err == nil {
  237. for _, extPeerIdAndAddr := range extPeerIDAndAddrs {
  238. extPeerIdAndAddr := extPeerIdAndAddr
  239. nodePeerMap[extPeerIdAndAddr.ID] = models.PeerRouteInfo{
  240. PeerAddr: net.IPNet{
  241. IP: net.ParseIP(extPeerIdAndAddr.Address),
  242. Mask: getCIDRMaskFromAddr(extPeerIdAndAddr.Address),
  243. },
  244. PeerKey: extPeerIdAndAddr.ID,
  245. Allow: true,
  246. ID: extPeerIdAndAddr.ID,
  247. }
  248. }
  249. }
  250. }
  251. if node.IsIngressGateway && peer.IsEgressGateway {
  252. fwUpdate.IngressInfo.EgressRanges = append(fwUpdate.IngressInfo.EgressRanges,
  253. peer.EgressGatewayRanges...)
  254. }
  255. nodePeerMap[peerHost.PublicKey.String()] = models.PeerRouteInfo{
  256. PeerAddr: net.IPNet{
  257. IP: net.ParseIP(peer.PrimaryAddress()),
  258. Mask: getCIDRMaskFromAddr(peer.PrimaryAddress()),
  259. },
  260. PeerKey: peerHost.PublicKey.String(),
  261. Allow: true,
  262. ID: peer.ID.String(),
  263. }
  264. }
  265. }
  266. var extPeerIDAndAddrs []models.IDandAddr
  267. if node.IsIngressGateway {
  268. fwUpdate.IsIngressGw = true
  269. _, extPeerIDAndAddrs, err = GetExtPeers(&node)
  270. if err == nil {
  271. for _, extPeerIdAndAddr := range extPeerIDAndAddrs {
  272. extPeerIdAndAddr := extPeerIdAndAddr
  273. nodePeerMap[extPeerIdAndAddr.ID] = models.PeerRouteInfo{
  274. PeerAddr: net.IPNet{
  275. IP: net.ParseIP(extPeerIdAndAddr.Address),
  276. Mask: getCIDRMaskFromAddr(extPeerIdAndAddr.Address),
  277. },
  278. PeerKey: extPeerIdAndAddr.ID,
  279. Allow: true,
  280. ID: extPeerIdAndAddr.ID,
  281. }
  282. }
  283. for _, extPeerIdAndAddr := range extPeerIDAndAddrs {
  284. extPeerIdAndAddr := extPeerIdAndAddr
  285. fwUpdate.IngressInfo.ExtPeers[extPeerIdAndAddr.ID] = models.ExtClientInfo{
  286. Masquerade: true,
  287. IngGwAddr: net.IPNet{
  288. IP: net.ParseIP(node.PrimaryAddress()),
  289. Mask: getCIDRMaskFromAddr(node.PrimaryAddress()),
  290. },
  291. Network: node.PrimaryNetworkRange(),
  292. ExtPeerAddr: net.IPNet{
  293. IP: net.ParseIP(extPeerIdAndAddr.Address),
  294. Mask: getCIDRMaskFromAddr(extPeerIdAndAddr.Address),
  295. },
  296. ExtPeerKey: extPeerIdAndAddr.ID,
  297. Peers: filterNodeMapForClientACLs(extPeerIdAndAddr.ID, node.Network, nodePeerMap),
  298. }
  299. }
  300. } else if !database.IsEmptyRecord(err) {
  301. logger.Log(1, "error retrieving external clients:", err.Error())
  302. }
  303. }
  304. if node.IsEgressGateway {
  305. fwUpdate.IsEgressGw = true
  306. fwUpdate.EgressInfo[node.ID.String()] = models.EgressInfo{
  307. EgressID: node.ID.String(),
  308. Network: node.PrimaryNetworkRange(),
  309. EgressGwAddr: net.IPNet{
  310. IP: net.ParseIP(node.PrimaryAddress()),
  311. Mask: getCIDRMaskFromAddr(node.PrimaryAddress()),
  312. },
  313. GwPeers: nodePeerMap,
  314. EgressGWCfg: node.EgressGatewayRequest,
  315. }
  316. }
  317. }
  318. return fwUpdate, nil
  319. }
  320. // getPeerWgListenPort - fetches the wg listen port for the host
  321. func getPeerWgListenPort(host *models.Host) int {
  322. peerPort := host.ListenPort
  323. if host.WgPublicListenPort != 0 {
  324. peerPort = host.WgPublicListenPort
  325. }
  326. return peerPort
  327. }
  328. // GetPeerListenPort - given a host, retrieve it's appropriate listening port
  329. func GetPeerListenPort(host *models.Host) int {
  330. peerPort := host.ListenPort
  331. if host.WgPublicListenPort != 0 {
  332. peerPort = host.WgPublicListenPort
  333. }
  334. return peerPort
  335. }
  336. func GetExtPeers(node *models.Node) ([]wgtypes.PeerConfig, []models.IDandAddr, error) {
  337. var peers []wgtypes.PeerConfig
  338. var idsAndAddr []models.IDandAddr
  339. extPeers, err := GetNetworkExtClients(node.Network)
  340. if err != nil {
  341. return peers, idsAndAddr, err
  342. }
  343. host, err := GetHost(node.HostID.String())
  344. if err != nil {
  345. return peers, idsAndAddr, err
  346. }
  347. for _, extPeer := range extPeers {
  348. extPeer := extPeer
  349. pubkey, err := wgtypes.ParseKey(extPeer.PublicKey)
  350. if err != nil {
  351. logger.Log(1, "error parsing ext pub key:", err.Error())
  352. continue
  353. }
  354. if host.PublicKey.String() == extPeer.PublicKey ||
  355. extPeer.IngressGatewayID != node.ID.String() || !extPeer.Enabled {
  356. continue
  357. }
  358. var allowedips []net.IPNet
  359. var peer wgtypes.PeerConfig
  360. if extPeer.Address != "" {
  361. var peeraddr = net.IPNet{
  362. IP: net.ParseIP(extPeer.Address),
  363. Mask: net.CIDRMask(32, 32),
  364. }
  365. if peeraddr.IP != nil && peeraddr.Mask != nil {
  366. allowedips = append(allowedips, peeraddr)
  367. }
  368. }
  369. if extPeer.Address6 != "" {
  370. var addr6 = net.IPNet{
  371. IP: net.ParseIP(extPeer.Address6),
  372. Mask: net.CIDRMask(128, 128),
  373. }
  374. if addr6.IP != nil && addr6.Mask != nil {
  375. allowedips = append(allowedips, addr6)
  376. }
  377. }
  378. primaryAddr := extPeer.Address
  379. if primaryAddr == "" {
  380. primaryAddr = extPeer.Address6
  381. }
  382. peer = wgtypes.PeerConfig{
  383. PublicKey: pubkey,
  384. ReplaceAllowedIPs: true,
  385. AllowedIPs: allowedips,
  386. }
  387. peers = append(peers, peer)
  388. idsAndAddr = append(idsAndAddr, models.IDandAddr{
  389. ID: peer.PublicKey.String(),
  390. Name: extPeer.ClientID,
  391. Address: primaryAddr,
  392. IsExtclient: true,
  393. })
  394. }
  395. return peers, idsAndAddr, nil
  396. }
  397. func getNodeByNetworkFromHost(h *models.Host, network string) *models.Node {
  398. for _, nodeID := range h.Nodes {
  399. node, err := GetNodeByID(nodeID)
  400. if err == nil && node.Network == network {
  401. return &node
  402. }
  403. }
  404. return nil
  405. }
  406. // GetAllowedIPs - calculates the wireguard allowedip field for a peer of a node based on the peer and node settings
  407. func GetAllowedIPs(client, peer models.Client) []net.IPNet {
  408. var allowedips []net.IPNet
  409. for _, nodeID := range peer.Host.Nodes {
  410. node, err := GetNodeByID(nodeID)
  411. if err != nil {
  412. continue
  413. }
  414. clientNode := getNodeByNetworkFromHost(&client.Host, node.Network)
  415. if clientNode == nil {
  416. continue
  417. }
  418. client.Node = *clientNode
  419. peer.Node = node
  420. if ShouldRemovePeer(*clientNode, peer.Node) {
  421. continue
  422. }
  423. if peer.Node.Address.IP != nil {
  424. allowed := net.IPNet{
  425. IP: peer.Node.Address.IP,
  426. Mask: net.CIDRMask(32, 32),
  427. }
  428. allowedips = append(allowedips, allowed)
  429. }
  430. if peer.Node.Address6.IP != nil {
  431. allowed := net.IPNet{
  432. IP: peer.Node.Address6.IP,
  433. Mask: net.CIDRMask(128, 128),
  434. }
  435. allowedips = append(allowedips, allowed)
  436. }
  437. // handle egress gateway peers
  438. if peer.Node.IsEgressGateway {
  439. allowedips = append(allowedips, getEgressIPs(peer)...)
  440. }
  441. if peer.Node.IsRelay {
  442. allowedips = append(allowedips, getRelayAllowedIPs(client, peer)...)
  443. }
  444. // handle ingress gateway peers
  445. if peer.Node.IsIngressGateway {
  446. allowedips = append(allowedips, getIngressIPs(peer)...)
  447. }
  448. }
  449. return allowedips
  450. }
  451. // GetNetworkAllowedIPs - calculates the wireguard allowedip field for a peer of a node based on the peer and node settings
  452. func GetNetworkAllowedIPs(client, peer models.Client) []net.IPNet {
  453. var allowedips []net.IPNet
  454. if peer.Node.Address.IP != nil {
  455. allowed := net.IPNet{
  456. IP: peer.Node.Address.IP,
  457. Mask: net.CIDRMask(32, 32),
  458. }
  459. allowedips = append(allowedips, allowed)
  460. }
  461. if peer.Node.Address6.IP != nil {
  462. allowed := net.IPNet{
  463. IP: peer.Node.Address6.IP,
  464. Mask: net.CIDRMask(128, 128),
  465. }
  466. allowedips = append(allowedips, allowed)
  467. }
  468. // handle egress gateway peers
  469. if peer.Node.IsEgressGateway {
  470. egressIPs := getEgressIPs(peer)
  471. allowedips = append(allowedips, egressIPs...)
  472. }
  473. if peer.Node.IsRelay {
  474. allowedips = append(allowedips, getRelayAllowedIPs(client, peer)...)
  475. }
  476. // handle ingress gateway peers
  477. if peer.Node.IsIngressGateway {
  478. allowedips = append(allowedips, getIngressIPs(peer)...)
  479. }
  480. return allowedips
  481. }
  482. // getEgressIPs - gets the egress IPs for a client
  483. func getEgressIPs(client models.Client) []net.IPNet {
  484. //check for internet gateway
  485. internetGateway := false
  486. if slices.Contains(client.Node.EgressGatewayRanges, "0.0.0.0/0") || slices.Contains(client.Node.EgressGatewayRanges, "::/0") {
  487. internetGateway = true
  488. }
  489. allowedips := []net.IPNet{}
  490. for _, iprange := range client.Node.EgressGatewayRanges { // go through each cidr for egress gateway
  491. ip, cidr, err := net.ParseCIDR(iprange) // confirming it's valid cidr
  492. if err != nil {
  493. logger.Log(1, "could not parse gateway IP range. Not adding ", iprange)
  494. continue // if can't parse CIDR
  495. }
  496. cidr.IP = ip
  497. // getting the public ip of node
  498. if cidr.Contains(client.Host.EndpointIP) && !internetGateway { // ensuring egress gateway range does not contain endpoint of node
  499. logger.Log(2, "egress IP range of ", iprange, " overlaps with ", client.Host.EndpointIP.String(), ", omitting")
  500. continue // skip adding egress range if overlaps with node's ip
  501. }
  502. // TODO: Could put in a lot of great logic to avoid conflicts / bad routes
  503. if cidr.Contains(client.Node.LocalAddress.IP) && !internetGateway { // ensuring egress gateway range does not contain public ip of node
  504. logger.Log(2, "egress IP range of ", iprange, " overlaps with ", client.Node.LocalAddress.String(), ", omitting")
  505. continue // skip adding egress range if overlaps with node's local ip
  506. }
  507. if err != nil {
  508. logger.Log(1, "error encountered when setting egress range", err.Error())
  509. } else {
  510. allowedips = append(allowedips, *cidr)
  511. }
  512. }
  513. return allowedips
  514. }
  515. func getCIDRMaskFromAddr(addr string) net.IPMask {
  516. cidr := net.CIDRMask(32, 32)
  517. ipAddr, err := netip.ParseAddr(addr)
  518. if err != nil {
  519. return cidr
  520. }
  521. if ipAddr.Is6() {
  522. cidr = net.CIDRMask(128, 128)
  523. }
  524. return cidr
  525. }
  526. // accounts for ext client ACLs
  527. func filterNodeMapForClientACLs(publicKey, network string, nodePeerMap map[string]models.PeerRouteInfo) map[string]models.PeerRouteInfo {
  528. if !isEE {
  529. return nodePeerMap
  530. }
  531. if nodePeerMap == nil {
  532. return map[string]models.PeerRouteInfo{}
  533. }
  534. if len(publicKey) == 0 || len(network) == 0 {
  535. return nodePeerMap
  536. }
  537. client, err := GetExtClientByPubKey(publicKey, network)
  538. if err != nil {
  539. return nodePeerMap
  540. }
  541. for k := range nodePeerMap {
  542. currNodePeer := nodePeerMap[k]
  543. if _, ok := client.ACLs[currNodePeer.ID]; ok {
  544. delete(nodePeerMap, k)
  545. }
  546. }
  547. return nodePeerMap
  548. }
  549. // getRelayAllowedIPs returns the list of allowedips for a peer that is a relay
  550. func getRelayAllowedIPs(client, relayPeer models.Client) []net.IPNet {
  551. var relayIPs []net.IPNet
  552. if !relayPeer.Node.IsRelay {
  553. logger.Log(0, "getRelayAllowedIPs called for a non-relay node", relayPeer.Host.Name)
  554. return relayIPs
  555. }
  556. for _, relayed := range relayPeer.Node.RelayedNodes {
  557. relayedNode, err := GetNodeByID(relayed)
  558. if err != nil {
  559. logger.Log(0, "retrieve relayed node", err.Error())
  560. continue
  561. }
  562. if relayedNode.ID == client.Node.ID {
  563. relayIPs = append(relayIPs, getAllowedIpsForRelayed(client, relayPeer)...)
  564. continue
  565. }
  566. if relayedNode.Address.IP != nil {
  567. relayedNode.Address.Mask = net.CIDRMask(32, 32)
  568. relayIPs = append(relayIPs, relayedNode.Address)
  569. }
  570. if relayedNode.Address6.IP != nil {
  571. relayedNode.Address.Mask = net.CIDRMask(128, 128)
  572. relayIPs = append(relayIPs, relayedNode.Address6)
  573. }
  574. if relayedNode.IsEgressGateway {
  575. relayedHost, err := GetHost(relayedNode.HostID.String())
  576. if err == nil {
  577. relayIPs = append(relayIPs, getEgressIPs(models.Client{
  578. Host: *relayedHost,
  579. Node: relayedNode,
  580. })...)
  581. }
  582. }
  583. }
  584. return relayIPs
  585. }
  586. // getAllowedIpsForRelayed - returns the peerConfig for a node relayed by relay
  587. func getAllowedIpsForRelayed(relayed, relay models.Client) (allowedIPs []net.IPNet) {
  588. if relayed.Node.RelayedBy != relay.Node.ID.String() {
  589. logger.Log(0, "RelayedByRelay called with invalid parameters")
  590. return
  591. }
  592. peers, err := GetNetworkClients(relay.Node.Network)
  593. if err != nil {
  594. logger.Log(0, "error getting network clients", err.Error())
  595. return
  596. }
  597. for _, peer := range peers {
  598. if peer.Node.ID == relayed.Node.ID || peer.Node.ID == relay.Node.ID {
  599. continue
  600. }
  601. if nodeacls.AreNodesAllowed(nodeacls.NetworkID(relayed.Node.Network), nodeacls.NodeID(relayed.Node.ID.String()), nodeacls.NodeID(peer.Node.ID.String())) {
  602. allowedIPs = append(allowedIPs, GetAllowedIPs(relayed, peer)...)
  603. }
  604. }
  605. return
  606. }
  607. // getIngressIPs returns the additional allowedips (ext client addresses) that need
  608. // to be included for an ingress gateway peer
  609. // TODO: add ExtraAllowedIPs
  610. func getIngressIPs(peer models.Client) []net.IPNet {
  611. var ingressIPs []net.IPNet
  612. extclients, err := GetNetworkExtClients(peer.Node.Network)
  613. if err != nil {
  614. return ingressIPs
  615. }
  616. for _, ec := range extclients {
  617. if ec.IngressGatewayID == peer.Node.ID.String() {
  618. if ec.Address != "" {
  619. var peeraddr = net.IPNet{
  620. IP: net.ParseIP(ec.Address),
  621. Mask: net.CIDRMask(32, 32),
  622. }
  623. if peeraddr.IP != nil && peeraddr.Mask != nil {
  624. ingressIPs = append(ingressIPs, peeraddr)
  625. }
  626. }
  627. if ec.Address6 != "" {
  628. var addr6 = net.IPNet{
  629. IP: net.ParseIP(ec.Address6),
  630. Mask: net.CIDRMask(128, 128),
  631. }
  632. if addr6.IP != nil && addr6.Mask != nil {
  633. ingressIPs = append(ingressIPs, addr6)
  634. }
  635. }
  636. }
  637. }
  638. return ingressIPs
  639. }