user_mgmt.go 29 KB

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