修复两处存在SQL注入漏洞问题

RuoYi

2021-05-27 af479c78385fc3ebdfd345ae12a42858b0f79d2e

 ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysMenuServiceImpl.java

@@ -8,18 +8,19 @@
import java.util.List;

import java.util.Set;

import java.util.stream.Collectors;



import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.stereotype.Service;



import com.ruoyi.common.core.constant.UserConstants;

import com.ruoyi.common.core.utils.SecurityUtils;

import com.ruoyi.common.core.utils.StringUtils;

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

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

import com.ruoyi.system.domain.SysMenu;

import com.ruoyi.system.domain.vo.MetaVo;

import com.ruoyi.system.domain.vo.RouterVo;

import com.ruoyi.system.domain.vo.TreeSelect;

import com.ruoyi.system.mapper.SysMenuMapper;

import com.ruoyi.system.mapper.SysRoleMapper;

import com.ruoyi.system.mapper.SysRoleMenuMapper;

import com.ruoyi.system.service.ISysMenuService;



@@ -35,6 +36,9 @@


    @Autowired

    private SysMenuMapper menuMapper;



    @Autowired

    private SysRoleMapper roleMapper;



    @Autowired

    private SysRoleMenuMapper roleMenuMapper;

@@ -105,7 +109,7 @@
    public List<SysMenu> selectMenuTreeByUserId(Long userId)

    {

        List<SysMenu> menus = null;

        if (SysUser.isAdmin(userId))

        if (SecurityUtils.isAdmin(userId))

        {

            menus = menuMapper.selectMenuTreeAll();

        }

@@ -122,9 +126,11 @@
     * @param roleId 角色ID

     * @return 选中菜单列表

     */

    @Override

    public List<Integer> selectMenuListByRoleId(Long roleId)

    {

        return menuMapper.selectMenuListByRoleId(roleId);

        SysRole role = roleMapper.selectRoleById(roleId);

        return menuMapper.selectMenuListByRoleId(roleId, role.isMenuCheckStrictly());

    }



    /**

@@ -141,16 +147,28 @@
        {

            RouterVo router = new RouterVo();

            router.setHidden("1".equals(menu.getVisible()));

            router.setName(StringUtils.capitalize(menu.getPath()));

            router.setName(getRouteName(menu));

            router.setPath(getRouterPath(menu));

            router.setComponent(StringUtils.isEmpty(menu.getComponent()) ? "Layout" : menu.getComponent());

            router.setMeta(new MetaVo(menu.getMenuName(), menu.getIcon()));

            router.setComponent(getComponent(menu));

            router.setMeta(new MetaVo(menu.getMenuName(), menu.getIcon(), StringUtils.equals("1", menu.getIsCache())));

            List<SysMenu> cMenus = menu.getChildren();

            if (!cMenus.isEmpty() && cMenus.size() > 0 && "M".equals(menu.getMenuType()))

            if (!cMenus.isEmpty() && cMenus.size() > 0 && UserConstants.TYPE_DIR.equals(menu.getMenuType()))

            {

                router.setAlwaysShow(true);

                router.setRedirect("noRedirect");

                router.setChildren(buildMenus(cMenus));

            }

            else if (isMenuFrame(menu))

            {

                router.setMeta(null);

                List<RouterVo> childrenList = new ArrayList<RouterVo>();

                RouterVo children = new RouterVo();

                children.setPath(menu.getPath());

                children.setComponent(menu.getComponent());

                children.setName(StringUtils.capitalize(menu.getPath()));

                children.setMeta(new MetaVo(menu.getMenuName(), menu.getIcon(), StringUtils.equals("1", menu.getIsCache())));

                childrenList.add(children);

                router.setChildren(childrenList);

            }

            routers.add(router);

        }

@@ -167,14 +185,19 @@
    public List<SysMenu> buildMenuTree(List<SysMenu> menus)

    {

        List<SysMenu> returnList = new ArrayList<SysMenu>();

        List<Long> tempList = new ArrayList<Long>();

        for (SysMenu dept : menus)

        {

            tempList.add(dept.getMenuId());

        }

        for (Iterator<SysMenu> iterator = menus.iterator(); iterator.hasNext();)

        {

            SysMenu t = (SysMenu) iterator.next();

            // 根据传入的某个父节点ID,遍历该父节点的所有子节点

            if (t.getParentId() == 0)

            SysMenu menu = (SysMenu) iterator.next();

            // 如果是顶级节点, 遍历该父节点的所有子节点

            if (!tempList.contains(menu.getParentId()))

            {

                recursionFn(menus, t);

                returnList.add(t);

                recursionFn(menus, menu);

                returnList.add(menu);

            }

        }

        if (returnList.isEmpty())

@@ -290,6 +313,23 @@
    }



    /**

     * 获取路由名称

     * 
     * @param menu 菜单信息

     * @return 路由名称

     */

    public String getRouteName(SysMenu menu)

    {

        String routerName = StringUtils.capitalize(menu.getPath());

        // 非外链并且是一级目录（类型为目录）

        if (isMenuFrame(menu))

        {

            routerName = StringUtils.EMPTY;

        }

        return routerName;

    }



    /**

     * 获取路由地址

     * 

     * @param menu 菜单信息

@@ -298,12 +338,61 @@
    public String getRouterPath(SysMenu menu)

    {

        String routerPath = menu.getPath();

        // 非外链并且是一级目录

        if (0 == menu.getParentId() && "1".equals(menu.getIsFrame()))

        // 非外链并且是一级目录（类型为目录）

        if (0 == menu.getParentId().intValue() && UserConstants.TYPE_DIR.equals(menu.getMenuType())

                && UserConstants.NO_FRAME.equals(menu.getIsFrame()))

        {

            routerPath = "/" + menu.getPath();

        }

        // 非外链并且是一级目录（类型为菜单）

        else if (isMenuFrame(menu))

        {

            routerPath = "/";

        }

        return routerPath;

    }



    /**

     * 获取组件信息

     * 
     * @param menu 菜单信息

     * @return 组件信息

     */

    public String getComponent(SysMenu menu)

    {

        String component = UserConstants.LAYOUT;

        if (StringUtils.isNotEmpty(menu.getComponent()) && !isMenuFrame(menu))

        {

            component = menu.getComponent();

        }

        else if (StringUtils.isEmpty(menu.getComponent()) && isParentView(menu))

        {

            component = UserConstants.PARENT_VIEW;

        }

        return component;

    }



    /**

     * 是否为菜单内部跳转

     * 
     * @param menu 菜单信息

     * @return 结果

     */

    public boolean isMenuFrame(SysMenu menu)

    {

        return menu.getParentId().intValue() == 0 && UserConstants.TYPE_MENU.equals(menu.getMenuType())

                && menu.getIsFrame().equals(UserConstants.NO_FRAME);

    }



    /**

     * 是否为parent_view组件

     * 
     * @param menu 菜单信息

     * @return 结果

     */

    public boolean isParentView(SysMenu menu)

    {

        return menu.getParentId().intValue() != 0 && UserConstants.TYPE_DIR.equals(menu.getMenuType());

    }



    /**

@@ -344,13 +433,7 @@
        {

            if (hasChild(list, tChild))

            {

                // 判断是否有子节点

                Iterator<SysMenu> it = childList.iterator();

                while (it.hasNext())

                {

                    SysMenu n = (SysMenu) it.next();

                    recursionFn(list, n);

                }

                recursionFn(list, tChild);

            }

        }

    }