-
Notifications
You must be signed in to change notification settings - Fork 1.9k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merged branch 'jetty-10.0.x' into 'jetty-11.0.x'.
Signed-off-by: Simone Bordet <simone.bordet@gmail.com>
- Loading branch information
Showing
8 changed files
with
225 additions
and
14 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
210 changes: 210 additions & 0 deletions
210
...client-transport/src/test/java/org/eclipse/jetty/http/client/TLSHandshakeFailureTest.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,210 @@ | ||
| // | ||
| // ======================================================================== | ||
| // Copyright (c) 1995 Mort Bay Consulting Pty Ltd and others. | ||
| // | ||
| // This program and the accompanying materials are made available under the | ||
| // terms of the Eclipse Public License v. 2.0 which is available at | ||
| // https://www.eclipse.org/legal/epl-2.0, or the Apache License, Version 2.0 | ||
| // which is available at https://www.apache.org/licenses/LICENSE-2.0. | ||
| // | ||
| // SPDX-License-Identifier: EPL-2.0 OR Apache-2.0 | ||
| // ======================================================================== | ||
| // | ||
|
|
||
| package org.eclipse.jetty.http.client; | ||
|
|
||
| import java.io.IOException; | ||
| import java.nio.ByteBuffer; | ||
| import java.util.List; | ||
| import java.util.concurrent.ExecutionException; | ||
| import java.util.concurrent.Executor; | ||
| import java.util.concurrent.ScheduledThreadPoolExecutor; | ||
| import java.util.concurrent.TimeUnit; | ||
| import java.util.concurrent.atomic.AtomicInteger; | ||
| import javax.net.ssl.SSLEngine; | ||
| import javax.net.ssl.SSLEngineResult; | ||
| import javax.net.ssl.SSLException; | ||
|
|
||
| import org.eclipse.jetty.client.HttpClient; | ||
| import org.eclipse.jetty.io.ByteBufferPool; | ||
| import org.eclipse.jetty.io.ClientConnectionFactory; | ||
| import org.eclipse.jetty.io.EndPoint; | ||
| import org.eclipse.jetty.io.ssl.SslClientConnectionFactory; | ||
| import org.eclipse.jetty.io.ssl.SslConnection; | ||
| import org.eclipse.jetty.util.ssl.SslContextFactory; | ||
| import org.eclipse.jetty.util.thread.ScheduledExecutorScheduler; | ||
| import org.junit.jupiter.params.ParameterizedTest; | ||
| import org.junit.jupiter.params.provider.ArgumentsSource; | ||
|
|
||
| import static org.hamcrest.MatcherAssert.assertThat; | ||
| import static org.hamcrest.Matchers.allOf; | ||
| import static org.hamcrest.Matchers.greaterThan; | ||
| import static org.hamcrest.Matchers.lessThan; | ||
| import static org.junit.jupiter.api.Assertions.assertThrows; | ||
| import static org.junit.jupiter.api.Assumptions.assumeTrue; | ||
|
|
||
| public class TLSHandshakeFailureTest extends AbstractTest<TransportScenario> | ||
| { | ||
| @Override | ||
| public void init(Transport transport) throws IOException | ||
| { | ||
| setScenario(new TransportScenario(transport)); | ||
| assumeTrue(scenario.transport.isTlsBased()); | ||
| } | ||
|
|
||
| @ParameterizedTest | ||
| @ArgumentsSource(TransportProvider.class) | ||
| public void testTLSWrapAbruptSSLEngineClose(Transport transport) throws Exception | ||
| { | ||
| TLSHandshakeAction action = SSLEngine::closeOutbound; | ||
| testTLSWrapFailure(transport, action, 1); | ||
| testTLSWrapFailure(transport, action, 2); | ||
| } | ||
|
|
||
| @ParameterizedTest | ||
| @ArgumentsSource(TransportProvider.class) | ||
| public void testTLSWrapAbruptSSLEngineFailure(Transport transport) throws Exception | ||
| { | ||
| TLSHandshakeAction action = sslEngine -> | ||
| { | ||
| throw new SSLException("test"); | ||
| }; | ||
| testTLSWrapFailure(transport, action, 1); | ||
| testTLSWrapFailure(transport, action, 2); | ||
| } | ||
|
|
||
| private void testTLSWrapFailure(Transport transport, TLSHandshakeAction action, int wrapCount) throws Exception | ||
| { | ||
| init(transport); | ||
| scenario.start(new EmptyServerHandler()); | ||
| scenario.client.stop(); | ||
| scenario.client = new HttpClient(scenario.client.getTransport()) | ||
| { | ||
| @Override | ||
| protected ClientConnectionFactory newSslClientConnectionFactory(SslContextFactory.Client sslContextFactory, ClientConnectionFactory connectionFactory) | ||
| { | ||
| if (sslContextFactory == null) | ||
| sslContextFactory = getSslContextFactory(); | ||
| return new SslClientConnectionFactory(sslContextFactory, getByteBufferPool(), getExecutor(), connectionFactory) | ||
| { | ||
| @Override | ||
| protected SslConnection newSslConnection(ByteBufferPool byteBufferPool, Executor executor, EndPoint endPoint, SSLEngine engine) | ||
| { | ||
| return new SslConnection(byteBufferPool, executor, endPoint, engine, isDirectBuffersForEncryption(), isDirectBuffersForDecryption()) | ||
| { | ||
| private final AtomicInteger wraps = new AtomicInteger(); | ||
|
|
||
| @Override | ||
| protected SSLEngineResult wrap(SSLEngine sslEngine, ByteBuffer[] input, ByteBuffer output) throws SSLException | ||
| { | ||
| if (wraps.incrementAndGet() == wrapCount) | ||
| action.accept(sslEngine); | ||
| return super.wrap(sslEngine, input, output); | ||
| } | ||
| }; | ||
| } | ||
| }; | ||
| } | ||
| }; | ||
| ScheduledThreadPoolExecutor schedulerService = new ScheduledThreadPoolExecutor(1); | ||
| schedulerService.setRemoveOnCancelPolicy(true); | ||
| scenario.client.setScheduler(new ScheduledExecutorScheduler(schedulerService)); | ||
| scenario.client.start(); | ||
|
|
||
| int count = 10; | ||
| for (int i = 0; i < count; ++i) | ||
| { | ||
| assertThrows(ExecutionException.class, () -> scenario.client.newRequest(scenario.newURI()) | ||
| .timeout(5, TimeUnit.SECONDS) | ||
| .send() | ||
| ); | ||
| } | ||
|
|
||
| // There should be a task scheduled by HttpDestination | ||
| // to expire HttpExchanges while they are queued. | ||
| // There may be also a task for the idle timeout of the connection. | ||
| List<Runnable> tasks = schedulerService.shutdownNow(); | ||
| assertThat(tasks.toString(), tasks.size(), allOf(greaterThan(0), lessThan(count + 1))); | ||
| } | ||
|
|
||
| @ParameterizedTest | ||
| @ArgumentsSource(TransportProvider.class) | ||
| public void testTLSUnwrapAbruptSSLEngineClose(Transport transport) throws Exception | ||
| { | ||
| TLSHandshakeAction action = SSLEngine::closeInbound; | ||
| testTLSUnwrapFailure(transport, action, 1); | ||
| testTLSUnwrapFailure(transport, action, 2); | ||
| } | ||
|
|
||
| @ParameterizedTest | ||
| @ArgumentsSource(TransportProvider.class) | ||
| public void testTLSUnwrapAbruptSSLEngineFailure(Transport transport) throws Exception | ||
| { | ||
| TLSHandshakeAction action = sslEngine -> | ||
| { | ||
| throw new SSLException("test"); | ||
| }; | ||
| testTLSUnwrapFailure(transport, action, 1); | ||
| testTLSUnwrapFailure(transport, action, 2); | ||
| } | ||
|
|
||
| private void testTLSUnwrapFailure(Transport transport, TLSHandshakeAction action, int unwrapCount) throws Exception | ||
| { | ||
| init(transport); | ||
| scenario.start(new EmptyServerHandler()); | ||
| scenario.client.stop(); | ||
| scenario.client = new HttpClient(scenario.client.getTransport()) | ||
| { | ||
| @Override | ||
| protected ClientConnectionFactory newSslClientConnectionFactory(SslContextFactory.Client sslContextFactory, ClientConnectionFactory connectionFactory) | ||
| { | ||
| if (sslContextFactory == null) | ||
| sslContextFactory = getSslContextFactory(); | ||
| return new SslClientConnectionFactory(sslContextFactory, getByteBufferPool(), getExecutor(), connectionFactory) | ||
| { | ||
| @Override | ||
| protected SslConnection newSslConnection(ByteBufferPool byteBufferPool, Executor executor, EndPoint endPoint, SSLEngine engine) | ||
| { | ||
| return new SslConnection(byteBufferPool, executor, endPoint, engine, isDirectBuffersForEncryption(), isDirectBuffersForDecryption()) | ||
| { | ||
| private final AtomicInteger unwraps = new AtomicInteger(); | ||
|
|
||
| @Override | ||
| protected SSLEngineResult unwrap(SSLEngine sslEngine, ByteBuffer input, ByteBuffer output) throws SSLException | ||
| { | ||
| if (unwraps.incrementAndGet() == unwrapCount) | ||
| action.accept(sslEngine); | ||
| return super.unwrap(sslEngine, input, output); | ||
| } | ||
| }; | ||
| } | ||
| }; | ||
| } | ||
| }; | ||
| ScheduledThreadPoolExecutor schedulerService = new ScheduledThreadPoolExecutor(1); | ||
| schedulerService.setRemoveOnCancelPolicy(true); | ||
| scenario.client.setScheduler(new ScheduledExecutorScheduler(schedulerService)); | ||
| scenario.client.start(); | ||
|
|
||
| int count = 10; | ||
| for (int i = 0; i < count; ++i) | ||
| { | ||
| assertThrows(ExecutionException.class, () -> scenario.client.newRequest(scenario.newURI()) | ||
| .timeout(5, TimeUnit.SECONDS) | ||
| .send() | ||
| ); | ||
| } | ||
|
|
||
| // There should be a task scheduled by HttpDestination | ||
| // to expire HttpExchanges while they are queued. | ||
| // There may be also a task for the idle timeout of the connection. | ||
| List<Runnable> tasks = schedulerService.shutdownNow(); | ||
| assertThat(tasks.toString(), tasks.size(), allOf(greaterThan(0), lessThan(count + 1))); | ||
| } | ||
|
|
||
| @FunctionalInterface | ||
| private interface TLSHandshakeAction | ||
| { | ||
| void accept(SSLEngine sslEngine) throws SSLException; | ||
| } | ||
| } |