netclient grpc registration working

This commit is contained in:
afeiszli
2021-05-29 15:12:15 -04:00
parent 6c13f06001
commit afa4f1e8bb
6 changed files with 89 additions and 13 deletions

View File

@@ -9,6 +9,7 @@ type IntClient struct {
Address6 string `json:"address6" bson:"address6"` Address6 string `json:"address6" bson:"address6"`
Network string `json:"network" bson:"network"` Network string `json:"network" bson:"network"`
ServerEndpoint string `json:"serverendpoint" bson:"serverendpoint"` ServerEndpoint string `json:"serverendpoint" bson:"serverendpoint"`
ServerAPIEndpoint string `json:"serverapiendpoint" bson:"serverapiendpoint"`
ServerAddress string `json:"serveraddress" bson:"serveraddress"` ServerAddress string `json:"serveraddress" bson:"serveraddress"`
ServerPort string `json:"serverport" bson:"serverport"` ServerPort string `json:"serverport" bson:"serverport"`
ServerKey string `json:"serverkey" bson:"serverkey"` ServerKey string `json:"serverkey" bson:"serverkey"`

View File

@@ -103,3 +103,12 @@ func Uninstall(cfg config.GlobalConfig) error {
err = functions.Unregister(cfg) err = functions.Unregister(cfg)
return err return err
} }
func Reregister(cfg config.GlobalConfig) error {
err := functions.Reregister(cfg)
return err
}
func Unregister(cfg config.GlobalConfig) error {
err := functions.Unregister(cfg)
return err
}

View File

