1 files changed, 91 insertions, 0 deletions
diff --git a/src/proguard/classfile/editor/BootstrapMethodsAttributeAdder.java b/src/proguard/classfile/editor/BootstrapMethodsAttributeAdder.java
new file mode 100644
index 0000000..1488b9b
--- /dev/null
+++ b/src/proguard/classfile/editor/BootstrapMethodsAttributeAdder.java
@@ -0,0 +1,91 @@
+/*
+ * ProGuard -- shrinking, optimization, obfuscation, and preverification
+ *             of Java bytecode.
+ *
+ * Copyright (c) 2002-2014 Eric Lafortune (eric@graphics.cornell.edu)
+ *
+ * This program is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU General Public License as published by the Free
+ * Software Foundation; either version 2 of the License, or (at your option)
+ * any later version.
+ *
+ * This program is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
+ * more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with this program; if not, write to the Free Software Foundation, Inc.,
+ * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+ */
+package proguard.classfile.editor;
+
+import proguard.classfile.*;
+import proguard.classfile.attribute.*;
+import proguard.classfile.attribute.visitor.BootstrapMethodInfoVisitor;
+
+/**
+ * This BootstrapMethodInfoVisitor adds all bootstrap methods that it visits to
+ * the given target class, creating a bootstrap methods attribute if necessary.
+ */
+public class BootstrapMethodsAttributeAdder
+implements   BootstrapMethodInfoVisitor
+{
+    private final ProgramClass             targetClass;
+    private final ConstantPoolEditor       constantPoolEditor;
+    private       BootstrapMethodInfoAdder bootstrapMethodInfoAdder;
+
+
+    /**
+     * Creates a new BootstrapMethodsAttributeAdder that will copy bootstrap
+     * methods into the given target class/
+     */
+    public BootstrapMethodsAttributeAdder(ProgramClass targetClass)
+    {
+        this.targetClass        = targetClass;
+        this.constantPoolEditor = new ConstantPoolEditor(targetClass);
+    }
+
+
+    /**
+     * Returns the index of the most recently added bootstrap method.
+     */
+    public int getBootstrapMethodIndex()
+    {
+        return bootstrapMethodInfoAdder.getBootstrapMethodIndex();
+    }
+
+
+    // Implementations for BootstrapMethodInfoVisitor.
+
+    public void visitBootstrapMethodInfo(Clazz clazz, BootstrapMethodInfo bootstrapMethodInfo)
+    {
+        // Make sure we have a bootstrap methods attribute adder.
+        if (bootstrapMethodInfoAdder == null)
+        {
+            // Make sure we have a target bootstrap methods attribute.
+            AttributesEditor attributesEditor =
+                new AttributesEditor(targetClass, false);
+
+            BootstrapMethodsAttribute targetBootstrapMethodsAttribute =
+                (BootstrapMethodsAttribute)attributesEditor.findAttribute(ClassConstants.ATTR_BootstrapMethods);
+
+            if (targetBootstrapMethodsAttribute == null)
+            {
+                targetBootstrapMethodsAttribute =
+                    new BootstrapMethodsAttribute(constantPoolEditor.addUtf8Constant(ClassConstants.ATTR_BootstrapMethods),
+                                                  0,
+                                                  new BootstrapMethodInfo[0]);
+
+                attributesEditor.addAttribute(targetBootstrapMethodsAttribute);
+            }
+
+            // Create a bootstrap method adder for it.
+            bootstrapMethodInfoAdder = new BootstrapMethodInfoAdder(targetClass,
+                                                                    targetBootstrapMethodsAttribute);
+        }
+
+        // Delegate to the bootstrap method adder.
+        bootstrapMethodInfoAdder.visitBootstrapMethodInfo(clazz, bootstrapMethodInfo);
+    }
+}
+\ No newline at end of file