saving netclient changes

netclient/functions/common.go

@@ -72,7 +72,7 @@ func GetFreePort(rangestart int32) (int32, error){
         return portno, err
 }
 
-func Install(accesskey string, password string, server string, network string, noauto bool, accesstoken string,  inputname string) error {
+func Install(accesskey string, password string, server string, network string, noauto bool, accesstoken string,  inputname string, dnsoff bool) error {
 
 	tserver := ""
 	tnetwork := ""
@@ -143,6 +143,8 @@ func Install(accesskey string, password string, server string, network string, n
 	servercfg := cfg.Server
 	fmt.Println("SERVER SETTINGS:")
 
+	nodecfg.DNSOff = dnsoff
+
 	if server == "" {
 		if servercfg.Address == "" && tserver == "" {
 			log.Fatal("no server provided")
@@ -588,6 +590,9 @@ func modConfig(node *nodepb.Node) error{
         if node.Address != ""{
                 nodecfg.WGAddress = node.Address
         }
+        if node.Address != ""{
+                nodecfg.WGAddress = node.Address
+        }
         if node.Postchanges != "" {
                 nodecfg.PostChanges = node.Postchanges
         }
@@ -952,6 +957,15 @@ func CheckIn(network string) error {
 	setupcheck := true
 	ipchange := false
 
+	if !nodecfg.DNSOff {
+		vals := strings.Split(servercfg.Address, ":")
+		server := vals[0]
+		err = SetDNS(server)
+		if err != nil {
+                        fmt.Printf("Error encountered setting dns: %v", err)
+		}
+	}
+
 	if !nodecfg.RoamingOff {
 		if !nodecfg.IsLocal {
 		fmt.Println("Checking to see if public addresses have changed")

netclient/functions/local.go

@@ -6,6 +6,7 @@ import (
         "io/ioutil"
 	"path/filepath"
         "io"
+	"strings"
         "log"
         "os"
         "os/exec"
@@ -20,6 +21,27 @@ func FileExists(f string) bool {
     return !info.IsDir()
 }
 
+func SetDNS(nameserver string) error {
+	bytes, err := ioutil.ReadFile("/etc/resolv.conf")
+	if err != nil {
+		return err
+	}
+	resolvstring := string(bytes)
+	// //check whether s contains substring text
+	hasdns := strings.Contains(resolvstring, nameserver)
+	if hasdns {
+		return nil
+	}
+	resolv, err := os.OpenFile("/etc/resolv.conf",os.O_APPEND|os.O_WRONLY, 0644)
+	if err != nil {
+		return err
+	}
+	defer resolv.Close()
+	_, err = resolv.WriteString("nameserver " + nameserver + "\n")
+
+	return err
+}
+
 func ConfigureSystemD(network string) error {
 	/*
 	path, err := os.Getwd()