mirror of
https://github.com/gravitl/netmaker.git
synced 2025-11-01 20:52:57 +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