|
|
@ -1,5 +1,5 @@ |
|
|
|
/* |
|
|
|
/* |
|
|
|
* Copyright 2000-2015 JetBrains s.r.o. |
|
|
|
* Copyright 2000-2016 JetBrains s.r.o. |
|
|
|
* |
|
|
|
* |
|
|
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
|
|
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
|
|
|
* you may not use this file except in compliance with the License. |
|
|
|
* you may not use this file except in compliance with the License. |
|
|
@ -27,20 +27,21 @@ import org.jetbrains.java.decompiler.util.DataInputFullStream; |
|
|
|
import java.io.DataInputStream; |
|
|
|
import java.io.DataInputStream; |
|
|
|
import java.io.IOException; |
|
|
|
import java.io.IOException; |
|
|
|
import java.util.ArrayList; |
|
|
|
import java.util.ArrayList; |
|
|
|
|
|
|
|
import java.util.BitSet; |
|
|
|
import java.util.List; |
|
|
|
import java.util.List; |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@SuppressWarnings("AssignmentToForLoopParameter") |
|
|
|
public class ConstantPool implements NewClassNameBuilder { |
|
|
|
public class ConstantPool implements NewClassNameBuilder { |
|
|
|
|
|
|
|
|
|
|
|
public static final int FIELD = 1; |
|
|
|
public static final int FIELD = 1; |
|
|
|
public static final int METHOD = 2; |
|
|
|
public static final int METHOD = 2; |
|
|
|
|
|
|
|
|
|
|
|
private final List<PooledConstant> pool = new ArrayList<PooledConstant>(); |
|
|
|
private final List<PooledConstant> pool; |
|
|
|
private final PoolInterceptor interceptor; |
|
|
|
private final PoolInterceptor interceptor; |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
public ConstantPool(DataInputStream in) throws IOException { |
|
|
|
public ConstantPool(DataInputStream in) throws IOException { |
|
|
|
int size = in.readUnsignedShort(); |
|
|
|
int size = in.readUnsignedShort(); |
|
|
|
int[] pass = new int[size]; |
|
|
|
pool = new ArrayList<PooledConstant>(size); |
|
|
|
|
|
|
|
BitSet[] nextPass = {new BitSet(size), new BitSet(size), new BitSet(size)}; |
|
|
|
|
|
|
|
|
|
|
|
// first dummy constant
|
|
|
|
// first dummy constant
|
|
|
|
pool.add(null); |
|
|
|
pool.add(null); |
|
|
@ -53,54 +54,59 @@ public class ConstantPool implements NewClassNameBuilder { |
|
|
|
case CodeConstants.CONSTANT_Utf8: |
|
|
|
case CodeConstants.CONSTANT_Utf8: |
|
|
|
pool.add(new PrimitiveConstant(CodeConstants.CONSTANT_Utf8, in.readUTF())); |
|
|
|
pool.add(new PrimitiveConstant(CodeConstants.CONSTANT_Utf8, in.readUTF())); |
|
|
|
break; |
|
|
|
break; |
|
|
|
|
|
|
|
|
|
|
|
case CodeConstants.CONSTANT_Integer: |
|
|
|
case CodeConstants.CONSTANT_Integer: |
|
|
|
pool.add(new PrimitiveConstant(CodeConstants.CONSTANT_Integer, new Integer(in.readInt()))); |
|
|
|
pool.add(new PrimitiveConstant(CodeConstants.CONSTANT_Integer, new Integer(in.readInt()))); |
|
|
|
break; |
|
|
|
break; |
|
|
|
|
|
|
|
|
|
|
|
case CodeConstants.CONSTANT_Float: |
|
|
|
case CodeConstants.CONSTANT_Float: |
|
|
|
pool.add(new PrimitiveConstant(CodeConstants.CONSTANT_Float, new Float(in.readFloat()))); |
|
|
|
pool.add(new PrimitiveConstant(CodeConstants.CONSTANT_Float, new Float(in.readFloat()))); |
|
|
|
break; |
|
|
|
break; |
|
|
|
|
|
|
|
|
|
|
|
case CodeConstants.CONSTANT_Long: |
|
|
|
case CodeConstants.CONSTANT_Long: |
|
|
|
pool.add(new PrimitiveConstant(CodeConstants.CONSTANT_Long, new Long(in.readLong()))); |
|
|
|
pool.add(new PrimitiveConstant(CodeConstants.CONSTANT_Long, new Long(in.readLong()))); |
|
|
|
pool.add(null); |
|
|
|
pool.add(null); |
|
|
|
i++; |
|
|
|
i++; |
|
|
|
break; |
|
|
|
break; |
|
|
|
|
|
|
|
|
|
|
|
case CodeConstants.CONSTANT_Double: |
|
|
|
case CodeConstants.CONSTANT_Double: |
|
|
|
pool.add(new PrimitiveConstant(CodeConstants.CONSTANT_Double, new Double(in.readDouble()))); |
|
|
|
pool.add(new PrimitiveConstant(CodeConstants.CONSTANT_Double, new Double(in.readDouble()))); |
|
|
|
pool.add(null); |
|
|
|
pool.add(null); |
|
|
|
i++; |
|
|
|
i++; |
|
|
|
break; |
|
|
|
break; |
|
|
|
|
|
|
|
|
|
|
|
case CodeConstants.CONSTANT_Class: |
|
|
|
case CodeConstants.CONSTANT_Class: |
|
|
|
case CodeConstants.CONSTANT_String: |
|
|
|
case CodeConstants.CONSTANT_String: |
|
|
|
case CodeConstants.CONSTANT_MethodType: |
|
|
|
case CodeConstants.CONSTANT_MethodType: |
|
|
|
pool.add(new PrimitiveConstant(tag, in.readUnsignedShort())); |
|
|
|
pool.add(new PrimitiveConstant(tag, in.readUnsignedShort())); |
|
|
|
pass[i] = 1; |
|
|
|
nextPass[0].set(i); |
|
|
|
break; |
|
|
|
break; |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
case CodeConstants.CONSTANT_NameAndType: |
|
|
|
|
|
|
|
pool.add(new LinkConstant(tag, in.readUnsignedShort(), in.readUnsignedShort())); |
|
|
|
|
|
|
|
nextPass[0].set(i); |
|
|
|
|
|
|
|
break; |
|
|
|
|
|
|
|
|
|
|
|
case CodeConstants.CONSTANT_Fieldref: |
|
|
|
case CodeConstants.CONSTANT_Fieldref: |
|
|
|
case CodeConstants.CONSTANT_Methodref: |
|
|
|
case CodeConstants.CONSTANT_Methodref: |
|
|
|
case CodeConstants.CONSTANT_InterfaceMethodref: |
|
|
|
case CodeConstants.CONSTANT_InterfaceMethodref: |
|
|
|
case CodeConstants.CONSTANT_NameAndType: |
|
|
|
|
|
|
|
case CodeConstants.CONSTANT_InvokeDynamic: |
|
|
|
case CodeConstants.CONSTANT_InvokeDynamic: |
|
|
|
pool.add(new LinkConstant(tag, in.readUnsignedShort(), in.readUnsignedShort())); |
|
|
|
pool.add(new LinkConstant(tag, in.readUnsignedShort(), in.readUnsignedShort())); |
|
|
|
if (tag == CodeConstants.CONSTANT_NameAndType) { |
|
|
|
nextPass[1].set(i); |
|
|
|
pass[i] = 1; |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
else { |
|
|
|
|
|
|
|
pass[i] = 2; |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
break; |
|
|
|
break; |
|
|
|
|
|
|
|
|
|
|
|
case CodeConstants.CONSTANT_MethodHandle: |
|
|
|
case CodeConstants.CONSTANT_MethodHandle: |
|
|
|
pool.add(new LinkConstant(tag, in.readUnsignedByte(), in.readUnsignedShort())); |
|
|
|
pool.add(new LinkConstant(tag, in.readUnsignedByte(), in.readUnsignedShort())); |
|
|
|
pass[i] = 3; |
|
|
|
nextPass[2].set(i); |
|
|
|
break; |
|
|
|
break; |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
// resolving complex pool elements
|
|
|
|
// resolving complex pool elements
|
|
|
|
for (int passIndex = 1; passIndex <= 3; passIndex++) { |
|
|
|
for (BitSet pass : nextPass) { |
|
|
|
for (int i = 1; i < size; i++) { |
|
|
|
int idx = 0; |
|
|
|
if (pass[i] == passIndex) { |
|
|
|
while ((idx = pass.nextSetBit(idx + 1)) > 0) { |
|
|
|
pool.get(i).resolveConstant(this); |
|
|
|
pool.get(idx).resolveConstant(this); |
|
|
|
} |
|
|
|
|
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
@ -116,6 +122,7 @@ public class ConstantPool implements NewClassNameBuilder { |
|
|
|
case CodeConstants.CONSTANT_Utf8: |
|
|
|
case CodeConstants.CONSTANT_Utf8: |
|
|
|
in.readUTF(); |
|
|
|
in.readUTF(); |
|
|
|
break; |
|
|
|
break; |
|
|
|
|
|
|
|
|
|
|
|
case CodeConstants.CONSTANT_Integer: |
|
|
|
case CodeConstants.CONSTANT_Integer: |
|
|
|
case CodeConstants.CONSTANT_Float: |
|
|
|
case CodeConstants.CONSTANT_Float: |
|
|
|
case CodeConstants.CONSTANT_Fieldref: |
|
|
|
case CodeConstants.CONSTANT_Fieldref: |
|
|
@ -125,16 +132,19 @@ public class ConstantPool implements NewClassNameBuilder { |
|
|
|
case CodeConstants.CONSTANT_InvokeDynamic: |
|
|
|
case CodeConstants.CONSTANT_InvokeDynamic: |
|
|
|
in.discard(4); |
|
|
|
in.discard(4); |
|
|
|
break; |
|
|
|
break; |
|
|
|
|
|
|
|
|
|
|
|
case CodeConstants.CONSTANT_Long: |
|
|
|
case CodeConstants.CONSTANT_Long: |
|
|
|
case CodeConstants.CONSTANT_Double: |
|
|
|
case CodeConstants.CONSTANT_Double: |
|
|
|
in.discard(8); |
|
|
|
in.discard(8); |
|
|
|
i++; |
|
|
|
i++; |
|
|
|
break; |
|
|
|
break; |
|
|
|
|
|
|
|
|
|
|
|
case CodeConstants.CONSTANT_Class: |
|
|
|
case CodeConstants.CONSTANT_Class: |
|
|
|
case CodeConstants.CONSTANT_String: |
|
|
|
case CodeConstants.CONSTANT_String: |
|
|
|
case CodeConstants.CONSTANT_MethodType: |
|
|
|
case CodeConstants.CONSTANT_MethodType: |
|
|
|
in.discard(2); |
|
|
|
in.discard(2); |
|
|
|
break; |
|
|
|
break; |
|
|
|
|
|
|
|
|
|
|
|
case CodeConstants.CONSTANT_MethodHandle: |
|
|
|
case CodeConstants.CONSTANT_MethodHandle: |
|
|
|
in.discard(3); |
|
|
|
in.discard(3); |
|
|
|
} |
|
|
|
} |
|
|
@ -155,7 +165,7 @@ public class ConstantPool implements NewClassNameBuilder { |
|
|
|
className = oldClassName; |
|
|
|
className = oldClassName; |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
String newElement = interceptor.getName(className + " " + elementName + " " + descriptor); |
|
|
|
String newElement = interceptor.getName(className + ' ' + elementName + ' ' + descriptor); |
|
|
|
if (newElement != null) { |
|
|
|
if (newElement != null) { |
|
|
|
elementName = newElement.split(" ")[1]; |
|
|
|
elementName = newElement.split(" ")[1]; |
|
|
|
} |
|
|
|
} |
|
|
@ -196,7 +206,7 @@ public class ConstantPool implements NewClassNameBuilder { |
|
|
|
ln.type == CodeConstants.CONSTANT_Methodref || |
|
|
|
ln.type == CodeConstants.CONSTANT_Methodref || |
|
|
|
ln.type == CodeConstants.CONSTANT_InterfaceMethodref)) { |
|
|
|
ln.type == CodeConstants.CONSTANT_InterfaceMethodref)) { |
|
|
|
String newClassName = buildNewClassname(ln.classname); |
|
|
|
String newClassName = buildNewClassname(ln.classname); |
|
|
|
String newElement = interceptor.getName(ln.classname + " " + ln.elementname + " " + ln.descriptor); |
|
|
|
String newElement = interceptor.getName(ln.classname + ' ' + ln.elementname + ' ' + ln.descriptor); |
|
|
|
String newDescriptor = buildNewDescriptor(ln.type == CodeConstants.CONSTANT_Fieldref, ln.descriptor); |
|
|
|
String newDescriptor = buildNewDescriptor(ln.type == CodeConstants.CONSTANT_Fieldref, ln.descriptor); |
|
|
|
|
|
|
|
|
|
|
|
if (newClassName != null || newElement != null || newDescriptor != null) { |
|
|
|
if (newClassName != null || newElement != null || newDescriptor != null) { |
|
|
@ -220,10 +230,10 @@ public class ConstantPool implements NewClassNameBuilder { |
|
|
|
|
|
|
|
|
|
|
|
if (vt.arrayDim > 0) { |
|
|
|
if (vt.arrayDim > 0) { |
|
|
|
for (int i = 0; i < vt.arrayDim; i++) { |
|
|
|
for (int i = 0; i < vt.arrayDim; i++) { |
|
|
|
buffer.append("["); |
|
|
|
buffer.append('['); |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
buffer.append("L").append(newName).append(";"); |
|
|
|
buffer.append('L').append(newName).append(';'); |
|
|
|
} |
|
|
|
} |
|
|
|
else { |
|
|
|
else { |
|
|
|
buffer.append(newName); |
|
|
|
buffer.append(newName); |
|
|
@ -243,4 +253,4 @@ public class ConstantPool implements NewClassNameBuilder { |
|
|
|
return MethodDescriptor.parseDescriptor(descriptor).buildNewDescriptor(this); |
|
|
|
return MethodDescriptor.parseDescriptor(descriptor).buildNewDescriptor(this); |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
} |