Merge pull request #1821 from fl4via/backport-features_2.4.x

[UNDERTOW-2634 / 1901 / 1870 / 1904 / 2254 / 2249] Backport features to 2.4.x

Author: fl4via

core/src/main/java/io/undertow/client/ajp/AjpClientConnection.java

@@ -237,6 +237,10 @@ public void sendRequest(final ClientRequest request, final ClientCallback<Client
             clientCallback.failed(UndertowClientMessages.MESSAGES.invalidConnectionState());
             return;
         }
+        if (anyAreSet(state, CLOSE_REQ | CLOSED)) {
+            clientCallback.failed(new ClosedChannelException());
+            return;
+        }
         final AjpClientExchange AjpClientExchange = new AjpClientExchange(clientCallback, request, this);
         if (currentRequest == null) {
             initiateRequest(AjpClientExchange);

core/src/main/java/io/undertow/client/http/HttpClientConnection.java

@@ -352,6 +352,10 @@ public void sendRequest(final ClientRequest request, final ClientCallback<Client
             clientCallback.failed(UndertowClientMessages.MESSAGES.invalidConnectionState());
             return;
         }
+        if (anyAreSet(state, CLOSE_REQ | CLOSED)) {
+            clientCallback.failed(new ClosedChannelException());
+            return;
+        }
         final HttpClientExchange httpClientExchange = new HttpClientExchange(clientCallback, request, this);
         boolean ssl = this.connection instanceof SslConnection;
         if(!ssl && !http2Tried && options.get(UndertowOptions.ENABLE_HTTP2, false) && !request.getRequestHeaders().contains(Headers.UPGRADE)) {

core/src/main/java/io/undertow/server/HttpServerExchange.java

@@ -40,6 +40,7 @@
 import io.undertow.util.ConduitFactory;
 import io.undertow.util.Cookies;
 import io.undertow.util.HeaderMap;
+import io.undertow.util.HeaderValues;
 import io.undertow.util.Headers;
 import io.undertow.util.HttpString;
 import io.undertow.util.Methods;
@@ -1929,6 +1930,16 @@ public XnioIoThread getIoThread() {
         return connection.getIoThread();
     }
 
+    public boolean isMultiPartExchange() {
+        //NOTE: should this include Range response?
+        final HeaderValues contentTypeHeaders = getRequestHeaders().get("Content-Type");
+        if(contentTypeHeaders != null && contentTypeHeaders.size() >0) {
+            return contentTypeHeaders.getFirst().startsWith("multipart");
+        } else {
+            return false;
+        }
+    }
+
     /**
      * @return The maximum entity size for this exchange
      */

core/src/main/java/io/undertow/server/handlers/proxy/LoadBalancingProxyClient.java

@@ -360,7 +360,7 @@ protected Host selectHost(HttpServerExchange exchange) {
             }
         }
 
-        int host = hostSelector.selectHost(hosts);
+        int host = hostSelector.selectHost(hosts, exchange);
 
         final int startHost = host; //if the all hosts have problems we come back to this one
         Host full = null;
@@ -473,16 +473,27 @@ private static class ExclusiveConnectionHolder {
 
     public interface HostSelector {
 
+        @Deprecated(forRemoval = true)
         int selectHost(Host[] availableHosts);
+
+        int selectHost(Host[] availableHosts, HttpServerExchange exchange);
+
     }
 
     static class RoundRobinHostSelector implements HostSelector {
 
         private final AtomicInteger currentHost = new AtomicInteger(0);
 
         @Override
+        @Deprecated(forRemoval = true)
         public int selectHost(Host[] availableHosts) {
             return currentHost.incrementAndGet() % availableHosts.length;
         }
+
+        @Override
+        public int selectHost(Host[] availableHosts, HttpServerExchange exchange) {
+            return selectHost(availableHosts);
+        }
+
     }
 }

core/src/main/java/io/undertow/server/session/InMemorySessionManager.java

@@ -671,5 +671,10 @@ private synchronized void destroy() {
             cancelTask = null;
         }
 
+        @Override
+        public boolean isInvalid() {
+            return this.invalid;
+        }
+
     }
 }

core/src/main/java/io/undertow/server/session/Session.java

@@ -183,4 +183,10 @@ public interface Session {
      * @return The new session ID
      */
     String changeSessionId(HttpServerExchange exchange, SessionConfig config);
+
+    /**
+     * Return state of session. NOTE: this method does not take into account if process of invalidation start.
+     * @return <ul><li><b>true</b> - if session is no longer valid</li><li><b>false</b> - otherwise</li></ul>
+     */
+    boolean isInvalid();
 }

core/src/main/java/io/undertow/util/MimeMappings.java

