rfield@1422: /*
katleman@1448:  * Copyright (c) 2012, Oracle and/or its affiliates. All rights reserved.
rfield@1422:  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
rfield@1422:  *
rfield@1422:  * This code is free software; you can redistribute it and/or modify it
rfield@1422:  * under the terms of the GNU General Public License version 2 only, as
rfield@1422:  * published by the Free Software Foundation.  Oracle designates this
rfield@1422:  * particular file as subject to the "Classpath" exception as provided
rfield@1422:  * by Oracle in the LICENSE file that accompanied this code.
rfield@1422:  *
rfield@1422:  * This code is distributed in the hope that it will be useful, but WITHOUT
rfield@1422:  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
rfield@1422:  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
rfield@1422:  * version 2 for more details (a copy is included in the LICENSE file that
rfield@1422:  * accompanied this code).
rfield@1422:  *
rfield@1422:  * You should have received a copy of the GNU General Public License version
rfield@1422:  * 2 along with this work; if not, write to the Free Software Foundation,
rfield@1422:  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
rfield@1422:  *
rfield@1422:  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
rfield@1422:  * or visit www.oracle.com if you need additional information or have any
rfield@1422:  * questions.
rfield@1422:  */
rfield@1422: 
rfield@1422: package org.openjdk.tests.separate;
rfield@1422: 
rfield@1422: import java.io.*;
rfield@1422: import java.util.*;
rfield@1422: 
rfield@1422: class CfInputStream extends ByteArrayInputStream {
rfield@1422:     private int ct;
rfield@1422:     public CfInputStream(byte[] input) {
rfield@1422:         super(input);
rfield@1422:     }
rfield@1422: 
rfield@1422:     byte u1() { return (byte)read(); }
rfield@1422:     short u2() {
rfield@1422:         int b0 = read() << 8;
rfield@1422:         int b1 = read();
rfield@1422:         return (short)(b0 | b1);
rfield@1422:     }
rfield@1422:     int u4() {
rfield@1422:         int b0 = read() << 24;
rfield@1422:         int b1 = read() << 16;
rfield@1422:         int b2 = read() << 8;
rfield@1422:         int b3 = read();
rfield@1422:         return b0 | b1 | b2 | b3;
rfield@1422:     }
rfield@1422:     byte[] array(int count) {
rfield@1422:         byte[] ret = new byte[count];
rfield@1422:         read(ret, 0, count);
rfield@1422:         return ret;
rfield@1422:     }
rfield@1422: };
rfield@1422: 
rfield@1422: class CfOutputStream extends ByteArrayOutputStream {
rfield@1422:     void u1(byte b) { write((int)b); }
rfield@1422:     void u2(short s) {
rfield@1422:         write((s >> 8) & 0xff);
rfield@1422:         write(s & 0xff);
rfield@1422:     }
rfield@1422:     void u4(int i) {
rfield@1422:         write((i >> 24) & 0xff);
rfield@1422:         write((i >> 16) & 0xff);
rfield@1422:         write((i >> 8) & 0xff);
rfield@1422:         write(i & 0xff);
rfield@1422:     }
rfield@1422:     void array(byte[] a) {
rfield@1422:         write(a, 0, a.length);
rfield@1422:     }
rfield@1422: 
rfield@1422:     public byte[] toByteArray() { return super.toByteArray(); }
rfield@1422: };
rfield@1422: 
rfield@1422: // A quick and dirty class file parser and representation
rfield@1422: public class ClassFile {
rfield@1422: 
rfield@1422:     int magic;
rfield@1422:     short minor_version;
rfield@1422:     short major_version;
rfield@1422:     ArrayList<CpEntry> constant_pool;
rfield@1422:     short access_flags;
rfield@1422:     short this_class;
rfield@1422:     short super_class;
rfield@1422:     ArrayList<Interface> interfaces;
rfield@1422:     ArrayList<Field> fields;
rfield@1422:     ArrayList<Method> methods;
rfield@1422:     ArrayList<Attribute> attributes;
rfield@1422: 
rfield@1422:     ClassFile(byte[] cf) {
rfield@1422:         CfInputStream in = new CfInputStream(cf);
rfield@1422: 
rfield@1422:         magic = in.u4();
rfield@1422:         minor_version = in.u2();
rfield@1422:         major_version = in.u2();
rfield@1422: 
rfield@1422:         short cpCount = in.u2();
rfield@1422:         constant_pool = new ArrayList<>();
rfield@1422:         constant_pool.add(new CpNull());
rfield@1422:         for (int i = 1; i < cpCount; ++i) {
rfield@1422:             constant_pool.add(CpEntry.newCpEntry(in));
rfield@1422:         }
rfield@1422: 
rfield@1422:         access_flags = in.u2();
rfield@1422:         this_class = in.u2();
rfield@1422:         super_class = in.u2();
rfield@1422: 
rfield@1422:         short ifaceCount = in.u2();
rfield@1422:         interfaces = new ArrayList<>();
rfield@1422:         for (int i = 0; i < ifaceCount; ++i) {
rfield@1422:             interfaces.add(new Interface(in));
rfield@1422:         }
rfield@1422: 
rfield@1422:         short fieldCount = in.u2();
rfield@1422:         fields = new ArrayList<>();
rfield@1422:         for (int i = 0; i < fieldCount; ++i) {
rfield@1422:             fields.add(new Field(in));
rfield@1422:         }
rfield@1422: 
rfield@1422:         short methodCount = in.u2();
rfield@1422:         methods = new ArrayList<>();
rfield@1422:         for (int i = 0; i < methodCount; ++i) {
rfield@1422:             methods.add(new Method(in));
rfield@1422:         }
rfield@1422: 
rfield@1422:         short attributeCount = in.u2();
rfield@1422:         attributes = new ArrayList<>();
rfield@1422:         for (int i = 0; i < attributeCount; ++i) {
rfield@1422:             attributes.add(new Attribute(in));
rfield@1422:         }
rfield@1422:     }
rfield@1422: 
rfield@1422:     byte[] toByteArray() {
rfield@1422:         CfOutputStream out = new CfOutputStream();
rfield@1422: 
rfield@1422:         out.u4(magic);
rfield@1422:         out.u2(minor_version);
rfield@1422:         out.u2(major_version);
rfield@1422: 
rfield@1422:         out.u2((short)(constant_pool.size()));
rfield@1422:         for (CpEntry cp : constant_pool) {
rfield@1422:             cp.write(out);
rfield@1422:         }
rfield@1422: 
rfield@1422:         out.u2(access_flags);
rfield@1422:         out.u2(this_class);
rfield@1422:         out.u2(super_class);
rfield@1422: 
rfield@1422:         out.u2((short)interfaces.size());
rfield@1422:         for (Interface iface : interfaces) {
rfield@1422:             iface.write(out);
rfield@1422:         }
rfield@1422: 
rfield@1422:         out.u2((short)fields.size());
rfield@1422:         for (Field field : fields) {
rfield@1422:             field.write(out);
rfield@1422:         }
rfield@1422: 
rfield@1422:         out.u2((short)methods.size());
rfield@1422:         for (Method method : methods) {
rfield@1422:             method.write(out);
rfield@1422:         }
rfield@1422: 
rfield@1422:         out.u2((short)attributes.size());
rfield@1422:         for (Attribute attribute : attributes) {
rfield@1422:             attribute.write(out);
rfield@1422:         }
rfield@1422: 
rfield@1422:         return out.toByteArray();
rfield@1422:     }
rfield@1422: 
rfield@1422:     static abstract class CpEntry {
rfield@1422:         byte tag;
rfield@1422: 
rfield@1422:         CpEntry(byte t) { tag = t; }
rfield@1422:         void write(CfOutputStream out) {
rfield@1422:             out.u1(tag);
rfield@1422:         }
rfield@1422: 
rfield@1422:         static CpEntry newCpEntry(CfInputStream in) {
rfield@1422:             byte tag = in.u1();
rfield@1422:             switch (tag) {
rfield@1422:                 case CpUtf8.TAG: return new CpUtf8(in);
rfield@1422:                 case CpInteger.TAG: return new CpInteger(in);
rfield@1422:                 case CpFloat.TAG: return new CpFloat(in);
rfield@1422:                 case CpLong.TAG: return new CpLong(in);
rfield@1422:                 case CpDouble.TAG: return new CpDouble(in);
rfield@1422:                 case CpClass.TAG: return new CpClass(in);
rfield@1422:                 case CpString.TAG: return new CpString(in);
rfield@1422:                 case CpFieldRef.TAG: return new CpFieldRef(in);
rfield@1422:                 case CpMethodRef.TAG: return new CpMethodRef(in);
rfield@1422:                 case CpInterfaceMethodRef.TAG:
rfield@1422:                     return new CpInterfaceMethodRef(in);
rfield@1422:                 case CpNameAndType.TAG: return new CpNameAndType(in);
rfield@1422:                 case CpMethodHandle.TAG: return new CpMethodHandle(in);
rfield@1422:                 case CpMethodType.TAG: return new CpMethodType(in);
rfield@1422:                 case CpInvokeDynamic.TAG: return new CpInvokeDynamic(in);
rfield@1422:                 default: throw new RuntimeException("Bad cp entry tag: " + tag);
rfield@1422:             }
rfield@1422:         }
rfield@1422:     }
rfield@1422: 
rfield@1422:     static class CpNull extends CpEntry {
rfield@1422:         CpNull() { super((byte)0); }
rfield@1422:         CpNull(CfInputStream in) { super((byte)0); }
rfield@1422:         void write(CfOutputStream out) {}
rfield@1422:     }
rfield@1422: 
rfield@1422:     static class CpUtf8 extends CpEntry {
rfield@1422:         static final byte TAG = 1;
rfield@1422:         byte[] bytes;
rfield@1422: 
rfield@1422:         CpUtf8() { super(TAG); }
rfield@1422:         CpUtf8(CfInputStream in) {
rfield@1422:             this();
rfield@1422:             short length = in.u2();
rfield@1422:             bytes = in.array(length);
rfield@1422:         }
rfield@1422:         void write(CfOutputStream out) {
rfield@1422:             super.write(out);
rfield@1422:             out.u2((short)bytes.length);
rfield@1422:             out.array(bytes);
rfield@1422:         }
rfield@1422:     }
rfield@1422: 
rfield@1422:     static class CpU4Constant extends CpEntry {
rfield@1422:         byte[] bytes;
rfield@1422: 
rfield@1422:         CpU4Constant(byte tag) { super(tag); }
rfield@1422:         CpU4Constant(byte tag, CfInputStream in) {
rfield@1422:             this(tag);
rfield@1422:             bytes = in.array(4);
rfield@1422:         }
rfield@1422:         void write(CfOutputStream out) { super.write(out); out.array(bytes); }
rfield@1422:     }
rfield@1422:     static class CpInteger extends CpU4Constant {
rfield@1422:         static final byte TAG = 3;
rfield@1422:         CpInteger() { super(TAG); }
rfield@1422:         CpInteger(CfInputStream in) { super(TAG, in); }
rfield@1422:     }
rfield@1422:     static class CpFloat extends CpU4Constant {
rfield@1422:         static final byte TAG = 4;
rfield@1422:         CpFloat() { super(TAG); }
rfield@1422:         CpFloat(CfInputStream in) { super(TAG, in); }
rfield@1422:     }
rfield@1422: 
rfield@1422:     static class CpU8Constant extends CpEntry {
rfield@1422:         byte[] bytes;
rfield@1422: 
rfield@1422:         CpU8Constant(byte tag) { super(tag); }
rfield@1422:         CpU8Constant(byte tag, CfInputStream in) {
rfield@1422:             this(tag);
rfield@1422:             bytes = in.array(8);
rfield@1422:         }
rfield@1422:         void write(CfOutputStream out) { super.write(out); out.array(bytes); }
rfield@1422:     }
rfield@1422:     static class CpLong extends CpU8Constant {
rfield@1422:         static final byte TAG = 5;
rfield@1422:         CpLong() { super(TAG); }
rfield@1422:         CpLong(CfInputStream in) { super(TAG, in); }
rfield@1422:     }
rfield@1422:     static class CpDouble extends CpU8Constant {
rfield@1422:         static final byte TAG = 6;
rfield@1422:         CpDouble() { super(TAG); }
rfield@1422:         CpDouble(CfInputStream in) { super(TAG, in); }
rfield@1422:     }
rfield@1422: 
rfield@1422:     static class CpClass extends CpEntry {
rfield@1422:         static final byte TAG = 7;
rfield@1422:         short name_index;
rfield@1422: 
rfield@1422:         CpClass() { super(TAG); }
rfield@1422:         CpClass(CfInputStream in) { super(TAG); name_index = in.u2(); }
rfield@1422:         void write(CfOutputStream out) {
rfield@1422:             super.write(out);
rfield@1422:             out.u2(name_index);
rfield@1422:         }
rfield@1422:     }
rfield@1422: 
rfield@1422:     static class CpString extends CpEntry {
rfield@1422:         static final byte TAG = 8;
rfield@1422:         short string_index;
rfield@1422: 
rfield@1422:         CpString() { super(TAG); }
rfield@1422:         CpString(CfInputStream in) { super(TAG); string_index = in.u2(); }
rfield@1422:         void write(CfOutputStream out) {
rfield@1422:             super.write(out);
rfield@1422:             out.u2(string_index);
rfield@1422:         }
rfield@1422:     }
rfield@1422: 
rfield@1422:     static class CpRef extends CpEntry {
rfield@1422:         short class_index;
rfield@1422:         short name_and_type_index;
rfield@1422: 
rfield@1422:         CpRef(byte tag) { super(tag); }
rfield@1422:         CpRef(byte tag, CfInputStream in) {
rfield@1422:             this(tag);
rfield@1422:             class_index = in.u2();
rfield@1422:             name_and_type_index = in.u2();
rfield@1422:         }
rfield@1422:         void write(CfOutputStream out) {
rfield@1422:             super.write(out);
rfield@1422:             out.u2(class_index);
rfield@1422:             out.u2(name_and_type_index);
rfield@1422:         }
rfield@1422:     }
rfield@1422:     static class CpFieldRef extends CpRef {
rfield@1422:         static final byte TAG = 9;
rfield@1422:         CpFieldRef() { super(TAG); }
rfield@1422:         CpFieldRef(CfInputStream in) { super(TAG, in); }
rfield@1422:     }
rfield@1422:     static class CpMethodRef extends CpRef {
rfield@1422:         static final byte TAG = 10;
rfield@1422:         CpMethodRef() { super(TAG); }
rfield@1422:         CpMethodRef(CfInputStream in) { super(TAG, in); }
rfield@1422:     }
rfield@1422:     static class CpInterfaceMethodRef extends CpRef {
rfield@1422:         static final byte TAG = 11;
rfield@1422:         CpInterfaceMethodRef() { super(TAG); }
rfield@1422:         CpInterfaceMethodRef(CfInputStream in) { super(TAG, in); }
rfield@1422:     }
rfield@1422: 
rfield@1422:     static class CpNameAndType extends CpEntry {
rfield@1422:         static final byte TAG = 12;
rfield@1422:         short name_index;
rfield@1422:         short descriptor_index;
rfield@1422: 
rfield@1422:         CpNameAndType() { super(TAG); }
rfield@1422:         CpNameAndType(CfInputStream in) {
rfield@1422:             this();
rfield@1422:             name_index = in.u2();
rfield@1422:             descriptor_index = in.u2();
rfield@1422:         }
rfield@1422:         void write(CfOutputStream out) {
rfield@1422:             super.write(out);
rfield@1422:             out.u2(name_index);
rfield@1422:             out.u2(descriptor_index);
rfield@1422:         }
rfield@1422:     }
rfield@1422: 
rfield@1422:     static class CpMethodHandle extends CpEntry {
rfield@1422:         static final byte TAG = 15;
rfield@1422:         byte reference_kind;
rfield@1422:         short reference_index;
rfield@1422: 
rfield@1422:         CpMethodHandle() { super(TAG); }
rfield@1422:         CpMethodHandle(CfInputStream in) {
rfield@1422:             this();
rfield@1422:             reference_kind = in.u1();
rfield@1422:             reference_index = in.u2();
rfield@1422:         }
rfield@1422:         void write(CfOutputStream out) {
rfield@1422:             super.write(out);
rfield@1422:             out.u1(reference_kind);
rfield@1422:             out.u2(reference_index);
rfield@1422:         }
rfield@1422:     }
rfield@1422: 
rfield@1422:     static class CpMethodType extends CpEntry {
rfield@1422:         static final byte TAG = 16;
rfield@1422:         short descriptor_index;
rfield@1422: 
rfield@1422:         CpMethodType() { super(TAG); }
rfield@1422:         CpMethodType(CfInputStream in) {
rfield@1422:             this();
rfield@1422:             descriptor_index = in.u2();
rfield@1422:         }
rfield@1422:         void write(CfOutputStream out) {
rfield@1422:             super.write(out);
rfield@1422:             out.u2(descriptor_index);
rfield@1422:         }
rfield@1422:     }
rfield@1422: 
rfield@1422:     static class CpInvokeDynamic extends CpEntry {
rfield@1422:         static final byte TAG = 18;
rfield@1422:         short bootstrap_index;
rfield@1422:         short name_and_type_index;
rfield@1422: 
rfield@1422:         CpInvokeDynamic() { super(TAG); }
rfield@1422:         CpInvokeDynamic(CfInputStream in) {
rfield@1422:             this();
rfield@1422:             bootstrap_index = in.u2();
rfield@1422:             name_and_type_index = in.u2();
rfield@1422:         }
rfield@1422:         void write(CfOutputStream out) {
rfield@1422:             super.write(out);
rfield@1422:             out.u2(bootstrap_index);
rfield@1422:             out.u2(name_and_type_index);
rfield@1422:         }
rfield@1422:     }
rfield@1422: 
rfield@1422:     static class Interface {
rfield@1422:         short index;
rfield@1422: 
rfield@1422:         Interface() {}
rfield@1422:         Interface(CfInputStream in) { index = in.u2(); }
rfield@1422:         void write(CfOutputStream out) { out.u2(index); }
rfield@1422:     }
rfield@1422: 
rfield@1422:     static class FieldOrMethod {
rfield@1422:         short access_flags;
rfield@1422:         short name_index;
rfield@1422:         short descriptor_index;
rfield@1422:         ArrayList<Attribute> attributes;
rfield@1422: 
rfield@1422:         FieldOrMethod() { attributes = new ArrayList<>(); }
rfield@1422:         FieldOrMethod(CfInputStream in) {
rfield@1422:             access_flags = in.u2();
rfield@1422:             name_index = in.u2();
rfield@1422:             descriptor_index = in.u2();
rfield@1422: 
rfield@1422:             short attrCount = in.u2();
rfield@1422:             attributes = new ArrayList<>();
rfield@1422:             for (int i = 0; i < attrCount; ++i) {
rfield@1422:                 attributes.add(new Attribute(in));
rfield@1422:             }
rfield@1422:         }
rfield@1422:         void write(CfOutputStream out) {
rfield@1422:             out.u2(access_flags);
rfield@1422:             out.u2(name_index);
rfield@1422:             out.u2(descriptor_index);
rfield@1422:             out.u2((short)attributes.size());
rfield@1422:             for (Attribute attribute : attributes) { attribute.write(out); }
rfield@1422:         }
rfield@1422:     }
rfield@1422: 
rfield@1422:     static class Field extends FieldOrMethod {
rfield@1422:         Field() {}
rfield@1422:         Field(CfInputStream in) { super(in); }
rfield@1422:     }
rfield@1422:     static class Method extends FieldOrMethod {
rfield@1422:         Method() {}
rfield@1422:         Method(CfInputStream in) { super(in); }
rfield@1422:     }
rfield@1422: 
rfield@1422:     static class Attribute {
rfield@1422:         short attribute_name_index;
rfield@1422:         byte[] info;
rfield@1422: 
rfield@1422:         Attribute() { info = new byte[0]; }
rfield@1422:         Attribute(CfInputStream in) {
rfield@1422:             attribute_name_index = in.u2();
rfield@1422:             int length = in.u4();
rfield@1422:             info = in.array(length);
rfield@1422:         }
rfield@1422:         void write(CfOutputStream out) {
rfield@1422:             out.u2(attribute_name_index);
rfield@1422:             out.u4(info.length);
rfield@1422:             out.array(info);
rfield@1422:         }
rfield@1422:     }
rfield@1422: }