The Tor Browser / file revision

 /*

  * ====================================================================

  * 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.

  * ====================================================================

  *

  * 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 ch.boye.httpclientandroidlib.impl.entity;

 import java.io.IOException;

 import ch.boye.httpclientandroidlib.Header;

 import ch.boye.httpclientandroidlib.HttpEntity;

 import ch.boye.httpclientandroidlib.HttpException;

 import ch.boye.httpclientandroidlib.HttpMessage;

 import ch.boye.httpclientandroidlib.entity.BasicHttpEntity;

 import ch.boye.httpclientandroidlib.entity.ContentLengthStrategy;

 import ch.boye.httpclientandroidlib.impl.io.ChunkedInputStream;

 import ch.boye.httpclientandroidlib.impl.io.ContentLengthInputStream;

 import ch.boye.httpclientandroidlib.impl.io.IdentityInputStream;

 import ch.boye.httpclientandroidlib.io.SessionInputBuffer;

 import ch.boye.httpclientandroidlib.protocol.HTTP;

 /**

  * HTTP entity deserializer.

  * <p>

  * This entity deserializer supports "chunked" and "identitiy" transfer-coding

  * and content length delimited content.

  * <p>

  * This class relies on a specific implementation of

  * {@link ContentLengthStrategy} to determine the content length or transfer

  * encoding of the entity.

  * <p>

  * This class generates an instance of {@link HttpEntity} based on

  * properties of the message. The content of the entity will be decoded

  * transparently for the consumer.

  *

  * @since 4.0

  */

 public class EntityDeserializer {

     private final ContentLengthStrategy lenStrategy;

     public EntityDeserializer(final ContentLengthStrategy lenStrategy) {

         super();

         if (lenStrategy == null) {

             throw new IllegalArgumentException("Content length strategy may not be null");

         }

         this.lenStrategy = lenStrategy;

     }

     /**

      * Creates a {@link BasicHttpEntity} based on properties of the given

      * message. The content of the entity is created by wrapping

      * {@link SessionInputBuffer} with a content decoder depending on the

      * transfer mechanism used by the message.

      * <p>

      * This method is called by the public

      * {@link #deserialize(SessionInputBuffer, HttpMessage)}.

      *

      * @param inbuffer the session input buffer.

      * @param message the message.

      * @return HTTP entity.

      * @throws HttpException in case of HTTP protocol violation.

      * @throws IOException in case of an I/O error.

      */

     protected BasicHttpEntity doDeserialize(

             final SessionInputBuffer inbuffer,

             final HttpMessage message) throws HttpException, IOException {

         BasicHttpEntity entity = new BasicHttpEntity();

         long len = this.lenStrategy.determineLength(message);

         if (len == ContentLengthStrategy.CHUNKED) {

             entity.setChunked(true);

             entity.setContentLength(-1);

             entity.setContent(new ChunkedInputStream(inbuffer));

         } else if (len == ContentLengthStrategy.IDENTITY) {

             entity.setChunked(false);

             entity.setContentLength(-1);

             entity.setContent(new IdentityInputStream(inbuffer));

         } else {

             entity.setChunked(false);

             entity.setContentLength(len);

             entity.setContent(new ContentLengthInputStream(inbuffer, len));

         }

         Header contentTypeHeader = message.getFirstHeader(HTTP.CONTENT_TYPE);

         if (contentTypeHeader != null) {

             entity.setContentType(contentTypeHeader);

         }

         Header contentEncodingHeader = message.getFirstHeader(HTTP.CONTENT_ENCODING);

         if (contentEncodingHeader != null) {

             entity.setContentEncoding(contentEncodingHeader);

         }

         return entity;

     }

     /**

      * Creates an {@link HttpEntity} based on properties of the given message.

      * The content of the entity is created by wrapping

      * {@link SessionInputBuffer} with a content decoder depending on the

      * transfer mechanism used by the message.

      * <p>

      * The content of the entity is NOT retrieved by this method.

      *

      * @param inbuffer the session input buffer.

      * @param message the message.

      * @return HTTP entity.

      * @throws HttpException in case of HTTP protocol violation.

      * @throws IOException in case of an I/O error.

      */

     public HttpEntity deserialize(

             final SessionInputBuffer inbuffer,

             final HttpMessage message) throws HttpException, IOException {

         if (inbuffer == null) {

             throw new IllegalArgumentException("Session input buffer may not be null");

         }

         if (message == null) {

             throw new IllegalArgumentException("HTTP message may not be null");

         }

         return doDeserialize(inbuffer, message);

     }

 }