zishuo/netmaker
1
0
Fork 0
mirror of https://github.com/gravitl/netmaker.git synced 2025-10-05 16:57:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

added local range logic

Browse Source
This commit is contained in:
0xdcarns
2022-01-12 19:39:10 -05:00
parent 685689df1d
commit 72206817bb
3 changed files with 38 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
logic/server.go
View File

@@ -26,9 +26,9 @@ const KUBERNETES_LISTEN_PORT = 31821
const KUBERNETES_SERVER_MTU = 1024
// ServerJoin - responsible for joining a server to a network
func ServerJoin(network string, serverID string, privateKey string) error {
func ServerJoin(networkSettings *models.Network, serverID string) error {
if network == "" {
if networkSettings == nil || networkSettings.NetID == "" {
return errors.New("no network provided")
}
@@ -54,6 +54,15 @@ func ServerJoin(network string, serverID string, privateKey string) error {
}
if node.Endpoint == "" {
if node.IsLocal == "yes" {
var localAddr, localErr = getServerLocalIP(networkSettings)
if localErr != nil {
logger.Log(1, "could not acquire local address", localErr.Error())
} else {
node.LocalAddress = localAddr
node.LocalRange = networkSettings.LocalRange
}
}
if node.IsLocal == "yes" && node.LocalAddress != "" {
node.Endpoint = node.LocalAddress
} else {
@@ -65,6 +74,8 @@ func ServerJoin(network string, serverID string, privateKey string) error {
}
}
var privateKey = ""
// Generate and set public/private WireGuard Keys
if privateKey == "" {
wgPrivatekey, err := wgtypes.GeneratePrivateKey()
@@ -76,7 +87,7 @@ func ServerJoin(network string, serverID string, privateKey string) error {
node.PublicKey = wgPrivatekey.PublicKey().String()
}
node.Network = network
node.Network = networkSettings.NetID
logger.Log(2, "adding a server instance on network", node.Network)
err = CreateNode(node)
@@ -421,3 +432,23 @@ func checkNodeActions(node *models.Node) string {
}
return ""
}
// best effort, or get public ip
func getServerLocalIP(networkSettings *models.Network) (string, error) {
var networkCIDR = networkSettings.AddressRange
var currentAddresses, err = net.InterfaceAddrs()
if err != nil { // attempt to use public IP
return "", err
}
var _, currentCIDR, cidrErr = net.ParseCIDR(networkCIDR)
if cidrErr != nil {
return "", err
}
for _, address := range currentAddresses {
if currentCIDR.Contains(net.IP(address.Network())) {
logger.Log(1, "setting local ip", address.String())
return address.String(), nil
}
}
return "", errors.New("could not find local ip")
}