123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312 |
- /*
- * Javassist, a Java-bytecode translator toolkit.
- * Copyright (C) 1999- Shigeru Chiba. All Rights Reserved.
- *
- * The contents of this file are subject to the Mozilla Public License Version
- * 1.1 (the "License"); you may not use this file except in compliance with
- * the License. Alternatively, the contents of this file may be used under
- * the terms of the GNU Lesser General Public License Version 2.1 or later,
- * or the Apache License Version 2.0.
- *
- * Software distributed under the License is distributed on an "AS IS" basis,
- * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
- * for the specific language governing rights and limitations under the
- * License.
- */
-
- package javassist.bytecode;
-
- import java.io.DataInputStream;
- import java.io.DataOutputStream;
- import java.io.IOException;
- import java.util.Map;
- import java.util.ArrayList;
- import java.util.ListIterator;
- import java.util.List;
- import java.util.Iterator;
-
- // Note: if you define a new subclass of AttributeInfo, then
- // update AttributeInfo.read(), .copy(), and (maybe) write().
-
- /**
- * <code>attribute_info</code> structure.
- *
- * @see ClassFile#getAttribute(String)
- * @see MethodInfo#getAttribute(String)
- * @see FieldInfo#getAttribute(String)
- */
- public class AttributeInfo {
- protected ConstPool constPool;
- int name;
- byte[] info;
-
- protected AttributeInfo(ConstPool cp, int attrname, byte[] attrinfo) {
- constPool = cp;
- name = attrname;
- info = attrinfo;
- }
-
- protected AttributeInfo(ConstPool cp, String attrname) {
- this(cp, attrname, (byte[])null);
- }
-
- /**
- * Constructs an <code>attribute_info</code> structure.
- *
- * @param cp constant pool table
- * @param attrname attribute name
- * @param attrinfo <code>info</code> field
- * of <code>attribute_info</code> structure.
- */
- public AttributeInfo(ConstPool cp, String attrname, byte[] attrinfo) {
- this(cp, cp.addUtf8Info(attrname), attrinfo);
- }
-
- protected AttributeInfo(ConstPool cp, int n, DataInputStream in)
- throws IOException
- {
- constPool = cp;
- name = n;
- int len = in.readInt();
- info = new byte[len];
- if (len > 0)
- in.readFully(info);
- }
-
- static AttributeInfo read(ConstPool cp, DataInputStream in)
- throws IOException
- {
- int name = in.readUnsignedShort();
- String nameStr = cp.getUtf8Info(name);
- char first = nameStr.charAt(0);
- if (first < 'M') {
- if (first < 'E') {
- if (nameStr.equals(AnnotationDefaultAttribute.tag))
- return new AnnotationDefaultAttribute(cp, name, in);
- else if (nameStr.equals(BootstrapMethodsAttribute.tag))
- return new BootstrapMethodsAttribute(cp, name, in);
- else if (nameStr.equals(CodeAttribute.tag))
- return new CodeAttribute(cp, name, in);
- else if (nameStr.equals(ConstantAttribute.tag))
- return new ConstantAttribute(cp, name, in);
- else if (nameStr.equals(DeprecatedAttribute.tag))
- return new DeprecatedAttribute(cp, name, in);
- }
- else {
- if (nameStr.equals(EnclosingMethodAttribute.tag))
- return new EnclosingMethodAttribute(cp, name, in);
- else if (nameStr.equals(ExceptionsAttribute.tag))
- return new ExceptionsAttribute(cp, name, in);
- else if (nameStr.equals(InnerClassesAttribute.tag))
- return new InnerClassesAttribute(cp, name, in);
- else if (nameStr.equals(LineNumberAttribute.tag))
- return new LineNumberAttribute(cp, name, in);
- else if (nameStr.equals(LocalVariableAttribute.tag))
- return new LocalVariableAttribute(cp, name, in);
- else if (nameStr.equals(LocalVariableTypeAttribute.tag))
- return new LocalVariableTypeAttribute(cp, name, in);
- }
- }
- else {
- if (first < 'S') {
- /* 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(AnnotationsAttribute.visibleTag)
- || nameStr.equals(AnnotationsAttribute.invisibleTag)) {
- // RuntimeVisibleAnnotations or RuntimeInvisibleAnnotations
- return new AnnotationsAttribute(cp, name, in);
- }
- else if (nameStr.equals(ParameterAnnotationsAttribute.visibleTag)
- || nameStr.equals(ParameterAnnotationsAttribute.invisibleTag))
- return new ParameterAnnotationsAttribute(cp, name, in);
- else if (nameStr.equals(TypeAnnotationsAttribute.visibleTag)
- || nameStr.equals(TypeAnnotationsAttribute.invisibleTag))
- return new TypeAnnotationsAttribute(cp, name, in);
- }
- else {
- if (nameStr.equals(SignatureAttribute.tag))
- return new SignatureAttribute(cp, name, in);
- else if (nameStr.equals(SourceFileAttribute.tag))
- return new SourceFileAttribute(cp, name, in);
- else if (nameStr.equals(SyntheticAttribute.tag))
- return new SyntheticAttribute(cp, name, in);
- else if (nameStr.equals(StackMap.tag))
- return new StackMap(cp, name, in);
- else if (nameStr.equals(StackMapTable.tag))
- return new StackMapTable(cp, name, in);
- }
- }
-
- return new AttributeInfo(cp, name, in);
- }
-
- /**
- * Returns an attribute name.
- */
- public String getName() {
- return constPool.getUtf8Info(name);
- }
-
- /**
- * Returns a constant pool table.
- */
- public ConstPool getConstPool() { return constPool; }
-
- /**
- * Returns the length of this <code>attribute_info</code>
- * structure.
- * The returned value is <code>attribute_length + 6</code>.
- */
- public int length() {
- return info.length + 6;
- }
-
- /**
- * Returns the <code>info</code> field
- * of this <code>attribute_info</code> structure.
- *
- * <p>This method is not available if the object is an instance
- * of <code>CodeAttribute</code>.
- */
- public byte[] get() { return info; }
-
- /**
- * Sets the <code>info</code> field
- * of this <code>attribute_info</code> structure.
- *
- * <p>This method is not available if the object is an instance
- * of <code>CodeAttribute</code>.
- */
- public void set(byte[] newinfo) { info = newinfo; }
-
- /**
- * Makes a copy. Class names are replaced according to the
- * given <code>Map</code> object.
- *
- * @param newCp the constant pool table used by the new copy.
- * @param classnames pairs of replaced and substituted
- * class names.
- */
- public AttributeInfo copy(ConstPool newCp, Map classnames) {
- int s = info.length;
- byte[] srcInfo = info;
- byte[] newInfo = new byte[s];
- for (int i = 0; i < s; ++i)
- newInfo[i] = srcInfo[i];
-
- return new AttributeInfo(newCp, getName(), newInfo);
- }
-
- void write(DataOutputStream out) throws IOException {
- out.writeShort(name);
- out.writeInt(info.length);
- if (info.length > 0)
- out.write(info);
- }
-
- static int getLength(ArrayList list) {
- int size = 0;
- int n = list.size();
- for (int i = 0; i < n; ++i) {
- AttributeInfo attr = (AttributeInfo)list.get(i);
- size += attr.length();
- }
-
- return size;
- }
-
- static AttributeInfo lookup(ArrayList list, String name) {
- if (list == null)
- return null;
-
- ListIterator iterator = list.listIterator();
- while (iterator.hasNext()) {
- AttributeInfo ai = (AttributeInfo)iterator.next();
- if (ai.getName().equals(name))
- return ai;
- }
-
- return null; // no such attribute
- }
-
- static synchronized AttributeInfo remove(ArrayList list, String name) {
- if (list == null)
- return null;
-
- AttributeInfo removed = null;
- ListIterator iterator = list.listIterator();
- while (iterator.hasNext()) {
- AttributeInfo ai = (AttributeInfo)iterator.next();
- if (ai.getName().equals(name)) {
- iterator.remove();
- removed = ai;
- }
- }
-
- return removed;
- }
-
- static void writeAll(ArrayList list, DataOutputStream out)
- throws IOException
- {
- if (list == null)
- return;
-
- int n = list.size();
- for (int i = 0; i < n; ++i) {
- AttributeInfo attr = (AttributeInfo)list.get(i);
- attr.write(out);
- }
- }
-
- static ArrayList copyAll(ArrayList list, ConstPool cp) {
- if (list == null)
- return null;
-
- ArrayList newList = new ArrayList();
- int n = list.size();
- for (int i = 0; i < n; ++i) {
- AttributeInfo attr = (AttributeInfo)list.get(i);
- newList.add(attr.copy(cp, null));
- }
-
- return newList;
- }
-
- /* The following two methods are used to implement
- * ClassFile.renameClass().
- * Only CodeAttribute, LocalVariableAttribute,
- * AnnotationsAttribute, and SignatureAttribute
- * override these methods.
- */
- void renameClass(String oldname, String newname) {}
- void renameClass(Map classnames) {}
-
- static void renameClass(List attributes, String oldname, String newname) {
- Iterator iterator = attributes.iterator();
- while (iterator.hasNext()) {
- AttributeInfo ai = (AttributeInfo)iterator.next();
- ai.renameClass(oldname, newname);
- }
- }
-
- static void renameClass(List attributes, Map classnames) {
- Iterator iterator = attributes.iterator();
- while (iterator.hasNext()) {
- AttributeInfo ai = (AttributeInfo)iterator.next();
- ai.renameClass(classnames);
- }
- }
-
- void getRefClasses(Map classnames) {}
-
- static void getRefClasses(List attributes, Map classnames) {
- Iterator iterator = attributes.iterator();
- while (iterator.hasNext()) {
- AttributeInfo ai = (AttributeInfo)iterator.next();
- ai.getRefClasses(classnames);
- }
- }
- }
|