SpringBoot之----了解Shiro安全框架整合Mybatis和thymeleaf，實作授權登陸認證-有解無憂

這里我做一個流程圖幫助大家理解Shiro安全框架整合Mybatis和thymeleaf，實作授權登陸認證的程序，大家先了解，然后看代碼，看完代碼再看流程圖，

在這里插入圖片描述

1.了解完那我們先看專案結構：

專案結構：
在這里插入圖片描述
資料庫截圖：

2.我們再看整個專案的代碼：

pom.xml

<?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 https://maven.apache.org/xsd/maven-4.0.0.xsd">
    <modelVersion>4.0.0</modelVersion>
    <parent>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-parent</artifactId>
        <version>2.6.2</version>
        <relativePath/> <!-- lookup parent from repository -->
    </parent>
    <groupId>com.wlm</groupId>
    <artifactId>shiro-springboot</artifactId>
    <version>0.0.1-SNAPSHOT</version>
    <name>shiro-springboot</name>
    <description>Demo project for Spring Boot</description>
    <properties>
        <java.version>1.8</java.version>
    </properties>
    <dependencies>

        <!--
            我們先看Shiro的核心三大物件
            Subject 用戶
            SecurityManager 管理所有用戶
            Realm 連接資料
        -->
        <dependency>
            <groupId>mysql</groupId>
            <artifactId>mysql-connector-java</artifactId>
        </dependency>
        <dependency>
            <groupId>log4j</groupId>
            <artifactId>log4j</artifactId>
            <version>1.2.17</version>
        </dependency>
        <dependency>
            <groupId>com.alibaba</groupId>
            <artifactId>druid</artifactId>
            <version>1.1.9</version>
        </dependency>
        <dependency>
            <groupId>org.mybatis.spring.boot</groupId>
            <artifactId>mybatis-spring-boot-starter</artifactId>
            <version>2.1.1</version>
        </dependency>
        <dependency>
            <groupId>org.projectlombok</groupId>
            <artifactId>lombok</artifactId>
        </dependency>
        <!--thymeleaf整合-->
        <dependency>
            <groupId>com.github.theborakompanioni</groupId>
            <artifactId>thymeleaf-extras-shiro</artifactId>
            <version>2.0.0</version>
        </dependency>
        <!-- 匯入Shiro整合包-->
        <dependency>
            <groupId>org.apache.shiro</groupId>
            <artifactId>shiro-spring</artifactId>
            <version>1.4.1</version>
        </dependency>
        <dependency>
            <groupId>org.apache.shiro</groupId>
            <artifactId>shiro-core</artifactId>
            <version>1.4.1</version>
        </dependency>

        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-thymeleaf</artifactId>
        </dependency>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-web</artifactId>
        </dependency>

        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-test</artifactId>
            <scope>test</scope>
        </dependency>
    </dependencies>

    <build>
        <plugins>
            <plugin>
                <groupId>org.springframework.boot</groupId>
                <artifactId>spring-boot-maven-plugin</artifactId>
            </plugin>
        </plugins>
    </build>

</project>

application.yml

spring:
  datasource:
    username: root
    password: admin
    url: jdbc:mysql://localhost:3306/wuting?serverTimezone=UTC&useUnicode=true&characterEncoding=utf-8
    driver-class-name: com.mysql.cj.jdbc.Driver
    type: com.alibaba.druid.pool.DruidDataSource # 自定義資料源

    #Spring Boot 默認是不注入這些屬性值的，需要自己系結
    #druid 資料源專有配置
    initialSize: 5
    minIdle: 5
    maxActive: 20
    maxWait: 60000
    timeBetweenEvictionRunsMillis: 60000
    minEvictableIdleTimeMillis: 300000
    validationQuery: SELECT 1 FROM DUAL
    testWhileIdle: true
    testOnBorrow: false
    testOnReturn: false
    poolPreparedStatements: true

    #配置監控統計攔截的filters，stat:監控統計、log4j：日志記錄、wall：防御sql注入
    #如果允許時報錯  java.lang.ClassNotFoundException: org.apache.log4j.Priority
    #則匯入 log4j 依賴即可，Maven 地址：https://mvnrepository.com/artifact/log4j/log4j
    filters: stat,wall,log4j
    maxPoolPreparedStatementPerConnectionSize: 20
    useGlobalDataSourceStat: true
    connectionProperties: druid.stat.mergeSql=true;druid.stat.slowSqlMillis=500

application.properties

