| /* |
| * Copyright (c) 1997, 2021 Oracle and/or its affiliates. All rights reserved. |
| * |
| * This program and the accompanying materials are made available under the |
| * terms of the Eclipse Distribution License v. 1.0, which is available at |
| * http://www.eclipse.org/org/documents/edl-v10.php. |
| * |
| * SPDX-License-Identifier: BSD-3-Clause |
| */ |
| |
| package org.glassfish.jaxb.runtime.v2.util; |
| |
| import org.glassfish.jaxb.runtime.v2.runtime.unmarshaller.Base64Data; |
| |
| import java.io.ByteArrayOutputStream; |
| import java.io.IOException; |
| import java.io.InputStream; |
| |
| /** |
| * {@link ByteArrayOutputStream} with access to its raw buffer. |
| */ |
| public final class ByteArrayOutputStreamEx extends ByteArrayOutputStream { |
| public ByteArrayOutputStreamEx() { |
| } |
| |
| public ByteArrayOutputStreamEx(int size) { |
| super(size); |
| } |
| |
| public void set(Base64Data dt, String mimeType) { |
| dt.set(buf,count,mimeType); |
| } |
| |
| public byte[] getBuffer() { |
| return buf; |
| } |
| |
| /** |
| * Reads the given {@link InputStream} completely into the buffer. |
| */ |
| public void readFrom(InputStream is) throws IOException { |
| while(true) { |
| if(count==buf.length) { |
| // realllocate |
| byte[] data = new byte[buf.length*2]; |
| System.arraycopy(buf,0,data,0,buf.length); |
| buf = data; |
| } |
| |
| int sz = is.read(buf,count,buf.length-count); |
| if(sz<0) return; |
| count += sz; |
| } |
| } |
| } |