From 62208360505f92a54248f88db2571910a758e789 Mon Sep 17 00:00:00 2001
From: Csaba Sarkadi <sarkadicsa@tutanota.de>
Date: Sat, 29 Jan 2022 15:26:28 +0100
Subject: [PATCH] utils: extract GetIPPrefixEndpoints from anonymous function

---
 utils.go | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/utils.go b/utils.go
index 3b18a40..babc69f 100644
--- a/utils.go
+++ b/utils.go
@@ -147,6 +147,13 @@ func (h *Headscale) getAvailableIPs() (ips MachineAddresses, err error) {
 	return
 }
 
+func GetIPPrefixEndpoints(na netaddr.IPPrefix) (network, broadcast netaddr.IP) {
+	ipRange := na.Range()
+	network = ipRange.From()
+	broadcast = ipRange.To()
+	return
+}
+
 // TODO: Is this concurrency safe?
 // What would happen if multiple hosts were to register at the same time?
 // Would we attempt to assign the same addresses to multiple nodes?
@@ -156,11 +163,7 @@ func (h *Headscale) getAvailableIP(ipPrefix netaddr.IPPrefix) (*netaddr.IP, erro
 		return nil, err
 	}
 
-	ipPrefixNetworkAddress, ipPrefixBroadcastAddress := func() (netaddr.IP, netaddr.IP) {
-		ipRange := ipPrefix.Range()
-
-		return ipRange.From(), ipRange.To()
-	}()
+	ipPrefixNetworkAddress, ipPrefixBroadcastAddress := GetIPPrefixEndpoints(ipPrefix)
 
 	// Get the first IP in our prefix
 	ip := ipPrefixNetworkAddress.Next()