io.soluble.pjb.bridge.http.ChunkedOutputStream Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of php-java-bridge Show documentation
PHPJavaBridge server (soluble fork)
There is a newer version: 7.1.3
/*-*- mode: Java; tab-width:8 -*-*/

package io.soluble.pjb.bridge.http;

/*
 * Copyright (C) 2003-2007 Jost Boekemeier
 *
 * Permission is hereby granted, free of charge, to any person obtaining a
 * copy of this software and associated documentation files (the "Software"),
 * to deal in the Software without restriction, including without limitation
 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
 * and/or sell copies of the Software, and to permit persons to whom the
 * Software is furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
 * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR
 * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
 * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
 * OTHER DEALINGS IN THE SOFTWARE.
 */

import java.io.BufferedOutputStream;
import java.io.FilterOutputStream;
import java.io.IOException;
import java.io.OutputStream;

import io.soluble.pjb.bridge.Util;

/**
 * An output stream which writes data in HTTP chunks.
 *
 * @author jostb
 */
public class ChunkedOutputStream extends FilterOutputStream {

    /**
     * Create new new chunked output stream
     *
     * @param out
     */
    public ChunkedOutputStream(OutputStream out) {
        super(new BufferedOutputStream(out));
    }

    private static final byte[] RN0 = "0\r\n\r\n".getBytes();

    private void writeEOF() throws IOException {
        out.write(RN0);
    }

    private byte[] buf = new byte[8];

    /**
     * Write a length as hex digits.
     *
     * @param length the length, must be > 0
     * @throws IOException
     */
    private void writeHex(int length) throws IOException {
        int i = buf.length - 1;
        for (; length > 0; i--) {
            buf[i] = Util.HEX_DIGITS[0xF & length];
            length >>>= 4;
        }
        i++;
        out.write(buf, i, buf.length - i);
    }

    /**
     * {@inheritDoc}
     */
    public void write(byte[] buf, int pos, int len) throws IOException {
        if (len == 0) {
            writeEOF();
        } else {
            writeHex(len);
            out.write(Util.RN);
            out.write(buf, pos, len);
            out.write(Util.RN);
        }
        out.flush();
    }

    /**
     * Write trailing 0\r\n and flush the underlying output stream
     */
    public void eof() throws IOException {
        writeEOF();
        super.flush();
    }
}