networkHttpController.go 25 KB

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