com.netflix.http4.ssl.KeyStoreAwareSocketFactory Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ribbon-httpclient Show documentation
ribbon-httpclient
There is a newer version: 2.7.18
/*
*
* Copyright 2013 Netflix, Inc.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
*/
package com.netflix.http4.ssl;


import java.security.KeyStore;
import java.security.KeyStoreException;
import java.security.NoSuchAlgorithmException;

import javax.net.ssl.SSLContext;

import org.apache.http.conn.ssl.SSLSocketFactory;
import org.apache.http.conn.ssl.X509HostnameVerifier;

import com.netflix.client.ssl.AbstractSslContextFactory;
import com.netflix.client.ssl.ClientSslSocketFactoryException;

/**
 *
 * SocketFactory that remembers what keystore and truststore being used,
 * allowing for that information to be queried later.
 *
 * @author jzarfoss
 *
 */
public class KeyStoreAwareSocketFactory extends SSLSocketFactory{


	private final KeyStore keyStore;
	private final KeyStore trustStore;


	public KeyStoreAwareSocketFactory(X509HostnameVerifier hostnameVerifier) throws NoSuchAlgorithmException, KeyStoreException{
		super(SSLContext.getDefault(), hostnameVerifier);

		this.keyStore = null;
		this.trustStore = null;
	}



	public KeyStoreAwareSocketFactory(final AbstractSslContextFactory abstractFactory) throws ClientSslSocketFactoryException, NoSuchAlgorithmException{
		super(abstractFactory == null ? SSLContext.getDefault() : abstractFactory.getSSLContext());

		if(abstractFactory == null){
			this.keyStore = null;
			this.trustStore = null;
		}else{
			this.keyStore = abstractFactory.getKeyStore();
			this.trustStore = abstractFactory.getTrustStore();
		}
	}


	public KeyStoreAwareSocketFactory(final AbstractSslContextFactory abstractFactory, X509HostnameVerifier hostnameVerifier) throws ClientSslSocketFactoryException, NoSuchAlgorithmException{
		super(abstractFactory == null ? SSLContext.getDefault() : abstractFactory.getSSLContext(), hostnameVerifier);

		if(abstractFactory == null){
			this.keyStore = null;
			this.trustStore = null;
		}else{
			this.keyStore = abstractFactory.getKeyStore();
			this.trustStore = abstractFactory.getTrustStore();
		}
	}

	public KeyStore getKeyStore(){
		return this.keyStore;
	}

	public KeyStore getTrustStore(){
		return this.trustStore;
	}



}