added mutex on log

This commit is contained in:
0xdcarns 2022-01-26 11:20:46 -05:00
parent cd4e2c57d6
commit f5d763ca9f
4 changed files with 60 additions and 66 deletions

View file

@ -6,6 +6,7 @@ import (
"sort" "sort"
"strconv" "strconv"
"strings" "strings"
"sync"
"time" "time"
) )
@ -36,6 +37,9 @@ func ResetLogs() {
// Log - handles adding logs // Log - handles adding logs
func Log(verbosity int, message ...string) { func Log(verbosity int, message ...string) {
var mu sync.Mutex
mu.Lock()
defer mu.Unlock()
var currentTime = time.Now() var currentTime = time.Now()
var currentMessage = makeString(message...) var currentMessage = makeString(message...)
if int32(verbosity) <= getVerbose() && getVerbose() >= 0 { if int32(verbosity) <= getVerbose() && getVerbose() >= 0 {

View file

@ -119,7 +119,8 @@ func ModConfig(node *models.Node) error {
modconfig.Node = (*node) modconfig.Node = (*node)
modconfig.NetworkSettings = node.NetworkSettings modconfig.NetworkSettings = node.NetworkSettings
log.Printf("%v \n", modconfig) log.Printf("%v \n", node.NetworkSettings)
log.Printf("%v \n", modconfig.NetworkSettings)
err = Write(&modconfig, network) err = Write(&modconfig, network)
return err return err
} }

View file

@ -59,10 +59,11 @@ func Daemon() error {
// SetupMQTT creates a connection to broker and return client // SetupMQTT creates a connection to broker and return client
func SetupMQTT(cfg *config.ClientConfig) mqtt.Client { func SetupMQTT(cfg *config.ClientConfig) mqtt.Client {
opts := mqtt.NewClientOptions() opts := mqtt.NewClientOptions()
for i, server := range cfg.Node.NetworkSettings.DefaultServerAddrs { for _, server := range cfg.Node.NetworkSettings.DefaultServerAddrs {
if server.Address != "" && server.IsLeader { if server.Address != "" && server.IsLeader {
ncutils.Log(fmt.Sprintf("adding server (%d) to listen on network %s \n", (i + 1), cfg.Node.Network)) ncutils.Log(fmt.Sprintf("adding server (%s) to listen on network %s \n", server.Address, cfg.Node.Network))
opts.AddBroker(server.Address + ":1883") opts.AddBroker(server.Address + ":1883")
break
} }
} }
opts.SetDefaultPublishHandler(All) opts.SetDefaultPublishHandler(All)

View file

@ -7,6 +7,7 @@ import (
"fmt" "fmt"
"log" "log"
"os/exec" "os/exec"
"runtime"
"github.com/google/uuid" "github.com/google/uuid"
nodepb "github.com/gravitl/netmaker/grpc" nodepb "github.com/gravitl/netmaker/grpc"
@ -101,8 +102,7 @@ func JoinNetwork(cfg config.ClientConfig, privateKey string) error {
// make sure name is appropriate, if not, give blank name // make sure name is appropriate, if not, give blank name
cfg.Node.Name = formatName(cfg.Node) cfg.Node.Name = formatName(cfg.Node)
// differentiate between client/server here // differentiate between client/server here
var node models.Node // fill this node with appropriate calls var node = models.Node{
postnode := &models.Node{
Password: cfg.Node.Password, Password: cfg.Node.Password,
ID: cfg.Node.ID, ID: cfg.Node.ID,
MacAddress: cfg.Node.MacAddress, MacAddress: cfg.Node.MacAddress,
@ -124,45 +124,18 @@ func JoinNetwork(cfg config.ClientConfig, privateKey string) error {
UDPHolePunch: cfg.Node.UDPHolePunch, UDPHolePunch: cfg.Node.UDPHolePunch,
} }
if cfg.Node.IsServer != "yes" { ncutils.Log("joining " + cfg.Network + " at " + cfg.Server.GRPCAddress)
ncutils.Log("joining " + cfg.Network + " at " + cfg.Server.GRPCAddress) var wcclient nodepb.NodeServiceClient
var wcclient nodepb.NodeServiceClient
conn, err := grpc.Dial(cfg.Server.GRPCAddress, conn, err := grpc.Dial(cfg.Server.GRPCAddress,
ncutils.GRPCRequestOpts(cfg.Server.GRPCSSL)) ncutils.GRPCRequestOpts(cfg.Server.GRPCSSL))
if err != nil { if err != nil {
log.Fatalf("Unable to establish client connection to "+cfg.Server.GRPCAddress+": %v", err) log.Fatalf("Unable to establish client connection to "+cfg.Server.GRPCAddress+": %v", err)
}
defer conn.Close()
wcclient = nodepb.NewNodeServiceClient(conn)
if err = config.ModConfig(postnode); err != nil {
return err
}
data, err := json.Marshal(postnode)
if err != nil {
return err
}
// Create node on server
res, err := wcclient.CreateNode(
context.TODO(),
&nodepb.Object{
Data: string(data),
Type: nodepb.NODE_TYPE,
},
)
if err != nil {
return err
}
ncutils.PrintLog("node created on remote server...updating configs", 1)
nodeData := res.Data
if err = json.Unmarshal([]byte(nodeData), &node); err != nil {
return err
}
log.Printf("%v \n", nodeData)
} }
defer conn.Close()
wcclient = nodepb.NewNodeServiceClient(conn)
// log.Printf("%v \n", nodeData)
// get free port based on returned default listen port // get free port based on returned default listen port
node.ListenPort, err = ncutils.GetFreePort(node.ListenPort) node.ListenPort, err = ncutils.GetFreePort(node.ListenPort)
@ -183,33 +156,48 @@ func JoinNetwork(cfg config.ClientConfig, privateKey string) error {
cfg.Node.IsStatic = "yes" cfg.Node.IsStatic = "yes"
} }
if node.IsServer != "yes" { // == handle client side == err = wireguard.StorePrivKey(privateKey, cfg.Network)
cfg.Node = node if err != nil {
err = config.ModConfig(&node) return err
if err != nil { }
return err if node.IsPending == "yes" {
} ncutils.Log("Node is marked as PENDING.")
err = wireguard.StorePrivKey(privateKey, cfg.Network) ncutils.Log("Awaiting approval from Admin before configuring WireGuard.")
if err != nil { if cfg.Daemon != "off" {
return err return daemon.InstallDaemon(cfg)
}
if node.IsPending == "yes" {
ncutils.Log("Node is marked as PENDING.")
ncutils.Log("Awaiting approval from Admin before configuring WireGuard.")
if cfg.Daemon != "off" {
return daemon.InstallDaemon(cfg)
}
}
// pushing any local changes to server before starting wireguard
err = Push(cfg.Network)
if err != nil {
return err
}
// attempt to make backup
if err = config.SaveBackup(node.Network); err != nil {
ncutils.Log("failed to make backup, node will not auto restore if config is corrupted")
} }
} }
data, err := json.Marshal(&node)
if err != nil {
return err
}
// Create node on server
res, err := wcclient.CreateNode(
context.TODO(),
&nodepb.Object{
Data: string(data),
Type: nodepb.NODE_TYPE,
},
)
if err != nil {
return err
}
ncutils.PrintLog("node created on remote server...updating configs", 1)
nodeData := res.Data
if err = json.Unmarshal([]byte(nodeData), &node); err != nil {
return err
}
node.OS = runtime.GOOS
cfg.Node = node
err = config.ModConfig(&node)
if err != nil {
return err
}
// attempt to make backup
if err = config.SaveBackup(node.Network); err != nil {
ncutils.Log("failed to make backup, node will not auto restore if config is corrupted")
}
ncutils.Log("retrieving peers") ncutils.Log("retrieving peers")
peers, hasGateway, gateways, err := server.GetPeers(node.MacAddress, cfg.Network, cfg.Server.GRPCAddress, node.IsDualStack == "yes", node.IsIngressGateway == "yes", node.IsServer == "yes") peers, hasGateway, gateways, err := server.GetPeers(node.MacAddress, cfg.Network, cfg.Server.GRPCAddress, node.IsDualStack == "yes", node.IsIngressGateway == "yes", node.IsServer == "yes")