package test;

import java.io.IOException;
import java.net.InetAddress;
import java.net.Socket;
import java.net.UnknownHostException;

import javax.net.ssl.SSLContext;
import javax.net.ssl.TrustManager;

import org.apache.commons.httpclient.ConnectTimeoutException;
import org.apache.commons.httpclient.HttpClientError;
import org.apache.commons.httpclient.params.HttpConnectionParams;
import org.apache.commons.httpclient.protocol.ControllerThreadSocketFactory;
import org.apache.commons.httpclient.protocol.SecureProtocolSocketFactory;

public class MySecureProtocolSocketFactory implements
??SecureProtocolSocketFactory {

?private SSLContext sslContext = null;

?/**
? * Constructor for MySecureProtocolSocketFactory.
? */
?public MySecureProtocolSocketFactory() {
?}

?/**
? *
? * @return
? */
?private static SSLContext createEasySSLContext() {
??try {
???SSLContext context = SSLContext.getInstance("SSL");
???context.init(null, new TrustManager[] { new MyX509TrustManager() },
?????null);
???return context;
??} catch (Exception e) {
???throw new HttpClientError(e.toString());
??}
?}

?/**
? *
? * @return
? */
?private SSLContext getSSLContext() {
??if (this.sslContext == null) {
???this.sslContext = createEasySSLContext();
??}
??return this.sslContext;
?}

?/*
? * (non-Javadoc)
? *
? * @see org.apache.commons.httpclient.protocol.ProtocolSocketFactory#createSocket(java.lang.String,
? *????? int, java.net.InetAddress, int)
? */
?public Socket createSocket(String host, int port, InetAddress clientHost,
???int clientPort) throws IOException, UnknownHostException {

??return getSSLContext().getSocketFactory().createSocket(host, port,
????clientHost, clientPort);
?}

?/*
? * (non-Javadoc)
? *
? * @see org.apache.commons.httpclient.protocol.ProtocolSocketFactory#createSocket(java.lang.String,
? *????? int, java.net.InetAddress, int,
? *????? org.apache.commons.httpclient.params.HttpConnectionParams)
? */
?public Socket createSocket(final String host, final int port,
???final InetAddress localAddress, final int localPort,
???final HttpConnectionParams params) throws IOException,
???UnknownHostException, ConnectTimeoutException {
??if (params == null) {
???throw new IllegalArgumentException("Parameters may not be null");
??}
??int timeout = params.getConnectionTimeout();
??if (timeout == 0) {
???return createSocket(host, port, localAddress, localPort);
??} else {???
???return ControllerThreadSocketFactory.createSocket(this, host, port,
?????localAddress, localPort, timeout);
??}
?}

?/*
? * (non-Javadoc)
? *
? * @see SecureProtocolSocketFactory#createSocket(java.lang.String,int)
? */
?public Socket createSocket(String host, int port) throws IOException,
???UnknownHostException {
??return getSSLContext().getSocketFactory().createSocket(host, port);
?}

?/*
? * (non-Javadoc)
? *
? * @see SecureProtocolSocketFactory#createSocket(java.net.Socket,java.lang.String,int,boolean)
? */
?public Socket createSocket(Socket socket, String host, int port,
???boolean autoClose) throws IOException, UnknownHostException {
??return getSSLContext().getSocketFactory().createSocket(socket, host,
????port, autoClose);
?}
}