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