mirror of
https://github.com/gravitl/netmaker.git
synced 2024-09-22 08:16:07 +08:00
102 lines
2.4 KiB
Go
102 lines
2.4 KiB
Go
package wireguard
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"os"
|
|
|
|
"github.com/gravitl/netmaker/models"
|
|
"github.com/gravitl/netmaker/netclient/config"
|
|
"github.com/gravitl/netmaker/netclient/ncutils"
|
|
"golang.zx2c4.com/wireguard/wgctrl/wgtypes"
|
|
)
|
|
|
|
// SetWGKeyConfig - sets the wg conf with a new private key
|
|
func SetWGKeyConfig(network string, serveraddr string) error {
|
|
|
|
cfg, err := config.ReadConfig(network)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
node := cfg.Node
|
|
|
|
privatekey, err := wgtypes.GeneratePrivateKey()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
privkeystring := privatekey.String()
|
|
publickey := privatekey.PublicKey()
|
|
|
|
node.PublicKey = publickey.String()
|
|
|
|
err = StorePrivKey(privkeystring, network)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if node.Action == models.NODE_UPDATE_KEY {
|
|
node.Action = models.NODE_NOOP
|
|
}
|
|
err = config.ModConfig(&node)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = SetWGConfig(network, false)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
// ApplyWGQuickConf - applies wg-quick commands if os supports
|
|
func ApplyWGQuickConf(confPath string) error {
|
|
_, _ = ncutils.RunCmd("wg-quick down "+confPath, false)
|
|
_, err := ncutils.RunCmd("wg-quick up "+confPath, false)
|
|
return err
|
|
}
|
|
|
|
// ApplyWGQuickConf - applies wg-quick commands if os supports
|
|
func SyncWGQuickConf(confPath string) error {
|
|
var tmpConf = confPath + ".sync.tmp"
|
|
conf, err := ncutils.RunCmd("wg-quick strip "+confPath, false)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = ioutil.WriteFile(tmpConf, []byte(conf), 0644)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, err = ncutils.RunCmd("wg sync "+confPath, false)
|
|
if err != nil {
|
|
ncutils.Log("error syncing conf, resetting")
|
|
err = ApplyWGQuickConf(confPath)
|
|
}
|
|
errN := os.Remove(tmpConf)
|
|
if errN != nil {
|
|
ncutils.Log(errN.Error())
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
// RemoveWGQuickConf - calls wg-quick down
|
|
func RemoveWGQuickConf(confPath string, printlog bool) error {
|
|
_, err := ncutils.RunCmd("wg-quick down "+confPath, printlog)
|
|
return err
|
|
}
|
|
|
|
// StorePrivKey - stores wg priv key on disk locally
|
|
func StorePrivKey(key string, network string) error {
|
|
var err error
|
|
d1 := []byte(key)
|
|
err = ioutil.WriteFile(ncutils.GetNetclientPathSpecific()+"wgkey-"+network, d1, 0644)
|
|
return err
|
|
}
|
|
|
|
// RetrievePrivKey - reads wg priv key from local disk
|
|
func RetrievePrivKey(network string) (string, error) {
|
|
dat, err := ioutil.ReadFile(ncutils.GetNetclientPathSpecific() + "wgkey-" + network)
|
|
return string(dat), err
|
|
}
|