ext_client.go 21 KB

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