1 /* 2 * ProGuard -- shrinking, optimization, obfuscation, and preverification 3 * of Java bytecode. 4 * 5 * Copyright (c) 2002-2014 Eric Lafortune (eric (at) graphics.cornell.edu) 6 * 7 * This program is free software; you can redistribute it and/or modify it 8 * under the terms of the GNU General Public License as published by the Free 9 * Software Foundation; either version 2 of the License, or (at your option) 10 * any later version. 11 * 12 * This program is distributed in the hope that it will be useful, but WITHOUT 13 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 14 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for 15 * more details. 16 * 17 * You should have received a copy of the GNU General Public License along 18 * with this program; if not, write to the Free Software Foundation, Inc., 19 * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA 20 */ 21 package proguard.classfile.editor; 22 23 import proguard.classfile.*; 24 import proguard.classfile.attribute.*; 25 import proguard.classfile.attribute.visitor.BootstrapMethodInfoVisitor; 26 27 /** 28 * This BootstrapMethodInfoVisitor adds all bootstrap methods that it visits to 29 * the given target class, creating a bootstrap methods attribute if necessary. 30 */ 31 public class BootstrapMethodsAttributeAdder 32 implements BootstrapMethodInfoVisitor 33 { 34 private final ProgramClass targetClass; 35 private final ConstantPoolEditor constantPoolEditor; 36 private BootstrapMethodInfoAdder bootstrapMethodInfoAdder; 37 38 39 /** 40 * Creates a new BootstrapMethodsAttributeAdder that will copy bootstrap 41 * methods into the given target class/ 42 */ 43 public BootstrapMethodsAttributeAdder(ProgramClass targetClass) 44 { 45 this.targetClass = targetClass; 46 this.constantPoolEditor = new ConstantPoolEditor(targetClass); 47 } 48 49 50 /** 51 * Returns the index of the most recently added bootstrap method. 52 */ 53 public int getBootstrapMethodIndex() 54 { 55 return bootstrapMethodInfoAdder.getBootstrapMethodIndex(); 56 } 57 58 59 // Implementations for BootstrapMethodInfoVisitor. 60 61 public void visitBootstrapMethodInfo(Clazz clazz, BootstrapMethodInfo bootstrapMethodInfo) 62 { 63 // Make sure we have a bootstrap methods attribute adder. 64 if (bootstrapMethodInfoAdder == null) 65 { 66 // Make sure we have a target bootstrap methods attribute. 67 AttributesEditor attributesEditor = 68 new AttributesEditor(targetClass, false); 69 70 BootstrapMethodsAttribute targetBootstrapMethodsAttribute = 71 (BootstrapMethodsAttribute)attributesEditor.findAttribute(ClassConstants.ATTR_BootstrapMethods); 72 73 if (targetBootstrapMethodsAttribute == null) 74 { 75 targetBootstrapMethodsAttribute = 76 new BootstrapMethodsAttribute(constantPoolEditor.addUtf8Constant(ClassConstants.ATTR_BootstrapMethods), 77 0, 78 new BootstrapMethodInfo[0]); 79 80 attributesEditor.addAttribute(targetBootstrapMethodsAttribute); 81 } 82 83 // Create a bootstrap method adder for it. 84 bootstrapMethodInfoAdder = new BootstrapMethodInfoAdder(targetClass, 85 targetBootstrapMethodsAttribute); 86 } 87 88 // Delegate to the bootstrap method adder. 89 bootstrapMethodInfoAdder.visitBootstrapMethodInfo(clazz, bootstrapMethodInfo); 90 } 91 }