package serverctl import ( "encoding/json" "errors" "net" "os" "strings" "github.com/gravitl/netmaker/database" "github.com/gravitl/netmaker/logger" "github.com/gravitl/netmaker/logic" "github.com/gravitl/netmaker/models" "github.com/gravitl/netmaker/netclient/ncutils" "github.com/gravitl/netmaker/servercfg" ) // GetServerWGConf - gets the server WG configuration func GetServerWGConf() (models.IntClient, error) { var server models.IntClient collection, err := database.FetchRecords(database.INT_CLIENTS_TABLE_NAME) if err != nil { return models.IntClient{}, errors.New("could not find comms server") } for _, value := range collection { json.Unmarshal([]byte(value), &server) if server.Network == "comms" && server.IsServer == "yes" { return server, nil } } return models.IntClient{}, errors.New("could not find comms server") } // FileExists - checks if local file exists func FileExists(f string) bool { info, err := os.Stat(f) if os.IsNotExist(err) { return false } return !info.IsDir() } // RemoveNetwork - removes a network locally on server func RemoveNetwork(network string) (bool, error) { err := logic.ServerLeave(network) return true, err } // InitServerNetclient - intializes the server netclient func InitServerNetclient() error { netclientDir := ncutils.GetNetclientPath() _, err := os.Stat(netclientDir + "/config") if os.IsNotExist(err) { os.MkdirAll(netclientDir+"/config", 0744) } else if err != nil { logger.Log(1, "could not find or create", netclientDir) return err } return nil } // HandleContainedClient - function for checkins on server func HandleContainedClient() error { servernets, err := logic.GetNetworks() if err != nil && !database.IsEmptyRecord(err) { return err } if len(servernets) > 0 { if err != nil { return err } for _, serverNet := range servernets { var serverID, err = logic.GetNetworkServerNodeID(serverNet.NetID) if err != nil { logger.Log(1, "error occurred during server checkin:", err.Error()) continue } err = logic.ServerCheckin(serverID, servercfg.GetNodeID(), serverNet.NetID) if err != nil { logger.Log(1, "error occurred during server checkin:", err.Error()) } else { logger.Log(3, "completed peers check of network", serverNet.NetID) } } syncErr := SyncNetworks(servernets[:]) if syncErr != nil { logger.Log(1, "error syncing networks:", syncErr.Error()) syncErr = nil } // logger.Log("completed a checkin call", 3) } return nil } // SyncNetworks - syncs the networks for servers func SyncNetworks(servernets []models.Network) error { localnets, err := net.Interfaces() if err != nil { return err } // check networks to join for _, servernet := range servernets { exists := false for _, localnet := range localnets { if servernet.DefaultInterface == localnet.Name { exists = true } } if !exists { success, err := AddNetwork(&servernet) if err != nil || !success { if err == nil { err = errors.New("network add failed for " + servernet.NetID) } if !strings.Contains(err.Error(), "macaddress_unique") { // ignore macaddress unique error throws logger.Log(1, "error adding network", servernet.NetID, "during sync:", err.Error()) } } } } // check networks to leave for _, localnet := range localnets { if strings.Contains(localnet.Name, "nm-") { var exists = "" for _, servernet := range servernets { if servernet.DefaultInterface == localnet.Name { exists = servernet.NetID } } if exists == "" { success, err := RemoveNetwork(exists) if err != nil || !success { if err == nil { err = errors.New("network delete failed for " + exists) } logger.Log(1, "error removing network", exists, "during sync", err.Error()) } } } } return nil } // AddNetwork - add a network to server in client mode func AddNetwork(networkSettings *models.Network) (bool, error) { var err = logic.ServerJoin(networkSettings) return true, err }