Spring Boot Shiro权限管理--自定义 FormAuthenticationFilter验证码整合

转载 2016年08月29日 23:25:44
验证码

思路
shiro使用FormAuthenticationFilter进行表单认证,验证校验的功能应该加在FormAuthenticationFilter中,在认证之前进行验证码校验。

需要写FormAuthenticationFilter的子类,继承FormAuthenticationFilter,改写它的认证方法,在认证之前进行验证码校验。

自定义FormAuthenticationFilter

package com.example.config.shiro;

import javax.servlet.ServletRequest;
import javax.servlet.ServletResponse;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpSession;

import org.apache.shiro.web.filter.authc.FormAuthenticationFilter;

public class CustomFormAuthenticationFilter extends FormAuthenticationFilter{
	@Override  
    protected boolean onAccessDenied(ServletRequest request, ServletResponse response) throws Exception {  
        // 在这里进行验证码的校验  
        HttpServletRequest httpServletRequest = (HttpServletRequest) request;  
        HttpSession session = httpServletRequest.getSession();  
  
        // 取出验证码  
        String validateCode = (String) session.getAttribute("validateCode");  
        // 取出页面的验证码  
        // 输入的验证和session中的验证进行对比  
        String randomcode = httpServletRequest.getParameter("randomcode");  
        if (randomcode != null && validateCode != null && !randomcode.equals(validateCode)) {  
            // 如果校验失败,将验证码错误失败信息,通过shiroLoginFailure设置到request中  
            httpServletRequest.setAttribute("shiroLoginFailure", "kaptchaValidateFailed");//自定义登录异常  
            // 拒绝访问,不再校验账号和密码  
            return true;  
        }  
        return super.onAccessDenied(request, response);  
    }  
}

在ShiroConfiguration.java中的shiroFilter方法注入自定义FormAuthenticationFilter

@Bean
	public ShiroFilterFactoryBean shiroFilter(SecurityManager securityManager) {
		System.out.println("ShiroConfiguration.shiroFilter()");
		ShiroFilterFactoryBean shiroFilterFactoryBean = new ShiroFilterFactoryBean();
		
		Map<String, Filter> filters = shiroFilterFactoryBean.getFilters();//获取filters
		filters.put("authc", new CustomFormAuthenticationFilter());//将自定义 的FormAuthenticationFilter注入shiroFilter中  
		// 必须设置SecuritManager
		shiroFilterFactoryBean.setSecurityManager(securityManager);

		// 拦截器
		Map<String, String> filterChainDefinitionMap = new LinkedHashMap<String, String>();

		// 配置退出过滤器,其中的具体代码Shiro已经替我们实现了
		filterChainDefinitionMap.put("/logout", "logout");
		//验证码可以匿名访问
		filterChainDefinitionMap.put("/validatecodeServlet", "anon");
		
		//配置记住我或认证通过可以访问的地址

        filterChainDefinitionMap.put("/index", "user");
        filterChainDefinitionMap.put("/", "user");
        
		// <!-- 过滤链定义,从上向下顺序执行,一般将 /**放在最为下边 -->:这是一个坑呢,一不小心代码就不好使了;
		// <!-- authc:所有url都必须认证通过才可以访问; anon:所有url都都可以匿名访问-->
		
		filterChainDefinitionMap.put("/**", "authc");

		// 如果不设置默认会自动寻找Web工程根目录下的"/login.jsp"页面
		shiroFilterFactoryBean.setLoginUrl("/login");
		// 登录成功后要跳转的链接
		shiroFilterFactoryBean.setSuccessUrl("/index");
		// 未授权界面;
		shiroFilterFactoryBean.setUnauthorizedUrl("/403");
		
		shiroFilterFactoryBean
				.setFilterChainDefinitionMap(filterChainDefinitionMap);
		return shiroFilterFactoryBean;

	}

ShiroFilterFactoryBean shiroFilterFactoryBean = new ShiroFilterFactoryBean();

Map<String, Filter> filters = shiroFilterFactoryBean.getFilters();//获取filters
filters.put("authc", new CustomFormAuthenticationFilter());//将自定义 的FormAuthenticationFilter注入shiroFilter中  


