本篇重点讨论如何实现自定义的登录方式,所以如何配制 spring-security-oauth2 就不说了.

一些说明

• 之前已经配好使用 spring-security-oauth2 + JWT 的认证方式,这次仅是改造

• 本篇部分涉及的代码都是认证服务器的代码

跟踪spring的登录逻辑发现,帐号密码的验证是在 tokenGranter 中完成的, 帐号密码对应的是 org.springframework.security.oauth2.provider.password.ResourceOwnerPasswordTokenGranter;
而spring找到对应的 tokenGranter 是通过登录时的一个表单参数"grant_type" 来找到的,这样的话,那我是不是可以通过扩展一个 TokenGranter 来达成我要的效果呢?

spring 默认是同时支持多重 grant_type 的(根据客户端的配制决定特定客户端支持特定的 grant_type), 而AuthorizationServerConfigurerAdapter 的配制中 tokenGranter 又只能设置一个,那么spirng是怎么实现多个的呢?经过折腾我发现了一个 org.springframework.security.oauth2.provider.CompositeTokenGranter , 原来是通过它来实现的. 那么又有思路了.

接下来研究如何向 CompositeTokenGranter  中增加我自定义的 TokenGranter , 结果发现spring在创建 CompositeTokenGranter 的时候已经把内置的 TokenGranter  写死了,没法通过它的机制扩展.唯一的方法就是我直接使用 CompositeTokenGranter . 那么我还是想要内置的 TokenGranter  也一起工作怎么办...?最后我无奈的选侧了把创建内置 TokenGranter  的代码copy了出来并修改了能用...不说了,直接上代码,有些需要注意的地方我会通过注释来描述.

由于是已有项目的改造,所以怎么配制 spring-security-oauth2 + JWT ,怎么配制认证服务器和资源服务器,甚至oauth2 是什么?我就不多说了,网上一搜一大把.如果有同学实在需要的话,我再另开一篇来写吧

