| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105 | package controllerimport (	"encoding/json"	"errors"	"fmt"	"net"	"net/http"	"reflect"	"strconv"	"strings"	"github.com/go-playground/validator/v10"	"github.com/gorilla/mux"	"github.com/gravitl/netmaker/database"	"github.com/gravitl/netmaker/logger"	"github.com/gravitl/netmaker/logic"	"github.com/gravitl/netmaker/servercfg"	"github.com/gravitl/netmaker/models"	"github.com/gravitl/netmaker/mq"	"github.com/skip2/go-qrcode"	"golang.org/x/exp/slices"	"golang.org/x/exp/slog"	"golang.zx2c4.com/wireguard/wgctrl/wgtypes")func extClientHandlers(r *mux.Router) {	r.HandleFunc("/api/extclients", logic.SecurityCheck(true, http.HandlerFunc(getAllExtClients))).		Methods(http.MethodGet)	r.HandleFunc("/api/extclients/{network}", logic.SecurityCheck(true, http.HandlerFunc(getNetworkExtClients))).		Methods(http.MethodGet)	r.HandleFunc("/api/extclients/{network}/{clientid}", logic.SecurityCheck(false, http.HandlerFunc(getExtClient))).		Methods(http.MethodGet)	r.HandleFunc("/api/extclients/{network}/{clientid}/{type}", logic.SecurityCheck(false, http.HandlerFunc(getExtClientConf))).		Methods(http.MethodGet)	r.HandleFunc("/api/extclients/{network}/{clientid}", logic.SecurityCheck(false, http.HandlerFunc(updateExtClient))).		Methods(http.MethodPut)	r.HandleFunc("/api/extclients/{network}/{clientid}", logic.SecurityCheck(false, http.HandlerFunc(deleteExtClient))).		Methods(http.MethodDelete)	r.HandleFunc("/api/extclients/{network}/{nodeid}", logic.SecurityCheck(false, checkFreeTierLimits(limitChoiceMachines, http.HandlerFunc(createExtClient)))).		Methods(http.MethodPost)	r.HandleFunc("/api/v1/client_conf/{network}", logic.SecurityCheck(false, http.HandlerFunc(getExtClientHAConf))).Methods(http.MethodGet)}func checkIngressExists(nodeID string) bool {	node, err := logic.GetNodeByID(nodeID)	if err != nil {		return false	}	return node.IsIngressGateway}// @Summary     Get all remote access client associated with network// @Router      /api/extclients/{network} [get]// @Tags        Remote Access Client// @Security    oauth2// @Success     200 {object} models.ExtClient// @Failure     500 {object} models.ErrorResponsefunc getNetworkExtClients(w http.ResponseWriter, r *http.Request) {	w.Header().Set("Content-Type", "application/json")	var extclients []models.ExtClient	var params = mux.Vars(r)	network := params["network"]	extclients, err := logic.GetNetworkExtClients(network)	if err != nil {		logger.Log(0, r.Header.Get("user"),			fmt.Sprintf("failed to get ext clients for network [%s]: %v", network, err))		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))		return	}	//Returns all the extclients in JSON format	w.WriteHeader(http.StatusOK)	json.NewEncoder(w).Encode(extclients)}// @Summary     Fetches All Remote Access Clients across all networks// @Router      /api/extclients [get]// @Tags        Remote Access Client// @Security    oauth2// @Success     200 {object} models.ExtClient// @Failure     500 {object} models.ErrorResponse// Not quite sure if this is necessary. Probably necessary based on front end but may// want to review after iteration 1 if it's being used or notfunc getAllExtClients(w http.ResponseWriter, r *http.Request) {	w.Header().Set("Content-Type", "application/json")	clients, err := logic.GetAllExtClients()	if err != nil && !database.IsEmptyRecord(err) {		logger.Log(0, "failed to get all extclients: ", err.Error())		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))		return	}	//Return all the extclients in JSON format	logic.SortExtClient(clients[:])	w.WriteHeader(http.StatusOK)	json.NewEncoder(w).Encode(clients)}// @Summary     Get an individual remote access client// @Router      /api/extclients/{network}/{clientid} [get]// @Tags        Remote Access Client// @Security    oauth2// @Success     200 {object} models.ExtClient// @Failure     500 {object} models.ErrorResponse// @Failure     403 {object} models.ErrorResponsefunc getExtClient(w http.ResponseWriter, r *http.Request) {	// set header.	w.Header().Set("Content-Type", "application/json")	var params = mux.Vars(r)	clientid := params["clientid"]	network := params["network"]	client, err := logic.GetExtClient(clientid, network)	if err != nil {		logger.Log(			0,			r.Header.Get("user"),			fmt.Sprintf("failed to get extclient for [%s] on network [%s]: %v",				clientid, network, err),		)		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))		return	}	w.WriteHeader(http.StatusOK)	json.NewEncoder(w).Encode(client)}// @Summary     Get an individual remote access client// @Router      /api/extclients/{network}/{clientid}/{type} [get]// @Tags        Remote Access Client// @Security    oauth2// @Success     200 {object} models.ExtClient// @Failure     500 {object} models.ErrorResponse// @Failure     403 {object} models.ErrorResponsefunc getExtClientConf(w http.ResponseWriter, r *http.Request) {	// set header.	w.Header().Set("Content-Type", "application/json")	var params = mux.Vars(r)	clientid := params["clientid"]	networkid := params["network"]	client, err := logic.GetExtClient(clientid, networkid)	if err != nil {		logger.Log(			0,			r.Header.Get("user"),			fmt.Sprintf("failed to get extclient for [%s] on network [%s]: %v",				clientid, networkid, err),		)		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))		return	}	gwnode, err := logic.GetNodeByID(client.IngressGatewayID)	if err != nil {		logger.Log(			0,			r.Header.Get("user"),			fmt.Sprintf(				"failed to get ingress gateway node [%s] info: %v",				client.IngressGatewayID,				err,			),		)		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))		return	}	host, err := logic.GetHost(gwnode.HostID.String())	if err != nil {		logger.Log(			0,			r.Header.Get("user"),			fmt.Sprintf(				"failed to get host for ingress gateway node [%s] info: %v",				client.IngressGatewayID,				err,			),		)		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))		return	}	network, err := logic.GetParentNetwork(client.Network)	if err != nil {		logger.Log(			1,			r.Header.Get("user"),			"Could not retrieve Ingress Gateway Network",			client.Network,		)		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))		return	}	preferredIp := strings.TrimSpace(r.URL.Query().Get("preferredip"))	if preferredIp != "" {		allowedPreferredIps := []string{}		for i := range gwnode.AdditionalRagIps {			allowedPreferredIps = append(allowedPreferredIps, gwnode.AdditionalRagIps[i].String())		}		allowedPreferredIps = append(allowedPreferredIps, host.EndpointIP.String())		allowedPreferredIps = append(allowedPreferredIps, host.EndpointIPv6.String())		if !slices.Contains(allowedPreferredIps, preferredIp) {			slog.Warn(				"preferred endpoint ip is not associated with the RAG. proceeding with preferred ip",				"preferred ip",				preferredIp,			)			logic.ReturnErrorResponse(				w,				r,				logic.FormatError(					errors.New("preferred endpoint ip is not associated with the RAG"),					"badrequest",				),			)			return		}		if net.ParseIP(preferredIp).To4() == nil {			preferredIp = fmt.Sprintf("[%s]", preferredIp)		}	}	addrString := client.Address	if addrString != "" {		addrString += "/32"	}	if client.Address6 != "" {		if addrString != "" {			addrString += ","		}		addrString += client.Address6 + "/128"	}	keepalive := ""	if network.DefaultKeepalive != 0 {		keepalive = "PersistentKeepalive = " + strconv.Itoa(int(network.DefaultKeepalive))	}	if gwnode.IngressPersistentKeepalive != 0 {		keepalive = "PersistentKeepalive = " + strconv.Itoa(int(gwnode.IngressPersistentKeepalive))	}	gwendpoint := ""	if preferredIp == "" {		if host.EndpointIP.To4() == nil {			gwendpoint = fmt.Sprintf("[%s]:%d", host.EndpointIPv6.String(), host.ListenPort)		} else {			gwendpoint = fmt.Sprintf("%s:%d", host.EndpointIP.String(), host.ListenPort)		}	} else {		gwendpoint = fmt.Sprintf("%s:%d", preferredIp, host.ListenPort)	}	var newAllowedIPs string	if logic.IsInternetGw(gwnode) || gwnode.InternetGwID != "" {		egressrange := "0.0.0.0/0"		if gwnode.Address6.IP != nil && client.Address6 != "" {			egressrange += "," + "::/0"		}		newAllowedIPs = egressrange	} else {		newAllowedIPs = network.AddressRange		if newAllowedIPs != "" && network.AddressRange6 != "" {			newAllowedIPs += ","		}		if network.AddressRange6 != "" {			newAllowedIPs += network.AddressRange6		}		if egressGatewayRanges, err := logic.GetEgressRangesOnNetwork(&client); err == nil {			for _, egressGatewayRange := range egressGatewayRanges {				newAllowedIPs += "," + egressGatewayRange			}		}	}	defaultDNS := ""	if client.DNS != "" {		defaultDNS = "DNS = " + client.DNS	} else if gwnode.IngressDNS != "" {		defaultDNS = "DNS = " + gwnode.IngressDNS	}	if client.DNS == "" {		if len(network.NameServers) > 0 {			if defaultDNS == "" {				defaultDNS = "DNS = " + strings.Join(network.NameServers, ",")			} else {				defaultDNS += "," + strings.Join(network.NameServers, ",")			}		}	}	// if servercfg.GetManageDNS() {	// 	if gwnode.Address6.IP != nil {	// 		if defaultDNS == "" {	// 			defaultDNS = "DNS = " + gwnode.Address6.IP.String()	// 		} else {	// 			defaultDNS = defaultDNS + ", " + gwnode.Address6.IP.String()	// 		}	// 	}	// 	if gwnode.Address.IP != nil {	// 		if defaultDNS == "" {	// 			defaultDNS = "DNS = " + gwnode.Address.IP.String()	// 		} else {	// 			defaultDNS = defaultDNS + ", " + gwnode.Address.IP.String()	// 		}	// 	}	// }	defaultMTU := 1420	if host.MTU != 0 {		defaultMTU = host.MTU	}	if gwnode.IngressMTU != 0 {		defaultMTU = int(gwnode.IngressMTU)	}	postUp := strings.Builder{}	if client.PostUp != "" && params["type"] != "qr" {		for _, loc := range strings.Split(client.PostUp, "\n") {			postUp.WriteString(fmt.Sprintf("PostUp = %s\n", loc))		}	}	postDown := strings.Builder{}	if client.PostDown != "" && params["type"] != "qr" {		for _, loc := range strings.Split(client.PostDown, "\n") {			postDown.WriteString(fmt.Sprintf("PostDown = %s\n", loc))		}	}	config := fmt.Sprintf(`[Interface]Address = %sPrivateKey = %sMTU = %d%s%s%s[Peer]PublicKey = %sAllowedIPs = %sEndpoint = %s%s`, addrString,		client.PrivateKey,		defaultMTU,		defaultDNS,		postUp.String(),		postDown.String(),		host.PublicKey,		newAllowedIPs,		gwendpoint,		keepalive,	)	if params["type"] == "qr" {		bytes, err := qrcode.Encode(config, qrcode.Medium, 220)		if err != nil {			logger.Log(1, r.Header.Get("user"), "failed to encode qr code: ", err.Error())			logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))			return		}		w.Header().Set("Content-Type", "image/png")		w.WriteHeader(http.StatusOK)		_, err = w.Write(bytes)		if err != nil {			logger.Log(1, r.Header.Get("user"), "response writer error (qr) ", err.Error())			logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))			return		}		return	}	if params["type"] == "file" {		name := client.ClientID + ".conf"		w.Header().Set("Content-Type", "application/config")		w.Header().Set("Content-Disposition", "attachment; filename=\""+name+"\"")		w.WriteHeader(http.StatusOK)		_, err := fmt.Fprint(w, config)		if err != nil {			logger.Log(1, r.Header.Get("user"), "response writer error (file) ", err.Error())			logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))		}		return	}	logger.Log(2, r.Header.Get("user"), "retrieved ext client config")	w.WriteHeader(http.StatusOK)	json.NewEncoder(w).Encode(client)}// @Summary     Get an individual remote access client// @Router      /api/extclients/{network}/{clientid}/{type} [get]// @Tags        Remote Access Client// @Security    oauth2// @Success     200 {object} models.ExtClient// @Failure     500 {object} models.ErrorResponse// @Failure     403 {object} models.ErrorResponsefunc getExtClientHAConf(w http.ResponseWriter, r *http.Request) {	var params = mux.Vars(r)	networkid := params["network"]	network, err := logic.GetParentNetwork(networkid)	if err != nil {		logger.Log(			1,			r.Header.Get("user"),			"Could not retrieve Ingress Gateway Network",			networkid,		)		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))		return	}	// fetch client based on availability	nodes, _ := logic.GetNetworkNodes(networkid)	defaultPolicy, _ := logic.GetDefaultPolicy(models.NetworkID(networkid), models.DevicePolicy)	var targetGwID string	var connectionCnt int = -1	for _, nodeI := range nodes {		if nodeI.IsGw {			// check health status			logic.GetNodeStatus(&nodeI, defaultPolicy.Enabled)			if nodeI.Status != models.OnlineSt {				continue			}			// Get Total connections on the gw			clients := logic.GetGwExtclients(nodeI.ID.String(), networkid)			if connectionCnt == -1 || len(clients) < connectionCnt {				connectionCnt = len(clients)				targetGwID = nodeI.ID.String()			}		}	}	gwnode, err := logic.GetNodeByID(targetGwID)	if err != nil {		logger.Log(			0,			r.Header.Get("user"),			fmt.Sprintf(				"failed to get ingress gateway node [%s] info: %v",				gwnode.ID,				err,			),		)		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))		return	}	host, err := logic.GetHost(gwnode.HostID.String())	if err != nil {		logger.Log(0, r.Header.Get("user"),			fmt.Sprintf("failed to get ingress gateway host for node [%s] info: %v", gwnode.ID, err))		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))		return	}	var userName string	if r.Header.Get("ismaster") == "yes" {		userName = logic.MasterUser	} else {		caller, err := logic.GetUser(r.Header.Get("user"))		if err != nil {			logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))			return		}		userName = caller.UserName	}	// create client	var extclient models.ExtClient	extclient.OwnerID = userName	extclient.IngressGatewayID = targetGwID	extclient.Network = networkid	extclient.Tags = make(map[models.TagID]struct{})	// extclient.Tags[models.TagID(fmt.Sprintf("%s.%s", extclient.Network,	// 	models.RemoteAccessTagName))] = struct{}{}	// set extclient dns to ingressdns if extclient dns is not explicitly set	if (extclient.DNS == "") && (gwnode.IngressDNS != "") {		network, _ := logic.GetNetwork(gwnode.Network)		dns := gwnode.IngressDNS		if len(network.NameServers) > 0 {			if dns == "" {				dns = strings.Join(network.NameServers, ",")			} else {				dns += "," + strings.Join(network.NameServers, ",")			}		}		extclient.DNS = dns	}	listenPort := logic.GetPeerListenPort(host)	extclient.IngressGatewayEndpoint = fmt.Sprintf("%s:%d", host.EndpointIP.String(), listenPort)	extclient.Enabled = true	if err = logic.CreateExtClient(&extclient); err != nil {		slog.Error(			"failed to create extclient",			"user",			r.Header.Get("user"),			"network",			networkid,			"error",			err,		)		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))		return	}	client, err := logic.GetExtClient(extclient.ClientID, networkid)	if err != nil {		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))		return	}	addrString := client.Address	if addrString != "" {		addrString += "/32"	}	if client.Address6 != "" {		if addrString != "" {			addrString += ","		}		addrString += client.Address6 + "/128"	}	keepalive := ""	if network.DefaultKeepalive != 0 {		keepalive = "PersistentKeepalive = " + strconv.Itoa(int(network.DefaultKeepalive))	}	if gwnode.IngressPersistentKeepalive != 0 {		keepalive = "PersistentKeepalive = " + strconv.Itoa(int(gwnode.IngressPersistentKeepalive))	}	var newAllowedIPs string	if logic.IsInternetGw(gwnode) || gwnode.InternetGwID != "" {		egressrange := "0.0.0.0/0"		if gwnode.Address6.IP != nil && client.Address6 != "" {			egressrange += "," + "::/0"		}		newAllowedIPs = egressrange	} else {		newAllowedIPs = network.AddressRange		if newAllowedIPs != "" && network.AddressRange6 != "" {			newAllowedIPs += ","		}		if network.AddressRange6 != "" {			newAllowedIPs += network.AddressRange6		}		if egressGatewayRanges, err := logic.GetEgressRangesOnNetwork(&client); err == nil {			for _, egressGatewayRange := range egressGatewayRanges {				newAllowedIPs += "," + egressGatewayRange			}		}	}	gwendpoint := ""	if host.EndpointIP.To4() == nil {		gwendpoint = fmt.Sprintf("[%s]:%d", host.EndpointIPv6.String(), host.ListenPort)	} else {		gwendpoint = fmt.Sprintf("%s:%d", host.EndpointIP.String(), host.ListenPort)	}	defaultDNS := ""	if client.DNS != "" {		defaultDNS = "DNS = " + client.DNS	} else if gwnode.IngressDNS != "" {		defaultDNS = "DNS = " + gwnode.IngressDNS	}	defaultMTU := 1420	if host.MTU != 0 {		defaultMTU = host.MTU	}	if gwnode.IngressMTU != 0 {		defaultMTU = int(gwnode.IngressMTU)	}	postUp := strings.Builder{}	if client.PostUp != "" && params["type"] != "qr" {		for _, loc := range strings.Split(client.PostUp, "\n") {			postUp.WriteString(fmt.Sprintf("PostUp = %s\n", loc))		}	}	postDown := strings.Builder{}	if client.PostDown != "" && params["type"] != "qr" {		for _, loc := range strings.Split(client.PostDown, "\n") {			postDown.WriteString(fmt.Sprintf("PostDown = %s\n", loc))		}	}	config := fmt.Sprintf(`[Interface]Address = %sPrivateKey = %sMTU = %d%s%s%s[Peer]PublicKey = %sAllowedIPs = %sEndpoint = %s%s`, addrString,		client.PrivateKey,		defaultMTU,		defaultDNS,		postUp.String(),		postDown.String(),		host.PublicKey,		newAllowedIPs,		gwendpoint,		keepalive,	)	go func() {		if err := logic.SetClientDefaultACLs(&extclient); err != nil {			slog.Error(				"failed to set default acls for extclient",				"user",				r.Header.Get("user"),				"network",				networkid,				"error",				err,			)			logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))			return		}		if err := mq.PublishPeerUpdate(false); err != nil {			logger.Log(1, "error publishing peer update ", err.Error())		}		if servercfg.IsDNSMode() {			logic.SetDNS()		}	}()	name := client.ClientID + ".conf"	w.Header().Set("Content-Type", "application/config")	w.Header().Set("Content-Disposition", "attachment; filename=\""+name+"\"")	w.WriteHeader(http.StatusOK)	_, err = fmt.Fprint(w, config)	if err != nil {		logger.Log(1, r.Header.Get("user"), "response writer error (file) ", err.Error())		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))	}}// @Summary     Create an individual remote access client// @Router      /api/extclients/{network}/{nodeid} [post]// @Tags        Remote Access Client// @Security    oauth2// @Success     200 {string} string "OK"// @Failure     500 {object} models.ErrorResponse// @Failure     400 {object} models.ErrorResponse// @Failure     403 {object} models.ErrorResponsefunc createExtClient(w http.ResponseWriter, r *http.Request) {	w.Header().Set("Content-Type", "application/json")	var params = mux.Vars(r)	nodeid := params["nodeid"]	ingressExists := checkIngressExists(nodeid)	if !ingressExists {		err := errors.New("ingress does not exist")		slog.Error("failed to create extclient", "user", r.Header.Get("user"), "error", err)		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))		return	}	var customExtClient models.CustomExtClient	if err := json.NewDecoder(r.Body).Decode(&customExtClient); err != nil {		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))		return	}	if err := validateCustomExtClient(&customExtClient, true); err != nil {		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))		return	}	var gateway models.EgressGatewayRequest	gateway.NetID = params["network"]	gateway.Ranges = customExtClient.ExtraAllowedIPs	err := logic.ValidateEgressRange(gateway)	if err != nil {		logger.Log(0, r.Header.Get("user"), "error validating egress range: ", err.Error())		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))		return	}	node, err := logic.GetNodeByID(nodeid)	if err != nil {		logger.Log(0, r.Header.Get("user"),			fmt.Sprintf("failed to get ingress gateway node [%s] info: %v", nodeid, err))		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))		return	}	var userName string	if r.Header.Get("ismaster") == "yes" {		userName = logic.MasterUser	} else {		caller, err := logic.GetUser(r.Header.Get("user"))		if err != nil {			logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))			return		}		userName = caller.UserName		// check if user has a config already for remote access client		extclients, err := logic.GetNetworkExtClients(node.Network)		if err != nil {			slog.Error("failed to get extclients", "error", err)			logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))			return		}		for _, extclient := range extclients {			if extclient.RemoteAccessClientID != "" &&				extclient.RemoteAccessClientID == customExtClient.RemoteAccessClientID && extclient.OwnerID == caller.UserName && nodeid == extclient.IngressGatewayID {				// extclient on the gw already exists for the remote access client				err = errors.New("remote client config already exists on the gateway")				slog.Error("failed to create extclient", "user", userName, "error", err)				logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))				return			}		}	}	extclient := logic.UpdateExtClient(&models.ExtClient{}, &customExtClient)	extclient.OwnerID = userName	extclient.RemoteAccessClientID = customExtClient.RemoteAccessClientID	extclient.IngressGatewayID = nodeid	extclient.Network = node.Network	extclient.Tags = make(map[models.TagID]struct{})	// extclient.Tags[models.TagID(fmt.Sprintf("%s.%s", extclient.Network,	// 	models.RemoteAccessTagName))] = struct{}{}	// set extclient dns to ingressdns if extclient dns is not explicitly set	if (extclient.DNS == "") && (node.IngressDNS != "") {		network, _ := logic.GetNetwork(node.Network)		dns := node.IngressDNS		if len(network.NameServers) > 0 {			if dns == "" {				dns = strings.Join(network.NameServers, ",")			} else {				dns += "," + strings.Join(network.NameServers, ",")			}		}		extclient.DNS = dns	}	host, err := logic.GetHost(node.HostID.String())	if err != nil {		logger.Log(0, r.Header.Get("user"),			fmt.Sprintf("failed to get ingress gateway host for node [%s] info: %v", nodeid, err))		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))		return	}	listenPort := logic.GetPeerListenPort(host)	extclient.IngressGatewayEndpoint = fmt.Sprintf("%s:%d", host.EndpointIP.String(), listenPort)	extclient.Enabled = true	parentNetwork, err := logic.GetNetwork(node.Network)	if err == nil { // check if parent network default ACL is enabled (yes) or not (no)		extclient.Enabled = parentNetwork.DefaultACL == "yes"	}	extclient.Os = customExtClient.Os	extclient.DeviceName = customExtClient.DeviceName	if customExtClient.IsAlreadyConnectedToInetGw {		slog.Warn("RAC/Client is already connected to internet gateway. this may mask their real IP address", "client IP", customExtClient.PublicEndpoint)	}	extclient.PublicEndpoint = customExtClient.PublicEndpoint	extclient.Country = customExtClient.Country	if err = logic.CreateExtClient(&extclient); err != nil {		slog.Error(			"failed to create extclient",			"user",			r.Header.Get("user"),			"network",			node.Network,			"error",			err,		)		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))		return	}	slog.Info(		"created extclient",		"user",		r.Header.Get("user"),		"network",		node.Network,		"clientid",		extclient.ClientID,	)	w.WriteHeader(http.StatusOK)	go func() {		if err := logic.SetClientDefaultACLs(&extclient); err != nil {			slog.Error(				"failed to set default acls for extclient",				"user",				r.Header.Get("user"),				"network",				node.Network,				"error",				err,			)			logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))			return		}		if err := mq.PublishPeerUpdate(false); err != nil {			logger.Log(1, "error setting ext peers on "+nodeid+": "+err.Error())		}		if servercfg.IsDNSMode() {			logic.SetDNS()		}	}()}// @Summary     Update an individual remote access client// @Router      /api/extclients/{network}/{clientid} [put]// @Tags        Remote Access Client// @Security    oauth2// @Success     200 {object} models.ExtClient// @Failure     500 {object} models.ErrorResponse// @Failure     400 {object} models.ErrorResponse// @Failure     403 {object} models.ErrorResponsefunc updateExtClient(w http.ResponseWriter, r *http.Request) {	w.Header().Set("Content-Type", "application/json")	var params = mux.Vars(r)	var update models.CustomExtClient	//var oldExtClient models.ExtClient	var sendPeerUpdate bool	var replacePeers bool	err := json.NewDecoder(r.Body).Decode(&update)	if err != nil {		logger.Log(0, r.Header.Get("user"), "error decoding request body: ",			err.Error())		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))		return	}	clientid := params["clientid"]	oldExtClient, err := logic.GetExtClientByName(clientid)	if err != nil {		slog.Error(			"failed to retrieve extclient",			"user",			r.Header.Get("user"),			"id",			clientid,			"error",			err,		)		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))		return	}	if oldExtClient.ClientID == update.ClientID {		if err := validateCustomExtClient(&update, false); err != nil {			logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))			return		}	} else {		if err := validateCustomExtClient(&update, true); err != nil {			logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))			return		}	}	var gateway models.EgressGatewayRequest	gateway.NetID = params["network"]	gateway.Ranges = update.ExtraAllowedIPs	err = logic.ValidateEgressRange(gateway)	if err != nil {		logger.Log(0, r.Header.Get("user"), "error validating egress range: ", err.Error())		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))		return	}	var changedID = update.ClientID != oldExtClient.ClientID	if !reflect.DeepEqual(update.DeniedACLs, oldExtClient.DeniedACLs) {		sendPeerUpdate = true		logic.SetClientACLs(&oldExtClient, update.DeniedACLs)	}	if !logic.IsSlicesEqual(update.ExtraAllowedIPs, oldExtClient.ExtraAllowedIPs) {		sendPeerUpdate = true	}	if update.Enabled != oldExtClient.Enabled {		sendPeerUpdate = true	}	if update.PublicKey != oldExtClient.PublicKey {		//remove old peer entry		sendPeerUpdate = true		replacePeers = true	}	newclient := logic.UpdateExtClient(&oldExtClient, &update)	if err := logic.DeleteExtClient(oldExtClient.Network, oldExtClient.ClientID); err != nil {		slog.Error(			"failed to delete ext client",			"user",			r.Header.Get("user"),			"id",			oldExtClient.ClientID,			"network",			oldExtClient.Network,			"error",			err,		)		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))		return	}	if err := logic.SaveExtClient(&newclient); err != nil {		slog.Error(			"failed to save ext client",			"user",			r.Header.Get("user"),			"id",			newclient.ClientID,			"network",			newclient.Network,			"error",			err,		)		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))		return	}	logger.Log(0, r.Header.Get("user"), "updated ext client", update.ClientID)	w.WriteHeader(http.StatusOK)	json.NewEncoder(w).Encode(newclient)	go func() {		if changedID && servercfg.IsDNSMode() {			logic.SetDNS()		}		if replacePeers {			if err := mq.PublishDeletedClientPeerUpdate(&oldExtClient); err != nil {				slog.Error("error deleting old ext peers", "error", err.Error())			}		}		if sendPeerUpdate { // need to send a peer update to the ingress node as enablement of one of it's clients has changed			ingressNode, err := logic.GetNodeByID(newclient.IngressGatewayID)			if err == nil {				if err = mq.PublishPeerUpdate(false); err != nil {					logger.Log(						1,						"error setting ext peers on",						ingressNode.ID.String(),						":",						err.Error(),					)				}			}			if !update.Enabled {				ingressHost, err := logic.GetHost(ingressNode.HostID.String())				if err != nil {					slog.Error(						"Failed to get ingress host",						"node",						ingressNode.ID.String(),						"error",						err,					)					return				}				nodes, err := logic.GetAllNodes()				if err != nil {					slog.Error("Failed to get nodes", "error", err)					return				}				go mq.PublishSingleHostPeerUpdate(ingressHost, nodes, nil, []models.ExtClient{oldExtClient}, false, nil)			}		}	}()}// @Summary     Delete an individual remote access client// @Router      /api/extclients/{network}/{clientid} [delete]// @Tags        Remote Access Client// @Security    oauth2// @Success     200// @Failure     500 {object} models.ErrorResponse// @Failure     403 {object} models.ErrorResponsefunc deleteExtClient(w http.ResponseWriter, r *http.Request) {	// Set header	w.Header().Set("Content-Type", "application/json")	// get params	var params = mux.Vars(r)	clientid := params["clientid"]	network := params["network"]	extclient, err := logic.GetExtClient(clientid, network)	if err != nil {		err = errors.New("Could not delete extclient " + params["clientid"])		logger.Log(0, r.Header.Get("user"),			fmt.Sprintf("failed to get extclient [%s],network [%s]: %v", clientid, network, err))		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))		return	}	ingressnode, err := logic.GetNodeByID(extclient.IngressGatewayID)	if err != nil {		logger.Log(			0,			r.Header.Get("user"),			fmt.Sprintf(				"failed to get ingress gateway node [%s] info: %v",				extclient.IngressGatewayID,				err,			),		)		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))		return	}	err = logic.DeleteExtClientAndCleanup(extclient)	if err != nil {		slog.Error("deleteExtClient: ", "Error", err.Error())		err = errors.New("Could not delete extclient " + params["clientid"])		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))		return	}	go func() {		if err := mq.PublishDeletedClientPeerUpdate(&extclient); err != nil {			slog.Error("error setting ext peers on " + ingressnode.ID.String() + ": " + err.Error())		}		if servercfg.IsDNSMode() {			logic.SetDNS()		}	}()	logger.Log(0, r.Header.Get("user"),		"Deleted extclient client", params["clientid"], "from network", params["network"])	logic.ReturnSuccessResponse(w, r, params["clientid"]+" deleted.")}// validateCustomExtClient	Validates the extclient objectfunc validateCustomExtClient(customExtClient *models.CustomExtClient, checkID bool) error {	v := validator.New()	err := v.Struct(customExtClient)	if err != nil {		return err	}	//validate clientid	if customExtClient.ClientID != "" {		if err := isValid(customExtClient.ClientID, checkID); err != nil {			return fmt.Errorf("client validation: %v", err)		}	}	//extclient.ClientID = customExtClient.ClientID	if len(customExtClient.PublicKey) > 0 {		if _, err := wgtypes.ParseKey(customExtClient.PublicKey); err != nil {			return errInvalidExtClientPubKey		}		//extclient.PublicKey = customExtClient.PublicKey	}	//validate extra ips	if len(customExtClient.ExtraAllowedIPs) > 0 {		for _, ip := range customExtClient.ExtraAllowedIPs {			if _, _, err := net.ParseCIDR(ip); err != nil {				return errInvalidExtClientExtraIP			}		}		//extclient.ExtraAllowedIPs = customExtClient.ExtraAllowedIPs	}	//validate DNS	if customExtClient.DNS != "" {		ips := strings.Split(customExtClient.DNS, ",")		for _, ip := range ips {			trimmedIp := strings.TrimSpace(ip)			if ip := net.ParseIP(trimmedIp); ip == nil {				return errInvalidExtClientDNS			}		}	}	return nil}// isValid	Checks if the clientid is validfunc isValid(clientid string, checkID bool) error {	if !validName(clientid) {		return errInvalidExtClientID	}	if checkID {		extclients, err := logic.GetAllExtClients()		if err != nil {			return fmt.Errorf("extclients isValid: %v", err)		}		for _, extclient := range extclients {			if clientid == extclient.ClientID {				return errDuplicateExtClientName			}		}	}	return nil}
 |