lambda/separate/ClassFile.java

/*
 * Copyright (c) 2012, 2013, 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.
 *
 * 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 separate;

import java.io.*;
import java.util.*;

class CfInputStream extends ByteArrayInputStream {
    private int ct;
    public CfInputStream(byte[] input) {
        super(input);
    }

    byte u1() { return (byte)read(); }
    short u2() {
        int b0 = read() << 8;
        int b1 = read();
        return (short)(b0 | b1);
    }
    int u4() {
        int b0 = read() << 24;
        int b1 = read() << 16;
        int b2 = read() << 8;
        int b3 = read();
        return b0 | b1 | b2 | b3;
    }
    byte[] array(int count) {
        byte[] ret = new byte[count];
        read(ret, 0, count);
        return ret;
    }
};

class CfOutputStream extends ByteArrayOutputStream {
    void u1(byte b) { write((int)b); }
    void u2(short s) {
        write((s >> 8) & 0xff);
        write(s & 0xff);
    }
    void u4(int i) {
        write((i >> 24) & 0xff);
        write((i >> 16) & 0xff);
        write((i >> 8) & 0xff);
        write(i & 0xff);
    }
    void array(byte[] a) {
        write(a, 0, a.length);
    }

    public byte[] toByteArray() { return super.toByteArray(); }
};

// A quick and dirty class file parser and representation
public class ClassFile {

    int magic;
    short minor_version;
    short major_version;
    ArrayList<CpEntry> constant_pool;
    short access_flags;
    short this_class;
    short super_class;
    ArrayList<Interface> interfaces;
    ArrayList<Field> fields;
    ArrayList<Method> methods;
    ArrayList<Attribute> attributes;

    ClassFile(byte[] cf) {
        CfInputStream in = new CfInputStream(cf);

        magic = in.u4();
        minor_version = in.u2();
        major_version = in.u2();

        short cpCount = in.u2();
        constant_pool = new ArrayList<>();
        constant_pool.add(new CpNull());
        for (int i = 1; i < cpCount; ++i) {
            constant_pool.add(CpEntry.newCpEntry(in));
        }

        access_flags = in.u2();
        this_class = in.u2();
        super_class = in.u2();

        short ifaceCount = in.u2();
        interfaces = new ArrayList<>();
        for (int i = 0; i < ifaceCount; ++i) {
            interfaces.add(new Interface(in));
        }

        short fieldCount = in.u2();
        fields = new ArrayList<>();
        for (int i = 0; i < fieldCount; ++i) {
            fields.add(new Field(in));
        }

        short methodCount = in.u2();
        methods = new ArrayList<>();
        for (int i = 0; i < methodCount; ++i) {
            methods.add(new Method(in));
        }

        short attributeCount = in.u2();
        attributes = new ArrayList<>();
        for (int i = 0; i < attributeCount; ++i) {
            attributes.add(new Attribute(in));
        }
    }

    byte[] toByteArray() {
        CfOutputStream out = new CfOutputStream();

        out.u4(magic);
        out.u2(minor_version);
        out.u2(major_version);

        out.u2((short)(constant_pool.size()));
        for (CpEntry cp : constant_pool) {
            cp.write(out);
        }

        out.u2(access_flags);
        out.u2(this_class);
        out.u2(super_class);

        out.u2((short)interfaces.size());
        for (Interface iface : interfaces) {
            iface.write(out);
        }

        out.u2((short)fields.size());
        for (Field field : fields) {
            field.write(out);
        }

        out.u2((short)methods.size());
        for (Method method : methods) {
            method.write(out);
        }

        out.u2((short)attributes.size());
        for (Attribute attribute : attributes) {
            attribute.write(out);
        }

        return out.toByteArray();
    }

    static abstract class CpEntry {
        byte tag;

        CpEntry(byte t) { tag = t; }
        void write(CfOutputStream out) {
            out.u1(tag);
        }

        static CpEntry newCpEntry(CfInputStream in) {
            byte tag = in.u1();
            switch (tag) {
                case CpUtf8.TAG: return new CpUtf8(in);
                case CpInteger.TAG: return new CpInteger(in);
                case CpFloat.TAG: return new CpFloat(in);
                case CpLong.TAG: return new CpLong(in);
                case CpDouble.TAG: return new CpDouble(in);
                case CpClass.TAG: return new CpClass(in);
                case CpString.TAG: return new CpString(in);
                case CpFieldRef.TAG: return new CpFieldRef(in);
                case CpMethodRef.TAG: return new CpMethodRef(in);
                case CpInterfaceMethodRef.TAG:
                    return new CpInterfaceMethodRef(in);
                case CpNameAndType.TAG: return new CpNameAndType(in);
                case CpMethodHandle.TAG: return new CpMethodHandle(in);
                case CpMethodType.TAG: return new CpMethodType(in);
                case CpInvokeDynamic.TAG: return new CpInvokeDynamic(in);
                default: throw new RuntimeException("Bad cp entry tag: " + tag);
            }
        }
    }

    static class CpNull extends CpEntry {
        CpNull() { super((byte)0); }
        CpNull(CfInputStream in) { super((byte)0); }
        void write(CfOutputStream out) {}
    }

    static class CpUtf8 extends CpEntry {
        static final byte TAG = 1;
        byte[] bytes;

        CpUtf8() { super(TAG); }
        CpUtf8(CfInputStream in) {
            this();
            short length = in.u2();
            bytes = in.array(length);
        }
        void write(CfOutputStream out) {
            super.write(out);
            out.u2((short)bytes.length);
            out.array(bytes);
        }
    }

    static class CpU4Constant extends CpEntry {
        byte[] bytes;

        CpU4Constant(byte tag) { super(tag); }
        CpU4Constant(byte tag, CfInputStream in) {
            this(tag);
            bytes = in.array(4);
        }
        void write(CfOutputStream out) { super.write(out); out.array(bytes); }
    }
    static class CpInteger extends CpU4Constant {
        static final byte TAG = 3;
        CpInteger() { super(TAG); }
        CpInteger(CfInputStream in) { super(TAG, in); }
    }
    static class CpFloat extends CpU4Constant {
        static final byte TAG = 4;
        CpFloat() { super(TAG); }
        CpFloat(CfInputStream in) { super(TAG, in); }
    }

    static class CpU8Constant extends CpEntry {
        byte[] bytes;

        CpU8Constant(byte tag) { super(tag); }
        CpU8Constant(byte tag, CfInputStream in) {
            this(tag);
            bytes = in.array(8);
        }
        void write(CfOutputStream out) { super.write(out); out.array(bytes); }
    }
    static class CpLong extends CpU8Constant {
        static final byte TAG = 5;
        CpLong() { super(TAG); }
        CpLong(CfInputStream in) { super(TAG, in); }
    }
    static class CpDouble extends CpU8Constant {
        static final byte TAG = 6;
        CpDouble() { super(TAG); }
        CpDouble(CfInputStream in) { super(TAG, in); }
    }

    static class CpClass extends CpEntry {
        static final byte TAG = 7;
        short name_index;

        CpClass() { super(TAG); }
        CpClass(CfInputStream in) { super(TAG); name_index = in.u2(); }
        void write(CfOutputStream out) {
            super.write(out);
            out.u2(name_index);
        }
    }

    static class CpString extends CpEntry {
        static final byte TAG = 8;
        short string_index;

        CpString() { super(TAG); }
        CpString(CfInputStream in) { super(TAG); string_index = in.u2(); }
        void write(CfOutputStream out) {
            super.write(out);
            out.u2(string_index);
        }
    }

    static class CpRef extends CpEntry {
        short class_index;
        short name_and_type_index;

        CpRef(byte tag) { super(tag); }
        CpRef(byte tag, CfInputStream in) {
            this(tag);
            class_index = in.u2();
            name_and_type_index = in.u2();
        }
        void write(CfOutputStream out) {
            super.write(out);
            out.u2(class_index);
            out.u2(name_and_type_index);
        }
    }
    static class CpFieldRef extends CpRef {
        static final byte TAG = 9;
        CpFieldRef() { super(TAG); }
        CpFieldRef(CfInputStream in) { super(TAG, in); }
    }
    static class CpMethodRef extends CpRef {
        static final byte TAG = 10;
        CpMethodRef() { super(TAG); }
        CpMethodRef(CfInputStream in) { super(TAG, in); }
    }
    static class CpInterfaceMethodRef extends CpRef {
        static final byte TAG = 11;
        CpInterfaceMethodRef() { super(TAG); }
        CpInterfaceMethodRef(CfInputStream in) { super(TAG, in); }
    }

    static class CpNameAndType extends CpEntry {
        static final byte TAG = 12;
        short name_index;
        short descriptor_index;

        CpNameAndType() { super(TAG); }
        CpNameAndType(CfInputStream in) {
            this();
            name_index = in.u2();
            descriptor_index = in.u2();
        }
        void write(CfOutputStream out) {
            super.write(out);
            out.u2(name_index);
            out.u2(descriptor_index);
        }
    }

    static class CpMethodHandle extends CpEntry {
        static final byte TAG = 15;
        byte reference_kind;
        short reference_index;

        CpMethodHandle() { super(TAG); }
        CpMethodHandle(CfInputStream in) {
            this();
            reference_kind = in.u1();
            reference_index = in.u2();
        }
        void write(CfOutputStream out) {
            super.write(out);
            out.u1(reference_kind);
            out.u2(reference_index);
        }
    }

    static class CpMethodType extends CpEntry {
        static final byte TAG = 16;
        short descriptor_index;

        CpMethodType() { super(TAG); }
        CpMethodType(CfInputStream in) {
            this();
            descriptor_index = in.u2();
        }
        void write(CfOutputStream out) {
            super.write(out);
            out.u2(descriptor_index);
        }
    }

    static class CpInvokeDynamic extends CpEntry {
        static final byte TAG = 18;
        short bootstrap_index;
        short name_and_type_index;

        CpInvokeDynamic() { super(TAG); }
        CpInvokeDynamic(CfInputStream in) {
            this();
            bootstrap_index = in.u2();
            name_and_type_index = in.u2();
        }
        void write(CfOutputStream out) {
            super.write(out);
            out.u2(bootstrap_index);
            out.u2(name_and_type_index);
        }
    }

    static class Interface {
        short index;

        Interface() {}
        Interface(CfInputStream in) { index = in.u2(); }
        void write(CfOutputStream out) { out.u2(index); }
    }

    static class FieldOrMethod {
        short access_flags;
        short name_index;
        short descriptor_index;
        ArrayList<Attribute> attributes;

        FieldOrMethod() { attributes = new ArrayList<>(); }
        FieldOrMethod(CfInputStream in) {
            access_flags = in.u2();
            name_index = in.u2();
            descriptor_index = in.u2();

            short attrCount = in.u2();
            attributes = new ArrayList<>();
            for (int i = 0; i < attrCount; ++i) {
                attributes.add(new Attribute(in));
            }
        }
        void write(CfOutputStream out) {
            out.u2(access_flags);
            out.u2(name_index);
            out.u2(descriptor_index);
            out.u2((short)attributes.size());
            for (Attribute attribute : attributes) { attribute.write(out); }
        }
    }

    static class Field extends FieldOrMethod {
        Field() {}
        Field(CfInputStream in) { super(in); }
    }
    static class Method extends FieldOrMethod {
        Method() {}
        Method(CfInputStream in) { super(in); }
    }

    static class Attribute {
        short attribute_name_index;
        byte[] info;

        Attribute() { info = new byte[0]; }
        Attribute(CfInputStream in) {
            attribute_name_index = in.u2();
            int length = in.u4();
            info = in.array(length);
        }
        void write(CfOutputStream out) {
            out.u2(attribute_name_index);
            out.u4(info.length);
            out.array(info);
        }
    }
}