networkHttpController.go 26 KB

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