peers.go 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774
  1. package logic
  2. import (
  3. "errors"
  4. "fmt"
  5. "log"
  6. "net"
  7. "net/netip"
  8. "time"
  9. "github.com/gravitl/netmaker/database"
  10. "github.com/gravitl/netmaker/logger"
  11. "github.com/gravitl/netmaker/logic/acls/nodeacls"
  12. "github.com/gravitl/netmaker/models"
  13. "github.com/gravitl/netmaker/servercfg"
  14. "golang.org/x/exp/slices"
  15. "golang.zx2c4.com/wireguard/wgctrl/wgtypes"
  16. )
  17. // GetPeersforProxy calculates the peers for a proxy
  18. // TODO ==========================
  19. // TODO ==========================
  20. // TODO ==========================
  21. // TODO ==========================
  22. // TODO ==========================
  23. // revisit this logic with new host/node models.
  24. func GetPeersForProxy(node *models.Node, onlyPeers bool) (models.ProxyManagerPayload, error) {
  25. proxyPayload := models.ProxyManagerPayload{}
  26. var peers []wgtypes.PeerConfig
  27. peerConfMap := make(map[string]models.PeerConf)
  28. var err error
  29. currentPeers, err := GetNetworkNodes(node.Network)
  30. if err != nil {
  31. return proxyPayload, err
  32. }
  33. if !onlyPeers {
  34. if node.IsRelayed {
  35. relayNode := FindRelay(node)
  36. relayHost, err := GetHost(relayNode.HostID.String())
  37. if err != nil {
  38. return proxyPayload, err
  39. }
  40. if relayNode != nil {
  41. host, err := GetHost(relayNode.HostID.String())
  42. if err != nil {
  43. logger.Log(0, "error retrieving host for relay node", relayNode.HostID.String(), err.Error())
  44. }
  45. relayEndpoint, err := net.ResolveUDPAddr("udp", fmt.Sprintf("%s:%d", relayHost.EndpointIP, host.ListenPort))
  46. if err != nil {
  47. logger.Log(1, "failed to resolve relay node endpoint: ", err.Error())
  48. }
  49. proxyPayload.IsRelayed = true
  50. proxyPayload.RelayedTo = relayEndpoint
  51. } else {
  52. logger.Log(0, "couldn't find relay node for: ", node.ID.String())
  53. }
  54. }
  55. if node.IsRelay {
  56. host, err := GetHost(node.HostID.String())
  57. if err != nil {
  58. logger.Log(0, "error retrieving host for relay node", node.ID.String(), err.Error())
  59. }
  60. relayedNodes, err := GetRelayedNodes(node)
  61. if err != nil {
  62. logger.Log(1, "failed to relayed nodes: ", node.ID.String(), err.Error())
  63. proxyPayload.IsRelay = false
  64. } else {
  65. relayPeersMap := make(map[string]models.RelayedConf)
  66. for _, relayedNode := range relayedNodes {
  67. relayedNode := relayedNode
  68. payload, err := GetPeersForProxy(&relayedNode, true)
  69. if err == nil {
  70. relayedHost, err := GetHost(relayedNode.HostID.String())
  71. if err != nil {
  72. logger.Log(0, "error retrieving host for relayNode", relayedNode.ID.String(), err.Error())
  73. }
  74. relayedEndpoint, udpErr := net.ResolveUDPAddr("udp", fmt.Sprintf("%s:%d", relayedHost.EndpointIP, host.ListenPort))
  75. if udpErr == nil {
  76. relayPeersMap[host.PublicKey.String()] = models.RelayedConf{
  77. RelayedPeerEndpoint: relayedEndpoint,
  78. RelayedPeerPubKey: relayedHost.PublicKey.String(),
  79. Peers: payload.Peers,
  80. }
  81. }
  82. }
  83. }
  84. proxyPayload.IsRelay = true
  85. proxyPayload.RelayedPeerConf = relayPeersMap
  86. }
  87. }
  88. }
  89. for _, peer := range currentPeers {
  90. if peer.ID == node.ID {
  91. //skip yourself
  92. continue
  93. }
  94. host, err := GetHost(peer.HostID.String())
  95. if err != nil {
  96. continue
  97. }
  98. proxyStatus := host.ProxyEnabled
  99. listenPort := host.ListenPort
  100. if proxyStatus {
  101. listenPort = host.ProxyListenPort
  102. if listenPort == 0 {
  103. listenPort = models.NmProxyPort
  104. }
  105. } else if listenPort == 0 {
  106. listenPort = host.ListenPort
  107. }
  108. endpoint, err := net.ResolveUDPAddr("udp", fmt.Sprintf("%s:%d", host.EndpointIP, listenPort))
  109. if err != nil {
  110. logger.Log(1, "failed to resolve udp addr for node: ", peer.ID.String(), host.EndpointIP.String(), err.Error())
  111. continue
  112. }
  113. allowedips := GetAllowedIPs(node, &peer, nil)
  114. var keepalive time.Duration
  115. if node.PersistentKeepalive != 0 {
  116. // set_keepalive
  117. keepalive = node.PersistentKeepalive
  118. }
  119. peers = append(peers, wgtypes.PeerConfig{
  120. PublicKey: host.PublicKey,
  121. Endpoint: endpoint,
  122. AllowedIPs: allowedips,
  123. PersistentKeepaliveInterval: &keepalive,
  124. ReplaceAllowedIPs: true,
  125. })
  126. peerConfMap[host.PublicKey.String()] = models.PeerConf{
  127. Address: net.ParseIP(peer.PrimaryAddress()),
  128. Proxy: proxyStatus,
  129. PublicListenPort: int32(listenPort),
  130. }
  131. if !onlyPeers && peer.IsRelayed {
  132. relayNode := FindRelay(&peer)
  133. if relayNode != nil {
  134. relayHost, err := GetHost(relayNode.HostID.String())
  135. if err != nil {
  136. logger.Log(0, "error retrieving host for relayNode", relayNode.ID.String(), err.Error())
  137. continue
  138. }
  139. relayTo, err := net.ResolveUDPAddr("udp", fmt.Sprintf("%s:%d", relayHost.EndpointIP, relayHost.ListenPort))
  140. if err == nil {
  141. peerConfMap[host.PublicKey.String()] = models.PeerConf{
  142. IsRelayed: true,
  143. RelayedTo: relayTo,
  144. Address: net.ParseIP(peer.PrimaryAddress()),
  145. Proxy: proxyStatus,
  146. PublicListenPort: int32(listenPort),
  147. }
  148. }
  149. }
  150. }
  151. }
  152. if node.IsIngressGateway {
  153. var extPeers []wgtypes.PeerConfig
  154. extPeers, peerConfMap, err = getExtPeersForProxy(node, peerConfMap)
  155. if err == nil {
  156. peers = append(peers, extPeers...)
  157. } else if !database.IsEmptyRecord(err) {
  158. logger.Log(1, "error retrieving external clients:", err.Error())
  159. }
  160. }
  161. proxyPayload.IsIngress = node.IsIngressGateway
  162. addr := node.Address
  163. if addr.String() == "" {
  164. addr = node.Address6
  165. }
  166. proxyPayload.Peers = peers
  167. proxyPayload.PeerMap = peerConfMap
  168. //proxyPayload.Network = node.Network
  169. //proxyPayload.InterfaceName = node.Interface
  170. //hardcode or read from host ??
  171. proxyPayload.InterfaceName = models.WIREGUARD_INTERFACE
  172. return proxyPayload, nil
  173. }
  174. // GetProxyUpdateForHost - gets the proxy update for host
  175. func GetProxyUpdateForHost(host *models.Host) (models.ProxyManagerPayload, error) {
  176. proxyPayload := models.ProxyManagerPayload{
  177. Action: models.ProxyUpdate,
  178. }
  179. peerConfMap := make(map[string]models.PeerConf)
  180. if host.IsRelayed {
  181. relayHost, err := GetHost(host.RelayedBy)
  182. if err == nil {
  183. relayEndpoint, err := net.ResolveUDPAddr("udp", fmt.Sprintf("%s:%d", relayHost.EndpointIP, GetPeerListenPort(relayHost)))
  184. if err != nil {
  185. logger.Log(1, "failed to resolve relay node endpoint: ", err.Error())
  186. }
  187. proxyPayload.IsRelayed = true
  188. proxyPayload.RelayedTo = relayEndpoint
  189. } else {
  190. logger.Log(0, "couldn't find relay host for: ", host.ID.String())
  191. }
  192. }
  193. if host.IsRelay {
  194. relayedHosts := GetRelayedHosts(host)
  195. relayPeersMap := make(map[string]models.RelayedConf)
  196. for _, relayedHost := range relayedHosts {
  197. relayedHost := relayedHost
  198. payload, err := GetPeerUpdateForHost("", &relayedHost)
  199. if err == nil {
  200. relayedEndpoint, udpErr := net.ResolveUDPAddr("udp", fmt.Sprintf("%s:%d", relayedHost.EndpointIP, GetPeerListenPort(&relayedHost)))
  201. if udpErr == nil {
  202. relayPeersMap[relayedHost.PublicKey.String()] = models.RelayedConf{
  203. RelayedPeerEndpoint: relayedEndpoint,
  204. RelayedPeerPubKey: relayedHost.PublicKey.String(),
  205. Peers: payload.Peers,
  206. }
  207. }
  208. }
  209. }
  210. proxyPayload.IsRelay = true
  211. proxyPayload.RelayedPeerConf = relayPeersMap
  212. }
  213. var ingressStatus bool
  214. for _, nodeID := range host.Nodes {
  215. node, err := GetNodeByID(nodeID)
  216. if err != nil {
  217. continue
  218. }
  219. currentPeers, err := GetNetworkNodes(node.Network)
  220. if err != nil {
  221. continue
  222. }
  223. for _, peer := range currentPeers {
  224. if peer.ID == node.ID {
  225. //skip yourself
  226. continue
  227. }
  228. peerHost, err := GetHost(peer.HostID.String())
  229. if err != nil {
  230. continue
  231. }
  232. var currPeerConf models.PeerConf
  233. var found bool
  234. if currPeerConf, found = peerConfMap[peerHost.PublicKey.String()]; !found {
  235. currPeerConf = models.PeerConf{
  236. Proxy: peerHost.ProxyEnabled,
  237. PublicListenPort: int32(GetPeerListenPort(peerHost)),
  238. }
  239. }
  240. if peerHost.IsRelayed && peerHost.RelayedBy != host.ID.String() {
  241. relayHost, err := GetHost(peerHost.RelayedBy)
  242. if err == nil {
  243. relayTo, err := net.ResolveUDPAddr("udp", fmt.Sprintf("%s:%d", relayHost.EndpointIP, GetPeerListenPort(relayHost)))
  244. if err == nil {
  245. currPeerConf.IsRelayed = true
  246. currPeerConf.RelayedTo = relayTo
  247. }
  248. }
  249. }
  250. peerConfMap[peerHost.PublicKey.String()] = currPeerConf
  251. }
  252. if node.IsIngressGateway {
  253. ingressStatus = true
  254. _, peerConfMap, err = getExtPeersForProxy(&node, peerConfMap)
  255. if err == nil {
  256. } else if !database.IsEmptyRecord(err) {
  257. logger.Log(1, "error retrieving external clients:", err.Error())
  258. }
  259. }
  260. }
  261. proxyPayload.IsIngress = ingressStatus
  262. proxyPayload.PeerMap = peerConfMap
  263. return proxyPayload, nil
  264. }
  265. // GetPeerUpdateForHost - gets the consolidated peer update for the host from all networks
  266. func GetPeerUpdateForHost(network string, host *models.Host) (models.HostPeerUpdate, error) {
  267. if host == nil {
  268. return models.HostPeerUpdate{}, errors.New("host is nil")
  269. }
  270. hostPeerUpdate := models.HostPeerUpdate{
  271. Host: *host,
  272. Server: servercfg.GetServer(),
  273. HostPeerIDs: make(models.HostPeerMap, 0),
  274. ServerVersion: servercfg.GetVersion(),
  275. ServerAddrs: []models.ServerAddr{},
  276. IngressInfo: models.IngressInfo{
  277. ExtPeers: make(map[string]models.ExtClientInfo),
  278. },
  279. EgressInfo: make(map[string]models.EgressInfo),
  280. PeerIDs: make(models.PeerMap, 0),
  281. Peers: []wgtypes.PeerConfig{},
  282. NodePeers: []wgtypes.PeerConfig{},
  283. }
  284. logger.Log(1, "peer update for host ", host.ID.String())
  285. peerIndexMap := make(map[string]int)
  286. for _, nodeID := range host.Nodes {
  287. node, err := GetNodeByID(nodeID)
  288. if err != nil {
  289. continue
  290. }
  291. if !node.Connected || node.Action == models.NODE_DELETE || node.PendingDelete {
  292. continue
  293. }
  294. currentPeers, err := GetNetworkNodes(node.Network)
  295. if err != nil {
  296. log.Println("no network nodes")
  297. return models.HostPeerUpdate{}, err
  298. }
  299. var nodePeerMap map[string]models.PeerRouteInfo
  300. if node.IsIngressGateway || node.IsEgressGateway {
  301. nodePeerMap = make(map[string]models.PeerRouteInfo)
  302. }
  303. for _, peer := range currentPeers {
  304. if peer.ID == node.ID {
  305. logger.Log(2, "peer update, skipping self")
  306. //skip yourself
  307. continue
  308. }
  309. var peerConfig wgtypes.PeerConfig
  310. peerHost, err := GetHost(peer.HostID.String())
  311. if err != nil {
  312. logger.Log(1, "no peer host", peer.HostID.String(), err.Error())
  313. return models.HostPeerUpdate{}, err
  314. }
  315. if !peer.Connected || peer.Action == models.NODE_DELETE || peer.PendingDelete {
  316. logger.Log(2, "peer update, skipping unconnected node", peer.ID.String())
  317. //skip unconnected nodes
  318. continue
  319. }
  320. if !nodeacls.AreNodesAllowed(nodeacls.NetworkID(node.Network), nodeacls.NodeID(node.ID.String()), nodeacls.NodeID(peer.ID.String())) {
  321. log.Println("peer update, skipping node for acl")
  322. //skip if not permitted by acl
  323. continue
  324. }
  325. peerConfig.PublicKey = peerHost.PublicKey
  326. peerConfig.PersistentKeepaliveInterval = &peer.PersistentKeepalive
  327. peerConfig.ReplaceAllowedIPs = true
  328. uselocal := false
  329. if host.EndpointIP.String() == peerHost.EndpointIP.String() {
  330. //peer is on same network
  331. // set to localaddress
  332. uselocal = true
  333. if node.LocalAddress.IP == nil {
  334. // use public endpint
  335. uselocal = false
  336. }
  337. if node.LocalAddress.String() == peer.LocalAddress.String() {
  338. uselocal = false
  339. }
  340. }
  341. peerConfig.Endpoint = &net.UDPAddr{
  342. IP: peerHost.EndpointIP,
  343. Port: GetPeerListenPort(peerHost),
  344. }
  345. if uselocal {
  346. peerConfig.Endpoint.IP = peer.LocalAddress.IP
  347. }
  348. allowedips := GetAllowedIPs(&node, &peer, nil)
  349. if peer.IsIngressGateway {
  350. for _, entry := range peer.IngressGatewayRange {
  351. _, cidr, err := net.ParseCIDR(string(entry))
  352. if err == nil {
  353. allowedips = append(allowedips, *cidr)
  354. }
  355. }
  356. }
  357. if peer.IsEgressGateway {
  358. allowedips = append(allowedips, getEgressIPs(&node, &peer)...)
  359. }
  360. peerConfig.AllowedIPs = allowedips
  361. if node.IsIngressGateway || node.IsEgressGateway {
  362. if peer.IsIngressGateway {
  363. _, extPeerIDAndAddrs, err := getExtPeers(&peer)
  364. if err == nil {
  365. for _, extPeerIdAndAddr := range extPeerIDAndAddrs {
  366. nodePeerMap[extPeerIdAndAddr.ID] = models.PeerRouteInfo{
  367. PeerAddr: net.IPNet{
  368. IP: net.ParseIP(extPeerIdAndAddr.Address),
  369. Mask: getCIDRMaskFromAddr(extPeerIdAndAddr.Address),
  370. },
  371. PeerKey: extPeerIdAndAddr.ID,
  372. Allow: true,
  373. }
  374. }
  375. }
  376. }
  377. if node.IsIngressGateway && peer.IsEgressGateway {
  378. hostPeerUpdate.IngressInfo.EgressRanges = append(hostPeerUpdate.IngressInfo.EgressRanges,
  379. peer.EgressGatewayRanges...)
  380. }
  381. nodePeerMap[peerHost.PublicKey.String()] = models.PeerRouteInfo{
  382. PeerAddr: net.IPNet{
  383. IP: net.ParseIP(peer.PrimaryAddress()),
  384. Mask: getCIDRMaskFromAddr(peer.PrimaryAddress()),
  385. },
  386. PeerKey: peerHost.PublicKey.String(),
  387. Allow: true,
  388. }
  389. }
  390. var nodePeer wgtypes.PeerConfig
  391. if _, ok := hostPeerUpdate.HostPeerIDs[peerHost.PublicKey.String()]; !ok {
  392. hostPeerUpdate.HostPeerIDs[peerHost.PublicKey.String()] = make(map[string]models.IDandAddr)
  393. hostPeerUpdate.Peers = append(hostPeerUpdate.Peers, peerConfig)
  394. peerIndexMap[peerHost.PublicKey.String()] = len(hostPeerUpdate.Peers) - 1
  395. hostPeerUpdate.HostPeerIDs[peerHost.PublicKey.String()][peer.ID.String()] = models.IDandAddr{
  396. ID: peer.ID.String(),
  397. Address: peer.PrimaryAddress(),
  398. Name: peerHost.Name,
  399. Network: peer.Network,
  400. }
  401. nodePeer = peerConfig
  402. } else {
  403. peerAllowedIPs := hostPeerUpdate.Peers[peerIndexMap[peerHost.PublicKey.String()]].AllowedIPs
  404. peerAllowedIPs = append(peerAllowedIPs, allowedips...)
  405. hostPeerUpdate.Peers[peerIndexMap[peerHost.PublicKey.String()]].AllowedIPs = peerAllowedIPs
  406. hostPeerUpdate.HostPeerIDs[peerHost.PublicKey.String()][peer.ID.String()] = models.IDandAddr{
  407. ID: peer.ID.String(),
  408. Address: peer.PrimaryAddress(),
  409. Name: peerHost.Name,
  410. Network: peer.Network,
  411. }
  412. nodePeer = hostPeerUpdate.Peers[peerIndexMap[peerHost.PublicKey.String()]]
  413. }
  414. if node.Network == network { // add to peers map for metrics
  415. hostPeerUpdate.PeerIDs[peerHost.PublicKey.String()] = models.IDandAddr{
  416. ID: peer.ID.String(),
  417. Address: peer.PrimaryAddress(),
  418. Name: peerHost.Name,
  419. Network: peer.Network,
  420. }
  421. hostPeerUpdate.NodePeers = append(hostPeerUpdate.NodePeers, nodePeer)
  422. }
  423. }
  424. var extPeers []wgtypes.PeerConfig
  425. var extPeerIDAndAddrs []models.IDandAddr
  426. if node.IsIngressGateway {
  427. extPeers, extPeerIDAndAddrs, err = getExtPeers(&node)
  428. if err == nil {
  429. for _, extPeerIdAndAddr := range extPeerIDAndAddrs {
  430. nodePeerMap[extPeerIdAndAddr.ID] = models.PeerRouteInfo{
  431. PeerAddr: net.IPNet{
  432. IP: net.ParseIP(extPeerIdAndAddr.Address),
  433. Mask: getCIDRMaskFromAddr(extPeerIdAndAddr.Address),
  434. },
  435. PeerKey: extPeerIdAndAddr.ID,
  436. Allow: true,
  437. }
  438. }
  439. hostPeerUpdate.Peers = append(hostPeerUpdate.Peers, extPeers...)
  440. for _, extPeerIdAndAddr := range extPeerIDAndAddrs {
  441. hostPeerUpdate.HostPeerIDs[extPeerIdAndAddr.ID] = make(map[string]models.IDandAddr)
  442. hostPeerUpdate.HostPeerIDs[extPeerIdAndAddr.ID][extPeerIdAndAddr.ID] = models.IDandAddr{
  443. ID: extPeerIdAndAddr.ID,
  444. Address: extPeerIdAndAddr.Address,
  445. Name: extPeerIdAndAddr.Name,
  446. Network: node.Network,
  447. }
  448. hostPeerUpdate.IngressInfo.ExtPeers[extPeerIdAndAddr.ID] = models.ExtClientInfo{
  449. Masquerade: true,
  450. IngGwAddr: net.IPNet{
  451. IP: net.ParseIP(node.PrimaryAddress()),
  452. Mask: getCIDRMaskFromAddr(node.PrimaryAddress()),
  453. },
  454. Network: node.PrimaryNetworkRange(),
  455. ExtPeerAddr: net.IPNet{
  456. IP: net.ParseIP(extPeerIdAndAddr.Address),
  457. Mask: getCIDRMaskFromAddr(extPeerIdAndAddr.Address),
  458. },
  459. ExtPeerKey: extPeerIdAndAddr.ID,
  460. Peers: nodePeerMap,
  461. }
  462. if node.Network == network {
  463. hostPeerUpdate.PeerIDs[extPeerIdAndAddr.ID] = extPeerIdAndAddr
  464. hostPeerUpdate.NodePeers = append(hostPeerUpdate.NodePeers, extPeers...)
  465. }
  466. }
  467. } else if !database.IsEmptyRecord(err) {
  468. logger.Log(1, "error retrieving external clients:", err.Error())
  469. }
  470. }
  471. if node.IsEgressGateway {
  472. hostPeerUpdate.EgressInfo[node.ID.String()] = models.EgressInfo{
  473. EgressID: node.ID.String(),
  474. Network: node.PrimaryNetworkRange(),
  475. EgressGwAddr: net.IPNet{
  476. IP: net.ParseIP(node.PrimaryAddress()),
  477. Mask: getCIDRMaskFromAddr(node.PrimaryAddress()),
  478. },
  479. GwPeers: nodePeerMap,
  480. EgressGWCfg: node.EgressGatewayRequest,
  481. }
  482. }
  483. }
  484. return hostPeerUpdate, nil
  485. }
  486. func GetPeerListenPort(host *models.Host) int {
  487. peerPort := host.ListenPort
  488. if host.ProxyEnabled {
  489. if host.PublicListenPort != 0 {
  490. peerPort = host.PublicListenPort
  491. } else if host.ProxyListenPort != 0 {
  492. peerPort = host.ProxyListenPort
  493. }
  494. }
  495. return peerPort
  496. }
  497. func getExtPeers(node *models.Node) ([]wgtypes.PeerConfig, []models.IDandAddr, error) {
  498. var peers []wgtypes.PeerConfig
  499. var idsAndAddr []models.IDandAddr
  500. extPeers, err := GetNetworkExtClients(node.Network)
  501. if err != nil {
  502. return peers, idsAndAddr, err
  503. }
  504. host, err := GetHost(node.HostID.String())
  505. if err != nil {
  506. return peers, idsAndAddr, err
  507. }
  508. for _, extPeer := range extPeers {
  509. pubkey, err := wgtypes.ParseKey(extPeer.PublicKey)
  510. if err != nil {
  511. logger.Log(1, "error parsing ext pub key:", err.Error())
  512. continue
  513. }
  514. if host.PublicKey.String() == extPeer.PublicKey ||
  515. extPeer.IngressGatewayID != node.ID.String() || !extPeer.Enabled {
  516. continue
  517. }
  518. var allowedips []net.IPNet
  519. var peer wgtypes.PeerConfig
  520. if extPeer.Address != "" {
  521. var peeraddr = net.IPNet{
  522. IP: net.ParseIP(extPeer.Address),
  523. Mask: net.CIDRMask(32, 32),
  524. }
  525. if peeraddr.IP != nil && peeraddr.Mask != nil {
  526. allowedips = append(allowedips, peeraddr)
  527. }
  528. }
  529. if extPeer.Address6 != "" {
  530. var addr6 = net.IPNet{
  531. IP: net.ParseIP(extPeer.Address6),
  532. Mask: net.CIDRMask(128, 128),
  533. }
  534. if addr6.IP != nil && addr6.Mask != nil {
  535. allowedips = append(allowedips, addr6)
  536. }
  537. }
  538. primaryAddr := extPeer.Address
  539. if primaryAddr == "" {
  540. primaryAddr = extPeer.Address6
  541. }
  542. peer = wgtypes.PeerConfig{
  543. PublicKey: pubkey,
  544. ReplaceAllowedIPs: true,
  545. AllowedIPs: allowedips,
  546. }
  547. peers = append(peers, peer)
  548. idsAndAddr = append(idsAndAddr, models.IDandAddr{
  549. ID: peer.PublicKey.String(),
  550. Name: extPeer.ClientID,
  551. Address: primaryAddr,
  552. })
  553. }
  554. return peers, idsAndAddr, nil
  555. }
  556. func getExtPeersForProxy(node *models.Node, proxyPeerConf map[string]models.PeerConf) ([]wgtypes.PeerConfig, map[string]models.PeerConf, error) {
  557. var peers []wgtypes.PeerConfig
  558. host, err := GetHost(node.HostID.String())
  559. if err != nil {
  560. logger.Log(0, "error retrieving host for node", node.ID.String(), err.Error())
  561. }
  562. extPeers, err := GetNetworkExtClients(node.Network)
  563. if err != nil {
  564. return peers, proxyPeerConf, err
  565. }
  566. for _, extPeer := range extPeers {
  567. pubkey, err := wgtypes.ParseKey(extPeer.PublicKey)
  568. if err != nil {
  569. logger.Log(1, "error parsing ext pub key:", err.Error())
  570. continue
  571. }
  572. if host.PublicKey.String() == extPeer.PublicKey ||
  573. extPeer.IngressGatewayID != node.ID.String() || !extPeer.Enabled {
  574. continue
  575. }
  576. var allowedips []net.IPNet
  577. var peer wgtypes.PeerConfig
  578. if extPeer.Address != "" {
  579. var peeraddr = net.IPNet{
  580. IP: net.ParseIP(extPeer.Address),
  581. Mask: net.CIDRMask(32, 32),
  582. }
  583. if peeraddr.IP != nil && peeraddr.Mask != nil {
  584. allowedips = append(allowedips, peeraddr)
  585. }
  586. }
  587. if extPeer.Address6 != "" {
  588. var addr6 = net.IPNet{
  589. IP: net.ParseIP(extPeer.Address6),
  590. Mask: net.CIDRMask(128, 128),
  591. }
  592. if addr6.IP != nil && addr6.Mask != nil {
  593. allowedips = append(allowedips, addr6)
  594. }
  595. }
  596. peer = wgtypes.PeerConfig{
  597. PublicKey: pubkey,
  598. ReplaceAllowedIPs: true,
  599. AllowedIPs: allowedips,
  600. }
  601. extConf := models.PeerConf{
  602. IsExtClient: true,
  603. Address: net.ParseIP(extPeer.Address),
  604. }
  605. proxyPeerConf[peer.PublicKey.String()] = extConf
  606. peers = append(peers, peer)
  607. }
  608. return peers, proxyPeerConf, nil
  609. }
  610. // GetAllowedIPs - calculates the wireguard allowedip field for a peer of a node based on the peer and node settings
  611. func GetAllowedIPs(node, peer *models.Node, metrics *models.Metrics) []net.IPNet {
  612. var allowedips []net.IPNet
  613. allowedips = getNodeAllowedIPs(peer, node)
  614. // handle ingress gateway peers
  615. if peer.IsIngressGateway {
  616. extPeers, _, err := getExtPeers(peer)
  617. if err != nil {
  618. logger.Log(2, "could not retrieve ext peers for ", peer.ID.String(), err.Error())
  619. }
  620. for _, extPeer := range extPeers {
  621. allowedips = append(allowedips, extPeer.AllowedIPs...)
  622. }
  623. // if node is a failover node, add allowed ips from nodes it is handling
  624. if metrics != nil && peer.Failover && metrics.FailoverPeers != nil {
  625. // traverse through nodes that need handling
  626. logger.Log(3, "peer", peer.ID.String(), "was found to be failover for", node.ID.String(), "checking failover peers...")
  627. for k := range metrics.FailoverPeers {
  628. // if FailoverNode is me for this node, add allowedips
  629. if metrics.FailoverPeers[k] == peer.ID.String() {
  630. // get original node so we can traverse the allowed ips
  631. nodeToFailover, err := GetNodeByID(k)
  632. if err == nil {
  633. failoverNodeMetrics, err := GetMetrics(nodeToFailover.ID.String())
  634. if err == nil && failoverNodeMetrics != nil {
  635. if len(failoverNodeMetrics.NodeName) > 0 {
  636. allowedips = append(allowedips, getNodeAllowedIPs(&nodeToFailover, peer)...)
  637. logger.Log(0, "failing over node", nodeToFailover.ID.String(), nodeToFailover.PrimaryAddress(), "to failover node", peer.ID.String())
  638. }
  639. }
  640. }
  641. }
  642. }
  643. }
  644. }
  645. return allowedips
  646. }
  647. func getEgressIPs(node, peer *models.Node) []net.IPNet {
  648. host, err := GetHost(node.HostID.String())
  649. if err != nil {
  650. logger.Log(0, "error retrieving host for node", node.ID.String(), err.Error())
  651. }
  652. peerHost, err := GetHost(peer.HostID.String())
  653. if err != nil {
  654. logger.Log(0, "error retrieving host for peer", peer.ID.String(), err.Error())
  655. }
  656. //check for internet gateway
  657. internetGateway := false
  658. if slices.Contains(peer.EgressGatewayRanges, "0.0.0.0/0") || slices.Contains(peer.EgressGatewayRanges, "::/0") {
  659. internetGateway = true
  660. }
  661. allowedips := []net.IPNet{}
  662. for _, iprange := range peer.EgressGatewayRanges { // go through each cidr for egress gateway
  663. _, ipnet, err := net.ParseCIDR(iprange) // confirming it's valid cidr
  664. if err != nil {
  665. logger.Log(1, "could not parse gateway IP range. Not adding ", iprange)
  666. continue // if can't parse CIDR
  667. }
  668. // getting the public ip of node
  669. if ipnet.Contains(peerHost.EndpointIP) && !internetGateway { // ensuring egress gateway range does not contain endpoint of node
  670. logger.Log(2, "egress IP range of ", iprange, " overlaps with ", host.EndpointIP.String(), ", omitting")
  671. continue // skip adding egress range if overlaps with node's ip
  672. }
  673. // TODO: Could put in a lot of great logic to avoid conflicts / bad routes
  674. if ipnet.Contains(node.LocalAddress.IP) && !internetGateway { // ensuring egress gateway range does not contain public ip of node
  675. logger.Log(2, "egress IP range of ", iprange, " overlaps with ", node.LocalAddress.String(), ", omitting")
  676. continue // skip adding egress range if overlaps with node's local ip
  677. }
  678. if err != nil {
  679. logger.Log(1, "error encountered when setting egress range", err.Error())
  680. } else {
  681. allowedips = append(allowedips, *ipnet)
  682. }
  683. }
  684. return allowedips
  685. }
  686. func getNodeAllowedIPs(peer, node *models.Node) []net.IPNet {
  687. var allowedips = []net.IPNet{}
  688. if peer.Address.IP != nil {
  689. allowed := net.IPNet{
  690. IP: peer.Address.IP,
  691. Mask: net.CIDRMask(32, 32),
  692. }
  693. allowedips = append(allowedips, allowed)
  694. }
  695. if peer.Address6.IP != nil {
  696. allowed := net.IPNet{
  697. IP: peer.Address6.IP,
  698. Mask: net.CIDRMask(128, 128),
  699. }
  700. allowedips = append(allowedips, allowed)
  701. }
  702. // handle egress gateway peers
  703. if peer.IsEgressGateway {
  704. //hasGateway = true
  705. egressIPs := getEgressIPs(node, peer)
  706. allowedips = append(allowedips, egressIPs...)
  707. }
  708. return allowedips
  709. }
  710. func getCIDRMaskFromAddr(addr string) net.IPMask {
  711. cidr := net.CIDRMask(32, 32)
  712. ipAddr, err := netip.ParseAddr(addr)
  713. if err != nil {
  714. return cidr
  715. }
  716. if ipAddr.Is6() {
  717. cidr = net.CIDRMask(128, 128)
  718. }
  719. return cidr
  720. }