mybatis.type-aliases-package=com.wlm.pojo
mybatis.mapper-locations=classpath:mapper/*.xml

<!DOCTYPE html>
<html lang="en" xmlns:th="http://www.thymeleaf.org">
<head>
    <meta charset="UTF-8">
    <title>Title</title>
</head>
<body>
<h1>登陸</h1>
<hr>
<p th:text="${msg}" style="color: red"></p>
<form th:action="@{/login}">
    <p>賬  號：<input type="text" name="username"></p>
    <p>密  碼：<input type="text" name="password"></p>
    <p><input type="submit"></p>
</form>
</body>
</html>

index.html

<!DOCTYPE html>
<html lang="en" xmlns:th="http://www.thymeleaf.org"
      xmlns:shiro="http://www.thymeleaf.org/thymeleaf-extras-shiro">
<head>
    <meta charset="UTF-8">
    <title>Title</title>
</head>
<body>

<h1>首頁</h1>
<!--shiro:guest=""也是有效滴，但是我們還是更需要了解session中判斷值！-->
<div th:if="${session.loginUser==null}">
    <a th:href="@{/toLogin}">登陸</a>
</div>
<p th:text="${msg}"></p>
<hr>
<div shiro:hasPermission="user:add">
    <a th:href="@{/user/add}">add</a>
</div>
<div shiro:hasPermission="user:update">
    <a th:href="@{/user/update}">update</a>
</div>
</body>
</html>

update.html

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>Title</title>
</head>
<body>

<h1>update</h1>

</body>
</html>

add.html

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>Title</title>
</head>
<body>

<h1>add</h1>

</body>
</html>

UserMapper.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">
<!--namespace=系結一個對應的Dao/Mapper介面-->
<mapper namespace="com.wlm.mapper.UserMapper">
    <!--    查詢陳述句 id相當與要重寫的方法名-->
    <select id="queryUserByName" resultType="User" parameterType="String">
        select * from wuting.wt where username = #{username}
    </select>

</mapper>

ShiroConfig.java

package com.wlm.config;

import at.pollux.thymeleaf.shiro.dialect.ShiroDialect;
import org.apache.shiro.spring.web.ShiroFilterFactoryBean;
import org.apache.shiro.web.mgt.DefaultWebSecurityManager;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import java.util.LinkedHashMap;
import java.util.Map;

//加上@Configuration注解主要是給我們的類加上了cglib代理，
//在執行我們的配置類的方法時，會執行cglib代理類中的方法
@Configuration
public class ShiroConfig {
    //shiroFilterFactoryBean：第三步
    @Bean
    public ShiroFilterFactoryBean getShiroFilterFactoryBean(@Qualifier("securityManager") DefaultWebSecurityManager defaultWebSecurityManager){
        ShiroFilterFactoryBean bean = new ShiroFilterFactoryBean();
        //設定安全管理器
        bean.setSecurityManager(defaultWebSecurityManager);
        //添加shiro的內置過濾器
        /*
            anon: 無需認證就可以訪問
            authc: 必須認證了才能訪問
            user: 必須擁有 記住我 功能才能用
            perms: 擁有對某個資源的權限才能訪問檔案
            role: 擁有某個角色權限才能訪問
        */
        Map<String, String> filterMap = new LinkedHashMap<>();

        //授權，正常的情況下，沒有授權會跳到未授權頁面
        filterMap.put("/user/add","perms[user:add]");
        filterMap.put("/user/update","perms[user:update]");

        //攔截
        filterMap.put("/user/*","authc");
        bean.setFilterChainDefinitionMap(filterMap);

        //設定登陸的請求
        bean.setLoginUrl("/toLogin");

        //未授權頁面
        bean.setUnauthorizedUrl("/noauth");
        return bean;
    }
    //DefaultWebSecurityManager：第二步
    @Bean(name="securityManager")
    public DefaultWebSecurityManager getdefaultWebSecurityManager(@Qualifier("userRealm") UserRealm userRealm){
        DefaultWebSecurityManager securityManager = new DefaultWebSecurityManager();
        //關聯userRealm
        securityManager.setRealm(userRealm);
        return securityManager;
    }

    //創建realm物件：第一步
    @Bean
    public UserRealm userRealm(){
        //需要自定義，交給spring托管
        return new UserRealm();
    }
    //整合ShiroDialect：用來整合shiro thymeleaf
    @Bean
    public ShiroDialect getShiroDialect(){
        return new ShiroDialect();
    }
}

UserRealm.java

package com.wlm.config;

import com.wlm.pojo.User;
import com.wlm.service.UserServiceImpl;
import org.apache.shiro.SecurityUtils;
import org.apache.shiro.authc.*;
import org.apache.shiro.authz.AuthorizationInfo;
import org.apache.shiro.authz.SimpleAuthorizationInfo;
import org.apache.shiro.realm.AuthorizingRealm;
import org.apache.shiro.session.Session;
import org.apache.shiro.subject.PrincipalCollection;
import org.apache.shiro.subject.Subject;
import org.springframework.beans.factory.annotation.Autowired;

//自定義的 UserRealm，繼承AuthorizingRealm，然后重寫授權和認證方法，跟SpringSecurity很像的，
public class UserRealm extends AuthorizingRealm {

