mirror of
https://github.com/gravitl/netmaker.git
synced 2025-10-02 01:45:23 +08:00
gen new key and re-register on pull
This commit is contained in:
parent
1ae9240d86
commit
e3206c5b4c
4 changed files with 22 additions and 20 deletions
|
@ -78,7 +78,7 @@ func Pull(cfg *config.ClientConfig) error {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
for _, network := range networks {
|
for _, network := range networks {
|
||||||
_, err = functions.Pull(network, true)
|
_, err = functions.Pull(network, true, true)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Log(1, "Error pulling network config for network: ", network, "\n", err.Error())
|
logger.Log(1, "Error pulling network config for network: ", network, "\n", err.Error())
|
||||||
} else {
|
} else {
|
||||||
|
@ -87,7 +87,7 @@ func Pull(cfg *config.ClientConfig) error {
|
||||||
}
|
}
|
||||||
err = nil
|
err = nil
|
||||||
} else {
|
} else {
|
||||||
_, err = functions.Pull(cfg.Network, true)
|
_, err = functions.Pull(cfg.Network, true, true)
|
||||||
}
|
}
|
||||||
logger.Log(1, "reset network and peer configs")
|
logger.Log(1, "reset network and peer configs")
|
||||||
if err == nil {
|
if err == nil {
|
||||||
|
|
|
@ -236,7 +236,7 @@ func setupMQTT(cfg *config.ClientConfig, server string, publish bool) mqtt.Clien
|
||||||
opts.SetResumeSubs(true)
|
opts.SetResumeSubs(true)
|
||||||
opts.SetConnectionLostHandler(func(c mqtt.Client, e error) {
|
opts.SetConnectionLostHandler(func(c mqtt.Client, e error) {
|
||||||
logger.Log(0, "detected broker connection lost, running pull for ", cfg.Node.Network)
|
logger.Log(0, "detected broker connection lost, running pull for ", cfg.Node.Network)
|
||||||
_, err := Pull(cfg.Node.Network, true)
|
_, err := Pull(cfg.Node.Network, true, false)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Log(0, "could not run pull, server unreachable: ", err.Error())
|
logger.Log(0, "could not run pull, server unreachable: ", err.Error())
|
||||||
logger.Log(0, "waiting to retry...")
|
logger.Log(0, "waiting to retry...")
|
||||||
|
@ -250,7 +250,7 @@ func setupMQTT(cfg *config.ClientConfig, server string, publish bool) mqtt.Clien
|
||||||
//if after 12 seconds, try a pull on the last try
|
//if after 12 seconds, try a pull on the last try
|
||||||
if time.Now().After(tperiod) {
|
if time.Now().After(tperiod) {
|
||||||
logger.Log(0, "running pull for ", cfg.Node.Network)
|
logger.Log(0, "running pull for ", cfg.Node.Network)
|
||||||
_, err := Pull(cfg.Node.Network, true)
|
_, err := Pull(cfg.Node.Network, true, false)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Log(0, "could not run pull, exiting ", cfg.Node.Network, " setup: ", err.Error())
|
logger.Log(0, "could not run pull, exiting ", cfg.Node.Network, " setup: ", err.Error())
|
||||||
return client
|
return client
|
||||||
|
@ -263,7 +263,7 @@ func setupMQTT(cfg *config.ClientConfig, server string, publish bool) mqtt.Clien
|
||||||
logger.Log(0, "could not connect to broker, exiting ", cfg.Node.Network, " setup: ", token.Error().Error())
|
logger.Log(0, "could not connect to broker, exiting ", cfg.Node.Network, " setup: ", token.Error().Error())
|
||||||
if strings.Contains(token.Error().Error(), "connectex") || strings.Contains(token.Error().Error(), "i/o timeout") {
|
if strings.Contains(token.Error().Error(), "connectex") || strings.Contains(token.Error().Error(), "i/o timeout") {
|
||||||
logger.Log(0, "connection issue detected.. pulling and restarting daemon")
|
logger.Log(0, "connection issue detected.. pulling and restarting daemon")
|
||||||
Pull(cfg.Node.Network, true)
|
Pull(cfg.Node.Network, true, false)
|
||||||
daemon.Restart()
|
daemon.Restart()
|
||||||
}
|
}
|
||||||
return client
|
return client
|
||||||
|
@ -296,7 +296,7 @@ func initialPull(network string) {
|
||||||
if !fileInfo.ModTime().IsZero() && time.Now().After(fileInfo.ModTime().Add(time.Minute)) {
|
if !fileInfo.ModTime().IsZero() && time.Now().After(fileInfo.ModTime().Add(time.Minute)) {
|
||||||
sleepTime := 2
|
sleepTime := 2
|
||||||
for {
|
for {
|
||||||
_, err := Pull(network, true)
|
_, err := Pull(network, true, false)
|
||||||
if err == nil {
|
if err == nil {
|
||||||
break
|
break
|
||||||
}
|
}
|
||||||
|
|
|
@ -105,7 +105,7 @@ func Hello(nodeCfg *config.ClientConfig) {
|
||||||
if err := publish(nodeCfg, fmt.Sprintf("ping/%s", nodeCfg.Node.ID), []byte(ncutils.Version), 0); err != nil {
|
if err := publish(nodeCfg, fmt.Sprintf("ping/%s", nodeCfg.Node.ID), []byte(ncutils.Version), 0); err != nil {
|
||||||
logger.Log(0, fmt.Sprintf("error publishing ping, %v", err))
|
logger.Log(0, fmt.Sprintf("error publishing ping, %v", err))
|
||||||
logger.Log(0, "running pull on "+nodeCfg.Node.Network+" to reconnect")
|
logger.Log(0, "running pull on "+nodeCfg.Node.Network+" to reconnect")
|
||||||
_, err := Pull(nodeCfg.Node.Network, true)
|
_, err := Pull(nodeCfg.Node.Network, true, false)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Log(0, "could not run pull on "+nodeCfg.Node.Network+", error: "+err.Error())
|
logger.Log(0, "could not run pull on "+nodeCfg.Node.Network+", error: "+err.Error())
|
||||||
}
|
}
|
||||||
|
|
|
@ -23,7 +23,7 @@ import (
|
||||||
)
|
)
|
||||||
|
|
||||||
// Pull - pulls the latest config from the server, if manual it will overwrite
|
// Pull - pulls the latest config from the server, if manual it will overwrite
|
||||||
func Pull(network string, manual bool) (*models.Node, error) {
|
func Pull(network string, iface bool, register bool) (*models.Node, error) {
|
||||||
cfg, err := config.ReadConfig(network)
|
cfg, err := config.ReadConfig(network)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
|
@ -56,7 +56,7 @@ func Pull(network string, manual bool) (*models.Node, error) {
|
||||||
}
|
}
|
||||||
// ensure that the OS never changes
|
// ensure that the OS never changes
|
||||||
resNode.OS = runtime.GOOS
|
resNode.OS = runtime.GOOS
|
||||||
if manual {
|
if iface {
|
||||||
// check for interface change
|
// check for interface change
|
||||||
if cfg.Node.Interface != resNode.Interface {
|
if cfg.Node.Interface != resNode.Interface {
|
||||||
if err = DeleteInterface(cfg.Node.Interface, cfg.Node.PostDown); err != nil {
|
if err = DeleteInterface(cfg.Node.Interface, cfg.Node.PostDown); err != nil {
|
||||||
|
@ -72,7 +72,7 @@ func Pull(network string, manual bool) (*models.Node, error) {
|
||||||
} else {
|
} else {
|
||||||
if err = wireguard.SetWGConfig(network, true); err != nil {
|
if err = wireguard.SetWGConfig(network, true); err != nil {
|
||||||
if errors.Is(err, os.ErrNotExist) && !ncutils.IsFreeBSD() {
|
if errors.Is(err, os.ErrNotExist) && !ncutils.IsFreeBSD() {
|
||||||
return Pull(network, true)
|
return Pull(network, true, false)
|
||||||
} else {
|
} else {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
@ -83,17 +83,19 @@ func Pull(network string, manual bool) (*models.Node, error) {
|
||||||
logger.Log(0, "unable to update backup file")
|
logger.Log(0, "unable to update backup file")
|
||||||
}
|
}
|
||||||
//generate new private key and re-register with server
|
//generate new private key and re-register with server
|
||||||
_, newKey, err := ed25519.GenerateKey(rand.Reader)
|
if register {
|
||||||
if err != nil {
|
_, newKey, err := ed25519.GenerateKey(rand.Reader)
|
||||||
return &resNode, err
|
if err != nil {
|
||||||
|
return &resNode, err
|
||||||
|
}
|
||||||
|
if err := tls.SaveKey(ncutils.GetNetclientPath(), "/client.key", newKey); err != nil {
|
||||||
|
return &resNode, err
|
||||||
|
}
|
||||||
|
if err = RegisterWithServer(&newKey, cfg); err != nil {
|
||||||
|
return &resNode, err
|
||||||
|
}
|
||||||
|
daemon.Restart()
|
||||||
}
|
}
|
||||||
if err := tls.SaveKey(ncutils.GetNetclientPath(), "/client.key", newKey); err != nil {
|
|
||||||
return &resNode, err
|
|
||||||
}
|
|
||||||
if err = RegisterWithServer(&newKey, cfg); err != nil {
|
|
||||||
return &resNode, err
|
|
||||||
}
|
|
||||||
daemon.Restart()
|
|
||||||
|
|
||||||
return &resNode, err
|
return &resNode, err
|
||||||
}
|
}
|
||||||
|
|
Loading…
Add table
Reference in a new issue