/*
 * The Apache Software License, Version 1.1
 *
 *
 * Copyright (c) 2001-2003 The Apache Software Foundation.  All rights
 * reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 *
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in
 *    the documentation and/or other materials provided with the
 *    distribution.
 *
 * 3. The end-user documentation included with the redistribution,
 *    if any, must include the following acknowledgment:
 *       "This product includes software developed by the
 *        Apache Software Foundation (http://www.apache.org/)."
 *    Alternately, this acknowledgment may appear in the software itself,
 *    if and wherever such third-party acknowledgments normally appear.
 *
 * 4. The names "Axis" and "Apache Software Foundation" must
 *    not be used to endorse or promote products derived from this
 *    software without prior written permission. For written
 *    permission, please contact apache@apache.org.
 *
 * 5. Products derived from this software may not be called "Apache",
 *    nor may "Apache" appear in their name, without prior written
 *    permission of the Apache Software Foundation.
 *
 * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESSED OR IMPLIED
 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED.  IN NO EVENT SHALL THE APACHE SOFTWARE FOUNDATION OR
 * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
 * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
 * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 * ====================================================================
 *
 * 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.jboss.axis.transport.http;


import java.io.IOException;
import java.io.InputStream;


/**
 * @author Rick Rineholt
 */

public class ChunkedInputStream extends java.io.FilterInputStream
{
   protected long chunkSize = 0l;
   protected volatile boolean closed = false;
   private static final int maxCharLong = (Long.toHexString(Long.MAX_VALUE))
           .toString().length();

   private ChunkedInputStream()
   {
      super(null);
   }

   public ChunkedInputStream(InputStream is)
   {
      super(is);
   }

   public int read()
           throws IOException
   {
      byte[] d = new byte[1];
      int rc = read(d, 0, 1);

      return rc > 0 ? (d[0] & 0xFF) : rc;
   }

   public int read(byte[] b)
           throws IOException
   {
      return read(b, 0, b.length);
   }

   public synchronized int read(byte[] b,
                                int off,
                                int len)
           throws IOException
   {
      if (closed) return -1;
      int cs = (int)Math.min(Integer.MAX_VALUE, chunkSize);
      int totalread = 0;
      int bytesread = 0;

      try
      {
         do
         {
            if (chunkSize < 1L)
            {
               if (0l == getChunked())
               {
                  if (totalread == 0)
                     return -1;
                  else
                     return totalread;
               }
            }
            bytesread = in.read(b, off + totalread, Math.min(len - totalread,
                    (int)Math.min(chunkSize, Integer.MAX_VALUE)));
            if (bytesread > 0)
            {
               totalread += bytesread;
               chunkSize -= bytesread;
            }
         }
         while (len - totalread > 0 && bytesread > -1);
      }
      catch (IOException e)
      {
         closed = true;
         throw e;
      }
      return totalread;
   }

   public long skip(final long n)
           throws IOException
   {
      if (closed) return 0;
      long skipped = 0l;
      byte[] b = new byte[1024];
      int bread = -1;

      do
      {
         bread = read(b, 0, b.length);
         if (bread > 0) skipped += bread;
      }
      while (bread != -1 && skipped < n);
      return skipped;
   }

   public int available()
           throws IOException
   {
      if (closed) return 0;
      int rc = (int)Math.min(chunkSize, Integer.MAX_VALUE);

      return Math.min(rc, in.available());
   }

   protected long getChunked() throws IOException
   {
      //StringBuffer buf= new StringBuffer(1024);
      byte[] buf = new byte[maxCharLong + 2];
      int bufsz = 0;

      chunkSize = -1L;
      int c = -1;

      do
      {
         c = in.read();
         if (c > -1)
         {
            if (c != '\r' && c != '\n' && c != ' ' && c != '\t')
            {
               buf[bufsz++] = ((byte)c);
            }
         }
      }
      while (c > -1 && (c != '\n' || bufsz == 0) && bufsz < buf.length);
      if (c < 0)
      {
         closed = true;
      }
      String sbuf = new String(buf, 0, bufsz);

      if (bufsz > maxCharLong)
      {
         closed = true;
         throw new IOException("Chunked input stream failed to receive valid chunk size:" + sbuf);
      }
      try
      {
         chunkSize = Long.parseLong(sbuf, 16);
      }
      catch (NumberFormatException ne)
      {
         closed = true;
         throw new IOException("'" + sbuf + "' " + ne.getMessage());
      }
      if (chunkSize < 1L) closed = true;
      if (chunkSize != 0L && c < 0)
      {
         //If chunk size is zero try and be tolerant that there maybe no cr or lf at the end.
         throw new IOException("HTTP Chunked stream closed in middle of chunk.");
      }
      if (chunkSize < 0L)
         throw new IOException("HTTP Chunk size received " +
                 chunkSize + " is less than zero.");
      return chunkSize;
   }

   public void close() throws IOException
   {

      synchronized (this)
      {
         if (closed) return;
         closed = true;
      }

      byte[] b = new byte[1024];
      int bread = -1;

      do
      {
         bread = read(b, 0, b.length);
      }
      while (bread != -1);
   }

   /*
    public void mark(int readlimit)
    {

    }
    */

   public void reset()
           throws IOException
   {
      throw new IOException("Don't support marked streams");
   }

   public boolean markSupported()
   {
      return false;
   }

}