SpringBoot淺析安全管理之高級配置

更新時間：2022年08月12日 15:33:57 作者：一只小熊貓呀

安全管理是軟件系統(tǒng)必不可少的的功能。根據(jù)經(jīng)典的“墨菲定律”——凡是可能，總會發(fā)生。如果系統(tǒng)存在安全隱患，最終必然會出現(xiàn)問題，這篇文章主要介紹了SpringBoot安全管理之高級配置

角色繼承

SpringBoot淺析安全管理之基于數(shù)據(jù)庫認(rèn)證中定義了三種角色，但是這三種角色之間不具備任何關(guān)系，一般來說角色之間是有關(guān)系的，例如 ROLE_admin 一般既具有 admin 權(quán)限，又具有 user 權(quán)限。那么如何配置這種角色繼承關(guān)系呢？只需要開發(fā)者在 Spring Security 的配置類中提供一個 RoleHierarchy 即可

@Bean
RoleHierarchy roleHierarchy() {
    RoleHierarchyImpl roleHierarchy = new RoleHierarchyImpl();
    String hierarchy = "ROLE_dba > ROLE_admin ROLE_admin > ROLE_user";
    roleHierarchy.setHierarchy(hierarchy);
    return roleHierarchy;
}

配置完 RoleHierarchy 之后，具有 ROLE_dba 角色的用戶就可以訪問所有資源了，具有 ROLE_admin 角色的用戶也可以訪問具有 ROLE_user 角色才能訪問的資源。

動態(tài)權(quán)限配置

使用 HttpSecurity 配置的認(rèn)證授權(quán)規(guī)則還是不夠靈活，無法實現(xiàn)資源和角色之間的動態(tài)調(diào)整，要實現(xiàn)動態(tài)配置 URL 權(quán)限，需要開發(fā)者自定義權(quán)限配置，配置步驟如下傳送門

1. 數(shù)據(jù)庫設(shè)計

在 10.2節(jié) 數(shù)據(jù)庫的基礎(chǔ)上再增加一張資源表和資源角色關(guān)聯(lián)表，資源表中定義了用戶能夠訪問的 URL 模式，資源角色表則定義了訪問該模式的 URL 需要什么樣的角色

建表語句

