networkHttpController.go 23 KB

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