peers.go 35 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121
  1. package logic
  2. import (
  3. "errors"
  4. "fmt"
  5. "log"
  6. "net"
  7. "sort"
  8. "strconv"
  9. "strings"
  10. "time"
  11. proxy_models "github.com/gravitl/netclient/nmproxy/models"
  12. "github.com/gravitl/netmaker/database"
  13. "github.com/gravitl/netmaker/logger"
  14. "github.com/gravitl/netmaker/logic/acls/nodeacls"
  15. "github.com/gravitl/netmaker/models"
  16. "github.com/gravitl/netmaker/netclient/ncutils"
  17. "github.com/gravitl/netmaker/servercfg"
  18. "golang.org/x/exp/slices"
  19. "golang.zx2c4.com/wireguard/wgctrl/wgtypes"
  20. )
  21. // GetPeersforProxy calculates the peers for a proxy
  22. // TODO ==========================
  23. // TODO ==========================
  24. // TODO ==========================
  25. // TODO ==========================
  26. // TODO ==========================
  27. // revisit this logic with new host/node models.
  28. func GetPeersForProxy(node *models.Node, onlyPeers bool) (proxy_models.ProxyManagerPayload, error) {
  29. proxyPayload := proxy_models.ProxyManagerPayload{}
  30. var peers []wgtypes.PeerConfig
  31. peerConfMap := make(map[string]proxy_models.PeerConf)
  32. var err error
  33. currentPeers, err := GetNetworkNodes(node.Network)
  34. if err != nil {
  35. return proxyPayload, err
  36. }
  37. if !onlyPeers {
  38. if node.IsRelayed {
  39. relayNode := FindRelay(node)
  40. relayHost, err := GetHost(relayNode.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]proxy_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()] = proxy_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 = proxy_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()] = proxy_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()] = proxy_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) (proxy_models.ProxyManagerPayload, error) {
  180. proxyPayload := proxy_models.ProxyManagerPayload{
  181. Action: proxy_models.ProxyUpdate,
  182. }
  183. peerConfMap := make(map[string]proxy_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]proxy_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()] = proxy_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 proxy_models.PeerConf
  237. var found bool
  238. if currPeerConf, found = peerConfMap[peerHost.PublicKey.String()]; !found {
  239. currPeerConf = proxy_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(peerHost)))
  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. Network: make(map[string]models.NetworkInfo),
  277. PeerIDs: make(models.HostPeerMap),
  278. ServerVersion: servercfg.GetVersion(),
  279. ServerAddrs: []models.ServerAddr{},
  280. }
  281. logger.Log(1, "peer update for host ", host.ID.String())
  282. peerIndexMap := make(map[string]int)
  283. for _, nodeID := range host.Nodes {
  284. node, err := GetNodeByID(nodeID)
  285. if err != nil {
  286. continue
  287. }
  288. if !node.Connected || node.Action == models.NODE_DELETE || node.PendingDelete {
  289. continue
  290. }
  291. hostPeerUpdate.Network[node.Network] = models.NetworkInfo{
  292. DNS: getPeerDNS(node.Network),
  293. }
  294. currentPeers, err := GetNetworkNodes(node.Network)
  295. if err != nil {
  296. log.Println("no network nodes")
  297. return models.HostPeerUpdate{}, err
  298. }
  299. for _, peer := range currentPeers {
  300. if peer.ID == node.ID {
  301. logger.Log(2, "peer update, skipping self")
  302. //skip yourself
  303. continue
  304. }
  305. var peerConfig wgtypes.PeerConfig
  306. peerHost, err := GetHost(peer.HostID.String())
  307. if err != nil {
  308. logger.Log(1, "no peer host", peer.HostID.String(), err.Error())
  309. return models.HostPeerUpdate{}, err
  310. }
  311. if !peer.Connected || peer.Action == models.NODE_DELETE || peer.PendingDelete {
  312. logger.Log(2, "peer update, skipping unconnected node", peer.ID.String())
  313. //skip unconnected nodes
  314. continue
  315. }
  316. if !nodeacls.AreNodesAllowed(nodeacls.NetworkID(node.Network), nodeacls.NodeID(node.ID.String()), nodeacls.NodeID(peer.ID.String())) {
  317. log.Println("peer update, skipping node for acl")
  318. //skip if not permitted by acl
  319. continue
  320. }
  321. peerConfig.PublicKey = peerHost.PublicKey
  322. peerConfig.PersistentKeepaliveInterval = &peer.PersistentKeepalive
  323. peerConfig.ReplaceAllowedIPs = true
  324. uselocal := false
  325. if host.EndpointIP.String() == peerHost.EndpointIP.String() {
  326. //peer is on same network
  327. // set to localaddress
  328. uselocal = true
  329. if node.LocalAddress.IP == nil {
  330. // use public endpint
  331. uselocal = false
  332. }
  333. if node.LocalAddress.String() == peer.LocalAddress.String() {
  334. uselocal = false
  335. }
  336. }
  337. peerConfig.Endpoint = &net.UDPAddr{
  338. IP: peerHost.EndpointIP,
  339. Port: getPeerListenPort(peerHost),
  340. }
  341. if uselocal {
  342. peerConfig.Endpoint.IP = peer.LocalAddress.IP
  343. }
  344. allowedips := GetAllowedIPs(&node, &peer, nil)
  345. if peer.IsIngressGateway {
  346. for _, entry := range peer.IngressGatewayRange {
  347. _, cidr, err := net.ParseCIDR(string(entry))
  348. if err == nil {
  349. allowedips = append(allowedips, *cidr)
  350. }
  351. }
  352. }
  353. if peer.IsEgressGateway {
  354. allowedips = append(allowedips, getEgressIPs(&node, &peer)...)
  355. }
  356. peerConfig.AllowedIPs = allowedips
  357. if _, ok := hostPeerUpdate.PeerIDs[peerHost.PublicKey.String()]; !ok {
  358. hostPeerUpdate.PeerIDs[peerHost.PublicKey.String()] = make(map[string]models.IDandAddr)
  359. hostPeerUpdate.Peers = append(hostPeerUpdate.Peers, peerConfig)
  360. peerIndexMap[peerHost.PublicKey.String()] = len(hostPeerUpdate.Peers) - 1
  361. hostPeerUpdate.PeerIDs[peerHost.PublicKey.String()][peer.ID.String()] = models.IDandAddr{
  362. ID: peer.ID.String(),
  363. Address: peer.PrimaryAddress(),
  364. Name: peerHost.Name,
  365. Network: peer.Network,
  366. }
  367. } else {
  368. peerAllowedIPs := hostPeerUpdate.Peers[peerIndexMap[peerHost.PublicKey.String()]].AllowedIPs
  369. peerAllowedIPs = append(peerAllowedIPs, allowedips...)
  370. hostPeerUpdate.Peers[peerIndexMap[peerHost.PublicKey.String()]].AllowedIPs = peerAllowedIPs
  371. hostPeerUpdate.PeerIDs[peerHost.PublicKey.String()][peer.ID.String()] = models.IDandAddr{
  372. ID: peer.ID.String(),
  373. Address: peer.PrimaryAddress(),
  374. Name: peerHost.Name,
  375. Network: peer.Network,
  376. }
  377. }
  378. }
  379. if node.IsIngressGateway {
  380. extPeers, extPeerIDAndAddrs, err := getExtPeers(&node)
  381. if err == nil {
  382. hostPeerUpdate.Peers = append(hostPeerUpdate.Peers, extPeers...)
  383. hostPeerUpdate.ExtPeers = extPeers
  384. for _, extPeerIdAndAddr := range extPeerIDAndAddrs {
  385. hostPeerUpdate.PeerIDs[extPeerIdAndAddr.ID] = make(map[string]models.IDandAddr)
  386. hostPeerUpdate.PeerIDs[extPeerIdAndAddr.ID][extPeerIdAndAddr.ID] = models.IDandAddr{
  387. ID: extPeerIdAndAddr.ID,
  388. Address: extPeerIdAndAddr.Address,
  389. Name: extPeerIdAndAddr.Name,
  390. Network: node.Network,
  391. }
  392. }
  393. } else if !database.IsEmptyRecord(err) {
  394. logger.Log(1, "error retrieving external clients:", err.Error())
  395. }
  396. }
  397. }
  398. return hostPeerUpdate, nil
  399. }
  400. func getPeerListenPort(host *models.Host) int {
  401. peerPort := host.ListenPort
  402. if host.ProxyEnabled {
  403. if host.PublicListenPort != 0 {
  404. peerPort = host.PublicListenPort
  405. } else if host.ProxyListenPort != 0 {
  406. peerPort = host.ProxyListenPort
  407. }
  408. }
  409. return peerPort
  410. }
  411. // GetPeerUpdate - gets a wireguard peer config for each peer of a node
  412. func GetPeerUpdate(node *models.Node, host *models.Host) (models.PeerUpdate, error) {
  413. log.Println("peer update for node ", node.ID)
  414. peerUpdate := models.PeerUpdate{
  415. Network: node.Network,
  416. ServerVersion: ncutils.Version,
  417. DNS: getPeerDNS(node.Network),
  418. PeerIDs: make(models.PeerMap),
  419. }
  420. currentPeers, err := GetNetworkNodes(node.Network)
  421. if err != nil {
  422. log.Println("no network nodes")
  423. return models.PeerUpdate{}, err
  424. }
  425. for _, peer := range currentPeers {
  426. var peerConfig wgtypes.PeerConfig
  427. peerHost, err := GetHost(peer.HostID.String())
  428. if err != nil {
  429. log.Println("no peer host", err)
  430. return models.PeerUpdate{}, err
  431. }
  432. if peer.ID == node.ID {
  433. log.Println("peer update, skipping self")
  434. //skip yourself
  435. continue
  436. }
  437. if !peer.Connected {
  438. log.Println("peer update, skipping unconnected node")
  439. //skip unconnected nodes
  440. continue
  441. }
  442. if !nodeacls.AreNodesAllowed(nodeacls.NetworkID(node.Network), nodeacls.NodeID(node.ID.String()), nodeacls.NodeID(peer.ID.String())) {
  443. log.Println("peer update, skipping node for acl")
  444. //skip if not permitted by acl
  445. continue
  446. }
  447. peerConfig.PublicKey = peerHost.PublicKey
  448. peerConfig.PersistentKeepaliveInterval = &peer.PersistentKeepalive
  449. peerConfig.ReplaceAllowedIPs = true
  450. uselocal := false
  451. if host.EndpointIP.String() == peerHost.EndpointIP.String() {
  452. //peer is on same network
  453. // set to localaddress
  454. uselocal = true
  455. if node.LocalAddress.IP == nil {
  456. // use public endpint
  457. uselocal = false
  458. }
  459. if node.LocalAddress.String() == peer.LocalAddress.String() {
  460. uselocal = false
  461. }
  462. }
  463. peerConfig.Endpoint = &net.UDPAddr{
  464. IP: peerHost.EndpointIP,
  465. Port: peerHost.ListenPort,
  466. }
  467. if peerHost.ProxyEnabled {
  468. peerConfig.Endpoint.Port = getPeerListenPort(peerHost)
  469. }
  470. if uselocal {
  471. peerConfig.Endpoint.IP = peer.LocalAddress.IP
  472. }
  473. allowedips := getNodeAllowedIPs(&peer, node)
  474. if peer.IsIngressGateway {
  475. for _, entry := range peer.IngressGatewayRange {
  476. _, cidr, err := net.ParseCIDR(string(entry))
  477. if err == nil {
  478. allowedips = append(allowedips, *cidr)
  479. }
  480. }
  481. }
  482. if peer.IsEgressGateway {
  483. allowedips = append(allowedips, getEgressIPs(node, &peer)...)
  484. }
  485. peerUpdate.PeerIDs[peerHost.PublicKey.String()] = models.IDandAddr{
  486. ID: peer.ID.String(),
  487. Address: peer.PrimaryAddress(),
  488. Name: peerHost.Name,
  489. Network: peer.Network,
  490. }
  491. peerConfig.AllowedIPs = allowedips
  492. peerUpdate.Peers = append(peerUpdate.Peers, peerConfig)
  493. }
  494. return peerUpdate, nil
  495. }
  496. // func getRelayAllowedIPs(node, peer *models.Node) []net.IPNet {
  497. // var allowedips []net.IPNet
  498. // var allowedip net.IPNet
  499. // for _, addr := range peer.RelayAddrs {
  500. // if node.Address.IP.String() == addr {
  501. // continue
  502. // }
  503. // if node.Address6.IP.String() == addr {
  504. // continue
  505. // }
  506. // allowedip.IP = net.ParseIP(addr)
  507. // allowedips = append(allowedips, allowedip)
  508. // }
  509. // return allowedips
  510. // }
  511. // GetPeerUpdateLegacy - gets a wireguard peer config for each peer of a node
  512. func GetPeerUpdateLegacy(node *models.Node) (models.PeerUpdate, error) {
  513. var peerUpdate models.PeerUpdate
  514. var peers []wgtypes.PeerConfig
  515. var serverNodeAddresses = []models.ServerAddr{}
  516. var peerMap = make(models.PeerMap)
  517. var metrics *models.Metrics
  518. if servercfg.Is_EE {
  519. metrics, _ = GetMetrics(node.ID.String())
  520. }
  521. if metrics == nil {
  522. metrics = &models.Metrics{}
  523. }
  524. if metrics.FailoverPeers == nil {
  525. metrics.FailoverPeers = make(map[string]string)
  526. }
  527. // udppeers = the peers parsed from the local interface
  528. // gives us correct port to reach
  529. udppeers, errN := database.GetPeers(node.Network)
  530. if errN != nil {
  531. logger.Log(2, errN.Error())
  532. }
  533. currentPeers, err := GetNetworkNodes(node.Network)
  534. if err != nil {
  535. return models.PeerUpdate{}, err
  536. }
  537. host, err := GetHost(node.HostID.String())
  538. if err != nil {
  539. return peerUpdate, err
  540. }
  541. if node.IsRelayed && !host.ProxyEnabled {
  542. return GetPeerUpdateForRelayedNode(node, udppeers)
  543. }
  544. // #1 Set Keepalive values: set_keepalive
  545. // #2 Set local address: set_local - could be a LOT BETTER and fix some bugs with additional logic
  546. // #3 Set allowedips: set_allowedips
  547. for _, peer := range currentPeers {
  548. peerHost, err := GetHost(peer.HostID.String())
  549. if err != nil {
  550. logger.Log(0, "error retrieving host for peer", node.ID.String(), err.Error())
  551. return models.PeerUpdate{}, err
  552. }
  553. if peer.ID == node.ID {
  554. //skip yourself
  555. continue
  556. }
  557. if node.Connected {
  558. //skip unconnected nodes
  559. continue
  560. }
  561. // if the node is not a server, set the endpoint
  562. var setEndpoint = true
  563. if peer.IsRelayed {
  564. if !peerHost.ProxyEnabled && !(node.IsRelay && ncutils.StringSliceContains(node.RelayAddrs, peer.PrimaryAddress())) {
  565. //skip -- will be added to relay
  566. continue
  567. } else if node.IsRelay && ncutils.StringSliceContains(node.RelayAddrs, peer.PrimaryAddress()) {
  568. // dont set peer endpoint if it's relayed by node
  569. setEndpoint = false
  570. }
  571. }
  572. if !nodeacls.AreNodesAllowed(nodeacls.NetworkID(node.Network), nodeacls.NodeID(node.ID.String()), nodeacls.NodeID(peer.ID.String())) {
  573. //skip if not permitted by acl
  574. continue
  575. }
  576. if len(metrics.FailoverPeers[peer.ID.String()]) > 0 && IsFailoverPresent(node.Network) {
  577. logger.Log(2, "peer", peer.ID.String(), peer.PrimaryAddress(), "was found to be in failover peers list for node", node.ID.String(), node.PrimaryAddress())
  578. continue
  579. }
  580. if err != nil {
  581. return models.PeerUpdate{}, err
  582. }
  583. host, err := GetHost(node.HostID.String())
  584. if err != nil {
  585. logger.Log(0, "error retrieving host for node", node.ID.String(), err.Error())
  586. return models.PeerUpdate{}, err
  587. }
  588. if host.EndpointIP.String() == peerHost.EndpointIP.String() {
  589. //peer is on same network
  590. // set_local
  591. if node.LocalAddress.String() != peer.LocalAddress.String() && peer.LocalAddress.IP != nil {
  592. peerHost.EndpointIP = peer.LocalAddress.IP
  593. if peerHost.ListenPort != 0 {
  594. peerHost.ListenPort = getPeerListenPort(peerHost)
  595. }
  596. } else {
  597. continue
  598. }
  599. }
  600. // set address if setEndpoint is true
  601. // otherwise, will get inserted as empty value
  602. var address *net.UDPAddr
  603. // Sets ListenPort to UDP Hole Punching Port assuming:
  604. // - UDP Hole Punching is enabled
  605. // - udppeers retrieval did not return an error
  606. // - the endpoint is valid
  607. if setEndpoint {
  608. var setUDPPort = false
  609. if CheckEndpoint(udppeers[peerHost.PublicKey.String()]) {
  610. endpointstring := udppeers[peerHost.PublicKey.String()]
  611. endpointarr := strings.Split(endpointstring, ":")
  612. if len(endpointarr) == 2 {
  613. port, err := strconv.Atoi(endpointarr[1])
  614. if err == nil {
  615. setUDPPort = true
  616. peerHost.ListenPort = port
  617. }
  618. }
  619. }
  620. // if udp hole punching is on, but udp hole punching did not set it, use the LocalListenPort instead
  621. // or, if port is for some reason zero use the LocalListenPort
  622. // but only do this if LocalListenPort is not zero
  623. if ((!setUDPPort) || peerHost.ListenPort == 0) && peerHost.ListenPort != 0 {
  624. peerHost.ListenPort = getPeerListenPort(peerHost)
  625. }
  626. endpoint := peerHost.EndpointIP.String() + ":" + strconv.FormatInt(int64(peerHost.ListenPort), 10)
  627. address, err = net.ResolveUDPAddr("udp", endpoint)
  628. if err != nil {
  629. return models.PeerUpdate{}, err
  630. }
  631. }
  632. allowedips := GetAllowedIPs(node, &peer, metrics)
  633. var keepalive time.Duration
  634. if node.PersistentKeepalive != 0 {
  635. // set_keepalive
  636. keepalive = node.PersistentKeepalive
  637. }
  638. var peerData = wgtypes.PeerConfig{
  639. PublicKey: peerHost.PublicKey,
  640. Endpoint: address,
  641. ReplaceAllowedIPs: true,
  642. AllowedIPs: allowedips,
  643. PersistentKeepaliveInterval: &keepalive,
  644. }
  645. peers = append(peers, peerData)
  646. peerMap[peerHost.PublicKey.String()] = models.IDandAddr{
  647. Name: peerHost.Name,
  648. ID: peer.ID.String(),
  649. Address: peer.PrimaryAddress(),
  650. IsServer: "no",
  651. }
  652. }
  653. if node.IsIngressGateway {
  654. extPeers, idsAndAddr, err := getExtPeers(node)
  655. if err == nil {
  656. peers = append(peers, extPeers...)
  657. for i := range idsAndAddr {
  658. peerMap[idsAndAddr[i].ID] = idsAndAddr[i]
  659. }
  660. } else if !database.IsEmptyRecord(err) {
  661. logger.Log(1, "error retrieving external clients:", err.Error())
  662. }
  663. }
  664. peerUpdate.Network = node.Network
  665. peerUpdate.ServerVersion = servercfg.Version
  666. sort.SliceStable(peers[:], func(i, j int) bool {
  667. return peers[i].PublicKey.String() < peers[j].PublicKey.String()
  668. })
  669. peerUpdate.Peers = peers
  670. peerUpdate.ServerAddrs = serverNodeAddresses
  671. peerUpdate.DNS = getPeerDNS(node.Network)
  672. peerUpdate.PeerIDs = peerMap
  673. return peerUpdate, nil
  674. }
  675. func getExtPeers(node *models.Node) ([]wgtypes.PeerConfig, []models.IDandAddr, error) {
  676. var peers []wgtypes.PeerConfig
  677. var idsAndAddr []models.IDandAddr
  678. extPeers, err := GetNetworkExtClients(node.Network)
  679. if err != nil {
  680. return peers, idsAndAddr, err
  681. }
  682. host, err := GetHost(node.HostID.String())
  683. if err != nil {
  684. return peers, idsAndAddr, err
  685. }
  686. for _, extPeer := range extPeers {
  687. pubkey, err := wgtypes.ParseKey(extPeer.PublicKey)
  688. if err != nil {
  689. logger.Log(1, "error parsing ext pub key:", err.Error())
  690. continue
  691. }
  692. if host.PublicKey.String() == extPeer.PublicKey ||
  693. extPeer.IngressGatewayID != node.ID.String() {
  694. continue
  695. }
  696. var allowedips []net.IPNet
  697. var peer wgtypes.PeerConfig
  698. if extPeer.Address != "" {
  699. var peeraddr = net.IPNet{
  700. IP: net.ParseIP(extPeer.Address),
  701. Mask: net.CIDRMask(32, 32),
  702. }
  703. if peeraddr.IP != nil && peeraddr.Mask != nil {
  704. allowedips = append(allowedips, peeraddr)
  705. }
  706. }
  707. if extPeer.Address6 != "" {
  708. var addr6 = net.IPNet{
  709. IP: net.ParseIP(extPeer.Address6),
  710. Mask: net.CIDRMask(128, 128),
  711. }
  712. if addr6.IP != nil && addr6.Mask != nil {
  713. allowedips = append(allowedips, addr6)
  714. }
  715. }
  716. primaryAddr := extPeer.Address
  717. if primaryAddr == "" {
  718. primaryAddr = extPeer.Address6
  719. }
  720. peer = wgtypes.PeerConfig{
  721. PublicKey: pubkey,
  722. ReplaceAllowedIPs: true,
  723. AllowedIPs: allowedips,
  724. }
  725. peers = append(peers, peer)
  726. idsAndAddr = append(idsAndAddr, models.IDandAddr{
  727. ID: peer.PublicKey.String(),
  728. Name: extPeer.ClientID,
  729. Address: primaryAddr,
  730. })
  731. }
  732. return peers, idsAndAddr, nil
  733. }
  734. func getExtPeersForProxy(node *models.Node, proxyPeerConf map[string]proxy_models.PeerConf) ([]wgtypes.PeerConfig, map[string]proxy_models.PeerConf, error) {
  735. var peers []wgtypes.PeerConfig
  736. host, err := GetHost(node.HostID.String())
  737. if err != nil {
  738. logger.Log(0, "error retrieving host for node", node.ID.String(), err.Error())
  739. }
  740. extPeers, err := GetNetworkExtClients(node.Network)
  741. if err != nil {
  742. return peers, proxyPeerConf, err
  743. }
  744. for _, extPeer := range extPeers {
  745. pubkey, err := wgtypes.ParseKey(extPeer.PublicKey)
  746. if err != nil {
  747. logger.Log(1, "error parsing ext pub key:", err.Error())
  748. continue
  749. }
  750. if host.PublicKey.String() == extPeer.PublicKey ||
  751. extPeer.IngressGatewayID != node.ID.String() {
  752. continue
  753. }
  754. var allowedips []net.IPNet
  755. var peer wgtypes.PeerConfig
  756. if extPeer.Address != "" {
  757. var peeraddr = net.IPNet{
  758. IP: net.ParseIP(extPeer.Address),
  759. Mask: net.CIDRMask(32, 32),
  760. }
  761. if peeraddr.IP != nil && peeraddr.Mask != nil {
  762. allowedips = append(allowedips, peeraddr)
  763. }
  764. }
  765. if extPeer.Address6 != "" {
  766. var addr6 = net.IPNet{
  767. IP: net.ParseIP(extPeer.Address6),
  768. Mask: net.CIDRMask(128, 128),
  769. }
  770. if addr6.IP != nil && addr6.Mask != nil {
  771. allowedips = append(allowedips, addr6)
  772. }
  773. }
  774. peer = wgtypes.PeerConfig{
  775. PublicKey: pubkey,
  776. ReplaceAllowedIPs: true,
  777. AllowedIPs: allowedips,
  778. }
  779. extConf := proxy_models.PeerConf{
  780. IsExtClient: true,
  781. Address: net.ParseIP(extPeer.Address),
  782. }
  783. proxyPeerConf[peer.PublicKey.String()] = extConf
  784. peers = append(peers, peer)
  785. }
  786. return peers, proxyPeerConf, nil
  787. }
  788. // GetAllowedIPs - calculates the wireguard allowedip field for a peer of a node based on the peer and node settings
  789. func GetAllowedIPs(node, peer *models.Node, metrics *models.Metrics) []net.IPNet {
  790. var allowedips []net.IPNet
  791. allowedips = getNodeAllowedIPs(peer, node)
  792. // handle ingress gateway peers
  793. if peer.IsIngressGateway {
  794. extPeers, _, err := getExtPeers(peer)
  795. if err != nil {
  796. logger.Log(2, "could not retrieve ext peers for ", peer.ID.String(), err.Error())
  797. }
  798. for _, extPeer := range extPeers {
  799. allowedips = append(allowedips, extPeer.AllowedIPs...)
  800. }
  801. // if node is a failover node, add allowed ips from nodes it is handling
  802. if metrics != nil && peer.Failover && metrics.FailoverPeers != nil {
  803. // traverse through nodes that need handling
  804. logger.Log(3, "peer", peer.ID.String(), "was found to be failover for", node.ID.String(), "checking failover peers...")
  805. for k := range metrics.FailoverPeers {
  806. // if FailoverNode is me for this node, add allowedips
  807. if metrics.FailoverPeers[k] == peer.ID.String() {
  808. // get original node so we can traverse the allowed ips
  809. nodeToFailover, err := GetNodeByID(k)
  810. if err == nil {
  811. failoverNodeMetrics, err := GetMetrics(nodeToFailover.ID.String())
  812. if err == nil && failoverNodeMetrics != nil {
  813. if len(failoverNodeMetrics.NodeName) > 0 {
  814. allowedips = append(allowedips, getNodeAllowedIPs(&nodeToFailover, peer)...)
  815. logger.Log(0, "failing over node", nodeToFailover.ID.String(), nodeToFailover.PrimaryAddress(), "to failover node", peer.ID.String())
  816. }
  817. }
  818. }
  819. }
  820. }
  821. }
  822. }
  823. return allowedips
  824. }
  825. func getPeerDNS(network string) string {
  826. var dns string
  827. if nodes, err := GetNetworkNodes(network); err == nil {
  828. for i, node := range nodes {
  829. host, err := GetHost(node.HostID.String())
  830. if err != nil {
  831. logger.Log(0, "error retrieving host for node", node.ID.String(), err.Error())
  832. continue
  833. }
  834. dns = dns + fmt.Sprintf("%s %s.%s\n", nodes[i].Address, host.Name, nodes[i].Network)
  835. }
  836. }
  837. if customDNSEntries, err := GetCustomDNS(network); err == nil {
  838. for _, entry := range customDNSEntries {
  839. // TODO - filter entries based on ACLs / given peers vs nodes in network
  840. dns = dns + fmt.Sprintf("%s %s.%s\n", entry.Address, entry.Name, entry.Network)
  841. }
  842. }
  843. return dns
  844. }
  845. // GetPeerUpdateForRelayedNode - calculates peer update for a relayed node by getting the relay
  846. // copying the relay node's allowed ips and making appropriate substitutions
  847. func GetPeerUpdateForRelayedNode(node *models.Node, udppeers map[string]string) (models.PeerUpdate, error) {
  848. var peerUpdate models.PeerUpdate
  849. var peers []wgtypes.PeerConfig
  850. var serverNodeAddresses = []models.ServerAddr{}
  851. var allowedips []net.IPNet
  852. //find node that is relaying us
  853. relay := FindRelay(node)
  854. if relay == nil {
  855. return models.PeerUpdate{}, errors.New("not found")
  856. }
  857. //add relay to lists of allowed ip
  858. if relay.Address.IP != nil {
  859. relayIP := relay.Address
  860. allowedips = append(allowedips, relayIP)
  861. }
  862. if relay.Address6.IP != nil {
  863. relayIP6 := relay.Address6
  864. allowedips = append(allowedips, relayIP6)
  865. }
  866. //get PeerUpdate for relayed node
  867. relayHost, err := GetHost(relay.HostID.String())
  868. if err != nil {
  869. return models.PeerUpdate{}, err
  870. }
  871. relayPeerUpdate, err := GetPeerUpdate(relay, relayHost)
  872. if err != nil {
  873. return models.PeerUpdate{}, err
  874. }
  875. //add the relays allowed ips from all of the relay's peers
  876. for _, peer := range relayPeerUpdate.Peers {
  877. allowedips = append(allowedips, peer.AllowedIPs...)
  878. }
  879. //delete any ips not permitted by acl
  880. for i := len(allowedips) - 1; i >= 0; i-- {
  881. target, err := findNode(allowedips[i].IP.String())
  882. if err != nil {
  883. logger.Log(0, "failed to find node for ip", allowedips[i].IP.String(), err.Error())
  884. continue
  885. }
  886. if target == nil {
  887. logger.Log(0, "failed to find node for ip", allowedips[i].IP.String())
  888. continue
  889. }
  890. if !nodeacls.AreNodesAllowed(nodeacls.NetworkID(node.Network), nodeacls.NodeID(node.ID.String()), nodeacls.NodeID(target.ID.String())) {
  891. logger.Log(0, "deleting node from relayednode per acl", node.ID.String(), target.ID.String())
  892. allowedips = append(allowedips[:i], allowedips[i+1:]...)
  893. }
  894. }
  895. //delete self from allowed ips
  896. for i := len(allowedips) - 1; i >= 0; i-- {
  897. if allowedips[i].IP.String() == node.Address.IP.String() || allowedips[i].IP.String() == node.Address6.IP.String() {
  898. allowedips = append(allowedips[:i], allowedips[i+1:]...)
  899. }
  900. }
  901. //delete egressrange from allowedip if we are egress gateway
  902. if node.IsEgressGateway {
  903. for i := len(allowedips) - 1; i >= 0; i-- {
  904. if StringSliceContains(node.EgressGatewayRanges, allowedips[i].String()) {
  905. allowedips = append(allowedips[:i], allowedips[i+1:]...)
  906. }
  907. }
  908. }
  909. //delete extclients from allowedip if we are ingress gateway
  910. if node.IsIngressGateway {
  911. for i := len(allowedips) - 1; i >= 0; i-- {
  912. if strings.Contains(node.IngressGatewayRange, allowedips[i].IP.String()) {
  913. allowedips = append(allowedips[:i], allowedips[i+1:]...)
  914. }
  915. }
  916. }
  917. //add egress range if relay is egress
  918. if relay.IsEgressGateway {
  919. var ip *net.IPNet
  920. for _, cidr := range relay.EgressGatewayRanges {
  921. _, ip, err = net.ParseCIDR(cidr)
  922. if err != nil {
  923. continue
  924. }
  925. }
  926. allowedips = append(allowedips, *ip)
  927. }
  928. var setUDPPort = false
  929. var listenPort int
  930. if CheckEndpoint(udppeers[relayHost.PublicKey.String()]) {
  931. endpointstring := udppeers[relayHost.PublicKey.String()]
  932. endpointarr := strings.Split(endpointstring, ":")
  933. if len(endpointarr) == 2 {
  934. port, err := strconv.Atoi(endpointarr[1])
  935. if err == nil {
  936. setUDPPort = true
  937. listenPort = port
  938. }
  939. }
  940. }
  941. // if udp hole punching is on, but udp hole punching did not set it, use the LocalListenPort instead
  942. // or, if port is for some reason zero use the LocalListenPort
  943. // but only do this if LocalListenPort is not zero
  944. if ((!setUDPPort) || relayHost.ListenPort == 0) && relayHost.ListenPort != 0 {
  945. listenPort = relayHost.ListenPort
  946. }
  947. endpoint := relayHost.EndpointIP.String() + ":" + strconv.FormatInt(int64(listenPort), 10)
  948. address, err := net.ResolveUDPAddr("udp", endpoint)
  949. if err != nil {
  950. return models.PeerUpdate{}, err
  951. }
  952. var keepalive time.Duration
  953. if node.PersistentKeepalive != 0 {
  954. // set_keepalive
  955. keepalive = node.PersistentKeepalive
  956. }
  957. var peerData = wgtypes.PeerConfig{
  958. PublicKey: relayHost.PublicKey,
  959. Endpoint: address,
  960. ReplaceAllowedIPs: true,
  961. AllowedIPs: allowedips,
  962. PersistentKeepaliveInterval: &keepalive,
  963. }
  964. peers = append(peers, peerData)
  965. //if ingress add extclients
  966. if node.IsIngressGateway {
  967. extPeers, _, err := getExtPeers(node)
  968. if err == nil {
  969. peers = append(peers, extPeers...)
  970. } else {
  971. logger.Log(2, "could not retrieve ext peers for ", node.ID.String(), err.Error())
  972. }
  973. }
  974. peerUpdate.Network = node.Network
  975. peerUpdate.ServerVersion = servercfg.Version
  976. sort.SliceStable(peers[:], func(i, j int) bool {
  977. return peers[i].PublicKey.String() < peers[j].PublicKey.String()
  978. })
  979. peerUpdate.Peers = peers
  980. peerUpdate.ServerAddrs = serverNodeAddresses
  981. peerUpdate.DNS = getPeerDNS(node.Network)
  982. return peerUpdate, nil
  983. }
  984. func getEgressIPs(node, peer *models.Node) []net.IPNet {
  985. host, err := GetHost(node.HostID.String())
  986. if err != nil {
  987. logger.Log(0, "error retrieving host for node", node.ID.String(), err.Error())
  988. }
  989. peerHost, err := GetHost(peer.HostID.String())
  990. if err != nil {
  991. logger.Log(0, "error retrieving host for peer", peer.ID.String(), err.Error())
  992. }
  993. //check for internet gateway
  994. internetGateway := false
  995. if slices.Contains(peer.EgressGatewayRanges, "0.0.0.0/0") || slices.Contains(peer.EgressGatewayRanges, "::/0") {
  996. internetGateway = true
  997. }
  998. allowedips := []net.IPNet{}
  999. for _, iprange := range peer.EgressGatewayRanges { // go through each cidr for egress gateway
  1000. _, ipnet, err := net.ParseCIDR(iprange) // confirming it's valid cidr
  1001. if err != nil {
  1002. logger.Log(1, "could not parse gateway IP range. Not adding ", iprange)
  1003. continue // if can't parse CIDR
  1004. }
  1005. nodeEndpointArr := strings.Split(peerHost.EndpointIP.String(), ":") // getting the public ip of node
  1006. if ipnet.Contains(net.ParseIP(nodeEndpointArr[0])) && !internetGateway { // ensuring egress gateway range does not contain endpoint of node
  1007. logger.Log(2, "egress IP range of ", iprange, " overlaps with ", host.EndpointIP.String(), ", omitting")
  1008. continue // skip adding egress range if overlaps with node's ip
  1009. }
  1010. // TODO: Could put in a lot of great logic to avoid conflicts / bad routes
  1011. if ipnet.Contains(net.ParseIP(node.LocalAddress.String())) && !internetGateway { // ensuring egress gateway range does not contain public ip of node
  1012. logger.Log(2, "egress IP range of ", iprange, " overlaps with ", node.LocalAddress.String(), ", omitting")
  1013. continue // skip adding egress range if overlaps with node's local ip
  1014. }
  1015. if err != nil {
  1016. logger.Log(1, "error encountered when setting egress range", err.Error())
  1017. } else {
  1018. allowedips = append(allowedips, *ipnet)
  1019. }
  1020. }
  1021. return allowedips
  1022. }
  1023. func getNodeAllowedIPs(peer, node *models.Node) []net.IPNet {
  1024. var allowedips = []net.IPNet{}
  1025. if peer.Address.IP != nil {
  1026. allowed := net.IPNet{
  1027. IP: peer.Address.IP,
  1028. Mask: net.CIDRMask(32, 32),
  1029. }
  1030. allowedips = append(allowedips, allowed)
  1031. }
  1032. if peer.Address6.IP != nil {
  1033. allowed := net.IPNet{
  1034. IP: peer.Address6.IP,
  1035. Mask: net.CIDRMask(128, 128),
  1036. }
  1037. allowedips = append(allowedips, allowed)
  1038. }
  1039. // handle egress gateway peers
  1040. if peer.IsEgressGateway {
  1041. //hasGateway = true
  1042. egressIPs := getEgressIPs(node, peer)
  1043. allowedips = append(allowedips, egressIPs...)
  1044. }
  1045. return allowedips
  1046. }