CREATE TABLE `menu` (
  `id` int(11) NOT NULL,
  `pattern` varchar(255) DEFAULT NULL,
  PRIMARY KEY (`id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8;
CREATE TABLE `menu_role` (
  `id` int(11) NOT NULL,
  `mid` int(11) DEFAULT NULL,
  `rid` int(11) DEFAULT NULL,
  PRIMARY KEY (`id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8;

初始化數(shù)據(jù)

INSERT INTO `menu`(`id`, `pattern`) VALUES (1, '/db/**');
INSERT INTO `menu`(`id`, `pattern`) VALUES (2, '/admin/**');
INSERT INTO `menu`(`id`, `pattern`) VALUES (3, '/user/**');
INSERT INTO `menu_role`(`id`, `mid`, `rid`) VALUES (1, 1, 1);
INSERT INTO `menu_role`(`id`, `mid`, `rid`) VALUES (2, 2, 2);
INSERT INTO `menu_role`(`id`, `mid`, `rid`) VALUES (3, 3, 3);

Menu 實體類

public class Menu {
    private Integer id;
    private String pattern;
    private List<Role> roles;
    public List<Role> getRoles() {
        return roles;
    }
    public void setRoles(List<Role> roles) {
        this.roles = roles;
    }
    public Integer getId() {
        return id;
    }
    public void setId(Integer id) {
        this.id = id;
    }
    public String getPattern() {
        return pattern;
    }
    public void setPattern(String pattern) {
        this.pattern = pattern;
    }
}

MenuMapper

@Mapper
public interface MenuMapper {
    List<Menu> getAllMenus();
}

MenuMapper.xml

<?xml version="1.0" encoding="UTF-8" ?>
<!DOCTYPE mapper
        PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
        "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
<mapper namespace="org.sang.mapper.MenuMapper">
    <resultMap id="BaseResultMap" type="org.sang.model.Menu">
        <id property="id" column="id"/>
        <result property="pattern" column="pattern"/>
        <collection property="roles" ofType="org.sang.model.Role">
            <id property="id" column="rid"/>
            <result property="name" column="rname"/>
            <result property="nameZh" column="rnameZh"/>
        </collection>
    </resultMap>
    <select id="getAllMenus" resultMap="BaseResultMap">
        SELECT m.*,r.id AS rid,r.name AS rname,r.nameZh AS rnameZh FROM menu m LEFT JOIN menu_role mr ON m.`id`=mr.`mid` LEFT JOIN role r ON mr.`rid`=r.`id`
    </select>
</mapper>

2. 自定義FilterInvocationSecurityMetadataSource

Spring Security 中通過 FilterInvocationSecurityMetadataSource 接口中的 getAttributes 方法來確定一個請求需要哪些角色，F(xiàn)ilterInvocationSecurityMetadataSource 接口默認(rèn)實現(xiàn)類是 DefaultFilterInvocationSecurityMetadataSource ，參考 DefaultFilterInvocationSecurityMetadataSource 的實現(xiàn)，開發(fā)者可以定義自己的 FilterInvocationSecurityMetadataSource ，如下：

@Component
public class CustomFilterInvocationSecurityMetadataSource
        implements FilterInvocationSecurityMetadataSource {
    AntPathMatcher antPathMatcher = new AntPathMatcher();
    @Autowired
    MenuMapper menuMapper;
    @Override
    public Collection<ConfigAttribute> getAttributes(Object object)
            throws IllegalArgumentException {
        String requestUrl = ((FilterInvocation) object).getRequestUrl();
        List<Menu> allMenus = menuMapper.getAllMenus();
        for (Menu menu : allMenus) {
            if (antPathMatcher.match(menu.getPattern(), requestUrl)) {
                List<Role> roles = menu.getRoles();
                String[] roleArr = new String[roles.size()];
                for (int i = 0; i < roleArr.length; i++) {
                    roleArr[i] = roles.get(i).getName();
                }
                return SecurityConfig.createList(roleArr);
            }
        }
        return SecurityConfig.createList("ROLE_LOGIN");
    }
    @Override
    public Collection<ConfigAttribute> getAllConfigAttributes() {
        return null;
    }
    @Override
    public boolean supports(Class<?> clazz) {
        return FilterInvocation.class.isAssignableFrom(clazz);
    }
}

代碼解釋：

開發(fā)者自定義 FilterInvocationSecurityMetadataSource 主要實現(xiàn)接口中的 getAttributes 方法，該方法的參數(shù)是一個 FilterInvocation ，開發(fā)者可以從 FilterInvocation 中提取當(dāng)前請求的 URL ，返回值是 Collection，表示當(dāng)前請求 URL 所需角色
創(chuàng)建一個 AntPathMatcher 主要用來實現(xiàn) ant 風(fēng)格的 URL 匹配
((FilterInvocation) object).getRequestUrl(); 從參數(shù)中提取當(dāng)前請求的 URL
menuMapper.getAllMenus(); 從數(shù)據(jù)庫中獲取所有的資源信息，即 menu 表以及 menu 所對應(yīng)的 role，在真實項目環(huán)境中，開發(fā)者可以將資源信息緩存在 Redis 或者其他緩存數(shù)據(jù)庫中
然后遍歷資源信息，遍歷過程中獲取當(dāng)前請求的 URL 所需要的角色信息并返回。如果當(dāng)前請求的 URL 在資源表中不存在相應(yīng)的模式，就假設(shè)該請求登錄后即可訪問，即直接返回 ROLE_LOGIN
getAllConfigAttributes 方法用來返回所有定義好的權(quán)限資源，Spring Security 在啟動時會校驗相關(guān)配置是否正確，如果不需要校驗，那么該方法直接返回 null 即可
supports 方法返回類對象是否支持校驗

3. 自定義 AccessDecisionManager

當(dāng)一個請求走完 FilterInvocationSecurityMetadataSource 中的 getAttributes 方法后，接下來就會來到 AccessDecisionManager 類中進(jìn)行角色信息的比對，自定義 AccessDecisionManager 如下：

@Component
public class CustomAccessDecisionManager implements AccessDecisionManager {
    @Override
    public void decide(Authentication auth,
                       Object object,
                       Collection<ConfigAttribute> ca) {
        Collection<? extends GrantedAuthority> auths = auth.getAuthorities();
        for (ConfigAttribute configAttribute : ca) {
            if ("ROLE_LOGIN".equals(configAttribute.getAttribute())
                    && auth instanceof UsernamePasswordAuthenticationToken) {
                return;
            }
            for (GrantedAuthority authority : auths) {
                if (configAttribute.getAttribute().equals(authority.getAuthority())) {
                    return;
                }
            }
        }
        throw new AccessDeniedException("權(quán)限不足");
    }
    @Override
    public boolean supports(ConfigAttribute attribute) {
        return true;
    }
    @Override
    public boolean supports(Class<?> clazz) {
        return true;
    }
}

代碼解釋：

自定義 AccessDecisionManager 并重寫 decide 方法，在該方法中判斷當(dāng)前登錄的用戶是否具備當(dāng)前請求 URL 所需要的角色信息，如果不具備，就拋出 AccessDeniedException 異常，否則不做任何事情
decide 有三個參數(shù)，第一個參數(shù)包含當(dāng)前登錄用戶的信息；第二個參數(shù)則是一個 FilterInvocation 對象，可以獲取當(dāng)前請求對象等；第三個參數(shù)就是 UsernamePasswordAuthenticationToken 的實例，說明當(dāng)前用戶已登錄，該方法到此結(jié)束，否則進(jìn)入正常的判斷流程，如果當(dāng)前用戶具備當(dāng)前請求需要的角色，那么方法結(jié)束

4. 配置

最后，在 Spring Security 中配置上邊的兩個自定義類，代碼如下：

@Configuration
public class WebSecurityConfig extends WebSecurityConfigurerAdapter {
    @Autowired
    UserService userService;
    @Bean
    PasswordEncoder passwordEncoder() {
        return new BCryptPasswordEncoder();
    }
    @Override
    protected void configure(AuthenticationManagerBuilder auth) throws Exception {
        auth.userDetailsService(userService);
    }
    @Bean
    RoleHierarchy roleHierarchy() {
        RoleHierarchyImpl roleHierarchy = new RoleHierarchyImpl();
        String hierarchy = "ROLE_dba > ROLE_admin ROLE_admin > ROLE_user";
        roleHierarchy.setHierarchy(hierarchy);
        return roleHierarchy;
    }
    @Override
    protected void configure(HttpSecurity http) throws Exception {
        http.authorizeRequests()
                .withObjectPostProcessor(new ObjectPostProcessor<FilterSecurityInterceptor>() {
                    @Override
                    public <O extends FilterSecurityInterceptor> O postProcess(O object) {
                        object.setSecurityMetadataSource(cfisms());
                        object.setAccessDecisionManager(cadm());
                        return object;
                    }
                })
                .and()
                .formLogin()
                .loginProcessingUrl("/login").permitAll()
                .and()
                .csrf().disable();
    }
    @Bean
    CustomFilterInvocationSecurityMetadataSource cfisms() {
        return new CustomFilterInvocationSecurityMetadataSource();
    }
    @Bean
    CustomAccessDecisionManager cadm() {
        return new CustomAccessDecisionManager();
    }
}

代碼解釋：

此處 WebSecurityConfig 類的定義是對 10.2節(jié) 中 WebSecurityConfig 定義的補充，主要修改了 configure(HttpSecurity http) 方法的實現(xiàn)并添加了兩個 Bean
object.setSecurityMetadataSource(cfisms()); object.setAccessDecisionManager(cadm());將定義的兩個實例設(shè)置進(jìn)去

到此這篇關(guān)于SpringBoot淺析安全管理之高級配置的文章就介紹到這了,更多相關(guān)SpringBoot高級配置內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: