Custom SSLSocketFactory (#80)

Author: rugameuser

centrifuge/src/main/java/io/github/centrifugal/centrifuge/Client.java

@@ -16,6 +16,8 @@
 import java.util.concurrent.ScheduledExecutorService;
 import java.util.concurrent.ScheduledFuture;
 import java.util.concurrent.TimeUnit;
+import javax.net.ssl.SSLSocketFactory;
+import javax.net.ssl.X509TrustManager;
 
 import io.github.centrifugal.centrifuge.internal.backoff.Backoff;
 import io.github.centrifugal.centrifuge.internal.protocol.Protocol;
@@ -30,7 +32,6 @@
 import okhttp3.WebSocketListener;
 import okio.ByteString;
 
-
 public class Client {
     private WebSocket ws;
     private final String endpoint;
@@ -265,6 +266,16 @@ private void _connect() {
             okHttpBuilder.dns(dns::resolve);
         }
 
+        SSLSocketFactory socketFactory = opts.getSSLSocketFactory();
+        if (socketFactory != null) {
+            X509TrustManager trustManager = opts.getTrustManager();
+            if (trustManager != null) {
+                okHttpBuilder.sslSocketFactory(socketFactory, trustManager);
+            } else {
+                okHttpBuilder.setSslSocketFactoryOrNull$okhttp(socketFactory);
+            }
+        }
+
         if (opts.getProxy() != null) {
             okHttpBuilder.proxy(opts.getProxy());
             if (this.opts.getProxyLogin() != null && this.opts.getProxyPassword() != null) {

centrifuge/src/main/java/io/github/centrifugal/centrifuge/Options.java

@@ -1,5 +1,7 @@
 package io.github.centrifugal.centrifuge;
 
+import javax.net.ssl.SSLSocketFactory;
+import javax.net.ssl.X509TrustManager;
 import java.net.Proxy;
 import java.util.Map;
 
@@ -190,4 +192,30 @@ public void setDns(Dns dns) {
     }
 
     private Dns dns;
+
+    /**
+     * Set custom SSLSocketFactory
+     */
+    public void setSSLSocketFactory(SSLSocketFactory sslSocketFactory) {
+        this.sslSocketFactory = sslSocketFactory;
+    }
+
+    /**
+     * Set custom SSLSocketFactory & X509TrustManager
+     */
+    public void setSSLSocketFactory(SSLSocketFactory sslSocketFactory, X509TrustManager trustManager) {
+        this.sslSocketFactory = sslSocketFactory;
+        this.trustManager = trustManager;
+    }
+
+    public SSLSocketFactory getSSLSocketFactory() {
+        return this.sslSocketFactory;
+    }
+
+    private SSLSocketFactory sslSocketFactory;
+    private X509TrustManager trustManager;
+
+    public X509TrustManager getTrustManager() {
+        return trustManager;
+    }
 }