fix: enforce role hierarchy on permission edits
This commit is contained in:
@@ -1212,6 +1212,7 @@ class RbacAdminServiceImpl(IRbacAdminService):
|
||||
routeIds = sorted(set(Body.route_ids))
|
||||
async with GetAsyncSession() as Session:
|
||||
await self._getAdminRouteMap(Session, EnsureSeeds=True)
|
||||
await self._assertCanManageTargetRole(Session, CurrentUserId, RoleId)
|
||||
result = await self._updateRoleRoutesInSession(Session, RoleId, routeIds, Body.permission)
|
||||
await Session.commit()
|
||||
PermissionServiceImpl.InvalidateAll()
|
||||
@@ -1243,6 +1244,7 @@ class RbacAdminServiceImpl(IRbacAdminService):
|
||||
await self._assertManageAndPermission(CurrentUserId, "rbac:role_permissions:write")
|
||||
async with GetAsyncSession() as Session:
|
||||
await self._getAdminRouteMap(Session, EnsureSeeds=True)
|
||||
await self._assertCanManageTargetRole(Session, CurrentUserId, Body.role_id)
|
||||
await self._saveRolePermissionsInSession(Session, Body.role_id, Body.permissions, Body.replace, Body.replace_scope_permission_ids)
|
||||
await Session.commit()
|
||||
PermissionServiceImpl.InvalidateAll()
|
||||
@@ -1259,6 +1261,7 @@ class RbacAdminServiceImpl(IRbacAdminService):
|
||||
]
|
||||
async with GetAsyncSession() as Session:
|
||||
await self._getAdminRouteMap(Session, EnsureSeeds=True)
|
||||
await self._assertCanManageTargetRole(Session, CurrentUserId, RoleId)
|
||||
routeResult = await self._updateRoleRoutesInSession(Session, RoleId, routeIds, Body.route_permission)
|
||||
await self._saveRolePermissionsInSession(Session, RoleId, permissionConfigs, True, Body.replace_scope_permission_ids)
|
||||
permissionResult = await self._getRolePermissionsInSession(Session, RoleId)
|
||||
@@ -1539,7 +1542,8 @@ class RbacAdminServiceImpl(IRbacAdminService):
|
||||
{tenant_name_select},
|
||||
COALESCE(bool_or(r.role_key IN ('super_admin', 'provincial_admin')), FALSE) AS is_global,
|
||||
COALESCE(bool_or(r.role_key IN ('super_admin', 'provincial_admin', 'admin')), FALSE) AS can_manage,
|
||||
COALESCE(bool_or(r.role_key = 'super_admin'), FALSE) AS is_super_admin
|
||||
COALESCE(bool_or(r.role_key = 'super_admin'), FALSE) AS is_super_admin,
|
||||
COALESCE(MAX(COALESCE(r.priority, 0)), 0) AS max_role_priority
|
||||
FROM sso_users u
|
||||
LEFT JOIN user_role ur ON ur.user_id = u.id
|
||||
LEFT JOIN roles r ON r.id = ur.role_id
|
||||
@@ -1566,6 +1570,7 @@ class RbacAdminServiceImpl(IRbacAdminService):
|
||||
"is_global": bool(row["is_global"]),
|
||||
"can_manage": bool(row["can_manage"]),
|
||||
"is_super_admin": bool(row["is_super_admin"]),
|
||||
"max_role_priority": int(row["max_role_priority"] or 0),
|
||||
}
|
||||
|
||||
async def _resolve_requested_scope(self, Session, RawValue: str | None) -> dict[str, str]:
|
||||
@@ -1869,6 +1874,21 @@ class RbacAdminServiceImpl(IRbacAdminService):
|
||||
raise LeauditException(StatusCodeEnum.HTTP_404_NOT_FOUND, "角色不存在")
|
||||
return row
|
||||
|
||||
async def _assertCanManageTargetRole(self, Session, CurrentUserId: int, RoleId: int) -> None:
|
||||
"""禁止低优先级角色编辑更高优先级角色的权限配置。"""
|
||||
context = await self._getCurrentUserContext(CurrentUserId)
|
||||
if context.get("is_super_admin"):
|
||||
return
|
||||
|
||||
target = await self._getRoleRow(Session, RoleId)
|
||||
currentPriority = int(context.get("max_role_priority") or 0)
|
||||
targetPriority = int(target.get("priority") or 0)
|
||||
if targetPriority > currentPriority:
|
||||
raise LeauditException(
|
||||
StatusCodeEnum.HTTP_403_FORBIDDEN,
|
||||
"下级角色不可编辑上级角色的路由权限",
|
||||
)
|
||||
|
||||
def _toRoleVo(self, Row) -> RoleVO:
|
||||
"""角色记录转 VO。"""
|
||||
return RoleVO(
|
||||
|
||||
+1
-1
Submodule legal-platform-frontend updated: ab61283579...1c20a2989f
@@ -216,6 +216,41 @@ def test_rbac_seed_cache_reuses_recent_route_map():
|
||||
assert service._get_cached_admin_seed_route_map() == route_map
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_rbac_rejects_lower_role_editing_higher_role_permissions(monkeypatch):
|
||||
service = RbacAdminServiceImpl()
|
||||
|
||||
async def fake_context(_current_user_id):
|
||||
return {"is_super_admin": False, "max_role_priority": 50}
|
||||
|
||||
async def fake_role_row(_session, _role_id):
|
||||
return {"id": 1, "priority": 90}
|
||||
|
||||
monkeypatch.setattr(service, "_getCurrentUserContext", fake_context)
|
||||
monkeypatch.setattr(service, "_getRoleRow", fake_role_row)
|
||||
|
||||
with pytest.raises(LeauditException) as exc:
|
||||
await service._assertCanManageTargetRole(None, 100, 1)
|
||||
|
||||
assert exc.value.status == StatusCodeEnum.HTTP_403_FORBIDDEN
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_rbac_allows_super_admin_editing_higher_role_permissions(monkeypatch):
|
||||
service = RbacAdminServiceImpl()
|
||||
|
||||
async def fake_context(_current_user_id):
|
||||
return {"is_super_admin": True, "max_role_priority": 100}
|
||||
|
||||
async def fail_if_called(_session, _role_id):
|
||||
raise AssertionError("super_admin should bypass target role lookup")
|
||||
|
||||
monkeypatch.setattr(service, "_getCurrentUserContext", fake_context)
|
||||
monkeypatch.setattr(service, "_getRoleRow", fail_if_called)
|
||||
|
||||
await service._assertCanManageTargetRole(None, 100, 1)
|
||||
|
||||
|
||||
def test_permission_cache_is_shared_and_can_invalidate_user():
|
||||
first = PermissionServiceImpl()
|
||||
second = PermissionServiceImpl()
|
||||
|
||||
Reference in New Issue
Block a user