- /*
- * $Header: /home/cvs/jakarta-commons/httpclient/src/java/org/apache/commons/httpclient/HttpMethodBase.java,v 1.215 2004/09/17 08:00:51 oglueck Exp $
- * $Revision: 1.215 $
- * $Date: 2004/09/17 08:00:51 $
- *
- * ====================================================================
- *
- * Copyright 1999-2004 The Apache Software Foundation
- *
- * 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 software consists of voluntary contributions made by many
- * individuals on behalf of the Apache Software Foundation. For more
- * information on the Apache Software Foundation, please see
- * <http://www.apache.org/>.
- *
- */
- package org.apache.commons.httpclient;
- import java.io.ByteArrayInputStream;
- import java.io.ByteArrayOutputStream;
- import java.io.IOException;
- import java.io.InputStream;
- import java.io.InterruptedIOException;
- import java.util.Collection;
- import org.apache.commons.httpclient.auth.AuthState;
- import org.apache.commons.httpclient.cookie.CookiePolicy;
- import org.apache.commons.httpclient.cookie.CookieSpec;
- import org.apache.commons.httpclient.cookie.MalformedCookieException;
- import org.apache.commons.httpclient.params.HttpMethodParams;
- import org.apache.commons.httpclient.protocol.Protocol;
- import org.apache.commons.httpclient.util.EncodingUtil;
- import org.apache.commons.logging.Log;
- import org.apache.commons.logging.LogFactory;
- /**
- * An abstract base implementation of HttpMethod.
- * <p>
- * At minimum, subclasses will need to override:
- * <ul>
- * <li>{@link #getName} to return the approriate name for this method
- * </li>
- * </ul>
- * </p>
- *
- * <p>
- * When a method requires additional request headers, subclasses will typically
- * want to override:
- * <ul>
- * <li>{@link #addRequestHeaders addRequestHeaders(HttpState,HttpConnection)}
- * to write those headers
- * </li>
- * </ul>
- * </p>
- *
- * <p>
- * When a method expects specific response headers, subclasses may want to
- * override:
- * <ul>
- * <li>{@link #processResponseHeaders processResponseHeaders(HttpState,HttpConnection)}
- * to handle those headers
- * </li>
- * </ul>
- * </p>
- *
- *
- * @author <a href="mailto:remm@apache.org">Remy Maucherat</a>
- * @author Rodney Waldhoff
- * @author Sean C. Sullivan
- * @author <a href="mailto:dion@apache.org">dIon Gillard</a>
- * @author <a href="mailto:jsdever@apache.org">Jeff Dever</a>
- * @author <a href="mailto:dims@apache.org">Davanum Srinivas</a>
- * @author Ortwin Glueck
- * @author Eric Johnson
- * @author Michael Becke
- * @author <a href="mailto:oleg@ural.ru">Oleg Kalnichevski</a>
- * @author <a href="mailto:mbowler@GargoyleSoftware.com">Mike Bowler</a>
- * @author <a href="mailto:ggregory@seagullsw.com">Gary Gregory</a>
- * @author Christian Kohlschuetter
- *
- * @version $Revision: 1.215 $ $Date: 2004/09/17 08:00:51 $
- */
- public abstract class HttpMethodBase implements HttpMethod {
- // -------------------------------------------------------------- Constants
- /** Log object for this class. */
- private static final Log LOG = LogFactory.getLog(HttpMethodBase.class);
- // ----------------------------------------------------- Instance variables
- /** Request headers, if any. */
- private HeaderGroup requestHeaders = new HeaderGroup();
- /** The Status-Line from the response. */
- private StatusLine statusLine = null;
- /** Response headers, if any. */
- private HeaderGroup responseHeaders = new HeaderGroup();
- /** Response trailer headers, if any. */
- private HeaderGroup responseTrailerHeaders = new HeaderGroup();
- /** Path of the HTTP method. */
- private String path = null;
- /** Query string of the HTTP method, if any. */
- private String queryString = null;
- /** The response body of the HTTP method, assuming it has not be
- * intercepted by a sub-class. */
- private InputStream responseStream = null;
- /** The connection that the response stream was read from. */
- private HttpConnection responseConnection = null;
- /** Buffer for the response */
- private byte[] responseBody = null;
- /** True if the HTTP method should automatically follow HTTP redirects.*/
- private boolean followRedirects = false;
- /** True if the HTTP method should automatically handle
- * HTTP authentication challenges. */
- private boolean doAuthentication = true;
- /** HTTP protocol parameters. */
- private HttpMethodParams params = new HttpMethodParams();
- /** Host authentication state */
- private AuthState hostAuthState = new AuthState();
- /** Proxy authentication state */
- private AuthState proxyAuthState = new AuthState();
- /** True if this method has already been executed. */
- private boolean used = false;
- /** Count of how many times did this HTTP method transparently handle
- * a recoverable exception. */
- private int recoverableExceptionCount = 0;
- /** the host configuration for this HTTP method, can be null */
- private HostConfiguration hostConfiguration;
- /**
- * Handles method retries
- *
- * @deprecated no loner used
- */
- private MethodRetryHandler methodRetryHandler;
- /** True if the connection must be closed when no longer needed */
- private boolean connectionCloseForced = false;
- /** Number of milliseconds to wait for 100-contunue response. */
- private static final int RESPONSE_WAIT_TIME_MS = 3000;
- /** HTTP protocol version used for execution of this method. */
- private HttpVersion effectiveVersion = null;
- /** Whether the execution of this method has been aborted */
- private transient boolean aborted = false;
- /** Whether the HTTP request has been transmitted to the target
- * server it its entirety */
- private boolean requestSent = false;
- /** Actual cookie policy */
- private CookieSpec cookiespec = null;
- /** Default initial size of the response buffer if content length is unknown. */
- private static final int DEFAULT_INITIAL_BUFFER_SIZE = 4*1024; // 4 kB
- // ----------------------------------------------------------- Constructors
- /**
- * No-arg constructor.
- */
- public HttpMethodBase() {
- }
- /**
- * Constructor specifying a URI.
- * It is responsibility of the caller to ensure that URI elements
- * (path & query parameters) are properly encoded (URL safe).
- *
- * @param uri either an absolute or relative URI. The URI is expected
- * to be URL-encoded
- *
- * @throws IllegalArgumentException when URI is invalid
- * @throws IllegalStateException when protocol of the absolute URI is not recognised
- */
- public HttpMethodBase(String uri)
- throws IllegalArgumentException, IllegalStateException {
- try {
- // create a URI and allow for null/empty uri values
- if (uri == null || uri.equals("")) {
- uri = "/";
- }
- setURI(new URI(uri, true));
- } catch (URIException e) {
- throw new IllegalArgumentException("Invalid uri '"
- + uri + "': " + e.getMessage()
- );
- }
- }
- // ------------------------------------------- Property Setters and Getters
- /**
- * Obtains the name of the HTTP method as used in the HTTP request line,
- * for example <tt>"GET"</tt> or <tt>"POST"</tt>.
- *
- * @return the name of this method
- */
- public abstract String getName();
- /**
- * Returns the URI of the HTTP method
- *
- * @return The URI
- *
- * @throws URIException If the URI cannot be created.
- *
- * @see org.apache.commons.httpclient.HttpMethod#getURI()
- */
- public URI getURI() throws URIException {
- if (hostConfiguration == null) {
- // just use a relative URI, the host hasn't been set
- URI tmpUri = new URI(null, null, path, null, null);
- tmpUri.setEscapedQuery(queryString);
- return tmpUri;
- } else {
- // we only want to include the port if it's not the default
- int port = hostConfiguration.getPort();
- if (port == hostConfiguration.getProtocol().getDefaultPort()) {
- port = -1;
- }
- URI tmpUri = new URI(
- hostConfiguration.getProtocol().getScheme(),
- null,
- hostConfiguration.getHost(),
- port,
- path,
- null // to set an escaped form
- );
- tmpUri.setEscapedQuery(queryString);
- return tmpUri;
- }
- }
- /**
- * Sets the URI for this method.
- *
- * @param uri URI to be set
- *
- * @throws URIException if a URI cannot be set
- *
- * @since 3.0
- */
- public void setURI(URI uri) throws URIException {
- // only set the host if specified by the URI
- if (uri.isAbsoluteURI()) {
- if (this.hostConfiguration == null) {
- this.hostConfiguration = new HostConfiguration();
- }
- this.hostConfiguration.setHost(
- uri.getHost(),
- uri.getPort(),
- uri.getScheme()
- );
- }
- // set the path, defaulting to root
- setPath(
- uri.getPath() == null
- ? "/"
- : uri.getEscapedPath()
- );
- setQueryString(uri.getEscapedQuery());
- }
- /**
- * Sets whether or not the HTTP method should automatically follow HTTP redirects
- * (status code 302, etc.)
- *
- * @param followRedirects <tt>true</tt> if the method will automatically follow redirects,
- * <tt>false</tt> otherwise.
- */
- public void setFollowRedirects(boolean followRedirects) {
- this.followRedirects = followRedirects;
- }
- /**
- * Returns <tt>true</tt> if the HTTP method should automatically follow HTTP redirects
- * (status code 302, etc.), <tt>false</tt> otherwise.
- *
- * @return <tt>true</tt> if the method will automatically follow HTTP redirects,
- * <tt>false</tt> otherwise.
- */
- public boolean getFollowRedirects() {
- return this.followRedirects;
- }
- /** Sets whether version 1.1 of the HTTP protocol should be used per default.
- *
- * @param http11 <tt>true</tt> to use HTTP/1.1, <tt>false</tt> to use 1.0
- *
- * @deprecated Use {@link HttpMethodParams#setVersion(HttpVersion)}
- */
- public void setHttp11(boolean http11) {
- if (http11) {
- this.params.setVersion(HttpVersion.HTTP_1_1);
- } else {
- this.params.setVersion(HttpVersion.HTTP_1_0);
- }
- }
- /**
- * Returns <tt>true</tt> if the HTTP method should automatically handle HTTP
- * authentication challenges (status code 401, etc.), <tt>false</tt> otherwise
- *
- * @return <tt>true</tt> if authentication challenges will be processed
- * automatically, <tt>false</tt> otherwise.
- *
- * @since 2.0
- */
- public boolean getDoAuthentication() {
- return doAuthentication;
- }
- /**
- * Sets whether or not the HTTP method should automatically handle HTTP
- * authentication challenges (status code 401, etc.)
- *
- * @param doAuthentication <tt>true</tt> to process authentication challenges
- * authomatically, <tt>false</tt> otherwise.
- *
- * @since 2.0
- */
- public void setDoAuthentication(boolean doAuthentication) {
- this.doAuthentication = doAuthentication;
- }
- // ---------------------------------------------- Protected Utility Methods
- /**
- * Returns <tt>true</tt> if version 1.1 of the HTTP protocol should be
- * used per default, <tt>false</tt> if version 1.0 should be used.
- *
- * @return <tt>true</tt> to use HTTP/1.1, <tt>false</tt> to use 1.0
- *
- * @deprecated Use {@link HttpMethodParams#getVersion()}
- */
- public boolean isHttp11() {
- return this.params.getVersion().equals(HttpVersion.HTTP_1_1);
- }
- /**
- * Sets the path of the HTTP method.
- * It is responsibility of the caller to ensure that the path is
- * properly encoded (URL safe).
- *
- * @param path the path of the HTTP method. The path is expected
- * to be URL-encoded
- */
- public void setPath(String path) {
- this.path = path;
- }
- /**
- * Adds the specified request header, NOT overwriting any previous value.
- * Note that header-name matching is case insensitive.
- *
- * @param header the header to add to the request
- */
- public void addRequestHeader(Header header) {
- LOG.trace("HttpMethodBase.addRequestHeader(Header)");
- if (header == null) {
- LOG.debug("null header value ignored");
- } else {
- getRequestHeaderGroup().addHeader(header);
- }
- }
- /**
- * Use this method internally to add footers.
- *
- * @param footer The footer to add.
- */
- public void addResponseFooter(Header footer) {
- getResponseTrailerHeaderGroup().addHeader(footer);
- }
- /**
- * Gets the path of this HTTP method.
- * Calling this method <em>after</em> the request has been executed will
- * return the <em>actual</em> path, following any redirects automatically
- * handled by this HTTP method.
- *
- * @return the path to request or "/" if the path is blank.
- */
- public String getPath() {
- return (path == null || path.equals("")) ? "/" : path;
- }
- /**
- * Sets the query string of this HTTP method. The caller must ensure that the string
- * is properly URL encoded. The query string should not start with the question
- * mark character.
- *
- * @param queryString the query string
- *
- * @see EncodingUtil#formUrlEncode(NameValuePair[], String)
- */
- public void setQueryString(String queryString) {
- this.queryString = queryString;
- }
- /**
- * Sets the query string of this HTTP method. The pairs are encoded as UTF-8 characters.
- * To use a different charset the parameters can be encoded manually using EncodingUtil
- * and set as a single String.
- *
- * @param params an array of {@link NameValuePair}s to add as query string
- * parameters. The name/value pairs will be automcatically
- * URL encoded
- *
- * @see EncodingUtil#formUrlEncode(NameValuePair[], String)
- * @see #setQueryString(String)
- */
- public void setQueryString(NameValuePair[] params) {
- LOG.trace("enter HttpMethodBase.setQueryString(NameValuePair[])");
- queryString = EncodingUtil.formUrlEncode(params, "UTF-8");
- }
- /**
- * Gets the query string of this HTTP method.
- *
- * @return The query string
- */
- public String getQueryString() {
- return queryString;
- }
- /**
- * Set the specified request header, overwriting any previous value. Note
- * that header-name matching is case-insensitive.
- *
- * @param headerName the header's name
- * @param headerValue the header's value
- */
- public void setRequestHeader(String headerName, String headerValue) {
- Header header = new Header(headerName, headerValue);
- setRequestHeader(header);
- }
- /**
- * Sets the specified request header, overwriting any previous value.
- * Note that header-name matching is case insensitive.
- *
- * @param header the header
- */
- public void setRequestHeader(Header header) {
- Header[] headers = getRequestHeaderGroup().getHeaders(header.getName());
- for (int i = 0; i < headers.length; i++) {
- getRequestHeaderGroup().removeHeader(headers[i]);
- }
- getRequestHeaderGroup().addHeader(header);
- }
- /**
- * Returns the specified request header. Note that header-name matching is
- * case insensitive. <tt>null</tt> will be returned if either
- * <i>headerName</i> is <tt>null</tt> or there is no matching header for
- * <i>headerName</i>.
- *
- * @param headerName The name of the header to be returned.
- *
- * @return The specified request header.
- *
- * @since 3.0
- */
- public Header getRequestHeader(String headerName) {
- if (headerName == null) {
- return null;
- } else {
- return getRequestHeaderGroup().getCondensedHeader(headerName);
- }
- }
- /**
- * Returns an array of the requests headers that the HTTP method currently has
- *
- * @return an array of my request headers.
- */
- public Header[] getRequestHeaders() {
- return getRequestHeaderGroup().getAllHeaders();
- }
- /**
- * @see org.apache.commons.httpclient.HttpMethod#getRequestHeaders(java.lang.String)
- */
- public Header[] getRequestHeaders(String headerName) {
- return getRequestHeaderGroup().getHeaders(headerName);
- }
- /**
- * Gets the {@link HeaderGroup header group} storing the request headers.
- *
- * @return a HeaderGroup
- *
- * @since 2.0beta1
- */
- protected HeaderGroup getRequestHeaderGroup() {
- return requestHeaders;
- }
- /**
- * Gets the {@link HeaderGroup header group} storing the response trailer headers
- * as per RFC 2616 section 3.6.1.
- *
- * @return a HeaderGroup
- *
- * @since 2.0beta1
- */
- protected HeaderGroup getResponseTrailerHeaderGroup() {
- return responseTrailerHeaders;
- }
- /**
- * Gets the {@link HeaderGroup header group} storing the response headers.
- *
- * @return a HeaderGroup
- *
- * @since 2.0beta1
- */
- protected HeaderGroup getResponseHeaderGroup() {
- return responseHeaders;
- }
- /**
- * @see org.apache.commons.httpclient.HttpMethod#getResponseHeaders(java.lang.String)
- *
- * @since 3.0
- */
- public Header[] getResponseHeaders(String headerName) {
- return getResponseHeaderGroup().getHeaders(headerName);
- }
- /**
- * Returns the response status code.
- *
- * @return the status code associated with the latest response.
- */
- public int getStatusCode() {
- return statusLine.getStatusCode();
- }
- /**
- * Provides access to the response status line.
- *
- * @return the status line object from the latest response.
- * @since 2.0
- */
- public StatusLine getStatusLine() {
- return statusLine;
- }
- /**
- * Checks if response data is available.
- * @return <tt>true</tt> if response data is available, <tt>false</tt> otherwise.
- */
- private boolean responseAvailable() {
- return (responseBody != null) || (responseStream != null);
- }
- /**
- * Returns an array of the response headers that the HTTP method currently has
- * in the order in which they were read.
- *
- * @return an array of response headers.
- */
- public Header[] getResponseHeaders() {
- return getResponseHeaderGroup().getAllHeaders();
- }
- /**
- * Gets the response header associated with the given name. Header name
- * matching is case insensitive. <tt>null</tt> will be returned if either
- * <i>headerName</i> is <tt>null</tt> or there is no matching header for
- * <i>headerName</i>.
- *
- * @param headerName the header name to match
- *
- * @return the matching header
- */
- public Header getResponseHeader(String headerName) {
- if (headerName == null) {
- return null;
- } else {
- return getResponseHeaderGroup().getCondensedHeader(headerName);
- }
- }
- /**
- * Return the length (in bytes) of the response body, as specified in a
- * <tt>Content-Length</tt> header.
- *
- * <p>
- * Return <tt>-1</tt> when the content-length is unknown.
- * </p>
- *
- * @return content length, if <tt>Content-Length</tt> header is available.
- * <tt>0</tt> indicates that the request has no body.
- * If <tt>Content-Length</tt> header is not present, the method
- * returns <tt>-1</tt>.
- */
- public long getResponseContentLength() {
- Header[] headers = getResponseHeaderGroup().getHeaders("Content-Length");
- if (headers.length == 0) {
- return -1;
- }
- if (headers.length > 1) {
- LOG.warn("Multiple content-length headers detected");
- }
- for (int i = headers.length - 1; i >= 0; i--) {
- Header header = headers[i];
- try {
- return Long.parseLong(header.getValue());
- } catch (NumberFormatException e) {
- if (LOG.isWarnEnabled()) {
- LOG.warn("Invalid content-length value: " + e.getMessage());
- }
- }
- // See if we can have better luck with another header, if present
- }
- return -1;
- }
- /**
- * Returns the response body of the HTTP method, if any, as an array of bytes.
- * If response body is not available or cannot be read, returns <tt>null</tt>
- *
- * Note: This will cause the entire response body to be buffered in memory. A
- * malicious server may easily exhaust all the VM memory. It is strongly
- * recommended, to use getResponseAsStream if the content length of the response
- * is unknown or resonably large.
- *
- * @return The response body.
- *
- * @throws IOException If an I/O (transport) problem occurs while obtaining the
- * response body.
- */
- public byte[] getResponseBody() throws IOException {
- if (this.responseBody == null) {
- InputStream instream = getResponseBodyAsStream();
- if (instream != null) {
- long contentLength = getResponseContentLength();
- if (contentLength > Integer.MAX_VALUE) { //guard below cast from overflow
- throw new IOException("Content too large to be buffered: "+ contentLength +" bytes");
- }
- int limit = getParams().getIntParameter(HttpMethodParams.BUFFER_WARN_TRIGGER_LIMIT, 1024*1024);
- if ((contentLength == -1) || (contentLength > limit)) {
- LOG.warn("Going to buffer response body of large or unknown size. "
- +"Using getResponseAsStream instead is recommended.");
- }
- LOG.debug("Buffering response body");
- ByteArrayOutputStream outstream = new ByteArrayOutputStream(
- contentLength > 0 ? (int) contentLength : DEFAULT_INITIAL_BUFFER_SIZE);
- byte[] buffer = new byte[4096];
- int len;
- while ((len = instream.read(buffer)) > 0) {
- outstream.write(buffer, 0, len);
- }
- outstream.close();
- setResponseStream(null);
- this.responseBody = outstream.toByteArray();
- }
- }
- return this.responseBody;
- }
- /**
- * Returns the response body of the HTTP method, if any, as an {@link InputStream}.
- * If response body is not available, returns <tt>null</tt>
- *
- * @return The response body
- *
- * @throws IOException If an I/O (transport) problem occurs while obtaining the
- * response body.
- */
- public InputStream getResponseBodyAsStream() throws IOException {
- if (responseStream != null) {
- return responseStream;
- }
- if (responseBody != null) {
- InputStream byteResponseStream = new ByteArrayInputStream(responseBody);
- LOG.debug("re-creating response stream from byte array");
- return byteResponseStream;
- }
- return null;
- }
- /**
- * Returns the response body of the HTTP method, if any, as a {@link String}.
- * If response body is not available or cannot be read, returns <tt>null</tt>
- * The string conversion on the data is done using the character encoding specified
- * in <tt>Content-Type</tt> header.
- *
- * Note: This will cause the entire response body to be buffered in memory. A
- * malicious server may easily exhaust all the VM memory. It is strongly
- * recommended, to use getResponseAsStream if the content length of the response
- * is unknown or resonably large.
- *
- * @return The response body.
- *
- * @throws IOException If an I/O (transport) problem occurs while obtaining the
- * response body.
- */
- public String getResponseBodyAsString() throws IOException {
- byte[] rawdata = null;
- if (responseAvailable()) {
- rawdata = getResponseBody();
- }
- if (rawdata != null) {
- return EncodingUtil.getString(rawdata, getResponseCharSet());
- } else {
- return null;
- }
- }
- /**
- * Returns an array of the response footers that the HTTP method currently has
- * in the order in which they were read.
- *
- * @return an array of footers
- */
- public Header[] getResponseFooters() {
- return getResponseTrailerHeaderGroup().getAllHeaders();
- }
- /**
- * Gets the response footer associated with the given name.
- * Footer name matching is case insensitive.
- * <tt>null</tt> will be returned if either <i>footerName</i> is
- * <tt>null</tt> or there is no matching footer for <i>footerName</i>
- * or there are no footers available. If there are multiple footers
- * with the same name, there values will be combined with the ',' separator
- * as specified by RFC2616.
- *
- * @param footerName the footer name to match
- * @return the matching footer
- */
- public Header getResponseFooter(String footerName) {
- if (footerName == null) {
- return null;
- } else {
- return getResponseTrailerHeaderGroup().getCondensedHeader(footerName);
- }
- }
- /**
- * Sets the response stream.
- * @param responseStream The new response stream.
- */
- protected void setResponseStream(InputStream responseStream) {
- this.responseStream = responseStream;
- }
- /**
- * Returns a stream from which the body of the current response may be read.
- * If the method has not yet been executed, if <code>responseBodyConsumed</code>
- * has been called, or if the stream returned by a previous call has been closed,
- * <code>null</code> will be returned.
- *
- * @return the current response stream
- */
- protected InputStream getResponseStream() {
- return responseStream;
- }
- /**
- * Returns the status text (or "reason phrase") associated with the latest
- * response.
- *
- * @return The status text.
- */
- public String getStatusText() {
- return statusLine.getReasonPhrase();
- }
- /**
- * Defines how strictly HttpClient follows the HTTP protocol specification
- * (RFC 2616 and other relevant RFCs). In the strict mode HttpClient precisely
- * implements the requirements of the specification, whereas in non-strict mode
- * it attempts to mimic the exact behaviour of commonly used HTTP agents,
- * which many HTTP servers expect.
- *
- * @param strictMode <tt>true</tt> for strict mode, <tt>false</tt> otherwise
- *
- * @deprecated Use {@link org.apache.commons.httpclient.params.HttpParams#setParameter(String, Object)}
- * to exercise a more granular control over HTTP protocol strictness.
- */
- public void setStrictMode(boolean strictMode) {
- if (strictMode) {
- this.params.makeStrict();
- } else {
- this.params.makeLenient();
- }
- }
- /**
- * @deprecated Use {@link org.apache.commons.httpclient.params.HttpParams#setParameter(String, Object)}
- * to exercise a more granular control over HTTP protocol strictness.
- *
- * @return <tt>false</tt>
- */
- public boolean isStrictMode() {
- return false;
- }
- /**
- * Adds the specified request header, NOT overwriting any previous value.
- * Note that header-name matching is case insensitive.
- *
- * @param headerName the header's name
- * @param headerValue the header's value
- */
- public void addRequestHeader(String headerName, String headerValue) {
- addRequestHeader(new Header(headerName, headerValue));
- }
- /**
- * Tests if the connection should be force-closed when no longer needed.
- *
- * @return <code>true</code> if the connection must be closed
- */
- protected boolean isConnectionCloseForced() {
- return this.connectionCloseForced;
- }
- /**
- * Sets whether or not the connection should be force-closed when no longer
- * needed. This value should only be set to <code>true</code> in abnormal
- * circumstances, such as HTTP protocol violations.
- *
- * @param b <code>true</code> if the connection must be closed, <code>false</code>
- * otherwise.
- */
- protected void setConnectionCloseForced(boolean b) {
- if (LOG.isDebugEnabled()) {
- LOG.debug("Force-close connection: " + b);
- }
- this.connectionCloseForced = b;
- }
- /**
- * Tests if the connection should be closed after the method has been executed.
- * The connection will be left open when using HTTP/1.1 or if <tt>Connection:
- * keep-alive</tt> header was sent.
- *
- * @param conn the connection in question
- *
- * @return boolean true if we should close the connection.
- */
- protected boolean shouldCloseConnection(HttpConnection conn) {
- // Connection must be closed due to an abnormal circumstance
- if (isConnectionCloseForced()) {
- LOG.debug("Should force-close connection.");
- return true;
- }
- Header connectionHeader = null;
- // In case being connected via a proxy server
- if (!conn.isTransparent()) {
- // Check for 'proxy-connection' directive
- connectionHeader = responseHeaders.getFirstHeader("proxy-connection");
- }
- // In all cases Check for 'connection' directive
- // some non-complaint proxy servers send it instread of
- // expected 'proxy-connection' directive
- if (connectionHeader == null) {
- connectionHeader = responseHeaders.getFirstHeader("connection");
- }
- if (connectionHeader != null) {
- if (connectionHeader.getValue().equalsIgnoreCase("close")) {
- if (LOG.isDebugEnabled()) {
- LOG.debug("Should close connection in response to "
- + connectionHeader.toExternalForm());
- }
- return true;
- } else if (connectionHeader.getValue().equalsIgnoreCase("keep-alive")) {
- if (LOG.isDebugEnabled()) {
- LOG.debug("Should NOT close connection in response to "
- + connectionHeader.toExternalForm());
- }
- return false;
- } else {
- if (LOG.isDebugEnabled()) {
- LOG.debug("Unknown directive: " + connectionHeader.toExternalForm());
- }
- }
- }
- LOG.debug("Resorting to protocol version default close connection policy");
- // missing or invalid connection header, do the default
- if (this.effectiveVersion.greaterEquals(HttpVersion.HTTP_1_1)) {
- if (LOG.isDebugEnabled()) {
- LOG.debug("Should NOT close connection, using " + this.effectiveVersion.toString());
- }
- } else {
- if (LOG.isDebugEnabled()) {
- LOG.debug("Should close connection, using " + this.effectiveVersion.toString());
- }
- }
- return this.effectiveVersion.lessEquals(HttpVersion.HTTP_1_0);
- }
- /**
- * Tests if the this method is ready to be executed.
- *
- * @param state the {@link HttpState state} information associated with this method
- * @param conn the {@link HttpConnection connection} to be used
- * @throws HttpException If the method is in invalid state.
- */
- private void checkExecuteConditions(HttpState state, HttpConnection conn)
- throws HttpException {
- if (state == null) {
- throw new IllegalArgumentException("HttpState parameter may not be null");
- }
- if (conn == null) {
- throw new IllegalArgumentException("HttpConnection parameter may not be null");
- }
- if (this.aborted) {
- throw new IllegalStateException("Method has been aborted");
- }
- if (!validate()) {
- throw new ProtocolException("HttpMethodBase object not valid");
- }
- }
- /**
- * Executes this method using the specified <code>HttpConnection</code> and
- * <code>HttpState</code>.
- *
- * @param state {@link HttpState state} information to associate with this
- * request. Must be non-null.
- * @param conn the {@link HttpConnection connection} to used to execute
- * this HTTP method. Must be non-null.
- *
- * @return the integer status code if one was obtained, or <tt>-1</tt>
- *
- * @throws IOException if an I/O (transport) error occurs
- * @throws HttpException if a protocol exception occurs.
- * @throws HttpRecoverableException if a recoverable transport error occurs.
- * Usually this kind of exceptions can be recovered from by
- * retrying the HTTP method
- */
- public int execute(HttpState state, HttpConnection conn)
- throws HttpException, HttpRecoverableException, IOException {
- LOG.trace("enter HttpMethodBase.execute(HttpState, HttpConnection)");
- // this is our connection now, assign it to a local variable so
- // that it can be released later
- this.responseConnection = conn;
- checkExecuteConditions(state, conn);
- this.statusLine = null;
- this.connectionCloseForced = false;
- conn.setLastResponseInputStream(null);
- // determine the effective protocol version
- if (this.effectiveVersion == null) {
- this.effectiveVersion = this.params.getVersion();
- }
- writeRequest(state, conn);
- this.requestSent = true;
- readResponse(state, conn);
- // the method has successfully executed
- used = true;
- return statusLine.getStatusCode();
- }
- /**
- * Aborts the execution of this method.
- *
- * @since 3.0
- */
- public void abort() {
- if (this.aborted) {
- return;
- }
- this.aborted = true;
- HttpConnection conn = this.responseConnection;
- if (conn != null) {
- conn.close();
- }
- }
- /**
- * Returns <tt>true</tt> if the HTTP method has been already {@link #execute executed},
- * but not {@link #recycle recycled}.
- *
- * @return <tt>true</tt> if the method has been executed, <tt>false</tt> otherwise
- */
- public boolean hasBeenUsed() {
- return used;
- }
- /**
- * Recycles the HTTP method so that it can be used again.
- * Note that all of the instance variables will be reset
- * once this method has been called. This method will also
- * release the connection being used by this HTTP method.
- *
- * @see #releaseConnection()
- *
- * @deprecated no longer supported and will be removed in the future
- * version of HttpClient
- */
- public void recycle() {
- LOG.trace("enter HttpMethodBase.recycle()");
- releaseConnection();
- path = null;
- followRedirects = false;
- doAuthentication = true;
- queryString = null;
- getRequestHeaderGroup().clear();
- getResponseHeaderGroup().clear();
- getResponseTrailerHeaderGroup().clear();
- statusLine = null;
- effectiveVersion = null;
- aborted = false;
- used = false;
- params = new HttpMethodParams();
- responseBody = null;
- recoverableExceptionCount = 0;
- connectionCloseForced = false;
- hostAuthState.invalidate();
- proxyAuthState.invalidate();
- cookiespec = null;
- requestSent = false;
- }
- /**
- * Releases the connection being used by this HTTP method. In particular the
- * connection is used to read the response(if there is one) and will be held
- * until the response has been read. If the connection can be reused by other
- * HTTP methods it is NOT closed at this point.
- *
- * @since 2.0
- */
- public void releaseConnection() {
- if (responseStream != null) {
- try {
- // FYI - this may indirectly invoke responseBodyConsumed.
- responseStream.close();
- } catch (IOException e) {
- // the connection may not have been released, let's make sure
- ensureConnectionRelease();
- }
- } else {
- // Make sure the connection has been released. If the response
- // stream has not been set, this is the only way to release the
- // connection.
- ensureConnectionRelease();
- }
- }
- /**
- * Remove the request header associated with the given name. Note that
- * header-name matching is case insensitive.
- *
- * @param headerName the header name
- */
- public void removeRequestHeader(String headerName) {
- Header[] headers = getRequestHeaderGroup().getHeaders(headerName);
- for (int i = 0; i < headers.length; i++) {
- getRequestHeaderGroup().removeHeader(headers[i]);
- }
- }
- /**
- * Removes the given request header.
- *
- * @param header the header
- */
- public void removeRequestHeader(final Header header) {
- if (header == null) {
- return;
- }
- getRequestHeaderGroup().removeHeader(header);
- }
- // ---------------------------------------------------------------- Queries
- /**
- * Returns <tt>true</tt> the method is ready to execute, <tt>false</tt> otherwise.
- *
- * @return This implementation always returns <tt>true</tt>.
- */
- public boolean validate() {
- return true;
- }
- /**
- * Returns the actual cookie policy
- *
- * @param state HTTP state. TODO: to be removed in the future
- *
- * @return cookie spec
- */
- private CookieSpec getCookieSpec(final HttpState state) {
- if (this.cookiespec == null) {
- int i = state.getCookiePolicy();
- if (i == -1) {
- this.cookiespec = CookiePolicy.getCookieSpec(this.params.getCookiePolicy());
- } else {
- this.cookiespec = CookiePolicy.getSpecByPolicy(i);
- }
- this.cookiespec.setValidDateFormats(
- (Collection)this.params.getParameter(HttpMethodParams.DATE_PATTERNS));
- }
- return this.cookiespec;
- }
- /**
- * Generates <tt>Cookie</tt> request headers for those {@link Cookie cookie}s
- * that match the given host, port and path.
- *
- * @param state the {@link HttpState state} information associated with this method
- * @param conn the {@link HttpConnection connection} used to execute
- * this HTTP method
- *
- * @throws IOException if an I/O (transport) error occurs. Some transport exceptions
- * can be recovered from.
- * @throws HttpException if a protocol exception occurs. Usually protocol exceptions
- * cannot be recovered from.
- */
- protected void addCookieRequestHeader(HttpState state, HttpConnection conn)
- throws IOException, HttpException {
- LOG.trace("enter HttpMethodBase.addCookieRequestHeader(HttpState, "
- + "HttpConnection)");
- Header[] cookieheaders = getRequestHeaderGroup().getHeaders("Cookie");
- for (int i = 0; i < cookieheaders.length; i++) {
- Header cookieheader = cookieheaders[i];
- if (cookieheader.isAutogenerated()) {
- getRequestHeaderGroup().removeHeader(cookieheader);
- }
- }
- CookieSpec matcher = getCookieSpec(state);
- Cookie[] cookies = matcher.match(conn.getHost(), conn.getPort(),
- getPath(), conn.isSecure(), state.getCookies());
- if ((cookies != null) && (cookies.length > 0)) {
- if (getParams().isParameterTrue(HttpMethodParams.SINGLE_COOKIE_HEADER)) {
- // In strict mode put all cookies on the same header
- String s = matcher.formatCookies(cookies);
- getRequestHeaderGroup().addHeader(new Header("Cookie", s, true));
- } else {
- // In non-strict mode put each cookie on a separate header
- for (int i = 0; i < cookies.length; i++) {
- String s = matcher.formatCookie(cookies[i]);
- getRequestHeaderGroup().addHeader(new Header("Cookie", s, true));
- }
- }
- }
- }
- /**
- * Generates <tt>Host</tt> request header, as long as no <tt>Host</tt> request
- * header already exists.
- *
- * @param state the {@link HttpState state} information associated with this method
- * @param conn the {@link HttpConnection connection} used to execute
- * this HTTP method
- *
- * @throws IOException if an I/O (transport) error occurs. Some transport exceptions
- * can be recovered from.
- * @throws HttpException if a protocol exception occurs. Usually protocol exceptions
- * cannot be recovered from.
- */
- protected void addHostRequestHeader(HttpState state, HttpConnection conn)
- throws IOException, HttpException {
- LOG.trace("enter HttpMethodBase.addHostRequestHeader(HttpState, "
- + "HttpConnection)");
- // Per 19.6.1.1 of RFC 2616, it is legal for HTTP/1.0 based
- // applications to send the Host request-header.
- // TODO: Add the ability to disable the sending of this header for
- // HTTP/1.0 requests.
- String host = conn.getVirtualHost();
- if (host != null) {
- LOG.debug("Using virtual host name: " + host);
- } else {
- host = conn.getHost();
- }
- int port = conn.getPort();
- // Note: RFC 2616 uses the term "internet host name" for what goes on the
- // host line. It would seem to imply that host should be blank if the
- // host is a number instead of an name. Based on the behavior of web
- // browsers, and the fact that RFC 2616 never defines the phrase "internet
- // host name", and the bad behavior of HttpClient that follows if we
- // send blank, I interpret this as a small misstatement in the RFC, where
- // they meant to say "internet host". So IP numbers get sent as host
- // entries too. -- Eric Johnson 12/13/2002
- if (LOG.isDebugEnabled()) {
- LOG.debug("Adding Host request header");
- }
- //appends the port only if not using the default port for the protocol
- if (conn.getProtocol().getDefaultPort() != port) {
- host += (":" + port);
- }
- setRequestHeader("Host", host);
- }
- /**
- * Generates <tt>Proxy-Connection: Keep-Alive</tt> request header when
- * communicating via a proxy server.
- *
- * @param state the {@link HttpState state} information associated with this method
- * @param conn the {@link HttpConnection connection} used to execute
- * this HTTP method
- *
- * @throws IOException if an I/O (transport) error occurs. Some transport exceptions
- * can be recovered from.
- * @throws HttpException if a protocol exception occurs. Usually protocol exceptions
- * cannot be recovered from.
- */
- protected void addProxyConnectionHeader(HttpState state,
- HttpConnection conn)
- throws IOException, HttpException {
- LOG.trace("enter HttpMethodBase.addProxyConnectionHeader("
- + "HttpState, HttpConnection)");
- if (!conn.isTransparent()) {
- setRequestHeader("Proxy-Connection", "Keep-Alive");
- }
- }
- /**
- * Generates all the required request {@link Header header}s
- * to be submitted via the given {@link HttpConnection connection}.
- *
- * <p>
- * This implementation adds <tt>User-Agent</tt>, <tt>Host</tt>,
- * <tt>Cookie</tt>, <tt>Authorization</tt>, <tt>Proxy-Authorization</tt>
- * and <tt>Proxy-Connection</tt> headers, when appropriate.
- * </p>
- *
- * <p>
- * Subclasses may want to override this method to to add additional
- * headers, and may choose to invoke this implementation (via
- * <tt>super</tt>) to add the "standard" headers.
- * </p>
- *
- * @param state the {@link HttpState state} information associated with this method
- * @param conn the {@link HttpConnection connection} used to execute
- * this HTTP method
- *
- * @throws IOException if an I/O (transport) error occurs. Some transport exceptions
- * can be recovered from.
- * @throws HttpException if a protocol exception occurs. Usually protocol exceptions
- * cannot be recovered from.
- *
- * @see #writeRequestHeaders
- */
- protected void addRequestHeaders(HttpState state, HttpConnection conn)
- throws IOException, HttpException {
- LOG.trace("enter HttpMethodBase.addRequestHeaders(HttpState, "
- + "HttpConnection)");
- addUserAgentRequestHeader(state, conn);
- addHostRequestHeader(state, conn);
- addCookieRequestHeader(state, conn);
- addProxyConnectionHeader(state, conn);
- }
- /**
- * Generates default <tt>User-Agent</tt> request header, as long as no
- * <tt>User-Agent</tt> request header already exists.
- *
- * @param state the {@link HttpState state} information associated with this method
- * @param conn the {@link HttpConnection connection} used to execute
- * this HTTP method
- *
- * @throws IOException if an I/O (transport) error occurs. Some transport exceptions
- * can be recovered from.
- * @throws HttpException if a protocol exception occurs. Usually protocol exceptions
- * cannot be recovered from.
- */
- protected void addUserAgentRequestHeader(HttpState state,
- HttpConnection conn)
- throws IOException, HttpException {
- LOG.trace("enter HttpMethodBase.addUserAgentRequestHeaders(HttpState, "
- + "HttpConnection)");
- if (getRequestHeader("User-Agent") == null) {
- String agent = (String)getParams().getParameter(HttpMethodParams.USER_AGENT);
- if (agent == null) {
- agent = "Jakarta Commons-HttpClient";
- }
- setRequestHeader("User-Agent", agent);
- }
- }
- /**
- * Throws an {@link IllegalStateException} if the HTTP method has been already
- * {@link #execute executed}, but not {@link #recycle recycled}.
- *
- * @throws IllegalStateException if the method has been used and not
- * recycled
- */
- protected void checkNotUsed() throws IllegalStateException {
- if (used) {
- throw new IllegalStateException("Already used.");
- }
- }
- /**
- * Throws an {@link IllegalStateException} if the HTTP method has not been
- * {@link #execute executed} since last {@link #recycle recycle}.
- *
- *
- * @throws IllegalStateException if not used
- */
- protected void checkUsed() throws IllegalStateException {
- if (!used) {
- throw new IllegalStateException("Not Used.");
- }
- }
- // ------------------------------------------------- Static Utility Methods
- /**
- * Generates HTTP request line according to the specified attributes.
- *
- * @param connection the {@link HttpConnection connection} used to execute
- * this HTTP method
- * @param name the method name generate a request for
- * @param requestPath the path string for the request
- * @param query the query string for the request
- * @param version the protocol version to use (e.g. HTTP/1.0)
- *
- * @return HTTP request line
- */
- protected static String generateRequestLine(HttpConnection connection,
- String name, String requestPath, String query, String version) {
- LOG.trace("enter HttpMethodBase.generateRequestLine(HttpConnection, "
- + "String, String, String, String)");
- StringBuffer buf = new StringBuffer();
- // Append method name
- buf.append(name);
- buf.append(" ");
- // Absolute or relative URL?
- if (!connection.isTransparent()) {
- Protocol protocol = connection.getProtocol();
- buf.append(protocol.getScheme().toLowerCase());
- buf.append("://");
- buf.append(connection.getHost());
- if ((connection.getPort() != -1)
- && (connection.getPort() != protocol.getDefaultPort())
- ) {
- buf.append(":");
- buf.append(connection.getPort());
- }
- }
- // Append path, if any
- if (requestPath == null) {
- buf.append("/");
- } else {
- if (!connection.isTransparent() && !requestPath.startsWith("/")) {
- buf.append("/");
- }
- buf.append(requestPath);
- }
- // Append query, if any
- if (query != null) {
- if (query.indexOf("?") != 0) {
- buf.append("?");
- }
- buf.append(query);
- }
- // Append protocol
- buf.append(" ");
- buf.append(version);
- buf.append("\r\n");
- return buf.toString();
- }
- /**
- * This method is invoked immediately after
- * {@link #readResponseBody(HttpState,HttpConnection)} and can be overridden by
- * sub-classes in order to provide custom body processing.
- *
- * <p>
- * This implementation does nothing.
- * </p>
- *
- * @param state the {@link HttpState state} information associated with this method
- * @param conn the {@link HttpConnection connection} used to execute
- * this HTTP method
- *
- * @see #readResponse
- * @see #readResponseBody
- */
- protected void processResponseBody(HttpState state, HttpConnection conn) {
- }
- /**
- * This method is invoked immediately after
- * {@link #readResponseHeaders(HttpState,HttpConnection)} and can be overridden by
- * sub-classes in order to provide custom response headers processing.
- * <p>
- * This implementation will handle the <tt>Set-Cookie</tt> and
- * <tt>Set-Cookie2</tt> headers, if any, adding the relevant cookies to
- * the given {@link HttpState}.
- * </p>
- *
- * @param state the {@link HttpState state} information associated with this method
- * @param conn the {@link HttpConnection connection} used to execute
- * this HTTP method
- *
- * @see #readResponse
- * @see #readResponseHeaders
- */
- protected void processResponseHeaders(HttpState state,
- HttpConnection conn) {
- LOG.trace("enter HttpMethodBase.processResponseHeaders(HttpState, "
- + "HttpConnection)");
- Header[] headers = getResponseHeaderGroup().getHeaders("set-cookie2");
- //Only process old style set-cookie headers if new style headres
- //are not present
- if (headers.length == 0) {
- headers = getResponseHeaderGroup().getHeaders("set-cookie");
- }
- CookieSpec parser = getCookieSpec(state);
- for (int i = 0; i < headers.length; i++) {
- Header header = headers[i];
- Cookie[] cookies = null;