helpers.go 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050
  1. //TODO: Consider restructuring this file/folder "github.com/gorilla/handlers"
  2. //It may make more sense to split into different files and not call it "helpers"
  3. package functions
  4. import (
  5. "context"
  6. "encoding/base64"
  7. "errors"
  8. "fmt"
  9. "log"
  10. "math/rand"
  11. "net"
  12. "strings"
  13. "time"
  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/bson/primitive"
  19. "go.mongodb.org/mongo-driver/mongo"
  20. "go.mongodb.org/mongo-driver/mongo/options"
  21. )
  22. func PrintUserLog(username string, message string, loglevel int) {
  23. log.SetFlags(log.Flags() &^ (log.Llongfile | log.Lshortfile))
  24. if int32(loglevel) <= servercfg.GetVerbose() && servercfg.GetVerbose() != 0 {
  25. log.Println(username, message)
  26. }
  27. }
  28. //Takes in an arbitrary field and value for field and checks to see if any other
  29. //node has that value for the same field within the network
  30. func SliceContains(slice []string, item string) bool {
  31. set := make(map[string]struct{}, len(slice))
  32. for _, s := range slice {
  33. set[s] = struct{}{}
  34. }
  35. _, ok := set[item]
  36. return ok
  37. }
  38. func CreateServerToken(netID string) (string, error) {
  39. var network models.Network
  40. var accesskey models.AccessKey
  41. network, err := GetParentNetwork(netID)
  42. if err != nil {
  43. return "", err
  44. }
  45. accesskey.Name = GenKeyName()
  46. accesskey.Value = GenKey()
  47. accesskey.Uses = 1
  48. address := "127.0.0.1:" + servercfg.GetGRPCPort()
  49. privAddr := ""
  50. if *network.IsLocal {
  51. privAddr = network.LocalRange
  52. }
  53. accessstringdec := address + "|" + address + "|" + address + "|" + netID + "|" + accesskey.Value + "|" + privAddr
  54. accesskey.AccessString = base64.StdEncoding.EncodeToString([]byte(accessstringdec))
  55. network.AccessKeys = append(network.AccessKeys, accesskey)
  56. collection := mongoconn.Client.Database("netmaker").Collection("networks")
  57. ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
  58. // Create filter
  59. filter := bson.M{"netid": netID}
  60. // prepare update model.
  61. update := bson.D{
  62. {"$set", bson.D{
  63. {"accesskeys", network.AccessKeys},
  64. }},
  65. }
  66. errN := collection.FindOneAndUpdate(ctx, filter, update).Decode(&network)
  67. defer cancel()
  68. if errN != nil {
  69. return "", errN
  70. }
  71. return accesskey.AccessString, nil
  72. }
  73. func GetPeersList(networkName string) ([]models.PeersResponse, error) {
  74. var peers []models.PeersResponse
  75. //Connection mongoDB with mongoconn class
  76. collection := mongoconn.Client.Database("netmaker").Collection("nodes")
  77. ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
  78. //Get all nodes in the relevant network which are NOT in pending state
  79. filter := bson.M{"network": networkName, "ispending": false}
  80. cur, err := collection.Find(ctx, filter)
  81. if err != nil {
  82. return peers, err
  83. }
  84. // Close the cursor once finished and cancel if it takes too long
  85. defer cancel()
  86. for cur.Next(context.TODO()) {
  87. var peer models.PeersResponse
  88. err := cur.Decode(&peer)
  89. if err != nil {
  90. log.Fatal(err)
  91. }
  92. // add the node to our node array
  93. //maybe better to just return this? But then that's just GetNodes...
  94. peers = append(peers, peer)
  95. }
  96. //Uh oh, fatal error! This needs some better error handling
  97. //TODO: needs appropriate error handling so the server doesnt shut down.
  98. if err := cur.Err(); err != nil {
  99. log.Fatal(err)
  100. }
  101. return peers, err
  102. }
  103. func GetIntPeersList() ([]models.PeersResponse, error) {
  104. var peers []models.PeersResponse
  105. collection := mongoconn.Client.Database("netmaker").Collection("intclients")
  106. ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
  107. filter := bson.M{"isserver": ""}
  108. cur, err := collection.Find(ctx, filter)
  109. if err != nil {
  110. return peers, err
  111. }
  112. // Close the cursor once finished and cancel if it takes too long
  113. defer cancel()
  114. for cur.Next(context.TODO()) {
  115. var peer models.PeersResponse
  116. err := cur.Decode(&peer)
  117. if err != nil {
  118. log.Fatal(err)
  119. }
  120. // add the node to our node array
  121. //maybe better to just return this? But then that's just GetNodes...
  122. peers = append(peers, peer)
  123. }
  124. //Uh oh, fatal error! This needs some better error handling
  125. //TODO: needs appropriate error handling so the server doesnt shut down.
  126. if err := cur.Err(); err != nil {
  127. log.Fatal(err)
  128. }
  129. return peers, err
  130. }
  131. func IsFieldUnique(network string, field string, value string) bool {
  132. var node models.Node
  133. isunique := true
  134. collection := mongoconn.Client.Database("netmaker").Collection("nodes")
  135. ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
  136. filter := bson.M{field: value, "network": network}
  137. err := collection.FindOne(ctx, filter).Decode(&node)
  138. defer cancel()
  139. if err != nil {
  140. return isunique
  141. }
  142. if node.Name != "" {
  143. isunique = false
  144. }
  145. return isunique
  146. }
  147. func ServerIntClientExists() (bool, error) {
  148. collection := mongoconn.Client.Database("netmaker").Collection("intclients")
  149. ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
  150. filter := bson.M{"isserver": "yes"}
  151. var result bson.M
  152. err := collection.FindOne(ctx, filter).Decode(&result)
  153. defer cancel()
  154. if err != nil {
  155. if err == mongo.ErrNoDocuments {
  156. return false, nil
  157. }
  158. }
  159. return true, err
  160. }
  161. func NetworkExists(name string) (bool, error) {
  162. collection := mongoconn.Client.Database("netmaker").Collection("networks")
  163. ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
  164. filter := bson.M{"netid": name}
  165. var result bson.M
  166. err := collection.FindOne(ctx, filter).Decode(&result)
  167. defer cancel()
  168. if err != nil {
  169. if err == mongo.ErrNoDocuments {
  170. return false, nil
  171. }
  172. }
  173. return true, err
  174. }
  175. //TODO: This is very inefficient (N-squared). Need to find a better way.
  176. //Takes a list of nodes in a network and iterates through
  177. //for each node, it gets a unique address. That requires checking against all other nodes once more
  178. func UpdateNetworkNodeAddresses(networkName string) error {
  179. //Connection mongoDB with mongoconn class
  180. collection := mongoconn.Client.Database("netmaker").Collection("nodes")
  181. ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
  182. filter := bson.M{"network": networkName}
  183. cur, err := collection.Find(ctx, filter)
  184. if err != nil {
  185. return err
  186. }
  187. defer cancel()
  188. for cur.Next(context.TODO()) {
  189. var node models.Node
  190. err := cur.Decode(&node)
  191. if err != nil {
  192. fmt.Println("error in node address assignment!")
  193. return err
  194. }
  195. ipaddr, iperr := UniqueAddress(networkName)
  196. if iperr != nil {
  197. fmt.Println("error in node address assignment!")
  198. return iperr
  199. }
  200. filter := bson.M{"macaddress": node.MacAddress}
  201. update := bson.D{{"$set", bson.D{{"address", ipaddr}}}}
  202. errN := collection.FindOneAndUpdate(ctx, filter, update).Decode(&node)
  203. defer cancel()
  204. if errN != nil {
  205. return errN
  206. }
  207. }
  208. return err
  209. }
  210. //TODO TODO TODO!!!!!
  211. func UpdateNetworkPrivateAddresses(networkName string) error {
  212. //Connection mongoDB with mongoconn class
  213. collection := mongoconn.Client.Database("netmaker").Collection("nodes")
  214. ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
  215. filter := bson.M{"network": networkName}
  216. cur, err := collection.Find(ctx, filter)
  217. if err != nil {
  218. return err
  219. }
  220. defer cancel()
  221. for cur.Next(context.TODO()) {
  222. var node models.Node
  223. err := cur.Decode(&node)
  224. if err != nil {
  225. fmt.Println("error in node address assignment!")
  226. return err
  227. }
  228. ipaddr, iperr := UniqueAddress(networkName)
  229. if iperr != nil {
  230. fmt.Println("error in node address assignment!")
  231. return iperr
  232. }
  233. filter := bson.M{"macaddress": node.MacAddress}
  234. update := bson.D{{"$set", bson.D{{"address", ipaddr}}}}
  235. errN := collection.FindOneAndUpdate(ctx, filter, update).Decode(&node)
  236. defer cancel()
  237. if errN != nil {
  238. return errN
  239. }
  240. }
  241. return err
  242. }
  243. //Checks to see if any other networks have the same name (id)
  244. func IsNetworkNameUnique(name string) (bool, error) {
  245. isunique := true
  246. dbs, err := ListNetworks()
  247. if err != nil {
  248. return false, err
  249. }
  250. for i := 0; i < len(dbs); i++ {
  251. if name == dbs[i].NetID {
  252. isunique = false
  253. }
  254. }
  255. return isunique, nil
  256. }
  257. func IsNetworkDisplayNameUnique(name string) (bool, error) {
  258. isunique := true
  259. dbs, err := ListNetworks()
  260. if err != nil {
  261. return false, err
  262. }
  263. for i := 0; i < len(dbs); i++ {
  264. if name == dbs[i].DisplayName {
  265. isunique = false
  266. }
  267. }
  268. return isunique, nil
  269. }
  270. func GetNetworkNodeNumber(networkName string) (int, error) {
  271. collection := mongoconn.Client.Database("netmaker").Collection("nodes")
  272. ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
  273. filter := bson.M{"network": networkName}
  274. count, err := collection.CountDocuments(ctx, filter)
  275. returncount := int(count)
  276. //not sure if this is the right way of handling this error...
  277. if err != nil {
  278. return 9999, err
  279. }
  280. defer cancel()
  281. return returncount, err
  282. }
  283. //Kind of a weird name. Should just be GetNetworks I think. Consider changing.
  284. //Anyway, returns all the networks
  285. func ListNetworks() ([]models.Network, error) {
  286. var networks []models.Network
  287. collection := mongoconn.Client.Database("netmaker").Collection("networks")
  288. ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
  289. cur, err := collection.Find(ctx, bson.M{}, options.Find().SetProjection(bson.M{"_id": 0}))
  290. if err != nil {
  291. return networks, err
  292. }
  293. defer cancel()
  294. for cur.Next(context.TODO()) {
  295. var network models.Network
  296. err := cur.Decode(&network)
  297. if err != nil {
  298. return networks, err
  299. }
  300. // add network our array
  301. networks = append(networks, network)
  302. }
  303. if err := cur.Err(); err != nil {
  304. return networks, err
  305. }
  306. return networks, err
  307. }
  308. //Checks to see if access key is valid
  309. //Does so by checking against all keys and seeing if any have the same value
  310. //may want to hash values before comparing...consider this
  311. //TODO: No error handling!!!!
  312. func IsKeyValid(networkname string, keyvalue string) bool {
  313. network, _ := GetParentNetwork(networkname)
  314. var key models.AccessKey
  315. foundkey := false
  316. isvalid := false
  317. for i := len(network.AccessKeys) - 1; i >= 0; i-- {
  318. currentkey := network.AccessKeys[i]
  319. if currentkey.Value == keyvalue {
  320. key = currentkey
  321. foundkey = true
  322. }
  323. }
  324. if foundkey {
  325. if key.Uses > 0 {
  326. isvalid = true
  327. }
  328. }
  329. return isvalid
  330. }
  331. func IsKeyValidGlobal(keyvalue string) bool {
  332. networks, _ := ListNetworks()
  333. var key models.AccessKey
  334. foundkey := false
  335. isvalid := false
  336. for _, network := range networks {
  337. for i := len(network.AccessKeys) - 1; i >= 0; i-- {
  338. currentkey := network.AccessKeys[i]
  339. if currentkey.Value == keyvalue {
  340. key = currentkey
  341. foundkey = true
  342. break
  343. }
  344. }
  345. if foundkey {
  346. break
  347. }
  348. }
  349. if foundkey {
  350. if key.Uses > 0 {
  351. isvalid = true
  352. }
  353. }
  354. return isvalid
  355. }
  356. //TODO: Contains a fatal error return. Need to change
  357. //This just gets a network object from a network name
  358. //Should probably just be GetNetwork. kind of a dumb name.
  359. //Used in contexts where it's not the Parent network.
  360. func GetParentNetwork(networkname string) (models.Network, error) {
  361. var network models.Network
  362. collection := mongoconn.Client.Database("netmaker").Collection("networks")
  363. ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
  364. filter := bson.M{"netid": networkname}
  365. err := collection.FindOne(ctx, filter).Decode(&network)
  366. defer cancel()
  367. if err != nil {
  368. return network, err
  369. }
  370. return network, nil
  371. }
  372. func IsIpNet(host string) bool {
  373. return net.ParseIP(host) != nil
  374. }
  375. //Similar to above but checks if Cidr range is valid
  376. //At least this guy's got some print statements
  377. //still not good error handling
  378. func IsIpCIDR(host string) bool {
  379. ip, ipnet, err := net.ParseCIDR(host)
  380. if err != nil {
  381. fmt.Println(err)
  382. fmt.Println("Address Range is not valid!")
  383. return false
  384. }
  385. return ip != nil && ipnet != nil
  386. }
  387. //This is used to validate public keys (make sure they're base64 encoded like all public keys should be).
  388. func IsBase64(s string) bool {
  389. _, err := base64.StdEncoding.DecodeString(s)
  390. return err == nil
  391. }
  392. //This should probably just be called GetNode
  393. //It returns a node based on the ID of the node.
  394. //Why do we need this?
  395. //TODO: Check references. This seems unnecessary.
  396. func GetNodeObj(id primitive.ObjectID) models.Node {
  397. var node models.Node
  398. collection := mongoconn.Client.Database("netmaker").Collection("nodes")
  399. ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
  400. filter := bson.M{"_id": id}
  401. err := collection.FindOne(ctx, filter).Decode(&node)
  402. defer cancel()
  403. if err != nil {
  404. fmt.Println(err)
  405. fmt.Println("Did not get the node...")
  406. return node
  407. }
  408. fmt.Println("Got node " + node.Name)
  409. return node
  410. }
  411. //This checks to make sure a network name is valid.
  412. //Switch to REGEX?
  413. func NameInNetworkCharSet(name string) bool {
  414. charset := "abcdefghijklmnopqrstuvwxyz1234567890-_."
  415. for _, char := range name {
  416. if !strings.Contains(charset, strings.ToLower(string(char))) {
  417. return false
  418. }
  419. }
  420. return true
  421. }
  422. func NameInDNSCharSet(name string) bool {
  423. charset := "abcdefghijklmnopqrstuvwxyz1234567890-."
  424. for _, char := range name {
  425. if !strings.Contains(charset, strings.ToLower(string(char))) {
  426. return false
  427. }
  428. }
  429. return true
  430. }
  431. func NameInNodeCharSet(name string) bool {
  432. charset := "abcdefghijklmnopqrstuvwxyz1234567890-"
  433. for _, char := range name {
  434. if !strings.Contains(charset, strings.ToLower(string(char))) {
  435. return false
  436. }
  437. }
  438. return true
  439. }
  440. //This returns a node based on its mac address.
  441. //The mac address acts as the Unique ID for nodes.
  442. //Is this a dumb thing to do? I thought it was cool but maybe it's dumb.
  443. //It doesn't really provide a tangible benefit over a random ID
  444. func GetNodeByMacAddress(network string, macaddress string) (models.Node, error) {
  445. var node models.Node
  446. filter := bson.M{"macaddress": macaddress, "network": network}
  447. collection := mongoconn.Client.Database("netmaker").Collection("nodes")
  448. ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
  449. err := collection.FindOne(ctx, filter).Decode(&node)
  450. defer cancel()
  451. if err != nil {
  452. return node, err
  453. }
  454. return node, nil
  455. }
  456. func DeleteAllIntClients() error {
  457. collection := mongoconn.Client.Database("netmaker").Collection("intclients")
  458. ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
  459. // Filter out them ID's again
  460. err := collection.Drop(ctx)
  461. if err != nil {
  462. return err
  463. }
  464. defer cancel()
  465. return nil
  466. }
  467. func GetAllIntClients() ([]models.IntClient, error) {
  468. var client models.IntClient
  469. var clients []models.IntClient
  470. collection := mongoconn.Client.Database("netmaker").Collection("intclients")
  471. ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
  472. // Filter out them ID's again
  473. cur, err := collection.Find(ctx, bson.M{}, options.Find().SetProjection(bson.M{"_id": 0}))
  474. if err != nil {
  475. return []models.IntClient{}, err
  476. }
  477. defer cancel()
  478. for cur.Next(context.TODO()) {
  479. err := cur.Decode(&client)
  480. if err != nil {
  481. return []models.IntClient{}, err
  482. }
  483. // add node to our array
  484. clients = append(clients, client)
  485. }
  486. //TODO: Fatal error
  487. if err := cur.Err(); err != nil {
  488. return []models.IntClient{}, err
  489. }
  490. return clients, nil
  491. }
  492. func GetAllExtClients() ([]models.ExtClient, error) {
  493. var extclient models.ExtClient
  494. var extclients []models.ExtClient
  495. collection := mongoconn.Client.Database("netmaker").Collection("extclients")
  496. ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
  497. // Filter out them ID's again
  498. cur, err := collection.Find(ctx, bson.M{}, options.Find().SetProjection(bson.M{"_id": 0}))
  499. if err != nil {
  500. return []models.ExtClient{}, err
  501. }
  502. defer cancel()
  503. for cur.Next(context.TODO()) {
  504. err := cur.Decode(&extclient)
  505. if err != nil {
  506. return []models.ExtClient{}, err
  507. }
  508. // add node to our array
  509. extclients = append(extclients, extclient)
  510. }
  511. //TODO: Fatal error
  512. if err := cur.Err(); err != nil {
  513. return []models.ExtClient{}, err
  514. }
  515. return extclients, nil
  516. }
  517. //This returns a unique address for a node to use
  518. //it iterates through the list of IP's in the subnet
  519. //and checks against all nodes to see if it's taken, until it finds one.
  520. //TODO: We do not handle a case where we run out of addresses.
  521. //We will need to handle that eventually
  522. func UniqueAddress(networkName string) (string, error) {
  523. var network models.Network
  524. network, err := GetParentNetwork(networkName)
  525. if err != nil {
  526. fmt.Println("UniqueAddress encountered an error")
  527. return "666", err
  528. }
  529. offset := true
  530. ip, ipnet, err := net.ParseCIDR(network.AddressRange)
  531. if err != nil {
  532. fmt.Println("UniqueAddress encountered an error")
  533. return "666", err
  534. }
  535. for ip := ip.Mask(ipnet.Mask); ipnet.Contains(ip); Inc(ip) {
  536. if offset {
  537. offset = false
  538. continue
  539. }
  540. if networkName == "comms" {
  541. if IsIPUniqueClients(networkName, ip.String()) {
  542. return ip.String(), err
  543. }
  544. } else {
  545. if IsIPUnique(networkName, ip.String()) && IsIPUniqueExtClients(networkName, ip.String()) {
  546. return ip.String(), err
  547. }
  548. }
  549. }
  550. //TODO
  551. err1 := errors.New("ERROR: No unique addresses available. Check network subnet.")
  552. return "W1R3: NO UNIQUE ADDRESSES AVAILABLE", err1
  553. }
  554. func UniqueAddress6(networkName string) (string, error) {
  555. var network models.Network
  556. network, err := GetParentNetwork(networkName)
  557. if err != nil {
  558. fmt.Println("Network Not Found")
  559. return "", err
  560. }
  561. if network.IsDualStack == nil || *network.IsDualStack == false {
  562. if networkName != "comms" {
  563. return "", nil
  564. }
  565. }
  566. offset := true
  567. ip, ipnet, err := net.ParseCIDR(network.AddressRange6)
  568. if err != nil {
  569. fmt.Println("UniqueAddress6 encountered an error")
  570. return "666", err
  571. }
  572. for ip := ip.Mask(ipnet.Mask); ipnet.Contains(ip); Inc(ip) {
  573. if offset {
  574. offset = false
  575. continue
  576. }
  577. if networkName == "comms" {
  578. if IsIP6UniqueClients(networkName, ip.String()) {
  579. return ip.String(), err
  580. }
  581. } else {
  582. if IsIP6Unique(networkName, ip.String()) {
  583. return ip.String(), err
  584. }
  585. }
  586. }
  587. //TODO
  588. err1 := errors.New("ERROR: No unique addresses available. Check network subnet.")
  589. return "W1R3: NO UNIQUE ADDRESSES AVAILABLE", err1
  590. }
  591. //generate an access key value
  592. func GenKey() string {
  593. var seededRand *rand.Rand = rand.New(
  594. rand.NewSource(time.Now().UnixNano()))
  595. length := 16
  596. charset := "abcdefghijklmnopqrstuvwxyz" + "ABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"
  597. b := make([]byte, length)
  598. for i := range b {
  599. b[i] = charset[seededRand.Intn(len(charset))]
  600. }
  601. return string(b)
  602. }
  603. //generate a key value
  604. //we should probably just have 1 random string generator
  605. //that can be used across all functions
  606. //have a "base string" a "length" and a "charset"
  607. func GenKeyName() string {
  608. var seededRand *rand.Rand = rand.New(
  609. rand.NewSource(time.Now().UnixNano()))
  610. length := 5
  611. charset := "abcdefghijklmnopqrstuvwxyz" + "ABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"
  612. b := make([]byte, length)
  613. for i := range b {
  614. b[i] = charset[seededRand.Intn(len(charset))]
  615. }
  616. return "key" + string(b)
  617. }
  618. func IsIPUniqueExtClients(network string, ip string) bool {
  619. var extclient models.ExtClient
  620. isunique := true
  621. collection := mongoconn.Client.Database("netmaker").Collection("extclients")
  622. ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
  623. filter := bson.M{"address": ip, "network": network}
  624. err := collection.FindOne(ctx, filter).Decode(&extclient)
  625. defer cancel()
  626. if err != nil {
  627. return isunique
  628. }
  629. if extclient.Address == ip {
  630. isunique = false
  631. }
  632. return isunique
  633. }
  634. //checks if IP is unique in the address range
  635. //used by UniqueAddress
  636. func IsIPUnique(network string, ip string) bool {
  637. var node models.Node
  638. isunique := true
  639. collection := mongoconn.Client.Database("netmaker").Collection("nodes")
  640. ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
  641. filter := bson.M{"address": ip, "network": network}
  642. err := collection.FindOne(ctx, filter).Decode(&node)
  643. defer cancel()
  644. if err != nil {
  645. return isunique
  646. }
  647. if node.Address == ip {
  648. isunique = false
  649. }
  650. return isunique
  651. }
  652. //checks if IP is unique in the address range
  653. //used by UniqueAddress
  654. func IsIP6Unique(network string, ip string) bool {
  655. var node models.Node
  656. isunique := true
  657. collection := mongoconn.Client.Database("netmaker").Collection("nodes")
  658. ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
  659. filter := bson.M{"address6": ip, "network": network}
  660. err := collection.FindOne(ctx, filter).Decode(&node)
  661. defer cancel()
  662. if err != nil {
  663. return isunique
  664. }
  665. if node.Address6 == ip {
  666. isunique = false
  667. }
  668. return isunique
  669. }
  670. //checks if IP is unique in the address range
  671. //used by UniqueAddress
  672. func IsIP6UniqueClients(network string, ip string) bool {
  673. var client models.IntClient
  674. isunique := true
  675. collection := mongoconn.Client.Database("netmaker").Collection("intclients")
  676. ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
  677. filter := bson.M{"address6": ip, "network": network}
  678. err := collection.FindOne(ctx, filter).Decode(&client)
  679. defer cancel()
  680. if err != nil {
  681. return isunique
  682. }
  683. if client.Address6 == ip {
  684. isunique = false
  685. }
  686. return isunique
  687. }
  688. //checks if IP is unique in the address range
  689. //used by UniqueAddress
  690. func IsIPUniqueClients(network string, ip string) bool {
  691. var client models.IntClient
  692. isunique := true
  693. collection := mongoconn.Client.Database("netmaker").Collection("intclients")
  694. ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
  695. filter := bson.M{"address": ip, "network": network}
  696. err := collection.FindOne(ctx, filter).Decode(&client)
  697. defer cancel()
  698. if err != nil {
  699. return isunique
  700. }
  701. if client.Address == ip {
  702. isunique = false
  703. }
  704. return isunique
  705. }
  706. //called once key has been used by createNode
  707. //reduces value by one and deletes if necessary
  708. func DecrimentKey(networkName string, keyvalue string) {
  709. var network models.Network
  710. network, err := GetParentNetwork(networkName)
  711. if err != nil {
  712. return
  713. }
  714. for i := len(network.AccessKeys) - 1; i >= 0; i-- {
  715. currentkey := network.AccessKeys[i]
  716. if currentkey.Value == keyvalue {
  717. network.AccessKeys[i].Uses--
  718. if network.AccessKeys[i].Uses < 1 {
  719. //this is the part where it will call the delete
  720. //not sure if there's edge cases I'm missing
  721. DeleteKey(network, i)
  722. return
  723. }
  724. }
  725. }
  726. collection := mongoconn.Client.Database("netmaker").Collection("networks")
  727. ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
  728. filter := bson.M{"netid": network.NetID}
  729. update := bson.D{
  730. {"$set", bson.D{
  731. {"accesskeys", network.AccessKeys},
  732. }},
  733. }
  734. errN := collection.FindOneAndUpdate(ctx, filter, update).Decode(&network)
  735. defer cancel()
  736. if errN != nil {
  737. return
  738. }
  739. }
  740. //takes the logic from controllers.deleteKey
  741. func DeleteKey(network models.Network, i int) {
  742. network.AccessKeys = append(network.AccessKeys[:i],
  743. network.AccessKeys[i+1:]...)
  744. collection := mongoconn.Client.Database("netmaker").Collection("networks")
  745. ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
  746. // Create filter
  747. filter := bson.M{"netid": network.NetID}
  748. // prepare update model.
  749. update := bson.D{
  750. {"$set", bson.D{
  751. {"accesskeys", network.AccessKeys},
  752. }},
  753. }
  754. errN := collection.FindOneAndUpdate(ctx, filter, update).Decode(&network)
  755. defer cancel()
  756. if errN != nil {
  757. return
  758. }
  759. }
  760. //increments an IP over the previous
  761. func Inc(ip net.IP) {
  762. for j := len(ip) - 1; j >= 0; j-- {
  763. ip[j]++
  764. if ip[j] > 0 {
  765. break
  766. }
  767. }
  768. }
  769. func GetAllNodes() ([]models.Node, error) {
  770. var node models.Node
  771. var nodes []models.Node
  772. collection := mongoconn.Client.Database("netmaker").Collection("nodes")
  773. ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
  774. // Filter out them ID's again
  775. cur, err := collection.Find(ctx, bson.M{}, options.Find().SetProjection(bson.M{"_id": 0}))
  776. if err != nil {
  777. return []models.Node{}, err
  778. }
  779. defer cancel()
  780. for cur.Next(context.TODO()) {
  781. err := cur.Decode(&node)
  782. if err != nil {
  783. return []models.Node{}, err
  784. }
  785. // add node to our array
  786. nodes = append(nodes, node)
  787. }
  788. //TODO: Fatal error
  789. if err := cur.Err(); err != nil {
  790. return []models.Node{}, err
  791. }
  792. return nodes, nil
  793. }