Upgrade to Undertow 1.0.15 and fix failing test

Issue: SPR-11777

Author: rstoyanchev

build.gradle

@@ -670,12 +670,12 @@ project("spring-websocket") {
 			exclude group: "javax.servlet", module: "javax.servlet"
 		}
 		optional("org.eclipse.jetty.websocket:websocket-client:${jettyVersion}")
-		optional("io.undertow:undertow-core:1.0.1.Final")
-		optional("io.undertow:undertow-servlet:1.0.1.Final") {
+		optional("io.undertow:undertow-core:1.0.15.Final")
+		optional("io.undertow:undertow-servlet:1.0.15.Final") {
 			exclude group: "org.jboss.spec.javax.servlet", module: "jboss-servlet-api_3.1_spec"
 			exclude group: "org.jboss.spec.javax.annotation", module: "jboss-annotations-api_1.2_spec"
 		}
-		optional("io.undertow:undertow-websockets-jsr:1.0.1.Final") {
+		optional("io.undertow:undertow-websockets-jsr:1.0.15.Final") {
 			exclude group: "org.jboss.spec.javax.websocket", module: "jboss-websocket-api_1.0_spec"
 		}
 		optional("com.fasterxml.jackson.core:jackson-databind:${jackson2Version}")

spring-websocket/src/main/java/org/springframework/web/socket/server/standard/UndertowRequestUpgradeStrategy.java

@@ -32,6 +32,7 @@
 import io.undertow.servlet.api.InstanceHandle;
 import io.undertow.servlet.websockets.ServletWebSocketHttpExchange;
 import io.undertow.websockets.core.WebSocketChannel;
+import io.undertow.websockets.core.WebSocketVersion;
 import io.undertow.websockets.core.protocol.Handshake;
 import io.undertow.websockets.core.protocol.version07.Hybi07Handshake;
 import io.undertow.websockets.core.protocol.version08.Hybi08Handshake;
@@ -41,6 +42,9 @@
 import io.undertow.websockets.jsr.EndpointSessionHandler;
 import io.undertow.websockets.jsr.ServerWebSocketContainer;
 import io.undertow.websockets.jsr.handshake.HandshakeUtil;
+import io.undertow.websockets.jsr.handshake.JsrHybi07Handshake;
+import io.undertow.websockets.jsr.handshake.JsrHybi08Handshake;
+import io.undertow.websockets.jsr.handshake.JsrHybi13Handshake;
 import org.xnio.StreamConnection;
 
 import org.springframework.http.server.ServerHttpRequest;
@@ -57,23 +61,12 @@
  */
 public class UndertowRequestUpgradeStrategy extends AbstractStandardUpgradeStrategy {
 
-	private final Handshake[] handshakes;
+	private final String[] supportedVersions = new String[] {
+			WebSocketVersion.V13.toHttpHeaderValue(),
+			WebSocketVersion.V08.toHttpHeaderValue(),
+			WebSocketVersion.V07.toHttpHeaderValue()
+	};
 
-	private final String[] supportedVersions;
-
-
-	public UndertowRequestUpgradeStrategy() {
-		this.handshakes = new Handshake[] {new Hybi13Handshake(), new Hybi08Handshake(), new Hybi07Handshake()};
-		this.supportedVersions = initSupportedVersions(this.handshakes);
-	}
-
-	private String[] initSupportedVersions(Handshake[] handshakes) {
-		String[] versions = new String[handshakes.length];
-		for (int i = 0; i < versions.length; i++) {
-			versions[i] = handshakes[i].getVersion().toHttpHeaderValue();
-		}
-		return versions;
-	}
 
 	@Override
 	public String[] getSupportedVersions() {
@@ -93,28 +86,34 @@ protected void upgradeInternal(ServerHttpRequest request, ServerHttpResponse res
 		ServerWebSocketContainer wsContainer = (ServerWebSocketContainer) getContainer(servletRequest);
 		final EndpointSessionHandler endpointSessionHandler = new EndpointSessionHandler(wsContainer);
 
-		final Handshake handshake = getHandshakeToUse(exchange);
-
 		final ConfiguredServerEndpoint configuredServerEndpoint = createConfiguredServerEndpoint(
 				selectedProtocol, selectedExtensions, endpoint, servletRequest);
 
+		final Handshake handshake = getHandshakeToUse(exchange, configuredServerEndpoint);
+
 		exchange.upgradeChannel(new HttpUpgradeListener() {
 			@Override
 			public void handleUpgrade(StreamConnection connection, HttpServerExchange serverExchange) {
 				WebSocketChannel channel = handshake.createChannel(exchange, connection, exchange.getBufferPool());
-				HandshakeUtil.setConfig(channel, configuredServerEndpoint);
 				endpointSessionHandler.onConnect(exchange, channel);
 			}
 		});
 
 		handshake.handshake(exchange);
 	}
 
-	private Handshake getHandshakeToUse(ServletWebSocketHttpExchange exchange) {
-		for (Handshake handshake : this.handshakes) {
-			if (handshake.matches(exchange)) {
-				return handshake;
-			}
+	private Handshake getHandshakeToUse(ServletWebSocketHttpExchange exchange, ConfiguredServerEndpoint endpoint) {
+		Handshake handshake = new JsrHybi13Handshake(endpoint);
+		if (handshake.matches(exchange)) {
+			return handshake;
+		}
+		handshake = new JsrHybi08Handshake(endpoint);
+		if (handshake.matches(exchange)) {
+			return handshake;
+		}
+		handshake = new JsrHybi07Handshake(endpoint);
+		if (handshake.matches(exchange)) {
+			return handshake;
 		}
 		// Should never occur
 		throw new HandshakeFailureException("No matching Undertow Handshake found: " + exchange.getRequestHeaders());