    @Autowired
    UserServiceImpl userService;
    //授權
    @Override
    protected AuthorizationInfo doGetAuthorizationInfo(PrincipalCollection principals) {
        System.out.println("doGetAuthorizationInfo授權=》咸魚_翻身");
        SimpleAuthorizationInfo info = new SimpleAuthorizationInfo();
        //info.addStringPermission("user:add");
        //拿到當前登陸的這個物件
        Subject subject = SecurityUtils.getSubject();
        User currentUser = (User) subject.getPrincipal();
        //設定當前用戶的權限
        info.addStringPermission(currentUser.getPerms());
        return info;
    }
    //認證
    @Override
    protected AuthenticationInfo doGetAuthenticationInfo(AuthenticationToken token) throws AuthenticationException {
        System.out.println("doGetAuthenticationInfo認證=》咸魚_翻身");

        UsernamePasswordToken userToken = (UsernamePasswordToken) token;

        //用戶名，密碼，在資料庫中取
        User user = userService.queryUserByName(userToken.getUsername());
        if (user==null){//沒有這個人的話
            return null;
        }
        Subject currentSubject = SecurityUtils.getSubject();
        Session session = currentSubject.getSession();
        session.setAttribute("loginUser",user);
        //密碼認證不用咱們做，交給shiro做就好了
        //我們得知道他是MD5加密或者MD5鹽值加密的（鹽值加密就是在MD5的基礎上面加一些用戶資訊）
        return new SimpleAuthenticationInfo( user,user.getPassword(),"");
    }
}

MyController.java

package com.wlm.controller;

import org.apache.shiro.SecurityUtils;
import org.apache.shiro.authc.IncorrectCredentialsException;
import org.apache.shiro.authc.UnknownAccountException;
import org.apache.shiro.authc.UsernamePasswordToken;
import org.apache.shiro.subject.Subject;
import org.springframework.stereotype.Controller;
import org.springframework.ui.Model;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.ResponseBody;

@Controller
public class MyController {

    @RequestMapping({"/","/index"})
    public String toIndex(Model model){
        model.addAttribute("msg","hello xian_yu");
        return "index";
    }

    @RequestMapping("/user/add")
    public String add(){
        return "user/add";
    }

    @RequestMapping("/user/update")
    public String update(){
        return "user/update";
    }

    @RequestMapping("/toLogin")
    public String toLogin(){
        return "login";
    }

    @RequestMapping("/login")
    public String login(String username, String password,Model model){
        //獲取當前的用戶
        Subject subject = SecurityUtils.getSubject();
        //封裝用戶的登陸資料
        UsernamePasswordToken token = new UsernamePasswordToken(username, password);
        try{
            //執行登陸的方法，如果沒有例外就說明可以了
            subject.login(token);
            return "index";
        }catch (UnknownAccountException e){//用戶名不存在
            model.addAttribute("msg","用戶名錯誤");
            return "login";
        }catch (IncorrectCredentialsException e){//密碼不存在
            model.addAttribute("msg","密碼錯誤");
            return "login";
        }
    }

    @RequestMapping("/noauth")
    @ResponseBody
    public String unauthorized(){
        return "沒有授權，無法訪問當前頁面";
    }

}

UserMapper介面

package com.wlm.mapper;

import com.wlm.pojo.User;
import org.apache.ibatis.annotations.Mapper;
import org.springframework.stereotype.Repository;

@Repository
@Mapper
public interface UserMapper {
    public User queryUserByName(String name);
}

User.java

package com.wlm.pojo;

import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;

@Data
@AllArgsConstructor
@NoArgsConstructor
public class User {

    private int id;
    private String username;
    private String password;
    private String perms;

}

UserService介面

package com.wlm.service;

import com.wlm.pojo.User;

public interface UserService {
    public User queryUserByName(String name);
}

UserServiceImpl.java

package com.wlm.service;

import com.wlm.mapper.UserMapper;
import com.wlm.pojo.User;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

@Service
public class UserServiceImpl implements UserService {

    @Autowired
    UserMapper userMapper;

    @Override
    public User queryUserByName(String name) {
        return userMapper.queryUserByName(name);
    }
}

3.運行結果：

點擊登陸：
在這里插入圖片描述
那么輸入我們資料庫設計好的賬號：wt 密碼：123456 點擊提交：

不難看出，這里只顯示了add，說明某些用戶獲取的權限和看到界面就是給過濾過的，點擊add就能看見我們寫好給規定用戶看的了：

那么我們換一個賬號密碼呢？賬號：xian_yu 密碼 123456
在這里插入圖片描述
當然我們也只能看見update界面的東西：

這時候我們再去回傳首頁就能看見首頁變成我們的update了：

轉載請註明出處，本文鏈接：https://www.uj5u.com/qita/413277.html

標籤：其他

上一篇：復現CVE-2016-4977——Spring Security OAuth2 遠程命令執行漏洞

下一篇：Qaseem安全團隊- 云函式搭建高質量代理池