fix processing multiple set-cookie headers

SurenSritharan · Mar 2, 2023 · 58669d2 · 58669d2
1 parent 94060bd
commit 58669d2
Showing 1 changed file with 12 additions and 7 deletions.
diff --git a/src/main/java/com/blazemeter/jmeter/http2/core/HTTP2JettyClient.java b/src/main/java/com/blazemeter/jmeter/http2/core/HTTP2JettyClient.java
@@ -595,13 +595,18 @@ private String extractRedirectLocation(ContentResponse contentResponse) {
     return redirectLocation;
   }
 
-  private void saveCookiesInCookieManager(ContentResponse response, URL url,
-      CookieManager cookieManager) {
-    String cookieHeader = response.getHeaders().get(HTTPConstants.HEADER_SET_COOKIE);
-    if (cookieHeader != null && cookieManager != null) {
-      cookieManager.addCookieFromHeader(cookieHeader, url);
-    }
-  }
+	private void saveCookiesInCookieManager(ContentResponse response, URL url, CookieManager cookieManager) {
+		Iterator<HttpField> iterator = response.getHeaders().iterator();
+		while (iterator.hasNext()) {
+			HttpField field = iterator.next();
+			if (field.is(HTTPConstants.HEADER_SET_COOKIE)) {
+				String cookieHeader = field.getValue();
+				if (cookieHeader != null && cookieManager != null) {
+					cookieManager.addCookieFromHeader(cookieHeader, url);
+				}
+			}
+		}
+	}
 
   public void clearCookies() {
     httpClient.getCookieStore().removeAll();