mirror of
https://github.com/gravitl/netmaker.git
synced 2025-10-31 20:22:44 +08:00
modify access key
This commit is contained in:
afeiszli
54
netclient/\
54
netclient/\
@@ -1,54 +0,0 @@
|
||||
package functions
|
||||
|
||||
import (
|
||||
"io/ioutil"
|
||||
"bytes"
|
||||
"github.com/gravitl/netmaker/netclient/config"
|
||||
"github.com/gravitl/netmaker/netclient/wireguard"
|
||||
"github.com/gravitl/netmaker/models"
|
||||
"encoding/json"
|
||||
"net/http"
|
||||
"errors"
|
||||
)
|
||||
|
||||
func Register(cfg config.GlobalConfig) error {
|
||||
|
||||
postclient := &models.ServerClient{
|
||||
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)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
jsonbytes := []byte(jsonstring)
|
||||
body := bytes.NewBuffer(jsonbytes)
|
||||
res, err := http.Post("http:/"+cfg.Client.ServerEndpoint+"/api/register","application/json",body)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if res.StatusCode != http.StatusOK {
|
||||
return errors.New("request to server failed: " + res.Status)
|
||||
}
|
||||
bodyBytes, err := ioutil.ReadAll(res.Body)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
var wgclient models.ServerClient
|
||||
json.Unmarshal(bodyBytes, &wgclient)
|
||||
err = config.ModGlobalConfig(wgclient)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
err = wireguard.InitGRPCWireguard(wgclient)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return err
|
||||
}
|
||||
@@ -1,196 +0,0 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"github.com/urfave/cli"
|
||||
"github.com/gravitl/netmaker/netclient/functions"
|
||||
"golang.zx2c4.com/wireguard/wgctrl"
|
||||
nodepb "github.com/gravitl/netmaker/grpc"
|
||||
"flag"
|
||||
"os"
|
||||
"os/exec"
|
||||
"strconv"
|
||||
"strings"
|
||||
"log"
|
||||
)
|
||||
|
||||
const (
|
||||
// name of the service
|
||||
name = "netclient"
|
||||
description = "Netmaker Daemon Service"
|
||||
)
|
||||
|
||||
var password string
|
||||
var network string
|
||||
var server string
|
||||
var accesskey string
|
||||
|
||||
var (
|
||||
wgclient *wgctrl.Client
|
||||
)
|
||||
|
||||
var (
|
||||
wcclient nodepb.NodeServiceClient
|
||||
)
|
||||
|
||||
func main() {
|
||||
|
||||
tpassword := flag.String("p", "changeme", "This node's password for accessing the server regularly")
|
||||
taccesskey := flag.String("k", "badkey", "an access key generated by the server and used for one-time access (install only)")
|
||||
taccesstoken := flag.String("t", "badtoken", "an token generated by the server and used for one-time access (install only)")
|
||||
tname := flag.String("name", "noname", "give the node a name at runtime")
|
||||
tserver := flag.String("s", "localhost:50051", "The location (including port) of the remote gRPC server.")
|
||||
tnetwork := flag.String("n", "nonetwork", "The node network you are attempting to join.")
|
||||
tdnsoff := flag.Bool("dnsoff", false, "DNS Mode. If true, netclient will not alter system dns. false by default.")
|
||||
tpublicip := flag.String("ip4", "nopubip", "The node network you are attempting to join.")
|
||||
tnoauto := flag.Bool("na", false, "No auto mode. If true, netmclient will not be installed as a system service and you will have to retrieve updates manually via checkin command.")
|
||||
tipforward := flag.String("nf", "on", "No Forward mode. If true, netclient will not check for IP forwarding. This may break functionality")
|
||||
command := flag.String("c", "required", "The command to run")
|
||||
|
||||
|
||||
flag.Parse()
|
||||
|
||||
|
||||
|
||||
getID := exec.Command("id", "-u")
|
||||
out, err := getID.Output()
|
||||
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
id, err := strconv.Atoi(string(out[:len(out)-1]))
|
||||
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
|
||||
if id != 0 {
|
||||
log.Fatal("This program must be run with elevated privileges (sudo). This program installs a SystemD service and configures WireGuard and networking rules. Please re-run with sudo/root.")
|
||||
}
|
||||
|
||||
|
||||
_, err = exec.LookPath("wg")
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
log.Fatal("WireGuard not installed. Please install WireGuard (wireguard-tools) and try again.")
|
||||
}
|
||||
|
||||
switch *command {
|
||||
case "getport":
|
||||
portno, err := functions.GetFreePort(51821)
|
||||
fmt.Printf("Port Number: %v", portno)
|
||||
fmt.Println("")
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
case "required":
|
||||
fmt.Println("command flag 'c' is required. Pick one of |install|checkin|update|remove|")
|
||||
os.Exit(1)
|
||||
log.Fatal("Exiting")
|
||||
case "install":
|
||||
|
||||
if *taccesstoken == "badtoken" && (*tnetwork == "nonetwork" || *tnetwork == "") {
|
||||
fmt.Println("Required, '-n'. No network provided. Exiting.")
|
||||
os.Exit(1)
|
||||
}
|
||||
/*
|
||||
if !*tnoforward {
|
||||
forward := exec.Command("sysctl", "net.ipv4.ip_forward")
|
||||
out, err := forward.Output()
|
||||
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
//s := strings.Split(string(out), " ", "\n")
|
||||
s := strings.Fields(string(out))
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
if s[2] != "1" {
|
||||
log.Fatal("It is recommended to enable IP Forwarding. Current status is: " + s[2] + ", but should be 1. if you would like to run without IP Forwarding, re-run with flag '-nf true'")
|
||||
}
|
||||
}
|
||||
*/
|
||||
fmt.Println("Beginning agent installation.")
|
||||
err := functions.Install(*taccesskey, *tpassword, *tserver, *tnetwork, *tnoauto, *taccesstoken, *tname, *tpublicip, *tdnsoff, *tipforward)
|
||||
if err != nil {
|
||||
fmt.Println("Error encountered while installing.")
|
||||
if !strings.Contains(err.Error(), "ALREADY_INSTALLED") {
|
||||
fmt.Println("Error installing: ", err)
|
||||
fmt.Println("Cleaning up (uninstall)")
|
||||
err = functions.Remove(*tnetwork)
|
||||
if err != nil {
|
||||
fmt.Println("Error uninstalling: ", err)
|
||||
fmt.Println("Wiping local.")
|
||||
err = functions.WipeLocal(*tnetwork)
|
||||
if err != nil {
|
||||
fmt.Println("Error removing artifacts: ", err)
|
||||
}
|
||||
err = functions.RemoveSystemDServices(*tnetwork)
|
||||
if err != nil {
|
||||
fmt.Println("Error removing services: ", err)
|
||||
}
|
||||
}
|
||||
os.Exit(1)
|
||||
} else {
|
||||
fmt.Println(err.Error())
|
||||
os.Exit(1)
|
||||
}
|
||||
}
|
||||
/*
|
||||
case "service-install":
|
||||
fmt.Println("Beginning service installation.")
|
||||
err := functions.ConfigureSystemD()
|
||||
if err != nil {
|
||||
fmt.Println("Error installing service: ", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
case "service-uninstall":
|
||||
fmt.Println("Beginning service uninstall.")
|
||||
err := functions.RemoveSystemDServices()
|
||||
if err != nil {
|
||||
fmt.Println("Error installing service: ", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
*/
|
||||
case "checkin":
|
||||
if *tnetwork == "nonetwork" || *tnetwork == "" {
|
||||
fmt.Println("Required, '-n'. No network provided. Exiting.")
|
||||
os.Exit(1)
|
||||
}
|
||||
fmt.Println("Beginning node check in for network " + *tnetwork)
|
||||
err := functions.CheckIn(*tnetwork)
|
||||
if err != nil {
|
||||
fmt.Println("Error checking in: ", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
case "remove":
|
||||
if *tnetwork == "nonetwork" || *tnetwork == "" {
|
||||
fmt.Println("Required, '-n'. No network provided. Exiting.")
|
||||
os.Exit(1)
|
||||
}
|
||||
fmt.Println("Beginning node cleanup.")
|
||||
err := functions.Remove(*tnetwork)
|
||||
if err != nil {
|
||||
/*
|
||||
fmt.Println("Error uninstalling: ", err)
|
||||
fmt.Println("Wiping local.")
|
||||
err = functions.WipeLocal()
|
||||
if err != nil {
|
||||
fmt.Println("Error removing artifacts: ", err)
|
||||
}
|
||||
err = functions.RemoveSystemDServices()
|
||||
if err != nil {
|
||||
fmt.Println("Error removing services: ", err)
|
||||
}
|
||||
*/
|
||||
fmt.Println("Error deleting node: ", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
default:
|
||||
fmt.Println("You must select from the following commands: install|remove|checkin", err)
|
||||
os.Exit(1)
|
||||
|
||||
}
|
||||
fmt.Println("Command " + *command + " Executed Successfully")
|
||||
}
|
||||
Reference in New Issue
Block a user