migrate.go 22 KB

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