libcore/luni/src/main/java/org/apache/harmony/luni/internal/net/www/protocol/http/HttpConnection.java - platform/dalvik - Git at Google

 /*
  *  Licensed to the Apache Software Foundation (ASF) under one or more
  *  contributor license agreements.  See the NOTICE file distributed with
  *  this work for additional information regarding copyright ownership.
  *  The ASF licenses this file to You 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.
  */

 // BEGIN android-note
 // This class was copied from a newer version of harmony
 // to improve reusability of URLConnections
 // END android-note

 package org.apache.harmony.luni.internal.net.www.protocol.http;

 import java.io.IOException;
 import java.io.InputStream;
 import java.io.InterruptedIOException;
 import java.io.OutputStream;
 import java.net.InetSocketAddress;
 import java.net.Proxy;
 import java.net.Socket;
 import java.net.SocketException;

 import javax.net.ssl.HostnameVerifier;
 import javax.net.ssl.SSLSocket;
 import javax.net.ssl.SSLSocketFactory;

 import org.apache.harmony.luni.internal.nls.Messages;

 /**
  * An <code>HttpConnection</code> represents a persistent http or https connection and contains
  * various utility methods to access that connection.
  */
 public class HttpConnection {
     private boolean usingSecureSocket = false;

     private Socket socket;
     private SSLSocket sslSocket;

     private InputStream inputStream;
     private OutputStream outputStream;
     private InputStream sslInputStream;
     private OutputStream sslOutputStream;

     private HttpConfiguration config;

     public HttpConnection(HttpConfiguration config, int connectTimeout) throws IOException {
         this.config = config;
         String hostName = config.getHostName();
         int hostPort = config.getHostPort();
         Proxy proxy = config.getProxy();
         if(proxy == null || proxy.type() == Proxy.Type.HTTP) {
             socket = new Socket();
         } else {
             socket = new Socket(proxy);
         }
         socket.connect(new InetSocketAddress(hostName, hostPort), connectTimeout);
     }

     public void closeSocketAndStreams() {
         if(usingSecureSocket) {
             if (null != sslOutputStream) {
                 OutputStream temp = sslOutputStream;
                 sslOutputStream = null;
                 try {
                     temp.close();
                 } catch (Exception ex) {
                     // ignored
                 }
             }

             if (null != sslInputStream) {
                 InputStream temp = sslInputStream;
                 sslInputStream = null;
                 try {
                     temp.close();
                 } catch (Exception ex) {
                     // ignored
                 }
             }

             if (null != sslSocket) {
                 Socket temp = sslSocket;
                 sslSocket = null;
                 try {
                     temp.close();
                 } catch (Exception ex) {
                     // ignored
                 }
             }
         }
         if (null != outputStream) {
             OutputStream temp = outputStream;
             outputStream = null;
             try {
                 temp.close();
             } catch (Exception ex) {
                 // ignored
             }
         }

         if (null != inputStream) {
             InputStream temp = inputStream;
             inputStream = null;
             try {
                 temp.close();
             } catch (Exception ex) {
                 // ignored
             }
         }

         if (null != socket) {
             Socket temp = socket;
             socket = null;
             try {
                 temp.close();
             } catch (Exception ex) {
                 // ignored
             }
         }
     }

     public void setSoTimeout(int readTimeout) throws SocketException {
         socket.setSoTimeout(readTimeout);
     }

     public OutputStream getOutputStream() throws IOException {
         if(usingSecureSocket) {
             if (sslOutputStream == null) {
                 sslOutputStream = sslSocket.getOutputStream();
             }
             return sslOutputStream;
         } else if(outputStream == null) {
             outputStream = socket.getOutputStream();
         }
         return outputStream;
     }

     public InputStream getInputStream() throws IOException {
         if(usingSecureSocket) {
             if (sslInputStream == null) {
                 sslInputStream = sslSocket.getInputStream();
             }
             return sslInputStream;
         } else if(inputStream == null) {
             inputStream = socket.getInputStream();
         }
         return inputStream;
     }

     public HttpConfiguration getHttpConfiguration() {
         return config;
     }

     public SSLSocket getSecureSocket(SSLSocketFactory sslSocketFactory, HostnameVerifier hostnameVerifier) throws IOException {
         if(!usingSecureSocket) {
             String hostName = config.getHostName();
             int port = config.getHostPort();
             // create the wrapper over connected socket
             sslSocket = (SSLSocket) sslSocketFactory.createSocket(socket,
                     hostName, port, true);
             sslSocket.setUseClientMode(true);
             sslSocket.startHandshake();
             if (!hostnameVerifier.verify(hostName, sslSocket.getSession())) {
                 throw new IOException(Messages.getString("luni.02", hostName)); //$NON-NLS-1$
             }
             usingSecureSocket = true;
         }
         return sslSocket;
     }

     Socket getSocket() {
         return socket;
     }

     /*
      * This method has been copied from the Apache Jakarta Commons HttpClient project
      * http://svn.apache.org/repos/asf/jakarta/commons/proper/httpclient/trunk/HttpClient/src/java/org/apache/commons/httpclient/HttpConnection.java r480424
      */
     protected boolean isStale() throws IOException {
         boolean isStale = true;
         // BEGIN android-note
         // The following line was expanded to check for input/output shutdown.
         // END android-note
         if (! (socket.isClosed() || socket.isInputShutdown()
                         || socket.isOutputShutdown())) {
             // the socket is open, but could still have been closed from the other end
             isStale = false;
             try {
                 if (inputStream.available() <= 0) {
                     int soTimeout = socket.getSoTimeout();
                     try {
                         socket.setSoTimeout(1);
                         inputStream.mark(1);
                         int byteRead = inputStream.read();
                         if (byteRead == -1) {
                             // again - if the socket is reporting all data read,
                             // probably stale
                             isStale = true;
                         } else {
                             inputStream.reset();
                         }
                     } finally {
                         socket.setSoTimeout(soTimeout);
                     }
                 }
             } catch (InterruptedIOException e) {
                 if (!isSocketTimeoutException(e)) {
                     throw e;
                 }
                 // aha - the connection is NOT stale - continue on!
             } catch (IOException e) {
                 // oops - the connection is stale, the read or soTimeout failed.
                 isStale = true;
             }
         }

         return isStale;
     }

     // BEGIN android-added
     /**
      * Returns whether this connection is eligible to be recycled. This
      * is like {@link #isStale} except that it doesn't try to actually
      * perform any I/O.
      *
      * @return <code>true</code> if the connection is eligible to be
      * recycled
      */
     protected boolean isEligibleForRecycling() {
         return ! (socket.isClosed() || socket.isInputShutdown()
                 || socket.isOutputShutdown());
     }
     // END android-added

     /*
      * This field has been copied from the Apache Jakarta Commons HttpClient project
      * http://svn.apache.org/repos/asf/jakarta/commons/proper/httpclient/trunk/HttpClient/src/java/org/apache/commons/httpclient/HttpConnection.java r480424
      */
     static private final Class SOCKET_TIMEOUT_CLASS = SocketTimeoutExceptionClass();

     /*
      * This method has been copied from the Apache Jakarta Commons HttpClient project
      * http://svn.apache.org/repos/asf/jakarta/commons/proper/httpclient/trunk/HttpClient/src/java/org/apache/commons/httpclient/HttpConnection.java r480424
      */
     public static boolean isSocketTimeoutException(final InterruptedIOException e) {
         if (SOCKET_TIMEOUT_CLASS != null) {
             return SOCKET_TIMEOUT_CLASS.isInstance(e);
         } else {
             return true;
         }
     }

     /*
      * This method has been copied from the Apache Jakarta Commons HttpClient project
      * http://svn.apache.org/repos/asf/jakarta/commons/proper/httpclient/trunk/HttpClient/src/java/org/apache/commons/httpclient/HttpConnection.java r480424
      */
     static private Class SocketTimeoutExceptionClass() {
         try {
             return Class.forName("java.net.SocketTimeoutException");
         } catch (ClassNotFoundException e) {
             return null;
         }
     }
 }
	/*
	* Licensed to the Apache Software Foundation (ASF) under one or more
	* contributor license agreements. See the NOTICE file distributed with
	* this work for additional information regarding copyright ownership.
	* The ASF licenses this file to You 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.
	*/

	// BEGIN android-note
	// This class was copied from a newer version of harmony
	// to improve reusability of URLConnections
	// END android-note

	package org.apache.harmony.luni.internal.net.www.protocol.http;

	import java.io.IOException;
	import java.io.InputStream;
	import java.io.InterruptedIOException;
	import java.io.OutputStream;
	import java.net.InetSocketAddress;
	import java.net.Proxy;
	import java.net.Socket;
	import java.net.SocketException;

	import javax.net.ssl.HostnameVerifier;
	import javax.net.ssl.SSLSocket;
	import javax.net.ssl.SSLSocketFactory;

	import org.apache.harmony.luni.internal.nls.Messages;

	/**
	* An <code>HttpConnection</code> represents a persistent http or https connection and contains
	* various utility methods to access that connection.
	*/
	public class HttpConnection {
	private boolean usingSecureSocket = false;

	private Socket socket;
	private SSLSocket sslSocket;

	private InputStream inputStream;
	private OutputStream outputStream;
	private InputStream sslInputStream;
	private OutputStream sslOutputStream;

	private HttpConfiguration config;

	public HttpConnection(HttpConfiguration config, int connectTimeout) throws IOException {
	this.config = config;
	String hostName = config.getHostName();
	int hostPort = config.getHostPort();
	Proxy proxy = config.getProxy();
	if(proxy == null \|\| proxy.type() == Proxy.Type.HTTP) {
	socket = new Socket();
	} else {
	socket = new Socket(proxy);
	}
	socket.connect(new InetSocketAddress(hostName, hostPort), connectTimeout);
	}

	public void closeSocketAndStreams() {
	if(usingSecureSocket) {
	if (null != sslOutputStream) {
	OutputStream temp = sslOutputStream;
	sslOutputStream = null;
	try {
	temp.close();
	} catch (Exception ex) {
	// ignored
	}
	}

	if (null != sslInputStream) {
	InputStream temp = sslInputStream;
	sslInputStream = null;
	try {
	temp.close();
	} catch (Exception ex) {
	// ignored
	}
	}

	if (null != sslSocket) {
	Socket temp = sslSocket;
	sslSocket = null;
	try {
	temp.close();
	} catch (Exception ex) {
	// ignored
	}
	}
	}
	if (null != outputStream) {
	OutputStream temp = outputStream;
	outputStream = null;
	try {
	temp.close();
	} catch (Exception ex) {
	// ignored
	}
	}

	if (null != inputStream) {
	InputStream temp = inputStream;
	inputStream = null;
	try {
	temp.close();
	} catch (Exception ex) {
	// ignored
	}
	}

	if (null != socket) {
	Socket temp = socket;
	socket = null;
	try {
	temp.close();
	} catch (Exception ex) {
	// ignored
	}
	}
	}

	public void setSoTimeout(int readTimeout) throws SocketException {
	socket.setSoTimeout(readTimeout);
	}

	public OutputStream getOutputStream() throws IOException {
	if(usingSecureSocket) {
	if (sslOutputStream == null) {
	sslOutputStream = sslSocket.getOutputStream();
	}
	return sslOutputStream;
	} else if(outputStream == null) {
	outputStream = socket.getOutputStream();
	}
	return outputStream;
	}

	public InputStream getInputStream() throws IOException {
	if(usingSecureSocket) {
	if (sslInputStream == null) {
	sslInputStream = sslSocket.getInputStream();
	}
	return sslInputStream;
	} else if(inputStream == null) {
	inputStream = socket.getInputStream();
	}
	return inputStream;
	}

	public HttpConfiguration getHttpConfiguration() {
	return config;
	}

	public SSLSocket getSecureSocket(SSLSocketFactory sslSocketFactory, HostnameVerifier hostnameVerifier) throws IOException {
	if(!usingSecureSocket) {
	String hostName = config.getHostName();
	int port = config.getHostPort();
	// create the wrapper over connected socket
	sslSocket = (SSLSocket) sslSocketFactory.createSocket(socket,
	hostName, port, true);
	sslSocket.setUseClientMode(true);
	sslSocket.startHandshake();
	if (!hostnameVerifier.verify(hostName, sslSocket.getSession())) {
	throw new IOException(Messages.getString("luni.02", hostName)); //$NON-NLS-1$
	}
	usingSecureSocket = true;
	}
	return sslSocket;
	}

	Socket getSocket() {
	return socket;
	}

	/*
	* This method has been copied from the Apache Jakarta Commons HttpClient project
	* http://svn.apache.org/repos/asf/jakarta/commons/proper/httpclient/trunk/HttpClient/src/java/org/apache/commons/httpclient/HttpConnection.java r480424
	*/
	protected boolean isStale() throws IOException {
	boolean isStale = true;
	// BEGIN android-note
	// The following line was expanded to check for input/output shutdown.
	// END android-note
	if (! (socket.isClosed() \|\| socket.isInputShutdown()
	\|\| socket.isOutputShutdown())) {
	// the socket is open, but could still have been closed from the other end
	isStale = false;
	try {
	if (inputStream.available() <= 0) {
	int soTimeout = socket.getSoTimeout();
	try {
	socket.setSoTimeout(1);
	inputStream.mark(1);
	int byteRead = inputStream.read();
	if (byteRead == -1) {
	// again - if the socket is reporting all data read,
	// probably stale
	isStale = true;
	} else {
	inputStream.reset();
	}
	} finally {
	socket.setSoTimeout(soTimeout);
	}
	}
	} catch (InterruptedIOException e) {
	if (!isSocketTimeoutException(e)) {
	throw e;
	}
	// aha - the connection is NOT stale - continue on!
	} catch (IOException e) {
	// oops - the connection is stale, the read or soTimeout failed.
	isStale = true;
	}
	}

	return isStale;
	}

	// BEGIN android-added
	/**
	* Returns whether this connection is eligible to be recycled. This
	* is like {@link #isStale} except that it doesn't try to actually
	* perform any I/O.
	*
	* @return <code>true</code> if the connection is eligible to be
	* recycled
	*/
	protected boolean isEligibleForRecycling() {
	return ! (socket.isClosed() \|\| socket.isInputShutdown()
	\|\| socket.isOutputShutdown());
	}
	// END android-added

	/*
	* This field has been copied from the Apache Jakarta Commons HttpClient project
	* http://svn.apache.org/repos/asf/jakarta/commons/proper/httpclient/trunk/HttpClient/src/java/org/apache/commons/httpclient/HttpConnection.java r480424
	*/
	static private final Class SOCKET_TIMEOUT_CLASS = SocketTimeoutExceptionClass();

	/*
	* This method has been copied from the Apache Jakarta Commons HttpClient project
	* http://svn.apache.org/repos/asf/jakarta/commons/proper/httpclient/trunk/HttpClient/src/java/org/apache/commons/httpclient/HttpConnection.java r480424
	*/
	public static boolean isSocketTimeoutException(final InterruptedIOException e) {
	if (SOCKET_TIMEOUT_CLASS != null) {
	return SOCKET_TIMEOUT_CLASS.isInstance(e);
	} else {
	return true;
	}
	}

	/*
	* This method has been copied from the Apache Jakarta Commons HttpClient project
	* http://svn.apache.org/repos/asf/jakarta/commons/proper/httpclient/trunk/HttpClient/src/java/org/apache/commons/httpclient/HttpConnection.java r480424
	*/
	static private Class SocketTimeoutExceptionClass() {
	try {
	return Class.forName("java.net.SocketTimeoutException");
	} catch (ClassNotFoundException e) {
	return null;
	}
	}
	}