Browse Source

adjusted values to pointers for a lot of functions

0xdcarns 3 years ago
parent
commit
6215ea2db8

+ 4 - 4
controllers/common_test.go

@@ -25,7 +25,7 @@ func TestGetPeerList(t *testing.T) {
 		assert.Equal(t, node.Address, peers[0].Address)
 		assert.Equal(t, node.Address, peers[0].Address)
 	})
 	})
 	t.Run("Multiple Nodes", func(t *testing.T) {
 	t.Run("Multiple Nodes", func(t *testing.T) {
-		createnode := models.Node{PublicKey: "RM5qhLAE20PG9BbfBCger+Ac9D2NDOwCtY1rbYDLf34=", Endpoint: "10.0.0.2", MacAddress: "02:02:03:04:05:06", Password: "password", Network: "skynet"}
+		createnode := &models.Node{PublicKey: "RM5qhLAE20PG9BbfBCger+Ac9D2NDOwCtY1rbYDLf34=", Endpoint: "10.0.0.2", MacAddress: "02:02:03:04:05:06", Password: "password", Network: "skynet"}
 		logic.CreateNode(createnode, "skynet")
 		logic.CreateNode(createnode, "skynet")
 		peers, err := logic.GetPeersList("skynet", false, "")
 		peers, err := logic.GetPeersList("skynet", false, "")
 		assert.Nil(t, err)
 		assert.Nil(t, err)
@@ -95,7 +95,7 @@ func TestCreateNode(t *testing.T) {
 	database.InitializeDatabase()
 	database.InitializeDatabase()
 	deleteAllNetworks()
 	deleteAllNetworks()
 	createNet()
 	createNet()
-	createnode := models.Node{PublicKey: "DM5qhLAE20PG9BbfBCger+Ac9D2NDOwCtY1rbYDLf34=", Endpoint: "10.0.0.1", MacAddress: "01:02:03:04:05:06", Password: "password", Network: "skynet"}
+	createnode := &models.Node{PublicKey: "DM5qhLAE20PG9BbfBCger+Ac9D2NDOwCtY1rbYDLf34=", Endpoint: "10.0.0.1", MacAddress: "01:02:03:04:05:06", Password: "password", Network: "skynet"}
 	//err := ValidateNodeCreate("skynet", createnode)
 	//err := ValidateNodeCreate("skynet", createnode)
 	//assert.Nil(t, err)
 	//assert.Nil(t, err)
 	node, err := logic.CreateNode(createnode, "skynet")
 	node, err := logic.CreateNode(createnode, "skynet")
@@ -123,8 +123,8 @@ func TestSetNetworkNodesLastModified(t *testing.T) {
 	})
 	})
 }
 }
 
 
-func createTestNode() models.Node {
+func createTestNode() *models.Node {
 	createnode := models.Node{PublicKey: "DM5qhLAE20PG9BbfBCger+Ac9D2NDOwCtY1rbYDLf34=", Name: "testnode", Endpoint: "10.0.0.1", MacAddress: "01:02:03:04:05:06", Password: "password", Network: "skynet"}
 	createnode := models.Node{PublicKey: "DM5qhLAE20PG9BbfBCger+Ac9D2NDOwCtY1rbYDLf34=", Name: "testnode", Endpoint: "10.0.0.1", MacAddress: "01:02:03:04:05:06", Password: "password", Network: "skynet"}
-	node, _ := logic.CreateNode(createnode, "skynet")
+	node, _ := logic.CreateNode(&createnode, "skynet")
 	return node
 	return node
 }
 }

+ 1 - 1
controllers/dns_test.go

