Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -113,10 +113,19 @@ public void saveToken(CsrfToken token, HttpServletRequest request, HttpServletRe
token = generateToken(request);
expire = true;
}
Cookie csrfCookie = new Cookie(token.getParameterName(), token.getToken());

boolean isSecure = secure || "https".equals(request.getScheme());
String baseName = token.getParameterName().startsWith("__Host-") ? token.getParameterName().substring("__Host-".length()) : token.getParameterName();
String cookieName = isSecure ? "__Host-" + baseName : baseName;

Cookie csrfCookie = new Cookie(cookieName, token.getToken());
csrfCookie.setHttpOnly(true);
csrfCookie.setSecure(secure || "https".equals(request.getScheme()));
csrfCookie.setPath(ofNullable(request.getContextPath()).orElse("") + "/");
csrfCookie.setSecure(isSecure);
if (cookieName.startsWith("__Host-")) {
csrfCookie.setPath("/");
} else {
csrfCookie.setPath(ofNullable(request.getContextPath()).orElse("") + "/");
}
Comment thread
joemahady-comm marked this conversation as resolved.
if (expire) {
csrfCookie.setMaxAge(0);
} else {
Expand All @@ -133,8 +142,11 @@ public CsrfToken loadToken(HttpServletRequest request) {
if (requiresCsrfProtection) {
Cookie[] cookies = request.getCookies();
if (cookies != null) {
for (Cookie cookie : request.getCookies()) {
if (getParameterName().equals(cookie.getName())) {
boolean isSecure = secure || "https".equals(request.getScheme());
String baseName = getParameterName().startsWith("__Host-") ? getParameterName().substring("__Host-".length()) : getParameterName();
String expectedCookieName = isSecure ? "__Host-" + baseName : baseName;
for (Cookie cookie : cookies) {
if (expectedCookieName.equals(cookie.getName())) {
return new DefaultCsrfToken(getHeaderName(), getParameterName(), cookie.getValue());
}
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -166,6 +166,21 @@ private MockHttpServletResponse saveTokenAndReturnResponse(boolean isSecure, Str
}

private Cookie saveTokenAndReturnCookie(boolean isSecure, String protocol) {
return saveTokenAndReturnResponse(isSecure, protocol).getCookie("X-Uaa-Csrf");
MockHttpServletResponse response = saveTokenAndReturnResponse(isSecure, protocol);
boolean expectSecure = isSecure || "https".equals(protocol);
String expectedCookieName = expectSecure ? "__Host-X-Uaa-Csrf" : "X-Uaa-Csrf";

String setCookie = response.getHeader("Set-Cookie");
if (setCookie == null || !setCookie.contains(expectedCookieName + "=")) {
throw new IllegalStateException("Expected Set-Cookie header for " + expectedCookieName + " but was: " + setCookie);
}

int valueStart = setCookie.indexOf(expectedCookieName + "=") + (expectedCookieName + "=").length();
int valueEnd = setCookie.indexOf(';', valueStart);
String value = valueEnd >= 0 ? setCookie.substring(valueStart, valueEnd) : setCookie.substring(valueStart);
Cookie cookie = new Cookie(expectedCookieName, value);
cookie.setSecure(setCookie.contains("Secure"));
cookie.setHttpOnly(setCookie.contains("HttpOnly"));
return cookie;
}
}
Loading