Extract OidcUserRequestUtils
This logic is shared by both reactive and non-reactive clients. Issue: gh-5330
This commit is contained in:
		
							parent
							
								
									4d1c8f26c5
								
							
						
					
					
						commit
						adb8c60173
					
				| 
						 | 
				
			
			@ -0,0 +1,70 @@
 | 
			
		|||
/*
 | 
			
		||||
 * Copyright 2002-2018 the original author or authors.
 | 
			
		||||
 *
 | 
			
		||||
 * Licensed under the Apache License, Version 2.0 (the "License");
 | 
			
		||||
 * you may not use this file except in compliance with the License.
 | 
			
		||||
 * You may obtain a copy of the License at
 | 
			
		||||
 *
 | 
			
		||||
 *      http://www.apache.org/licenses/LICENSE-2.0
 | 
			
		||||
 *
 | 
			
		||||
 * Unless required by applicable law or agreed to in writing, software
 | 
			
		||||
 * distributed under the License is distributed on an "AS IS" BASIS,
 | 
			
		||||
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 | 
			
		||||
 * See the License for the specific language governing permissions and
 | 
			
		||||
 * limitations under the License.
 | 
			
		||||
 */
 | 
			
		||||
 | 
			
		||||
package org.springframework.security.oauth2.client.oidc.userinfo;
 | 
			
		||||
 | 
			
		||||
import org.springframework.security.oauth2.client.registration.ClientRegistration;
 | 
			
		||||
import org.springframework.security.oauth2.core.AuthorizationGrantType;
 | 
			
		||||
import org.springframework.util.CollectionUtils;
 | 
			
		||||
import org.springframework.util.StringUtils;
 | 
			
		||||
 | 
			
		||||
/**
 | 
			
		||||
 * Utilities for working with the {@link OidcUserRequest}
 | 
			
		||||
 *
 | 
			
		||||
 * @author Rob Winch
 | 
			
		||||
 * @since 5.1
 | 
			
		||||
 */
 | 
			
		||||
final class OidcUserRequestUtils {
 | 
			
		||||
 | 
			
		||||
	/**
 | 
			
		||||
	 * Determines if an {@link OidcUserRequest} should attempt to retrieve the user info endpoint. Will return true if
 | 
			
		||||
	 * all of the following are true:
 | 
			
		||||
	 *
 | 
			
		||||
	 * <ul>
 | 
			
		||||
	 *     <li>The user info endpoint is defined on the ClientRegistration</li>
 | 
			
		||||
	 *     <li>The Client Registration uses the {@link AuthorizationGrantType#AUTHORIZATION_CODE} and scopes in the
 | 
			
		||||
	 *       access token are defined in the {@link ClientRegistration}</li>
 | 
			
		||||
	 * </ul>
 | 
			
		||||
	 * @param userRequest
 | 
			
		||||
	 * @return
 | 
			
		||||
	 */
 | 
			
		||||
	static boolean shouldRetrieveUserInfo(OidcUserRequest userRequest) {
 | 
			
		||||
		// Auto-disabled if UserInfo Endpoint URI is not provided
 | 
			
		||||
		ClientRegistration clientRegistration = userRequest.getClientRegistration();
 | 
			
		||||
		if (StringUtils.isEmpty(clientRegistration.getProviderDetails()
 | 
			
		||||
				.getUserInfoEndpoint().getUri())) {
 | 
			
		||||
 | 
			
		||||
			return false;
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		// The Claims requested by the profile, email, address, and phone scope values
 | 
			
		||||
		// are returned from the UserInfo Endpoint (as described in Section 5.3.2),
 | 
			
		||||
		// when a response_type value is used that results in an Access Token being issued.
 | 
			
		||||
		// However, when no Access Token is issued, which is the case for the response_type=id_token,
 | 
			
		||||
		// the resulting Claims are returned in the ID Token.
 | 
			
		||||
		// The Authorization Code Grant Flow, which is response_type=code, results in an Access Token being issued.
 | 
			
		||||
		if (AuthorizationGrantType.AUTHORIZATION_CODE.equals(clientRegistration.getAuthorizationGrantType())) {
 | 
			
		||||
 | 
			
		||||
			// Return true if there is at least one match between the authorized scope(s) and UserInfo scope(s)
 | 
			
		||||
			return CollectionUtils
 | 
			
		||||
					.containsAny(userRequest.getAccessToken().getScopes(), userRequest.getClientRegistration().getScopes());
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		return false;
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	private OidcUserRequestUtils() {}
 | 
			
		||||
}
 | 
			
		||||
| 
						 | 
				
			
			@ -0,0 +1,89 @@
 | 
			
		|||
/*
 | 
			
		||||
 * Copyright 2002-2018 the original author or authors.
 | 
			
		||||
 *
 | 
			
		||||
 * Licensed under the Apache License, Version 2.0 (the "License");
 | 
			
		||||
 * you may not use this file except in compliance with the License.
 | 
			
		||||
 * You may obtain a copy of the License at
 | 
			
		||||
 *
 | 
			
		||||
 *      http://www.apache.org/licenses/LICENSE-2.0
 | 
			
		||||
 *
 | 
			
		||||
 * Unless required by applicable law or agreed to in writing, software
 | 
			
		||||
 * distributed under the License is distributed on an "AS IS" BASIS,
 | 
			
		||||
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 | 
			
		||||
 * See the License for the specific language governing permissions and
 | 
			
		||||
 * limitations under the License.
 | 
			
		||||
 */
 | 
			
		||||
 | 
			
		||||
package org.springframework.security.oauth2.client.oidc.userinfo;
 | 
			
		||||
 | 
			
		||||
import org.junit.Test;
 | 
			
		||||
import org.springframework.security.oauth2.client.registration.ClientRegistration;
 | 
			
		||||
import org.springframework.security.oauth2.core.AuthorizationGrantType;
 | 
			
		||||
import org.springframework.security.oauth2.core.ClientAuthenticationMethod;
 | 
			
		||||
import org.springframework.security.oauth2.core.OAuth2AccessToken;
 | 
			
		||||
import org.springframework.security.oauth2.core.oidc.IdTokenClaimNames;
 | 
			
		||||
import org.springframework.security.oauth2.core.oidc.OidcIdToken;
 | 
			
		||||
 | 
			
		||||
import java.time.Duration;
 | 
			
		||||
import java.time.Instant;
 | 
			
		||||
import java.util.Collections;
 | 
			
		||||
 | 
			
		||||
import static org.assertj.core.api.Assertions.*;
 | 
			
		||||
 | 
			
		||||
/**
 | 
			
		||||
 * @author Rob Winch
 | 
			
		||||
 * @since 5.1
 | 
			
		||||
 */
 | 
			
		||||
public class OidcUserRequestUtilsTests {
 | 
			
		||||
	private ClientRegistration.Builder registration = ClientRegistration.withRegistrationId("id")
 | 
			
		||||
			.clientAuthenticationMethod(ClientAuthenticationMethod.BASIC)
 | 
			
		||||
			.authorizationUri("https://example.com/oauth2/authorize")
 | 
			
		||||
			.authorizationGrantType(AuthorizationGrantType.AUTHORIZATION_CODE)
 | 
			
		||||
			.userInfoUri("https://example.com/users/me")
 | 
			
		||||
			.clientId("client-id")
 | 
			
		||||
			.clientName("client-name")
 | 
			
		||||
			.clientSecret("client-secret")
 | 
			
		||||
			.redirectUriTemplate("{baseUrl}/login/oauth2/code/{registrationId}")
 | 
			
		||||
			.scope("user")
 | 
			
		||||
			.tokenUri("https://example.com/oauth/access_token");
 | 
			
		||||
 | 
			
		||||
	OidcIdToken idToken = new OidcIdToken("token123", Instant.now(),
 | 
			
		||||
			Instant.now().plusSeconds(3600), Collections
 | 
			
		||||
			.singletonMap(IdTokenClaimNames.SUB, "sub123"));
 | 
			
		||||
 | 
			
		||||
	OAuth2AccessToken accessToken = new OAuth2AccessToken(OAuth2AccessToken.TokenType.BEARER,
 | 
			
		||||
			"token",
 | 
			
		||||
			Instant.now(),
 | 
			
		||||
			Instant.now().plus(Duration.ofDays(1)),
 | 
			
		||||
			Collections.singleton("user"));
 | 
			
		||||
 | 
			
		||||
	@Test
 | 
			
		||||
	public void shouldRetrieveUserInfoWhenEndpointDefinedAndScopesOverlapThenTrue() {
 | 
			
		||||
		assertThat(OidcUserRequestUtils.shouldRetrieveUserInfo(userRequest())).isTrue();
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	@Test
 | 
			
		||||
	public void shouldRetrieveUserInfoWhenNoUserInfoUriThenFalse() {
 | 
			
		||||
		this.registration.userInfoUri(null);
 | 
			
		||||
 | 
			
		||||
		assertThat(OidcUserRequestUtils.shouldRetrieveUserInfo(userRequest())).isFalse();
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	@Test
 | 
			
		||||
	public void shouldRetrieveUserInfoWhenDifferentScopesThenFalse() {
 | 
			
		||||
		this.registration.scope("notintoken");
 | 
			
		||||
 | 
			
		||||
		assertThat(OidcUserRequestUtils.shouldRetrieveUserInfo(userRequest())).isFalse();
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	@Test
 | 
			
		||||
	public void shouldRetrieveUserInfoWhenNotAuthorizationCodeThenFalse() {
 | 
			
		||||
		this.registration.authorizationGrantType(AuthorizationGrantType.IMPLICIT);
 | 
			
		||||
 | 
			
		||||
		assertThat(OidcUserRequestUtils.shouldRetrieveUserInfo(userRequest())).isFalse();
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	private OidcUserRequest userRequest() {
 | 
			
		||||
		return new OidcUserRequest(this.registration.build(), this.accessToken, this.idToken);
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
		Loading…
	
		Reference in New Issue