1b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato/* 2b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * ProGuard -- shrinking, optimization, obfuscation, and preverification 3b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * of Java bytecode. 4b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * 59f606f95f03a75961498803e24bee6799a7c0885Ying Wang * Copyright (c) 2002-2009 Eric Lafortune (eric@graphics.cornell.edu) 6b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * 7b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * This program is free software; you can redistribute it and/or modify it 8b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * under the terms of the GNU General Public License as published by the Free 9b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * Software Foundation; either version 2 of the License, or (at your option) 10b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * any later version. 11b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * 12b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * This program is distributed in the hope that it will be useful, but WITHOUT 13b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 14b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for 15b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * more details. 16b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * 17b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * You should have received a copy of the GNU General Public License along 18b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * with this program; if not, write to the Free Software Foundation, Inc., 19b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA 20b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato */ 21b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onoratopackage proguard.classfile.editor; 22b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 23b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onoratoimport proguard.classfile.*; 24b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onoratoimport proguard.classfile.constant.*; 25b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onoratoimport proguard.classfile.constant.visitor.ConstantVisitor; 26b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 27b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato/** 28b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * This ConstantVisitor adds all constants that it visits to the constant pool 29b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * of a given target class. 30b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * 31b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * @author Eric Lafortune 32b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato */ 33b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onoratopublic class ConstantAdder 34b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onoratoimplements ConstantVisitor 35b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato{ 36b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato private final ConstantPoolEditor constantPoolEditor; 37b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 38b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato private int constantIndex; 39b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 40b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 41b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato /** 42b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * Creates a new ConstantAdder that will copy constants into the given 43b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * target class. 44b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato */ 45b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato public ConstantAdder(ProgramClass targetClass) 46b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 47b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato constantPoolEditor = new ConstantPoolEditor(targetClass); 48b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 49b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 50b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 51b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato /** 52b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * Adds a copy of the specified constant in the given class and returns 53b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * its index. If the specified index is 0, the returned value is 0 too. 54b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato */ 55b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato public int addConstant(Clazz clazz, int constantIndex) 56b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 57b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato clazz.constantPoolEntryAccept(constantIndex, this); 58b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 59b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato return this.constantIndex; 60b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 61b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 62b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 63b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato /** 64b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * Adds a copy of the given constant in the given class and returns 65b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * its index. 66b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato */ 67b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato public int addConstant(Clazz clazz, Constant constant) 68b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 69b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato constant.accept(clazz, this); 70b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 71b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato return this.constantIndex; 72b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 73b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 74b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 75b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato /** 76b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * Returns the index of the most recently created constant in the constant 77b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * pool of the target class. 78b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato */ 79b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato public int getConstantIndex() 80b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 81b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato return constantIndex; 82b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 83b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 84b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 85b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Implementations for ConstantVisitor. 86b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 87b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato public void visitIntegerConstant(Clazz clazz, IntegerConstant integerConstant) 88b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 89b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato constantIndex = 90b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato constantPoolEditor.addIntegerConstant(integerConstant.getValue()); 91b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 92b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 93b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 94b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato public void visitLongConstant(Clazz clazz, LongConstant longConstant) 95b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 96b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato constantIndex = 97b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato constantPoolEditor.addLongConstant(longConstant.getValue()); 98b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 99b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 100b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 101b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato public void visitFloatConstant(Clazz clazz, FloatConstant floatConstant) 102b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 103b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato constantIndex = 104b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato constantPoolEditor.addFloatConstant(floatConstant.getValue()); 105b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 106b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 107b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 108b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato public void visitDoubleConstant(Clazz clazz, DoubleConstant doubleConstant) 109b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 110b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato constantIndex = 111b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato constantPoolEditor.addDoubleConstant(doubleConstant.getValue()); 112b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 113b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 114b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 115b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato public void visitStringConstant(Clazz clazz, StringConstant stringConstant) 116b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 117b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato constantIndex = 118b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato constantPoolEditor.addStringConstant(stringConstant.getString(clazz), 119b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato stringConstant.referencedClass, 120b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato stringConstant.referencedMember); 121b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 122b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 123b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 124b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato public void visitUtf8Constant(Clazz clazz, Utf8Constant utf8Constant) 125b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 126b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato constantIndex = 127b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato constantPoolEditor.addUtf8Constant(utf8Constant.getString()); 128b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 129b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 130b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 131b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato public void visitFieldrefConstant(Clazz clazz, FieldrefConstant fieldrefConstant) 132b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 133b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // First add the referenced class constant, with its own referenced class. 134b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato clazz.constantPoolEntryAccept(fieldrefConstant.u2classIndex, this); 135b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 136b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Then add the actual field reference constant, with its referenced 137b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // class and class member. 138b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato constantIndex = 139b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato constantPoolEditor.addFieldrefConstant(constantIndex, 140b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato fieldrefConstant.getName(clazz), 141b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato fieldrefConstant.getType(clazz), 142b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato fieldrefConstant.referencedClass, 143b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato fieldrefConstant.referencedMember); 144b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 145b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 146b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 147b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato public void visitInterfaceMethodrefConstant(Clazz clazz, InterfaceMethodrefConstant interfaceMethodrefConstant) 148b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 149b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // First add the referenced class constant, with its own referenced class. 150b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato clazz.constantPoolEntryAccept(interfaceMethodrefConstant.u2classIndex, this); 151b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 152b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Then add the actual interface method reference constant, with its 153b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // referenced class and class member. 154b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato constantIndex = 155b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato constantPoolEditor.addInterfaceMethodrefConstant(constantIndex, 156b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato interfaceMethodrefConstant.getName(clazz), 157b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato interfaceMethodrefConstant.getType(clazz), 158b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato interfaceMethodrefConstant.referencedClass, 159b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato interfaceMethodrefConstant.referencedMember); 160b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 161b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 162b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 163b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato public void visitMethodrefConstant(Clazz clazz, MethodrefConstant methodrefConstant) 164b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 165b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // First add the referenced class constant, with its own referenced class. 166b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato clazz.constantPoolEntryAccept(methodrefConstant.u2classIndex, this); 167b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 168b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Then add the actual method reference constant, with its referenced 169b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // class and class member. 170b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato constantIndex = 171b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato constantPoolEditor.addMethodrefConstant(constantIndex, 172b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato methodrefConstant.getName(clazz), 173b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato methodrefConstant.getType(clazz), 174b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato methodrefConstant.referencedClass, 175b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato methodrefConstant.referencedMember); 176b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 177b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 178b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 179b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato public void visitClassConstant(Clazz clazz, ClassConstant classConstant) 180b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 181b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Add the class constant, with its referenced class.. 182b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato constantIndex = 183b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato constantPoolEditor.addClassConstant(classConstant.getName(clazz), 184b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato classConstant.referencedClass); 185b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 186b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 187b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 188b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato public void visitNameAndTypeConstant(Clazz clazz, NameAndTypeConstant nameAndTypeConstant) 189b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 190b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato constantIndex = 191b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato constantPoolEditor.addNameAndTypeConstant(nameAndTypeConstant.getName(clazz), 192b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato nameAndTypeConstant.getType(clazz)); 193b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 194b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato} 195