Cwikius-Spring
/
spring-security
mirror of https://github.com/spring-projects/spring-security.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

ServerRedirectCache.getRequest->getRedirectUri 

Issue: gh-4789
This commit is contained in:
Rob Winch 2017-11-15 13:21:25 -06:00
parent 4039cd285e
commit 64ad08e96d
5 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
web/src/main/java/org/springframework/security/web/server/authentication/RedirectServerAuthenticationSuccessHandler.java
View File

@ -55,7 +55,7 @@ public class RedirectServerAuthenticationSuccessHandler
public Mono<Void> onAuthenticationSuccess(WebFilterExchange webFilterExchange,
Authentication authentication) {
ServerWebExchange exchange = webFilterExchange.getExchange();
return this.requestCache.getRequest(exchange)
return this.requestCache.getRedirectUri(exchange)
.defaultIfEmpty(this.location)
.flatMap(location -> this.redirectStrategy.sendRedirect(exchange, location));
}

2
web/src/main/java/org/springframework/security/web/server/savedrequest/NoOpServerRequestCache.java
View File

@ -33,7 +33,7 @@ public class NoOpServerRequestCache implements ServerRequestCache {
}
@Override
public Mono<URI> getRequest(ServerWebExchange exchange) {
public Mono<URI> getRedirectUri(ServerWebExchange exchange) {
return Mono.empty();
}

6
web/src/main/java/org/springframework/security/web/server/savedrequest/ServerRequestCache.java
View File

@ -40,11 +40,11 @@ public interface ServerRequestCache {
Mono<Void> saveRequest(ServerWebExchange exchange);
/**
* Get the saved {@link ServerHttpRequest}
* Get the URI that can be redirected to trigger the saved request to be used
* @param exchange the exchange to obtain the saved {@link ServerHttpRequest} from
* @return the {@link ServerHttpRequest}
* @return the URI that can be redirected to trigger the saved request to be used
*/
Mono<URI> getRequest(ServerWebExchange exchange);
Mono<URI> getRedirectUri(ServerWebExchange exchange);
/**
* If the provided {@link ServerWebExchange} matches the saved {@link ServerHttpRequest}

4
web/src/main/java/org/springframework/security/web/server/savedrequest/WebSessionServerRequestCache.java
View File

@ -70,7 +70,7 @@ public class WebSessionServerRequestCache implements ServerRequestCache {
}
@Override
public Mono<URI> getRequest(ServerWebExchange exchange) {
public Mono<URI> getRedirectUri(ServerWebExchange exchange) {
return exchange.getSession()
.flatMap(session -> Mono.justOrEmpty(session.<String>getAttribute(this.sessionAttrName)))
.map(URI::create);
@ -79,7 +79,7 @@ public class WebSessionServerRequestCache implements ServerRequestCache {
@Override
public Mono<ServerHttpRequest> getMatchingRequest(
ServerWebExchange exchange) {
return getRequest(exchange)
return getRedirectUri(exchange)
.map(URI::toASCIIString)
.map(path -> exchange.getRequest().mutate().path(path).build())
.filter( request -> pathInApplication(request).equals(

8
web/src/test/java/org/springframework/security/web/server/savedrequest/WebSessionServerRequestCacheTests.java
View File

@ -38,7 +38,7 @@ public class WebSessionServerRequestCacheTests {
MockServerWebExchange exchange = MockServerWebExchange.from(MockServerHttpRequest.get("/secured/"));
this.cache.saveRequest(exchange).block();
URI saved = this.cache.getRequest(exchange).block();
URI saved = this.cache.getRedirectUri(exchange).block();
assertThat(saved).isEqualTo(exchange.getRequest().getURI());
}
@ -48,7 +48,7 @@ public class WebSessionServerRequestCacheTests {
MockServerWebExchange exchange = MockServerWebExchange.from(MockServerHttpRequest.post("/secured/"));
this.cache.saveRequest(exchange).block();
assertThat(this.cache.getRequest(exchange).block()).isNull();
assertThat(this.cache.getRedirectUri(exchange).block()).isNull();
}
@Test
@ -57,7 +57,7 @@ public class WebSessionServerRequestCacheTests {
MockServerWebExchange exchange = MockServerWebExchange.from(MockServerHttpRequest.post("/secured/"));
this.cache.saveRequest(exchange).block();
URI saved = this.cache.getRequest(exchange).block();
URI saved = this.cache.getRedirectUri(exchange).block();
assertThat(saved).isEqualTo(exchange.getRequest().getURI());
}
@ -79,6 +79,6 @@ public class WebSessionServerRequestCacheTests {
this.cache.removeRequest(exchange).block();
assertThat(this.cache.getRequest(exchange).block()).isNull();
assertThat(this.cache.getRedirectUri(exchange).block()).isNull();
}
}