Commit 32454b83 authored by Vedran Pavic's avatar Vedran Pavic Committed by Andy Wilkinson

Add support for customizing Spring Session's cookie serializer

This commit introduces a CookieSerializerCustomizer callback that
allows the customization of the auto-configured
DefaultCookieSerializer bean. This is particularly useful for
configuring cookie serializer's capabilities, such as SameSite, that
are not supported by the Servlet API and therefore not exposed via
server.servlet.session.cookie.* properties.

See gh-20961
parent f3d717e9
/*
* Copyright 2012-2020 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
*
* https://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.boot.autoconfigure.session;
import org.springframework.session.web.http.DefaultCookieSerializer;
/**
* Callback interface that can be implemented by beans wishing to customize the
* {@link DefaultCookieSerializer} configuration.
*
* @author Vedran Pavic
* @since 2.3.0
*/
@FunctionalInterface
public interface CookieSerializerCustomizer {
/**
* Customize the cookie serializer.
* @param cookieSerializer the {@code CookieSerializer} to customize
*/
void customize(DefaultCookieSerializer cookieSerializer);
}
...@@ -53,6 +53,7 @@ import org.springframework.context.annotation.Configuration; ...@@ -53,6 +53,7 @@ import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Import; import org.springframework.context.annotation.Import;
import org.springframework.context.annotation.ImportSelector; import org.springframework.context.annotation.ImportSelector;
import org.springframework.core.type.AnnotationMetadata; import org.springframework.core.type.AnnotationMetadata;
import org.springframework.security.web.authentication.RememberMeServices;
import org.springframework.session.ReactiveSessionRepository; import org.springframework.session.ReactiveSessionRepository;
import org.springframework.session.Session; import org.springframework.session.Session;
import org.springframework.session.SessionRepository; import org.springframework.session.SessionRepository;
...@@ -61,7 +62,6 @@ import org.springframework.session.web.http.CookieHttpSessionIdResolver; ...@@ -61,7 +62,6 @@ import org.springframework.session.web.http.CookieHttpSessionIdResolver;
import org.springframework.session.web.http.CookieSerializer; import org.springframework.session.web.http.CookieSerializer;
import org.springframework.session.web.http.DefaultCookieSerializer; import org.springframework.session.web.http.DefaultCookieSerializer;
import org.springframework.session.web.http.HttpSessionIdResolver; import org.springframework.session.web.http.HttpSessionIdResolver;
import org.springframework.util.ClassUtils;
/** /**
* {@link EnableAutoConfiguration Auto-configuration} for Spring Session. * {@link EnableAutoConfiguration Auto-configuration} for Spring Session.
...@@ -83,8 +83,6 @@ import org.springframework.util.ClassUtils; ...@@ -83,8 +83,6 @@ import org.springframework.util.ClassUtils;
@AutoConfigureBefore(HttpHandlerAutoConfiguration.class) @AutoConfigureBefore(HttpHandlerAutoConfiguration.class)
public class SessionAutoConfiguration { public class SessionAutoConfiguration {
private static final String REMEMBER_ME_SERVICES_CLASS = "org.springframework.security.web.authentication.RememberMeServices";
@Configuration(proxyBeanMethods = false) @Configuration(proxyBeanMethods = false)
@ConditionalOnWebApplication(type = Type.SERVLET) @ConditionalOnWebApplication(type = Type.SERVLET)
@Import({ ServletSessionRepositoryValidator.class, SessionRepositoryFilterConfiguration.class }) @Import({ ServletSessionRepositoryValidator.class, SessionRepositoryFilterConfiguration.class })
...@@ -92,7 +90,8 @@ public class SessionAutoConfiguration { ...@@ -92,7 +90,8 @@ public class SessionAutoConfiguration {
@Bean @Bean
@Conditional(DefaultCookieSerializerCondition.class) @Conditional(DefaultCookieSerializerCondition.class)
DefaultCookieSerializer cookieSerializer(ServerProperties serverProperties) { DefaultCookieSerializer cookieSerializer(ServerProperties serverProperties,
ObjectProvider<CookieSerializerCustomizer> cookieSerializerCustomizers) {
Cookie cookie = serverProperties.getServlet().getSession().getCookie(); Cookie cookie = serverProperties.getServlet().getSession().getCookie();
DefaultCookieSerializer cookieSerializer = new DefaultCookieSerializer(); DefaultCookieSerializer cookieSerializer = new DefaultCookieSerializer();
PropertyMapper map = PropertyMapper.get().alwaysApplyingWhenNonNull(); PropertyMapper map = PropertyMapper.get().alwaysApplyingWhenNonNull();
...@@ -102,12 +101,21 @@ public class SessionAutoConfiguration { ...@@ -102,12 +101,21 @@ public class SessionAutoConfiguration {
map.from(cookie::getHttpOnly).to(cookieSerializer::setUseHttpOnlyCookie); map.from(cookie::getHttpOnly).to(cookieSerializer::setUseHttpOnlyCookie);
map.from(cookie::getSecure).to(cookieSerializer::setUseSecureCookie); map.from(cookie::getSecure).to(cookieSerializer::setUseSecureCookie);
map.from(cookie::getMaxAge).to((maxAge) -> cookieSerializer.setCookieMaxAge((int) maxAge.getSeconds())); map.from(cookie::getMaxAge).to((maxAge) -> cookieSerializer.setCookieMaxAge((int) maxAge.getSeconds()));
if (ClassUtils.isPresent(REMEMBER_ME_SERVICES_CLASS, getClass().getClassLoader())) { cookieSerializerCustomizers.orderedStream().forEach((customizer) -> customizer.customize(cookieSerializer));
new RememberMeServicesCookieSerializerCustomizer().apply(cookieSerializer);
}
return cookieSerializer; return cookieSerializer;
} }
@Configuration(proxyBeanMethods = false)
@ConditionalOnClass(RememberMeServices.class)
static class RememberMeServicesConfiguration {
@Bean
RememberMeServicesCookieSerializerCustomizer rememberMeServicesCookieSerializerCustomizer() {
return new RememberMeServicesCookieSerializerCustomizer();
}
}
@Configuration(proxyBeanMethods = false) @Configuration(proxyBeanMethods = false)
@ConditionalOnMissingBean(SessionRepository.class) @ConditionalOnMissingBean(SessionRepository.class)
@Import({ ServletSessionRepositoryImplementationValidator.class, @Import({ ServletSessionRepositoryImplementationValidator.class,
...@@ -137,9 +145,10 @@ public class SessionAutoConfiguration { ...@@ -137,9 +145,10 @@ public class SessionAutoConfiguration {
* Customization for {@link SpringSessionRememberMeServices} that is only instantiated * Customization for {@link SpringSessionRememberMeServices} that is only instantiated
* when Spring Security is on the classpath. * when Spring Security is on the classpath.
*/ */
static class RememberMeServicesCookieSerializerCustomizer { static class RememberMeServicesCookieSerializerCustomizer implements CookieSerializerCustomizer {
void apply(DefaultCookieSerializer cookieSerializer) { @Override
public void customize(DefaultCookieSerializer cookieSerializer) {
cookieSerializer.setRememberMeRequestAttribute(SpringSessionRememberMeServices.REMEMBER_ME_LOGIN_ATTR); cookieSerializer.setRememberMeRequestAttribute(SpringSessionRememberMeServices.REMEMBER_ME_LOGIN_ATTR);
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment