| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139 | package controllerimport (	"context"	"encoding/json"	"errors"	"fmt"	"net"	"net/http"	"os"	"reflect"	"strconv"	"strings"	"github.com/go-playground/validator/v10"	"github.com/gorilla/mux"	"github.com/gravitl/netmaker/database"	"github.com/gravitl/netmaker/db"	"github.com/gravitl/netmaker/logger"	"github.com/gravitl/netmaker/logic"	"github.com/gravitl/netmaker/schema"	"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	}	eli, _ := (&schema.Egress{Network: gwnode.Network}).ListByNetwork(db.WithContext(context.TODO()))	acls, _ := logic.ListAclsByNetwork(models.NetworkID(client.Network))	logic.GetNodeEgressInfo(&gwnode, eli, acls)	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) {		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.NetID, gateway.Ranges)	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 customExtClient.RemoteAccessClientID != "" && customExtClient.Location == "" {		extclient.Location = logic.GetHostLocInfo(logic.GetClientIP(r), os.Getenv("IP_INFO_TOKEN"))	}	extclient.Location = customExtClient.Location	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,	)	if extclient.RemoteAccessClientID != "" {		// if created by user from client app, log event		logic.LogEvent(&models.Event{			Action: models.Connect,			Source: models.Subject{				ID:   userName,				Name: userName,				Type: models.UserSub,			},			TriggeredBy: userName,			Target: models.Subject{				ID:   extclient.Network,				Name: extclient.Network,				Type: models.NetworkSub,				Info: extclient,			},			NetworkID: models.NetworkID(extclient.Network),			Origin:    models.ClientApp,		})	}	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.NetID, gateway.Ranges)	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	}	if update.RemoteAccessClientID != "" && update.Location == "" {		update.Location = logic.GetHostLocInfo(logic.GetClientIP(r), os.Getenv("IP_INFO_TOKEN"))	}	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}
 |