spring oauth2 自定义短信验证

本文介绍了如何在Spring OAuth2中实现自定义的短信验证码授权模式。通过继承AbstractTokenGranter并修改grantType为'sms_code',创建自定义的SmsAuthenticationToken。同时,需要实现AuthenticationProvider来设定短信验证码的校验规则。最后,将自定义的TokenGranter和AuthenticationProvider添加到系统中,并确保数据库中grant_type字段包含'sms_code'。通过POSTMAN请求验证,整个流程完成。
摘要由CSDN通过智能技术生成

AbstractTokenGranter 是一个抽象类,它由四个默认实现方式,分别对应了四种默认授权模式
所有如果需要新增模式 继承这个类
ResourceOwnerPasswordTokenGranter
参照这个类
1. 修改 grantyType —> sms_code
2. 修改 UsernamePasswordAuthenticationToken —> SmsAuthenticationToken(自定义)


public class SmsCodeTokenGranter extends AbstractTokenGranter {
   

    private static final String GRANT_TYPE = "sms_code";

    private final AuthenticationManager authenticationManager;

    public SmsCodeTokenGranter(
            AuthenticationManager authenticationManager,
            AuthorizationServerTokenServices tokenServices,
            ClientDetailsService clientDetailsService,
            OAuth2RequestFactory requestFactory) {
   
        this(authenticationManager, tokenServices, clientDetailsService, requestFactory, GRANT_TYPE);
    }

    protected SmsCodeTokenGranter(
            AuthenticationManager authenticationManager,
            AuthorizationServerTokenServices tokenServices,
            ClientDetailsService clientDetailsService,
            OAuth2RequestFactory requestFactory,
            String grantType) {
   
        super(tokenServices, clientDetailsService, requestFactory, grantType);
        this.authenticationManager = authenticationManager;
    }

