user_mgmt.go 30 KB

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