peers.go 37 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179
  1. package logic
  2. import (
  3. "errors"
  4. "fmt"
  5. "log"
  6. "net"
  7. "net/netip"
  8. "sort"
  9. "strconv"
  10. "strings"
  11. "time"
  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) (models.ProxyManagerPayload, error) {
  29. proxyPayload := models.ProxyManagerPayload{}
  30. var peers []wgtypes.PeerConfig
  31. peerConfMap := make(map[string]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.HostID.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.ListenPort))
  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]models.RelayedConf)
  70. for _, relayedNode := range relayedNodes {
  71. relayedNode := relayedNode
  72. payload, err := GetPeersForProxy(&relayedNode, true)
  73. if err == nil {
  74. relayedHost, err := GetHost(relayedNode.HostID.String())
  75. if err != nil {
  76. logger.Log(0, "error retrieving host for relayNode", relayedNode.ID.String(), err.Error())
  77. }
  78. relayedEndpoint, udpErr := net.ResolveUDPAddr("udp", fmt.Sprintf("%s:%d", relayedHost.EndpointIP, host.ListenPort))
  79. if udpErr == nil {
  80. relayPeersMap[host.PublicKey.String()] = models.RelayedConf{
  81. RelayedPeerEndpoint: relayedEndpoint,
  82. RelayedPeerPubKey: relayedHost.PublicKey.String(),
  83. Peers: payload.Peers,
  84. }
  85. }
  86. }
  87. }
  88. proxyPayload.IsRelay = true
  89. proxyPayload.RelayedPeerConf = relayPeersMap
  90. }
  91. }
  92. }
  93. for _, peer := range currentPeers {
  94. if peer.ID == node.ID {
  95. //skip yourself
  96. continue
  97. }
  98. host, err := GetHost(peer.HostID.String())
  99. if err != nil {
  100. continue
  101. }
  102. proxyStatus := host.ProxyEnabled
  103. listenPort := host.ListenPort
  104. if proxyStatus {
  105. listenPort = host.ProxyListenPort
  106. if listenPort == 0 {
  107. listenPort = models.NmProxyPort
  108. }
  109. } else if listenPort == 0 {
  110. listenPort = host.ListenPort
  111. }
  112. endpoint, err := net.ResolveUDPAddr("udp", fmt.Sprintf("%s:%d", host.EndpointIP, listenPort))
  113. if err != nil {
  114. logger.Log(1, "failed to resolve udp addr for node: ", peer.ID.String(), host.EndpointIP.String(), err.Error())
  115. continue
  116. }
  117. allowedips := GetAllowedIPs(node, &peer, nil)
  118. var keepalive time.Duration
  119. if node.PersistentKeepalive != 0 {
  120. // set_keepalive
  121. keepalive = node.PersistentKeepalive
  122. }
  123. peers = append(peers, wgtypes.PeerConfig{
  124. PublicKey: host.PublicKey,
  125. Endpoint: endpoint,
  126. AllowedIPs: allowedips,
  127. PersistentKeepaliveInterval: &keepalive,
  128. ReplaceAllowedIPs: true,
  129. })
  130. peerConfMap[host.PublicKey.String()] = models.PeerConf{
  131. Address: net.ParseIP(peer.PrimaryAddress()),
  132. Proxy: proxyStatus,
  133. PublicListenPort: int32(listenPort),
  134. }
  135. if !onlyPeers && peer.IsRelayed {
  136. relayNode := FindRelay(&peer)
  137. if relayNode != nil {
  138. relayHost, err := GetHost(relayNode.HostID.String())
  139. if err != nil {
  140. logger.Log(0, "error retrieving host for relayNode", relayNode.ID.String(), err.Error())
  141. continue
  142. }
  143. relayTo, err := net.ResolveUDPAddr("udp", fmt.Sprintf("%s:%d", relayHost.EndpointIP, relayHost.ListenPort))
  144. if err == nil {
  145. peerConfMap[host.PublicKey.String()] = models.PeerConf{
  146. IsRelayed: true,
  147. RelayedTo: relayTo,
  148. Address: net.ParseIP(peer.PrimaryAddress()),
  149. Proxy: proxyStatus,
  150. PublicListenPort: int32(listenPort),
  151. }
  152. }
  153. }
  154. }
  155. }
  156. if node.IsIngressGateway {
  157. var extPeers []wgtypes.PeerConfig
  158. extPeers, peerConfMap, err = getExtPeersForProxy(node, peerConfMap)
  159. if err == nil {
  160. peers = append(peers, extPeers...)
  161. } else if !database.IsEmptyRecord(err) {
  162. logger.Log(1, "error retrieving external clients:", err.Error())
  163. }
  164. }
  165. proxyPayload.IsIngress = node.IsIngressGateway
  166. addr := node.Address
  167. if addr.String() == "" {
  168. addr = node.Address6
  169. }
  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. // GetProxyUpdateForHost - gets the proxy update for host
  179. func GetProxyUpdateForHost(host *models.Host) (models.ProxyManagerPayload, error) {
  180. proxyPayload := models.ProxyManagerPayload{
  181. Action: models.ProxyUpdate,
  182. }
  183. peerConfMap := make(map[string]models.PeerConf)
  184. if host.IsRelayed {
  185. relayHost, err := GetHost(host.RelayedBy)
  186. if err == nil {
  187. relayEndpoint, err := net.ResolveUDPAddr("udp", fmt.Sprintf("%s:%d", relayHost.EndpointIP, GetPeerListenPort(relayHost)))
  188. if err != nil {
  189. logger.Log(1, "failed to resolve relay node endpoint: ", err.Error())
  190. }
  191. proxyPayload.IsRelayed = true
  192. proxyPayload.RelayedTo = relayEndpoint
  193. } else {
  194. logger.Log(0, "couldn't find relay host for: ", host.ID.String())
  195. }
  196. }
  197. if host.IsRelay {
  198. relayedHosts := GetRelayedHosts(host)
  199. relayPeersMap := make(map[string]models.RelayedConf)
  200. for _, relayedHost := range relayedHosts {
  201. relayedHost := relayedHost
  202. payload, err := GetPeerUpdateForHost(&relayedHost)
  203. if err == nil {
  204. relayedEndpoint, udpErr := net.ResolveUDPAddr("udp", fmt.Sprintf("%s:%d", relayedHost.EndpointIP, GetPeerListenPort(&relayedHost)))
  205. if udpErr == nil {
  206. relayPeersMap[relayedHost.PublicKey.String()] = models.RelayedConf{
  207. RelayedPeerEndpoint: relayedEndpoint,
  208. RelayedPeerPubKey: relayedHost.PublicKey.String(),
  209. Peers: payload.Peers,
  210. }
  211. }
  212. }
  213. }
  214. proxyPayload.IsRelay = true
  215. proxyPayload.RelayedPeerConf = relayPeersMap
  216. }
  217. var ingressStatus bool
  218. for _, nodeID := range host.Nodes {
  219. node, err := GetNodeByID(nodeID)
  220. if err != nil {
  221. continue
  222. }
  223. currentPeers, err := GetNetworkNodes(node.Network)
  224. if err != nil {
  225. continue
  226. }
  227. for _, peer := range currentPeers {
  228. if peer.ID == node.ID {
  229. //skip yourself
  230. continue
  231. }
  232. peerHost, err := GetHost(peer.HostID.String())
  233. if err != nil {
  234. continue
  235. }
  236. var currPeerConf models.PeerConf
  237. var found bool
  238. if currPeerConf, found = peerConfMap[peerHost.PublicKey.String()]; !found {
  239. currPeerConf = models.PeerConf{
  240. Proxy: peerHost.ProxyEnabled,
  241. PublicListenPort: int32(GetPeerListenPort(peerHost)),
  242. }
  243. }
  244. if peerHost.IsRelayed && peerHost.RelayedBy != host.ID.String() {
  245. relayHost, err := GetHost(peerHost.RelayedBy)
  246. if err == nil {
  247. relayTo, err := net.ResolveUDPAddr("udp", fmt.Sprintf("%s:%d", relayHost.EndpointIP, GetPeerListenPort(relayHost)))
  248. if err == nil {
  249. currPeerConf.IsRelayed = true
  250. currPeerConf.RelayedTo = relayTo
  251. }
  252. }
  253. }
  254. peerConfMap[peerHost.PublicKey.String()] = currPeerConf
  255. }
  256. if node.IsIngressGateway {
  257. ingressStatus = true
  258. _, peerConfMap, err = getExtPeersForProxy(&node, peerConfMap)
  259. if err == nil {
  260. } else if !database.IsEmptyRecord(err) {
  261. logger.Log(1, "error retrieving external clients:", err.Error())
  262. }
  263. }
  264. }
  265. proxyPayload.IsIngress = ingressStatus
  266. proxyPayload.PeerMap = peerConfMap
  267. return proxyPayload, nil
  268. }
  269. // GetPeerUpdateForHost - gets the consolidated peer update for the host from all networks
  270. func GetPeerUpdateForHost(host *models.Host) (models.HostPeerUpdate, error) {
  271. if host == nil {
  272. return models.HostPeerUpdate{}, errors.New("host is nil")
  273. }
  274. hostPeerUpdate := models.HostPeerUpdate{
  275. Host: *host,
  276. Server: servercfg.GetServer(),
  277. PeerIDs: make(models.HostPeerMap),
  278. ServerVersion: servercfg.GetVersion(),
  279. ServerAddrs: []models.ServerAddr{},
  280. IngressInfo: models.IngressInfo{
  281. ExtPeers: make(map[string]models.ExtClientInfo),
  282. },
  283. EgressInfo: make(map[string]models.EgressInfo),
  284. }
  285. logger.Log(1, "peer update for host ", host.ID.String())
  286. peerIndexMap := make(map[string]int)
  287. for _, nodeID := range host.Nodes {
  288. node, err := GetNodeByID(nodeID)
  289. if err != nil {
  290. continue
  291. }
  292. if !node.Connected || node.Action == models.NODE_DELETE || node.PendingDelete {
  293. continue
  294. }
  295. currentPeers, err := GetNetworkNodes(node.Network)
  296. if err != nil {
  297. log.Println("no network nodes")
  298. return models.HostPeerUpdate{}, err
  299. }
  300. var nodePeerMap map[string]models.PeerRouteInfo
  301. if node.IsIngressGateway || node.IsEgressGateway {
  302. nodePeerMap = make(map[string]models.PeerRouteInfo)
  303. }
  304. for _, peer := range currentPeers {
  305. if peer.ID == node.ID {
  306. logger.Log(2, "peer update, skipping self")
  307. //skip yourself
  308. continue
  309. }
  310. var peerConfig wgtypes.PeerConfig
  311. peerHost, err := GetHost(peer.HostID.String())
  312. if err != nil {
  313. logger.Log(1, "no peer host", peer.HostID.String(), err.Error())
  314. return models.HostPeerUpdate{}, err
  315. }
  316. if !peer.Connected || peer.Action == models.NODE_DELETE || peer.PendingDelete {
  317. logger.Log(2, "peer update, skipping unconnected node", peer.ID.String())
  318. //skip unconnected nodes
  319. continue
  320. }
  321. if !nodeacls.AreNodesAllowed(nodeacls.NetworkID(node.Network), nodeacls.NodeID(node.ID.String()), nodeacls.NodeID(peer.ID.String())) {
  322. log.Println("peer update, skipping node for acl")
  323. //skip if not permitted by acl
  324. continue
  325. }
  326. peerConfig.PublicKey = peerHost.PublicKey
  327. peerConfig.PersistentKeepaliveInterval = &peer.PersistentKeepalive
  328. peerConfig.ReplaceAllowedIPs = true
  329. uselocal := false
  330. if host.EndpointIP.String() == peerHost.EndpointIP.String() {
  331. //peer is on same network
  332. // set to localaddress
  333. uselocal = true
  334. if node.LocalAddress.IP == nil {
  335. // use public endpint
  336. uselocal = false
  337. }
  338. if node.LocalAddress.String() == peer.LocalAddress.String() {
  339. uselocal = false
  340. }
  341. }
  342. peerConfig.Endpoint = &net.UDPAddr{
  343. IP: peerHost.EndpointIP,
  344. Port: GetPeerListenPort(peerHost),
  345. }
  346. if uselocal {
  347. peerConfig.Endpoint.IP = peer.LocalAddress.IP
  348. }
  349. allowedips := GetAllowedIPs(&node, &peer, nil)
  350. if peer.IsIngressGateway {
  351. for _, entry := range peer.IngressGatewayRange {
  352. _, cidr, err := net.ParseCIDR(string(entry))
  353. if err == nil {
  354. allowedips = append(allowedips, *cidr)
  355. }
  356. }
  357. }
  358. if peer.IsEgressGateway {
  359. allowedips = append(allowedips, getEgressIPs(&node, &peer)...)
  360. }
  361. peerConfig.AllowedIPs = allowedips
  362. if node.IsIngressGateway || node.IsEgressGateway {
  363. if peer.IsIngressGateway {
  364. _, extPeerIDAndAddrs, err := getExtPeers(&peer)
  365. if err == nil {
  366. for _, extPeerIdAndAddr := range extPeerIDAndAddrs {
  367. nodePeerMap[extPeerIdAndAddr.ID] = models.PeerRouteInfo{
  368. PeerAddr: net.IPNet{
  369. IP: net.ParseIP(extPeerIdAndAddr.Address),
  370. Mask: getCIDRMaskFromAddr(extPeerIdAndAddr.Address),
  371. },
  372. PeerKey: extPeerIdAndAddr.ID,
  373. Allow: true,
  374. }
  375. }
  376. }
  377. }
  378. if node.IsIngressGateway && peer.IsEgressGateway {
  379. hostPeerUpdate.IngressInfo.EgressRanges = append(hostPeerUpdate.IngressInfo.EgressRanges,
  380. peer.EgressGatewayRanges...)
  381. }
  382. nodePeerMap[peerHost.PublicKey.String()] = models.PeerRouteInfo{
  383. PeerAddr: net.IPNet{
  384. IP: net.ParseIP(peer.PrimaryAddress()),
  385. Mask: getCIDRMaskFromAddr(peer.PrimaryAddress()),
  386. },
  387. PeerKey: peerHost.PublicKey.String(),
  388. Allow: true,
  389. }
  390. }
  391. if _, ok := hostPeerUpdate.PeerIDs[peerHost.PublicKey.String()]; !ok {
  392. hostPeerUpdate.PeerIDs[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.PeerIDs[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. } else {
  402. peerAllowedIPs := hostPeerUpdate.Peers[peerIndexMap[peerHost.PublicKey.String()]].AllowedIPs
  403. peerAllowedIPs = append(peerAllowedIPs, allowedips...)
  404. hostPeerUpdate.Peers[peerIndexMap[peerHost.PublicKey.String()]].AllowedIPs = peerAllowedIPs
  405. hostPeerUpdate.PeerIDs[peerHost.PublicKey.String()][peer.ID.String()] = models.IDandAddr{
  406. ID: peer.ID.String(),
  407. Address: peer.PrimaryAddress(),
  408. Name: peerHost.Name,
  409. Network: peer.Network,
  410. }
  411. }
  412. }
  413. var extPeers []wgtypes.PeerConfig
  414. var extPeerIDAndAddrs []models.IDandAddr
  415. if node.IsIngressGateway {
  416. extPeers, extPeerIDAndAddrs, err = getExtPeers(&node)
  417. if err == nil {
  418. for _, extPeerIdAndAddr := range extPeerIDAndAddrs {
  419. nodePeerMap[extPeerIdAndAddr.ID] = models.PeerRouteInfo{
  420. PeerAddr: net.IPNet{
  421. IP: net.ParseIP(extPeerIdAndAddr.Address),
  422. Mask: getCIDRMaskFromAddr(extPeerIdAndAddr.Address),
  423. },
  424. PeerKey: extPeerIdAndAddr.ID,
  425. Allow: true,
  426. }
  427. }
  428. hostPeerUpdate.Peers = append(hostPeerUpdate.Peers, extPeers...)
  429. for _, extPeerIdAndAddr := range extPeerIDAndAddrs {
  430. hostPeerUpdate.PeerIDs[extPeerIdAndAddr.ID] = make(map[string]models.IDandAddr)
  431. hostPeerUpdate.PeerIDs[extPeerIdAndAddr.ID][extPeerIdAndAddr.ID] = models.IDandAddr{
  432. ID: extPeerIdAndAddr.ID,
  433. Address: extPeerIdAndAddr.Address,
  434. Name: extPeerIdAndAddr.Name,
  435. Network: node.Network,
  436. }
  437. hostPeerUpdate.IngressInfo.ExtPeers[extPeerIdAndAddr.ID] = models.ExtClientInfo{
  438. Masquerade: true,
  439. IngGwAddr: net.IPNet{
  440. IP: net.ParseIP(node.PrimaryAddress()),
  441. Mask: getCIDRMaskFromAddr(node.PrimaryAddress()),
  442. },
  443. Network: node.PrimaryNetworkRange(),
  444. ExtPeerAddr: net.IPNet{
  445. IP: net.ParseIP(extPeerIdAndAddr.Address),
  446. Mask: getCIDRMaskFromAddr(extPeerIdAndAddr.Address),
  447. },
  448. ExtPeerKey: extPeerIdAndAddr.ID,
  449. Peers: nodePeerMap,
  450. }
  451. }
  452. } else if !database.IsEmptyRecord(err) {
  453. logger.Log(1, "error retrieving external clients:", err.Error())
  454. }
  455. }
  456. if node.IsEgressGateway {
  457. hostPeerUpdate.EgressInfo[node.ID.String()] = models.EgressInfo{
  458. EgressID: node.ID.String(),
  459. Network: node.PrimaryNetworkRange(),
  460. EgressGwAddr: net.IPNet{
  461. IP: net.ParseIP(node.PrimaryAddress()),
  462. Mask: getCIDRMaskFromAddr(node.PrimaryAddress()),
  463. },
  464. GwPeers: nodePeerMap,
  465. EgressGWCfg: node.EgressGatewayRequest,
  466. }
  467. }
  468. }
  469. return hostPeerUpdate, nil
  470. }
  471. func GetPeerListenPort(host *models.Host) int {
  472. peerPort := host.ListenPort
  473. if host.ProxyEnabled {
  474. if host.PublicListenPort != 0 {
  475. peerPort = host.PublicListenPort
  476. } else if host.ProxyListenPort != 0 {
  477. peerPort = host.ProxyListenPort
  478. }
  479. }
  480. return peerPort
  481. }
  482. // GetPeerUpdate - gets a wireguard peer config for each peer of a node
  483. func GetPeerUpdate(node *models.Node, host *models.Host) (models.PeerUpdate, error) {
  484. log.Println("peer update for node ", node.ID)
  485. peerUpdate := models.PeerUpdate{
  486. Network: node.Network,
  487. ServerVersion: ncutils.Version,
  488. PeerIDs: make(models.PeerMap),
  489. }
  490. currentPeers, err := GetNetworkNodes(node.Network)
  491. if err != nil {
  492. log.Println("no network nodes")
  493. return models.PeerUpdate{}, err
  494. }
  495. for _, peer := range currentPeers {
  496. var peerConfig wgtypes.PeerConfig
  497. peerHost, err := GetHost(peer.HostID.String())
  498. if err != nil {
  499. log.Println("no peer host", err)
  500. return models.PeerUpdate{}, err
  501. }
  502. if peer.ID == node.ID {
  503. log.Println("peer update, skipping self")
  504. //skip yourself
  505. continue
  506. }
  507. if !peer.Connected {
  508. log.Println("peer update, skipping unconnected node")
  509. //skip unconnected nodes
  510. continue
  511. }
  512. if !nodeacls.AreNodesAllowed(nodeacls.NetworkID(node.Network), nodeacls.NodeID(node.ID.String()), nodeacls.NodeID(peer.ID.String())) {
  513. log.Println("peer update, skipping node for acl")
  514. //skip if not permitted by acl
  515. continue
  516. }
  517. peerConfig.PublicKey = peerHost.PublicKey
  518. peerConfig.PersistentKeepaliveInterval = &peer.PersistentKeepalive
  519. peerConfig.ReplaceAllowedIPs = true
  520. uselocal := false
  521. if host.EndpointIP.String() == peerHost.EndpointIP.String() {
  522. //peer is on same network
  523. // set to localaddress
  524. uselocal = true
  525. if node.LocalAddress.IP == nil {
  526. // use public endpint
  527. uselocal = false
  528. }
  529. if node.LocalAddress.String() == peer.LocalAddress.String() {
  530. uselocal = false
  531. }
  532. }
  533. peerConfig.Endpoint = &net.UDPAddr{
  534. IP: peerHost.EndpointIP,
  535. Port: peerHost.ListenPort,
  536. }
  537. if peerHost.ProxyEnabled {
  538. peerConfig.Endpoint.Port = GetPeerListenPort(peerHost)
  539. }
  540. if uselocal {
  541. peerConfig.Endpoint.IP = peer.LocalAddress.IP
  542. }
  543. allowedips := getNodeAllowedIPs(&peer, node)
  544. if peer.IsIngressGateway {
  545. for _, entry := range peer.IngressGatewayRange {
  546. _, cidr, err := net.ParseCIDR(string(entry))
  547. if err == nil {
  548. allowedips = append(allowedips, *cidr)
  549. }
  550. }
  551. }
  552. if peer.IsEgressGateway {
  553. allowedips = append(allowedips, getEgressIPs(node, &peer)...)
  554. }
  555. peerUpdate.PeerIDs[peerHost.PublicKey.String()] = models.IDandAddr{
  556. ID: peer.ID.String(),
  557. Address: peer.PrimaryAddress(),
  558. Name: peerHost.Name,
  559. Network: peer.Network,
  560. }
  561. peerConfig.AllowedIPs = allowedips
  562. peerUpdate.Peers = append(peerUpdate.Peers, peerConfig)
  563. }
  564. return peerUpdate, nil
  565. }
  566. // func getRelayAllowedIPs(node, peer *models.Node) []net.IPNet {
  567. // var allowedips []net.IPNet
  568. // var allowedip net.IPNet
  569. // for _, addr := range peer.RelayAddrs {
  570. // if node.Address.IP.String() == addr {
  571. // continue
  572. // }
  573. // if node.Address6.IP.String() == addr {
  574. // continue
  575. // }
  576. // allowedip.IP = net.ParseIP(addr)
  577. // allowedips = append(allowedips, allowedip)
  578. // }
  579. // return allowedips
  580. // }
  581. // GetPeerUpdateLegacy - gets a wireguard peer config for each peer of a node
  582. func GetPeerUpdateLegacy(node *models.Node) (models.PeerUpdate, error) {
  583. var peerUpdate models.PeerUpdate
  584. var peers []wgtypes.PeerConfig
  585. var serverNodeAddresses = []models.ServerAddr{}
  586. var peerMap = make(models.PeerMap)
  587. var metrics *models.Metrics
  588. if servercfg.Is_EE {
  589. metrics, _ = GetMetrics(node.ID.String())
  590. }
  591. if metrics == nil {
  592. metrics = &models.Metrics{}
  593. }
  594. if metrics.FailoverPeers == nil {
  595. metrics.FailoverPeers = make(map[string]string)
  596. }
  597. // udppeers = the peers parsed from the local interface
  598. // gives us correct port to reach
  599. udppeers, errN := database.GetPeers(node.Network)
  600. if errN != nil {
  601. logger.Log(2, errN.Error())
  602. }
  603. currentPeers, err := GetNetworkNodes(node.Network)
  604. if err != nil {
  605. return models.PeerUpdate{}, err
  606. }
  607. host, err := GetHost(node.HostID.String())
  608. if err != nil {
  609. return peerUpdate, err
  610. }
  611. if node.IsRelayed && !host.ProxyEnabled {
  612. return GetPeerUpdateForRelayedNode(node, udppeers)
  613. }
  614. // #1 Set Keepalive values: set_keepalive
  615. // #2 Set local address: set_local - could be a LOT BETTER and fix some bugs with additional logic
  616. // #3 Set allowedips: set_allowedips
  617. for _, peer := range currentPeers {
  618. peerHost, err := GetHost(peer.HostID.String())
  619. if err != nil {
  620. logger.Log(0, "error retrieving host for peer", node.ID.String(), err.Error())
  621. return models.PeerUpdate{}, err
  622. }
  623. if peer.ID == node.ID {
  624. //skip yourself
  625. continue
  626. }
  627. if node.Connected {
  628. //skip unconnected nodes
  629. continue
  630. }
  631. // if the node is not a server, set the endpoint
  632. var setEndpoint = true
  633. if peer.IsRelayed {
  634. if !peerHost.ProxyEnabled && !(node.IsRelay && ncutils.StringSliceContains(node.RelayAddrs, peer.PrimaryAddress())) {
  635. //skip -- will be added to relay
  636. continue
  637. } else if node.IsRelay && ncutils.StringSliceContains(node.RelayAddrs, peer.PrimaryAddress()) {
  638. // dont set peer endpoint if it's relayed by node
  639. setEndpoint = false
  640. }
  641. }
  642. if !nodeacls.AreNodesAllowed(nodeacls.NetworkID(node.Network), nodeacls.NodeID(node.ID.String()), nodeacls.NodeID(peer.ID.String())) {
  643. //skip if not permitted by acl
  644. continue
  645. }
  646. if len(metrics.FailoverPeers[peer.ID.String()]) > 0 && IsFailoverPresent(node.Network) {
  647. logger.Log(2, "peer", peer.ID.String(), peer.PrimaryAddress(), "was found to be in failover peers list for node", node.ID.String(), node.PrimaryAddress())
  648. continue
  649. }
  650. if err != nil {
  651. return models.PeerUpdate{}, err
  652. }
  653. host, err := GetHost(node.HostID.String())
  654. if err != nil {
  655. logger.Log(0, "error retrieving host for node", node.ID.String(), err.Error())
  656. return models.PeerUpdate{}, err
  657. }
  658. if host.EndpointIP.String() == peerHost.EndpointIP.String() {
  659. //peer is on same network
  660. // set_local
  661. if node.LocalAddress.String() != peer.LocalAddress.String() && peer.LocalAddress.IP != nil {
  662. peerHost.EndpointIP = peer.LocalAddress.IP
  663. if peerHost.ListenPort != 0 {
  664. peerHost.ListenPort = GetPeerListenPort(peerHost)
  665. }
  666. } else {
  667. continue
  668. }
  669. }
  670. // set address if setEndpoint is true
  671. // otherwise, will get inserted as empty value
  672. var address *net.UDPAddr
  673. // Sets ListenPort to UDP Hole Punching Port assuming:
  674. // - UDP Hole Punching is enabled
  675. // - udppeers retrieval did not return an error
  676. // - the endpoint is valid
  677. if setEndpoint {
  678. var setUDPPort = false
  679. if CheckEndpoint(udppeers[peerHost.PublicKey.String()]) {
  680. endpointstring := udppeers[peerHost.PublicKey.String()]
  681. endpointarr := strings.Split(endpointstring, ":")
  682. if len(endpointarr) == 2 {
  683. port, err := strconv.Atoi(endpointarr[1])
  684. if err == nil {
  685. setUDPPort = true
  686. peerHost.ListenPort = port
  687. }
  688. }
  689. }
  690. // if udp hole punching is on, but udp hole punching did not set it, use the LocalListenPort instead
  691. // or, if port is for some reason zero use the LocalListenPort
  692. // but only do this if LocalListenPort is not zero
  693. if ((!setUDPPort) || peerHost.ListenPort == 0) && peerHost.ListenPort != 0 {
  694. peerHost.ListenPort = GetPeerListenPort(peerHost)
  695. }
  696. endpoint := peerHost.EndpointIP.String() + ":" + strconv.FormatInt(int64(peerHost.ListenPort), 10)
  697. address, err = net.ResolveUDPAddr("udp", endpoint)
  698. if err != nil {
  699. return models.PeerUpdate{}, err
  700. }
  701. }
  702. allowedips := GetAllowedIPs(node, &peer, metrics)
  703. var keepalive time.Duration
  704. if node.PersistentKeepalive != 0 {
  705. // set_keepalive
  706. keepalive = node.PersistentKeepalive
  707. }
  708. var peerData = wgtypes.PeerConfig{
  709. PublicKey: peerHost.PublicKey,
  710. Endpoint: address,
  711. ReplaceAllowedIPs: true,
  712. AllowedIPs: allowedips,
  713. PersistentKeepaliveInterval: &keepalive,
  714. }
  715. peers = append(peers, peerData)
  716. peerMap[peerHost.PublicKey.String()] = models.IDandAddr{
  717. Name: peerHost.Name,
  718. ID: peer.ID.String(),
  719. Address: peer.PrimaryAddress(),
  720. IsServer: "no",
  721. }
  722. }
  723. if node.IsIngressGateway {
  724. extPeers, idsAndAddr, err := getExtPeers(node)
  725. if err == nil {
  726. peers = append(peers, extPeers...)
  727. for i := range idsAndAddr {
  728. peerMap[idsAndAddr[i].ID] = idsAndAddr[i]
  729. }
  730. } else if !database.IsEmptyRecord(err) {
  731. logger.Log(1, "error retrieving external clients:", err.Error())
  732. }
  733. }
  734. peerUpdate.Network = node.Network
  735. peerUpdate.ServerVersion = servercfg.Version
  736. sort.SliceStable(peers[:], func(i, j int) bool {
  737. return peers[i].PublicKey.String() < peers[j].PublicKey.String()
  738. })
  739. peerUpdate.Peers = peers
  740. peerUpdate.ServerAddrs = serverNodeAddresses
  741. peerUpdate.PeerIDs = peerMap
  742. return peerUpdate, nil
  743. }
  744. func getExtPeers(node *models.Node) ([]wgtypes.PeerConfig, []models.IDandAddr, error) {
  745. var peers []wgtypes.PeerConfig
  746. var idsAndAddr []models.IDandAddr
  747. extPeers, err := GetNetworkExtClients(node.Network)
  748. if err != nil {
  749. return peers, idsAndAddr, err
  750. }
  751. host, err := GetHost(node.HostID.String())
  752. if err != nil {
  753. return peers, idsAndAddr, err
  754. }
  755. for _, extPeer := range extPeers {
  756. pubkey, err := wgtypes.ParseKey(extPeer.PublicKey)
  757. if err != nil {
  758. logger.Log(1, "error parsing ext pub key:", err.Error())
  759. continue
  760. }
  761. if host.PublicKey.String() == extPeer.PublicKey ||
  762. extPeer.IngressGatewayID != node.ID.String() || !extPeer.Enabled {
  763. continue
  764. }
  765. var allowedips []net.IPNet
  766. var peer wgtypes.PeerConfig
  767. if extPeer.Address != "" {
  768. var peeraddr = net.IPNet{
  769. IP: net.ParseIP(extPeer.Address),
  770. Mask: net.CIDRMask(32, 32),
  771. }
  772. if peeraddr.IP != nil && peeraddr.Mask != nil {
  773. allowedips = append(allowedips, peeraddr)
  774. }
  775. }
  776. if extPeer.Address6 != "" {
  777. var addr6 = net.IPNet{
  778. IP: net.ParseIP(extPeer.Address6),
  779. Mask: net.CIDRMask(128, 128),
  780. }
  781. if addr6.IP != nil && addr6.Mask != nil {
  782. allowedips = append(allowedips, addr6)
  783. }
  784. }
  785. primaryAddr := extPeer.Address
  786. if primaryAddr == "" {
  787. primaryAddr = extPeer.Address6
  788. }
  789. peer = wgtypes.PeerConfig{
  790. PublicKey: pubkey,
  791. ReplaceAllowedIPs: true,
  792. AllowedIPs: allowedips,
  793. }
  794. peers = append(peers, peer)
  795. idsAndAddr = append(idsAndAddr, models.IDandAddr{
  796. ID: peer.PublicKey.String(),
  797. Name: extPeer.ClientID,
  798. Address: primaryAddr,
  799. })
  800. }
  801. return peers, idsAndAddr, nil
  802. }
  803. func getExtPeersForProxy(node *models.Node, proxyPeerConf map[string]models.PeerConf) ([]wgtypes.PeerConfig, map[string]models.PeerConf, error) {
  804. var peers []wgtypes.PeerConfig
  805. host, err := GetHost(node.HostID.String())
  806. if err != nil {
  807. logger.Log(0, "error retrieving host for node", node.ID.String(), err.Error())
  808. }
  809. extPeers, err := GetNetworkExtClients(node.Network)
  810. if err != nil {
  811. return peers, proxyPeerConf, err
  812. }
  813. for _, extPeer := range extPeers {
  814. pubkey, err := wgtypes.ParseKey(extPeer.PublicKey)
  815. if err != nil {
  816. logger.Log(1, "error parsing ext pub key:", err.Error())
  817. continue
  818. }
  819. if host.PublicKey.String() == extPeer.PublicKey ||
  820. extPeer.IngressGatewayID != node.ID.String() || !extPeer.Enabled {
  821. continue
  822. }
  823. var allowedips []net.IPNet
  824. var peer wgtypes.PeerConfig
  825. if extPeer.Address != "" {
  826. var peeraddr = net.IPNet{
  827. IP: net.ParseIP(extPeer.Address),
  828. Mask: net.CIDRMask(32, 32),
  829. }
  830. if peeraddr.IP != nil && peeraddr.Mask != nil {
  831. allowedips = append(allowedips, peeraddr)
  832. }
  833. }
  834. if extPeer.Address6 != "" {
  835. var addr6 = net.IPNet{
  836. IP: net.ParseIP(extPeer.Address6),
  837. Mask: net.CIDRMask(128, 128),
  838. }
  839. if addr6.IP != nil && addr6.Mask != nil {
  840. allowedips = append(allowedips, addr6)
  841. }
  842. }
  843. peer = wgtypes.PeerConfig{
  844. PublicKey: pubkey,
  845. ReplaceAllowedIPs: true,
  846. AllowedIPs: allowedips,
  847. }
  848. extConf := models.PeerConf{
  849. IsExtClient: true,
  850. Address: net.ParseIP(extPeer.Address),
  851. }
  852. proxyPeerConf[peer.PublicKey.String()] = extConf
  853. peers = append(peers, peer)
  854. }
  855. return peers, proxyPeerConf, nil
  856. }
  857. // GetAllowedIPs - calculates the wireguard allowedip field for a peer of a node based on the peer and node settings
  858. func GetAllowedIPs(node, peer *models.Node, metrics *models.Metrics) []net.IPNet {
  859. var allowedips []net.IPNet
  860. allowedips = getNodeAllowedIPs(peer, node)
  861. // handle ingress gateway peers
  862. if peer.IsIngressGateway {
  863. extPeers, _, err := getExtPeers(peer)
  864. if err != nil {
  865. logger.Log(2, "could not retrieve ext peers for ", peer.ID.String(), err.Error())
  866. }
  867. for _, extPeer := range extPeers {
  868. allowedips = append(allowedips, extPeer.AllowedIPs...)
  869. }
  870. // if node is a failover node, add allowed ips from nodes it is handling
  871. if metrics != nil && peer.Failover && metrics.FailoverPeers != nil {
  872. // traverse through nodes that need handling
  873. logger.Log(3, "peer", peer.ID.String(), "was found to be failover for", node.ID.String(), "checking failover peers...")
  874. for k := range metrics.FailoverPeers {
  875. // if FailoverNode is me for this node, add allowedips
  876. if metrics.FailoverPeers[k] == peer.ID.String() {
  877. // get original node so we can traverse the allowed ips
  878. nodeToFailover, err := GetNodeByID(k)
  879. if err == nil {
  880. failoverNodeMetrics, err := GetMetrics(nodeToFailover.ID.String())
  881. if err == nil && failoverNodeMetrics != nil {
  882. if len(failoverNodeMetrics.NodeName) > 0 {
  883. allowedips = append(allowedips, getNodeAllowedIPs(&nodeToFailover, peer)...)
  884. logger.Log(0, "failing over node", nodeToFailover.ID.String(), nodeToFailover.PrimaryAddress(), "to failover node", peer.ID.String())
  885. }
  886. }
  887. }
  888. }
  889. }
  890. }
  891. }
  892. return allowedips
  893. }
  894. // GetPeerUpdateForRelayedNode - calculates peer update for a relayed node by getting the relay
  895. // copying the relay node's allowed ips and making appropriate substitutions
  896. func GetPeerUpdateForRelayedNode(node *models.Node, udppeers map[string]string) (models.PeerUpdate, error) {
  897. var peerUpdate models.PeerUpdate
  898. var peers []wgtypes.PeerConfig
  899. var serverNodeAddresses = []models.ServerAddr{}
  900. var allowedips []net.IPNet
  901. //find node that is relaying us
  902. relay := FindRelay(node)
  903. if relay == nil {
  904. return models.PeerUpdate{}, errors.New("not found")
  905. }
  906. //add relay to lists of allowed ip
  907. if relay.Address.IP != nil {
  908. relayIP := relay.Address
  909. allowedips = append(allowedips, relayIP)
  910. }
  911. if relay.Address6.IP != nil {
  912. relayIP6 := relay.Address6
  913. allowedips = append(allowedips, relayIP6)
  914. }
  915. //get PeerUpdate for relayed node
  916. relayHost, err := GetHost(relay.HostID.String())
  917. if err != nil {
  918. return models.PeerUpdate{}, err
  919. }
  920. relayPeerUpdate, err := GetPeerUpdate(relay, relayHost)
  921. if err != nil {
  922. return models.PeerUpdate{}, err
  923. }
  924. //add the relays allowed ips from all of the relay's peers
  925. for _, peer := range relayPeerUpdate.Peers {
  926. allowedips = append(allowedips, peer.AllowedIPs...)
  927. }
  928. //delete any ips not permitted by acl
  929. for i := len(allowedips) - 1; i >= 0; i-- {
  930. target, err := findNode(allowedips[i].IP.String())
  931. if err != nil {
  932. logger.Log(0, "failed to find node for ip", allowedips[i].IP.String(), err.Error())
  933. continue
  934. }
  935. if target == nil {
  936. logger.Log(0, "failed to find node for ip", allowedips[i].IP.String())
  937. continue
  938. }
  939. if !nodeacls.AreNodesAllowed(nodeacls.NetworkID(node.Network), nodeacls.NodeID(node.ID.String()), nodeacls.NodeID(target.ID.String())) {
  940. logger.Log(0, "deleting node from relayednode per acl", node.ID.String(), target.ID.String())
  941. allowedips = append(allowedips[:i], allowedips[i+1:]...)
  942. }
  943. }
  944. //delete self from allowed ips
  945. for i := len(allowedips) - 1; i >= 0; i-- {
  946. if allowedips[i].IP.String() == node.Address.IP.String() || allowedips[i].IP.String() == node.Address6.IP.String() {
  947. allowedips = append(allowedips[:i], allowedips[i+1:]...)
  948. }
  949. }
  950. //delete egressrange from allowedip if we are egress gateway
  951. if node.IsEgressGateway {
  952. for i := len(allowedips) - 1; i >= 0; i-- {
  953. if StringSliceContains(node.EgressGatewayRanges, allowedips[i].String()) {
  954. allowedips = append(allowedips[:i], allowedips[i+1:]...)
  955. }
  956. }
  957. }
  958. //delete extclients from allowedip if we are ingress gateway
  959. if node.IsIngressGateway {
  960. for i := len(allowedips) - 1; i >= 0; i-- {
  961. if strings.Contains(node.IngressGatewayRange, allowedips[i].IP.String()) {
  962. allowedips = append(allowedips[:i], allowedips[i+1:]...)
  963. }
  964. }
  965. }
  966. //add egress range if relay is egress
  967. if relay.IsEgressGateway {
  968. var ip *net.IPNet
  969. for _, cidr := range relay.EgressGatewayRanges {
  970. _, ip, err = net.ParseCIDR(cidr)
  971. if err != nil {
  972. continue
  973. }
  974. }
  975. allowedips = append(allowedips, *ip)
  976. }
  977. var setUDPPort = false
  978. var listenPort int
  979. if CheckEndpoint(udppeers[relayHost.PublicKey.String()]) {
  980. endpointstring := udppeers[relayHost.PublicKey.String()]
  981. endpointarr := strings.Split(endpointstring, ":")
  982. if len(endpointarr) == 2 {
  983. port, err := strconv.Atoi(endpointarr[1])
  984. if err == nil {
  985. setUDPPort = true
  986. listenPort = port
  987. }
  988. }
  989. }
  990. // if udp hole punching is on, but udp hole punching did not set it, use the LocalListenPort instead
  991. // or, if port is for some reason zero use the LocalListenPort
  992. // but only do this if LocalListenPort is not zero
  993. if ((!setUDPPort) || relayHost.ListenPort == 0) && relayHost.ListenPort != 0 {
  994. listenPort = relayHost.ListenPort
  995. }
  996. endpoint := relayHost.EndpointIP.String() + ":" + strconv.FormatInt(int64(listenPort), 10)
  997. address, err := net.ResolveUDPAddr("udp", endpoint)
  998. if err != nil {
  999. return models.PeerUpdate{}, err
  1000. }
  1001. var keepalive time.Duration
  1002. if node.PersistentKeepalive != 0 {
  1003. // set_keepalive
  1004. keepalive = node.PersistentKeepalive
  1005. }
  1006. var peerData = wgtypes.PeerConfig{
  1007. PublicKey: relayHost.PublicKey,
  1008. Endpoint: address,
  1009. ReplaceAllowedIPs: true,
  1010. AllowedIPs: allowedips,
  1011. PersistentKeepaliveInterval: &keepalive,
  1012. }
  1013. peers = append(peers, peerData)
  1014. //if ingress add extclients
  1015. if node.IsIngressGateway {
  1016. extPeers, _, err := getExtPeers(node)
  1017. if err == nil {
  1018. peers = append(peers, extPeers...)
  1019. } else {
  1020. logger.Log(2, "could not retrieve ext peers for ", node.ID.String(), err.Error())
  1021. }
  1022. }
  1023. peerUpdate.Network = node.Network
  1024. peerUpdate.ServerVersion = servercfg.Version
  1025. sort.SliceStable(peers[:], func(i, j int) bool {
  1026. return peers[i].PublicKey.String() < peers[j].PublicKey.String()
  1027. })
  1028. peerUpdate.Peers = peers
  1029. peerUpdate.ServerAddrs = serverNodeAddresses
  1030. return peerUpdate, nil
  1031. }
  1032. func getEgressIPs(node, peer *models.Node) []net.IPNet {
  1033. host, err := GetHost(node.HostID.String())
  1034. if err != nil {
  1035. logger.Log(0, "error retrieving host for node", node.ID.String(), err.Error())
  1036. }
  1037. peerHost, err := GetHost(peer.HostID.String())
  1038. if err != nil {
  1039. logger.Log(0, "error retrieving host for peer", peer.ID.String(), err.Error())
  1040. }
  1041. //check for internet gateway
  1042. internetGateway := false
  1043. if slices.Contains(peer.EgressGatewayRanges, "0.0.0.0/0") || slices.Contains(peer.EgressGatewayRanges, "::/0") {
  1044. internetGateway = true
  1045. }
  1046. allowedips := []net.IPNet{}
  1047. for _, iprange := range peer.EgressGatewayRanges { // go through each cidr for egress gateway
  1048. _, ipnet, err := net.ParseCIDR(iprange) // confirming it's valid cidr
  1049. if err != nil {
  1050. logger.Log(1, "could not parse gateway IP range. Not adding ", iprange)
  1051. continue // if can't parse CIDR
  1052. }
  1053. // getting the public ip of node
  1054. if ipnet.Contains(peerHost.EndpointIP) && !internetGateway { // ensuring egress gateway range does not contain endpoint of node
  1055. logger.Log(2, "egress IP range of ", iprange, " overlaps with ", host.EndpointIP.String(), ", omitting")
  1056. continue // skip adding egress range if overlaps with node's ip
  1057. }
  1058. // TODO: Could put in a lot of great logic to avoid conflicts / bad routes
  1059. if ipnet.Contains(node.LocalAddress.IP) && !internetGateway { // ensuring egress gateway range does not contain public ip of node
  1060. logger.Log(2, "egress IP range of ", iprange, " overlaps with ", node.LocalAddress.String(), ", omitting")
  1061. continue // skip adding egress range if overlaps with node's local ip
  1062. }
  1063. if err != nil {
  1064. logger.Log(1, "error encountered when setting egress range", err.Error())
  1065. } else {
  1066. allowedips = append(allowedips, *ipnet)
  1067. }
  1068. }
  1069. return allowedips
  1070. }
  1071. func getNodeAllowedIPs(peer, node *models.Node) []net.IPNet {
  1072. var allowedips = []net.IPNet{}
  1073. if peer.Address.IP != nil {
  1074. allowed := net.IPNet{
  1075. IP: peer.Address.IP,
  1076. Mask: net.CIDRMask(32, 32),
  1077. }
  1078. allowedips = append(allowedips, allowed)
  1079. }
  1080. if peer.Address6.IP != nil {
  1081. allowed := net.IPNet{
  1082. IP: peer.Address6.IP,
  1083. Mask: net.CIDRMask(128, 128),
  1084. }
  1085. allowedips = append(allowedips, allowed)
  1086. }
  1087. // handle egress gateway peers
  1088. if peer.IsEgressGateway {
  1089. //hasGateway = true
  1090. egressIPs := getEgressIPs(node, peer)
  1091. allowedips = append(allowedips, egressIPs...)
  1092. }
  1093. return allowedips
  1094. }
  1095. func getCIDRMaskFromAddr(addr string) net.IPMask {
  1096. cidr := net.CIDRMask(32, 32)
  1097. ipAddr, err := netip.ParseAddr(addr)
  1098. if err != nil {
  1099. return cidr
  1100. }
  1101. if ipAddr.Is6() {
  1102. cidr = net.CIDRMask(128, 128)
  1103. }
  1104. return cidr
  1105. }