ext_client.go 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720
  1. package controller
  2. import (
  3. "encoding/json"
  4. "errors"
  5. "fmt"
  6. "net"
  7. "net/http"
  8. "strconv"
  9. "github.com/gorilla/mux"
  10. "github.com/gravitl/netmaker/database"
  11. "github.com/gravitl/netmaker/functions"
  12. "github.com/gravitl/netmaker/logger"
  13. "github.com/gravitl/netmaker/logic"
  14. "github.com/gravitl/netmaker/logic/pro"
  15. "github.com/gravitl/netmaker/models"
  16. "github.com/gravitl/netmaker/models/promodels"
  17. "github.com/gravitl/netmaker/mq"
  18. "github.com/skip2/go-qrcode"
  19. "golang.org/x/exp/slog"
  20. "golang.zx2c4.com/wireguard/wgctrl/wgtypes"
  21. )
  22. func extClientHandlers(r *mux.Router) {
  23. r.HandleFunc("/api/extclients", logic.SecurityCheck(false, http.HandlerFunc(getAllExtClients))).Methods(http.MethodGet)
  24. r.HandleFunc("/api/extclients/{network}", logic.SecurityCheck(false, http.HandlerFunc(getNetworkExtClients))).Methods(http.MethodGet)
  25. r.HandleFunc("/api/extclients/{network}/{clientid}", logic.SecurityCheck(false, http.HandlerFunc(getExtClient))).Methods(http.MethodGet)
  26. r.HandleFunc("/api/extclients/{network}/{clientid}/{type}", logic.NetUserSecurityCheck(false, true, http.HandlerFunc(getExtClientConf))).Methods(http.MethodGet)
  27. r.HandleFunc("/api/extclients/{network}/{clientid}", logic.NetUserSecurityCheck(false, true, http.HandlerFunc(updateExtClient))).Methods(http.MethodPut)
  28. r.HandleFunc("/api/extclients/{network}/{clientid}", logic.NetUserSecurityCheck(false, true, http.HandlerFunc(deleteExtClient))).Methods(http.MethodDelete)
  29. r.HandleFunc("/api/extclients/{network}/{nodeid}", logic.NetUserSecurityCheck(false, true, checkFreeTierLimits(clients_l, http.HandlerFunc(createExtClient)))).Methods(http.MethodPost)
  30. }
  31. func checkIngressExists(nodeID string) bool {
  32. node, err := logic.GetNodeByID(nodeID)
  33. if err != nil {
  34. return false
  35. }
  36. return node.IsIngressGateway
  37. }
  38. // swagger:route GET /api/extclients/{network} ext_client getNetworkExtClients
  39. //
  40. // Get all extclients associated with network.
  41. // Gets all extclients associated with network, including pending extclients.
  42. //
  43. // Schemes: https
  44. //
  45. // Security:
  46. // oauth
  47. //
  48. // Responses:
  49. // 200: extClientSliceResponse
  50. func getNetworkExtClients(w http.ResponseWriter, r *http.Request) {
  51. w.Header().Set("Content-Type", "application/json")
  52. var extclients []models.ExtClient
  53. var params = mux.Vars(r)
  54. network := params["network"]
  55. extclients, err := logic.GetNetworkExtClients(network)
  56. if err != nil {
  57. logger.Log(0, r.Header.Get("user"),
  58. fmt.Sprintf("failed to get ext clients for network [%s]: %v", network, err))
  59. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  60. return
  61. }
  62. //Returns all the extclients in JSON format
  63. w.WriteHeader(http.StatusOK)
  64. json.NewEncoder(w).Encode(extclients)
  65. }
  66. // swagger:route GET /api/extclients ext_client getAllExtClients
  67. //
  68. // A separate function to get all extclients, not just extclients for a particular network.
  69. //
  70. // Schemes: https
  71. //
  72. // Security:
  73. // oauth
  74. //
  75. // Responses:
  76. // 200: extClientSliceResponse
  77. //
  78. // Not quite sure if this is necessary. Probably necessary based on front end but may
  79. // want to review after iteration 1 if it's being used or not
  80. func getAllExtClients(w http.ResponseWriter, r *http.Request) {
  81. w.Header().Set("Content-Type", "application/json")
  82. headerNetworks := r.Header.Get("networks")
  83. networksSlice := []string{}
  84. marshalErr := json.Unmarshal([]byte(headerNetworks), &networksSlice)
  85. if marshalErr != nil {
  86. logger.Log(0, "error unmarshalling networks: ",
  87. marshalErr.Error())
  88. logic.ReturnErrorResponse(w, r, logic.FormatError(marshalErr, "internal"))
  89. return
  90. }
  91. clients := []models.ExtClient{}
  92. var err error
  93. if len(networksSlice) > 0 && networksSlice[0] == logic.ALL_NETWORK_ACCESS {
  94. clients, err = functions.GetAllExtClients()
  95. if err != nil && !database.IsEmptyRecord(err) {
  96. logger.Log(0, "failed to get all extclients: ", err.Error())
  97. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  98. return
  99. }
  100. } else {
  101. for _, network := range networksSlice {
  102. extclients, err := logic.GetNetworkExtClients(network)
  103. if err == nil {
  104. clients = append(clients, extclients...)
  105. }
  106. }
  107. }
  108. //Return all the extclients in JSON format
  109. logic.SortExtClient(clients[:])
  110. w.WriteHeader(http.StatusOK)
  111. json.NewEncoder(w).Encode(clients)
  112. }
  113. // swagger:route GET /api/extclients/{network}/{clientid} ext_client getExtClient
  114. //
  115. // Get an individual extclient.
  116. //
  117. // Schemes: https
  118. //
  119. // Security:
  120. // oauth
  121. //
  122. // Responses:
  123. // 200: extClientResponse
  124. func getExtClient(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. network := params["network"]
  130. client, err := logic.GetExtClient(clientid, network)
  131. if err != nil {
  132. logger.Log(0, r.Header.Get("user"), fmt.Sprintf("failed to get extclient for [%s] on network [%s]: %v",
  133. clientid, network, err))
  134. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  135. return
  136. }
  137. w.WriteHeader(http.StatusOK)
  138. json.NewEncoder(w).Encode(client)
  139. }
  140. // swagger:route GET /api/extclients/{network}/{clientid}/{type} ext_client getExtClientConf
  141. //
  142. // Get an individual extclient.
  143. //
  144. // Schemes: https
  145. //
  146. // Security:
  147. // oauth
  148. //
  149. // Responses:
  150. // 200: extClientResponse
  151. func getExtClientConf(w http.ResponseWriter, r *http.Request) {
  152. // set header.
  153. w.Header().Set("Content-Type", "application/json")
  154. var params = mux.Vars(r)
  155. clientid := params["clientid"]
  156. networkid := params["network"]
  157. client, err := logic.GetExtClient(clientid, networkid)
  158. if err != nil {
  159. logger.Log(0, r.Header.Get("user"), fmt.Sprintf("failed to get extclient for [%s] on network [%s]: %v",
  160. clientid, networkid, err))
  161. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  162. return
  163. }
  164. gwnode, err := logic.GetNodeByID(client.IngressGatewayID)
  165. if err != nil {
  166. logger.Log(0, r.Header.Get("user"),
  167. fmt.Sprintf("failed to get ingress gateway node [%s] info: %v", client.IngressGatewayID, err))
  168. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  169. return
  170. }
  171. host, err := logic.GetHost(gwnode.HostID.String())
  172. if err != nil {
  173. logger.Log(0, r.Header.Get("user"),
  174. fmt.Sprintf("failed to get host for ingress gateway node [%s] info: %v", client.IngressGatewayID, err))
  175. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  176. return
  177. }
  178. network, err := logic.GetParentNetwork(client.Network)
  179. if err != nil {
  180. logger.Log(1, r.Header.Get("user"), "Could not retrieve Ingress Gateway Network", client.Network)
  181. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  182. return
  183. }
  184. addrString := client.Address
  185. if addrString != "" {
  186. addrString += "/32"
  187. }
  188. if client.Address6 != "" {
  189. if addrString != "" {
  190. addrString += ","
  191. }
  192. addrString += client.Address6 + "/128"
  193. }
  194. keepalive := ""
  195. if network.DefaultKeepalive != 0 {
  196. keepalive = "PersistentKeepalive = " + strconv.Itoa(int(network.DefaultKeepalive))
  197. }
  198. gwendpoint := host.EndpointIP.String() + ":" + strconv.Itoa(host.ListenPort)
  199. newAllowedIPs := network.AddressRange
  200. if newAllowedIPs != "" && network.AddressRange6 != "" {
  201. newAllowedIPs += ","
  202. }
  203. if network.AddressRange6 != "" {
  204. newAllowedIPs += network.AddressRange6
  205. }
  206. if egressGatewayRanges, err := logic.GetEgressRangesOnNetwork(&client); err == nil {
  207. for _, egressGatewayRange := range egressGatewayRanges {
  208. newAllowedIPs += "," + egressGatewayRange
  209. }
  210. }
  211. defaultDNS := ""
  212. if client.DNS != "" {
  213. defaultDNS = "DNS = " + client.DNS
  214. } else if gwnode.IngressDNS != "" {
  215. defaultDNS = "DNS = " + gwnode.IngressDNS
  216. }
  217. defaultMTU := 1420
  218. if host.MTU != 0 {
  219. defaultMTU = host.MTU
  220. }
  221. config := fmt.Sprintf(`[Interface]
  222. Address = %s
  223. PrivateKey = %s
  224. MTU = %d
  225. %s
  226. [Peer]
  227. PublicKey = %s
  228. AllowedIPs = %s
  229. Endpoint = %s
  230. %s
  231. `, addrString,
  232. client.PrivateKey,
  233. defaultMTU,
  234. defaultDNS,
  235. host.PublicKey,
  236. newAllowedIPs,
  237. gwendpoint,
  238. keepalive)
  239. if params["type"] == "qr" {
  240. bytes, err := qrcode.Encode(config, qrcode.Medium, 220)
  241. if err != nil {
  242. logger.Log(1, r.Header.Get("user"), "failed to encode qr code: ", err.Error())
  243. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  244. return
  245. }
  246. w.Header().Set("Content-Type", "image/png")
  247. w.WriteHeader(http.StatusOK)
  248. _, err = w.Write(bytes)
  249. if err != nil {
  250. logger.Log(1, r.Header.Get("user"), "response writer error (qr) ", err.Error())
  251. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  252. return
  253. }
  254. return
  255. }
  256. if params["type"] == "file" {
  257. name := client.ClientID + ".conf"
  258. w.Header().Set("Content-Type", "application/config")
  259. w.Header().Set("Content-Disposition", "attachment; filename=\""+name+"\"")
  260. w.WriteHeader(http.StatusOK)
  261. _, err := fmt.Fprint(w, config)
  262. if err != nil {
  263. logger.Log(1, r.Header.Get("user"), "response writer error (file) ", err.Error())
  264. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  265. }
  266. return
  267. }
  268. logger.Log(2, r.Header.Get("user"), "retrieved ext client config")
  269. w.WriteHeader(http.StatusOK)
  270. json.NewEncoder(w).Encode(client)
  271. }
  272. // swagger:route POST /api/extclients/{network}/{nodeid} ext_client createExtClient
  273. //
  274. // Create an individual extclient. Must have valid key and be unique.
  275. //
  276. // Schemes: https
  277. //
  278. // Security:
  279. // oauth
  280. func createExtClient(w http.ResponseWriter, r *http.Request) {
  281. w.Header().Set("Content-Type", "application/json")
  282. var params = mux.Vars(r)
  283. networkName := params["network"]
  284. nodeid := params["nodeid"]
  285. ingressExists := checkIngressExists(nodeid)
  286. if !ingressExists {
  287. err := errors.New("ingress does not exist")
  288. logger.Log(0, r.Header.Get("user"),
  289. fmt.Sprintf("failed to create extclient on network [%s]: %v", networkName, err))
  290. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  291. return
  292. }
  293. var extclient models.ExtClient
  294. var customExtClient models.CustomExtClient
  295. if err := json.NewDecoder(r.Body).Decode(&customExtClient); err != nil {
  296. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  297. return
  298. }
  299. if err := validateExtClient(&extclient, &customExtClient); err != nil {
  300. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  301. return
  302. }
  303. extclient.Network = networkName
  304. extclient.IngressGatewayID = nodeid
  305. node, err := logic.GetNodeByID(nodeid)
  306. if err != nil {
  307. logger.Log(0, r.Header.Get("user"),
  308. fmt.Sprintf("failed to get ingress gateway node [%s] info: %v", nodeid, err))
  309. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  310. return
  311. }
  312. host, err := logic.GetHost(node.HostID.String())
  313. if err != nil {
  314. logger.Log(0, r.Header.Get("user"),
  315. fmt.Sprintf("failed to get ingress gateway host for node [%s] info: %v", nodeid, err))
  316. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  317. return
  318. }
  319. listenPort := host.ListenPort
  320. if host.ProxyEnabled {
  321. listenPort = host.ProxyListenPort
  322. }
  323. extclient.IngressGatewayEndpoint = host.EndpointIP.String() + ":" + strconv.FormatInt(int64(listenPort), 10)
  324. extclient.Enabled = true
  325. parentNetwork, err := logic.GetNetwork(networkName)
  326. if err == nil { // check if parent network default ACL is enabled (yes) or not (no)
  327. extclient.Enabled = parentNetwork.DefaultACL == "yes"
  328. }
  329. if err := logic.SetClientDefaultACLs(&extclient); err != nil {
  330. logger.Log(0, r.Header.Get("user"),
  331. fmt.Sprintf("failed to assign ACLs to new ext client on network [%s]: %v", networkName, err))
  332. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  333. return
  334. }
  335. if err = logic.CreateExtClient(&extclient); err != nil {
  336. logger.Log(0, r.Header.Get("user"),
  337. fmt.Sprintf("failed to create new ext client on network [%s]: %v", networkName, err))
  338. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  339. return
  340. }
  341. var isAdmin bool
  342. if r.Header.Get("ismaster") != "yes" {
  343. userID := r.Header.Get("user")
  344. if isAdmin, err = checkProClientAccess(userID, extclient.ClientID, &parentNetwork); err != nil {
  345. logger.Log(0, userID, "attempted to create a client on network", networkName, "but they lack access")
  346. logic.DeleteExtClient(networkName, extclient.ClientID)
  347. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  348. return
  349. }
  350. if !isAdmin {
  351. if err = pro.AssociateNetworkUserClient(userID, networkName, extclient.ClientID); err != nil {
  352. logger.Log(0, "failed to associate client", extclient.ClientID, "to user", userID)
  353. }
  354. extclient.OwnerID = userID
  355. if err := logic.SaveExtClient(&extclient); err != nil {
  356. logger.Log(0, "failed to add owner id", userID, "to client", extclient.ClientID)
  357. }
  358. }
  359. }
  360. logger.Log(0, r.Header.Get("user"), "created new ext client on network", networkName)
  361. w.WriteHeader(http.StatusOK)
  362. go func() {
  363. peers, err := logic.GetNetworkClients(node.Network)
  364. if err != nil {
  365. slog.Warn("error getting network clients: ", "error", err)
  366. }
  367. for _, client := range peers {
  368. mq.PubPeerUpdateToHost(&client.Host)
  369. }
  370. f, err := logic.GetFwUpdate(host)
  371. if err == nil {
  372. mq.PublishFwUpdate(host, &f)
  373. }
  374. if err := mq.PublishExtCLientDNS(&extclient); err != nil {
  375. logger.Log(1, "error publishing extclient dns", err.Error())
  376. }
  377. }()
  378. }
  379. // swagger:route PUT /api/extclients/{network}/{clientid} ext_client updateExtClient
  380. //
  381. // Update an individual extclient.
  382. //
  383. // Schemes: https
  384. //
  385. // Security:
  386. // oauth
  387. //
  388. // Responses:
  389. // 200: extClientResponse
  390. func updateExtClient(w http.ResponseWriter, r *http.Request) {
  391. w.Header().Set("Content-Type", "application/json")
  392. var params = mux.Vars(r)
  393. var update models.CustomExtClient
  394. var oldExtClient models.ExtClient
  395. err := json.NewDecoder(r.Body).Decode(&update)
  396. if err != nil {
  397. logger.Log(0, r.Header.Get("user"), "error decoding request body: ",
  398. err.Error())
  399. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  400. return
  401. }
  402. clientid := params["clientid"]
  403. network := params["network"]
  404. key, err := logic.GetRecordKey(clientid, network)
  405. if err != nil {
  406. logger.Log(0, r.Header.Get("user"),
  407. fmt.Sprintf("failed to get record key for client [%s], network [%s]: %v",
  408. clientid, network, err))
  409. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  410. return
  411. }
  412. if err := validateExtClient(&oldExtClient, &update); err != nil {
  413. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  414. return
  415. }
  416. data, err := database.FetchRecord(database.EXT_CLIENT_TABLE_NAME, key)
  417. if err != nil {
  418. logger.Log(0, r.Header.Get("user"),
  419. fmt.Sprintf("failed to fetch ext client record key [%s] from db for client [%s], network [%s]: %v",
  420. key, clientid, network, err))
  421. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  422. return
  423. }
  424. if err = json.Unmarshal([]byte(data), &oldExtClient); err != nil {
  425. logger.Log(0, "error unmarshalling extclient: ",
  426. err.Error())
  427. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  428. return
  429. }
  430. // == PRO ==
  431. networkName := params["network"]
  432. var changedID = update.ClientID != oldExtClient.ClientID
  433. if r.Header.Get("ismaster") != "yes" {
  434. userID := r.Header.Get("user")
  435. _, doesOwn := doesUserOwnClient(userID, params["clientid"], networkName)
  436. if !doesOwn {
  437. logic.ReturnErrorResponse(w, r, logic.FormatError(fmt.Errorf("user not permitted"), "internal"))
  438. return
  439. }
  440. }
  441. if changedID && oldExtClient.OwnerID != "" {
  442. if err := pro.DissociateNetworkUserClient(oldExtClient.OwnerID, networkName, oldExtClient.ClientID); err != nil {
  443. logger.Log(0, "failed to dissociate client", oldExtClient.ClientID, "from user", oldExtClient.OwnerID)
  444. }
  445. if err := pro.AssociateNetworkUserClient(oldExtClient.OwnerID, networkName, update.ClientID); err != nil {
  446. logger.Log(0, "failed to associate client", update.ClientID, "to user", oldExtClient.OwnerID)
  447. }
  448. }
  449. // == END PRO ==
  450. var changedEnabled = (update.Enabled != oldExtClient.Enabled) // indicates there was a change in enablement
  451. // extra var need as logic.Update changes oldExtClient
  452. currentClient := oldExtClient
  453. newclient, replaceOldClient, err := logic.UpdateExtClient(&oldExtClient, &update)
  454. if err != nil {
  455. logger.Log(0, r.Header.Get("user"),
  456. fmt.Sprintf("failed to update ext client [%s], network [%s]: %v",
  457. clientid, network, err))
  458. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  459. return
  460. }
  461. logger.Log(0, r.Header.Get("user"), "updated ext client", update.ClientID)
  462. w.WriteHeader(http.StatusOK)
  463. json.NewEncoder(w).Encode(newclient)
  464. go func() {
  465. if ingressNode, err := logic.GetNodeByID(newclient.IngressGatewayID); err == nil {
  466. if ingressHost, err := logic.GetHost(ingressNode.HostID.String()); err == nil {
  467. if replaceOldClient || !update.Enabled {
  468. peers, err := logic.GetNetworkClients(ingressNode.Network)
  469. if err != nil {
  470. slog.Warn("error getting network clients: ", "error", err)
  471. }
  472. for _, client := range peers {
  473. mq.PubPeerUpdateToHost(&client.Host)
  474. }
  475. }
  476. if replaceOldClient || changedEnabled {
  477. // broadcast update
  478. peers, err := logic.GetNetworkClients(ingressNode.Network)
  479. if err != nil {
  480. slog.Warn("error getting network clients: ", "error", err)
  481. }
  482. for _, client := range peers {
  483. mq.PubPeerUpdateToHost(&client.Host)
  484. }
  485. }
  486. f, err := logic.GetFwUpdate(ingressHost)
  487. if err == nil {
  488. mq.PublishFwUpdate(ingressHost, &f)
  489. }
  490. }
  491. }
  492. if changedID {
  493. if err := mq.PublishExtClientDNSUpdate(currentClient, *newclient, networkName); err != nil {
  494. logger.Log(1, "error pubishing dns update for extcient update", err.Error())
  495. }
  496. }
  497. }()
  498. }
  499. // swagger:route DELETE /api/extclients/{network}/{clientid} ext_client deleteExtClient
  500. //
  501. // Delete an individual extclient.
  502. //
  503. // Schemes: https
  504. //
  505. // Security:
  506. // oauth
  507. //
  508. // Responses:
  509. // 200: successResponse
  510. func deleteExtClient(w http.ResponseWriter, r *http.Request) {
  511. // Set header
  512. w.Header().Set("Content-Type", "application/json")
  513. // get params
  514. var params = mux.Vars(r)
  515. clientid := params["clientid"]
  516. network := params["network"]
  517. extclient, err := logic.GetExtClient(clientid, network)
  518. if err != nil {
  519. err = errors.New("Could not delete extclient " + params["clientid"])
  520. logger.Log(0, r.Header.Get("user"),
  521. fmt.Sprintf("failed to delete extclient [%s],network [%s]: %v", clientid, network, err))
  522. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  523. return
  524. }
  525. ingressnode, err := logic.GetNodeByID(extclient.IngressGatewayID)
  526. if err != nil {
  527. logger.Log(0, r.Header.Get("user"),
  528. fmt.Sprintf("failed to get ingress gateway node [%s] info: %v", extclient.IngressGatewayID, err))
  529. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  530. return
  531. }
  532. // == PRO ==
  533. if r.Header.Get("ismaster") != "yes" {
  534. userID, clientID, networkName := r.Header.Get("user"), params["clientid"], params["network"]
  535. _, doesOwn := doesUserOwnClient(userID, clientID, networkName)
  536. if !doesOwn {
  537. logic.ReturnErrorResponse(w, r, logic.FormatError(fmt.Errorf("user not permitted"), "internal"))
  538. return
  539. }
  540. }
  541. if extclient.OwnerID != "" {
  542. if err = pro.DissociateNetworkUserClient(extclient.OwnerID, extclient.Network, extclient.ClientID); err != nil {
  543. logger.Log(0, "failed to dissociate client", extclient.ClientID, "from user", extclient.OwnerID)
  544. }
  545. }
  546. // == END PRO ==
  547. err = logic.DeleteExtClient(params["network"], params["clientid"])
  548. if err != nil {
  549. logger.Log(0, r.Header.Get("user"),
  550. fmt.Sprintf("failed to delete extclient [%s],network [%s]: %v", clientid, network, err))
  551. err = errors.New("Could not delete extclient " + params["clientid"])
  552. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  553. return
  554. }
  555. go func() {
  556. ingressHost, err := logic.GetHost(ingressnode.HostID.String())
  557. if err == nil {
  558. peers, err := logic.GetNetworkClients(ingressnode.Network)
  559. if err != nil {
  560. slog.Warn("error getting network clients: ", "error", err)
  561. }
  562. for _, client := range peers {
  563. mq.PubPeerUpdateToHost(&client.Host)
  564. }
  565. f, err := logic.GetFwUpdate(ingressHost)
  566. if err == nil {
  567. mq.PublishFwUpdate(ingressHost, &f)
  568. }
  569. }
  570. if err = mq.PublishDeleteExtClientDNS(&extclient); err != nil {
  571. logger.Log(1, "error publishing dns update for extclient deletion", err.Error())
  572. }
  573. }()
  574. logger.Log(0, r.Header.Get("user"),
  575. "Deleted extclient client", params["clientid"], "from network", params["network"])
  576. logic.ReturnSuccessResponse(w, r, params["clientid"]+" deleted.")
  577. }
  578. func checkProClientAccess(username, clientID string, network *models.Network) (bool, error) {
  579. u, err := logic.GetUser(username)
  580. if err != nil {
  581. return false, err
  582. }
  583. if u.IsAdmin {
  584. return true, nil
  585. }
  586. netUser, err := pro.GetNetworkUser(network.NetID, promodels.NetworkUserID(u.UserName))
  587. if err != nil {
  588. return false, err
  589. }
  590. if netUser.AccessLevel == pro.NET_ADMIN {
  591. return false, nil
  592. }
  593. if netUser.AccessLevel == pro.NO_ACCESS {
  594. return false, fmt.Errorf("user does not have access")
  595. }
  596. if !(len(netUser.Clients) < netUser.ClientLimit) {
  597. return false, fmt.Errorf("user can not create more clients")
  598. }
  599. if netUser.AccessLevel < pro.NO_ACCESS {
  600. netUser.Clients = append(netUser.Clients, clientID)
  601. if err = pro.UpdateNetworkUser(network.NetID, netUser); err != nil {
  602. return false, err
  603. }
  604. }
  605. return false, nil
  606. }
  607. // checks if net user owns an ext client or is an admin
  608. func doesUserOwnClient(username, clientID, network string) (bool, bool) {
  609. u, err := logic.GetUser(username)
  610. if err != nil {
  611. return false, false
  612. }
  613. if u.IsAdmin {
  614. return true, true
  615. }
  616. netUser, err := pro.GetNetworkUser(network, promodels.NetworkUserID(u.UserName))
  617. if err != nil {
  618. return false, false
  619. }
  620. if netUser.AccessLevel == pro.NET_ADMIN {
  621. return false, true
  622. }
  623. return false, logic.StringSliceContains(netUser.Clients, clientID)
  624. }
  625. // validateExtClient Validates the extclient object
  626. func validateExtClient(extclient *models.ExtClient, customExtClient *models.CustomExtClient) error {
  627. //validate clientid
  628. if customExtClient.ClientID != "" && !validName(customExtClient.ClientID) {
  629. return errInvalidExtClientID
  630. }
  631. extclient.ClientID = customExtClient.ClientID
  632. if len(customExtClient.PublicKey) > 0 {
  633. if _, err := wgtypes.ParseKey(customExtClient.PublicKey); err != nil {
  634. return errInvalidExtClientPubKey
  635. }
  636. extclient.PublicKey = customExtClient.PublicKey
  637. }
  638. //validate extra ips
  639. if len(customExtClient.ExtraAllowedIPs) > 0 {
  640. for _, ip := range customExtClient.ExtraAllowedIPs {
  641. if _, _, err := net.ParseCIDR(ip); err != nil {
  642. return errInvalidExtClientExtraIP
  643. }
  644. }
  645. extclient.ExtraAllowedIPs = customExtClient.ExtraAllowedIPs
  646. }
  647. //validate DNS
  648. if customExtClient.DNS != "" {
  649. if ip := net.ParseIP(customExtClient.DNS); ip == nil {
  650. return errInvalidExtClientDNS
  651. }
  652. extclient.DNS = customExtClient.DNS
  653. }
  654. return nil
  655. }