用户访问控制时校验数据权限，防止越权

RuoYi

2022-01-27 b49cdbbc9516352d1f4511e4223fa9d089b057b9

 src/main/java/com/ruoyi/project/system/service/impl/SysRoleServiceImpl.java

@@ -10,12 +10,14 @@
import org.springframework.transaction.annotation.Transactional;

import com.ruoyi.common.constant.UserConstants;

import com.ruoyi.common.exception.ServiceException;

import com.ruoyi.common.utils.SecurityUtils;

import com.ruoyi.common.utils.StringUtils;

import com.ruoyi.common.utils.spring.SpringUtils;

import com.ruoyi.framework.aspectj.lang.annotation.DataScope;

import com.ruoyi.project.system.domain.SysRole;

import com.ruoyi.project.system.domain.SysRoleDept;

import com.ruoyi.project.system.domain.SysRoleMenu;

import com.ruoyi.project.system.domain.SysUser;

import com.ruoyi.project.system.domain.SysUserRole;

import com.ruoyi.project.system.mapper.SysRoleDeptMapper;

import com.ruoyi.project.system.mapper.SysRoleMapper;

@@ -120,7 +122,7 @@
     * @return 选中角色ID列表

     */

    @Override

    public List<Integer> selectRoleListByUserId(Long userId)

    public List<Long> selectRoleListByUserId(Long userId)

    {

        return roleMapper.selectRoleListByUserId(userId);

    }

@@ -184,6 +186,26 @@
        if (StringUtils.isNotNull(role.getRoleId()) && role.isAdmin())

        {

            throw new ServiceException("不允许操作超级管理员角色");

        }

    }



    /**

     * 校验角色是否有数据权限

     * 
     * @param roleId 角色id

     */

    @Override

    public void checkRoleDataScope(Long roleId)

    {

        if (!SysUser.isAdmin(SecurityUtils.getUserId()))

        {

            SysRole role = new SysRole();

            role.setRoleId(roleId);

            List<SysRole> roles = SpringUtils.getAopProxy(this).selectRoleList(role);

            if (StringUtils.isEmpty(roles))

            {

                throw new ServiceException("没有权限访问角色数据！");

            }

        }

    }



@@ -339,6 +361,7 @@
        for (Long roleId : roleIds)

        {

            checkRoleAllowed(new SysRole(roleId));

            checkRoleDataScope(roleId);

            SysRole role = selectRoleById(roleId);

            if (countUserRoleByRoleId(roleId) > 0)

            {