helpers.go 24 KB

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