ext_client.go 22 KB

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