internal/jimage/ImageStream.java

/*
 * Copyright (c) 2014, 2016, 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 jdk.internal.jimage;

import java.nio.ByteBuffer;
import java.nio.ByteOrder;
import java.util.Arrays;
import java.util.Objects;

/**
 * @implNote This class needs to maintain JDK 8 source compatibility.
 *
 * It is used internally in the JDK to implement jimage/jrtfs access,
 * but also compiled and delivered as part of the jrtfs.jar to support access
 * to the jimage file provided by the shipped JDK by tools running on JDK 8.
 */
public class ImageStream {
    private ByteBuffer buffer;

    public ImageStream() {
        this(1024, ByteOrder.nativeOrder());
    }

    public ImageStream(int size) {
        this(size, ByteOrder.nativeOrder());
    }

    public ImageStream(byte[] bytes) {
       this(bytes, ByteOrder.nativeOrder());
    }

    public ImageStream(ByteOrder byteOrder) {
        this(1024, byteOrder);
    }

    public ImageStream(int size, ByteOrder byteOrder) {
        buffer = ByteBuffer.allocate(size);
        buffer.order(Objects.requireNonNull(byteOrder));
    }

    public ImageStream(byte[] bytes, ByteOrder byteOrder) {
        buffer = ByteBuffer.wrap(Objects.requireNonNull(bytes));
        buffer.order(Objects.requireNonNull(byteOrder));
    }

    public ImageStream(ByteBuffer buffer) {
        this.buffer = Objects.requireNonNull(buffer);
    }

    public ImageStream align(int alignment) {
        int padding = (getSize() - 1) & ((1 << alignment) - 1);

        for (int i = 0; i < padding; i++) {
            put((byte)0);
        }

        return this;
    }

    public void ensure(int needs) {
        if (needs < 0) {
            throw new IndexOutOfBoundsException("Bad value: " + needs);
        }

        if (needs > buffer.remaining()) {
            byte[] bytes = buffer.array();
            ByteOrder byteOrder = buffer.order();
            int position = buffer.position();
            int newSize = needs <= bytes.length ? bytes.length << 1 : position + needs;
            buffer = ByteBuffer.allocate(newSize);
            buffer.order(byteOrder);
            buffer.put(bytes, 0, position);
        }
    }

    public boolean hasByte() {
        return buffer.remaining() != 0;
    }

    public boolean hasBytes(int needs) {
        return needs <= buffer.remaining();
    }

    public void skip(int n) {
        if (n < 0) {
            throw new IndexOutOfBoundsException("skip value = " + n);
        }

        buffer.position(buffer.position() + n);
    }

    public int get() {
        return buffer.get() & 0xFF;
    }

    public void get(byte bytes[], int offset, int size) {
        buffer.get(bytes, offset, size);
    }

    public int getShort() {
        return buffer.getShort();
    }

    public int getInt() {
        return buffer.getInt();
    }

    public long getLong() {
        return buffer.getLong();
    }

    public ImageStream put(byte byt) {
        ensure(1);
        buffer.put(byt);

        return this;
    }

    public ImageStream put(int byt) {
        return put((byte)byt);
    }

    public ImageStream put(byte bytes[], int offset, int size) {
        ensure(size);
        buffer.put(bytes, offset, size);

        return this;
    }

    public ImageStream put(ImageStream stream) {
        put(stream.buffer.array(), 0, stream.buffer.position());

        return this;
    }

    public ImageStream putShort(short value) {
        ensure(2);
        buffer.putShort(value);

        return this;
    }

    public ImageStream putShort(int value) {
        return putShort((short)value);
    }

    public ImageStream putInt(int value) {
        ensure(4);
        buffer.putInt(value);

        return this;
    }

    public ImageStream putLong(long value) {
        ensure(8);
        buffer.putLong(value);

        return this;
    }

    public ByteBuffer getBuffer() {
        return buffer;
    }

    public int getPosition() {
        return buffer.position();
    }

    public int getSize() {
        return buffer.position();
    }

    public byte[] getBytes() {
        return buffer.array();
    }

    public void setPosition(int offset) {
        buffer.position(offset);
    }

    public byte[] toArray() {
        return Arrays.copyOf(buffer.array(), buffer.position());
    }
}