@@ -24,6 +24,7 @@ func JoinNetwork(cfg config.ClientConfig) error {
err := errors.New("ALREADY_INSTALLED. Netclient appears to already be installed for cfg.Network " + cfg.Network + ". To re-install, please remove by executing 'sudo netclient -c remove -n " + cfg.Network + "'. Then re-run the install command.") err := errors.New("ALREADY_INSTALLED. Netclient appears to already be installed for cfg.Network " + cfg.Network + ". To re-install, please remove by executing 'sudo netclient -c remove -n " + cfg.Network + "'. Then re-run the install command.")
return err return err
} }
log.Println("attempting to joining " + cfg.Network + " at " + cfg.Server.GRPCAddress)
err := config.Write(&cfg, cfg.Network) err := config.Write(&cfg, cfg.Network)
if err != nil { if err != nil {
return err return err
@@ -42,7 +43,7 @@ func JoinNetwork(cfg config.ClientConfig) error {
return err return err
} }
_, localrange, err := net.ParseCIDR(cfg.Node.LocalRange) _, localrange, err := net.ParseCIDR(cfg.Node.LocalRange)
if err != nil { if err != nil {
return err return err
} }

View File

@@ -2,6 +2,7 @@ package functions
import ( import (
"time" "time"
"os"
"log" "log"
"io/ioutil" "io/ioutil"
"bytes" "bytes"
@@ -17,6 +18,14 @@ import (
func Register(cfg config.GlobalConfig) error { func Register(cfg config.GlobalConfig) error {
_, err := os.Stat("/etc/netclient")
if os.IsNotExist(err) {
os.Mkdir("/etc/netclient", 744)
} else if err != nil {
log.Println("couldnt find or create /etc/netclient")
return err
}
postclient := &models.IntClient{ postclient := &models.IntClient{
AccessKey: cfg.Client.AccessKey, AccessKey: cfg.Client.AccessKey,
PublicKey: cfg.Client.PublicKey, PublicKey: cfg.Client.PublicKey,
@@ -31,8 +40,7 @@ func Register(cfg config.GlobalConfig) error {
} }
jsonbytes := []byte(jsonstring) jsonbytes := []byte(jsonstring)
body := bytes.NewBuffer(jsonbytes) body := bytes.NewBuffer(jsonbytes)
log.Println(jsonstring) log.Println("registering to http://"+cfg.Client.ServerAPIEndpoint+"/api/client/register")
log.Println("http://"+cfg.Client.ServerEndpoint+"/api/client/register","application/json")
res, err := http.Post("http://"+cfg.Client.ServerEndpoint+"/api/intclient/register","application/json",body) res, err := http.Post("http://"+cfg.Client.ServerEndpoint+"/api/intclient/register","application/json",body)
if err != nil { if err != nil {
return err return err
@@ -62,21 +70,36 @@ func Register(cfg config.GlobalConfig) error {
func Unregister(cfg config.GlobalConfig) error { func Unregister(cfg config.GlobalConfig) error {
client := &http.Client{ Timeout: 7 * time.Second,} client := &http.Client{ Timeout: 7 * time.Second,}
req, err := http.NewRequest("DELETE", "http://"+cfg.Client.ServerEndpoint+"/api/intclient/"+cfg.Client.ClientID, nil) req, err := http.NewRequest("DELETE", "http://"+cfg.Client.ServerAPIEndpoint+"/api/intclient/"+cfg.Client.ClientID, nil)
if err != nil { if err != nil {
return err return err
} }
res, err := client.Do(req) res, err := client.Do(req)
if res == nil { if res == nil {
return errors.New("server not reachable at " + "http://"+cfg.Client.ServerAPIEndpoint+"/api/intclient/"+cfg.Client.ClientID)
} else if res.StatusCode != http.StatusOK {
return errors.New("request to server failed: " + res.Status)
defer res.Body.Close()
} else {
err = local.WipeGRPCClient() err = local.WipeGRPCClient()
if err == nil { if err == nil {
log.Println("successfully removed grpc client interface") log.Println("successfully removed grpc client interface")
} }
} else {
if res.StatusCode != http.StatusOK {
return errors.New("request to server failed: " + res.Status)
defer res.Body.Close()
}
} }
return err return err
} }
func Reregister(cfg config.GlobalConfig) error {
err := Unregister(cfg)
if err != nil {
log.Println("failed to un-register")
return err
}
err = Register(cfg)
if err != nil {
log.Println("failed to re-register after unregistering")
}
return err
}

View File

@@ -199,6 +199,22 @@ func main() {
return err return err
}, },
}, },
{
Name: "reregister",
Usage: "Re-Register with Netmaker Server for secure GRPC communications.",
Flags: cliFlags,
Action: func(c *cli.Context) error {
cfg, err := config.ReadGlobalConfig()
if err != nil {
return err
}
var gconf config.GlobalConfig
gconf = *cfg
err = command.Reregister(gconf)
return err
},
},
{ {
Name: "join", Name: "join",
Usage: "Join a Netmaker network.", Usage: "Join a Netmaker network.",
@@ -312,6 +328,23 @@ func main() {
return err return err
}, },
}, },
{
Name: "unregister",
Usage: "Unregister the netclient from secure server GRPC.",
Flags: cliFlags,
// the action, or code that will be executed when
// we execute our `ns` command
Action: func(c *cli.Context) error {
cfg, err := config.ReadGlobalConfig()
if err != nil {
return err
}
var gconf config.GlobalConfig
gconf = *cfg
err = command.Unregister(gconf)
return err
},
},
} }
// start our application // start our application

View File

@@ -48,9 +48,9 @@ func InitGRPCWireguard(client models.IntClient) error {
return errors.New("no address to configure") return errors.New("no address to configure")
} }
cmdIPDevLinkAdd := exec.Command("ip","link", "add", "dev", ifacename, "type", "wireguard" ) cmdIPDevLinkAdd := exec.Command("ip","link", "add", "dev", ifacename, "type", "wireguard" )
cmdIPAddrAdd := exec.Command("ip", "address", "add", "dev", ifacename, client.Address+"/24") cmdIPAddrAdd := exec.Command("ip", "address", "add", "dev", ifacename, client.Address+"/24")
cmdIPAddr6Add := exec.Command("ip", "address", "add", "dev", ifacename, client.Address6+"/64") cmdIPAddr6Add := exec.Command("ip", "address", "add", "dev", ifacename, client.Address6+"/64")
currentiface, err := net.InterfaceByName(ifacename) currentiface, err := net.InterfaceByName(ifacename)
if err != nil { if err != nil {
err = cmdIPDevLinkAdd.Run() err = cmdIPDevLinkAdd.Run()
if err != nil && !strings.Contains(err.Error(), "exists") { if err != nil && !strings.Contains(err.Error(), "exists") {
@@ -87,7 +87,7 @@ func InitGRPCWireguard(client models.IntClient) error {
var peers []wgtypes.PeerConfig var peers []wgtypes.PeerConfig
var peeraddr = net.IPNet{ var peeraddr = net.IPNet{
IP: net.ParseIP(client.ServerAddress), IP: net.ParseIP(client.ServerAddress),
Mask: net.CIDRMask(64, 128), Mask: net.CIDRMask(32, 32),
} }
var allowedips []net.IPNet var allowedips []net.IPNet
allowedips = append(allowedips, peeraddr) allowedips = append(allowedips, peeraddr)
@@ -126,6 +126,15 @@ func InitGRPCWireguard(client models.IntClient) error {
log.Printf("This is inconvenient: %v", err) log.Printf("This is inconvenient: %v", err)
} }
} }
cmdIPLinkUp := exec.Command("ip", "link", "set", "up", "dev", ifacename)
cmdIPLinkDown := exec.Command("ip", "link", "set", "down", "dev", ifacename)
err = cmdIPLinkDown.Run()
err = cmdIPLinkUp.Run()
if err != nil {
return err
}
return err return err
} }