mirror of
https://github.com/spring-projects/spring-framework
synced 2026-06-08 17:33:33 +00:00
Polishing contribution
Closes gh-34683
This commit is contained in:
+5
-40
@@ -67,8 +67,6 @@ class ForwardedHeaderFilterTests {
|
||||
|
||||
private static final String X_FORWARDED_FOR = "x-forwarded-for";
|
||||
|
||||
private static final String X_FORWARDED_BY = "x-forwarded-by";
|
||||
|
||||
|
||||
private final ForwardedHeaderFilter filter = new ForwardedHeaderFilter();
|
||||
|
||||
@@ -96,7 +94,6 @@ class ForwardedHeaderFilterTests {
|
||||
testShouldFilter(X_FORWARDED_SSL);
|
||||
testShouldFilter(X_FORWARDED_PREFIX);
|
||||
testShouldFilter(X_FORWARDED_FOR);
|
||||
testShouldFilter(X_FORWARDED_BY);
|
||||
}
|
||||
|
||||
private void testShouldFilter(String headerName) {
|
||||
@@ -119,7 +116,6 @@ class ForwardedHeaderFilterTests {
|
||||
this.request.addHeader(X_FORWARDED_PORT, "443");
|
||||
this.request.addHeader("foo", "bar");
|
||||
this.request.addHeader(X_FORWARDED_FOR, "[203.0.113.195]");
|
||||
this.request.addHeader(X_FORWARDED_BY, "[203.0.113.196]");
|
||||
|
||||
this.filter.doFilter(this.request, new MockHttpServletResponse(), this.filterChain);
|
||||
HttpServletRequest actual = (HttpServletRequest) this.filterChain.getRequest();
|
||||
@@ -131,13 +127,11 @@ class ForwardedHeaderFilterTests {
|
||||
assertThat(actual.getServerPort()).isEqualTo(443);
|
||||
assertThat(actual.isSecure()).isTrue();
|
||||
assertThat(actual.getRemoteAddr()).isEqualTo(actual.getRemoteHost()).isEqualTo("[203.0.113.195]");
|
||||
assertThat(actual.getLocalAddr()).isEqualTo(actual.getLocalAddr()).isEqualTo("[203.0.113.196]");
|
||||
|
||||
assertThat(actual.getHeader(X_FORWARDED_PROTO)).isNull();
|
||||
assertThat(actual.getHeader(X_FORWARDED_HOST)).isNull();
|
||||
assertThat(actual.getHeader(X_FORWARDED_PORT)).isNull();
|
||||
assertThat(actual.getHeader(X_FORWARDED_FOR)).isNull();
|
||||
assertThat(actual.getHeader(X_FORWARDED_BY)).isNull();
|
||||
assertThat(actual.getHeader("foo")).isEqualTo("bar");
|
||||
}
|
||||
|
||||
@@ -150,7 +144,6 @@ class ForwardedHeaderFilterTests {
|
||||
this.request.addHeader(X_FORWARDED_SSL, "on");
|
||||
this.request.addHeader("foo", "bar");
|
||||
this.request.addHeader(X_FORWARDED_FOR, "203.0.113.195");
|
||||
this.request.addHeader(X_FORWARDED_BY, "203.0.113.196");
|
||||
|
||||
this.filter.setRemoveOnly(true);
|
||||
this.filter.doFilter(this.request, new MockHttpServletResponse(), this.filterChain);
|
||||
@@ -171,7 +164,6 @@ class ForwardedHeaderFilterTests {
|
||||
assertThat(actual.getHeader(X_FORWARDED_PORT)).isNull();
|
||||
assertThat(actual.getHeader(X_FORWARDED_SSL)).isNull();
|
||||
assertThat(actual.getHeader(X_FORWARDED_FOR)).isNull();
|
||||
assertThat(actual.getHeader(X_FORWARDED_BY)).isNull();
|
||||
assertThat(actual.getHeader("foo")).isEqualTo("bar");
|
||||
}
|
||||
|
||||
@@ -555,34 +547,7 @@ class ForwardedHeaderFilterTests {
|
||||
class ForwardedBy {
|
||||
|
||||
@Test
|
||||
void xForwardedForEmpty() throws Exception {
|
||||
request.addHeader(X_FORWARDED_BY, "");
|
||||
HttpServletRequest actual = filterAndGetWrappedRequest();
|
||||
|
||||
assertThat(actual.getLocalAddr()).isEqualTo(MockHttpServletRequest.DEFAULT_SERVER_ADDR);
|
||||
assertThat(actual.getLocalPort()).isEqualTo(MockHttpServletRequest.DEFAULT_SERVER_PORT);
|
||||
}
|
||||
|
||||
@Test
|
||||
void xForwardedForSingleIdentifier() throws Exception {
|
||||
request.addHeader(X_FORWARDED_BY, "203.0.113.195");
|
||||
HttpServletRequest actual = filterAndGetWrappedRequest();
|
||||
|
||||
assertThat(actual.getLocalAddr()).isEqualTo(actual.getLocalAddr()).isEqualTo("203.0.113.195");
|
||||
assertThat(actual.getLocalPort()).isEqualTo(MockHttpServletRequest.DEFAULT_SERVER_PORT);
|
||||
}
|
||||
|
||||
@Test
|
||||
void xForwardedForMultipleIdentifiers() throws Exception {
|
||||
request.addHeader(X_FORWARDED_BY, "203.0.113.195, 70.41.3.18, 150.172.238.178");
|
||||
HttpServletRequest actual = filterAndGetWrappedRequest();
|
||||
|
||||
assertThat(actual.getLocalAddr()).isEqualTo(actual.getLocalAddr()).isEqualTo("203.0.113.195");
|
||||
assertThat(actual.getLocalPort()).isEqualTo(MockHttpServletRequest.DEFAULT_SERVER_PORT);
|
||||
}
|
||||
|
||||
@Test
|
||||
void forwardedForIpV4Identifier() throws Exception {
|
||||
void forwardedByIpV4Identifier() throws Exception {
|
||||
request.addHeader(FORWARDED, "By=203.0.113.195");
|
||||
HttpServletRequest actual = filterAndGetWrappedRequest();
|
||||
|
||||
@@ -591,7 +556,7 @@ class ForwardedHeaderFilterTests {
|
||||
}
|
||||
|
||||
@Test
|
||||
void forwardedForIpV6Identifier() throws Exception {
|
||||
void forwardedByIpV6Identifier() throws Exception {
|
||||
request.addHeader(FORWARDED, "By=\"[2001:db8:cafe::17]\"");
|
||||
HttpServletRequest actual = filterAndGetWrappedRequest();
|
||||
|
||||
@@ -600,7 +565,7 @@ class ForwardedHeaderFilterTests {
|
||||
}
|
||||
|
||||
@Test
|
||||
void forwardedForIpV4IdentifierWithPort() throws Exception {
|
||||
void forwardedByIpV4IdentifierWithPort() throws Exception {
|
||||
request.addHeader(FORWARDED, "By=\"203.0.113.195:47011\"");
|
||||
HttpServletRequest actual = filterAndGetWrappedRequest();
|
||||
|
||||
@@ -609,7 +574,7 @@ class ForwardedHeaderFilterTests {
|
||||
}
|
||||
|
||||
@Test
|
||||
void forwardedForIpV6IdentifierWithPort() throws Exception {
|
||||
void forwardedByIpV6IdentifierWithPort() throws Exception {
|
||||
request.addHeader(FORWARDED, "By=\"[2001:db8:cafe::17]:47011\"");
|
||||
HttpServletRequest actual = filterAndGetWrappedRequest();
|
||||
|
||||
@@ -618,7 +583,7 @@ class ForwardedHeaderFilterTests {
|
||||
}
|
||||
|
||||
@Test
|
||||
void forwardedForMultipleIdentifiers() throws Exception {
|
||||
void forwardedByMultipleIdentifiers() throws Exception {
|
||||
request.addHeader(FORWARDED, "by=203.0.113.195;proto=http, by=\"[2001:db8:cafe::17]\", by=unknown");
|
||||
HttpServletRequest actual = filterAndGetWrappedRequest();
|
||||
|
||||
|
||||
-16
@@ -53,7 +53,6 @@ class ForwardedHeaderTransformerTests {
|
||||
headers.add("X-Forwarded-Prefix", "prefix");
|
||||
headers.add("X-Forwarded-Ssl", "on");
|
||||
headers.add("X-Forwarded-For", "203.0.113.195");
|
||||
headers.add("X-Forwarded-By", "203.0.113.196");
|
||||
ServerHttpRequest request = this.requestMutator.apply(getRequest(headers));
|
||||
|
||||
assertForwardedHeadersRemoved(request);
|
||||
@@ -254,21 +253,6 @@ class ForwardedHeaderTransformerTests {
|
||||
assertThat(request.getLocalAddress().getPort()).isEqualTo(4711);
|
||||
}
|
||||
|
||||
@Test
|
||||
void xForwardedBy() {
|
||||
HttpHeaders headers = new HttpHeaders();
|
||||
headers.add("x-forwarded-by", "203.0.113.195, 70.41.3.18, 150.172.238.178");
|
||||
|
||||
ServerHttpRequest request = MockServerHttpRequest
|
||||
.method(HttpMethod.GET, URI.create("https://example.com/a%20b?q=a%2Bb"))
|
||||
.headers(headers)
|
||||
.build();
|
||||
|
||||
request = this.requestMutator.apply(request);
|
||||
assertThat(request.getLocalAddress()).isNotNull();
|
||||
assertThat(request.getLocalAddress().getHostName()).isEqualTo("203.0.113.195");
|
||||
}
|
||||
|
||||
|
||||
private MockServerHttpRequest getRequest(HttpHeaders headers) {
|
||||
return MockServerHttpRequest.get(BASE_URL).headers(headers).build();
|
||||
|
||||
@@ -551,4 +551,15 @@ class ForwardedHeaderUtilsTests {
|
||||
assertThat(address.getHostName()).isEqualTo("[fd00:fefe:1::4]");
|
||||
}
|
||||
|
||||
@Test
|
||||
void parseForwardedByHeader() {
|
||||
HttpHeaders headers = new HttpHeaders();
|
||||
headers.add("Forwarded", "by=[fd00:fefe:1::4], 192.168.0.1");
|
||||
|
||||
InetSocketAddress address =
|
||||
ForwardedHeaderUtils.parseForwardedBy(URI.create("https://example.com"), headers, null);
|
||||
|
||||
assertThat(address.getHostName()).isEqualTo("[fd00:fefe:1::4]");
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user