Spring/spring-framework
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Retain brackets for IPV6 address in MockHttpServletRequest

Browse Source 
According to the Javadoc for ServletRequest's getServerName() method,
when the `Host` header is set, the server name is "the value of the
part before ':' in the Host header value ...". For a value representing
an IPV6 address such as `[::ffff:abcd:abcd]`, the enclosing square
brackets should therefore not be stripped from the enclosed IPV6
address.

However, the changes made in conjunction with gh-16704 introduced a
regression in Spring Framework 4.1 for the getServerName() method in
MockHttpServletRequest by stripping the enclosing brackets from the
IPV6 address in the `Host` header. Similarly, the changes made in
conjunction with gh-20686 introduced a regression in Spring Framework
4.3.13 and 5.0.2 in the getRequestURL() method in
MockHttpServletRequest by delegating to the getServerName() method
which strips the enclosing brackets.

This commit fixes the implementation of getServerName() so that the
enclosing brackets are no longer stripped from an IPV6 address in the
`Host` header. The implementation of getRequestURL() is therefore also
fixed.

In addition, in order to avoid a NullPointerException, the
implementations of getServerName() and getServerPort() now assert that
an IPV6 address present in the `Host` header correctly contains an
opening and closing bracket and throw an IllegalStateException if that
is not the case.

Closes gh-24916
This commit is contained in:
Sam Brannen
2020-04-17 15:01:35 +02:00
parent 0017256968
commit 6b410df45b
2 changed files with 79 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

76
spring-test/src/test/java/org/springframework/mock/web/MockHttpServletRequestTests.java
View File

@@ -1,5 +1,5 @@
/*
* Copyright 2002-2019 the original author or authors.
* Copyright 2002-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.
@@ -17,6 +17,7 @@
package org.springframework.mock.web;
import java.io.IOException;
import java.net.URL;
import java.nio.charset.Charset;
import java.util.ArrayList;
import java.util.Arrays;
@@ -37,6 +38,7 @@ import org.springframework.util.FileCopyUtils;
import org.springframework.util.StreamUtils;
import static org.assertj.core.api.Assertions.assertThat;
import static org.assertj.core.api.Assertions.assertThatExceptionOfType;
import static org.assertj.core.api.Assertions.assertThatIllegalArgumentException;
import static org.assertj.core.api.Assertions.assertThatIllegalStateException;
@@ -389,18 +391,25 @@ class MockHttpServletRequestTests {
assertThat(request.getServerName()).isEqualTo(testServer);
}
@Test
void getServerNameWithInvalidIpv6AddressViaHostHeader() {
request.addHeader(HOST, "[::ffff:abcd:abcd"); // missing closing bracket
assertThatIllegalStateException()
.isThrownBy(() -> request.getServerName())
.withMessageStartingWith("Invalid Host header: ");
}
@Test
void getServerNameViaHostHeaderAsIpv6AddressWithoutPort() {
String ipv6Address = "[2001:db8:0:1]";
request.addHeader(HOST, ipv6Address);
assertThat(request.getServerName()).isEqualTo("2001:db8:0:1");
String host = "[2001:db8:0:1]";
request.addHeader(HOST, host);
assertThat(request.getServerName()).isEqualTo(host);
}
@Test
void getServerNameViaHostHeaderAsIpv6AddressWithPort() {
String ipv6Address = "[2001:db8:0:1]:8081";
request.addHeader(HOST, ipv6Address);
assertThat(request.getServerName()).isEqualTo("2001:db8:0:1");
request.addHeader(HOST, "[2001:db8:0:1]:8081");
assertThat(request.getServerName()).isEqualTo("[2001:db8:0:1]");
}
@Test
@@ -414,6 +423,22 @@ class MockHttpServletRequestTests {
assertThat(request.getServerPort()).isEqualTo(8080);
}
@Test
void getServerPortWithInvalidIpv6AddressViaHostHeader() {
request.addHeader(HOST, "[::ffff:abcd:abcd:8080"); // missing closing bracket
assertThatIllegalStateException()
.isThrownBy(() -> request.getServerPort())
.withMessageStartingWith("Invalid Host header: ");
}
@Test
void getServerPortWithIpv6AddressAndInvalidPortViaHostHeader() {
request.addHeader(HOST, "[::ffff:abcd:abcd]:bogus"); // "bogus" is not a port number
assertThatExceptionOfType(NumberFormatException.class)
.isThrownBy(() -> request.getServerPort())
.withMessageContaining("bogus");
}
@Test
void getServerPortViaHostHeaderAsIpv6AddressWithoutPort() {
String testServer = "[2001:db8:0:1]";
@@ -478,6 +503,43 @@ class MockHttpServletRequestTests {
assertThat(requestURL.toString()).isEqualTo(("http://" + testServer));
}
@Test
void getRequestURLWithIpv6AddressViaServerNameWithoutPort() throws Exception {
request.setServerName("[::ffff:abcd:abcd]");
URL url = new java.net.URL(request.getRequestURL().toString());
assertThat(url).asString().isEqualTo("http://[::ffff:abcd:abcd]");
}
@Test
void getRequestURLWithIpv6AddressViaServerNameWithPort() throws Exception {
request.setServerName("[::ffff:abcd:abcd]");
request.setServerPort(9999);
URL url = new java.net.URL(request.getRequestURL().toString());
assertThat(url).asString().isEqualTo("http://[::ffff:abcd:abcd]:9999");
}
@Test
void getRequestURLWithInvalidIpv6AddressViaHostHeader() {
request.addHeader(HOST, "[::ffff:abcd:abcd"); // missing closing bracket
assertThatIllegalStateException()
.isThrownBy(() -> request.getRequestURL())
.withMessageStartingWith("Invalid Host header: ");
}
@Test
void getRequestURLWithIpv6AddressViaHostHeaderWithoutPort() throws Exception {
request.addHeader(HOST, "[::ffff:abcd:abcd]");
URL url = new java.net.URL(request.getRequestURL().toString());
assertThat(url).asString().isEqualTo("http://[::ffff:abcd:abcd]");
}
@Test
void getRequestURLWithIpv6AddressViaHostHeaderWithPort() throws Exception {
request.addHeader(HOST, "[::ffff:abcd:abcd]:9999");
URL url = new java.net.URL(request.getRequestURL().toString());
assertThat(url).asString().isEqualTo("http://[::ffff:abcd:abcd]:9999");
}
@Test
void getRequestURLWithNullRequestUri() {
request.setRequestURI(null);