import com.google.gson.GsonBuilder;
import java.io.UnsupportedEncodingException;
import java.lang.reflect.Type;
+import java.nio.file.Path;
import java.util.HashMap;
import java.util.Map;
import java.util.Optional;
return empty();
}
- String sanitizedUrl = url.trim();
- boolean isValidUrl = VALID_RETURN_TO.matcher(sanitizedUrl).matches();
+ String trimmedUrl = url.trim();
+ boolean isValidUrl = VALID_RETURN_TO.matcher(trimmedUrl).matches();
if (!isValidUrl) {
return empty();
}
- return Optional.of(sanitizedUrl);
+ Path sanitizedPath = escapePathTraversalChars(trimmedUrl);
+ return Optional.of(sanitizedPath.toString());
+ }
+
+ private static Path escapePathTraversalChars(String sanitizedUrl) {
+ return Path.of(sanitizedUrl).normalize();
}
}
import com.tngtech.java.junit.dataprovider.DataProvider;
import com.tngtech.java.junit.dataprovider.DataProviderRunner;
+import com.tngtech.java.junit.dataprovider.UseDataProvider;
import java.util.Optional;
+import javax.annotation.Nullable;
import javax.servlet.http.Cookie;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
@Test
public void init_create_cookie() {
- when(request.getParameter("return_to")).thenReturn("/settings");
+ when(request.getParameter("return_to")).thenReturn("/admin/settings");
underTest.init(request, response);
@Test
public void get_return_to_is_empty_when_no_cookie() {
- when(request.getCookies()).thenReturn(new Cookie[] {});
+ when(request.getCookies()).thenReturn(new Cookie[]{});
Optional<String> redirection = underTest.getReturnTo(request);
@Test
public void get_return_to_is_empty_when_no_value() {
- when(request.getCookies()).thenReturn(new Cookie[] {new Cookie(AUTHENTICATION_COOKIE_NAME, "{}")});
+ when(request.getCookies()).thenReturn(new Cookie[]{new Cookie(AUTHENTICATION_COOKIE_NAME, "{}")});
Optional<String> redirection = underTest.getReturnTo(request);
@Test
public void delete() {
- when(request.getCookies()).thenReturn(new Cookie[] {new Cookie(AUTHENTICATION_COOKIE_NAME, "{\"return_to\":\"/settings\"}")});
+ when(request.getCookies()).thenReturn(new Cookie[]{new Cookie(AUTHENTICATION_COOKIE_NAME, "{\"return_to\":\"/settings\"}")});
underTest.delete(request, response);
assertThat(updatedCookie.getPath()).isEqualTo("/");
assertThat(updatedCookie.getMaxAge()).isZero();
}
+
+ @DataProvider
+ public static Object[][] payloadToSanitizeAndExpectedOutcome() {
+ return new Object[][]{
+ {generatePath("/admin/settings"), "/admin/settings"},
+ {generatePath("/admin/../../settings"), "/settings"},
+ {generatePath("/admin/../settings"), "/settings"},
+ {generatePath("/admin/settings/.."), "/admin"},
+ {generatePath("/admin/..%2fsettings/"), "/settings"},
+ {generatePath("/admin/%2e%2e%2fsettings/"), "/settings"},
+ {generatePath("../admin/settings"), null},
+ };
+ }
+
+ private static String generatePath(String returnTo) {
+ return "{\"return_to\":\"" + returnTo + "\"}";
+ }
+
+ @Test
+ @UseDataProvider("payloadToSanitizeAndExpectedOutcome")
+ public void getReturnTo_whenContainingPathTraversalCharacters_sanitizeThem(String payload, @Nullable String expectedSanitizedUrl) {
+ when(request.getCookies()).thenReturn(new Cookie[]{wrapCookie(AUTHENTICATION_COOKIE_NAME, payload)});
+
+ Optional<String> redirection = underTest.getReturnTo(request);
+
+ assertThat(redirection).isEqualTo(Optional.ofNullable(expectedSanitizedUrl));
+ }
+
+ private Cookie wrapCookie(String name, String value) {
+ return new javax.servlet.http.Cookie(name, value);
+ }
}