Browse Source

minor cleanups

0xdcarns 3 years ago
parent
commit
90cbbfdf4a
5 changed files with 17 additions and 46 deletions
  1. 0 1
      controllers/nodeGrpcController.go
  2. 0 1
      docker/Dockerfile-netmaker-slim
  3. 3 0
      main.go
  4. 0 19
      models/node.go
  5. 14 25
      serverctl/serverctl.go

+ 0 - 1
controllers/nodeGrpcController.go

@@ -33,7 +33,6 @@ func (s *NodeServiceServer) ReadNode(ctx context.Context, req *nodepb.Object) (*
 		return nil, err
 		return nil, err
 	}
 	}
 	node.Update(&node)
 	node.Update(&node)
-	node.SetDefaults()
 	response := &nodepb.Object{
 	response := &nodepb.Object{
 		Data: string(nodeData),
 		Data: string(nodeData),
 		Type: nodepb.NODE_TYPE,
 		Type: nodepb.NODE_TYPE,

+ 0 - 1
docker/Dockerfile-netmaker-slim

@@ -11,7 +11,6 @@ RUN GOOS=linux GOARCH=amd64 CGO_ENABLED=1 /usr/local/go/bin/go build -ldflags="-
 
 
 WORKDIR /app/netclient
 WORKDIR /app/netclient
 
 
-
 RUN GOOS=linux GOARCH=amd64 CGO_ENABLED=0 /usr/local/go/bin/go build -ldflags="-w -s" -o netclient main.go
 RUN GOOS=linux GOARCH=amd64 CGO_ENABLED=0 /usr/local/go/bin/go build -ldflags="-w -s" -o netclient main.go
 #second stage
 #second stage
 
 

+ 3 - 0
main.go

@@ -53,6 +53,9 @@ func initialize() { // Client Mode Prereq Check
 		if uid != 0 {
 		if uid != 0 {
 			log.Fatal("To run in client mode requires root privileges. Either disable client mode or run with sudo.")
 			log.Fatal("To run in client mode requires root privileges. Either disable client mode or run with sudo.")
 		}
 		}
+		if err := serverctl.InitServerNetclient(); err != nil {
+			log.Fatal("Did not find netclient to use CLIENT_MODE")
+		}
 	}
 	}
 
 
 	if servercfg.IsDNSMode() {
 	if servercfg.IsDNSMode() {

+ 0 - 19
models/node.go

@@ -532,25 +532,6 @@ func GetAllNodes() ([]Node, error) {
 	return nodes, nil
 	return nodes, nil
 }
 }
 
 
-func GetNode(macaddress string, network string) (Node, error) {
-
-	var node Node
-
-	key, err := GetID(macaddress, network)
-	if err != nil {
-		return node, err
-	}
-	data, err := database.FetchRecord(database.NODES_TABLE_NAME, key)
-	if err != nil {
-		return node, err
-	}
-	if err = json.Unmarshal([]byte(data), &node); err != nil {
-		return node, err
-	}
-
-	return node, err
-}
-
 func GetID(macaddress string, network string) (string, error) {
 func GetID(macaddress string, network string) (string, error) {
 	if macaddress == "" || network == "" {
 	if macaddress == "" || network == "" {
 		return "", errors.New("unable to get record key")
 		return "", errors.New("unable to get record key")

+ 14 - 25
serverctl/serverctl.go

@@ -100,8 +100,7 @@ func RemoveNetwork(network string) (bool, error) {
 
 
 }
 }
 
 
-func HandleContainedClient() error {
-	log.SetFlags(log.Flags() &^ (log.Llongfile | log.Lshortfile))
+func InitServerNetclient() error {
 	netclientDir := ncutils.GetNetclientPath()
 	netclientDir := ncutils.GetNetclientPath()
 	netclientPath := ncutils.GetNetclientPathSpecific()
 	netclientPath := ncutils.GetNetclientPathSpecific()
 	_, err := os.Stat(netclientDir)
 	_, err := os.Stat(netclientDir)
@@ -123,11 +122,19 @@ func HandleContainedClient() error {
 		log.Println("could not change netclient binary permissions")
 		log.Println("could not change netclient binary permissions")
 		return err
 		return err
 	}
 	}
+	return nil
+}
+
+func HandleContainedClient() error {
+	log.SetFlags(log.Flags() &^ (log.Llongfile | log.Lshortfile))
 
 
+	netclientPath := ncutils.GetNetclientPathSpecific()
 	checkinCMD := exec.Command(netclientPath+"netclient", "checkin", "-n", "all")
 	checkinCMD := exec.Command(netclientPath+"netclient", "checkin", "-n", "all")
-	checkinCMD.Stdout = os.Stdout
+	if servercfg.GetVerbose() >= 2 {
+		checkinCMD.Stdout = os.Stdout
+	}
 	checkinCMD.Stderr = os.Stderr
 	checkinCMD.Stderr = os.Stderr
-	err = checkinCMD.Start()
+	err := checkinCMD.Start()
 	if err != nil {
 	if err != nil {
 		if servercfg.GetVerbose() >= 2 {
 		if servercfg.GetVerbose() >= 2 {
 			log.Println(err)
 			log.Println(err)
@@ -151,36 +158,18 @@ func AddNetwork(network string) (bool, error) {
 		log.Println("could not get public IP.")
 		log.Println("could not get public IP.")
 		return false, err
 		return false, err
 	}
 	}
-	netclientDir := ncutils.GetNetclientPath()
 	netclientPath := ncutils.GetNetclientPathSpecific()
 	netclientPath := ncutils.GetNetclientPathSpecific()
-	_, err = os.Stat(netclientDir)
-	if os.IsNotExist(err) {
-		os.Mkdir(netclientDir, 744)
-	} else if err != nil {
-		log.Println("could not find or create", netclientDir)
-		return false, err
-	}
+
 	token, err := functions.CreateServerToken(network)
 	token, err := functions.CreateServerToken(network)
 	if err != nil {
 	if err != nil {
 		log.Println("could not create server token for " + network)
 		log.Println("could not create server token for " + network)
 		return false, err
 		return false, err
 	}
 	}
-	_, err = os.Stat(netclientPath + "netclient")
-	if os.IsNotExist(err) {
-		err = InstallNetclient()
-		if err != nil {
-			return false, err
-		}
-	}
-	err = os.Chmod(netclientPath+"netclient", 0755)
-	if err != nil {
-		log.Println("could not change netclient directory permissions")
-		return false, err
-	}
+
 	functions.PrintUserLog(models.NODE_SERVER_NAME, "executing network join: "+netclientPath+"netclient "+"join "+"-t "+token+" -name "+models.NODE_SERVER_NAME+" -endpoint "+pubip, 0)
 	functions.PrintUserLog(models.NODE_SERVER_NAME, "executing network join: "+netclientPath+"netclient "+"join "+"-t "+token+" -name "+models.NODE_SERVER_NAME+" -endpoint "+pubip, 0)
 	var joinCMD *exec.Cmd
 	var joinCMD *exec.Cmd
 	if servercfg.IsClientMode() == "contained" {
 	if servercfg.IsClientMode() == "contained" {
-		joinCMD = exec.Command(netclientPath+"netclient", "join", "-t", token, "-name", models.NODE_SERVER_NAME, "-endpoint", pubip, "-daemon", "off")
+		joinCMD = exec.Command(netclientPath+"netclient", "join", "-t", token, "-name", models.NODE_SERVER_NAME, "-endpoint", pubip, "-daemon", "off", "-dnson", "no")
 	} else {
 	} else {
 		joinCMD = exec.Command(netclientPath+"netclient", "join", "-t", token, "-name", models.NODE_SERVER_NAME, "-endpoint", pubip)
 		joinCMD = exec.Command(netclientPath+"netclient", "join", "-t", token, "-name", models.NODE_SERVER_NAME, "-endpoint", pubip)
 	}
 	}