user_mgmt.go 34 KB

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