2021-05-26 00:48:04 +08:00
package functions
import (
2021-07-26 02:22:20 +08:00
"context"
2021-08-03 06:06:26 +08:00
"encoding/json"
2021-05-26 00:48:04 +08:00
"errors"
2021-07-26 02:22:20 +08:00
"fmt"
2021-05-26 00:48:04 +08:00
"log"
2021-07-26 02:22:20 +08:00
nodepb "github.com/gravitl/netmaker/grpc"
2021-08-03 06:06:26 +08:00
"github.com/gravitl/netmaker/models"
"github.com/gravitl/netmaker/netclient/auth"
2021-07-26 02:22:20 +08:00
"github.com/gravitl/netmaker/netclient/config"
2021-09-20 02:03:47 +08:00
"github.com/gravitl/netmaker/netclient/daemon"
2021-07-26 02:22:20 +08:00
"github.com/gravitl/netmaker/netclient/local"
2021-09-20 02:03:47 +08:00
"github.com/gravitl/netmaker/netclient/ncutils"
2021-07-26 02:22:20 +08:00
"github.com/gravitl/netmaker/netclient/server"
"github.com/gravitl/netmaker/netclient/wireguard"
2021-05-26 00:48:04 +08:00
"golang.zx2c4.com/wireguard/wgctrl/wgtypes"
2021-07-26 02:22:20 +08:00
"google.golang.org/grpc"
2021-05-26 00:48:04 +08:00
)
2021-08-03 06:06:26 +08:00
func JoinNetwork ( cfg config . ClientConfig , privateKey string ) error {
2021-05-26 00:48:04 +08:00
hasnet := local . HasNetwork ( cfg . Network )
2021-06-02 23:00:10 +08:00
if hasnet {
2021-07-26 02:22:20 +08:00
err := errors . New ( "ALREADY_INSTALLED. Netclient appears to already be installed for " + cfg . Network + ". To re-install, please remove by executing 'sudo netclient leave -n " + cfg . Network + "'. Then re-run the install command." )
2021-05-26 00:48:04 +08:00
return err
}
2021-09-11 03:48:18 +08:00
2021-09-20 02:03:47 +08:00
ncutils . Log ( "joining " + cfg . Network + " at " + cfg . Server . GRPCAddress )
2021-05-26 00:48:04 +08:00
err := config . Write ( & cfg , cfg . Network )
if err != nil {
return err
}
2021-06-02 23:00:10 +08:00
if cfg . Node . Network == "" {
return errors . New ( "no network provided" )
}
2021-05-26 00:48:04 +08:00
2021-09-11 03:48:18 +08:00
if cfg . Node . LocalRange != "" && cfg . Node . LocalAddress == "" {
log . Println ( "local vpn, getting local address from range: " + cfg . Node . LocalRange )
cfg . Node . LocalAddress = getLocalIP ( cfg . Node )
2021-05-26 00:48:04 +08:00
}
2021-06-02 23:00:10 +08:00
if cfg . Node . Password == "" {
2021-09-20 02:03:47 +08:00
cfg . Node . Password = ncutils . GenPass ( )
2021-06-02 23:00:10 +08:00
}
2021-08-06 01:19:55 +08:00
auth . StoreSecret ( cfg . Node . Password , cfg . Node . Network )
2021-09-11 03:48:18 +08:00
2021-09-20 02:03:47 +08:00
// set endpoint if blank. set to local if local net, retrieve from function if not
2021-07-26 02:22:20 +08:00
if cfg . Node . Endpoint == "" {
2021-05-26 00:48:04 +08:00
if cfg . Node . IsLocal == "yes" && cfg . Node . LocalAddress != "" {
cfg . Node . Endpoint = cfg . Node . LocalAddress
} else {
2021-09-20 02:03:47 +08:00
cfg . Node . Endpoint , err = ncutils . GetPublicIP ( )
2021-09-11 03:48:18 +08:00
}
if err != nil || cfg . Node . Endpoint == "" {
2021-09-20 02:03:47 +08:00
ncutils . Log ( "Error setting cfg.Node.Endpoint." )
2021-09-11 03:48:18 +08:00
return err
2021-05-26 00:48:04 +08:00
}
2021-07-26 02:22:20 +08:00
}
2021-09-11 03:48:18 +08:00
// Generate and set public/private WireGuard Keys
2021-08-03 06:06:26 +08:00
if privateKey == "" {
wgPrivatekey , err := wgtypes . GeneratePrivateKey ( )
2021-05-26 00:48:04 +08:00
if err != nil {
log . Fatal ( err )
}
2021-08-03 06:06:26 +08:00
privateKey = wgPrivatekey . String ( )
cfg . Node . PublicKey = wgPrivatekey . PublicKey ( ) . String ( )
2021-05-26 00:48:04 +08:00
}
2021-09-11 03:48:18 +08:00
// Find and set node MacAddress
2021-05-26 00:48:04 +08:00
if cfg . Node . MacAddress == "" {
2021-09-20 02:03:47 +08:00
macs , err := ncutils . GetMacAddr ( )
2021-05-26 00:48:04 +08:00
if err != nil {
return err
} else if len ( macs ) == 0 {
2021-09-11 03:48:18 +08:00
log . Fatal ( "could not retrieve mac address" )
2021-05-26 00:48:04 +08:00
} else {
2021-07-26 02:22:20 +08:00
cfg . Node . MacAddress = macs [ 0 ]
2021-05-26 00:48:04 +08:00
}
}
2021-08-19 06:59:17 +08:00
2021-05-26 00:48:04 +08:00
var wcclient nodepb . NodeServiceClient
2021-09-11 03:48:18 +08:00
2021-09-20 02:03:47 +08:00
conn , err := grpc . Dial ( cfg . Server . GRPCAddress ,
ncutils . GRPCRequestOpts ( cfg . Server . GRPCSSL ) )
2021-07-11 08:11:19 +08:00
2021-07-26 02:22:20 +08:00
if err != nil {
log . Fatalf ( "Unable to establish client connection to " + cfg . Server . GRPCAddress + ": %v" , err )
}
2021-09-23 21:04:33 +08:00
defer conn . Close ( )
2021-07-26 02:22:20 +08:00
wcclient = nodepb . NewNodeServiceClient ( conn )
2021-05-26 00:48:04 +08:00
2021-08-10 23:52:50 +08:00
postnode := & models . Node {
Password : cfg . Node . Password ,
MacAddress : cfg . Node . MacAddress ,
AccessKey : cfg . Server . AccessKey ,
Network : cfg . Network ,
ListenPort : cfg . Node . ListenPort ,
PostUp : cfg . Node . PostUp ,
PostDown : cfg . Node . PostDown ,
PersistentKeepalive : cfg . Node . PersistentKeepalive ,
LocalAddress : cfg . Node . LocalAddress ,
Interface : cfg . Node . Interface ,
PublicKey : cfg . Node . PublicKey ,
2021-09-29 22:02:34 +08:00
DNSOn : cfg . Node . DNSOn ,
2021-08-10 23:52:50 +08:00
Name : cfg . Node . Name ,
Endpoint : cfg . Node . Endpoint ,
SaveConfig : cfg . Node . SaveConfig ,
UDPHolePunch : cfg . Node . UDPHolePunch ,
}
2021-08-19 06:59:17 +08:00
2021-08-10 23:52:50 +08:00
if err = config . ModConfig ( postnode ) ; err != nil {
2021-08-03 06:06:26 +08:00
return err
2021-07-26 02:22:20 +08:00
}
2021-08-10 23:52:50 +08:00
data , err := json . Marshal ( postnode )
2021-07-26 02:22:20 +08:00
if err != nil {
2021-05-26 00:48:04 +08:00
return err
2021-07-26 02:22:20 +08:00
}
2021-09-11 03:48:18 +08:00
// Create node on server
2021-07-26 02:22:20 +08:00
res , err := wcclient . CreateNode (
context . TODO ( ) ,
2021-08-03 06:06:26 +08:00
& nodepb . Object {
Data : string ( data ) ,
Type : nodepb . NODE_TYPE ,
2021-07-26 02:22:20 +08:00
} ,
)
if err != nil {
return err
}
2021-09-20 02:03:47 +08:00
ncutils . PrintLog ( "node created on remote server...updating configs" , 1 )
2021-08-03 06:06:26 +08:00
nodeData := res . Data
var node models . Node
if err = json . Unmarshal ( [ ] byte ( nodeData ) , & node ) ; err != nil {
2021-07-26 02:22:20 +08:00
return err
}
2021-05-26 00:48:04 +08:00
2021-09-11 03:48:18 +08:00
// get free port based on returned default listen port
2021-09-20 02:03:47 +08:00
node . ListenPort , err = ncutils . GetFreePort ( node . ListenPort )
2021-09-11 03:48:18 +08:00
if err != nil {
fmt . Printf ( "Error retrieving port: %v" , err )
2021-05-26 00:48:04 +08:00
}
2021-09-20 02:03:47 +08:00
2021-09-11 03:48:18 +08:00
// safety check. If returned node from server is local, but not currently configured as local, set to local addr
if cfg . Node . IsLocal != "yes" && node . IsLocal == "yes" && node . LocalRange != "" {
2021-09-20 02:03:47 +08:00
node . LocalAddress , err = ncutils . GetLocalIP ( node . LocalRange )
2021-05-26 00:48:04 +08:00
if err != nil {
return err
}
2021-08-03 06:06:26 +08:00
node . Endpoint = node . LocalAddress
2021-05-26 00:48:04 +08:00
}
2021-09-29 22:02:34 +08:00
2021-08-03 06:06:26 +08:00
err = config . ModConfig ( & node )
2021-07-26 02:22:20 +08:00
if err != nil {
return err
}
2021-05-26 00:48:04 +08:00
2021-08-10 22:55:48 +08:00
err = wireguard . StorePrivKey ( privateKey , cfg . Network )
if err != nil {
return err
}
2021-09-20 02:03:47 +08:00
// pushing any local changes to server before starting wireguard
2021-09-11 03:48:18 +08:00
err = Push ( cfg . Network )
if err != nil {
return err
}
2021-08-03 06:06:26 +08:00
if node . IsPending == "yes" {
2021-09-20 02:03:47 +08:00
ncutils . Log ( "Node is marked as PENDING." )
ncutils . Log ( "Awaiting approval from Admin before configuring WireGuard." )
2021-07-26 02:22:20 +08:00
if cfg . Daemon != "off" {
2021-09-20 02:03:47 +08:00
return daemon . InstallDaemon ( cfg )
2021-05-26 00:48:04 +08:00
}
}
2021-09-11 03:48:18 +08:00
2021-09-20 02:03:47 +08:00
ncutils . Log ( "retrieving remote peers" )
2021-08-03 06:06:26 +08:00
peers , hasGateway , gateways , err := server . GetPeers ( node . MacAddress , cfg . Network , cfg . Server . GRPCAddress , node . IsDualStack == "yes" , node . IsIngressGateway == "yes" )
2021-05-26 00:48:04 +08:00
2021-09-20 02:03:47 +08:00
if err != nil && ! ncutils . IsEmptyRecord ( err ) {
ncutils . Log ( "failed to retrieve peers" )
2021-07-26 02:22:20 +08:00
return err
}
2021-08-10 22:55:48 +08:00
2021-09-20 02:03:47 +08:00
ncutils . Log ( "starting wireguard" )
2021-08-03 06:06:26 +08:00
err = wireguard . InitWireguard ( & node , privateKey , peers , hasGateway , gateways )
2021-07-26 02:22:20 +08:00
if err != nil {
return err
}
2021-06-02 00:23:05 +08:00
if cfg . Daemon != "off" {
2021-09-20 02:03:47 +08:00
err = daemon . InstallDaemon ( cfg )
2021-05-26 00:48:04 +08:00
}
2021-07-26 02:22:20 +08:00
if err != nil {
return err
}
2021-05-26 00:48:04 +08:00
return err
}