Add support for the new Dynamic constant (17) created in java 11

3 files changed, 208 insertions, 48 deletions

diff --git a/src/main/javassist/bytecode/Bytecode.java b/src/main/javassist/bytecode/Bytecode.java
index 37a6d127..193e3f8a 100644
--- a/src/main/javassist/bytecode/Bytecode.java
+++ b/src/main/javassist/bytecode/Bytecode.java
@@ -1241,6 +1241,25 @@ public class Bytecode extends ByteVector implements Cloneable, Opcode {
     }
 
     /**
+     * Appends DYNAMIC.
+     *
+     * @param bootstrap     an index into the <code>bootstrap_methods</code> array
+     *                      of the bootstrap method table.
+     * @param name          the method name.
+     * @param desc          the method descriptor.
+     * @see Descriptor#ofMethod(CtClass,CtClass[])
+     * @since 3.17
+     */
+    public void addDynamic(int bootstrap, String name, String desc) {
+        int nt = constPool.addNameAndTypeInfo(name, desc);
+        int dyn = constPool.addDynamicInfo(bootstrap, nt);
+        add(INVOKEDYNAMIC);
+        addIndex(dyn);
+        add(0, 0);
+        growStack(Descriptor.dataSize(desc));   // assume ConstPool#REF_invokeStatic
+    }
+
+    /**
      * Appends LDC or LDC_W.  The pushed item is a <code>String</code>
      * object.
      *
diff --git a/src/main/javassist/bytecode/ClassFileWriter.java b/src/main/javassist/bytecode/ClassFileWriter.java
index 931ffcb4..03650b75 100644
--- a/src/main/javassist/bytecode/ClassFileWriter.java
+++ b/src/main/javassist/bytecode/ClassFileWriter.java
@@ -693,7 +693,7 @@ public class ClassFileWriter {
          * @since 3.17.1
          */
         public int addInvokeDynamicInfo(int bootstrap,
-                                        int nameAndTypeInfo) {
+            int nameAndTypeInfo) {
             output.write(InvokeDynamicInfo.tag);
             output.writeShort(bootstrap);
             output.writeShort(nameAndTypeInfo);
@@ -701,6 +701,24 @@ public class ClassFileWriter {
         }
 
         /**
+         * Adds a new <code>CONSTANT_InvokeDynamic_info</code>
+         * structure.
+         *
+         * @param bootstrap         <code>bootstrap_method_attr_index</code>.
+         * @param nameAndTypeInfo   <code>name_and_type_index</code>.
+         * @return                  the index of the added entry.
+         *
+         * @since 3.17.1
+         */
+        public int addDynamicInfo(int bootstrap,
+            int nameAndTypeInfo) {
+            output.write(DynamicInfo.tag);
+            output.writeShort(bootstrap);
+            output.writeShort(nameAndTypeInfo);
+            return num++;
+        }
+
+        /**
          * Adds a new <code>CONSTANT_String_info</code>
          * structure.
          *
diff --git a/src/main/javassist/bytecode/ConstPool.java b/src/main/javassist/bytecode/ConstPool.java
index 0447ece7..08985089 100644
--- a/src/main/javassist/bytecode/ConstPool.java
+++ b/src/main/javassist/bytecode/ConstPool.java
@@ -716,53 +716,99 @@ public final class ConstPool
         return mtinfo.descriptor;
     }
 
-    /**
-     * Reads the <code>bootstrap_method_attr_index</code> field of the
-     * <code>CONSTANT_InvokeDynamic_info</code> structure
-     * at the given index.
-     *
-     * @since 3.17
-     */
-    public int getInvokeDynamicBootstrap(int index)
-    {
-        InvokeDynamicInfo iv = (InvokeDynamicInfo)getItem(index);
-        return iv.bootstrap;
-    }
-
-    /**
-     * Reads the <code>name_and_type_index</code> field of the
-     * <code>CONSTANT_InvokeDynamic_info</code> structure
-     * at the given index.
-     *
-     * @since 3.17
-     */
-    public int getInvokeDynamicNameAndType(int index)
-    {
-        InvokeDynamicInfo iv = (InvokeDynamicInfo)getItem(index);
-        return iv.nameAndType;
-    }
-
-    /**
-     * Reads the <code>descriptor_index</code> field of the
-     * <code>CONSTANT_NameAndType_info</code> structure
-     * indirectly specified by the given index.
-     *
-     * @param index     an index to a <code>CONSTANT_InvokeDynamic_info</code>.
-     * @return  the descriptor of the method.
-     * @since 3.17
-     */
-    public String getInvokeDynamicType(int index)
-    {
-        InvokeDynamicInfo iv = (InvokeDynamicInfo)getItem(index);
-        if (iv == null)
-            return null;
-        NameAndTypeInfo n = (NameAndTypeInfo)getItem(iv.nameAndType);
-        if(n == null)
-            return null;
-        return getUtf8Info(n.typeDescriptor);
-    }
-
-    /**
+  /**
+   * Reads the <code>bootstrap_method_attr_index</code> field of the
+   * <code>CONSTANT_InvokeDynamic_info</code> structure
+   * at the given index.
+   *
+   * @since 3.17
+   */
+  public int getInvokeDynamicBootstrap(int index)
+  {
+    InvokeDynamicInfo iv = (InvokeDynamicInfo)getItem(index);
+    return iv.bootstrap;
+  }
+
+  /**
+   * Reads the <code>name_and_type_index</code> field of the
+   * <code>CONSTANT_InvokeDynamic_info</code> structure
+   * at the given index.
+   *
+   * @since 3.17
+   */
+  public int getInvokeDynamicNameAndType(int index)
+  {
+    InvokeDynamicInfo iv = (InvokeDynamicInfo)getItem(index);
+    return iv.nameAndType;
+  }
+
+  /**
+   * Reads the <code>descriptor_index</code> field of the
+   * <code>CONSTANT_NameAndType_info</code> structure
+   * indirectly specified by the given index.
+   *
+   * @param index     an index to a <code>CONSTANT_InvokeDynamic_info</code>.
+   * @return  the descriptor of the method.
+   * @since 3.17
+   */
+  public String getInvokeDynamicType(int index)
+  {
+    InvokeDynamicInfo iv = (InvokeDynamicInfo)getItem(index);
+    if (iv == null)
+      return null;
+    NameAndTypeInfo n = (NameAndTypeInfo)getItem(iv.nameAndType);
+    if(n == null)
+      return null;
+    return getUtf8Info(n.typeDescriptor);
+  }
+
+  /**
+   * Reads the <code>bootstrap_method_attr_index</code> field of the
+   * <code>CONSTANT_Dynamic_info</code> structure
+   * at the given index.
+   *
+   * @since 3.17
+   */
+  public int getDynamicBootstrap(int index)
+  {
+    DynamicInfo iv = (DynamicInfo)getItem(index);
+    return iv.bootstrap;
+  }
+
+  /**
+   * Reads the <code>name_and_type_index</code> field of the
+   * <code>CONSTANT_Dynamic_info</code> structure
+   * at the given index.
+   *
+   * @since 3.17
+   */
+  public int getDynamicNameAndType(int index)
+  {
+    DynamicInfo iv = (DynamicInfo)getItem(index);
+    return iv.nameAndType;
+  }
+
+  /**
+   * Reads the <code>descriptor_index</code> field of the
+   * <code>CONSTANT_NameAndType_info</code> structure
+   * indirectly specified by the given index.
+   *
+   * @param index     an index to a <code>CONSTANT_Dynamic_info</code>.
+   * @return  the descriptor of the method.
+   * @since 3.17
+   */
+  public String getDynamicType(int index)
+  {
+    DynamicInfo iv = (DynamicInfo)getItem(index);
+    if (iv == null)
+      return null;
+    NameAndTypeInfo n = (NameAndTypeInfo)getItem(iv.nameAndType);
+    if(n == null)
+      return null;
+    return getUtf8Info(n.typeDescriptor);
+  }
+
+  /**
      * Reads the <code>name_index</code> field of the
      * <code>CONSTANT_Module_info</code> structure at the given index.
      *
@@ -1196,6 +1242,18 @@ public final class ConstPool
     }
 
     /**
+   * Adds a new <code>CONSTANT_InvokeDynamic_info</code> structure.
+   *
+   * @param bootstrap   <code>bootstrap_method_attr_index</code>.
+   * @param nameAndType <code>name_and_type_index</code>.
+   * @return the index of the added entry.
+   * @since 3.17
+   */
+  public int addDynamicInfo(int bootstrap, int nameAndType) {
+    return addItem(new DynamicInfo(bootstrap, nameAndType, numOfItems));
+  }
+
+  /**
      * Adds a new <code>CONSTANT_Module_info</code>
      * @param nameIndex         the index of the Utf8 entry.
      * @return          the index of the added entry.
@@ -1342,6 +1400,9 @@ public final class ConstPool
         case MethodTypeInfo.tag :               // 16
             info = new MethodTypeInfo(in, numOfItems);
             break;
+            case DynamicInfo.tag :            // 17
+                info = new DynamicInfo(in, numOfItems);
+                break;
         case InvokeDynamicInfo.tag :            // 18
             info = new InvokeDynamicInfo(in, numOfItems);
             break;
@@ -2322,6 +2383,68 @@ class InvokeDynamicInfo extends ConstInfo
     }
 }
 
+class DynamicInfo extends ConstInfo {
+
+  static final int tag = 17;
+  int bootstrap, nameAndType;
+
+  public DynamicInfo(int bootstrapMethod,
+      int ntIndex, int index) {
+    super(index);
+    bootstrap = bootstrapMethod;
+    nameAndType = ntIndex;
+  }
+
+  public DynamicInfo(DataInputStream in, int index)
+      throws IOException {
+    super(index);
+    bootstrap = in.readUnsignedShort();
+    nameAndType = in.readUnsignedShort();
+  }
+
+  @Override
+  public int hashCode() {
+    return (bootstrap << 16) ^ nameAndType;
+  }
+
+  @Override
+  public boolean equals(Object obj) {
+    if (obj instanceof DynamicInfo) {
+      DynamicInfo iv = (DynamicInfo) obj;
+      return iv.bootstrap == bootstrap
+          && iv.nameAndType == nameAndType;
+    }
+    return false;
+  }
+
+  @Override
+  public int getTag() {
+    return tag;
+  }
+
+  @Override
+  public int copy(ConstPool src, ConstPool dest,
+      Map<String, String> map) {
+    return dest.addDynamicInfo(bootstrap,
+        src.getItem(nameAndType).copy(src, dest, map));
+  }
+
+  @Override
+  public void write(DataOutputStream out) throws IOException {
+    out.writeByte(tag);
+    out.writeShort(bootstrap);
+    out.writeShort(nameAndType);
+  }
+
+  @Override
+  public void print(PrintWriter out) {
+    out.print("Dynamic #");
+    out.print(bootstrap);
+    out.print(", name&type #");
+    out.println(nameAndType);
+  }
+}
+
 class ModuleInfo extends ConstInfo
 {
     static final int tag = 19;