登录方法加入自定义的异常kaptchaValidateFailed

@RequestMapping(value = "/login", method = RequestMethod.POST)  
    public String login(HttpServletRequest request, Map<String, Object> map) {  
        System.out.println("HomeController.login");  
        // 登录失败从request中获取shiro处理的异常信息  
        // shiroLoginFailure:就是shiro异常类的全类名  
        String exception = (String) request.getAttribute("shiroLoginFailure");  
        String msg = "";  
        if (exception != null) {  
            if (UnknownAccountException.class.getName().equals(exception)) {  
                System.out.println("UnknownAccountException -->帐号不存在:");  
                msg = "UnknownAccountException -->帐号不存在:";  
            } else if (IncorrectCredentialsException.class.getName().equals(exception)) {  
                System.out.println("IncorrectCredentialsException -- > 密码不正确:");  
                msg = "IncorrectCredentialsException -- > 密码不正确:";  
            } else if ("kaptchaValidateFailed".equals(exception)) {  
                System.out.println("kaptchaValidateFailed -- > 验证码错误");  
                msg = "kaptchaValidateFailed -- > 验证码错误";  
            } else {  
                msg = "else >> " + exception;  
                System.out.println("else -- >" + exception);  
            }  
        }  
        map.put("msg", msg);  
        // 此方法不处理登录成功,由shiro进行处理.  
        return "/login";  
    }  
login.html

<!DOCTYPE html>  
<html xmlns="http://www.w3.org/1999/xhtml" xmlns:th="http://www.thymeleaf.org"
      xmlns:sec="http://www.thymeleaf.org/thymeleaf-extras-springsecurity3">
<head>  
<meta charset="UTF-8" />  
<title>Insert title here</title>  
</head>  
<body>  
            错误信息:<h4 th:text="${msg}"></h4>  
       <form action="" method="post">  
           <p>账号:<input type="text" name="username" value="admin"/></p>  
           <p>密码:<input type="text" name="password" value="123456"/></p>  
           <p>验证码:<input type="text" name="randomcode"/>
           <img th:src="@{/validatecodeServlet}" height="20px" width="60px" onclick="random(this)"/></p> 
           <P><input type="checkbox" name="rememberMe" />记住我</P> 
           <p><input type="submit" value="登录"/></p>  
       </form>  
       <script th:inline="javascript">
          function random(tmp){
               tmp.src="/validatecodeServlet?rnd="+Math.random();
          }
       </script>
</body>  
</html> 

验证码Servlet

package com.example.servlet;

import java.awt.Color;
import java.awt.Font;
import java.awt.Graphics;
import java.awt.image.BufferedImage;
import java.io.IOException;
import java.io.OutputStream;
import java.util.Random;

import javax.imageio.ImageIO;
import javax.servlet.ServletException;
import javax.servlet.annotation.WebServlet;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;


@WebServlet(urlPatterns="/validatecodeServlet")
public class ValidatecodeServlet extends HttpServlet{
	 /**
	 * 
	 */
	private static final long serialVersionUID = 1L;

	@Override  
	    protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {  
	        System.out.println(">>>>>>>>>>doGet()<<<<<<<<<<<");  
	        doPost(req, resp);  
	    }  
	  
