networkHttpController.go 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827
  1. package controller
  2. import (
  3. "encoding/base64"
  4. "encoding/json"
  5. "errors"
  6. "fmt"
  7. "net/http"
  8. "strings"
  9. "time"
  10. "github.com/go-playground/validator/v10"
  11. "github.com/gorilla/mux"
  12. "github.com/gravitl/netmaker/database"
  13. "github.com/gravitl/netmaker/functions"
  14. "github.com/gravitl/netmaker/models"
  15. "github.com/gravitl/netmaker/servercfg"
  16. )
  17. const ALL_NETWORK_ACCESS = "THIS_USER_HAS_ALL"
  18. const NO_NETWORKS_PRESENT = "THIS_USER_HAS_NONE"
  19. func networkHandlers(r *mux.Router) {
  20. r.HandleFunc("/api/networks", securityCheck(false, http.HandlerFunc(getNetworks))).Methods("GET")
  21. r.HandleFunc("/api/networks", securityCheck(true, http.HandlerFunc(createNetwork))).Methods("POST")
  22. r.HandleFunc("/api/networks/{networkname}", securityCheck(false, http.HandlerFunc(getNetwork))).Methods("GET")
  23. r.HandleFunc("/api/networks/{networkname}", securityCheck(false, http.HandlerFunc(updateNetwork))).Methods("PUT")
  24. r.HandleFunc("/api/networks/{networkname}/nodelimit", securityCheck(true, http.HandlerFunc(updateNetworkNodeLimit))).Methods("PUT")
  25. r.HandleFunc("/api/networks/{networkname}", securityCheck(true, http.HandlerFunc(deleteNetwork))).Methods("DELETE")
  26. r.HandleFunc("/api/networks/{networkname}/keyupdate", securityCheck(false, http.HandlerFunc(keyUpdate))).Methods("POST")
  27. r.HandleFunc("/api/networks/{networkname}/keys", securityCheck(false, http.HandlerFunc(createAccessKey))).Methods("POST")
  28. r.HandleFunc("/api/networks/{networkname}/keys", securityCheck(false, http.HandlerFunc(getAccessKeys))).Methods("GET")
  29. r.HandleFunc("/api/networks/{networkname}/signuptoken", securityCheck(false, http.HandlerFunc(getSignupToken))).Methods("GET")
  30. r.HandleFunc("/api/networks/{networkname}/keys/{name}", securityCheck(false, http.HandlerFunc(deleteAccessKey))).Methods("DELETE")
  31. }
  32. //Security check is middleware for every function and just checks to make sure that its the master calling
  33. //Only admin should have access to all these network-level actions
  34. //or maybe some Users once implemented
  35. func securityCheck(reqAdmin bool, next http.Handler) http.HandlerFunc {
  36. return func(w http.ResponseWriter, r *http.Request) {
  37. var errorResponse = models.ErrorResponse{
  38. Code: http.StatusUnauthorized, Message: "W1R3: It's not you it's me.",
  39. }
  40. var params = mux.Vars(r)
  41. bearerToken := r.Header.Get("Authorization")
  42. err, networks, username := SecurityCheck(reqAdmin, params["networkname"], bearerToken)
  43. if err != nil {
  44. if strings.Contains(err.Error(), "does not exist") {
  45. errorResponse.Code = http.StatusNotFound
  46. }
  47. errorResponse.Message = err.Error()
  48. returnErrorResponse(w, r, errorResponse)
  49. return
  50. }
  51. networksJson, err := json.Marshal(&networks)
  52. if err != nil {
  53. errorResponse.Message = err.Error()
  54. returnErrorResponse(w, r, errorResponse)
  55. return
  56. }
  57. r.Header.Set("user", username)
  58. r.Header.Set("networks", string(networksJson))
  59. next.ServeHTTP(w, r)
  60. }
  61. }
  62. func SecurityCheck(reqAdmin bool, netname, token string) (error, []string, string) {
  63. networkexists, err := functions.NetworkExists(netname)
  64. if err != nil {
  65. return err, nil, ""
  66. }
  67. if netname != "" && !networkexists {
  68. return errors.New("This network does not exist"), nil, ""
  69. }
  70. var hasBearer = true
  71. var tokenSplit = strings.Split(token, " ")
  72. var authToken = ""
  73. if len(tokenSplit) < 2 {
  74. hasBearer = false
  75. } else {
  76. authToken = tokenSplit[1]
  77. }
  78. userNetworks := []string{}
  79. //all endpoints here require master so not as complicated
  80. isMasterAuthenticated := authenticateMaster(authToken)
  81. username := ""
  82. if !hasBearer || !isMasterAuthenticated {
  83. userName, networks, isadmin, err := functions.VerifyUserToken(authToken)
  84. username = userName
  85. if err != nil {
  86. return errors.New("Error verifying user token"), nil, username
  87. }
  88. if !isadmin && reqAdmin {
  89. return errors.New("You are unauthorized to access this endpoint"), nil, username
  90. }
  91. userNetworks = networks
  92. if isadmin {
  93. userNetworks = []string{ALL_NETWORK_ACCESS}
  94. }
  95. } else if isMasterAuthenticated {
  96. userNetworks = []string{ALL_NETWORK_ACCESS}
  97. }
  98. if len(userNetworks) == 0 {
  99. userNetworks = append(userNetworks, NO_NETWORKS_PRESENT)
  100. }
  101. return nil, userNetworks, username
  102. }
  103. //Consider a more secure way of setting master key
  104. func authenticateMaster(tokenString string) bool {
  105. if tokenString == servercfg.GetMasterKey() {
  106. return true
  107. }
  108. return false
  109. }
  110. //simple get all networks function
  111. func getNetworks(w http.ResponseWriter, r *http.Request) {
  112. headerNetworks := r.Header.Get("networks")
  113. networksSlice := []string{}
  114. marshalErr := json.Unmarshal([]byte(headerNetworks), &networksSlice)
  115. if marshalErr != nil {
  116. returnErrorResponse(w, r, formatError(marshalErr, "internal"))
  117. return
  118. }
  119. allnetworks := []models.Network{}
  120. err := errors.New("Networks Error")
  121. if networksSlice[0] == ALL_NETWORK_ACCESS {
  122. allnetworks, err = functions.ListNetworks()
  123. if err != nil {
  124. returnErrorResponse(w, r, formatError(err, "internal"))
  125. return
  126. }
  127. } else {
  128. for _, network := range networksSlice {
  129. netObject, parentErr := functions.GetParentNetwork(network)
  130. if parentErr == nil {
  131. allnetworks = append(allnetworks, netObject)
  132. }
  133. }
  134. }
  135. networks := RemoveComms(allnetworks)
  136. functions.PrintUserLog(r.Header.Get("user"), "fetched networks.", 2)
  137. w.WriteHeader(http.StatusOK)
  138. json.NewEncoder(w).Encode(networks)
  139. }
  140. func RemoveComms(networks []models.Network) []models.Network {
  141. var index int = 100000001
  142. for ind, net := range networks {
  143. if net.NetID == "comms" {
  144. index = ind
  145. }
  146. }
  147. if index == 100000001 {
  148. return networks
  149. }
  150. returnable := make([]models.Network, 0)
  151. returnable = append(returnable, networks[:index]...)
  152. return append(returnable, networks[index+1:]...)
  153. }
  154. func ValidateNetworkUpdate(network models.NetworkUpdate) error {
  155. v := validator.New()
  156. _ = v.RegisterValidation("netid_valid", func(fl validator.FieldLevel) bool {
  157. if fl.Field().String() == "" {
  158. return true
  159. }
  160. inCharSet := functions.NameInNetworkCharSet(fl.Field().String())
  161. return inCharSet
  162. })
  163. // _ = v.RegisterValidation("addressrange_valid", func(fl validator.FieldLevel) bool {
  164. // isvalid := fl.Field().String() == "" || functions.IsIpCIDR(fl.Field().String())
  165. // return isvalid
  166. // })
  167. //_ = v.RegisterValidation("addressrange6_valid", func(fl validator.FieldLevel) bool {
  168. // isvalid := fl.Field().String() == "" || functions.IsIpCIDR(fl.Field().String())
  169. // return isvalid
  170. // })
  171. // _ = v.RegisterValidation("localrange_valid", func(fl validator.FieldLevel) bool {
  172. // isvalid := fl.Field().String() == "" || functions.IsIpCIDR(fl.Field().String())
  173. // return isvalid
  174. // })
  175. // _ = v.RegisterValidation("netid_valid", func(fl validator.FieldLevel) bool {
  176. // return true
  177. // })
  178. // _ = v.RegisterValidation("displayname_unique", func(fl validator.FieldLevel) bool {
  179. // return true
  180. // })
  181. err := v.Struct(network)
  182. if err != nil {
  183. for _, e := range err.(validator.ValidationErrors) {
  184. fmt.Println(e)
  185. }
  186. }
  187. return err
  188. }
  189. func ValidateNetworkCreate(network models.Network) error {
  190. v := validator.New()
  191. // _ = v.RegisterValidation("addressrange_valid", func(fl validator.FieldLevel) bool {
  192. // isvalid := functions.IsIpCIDR(fl.Field().String())
  193. // return isvalid
  194. // })
  195. _ = v.RegisterValidation("addressrange6_valid", func(fl validator.FieldLevel) bool {
  196. isvalid := true
  197. if *network.IsDualStack {
  198. isvalid = functions.IsIpCIDR(fl.Field().String())
  199. }
  200. return isvalid
  201. })
  202. //
  203. // _ = v.RegisterValidation("localrange_valid", func(fl validator.FieldLevel) bool {
  204. // isvalid := fl.Field().String() == "" || functions.IsIpCIDR(fl.Field().String())
  205. // return isvalid
  206. // })
  207. //
  208. _ = v.RegisterValidation("netid_valid", func(fl validator.FieldLevel) bool {
  209. isFieldUnique, _ := functions.IsNetworkNameUnique(fl.Field().String())
  210. inCharSet := functions.NameInNetworkCharSet(fl.Field().String())
  211. return isFieldUnique && inCharSet
  212. })
  213. //
  214. _ = v.RegisterValidation("displayname_valid", func(fl validator.FieldLevel) bool {
  215. isFieldUnique, _ := functions.IsNetworkDisplayNameUnique(fl.Field().String())
  216. inCharSet := functions.NameInNetworkCharSet(fl.Field().String())
  217. return isFieldUnique && inCharSet
  218. })
  219. err := v.Struct(network)
  220. if err != nil {
  221. for _, e := range err.(validator.ValidationErrors) {
  222. fmt.Println(e)
  223. }
  224. }
  225. return err
  226. }
  227. //Simple get network function
  228. func getNetwork(w http.ResponseWriter, r *http.Request) {
  229. // set header.
  230. w.Header().Set("Content-Type", "application/json")
  231. var params = mux.Vars(r)
  232. netname := params["networkname"]
  233. network, err := GetNetwork(netname)
  234. if err != nil {
  235. returnErrorResponse(w, r, formatError(err, "internal"))
  236. return
  237. }
  238. functions.PrintUserLog(r.Header.Get("user"), "fetched network "+netname, 2)
  239. w.WriteHeader(http.StatusOK)
  240. json.NewEncoder(w).Encode(network)
  241. }
  242. func GetNetwork(name string) (models.Network, error) {
  243. var network models.Network
  244. record, err := database.FetchRecord(database.NETWORKS_TABLE_NAME, name)
  245. if err != nil {
  246. return network, err
  247. }
  248. if err = json.Unmarshal([]byte(record), &network); err != nil {
  249. return models.Network{}, err
  250. }
  251. return network, nil
  252. }
  253. func keyUpdate(w http.ResponseWriter, r *http.Request) {
  254. w.Header().Set("Content-Type", "application/json")
  255. var params = mux.Vars(r)
  256. netname := params["networkname"]
  257. network, err := KeyUpdate(netname)
  258. if err != nil {
  259. returnErrorResponse(w, r, formatError(err, "internal"))
  260. return
  261. }
  262. functions.PrintUserLog(r.Header.Get("user"), "updated key on network "+netname, 2)
  263. w.WriteHeader(http.StatusOK)
  264. json.NewEncoder(w).Encode(network)
  265. }
  266. func KeyUpdate(netname string) (models.Network, error) {
  267. network, err := functions.GetParentNetwork(netname)
  268. if err != nil {
  269. return models.Network{}, err
  270. }
  271. network.KeyUpdateTimeStamp = time.Now().Unix()
  272. data, err := json.Marshal(&network)
  273. if err != nil {
  274. return models.Network{}, err
  275. }
  276. database.Insert(netname, string(data), database.NETWORKS_TABLE_NAME)
  277. return network, nil
  278. }
  279. //Update a network
  280. func AlertNetwork(netid string) error {
  281. var network models.Network
  282. network, err := functions.GetParentNetwork(netid)
  283. if err != nil {
  284. return err
  285. }
  286. updatetime := time.Now().Unix()
  287. network.NodesLastModified = updatetime
  288. network.NetworkLastModified = updatetime
  289. data, err := json.Marshal(&network)
  290. if err != nil {
  291. return err
  292. }
  293. database.Insert(netid, string(data), database.NETWORKS_TABLE_NAME)
  294. return nil
  295. }
  296. //Update a network
  297. func updateNetwork(w http.ResponseWriter, r *http.Request) {
  298. w.Header().Set("Content-Type", "application/json")
  299. var params = mux.Vars(r)
  300. var network models.Network
  301. netname := params["networkname"]
  302. network, err := functions.GetParentNetwork(netname)
  303. if err != nil {
  304. returnErrorResponse(w, r, formatError(err, "internal"))
  305. return
  306. }
  307. var networkChange models.NetworkUpdate
  308. _ = json.NewDecoder(r.Body).Decode(&networkChange)
  309. if networkChange.AddressRange == "" {
  310. networkChange.AddressRange = network.AddressRange
  311. }
  312. if networkChange.AddressRange6 == "" {
  313. networkChange.AddressRange6 = network.AddressRange6
  314. }
  315. if networkChange.NetID == "" {
  316. networkChange.NetID = network.NetID
  317. }
  318. err = ValidateNetworkUpdate(networkChange)
  319. if err != nil {
  320. returnErrorResponse(w, r, formatError(err, "badrequest"))
  321. return
  322. }
  323. returnednetwork, err := UpdateNetwork(networkChange, network)
  324. if err != nil {
  325. returnErrorResponse(w, r, formatError(err, "badrequest"))
  326. return
  327. }
  328. functions.PrintUserLog(r.Header.Get("user"), "updated network "+netname, 1)
  329. w.WriteHeader(http.StatusOK)
  330. json.NewEncoder(w).Encode(returnednetwork)
  331. }
  332. func updateNetworkNodeLimit(w http.ResponseWriter, r *http.Request) {
  333. w.Header().Set("Content-Type", "application/json")
  334. var params = mux.Vars(r)
  335. var network models.Network
  336. netname := params["networkname"]
  337. network, err := functions.GetParentNetwork(netname)
  338. if err != nil {
  339. returnErrorResponse(w, r, formatError(err, "internal"))
  340. return
  341. }
  342. var networkChange models.NetworkUpdate
  343. _ = json.NewDecoder(r.Body).Decode(&networkChange)
  344. if networkChange.NodeLimit != 0 {
  345. network.NodeLimit = networkChange.NodeLimit
  346. data, err := json.Marshal(&network)
  347. if err != nil {
  348. returnErrorResponse(w, r, formatError(err, "badrequest"))
  349. return
  350. }
  351. database.Insert(network.NetID, string(data), database.NETWORKS_TABLE_NAME)
  352. functions.PrintUserLog(r.Header.Get("user"), "updated network node limit on, "+netname, 1)
  353. }
  354. w.WriteHeader(http.StatusOK)
  355. json.NewEncoder(w).Encode(network)
  356. }
  357. func UpdateNetwork(networkChange models.NetworkUpdate, network models.Network) (models.Network, error) {
  358. //NOTE: Network.NetID is intentionally NOT editable. It acts as a static ID for the network.
  359. //DisplayName can be changed instead, which is what shows on the front end
  360. if networkChange.NetID != network.NetID {
  361. return models.Network{}, errors.New("NetID is not editable")
  362. }
  363. haschange := false
  364. hasrangeupdate := false
  365. haslocalrangeupdate := false
  366. if networkChange.AddressRange != "" {
  367. haschange = true
  368. hasrangeupdate = true
  369. network.AddressRange = networkChange.AddressRange
  370. }
  371. if networkChange.LocalRange != "" {
  372. haschange = true
  373. haslocalrangeupdate = true
  374. network.LocalRange = networkChange.LocalRange
  375. }
  376. if networkChange.IsLocal != nil {
  377. network.IsLocal = networkChange.IsLocal
  378. }
  379. if networkChange.IsDualStack != nil {
  380. network.IsDualStack = networkChange.IsDualStack
  381. }
  382. if networkChange.DefaultListenPort != 0 {
  383. network.DefaultListenPort = networkChange.DefaultListenPort
  384. haschange = true
  385. }
  386. if networkChange.DefaultPostDown != "" {
  387. network.DefaultPostDown = networkChange.DefaultPostDown
  388. haschange = true
  389. }
  390. if networkChange.DefaultInterface != "" {
  391. network.DefaultInterface = networkChange.DefaultInterface
  392. haschange = true
  393. }
  394. if networkChange.DefaultPostUp != "" {
  395. network.DefaultPostUp = networkChange.DefaultPostUp
  396. haschange = true
  397. }
  398. if networkChange.DefaultKeepalive != 0 {
  399. network.DefaultKeepalive = networkChange.DefaultKeepalive
  400. haschange = true
  401. }
  402. if networkChange.DisplayName != "" {
  403. network.DisplayName = networkChange.DisplayName
  404. haschange = true
  405. }
  406. if networkChange.DefaultCheckInInterval != 0 {
  407. network.DefaultCheckInInterval = networkChange.DefaultCheckInInterval
  408. haschange = true
  409. }
  410. if networkChange.AllowManualSignUp != nil {
  411. network.AllowManualSignUp = networkChange.AllowManualSignUp
  412. haschange = true
  413. }
  414. if haschange {
  415. network.SetNetworkLastModified()
  416. }
  417. data, err := json.Marshal(&network)
  418. if err != nil {
  419. return models.Network{}, err
  420. }
  421. database.Insert(network.NetID, string(data), database.NETWORKS_TABLE_NAME)
  422. //Cycles through nodes and gives them new IP's based on the new range
  423. //Pretty cool, but also pretty inefficient currently
  424. if hasrangeupdate {
  425. err = functions.UpdateNetworkNodeAddresses(network.NetID)
  426. if err != nil {
  427. return models.Network{}, err
  428. }
  429. }
  430. if haslocalrangeupdate {
  431. err = functions.UpdateNetworkLocalAddresses(network.NetID)
  432. if err != nil {
  433. return models.Network{}, err
  434. }
  435. }
  436. returnnetwork, err := functions.GetParentNetwork(network.NetID)
  437. if err != nil {
  438. return models.Network{}, err
  439. }
  440. return returnnetwork, nil
  441. }
  442. //Delete a network
  443. //Will stop you if there's any nodes associated
  444. func deleteNetwork(w http.ResponseWriter, r *http.Request) {
  445. // Set header
  446. w.Header().Set("Content-Type", "application/json")
  447. var params = mux.Vars(r)
  448. network := params["networkname"]
  449. err := DeleteNetwork(network)
  450. if err != nil {
  451. errtype := "badrequest"
  452. if strings.Contains(err.Error(), "Node check failed") {
  453. errtype = "forbidden"
  454. }
  455. returnErrorResponse(w, r, formatError(err, errtype))
  456. return
  457. }
  458. functions.PrintUserLog(r.Header.Get("user"), "deleted network "+network, 1)
  459. w.WriteHeader(http.StatusOK)
  460. json.NewEncoder(w).Encode("success")
  461. }
  462. func DeleteNetwork(network string) error {
  463. nodecount, err := functions.GetNetworkNodeNumber(network)
  464. if err != nil {
  465. return err
  466. } else if nodecount > 0 {
  467. return errors.New("node check failed. All nodes must be deleted before deleting network")
  468. }
  469. database.DeleteRecord(database.NETWORKS_TABLE_NAME, network)
  470. return err
  471. }
  472. //Create a network
  473. //Pretty simple
  474. func createNetwork(w http.ResponseWriter, r *http.Request) {
  475. w.Header().Set("Content-Type", "application/json")
  476. var network models.Network
  477. // we decode our body request params
  478. err := json.NewDecoder(r.Body).Decode(&network)
  479. if err != nil {
  480. returnErrorResponse(w, r, formatError(err, "internal"))
  481. return
  482. }
  483. err = CreateNetwork(network)
  484. if err != nil {
  485. returnErrorResponse(w, r, formatError(err, "badrequest"))
  486. return
  487. }
  488. functions.PrintUserLog(r.Header.Get("user"), "created network "+network.NetID, 1)
  489. w.WriteHeader(http.StatusOK)
  490. //json.NewEncoder(w).Encode(result)
  491. }
  492. func CreateNetwork(network models.Network) error {
  493. //TODO: Not really doing good validation here. Same as createNode, updateNode, and updateNetwork
  494. //Need to implement some better validation across the board
  495. if network.IsLocal == nil {
  496. falsevar := false
  497. network.IsLocal = &falsevar
  498. }
  499. if network.IsDualStack == nil {
  500. falsevar := false
  501. network.IsDualStack = &falsevar
  502. }
  503. err := ValidateNetworkCreate(network)
  504. if err != nil {
  505. //returnErrorResponse(w, r, formatError(err, "badrequest"))
  506. return err
  507. }
  508. network.SetDefaults()
  509. network.SetNodesLastModified()
  510. network.SetNetworkLastModified()
  511. network.KeyUpdateTimeStamp = time.Now().Unix()
  512. data, err := json.Marshal(&network)
  513. if err != nil {
  514. return err
  515. }
  516. if err = database.Insert(network.NetID, string(data), database.NETWORKS_TABLE_NAME); err != nil {
  517. return err
  518. }
  519. return nil
  520. }
  521. // BEGIN KEY MANAGEMENT SECTION
  522. //TODO: Very little error handling
  523. //accesskey is created as a json string inside the Network collection item in mongo
  524. func createAccessKey(w http.ResponseWriter, r *http.Request) {
  525. w.Header().Set("Content-Type", "application/json")
  526. var params = mux.Vars(r)
  527. var accesskey models.AccessKey
  528. //start here
  529. netname := params["networkname"]
  530. network, err := functions.GetParentNetwork(netname)
  531. if err != nil {
  532. returnErrorResponse(w, r, formatError(err, "internal"))
  533. return
  534. }
  535. err = json.NewDecoder(r.Body).Decode(&accesskey)
  536. if err != nil {
  537. returnErrorResponse(w, r, formatError(err, "internal"))
  538. return
  539. }
  540. key, err := CreateAccessKey(accesskey, network)
  541. if err != nil {
  542. returnErrorResponse(w, r, formatError(err, "badrequest"))
  543. return
  544. }
  545. functions.PrintUserLog(r.Header.Get("user"), "created access key "+netname, 1)
  546. w.WriteHeader(http.StatusOK)
  547. json.NewEncoder(w).Encode(key)
  548. //w.Write([]byte(accesskey.AccessString))
  549. }
  550. func CreateAccessKey(accesskey models.AccessKey, network models.Network) (models.AccessKey, error) {
  551. if accesskey.Name == "" {
  552. accesskey.Name = functions.GenKeyName()
  553. }
  554. if accesskey.Value == "" {
  555. accesskey.Value = functions.GenKey()
  556. }
  557. if accesskey.Uses == 0 {
  558. accesskey.Uses = 1
  559. }
  560. checkkeys, err := GetKeys(network.NetID)
  561. if err != nil {
  562. return models.AccessKey{}, errors.New("could not retrieve network keys")
  563. }
  564. for _, key := range checkkeys {
  565. if key.Name == accesskey.Name {
  566. return models.AccessKey{}, errors.New("duplicate AccessKey Name")
  567. }
  568. }
  569. privAddr := ""
  570. if network.IsLocal != nil {
  571. if *network.IsLocal {
  572. privAddr = network.LocalRange
  573. }
  574. }
  575. netID := network.NetID
  576. var accessToken models.AccessToken
  577. s := servercfg.GetServerConfig()
  578. w := servercfg.GetWGConfig()
  579. servervals := models.ServerConfig{
  580. CoreDNSAddr: s.CoreDNSAddr,
  581. APIConnString: s.APIConnString,
  582. APIHost: s.APIHost,
  583. APIPort: s.APIPort,
  584. GRPCConnString: s.GRPCConnString,
  585. GRPCHost: s.GRPCHost,
  586. GRPCPort: s.GRPCPort,
  587. GRPCSSL: s.GRPCSSL,
  588. }
  589. wgvals := models.WG{
  590. GRPCWireGuard: w.GRPCWireGuard,
  591. GRPCWGAddress: w.GRPCWGAddress,
  592. GRPCWGPort: w.GRPCWGPort,
  593. GRPCWGPubKey: w.GRPCWGPubKey,
  594. GRPCWGEndpoint: s.APIHost,
  595. }
  596. accessToken.ServerConfig = servervals
  597. accessToken.WG = wgvals
  598. accessToken.ClientConfig.Network = netID
  599. accessToken.ClientConfig.Key = accesskey.Value
  600. accessToken.ClientConfig.LocalRange = privAddr
  601. tokenjson, err := json.Marshal(accessToken)
  602. if err != nil {
  603. return accesskey, err
  604. }
  605. accesskey.AccessString = base64.StdEncoding.EncodeToString([]byte(tokenjson))
  606. //validate accesskey
  607. v := validator.New()
  608. err = v.Struct(accesskey)
  609. if err != nil {
  610. for _, e := range err.(validator.ValidationErrors) {
  611. fmt.Println(e)
  612. }
  613. return models.AccessKey{}, err
  614. }
  615. network.AccessKeys = append(network.AccessKeys, accesskey)
  616. data, err := json.Marshal(&network)
  617. if err != nil {
  618. return models.AccessKey{}, err
  619. }
  620. if err = database.Insert(network.NetID, string(data), database.NETWORKS_TABLE_NAME); err != nil {
  621. return models.AccessKey{}, err
  622. }
  623. return accesskey, nil
  624. }
  625. func GetSignupToken(netID string) (models.AccessKey, error) {
  626. var accesskey models.AccessKey
  627. var accessToken models.AccessToken
  628. s := servercfg.GetServerConfig()
  629. w := servercfg.GetWGConfig()
  630. servervals := models.ServerConfig{
  631. APIConnString: s.APIConnString,
  632. APIHost: s.APIHost,
  633. APIPort: s.APIPort,
  634. GRPCConnString: s.GRPCConnString,
  635. GRPCHost: s.GRPCHost,
  636. GRPCPort: s.GRPCPort,
  637. GRPCSSL: s.GRPCSSL,
  638. }
  639. wgvals := models.WG{
  640. GRPCWireGuard: w.GRPCWireGuard,
  641. GRPCWGAddress: w.GRPCWGAddress,
  642. GRPCWGPort: w.GRPCWGPort,
  643. GRPCWGPubKey: w.GRPCWGPubKey,
  644. GRPCWGEndpoint: s.APIHost,
  645. }
  646. accessToken.ServerConfig = servervals
  647. accessToken.WG = wgvals
  648. tokenjson, err := json.Marshal(accessToken)
  649. if err != nil {
  650. return accesskey, err
  651. }
  652. accesskey.AccessString = base64.StdEncoding.EncodeToString([]byte(tokenjson))
  653. return accesskey, nil
  654. }
  655. func getSignupToken(w http.ResponseWriter, r *http.Request) {
  656. w.Header().Set("Content-Type", "application/json")
  657. var params = mux.Vars(r)
  658. netID := params["networkname"]
  659. token, err := GetSignupToken(netID)
  660. if err != nil {
  661. returnErrorResponse(w, r, formatError(err, "internal"))
  662. return
  663. }
  664. functions.PrintUserLog(r.Header.Get("user"), "got signup token "+netID, 2)
  665. w.WriteHeader(http.StatusOK)
  666. json.NewEncoder(w).Encode(token)
  667. }
  668. //pretty simple get
  669. func getAccessKeys(w http.ResponseWriter, r *http.Request) {
  670. w.Header().Set("Content-Type", "application/json")
  671. var params = mux.Vars(r)
  672. network := params["networkname"]
  673. keys, err := GetKeys(network)
  674. if err != nil {
  675. returnErrorResponse(w, r, formatError(err, "internal"))
  676. return
  677. }
  678. functions.PrintUserLog(r.Header.Get("user"), "fetched access keys on network "+network, 2)
  679. w.WriteHeader(http.StatusOK)
  680. json.NewEncoder(w).Encode(keys)
  681. }
  682. func GetKeys(net string) ([]models.AccessKey, error) {
  683. record, err := database.FetchRecord(database.NETWORKS_TABLE_NAME, net)
  684. if err != nil {
  685. return []models.AccessKey{}, err
  686. }
  687. network, err := functions.ParseNetwork(record)
  688. if err != nil {
  689. return []models.AccessKey{}, err
  690. }
  691. return network.AccessKeys, nil
  692. }
  693. //delete key. Has to do a little funky logic since it's not a collection item
  694. func deleteAccessKey(w http.ResponseWriter, r *http.Request) {
  695. w.Header().Set("Content-Type", "application/json")
  696. var params = mux.Vars(r)
  697. keyname := params["name"]
  698. netname := params["networkname"]
  699. err := DeleteKey(keyname, netname)
  700. if err != nil {
  701. returnErrorResponse(w, r, formatError(err, "badrequest"))
  702. return
  703. }
  704. functions.PrintUserLog(r.Header.Get("user"), "deleted access key "+keyname+" on network "+netname, 1)
  705. w.WriteHeader(http.StatusOK)
  706. }
  707. func DeleteKey(keyname, netname string) error {
  708. network, err := functions.GetParentNetwork(netname)
  709. if err != nil {
  710. return err
  711. }
  712. //basically, turn the list of access keys into the list of access keys before and after the item
  713. //have not done any error handling for if there's like...1 item. I think it works? need to test.
  714. found := false
  715. var updatedKeys []models.AccessKey
  716. for _, currentkey := range network.AccessKeys {
  717. if currentkey.Name == keyname {
  718. found = true
  719. } else {
  720. updatedKeys = append(updatedKeys, currentkey)
  721. }
  722. }
  723. if !found {
  724. return errors.New("key " + keyname + " does not exist")
  725. }
  726. network.AccessKeys = updatedKeys
  727. data, err := json.Marshal(&network)
  728. if err != nil {
  729. return err
  730. }
  731. if err := database.Insert(network.NetID, string(data), database.NETWORKS_TABLE_NAME); err != nil {
  732. return err
  733. }
  734. return nil
  735. }