user_mgmt.go 42 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533
  1. package logic
  2. import (
  3. "encoding/json"
  4. "errors"
  5. "fmt"
  6. "time"
  7. "github.com/google/uuid"
  8. "github.com/gravitl/netmaker/database"
  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 (
  16. globalNetworksAdminGroupID = models.UserGroupID(fmt.Sprintf("global-%s-grp", models.NetworkAdmin))
  17. globalNetworksUserGroupID = models.UserGroupID(fmt.Sprintf("global-%s-grp", models.NetworkUser))
  18. globalNetworksAdminRoleID = models.UserRoleID(fmt.Sprintf("global-%s", models.NetworkAdmin))
  19. globalNetworksUserRoleID = models.UserRoleID(fmt.Sprintf("global-%s", models.NetworkUser))
  20. )
  21. var ServiceUserPermissionTemplate = models.UserRolePermissionTemplate{
  22. ID: models.ServiceUser,
  23. Default: true,
  24. FullAccess: false,
  25. DenyDashboardAccess: true,
  26. }
  27. var PlatformUserUserPermissionTemplate = models.UserRolePermissionTemplate{
  28. ID: models.PlatformUser,
  29. Default: true,
  30. FullAccess: false,
  31. GlobalLevelAccess: map[models.RsrcType]map[models.RsrcID]models.RsrcPermissionScope{
  32. models.UserRsrc: {
  33. models.AllUserRsrcID: models.RsrcPermissionScope{
  34. Read: true,
  35. },
  36. },
  37. },
  38. }
  39. var NetworkAdminAllPermissionTemplate = models.UserRolePermissionTemplate{
  40. ID: globalNetworksAdminRoleID,
  41. Name: "Network Admins",
  42. MetaData: "can manage configuration of all networks",
  43. Default: true,
  44. FullAccess: true,
  45. NetworkID: models.AllNetworks,
  46. }
  47. var NetworkUserAllPermissionTemplate = models.UserRolePermissionTemplate{
  48. ID: globalNetworksUserRoleID,
  49. Name: "Network Users",
  50. MetaData: "Can connect to nodes in your networks via Netmaker Desktop App.",
  51. Default: true,
  52. FullAccess: false,
  53. NetworkID: models.AllNetworks,
  54. NetworkLevelAccess: map[models.RsrcType]map[models.RsrcID]models.RsrcPermissionScope{
  55. models.HostRsrc: {
  56. models.AllHostRsrcID: models.RsrcPermissionScope{
  57. Read: true,
  58. },
  59. },
  60. models.RemoteAccessGwRsrc: {
  61. models.AllRemoteAccessGwRsrcID: models.RsrcPermissionScope{
  62. Read: true,
  63. VPNaccess: true,
  64. },
  65. },
  66. models.ExtClientsRsrc: {
  67. models.AllExtClientsRsrcID: models.RsrcPermissionScope{
  68. Read: true,
  69. Create: true,
  70. Update: true,
  71. Delete: true,
  72. SelfOnly: true,
  73. },
  74. },
  75. models.DnsRsrc: {
  76. models.AllDnsRsrcID: models.RsrcPermissionScope{
  77. Read: true,
  78. },
  79. },
  80. models.AclRsrc: {
  81. models.AllAclsRsrcID: models.RsrcPermissionScope{
  82. Read: true,
  83. },
  84. },
  85. models.EgressGwRsrc: {
  86. models.AllEgressGwRsrcID: models.RsrcPermissionScope{
  87. Read: true,
  88. },
  89. },
  90. models.InetGwRsrc: {
  91. models.AllInetGwRsrcID: models.RsrcPermissionScope{
  92. Read: true,
  93. },
  94. },
  95. models.RelayRsrc: {
  96. models.AllRelayRsrcID: models.RsrcPermissionScope{
  97. Read: true,
  98. },
  99. },
  100. models.TagRsrc: {
  101. models.AllTagsRsrcID: models.RsrcPermissionScope{
  102. Read: true,
  103. },
  104. },
  105. },
  106. }
  107. func UserRolesInit() {
  108. d, _ := json.Marshal(logic.SuperAdminPermissionTemplate)
  109. database.Insert(logic.SuperAdminPermissionTemplate.ID.String(), string(d), database.USER_PERMISSIONS_TABLE_NAME)
  110. d, _ = json.Marshal(logic.AdminPermissionTemplate)
  111. database.Insert(logic.AdminPermissionTemplate.ID.String(), string(d), database.USER_PERMISSIONS_TABLE_NAME)
  112. d, _ = json.Marshal(ServiceUserPermissionTemplate)
  113. database.Insert(ServiceUserPermissionTemplate.ID.String(), string(d), database.USER_PERMISSIONS_TABLE_NAME)
  114. d, _ = json.Marshal(PlatformUserUserPermissionTemplate)
  115. database.Insert(PlatformUserUserPermissionTemplate.ID.String(), string(d), database.USER_PERMISSIONS_TABLE_NAME)
  116. d, _ = json.Marshal(NetworkAdminAllPermissionTemplate)
  117. database.Insert(NetworkAdminAllPermissionTemplate.ID.String(), string(d), database.USER_PERMISSIONS_TABLE_NAME)
  118. d, _ = json.Marshal(NetworkUserAllPermissionTemplate)
  119. database.Insert(NetworkUserAllPermissionTemplate.ID.String(), string(d), database.USER_PERMISSIONS_TABLE_NAME)
  120. }
  121. func UserGroupsInit() {
  122. // create default network groups
  123. var NetworkGlobalAdminGroup = models.UserGroup{
  124. ID: globalNetworksAdminGroupID,
  125. Default: true,
  126. Name: "All Networks Admin Group",
  127. MetaData: "can manage configuration of all networks",
  128. NetworkRoles: map[models.NetworkID]map[models.UserRoleID]struct{}{
  129. models.AllNetworks: {
  130. globalNetworksAdminRoleID: {},
  131. },
  132. },
  133. }
  134. var NetworkGlobalUserGroup = models.UserGroup{
  135. ID: globalNetworksUserGroupID,
  136. Name: "All Networks User Group",
  137. Default: true,
  138. NetworkRoles: map[models.NetworkID]map[models.UserRoleID]struct{}{
  139. models.AllNetworks: {
  140. globalNetworksUserRoleID: {},
  141. },
  142. },
  143. MetaData: "Provides read-only dashboard access to platform users and allows connection to network nodes via the Netmaker Desktop App.",
  144. }
  145. d, _ := json.Marshal(NetworkGlobalAdminGroup)
  146. database.Insert(NetworkGlobalAdminGroup.ID.String(), string(d), database.USER_GROUPS_TABLE_NAME)
  147. d, _ = json.Marshal(NetworkGlobalUserGroup)
  148. database.Insert(NetworkGlobalUserGroup.ID.String(), string(d), database.USER_GROUPS_TABLE_NAME)
  149. }
  150. func CreateDefaultNetworkRolesAndGroups(netID models.NetworkID) {
  151. if netID.String() == "" {
  152. return
  153. }
  154. var NetworkAdminPermissionTemplate = models.UserRolePermissionTemplate{
  155. ID: GetDefaultNetworkAdminRoleID(netID),
  156. Name: fmt.Sprintf("%s Admin", netID),
  157. MetaData: fmt.Sprintf("can manage your network `%s` configuration.", netID),
  158. Default: true,
  159. NetworkID: netID,
  160. FullAccess: true,
  161. NetworkLevelAccess: make(map[models.RsrcType]map[models.RsrcID]models.RsrcPermissionScope),
  162. }
  163. var NetworkUserPermissionTemplate = models.UserRolePermissionTemplate{
  164. ID: GetDefaultNetworkUserRoleID(netID),
  165. Name: fmt.Sprintf("%s User", netID),
  166. MetaData: fmt.Sprintf("Can connect to nodes in your network `%s` via Netmaker Desktop App.", netID),
  167. Default: true,
  168. FullAccess: false,
  169. NetworkID: netID,
  170. DenyDashboardAccess: false,
  171. NetworkLevelAccess: map[models.RsrcType]map[models.RsrcID]models.RsrcPermissionScope{
  172. models.HostRsrc: {
  173. models.AllHostRsrcID: models.RsrcPermissionScope{
  174. Read: true,
  175. },
  176. },
  177. models.RemoteAccessGwRsrc: {
  178. models.AllRemoteAccessGwRsrcID: models.RsrcPermissionScope{
  179. Read: true,
  180. VPNaccess: true,
  181. },
  182. },
  183. models.ExtClientsRsrc: {
  184. models.AllExtClientsRsrcID: models.RsrcPermissionScope{
  185. Read: true,
  186. Create: true,
  187. Update: true,
  188. Delete: true,
  189. SelfOnly: true,
  190. },
  191. },
  192. models.DnsRsrc: {
  193. models.AllDnsRsrcID: models.RsrcPermissionScope{
  194. Read: true,
  195. },
  196. },
  197. models.AclRsrc: {
  198. models.AllAclsRsrcID: models.RsrcPermissionScope{
  199. Read: true,
  200. },
  201. },
  202. models.EgressGwRsrc: {
  203. models.AllEgressGwRsrcID: models.RsrcPermissionScope{
  204. Read: true,
  205. },
  206. },
  207. models.InetGwRsrc: {
  208. models.AllInetGwRsrcID: models.RsrcPermissionScope{
  209. Read: true,
  210. },
  211. },
  212. models.RelayRsrc: {
  213. models.AllRelayRsrcID: models.RsrcPermissionScope{
  214. Read: true,
  215. },
  216. },
  217. models.TagRsrc: {
  218. models.AllTagsRsrcID: models.RsrcPermissionScope{
  219. Read: true,
  220. },
  221. },
  222. },
  223. }
  224. d, _ := json.Marshal(NetworkAdminPermissionTemplate)
  225. database.Insert(NetworkAdminPermissionTemplate.ID.String(), string(d), database.USER_PERMISSIONS_TABLE_NAME)
  226. d, _ = json.Marshal(NetworkUserPermissionTemplate)
  227. database.Insert(NetworkUserPermissionTemplate.ID.String(), string(d), database.USER_PERMISSIONS_TABLE_NAME)
  228. // create default network groups
  229. var NetworkAdminGroup = models.UserGroup{
  230. ID: GetDefaultNetworkAdminGroupID(netID),
  231. Name: fmt.Sprintf("%s Admin Group", netID),
  232. Default: true,
  233. NetworkRoles: map[models.NetworkID]map[models.UserRoleID]struct{}{
  234. netID: {
  235. GetDefaultNetworkAdminRoleID(netID): {},
  236. },
  237. },
  238. MetaData: fmt.Sprintf("can manage your network `%s` configuration including adding and removing devices.", netID),
  239. }
  240. var NetworkUserGroup = models.UserGroup{
  241. ID: GetDefaultNetworkUserGroupID(netID),
  242. Name: fmt.Sprintf("%s User Group", netID),
  243. Default: true,
  244. NetworkRoles: map[models.NetworkID]map[models.UserRoleID]struct{}{
  245. netID: {
  246. GetDefaultNetworkUserRoleID(netID): {},
  247. },
  248. },
  249. 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),
  250. }
  251. d, _ = json.Marshal(NetworkAdminGroup)
  252. database.Insert(NetworkAdminGroup.ID.String(), string(d), database.USER_GROUPS_TABLE_NAME)
  253. d, _ = json.Marshal(NetworkUserGroup)
  254. database.Insert(NetworkUserGroup.ID.String(), string(d), database.USER_GROUPS_TABLE_NAME)
  255. }
  256. func DeleteNetworkRoles(netID string) {
  257. users, err := logic.GetUsersDB()
  258. if err != nil {
  259. return
  260. }
  261. defaultAdminGrpID := GetDefaultNetworkAdminGroupID(models.NetworkID(netID))
  262. defaultUserGrpID := GetDefaultNetworkUserGroupID(models.NetworkID(netID))
  263. for _, user := range users {
  264. var upsert bool
  265. if _, ok := user.NetworkRoles[models.NetworkID(netID)]; ok {
  266. delete(user.NetworkRoles, models.NetworkID(netID))
  267. upsert = true
  268. }
  269. if _, ok := user.UserGroups[defaultUserGrpID]; ok {
  270. delete(user.UserGroups, defaultUserGrpID)
  271. upsert = true
  272. }
  273. if _, ok := user.UserGroups[defaultAdminGrpID]; ok {
  274. delete(user.UserGroups, defaultAdminGrpID)
  275. upsert = true
  276. }
  277. if upsert {
  278. logic.UpsertUser(user)
  279. }
  280. }
  281. database.DeleteRecord(database.USER_GROUPS_TABLE_NAME, defaultUserGrpID.String())
  282. database.DeleteRecord(database.USER_GROUPS_TABLE_NAME, defaultAdminGrpID.String())
  283. userGs, _ := ListUserGroups()
  284. for _, userGI := range userGs {
  285. if _, ok := userGI.NetworkRoles[models.NetworkID(netID)]; ok {
  286. delete(userGI.NetworkRoles, models.NetworkID(netID))
  287. UpdateUserGroup(userGI)
  288. }
  289. }
  290. roles, _ := ListNetworkRoles()
  291. for _, role := range roles {
  292. if role.NetworkID.String() == netID {
  293. database.DeleteRecord(database.USER_PERMISSIONS_TABLE_NAME, role.ID.String())
  294. }
  295. }
  296. }
  297. // ListNetworkRoles - lists user network roles permission templates
  298. func ListNetworkRoles() ([]models.UserRolePermissionTemplate, error) {
  299. data, err := database.FetchRecords(database.USER_PERMISSIONS_TABLE_NAME)
  300. if err != nil && !database.IsEmptyRecord(err) {
  301. return []models.UserRolePermissionTemplate{}, err
  302. }
  303. userRoles := []models.UserRolePermissionTemplate{}
  304. for _, dataI := range data {
  305. userRole := models.UserRolePermissionTemplate{}
  306. err := json.Unmarshal([]byte(dataI), &userRole)
  307. if err != nil {
  308. continue
  309. }
  310. if userRole.NetworkID == "" {
  311. continue
  312. }
  313. userRoles = append(userRoles, userRole)
  314. }
  315. return userRoles, nil
  316. }
  317. func ValidateCreateRoleReq(userRole *models.UserRolePermissionTemplate) error {
  318. // check if role exists with this id
  319. _, err := logic.GetRole(userRole.ID)
  320. if err == nil {
  321. return fmt.Errorf("role with id `%s` exists already", userRole.ID.String())
  322. }
  323. if len(userRole.NetworkLevelAccess) > 0 {
  324. for rsrcType := range userRole.NetworkLevelAccess {
  325. if _, ok := models.RsrcTypeMap[rsrcType]; !ok {
  326. return errors.New("invalid rsrc type " + rsrcType.String())
  327. }
  328. if rsrcType == models.RemoteAccessGwRsrc {
  329. userRsrcPermissions := userRole.NetworkLevelAccess[models.RemoteAccessGwRsrc]
  330. var vpnAccess bool
  331. for _, scope := range userRsrcPermissions {
  332. if scope.VPNaccess {
  333. vpnAccess = true
  334. break
  335. }
  336. }
  337. if vpnAccess {
  338. userRole.NetworkLevelAccess[models.ExtClientsRsrc] = map[models.RsrcID]models.RsrcPermissionScope{
  339. models.AllExtClientsRsrcID: {
  340. Read: true,
  341. Create: true,
  342. Update: true,
  343. Delete: true,
  344. SelfOnly: true,
  345. },
  346. }
  347. }
  348. }
  349. }
  350. }
  351. if userRole.NetworkID == "" {
  352. return errors.New("only network roles are allowed to be created")
  353. }
  354. return nil
  355. }
  356. func ValidateUpdateRoleReq(userRole *models.UserRolePermissionTemplate) error {
  357. roleInDB, err := logic.GetRole(userRole.ID)
  358. if err != nil {
  359. return err
  360. }
  361. if roleInDB.NetworkID != userRole.NetworkID {
  362. return errors.New("network id mismatch")
  363. }
  364. if roleInDB.Default {
  365. return errors.New("cannot update default role")
  366. }
  367. if len(userRole.NetworkLevelAccess) > 0 {
  368. for rsrcType := range userRole.NetworkLevelAccess {
  369. if _, ok := models.RsrcTypeMap[rsrcType]; !ok {
  370. return errors.New("invalid rsrc type " + rsrcType.String())
  371. }
  372. if rsrcType == models.RemoteAccessGwRsrc {
  373. userRsrcPermissions := userRole.NetworkLevelAccess[models.RemoteAccessGwRsrc]
  374. var vpnAccess bool
  375. for _, scope := range userRsrcPermissions {
  376. if scope.VPNaccess {
  377. vpnAccess = true
  378. break
  379. }
  380. }
  381. if vpnAccess {
  382. userRole.NetworkLevelAccess[models.ExtClientsRsrc] = map[models.RsrcID]models.RsrcPermissionScope{
  383. models.AllExtClientsRsrcID: {
  384. Read: true,
  385. Create: true,
  386. Update: true,
  387. Delete: true,
  388. SelfOnly: true,
  389. },
  390. }
  391. }
  392. }
  393. }
  394. }
  395. return nil
  396. }
  397. // CreateRole - inserts new role into DB
  398. func CreateRole(r models.UserRolePermissionTemplate) error {
  399. // default roles are currently created directly in the db.
  400. // this check is only to prevent future errors.
  401. if r.Default && r.ID == "" {
  402. return errors.New("role id cannot be empty for default role")
  403. }
  404. if !r.Default {
  405. r.ID = models.UserRoleID(uuid.NewString())
  406. }
  407. // check if the role already exists
  408. if r.Name == "" {
  409. return errors.New("role name cannot be empty")
  410. }
  411. roles, err := ListNetworkRoles()
  412. if err != nil {
  413. return err
  414. }
  415. for _, role := range roles {
  416. if role.Name == r.Name {
  417. return errors.New("role already exists")
  418. }
  419. }
  420. d, err := json.Marshal(r)
  421. if err != nil {
  422. return err
  423. }
  424. return database.Insert(r.ID.String(), string(d), database.USER_PERMISSIONS_TABLE_NAME)
  425. }
  426. // UpdateRole - updates role template
  427. func UpdateRole(r models.UserRolePermissionTemplate) error {
  428. if r.ID.String() == "" {
  429. return errors.New("role id cannot be empty")
  430. }
  431. _, err := database.FetchRecord(database.USER_PERMISSIONS_TABLE_NAME, r.ID.String())
  432. if err != nil {
  433. return err
  434. }
  435. d, err := json.Marshal(r)
  436. if err != nil {
  437. return err
  438. }
  439. return database.Insert(r.ID.String(), string(d), database.USER_PERMISSIONS_TABLE_NAME)
  440. }
  441. // DeleteRole - deletes user role
  442. func DeleteRole(rid models.UserRoleID, force bool) error {
  443. if rid.String() == "" {
  444. return errors.New("role id cannot be empty")
  445. }
  446. users, err := logic.GetUsersDB()
  447. if err != nil {
  448. return err
  449. }
  450. role, err := logic.GetRole(rid)
  451. if err != nil {
  452. return err
  453. }
  454. if role.NetworkID == "" {
  455. return errors.New("cannot delete platform role")
  456. }
  457. // allow deletion of default network roles if network doesn't exist
  458. if role.NetworkID == models.AllNetworks {
  459. return errors.New("cannot delete default network role")
  460. }
  461. // check if network exists
  462. exists, _ := logic.NetworkExists(role.NetworkID.String())
  463. if role.Default {
  464. if exists && !force {
  465. return errors.New("cannot delete default role")
  466. }
  467. }
  468. for _, user := range users {
  469. for userG := range user.UserGroups {
  470. ug, err := GetUserGroup(userG)
  471. if err == nil {
  472. if role.NetworkID != "" {
  473. for netID, networkRoles := range ug.NetworkRoles {
  474. if _, ok := networkRoles[rid]; ok {
  475. delete(networkRoles, rid)
  476. ug.NetworkRoles[netID] = networkRoles
  477. UpdateUserGroup(ug)
  478. }
  479. }
  480. }
  481. }
  482. }
  483. if user.PlatformRoleID == rid {
  484. err = errors.New("active roles cannot be deleted.switch existing users to a new role before deleting")
  485. return err
  486. }
  487. if role.NetworkID != "" {
  488. for netID, networkRoles := range user.NetworkRoles {
  489. if _, ok := networkRoles[rid]; ok {
  490. delete(networkRoles, rid)
  491. user.NetworkRoles[netID] = networkRoles
  492. logic.UpsertUser(user)
  493. }
  494. }
  495. }
  496. }
  497. return database.DeleteRecord(database.USER_PERMISSIONS_TABLE_NAME, rid.String())
  498. }
  499. func ValidateCreateGroupReq(g models.UserGroup) error {
  500. // check if network roles are valid
  501. for _, roleMap := range g.NetworkRoles {
  502. for roleID := range roleMap {
  503. role, err := logic.GetRole(roleID)
  504. if err != nil {
  505. return fmt.Errorf("invalid network role %s", roleID)
  506. }
  507. if role.NetworkID == "" {
  508. return errors.New("platform role cannot be used as network role")
  509. }
  510. }
  511. }
  512. return nil
  513. }
  514. func ValidateUpdateGroupReq(g models.UserGroup) error {
  515. for networkID := range g.NetworkRoles {
  516. userRolesMap := g.NetworkRoles[networkID]
  517. for roleID := range userRolesMap {
  518. netRole, err := logic.GetRole(roleID)
  519. if err != nil {
  520. err = fmt.Errorf("invalid network role")
  521. return err
  522. }
  523. if netRole.NetworkID == "" {
  524. return errors.New("platform role cannot be used as network role")
  525. }
  526. }
  527. }
  528. return nil
  529. }
  530. // CreateUserGroup - creates new user group
  531. func CreateUserGroup(g *models.UserGroup) error {
  532. // default groups are currently created directly in the db.
  533. // this check is only to prevent future errors.
  534. if g.Default && g.ID == "" {
  535. return errors.New("group id cannot be empty for default group")
  536. }
  537. if !g.Default {
  538. g.ID = models.UserGroupID(uuid.NewString())
  539. }
  540. // check if the group already exists
  541. if g.Name == "" {
  542. return errors.New("group name cannot be empty")
  543. }
  544. groups, err := ListUserGroups()
  545. if err != nil {
  546. return err
  547. }
  548. for _, group := range groups {
  549. if group.Name == g.Name {
  550. return errors.New("group already exists")
  551. }
  552. }
  553. d, err := json.Marshal(g)
  554. if err != nil {
  555. return err
  556. }
  557. err = database.Insert(g.ID.String(), string(d), database.USER_GROUPS_TABLE_NAME)
  558. if err != nil {
  559. return err
  560. }
  561. // create default network gateway policies
  562. CreateDefaultUserGroupNetworkPolicies(*g)
  563. return nil
  564. }
  565. // GetUserGroup - fetches user group
  566. func GetUserGroup(gid models.UserGroupID) (models.UserGroup, error) {
  567. d, err := database.FetchRecord(database.USER_GROUPS_TABLE_NAME, gid.String())
  568. if err != nil {
  569. return models.UserGroup{}, err
  570. }
  571. var ug models.UserGroup
  572. err = json.Unmarshal([]byte(d), &ug)
  573. if err != nil {
  574. return ug, err
  575. }
  576. return ug, nil
  577. }
  578. func GetDefaultGlobalAdminGroupID() models.UserGroupID {
  579. return globalNetworksAdminGroupID
  580. }
  581. func GetDefaultGlobalUserGroupID() models.UserGroupID {
  582. return globalNetworksUserGroupID
  583. }
  584. func GetDefaultGlobalAdminRoleID() models.UserRoleID {
  585. return globalNetworksAdminRoleID
  586. }
  587. func GetDefaultGlobalUserRoleID() models.UserRoleID {
  588. return globalNetworksUserRoleID
  589. }
  590. func GetDefaultNetworkAdminGroupID(networkID models.NetworkID) models.UserGroupID {
  591. return models.UserGroupID(fmt.Sprintf("%s-%s-grp", networkID, models.NetworkAdmin))
  592. }
  593. func GetDefaultNetworkUserGroupID(networkID models.NetworkID) models.UserGroupID {
  594. return models.UserGroupID(fmt.Sprintf("%s-%s-grp", networkID, models.NetworkUser))
  595. }
  596. func GetDefaultNetworkAdminRoleID(networkID models.NetworkID) models.UserRoleID {
  597. return models.UserRoleID(fmt.Sprintf("%s-%s", networkID, models.NetworkAdmin))
  598. }
  599. func GetDefaultNetworkUserRoleID(networkID models.NetworkID) models.UserRoleID {
  600. return models.UserRoleID(fmt.Sprintf("%s-%s", networkID, models.NetworkUser))
  601. }
  602. // ListUserGroups - lists user groups
  603. func ListUserGroups() ([]models.UserGroup, error) {
  604. data, err := database.FetchRecords(database.USER_GROUPS_TABLE_NAME)
  605. if err != nil && !database.IsEmptyRecord(err) {
  606. return []models.UserGroup{}, err
  607. }
  608. userGroups := []models.UserGroup{}
  609. for _, dataI := range data {
  610. userGroup := models.UserGroup{}
  611. err := json.Unmarshal([]byte(dataI), &userGroup)
  612. if err != nil {
  613. continue
  614. }
  615. userGroups = append(userGroups, userGroup)
  616. }
  617. return userGroups, nil
  618. }
  619. // UpdateUserGroup - updates new user group
  620. func UpdateUserGroup(g models.UserGroup) error {
  621. // check if the group exists
  622. if g.ID == "" {
  623. return errors.New("group id cannot be empty")
  624. }
  625. _, err := database.FetchRecord(database.USER_GROUPS_TABLE_NAME, g.ID.String())
  626. if err != nil {
  627. return err
  628. }
  629. d, err := json.Marshal(g)
  630. if err != nil {
  631. return err
  632. }
  633. return database.Insert(g.ID.String(), string(d), database.USER_GROUPS_TABLE_NAME)
  634. }
  635. func DeleteAndCleanUpGroup(group *models.UserGroup) error {
  636. err := DeleteUserGroup(group.ID)
  637. if err != nil {
  638. return err
  639. }
  640. go func() {
  641. var replacePeers bool
  642. for networkID := range group.NetworkRoles {
  643. acls, err := logic.ListAclsByNetwork(networkID)
  644. if err != nil {
  645. continue
  646. }
  647. for _, acl := range acls {
  648. var hasGroupSrc bool
  649. newAclSrc := make([]models.AclPolicyTag, 0)
  650. for _, src := range acl.Src {
  651. if src.ID == models.UserGroupAclID && src.Value == group.ID.String() {
  652. hasGroupSrc = true
  653. } else {
  654. newAclSrc = append(newAclSrc, src)
  655. }
  656. }
  657. if hasGroupSrc {
  658. if len(newAclSrc) == 0 {
  659. // no other src exists, delete acl.
  660. _ = logic.DeleteAcl(acl)
  661. } else {
  662. // other sources exist, update acl.
  663. acl.Src = newAclSrc
  664. _ = logic.UpsertAcl(acl)
  665. }
  666. replacePeers = true
  667. }
  668. }
  669. }
  670. go UpdatesUserGwAccessOnGrpUpdates(group.ID, group.NetworkRoles, make(map[models.NetworkID]map[models.UserRoleID]struct{}))
  671. go mq.PublishPeerUpdate(replacePeers)
  672. }()
  673. return nil
  674. }
  675. // DeleteUserGroup - deletes user group
  676. func DeleteUserGroup(gid models.UserGroupID) error {
  677. g, err := GetUserGroup(gid)
  678. if err != nil {
  679. return err
  680. }
  681. users, err := logic.GetUsersDB()
  682. if err != nil && !database.IsEmptyRecord(err) {
  683. return err
  684. }
  685. for _, user := range users {
  686. delete(user.UserGroups, gid)
  687. logic.UpsertUser(user)
  688. }
  689. // create default network gateway policies
  690. DeleteDefaultUserGroupNetworkPolicies(g)
  691. return database.DeleteRecord(database.USER_GROUPS_TABLE_NAME, gid.String())
  692. }
  693. func HasNetworkRsrcScope(permissionTemplate models.UserRolePermissionTemplate, netid string, rsrcType models.RsrcType, rsrcID models.RsrcID, op string) bool {
  694. if permissionTemplate.FullAccess {
  695. return true
  696. }
  697. rsrcScope, ok := permissionTemplate.NetworkLevelAccess[rsrcType]
  698. if !ok {
  699. return false
  700. }
  701. _, ok = rsrcScope[rsrcID]
  702. return ok
  703. }
  704. func GetUserRAGNodesV1(user models.User) (gws map[string]models.Node) {
  705. gws = make(map[string]models.Node)
  706. nodes, err := logic.GetAllNodes()
  707. if err != nil {
  708. return
  709. }
  710. if user.PlatformRoleID == models.AdminRole || user.PlatformRoleID == models.SuperAdminRole {
  711. for _, node := range nodes {
  712. if node.IsIngressGateway {
  713. gws[node.ID.String()] = node
  714. }
  715. }
  716. }
  717. tagNodesMap := GetTagMapWithNodes()
  718. accessPolices := ListUserPolicies(user)
  719. for _, policyI := range accessPolices {
  720. if !policyI.Enabled {
  721. continue
  722. }
  723. for _, dstI := range policyI.Dst {
  724. if dstI.Value == "*" {
  725. networkNodes := logic.GetNetworkNodesMemory(nodes, policyI.NetworkID.String())
  726. for _, node := range networkNodes {
  727. if node.IsIngressGateway {
  728. gws[node.ID.String()] = node
  729. }
  730. }
  731. }
  732. if nodes, ok := tagNodesMap[models.TagID(dstI.Value)]; ok {
  733. for _, node := range nodes {
  734. if node.IsIngressGateway {
  735. gws[node.ID.String()] = node
  736. }
  737. }
  738. }
  739. }
  740. }
  741. return
  742. }
  743. func GetUserRAGNodes(user models.User) (gws map[string]models.Node) {
  744. gws = make(map[string]models.Node)
  745. nodes, err := logic.GetAllNodes()
  746. if err != nil {
  747. return
  748. }
  749. for _, node := range nodes {
  750. if !node.IsGw {
  751. continue
  752. }
  753. if user.PlatformRoleID == models.AdminRole || user.PlatformRoleID == models.SuperAdminRole {
  754. if ok, _ := IsUserAllowedToCommunicate(user.UserName, node); ok {
  755. gws[node.ID.String()] = node
  756. continue
  757. }
  758. } else {
  759. // check if user has network role assigned
  760. if roles, ok := user.NetworkRoles[models.NetworkID(node.Network)]; ok && len(roles) > 0 {
  761. if ok, _ := IsUserAllowedToCommunicate(user.UserName, node); ok {
  762. gws[node.ID.String()] = node
  763. continue
  764. }
  765. }
  766. if roles, ok := user.NetworkRoles[models.AllNetworks]; ok && len(roles) > 0 {
  767. if ok, _ := IsUserAllowedToCommunicate(user.UserName, node); ok {
  768. gws[node.ID.String()] = node
  769. continue
  770. }
  771. }
  772. for groupID := range user.UserGroups {
  773. userGrp, err := logic.GetUserGroup(groupID)
  774. if err == nil {
  775. if roles, ok := userGrp.NetworkRoles[models.NetworkID(node.Network)]; ok && len(roles) > 0 {
  776. if ok, _ := IsUserAllowedToCommunicate(user.UserName, node); ok {
  777. gws[node.ID.String()] = node
  778. break
  779. }
  780. }
  781. if roles, ok := userGrp.NetworkRoles[models.AllNetworks]; ok && len(roles) > 0 {
  782. if ok, _ := IsUserAllowedToCommunicate(user.UserName, node); ok {
  783. gws[node.ID.String()] = node
  784. break
  785. }
  786. }
  787. }
  788. }
  789. }
  790. }
  791. return
  792. }
  793. // GetUserNetworkRoles - get user network roles
  794. func GetUserNetworkRolesWithRemoteVPNAccess(user models.User) (gwAccess map[models.NetworkID]map[models.RsrcID]models.RsrcPermissionScope) {
  795. gwAccess = make(map[models.NetworkID]map[models.RsrcID]models.RsrcPermissionScope)
  796. platformRole, err := logic.GetRole(user.PlatformRoleID)
  797. if err != nil {
  798. return
  799. }
  800. if platformRole.FullAccess {
  801. gwAccess[models.NetworkID("*")] = make(map[models.RsrcID]models.RsrcPermissionScope)
  802. return
  803. }
  804. if _, ok := user.NetworkRoles[models.AllNetworks]; ok {
  805. gwAccess[models.NetworkID("*")] = make(map[models.RsrcID]models.RsrcPermissionScope)
  806. return
  807. }
  808. if len(user.UserGroups) > 0 {
  809. for gID := range user.UserGroups {
  810. userG, err := GetUserGroup(gID)
  811. if err != nil {
  812. continue
  813. }
  814. if _, ok := userG.NetworkRoles[models.AllNetworks]; ok {
  815. gwAccess[models.NetworkID("*")] = make(map[models.RsrcID]models.RsrcPermissionScope)
  816. return
  817. }
  818. for netID, roleMap := range userG.NetworkRoles {
  819. for roleID := range roleMap {
  820. role, err := logic.GetRole(roleID)
  821. if err == nil {
  822. if role.FullAccess {
  823. gwAccess[netID] = map[models.RsrcID]models.RsrcPermissionScope{
  824. models.AllRemoteAccessGwRsrcID: {
  825. Create: true,
  826. Read: true,
  827. Update: true,
  828. VPNaccess: true,
  829. Delete: true,
  830. },
  831. models.AllExtClientsRsrcID: {
  832. Create: true,
  833. Read: true,
  834. Update: true,
  835. Delete: true,
  836. },
  837. }
  838. break
  839. }
  840. if rsrcsMap, ok := role.NetworkLevelAccess[models.RemoteAccessGwRsrc]; ok {
  841. if permissions, ok := rsrcsMap[models.AllRemoteAccessGwRsrcID]; ok && permissions.VPNaccess {
  842. if len(gwAccess[netID]) == 0 {
  843. gwAccess[netID] = make(map[models.RsrcID]models.RsrcPermissionScope)
  844. }
  845. gwAccess[netID][models.AllRemoteAccessGwRsrcID] = permissions
  846. break
  847. } else {
  848. for gwID, scope := range rsrcsMap {
  849. if scope.VPNaccess {
  850. if len(gwAccess[netID]) == 0 {
  851. gwAccess[netID] = make(map[models.RsrcID]models.RsrcPermissionScope)
  852. }
  853. gwAccess[netID][gwID] = scope
  854. }
  855. }
  856. }
  857. }
  858. }
  859. }
  860. }
  861. }
  862. }
  863. for netID, roleMap := range user.NetworkRoles {
  864. for roleID := range roleMap {
  865. role, err := logic.GetRole(roleID)
  866. if err == nil {
  867. if role.FullAccess {
  868. gwAccess[netID] = map[models.RsrcID]models.RsrcPermissionScope{
  869. models.AllRemoteAccessGwRsrcID: {
  870. Create: true,
  871. Read: true,
  872. Update: true,
  873. VPNaccess: true,
  874. Delete: true,
  875. },
  876. models.AllExtClientsRsrcID: {
  877. Create: true,
  878. Read: true,
  879. Update: true,
  880. Delete: true,
  881. },
  882. }
  883. break
  884. }
  885. if rsrcsMap, ok := role.NetworkLevelAccess[models.RemoteAccessGwRsrc]; ok {
  886. if permissions, ok := rsrcsMap[models.AllRemoteAccessGwRsrcID]; ok && permissions.VPNaccess {
  887. if len(gwAccess[netID]) == 0 {
  888. gwAccess[netID] = make(map[models.RsrcID]models.RsrcPermissionScope)
  889. }
  890. gwAccess[netID][models.AllRemoteAccessGwRsrcID] = permissions
  891. break
  892. } else {
  893. for gwID, scope := range rsrcsMap {
  894. if scope.VPNaccess {
  895. if len(gwAccess[netID]) == 0 {
  896. gwAccess[netID] = make(map[models.RsrcID]models.RsrcPermissionScope)
  897. }
  898. gwAccess[netID][gwID] = scope
  899. }
  900. }
  901. }
  902. }
  903. }
  904. }
  905. }
  906. return
  907. }
  908. func GetFilteredNodesByUserAccess(user models.User, nodes []models.Node) (filteredNodes []models.Node) {
  909. return filteredNodes
  910. }
  911. func FilterNetworksByRole(allnetworks []models.Network, user models.User) []models.Network {
  912. platformRole, err := logic.GetRole(user.PlatformRoleID)
  913. if err != nil {
  914. return []models.Network{}
  915. }
  916. if !platformRole.FullAccess {
  917. allNetworkRoles := make(map[models.NetworkID]struct{})
  918. if len(user.NetworkRoles) > 0 {
  919. for netID := range user.NetworkRoles {
  920. if netID == models.AllNetworks {
  921. return allnetworks
  922. }
  923. allNetworkRoles[netID] = struct{}{}
  924. }
  925. }
  926. if len(user.UserGroups) > 0 {
  927. for userGID := range user.UserGroups {
  928. userG, err := GetUserGroup(userGID)
  929. if err == nil {
  930. if len(userG.NetworkRoles) > 0 {
  931. for netID := range userG.NetworkRoles {
  932. if netID == models.AllNetworks {
  933. return allnetworks
  934. }
  935. allNetworkRoles[netID] = struct{}{}
  936. }
  937. }
  938. }
  939. }
  940. }
  941. filteredNetworks := []models.Network{}
  942. for _, networkI := range allnetworks {
  943. if _, ok := allNetworkRoles[models.NetworkID(networkI.NetID)]; ok {
  944. filteredNetworks = append(filteredNetworks, networkI)
  945. }
  946. }
  947. allnetworks = filteredNetworks
  948. }
  949. return allnetworks
  950. }
  951. func IsGroupsValid(groups map[models.UserGroupID]struct{}) error {
  952. for groupID := range groups {
  953. _, err := GetUserGroup(groupID)
  954. if err != nil {
  955. return fmt.Errorf("user group `%s` not found", groupID)
  956. }
  957. }
  958. return nil
  959. }
  960. func IsGroupValid(groupID models.UserGroupID) error {
  961. _, err := GetUserGroup(groupID)
  962. if err != nil {
  963. return fmt.Errorf("user group `%s` not found", groupID)
  964. }
  965. return nil
  966. }
  967. func IsNetworkRolesValid(networkRoles map[models.NetworkID]map[models.UserRoleID]struct{}) error {
  968. for netID, netRoles := range networkRoles {
  969. if netID != models.AllNetworks {
  970. _, err := logic.GetNetwork(netID.String())
  971. if err != nil {
  972. return fmt.Errorf("failed to fetch network %s ", netID)
  973. }
  974. }
  975. for netRoleID := range netRoles {
  976. role, err := logic.GetRole(netRoleID)
  977. if err != nil {
  978. return fmt.Errorf("failed to fetch role %s ", netRoleID)
  979. }
  980. if role.NetworkID == "" {
  981. return fmt.Errorf("cannot use platform as network role %s", netRoleID)
  982. }
  983. }
  984. }
  985. return nil
  986. }
  987. // PrepareOauthUserFromInvite - init oauth user before create
  988. func PrepareOauthUserFromInvite(in models.UserInvite) (models.User, error) {
  989. var newPass, fetchErr = logic.FetchPassValue("")
  990. if fetchErr != nil {
  991. return models.User{}, fetchErr
  992. }
  993. user := models.User{
  994. UserName: in.Email,
  995. Password: newPass,
  996. }
  997. user.UserGroups = in.UserGroups
  998. user.NetworkRoles = in.NetworkRoles
  999. user.PlatformRoleID = models.UserRoleID(in.PlatformRoleID)
  1000. if user.PlatformRoleID == "" {
  1001. user.PlatformRoleID = models.ServiceUser
  1002. }
  1003. return user, nil
  1004. }
  1005. func UpdatesUserGwAccessOnRoleUpdates(currNetworkAccess,
  1006. changeNetworkAccess map[models.RsrcType]map[models.RsrcID]models.RsrcPermissionScope, netID string) {
  1007. networkChangeMap := make(map[models.RsrcID]models.RsrcPermissionScope)
  1008. for rsrcType, RsrcPermsMap := range currNetworkAccess {
  1009. if rsrcType != models.RemoteAccessGwRsrc {
  1010. continue
  1011. }
  1012. if _, ok := changeNetworkAccess[rsrcType]; !ok {
  1013. for rsrcID, scope := range RsrcPermsMap {
  1014. networkChangeMap[rsrcID] = scope
  1015. }
  1016. } else {
  1017. for rsrcID, scope := range RsrcPermsMap {
  1018. if _, ok := changeNetworkAccess[rsrcType][rsrcID]; !ok {
  1019. networkChangeMap[rsrcID] = scope
  1020. }
  1021. }
  1022. }
  1023. }
  1024. extclients, err := logic.GetAllExtClients()
  1025. if err != nil {
  1026. slog.Error("failed to fetch extclients", "error", err)
  1027. return
  1028. }
  1029. userMap, err := logic.GetUserMap()
  1030. if err != nil {
  1031. return
  1032. }
  1033. for _, extclient := range extclients {
  1034. if extclient.Network != netID {
  1035. continue
  1036. }
  1037. if _, ok := networkChangeMap[models.AllRemoteAccessGwRsrcID]; ok {
  1038. if user, ok := userMap[extclient.OwnerID]; ok {
  1039. if user.PlatformRoleID != models.ServiceUser {
  1040. continue
  1041. }
  1042. err = logic.DeleteExtClientAndCleanup(extclient)
  1043. if err != nil {
  1044. slog.Error("failed to delete extclient",
  1045. "id", extclient.ClientID, "owner", user.UserName, "error", err)
  1046. } else {
  1047. if err := mq.PublishDeletedClientPeerUpdate(&extclient); err != nil {
  1048. slog.Error("error setting ext peers: " + err.Error())
  1049. }
  1050. }
  1051. }
  1052. continue
  1053. }
  1054. if _, ok := networkChangeMap[models.RsrcID(extclient.IngressGatewayID)]; ok {
  1055. if user, ok := userMap[extclient.OwnerID]; ok {
  1056. if user.PlatformRoleID != models.ServiceUser {
  1057. continue
  1058. }
  1059. err = logic.DeleteExtClientAndCleanup(extclient)
  1060. if err != nil {
  1061. slog.Error("failed to delete extclient",
  1062. "id", extclient.ClientID, "owner", user.UserName, "error", err)
  1063. } else {
  1064. if err := mq.PublishDeletedClientPeerUpdate(&extclient); err != nil {
  1065. slog.Error("error setting ext peers: " + err.Error())
  1066. }
  1067. }
  1068. }
  1069. }
  1070. }
  1071. if servercfg.IsDNSMode() {
  1072. logic.SetDNS()
  1073. }
  1074. }
  1075. func UpdatesUserGwAccessOnGrpUpdates(groupID models.UserGroupID, oldNetworkRoles, newNetworkRoles map[models.NetworkID]map[models.UserRoleID]struct{}) {
  1076. networkRemovedMap := make(map[models.NetworkID]struct{})
  1077. for netID := range oldNetworkRoles {
  1078. if _, ok := newNetworkRoles[netID]; !ok {
  1079. networkRemovedMap[netID] = struct{}{}
  1080. }
  1081. }
  1082. extclients, err := logic.GetAllExtClients()
  1083. if err != nil {
  1084. slog.Error("failed to fetch extclients", "error", err)
  1085. return
  1086. }
  1087. userMap, err := logic.GetUserMap()
  1088. if err != nil {
  1089. return
  1090. }
  1091. for _, extclient := range extclients {
  1092. var shouldDelete bool
  1093. user, ok := userMap[extclient.OwnerID]
  1094. if !ok {
  1095. // user does not exist, delete extclient.
  1096. shouldDelete = true
  1097. } else {
  1098. if user.PlatformRoleID == models.SuperAdminRole || user.PlatformRoleID == models.AdminRole {
  1099. // Super-admin and Admin's access is not determined by group membership
  1100. // or network roles. Even if a network is removed from the group, they
  1101. // continue to have access to the network.
  1102. // So, no need to delete the extclient.
  1103. shouldDelete = false
  1104. } else {
  1105. _, hasAccess := user.NetworkRoles[models.NetworkID(extclient.Network)]
  1106. if hasAccess {
  1107. // The user has access to the network by themselves and not by
  1108. // virtue of being a member of the group.
  1109. // So, no need to delete the extclient.
  1110. shouldDelete = false
  1111. } else {
  1112. _, userInGroup := user.UserGroups[groupID]
  1113. _, networkRemoved := networkRemovedMap[models.NetworkID(extclient.Network)]
  1114. if userInGroup && networkRemoved {
  1115. // This group no longer provides it's members access to the
  1116. // network.
  1117. // This user is a member of the group and has no direct
  1118. // access to the network (either by its platform role or by
  1119. // network roles).
  1120. // So, delete the extclient.
  1121. shouldDelete = true
  1122. }
  1123. }
  1124. }
  1125. }
  1126. if shouldDelete {
  1127. err = logic.DeleteExtClientAndCleanup(extclient)
  1128. if err != nil {
  1129. slog.Error("failed to delete extclient",
  1130. "id", extclient.ClientID, "owner", user.UserName, "error", err)
  1131. } else {
  1132. if err := mq.PublishDeletedClientPeerUpdate(&extclient); err != nil {
  1133. slog.Error("error setting ext peers: " + err.Error())
  1134. }
  1135. }
  1136. }
  1137. }
  1138. if servercfg.IsDNSMode() {
  1139. logic.SetDNS()
  1140. }
  1141. }
  1142. func UpdateUserGwAccess(currentUser, changeUser models.User) {
  1143. if changeUser.PlatformRoleID != models.ServiceUser {
  1144. return
  1145. }
  1146. networkChangeMap := make(map[models.NetworkID]map[models.UserRoleID]struct{})
  1147. for netID, networkUserRoles := range currentUser.NetworkRoles {
  1148. if _, ok := changeUser.NetworkRoles[netID]; !ok {
  1149. for netRoleID := range networkUserRoles {
  1150. if _, ok := networkChangeMap[netID]; !ok {
  1151. networkChangeMap[netID] = make(map[models.UserRoleID]struct{})
  1152. }
  1153. networkChangeMap[netID][netRoleID] = struct{}{}
  1154. }
  1155. } else {
  1156. for netRoleID := range networkUserRoles {
  1157. if _, ok := changeUser.NetworkRoles[netID][netRoleID]; !ok {
  1158. if _, ok := networkChangeMap[netID]; !ok {
  1159. networkChangeMap[netID] = make(map[models.UserRoleID]struct{})
  1160. }
  1161. networkChangeMap[netID][netRoleID] = struct{}{}
  1162. }
  1163. }
  1164. }
  1165. }
  1166. for gID := range currentUser.UserGroups {
  1167. if _, ok := changeUser.UserGroups[gID]; ok {
  1168. continue
  1169. }
  1170. userG, err := GetUserGroup(gID)
  1171. if err == nil {
  1172. for netID, networkUserRoles := range userG.NetworkRoles {
  1173. for netRoleID := range networkUserRoles {
  1174. if _, ok := networkChangeMap[netID]; !ok {
  1175. networkChangeMap[netID] = make(map[models.UserRoleID]struct{})
  1176. }
  1177. networkChangeMap[netID][netRoleID] = struct{}{}
  1178. }
  1179. }
  1180. }
  1181. }
  1182. if len(networkChangeMap) == 0 {
  1183. return
  1184. }
  1185. // TODO - cleanup gw access when role and groups are updated
  1186. //removedGwAccess
  1187. extclients, err := logic.GetAllExtClients()
  1188. if err != nil {
  1189. slog.Error("failed to fetch extclients", "error", err)
  1190. return
  1191. }
  1192. for _, extclient := range extclients {
  1193. if extclient.OwnerID == currentUser.UserName {
  1194. if _, ok := networkChangeMap[models.NetworkID(extclient.Network)]; ok {
  1195. err = logic.DeleteExtClientAndCleanup(extclient)
  1196. if err != nil {
  1197. slog.Error("failed to delete extclient",
  1198. "id", extclient.ClientID, "owner", changeUser.UserName, "error", err)
  1199. } else {
  1200. if err := mq.PublishDeletedClientPeerUpdate(&extclient); err != nil {
  1201. slog.Error("error setting ext peers: " + err.Error())
  1202. }
  1203. }
  1204. }
  1205. }
  1206. }
  1207. if servercfg.IsDNSMode() {
  1208. logic.SetDNS()
  1209. }
  1210. }
  1211. func CreateDefaultUserGroupNetworkPolicies(g models.UserGroup) {
  1212. for networkID := range g.NetworkRoles {
  1213. network, err := logic.GetNetwork(networkID.String())
  1214. if err != nil {
  1215. continue
  1216. }
  1217. acl := models.Acl{
  1218. ID: uuid.New().String(),
  1219. Name: fmt.Sprintf("%s group", g.Name),
  1220. MetaData: "This Policy allows user group to communicate with all gateways",
  1221. Default: true,
  1222. ServiceType: models.Any,
  1223. NetworkID: models.NetworkID(network.NetID),
  1224. Proto: models.ALL,
  1225. RuleType: models.UserPolicy,
  1226. Src: []models.AclPolicyTag{
  1227. {
  1228. ID: models.UserGroupAclID,
  1229. Value: g.ID.String(),
  1230. },
  1231. },
  1232. Dst: []models.AclPolicyTag{
  1233. {
  1234. ID: models.NodeTagID,
  1235. Value: fmt.Sprintf("%s.%s", models.NetworkID(network.NetID), models.GwTagName),
  1236. }},
  1237. AllowedDirection: models.TrafficDirectionUni,
  1238. Enabled: true,
  1239. CreatedBy: "auto",
  1240. CreatedAt: time.Now().UTC(),
  1241. }
  1242. logic.InsertAcl(acl)
  1243. }
  1244. }
  1245. func DeleteDefaultUserGroupNetworkPolicies(g models.UserGroup) {
  1246. for networkID := range g.NetworkRoles {
  1247. acls, err := logic.ListAclsByNetwork(networkID)
  1248. if err != nil {
  1249. continue
  1250. }
  1251. for _, acl := range acls {
  1252. var hasGroupSrc bool
  1253. newAclSrc := make([]models.AclPolicyTag, 0)
  1254. for _, src := range acl.Src {
  1255. if src.ID == models.UserGroupAclID && src.Value == g.ID.String() {
  1256. hasGroupSrc = true
  1257. } else {
  1258. newAclSrc = append(newAclSrc, src)
  1259. }
  1260. }
  1261. if hasGroupSrc {
  1262. if len(newAclSrc) == 0 {
  1263. // no other src exists, delete acl.
  1264. _ = logic.DeleteAcl(acl)
  1265. } else {
  1266. // other sources exist, update acl.
  1267. acl.Src = newAclSrc
  1268. _ = logic.UpsertAcl(acl)
  1269. }
  1270. }
  1271. }
  1272. }
  1273. }
  1274. func CreateDefaultUserPolicies(netID models.NetworkID) {
  1275. if netID.String() == "" {
  1276. return
  1277. }
  1278. if !logic.IsAclExists(fmt.Sprintf("%s.%s", netID, "all-users")) {
  1279. defaultUserAcl := models.Acl{
  1280. ID: fmt.Sprintf("%s.%s", netID, "all-users"),
  1281. Default: true,
  1282. Name: "All Users",
  1283. MetaData: "This policy gives access to everything in the network for an user",
  1284. NetworkID: netID,
  1285. Proto: models.ALL,
  1286. ServiceType: models.Any,
  1287. Port: []string{},
  1288. RuleType: models.UserPolicy,
  1289. Src: []models.AclPolicyTag{
  1290. {
  1291. ID: models.UserAclID,
  1292. Value: "*",
  1293. },
  1294. },
  1295. Dst: []models.AclPolicyTag{{
  1296. ID: models.NodeTagID,
  1297. Value: "*",
  1298. }},
  1299. AllowedDirection: models.TrafficDirectionUni,
  1300. Enabled: true,
  1301. CreatedBy: "auto",
  1302. CreatedAt: time.Now().UTC(),
  1303. }
  1304. logic.InsertAcl(defaultUserAcl)
  1305. }
  1306. if !logic.IsAclExists(fmt.Sprintf("%s.%s-grp", netID, models.NetworkAdmin)) {
  1307. networkAdminGroupID := GetDefaultNetworkAdminGroupID(netID)
  1308. defaultUserAcl := models.Acl{
  1309. ID: fmt.Sprintf("%s.%s-grp", netID, models.NetworkAdmin),
  1310. Name: "Network Admin",
  1311. MetaData: "This Policy allows all network admins to communicate with all gateways",
  1312. Default: true,
  1313. ServiceType: models.Any,
  1314. NetworkID: netID,
  1315. Proto: models.ALL,
  1316. RuleType: models.UserPolicy,
  1317. Src: []models.AclPolicyTag{
  1318. {
  1319. ID: models.UserGroupAclID,
  1320. Value: globalNetworksAdminGroupID.String(),
  1321. },
  1322. {
  1323. ID: models.UserGroupAclID,
  1324. Value: networkAdminGroupID.String(),
  1325. },
  1326. },
  1327. Dst: []models.AclPolicyTag{
  1328. {
  1329. ID: models.NodeTagID,
  1330. Value: fmt.Sprintf("%s.%s", netID, models.GwTagName),
  1331. }},
  1332. AllowedDirection: models.TrafficDirectionUni,
  1333. Enabled: true,
  1334. CreatedBy: "auto",
  1335. CreatedAt: time.Now().UTC(),
  1336. }
  1337. logic.InsertAcl(defaultUserAcl)
  1338. }
  1339. if !logic.IsAclExists(fmt.Sprintf("%s.%s-grp", netID, models.NetworkUser)) {
  1340. networkUserGroupID := GetDefaultNetworkUserGroupID(netID)
  1341. defaultUserAcl := models.Acl{
  1342. ID: fmt.Sprintf("%s.%s-grp", netID, models.NetworkUser),
  1343. Name: "Network User",
  1344. MetaData: "This Policy allows all network users to communicate with all gateways",
  1345. Default: true,
  1346. ServiceType: models.Any,
  1347. NetworkID: netID,
  1348. Proto: models.ALL,
  1349. RuleType: models.UserPolicy,
  1350. Src: []models.AclPolicyTag{
  1351. {
  1352. ID: models.UserGroupAclID,
  1353. Value: globalNetworksUserGroupID.String(),
  1354. },
  1355. {
  1356. ID: models.UserGroupAclID,
  1357. Value: networkUserGroupID.String(),
  1358. },
  1359. },
  1360. Dst: []models.AclPolicyTag{
  1361. {
  1362. ID: models.NodeTagID,
  1363. Value: fmt.Sprintf("%s.%s", netID, models.GwTagName),
  1364. }},
  1365. AllowedDirection: models.TrafficDirectionUni,
  1366. Enabled: true,
  1367. CreatedBy: "auto",
  1368. CreatedAt: time.Now().UTC(),
  1369. }
  1370. logic.InsertAcl(defaultUserAcl)
  1371. }
  1372. }
  1373. func GetUserGroupsInNetwork(netID models.NetworkID) (networkGrps map[models.UserGroupID]models.UserGroup) {
  1374. groups, _ := ListUserGroups()
  1375. networkGrps = make(map[models.UserGroupID]models.UserGroup)
  1376. for _, grp := range groups {
  1377. if _, ok := grp.NetworkRoles[models.AllNetworks]; ok {
  1378. networkGrps[grp.ID] = grp
  1379. continue
  1380. }
  1381. if _, ok := grp.NetworkRoles[netID]; ok {
  1382. networkGrps[grp.ID] = grp
  1383. }
  1384. }
  1385. return
  1386. }
  1387. func AddGlobalNetRolesToAdmins(u *models.User) {
  1388. if u.PlatformRoleID != models.SuperAdminRole && u.PlatformRoleID != models.AdminRole {
  1389. return
  1390. }
  1391. if len(u.UserGroups) == 0 {
  1392. u.UserGroups = make(map[models.UserGroupID]struct{})
  1393. }
  1394. u.UserGroups[globalNetworksAdminGroupID] = struct{}{}
  1395. }
  1396. func GetUserGrpMap() map[models.UserGroupID]map[string]struct{} {
  1397. grpUsersMap := make(map[models.UserGroupID]map[string]struct{})
  1398. users, _ := logic.GetUsersDB()
  1399. for _, user := range users {
  1400. for gID := range user.UserGroups {
  1401. if grpUsers, ok := grpUsersMap[gID]; ok {
  1402. grpUsers[user.UserName] = struct{}{}
  1403. grpUsersMap[gID] = grpUsers
  1404. } else {
  1405. grpUsersMap[gID] = make(map[string]struct{})
  1406. grpUsersMap[gID][user.UserName] = struct{}{}
  1407. }
  1408. }
  1409. }
  1410. return grpUsersMap
  1411. }