ext_client.go 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825
  1. package controller
  2. import (
  3. "encoding/json"
  4. "errors"
  5. "fmt"
  6. "net"
  7. "net/http"
  8. "reflect"
  9. "strconv"
  10. "strings"
  11. "github.com/go-playground/validator/v10"
  12. "github.com/gorilla/mux"
  13. "github.com/gravitl/netmaker/database"
  14. "github.com/gravitl/netmaker/logger"
  15. "github.com/gravitl/netmaker/logic"
  16. "github.com/gravitl/netmaker/servercfg"
  17. "github.com/gravitl/netmaker/models"
  18. "github.com/gravitl/netmaker/mq"
  19. "github.com/skip2/go-qrcode"
  20. "golang.org/x/exp/slices"
  21. "golang.org/x/exp/slog"
  22. "golang.zx2c4.com/wireguard/wgctrl/wgtypes"
  23. )
  24. func extClientHandlers(r *mux.Router) {
  25. r.HandleFunc("/api/extclients", logic.SecurityCheck(true, http.HandlerFunc(getAllExtClients))).
  26. Methods(http.MethodGet)
  27. r.HandleFunc("/api/extclients/{network}", logic.SecurityCheck(true, http.HandlerFunc(getNetworkExtClients))).
  28. Methods(http.MethodGet)
  29. r.HandleFunc("/api/extclients/{network}/{clientid}", logic.SecurityCheck(false, http.HandlerFunc(getExtClient))).
  30. Methods(http.MethodGet)
  31. r.HandleFunc("/api/extclients/{network}/{clientid}/{type}", logic.SecurityCheck(false, http.HandlerFunc(getExtClientConf))).
  32. Methods(http.MethodGet)
  33. r.HandleFunc("/api/extclients/{network}/{clientid}", logic.SecurityCheck(false, http.HandlerFunc(updateExtClient))).
  34. Methods(http.MethodPut)
  35. r.HandleFunc("/api/extclients/{network}/{clientid}", logic.SecurityCheck(false, http.HandlerFunc(deleteExtClient))).
  36. Methods(http.MethodDelete)
  37. r.HandleFunc("/api/extclients/{network}/{nodeid}", logic.SecurityCheck(false, checkFreeTierLimits(limitChoiceMachines, http.HandlerFunc(createExtClient)))).
  38. Methods(http.MethodPost)
  39. }
  40. func checkIngressExists(nodeID string) bool {
  41. node, err := logic.GetNodeByID(nodeID)
  42. if err != nil {
  43. return false
  44. }
  45. return node.IsIngressGateway
  46. }
  47. // @Summary Get all remote access client associated with network
  48. // @Router /api/extclients/{network} [get]
  49. // @Tags Remote Access Client
  50. // @Security oauth2
  51. // @Success 200 {object} models.ExtClient
  52. // @Failure 500 {object} models.ErrorResponse
  53. func getNetworkExtClients(w http.ResponseWriter, r *http.Request) {
  54. w.Header().Set("Content-Type", "application/json")
  55. var extclients []models.ExtClient
  56. var params = mux.Vars(r)
  57. network := params["network"]
  58. extclients, err := logic.GetNetworkExtClients(network)
  59. if err != nil {
  60. logger.Log(0, r.Header.Get("user"),
  61. fmt.Sprintf("failed to get ext clients for network [%s]: %v", network, err))
  62. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  63. return
  64. }
  65. //Returns all the extclients in JSON format
  66. w.WriteHeader(http.StatusOK)
  67. json.NewEncoder(w).Encode(extclients)
  68. }
  69. // @Summary Fetches All Remote Access Clients across all networks
  70. // @Router /api/extclients [get]
  71. // @Tags Remote Access Client
  72. // @Security oauth2
  73. // @Success 200 {object} models.ExtClient
  74. // @Failure 500 {object} models.ErrorResponse
  75. // Not quite sure if this is necessary. Probably necessary based on front end but may
  76. // want to review after iteration 1 if it's being used or not
  77. func getAllExtClients(w http.ResponseWriter, r *http.Request) {
  78. w.Header().Set("Content-Type", "application/json")
  79. clients, err := logic.GetAllExtClients()
  80. if err != nil && !database.IsEmptyRecord(err) {
  81. logger.Log(0, "failed to get all extclients: ", err.Error())
  82. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  83. return
  84. }
  85. //Return all the extclients in JSON format
  86. logic.SortExtClient(clients[:])
  87. w.WriteHeader(http.StatusOK)
  88. json.NewEncoder(w).Encode(clients)
  89. }
  90. // @Summary Get an individual remote access client
  91. // @Router /api/extclients/{network}/{clientid} [get]
  92. // @Tags Remote Access Client
  93. // @Security oauth2
  94. // @Success 200 {object} models.ExtClient
  95. // @Failure 500 {object} models.ErrorResponse
  96. // @Failure 403 {object} models.ErrorResponse
  97. func getExtClient(w http.ResponseWriter, r *http.Request) {
  98. // set header.
  99. w.Header().Set("Content-Type", "application/json")
  100. var params = mux.Vars(r)
  101. clientid := params["clientid"]
  102. network := params["network"]
  103. client, err := logic.GetExtClient(clientid, network)
  104. if err != nil {
  105. logger.Log(
  106. 0,
  107. r.Header.Get("user"),
  108. fmt.Sprintf("failed to get extclient for [%s] on network [%s]: %v",
  109. clientid, network, err),
  110. )
  111. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  112. return
  113. }
  114. w.WriteHeader(http.StatusOK)
  115. json.NewEncoder(w).Encode(client)
  116. }
  117. // @Summary Get an individual remote access client
  118. // @Router /api/extclients/{network}/{clientid}/{type} [get]
  119. // @Tags Remote Access Client
  120. // @Security oauth2
  121. // @Success 200 {object} models.ExtClient
  122. // @Failure 500 {object} models.ErrorResponse
  123. // @Failure 403 {object} models.ErrorResponse
  124. func getExtClientConf(w http.ResponseWriter, r *http.Request) {
  125. // set header.
  126. w.Header().Set("Content-Type", "application/json")
  127. var params = mux.Vars(r)
  128. clientid := params["clientid"]
  129. networkid := params["network"]
  130. client, err := logic.GetExtClient(clientid, networkid)
  131. if err != nil {
  132. logger.Log(
  133. 0,
  134. r.Header.Get("user"),
  135. fmt.Sprintf("failed to get extclient for [%s] on network [%s]: %v",
  136. clientid, networkid, err),
  137. )
  138. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  139. return
  140. }
  141. gwnode, err := logic.GetNodeByID(client.IngressGatewayID)
  142. if err != nil {
  143. logger.Log(
  144. 0,
  145. r.Header.Get("user"),
  146. fmt.Sprintf(
  147. "failed to get ingress gateway node [%s] info: %v",
  148. client.IngressGatewayID,
  149. err,
  150. ),
  151. )
  152. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  153. return
  154. }
  155. host, err := logic.GetHost(gwnode.HostID.String())
  156. if err != nil {
  157. logger.Log(
  158. 0,
  159. r.Header.Get("user"),
  160. fmt.Sprintf(
  161. "failed to get host for ingress gateway node [%s] info: %v",
  162. client.IngressGatewayID,
  163. err,
  164. ),
  165. )
  166. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  167. return
  168. }
  169. network, err := logic.GetParentNetwork(client.Network)
  170. if err != nil {
  171. logger.Log(
  172. 1,
  173. r.Header.Get("user"),
  174. "Could not retrieve Ingress Gateway Network",
  175. client.Network,
  176. )
  177. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  178. return
  179. }
  180. preferredIp := strings.TrimSpace(r.URL.Query().Get("preferredip"))
  181. if preferredIp != "" {
  182. allowedPreferredIps := []string{}
  183. for i := range gwnode.AdditionalRagIps {
  184. allowedPreferredIps = append(allowedPreferredIps, gwnode.AdditionalRagIps[i].String())
  185. }
  186. allowedPreferredIps = append(allowedPreferredIps, host.EndpointIP.String())
  187. allowedPreferredIps = append(allowedPreferredIps, host.EndpointIPv6.String())
  188. if !slices.Contains(allowedPreferredIps, preferredIp) {
  189. slog.Warn(
  190. "preferred endpoint ip is not associated with the RAG. proceeding with preferred ip",
  191. "preferred ip",
  192. preferredIp,
  193. )
  194. logic.ReturnErrorResponse(
  195. w,
  196. r,
  197. logic.FormatError(
  198. errors.New("preferred endpoint ip is not associated with the RAG"),
  199. "badrequest",
  200. ),
  201. )
  202. return
  203. }
  204. if net.ParseIP(preferredIp).To4() == nil {
  205. preferredIp = fmt.Sprintf("[%s]", preferredIp)
  206. }
  207. }
  208. addrString := client.Address
  209. if addrString != "" {
  210. addrString += "/32"
  211. }
  212. if client.Address6 != "" {
  213. if addrString != "" {
  214. addrString += ","
  215. }
  216. addrString += client.Address6 + "/128"
  217. }
  218. keepalive := ""
  219. if network.DefaultKeepalive != 0 {
  220. keepalive = "PersistentKeepalive = " + strconv.Itoa(int(network.DefaultKeepalive))
  221. }
  222. if gwnode.IngressPersistentKeepalive != 0 {
  223. keepalive = "PersistentKeepalive = " + strconv.Itoa(int(gwnode.IngressPersistentKeepalive))
  224. }
  225. gwendpoint := ""
  226. if preferredIp == "" {
  227. if host.EndpointIP.To4() == nil {
  228. gwendpoint = fmt.Sprintf("[%s]:%d", host.EndpointIPv6.String(), host.ListenPort)
  229. } else {
  230. gwendpoint = fmt.Sprintf("%s:%d", host.EndpointIP.String(), host.ListenPort)
  231. }
  232. } else {
  233. gwendpoint = fmt.Sprintf("%s:%d", preferredIp, host.ListenPort)
  234. }
  235. var newAllowedIPs string
  236. if logic.IsInternetGw(gwnode) || gwnode.InternetGwID != "" {
  237. egressrange := "0.0.0.0/0"
  238. if gwnode.Address6.IP != nil && client.Address6 != "" {
  239. egressrange += "," + "::/0"
  240. }
  241. newAllowedIPs = egressrange
  242. } else {
  243. newAllowedIPs = network.AddressRange
  244. if newAllowedIPs != "" && network.AddressRange6 != "" {
  245. newAllowedIPs += ","
  246. }
  247. if network.AddressRange6 != "" {
  248. newAllowedIPs += network.AddressRange6
  249. }
  250. if egressGatewayRanges, err := logic.GetEgressRangesOnNetwork(&client); err == nil {
  251. for _, egressGatewayRange := range egressGatewayRanges {
  252. newAllowedIPs += "," + egressGatewayRange
  253. }
  254. }
  255. }
  256. defaultDNS := ""
  257. if client.DNS != "" {
  258. defaultDNS = "DNS = " + client.DNS
  259. } else if gwnode.IngressDNS != "" {
  260. defaultDNS = "DNS = " + gwnode.IngressDNS
  261. }
  262. // if servercfg.GetManageDNS() {
  263. // if gwnode.Address6.IP != nil {
  264. // if defaultDNS == "" {
  265. // defaultDNS = "DNS = " + gwnode.Address6.IP.String()
  266. // } else {
  267. // defaultDNS = defaultDNS + ", " + gwnode.Address6.IP.String()
  268. // }
  269. // }
  270. // if gwnode.Address.IP != nil {
  271. // if defaultDNS == "" {
  272. // defaultDNS = "DNS = " + gwnode.Address.IP.String()
  273. // } else {
  274. // defaultDNS = defaultDNS + ", " + gwnode.Address.IP.String()
  275. // }
  276. // }
  277. // }
  278. defaultMTU := 1420
  279. if host.MTU != 0 {
  280. defaultMTU = host.MTU
  281. }
  282. if gwnode.IngressMTU != 0 {
  283. defaultMTU = int(gwnode.IngressMTU)
  284. }
  285. postUp := strings.Builder{}
  286. if client.PostUp != "" && params["type"] != "qr" {
  287. for _, loc := range strings.Split(client.PostUp, "\n") {
  288. postUp.WriteString(fmt.Sprintf("PostUp = %s\n", loc))
  289. }
  290. }
  291. postDown := strings.Builder{}
  292. if client.PostDown != "" && params["type"] != "qr" {
  293. for _, loc := range strings.Split(client.PostDown, "\n") {
  294. postDown.WriteString(fmt.Sprintf("PostDown = %s\n", loc))
  295. }
  296. }
  297. config := fmt.Sprintf(`[Interface]
  298. Address = %s
  299. PrivateKey = %s
  300. MTU = %d
  301. %s
  302. %s
  303. %s
  304. [Peer]
  305. PublicKey = %s
  306. AllowedIPs = %s
  307. Endpoint = %s
  308. %s
  309. `, addrString,
  310. client.PrivateKey,
  311. defaultMTU,
  312. defaultDNS,
  313. postUp.String(),
  314. postDown.String(),
  315. host.PublicKey,
  316. newAllowedIPs,
  317. gwendpoint,
  318. keepalive,
  319. )
  320. if params["type"] == "qr" {
  321. bytes, err := qrcode.Encode(config, qrcode.Medium, 220)
  322. if err != nil {
  323. logger.Log(1, r.Header.Get("user"), "failed to encode qr code: ", err.Error())
  324. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  325. return
  326. }
  327. w.Header().Set("Content-Type", "image/png")
  328. w.WriteHeader(http.StatusOK)
  329. _, err = w.Write(bytes)
  330. if err != nil {
  331. logger.Log(1, r.Header.Get("user"), "response writer error (qr) ", err.Error())
  332. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  333. return
  334. }
  335. return
  336. }
  337. if params["type"] == "file" {
  338. name := client.ClientID + ".conf"
  339. w.Header().Set("Content-Type", "application/config")
  340. w.Header().Set("Content-Disposition", "attachment; filename=\""+name+"\"")
  341. w.WriteHeader(http.StatusOK)
  342. _, err := fmt.Fprint(w, config)
  343. if err != nil {
  344. logger.Log(1, r.Header.Get("user"), "response writer error (file) ", err.Error())
  345. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  346. }
  347. return
  348. }
  349. logger.Log(2, r.Header.Get("user"), "retrieved ext client config")
  350. w.WriteHeader(http.StatusOK)
  351. json.NewEncoder(w).Encode(client)
  352. }
  353. // @Summary Create an individual remote access client
  354. // @Router /api/extclients/{network}/{nodeid} [post]
  355. // @Tags Remote Access Client
  356. // @Security oauth2
  357. // @Success 200 {string} string "OK"
  358. // @Failure 500 {object} models.ErrorResponse
  359. // @Failure 400 {object} models.ErrorResponse
  360. // @Failure 403 {object} models.ErrorResponse
  361. func createExtClient(w http.ResponseWriter, r *http.Request) {
  362. w.Header().Set("Content-Type", "application/json")
  363. var params = mux.Vars(r)
  364. nodeid := params["nodeid"]
  365. ingressExists := checkIngressExists(nodeid)
  366. if !ingressExists {
  367. err := errors.New("ingress does not exist")
  368. slog.Error("failed to create extclient", "user", r.Header.Get("user"), "error", err)
  369. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  370. return
  371. }
  372. var customExtClient models.CustomExtClient
  373. if err := json.NewDecoder(r.Body).Decode(&customExtClient); err != nil {
  374. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  375. return
  376. }
  377. if err := validateCustomExtClient(&customExtClient, true); err != nil {
  378. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  379. return
  380. }
  381. var gateway models.EgressGatewayRequest
  382. gateway.NetID = params["network"]
  383. gateway.Ranges = customExtClient.ExtraAllowedIPs
  384. err := logic.ValidateEgressRange(gateway)
  385. if err != nil {
  386. logger.Log(0, r.Header.Get("user"), "error validating egress range: ", err.Error())
  387. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  388. return
  389. }
  390. node, err := logic.GetNodeByID(nodeid)
  391. if err != nil {
  392. logger.Log(0, r.Header.Get("user"),
  393. fmt.Sprintf("failed to get ingress gateway node [%s] info: %v", nodeid, err))
  394. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  395. return
  396. }
  397. var userName string
  398. if r.Header.Get("ismaster") == "yes" {
  399. userName = logic.MasterUser
  400. } else {
  401. caller, err := logic.GetUser(r.Header.Get("user"))
  402. if err != nil {
  403. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  404. return
  405. }
  406. userName = caller.UserName
  407. // check if user has a config already for remote access client
  408. extclients, err := logic.GetNetworkExtClients(node.Network)
  409. if err != nil {
  410. slog.Error("failed to get extclients", "error", err)
  411. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  412. return
  413. }
  414. for _, extclient := range extclients {
  415. if extclient.RemoteAccessClientID != "" &&
  416. extclient.RemoteAccessClientID == customExtClient.RemoteAccessClientID && extclient.OwnerID == caller.UserName && nodeid == extclient.IngressGatewayID {
  417. // extclient on the gw already exists for the remote access client
  418. err = errors.New("remote client config already exists on the gateway")
  419. slog.Error("failed to create extclient", "user", userName, "error", err)
  420. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  421. return
  422. }
  423. }
  424. }
  425. extclient := logic.UpdateExtClient(&models.ExtClient{}, &customExtClient)
  426. extclient.OwnerID = userName
  427. extclient.RemoteAccessClientID = customExtClient.RemoteAccessClientID
  428. extclient.IngressGatewayID = nodeid
  429. extclient.Network = node.Network
  430. extclient.Tags = make(map[models.TagID]struct{})
  431. // extclient.Tags[models.TagID(fmt.Sprintf("%s.%s", extclient.Network,
  432. // models.RemoteAccessTagName))] = struct{}{}
  433. // set extclient dns to ingressdns if extclient dns is not explicitly set
  434. if (extclient.DNS == "") && (node.IngressDNS != "") {
  435. extclient.DNS = node.IngressDNS
  436. }
  437. host, err := logic.GetHost(node.HostID.String())
  438. if err != nil {
  439. logger.Log(0, r.Header.Get("user"),
  440. fmt.Sprintf("failed to get ingress gateway host for node [%s] info: %v", nodeid, err))
  441. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  442. return
  443. }
  444. listenPort := logic.GetPeerListenPort(host)
  445. extclient.IngressGatewayEndpoint = fmt.Sprintf("%s:%d", host.EndpointIP.String(), listenPort)
  446. extclient.Enabled = true
  447. parentNetwork, err := logic.GetNetwork(node.Network)
  448. if err == nil { // check if parent network default ACL is enabled (yes) or not (no)
  449. extclient.Enabled = parentNetwork.DefaultACL == "yes"
  450. }
  451. extclient.Os = customExtClient.Os
  452. extclient.DeviceName = customExtClient.DeviceName
  453. if customExtClient.IsAlreadyConnectedToInetGw {
  454. slog.Warn("RAC/Client is already connected to internet gateway. this may mask their real IP address", "client IP", customExtClient.PublicEndpoint)
  455. }
  456. extclient.PublicEndpoint = customExtClient.PublicEndpoint
  457. extclient.Country = customExtClient.Country
  458. if err = logic.CreateExtClient(&extclient); err != nil {
  459. slog.Error(
  460. "failed to create extclient",
  461. "user",
  462. r.Header.Get("user"),
  463. "network",
  464. node.Network,
  465. "error",
  466. err,
  467. )
  468. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  469. return
  470. }
  471. slog.Info(
  472. "created extclient",
  473. "user",
  474. r.Header.Get("user"),
  475. "network",
  476. node.Network,
  477. "clientid",
  478. extclient.ClientID,
  479. )
  480. w.WriteHeader(http.StatusOK)
  481. go func() {
  482. if err := logic.SetClientDefaultACLs(&extclient); err != nil {
  483. slog.Error(
  484. "failed to set default acls for extclient",
  485. "user",
  486. r.Header.Get("user"),
  487. "network",
  488. node.Network,
  489. "error",
  490. err,
  491. )
  492. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  493. return
  494. }
  495. if err := mq.PublishPeerUpdate(false); err != nil {
  496. logger.Log(1, "error setting ext peers on "+nodeid+": "+err.Error())
  497. }
  498. if servercfg.IsDNSMode() {
  499. logic.SetDNS()
  500. }
  501. }()
  502. }
  503. // @Summary Update an individual remote access client
  504. // @Router /api/extclients/{network}/{clientid} [put]
  505. // @Tags Remote Access Client
  506. // @Security oauth2
  507. // @Success 200 {object} models.ExtClient
  508. // @Failure 500 {object} models.ErrorResponse
  509. // @Failure 400 {object} models.ErrorResponse
  510. // @Failure 403 {object} models.ErrorResponse
  511. func updateExtClient(w http.ResponseWriter, r *http.Request) {
  512. w.Header().Set("Content-Type", "application/json")
  513. var params = mux.Vars(r)
  514. var update models.CustomExtClient
  515. //var oldExtClient models.ExtClient
  516. var sendPeerUpdate bool
  517. var replacePeers bool
  518. err := json.NewDecoder(r.Body).Decode(&update)
  519. if err != nil {
  520. logger.Log(0, r.Header.Get("user"), "error decoding request body: ",
  521. err.Error())
  522. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  523. return
  524. }
  525. clientid := params["clientid"]
  526. oldExtClient, err := logic.GetExtClientByName(clientid)
  527. if err != nil {
  528. slog.Error(
  529. "failed to retrieve extclient",
  530. "user",
  531. r.Header.Get("user"),
  532. "id",
  533. clientid,
  534. "error",
  535. err,
  536. )
  537. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  538. return
  539. }
  540. if oldExtClient.ClientID == update.ClientID {
  541. if err := validateCustomExtClient(&update, false); err != nil {
  542. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  543. return
  544. }
  545. } else {
  546. if err := validateCustomExtClient(&update, true); err != nil {
  547. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  548. return
  549. }
  550. }
  551. var gateway models.EgressGatewayRequest
  552. gateway.NetID = params["network"]
  553. gateway.Ranges = update.ExtraAllowedIPs
  554. err = logic.ValidateEgressRange(gateway)
  555. if err != nil {
  556. logger.Log(0, r.Header.Get("user"), "error validating egress range: ", err.Error())
  557. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  558. return
  559. }
  560. var changedID = update.ClientID != oldExtClient.ClientID
  561. if !reflect.DeepEqual(update.DeniedACLs, oldExtClient.DeniedACLs) {
  562. sendPeerUpdate = true
  563. logic.SetClientACLs(&oldExtClient, update.DeniedACLs)
  564. }
  565. if !logic.IsSlicesEqual(update.ExtraAllowedIPs, oldExtClient.ExtraAllowedIPs) {
  566. sendPeerUpdate = true
  567. }
  568. if update.Enabled != oldExtClient.Enabled {
  569. sendPeerUpdate = true
  570. }
  571. if update.PublicKey != oldExtClient.PublicKey {
  572. //remove old peer entry
  573. sendPeerUpdate = true
  574. replacePeers = true
  575. }
  576. newclient := logic.UpdateExtClient(&oldExtClient, &update)
  577. if err := logic.DeleteExtClient(oldExtClient.Network, oldExtClient.ClientID); err != nil {
  578. slog.Error(
  579. "failed to delete ext client",
  580. "user",
  581. r.Header.Get("user"),
  582. "id",
  583. oldExtClient.ClientID,
  584. "network",
  585. oldExtClient.Network,
  586. "error",
  587. err,
  588. )
  589. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  590. return
  591. }
  592. if err := logic.SaveExtClient(&newclient); err != nil {
  593. slog.Error(
  594. "failed to save ext client",
  595. "user",
  596. r.Header.Get("user"),
  597. "id",
  598. newclient.ClientID,
  599. "network",
  600. newclient.Network,
  601. "error",
  602. err,
  603. )
  604. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  605. return
  606. }
  607. logger.Log(0, r.Header.Get("user"), "updated ext client", update.ClientID)
  608. w.WriteHeader(http.StatusOK)
  609. json.NewEncoder(w).Encode(newclient)
  610. go func() {
  611. if changedID && servercfg.IsDNSMode() {
  612. logic.SetDNS()
  613. }
  614. if replacePeers {
  615. if err := mq.PublishDeletedClientPeerUpdate(&oldExtClient); err != nil {
  616. slog.Error("error deleting old ext peers", "error", err.Error())
  617. }
  618. }
  619. if sendPeerUpdate { // need to send a peer update to the ingress node as enablement of one of it's clients has changed
  620. ingressNode, err := logic.GetNodeByID(newclient.IngressGatewayID)
  621. if err == nil {
  622. if err = mq.PublishPeerUpdate(false); err != nil {
  623. logger.Log(
  624. 1,
  625. "error setting ext peers on",
  626. ingressNode.ID.String(),
  627. ":",
  628. err.Error(),
  629. )
  630. }
  631. }
  632. if !update.Enabled {
  633. ingressHost, err := logic.GetHost(ingressNode.HostID.String())
  634. if err != nil {
  635. slog.Error(
  636. "Failed to get ingress host",
  637. "node",
  638. ingressNode.ID.String(),
  639. "error",
  640. err,
  641. )
  642. return
  643. }
  644. nodes, err := logic.GetAllNodes()
  645. if err != nil {
  646. slog.Error("Failed to get nodes", "error", err)
  647. return
  648. }
  649. go mq.PublishSingleHostPeerUpdate(ingressHost, nodes, nil, []models.ExtClient{oldExtClient}, false, nil)
  650. }
  651. }
  652. }()
  653. }
  654. // @Summary Delete an individual remote access client
  655. // @Router /api/extclients/{network}/{clientid} [delete]
  656. // @Tags Remote Access Client
  657. // @Security oauth2
  658. // @Success 200
  659. // @Failure 500 {object} models.ErrorResponse
  660. // @Failure 403 {object} models.ErrorResponse
  661. func deleteExtClient(w http.ResponseWriter, r *http.Request) {
  662. // Set header
  663. w.Header().Set("Content-Type", "application/json")
  664. // get params
  665. var params = mux.Vars(r)
  666. clientid := params["clientid"]
  667. network := params["network"]
  668. extclient, err := logic.GetExtClient(clientid, network)
  669. if err != nil {
  670. err = errors.New("Could not delete extclient " + params["clientid"])
  671. logger.Log(0, r.Header.Get("user"),
  672. fmt.Sprintf("failed to get extclient [%s],network [%s]: %v", clientid, network, err))
  673. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  674. return
  675. }
  676. ingressnode, err := logic.GetNodeByID(extclient.IngressGatewayID)
  677. if err != nil {
  678. logger.Log(
  679. 0,
  680. r.Header.Get("user"),
  681. fmt.Sprintf(
  682. "failed to get ingress gateway node [%s] info: %v",
  683. extclient.IngressGatewayID,
  684. err,
  685. ),
  686. )
  687. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  688. return
  689. }
  690. err = logic.DeleteExtClientAndCleanup(extclient)
  691. if err != nil {
  692. slog.Error("deleteExtClient: ", "Error", err.Error())
  693. err = errors.New("Could not delete extclient " + params["clientid"])
  694. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  695. return
  696. }
  697. go func() {
  698. if err := mq.PublishDeletedClientPeerUpdate(&extclient); err != nil {
  699. slog.Error("error setting ext peers on " + ingressnode.ID.String() + ": " + err.Error())
  700. }
  701. if servercfg.IsDNSMode() {
  702. logic.SetDNS()
  703. }
  704. }()
  705. logger.Log(0, r.Header.Get("user"),
  706. "Deleted extclient client", params["clientid"], "from network", params["network"])
  707. logic.ReturnSuccessResponse(w, r, params["clientid"]+" deleted.")
  708. }
  709. // validateCustomExtClient Validates the extclient object
  710. func validateCustomExtClient(customExtClient *models.CustomExtClient, checkID bool) error {
  711. v := validator.New()
  712. err := v.Struct(customExtClient)
  713. if err != nil {
  714. return err
  715. }
  716. //validate clientid
  717. if customExtClient.ClientID != "" {
  718. if err := isValid(customExtClient.ClientID, checkID); err != nil {
  719. return fmt.Errorf("client validation: %v", err)
  720. }
  721. }
  722. //extclient.ClientID = customExtClient.ClientID
  723. if len(customExtClient.PublicKey) > 0 {
  724. if _, err := wgtypes.ParseKey(customExtClient.PublicKey); err != nil {
  725. return errInvalidExtClientPubKey
  726. }
  727. //extclient.PublicKey = customExtClient.PublicKey
  728. }
  729. //validate extra ips
  730. if len(customExtClient.ExtraAllowedIPs) > 0 {
  731. for _, ip := range customExtClient.ExtraAllowedIPs {
  732. if _, _, err := net.ParseCIDR(ip); err != nil {
  733. return errInvalidExtClientExtraIP
  734. }
  735. }
  736. //extclient.ExtraAllowedIPs = customExtClient.ExtraAllowedIPs
  737. }
  738. //validate DNS
  739. if customExtClient.DNS != "" {
  740. if ip := net.ParseIP(customExtClient.DNS); ip == nil {
  741. return errInvalidExtClientDNS
  742. }
  743. //extclient.DNS = customExtClient.DNS
  744. }
  745. return nil
  746. }
  747. // isValid Checks if the clientid is valid
  748. func isValid(clientid string, checkID bool) error {
  749. if !validName(clientid) {
  750. return errInvalidExtClientID
  751. }
  752. if checkID {
  753. extclients, err := logic.GetAllExtClients()
  754. if err != nil {
  755. return fmt.Errorf("extclients isValid: %v", err)
  756. }
  757. for _, extclient := range extclients {
  758. if clientid == extclient.ClientID {
  759. return errDuplicateExtClientName
  760. }
  761. }
  762. }
  763. return nil
  764. }