    @Override
    public OAuth2Authentication getOAuth2Authentication(ClientDetails client, TokenRequest tokenRequest) {
   
        Map<String, String> parameters = new LinkedHashMap<String, String>(tokenRequest.getRequestParameters());
        // 获取参数
        String mobile = parameters.get("mobile");
        String smsCode = parameters.get("sms_code");
        
        Authentication userAuth = new SmsAuthenticationToken(mobile,smsCode);
        ((AbstractAuthenticationToken) userAuth).setDetails(parameters);
        try {
   
            userAuth = authenticationManager.authenticate(userAuth);
        } catch (AccountStatusException | BadCredentialsException ase) {
   
            throw new InvalidGra
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
Spring OAuth2提供了一个框架,可以使我们很容易地实现短信验证码登录。下面是实现该功能的简单步骤: 1. 添加依赖 在pom.xml文件中添加以下依赖项: ``` <dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-security</artifactId> </dependency> <dependency> <groupId>org.springframework.security.oauth.boot</groupId> <artifactId>spring-security-oauth2-autoconfigure</artifactId> <version>2.1.0.RELEASE</version> </dependency> ``` 2. 配置Security 在Spring Security配置文件中添加以下配置: ``` @Configuration @EnableWebSecurity public class SecurityConfig extends WebSecurityConfigurerAdapter { @Autowired private UserDetailsService userDetailsService; @Override protected void configure(AuthenticationManagerBuilder auth) throws Exception { auth.userDetailsService(userDetailsService).passwordEncoder(passwordEncoder()); } @Bean public PasswordEncoder passwordEncoder(){ return new BCryptPasswordEncoder(); } @Override protected void configure(HttpSecurity http) throws Exception { http.csrf().disable() .authorizeRequests() .antMatchers("/oauth/token").permitAll() .anyRequest().authenticated() .and() .formLogin().permitAll() .and() .logout().permitAll(); } } ``` 这将启用Web安全性,并配置Spring Security以使用我们提供的用户详细信息服务和密码编码器。它还允许所有人访问OAuth2令牌端点(/oauth/token)。 3. 实现用户详细信息服务 我们需要实现UserDetailsService接口来从我们的数据库或其他数据源中获取用户详细信息。以下是一个示例: ``` @Service public class UserDetailsServiceImpl implements UserDetailsService { @Autowired private UserRepository userRepository; @Override public UserDetails loadUserByUsername(String username) throws UsernameNotFoundException { User user = userRepository.findByUsername(username); if(user == null){ throw new UsernameNotFoundException("User not found with username: " + username); } return new org.springframework.security.core.userdetails.User(user.getUsername(), user.getPassword(), new ArrayList<>()); } } ``` 此服务返回Spring Security的UserDetails对象,该对象包含有关用户的详细信息。 4. 实现自定义短信验证码登录过滤器 我们需要实现一个自定义的过滤器来处理短信验证码登录。以下是一个示例: ``` public class SmsCodeAuthenticationFilter extends AbstractAuthenticationProcessingFilter { public static final String SPRING_SECURITY_FORM_MOBILE_KEY = "mobile"; public static final String SPRING_SECURITY_FORM_CODE_KEY = "code"; private String mobileParameter = SPRING_SECURITY_FORM_MOBILE_KEY; private String codeParameter = SPRING_SECURITY_FORM_CODE_KEY; private boolean postOnly = true; public SmsCodeAuthenticationFilter() { super(new AntPathRequestMatcher("/login/mobile", "POST")); } @Override public Authentication attemptAuthentication(HttpServletRequest request, HttpServletResponse response) throws AuthenticationException, IOException, ServletException { if (postOnly && !request.getMethod().equals("POST")) { throw new AuthenticationServiceException( "Authentication method not supported: " + request.getMethod()); } String mobile = obtainMobile(request); String code = obtainCode(request); if (mobile == null) { mobile = ""; } if (code == null) { code = ""; } mobile = mobile.trim(); SmsCodeAuthenticationToken authRequest = new SmsCodeAuthenticationToken(mobile, code); setDetails(request, authRequest); return this.getAuthenticationManager().authenticate(authRequest); } protected String obtainMobile(HttpServletRequest request) { return request.getParameter(mobileParameter); } protected String obtainCode(HttpServletRequest request) { return request.getParameter(codeParameter); } protected void setDetails(HttpServletRequest request, SmsCodeAuthenticationToken authRequest) { authRequest.setDetails(authenticationDetailsSource.buildDetails(request)); } public void setMobileParameter(String mobileParameter) { Assert.hasText(mobileParameter, "Mobile parameter must not be empty or null"); this.mobileParameter = mobileParameter; } public void setPostOnly(boolean postOnly) { this.postOnly = postOnly; } public final String getMobileParameter() { return mobileParameter; } public String getCodeParameter() { return codeParameter; } public void setCodeParameter(String codeParameter) { Assert.hasText(codeParameter, "Code parameter must not be empty or null"); this.codeParameter = codeParameter; } } ``` 此过滤器将处理POST请求,其中包含用户的手机号码和短信验证码。它将创建一个SmsCodeAuthenticationToken对象,该对象将包含有关用户的信息。在这里,我们还可以设置其他详细信息,例如IP地址和请求的时间戳等。 5. 配置认证管理器 在Spring Security配置文件中添加以下配置: ``` @Configuration @EnableWebSecurity public class SecurityConfig extends WebSecurityConfigurerAdapter { @Autowired private UserDetailsService userDetailsService; @Autowired private SmsCodeAuthenticationProvider smsCodeAuthenticationProvider; @Override protected void configure(AuthenticationManagerBuilder auth) throws Exception { auth.userDetailsService(userDetailsService).passwordEncoder(passwordEncoder()); auth.authenticationProvider(smsCodeAuthenticationProvider); } @Bean public PasswordEncoder passwordEncoder(){ return new BCryptPasswordEncoder(); } @Override protected void configure(HttpSecurity http) throws Exception { http.csrf().disable() .authorizeRequests() .antMatchers("/oauth/token").permitAll() .antMatchers("/login/mobile").permitAll() .anyRequest().authenticated() .and() .formLogin().permitAll() .and() .logout().permitAll(); http.addFilterBefore(smsCodeAuthenticationFilter(), UsernamePasswordAuthenticationFilter.class); } @Bean public SmsCodeAuthenticationFilter smsCodeAuthenticationFilter() throws Exception { SmsCodeAuthenticationFilter filter = new SmsCodeAuthenticationFilter(); filter.setAuthenticationManager(authenticationManagerBean()); return filter; } @Bean public SmsCodeAuthenticationProvider smsCodeAuthenticationProvider(){ return new SmsCodeAuthenticationProvider(); } @Bean @Override public AuthenticationManager authenticationManagerBean() throws Exception { return super.authenticationManagerBean(); } } ``` 这将配置我们的认证管理器,并告诉Spring Security使用我们自定义SmsCodeAuthenticationProvider来验证短信验证码。我们还需要将自定义SmsCodeAuthenticationFilter添加到过滤器链中。 6. 实现自定义短信验证码认证提供程序 我们需要实现一个自定义的认证提供程序来验证短信验证码。以下是一个示例: ``` public class SmsCodeAuthenticationProvider implements AuthenticationProvider { @Autowired private UserDetailsService userDetailsService; @Override public Authentication authenticate(Authentication authentication) throws AuthenticationException { SmsCodeAuthenticationToken authenticationToken = (SmsCodeAuthenticationToken) authentication; UserDetails userDetails = userDetailsService.loadUserByUsername((String) authenticationToken.getPrincipal()); if(userDetails == null){ throw new InternalAuthenticationServiceException("无法获取用户信息"); } SmsCode smsCode = (SmsCode) authenticationToken.getCredentials(); if(!smsCode.getCode().equals("123456")){ throw new BadCredentialsException("短信验证码不正确"); } SmsCodeAuthenticationToken authenticationResult = new SmsCodeAuthenticationToken(userDetails, userDetails.getAuthorities()); authenticationResult.setDetails(authenticationToken.getDetails()); return authenticationResult; } @Override public boolean supports(Class<?> authentication) { return SmsCodeAuthenticationToken.class.isAssignableFrom(authentication); } } ``` 此提供程序将验证提供的短信验证码,并返回一个已验证的SmsCodeAuthenticationToken对象。 最后,我们就可以使用短信验证码登录了。用户只需在登录页面上输入其手机号码和收到的短信验证码即可成功登录。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值