@@ -38,6 +38,7 @@ public class MimeMappings {
         Map<String, String> defaultMappings = new HashMap<>(101);
         defaultMappings.put("txt", "text/plain");
         defaultMappings.put("css", "text/css");
+        defaultMappings.put("csv", "text/csv");
         defaultMappings.put("html", "text/html");
         defaultMappings.put("htm", "text/html");
         defaultMappings.put("gif", "image/gif");
@@ -119,6 +120,10 @@ public class MimeMappings {
         defaultMappings.put("mpeg", "video/mpeg");
         defaultMappings.put("mpg", "video/mpeg");
         defaultMappings.put("mpe", "video/mpeg");
+        defaultMappings.put("mp4", "video/mp4");
+        defaultMappings.put("webm", "video/webm");
+        defaultMappings.put("flac", "audio/flac");
+        defaultMappings.put("weba", "audio/webm");
         defaultMappings.put("qt", "video/quicktime");
         defaultMappings.put("mov", "video/quicktime");
         defaultMappings.put("avi", "video/x-msvideo");
@@ -127,7 +132,6 @@ public class MimeMappings {
         defaultMappings.put("wrl", "x-world/x-vrml");
         defaultMappings.put("mpv2", "video/mpeg2");
         defaultMappings.put("jnlp", "application/x-java-jnlp-file");
-
         defaultMappings.put("eot", "application/vnd.ms-fontobject");
         defaultMappings.put("woff", "application/font-woff");
         defaultMappings.put("woff2", "application/font-woff2");

core/src/test/java/io/undertow/server/handlers/proxy/LoadBalancingProxyWithCustomHostSelectorTestCase.java

@@ -2,6 +2,7 @@
 
 import io.undertow.Undertow;
 import io.undertow.client.UndertowClient;
+import io.undertow.server.HttpServerExchange;
 import io.undertow.server.session.InMemorySessionManager;
 import io.undertow.server.session.SessionAttachmentHandler;
 import io.undertow.server.session.SessionCookieConfig;
@@ -53,10 +54,17 @@ public static void setup() throws URISyntaxException {
         server2.start();
 
         LoadBalancingProxyClient.HostSelector hostSelector = new LoadBalancingProxyClient.HostSelector() {
+
             @Override
             public int selectHost(LoadBalancingProxyClient.Host[] availableHosts) {
                 return 0;
             }
+
+            @Override
+            public int selectHost(LoadBalancingProxyClient.Host[] availableHosts, HttpServerExchange exchange) {
+                return 0;
+            }
+
         };
 
         DefaultServer.setRootHandler(ProxyHandler.builder().setProxyClient(new LoadBalancingProxyClient(UndertowClient.getInstance(), null, hostSelector)

servlet/src/main/java/io/undertow/servlet/api/DeploymentInfo.java

@@ -86,6 +86,7 @@ public class DeploymentInfo implements Cloneable {
     private ConfidentialPortManager confidentialPortManager;
     private boolean allowNonStandardWrappers = false;
     private int defaultSessionTimeout = 60 * 30;
+    private long defaultAsyncContextTimeout = 30000;
     private ConcurrentMap<String, Object> servletContextAttributeBackingMap;
     private ServletSessionConfig servletSessionConfig;
     private String hostName = "localhost";
@@ -319,6 +320,18 @@ public DeploymentInfo setDefaultSessionTimeout(final int defaultSessionTimeout)
         return this;
     }
 
+    public long getDefaultAsyncConextTimeout() {
+        return defaultAsyncContextTimeout;
+    }
+
+    /**
+     * @param defaultAsyncContextTimeout The default async context timeout, in milliseconds
+     */
+    public DeploymentInfo setDefaultAsyncConextTimeout(final long defaultAsyncContextTimeout) {
+        this.defaultAsyncContextTimeout = defaultAsyncContextTimeout;
+        return this;
+    }
+
     public String getDefaultEncoding() {
         return defaultEncoding;
     }
@@ -1468,6 +1481,7 @@ public DeploymentInfo clone() {
         info.notificationReceivers.addAll(notificationReceivers);
         info.allowNonStandardWrappers = allowNonStandardWrappers;
         info.defaultSessionTimeout = defaultSessionTimeout;
+        info.defaultAsyncContextTimeout = defaultAsyncContextTimeout;
         info.servletContextAttributeBackingMap = servletContextAttributeBackingMap;
         info.servletSessionConfig = servletSessionConfig;
         info.hostName = hostName;

servlet/src/main/java/io/undertow/servlet/core/ManagedServlet.java

@@ -232,12 +232,22 @@ public ServletInfo getServletInfo() {
 
     /**
      * This value determines max multipart message size
+     * @deprecated
+     * @see #getMaxMultipartRequestSize()
      * @return
      */
     public long getMaxRequestSize() {
         return maxMultipartRequestSize;
     }
 
+    /**
+     * This value determines max multipart message size
+     * @return
+     */
+    public long getMaxMultipartRequestSize() {
+        return maxMultipartRequestSize;
+    }
+
     public FormParserFactory getFormParserFactory() {
         return formParserFactory;
     }