| /* |
| * Portions Copyright 2006 Sun Microsystems, Inc. All Rights Reserved. |
| * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. |
| * |
| * This code is free software; you can redistribute it and/or modify it |
| * under the terms of the GNU General Public License version 2 only, as |
| * published by the Free Software Foundation. Sun designates this |
| * particular file as subject to the "Classpath" exception as provided |
| * by Sun in the LICENSE file that accompanied this code. |
| * |
| * This code is distributed in the hope that it will be useful, but WITHOUT |
| * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or |
| * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License |
| * version 2 for more details (a copy is included in the LICENSE file that |
| * accompanied this code). |
| * |
| * You should have received a copy of the GNU General Public License version |
| * 2 along with this work; if not, write to the Free Software Foundation, |
| * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. |
| * |
| * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara, |
| * CA 95054 USA or visit www.sun.com if you need additional information or |
| * have any questions. |
| */ |
| |
| package com.sun.xml.internal.ws.util; |
| |
| import java.io.ByteArrayInputStream; |
| import java.io.ByteArrayOutputStream; |
| import java.io.IOException; |
| import java.io.InputStream; |
| import java.io.OutputStream; |
| |
| /** |
| * Read/write buffer that stores a sequence of bytes. |
| * |
| * <p> |
| * It works in a way similar to {@link ByteArrayOutputStream} but |
| * this class works better in the following ways: |
| * |
| * <ol> |
| * <li>no synchronization |
| * <li>offers a {@link #newInputStream()} that creates a new {@link InputStream} |
| * that won't cause buffer reallocation. |
| * <li>less parameter correctness checking |
| * <li>offers a {@link #write(InputStream)} method that reads the entirety of the |
| * given {@link InputStream} without using a temporary buffer. |
| * </ol> |
| * |
| * @author Kohsuke Kawaguchi |
| */ |
| public final class ByteArrayBuffer extends OutputStream { |
| /** |
| * The buffer where data is stored. |
| */ |
| private byte buf[]; |
| |
| /** |
| * The number of valid bytes in the buffer. |
| */ |
| private int count; |
| |
| /** |
| * Creates a new byte array output stream. The buffer capacity is |
| * initially 32 bytes, though its size increases if necessary. |
| */ |
| public ByteArrayBuffer() { |
| this(32); |
| } |
| |
| /** |
| * Creates a new byte array output stream, with a buffer capacity of |
| * the specified size, in bytes. |
| * |
| * @param size the initial size. |
| * @throws IllegalArgumentException if size is negative. |
| */ |
| public ByteArrayBuffer(int size) { |
| if (size <= 0) |
| throw new IllegalArgumentException(); |
| buf = new byte[size]; |
| } |
| |
| public ByteArrayBuffer(byte[] data) { |
| this.buf = data; |
| } |
| |
| /** |
| * Reads all the data of the given {@link InputStream} and appends them |
| * into this buffer. |
| * |
| * @throws IOException |
| * if the read operation fails with an {@link IOException}. |
| */ |
| public void write(InputStream in) throws IOException { |
| while(true) { |
| int cap = buf.length-count; // the remaining buffer space |
| int sz = in.read(buf,count,cap); |
| if(sz<0) return; // hit EOS |
| count += sz; |
| |
| |
| if(cap==sz) |
| ensureCapacity(buf.length*2); // buffer filled up. |
| } |
| } |
| |
| public void write(int b) { |
| int newcount = count + 1; |
| ensureCapacity(newcount); |
| buf[count] = (byte) b; |
| count = newcount; |
| } |
| |
| public void write(byte b[], int off, int len) { |
| int newcount = count + len; |
| ensureCapacity(newcount); |
| System.arraycopy(b, off, buf, count, len); |
| count = newcount; |
| } |
| |
| private void ensureCapacity(int newcount) { |
| if (newcount > buf.length) { |
| byte newbuf[] = new byte[Math.max(buf.length << 1, newcount)]; |
| System.arraycopy(buf, 0, newbuf, 0, count); |
| buf = newbuf; |
| } |
| } |
| |
| public void writeTo(OutputStream out) throws IOException { |
| out.write(buf, 0, count); |
| } |
| |
| public void reset() { |
| count = 0; |
| } |
| |
| /** |
| * Gets the <b>copy</b> of exact-size byte[] that represents the written data. |
| * |
| * <p> |
| * Since this method needs to allocate a new byte[], this method will be costly. |
| * |
| * @deprecated |
| * this method causes a buffer reallocation. Use it only when |
| * you have to. |
| */ |
| public byte toByteArray()[] { |
| byte newbuf[] = new byte[count]; |
| System.arraycopy(buf, 0, newbuf, 0, count); |
| return newbuf; |
| } |
| |
| public int size() { |
| return count; |
| } |
| |
| public void close() { |
| } |
| |
| /** |
| * Creates a new {@link InputStream} that reads from this buffer. |
| */ |
| public InputStream newInputStream() { |
| return new ByteArrayInputStream(buf,0,count); |
| } |
| |
| /** |
| * Creates a new {@link InputStream} that reads a part of this bfufer. |
| */ |
| public InputStream newInputStream(int start, int length) { |
| return new ByteArrayInputStream(buf,start,length); |
| } |
| } |