diff options
author | HÃ¥kon Hallingstad <hakon@verizonmedia.com> | 2019-04-08 11:23:51 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2019-04-08 11:23:51 +0200 |
commit | e26fe5c363a5eea163635ff59352ba9cde2a855e (patch) | |
tree | f232c3e29701395c65cf9af0b691b823bd3c0bbf /http-utils/src | |
parent | 759eb69f16a04488d8661b24087d7da8a7cccd91 (diff) |
Revert "Bjorncs/http utils"
Diffstat (limited to 'http-utils/src')
-rw-r--r-- | http-utils/src/main/java/ai/vespa/util/http/VespaHttpClientBuilder.java | 135 | ||||
-rw-r--r-- | http-utils/src/test/java/ai/vespa/util/http/VespaHttpClientBuilderTest.java | 39 |
2 files changed, 0 insertions, 174 deletions
diff --git a/http-utils/src/main/java/ai/vespa/util/http/VespaHttpClientBuilder.java b/http-utils/src/main/java/ai/vespa/util/http/VespaHttpClientBuilder.java deleted file mode 100644 index 5e7a9441fc8..00000000000 --- a/http-utils/src/main/java/ai/vespa/util/http/VespaHttpClientBuilder.java +++ /dev/null @@ -1,135 +0,0 @@ -// Copyright 2019 Oath Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root. -package ai.vespa.util.http; - -import com.yahoo.security.tls.MixedMode; -import com.yahoo.security.tls.TlsContext; -import com.yahoo.security.tls.TransportSecurityUtils; -import org.apache.http.HttpRequest; -import org.apache.http.HttpRequestInterceptor; -import org.apache.http.client.methods.HttpRequestBase; -import org.apache.http.client.utils.URIBuilder; -import org.apache.http.config.Registry; -import org.apache.http.config.RegistryBuilder; -import org.apache.http.conn.HttpClientConnectionManager; -import org.apache.http.conn.socket.ConnectionSocketFactory; -import org.apache.http.conn.socket.PlainConnectionSocketFactory; -import org.apache.http.conn.ssl.NoopHostnameVerifier; -import org.apache.http.conn.ssl.SSLConnectionSocketFactory; -import org.apache.http.impl.client.HttpClientBuilder; -import org.apache.http.impl.conn.BasicHttpClientConnectionManager; -import org.apache.http.protocol.HttpContext; - -import javax.net.ssl.SSLParameters; -import java.net.URI; -import java.net.URISyntaxException; -import java.util.logging.Level; -import java.util.logging.Logger; - -/** - * Http client builder for internal Vespa communications over http/https. - * - * Notes: - * - hostname verification is not enabled - CN/SAN verification is assumed to be handled by the underlying x509 trust manager. - * - custom connection managers must be configured through {@link #createBuilder(ConnectionManagerFactory)}. Do not call {@link HttpClientBuilder#setConnectionManager(HttpClientConnectionManager)}. - * - * @author bjorncs - */ -public class VespaHttpClientBuilder { - - private static final Logger log = Logger.getLogger(VespaHttpClientBuilder.class.getName()); - - public interface ConnectionManagerFactory { - HttpClientConnectionManager create(Registry<ConnectionSocketFactory> socketFactoryRegistry); - } - - private VespaHttpClientBuilder() {} - - /** - * Create a client builder with default connection manager. - */ - public static HttpClientBuilder create() { - return createBuilder(null); - } - - /** - * Create a client builder with a user specified connection manager. - */ - public static HttpClientBuilder create(ConnectionManagerFactory connectionManagerFactory) { - return createBuilder(connectionManagerFactory); - } - - /** - * Creates a client builder with a {@link BasicHttpClientConnectionManager} configured. - * This connection manager uses a single connection for all requests. See Javadoc for details. - */ - public static HttpClientBuilder createWithBasicConnectionManager() { - return createBuilder(BasicHttpClientConnectionManager::new); - } - - private static HttpClientBuilder createBuilder(ConnectionManagerFactory connectionManagerFactory) { - var builder = HttpClientBuilder.create(); - addSslSocketFactory(builder, connectionManagerFactory); - addTlsAwareRequestInterceptor(builder); - return builder; - } - - private static void addSslSocketFactory(HttpClientBuilder builder, ConnectionManagerFactory connectionManagerFactory) { - TransportSecurityUtils.createTlsContext() - .ifPresent(tlsContext -> { - log.log(Level.FINE, "Adding ssl socket factory to client"); - SSLConnectionSocketFactory socketFactory = createSslSocketFactory(tlsContext); - if (connectionManagerFactory != null) { - builder.setConnectionManager(connectionManagerFactory.create(createRegistry(socketFactory))); - } else { - builder.setSSLSocketFactory(socketFactory); - } - }); - } - - private static void addTlsAwareRequestInterceptor(HttpClientBuilder builder) { - if (TransportSecurityUtils.isTransportSecurityEnabled() - && TransportSecurityUtils.getInsecureMixedMode() != MixedMode.PLAINTEXT_CLIENT_MIXED_SERVER) { - log.log(Level.FINE, "Adding request interceptor to client"); - builder.addInterceptorFirst(new HttpToHttpsRewritingRequestInterceptor()); - } - } - - private static SSLConnectionSocketFactory createSslSocketFactory(TlsContext tlsContext) { - SSLParameters parameters = tlsContext.parameters(); - return new SSLConnectionSocketFactory(tlsContext.context(), parameters.getProtocols(), parameters.getCipherSuites(), new NoopHostnameVerifier()); - } - - private static Registry<ConnectionSocketFactory> createRegistry(SSLConnectionSocketFactory sslSocketFactory) { - return RegistryBuilder.<ConnectionSocketFactory>create() - .register("https", sslSocketFactory) - .register("http", PlainConnectionSocketFactory.getSocketFactory()) - .build(); - } - - static class HttpToHttpsRewritingRequestInterceptor implements HttpRequestInterceptor { - @Override - public void process(HttpRequest request, HttpContext context) { - if (request instanceof HttpRequestBase) { - HttpRequestBase httpUriRequest = (HttpRequestBase) request; - httpUriRequest.setURI(rewriteUri(httpUriRequest.getURI())); - } else { - log.log(Level.FINE, () -> "Not a HttpRequestBase - skipping URI rewriting: " + request.getClass().getName()); - } - } - - private static URI rewriteUri(URI originalUri) { - if (!originalUri.getScheme().equals("http")) { - return originalUri; - } - int port = originalUri.getPort(); - int rewrittenPort = port != -1 ? port : 80; - try { - URI rewrittenUri = new URIBuilder(originalUri).setScheme("https").setPort(rewrittenPort).build(); - log.log(Level.FINE, () -> String.format("Uri rewritten from '%s' to '%s'", originalUri, rewrittenUri)); - return rewrittenUri; - } catch (URISyntaxException e) { - throw new RuntimeException(e); - } - } - } -} diff --git a/http-utils/src/test/java/ai/vespa/util/http/VespaHttpClientBuilderTest.java b/http-utils/src/test/java/ai/vespa/util/http/VespaHttpClientBuilderTest.java deleted file mode 100644 index 7ffd0e459b0..00000000000 --- a/http-utils/src/test/java/ai/vespa/util/http/VespaHttpClientBuilderTest.java +++ /dev/null @@ -1,39 +0,0 @@ -// Copyright 2019 Oath Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root. -package ai.vespa.util.http; - -import ai.vespa.util.http.VespaHttpClientBuilder.HttpToHttpsRewritingRequestInterceptor; -import org.apache.http.client.methods.HttpGet; -import org.apache.http.protocol.BasicHttpContext; -import org.junit.Test; - -import java.net.URI; - -import static org.assertj.core.api.AssertionsForClassTypes.assertThat; - -/** - * @author bjorncs - */ -public class VespaHttpClientBuilderTest { - - @Test - public void request_interceptor_modifies_scheme_of_requests() { - verifyProcessedUriMatchesExpectedOutput("http://dummyhostname:8080/a/path/to/resource?query=value", - "https://dummyhostname:8080/a/path/to/resource?query=value"); - } - - @Test - public void request_interceptor_add_handles_implicit_http_port() { - verifyProcessedUriMatchesExpectedOutput("http://dummyhostname/a/path/to/resource?query=value", - "https://dummyhostname:80/a/path/to/resource?query=value"); - } - - private static void verifyProcessedUriMatchesExpectedOutput(String inputUri, String expectedOutputUri) { - var interceptor = new HttpToHttpsRewritingRequestInterceptor(); - HttpGet request = new HttpGet(inputUri); - interceptor.process(request, new BasicHttpContext()); - URI modifiedUri = request.getURI(); - URI expectedUri = URI.create(expectedOutputUri); - assertThat(modifiedUri).isEqualTo(expectedUri); - } - -}
\ No newline at end of file |