com.caucho.quercus.lib.curl.CurlHttpConnection Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of quercus Show documentation
Show all versions of quercus Show documentation
A PHP engine implemented in 100% Java
/*
* Copyright (c) 1998-2012 Caucho Technology -- all rights reserved
*
* This file is part of Resin(R) Open Source
*
* Each copy or derived work must preserve the copyright notice and this
* notice unmodified.
*
* Resin Open Source is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* Resin Open Source is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE, or any warranty
* of NON-INFRINGEMENT. See the GNU General Public License for more
* details.
*
* You should have received a copy of the GNU General Public License
* along with Resin Open Source; if not, write to the
*
* Free Software Foundation, Inc.
* 59 Temple Place, Suite 330
* Boston, MA 02111-1307 USA
*
* @author Nam Nguyen
*/
package com.caucho.quercus.lib.curl;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.ConnectException;
import java.net.HttpURLConnection;
import java.net.InetSocketAddress;
import java.net.ProtocolException;
import java.net.Proxy;
import java.net.SocketTimeoutException;
import java.net.URL;
import java.net.URLConnection;
/**
* Represents a HttpURLConnection wrapper.
*/
public class CurlHttpConnection
{
private URLConnection _conn;
private HttpURLConnection _httpConn;
private URL _url;
private String _username;
private String _password;
private URL _proxyURL;
private String _proxyUsername;
private String _proxyPassword;
private String _proxyType;
private int _responseCode;
private boolean _hadSentAuthorization = false;
private boolean _hadSentProxyAuthorization = false;
private String _authorization;
private String _proxyAuthorization;
protected CurlHttpConnection(URL url,
String username,
String password)
throws IOException
{
_url = url;
_username = username;
_password = password;
}
public CurlHttpConnection(URL url,
String username,
String password,
URL proxyURL,
String proxyUsername,
String proxyPassword,
String proxyType)
throws IOException
{
_url = url;
_proxyURL = proxyURL;
_proxyType = proxyType;
_username = username;
_password = password;
_proxyUsername = proxyUsername;
_proxyPassword = proxyPassword;
}
protected void init(CurlResource curl)
throws IOException
{
Proxy proxy = getProxy();
if (proxy != null)
setConnection(_url.openConnection(proxy));
else
setConnection(_url.openConnection());
}
public final static CurlHttpConnection createConnection(URL url,
String username,
String password,
CurlResource curl,
URL proxyURL,
String proxyUsername,
String proxyPassword,
String proxyType)
throws IOException
{
CurlHttpConnection conn;
if (url.getProtocol().equals("https")) {
HttpsConnection secureConn
= new HttpsConnection(url, username, password);
conn = secureConn;
}
else {
conn = new CurlHttpConnection(url, username, password);
}
conn._proxyURL = proxyURL;
conn._proxyUsername = proxyUsername;
conn._proxyPassword = proxyPassword;
conn._proxyType = proxyType;
conn.init(curl);
return conn;
}
public final static CurlHttpConnection createConnection(URL url,
String username,
String password,
CurlResource curl)
throws IOException
{
CurlHttpConnection conn;
if (url.getProtocol().equals("https")) {
HttpsConnection secureConn
= new HttpsConnection(url, username, password);
conn = secureConn;
}
else {
conn = new CurlHttpConnection(url, username, password);
}
conn.init(curl);
return conn;
}
public void setConnectTimeout(int time)
{
_conn.setConnectTimeout(time);
}
public void setDoOutput(boolean doOutput)
{
_conn.setDoOutput(doOutput);
}
public void setInstanceFollowRedirects(boolean isToFollowRedirects)
{
getHttpConnection().setInstanceFollowRedirects(isToFollowRedirects);
}
public void setReadTimeout(int time)
{
getConnection().setReadTimeout(time);
}
public void setRequestMethod(String method)
throws ProtocolException
{
getHttpConnection().setRequestMethod(method);
}
public String getRequestProperty(String key)
{
return _conn.getRequestProperty(key);
}
public void setRequestProperty(String key, String value)
{
_conn.setRequestProperty(key, value);
}
protected final Proxy getProxy()
{
if (_proxyURL == null || _proxyURL.getPort() < 0)
return null;
InetSocketAddress address
= new InetSocketAddress(_proxyURL.getHost(), _proxyURL.getPort());
return new Proxy(Proxy.Type.valueOf(_proxyType), address);
}
protected final URL getURL()
{
return _url;
}
protected final URLConnection getConnection()
{
return _conn;
}
protected final HttpURLConnection getHttpConnection()
{
if (_httpConn == null)
throw new ClassCastException(_conn + " is not a HttpURLConnection");
return _httpConn;
}
protected final void setConnection(URLConnection conn)
{
_conn = conn;
if (conn instanceof HttpURLConnection) {
_httpConn = (HttpURLConnection) conn;
}
}
/**
* Connects to the server.
*/
public void connect(CurlResource curl)
throws ConnectException, ProtocolException, SocketTimeoutException,
IOException
{
authenticate();
_conn.connect();
}
/**
* Handles the authentication for this connection.
*/
public void authenticate()
throws ConnectException, ProtocolException, SocketTimeoutException,
IOException
{
if (_username != null || _proxyUsername != null)
authenticateImpl();
if (_proxyAuthorization != null)
_conn.setRequestProperty("Proxy-Authorization", _proxyAuthorization);
if (_authorization != null)
_conn.setRequestProperty("Authorization", _authorization);
}
/**
* Handles the authentication for this connection.
*/
public void authenticateImpl()
throws ConnectException, ProtocolException, SocketTimeoutException,
IOException
{
Proxy proxy = Proxy.NO_PROXY;
if (_proxyURL != null) {
InetSocketAddress address
= new InetSocketAddress(_proxyURL.getHost(), _proxyURL.getPort());
proxy = new Proxy(Proxy.Type.valueOf(_proxyType), address);
}
HttpURLConnection headConn = (HttpURLConnection)_url.openConnection(proxy);
headConn.setRequestMethod("HEAD");
if (_proxyAuthorization != null)
headConn.setRequestProperty("Proxy-Authorization", _proxyAuthorization);
if (_authorization != null)
headConn.setRequestProperty("Authorization", _authorization);
headConn.connect();
int responseCode = headConn.getResponseCode();
if (responseCode == HttpURLConnection.HTTP_PROXY_AUTH
&& _proxyAuthorization == null)
{
String header = headConn.getHeaderField("Proxy-Authenticate");
_proxyAuthorization = getAuthorization(_url,
getHttpConnection().getRequestMethod(),
header,
"Proxy-Authorization",
_proxyUsername,
_proxyPassword);
authenticateImpl();
}
else if (responseCode == HttpURLConnection.HTTP_UNAUTHORIZED
&& _authorization == null)
{
String header = headConn.getHeaderField("WWW-Authenticate");
_authorization = getAuthorization(_url,
getHttpConnection().getRequestMethod(),
header,
"Authorization",
_username,
_password);
authenticateImpl();
}
headConn.disconnect();
}
/**
* Returns the authorization response.
*/
private final String getAuthorization(URL url,
String requestMethod,
String header,
String clientField,
String username,
String password)
throws ConnectException, SocketTimeoutException, IOException
{
if (username == null || password == null)
return "";
String uri = url.getFile();
if (uri.length() == 0)
uri = "/";
String auth = Authentication.getAuthorization(username,
password,
requestMethod,
uri,
header);
return auth;
}
public int getContentLength()
{
return _conn.getContentLength();
}
public InputStream getErrorStream()
{
return getHttpConnection().getErrorStream();
}
public String getHeaderField(String key)
{
return _conn.getHeaderField(key);
}
public String getHeaderField(int i)
{
return _conn.getHeaderField(i);
}
public String getHeaderFieldKey(int i)
{
return _conn.getHeaderFieldKey(i);
}
public InputStream getInputStream()
throws IOException
{
return _conn.getInputStream();
}
public OutputStream getOutputStream()
throws IOException
{
return _conn.getOutputStream();
}
public int getResponseCode()
throws IOException
{
return getHttpConnection().getResponseCode();
}
public String getResponseMessage()
throws IOException
{
return getHttpConnection().getResponseMessage();
}
public void disconnect()
{
close();
}
public void close()
{
if (_httpConn != null)
_httpConn.disconnect();
}
}