allow for sni enabled endpoints re issue Netflix#337

clatko · Nov 14, 2018 · 152b491 · 152b491
1 parent 304e38c
commit 152b491
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/...nsport/src/main/java/com/netflix/ribbon/transport/netty/http/LoadBalancingHttpClient.java b/...nsport/src/main/java/com/netflix/ribbon/transport/netty/http/LoadBalancingHttpClient.java
@@ -38,8 +38,12 @@
 import io.reactivex.netty.protocol.http.client.HttpClientResponse;
 import io.reactivex.netty.servo.http.HttpClientListener;
 
+import java.net.MalformedURLException;
 import java.net.URI;
+import java.net.URL;
 import java.net.URISyntaxException;
+import java.util.ArrayList;
+import java.util.Arrays;
 import java.util.Collections;
 import java.util.List;
 import java.util.Map;
@@ -48,7 +52,10 @@
 import java.util.concurrent.TimeUnit;
 import java.util.concurrent.atomic.AtomicInteger;
 
+import javax.net.ssl.SNIHostName;
+import javax.net.ssl.SNIServerName;
 import javax.net.ssl.SSLEngine;
+import javax.net.ssl.SSLParameters;
 
 import rx.Observable;
 import rx.functions.Func1;
@@ -511,6 +518,23 @@ protected HttpClient<I, O> createRxClient(Server server) {
                     @Override
                     public SSLEngine createSSLEngine(ByteBufAllocator allocator) {
                         SSLEngine myEngine = super.createSSLEngine(allocator);
+
+                        URL url = null;
+                        if(null!=getRxClients()) {
+                          try {
+                            Server server = getRxClients().entrySet().iterator().next().getKey();
+                            url = new URL(server.getScheme() + "://" + server.getHost());
+
+                            SSLParameters sslParameters = new SSLParameters();
+                            List<SNIServerName> sniServerNames = new ArrayList<>();
+                            sniServerNames.add(new SNIHostName(url.getHost()));
+                            sslParameters.setServerNames(sniServerNames);
+                            myEngine.setSSLParameters(sslParameters);
+                          } catch (MalformedURLException e) {
+                            e.printStackTrace();
+                          }
+                        }
+
                         myEngine.setUseClientMode(true);
                         return myEngine;
                     }