    1   /*
    2    * $HeadURL: https://svn.apache.org/repos/asf/jakarta/httpcomponents/oac.hc3x/tags/HTTPCLIENT_3_1/src/test/org/apache/commons/httpclient/TestBadContentLength.java $
    3    * $Revision: 480424 $
    4    * $Date: 2006-11-29 06:56:49 +0100 (Wed, 29 Nov 2006) $
    5    * ====================================================================
    6    *
    7    *  Licensed to the Apache Software Foundation (ASF) under one or more
    8    *  contributor license agreements.  See the NOTICE file distributed with
    9    *  this work for additional information regarding copyright ownership.
    10   *  The ASF licenses this file to You under the Apache License, Version 2.0
    11   *  (the "License"); you may not use this file except in compliance with
    12   *  the License.  You may obtain a copy of the License at
    13   *
    14   *      http://www.apache.org/licenses/LICENSE-2.0
    15   *
    16   *  Unless required by applicable law or agreed to in writing, software
    17   *  distributed under the License is distributed on an "AS IS" BASIS,
    18   *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    19   *  See the License for the specific language governing permissions and
    20   *  limitations under the License.
    21   * ====================================================================
    22   *
    23   * This software consists of voluntary contributions made by many
    24   * individuals on behalf of the Apache Software Foundation.  For more
    25   * information on the Apache Software Foundation, please see
    26   * <http://www.apache.org/>.
    27   *
    28   */
    29  package org.apache.commons.httpclient;
    30
    31  import java.io.IOException;
    32  import java.io.InputStream;
    33
    34  import junit.framework.Test;
    35  import junit.framework.TestCase;
    36  import junit.framework.TestSuite;
    37
    38  import org.apache.commons.httpclient.methods.GetMethod;
    39  import org.apache.commons.httpclient.server.HttpRequestHandler;
    40  import org.apache.commons.httpclient.server.RequestLine;
    41  import org.apache.commons.httpclient.server.ResponseWriter;
    42  import org.apache.commons.httpclient.server.SimpleHttpServer;
    43  import org.apache.commons.httpclient.server.SimpleHttpServerConnection;
    44  import org.apache.commons.httpclient.server.SimpleRequest;
    45
    46  /**
    47   * Tests HttpClient's behaviour when receiving more response data than expected.
    48   * <p>
    49   * A very simple HTTP Server will be setup on a free port during testing, which
    50   * returns an incorrect response Content-Length, sending surplus response data,
    51   * which may contain malicious/fake response headers.
    52   * </p>
    53   *
    54   * @author Christian Kohlschuetter
    55   * @version $Id: TestBadContentLength.java 480424 2006-11-29 05:56:49Z bayard $
    56   */
    57  public class TestBadContentLength extends TestCase {
    58      private HttpClient client = null;
    59      private SimpleHttpServer server = null;
    60
    61      // ------------------------------------------------------------ Constructor
    62      public TestBadContentLength(String testName) {
    63          super(testName);
    64      }
    65
    66      // ------------------------------------------------------------------- Main
    67      public static void main(String args[]) {
    68          String[] testCaseName = { TestBadContentLength.class.getName()};
    69          junit.textui.TestRunner.main(testCaseName);
    70      }
    71
    72      // ------------------------------------------------------- TestCase Methods
    73
    74      public static Test suite() {
    75          return new TestSuite(TestBadContentLength.class);
    76      }
    77
    78      // ----------------------------------------------------------- Test Methods
    79
    80      public void setUp() throws IOException {
    81          client = new HttpClient();
    82          server = new SimpleHttpServer(); // use arbitrary port
    83          server.setTestname(getName());
    84          server.setRequestHandler(new MyHttpRequestHandler());
    85      }
    86
    87      public void tearDown() throws IOException {
    88          client = null;
    89
    90          server.destroy();
    91      }
    92
    93      /**
    94       * HttpClient connects to the test server and performs two subsequent
    95       * requests to the same URI in <u>lenient</u> mode.
    96       *
    97       * Expected behavior:
    98       * For both requests, status code 200 and a response body of "12345"
    99       * should be returned.
    100      *
    101      * @throws IOException
    102      */
    103     public void test1Lenient() throws IOException {
    104     	client.getParams().makeLenient();
    105
    106         GetMethod m =
    107             new GetMethod("http://localhost:" + server.getLocalPort() + "/");
    108
    109         client.executeMethod(m);
    110         assertEquals(200, m.getStatusCode());
    111         assertEquals("12345", m.getResponseBodyAsString());
    112
    113         m = new GetMethod("http://localhost:" + server.getLocalPort() + "/");
    114
    115         client.executeMethod(m);
    116         assertEquals(200, m.getStatusCode());
    117         assertEquals("12345", m.getResponseBodyAsString());
    118         m.releaseConnection();
    119     }
    120
    121     /**
    122      * HttpClient connects to the test server and performs two subsequent
    123      * requests to the same URI in <u>strict</u> mode.
    124      * <p>
    125      * The first response body will be read with getResponseBodyAsString(),
    126      * which returns null if an error occured.
    127      * </p>
    128      * <p>
    129      * The second response body will be read using an InputStream, which
    130      * throws an IOException if something went wrong.
    131      * </p>
    132      * Expected behavior:
    133      * For both requests, status code 200 should be returned.<br />
    134      * For request 1, a <code>null</code> response body should be returned.<br />
    135      * For request 2, a {@link ProtocolException} is expected.
    136      *
    137      * @throws IOException
    138      */
    139     public void test1Strict() throws IOException {
    140         client.getParams().makeStrict();
    141
    142         GetMethod m =
    143             new GetMethod("http://localhost:" + server.getLocalPort() + "/");
    144
    145         client.executeMethod(m);
    146         assertEquals(200, m.getStatusCode());
    147         assertEquals("12345", m.getResponseBodyAsString());
    148
    149         m = new GetMethod("http://localhost:" + server.getLocalPort() + "/");
    150
    151         client.executeMethod(m);
    152         assertEquals(200, m.getStatusCode());
    153
    154         InputStream in = m.getResponseBodyAsStream();
    155         while (in.read() != -1) {
    156         }
    157
    158         m.releaseConnection();
    159     }
    160
    161     public void enableThisTestForDebuggingOnly()
    162         throws InterruptedException {
    163         while (server.isRunning()) {
    164             Thread.sleep(100);
    165         }
    166     }
    167
    168     private class MyHttpRequestHandler implements HttpRequestHandler {
    169         private int requestNo = 0;
    170
    171         public boolean processRequest(
    172             final SimpleHttpServerConnection conn,
    173             final SimpleRequest request) throws IOException
    174         {
    175         	RequestLine requestLine = request.getRequestLine();
    176         	ResponseWriter out = conn.getWriter();
    177             if ("GET".equals(requestLine.getMethod())
    178                 && "/".equals(requestLine.getUri())) {
    179
    180                 requestNo++;
    181
    182                 out.println("HTTP/1.1 200 OK");
    183                 out.println("Content-Type: text/html");
    184                 out.println("Content-Length: 5");
    185                 out.println("Connection: keep-alive");
    186                 out.println();
    187                 out.println("12345"); // send exactly 5 bytes
    188
    189                 // and some more garbage!
    190                 out.println("AND SOME MORE\r\nGARBAGE!");
    191                 out.println("HTTP/1.0 404 Not Found");
    192                 out.println("Content-Type: text/plain");
    193                 out.println("");
    194                 out.println("THIS-IS-A-FAKE-RESPONSE!");
    195
    196                 out.flush();
    197                 // process max. 2 subsequents requests per connection
    198                 if (requestNo < 2) {
    199                     conn.setKeepAlive(true);
    200                 }
    201             }
    202             return true;
    203         }
    204     }
    205
    206 }