import java.util.ArrayList;import java.util.Arrays;import java.util.HashMap;import java.util.List;import java.util.Map;import org.springframework.beans.factory.annotation.Autowired;import org.springframework.context.annotation.Bean;import org.springframework.context.annotation.Configuration;import org.springframework.context.annotation.Primary;import org.springframework.core.io.ClassPathResource;import org.springframework.http.HttpMethod;import org.springframework.security.authentication.AuthenticationManager;import org.springframework.security.core.authority.AuthorityUtils;import org.springframework.security.core.userdetails.UserDetailsService;import org.springframework.security.oauth2.common.OAuth2AccessToken;import org.springframework.security.oauth2.config.annotation.web.configuration.AuthorizationServerConfigurerAdapter;import org.springframework.security.oauth2.config.annotation.web.configuration.EnableAuthorizationServer;import org.springframework.security.oauth2.config.annotation.web.configurers.AuthorizationServerEndpointsConfigurer;import org.springframework.security.oauth2.config.annotation.web.configurers.AuthorizationServerSecurityConfigurer;import org.springframework.security.oauth2.provider.ClientDetails;import org.springframework.security.oauth2.provider.ClientDetailsService;import org.springframework.security.oauth2.provider.CompositeTokenGranter;import org.springframework.security.oauth2.provider.OAuth2RequestFactory;import org.springframework.security.oauth2.provider.TokenGranter;import org.springframework.security.oauth2.provider.TokenRequest;import org.springframework.security.oauth2.provider.client.BaseClientDetails;import org.springframework.security.oauth2.provider.client.ClientCredentialsTokenGranter;import org.springframework.security.oauth2.provider.client.InMemoryClientDetailsService;import org.springframework.security.oauth2.provider.code.AuthorizationCodeServices;import org.springframework.security.oauth2.provider.code.AuthorizationCodeTokenGranter;import org.springframework.security.oauth2.provider.code.InMemoryAuthorizationCodeServices;import org.springframework.security.oauth2.provider.implicit.ImplicitTokenGranter;import org.springframework.security.oauth2.provider.password.ResourceOwnerPasswordTokenGranter;import org.springframework.security.oauth2.provider.refresh.RefreshTokenGranter;import org.springframework.security.oauth2.provider.request.DefaultOAuth2RequestFactory;import org.springframework.security.oauth2.provider.token.AuthorizationServerTokenServices;import org.springframework.security.oauth2.provider.token.DefaultTokenServices;import org.springframework.security.oauth2.provider.token.TokenEnhancer;import org.springframework.security.oauth2.provider.token.TokenEnhancerChain;import org.springframework.security.oauth2.provider.token.TokenStore;import org.springframework.security.oauth2.provider.token.store.JwtAccessTokenConverter;import org.springframework.security.oauth2.provider.token.store.JwtTokenStore;import org.springframework.security.oauth2.provider.token.store.KeyStoreKeyFactory;/**
 * @ClassName: OAuth2AuthorizationServerConfig
 * @Description: spring-security OAuth2 配制,使用 jwt
 */@Configuration@EnableAuthorizationServerpublic class OAuth2AuthorizationServerConfig extends AuthorizationServerConfigurerAdapter {    @Autowired
    private AuthenticationManager authenticationManager;    
    @Autowired
    private UserDetailsService userDetailsService;  // 这是提供根据用户名查用户的方式给spring使用的
    
    @Bean
    /**
    之前有个  public void configure(ClientDetailsServiceConfigurer clients) 配制客户端的方法,但是因为直接使  
    用 CompositeTokenGranter ,所以它不生效了,就在这里配制,同时使用这样的配制方式,后面可以改成从库里获取,自己实现一个 ClientDetailsService 就行
    由于之前的 Builder方式只能在 ClientDetailsServiceConfigurer 中使用,所以这里暂时先这样了,后面我要改为从库里获取
    */
    public ClientDetailsService clientDetailsService() {
    BaseClientDetails result = new BaseClientDetails();
    result.setClientId("weixin_client");
    List<String> authorizedGrantTypes = new ArrayList<>();
    authorizedGrantTypes.add("password");
    authorizedGrantTypes.add("refresh_token");
    result.setAuthorizedGrantTypes(authorizedGrantTypes);  // 这个 client 支持的 grant_type 
    result.setClientSecret("$2a$10$9s0p62wfKh7WT64a/VYFpOAk19GsrHh5C7Ty9.wPRWX40cjq7Rmu."); // 这个密码是用  org.springframework.security.crypto.bcrypt.BCryptPasswordEncoder 搞出来的,明文是 123456
    List<String> scopes = new ArrayList<>();
    scopes.add("select");
    result.setScope(scopes);
    result.setAuthorities(AuthorityUtils.createAuthorityList("client"));

    Map<String, ClientDetails> clientDetails = new HashMap<String, ClientDetails>();
    clientDetails.put(result.getClientId(), result);

    InMemoryClientDetailsService clientDetailsService = new InMemoryClientDetailsService();
    clientDetailsService.setClientDetailsStore(clientDetails);    return clientDetailsService;
    }    private AuthorizationCodeServices authorizationCodeServices() {      return new InMemoryAuthorizationCodeServices();  //使用默认
    }    private OAuth2RequestFactory requestFactory() {      return new DefaultOAuth2RequestFactory(clientDetailsService());  //使用默认
    }/**
这是从spring 的代码中 copy出来的,默认的几个 TokenGranter, 我们自定义的就加到这里就行了,目前我还没有加
*/
    private List<TokenGranter> getDefaultTokenGranters() { 
    ClientDetailsService clientDetails = clientDetailsService();
    AuthorizationServerTokenServices tokenServices = tokenServices();
    AuthorizationCodeServices authorizationCodeServices = authorizationCodeServices();
    OAuth2RequestFactory requestFactory = requestFactory();

    List<TokenGranter> tokenGranters = new ArrayList<TokenGranter>();
    tokenGranters.add(new AuthorizationCodeTokenGranter(tokenServices,
        authorizationCodeServices, clientDetails, requestFactory));
    tokenGranters.add(new RefreshTokenGranter(tokenServices, clientDetails, requestFactory));
    ImplicitTokenGranter implicit = new ImplicitTokenGranter(tokenServices, clientDetails,
        requestFactory);
    tokenGranters.add(implicit);
    tokenGranters.add(        new ClientCredentialsTokenGranter(tokenServices, clientDetails, requestFactory));    if (authenticationManager != null) {
        tokenGranters.add(new ResourceOwnerPasswordTokenGranter(authenticationManager,
            tokenServices, clientDetails, requestFactory));
    }    return tokenGranters;
    }/**
通过 tokenGranter 塞进去的就是它了
*/
    private TokenGranter tokenGranter() {
    TokenGranter tokenGranter = new TokenGranter() {        private CompositeTokenGranter delegate;        @Override
        public OAuth2AccessToken grant(String grantType, TokenRequest tokenRequest) {        if (delegate == null) {
            delegate = new CompositeTokenGranter(getDefaultTokenGranters());
        }        return delegate.grant(grantType, tokenRequest);
        }
    };    return tokenGranter;
    }    
    @Override
    public void configure(AuthorizationServerEndpointsConfigurer endpoints) throws Exception {
        endpoints.tokenStore(tokenStore())//      .accessTokenConverter(accessTokenConverter())
            .tokenGranter(tokenGranter())//      .tokenEnhancer(tokenEnhancerChain)  // 设了 tokenGranter 后该配制失效,需要在 tokenServices() 中设置
        .authenticationManager(authenticationManager)
        .userDetailsService(userDetailsService)  //refresh_token 需要配制它,否则会 UserDetailsService is required
        .allowedTokenEndpointRequestMethods(HttpMethod.POST);
    }    
    @Bean
    public TokenEnhancer tokenEnhancer() {
    TokenEnhancerChain tokenEnhancerChain = new TokenEnhancerChain();
    tokenEnhancerChain.setTokenEnhancers(Arrays.asList(new CustomTokenEnhancer(), accessTokenConverter()));   // CustomTokenEnhancer 是我自定义一些数据放到token里用的
        return tokenEnhancerChain;
    }    
    @Override
    public void configure(AuthorizationServerSecurityConfigurer oauthServer) throws Exception {        //允许表单认证
        oauthServer.allowFormAuthenticationForClients();//        .checkTokenAccess("permitAll()");  // 允许 check_token, 因为用了JWT,客户端可以验证签名,生产中可以不用
    }    

    @Bean
    public TokenStore tokenStore() {        return new JwtTokenStore(accessTokenConverter());
    }    @Bean
    public JwtAccessTokenConverter accessTokenConverter() {
        JwtAccessTokenConverter converter = new JwtAccessTokenConverter();
    KeyStoreKeyFactory keyStoreKeyFactory = new KeyStoreKeyFactory(new ClassPathResource("authorizationKey.jks"), "123456".toCharArray());
    converter.setKeyPair(keyStoreKeyFactory.getKeyPair("klw"));    return converter;
    }    @Bean
    @Primary
    public DefaultTokenServices tokenServices() {
        DefaultTokenServices defaultTokenServices = new DefaultTokenServices();
        defaultTokenServices.setTokenStore(tokenStore());
        defaultTokenServices.setSupportRefreshToken(true);
        defaultTokenServices.setTokenEnhancer(tokenEnhancer());   // 如果没有设置它,JWT就失效了.
        return defaultTokenServices;
    }
    
    
}

作者：邪影oO
链接：https://www.jianshu.com/p/99552e456311