gen new key and re-register on pull

This commit is contained in:
Matthew R. Kasun 2022-04-25 14:04:36 -04:00
parent 1ae9240d86
commit e3206c5b4c
4 changed files with 22 additions and 20 deletions

View file

@ -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 {

View file

@ -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
} }

View file

@ -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())
} }

View file

@ -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
} }