	    @Override  
	    protected void doPost(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {  
	        System.out.println(">>>>>>>>>>doPost()<<<<<<<<<<<");  
	        int width = 60;
	    	int height = 32;
	    	//create the image
	    	BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
	    	Graphics g = image.getGraphics();
	    	// set the background color
	    	g.setColor(new Color(0xDCDCDC));
	    	g.fillRect(0, 0, width, height);
	    	// draw the border
	    	g.setColor(Color.black);
	    	g.drawRect(0, 0, width - 1, height - 1);
	    	// create a random instance to generate the codes
	    	Random rdm = new Random();
	    	String hash1 = Integer.toHexString(rdm.nextInt());
	    	System.out.print(hash1);
	    	// make some confusion
	    	for (int i = 0; i < 50; i++) {
	    		int x = rdm.nextInt(width);
	    		int y = rdm.nextInt(height);
	    		g.drawOval(x, y, 0, 0);
	    	}
	    	// generate a random code
	    	String capstr = hash1.substring(0, 4);
	    	HttpSession session = req.getSession(true);
	    	//将生成的验证码存入session
	    	session.setAttribute("validateCode", capstr);
	    	g.setColor(new Color(0, 100, 0));
	    	g.setFont(new Font("Candara", Font.BOLD, 24));
	    	g.drawString(capstr, 8, 24);
	    	g.dispose();
	    	//输出图片
	    	resp.setContentType("image/jpeg");
	    	OutputStream strm = resp.getOutputStream();
	    	ImageIO.write(image, "jpeg", strm);
	    	strm.close();   
	       
	    }  
}

Application.java

package com.example;  
  
import org.springframework.boot.SpringApplication;  
import org.springframework.boot.autoconfigure.SpringBootApplication;  
import org.springframework.boot.web.servlet.ServletComponentScan;
@ServletComponentScan 
@SpringBootApplication  
public class Application {  
     public static void main(String[] args) {  
        SpringApplication.run(Application.class, args);  
    }  
} 




相关文章推荐

Spring Boot Shiro 权限管理

本来是打算接着写关于数据库方面,集成MyBatis的,刚好赶上朋友问到Shiro权限管理,就先总结下发出来了。使用Shiro之前用在Spring MVC中,是通过XML文件进行配置。 既然现在在写S...
  • catoop
  • catoop
  • 2016-01-14 23:44
  • 71065

shiro实现系统的退出功能

思路: 1.shiro有许多过滤器,其中就有一个LogoutFilter的过滤器,只需要继承并重写preHandle方法 2.在配置文件中注入自定义实现的过滤器,在此配置当退出的时候应重定向到某个...

Spring Boot 结合shiro做第三方登录验证

Spring Boot 结合shiro做第三方登录验证1、首先,说一下我的具体实现思路。在做spring boot拦截器的过程中,开始我准备用spring security来实现,但是研究了一段时间之...
  • he1lo
  • he1lo
  • 2016-08-18 19:47
  • 4216

spring-boot--使用thymeleaf模板

转:http://jisonami.iteye.com/blog/2301387 本文记录一下几点:  一、资源文件的约定目录结构  二、Maven配置  三、开发时修改thymeleaf模板自动...

Spring Boot Shiro权限管理

转:http://412887952-qq-com.iteye.com/blog/2299777 (1). Shiro简单介绍 Shiro是Apache下的一个开源项目,我们称之为Apache ...

Apache Shiro 标签模式授权

Apache Shiro 标签方式授权 Shiro提供了一套JSP标签库来实现页面级的授权控制。 在使用Shiro标签库前,首先需要在JSP引入shiro标签: 下面一一介绍Shi...

Spring Boot Shiro 权限信息缓存处理,记住我,thymleaf使用shiro标签

转: http://412887952-qq-com.iteye.com/blog/2299784 权限信息缓存处理 实际中我们的权限信息是不怎么会改变的,所以我们希望是第一次访问,然后进行缓存处理,...

Spring boot + thymeleaf + Shiro 会话过期返回登录界面片段之解决方案

解决方案的主导思想是:检测当会话过期时,判断是否为ajax请求,若是ajax请求,则将该url请求结果的状态置为401,并且不保存此次访问请求的url,当前端检测到访问结果为401时,跳转至登录界面,...

Spring Boot普通类调用bean

转:http://412887952-qq-com.iteye.com/blog/2292388 我们知道如果我们要在一个类使用spring提供的bean对象,我们需要把这个类注入到spring容器中...

Spring Boot 集成Shiro和CAS

请大家在看本文之前,先了解如下知识点: 1、Shiro 是什么?怎么用? 2、Cas 是什么?怎么用? 3、最好有Spring基础可以先看看这两篇文章,按照这2篇文章的内容做一遍: Sprin...
  • catoop
  • catoop
  • 2016-01-17 23:03
  • 31822
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:深度学习:神经网络中的前向传播和反向传播算法推导
举报原因:
原因补充:

(最多只允许输入30个字)