networkHttpController.go 23 KB

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