Spring Security源碼分析十:初識(shí)Spring Security OAuth2

OAuth 是一個(gè)開放標(biāo)準(zhǔn),允許用戶讓第三方應(yīng)用訪問該用戶在某一網(wǎng)站上存儲(chǔ)的私密的資源(如照片,視頻,聯(lián)系人列表),而不需要將用戶名和密碼提供給第三方應(yīng)用。OAuth允許用戶提供一個(gè)令牌,而不是用戶名和密碼來訪問他們存放在特定服務(wù)提供者的數(shù)據(jù)。每一個(gè)令牌授權(quán)一個(gè)特定的網(wǎng)站在特定的時(shí)段內(nèi)訪問特定的資源。這樣,OAuth讓用戶可以授權(quán)第三方網(wǎng)站訪問他們存儲(chǔ)在另外服務(wù)提供者的某些特定信息。更多OAuth2請參考理解OAuth 2.0

項(xiàng)目準(zhǔn)備

  1. 添加依賴

    <dependency>
                <groupId>org.springframework.boot</groupId>
                <artifactId>spring-boot-starter-security</artifactId>
            </dependency>
            <dependency>
                <groupId>org.springframework.boot</groupId>
                <artifactId>spring-boot-starter-web</artifactId>
            </dependency>
            <dependency>
                <groupId>org.springframework.security.oauth</groupId>
                <artifactId>spring-security-oauth2</artifactId>
            </dependency>
            <dependency>
                <groupId>org.springframework.boot</groupId>
                <artifactId>spring-boot-starter-test</artifactId>
            </dependency>
    
  2. 配置認(rèn)證服務(wù)器

    @Configuration
    @EnableAuthorizationServer//是的,沒做,就這么一個(gè)注解
    public class MerryyouAuthorizationServerConfig {
    
    }
    
  3. 配置資源服務(wù)器

    @Configuration
    @EnableResourceServer//咦,沒錯(cuò)還是一個(gè)注解
    public class MerryyouResourceServerConfig {
    }
    
  4. 配置application.yml客戶端信息(不配置的話,控制臺(tái)會(huì)默認(rèn)打印clientid和clietSecret)

    security:
      oauth2:
        client:
          client-id: merryyou
          client-secret: merryyou
    
  5. 定義MyUserDetailsService

    @Component
    public class MyUserDetailsService implements UserDetailsService {
    
       @Override
       public UserDetails loadUserByUsername(String username) throws UsernameNotFoundException {
           return new User(username, "123456", AuthorityUtils.commaSeparatedStringToAuthorityList("ROLE_USER"));
       }
    }
    
  6. 添加測試類SecurityOauth2Test(用戶名密碼模式)

    @RunWith(SpringRunner.class)
    @SpringBootTest
    @Slf4j
    public class SecurityOauth2Test {
        //端口
        final static long PORT = 9090;
        //clientId
        final static String CLIENT_ID = "merryyou";
        //clientSecret
        final static String CLIENT_SECRET = "merryyou";
        //用戶名
        final static String USERNAME = "admin";
        //密碼
        final static String PASSWORD = "123456";
        //獲取accessToken得URI
        final static String TOKEN_REQUEST_URI = "http://localhost:"+PORT+"/oauth/token?grant_type=password&username=" + USERNAME + "&password=" + PASSWORD+"&scope=all";
        //獲取用戶信息得URL
        final static String USER_INFO_URI = "http://localhost:"+PORT+"/user";
    
        @Test
        public void getUserInfo() throws Exception{
            RestTemplate rest = new RestTemplate();
            HttpHeaders headers = new HttpHeaders();
            headers.add( "authorization", "Bearer " + getAccessToken() );
            HttpEntity<String> entity = new HttpEntity<String>(null, headers);
            // pay attention, if using get with headers, should use exchange instead of getForEntity / getForObject
            ResponseEntity<String> result = rest.exchange( USER_INFO_URI, HttpMethod.GET, entity, String.class, new Object[]{ null } );
            log.info("用戶信息返回的結(jié)果={}",JsonUtil.toJson(result));
        }
    
        /**
         * 獲取accessToken
         * @return
         */
        private String getAccessToken(){
            RestTemplate rest = new RestTemplate();
            HttpHeaders headers = new HttpHeaders();
            headers.setContentType( MediaType.TEXT_PLAIN );
            headers.add("authorization", getBasicAuthHeader());
            HttpEntity<String> entity = new HttpEntity<String>(null, headers);
            ResponseEntity<OAuth2AccessToken> resp = rest.postForEntity( TOKEN_REQUEST_URI, entity, OAuth2AccessToken.class);
            if( !resp.getStatusCode().equals( HttpStatus.OK )){
                throw new RuntimeException( resp.toString() );
            }
            OAuth2AccessToken t = resp.getBody();
            log.info("accessToken={}",JsonUtil.toJson(t));
            log.info("the response, access_token: " + t.getValue() +"; token_type: " + t.getTokenType() +"; "
                    + "refresh_token: " + t.getRefreshToken() +"; expiration: " + t.getExpiresIn() +", expired when:" + t.getExpiration() );
            return t.getValue();
    
        }
    
        /**
         * 構(gòu)建header
         * @return
         */
        private String getBasicAuthHeader(){
            String auth = CLIENT_ID + ":" + CLIENT_SECRET;
            byte[] encodedAuth = Base64.encodeBase64(auth.getBytes());
            String authHeader = "Basic " + new String(encodedAuth);
            return authHeader;
        }
    }
    