@@ -54,7 +54,7 @@ func TestGetNodeDNS(t *testing.T) {
 		assert.Equal(t, "10.0.0.1", dns[0].Address)
 		assert.Equal(t, "10.0.0.1", dns[0].Address)
 	})
 	})
 	t.Run("MultipleNodes", func(t *testing.T) {
 	t.Run("MultipleNodes", func(t *testing.T) {
-		createnode := models.Node{PublicKey: "DM5qhLAE20PG9BbfBCger+Ac9D2NDOwCtY1rbYDLf34=", Endpoint: "10.100.100.3", MacAddress: "01:02:03:04:05:07", Password: "password", Network: "skynet"}
+		createnode := &models.Node{PublicKey: "DM5qhLAE20PG9BbfBCger+Ac9D2NDOwCtY1rbYDLf34=", Endpoint: "10.100.100.3", MacAddress: "01:02:03:04:05:07", Password: "password", Network: "skynet"}
 		_, err := logic.CreateNode(createnode, "skynet")
 		_, err := logic.CreateNode(createnode, "skynet")
 		assert.Nil(t, err)
 		assert.Nil(t, err)
 		dns, err := logic.GetNodeDNS("skynet")
 		dns, err := logic.GetNodeDNS("skynet")

+ 3 - 3
controllers/node.go

@@ -379,10 +379,10 @@ func createNode(w http.ResponseWriter, r *http.Request) {
 		return
 		return
 	}
 	}
 
 
-	var node models.Node
+	var node = &models.Node{}
 
 
 	//get node from body of request
 	//get node from body of request
-	err = json.NewDecoder(r.Body).Decode(&node)
+	err = json.NewDecoder(r.Body).Decode(node)
 	if err != nil {
 	if err != nil {
 		returnErrorResponse(w, r, formatError(err, "internal"))
 		returnErrorResponse(w, r, formatError(err, "internal"))
 		return
 		return
@@ -390,7 +390,7 @@ func createNode(w http.ResponseWriter, r *http.Request) {
 
 
 	node.Network = networkName
 	node.Network = networkName
 
 
-	network, err := logic.GetNetworkByNode(&node)
+	network, err := logic.GetNetworkByNode(node)
 	if err != nil {
 	if err != nil {
 		returnErrorResponse(w, r, formatError(err, "internal"))
 		returnErrorResponse(w, r, formatError(err, "internal"))
 		return
 		return

+ 2 - 2
controllers/node_grpc.go

@@ -51,9 +51,9 @@ func (s *NodeServiceServer) ReadNode(ctx context.Context, req *nodepb.Object) (*
 func (s *NodeServiceServer) CreateNode(ctx context.Context, req *nodepb.Object) (*nodepb.Object, error) {
 func (s *NodeServiceServer) CreateNode(ctx context.Context, req *nodepb.Object) (*nodepb.Object, error) {
 	// Get the protobuf node type from the protobuf request type
 	// Get the protobuf node type from the protobuf request type
 	// Essentially doing req.Node to access the struct with a nil check
 	// Essentially doing req.Node to access the struct with a nil check
-	var node models.Node
+	var node = &models.Node{}
 	data := req.GetData()
 	data := req.GetData()
-	if err := json.Unmarshal([]byte(data), &node); err != nil {
+	if err := json.Unmarshal([]byte(data), node); err != nil {
 		return nil, err
 		return nil, err
 	}
 	}
 
 

+ 15 - 30
logic/server.go

@@ -31,8 +31,7 @@ func ServerJoin(network string, serverID string, privateKey string) error {
 	}
 	}
 
 
 	var err error
 	var err error
-	var node *models.Node // fill this object with server node specifics
-	node = &models.Node{
+	var node = &models.Node{
 		IsServer:     "yes",
 		IsServer:     "yes",
 		DNSOn:        "no",
 		DNSOn:        "no",
 		IsStatic:     "yes",
 		IsStatic:     "yes",
@@ -74,30 +73,11 @@ func ServerJoin(network string, serverID string, privateKey string) error {
 		privateKey = wgPrivatekey.String()
 		privateKey = wgPrivatekey.String()
 		node.PublicKey = wgPrivatekey.PublicKey().String()
 		node.PublicKey = wgPrivatekey.PublicKey().String()
 	}
 	}
-	// should never set mac address for server anymore
-
-	var postnode *models.Node
-	postnode = &models.Node{
-		Password:            node.Password,
-		MacAddress:          node.MacAddress,
-		AccessKey:           node.AccessKey,
-		Network:             network,
-		ListenPort:          node.ListenPort,
-		PostUp:              node.PostUp,
-		PostDown:            node.PostDown,
-		PersistentKeepalive: node.PersistentKeepalive,
-		LocalAddress:        node.LocalAddress,
-		Interface:           node.Interface,
-		PublicKey:           node.PublicKey,
-		DNSOn:               node.DNSOn,
-		Name:                node.Name,
-		Endpoint:            node.Endpoint,
-		SaveConfig:          node.SaveConfig,
-		UDPHolePunch:        node.UDPHolePunch,
-	}
 
 
-	logger.Log(2, "adding a server instance on network", postnode.Network)
-	*node, err = CreateNode(*postnode, network)
+	node.Network = network
+
+	logger.Log(2, "adding a server instance on network", node.Network)
+	node, err = CreateNode(node, network)
 	if err != nil {
 	if err != nil {
 		return err
 		return err
 	}
 	}
@@ -194,7 +174,7 @@ func ServerPull(serverNode *models.Node, onErr bool) (*models.Node, error) {
 			logger.Log(1, "removed old interface", oldIfaceName)
 			logger.Log(1, "removed old interface", oldIfaceName)
 		}
 		}
 		serverNode.PullChanges = "no"
 		serverNode.PullChanges = "no"
-		if err = setWGConfig(*serverNode, serverNode.Network, false); err != nil {
+		if err = setWGConfig(serverNode, serverNode.Network, false); err != nil {
 			return serverNode, err
 			return serverNode, err
 		}
 		}
 		// handle server side update
 		// handle server side update
@@ -202,7 +182,7 @@ func ServerPull(serverNode *models.Node, onErr bool) (*models.Node, error) {
 			return serverNode, err
 			return serverNode, err
 		}
 		}
 	} else {
 	} else {
-		if err = setWGConfig(*serverNode, serverNode.Network, true); err != nil {
+		if err = setWGConfig(serverNode, serverNode.Network, true); err != nil {
 			if errors.Is(err, os.ErrNotExist) {
 			if errors.Is(err, os.ErrNotExist) {
 				return ServerPull(serverNode, true)
 				return ServerPull(serverNode, true)
 			} else {
 			} else {
@@ -234,6 +214,11 @@ func ServerLeave(mac string, network string) error {
 	return DeleteNode(&serverNode, true)
 	return DeleteNode(&serverNode, true)
 }
 }
 
 
+/**
+ * Below function needs major refactor
+ *
+ */
+
 // GetServerPeers - gets peers of server
 // GetServerPeers - gets peers of server
 func GetServerPeers(macaddress string, network string, dualstack bool, isIngressGateway bool) ([]wgtypes.PeerConfig, bool, []string, error) {
 func GetServerPeers(macaddress string, network string, dualstack bool, isIngressGateway bool) ([]wgtypes.PeerConfig, bool, []string, error) {
 	hasGateway := false
 	hasGateway := false
@@ -373,7 +358,7 @@ func GetServerPeers(macaddress string, network string, dualstack bool, isIngress
 			logger.Log(1, "ERROR RETRIEVING EXTERNAL PEERS ON SERVER:", err.Error())
 			logger.Log(1, "ERROR RETRIEVING EXTERNAL PEERS ON SERVER:", err.Error())
 		}
 		}
 	}
 	}
-	return peers, hasGateway, gateways, err
+	return peers[:], hasGateway, gateways[:], err
 }
 }
 
 
 // GetServerExtPeers - gets the extpeers for a client
 // GetServerExtPeers - gets the extpeers for a client
@@ -436,7 +421,7 @@ func GetServerExtPeers(macaddress string, network string, dualstack bool) ([]wgt
 		}
 		}
 		peers = append(peers, peer)
 		peers = append(peers, peer)
 	}
 	}
-	return peers, err
+	return peers[:], err
 }
 }
 
 
 // == Private ==
 // == Private ==
@@ -448,7 +433,7 @@ func isDeleteError(err error) bool {
 func checkNodeActions(node *models.Node, networkName string, localNode *models.Node) string {
 func checkNodeActions(node *models.Node, networkName string, localNode *models.Node) string {
 	if (node.Action == models.NODE_UPDATE_KEY || localNode.Action == models.NODE_UPDATE_KEY) &&
 	if (node.Action == models.NODE_UPDATE_KEY || localNode.Action == models.NODE_UPDATE_KEY) &&
 		node.IsStatic != "yes" {
 		node.IsStatic != "yes" {
-		err := setWGKeyConfig(*node)
+		err := setWGKeyConfig(node)
 		if err != nil {
 		if err != nil {
 			logger.Log(1, "unable to process reset keys request:", err.Error())
 			logger.Log(1, "unable to process reset keys request:", err.Error())
 			return ""
 			return ""

+ 3 - 3
logic/util.go

@@ -76,7 +76,7 @@ func DeleteNode(node *models.Node, exterminate bool) error {
 }
 }
 
 
 // CreateNode - creates a node in database
 // CreateNode - creates a node in database
-func CreateNode(node models.Node, networkName string) (models.Node, error) {
+func CreateNode(node *models.Node, networkName string) (*models.Node, error) {
 
 
 	//encrypt that password so we never see it
 	//encrypt that password so we never see it
 	hash, err := bcrypt.GenerateFromPassword([]byte(node.Password), 5)
 	hash, err := bcrypt.GenerateFromPassword([]byte(node.Password), 5)
@@ -98,7 +98,7 @@ func CreateNode(node models.Node, networkName string) (models.Node, error) {
 			node.DNSOn = "no"
 			node.DNSOn = "no"
 		}
 		}
 	}
 	}
-	SetNodeDefaults(&node)
+	SetNodeDefaults(node)
 	node.Address, err = UniqueAddress(networkName)
 	node.Address, err = UniqueAddress(networkName)
 	if err != nil {
 	if err != nil {
 		return node, err
 		return node, err
@@ -113,7 +113,7 @@ func CreateNode(node models.Node, networkName string) (models.Node, error) {
 		//returnErrorResponse(w, r, errorResponse)
 		//returnErrorResponse(w, r, errorResponse)
 		return node, err
 		return node, err
 	}
 	}
-	err = ValidateNode(&node, false)
+	err = ValidateNode(node, false)
 	if err != nil {
 	if err != nil {
 		return node, err
 		return node, err
 	}
 	}

+ 3 - 3
logic/wireguard.go

@@ -280,7 +280,7 @@ func setServerPeers(iface string, keepalive int32, peers []wgtypes.PeerConfig) e
 	return nil
 	return nil
 }
 }
 
 
-func setWGConfig(node models.Node, network string, peerupdate bool) error {
+func setWGConfig(node *models.Node, network string, peerupdate bool) error {
 
 
 	node.SetID()
 	node.SetID()
 	peers, hasGateway, gateways, err := GetServerPeers(node.MacAddress, node.Network, node.IsDualStack == "yes", node.IsIngressGateway == "yes")
 	peers, hasGateway, gateways, err := GetServerPeers(node.MacAddress, node.Network, node.IsDualStack == "yes", node.IsIngressGateway == "yes")
@@ -296,13 +296,13 @@ func setWGConfig(node models.Node, network string, peerupdate bool) error {
 		err = setServerPeers(iface, node.PersistentKeepalive, peers)
 		err = setServerPeers(iface, node.PersistentKeepalive, peers)
 		logger.Log(2, "updated peers on server", node.Name)
 		logger.Log(2, "updated peers on server", node.Name)
 	} else {
 	} else {
-		err = initWireguard(&node, privkey, peers, hasGateway, gateways)
+		err = initWireguard(node, privkey, peers, hasGateway, gateways)
 		logger.Log(3, "finished setting wg config on server", node.Name)
 		logger.Log(3, "finished setting wg config on server", node.Name)
 	}
 	}
 	return err
 	return err
 }
 }
 
 
-func setWGKeyConfig(node models.Node) error {
+func setWGKeyConfig(node *models.Node) error {
 
 
 	node.SetID()
 	node.SetID()
 	privatekey, err := wgtypes.GeneratePrivateKey()
 	privatekey, err := wgtypes.GeneratePrivateKey()

+ 1 - 1
serverctl/serverctl.go

@@ -77,7 +77,7 @@ func HandleContainedClient() error {
 				logger.Log(3, "completed peers check of network", serverNet.NetID)
 				logger.Log(3, "completed peers check of network", serverNet.NetID)
 			}
 			}
 		}
 		}
-		err := SyncNetworks(servernets)
+		err := SyncNetworks(servernets[:])
 		if err != nil {
 		if err != nil {
 			logger.Log(1, "error syncing networks:", err.Error())
 			logger.Log(1, "error syncing networks:", err.Error())
 		}
 		}