peers.go 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923
  1. package logic
  2. import (
  3. "context"
  4. "errors"
  5. "fmt"
  6. "net"
  7. "net/netip"
  8. "github.com/gravitl/netmaker/database"
  9. "github.com/gravitl/netmaker/logger"
  10. "github.com/gravitl/netmaker/logic/acls/nodeacls"
  11. "github.com/gravitl/netmaker/models"
  12. "github.com/gravitl/netmaker/servercfg"
  13. "golang.org/x/exp/slices"
  14. "golang.zx2c4.com/wireguard/wgctrl/wgtypes"
  15. )
  16. var (
  17. // PeerUpdateCtx context to send to host peer updates
  18. PeerUpdateCtx context.Context
  19. // PeerUpdateStop - the cancel for PeerUpdateCtx
  20. PeerUpdateStop context.CancelFunc
  21. )
  22. // GetProxyUpdateForHost - gets the proxy update for host
  23. func GetProxyUpdateForHost(ctx context.Context, host *models.Host) (models.ProxyManagerPayload, error) {
  24. proxyPayload := models.ProxyManagerPayload{
  25. Action: models.ProxyUpdate,
  26. }
  27. peerConfMap := make(map[string]models.PeerConf)
  28. var ingressStatus bool
  29. for _, nodeID := range host.Nodes {
  30. node, err := GetNodeByID(nodeID)
  31. if err != nil {
  32. continue
  33. }
  34. if !node.Connected || node.PendingDelete || node.Action == models.NODE_DELETE {
  35. continue
  36. }
  37. currentPeers, err := GetNetworkNodes(node.Network)
  38. if err != nil {
  39. continue
  40. }
  41. for _, peer := range currentPeers {
  42. if peer.ID == node.ID {
  43. //skip yourself
  44. continue
  45. }
  46. if !peer.Connected || peer.PendingDelete || peer.Action == models.NODE_DELETE {
  47. continue
  48. }
  49. peerHost, err := GetHost(peer.HostID.String())
  50. if err != nil {
  51. continue
  52. }
  53. var currPeerConf models.PeerConf
  54. var found bool
  55. if currPeerConf, found = peerConfMap[peerHost.PublicKey.String()]; !found {
  56. currPeerConf = models.PeerConf{
  57. Proxy: peerHost.ProxyEnabled,
  58. PublicListenPort: int32(GetPeerListenPort(peerHost)),
  59. ProxyListenPort: GetProxyListenPort(peerHost),
  60. NatType: peerHost.NatType,
  61. }
  62. }
  63. peerConfMap[peerHost.PublicKey.String()] = currPeerConf
  64. }
  65. if node.IsIngressGateway {
  66. ingressStatus = true
  67. _, peerConfMap, err = getExtPeersForProxy(&node, peerConfMap)
  68. if err == nil {
  69. } else if !database.IsEmptyRecord(err) {
  70. logger.Log(1, "error retrieving external clients:", err.Error())
  71. }
  72. }
  73. }
  74. proxyPayload.IsIngress = ingressStatus
  75. proxyPayload.PeerMap = peerConfMap
  76. return proxyPayload, nil
  77. }
  78. // ResetPeerUpdateContext - kills any current peer updates and resets the context
  79. func ResetPeerUpdateContext() {
  80. if PeerUpdateCtx != nil && PeerUpdateStop != nil {
  81. PeerUpdateStop() // tell any current peer updates to stop
  82. }
  83. PeerUpdateCtx, PeerUpdateStop = context.WithCancel(context.Background())
  84. }
  85. // GetPeerUpdateForHost - gets the consolidated peer update for the host from all networks
  86. func GetPeerUpdateForHost(ctx context.Context, network string, host *models.Host, deletedNode *models.Node, deletedClients []models.ExtClient) (models.HostPeerUpdate, error) {
  87. if host == nil {
  88. return models.HostPeerUpdate{}, errors.New("host is nil")
  89. }
  90. allNodes, err := GetAllNodes()
  91. if err != nil {
  92. return models.HostPeerUpdate{}, err
  93. }
  94. // track which nodes are deleted
  95. // after peer calculation, if peer not in list, add delete config of peer
  96. hostPeerUpdate := models.HostPeerUpdate{
  97. Host: *host,
  98. Server: servercfg.GetServer(),
  99. HostPeerIDs: make(models.HostPeerMap, 0),
  100. ServerVersion: servercfg.GetVersion(),
  101. ServerAddrs: []models.ServerAddr{},
  102. PeerIDs: make(models.PeerMap, 0),
  103. Peers: []wgtypes.PeerConfig{},
  104. NodePeers: []wgtypes.PeerConfig{},
  105. HostNetworkInfo: models.HostInfoMap{},
  106. }
  107. logger.Log(1, "peer update for host", host.ID.String())
  108. peerIndexMap := make(map[string]int)
  109. for _, nodeID := range host.Nodes {
  110. nodeID := nodeID
  111. node, err := GetNodeByID(nodeID)
  112. if err != nil {
  113. continue
  114. }
  115. if !node.Connected || node.PendingDelete || node.Action == models.NODE_DELETE {
  116. continue
  117. }
  118. currentPeers := GetNetworkNodesMemory(allNodes, node.Network)
  119. for _, peer := range currentPeers {
  120. select {
  121. case <-ctx.Done():
  122. logger.Log(2, "cancelled peer update for host", host.Name, host.ID.String())
  123. return models.HostPeerUpdate{}, fmt.Errorf("peer update cancelled")
  124. default:
  125. peer := peer
  126. if peer.ID.String() == node.ID.String() {
  127. logger.Log(2, "peer update, skipping self")
  128. //skip yourself
  129. continue
  130. }
  131. if peer.IsRelayed {
  132. // skip relayed peers; will be included in relay peer
  133. continue
  134. }
  135. var peerConfig wgtypes.PeerConfig
  136. peerHost, err := GetHost(peer.HostID.String())
  137. if err != nil {
  138. logger.Log(1, "no peer host", peer.HostID.String(), err.Error())
  139. return models.HostPeerUpdate{}, err
  140. }
  141. peerConfig.PublicKey = peerHost.PublicKey
  142. peerConfig.PersistentKeepaliveInterval = &peer.PersistentKeepalive
  143. peerConfig.ReplaceAllowedIPs = true
  144. uselocal := false
  145. if host.EndpointIP.String() == peerHost.EndpointIP.String() {
  146. // peer is on same network
  147. // set to localaddress
  148. uselocal = true
  149. if node.LocalAddress.IP == nil {
  150. // use public endpint
  151. uselocal = false
  152. }
  153. if node.LocalAddress.String() == peer.LocalAddress.String() {
  154. uselocal = false
  155. }
  156. }
  157. peerConfig.Endpoint = &net.UDPAddr{
  158. IP: peerHost.EndpointIP,
  159. Port: peerHost.ListenPort,
  160. }
  161. if uselocal {
  162. peerConfig.Endpoint.IP = peer.LocalAddress.IP
  163. }
  164. allowedips := GetAllowedIPs(&node, &peer, nil)
  165. if peer.IsIngressGateway {
  166. for _, entry := range peer.IngressGatewayRange {
  167. _, cidr, err := net.ParseCIDR(string(entry))
  168. if err == nil {
  169. allowedips = append(allowedips, *cidr)
  170. }
  171. }
  172. }
  173. if peer.IsEgressGateway {
  174. host, err := GetHost(peer.HostID.String())
  175. if err == nil {
  176. allowedips = append(allowedips, getEgressIPs(
  177. &models.Client{
  178. Host: *host,
  179. Node: peer,
  180. })...)
  181. }
  182. }
  183. if peer.Action != models.NODE_DELETE &&
  184. !peer.PendingDelete &&
  185. peer.Connected &&
  186. nodeacls.AreNodesAllowed(nodeacls.NetworkID(node.Network), nodeacls.NodeID(node.ID.String()), nodeacls.NodeID(peer.ID.String())) &&
  187. (deletedNode == nil || (deletedNode != nil && peer.ID.String() != deletedNode.ID.String())) {
  188. peerConfig.AllowedIPs = allowedips // only append allowed IPs if valid connection
  189. }
  190. peerProxyPort := GetProxyListenPort(peerHost)
  191. var nodePeer wgtypes.PeerConfig
  192. if _, ok := hostPeerUpdate.HostPeerIDs[peerHost.PublicKey.String()]; !ok {
  193. hostPeerUpdate.HostPeerIDs[peerHost.PublicKey.String()] = make(map[string]models.IDandAddr)
  194. hostPeerUpdate.Peers = append(hostPeerUpdate.Peers, peerConfig)
  195. peerIndexMap[peerHost.PublicKey.String()] = len(hostPeerUpdate.Peers) - 1
  196. hostPeerUpdate.HostPeerIDs[peerHost.PublicKey.String()][peer.ID.String()] = models.IDandAddr{
  197. ID: peer.ID.String(),
  198. Address: peer.PrimaryAddress(),
  199. Name: peerHost.Name,
  200. Network: peer.Network,
  201. ProxyListenPort: peerProxyPort,
  202. }
  203. hostPeerUpdate.HostNetworkInfo[peerHost.PublicKey.String()] = models.HostNetworkInfo{
  204. Interfaces: peerHost.Interfaces,
  205. ProxyListenPort: peerProxyPort,
  206. }
  207. nodePeer = peerConfig
  208. } else {
  209. peerAllowedIPs := hostPeerUpdate.Peers[peerIndexMap[peerHost.PublicKey.String()]].AllowedIPs
  210. peerAllowedIPs = append(peerAllowedIPs, allowedips...)
  211. hostPeerUpdate.Peers[peerIndexMap[peerHost.PublicKey.String()]].AllowedIPs = peerAllowedIPs
  212. hostPeerUpdate.HostPeerIDs[peerHost.PublicKey.String()][peer.ID.String()] = models.IDandAddr{
  213. ID: peer.ID.String(),
  214. Address: peer.PrimaryAddress(),
  215. Name: peerHost.Name,
  216. Network: peer.Network,
  217. ProxyListenPort: GetProxyListenPort(peerHost),
  218. }
  219. hostPeerUpdate.HostNetworkInfo[peerHost.PublicKey.String()] = models.HostNetworkInfo{
  220. Interfaces: peerHost.Interfaces,
  221. ProxyListenPort: peerProxyPort,
  222. }
  223. nodePeer = hostPeerUpdate.Peers[peerIndexMap[peerHost.PublicKey.String()]]
  224. }
  225. if node.Network == network { // add to peers map for metrics
  226. hostPeerUpdate.PeerIDs[peerHost.PublicKey.String()] = models.IDandAddr{
  227. ID: peer.ID.String(),
  228. Address: peer.PrimaryAddress(),
  229. Name: peerHost.Name,
  230. Network: peer.Network,
  231. ProxyListenPort: peerHost.ProxyListenPort,
  232. }
  233. hostPeerUpdate.NodePeers = append(hostPeerUpdate.NodePeers, nodePeer)
  234. }
  235. }
  236. }
  237. var extPeers []wgtypes.PeerConfig
  238. var extPeerIDAndAddrs []models.IDandAddr
  239. if node.IsIngressGateway {
  240. extPeers, extPeerIDAndAddrs, err = GetExtPeers(&node)
  241. if err == nil {
  242. hostPeerUpdate.Peers = append(hostPeerUpdate.Peers, extPeers...)
  243. for _, extPeerIdAndAddr := range extPeerIDAndAddrs {
  244. extPeerIdAndAddr := extPeerIdAndAddr
  245. hostPeerUpdate.HostPeerIDs[extPeerIdAndAddr.ID] = make(map[string]models.IDandAddr)
  246. hostPeerUpdate.HostPeerIDs[extPeerIdAndAddr.ID][extPeerIdAndAddr.ID] = models.IDandAddr{
  247. ID: extPeerIdAndAddr.ID,
  248. Address: extPeerIdAndAddr.Address,
  249. Name: extPeerIdAndAddr.Name,
  250. Network: node.Network,
  251. }
  252. if node.Network == network {
  253. hostPeerUpdate.PeerIDs[extPeerIdAndAddr.ID] = extPeerIdAndAddr
  254. hostPeerUpdate.NodePeers = append(hostPeerUpdate.NodePeers, extPeers...)
  255. }
  256. }
  257. } else if !database.IsEmptyRecord(err) {
  258. logger.Log(1, "error retrieving external clients:", err.Error())
  259. }
  260. }
  261. }
  262. // == post peer calculations ==
  263. // indicate removal if no allowed IPs were calculated
  264. for i := range hostPeerUpdate.Peers {
  265. peer := hostPeerUpdate.Peers[i]
  266. if len(peer.AllowedIPs) == 0 {
  267. peer.Remove = true
  268. }
  269. hostPeerUpdate.Peers[i] = peer
  270. }
  271. for i := range hostPeerUpdate.NodePeers {
  272. peer := hostPeerUpdate.NodePeers[i]
  273. if len(peer.AllowedIPs) == 0 {
  274. peer.Remove = true
  275. }
  276. hostPeerUpdate.NodePeers[i] = peer
  277. }
  278. if len(deletedClients) > 0 {
  279. for i := range deletedClients {
  280. deletedClient := deletedClients[i]
  281. key, err := wgtypes.ParseKey(deletedClient.PublicKey)
  282. if err == nil {
  283. hostPeerUpdate.Peers = append(hostPeerUpdate.Peers, wgtypes.PeerConfig{
  284. PublicKey: key,
  285. Remove: true,
  286. })
  287. }
  288. }
  289. }
  290. return hostPeerUpdate, nil
  291. }
  292. // GetFwUpdate - fetches the firewall update for the gateway nodes on the host
  293. func GetFwUpdate(host *models.Host) (models.FwUpdate, error) {
  294. fwUpdate := models.FwUpdate{
  295. IngressInfo: models.IngressInfo{
  296. ExtPeers: make(map[string]models.ExtClientInfo),
  297. },
  298. EgressInfo: make(map[string]models.EgressInfo),
  299. }
  300. allNodes, err := GetAllNodes()
  301. if err != nil {
  302. return fwUpdate, err
  303. }
  304. for _, nodeID := range host.Nodes {
  305. nodeID := nodeID
  306. node, err := GetNodeByID(nodeID)
  307. if err != nil {
  308. continue
  309. }
  310. if !node.Connected || node.PendingDelete || node.Action == models.NODE_DELETE {
  311. continue
  312. }
  313. currentPeers := GetNetworkNodesMemory(allNodes, node.Network)
  314. var nodePeerMap map[string]models.PeerRouteInfo
  315. if node.IsIngressGateway || node.IsEgressGateway {
  316. nodePeerMap = make(map[string]models.PeerRouteInfo)
  317. }
  318. for _, peer := range currentPeers {
  319. peer := peer
  320. if peer.ID.String() == node.ID.String() {
  321. logger.Log(2, "fw update, skipping self")
  322. //skip yourself
  323. continue
  324. }
  325. peerHost, err := GetHost(peer.HostID.String())
  326. if err != nil {
  327. logger.Log(1, "no peer host", peer.HostID.String(), err.Error())
  328. continue
  329. }
  330. if node.IsIngressGateway || node.IsEgressGateway {
  331. if peer.IsIngressGateway {
  332. _, extPeerIDAndAddrs, err := GetExtPeers(&peer)
  333. if err == nil {
  334. for _, extPeerIdAndAddr := range extPeerIDAndAddrs {
  335. extPeerIdAndAddr := extPeerIdAndAddr
  336. nodePeerMap[extPeerIdAndAddr.ID] = models.PeerRouteInfo{
  337. PeerAddr: net.IPNet{
  338. IP: net.ParseIP(extPeerIdAndAddr.Address),
  339. Mask: getCIDRMaskFromAddr(extPeerIdAndAddr.Address),
  340. },
  341. PeerKey: extPeerIdAndAddr.ID,
  342. Allow: true,
  343. ID: extPeerIdAndAddr.ID,
  344. }
  345. }
  346. }
  347. }
  348. if node.IsIngressGateway && peer.IsEgressGateway {
  349. fwUpdate.IngressInfo.EgressRanges = append(fwUpdate.IngressInfo.EgressRanges,
  350. peer.EgressGatewayRanges...)
  351. }
  352. nodePeerMap[peerHost.PublicKey.String()] = models.PeerRouteInfo{
  353. PeerAddr: net.IPNet{
  354. IP: net.ParseIP(peer.PrimaryAddress()),
  355. Mask: getCIDRMaskFromAddr(peer.PrimaryAddress()),
  356. },
  357. PeerKey: peerHost.PublicKey.String(),
  358. Allow: true,
  359. ID: peer.ID.String(),
  360. }
  361. }
  362. }
  363. var extPeerIDAndAddrs []models.IDandAddr
  364. if node.IsIngressGateway {
  365. fwUpdate.IsIngressGw = true
  366. _, extPeerIDAndAddrs, err = GetExtPeers(&node)
  367. if err == nil {
  368. for _, extPeerIdAndAddr := range extPeerIDAndAddrs {
  369. extPeerIdAndAddr := extPeerIdAndAddr
  370. nodePeerMap[extPeerIdAndAddr.ID] = models.PeerRouteInfo{
  371. PeerAddr: net.IPNet{
  372. IP: net.ParseIP(extPeerIdAndAddr.Address),
  373. Mask: getCIDRMaskFromAddr(extPeerIdAndAddr.Address),
  374. },
  375. PeerKey: extPeerIdAndAddr.ID,
  376. Allow: true,
  377. ID: extPeerIdAndAddr.ID,
  378. }
  379. }
  380. for _, extPeerIdAndAddr := range extPeerIDAndAddrs {
  381. extPeerIdAndAddr := extPeerIdAndAddr
  382. fwUpdate.IngressInfo.ExtPeers[extPeerIdAndAddr.ID] = models.ExtClientInfo{
  383. Masquerade: true,
  384. IngGwAddr: net.IPNet{
  385. IP: net.ParseIP(node.PrimaryAddress()),
  386. Mask: getCIDRMaskFromAddr(node.PrimaryAddress()),
  387. },
  388. Network: node.PrimaryNetworkRange(),
  389. ExtPeerAddr: net.IPNet{
  390. IP: net.ParseIP(extPeerIdAndAddr.Address),
  391. Mask: getCIDRMaskFromAddr(extPeerIdAndAddr.Address),
  392. },
  393. ExtPeerKey: extPeerIdAndAddr.ID,
  394. Peers: filterNodeMapForClientACLs(extPeerIdAndAddr.ID, node.Network, nodePeerMap),
  395. }
  396. }
  397. } else if !database.IsEmptyRecord(err) {
  398. logger.Log(1, "error retrieving external clients:", err.Error())
  399. }
  400. }
  401. if node.IsEgressGateway {
  402. fwUpdate.IsEgressGw = true
  403. fwUpdate.EgressInfo[node.ID.String()] = models.EgressInfo{
  404. EgressID: node.ID.String(),
  405. Network: node.PrimaryNetworkRange(),
  406. EgressGwAddr: net.IPNet{
  407. IP: net.ParseIP(node.PrimaryAddress()),
  408. Mask: getCIDRMaskFromAddr(node.PrimaryAddress()),
  409. },
  410. GwPeers: nodePeerMap,
  411. EgressGWCfg: node.EgressGatewayRequest,
  412. }
  413. }
  414. }
  415. return fwUpdate, nil
  416. }
  417. // GetPeerListenPort - given a host, retrieve it's appropriate listening port
  418. func GetPeerListenPort(host *models.Host) int {
  419. peerPort := host.ListenPort
  420. if host.ProxyEnabled && host.ProxyListenPort != 0 {
  421. peerPort = host.ProxyListenPort
  422. }
  423. if host.PublicListenPort != 0 {
  424. peerPort = host.PublicListenPort
  425. }
  426. return peerPort
  427. }
  428. // GetProxyListenPort - fetches the proxy listen port
  429. func GetProxyListenPort(host *models.Host) int {
  430. proxyPort := host.ProxyListenPort
  431. if host.PublicListenPort != 0 {
  432. proxyPort = host.PublicListenPort
  433. }
  434. return proxyPort
  435. }
  436. func GetExtPeers(node *models.Node) ([]wgtypes.PeerConfig, []models.IDandAddr, error) {
  437. var peers []wgtypes.PeerConfig
  438. var idsAndAddr []models.IDandAddr
  439. extPeers, err := GetNetworkExtClients(node.Network)
  440. if err != nil {
  441. return peers, idsAndAddr, err
  442. }
  443. host, err := GetHost(node.HostID.String())
  444. if err != nil {
  445. return peers, idsAndAddr, err
  446. }
  447. for _, extPeer := range extPeers {
  448. extPeer := extPeer
  449. pubkey, err := wgtypes.ParseKey(extPeer.PublicKey)
  450. if err != nil {
  451. logger.Log(1, "error parsing ext pub key:", err.Error())
  452. continue
  453. }
  454. if host.PublicKey.String() == extPeer.PublicKey ||
  455. extPeer.IngressGatewayID != node.ID.String() || !extPeer.Enabled {
  456. continue
  457. }
  458. var allowedips []net.IPNet
  459. var peer wgtypes.PeerConfig
  460. if extPeer.Address != "" {
  461. var peeraddr = net.IPNet{
  462. IP: net.ParseIP(extPeer.Address),
  463. Mask: net.CIDRMask(32, 32),
  464. }
  465. if peeraddr.IP != nil && peeraddr.Mask != nil {
  466. allowedips = append(allowedips, peeraddr)
  467. }
  468. }
  469. if extPeer.Address6 != "" {
  470. var addr6 = net.IPNet{
  471. IP: net.ParseIP(extPeer.Address6),
  472. Mask: net.CIDRMask(128, 128),
  473. }
  474. if addr6.IP != nil && addr6.Mask != nil {
  475. allowedips = append(allowedips, addr6)
  476. }
  477. }
  478. primaryAddr := extPeer.Address
  479. if primaryAddr == "" {
  480. primaryAddr = extPeer.Address6
  481. }
  482. peer = wgtypes.PeerConfig{
  483. PublicKey: pubkey,
  484. ReplaceAllowedIPs: true,
  485. AllowedIPs: allowedips,
  486. }
  487. peers = append(peers, peer)
  488. idsAndAddr = append(idsAndAddr, models.IDandAddr{
  489. ID: peer.PublicKey.String(),
  490. Name: extPeer.ClientID,
  491. Address: primaryAddr,
  492. })
  493. }
  494. return peers, idsAndAddr, nil
  495. }
  496. func getExtPeersForProxy(node *models.Node, proxyPeerConf map[string]models.PeerConf) ([]wgtypes.PeerConfig, map[string]models.PeerConf, error) {
  497. var peers []wgtypes.PeerConfig
  498. host, err := GetHost(node.HostID.String())
  499. if err != nil {
  500. logger.Log(0, "error retrieving host for node", node.ID.String(), err.Error())
  501. }
  502. extPeers, err := GetNetworkExtClients(node.Network)
  503. if err != nil {
  504. return peers, proxyPeerConf, err
  505. }
  506. for _, extPeer := range extPeers {
  507. pubkey, err := wgtypes.ParseKey(extPeer.PublicKey)
  508. if err != nil {
  509. logger.Log(1, "error parsing ext pub key:", err.Error())
  510. continue
  511. }
  512. if host.PublicKey.String() == extPeer.PublicKey ||
  513. extPeer.IngressGatewayID != node.ID.String() || !extPeer.Enabled {
  514. continue
  515. }
  516. var allowedips []net.IPNet
  517. var peer wgtypes.PeerConfig
  518. if extPeer.Address != "" {
  519. var peeraddr = net.IPNet{
  520. IP: net.ParseIP(extPeer.Address),
  521. Mask: net.CIDRMask(32, 32),
  522. }
  523. if peeraddr.IP != nil && peeraddr.Mask != nil {
  524. allowedips = append(allowedips, peeraddr)
  525. }
  526. }
  527. if extPeer.Address6 != "" {
  528. var addr6 = net.IPNet{
  529. IP: net.ParseIP(extPeer.Address6),
  530. Mask: net.CIDRMask(128, 128),
  531. }
  532. if addr6.IP != nil && addr6.Mask != nil {
  533. allowedips = append(allowedips, addr6)
  534. }
  535. }
  536. peer = wgtypes.PeerConfig{
  537. PublicKey: pubkey,
  538. ReplaceAllowedIPs: true,
  539. AllowedIPs: allowedips,
  540. }
  541. extConf := models.PeerConf{
  542. IsExtClient: true,
  543. Address: net.ParseIP(extPeer.Address),
  544. }
  545. proxyPeerConf[peer.PublicKey.String()] = extConf
  546. peers = append(peers, peer)
  547. }
  548. return peers, proxyPeerConf, nil
  549. }
  550. // GetAllowedIPs - calculates the wireguard allowedip field for a peer of a node based on the peer and node settings
  551. func GetAllowedIPs(node, peer *models.Node, metrics *models.Metrics) []net.IPNet {
  552. var allowedips []net.IPNet
  553. allowedips = getNodeAllowedIPs(peer, node)
  554. // handle ingress gateway peers
  555. if peer.IsIngressGateway {
  556. extPeers, _, err := GetExtPeers(peer)
  557. if err != nil {
  558. logger.Log(2, "could not retrieve ext peers for ", peer.ID.String(), err.Error())
  559. }
  560. for _, extPeer := range extPeers {
  561. allowedips = append(allowedips, extPeer.AllowedIPs...)
  562. }
  563. // if node is a failover node, add allowed ips from nodes it is handling
  564. if metrics != nil && peer.Failover && metrics.FailoverPeers != nil {
  565. // traverse through nodes that need handling
  566. logger.Log(3, "peer", peer.ID.String(), "was found to be failover for", node.ID.String(), "checking failover peers...")
  567. for k := range metrics.FailoverPeers {
  568. // if FailoverNode is me for this node, add allowedips
  569. if metrics.FailoverPeers[k] == peer.ID.String() {
  570. // get original node so we can traverse the allowed ips
  571. nodeToFailover, err := GetNodeByID(k)
  572. if err == nil {
  573. failoverNodeMetrics, err := GetMetrics(nodeToFailover.ID.String())
  574. if err == nil && failoverNodeMetrics != nil {
  575. if len(failoverNodeMetrics.NodeName) > 0 {
  576. allowedips = append(allowedips, getNodeAllowedIPs(&nodeToFailover, peer)...)
  577. logger.Log(0, "failing over node", nodeToFailover.ID.String(), nodeToFailover.PrimaryAddress(), "to failover node", peer.ID.String())
  578. }
  579. }
  580. }
  581. }
  582. }
  583. }
  584. }
  585. return allowedips
  586. }
  587. // getEgressIPs - gets the egress IPs for a client
  588. func getEgressIPs(client *models.Client) []net.IPNet {
  589. //check for internet gateway
  590. internetGateway := false
  591. if slices.Contains(client.Node.EgressGatewayRanges, "0.0.0.0/0") || slices.Contains(client.Node.EgressGatewayRanges, "::/0") {
  592. internetGateway = true
  593. }
  594. allowedips := []net.IPNet{}
  595. for _, iprange := range client.Node.EgressGatewayRanges { // go through each cidr for egress gateway
  596. ip, cidr, err := net.ParseCIDR(iprange) // confirming it's valid cidr
  597. if err != nil {
  598. logger.Log(1, "could not parse gateway IP range. Not adding ", iprange)
  599. continue // if can't parse CIDR
  600. }
  601. cidr.IP = ip
  602. // getting the public ip of node
  603. if cidr.Contains(client.Host.EndpointIP) && !internetGateway { // ensuring egress gateway range does not contain endpoint of node
  604. logger.Log(2, "egress IP range of ", iprange, " overlaps with ", client.Host.EndpointIP.String(), ", omitting")
  605. continue // skip adding egress range if overlaps with node's ip
  606. }
  607. // TODO: Could put in a lot of great logic to avoid conflicts / bad routes
  608. if cidr.Contains(client.Node.LocalAddress.IP) && !internetGateway { // ensuring egress gateway range does not contain public ip of node
  609. logger.Log(2, "egress IP range of ", iprange, " overlaps with ", client.Node.LocalAddress.String(), ", omitting")
  610. continue // skip adding egress range if overlaps with node's local ip
  611. }
  612. if err != nil {
  613. logger.Log(1, "error encountered when setting egress range", err.Error())
  614. } else {
  615. allowedips = append(allowedips, *cidr)
  616. }
  617. }
  618. return allowedips
  619. }
  620. func getNodeAllowedIPs(peer, node *models.Node) []net.IPNet {
  621. var allowedips = []net.IPNet{}
  622. if peer.Address.IP != nil {
  623. allowed := net.IPNet{
  624. IP: peer.Address.IP,
  625. Mask: net.CIDRMask(32, 32),
  626. }
  627. allowedips = append(allowedips, allowed)
  628. }
  629. if peer.Address6.IP != nil {
  630. allowed := net.IPNet{
  631. IP: peer.Address6.IP,
  632. Mask: net.CIDRMask(128, 128),
  633. }
  634. allowedips = append(allowedips, allowed)
  635. }
  636. // handle egress gateway peers
  637. if peer.IsEgressGateway {
  638. //hasGateway = true
  639. host, err := GetHost(peer.HostID.String())
  640. if err == nil {
  641. egressIPs := getEgressIPs(
  642. &models.Client{
  643. Host: *host,
  644. Node: *peer,
  645. })
  646. allowedips = append(allowedips, egressIPs...)
  647. }
  648. }
  649. if peer.IsRelay {
  650. for _, relayed := range peer.RelayedNodes {
  651. allowed := getRelayedAddresses(relayed)
  652. allowedips = append(allowedips, allowed...)
  653. }
  654. }
  655. return allowedips
  656. }
  657. func getCIDRMaskFromAddr(addr string) net.IPMask {
  658. cidr := net.CIDRMask(32, 32)
  659. ipAddr, err := netip.ParseAddr(addr)
  660. if err != nil {
  661. return cidr
  662. }
  663. if ipAddr.Is6() {
  664. cidr = net.CIDRMask(128, 128)
  665. }
  666. return cidr
  667. }
  668. // accounts for ext client ACLs
  669. func filterNodeMapForClientACLs(publicKey, network string, nodePeerMap map[string]models.PeerRouteInfo) map[string]models.PeerRouteInfo {
  670. if !isEE {
  671. return nodePeerMap
  672. }
  673. if nodePeerMap == nil {
  674. return map[string]models.PeerRouteInfo{}
  675. }
  676. if len(publicKey) == 0 || len(network) == 0 {
  677. return nodePeerMap
  678. }
  679. client, err := GetExtClientByPubKey(publicKey, network)
  680. if err != nil {
  681. return nodePeerMap
  682. }
  683. for k := range nodePeerMap {
  684. currNodePeer := nodePeerMap[k]
  685. if _, ok := client.ACLs[currNodePeer.ID]; ok {
  686. delete(nodePeerMap, k)
  687. }
  688. }
  689. return nodePeerMap
  690. }
  691. func GetPeerUpdate(host *models.Host) []wgtypes.PeerConfig {
  692. peerUpdate := []wgtypes.PeerConfig{}
  693. for _, nodeStr := range host.Nodes {
  694. node, err := GetNodeByID(nodeStr)
  695. if err != nil {
  696. continue
  697. }
  698. client := models.Client{Host: *host, Node: node}
  699. peers, err := GetNetworkClients(node.Network)
  700. if err != nil {
  701. continue
  702. }
  703. if node.IsRelayed {
  704. peerUpdate = append(peerUpdate, peerUpdateForRelayed(&client, peers)...)
  705. continue
  706. }
  707. if node.IsRelay {
  708. peerUpdate = append(peerUpdate, peerUpdateForRelay(&client, peers)...)
  709. continue
  710. }
  711. for _, peer := range peers {
  712. if peer.Host.ID == client.Host.ID {
  713. continue
  714. }
  715. // if peer is relayed by some other node, remove it from the peer list, it
  716. // will be added to allowedips of relay peer
  717. if peer.Node.IsRelayed {
  718. update := wgtypes.PeerConfig{
  719. PublicKey: peer.Host.PublicKey,
  720. Remove: true,
  721. }
  722. peerUpdate = append(peerUpdate, update)
  723. continue
  724. }
  725. update := wgtypes.PeerConfig{
  726. PublicKey: peer.Host.PublicKey,
  727. ReplaceAllowedIPs: true,
  728. Endpoint: &net.UDPAddr{
  729. IP: peer.Host.EndpointIP,
  730. Port: peer.Host.ListenPort,
  731. },
  732. PersistentKeepaliveInterval: &peer.Node.PersistentKeepalive,
  733. }
  734. // if peer is a relay that relays us, don't do anything
  735. if peer.Node.IsRelay && client.Node.RelayedBy == peer.Node.ID.String() {
  736. continue
  737. } else {
  738. update.AllowedIPs = append(update.AllowedIPs, getRelayAllowedIPs(&peer)...)
  739. }
  740. //normal peer
  741. update.AllowedIPs = append(update.AllowedIPs, AddAllowedIPs(&peer)...)
  742. peerUpdate = append(peerUpdate, update)
  743. }
  744. }
  745. return peerUpdate
  746. }
  747. func AddAllowedIPs(peer *models.Client) []net.IPNet {
  748. allowedIPs := []net.IPNet{}
  749. if peer.Node.Address.IP != nil {
  750. peer.Node.Address.Mask = net.CIDRMask(32, 32)
  751. allowedIPs = append(allowedIPs, peer.Node.Address)
  752. }
  753. if peer.Node.Address6.IP != nil {
  754. peer.Node.Address6.Mask = net.CIDRMask(128, 128)
  755. allowedIPs = append(allowedIPs, peer.Node.Address6)
  756. }
  757. if peer.Node.IsEgressGateway {
  758. allowedIPs = append(allowedIPs, getEgressIPs(peer)...)
  759. }
  760. if peer.Node.IsIngressGateway {
  761. allowedIPs = append(allowedIPs, getIngressIPs(peer)...)
  762. }
  763. return allowedIPs
  764. }
  765. // getRelayAllowedIPs returns the list of allowedips for a peer that is a relay
  766. func getRelayAllowedIPs(peer *models.Client) []net.IPNet {
  767. var relayIPs []net.IPNet
  768. if !peer.Node.IsRelay {
  769. logger.Log(0, "getRelayAllowedIPs called for a non-relay node", peer.Host.Name)
  770. return relayIPs
  771. }
  772. //if !client.Node.IsRelayed || client.Node.RelayedBy != peer.Node.ID.String() {
  773. //logger.Log(0, "getRelayAllowedIPs called for non-relayed node", client.Host.Name, peer.Host.Name)
  774. //return relayIPs
  775. //}
  776. for _, relayed := range peer.Node.RelayedNodes {
  777. relayedNode, err := GetNodeByID(relayed)
  778. if err != nil {
  779. logger.Log(0, "retrieve relayed node", err.Error())
  780. continue
  781. }
  782. if relayedNode.Address.IP != nil {
  783. relayedNode.Address.Mask = net.CIDRMask(32, 32)
  784. relayIPs = append(relayIPs, relayedNode.Address)
  785. }
  786. if relayedNode.Address6.IP != nil {
  787. relayedNode.Address.Mask = net.CIDRMask(128, 128)
  788. relayIPs = append(relayIPs, relayedNode.Address6)
  789. }
  790. host, err := GetHost(relayedNode.HostID.String())
  791. if err == nil {
  792. if relayedNode.IsRelay {
  793. relayIPs = append(relayIPs, getRelayAllowedIPs(
  794. &models.Client{
  795. Host: *host,
  796. Node: relayedNode,
  797. })...)
  798. }
  799. if relayedNode.IsEgressGateway {
  800. relayIPs = append(relayIPs, getEgressIPs(
  801. &models.Client{
  802. Host: *host,
  803. Node: relayedNode,
  804. })...)
  805. }
  806. if relayedNode.IsIngressGateway {
  807. relayIPs = append(relayIPs, getIngressIPs(
  808. &models.Client{
  809. Host: *host,
  810. Node: relayedNode,
  811. })...)
  812. }
  813. }
  814. }
  815. return relayIPs
  816. }
  817. // getIngressIPs returns the additional allowedips (ext client addresses) that need
  818. // to be included for an ingress gateway peer
  819. // TODO: add ExtraAllowedIPs
  820. func getIngressIPs(peer *models.Client) []net.IPNet {
  821. var ingressIPs []net.IPNet
  822. extclients, err := GetNetworkExtClients(peer.Node.Network)
  823. if err != nil {
  824. return ingressIPs
  825. }
  826. for _, ec := range extclients {
  827. if ec.IngressGatewayID == peer.Node.ID.String() {
  828. if ec.Address != "" {
  829. ip, cidr, err := net.ParseCIDR(ec.Address)
  830. if err != nil {
  831. continue
  832. }
  833. cidr.IP = ip
  834. ingressIPs = append(ingressIPs, *cidr)
  835. }
  836. if ec.Address6 != "" {
  837. ip, cidr, err := net.ParseCIDR(ec.Address6)
  838. if err != nil {
  839. continue
  840. }
  841. cidr.IP = ip
  842. ingressIPs = append(ingressIPs, *cidr)
  843. }
  844. }
  845. }
  846. return ingressIPs
  847. }
  848. // GetPeerUpdateForRelay - returns the peer update for a relay node
  849. func GetPeerUpdateForRelay(client *models.Client, peers []models.Client) []wgtypes.PeerConfig {
  850. peerConfig := []wgtypes.PeerConfig{}
  851. if !client.Node.IsRelay {
  852. return []wgtypes.PeerConfig{}
  853. }
  854. for _, peer := range peers {
  855. if peer.Host.ID == client.Host.ID {
  856. continue
  857. }
  858. update := wgtypes.PeerConfig{
  859. PublicKey: peer.Host.PublicKey,
  860. ReplaceAllowedIPs: true,
  861. Remove: false,
  862. Endpoint: &net.UDPAddr{
  863. IP: peer.Host.EndpointIP,
  864. Port: peer.Host.ListenPort,
  865. },
  866. PersistentKeepaliveInterval: &peer.Node.PersistentKeepalive,
  867. }
  868. update.AllowedIPs = append(update.AllowedIPs, AddAllowedIPs(&peer)...)
  869. peerConfig = append(peerConfig, update)
  870. }
  871. return peerConfig
  872. }