AjxpHttpClient.java

/**
 *  Copyright 2012 Charles du Jeu
 *
 *  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.
 *  
 *  This file is part of the AjaXplorer Java Client
 *  More info on http://ajaxplorer.info/
 */
package info.ajaxplorer.client.http;


import java.io.IOException;
import java.net.URI;
import java.util.ArrayList;
import java.util.List;

import org.apache.http.HttpResponse;
import org.apache.http.auth.AuthScope;
import org.apache.http.auth.UsernamePasswordCredentials;
import org.apache.http.client.ClientProtocolException;
import org.apache.http.client.CookieStore;
import org.apache.http.client.methods.HttpRequestBase;
import org.apache.http.client.params.ClientPNames;
import org.apache.http.client.protocol.ClientContext;
import org.apache.http.conn.ClientConnectionManager;
import org.apache.http.conn.scheme.PlainSocketFactory;
import org.apache.http.conn.scheme.Scheme;
import org.apache.http.conn.scheme.SchemeRegistry;
import org.apache.http.conn.ssl.SSLSocketFactory;
import org.apache.http.cookie.Cookie;
import org.apache.http.impl.client.BasicCookieStore;
import org.apache.http.impl.client.DefaultHttpClient;
import org.apache.http.impl.conn.SingleClientConnManager;
import org.apache.http.protocol.BasicHttpContext;
import org.apache.http.protocol.HttpContext;

public class AjxpHttpClient extends DefaultHttpClient {

	boolean trustSelfSignedSSL;
	static HttpContext localContext = new BasicHttpContext();
	public static CookieStore cookieStore = new BasicCookieStore();

	public AjxpHttpClient(boolean trustSSL) {
		super();
		this.trustSelfSignedSSL = trustSSL;
		localContext.setAttribute(ClientContext.COOKIE_STORE, cookieStore);
		this.getParams().setParameter(ClientPNames.ALLOW_CIRCULAR_REDIRECTS,true);        
	}
	
	public void refreshCredentials(String user, String pass){
		this.getCredentialsProvider().setCredentials(
                new AuthScope(AuthScope.ANY_HOST, AuthScope.ANY_PORT, AuthScope.ANY_REALM),
                new UsernamePasswordCredentials(user, pass)
        );		
	}
	
	@Override
	protected ClientConnectionManager createClientConnectionManager() {
		
		SchemeRegistry registry = new SchemeRegistry();
		registry.register(new Scheme("http", PlainSocketFactory.getSocketFactory(), 80));
		
		if(trustSelfSignedSSL){
			//HostnameVerifier hostnameVerifier = org.apache.http.conn.ssl.SSLSocketFactory.ALLOW_ALL_HOSTNAME_VERIFIER;
		    //socketFactory.setHostnameVerifier((X509HostnameVerifier) hostnameVerifier);
			//HttpsURLConnection.setDefaultHostnameVerifier(hostnameVerifier);
			/*
			HttpsURLConnection.setDefaultHostnameVerifier(new HostnameVerifier() {				
				public boolean verify(String hostname, SSLSession session) {
					return true;
				}
			});*/
			registry.register(new Scheme("https", new AjxpSSLSocketFactory(), 443));
			
		}else{
			SSLSocketFactory socketFactory = SSLSocketFactory.getSocketFactory();
			registry.register(new Scheme("https", socketFactory, 443));
		}
		
		return new SingleClientConnManager(getParams(), registry);		
	}

	public void clearCookies() {
		if(cookieStore != null) cookieStore.clear();
	}

	public static void clearCookiesStatic() {
		if(AjxpHttpClient.cookieStore != null) AjxpHttpClient.cookieStore.clear();
	}

	public HttpResponse executeInContext(HttpRequestBase request)
			throws ClientProtocolException, IOException {
		return execute(request, localContext);
	}

	public static List<Cookie> getCookies(URI uri) {
		// Duplicate and prune
		List<Cookie> originalCookies = cookieStore.getCookies();
		List<Cookie> cookies = new ArrayList<Cookie>(originalCookies);
		for (int i = 0; i < originalCookies.size(); i++) {
			if (!originalCookies.get(i).getDomain().equals(uri.getHost())) {
				cookies.remove(originalCookies.get(i));
			}
		}
		return cookies;
	}
	
	public void destroy() {
		// TODO Auto-generated method stub
		
	}
}