com.hubspot.horizon.SSLConfig Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of HorizonCore Show documentation
There is a newer version: 0.3.0
package com.hubspot.horizon;

import com.google.common.base.Throwables;

import javax.annotation.Nullable;
import javax.net.ssl.KeyManager;
import javax.net.ssl.KeyManagerFactory;
import javax.net.ssl.TrustManager;
import javax.net.ssl.TrustManagerFactory;
import java.io.InputStream;
import java.security.KeyStore;

public class SSLConfig {
  private final KeyManager[] keyManagers;
  private final TrustManager[] trustManagers;
  private final boolean acceptAllSSL;

  private SSLConfig(@Nullable KeyManager[] keyManagers, @Nullable TrustManager[] trustManagers, boolean acceptAllSSL) {
    this.keyManagers = keyManagers;
    this.trustManagers = trustManagers;
    this.acceptAllSSL = acceptAllSSL;
  }

  @Nullable
  public KeyManager[] getKeyManagers() {
    return keyManagers == null ? null : keyManagers.clone();
  }

  @Nullable
  public TrustManager[] getTrustManagers() {
    return trustManagers == null ? null : trustManagers.clone();
  }

  public boolean isAcceptAllSSL() {
    return acceptAllSSL;
  }

  public static SSLConfig standard() {
    return new SSLConfig(null, null, false);
  }

  public static SSLConfig acceptAll() {
    return new SSLConfig(null, null, true);
  }

  public static SSLConfig forKeyStore(InputStream keyStoreStream, String password) {
    try {
      KeyStore keyStore = KeyStore.getInstance(KeyStore.getDefaultType());
      keyStore.load(keyStoreStream, password.toCharArray());

      KeyManagerFactory keyManagerFactory = KeyManagerFactory.getInstance(KeyManagerFactory.getDefaultAlgorithm());
      keyManagerFactory.init(keyStore, password.toCharArray());
      KeyManager[] keyManagers = keyManagerFactory.getKeyManagers();

      TrustManagerFactory trustManagerFactory = TrustManagerFactory.getInstance(TrustManagerFactory.getDefaultAlgorithm());
      trustManagerFactory.init(keyStore);
      TrustManager[] trustManagers = trustManagerFactory.getTrustManagers();

      return custom(keyManagers, trustManagers);
    } catch (Exception e) {
      throw Throwables.propagate(e);
    }
  }

  public static SSLConfig custom(@Nullable KeyManager keyManager, @Nullable TrustManager trustManager) {
    return custom(new KeyManager[]{ keyManager }, new TrustManager[]{ trustManager });
  }

  public static SSLConfig custom(@Nullable KeyManager[] keyManagers, @Nullable TrustManager[] trustManagers) {
    return new SSLConfig(keyManagers, trustManagers, false);
  }
}