networkHttpController.go 28 KB

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