helpers.go 25 KB

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