授權(quán)碼模式效果如下:

授權(quán)鏈接:
http://localhost:9090/oauth/authorize?response_type=code&client_id=merryyou&redirect_uri=http://merryyou.cn&scope=all

[圖片上傳失敗...(image-fd646d-1516448982425)]

測試類打印accessToken信息

2018-01-20 18:16:49.900  INFO 16136 --- [           main] cn.merryyou.security.SecurityOauth2Test  : accessToken={
  "value": "8e5ea72c-d153-48f5-8ee7-9b5616fc43dc",
  "expiration": "Jan 21, 2018 6:10:25 AM",
  "tokenType": "bearer",
  "refreshToken": {
    "value": "7adfefec-c80c-4ff4-913c-4f161c47fbf1"
  },
  "scope": [
    "all"
  ],
  "additionalInformation": {}
}

spring security oauth2 登錄核心源碼

TokenEndpoint

//#1.處理/oauth/token請求
@RequestMapping(value = "/oauth/token", method=RequestMethod.POST)
    public ResponseEntity<OAuth2AccessToken> postAccessToken(Principal principal, @RequestParam
    Map<String, String> parameters) throws HttpRequestMethodNotSupportedException {

        if (!(principal instanceof Authentication)) {
            throw new InsufficientAuthenticationException(
                    "There is no client authentication. Try adding an appropriate authentication filter.");
        }
        //獲取clientId
        String clientId = getClientId(principal);
        //獲取第三方應(yīng)用的詳細(xì)配置信息
        ClientDetails authenticatedClient = getClientDetailsService().loadClientByClientId(clientId);
        //使用第三方應(yīng)用信息創(chuàng)建TokenRequest
        TokenRequest tokenRequest = getOAuth2RequestFactory().createTokenRequest(parameters, authenticatedClient);
        //有沒有傳clientId
        if (clientId != null && !clientId.equals("")) {
            // Only validate the client details if a client authenticated during this
            // request.
            //與配置里面的是否匹配
            if (!clientId.equals(tokenRequest.getClientId())) {
                // double check to make sure that the client ID in the token request is the same as that in the
                // authenticated client
                throw new InvalidClientException("Given client ID does not match authenticated client");
            }
        }
        if (authenticatedClient != null) {
            //檢查scope
            oAuth2RequestValidator.validateScope(tokenRequest, authenticatedClient);
        }
        //grant_type是否存在值,對應(yīng)四種授權(quán)模式和刷新token
        if (!StringUtils.hasText(tokenRequest.getGrantType())) {
            throw new InvalidRequestException("Missing grant type");
        }
        //是否簡化模式
        if (tokenRequest.getGrantType().equals("implicit")) {
            throw new InvalidGrantException("Implicit grant type not supported from token endpoint");
        }
        //是否是授權(quán)碼模式
        if (isAuthCodeRequest(parameters)) {
            // The scope was requested or determined during the authorization step
            if (!tokenRequest.getScope().isEmpty()) {
                logger.debug("Clearing scope of incoming token request");
                //如果是授權(quán)碼模式scope設(shè)置為空,根據(jù)獲取code時(shí)的scope設(shè)置
                tokenRequest.setScope(Collections.<String> emptySet());
            }
        }
        //是否刷新令牌
        if (isRefreshTokenRequest(parameters)) {
            // A refresh token has its own default scopes, so we should ignore any added by the factory here.
            //設(shè)置scope
            tokenRequest.setScope(OAuth2Utils.parseParameterList(parameters.get(OAuth2Utils.SCOPE)));
        }
        //獲取OAuth2AccessToken
        OAuth2AccessToken token = getTokenGranter().grant(tokenRequest.getGrantType(), tokenRequest);
        if (token == null) {
            throw new UnsupportedGrantTypeException("Unsupported grant type: " + tokenRequest.getGrantType());
        }

        return getResponse(token);

    }

