|
@@ -1,95 +1,95 @@
|
|
package functions
|
|
package functions
|
|
|
|
|
|
import (
|
|
import (
|
|
- "time"
|
|
|
|
- "os"
|
|
|
|
- "net"
|
|
|
|
- "log"
|
|
|
|
- "io/ioutil"
|
|
|
|
"bytes"
|
|
"bytes"
|
|
- "github.com/gravitl/netmaker/netclient/config"
|
|
|
|
- "github.com/gravitl/netmaker/netclient/local"
|
|
|
|
- "github.com/gravitl/netmaker/netclient/wireguard"
|
|
|
|
- "github.com/gravitl/netmaker/models"
|
|
|
|
"encoding/json"
|
|
"encoding/json"
|
|
- "net/http"
|
|
|
|
"errors"
|
|
"errors"
|
|
-// "github.com/davecgh/go-spew/spew"
|
|
|
|
|
|
+ "io/ioutil"
|
|
|
|
+ "log"
|
|
|
|
+ "net"
|
|
|
|
+ "net/http"
|
|
|
|
+ "os"
|
|
|
|
+ "time"
|
|
|
|
+
|
|
|
|
+ "github.com/gravitl/netmaker/models"
|
|
|
|
+ "github.com/gravitl/netmaker/netclient/config"
|
|
|
|
+ "github.com/gravitl/netmaker/netclient/local"
|
|
|
|
+ // "github.com/davecgh/go-spew/spew"
|
|
)
|
|
)
|
|
|
|
|
|
func Register(cfg config.GlobalConfig) error {
|
|
func Register(cfg config.GlobalConfig) error {
|
|
|
|
|
|
_, err := os.Stat("/etc/netclient")
|
|
_, 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
|
|
|
|
- }
|
|
|
|
|
|
+ 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{
|
|
|
|
- AccessKey: cfg.Client.AccessKey,
|
|
|
|
- PublicKey: cfg.Client.PublicKey,
|
|
|
|
- PrivateKey: cfg.Client.PublicKey,
|
|
|
|
- Address: cfg.Client.Address,
|
|
|
|
- Address6: cfg.Client.Address6,
|
|
|
|
- Network: "comms",
|
|
|
|
|
|
+ postclient := &models.IntClient{
|
|
|
|
+ AccessKey: cfg.Client.AccessKey,
|
|
|
|
+ PublicKey: cfg.Client.PublicKey,
|
|
|
|
+ PrivateKey: cfg.Client.PublicKey,
|
|
|
|
+ Address: cfg.Client.Address,
|
|
|
|
+ Address6: cfg.Client.Address6,
|
|
|
|
+ Network: "comms",
|
|
}
|
|
}
|
|
|
|
|
|
jsonstring, err := json.Marshal(postclient)
|
|
jsonstring, err := json.Marshal(postclient)
|
|
- if err != nil {
|
|
|
|
- return err
|
|
|
|
- }
|
|
|
|
|
|
+ if err != nil {
|
|
|
|
+ return err
|
|
|
|
+ }
|
|
jsonbytes := []byte(jsonstring)
|
|
jsonbytes := []byte(jsonstring)
|
|
body := bytes.NewBuffer(jsonbytes)
|
|
body := bytes.NewBuffer(jsonbytes)
|
|
publicaddress := net.JoinHostPort(cfg.Client.ServerPublicEndpoint, cfg.Client.ServerAPIPort)
|
|
publicaddress := net.JoinHostPort(cfg.Client.ServerPublicEndpoint, cfg.Client.ServerAPIPort)
|
|
|
|
|
|
- res, err := http.Post("http://"+publicaddress+"/api/intclient/register","application/json",body)
|
|
|
|
- if err != nil {
|
|
|
|
- log.Println("Failed to register to http://"+publicaddress+"/api/client/register")
|
|
|
|
- return err
|
|
|
|
- }
|
|
|
|
|
|
+ res, err := http.Post("http://"+publicaddress+"/api/intclient/register", "application/json", body)
|
|
|
|
+ if err != nil {
|
|
|
|
+ log.Println("Failed to register to http://" + publicaddress + "/api/client/register")
|
|
|
|
+ return err
|
|
|
|
+ }
|
|
if res.StatusCode != http.StatusOK {
|
|
if res.StatusCode != http.StatusOK {
|
|
- log.Println("Failed to register to http://"+publicaddress+"/api/client/register")
|
|
|
|
|
|
+ log.Println("Failed to register to http://" + publicaddress + "/api/client/register")
|
|
return errors.New("request to server failed: " + res.Status)
|
|
return errors.New("request to server failed: " + res.Status)
|
|
}
|
|
}
|
|
bodyBytes, err := ioutil.ReadAll(res.Body)
|
|
bodyBytes, err := ioutil.ReadAll(res.Body)
|
|
- //bodyString := string(bodyBytes)
|
|
|
|
|
|
+ //bodyString := string(bodyBytes)
|
|
//spew.Dump(bodyString)
|
|
//spew.Dump(bodyString)
|
|
if err != nil {
|
|
if err != nil {
|
|
return err
|
|
return err
|
|
}
|
|
}
|
|
var wgclient models.IntClient
|
|
var wgclient models.IntClient
|
|
json.Unmarshal(bodyBytes, &wgclient)
|
|
json.Unmarshal(bodyBytes, &wgclient)
|
|
- //spew.Dump(wgclient)
|
|
|
|
|
|
+ //spew.Dump(wgclient)
|
|
err = config.ModGlobalConfig(wgclient)
|
|
err = config.ModGlobalConfig(wgclient)
|
|
- if err != nil {
|
|
|
|
- return err
|
|
|
|
- }
|
|
|
|
- //spew.Dump(wgclient)
|
|
|
|
- err = wireguard.InitGRPCWireguard(wgclient)
|
|
|
|
- if err != nil {
|
|
|
|
- return err
|
|
|
|
- }
|
|
|
|
|
|
+ if err != nil {
|
|
|
|
+ return err
|
|
|
|
+ }
|
|
|
|
+ //spew.Dump(wgclient)
|
|
|
|
+ // err = wireguard.InitGRPCWireguard(wgclient)
|
|
|
|
+ // if err != nil {
|
|
|
|
+ // return err
|
|
|
|
+ // }
|
|
log.Println("registered netclient to " + cfg.Client.ServerPrivateAddress)
|
|
log.Println("registered netclient to " + cfg.Client.ServerPrivateAddress)
|
|
return err
|
|
return err
|
|
}
|
|
}
|
|
|
|
|
|
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}
|
|
publicaddress := net.JoinHostPort(cfg.Client.ServerPublicEndpoint, cfg.Client.ServerAPIPort)
|
|
publicaddress := net.JoinHostPort(cfg.Client.ServerPublicEndpoint, cfg.Client.ServerAPIPort)
|
|
- log.Println("sending delete request to: " + "http://"+publicaddress+"/api/intclient/"+cfg.Client.ClientID)
|
|
|
|
|
|
+ log.Println("sending delete request to: " + "http://" + publicaddress + "/api/intclient/" + cfg.Client.ClientID)
|
|
req, err := http.NewRequest("DELETE", "http://"+publicaddress+"/api/intclient/"+cfg.Client.ClientID, nil)
|
|
req, err := http.NewRequest("DELETE", "http://"+publicaddress+"/api/intclient/"+cfg.Client.ClientID, nil)
|
|
if err != nil {
|
|
if err != nil {
|
|
- log.Println(err)
|
|
|
|
- } else {
|
|
|
|
|
|
+ log.Println(err)
|
|
|
|
+ } else {
|
|
res, err := client.Do(req)
|
|
res, err := client.Do(req)
|
|
if res == nil {
|
|
if res == nil {
|
|
- err = errors.New("server not reachable at " + "http://"+publicaddress+"/api/intclient/"+cfg.Client.ClientID)
|
|
|
|
|
|
+ err = errors.New("server not reachable at " + "http://" + publicaddress + "/api/intclient/" + cfg.Client.ClientID)
|
|
log.Println(err)
|
|
log.Println(err)
|
|
} else if res.StatusCode != http.StatusOK {
|
|
} else if res.StatusCode != http.StatusOK {
|
|
err = errors.New("request to server failed: " + res.Status)
|
|
err = errors.New("request to server failed: " + res.Status)
|
|
- log.Println(err)
|
|
|
|
|
|
+ log.Println(err)
|
|
defer res.Body.Close()
|
|
defer res.Body.Close()
|
|
}
|
|
}
|
|
}
|
|
}
|