networkHttpController.go 23 KB

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