ClientDetails

TokenRequest

CompositeTokenGranter#grant

    //四種授權(quán)模式+刷新令牌的模式根據(jù)grant_type判斷
    public OAuth2AccessToken grant(String grantType, TokenRequest tokenRequest) {
        for (TokenGranter granter : tokenGranters) {
            OAuth2AccessToken grant = granter.grant(grantType, tokenRequest);
            if (grant!=null) {
                return grant;
            }
        }
        return null;
    }

tokenGranters

AbstractTokenGranter#grant

public OAuth2AccessToken grant(String grantType, TokenRequest tokenRequest) {
        //判斷當(dāng)前的授權(quán)類型和傳入的是否匹配
        if (!this.grantType.equals(grantType)) {
            return null;
        }
        //獲取clientId
        String clientId = tokenRequest.getClientId();
        ClientDetails client = clientDetailsService.loadClientByClientId(clientId);
        //校驗(yàn)
        validateGrantType(grantType, client);
        
        logger.debug("Getting access token for: " + clientId);
        //產(chǎn)生令牌
        return getAccessToken(client, tokenRequest);

    }

AbstractTokenGranter#getAccessToken

protected OAuth2AccessToken getAccessToken(ClientDetails client, TokenRequest tokenRequest) {
        return tokenServices.createAccessToken(getOAuth2Authentication(client, tokenRequest));
    }

DefaultTokenServices#createAccessToken

public OAuth2AccessToken createAccessToken(OAuth2Authentication authentication) throws AuthenticationException {
        //從tokenStore獲取OAuth2AccessToken (如果令牌存在,不同的授權(quán)模式下將返回同一個(gè)令牌)
        OAuth2AccessToken existingAccessToken = tokenStore.getAccessToken(authentication);
        OAuth2RefreshToken refreshToken = null;
        //判斷是否過期
        if (existingAccessToken != null) {
            if (existingAccessToken.isExpired()) {
                if (existingAccessToken.getRefreshToken() != null) {
                    //刪除過期的令牌
                    refreshToken = existingAccessToken.getRefreshToken();
                    // The token store could remove the refresh token when the
                    // access token is removed, but we want to
                    // be sure...

                    tokenStore.removeRefreshToken(refreshToken);
                }
                tokenStore.removeAccessToken(existingAccessToken);
            }
            else {
                //如果令牌存在則從新存儲(chǔ)一下
                // Re-store the access token in case the authentication has changed
                tokenStore.storeAccessToken(existingAccessToken, authentication);
                //存儲(chǔ)完直接返回
                return existingAccessToken;
            }
        }

        // Only create a new refresh token if there wasn't an existing one
        // associated with an expired access token.
        // Clients might be holding existing refresh tokens, so we re-use it in
        // the case that the old access token
        // expired.
        //判斷刷新令牌不存在
        if (refreshToken == null) {
            //創(chuàng)建刷新令牌
            refreshToken = createRefreshToken(authentication);
        }
        // But the refresh token itself might need to be re-issued if it has
        // expired.
        else if (refreshToken instanceof ExpiringOAuth2RefreshToken) {
            //過期
            ExpiringOAuth2RefreshToken expiring = (ExpiringOAuth2RefreshToken) refreshToken;
            if (System.currentTimeMillis() > expiring.getExpiration().getTime()) {
                refreshToken = createRefreshToken(authentication);
            }
        }
        //根據(jù)刷新令牌創(chuàng)建OAuth2AccessToken
        OAuth2AccessToken accessToken = createAccessToken(authentication, refreshToken);
        tokenStore.storeAccessToken(accessToken, authentication);
        // In case it was modified
        refreshToken = accessToken.getRefreshToken();
        if (refreshToken != null) {
            tokenStore.storeRefreshToken(refreshToken, authentication);
        }
        //返回OAuth2AccessToken
        return accessToken;

    }

代碼下載

從我的 github 中下載,https://github.com/longfeizheng/security-oauth2

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容