migrate.go 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909
  1. package migrate
  2. import (
  3. "context"
  4. "encoding/json"
  5. "fmt"
  6. "log"
  7. "net"
  8. "os"
  9. "time"
  10. "golang.org/x/exp/slog"
  11. "gorm.io/datatypes"
  12. "github.com/google/uuid"
  13. "github.com/gravitl/netmaker/database"
  14. "github.com/gravitl/netmaker/db"
  15. "github.com/gravitl/netmaker/logger"
  16. "github.com/gravitl/netmaker/logic"
  17. "github.com/gravitl/netmaker/logic/acls"
  18. "github.com/gravitl/netmaker/logic/acls/nodeacls"
  19. "github.com/gravitl/netmaker/models"
  20. "github.com/gravitl/netmaker/mq"
  21. "github.com/gravitl/netmaker/schema"
  22. "github.com/gravitl/netmaker/servercfg"
  23. )
  24. // Run - runs all migrations
  25. func Run() {
  26. migrateSettings()
  27. updateEnrollmentKeys()
  28. assignSuperAdmin()
  29. createDefaultTagsAndPolicies()
  30. removeOldUserGrps()
  31. migrateToUUIDs()
  32. syncUsers()
  33. updateHosts()
  34. updateNodes()
  35. checkAndDeprecateOldAcls()
  36. updateAcls()
  37. updateNewAcls()
  38. logic.MigrateToGws()
  39. migrateToEgressV1()
  40. updateNetworks()
  41. migrateNameservers()
  42. resync()
  43. deleteOldExtclients()
  44. }
  45. func checkAndDeprecateOldAcls() {
  46. // check if everything is allowed on old acl and disable old acls
  47. nets, _ := logic.GetNetworks()
  48. disableOldAcls := true
  49. for _, netI := range nets {
  50. networkACL, err := nodeacls.FetchAllACLs(nodeacls.NetworkID(netI.NetID))
  51. if err != nil {
  52. continue
  53. }
  54. for id, aclNode := range networkACL {
  55. if !aclNode.IsAllowed(id) {
  56. disableOldAcls = false
  57. }
  58. }
  59. }
  60. if disableOldAcls {
  61. settings := logic.GetServerSettings()
  62. settings.OldAClsSupport = false
  63. logic.UpsertServerSettings(settings)
  64. }
  65. }
  66. func updateNetworks() {
  67. nets, _ := logic.GetNetworks()
  68. for _, netI := range nets {
  69. if netI.AutoJoin == "" {
  70. netI.AutoJoin = "true"
  71. logic.UpsertNetwork(netI)
  72. }
  73. }
  74. }
  75. func migrateNameservers() {
  76. nets, _ := logic.GetNetworks()
  77. user, err := logic.GetSuperAdmin()
  78. if err != nil {
  79. return
  80. }
  81. for _, netI := range nets {
  82. _, cidr, err := net.ParseCIDR(netI.AddressRange)
  83. if err != nil {
  84. continue
  85. }
  86. if len(netI.NameServers) > 0 {
  87. ns := schema.Nameserver{
  88. ID: uuid.NewString(),
  89. Name: "upstream nameservers",
  90. NetworkID: netI.NetID,
  91. Servers: []string{},
  92. MatchAll: true,
  93. MatchDomains: []string{"."},
  94. Tags: datatypes.JSONMap{
  95. "*": struct{}{},
  96. },
  97. Nodes: make(datatypes.JSONMap),
  98. Status: true,
  99. CreatedBy: user.UserName,
  100. }
  101. for _, nsIP := range netI.NameServers {
  102. if net.ParseIP(nsIP) == nil {
  103. continue
  104. }
  105. if !cidr.Contains(net.ParseIP(nsIP)) {
  106. ns.Servers = append(ns.Servers, nsIP)
  107. }
  108. }
  109. ns.Create(db.WithContext(context.TODO()))
  110. netI.NameServers = []string{}
  111. logic.SaveNetwork(&netI)
  112. }
  113. }
  114. nodes, _ := logic.GetAllNodes()
  115. for _, node := range nodes {
  116. if !node.IsGw {
  117. continue
  118. }
  119. if node.IngressDNS != "" {
  120. if (node.Address.IP != nil && node.Address.IP.String() == node.IngressDNS) ||
  121. (node.Address6.IP != nil && node.Address6.IP.String() == node.IngressDNS) {
  122. continue
  123. }
  124. if node.IngressDNS == "8.8.8.8" || node.IngressDNS == "1.1.1.1" || node.IngressDNS == "9.9.9.9" {
  125. continue
  126. }
  127. h, err := logic.GetHost(node.HostID.String())
  128. if err != nil {
  129. continue
  130. }
  131. ns := schema.Nameserver{
  132. ID: uuid.NewString(),
  133. Name: fmt.Sprintf("%s gw nameservers", h.Name),
  134. NetworkID: node.Network,
  135. Servers: []string{node.IngressDNS},
  136. MatchAll: true,
  137. MatchDomains: []string{"."},
  138. Nodes: datatypes.JSONMap{
  139. node.ID.String(): struct{}{},
  140. },
  141. Tags: make(datatypes.JSONMap),
  142. Status: true,
  143. CreatedBy: user.UserName,
  144. }
  145. ns.Create(db.WithContext(context.TODO()))
  146. node.IngressDNS = ""
  147. logic.UpsertNode(&node)
  148. }
  149. }
  150. }
  151. // removes if any stale configurations from previous run.
  152. func resync() {
  153. nodes, _ := logic.GetAllNodes()
  154. for _, node := range nodes {
  155. if !node.IsGw {
  156. if len(node.RelayedNodes) > 0 {
  157. logic.DeleteRelay(node.Network, node.ID.String())
  158. }
  159. if node.IsIngressGateway {
  160. logic.DeleteIngressGateway(node.ID.String())
  161. }
  162. if len(node.InetNodeReq.InetNodeClientIDs) > 0 || node.IsInternetGateway {
  163. logic.UnsetInternetGw(&node)
  164. logic.UpsertNode(&node)
  165. }
  166. }
  167. if node.IsRelayed {
  168. if node.RelayedBy == "" {
  169. node.IsRelayed = false
  170. node.InternetGwID = ""
  171. logic.UpsertNode(&node)
  172. }
  173. if node.RelayedBy != "" {
  174. // check if node exists
  175. _, err := logic.GetNodeByID(node.RelayedBy)
  176. if err != nil {
  177. node.RelayedBy = ""
  178. node.InternetGwID = ""
  179. logic.UpsertNode(&node)
  180. }
  181. }
  182. }
  183. if node.InternetGwID != "" {
  184. _, err := logic.GetNodeByID(node.InternetGwID)
  185. if err != nil {
  186. node.InternetGwID = ""
  187. logic.UpsertNode(&node)
  188. }
  189. }
  190. }
  191. }
  192. func assignSuperAdmin() {
  193. users, err := logic.GetUsers()
  194. if err != nil || len(users) == 0 {
  195. return
  196. }
  197. if ok, _ := logic.HasSuperAdmin(); ok {
  198. return
  199. }
  200. createdSuperAdmin := false
  201. owner := servercfg.GetOwnerEmail()
  202. if owner != "" {
  203. user, err := logic.GetUser(owner)
  204. if err != nil {
  205. log.Fatal("error getting user", "user", owner, "error", err.Error())
  206. }
  207. user.PlatformRoleID = models.SuperAdminRole
  208. err = logic.UpsertUser(*user)
  209. if err != nil {
  210. log.Fatal(
  211. "error updating user to superadmin",
  212. "user",
  213. user.UserName,
  214. "error",
  215. err.Error(),
  216. )
  217. }
  218. return
  219. }
  220. for _, u := range users {
  221. var isAdmin bool
  222. if u.PlatformRoleID == models.AdminRole {
  223. isAdmin = true
  224. }
  225. if u.PlatformRoleID == "" && u.IsAdmin {
  226. isAdmin = true
  227. }
  228. if isAdmin {
  229. user, err := logic.GetUser(u.UserName)
  230. if err != nil {
  231. slog.Error("error getting user", "user", u.UserName, "error", err.Error())
  232. continue
  233. }
  234. user.PlatformRoleID = models.SuperAdminRole
  235. user.IsSuperAdmin = true
  236. err = logic.UpsertUser(*user)
  237. if err != nil {
  238. slog.Error(
  239. "error updating user to superadmin",
  240. "user",
  241. user.UserName,
  242. "error",
  243. err.Error(),
  244. )
  245. continue
  246. } else {
  247. createdSuperAdmin = true
  248. }
  249. break
  250. }
  251. }
  252. if !createdSuperAdmin {
  253. slog.Error("failed to create superadmin!!")
  254. }
  255. }
  256. func updateEnrollmentKeys() {
  257. rows, err := database.FetchRecords(database.ENROLLMENT_KEYS_TABLE_NAME)
  258. if err != nil {
  259. return
  260. }
  261. for _, row := range rows {
  262. var key models.EnrollmentKey
  263. if err = json.Unmarshal([]byte(row), &key); err != nil {
  264. continue
  265. }
  266. if key.Type != models.Undefined {
  267. logger.Log(2, "migration: enrollment key type already set")
  268. continue
  269. } else {
  270. logger.Log(2, "migration: updating enrollment key type")
  271. if key.Unlimited {
  272. key.Type = models.Unlimited
  273. } else if key.UsesRemaining > 0 {
  274. key.Type = models.Uses
  275. } else if !key.Expiration.IsZero() {
  276. key.Type = models.TimeExpiration
  277. }
  278. }
  279. data, err := json.Marshal(key)
  280. if err != nil {
  281. logger.Log(0, "migration: marshalling enrollment key: "+err.Error())
  282. continue
  283. }
  284. if err = database.Insert(key.Value, string(data), database.ENROLLMENT_KEYS_TABLE_NAME); err != nil {
  285. logger.Log(0, "migration: inserting enrollment key: "+err.Error())
  286. continue
  287. }
  288. }
  289. existingKeys, err := logic.GetAllEnrollmentKeys()
  290. if err != nil {
  291. return
  292. }
  293. // check if any tags are duplicate
  294. existingTags := make(map[string]struct{})
  295. for _, existingKey := range existingKeys {
  296. for _, t := range existingKey.Tags {
  297. existingTags[t] = struct{}{}
  298. }
  299. }
  300. networks, _ := logic.GetNetworks()
  301. for _, network := range networks {
  302. if _, ok := existingTags[network.NetID]; ok {
  303. continue
  304. }
  305. _, _ = logic.CreateEnrollmentKey(
  306. 0,
  307. time.Time{},
  308. []string{network.NetID},
  309. []string{network.NetID},
  310. []models.TagID{},
  311. true,
  312. uuid.Nil,
  313. true,
  314. false,
  315. false,
  316. )
  317. }
  318. }
  319. func removeOldUserGrps() {
  320. rows, err := database.FetchRecords(database.USER_GROUPS_TABLE_NAME)
  321. if err != nil {
  322. return
  323. }
  324. for key, row := range rows {
  325. userG := models.UserGroup{}
  326. _ = json.Unmarshal([]byte(row), &userG)
  327. if userG.ID == "" {
  328. database.DeleteRecord(database.USER_GROUPS_TABLE_NAME, key)
  329. }
  330. }
  331. }
  332. func updateHosts() {
  333. rows, err := database.FetchRecords(database.HOSTS_TABLE_NAME)
  334. if err != nil {
  335. logger.Log(0, "failed to fetch database records for hosts")
  336. }
  337. for _, row := range rows {
  338. var host models.Host
  339. if err := json.Unmarshal([]byte(row), &host); err != nil {
  340. logger.Log(0, "failed to unmarshal database row to host", "row", row)
  341. continue
  342. }
  343. if host.PersistentKeepalive == 0 {
  344. host.PersistentKeepalive = models.DefaultPersistentKeepAlive
  345. if err := logic.UpsertHost(&host); err != nil {
  346. logger.Log(0, "failed to upsert host", host.ID.String())
  347. continue
  348. }
  349. }
  350. if host.DNS == "" || (host.DNS != "yes" && host.DNS != "no") {
  351. if logic.GetServerSettings().ManageDNS {
  352. host.DNS = "yes"
  353. } else {
  354. host.DNS = "no"
  355. }
  356. if host.IsDefault {
  357. host.DNS = "yes"
  358. }
  359. logic.UpsertHost(&host)
  360. }
  361. if host.IsDefault && !host.AutoUpdate {
  362. host.AutoUpdate = true
  363. logic.UpsertHost(&host)
  364. }
  365. if servercfg.IsPro && host.Location == "" {
  366. if host.EndpointIP != nil {
  367. host.Location = logic.GetHostLocInfo(host.EndpointIP.String(), os.Getenv("IP_INFO_TOKEN"))
  368. } else if host.EndpointIPv6 != nil {
  369. host.Location = logic.GetHostLocInfo(host.EndpointIPv6.String(), os.Getenv("IP_INFO_TOKEN"))
  370. }
  371. if host.Location != "" {
  372. logic.UpsertHost(&host)
  373. }
  374. }
  375. }
  376. }
  377. func updateNodes() {
  378. nodes, err := logic.GetAllNodes()
  379. if err != nil {
  380. slog.Error("migration failed for nodes", "error", err)
  381. return
  382. }
  383. for _, node := range nodes {
  384. node := node
  385. if node.Tags == nil {
  386. node.Tags = make(map[models.TagID]struct{})
  387. logic.UpsertNode(&node)
  388. }
  389. // deprecate failover and initialise auto relay fields
  390. if node.IsFailOver {
  391. node.IsFailOver = false
  392. node.FailOverPeers = make(map[string]struct{})
  393. node.FailedOverBy = uuid.Nil
  394. node.AutoRelayedPeers = make(map[string]struct{})
  395. logic.UpsertNode(&node)
  396. }
  397. if node.FailedOverBy != uuid.Nil || len(node.FailOverPeers) > 0 {
  398. node.FailOverPeers = make(map[string]struct{})
  399. node.FailedOverBy = uuid.Nil
  400. node.AutoRelayedPeers = make(map[string]struct{})
  401. logic.UpsertNode(&node)
  402. }
  403. if node.IsIngressGateway {
  404. host, err := logic.GetHost(node.HostID.String())
  405. if err == nil {
  406. go logic.DeleteRole(models.GetRAGRoleID(node.Network, host.ID.String()), true)
  407. }
  408. }
  409. if node.IsEgressGateway {
  410. egressRanges, update := removeInterGw(node.EgressGatewayRanges)
  411. if update {
  412. node.EgressGatewayRequest.Ranges = egressRanges
  413. node.EgressGatewayRanges = egressRanges
  414. logic.UpsertNode(&node)
  415. }
  416. if len(node.EgressGatewayRequest.Ranges) > 0 && len(node.EgressGatewayRequest.RangesWithMetric) == 0 {
  417. for _, egressRangeI := range node.EgressGatewayRequest.Ranges {
  418. node.EgressGatewayRequest.RangesWithMetric = append(node.EgressGatewayRequest.RangesWithMetric, models.EgressRangeMetric{
  419. Network: egressRangeI,
  420. RouteMetric: 256,
  421. })
  422. }
  423. logic.UpsertNode(&node)
  424. }
  425. }
  426. }
  427. extclients, _ := logic.GetAllExtClients()
  428. for _, extclient := range extclients {
  429. if extclient.Tags == nil {
  430. extclient.Tags = make(map[models.TagID]struct{})
  431. logic.SaveExtClient(&extclient)
  432. }
  433. }
  434. }
  435. func removeInterGw(egressRanges []string) ([]string, bool) {
  436. update := false
  437. for i := len(egressRanges) - 1; i >= 0; i-- {
  438. if egressRanges[i] == "0.0.0.0/0" || egressRanges[i] == "::/0" {
  439. update = true
  440. egressRanges = append(egressRanges[:i], egressRanges[i+1:]...)
  441. }
  442. }
  443. return egressRanges, update
  444. }
  445. func updateAcls() {
  446. // get all networks
  447. if !logic.GetServerSettings().OldAClsSupport {
  448. return
  449. }
  450. networks, err := logic.GetNetworks()
  451. if err != nil && !database.IsEmptyRecord(err) {
  452. slog.Error("acls migration failed. error getting networks", "error", err)
  453. return
  454. }
  455. // get current acls per network
  456. for _, network := range networks {
  457. var networkAcl acls.ACLContainer
  458. networkAcl, err := networkAcl.Get(acls.ContainerID(network.NetID))
  459. if err != nil {
  460. if database.IsEmptyRecord(err) {
  461. continue
  462. }
  463. slog.Error(fmt.Sprintf("error during acls migration. error getting acls for network: %s", network.NetID), "error", err)
  464. continue
  465. }
  466. // convert old acls to new acls with clients
  467. // TODO: optimise O(n^2) operation
  468. clients, err := logic.GetNetworkExtClients(network.NetID)
  469. if err != nil {
  470. slog.Error(fmt.Sprintf("error during acls migration. error getting clients for network: %s", network.NetID), "error", err)
  471. continue
  472. }
  473. clientsIdMap := make(map[string]struct{})
  474. for _, client := range clients {
  475. clientsIdMap[client.ClientID] = struct{}{}
  476. }
  477. nodeIdsMap := make(map[string]struct{})
  478. for nodeId := range networkAcl {
  479. nodeIdsMap[string(nodeId)] = struct{}{}
  480. }
  481. /*
  482. initially, networkACL has only node acls so we add client acls to it
  483. final shape:
  484. {
  485. "node1": {
  486. "node2": 2,
  487. "client1": 2,
  488. "client2": 1,
  489. },
  490. "node2": {
  491. "node1": 2,
  492. "client1": 2,
  493. "client2": 1,
  494. },
  495. "client1": {
  496. "node1": 2,
  497. "node2": 2,
  498. "client2": 1,
  499. },
  500. "client2": {
  501. "node1": 1,
  502. "node2": 1,
  503. "client1": 1,
  504. },
  505. }
  506. */
  507. for _, client := range clients {
  508. networkAcl[acls.AclID(client.ClientID)] = acls.ACL{}
  509. // add client values to node acls and create client acls with node values
  510. for id, nodeAcl := range networkAcl {
  511. // skip if not a node
  512. if _, ok := nodeIdsMap[string(id)]; !ok {
  513. continue
  514. }
  515. if nodeAcl == nil {
  516. slog.Warn("acls migration bad data: nil node acl", "node", id, "network", network.NetID)
  517. continue
  518. }
  519. nodeAcl[acls.AclID(client.ClientID)] = acls.Allowed
  520. networkAcl[acls.AclID(client.ClientID)][id] = acls.Allowed
  521. if client.DeniedACLs == nil {
  522. continue
  523. } else if _, ok := client.DeniedACLs[string(id)]; ok {
  524. nodeAcl[acls.AclID(client.ClientID)] = acls.NotAllowed
  525. networkAcl[acls.AclID(client.ClientID)][id] = acls.NotAllowed
  526. }
  527. }
  528. // add clients to client acls response
  529. for _, c := range clients {
  530. if c.ClientID == client.ClientID {
  531. continue
  532. }
  533. networkAcl[acls.AclID(client.ClientID)][acls.AclID(c.ClientID)] = acls.Allowed
  534. if client.DeniedACLs == nil {
  535. continue
  536. } else if _, ok := client.DeniedACLs[c.ClientID]; ok {
  537. networkAcl[acls.AclID(client.ClientID)][acls.AclID(c.ClientID)] = acls.NotAllowed
  538. }
  539. }
  540. // delete oneself from its own acl
  541. delete(networkAcl[acls.AclID(client.ClientID)], acls.AclID(client.ClientID))
  542. }
  543. // remove non-existent client and node acls
  544. for objId := range networkAcl {
  545. if _, ok := nodeIdsMap[string(objId)]; ok {
  546. continue
  547. }
  548. if _, ok := clientsIdMap[string(objId)]; ok {
  549. continue
  550. }
  551. // remove all occurances of objId from all acls
  552. for objId2 := range networkAcl {
  553. delete(networkAcl[objId2], objId)
  554. }
  555. delete(networkAcl, objId)
  556. }
  557. // save new acls
  558. slog.Debug(fmt.Sprintf("(migration) saving new acls for network: %s", network.NetID), "networkAcl", networkAcl)
  559. if _, err := networkAcl.Save(acls.ContainerID(network.NetID)); err != nil {
  560. slog.Error(fmt.Sprintf("error during acls migration. error saving new acls for network: %s", network.NetID), "error", err)
  561. continue
  562. }
  563. slog.Info(fmt.Sprintf("(migration) successfully saved new acls for network: %s", network.NetID))
  564. }
  565. }
  566. func updateNewAcls() {
  567. if servercfg.IsPro {
  568. userGroups, _ := logic.ListUserGroups()
  569. userGroupMap := make(map[models.UserGroupID]models.UserGroup)
  570. for _, userGroup := range userGroups {
  571. userGroupMap[userGroup.ID] = userGroup
  572. }
  573. acls := logic.ListAcls()
  574. for _, acl := range acls {
  575. aclSrc := make([]models.AclPolicyTag, 0)
  576. for _, src := range acl.Src {
  577. if src.ID == models.UserGroupAclID {
  578. userGroup, ok := userGroupMap[models.UserGroupID(src.Value)]
  579. if !ok {
  580. // if the group doesn't exist, don't add it to the acl's src.
  581. continue
  582. } else {
  583. _, ok := userGroup.NetworkRoles[acl.NetworkID]
  584. if !ok {
  585. // if the group doesn't have permissions for the acl's
  586. // network, don't add it to the acl's src.
  587. continue
  588. }
  589. }
  590. }
  591. aclSrc = append(aclSrc, src)
  592. }
  593. if len(aclSrc) == 0 {
  594. // if there are no acl sources, delete the acl.
  595. _ = logic.DeleteAcl(acl)
  596. } else if len(aclSrc) != len(acl.Src) {
  597. // if some user groups were removed from the acl source,
  598. // update the acl.
  599. acl.Src = aclSrc
  600. _ = logic.UpsertAcl(acl)
  601. }
  602. }
  603. }
  604. }
  605. func MigrateEmqx() {
  606. err := mq.SendPullSYN()
  607. if err != nil {
  608. logger.Log(0, "failed to send pull syn to clients", "error", err.Error())
  609. }
  610. time.Sleep(time.Second * 3)
  611. slog.Info("proceeding to kicking out clients from emqx")
  612. err = mq.KickOutClients()
  613. if err != nil {
  614. logger.Log(2, "failed to migrate emqx: ", "kickout-error", err.Error())
  615. }
  616. }
  617. func migrateToUUIDs() {
  618. logic.MigrateToUUIDs()
  619. }
  620. func syncUsers() {
  621. // create default network user roles for existing networks
  622. if servercfg.IsPro {
  623. networks, _ := logic.GetNetworks()
  624. for _, netI := range networks {
  625. logic.CreateDefaultNetworkRolesAndGroups(models.NetworkID(netI.NetID))
  626. }
  627. }
  628. users, err := logic.GetUsersDB()
  629. if err == nil {
  630. for _, user := range users {
  631. user := user
  632. if user.PlatformRoleID == models.AdminRole && !user.IsAdmin {
  633. user.IsAdmin = true
  634. user.IsSuperAdmin = false
  635. logic.UpsertUser(user)
  636. }
  637. if user.PlatformRoleID == models.SuperAdminRole && !user.IsSuperAdmin {
  638. user.IsSuperAdmin = true
  639. user.IsAdmin = true
  640. logic.UpsertUser(user)
  641. }
  642. if user.PlatformRoleID == models.PlatformUser || user.PlatformRoleID == models.ServiceUser {
  643. user.IsSuperAdmin = false
  644. user.IsAdmin = false
  645. logic.UpsertUser(user)
  646. }
  647. if user.PlatformRoleID.String() != "" {
  648. logic.MigrateUserRoleAndGroups(user)
  649. logic.AddGlobalNetRolesToAdmins(&user)
  650. logic.UpsertUser(user)
  651. continue
  652. }
  653. user.AuthType = models.BasicAuth
  654. if logic.IsOauthUser(&user) == nil {
  655. user.AuthType = models.OAuth
  656. }
  657. if len(user.NetworkRoles) == 0 {
  658. user.NetworkRoles = make(map[models.NetworkID]map[models.UserRoleID]struct{})
  659. }
  660. if len(user.UserGroups) == 0 {
  661. user.UserGroups = make(map[models.UserGroupID]struct{})
  662. }
  663. // We reach here only if the platform role id has not been set.
  664. //
  665. // Thus, we use the boolean fields to assign the role.
  666. if user.IsSuperAdmin {
  667. user.PlatformRoleID = models.SuperAdminRole
  668. } else if user.IsAdmin {
  669. user.PlatformRoleID = models.AdminRole
  670. } else {
  671. user.PlatformRoleID = models.ServiceUser
  672. }
  673. logic.AddGlobalNetRolesToAdmins(&user)
  674. logic.MigrateUserRoleAndGroups(user)
  675. logic.UpsertUser(user)
  676. }
  677. }
  678. }
  679. func createDefaultTagsAndPolicies() {
  680. networks, err := logic.GetNetworks()
  681. if err != nil {
  682. return
  683. }
  684. for _, network := range networks {
  685. logic.CreateDefaultTags(models.NetworkID(network.NetID))
  686. logic.CreateDefaultAclNetworkPolicies(models.NetworkID(network.NetID))
  687. // delete old remote access gws policy
  688. logic.DeleteAcl(models.Acl{ID: fmt.Sprintf("%s.%s", network.NetID, "all-remote-access-gws")})
  689. }
  690. logic.MigrateAclPolicies()
  691. if !servercfg.IsPro {
  692. nodes, _ := logic.GetAllNodes()
  693. for _, node := range nodes {
  694. if node.IsGw {
  695. node.Tags = make(map[models.TagID]struct{})
  696. node.Tags[models.TagID(fmt.Sprintf("%s.%s", node.Network, models.GwTagName))] = struct{}{}
  697. logic.UpsertNode(&node)
  698. }
  699. }
  700. }
  701. }
  702. func migrateToEgressV1() {
  703. nodes, _ := logic.GetAllNodes()
  704. user, err := logic.GetSuperAdmin()
  705. if err != nil {
  706. return
  707. }
  708. for _, node := range nodes {
  709. if node.IsEgressGateway {
  710. _, err := logic.GetHost(node.HostID.String())
  711. if err != nil {
  712. continue
  713. }
  714. for _, rangeMetric := range node.EgressGatewayRequest.RangesWithMetric {
  715. e := &schema.Egress{Range: rangeMetric.Network}
  716. if err := e.DoesEgressRouteExists(db.WithContext(context.TODO())); err == nil {
  717. e.Nodes[node.ID.String()] = rangeMetric.RouteMetric
  718. e.Update(db.WithContext(context.TODO()))
  719. continue
  720. }
  721. e = &schema.Egress{
  722. ID: uuid.New().String(),
  723. Name: fmt.Sprintf("%s egress", rangeMetric.Network),
  724. Description: "",
  725. Network: node.Network,
  726. Nodes: datatypes.JSONMap{
  727. node.ID.String(): rangeMetric.RouteMetric,
  728. },
  729. Tags: make(datatypes.JSONMap),
  730. Range: rangeMetric.Network,
  731. Nat: node.EgressGatewayRequest.NatEnabled == "yes",
  732. Status: true,
  733. CreatedBy: user.UserName,
  734. CreatedAt: time.Now().UTC(),
  735. }
  736. err = e.Create(db.WithContext(context.TODO()))
  737. if err == nil {
  738. acl := models.Acl{
  739. ID: uuid.New().String(),
  740. Name: "egress node policy",
  741. MetaData: "",
  742. Default: false,
  743. ServiceType: models.Any,
  744. NetworkID: models.NetworkID(node.Network),
  745. Proto: models.ALL,
  746. RuleType: models.DevicePolicy,
  747. Src: []models.AclPolicyTag{
  748. {
  749. ID: models.NodeTagID,
  750. Value: "*",
  751. },
  752. },
  753. Dst: []models.AclPolicyTag{
  754. {
  755. ID: models.EgressID,
  756. Value: e.ID,
  757. },
  758. },
  759. AllowedDirection: models.TrafficDirectionBi,
  760. Enabled: true,
  761. CreatedBy: "auto",
  762. CreatedAt: time.Now().UTC(),
  763. }
  764. logic.InsertAcl(acl)
  765. acl = models.Acl{
  766. ID: uuid.New().String(),
  767. Name: "egress node policy",
  768. MetaData: "",
  769. Default: false,
  770. ServiceType: models.Any,
  771. NetworkID: models.NetworkID(node.Network),
  772. Proto: models.ALL,
  773. RuleType: models.UserPolicy,
  774. Src: []models.AclPolicyTag{
  775. {
  776. ID: models.UserAclID,
  777. Value: "*",
  778. },
  779. },
  780. Dst: []models.AclPolicyTag{
  781. {
  782. ID: models.EgressID,
  783. Value: e.ID,
  784. },
  785. },
  786. AllowedDirection: models.TrafficDirectionBi,
  787. Enabled: true,
  788. CreatedBy: "auto",
  789. CreatedAt: time.Now().UTC(),
  790. }
  791. logic.InsertAcl(acl)
  792. }
  793. }
  794. node.IsEgressGateway = false
  795. node.EgressGatewayRequest = models.EgressGatewayRequest{}
  796. node.EgressGatewayNatEnabled = false
  797. node.EgressGatewayRanges = []string{}
  798. logic.UpsertNode(&node)
  799. }
  800. }
  801. }
  802. func migrateSettings() {
  803. settingsD := make(map[string]interface{})
  804. data, err := database.FetchRecord(database.SERVER_SETTINGS, logic.ServerSettingsDBKey)
  805. if database.IsEmptyRecord(err) {
  806. logic.UpsertServerSettings(logic.GetServerSettingsFromEnv())
  807. } else if err == nil {
  808. json.Unmarshal([]byte(data), &settingsD)
  809. }
  810. settings := logic.GetServerSettings()
  811. if _, ok := settingsD["old_acl_support"]; !ok {
  812. settings.OldAClsSupport = servercfg.IsOldAclEnabled()
  813. }
  814. if settings.AuditLogsRetentionPeriodInDays == 0 {
  815. settings.AuditLogsRetentionPeriodInDays = 7
  816. }
  817. if settings.DefaultDomain == "" {
  818. settings.DefaultDomain = servercfg.GetDefaultDomain()
  819. }
  820. if settings.JwtValidityDurationClients == 0 {
  821. settings.JwtValidityDurationClients = servercfg.GetJwtValidityDurationFromEnv() / 60
  822. }
  823. if settings.StunServers == "" {
  824. settings.StunServers = servercfg.GetStunServers()
  825. }
  826. logic.UpsertServerSettings(settings)
  827. }
  828. func deleteOldExtclients() {
  829. extclients, _ := logic.GetAllExtClients()
  830. userExtclientMap := make(map[string][]models.ExtClient)
  831. for _, extclient := range extclients {
  832. if extclient.RemoteAccessClientID == "" {
  833. continue
  834. }
  835. if extclient.Enabled {
  836. continue
  837. }
  838. if _, ok := userExtclientMap[extclient.OwnerID]; !ok {
  839. userExtclientMap[extclient.OwnerID] = make([]models.ExtClient, 0)
  840. }
  841. userExtclientMap[extclient.OwnerID] = append(userExtclientMap[extclient.OwnerID], extclient)
  842. }
  843. for _, userExtclients := range userExtclientMap {
  844. if len(userExtclients) > 1 {
  845. for _, extclient := range userExtclients[1:] {
  846. _ = logic.DeleteExtClient(extclient.Network, extclient.Network, false)
  847. }
  848. }
  849. }
  850. }