diff --git a/src/main/javassist/bytecode/AttributeInfo.java b/src/main/javassist/bytecode/AttributeInfo.java index 4bfd0dbb..be6e2a21 100644 --- a/src/main/javassist/bytecode/AttributeInfo.java +++ b/src/main/javassist/bytecode/AttributeInfo.java @@ -108,6 +108,10 @@ else if (nameStr.equals(LocalVariableTypeAttribute.tag)) /* Note that the names of Annotations attributes begin with 'R'. */ if (nameStr.equals(MethodParametersAttribute.tag)) return new MethodParametersAttribute(cp, name, in); + else if (nameStr.equals(NestHostAttribute.tag)) + return new NestHostAttribute(cp, name, in); + else if (nameStr.equals(NestMembersAttribute.tag)) + return new NestMembersAttribute(cp, name, in); else if (nameStr.equals(AnnotationsAttribute.visibleTag) || nameStr.equals(AnnotationsAttribute.invisibleTag)) // RuntimeVisibleAnnotations or RuntimeInvisibleAnnotations diff --git a/src/main/javassist/bytecode/NestHostAttribute.java b/src/main/javassist/bytecode/NestHostAttribute.java new file mode 100644 index 00000000..ae9f1e5d --- /dev/null +++ b/src/main/javassist/bytecode/NestHostAttribute.java @@ -0,0 +1,39 @@ +package javassist.bytecode; + +import java.io.DataInputStream; +import java.io.IOException; +import java.util.Map; + +/** + * NestHost_attribute. + */ +public class NestHostAttribute extends AttributeInfo { + /** + * The name of this attribute "NestHost". + */ + public static final String tag = "NestHost"; + + NestHostAttribute(ConstPool cp, int n, DataInputStream in) throws IOException { + super(cp, n, in); + } + + private NestHostAttribute(ConstPool cp, int hostIndex) { + super(cp, tag, new byte[2]); + ByteArray.write16bit(hostIndex, get(), 0); + } + + /** + * Makes a copy. Class names are replaced according to the + * given Map object. + * + * @param newCp the constant pool table used by the new copy. + * @param classnames pairs of replaced and substituted + * class names. + */ + @Override + public AttributeInfo copy(ConstPool newCp, Map classnames) { + int hostIndex = ByteArray.readU16bit(get(), 0); + int newHostIndex = getConstPool().copy(hostIndex, newCp, classnames); + return new NestHostAttribute(newCp, newHostIndex); + } +} diff --git a/src/main/javassist/bytecode/NestMembersAttribute.java b/src/main/javassist/bytecode/NestMembersAttribute.java new file mode 100644 index 00000000..679dc28c --- /dev/null +++ b/src/main/javassist/bytecode/NestMembersAttribute.java @@ -0,0 +1,49 @@ +package javassist.bytecode; + +import java.io.DataInputStream; +import java.io.IOException; +import java.util.Map; + +/** + * NestMembers_attribute. + */ +public class NestMembersAttribute extends AttributeInfo { + /** + * The name of this attribute "NestMembers". + */ + public static final String tag = "NestMembers"; + + NestMembersAttribute(ConstPool cp, int n, DataInputStream in) throws IOException { + super(cp, n, in); + } + + private NestMembersAttribute(ConstPool cp, byte[] info) { + super(cp, tag, info); + } + + /** + * Makes a copy. Class names are replaced according to the + * given Map object. + * + * @param newCp the constant pool table used by the new copy. + * @param classnames pairs of replaced and substituted + * class names. + */ + @Override + public AttributeInfo copy(ConstPool newCp, Map classnames) { + byte[] src = get(); + byte[] dest = new byte[src.length]; + ConstPool cp = getConstPool(); + + int n = ByteArray.readU16bit(src, 0); + ByteArray.write16bit(n, dest, 0); + + for (int i = 0, j = 2; i < n; ++i, j += 2) { + int index = ByteArray.readU16bit(src, j); + int newIndex = cp.copy(index, newCp, classnames); + ByteArray.write16bit(newIndex, dest, j); + } + + return new NestMembersAttribute(newCp, dest); + } +}