user_mgmt.go 31 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093
  1. package logic
  2. import (
  3. "encoding/json"
  4. "errors"
  5. "fmt"
  6. "github.com/gravitl/netmaker/database"
  7. "github.com/gravitl/netmaker/logger"
  8. "github.com/gravitl/netmaker/logic"
  9. "github.com/gravitl/netmaker/models"
  10. "github.com/gravitl/netmaker/mq"
  11. "github.com/gravitl/netmaker/servercfg"
  12. "golang.org/x/exp/slog"
  13. )
  14. var ServiceUserPermissionTemplate = models.UserRolePermissionTemplate{
  15. ID: models.ServiceUser,
  16. Default: true,
  17. FullAccess: false,
  18. DenyDashboardAccess: true,
  19. }
  20. var PlatformUserUserPermissionTemplate = models.UserRolePermissionTemplate{
  21. ID: models.PlatformUser,
  22. Default: true,
  23. FullAccess: false,
  24. }
  25. var NetworkAdminAllPermissionTemplate = models.UserRolePermissionTemplate{
  26. ID: models.UserRoleID(fmt.Sprintf("global-%s", models.NetworkAdmin)),
  27. Default: true,
  28. FullAccess: true,
  29. NetworkID: models.AllNetworks,
  30. }
  31. var NetworkUserAllPermissionTemplate = models.UserRolePermissionTemplate{
  32. ID: models.UserRoleID(fmt.Sprintf("global-%s", models.NetworkUser)),
  33. Default: true,
  34. FullAccess: false,
  35. NetworkID: models.AllNetworks,
  36. NetworkLevelAccess: map[models.RsrcType]map[models.RsrcID]models.RsrcPermissionScope{
  37. models.RemoteAccessGwRsrc: {
  38. models.AllRemoteAccessGwRsrcID: models.RsrcPermissionScope{
  39. Read: true,
  40. VPNaccess: true,
  41. },
  42. },
  43. models.ExtClientsRsrc: {
  44. models.AllExtClientsRsrcID: models.RsrcPermissionScope{
  45. Read: true,
  46. Create: true,
  47. Update: true,
  48. Delete: true,
  49. SelfOnly: true,
  50. },
  51. },
  52. },
  53. }
  54. func UserRolesInit() {
  55. d, _ := json.Marshal(logic.SuperAdminPermissionTemplate)
  56. database.Insert(logic.SuperAdminPermissionTemplate.ID.String(), string(d), database.USER_PERMISSIONS_TABLE_NAME)
  57. d, _ = json.Marshal(logic.AdminPermissionTemplate)
  58. database.Insert(logic.AdminPermissionTemplate.ID.String(), string(d), database.USER_PERMISSIONS_TABLE_NAME)
  59. d, _ = json.Marshal(ServiceUserPermissionTemplate)
  60. database.Insert(ServiceUserPermissionTemplate.ID.String(), string(d), database.USER_PERMISSIONS_TABLE_NAME)
  61. d, _ = json.Marshal(PlatformUserUserPermissionTemplate)
  62. database.Insert(PlatformUserUserPermissionTemplate.ID.String(), string(d), database.USER_PERMISSIONS_TABLE_NAME)
  63. d, _ = json.Marshal(NetworkAdminAllPermissionTemplate)
  64. database.Insert(NetworkAdminAllPermissionTemplate.ID.String(), string(d), database.USER_PERMISSIONS_TABLE_NAME)
  65. d, _ = json.Marshal(NetworkUserAllPermissionTemplate)
  66. database.Insert(NetworkUserAllPermissionTemplate.ID.String(), string(d), database.USER_PERMISSIONS_TABLE_NAME)
  67. }
  68. func CreateDefaultNetworkRolesAndGroups(netID models.NetworkID) {
  69. if netID.String() == "" {
  70. return
  71. }
  72. var NetworkAdminPermissionTemplate = models.UserRolePermissionTemplate{
  73. ID: models.UserRoleID(fmt.Sprintf("%s-%s", netID, models.NetworkAdmin)),
  74. Default: true,
  75. NetworkID: netID,
  76. FullAccess: true,
  77. NetworkLevelAccess: make(map[models.RsrcType]map[models.RsrcID]models.RsrcPermissionScope),
  78. }
  79. var NetworkUserPermissionTemplate = models.UserRolePermissionTemplate{
  80. ID: models.UserRoleID(fmt.Sprintf("%s-%s", netID, models.NetworkUser)),
  81. Default: true,
  82. FullAccess: false,
  83. NetworkID: netID,
  84. DenyDashboardAccess: false,
  85. NetworkLevelAccess: map[models.RsrcType]map[models.RsrcID]models.RsrcPermissionScope{
  86. models.RemoteAccessGwRsrc: {
  87. models.AllRemoteAccessGwRsrcID: models.RsrcPermissionScope{
  88. Read: true,
  89. VPNaccess: true,
  90. },
  91. },
  92. models.ExtClientsRsrc: {
  93. models.AllExtClientsRsrcID: models.RsrcPermissionScope{
  94. Read: true,
  95. Create: true,
  96. Update: true,
  97. Delete: true,
  98. SelfOnly: true,
  99. },
  100. },
  101. },
  102. }
  103. d, _ := json.Marshal(NetworkAdminPermissionTemplate)
  104. database.Insert(NetworkAdminPermissionTemplate.ID.String(), string(d), database.USER_PERMISSIONS_TABLE_NAME)
  105. d, _ = json.Marshal(NetworkUserPermissionTemplate)
  106. database.Insert(NetworkUserPermissionTemplate.ID.String(), string(d), database.USER_PERMISSIONS_TABLE_NAME)
  107. // create default network groups
  108. var NetworkAdminGroup = models.UserGroup{
  109. ID: models.UserGroupID(fmt.Sprintf("%s-%s-grp", netID, models.NetworkAdmin)),
  110. NetworkRoles: map[models.NetworkID]map[models.UserRoleID]struct{}{
  111. netID: {
  112. models.UserRoleID(fmt.Sprintf("%s-%s", netID, models.NetworkAdmin)): {},
  113. },
  114. },
  115. MetaData: "The network group was automatically created by Netmaker.",
  116. }
  117. var NetworkUserGroup = models.UserGroup{
  118. ID: models.UserGroupID(fmt.Sprintf("%s-%s-grp", netID, models.NetworkUser)),
  119. NetworkRoles: map[models.NetworkID]map[models.UserRoleID]struct{}{
  120. netID: {
  121. models.UserRoleID(fmt.Sprintf("%s-%s", netID, models.NetworkUser)): {},
  122. },
  123. },
  124. MetaData: "The network group was automatically created by Netmaker.",
  125. }
  126. d, _ = json.Marshal(NetworkAdminGroup)
  127. database.Insert(NetworkAdminGroup.ID.String(), string(d), database.USER_GROUPS_TABLE_NAME)
  128. d, _ = json.Marshal(NetworkUserGroup)
  129. database.Insert(NetworkUserGroup.ID.String(), string(d), database.USER_GROUPS_TABLE_NAME)
  130. }
  131. func DeleteNetworkRoles(netID string) {
  132. users, err := logic.GetUsersDB()
  133. if err != nil {
  134. return
  135. }
  136. defaultUserGrp := fmt.Sprintf("%s-%s-grp", netID, models.NetworkUser)
  137. defaultAdminGrp := fmt.Sprintf("%s-%s-grp", netID, models.NetworkAdmin)
  138. for _, user := range users {
  139. var upsert bool
  140. if _, ok := user.NetworkRoles[models.NetworkID(netID)]; ok {
  141. delete(user.NetworkRoles, models.NetworkID(netID))
  142. upsert = true
  143. }
  144. if _, ok := user.UserGroups[models.UserGroupID(defaultUserGrp)]; ok {
  145. delete(user.UserGroups, models.UserGroupID(defaultUserGrp))
  146. upsert = true
  147. }
  148. if _, ok := user.UserGroups[models.UserGroupID(defaultAdminGrp)]; ok {
  149. delete(user.UserGroups, models.UserGroupID(defaultAdminGrp))
  150. upsert = true
  151. }
  152. if upsert {
  153. logic.UpsertUser(user)
  154. }
  155. }
  156. database.DeleteRecord(database.USER_GROUPS_TABLE_NAME, defaultUserGrp)
  157. database.DeleteRecord(database.USER_GROUPS_TABLE_NAME, defaultAdminGrp)
  158. userGs, _ := ListUserGroups()
  159. for _, userGI := range userGs {
  160. if _, ok := userGI.NetworkRoles[models.NetworkID(netID)]; ok {
  161. delete(userGI.NetworkRoles, models.NetworkID(netID))
  162. UpdateUserGroup(userGI)
  163. }
  164. }
  165. roles, _ := ListNetworkRoles()
  166. for _, role := range roles {
  167. if role.NetworkID.String() == netID {
  168. database.DeleteRecord(database.USER_PERMISSIONS_TABLE_NAME, role.ID.String())
  169. }
  170. }
  171. }
  172. // ListNetworkRoles - lists user network roles permission templates
  173. func ListNetworkRoles() ([]models.UserRolePermissionTemplate, error) {
  174. data, err := database.FetchRecords(database.USER_PERMISSIONS_TABLE_NAME)
  175. if err != nil && !database.IsEmptyRecord(err) {
  176. return []models.UserRolePermissionTemplate{}, err
  177. }
  178. userRoles := []models.UserRolePermissionTemplate{}
  179. for _, dataI := range data {
  180. userRole := models.UserRolePermissionTemplate{}
  181. err := json.Unmarshal([]byte(dataI), &userRole)
  182. if err != nil {
  183. continue
  184. }
  185. if userRole.NetworkID == "" {
  186. continue
  187. }
  188. userRoles = append(userRoles, userRole)
  189. }
  190. return userRoles, nil
  191. }
  192. func ValidateCreateRoleReq(userRole *models.UserRolePermissionTemplate) error {
  193. // check if role exists with this id
  194. _, err := logic.GetRole(userRole.ID)
  195. if err == nil {
  196. return fmt.Errorf("role with id `%s` exists already", userRole.ID.String())
  197. }
  198. if len(userRole.NetworkLevelAccess) > 0 {
  199. for rsrcType := range userRole.NetworkLevelAccess {
  200. if _, ok := models.RsrcTypeMap[rsrcType]; !ok {
  201. return errors.New("invalid rsrc type " + rsrcType.String())
  202. }
  203. if rsrcType == models.RemoteAccessGwRsrc {
  204. userRsrcPermissions := userRole.NetworkLevelAccess[models.RemoteAccessGwRsrc]
  205. var vpnAccess bool
  206. for _, scope := range userRsrcPermissions {
  207. if scope.VPNaccess {
  208. vpnAccess = true
  209. break
  210. }
  211. }
  212. if vpnAccess {
  213. userRole.NetworkLevelAccess[models.ExtClientsRsrc] = map[models.RsrcID]models.RsrcPermissionScope{
  214. models.AllExtClientsRsrcID: {
  215. Read: true,
  216. Create: true,
  217. Update: true,
  218. Delete: true,
  219. SelfOnly: true,
  220. },
  221. }
  222. }
  223. }
  224. }
  225. }
  226. if userRole.NetworkID == "" {
  227. return errors.New("only network roles are allowed to be created")
  228. }
  229. return nil
  230. }
  231. func ValidateUpdateRoleReq(userRole *models.UserRolePermissionTemplate) error {
  232. roleInDB, err := logic.GetRole(userRole.ID)
  233. if err != nil {
  234. return err
  235. }
  236. if roleInDB.NetworkID != userRole.NetworkID {
  237. return errors.New("network id mismatch")
  238. }
  239. if roleInDB.Default {
  240. return errors.New("cannot update default role")
  241. }
  242. if len(userRole.NetworkLevelAccess) > 0 {
  243. for rsrcType := range userRole.NetworkLevelAccess {
  244. if _, ok := models.RsrcTypeMap[rsrcType]; !ok {
  245. return errors.New("invalid rsrc type " + rsrcType.String())
  246. }
  247. if rsrcType == models.RemoteAccessGwRsrc {
  248. userRsrcPermissions := userRole.NetworkLevelAccess[models.RemoteAccessGwRsrc]
  249. var vpnAccess bool
  250. for _, scope := range userRsrcPermissions {
  251. if scope.VPNaccess {
  252. vpnAccess = true
  253. break
  254. }
  255. }
  256. if vpnAccess {
  257. userRole.NetworkLevelAccess[models.ExtClientsRsrc] = map[models.RsrcID]models.RsrcPermissionScope{
  258. models.AllExtClientsRsrcID: {
  259. Read: true,
  260. Create: true,
  261. Update: true,
  262. Delete: true,
  263. SelfOnly: true,
  264. },
  265. }
  266. }
  267. }
  268. }
  269. }
  270. return nil
  271. }
  272. // CreateRole - inserts new role into DB
  273. func CreateRole(r models.UserRolePermissionTemplate) error {
  274. // check if role already exists
  275. if r.ID.String() == "" {
  276. return errors.New("role id cannot be empty")
  277. }
  278. _, err := database.FetchRecord(database.USER_PERMISSIONS_TABLE_NAME, r.ID.String())
  279. if err == nil {
  280. return errors.New("role already exists")
  281. }
  282. d, err := json.Marshal(r)
  283. if err != nil {
  284. return err
  285. }
  286. return database.Insert(r.ID.String(), string(d), database.USER_PERMISSIONS_TABLE_NAME)
  287. }
  288. // UpdateRole - updates role template
  289. func UpdateRole(r models.UserRolePermissionTemplate) error {
  290. if r.ID.String() == "" {
  291. return errors.New("role id cannot be empty")
  292. }
  293. _, err := database.FetchRecord(database.USER_PERMISSIONS_TABLE_NAME, r.ID.String())
  294. if err != nil {
  295. return err
  296. }
  297. d, err := json.Marshal(r)
  298. if err != nil {
  299. return err
  300. }
  301. return database.Insert(r.ID.String(), string(d), database.USER_PERMISSIONS_TABLE_NAME)
  302. }
  303. // DeleteRole - deletes user role
  304. func DeleteRole(rid models.UserRoleID, force bool) error {
  305. if rid.String() == "" {
  306. return errors.New("role id cannot be empty")
  307. }
  308. users, err := logic.GetUsersDB()
  309. if err != nil {
  310. return err
  311. }
  312. role, err := logic.GetRole(rid)
  313. if err != nil {
  314. return err
  315. }
  316. if role.NetworkID == "" {
  317. return errors.New("cannot delete platform role")
  318. }
  319. // allow deletion of default network roles if network doesn't exist
  320. if role.NetworkID == models.AllNetworks {
  321. return errors.New("cannot delete default network role")
  322. }
  323. // check if network exists
  324. exists, _ := logic.NetworkExists(role.NetworkID.String())
  325. if role.Default {
  326. if exists && !force {
  327. return errors.New("cannot delete default role")
  328. }
  329. }
  330. for _, user := range users {
  331. for userG := range user.UserGroups {
  332. ug, err := GetUserGroup(userG)
  333. if err == nil {
  334. if role.NetworkID != "" {
  335. for netID, networkRoles := range ug.NetworkRoles {
  336. if _, ok := networkRoles[rid]; ok {
  337. delete(networkRoles, rid)
  338. ug.NetworkRoles[netID] = networkRoles
  339. UpdateUserGroup(ug)
  340. }
  341. }
  342. }
  343. }
  344. }
  345. if user.PlatformRoleID == rid {
  346. err = errors.New("active roles cannot be deleted.switch existing users to a new role before deleting")
  347. return err
  348. }
  349. if role.NetworkID != "" {
  350. for netID, networkRoles := range user.NetworkRoles {
  351. if _, ok := networkRoles[rid]; ok {
  352. delete(networkRoles, rid)
  353. user.NetworkRoles[netID] = networkRoles
  354. logic.UpsertUser(user)
  355. }
  356. }
  357. }
  358. }
  359. return database.DeleteRecord(database.USER_PERMISSIONS_TABLE_NAME, rid.String())
  360. }
  361. func ValidateCreateGroupReq(g models.UserGroup) error {
  362. // check if network roles are valid
  363. for _, roleMap := range g.NetworkRoles {
  364. for roleID := range roleMap {
  365. role, err := logic.GetRole(roleID)
  366. if err != nil {
  367. return fmt.Errorf("invalid network role %s", roleID)
  368. }
  369. if role.NetworkID == "" {
  370. return errors.New("platform role cannot be used as network role")
  371. }
  372. }
  373. }
  374. return nil
  375. }
  376. func ValidateUpdateGroupReq(g models.UserGroup) error {
  377. for networkID := range g.NetworkRoles {
  378. userRolesMap := g.NetworkRoles[networkID]
  379. for roleID := range userRolesMap {
  380. netRole, err := logic.GetRole(roleID)
  381. if err != nil {
  382. err = fmt.Errorf("invalid network role")
  383. return err
  384. }
  385. if netRole.NetworkID == "" {
  386. return errors.New("platform role cannot be used as network role")
  387. }
  388. }
  389. }
  390. return nil
  391. }
  392. // CreateUserGroup - creates new user group
  393. func CreateUserGroup(g models.UserGroup) error {
  394. // check if role already exists
  395. if g.ID == "" {
  396. return errors.New("group id cannot be empty")
  397. }
  398. _, err := database.FetchRecord(database.USER_GROUPS_TABLE_NAME, g.ID.String())
  399. if err == nil {
  400. return errors.New("group already exists")
  401. }
  402. d, err := json.Marshal(g)
  403. if err != nil {
  404. return err
  405. }
  406. return database.Insert(g.ID.String(), string(d), database.USER_GROUPS_TABLE_NAME)
  407. }
  408. // GetUserGroup - fetches user group
  409. func GetUserGroup(gid models.UserGroupID) (models.UserGroup, error) {
  410. d, err := database.FetchRecord(database.USER_GROUPS_TABLE_NAME, gid.String())
  411. if err != nil {
  412. return models.UserGroup{}, err
  413. }
  414. var ug models.UserGroup
  415. err = json.Unmarshal([]byte(d), &ug)
  416. if err != nil {
  417. return ug, err
  418. }
  419. return ug, nil
  420. }
  421. // ListUserGroups - lists user groups
  422. func ListUserGroups() ([]models.UserGroup, error) {
  423. data, err := database.FetchRecords(database.USER_GROUPS_TABLE_NAME)
  424. if err != nil && !database.IsEmptyRecord(err) {
  425. return []models.UserGroup{}, err
  426. }
  427. userGroups := []models.UserGroup{}
  428. for _, dataI := range data {
  429. userGroup := models.UserGroup{}
  430. err := json.Unmarshal([]byte(dataI), &userGroup)
  431. if err != nil {
  432. continue
  433. }
  434. userGroups = append(userGroups, userGroup)
  435. }
  436. return userGroups, nil
  437. }
  438. // UpdateUserGroup - updates new user group
  439. func UpdateUserGroup(g models.UserGroup) error {
  440. // check if group exists
  441. if g.ID == "" {
  442. return errors.New("group id cannot be empty")
  443. }
  444. _, err := database.FetchRecord(database.USER_GROUPS_TABLE_NAME, g.ID.String())
  445. if err != nil {
  446. return err
  447. }
  448. d, err := json.Marshal(g)
  449. if err != nil {
  450. return err
  451. }
  452. return database.Insert(g.ID.String(), string(d), database.USER_GROUPS_TABLE_NAME)
  453. }
  454. // DeleteUserGroup - deletes user group
  455. func DeleteUserGroup(gid models.UserGroupID) error {
  456. users, err := logic.GetUsersDB()
  457. if err != nil {
  458. return err
  459. }
  460. for _, user := range users {
  461. delete(user.UserGroups, gid)
  462. logic.UpsertUser(user)
  463. }
  464. return database.DeleteRecord(database.USER_GROUPS_TABLE_NAME, gid.String())
  465. }
  466. func HasNetworkRsrcScope(permissionTemplate models.UserRolePermissionTemplate, netid string, rsrcType models.RsrcType, rsrcID models.RsrcID, op string) bool {
  467. if permissionTemplate.FullAccess {
  468. return true
  469. }
  470. rsrcScope, ok := permissionTemplate.NetworkLevelAccess[rsrcType]
  471. if !ok {
  472. return false
  473. }
  474. _, ok = rsrcScope[rsrcID]
  475. return ok
  476. }
  477. func GetUserRAGNodesV1(user models.User) (gws map[string]models.Node) {
  478. gws = make(map[string]models.Node)
  479. tagNodesMap := logic.GetTagMapWithNodes()
  480. accessPolices := logic.ListUserPolicies(user)
  481. for _, policyI := range accessPolices {
  482. for _, dstI := range policyI.Dst {
  483. if nodes, ok := tagNodesMap[models.TagID(dstI.Value)]; ok {
  484. for _, node := range nodes {
  485. gws[node.ID.String()] = node
  486. }
  487. }
  488. }
  489. }
  490. return
  491. }
  492. func DoesUserHaveAccessToRAGNode(user models.User, node models.Node) bool {
  493. userGwAccessScope := GetUserNetworkRolesWithRemoteVPNAccess(user)
  494. logger.Log(3, fmt.Sprintf("User Gw Access Scope: %+v", userGwAccessScope))
  495. _, allNetAccess := userGwAccessScope["*"]
  496. if node.IsIngressGateway && !node.PendingDelete {
  497. if allNetAccess {
  498. return true
  499. } else {
  500. gwRsrcMap := userGwAccessScope[models.NetworkID(node.Network)]
  501. scope, ok := gwRsrcMap[models.AllRemoteAccessGwRsrcID]
  502. if !ok {
  503. if scope, ok = gwRsrcMap[models.RsrcID(node.ID.String())]; !ok {
  504. return false
  505. }
  506. }
  507. if scope.VPNaccess {
  508. return true
  509. }
  510. }
  511. }
  512. return false
  513. }
  514. func GetUserRAGNodes(user models.User) (gws map[string]models.Node) {
  515. gws = make(map[string]models.Node)
  516. userGwAccessScope := GetUserNetworkRolesWithRemoteVPNAccess(user)
  517. logger.Log(3, fmt.Sprintf("User Gw Access Scope: %+v", userGwAccessScope))
  518. _, allNetAccess := userGwAccessScope["*"]
  519. nodes, err := logic.GetAllNodes()
  520. if err != nil {
  521. return
  522. }
  523. for _, node := range nodes {
  524. if node.IsIngressGateway && !node.PendingDelete {
  525. if allNetAccess {
  526. gws[node.ID.String()] = node
  527. } else {
  528. gwRsrcMap := userGwAccessScope[models.NetworkID(node.Network)]
  529. scope, ok := gwRsrcMap[models.AllRemoteAccessGwRsrcID]
  530. if !ok {
  531. if scope, ok = gwRsrcMap[models.RsrcID(node.ID.String())]; !ok {
  532. continue
  533. }
  534. }
  535. if scope.VPNaccess {
  536. gws[node.ID.String()] = node
  537. }
  538. }
  539. }
  540. }
  541. return
  542. }
  543. // GetUserNetworkRoles - get user network roles
  544. func GetUserNetworkRolesWithRemoteVPNAccess(user models.User) (gwAccess map[models.NetworkID]map[models.RsrcID]models.RsrcPermissionScope) {
  545. gwAccess = make(map[models.NetworkID]map[models.RsrcID]models.RsrcPermissionScope)
  546. platformRole, err := logic.GetRole(user.PlatformRoleID)
  547. if err != nil {
  548. return
  549. }
  550. if platformRole.FullAccess {
  551. gwAccess[models.NetworkID("*")] = make(map[models.RsrcID]models.RsrcPermissionScope)
  552. return
  553. }
  554. if _, ok := user.NetworkRoles[models.AllNetworks]; ok {
  555. gwAccess[models.NetworkID("*")] = make(map[models.RsrcID]models.RsrcPermissionScope)
  556. }
  557. if len(user.UserGroups) > 0 {
  558. for gID := range user.UserGroups {
  559. userG, err := GetUserGroup(gID)
  560. if err != nil {
  561. continue
  562. }
  563. for netID, roleMap := range userG.NetworkRoles {
  564. for roleID := range roleMap {
  565. role, err := logic.GetRole(roleID)
  566. if err == nil {
  567. if role.FullAccess {
  568. gwAccess[netID] = map[models.RsrcID]models.RsrcPermissionScope{
  569. models.AllRemoteAccessGwRsrcID: {
  570. Create: true,
  571. Read: true,
  572. Update: true,
  573. VPNaccess: true,
  574. Delete: true,
  575. },
  576. models.AllExtClientsRsrcID: {
  577. Create: true,
  578. Read: true,
  579. Update: true,
  580. Delete: true,
  581. },
  582. }
  583. break
  584. }
  585. if rsrcsMap, ok := role.NetworkLevelAccess[models.RemoteAccessGwRsrc]; ok {
  586. if permissions, ok := rsrcsMap[models.AllRemoteAccessGwRsrcID]; ok && permissions.VPNaccess {
  587. if len(gwAccess[netID]) == 0 {
  588. gwAccess[netID] = make(map[models.RsrcID]models.RsrcPermissionScope)
  589. }
  590. gwAccess[netID][models.AllRemoteAccessGwRsrcID] = permissions
  591. break
  592. } else {
  593. for gwID, scope := range rsrcsMap {
  594. if scope.VPNaccess {
  595. if len(gwAccess[netID]) == 0 {
  596. gwAccess[netID] = make(map[models.RsrcID]models.RsrcPermissionScope)
  597. }
  598. gwAccess[netID][gwID] = scope
  599. }
  600. }
  601. }
  602. }
  603. }
  604. }
  605. }
  606. }
  607. }
  608. for netID, roleMap := range user.NetworkRoles {
  609. for roleID := range roleMap {
  610. role, err := logic.GetRole(roleID)
  611. if err == nil {
  612. if role.FullAccess {
  613. gwAccess[netID] = map[models.RsrcID]models.RsrcPermissionScope{
  614. models.AllRemoteAccessGwRsrcID: {
  615. Create: true,
  616. Read: true,
  617. Update: true,
  618. VPNaccess: true,
  619. Delete: true,
  620. },
  621. models.AllExtClientsRsrcID: {
  622. Create: true,
  623. Read: true,
  624. Update: true,
  625. Delete: true,
  626. },
  627. }
  628. break
  629. }
  630. if rsrcsMap, ok := role.NetworkLevelAccess[models.RemoteAccessGwRsrc]; ok {
  631. if permissions, ok := rsrcsMap[models.AllRemoteAccessGwRsrcID]; ok && permissions.VPNaccess {
  632. if len(gwAccess[netID]) == 0 {
  633. gwAccess[netID] = make(map[models.RsrcID]models.RsrcPermissionScope)
  634. }
  635. gwAccess[netID][models.AllRemoteAccessGwRsrcID] = permissions
  636. break
  637. } else {
  638. for gwID, scope := range rsrcsMap {
  639. if scope.VPNaccess {
  640. if len(gwAccess[netID]) == 0 {
  641. gwAccess[netID] = make(map[models.RsrcID]models.RsrcPermissionScope)
  642. }
  643. gwAccess[netID][gwID] = scope
  644. }
  645. }
  646. }
  647. }
  648. }
  649. }
  650. }
  651. return
  652. }
  653. func GetFilteredNodesByUserAccess(user models.User, nodes []models.Node) (filteredNodes []models.Node) {
  654. nodesMap := make(map[string]struct{})
  655. allNetworkRoles := make(map[models.UserRoleID]struct{})
  656. defer func() {
  657. filteredNodes = logic.AddStaticNodestoList(filteredNodes)
  658. }()
  659. if len(user.NetworkRoles) > 0 {
  660. for _, netRoles := range user.NetworkRoles {
  661. for netRoleI := range netRoles {
  662. allNetworkRoles[netRoleI] = struct{}{}
  663. }
  664. }
  665. }
  666. if _, ok := user.NetworkRoles[models.AllNetworks]; ok {
  667. filteredNodes = nodes
  668. return
  669. }
  670. if len(user.UserGroups) > 0 {
  671. for userGID := range user.UserGroups {
  672. userG, err := GetUserGroup(userGID)
  673. if err == nil {
  674. if len(userG.NetworkRoles) > 0 {
  675. if _, ok := userG.NetworkRoles[models.AllNetworks]; ok {
  676. filteredNodes = nodes
  677. return
  678. }
  679. for _, netRoles := range userG.NetworkRoles {
  680. for netRoleI := range netRoles {
  681. allNetworkRoles[netRoleI] = struct{}{}
  682. }
  683. }
  684. }
  685. }
  686. }
  687. }
  688. for networkRoleID := range allNetworkRoles {
  689. userPermTemplate, err := logic.GetRole(networkRoleID)
  690. if err != nil {
  691. continue
  692. }
  693. networkNodes := logic.GetNetworkNodesMemory(nodes, userPermTemplate.NetworkID.String())
  694. if userPermTemplate.FullAccess {
  695. for _, node := range networkNodes {
  696. if _, ok := nodesMap[node.ID.String()]; ok {
  697. continue
  698. }
  699. nodesMap[node.ID.String()] = struct{}{}
  700. filteredNodes = append(filteredNodes, node)
  701. }
  702. continue
  703. }
  704. if rsrcPerms, ok := userPermTemplate.NetworkLevelAccess[models.RemoteAccessGwRsrc]; ok {
  705. if _, ok := rsrcPerms[models.AllRemoteAccessGwRsrcID]; ok {
  706. for _, node := range networkNodes {
  707. if _, ok := nodesMap[node.ID.String()]; ok {
  708. continue
  709. }
  710. if node.IsIngressGateway {
  711. nodesMap[node.ID.String()] = struct{}{}
  712. filteredNodes = append(filteredNodes, node)
  713. }
  714. }
  715. } else {
  716. for gwID, scope := range rsrcPerms {
  717. if _, ok := nodesMap[gwID.String()]; ok {
  718. continue
  719. }
  720. if scope.Read {
  721. gwNode, err := logic.GetNodeByID(gwID.String())
  722. if err == nil && gwNode.IsIngressGateway {
  723. nodesMap[gwNode.ID.String()] = struct{}{}
  724. filteredNodes = append(filteredNodes, gwNode)
  725. }
  726. }
  727. }
  728. }
  729. }
  730. }
  731. return
  732. }
  733. func FilterNetworksByRole(allnetworks []models.Network, user models.User) []models.Network {
  734. platformRole, err := logic.GetRole(user.PlatformRoleID)
  735. if err != nil {
  736. return []models.Network{}
  737. }
  738. if !platformRole.FullAccess {
  739. allNetworkRoles := make(map[models.NetworkID]struct{})
  740. if len(user.NetworkRoles) > 0 {
  741. for netID := range user.NetworkRoles {
  742. if netID == models.AllNetworks {
  743. return allnetworks
  744. }
  745. allNetworkRoles[netID] = struct{}{}
  746. }
  747. }
  748. if len(user.UserGroups) > 0 {
  749. for userGID := range user.UserGroups {
  750. userG, err := GetUserGroup(userGID)
  751. if err == nil {
  752. if len(userG.NetworkRoles) > 0 {
  753. for netID := range userG.NetworkRoles {
  754. if netID == models.AllNetworks {
  755. return allnetworks
  756. }
  757. allNetworkRoles[netID] = struct{}{}
  758. }
  759. }
  760. }
  761. }
  762. }
  763. filteredNetworks := []models.Network{}
  764. for _, networkI := range allnetworks {
  765. if _, ok := allNetworkRoles[models.NetworkID(networkI.NetID)]; ok {
  766. filteredNetworks = append(filteredNetworks, networkI)
  767. }
  768. }
  769. allnetworks = filteredNetworks
  770. }
  771. return allnetworks
  772. }
  773. func IsGroupsValid(groups map[models.UserGroupID]struct{}) error {
  774. for groupID := range groups {
  775. _, err := GetUserGroup(groupID)
  776. if err != nil {
  777. return fmt.Errorf("user group `%s` not found", groupID)
  778. }
  779. }
  780. return nil
  781. }
  782. func IsGroupValid(groupID models.UserGroupID) error {
  783. _, err := GetUserGroup(groupID)
  784. if err != nil {
  785. return fmt.Errorf("user group `%s` not found", groupID)
  786. }
  787. return nil
  788. }
  789. func IsNetworkRolesValid(networkRoles map[models.NetworkID]map[models.UserRoleID]struct{}) error {
  790. for netID, netRoles := range networkRoles {
  791. if netID != models.AllNetworks {
  792. _, err := logic.GetNetwork(netID.String())
  793. if err != nil {
  794. return fmt.Errorf("failed to fetch network %s ", netID)
  795. }
  796. }
  797. for netRoleID := range netRoles {
  798. role, err := logic.GetRole(netRoleID)
  799. if err != nil {
  800. return fmt.Errorf("failed to fetch role %s ", netRoleID)
  801. }
  802. if role.NetworkID == "" {
  803. return fmt.Errorf("cannot use platform as network role %s", netRoleID)
  804. }
  805. }
  806. }
  807. return nil
  808. }
  809. // PrepareOauthUserFromInvite - init oauth user before create
  810. func PrepareOauthUserFromInvite(in models.UserInvite) (models.User, error) {
  811. var newPass, fetchErr = logic.FetchPassValue("")
  812. if fetchErr != nil {
  813. return models.User{}, fetchErr
  814. }
  815. user := models.User{
  816. UserName: in.Email,
  817. Password: newPass,
  818. }
  819. user.UserGroups = in.UserGroups
  820. user.NetworkRoles = in.NetworkRoles
  821. user.PlatformRoleID = models.UserRoleID(in.PlatformRoleID)
  822. if user.PlatformRoleID == "" {
  823. user.PlatformRoleID = models.ServiceUser
  824. }
  825. return user, nil
  826. }
  827. func UpdatesUserGwAccessOnRoleUpdates(currNetworkAccess,
  828. changeNetworkAccess map[models.RsrcType]map[models.RsrcID]models.RsrcPermissionScope, netID string) {
  829. networkChangeMap := make(map[models.RsrcID]models.RsrcPermissionScope)
  830. for rsrcType, RsrcPermsMap := range currNetworkAccess {
  831. if rsrcType != models.RemoteAccessGwRsrc {
  832. continue
  833. }
  834. if _, ok := changeNetworkAccess[rsrcType]; !ok {
  835. for rsrcID, scope := range RsrcPermsMap {
  836. networkChangeMap[rsrcID] = scope
  837. }
  838. } else {
  839. for rsrcID, scope := range RsrcPermsMap {
  840. if _, ok := changeNetworkAccess[rsrcType][rsrcID]; !ok {
  841. networkChangeMap[rsrcID] = scope
  842. }
  843. }
  844. }
  845. }
  846. extclients, err := logic.GetAllExtClients()
  847. if err != nil {
  848. slog.Error("failed to fetch extclients", "error", err)
  849. return
  850. }
  851. userMap, err := logic.GetUserMap()
  852. if err != nil {
  853. return
  854. }
  855. for _, extclient := range extclients {
  856. if extclient.Network != netID {
  857. continue
  858. }
  859. if _, ok := networkChangeMap[models.AllRemoteAccessGwRsrcID]; ok {
  860. if user, ok := userMap[extclient.OwnerID]; ok {
  861. if user.PlatformRoleID != models.ServiceUser {
  862. continue
  863. }
  864. err = logic.DeleteExtClientAndCleanup(extclient)
  865. if err != nil {
  866. slog.Error("failed to delete extclient",
  867. "id", extclient.ClientID, "owner", user.UserName, "error", err)
  868. } else {
  869. if err := mq.PublishDeletedClientPeerUpdate(&extclient); err != nil {
  870. slog.Error("error setting ext peers: " + err.Error())
  871. }
  872. }
  873. }
  874. continue
  875. }
  876. if _, ok := networkChangeMap[models.RsrcID(extclient.IngressGatewayID)]; ok {
  877. if user, ok := userMap[extclient.OwnerID]; ok {
  878. if user.PlatformRoleID != models.ServiceUser {
  879. continue
  880. }
  881. err = logic.DeleteExtClientAndCleanup(extclient)
  882. if err != nil {
  883. slog.Error("failed to delete extclient",
  884. "id", extclient.ClientID, "owner", user.UserName, "error", err)
  885. } else {
  886. if err := mq.PublishDeletedClientPeerUpdate(&extclient); err != nil {
  887. slog.Error("error setting ext peers: " + err.Error())
  888. }
  889. }
  890. }
  891. }
  892. }
  893. if servercfg.IsDNSMode() {
  894. logic.SetDNS()
  895. }
  896. }
  897. func UpdatesUserGwAccessOnGrpUpdates(currNetworkRoles, changeNetworkRoles map[models.NetworkID]map[models.UserRoleID]struct{}) {
  898. networkChangeMap := make(map[models.NetworkID]map[models.UserRoleID]struct{})
  899. for netID, networkUserRoles := range currNetworkRoles {
  900. if _, ok := changeNetworkRoles[netID]; !ok {
  901. for netRoleID := range networkUserRoles {
  902. if _, ok := networkChangeMap[netID]; !ok {
  903. networkChangeMap[netID] = make(map[models.UserRoleID]struct{})
  904. }
  905. networkChangeMap[netID][netRoleID] = struct{}{}
  906. }
  907. } else {
  908. for netRoleID := range networkUserRoles {
  909. if _, ok := changeNetworkRoles[netID][netRoleID]; !ok {
  910. if _, ok := networkChangeMap[netID]; !ok {
  911. networkChangeMap[netID] = make(map[models.UserRoleID]struct{})
  912. }
  913. networkChangeMap[netID][netRoleID] = struct{}{}
  914. }
  915. }
  916. }
  917. }
  918. extclients, err := logic.GetAllExtClients()
  919. if err != nil {
  920. slog.Error("failed to fetch extclients", "error", err)
  921. return
  922. }
  923. userMap, err := logic.GetUserMap()
  924. if err != nil {
  925. return
  926. }
  927. for _, extclient := range extclients {
  928. if _, ok := networkChangeMap[models.NetworkID(extclient.Network)]; ok {
  929. if user, ok := userMap[extclient.OwnerID]; ok {
  930. if user.PlatformRoleID != models.ServiceUser {
  931. continue
  932. }
  933. err = logic.DeleteExtClientAndCleanup(extclient)
  934. if err != nil {
  935. slog.Error("failed to delete extclient",
  936. "id", extclient.ClientID, "owner", user.UserName, "error", err)
  937. } else {
  938. if err := mq.PublishDeletedClientPeerUpdate(&extclient); err != nil {
  939. slog.Error("error setting ext peers: " + err.Error())
  940. }
  941. }
  942. }
  943. }
  944. }
  945. if servercfg.IsDNSMode() {
  946. logic.SetDNS()
  947. }
  948. }
  949. func UpdateUserGwAccess(currentUser, changeUser models.User) {
  950. if changeUser.PlatformRoleID != models.ServiceUser {
  951. return
  952. }
  953. networkChangeMap := make(map[models.NetworkID]map[models.UserRoleID]struct{})
  954. for netID, networkUserRoles := range currentUser.NetworkRoles {
  955. if _, ok := changeUser.NetworkRoles[netID]; !ok {
  956. for netRoleID := range networkUserRoles {
  957. if _, ok := networkChangeMap[netID]; !ok {
  958. networkChangeMap[netID] = make(map[models.UserRoleID]struct{})
  959. }
  960. networkChangeMap[netID][netRoleID] = struct{}{}
  961. }
  962. } else {
  963. for netRoleID := range networkUserRoles {
  964. if _, ok := changeUser.NetworkRoles[netID][netRoleID]; !ok {
  965. if _, ok := networkChangeMap[netID]; !ok {
  966. networkChangeMap[netID] = make(map[models.UserRoleID]struct{})
  967. }
  968. networkChangeMap[netID][netRoleID] = struct{}{}
  969. }
  970. }
  971. }
  972. }
  973. for gID := range currentUser.UserGroups {
  974. if _, ok := changeUser.UserGroups[gID]; ok {
  975. continue
  976. }
  977. userG, err := GetUserGroup(gID)
  978. if err == nil {
  979. for netID, networkUserRoles := range userG.NetworkRoles {
  980. for netRoleID := range networkUserRoles {
  981. if _, ok := networkChangeMap[netID]; !ok {
  982. networkChangeMap[netID] = make(map[models.UserRoleID]struct{})
  983. }
  984. networkChangeMap[netID][netRoleID] = struct{}{}
  985. }
  986. }
  987. }
  988. }
  989. if len(networkChangeMap) == 0 {
  990. return
  991. }
  992. // TODO - cleanup gw access when role and groups are updated
  993. //removedGwAccess
  994. extclients, err := logic.GetAllExtClients()
  995. if err != nil {
  996. slog.Error("failed to fetch extclients", "error", err)
  997. return
  998. }
  999. for _, extclient := range extclients {
  1000. if extclient.OwnerID == currentUser.UserName {
  1001. if _, ok := networkChangeMap[models.NetworkID(extclient.Network)]; ok {
  1002. err = logic.DeleteExtClientAndCleanup(extclient)
  1003. if err != nil {
  1004. slog.Error("failed to delete extclient",
  1005. "id", extclient.ClientID, "owner", changeUser.UserName, "error", err)
  1006. } else {
  1007. if err := mq.PublishDeletedClientPeerUpdate(&extclient); err != nil {
  1008. slog.Error("error setting ext peers: " + err.Error())
  1009. }
  1010. }
  1011. }
  1012. }
  1013. }
  1014. if servercfg.IsDNSMode() {
  1015. logic.SetDNS()
  1016. }
  1017. }