networkHttpController.go 22 KB

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