| /* |
| * Copyright (c) 1997, 2012, Oracle and/or its affiliates. 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. Oracle designates this |
| * particular file as subject to the "Classpath" exception as provided |
| * by Oracle 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 Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA |
| * or visit www.oracle.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 class ByteArrayBuffer extends OutputStream { |
| /** |
| * The buffer where data is stored. |
| */ |
| protected byte[] buf; |
| |
| /** |
| * The number of valid bytes in the buffer. |
| */ |
| private int count; |
| |
| private static final int CHUNK_SIZE = 4096; |
| |
| /** |
| * 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(data,data.length); |
| } |
| |
| public ByteArrayBuffer(byte[] data, int length) { |
| this.buf = data; |
| this.count = length; |
| } |
| |
| /** |
| * 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 final 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 final void write(int b) { |
| int newcount = count + 1; |
| ensureCapacity(newcount); |
| buf[count] = (byte) b; |
| count = newcount; |
| } |
| |
| public final 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 final void writeTo(OutputStream out) throws IOException { |
| // Instead of writing out.write(buf, 0, count) |
| // Writing it in chunks that would help larger payloads |
| // Also if out is System.out on windows, it doesn't show on the console |
| // for larger data. |
| int remaining = count; |
| int off = 0; |
| while(remaining > 0) { |
| int chunk = (remaining > CHUNK_SIZE) ? CHUNK_SIZE : remaining; |
| out.write(buf, off, chunk); |
| remaining -= chunk; |
| off += chunk; |
| } |
| } |
| |
| public final 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 final byte[] toByteArray() { |
| byte newbuf[] = new byte[count]; |
| System.arraycopy(buf, 0, newbuf, 0, count); |
| return newbuf; |
| } |
| |
| public final int size() { |
| return count; |
| } |
| |
| /** |
| * Gets the underlying buffer that this {@link ByteArrayBuffer} uses. |
| * It's never small than its {@link #size()}. |
| * |
| * Use with caution. |
| */ |
| public final byte[] getRawData() { |
| return buf; |
| } |
| |
| public void close() throws IOException { |
| } |
| |
| /** |
| * Creates a new {@link InputStream} that reads from this buffer. |
| */ |
| public final InputStream newInputStream() { |
| return new ByteArrayInputStream(buf,0,count); |
| } |
| |
| /** |
| * Creates a new {@link InputStream} that reads a part of this bfufer. |
| */ |
| public final InputStream newInputStream(int start, int length) { |
| return new ByteArrayInputStream(buf,start,length); |
| } |
| |
| /** |
| * Decodes the contents of this buffer by the default encoding |
| * and returns it as a string. |
| * |
| * <p> |
| * Meant to aid debugging, but no more. |
| */ |
| public String toString() { |
| return new String(buf, 0, count); |
| } |
| } |