WinstoneInputStream.java

/*
 * Copyright 2003-2006 Rick Knowles <winstone-devel at lists sourceforge net>
 * Distributed under the terms of either:
 * - the common development and distribution license (CDDL), v1.0; or
 * - the GNU Lesser General Public License, v2.1 or later
 */
package winstone;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;

/**
 * The request stream management class.
 * 
 * @author <a href="mailto:rick_knowles@hotmail.com">Rick Knowles</a>
 * @version $Id: WinstoneInputStream.java,v 1.4 2006/02/28 07:32:47 rickknowles Exp $
 */
public class WinstoneInputStream extends javax.servlet.ServletInputStream {
    final int BUFFER_SIZE = 4096;
    private InputStream inData;
    private Integer contentLength;
    private int readSoFar;
    private ByteArrayOutputStream dump;
    
    /**
     * Constructor
     */
    public WinstoneInputStream(InputStream inData) {
        super();
        this.inData = inData;
        this.dump = new ByteArrayOutputStream();
    }

    public WinstoneInputStream(byte inData[]) {
        this(new ByteArrayInputStream(inData));
    }

    public InputStream getRawInputStream() {
        return this.inData;
    }

    public void setContentLength(int length) {
        this.contentLength = new Integer(length);
        this.readSoFar = 0;
    }

    public int read() throws IOException {
        if (this.contentLength == null) {
            int data = this.inData.read();
            this.dump.write(data);
//            System.out.println("Char: " + (char) data);
            return data;
        } else if (this.contentLength.intValue() > this.readSoFar) {
            this.readSoFar++;
            int data = this.inData.read();
            this.dump.write(data);
//            System.out.println("Char: " + (char) data);
            return data;
        } else
            return -1;
    }

    public void finishRequest() {
        // this.inData = null;
        // byte content[] = this.dump.toByteArray();
        // com.rickknowles.winstone.ajp13.Ajp13Listener.packetDump(content,
        // content.length);
    }

    public int available() throws IOException {
        return this.inData.available();
    }

    /**
     * Wrapper for the servletInputStream's readline method
     */
    public byte[] readLine() throws IOException {
        // System.out.println("ReadLine()");
        byte buffer[] = new byte[BUFFER_SIZE];
        int charsRead = super.readLine(buffer, 0, BUFFER_SIZE);
        if (charsRead == -1) {
            Logger.log(Logger.DEBUG, Launcher.RESOURCES,
                    "WinstoneInputStream.EndOfStream");
            return new byte[0];
        }
        byte outBuf[] = new byte[charsRead];
        System.arraycopy(buffer, 0, outBuf, 0, charsRead);
        return outBuf;
    }

}