com.github.rapidark.framework.thirdparty.asm.ClassWriter Maven / Gradle / Ivy
The newest version!
package com.github.rapidark.framework.thirdparty.asm;
public class ClassWriter extends ClassVisitor {
public static final int COMPUTE_MAXS = 1;
public static final int COMPUTE_FRAMES = 2;
static final int TYPE_UNINIT = 31;
Item[] typeTable;
FieldWriter firstField;
FieldWriter lastField;
MethodWriter firstMethod;
MethodWriter lastMethod;
boolean invalidFrames;
static final int ACC_SYNTHETIC_ATTRIBUTE = 262144;
static final int TO_ACC_SYNTHETIC = 64;
static final int NOARG_INSN = 0;
static final int SBYTE_INSN = 1;
static final int SHORT_INSN = 2;
static final int VAR_INSN = 3;
static final int IMPLVAR_INSN = 4;
static final int TYPE_INSN = 5;
static final int FIELDORMETH_INSN = 6;
static final int ITFMETH_INSN = 7;
static final int INDYMETH_INSN = 8;
static final int LABEL_INSN = 9;
static final int LABELW_INSN = 10;
static final int LDC_INSN = 11;
static final int LDCW_INSN = 12;
static final int IINC_INSN = 13;
static final int TABL_INSN = 14;
static final int LOOK_INSN = 15;
static final int MANA_INSN = 16;
static final int WIDE_INSN = 17;
static final byte[] TYPE;
static final int CLASS = 7;
static final int FIELD = 9;
static final int METH = 10;
static final int IMETH = 11;
static final int STR = 8;
static final int INT = 3;
static final int FLOAT = 4;
static final int LONG = 5;
static final int DOUBLE = 6;
static final int NAME_TYPE = 12;
static final int UTF8 = 1;
static final int MTYPE = 16;
static final int HANDLE = 15;
static final int INDY = 18;
static final int HANDLE_BASE = 20;
static final int TYPE_NORMAL = 30;
static final int TYPE_MERGED = 32;
static final int BSM = 33;
ClassReader cr;
int version;
int index;
final ByteVector pool;
Item[] items;
int threshold;
final Item key;
final Item key2;
final Item key3;
final Item key4;
private short typeCount;
private int access;
private int name;
String thisName;
private int signature;
private int superName;
private int interfaceCount;
private int[] interfaces;
private int sourceFile;
private ByteVector sourceDebug;
private int enclosingMethodOwner;
private int enclosingMethod;
private AnnotationWriter anns;
private AnnotationWriter ianns;
private Attribute attrs;
private int innerClassesCount;
private ByteVector innerClasses;
int bootstrapMethodsCount;
ByteVector bootstrapMethods;
private final boolean computeMaxs;
private final boolean computeFrames;
static {
byte[] b = new byte['Ü'];
String s = "AAAAAAAAAAAAAAAABCLMMDDDDDEEEEEEEEEEEEEEEEEEEEAAAAAAAADDDDDEEEEEEEEEEEEEEEEEEEEAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAANAAAAAAAAAAAAAAAAAAAAJJJJJJJJJJJJJJJJDOPAAAAAAGGGGGGGHIFBFAAFFAARQJJKKJJJJJJJJJJJJJJJJJJ";
for (int i = 0; i < b.length; i++) {
b[i] = ((byte) (s.charAt(i) - 'A'));
}
TYPE = b;
}
public ClassWriter(int flags) {
super(262144);
this.index = 1;
this.pool = new ByteVector();
this.items = new Item['Ā'];
this.threshold = ((int) (0.75D * this.items.length));
this.key = new Item();
this.key2 = new Item();
this.key3 = new Item();
this.key4 = new Item();
this.computeMaxs = ((flags & 0x1) != 0);
this.computeFrames = ((flags & 0x2) != 0);
}
public ClassWriter(ClassReader classReader, int flags) {
this(flags);
classReader.copyPool(this);
this.cr = classReader;
}
public final void visit(int version, int access, String name, String signature, String superName, String[] interfaces) {
this.version = version;
this.access = access;
this.name = newClass(name);
this.thisName = name;
if (signature != null) {
this.signature = newUTF8(signature);
}
this.superName = (superName == null ? 0 : newClass(superName));
if ((interfaces != null) && (interfaces.length > 0)) {
this.interfaceCount = interfaces.length;
this.interfaces = new int[this.interfaceCount];
for (int i = 0; i < this.interfaceCount; i++) {
this.interfaces[i] = newClass(interfaces[i]);
}
}
}
public final void visitSource(String file, String debug) {
if (file != null) {
this.sourceFile = newUTF8(file);
}
if (debug != null) {
this.sourceDebug = new ByteVector().putUTF8(debug);
}
}
public final void visitOuterClass(String owner, String name, String desc) {
this.enclosingMethodOwner = newClass(owner);
if ((name != null) && (desc != null)) {
this.enclosingMethod = newNameType(name, desc);
}
}
public final AnnotationVisitor visitAnnotation(String desc, boolean visible) {
ByteVector bv = new ByteVector();
bv.putShort(newUTF8(desc)).putShort(0);
AnnotationWriter aw = new AnnotationWriter(this, true, bv, bv, 2);
if (visible) {
aw.next = this.anns;
this.anns = aw;
} else {
aw.next = this.ianns;
this.ianns = aw;
}
return aw;
}
public final void visitAttribute(Attribute attr) {
attr.next = this.attrs;
this.attrs = attr;
}
public final void visitInnerClass(String name, String outerName, String innerName, int access) {
if (this.innerClasses == null) {
this.innerClasses = new ByteVector();
}
this.innerClassesCount += 1;
this.innerClasses.putShort(name == null ? 0 : newClass(name));
this.innerClasses.putShort(outerName == null ? 0 : newClass(outerName));
this.innerClasses.putShort(innerName == null ? 0 : newUTF8(innerName));
this.innerClasses.putShort(access);
}
public final FieldVisitor visitField(int access, String name, String desc, String signature, Object value) {
return new FieldWriter(this, access, name, desc, signature, value);
}
public final MethodVisitor visitMethod(int access, String name, String desc, String signature, String[] exceptions) {
return new MethodWriter(this, access, name, desc, signature, exceptions, this.computeMaxs, this.computeFrames);
}
public final void visitEnd() {
}
public byte[] toByteArray() {
if (this.index > 65535) {
throw new RuntimeException("Class file too large!");
}
int size = 24 + 2 * this.interfaceCount;
int nbFields = 0;
FieldWriter fb = this.firstField;
while (fb != null) {
nbFields++;
size += fb.getSize();
fb = (FieldWriter) fb.fv;
}
int nbMethods = 0;
MethodWriter mb = this.firstMethod;
while (mb != null) {
nbMethods++;
size += mb.getSize();
mb = (MethodWriter) mb.mv;
}
int attributeCount = 0;
if (this.bootstrapMethods != null) {
attributeCount++;
size += 8 + this.bootstrapMethods.length;
newUTF8("BootstrapMethods");
}
if (this.signature != 0) {
attributeCount++;
size += 8;
newUTF8("Signature");
}
if (this.sourceFile != 0) {
attributeCount++;
size += 8;
newUTF8("SourceFile");
}
if (this.sourceDebug != null) {
attributeCount++;
size += this.sourceDebug.length + 4;
newUTF8("SourceDebugExtension");
}
if (this.enclosingMethodOwner != 0) {
attributeCount++;
size += 10;
newUTF8("EnclosingMethod");
}
if ((this.access & 0x20000) != 0) {
attributeCount++;
size += 6;
newUTF8("Deprecated");
}
if (((this.access & 0x1000) != 0) && (((this.version & 0xFFFF) < 49) || ((this.access & 0x40000) != 0))) {
attributeCount++;
size += 6;
newUTF8("Synthetic");
}
if (this.innerClasses != null) {
attributeCount++;
size += 8 + this.innerClasses.length;
newUTF8("InnerClasses");
}
if (this.anns != null) {
attributeCount++;
size += 8 + this.anns.getSize();
newUTF8("RuntimeVisibleAnnotations");
}
if (this.ianns != null) {
attributeCount++;
size += 8 + this.ianns.getSize();
newUTF8("RuntimeInvisibleAnnotations");
}
if (this.attrs != null) {
attributeCount += this.attrs.getCount();
size += this.attrs.getSize(this, null, 0, -1, -1);
}
size += this.pool.length;
ByteVector out = new ByteVector(size);
out.putInt(-889275714).putInt(this.version);
out.putShort(this.index).putByteArray(this.pool.data, 0, this.pool.length);
int mask = 0x60000 | (this.access & 0x40000) / 64;
out.putShort(this.access & (mask ^ 0xFFFFFFFF)).putShort(this.name).putShort(this.superName);
out.putShort(this.interfaceCount);
for (int i = 0; i < this.interfaceCount; i++) {
out.putShort(this.interfaces[i]);
}
out.putShort(nbFields);
fb = this.firstField;
while (fb != null) {
fb.put(out);
fb = (FieldWriter) fb.fv;
}
out.putShort(nbMethods);
mb = this.firstMethod;
while (mb != null) {
mb.put(out);
mb = (MethodWriter) mb.mv;
}
out.putShort(attributeCount);
if (this.bootstrapMethods != null) {
out.putShort(newUTF8("BootstrapMethods"));
out.putInt(this.bootstrapMethods.length + 2).putShort(this.bootstrapMethodsCount);
out.putByteArray(this.bootstrapMethods.data, 0, this.bootstrapMethods.length);
}
if (this.signature != 0) {
out.putShort(newUTF8("Signature")).putInt(2).putShort(this.signature);
}
if (this.sourceFile != 0) {
out.putShort(newUTF8("SourceFile")).putInt(2).putShort(this.sourceFile);
}
if (this.sourceDebug != null) {
int len = this.sourceDebug.length - 2;
out.putShort(newUTF8("SourceDebugExtension")).putInt(len);
out.putByteArray(this.sourceDebug.data, 2, len);
}
if (this.enclosingMethodOwner != 0) {
out.putShort(newUTF8("EnclosingMethod")).putInt(4);
out.putShort(this.enclosingMethodOwner).putShort(this.enclosingMethod);
}
if ((this.access & 0x20000) != 0) {
out.putShort(newUTF8("Deprecated")).putInt(0);
}
if (((this.access & 0x1000) != 0) && (((this.version & 0xFFFF) < 49) || ((this.access & 0x40000) != 0))) {
out.putShort(newUTF8("Synthetic")).putInt(0);
}
if (this.innerClasses != null) {
out.putShort(newUTF8("InnerClasses"));
out.putInt(this.innerClasses.length + 2).putShort(this.innerClassesCount);
out.putByteArray(this.innerClasses.data, 0, this.innerClasses.length);
}
if (this.anns != null) {
out.putShort(newUTF8("RuntimeVisibleAnnotations"));
this.anns.put(out);
}
if (this.ianns != null) {
out.putShort(newUTF8("RuntimeInvisibleAnnotations"));
this.ianns.put(out);
}
if (this.attrs != null) {
this.attrs.put(this, null, 0, -1, -1, out);
}
if (this.invalidFrames) {
ClassWriter cw = new ClassWriter(2);
new ClassReader(out.data).accept(cw, 4);
return cw.toByteArray();
}
return out.data;
}
Item newConstItem(Object cst) {
if ((cst instanceof Integer)) {
int val = ((Integer) cst).intValue();
return newInteger(val);
}
if ((cst instanceof Byte)) {
int val = ((Byte) cst).intValue();
return newInteger(val);
}
if ((cst instanceof Character)) {
int val = ((Character) cst).charValue();
return newInteger(val);
}
if ((cst instanceof Short)) {
int val = ((Short) cst).intValue();
return newInteger(val);
}
if ((cst instanceof Boolean)) {
int val = ((Boolean) cst).booleanValue() ? 1 : 0;
return newInteger(val);
}
if ((cst instanceof Float)) {
float val = ((Float) cst).floatValue();
return newFloat(val);
}
if ((cst instanceof Long)) {
long val = ((Long) cst).longValue();
return newLong(val);
}
if ((cst instanceof Double)) {
double val = ((Double) cst).doubleValue();
return newDouble(val);
}
if ((cst instanceof String)) {
return newString((String) cst);
}
if ((cst instanceof Type)) {
Type t = (Type) cst;
int s = t.getSort();
if (s == 10) {
return newClassItem(t.getInternalName());
}
if (s == 11) {
return newMethodTypeItem(t.getDescriptor());
}
return newClassItem(t.getDescriptor());
}
if ((cst instanceof Handle)) {
Handle h = (Handle) cst;
return newHandleItem(h.tag, h.owner, h.name, h.desc);
}
throw new IllegalArgumentException("value " + cst);
}
public int newConst(Object cst) {
return newConstItem(cst).index;
}
public int newUTF8(String value) {
this.key.set(1, value, null, null);
Item result = get(this.key);
if (result == null) {
this.pool.putByte(1).putUTF8(value);
result = new Item(this.index++, this.key);
put(result);
}
return result.index;
}
Item newClassItem(String value) {
this.key2.set(7, value, null, null);
Item result = get(this.key2);
if (result == null) {
this.pool.put12(7, newUTF8(value));
result = new Item(this.index++, this.key2);
put(result);
}
return result;
}
public int newClass(String value) {
return newClassItem(value).index;
}
Item newMethodTypeItem(String methodDesc) {
this.key2.set(16, methodDesc, null, null);
Item result = get(this.key2);
if (result == null) {
this.pool.put12(16, newUTF8(methodDesc));
result = new Item(this.index++, this.key2);
put(result);
}
return result;
}
public int newMethodType(String methodDesc) {
return newMethodTypeItem(methodDesc).index;
}
Item newHandleItem(int tag, String owner, String name, String desc) {
this.key4.set(20 + tag, owner, name, desc);
Item result = get(this.key4);
if (result == null) {
if (tag <= 4) {
put112(15, tag, newField(owner, name, desc));
} else {
put112(15, tag, newMethod(owner, name, desc, tag == 9));
}
result = new Item(this.index++, this.key4);
put(result);
}
return result;
}
public int newHandle(int tag, String owner, String name, String desc) {
return newHandleItem(tag, owner, name, desc).index;
}
Item newInvokeDynamicItem(String name, String desc, Handle bsm, Object... bsmArgs) {
ByteVector bootstrapMethods = this.bootstrapMethods;
if (bootstrapMethods == null) {
bootstrapMethods = this.bootstrapMethods = new ByteVector();
}
int position = bootstrapMethods.length;
int hashCode = bsm.hashCode();
bootstrapMethods.putShort(newHandle(bsm.tag, bsm.owner, bsm.name, bsm.desc));
int argsLength = bsmArgs.length;
bootstrapMethods.putShort(argsLength);
for (int i = 0; i < argsLength; i++) {
Object bsmArg = bsmArgs[i];
hashCode ^= bsmArg.hashCode();
bootstrapMethods.putShort(newConst(bsmArg));
}
byte[] data = bootstrapMethods.data;
int length = 2 + argsLength << 1;
hashCode &= 0x7FFFFFFF;
Item result = this.items[(hashCode % this.items.length)];
while (result != null) {
if ((result.type != 33) || (result.hashCode != hashCode)) {
result = result.next;
} else {
int resultPosition = result.intVal;
for (int p = 0; p < length; p++) {
if (data[(position + p)] != data[(resultPosition + p)]) {
result = result.next;
break;
}
}
break;
}
}
int bootstrapMethodIndex;
if (result != null) {
bootstrapMethodIndex = result.index;
bootstrapMethods.length = position;
} else {
bootstrapMethodIndex = this.bootstrapMethodsCount++;
result = new Item(bootstrapMethodIndex);
result.set(position, hashCode);
put(result);
}
this.key3.set(name, desc, bootstrapMethodIndex);
result = get(this.key3);
if (result == null) {
put122(18, bootstrapMethodIndex, newNameType(name, desc));
result = new Item(this.index++, this.key3);
put(result);
}
return result;
}
public int newInvokeDynamic(String name, String desc, Handle bsm, Object... bsmArgs) {
return newInvokeDynamicItem(name, desc, bsm, bsmArgs).index;
}
Item newFieldItem(String owner, String name, String desc) {
this.key3.set(9, owner, name, desc);
Item result = get(this.key3);
if (result == null) {
put122(9, newClass(owner), newNameType(name, desc));
result = new Item(this.index++, this.key3);
put(result);
}
return result;
}
public int newField(String owner, String name, String desc) {
return newFieldItem(owner, name, desc).index;
}
Item newMethodItem(String owner, String name, String desc, boolean itf) {
int type = itf ? 11 : 10;
this.key3.set(type, owner, name, desc);
Item result = get(this.key3);
if (result == null) {
put122(type, newClass(owner), newNameType(name, desc));
result = new Item(this.index++, this.key3);
put(result);
}
return result;
}
public int newMethod(String owner, String name, String desc, boolean itf) {
return newMethodItem(owner, name, desc, itf).index;
}
Item newInteger(int value) {
this.key.set(value);
Item result = get(this.key);
if (result == null) {
this.pool.putByte(3).putInt(value);
result = new Item(this.index++, this.key);
put(result);
}
return result;
}
Item newFloat(float value) {
this.key.set(value);
Item result = get(this.key);
if (result == null) {
this.pool.putByte(4).putInt(this.key.intVal);
result = new Item(this.index++, this.key);
put(result);
}
return result;
}
Item newLong(long value) {
this.key.set(value);
Item result = get(this.key);
if (result == null) {
this.pool.putByte(5).putLong(value);
result = new Item(this.index, this.key);
this.index += 2;
put(result);
}
return result;
}
Item newDouble(double value) {
this.key.set(value);
Item result = get(this.key);
if (result == null) {
this.pool.putByte(6).putLong(this.key.longVal);
result = new Item(this.index, this.key);
this.index += 2;
put(result);
}
return result;
}
private Item newString(String value) {
this.key2.set(8, value, null, null);
Item result = get(this.key2);
if (result == null) {
this.pool.put12(8, newUTF8(value));
result = new Item(this.index++, this.key2);
put(result);
}
return result;
}
public int newNameType(String name, String desc) {
return newNameTypeItem(name, desc).index;
}
Item newNameTypeItem(String name, String desc) {
this.key2.set(12, name, desc, null);
Item result = get(this.key2);
if (result == null) {
put122(12, newUTF8(name), newUTF8(desc));
result = new Item(this.index++, this.key2);
put(result);
}
return result;
}
int addType(String type) {
this.key.set(30, type, null, null);
Item result = get(this.key);
if (result == null) {
result = addType(this.key);
}
return result.index;
}
int addUninitializedType(String type, int offset) {
this.key.type = 31;
this.key.intVal = offset;
this.key.strVal1 = type;
this.key.hashCode = (0x7FFFFFFF & 31 + type.hashCode() + offset);
Item result = get(this.key);
if (result == null) {
result = addType(this.key);
}
return result.index;
}
private Item addType(Item item) {
this.typeCount = ((short) (this.typeCount + 1));
Item result = new Item(this.typeCount, this.key);
put(result);
if (this.typeTable == null) {
this.typeTable = new Item[16];
}
if (this.typeCount == this.typeTable.length) {
Item[] newTable = new Item[2 * this.typeTable.length];
System.arraycopy(this.typeTable, 0, newTable, 0, this.typeTable.length);
this.typeTable = newTable;
}
this.typeTable[this.typeCount] = result;
return result;
}
int getMergedType(int type1, int type2) {
this.key2.type = 32;
this.key2.longVal = (type1 | type2 << 32);
this.key2.hashCode = (0x7FFFFFFF & 32 + type1 + type2);
Item result = get(this.key2);
if (result == null) {
String t = this.typeTable[type1].strVal1;
String u = this.typeTable[type2].strVal1;
this.key2.intVal = addType(getCommonSuperClass(t, u));
result = new Item(0, this.key2);
put(result);
}
return result.intVal;
}
protected String getCommonSuperClass(String type1, String type2) {
ClassLoader classLoader = getClass().getClassLoader();
Class d;
Class c;
try {
c = Class.forName(type1.replace('/', '.'), false, classLoader);
d = Class.forName(type2.replace('/', '.'), false, classLoader);
} catch (Exception e) {
throw new RuntimeException(e.toString());
}
if (c.isAssignableFrom(d)) {
return type1;
}
if (d.isAssignableFrom(c)) {
return type2;
}
if ((c.isInterface()) || (d.isInterface())) {
return "java/lang/Object";
}
do {
c = c.getSuperclass();
} while (!
c.isAssignableFrom(d));
return c.getName().replace('.', '/');
}
private Item get(Item key) {
Item i = this.items[(key.hashCode % this.items.length)];
while ((i != null) && ((i.type != key.type) || (!key.isEqualTo(i)))) {
i = i.next;
}
return i;
}
private void put(Item i) {
if (this.index + this.typeCount > this.threshold) {
int ll = this.items.length;
int nl = ll * 2 + 1;
Item[] newItems = new Item[nl];
for (int l = ll - 1; l >= 0; l--) {
Item j = this.items[l];
while (j != null) {
int index = j.hashCode % newItems.length;
Item k = j.next;
j.next = newItems[index];
newItems[index] = j;
j = k;
}
}
this.items = newItems;
this.threshold = ((int) (nl * 0.75D));
}
int index = i.hashCode % this.items.length;
i.next = this.items[index];
this.items[index] = i;
}
private void put122(int b, int s1, int s2) {
this.pool.put12(b, s1).putShort(s2);
}
private void put112(int b1, int b2, int s) {
this.pool.put11(b1, b2).putShort(s);
}
}