1b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato/* 2b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * ProGuard -- shrinking, optimization, obfuscation, and preverification 3b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * of Java bytecode. 4b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * 5b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang * Copyright (c) 2002-2013 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.attribute.Attribute; 25b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onoratoimport proguard.classfile.util.SimplifiedVisitor; 26b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onoratoimport proguard.classfile.visitor.MemberVisitor; 27b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 28b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato/** 29b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang * This MemberVisitor copies all class members that it visits to the given 30b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang * target class. Their visitor info is set to the class members from which they 31b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang * were copied. 32b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * 33b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * @author Eric Lafortune 34b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato */ 35b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onoratopublic class MemberAdder 36b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onoratoextends SimplifiedVisitor 37b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onoratoimplements MemberVisitor 38b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato{ 39b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato //* 40b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato private static final boolean DEBUG = false; 41b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato /*/ 42b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato private static boolean DEBUG = true; 43b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato //*/ 44b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 45b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 46b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato private static final Attribute[] EMPTY_ATTRIBUTES = new Attribute[0]; 47b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 48b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 49b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang private final ProgramClass targetClass; 50b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang// private final boolean addFields; 51b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang private final MemberVisitor extraMemberVisitor; 52b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 53b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato private final ConstantAdder constantAdder; 54b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato private final ClassEditor classEditor; 55b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato private final ConstantPoolEditor constantPoolEditor; 56b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 57b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 58b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato /** 59b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * Creates a new MemberAdder that will copy methods into the given target 60b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * class. 61b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * @param targetClass the class to which all visited class members will be 62b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * added. 63b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato */ 64b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang public MemberAdder(ProgramClass targetClass) 65b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang { 66b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang this(targetClass, null); 67b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang } 68b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang 69b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang 70b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang /** 71b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang * Creates a new MemberAdder that will copy methods into the given target 72b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang * class. 73b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang * @param targetClass the class to which all visited class members 74b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang * will be added. 75b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang * @param extraMemberVisitor an optional member visitor that visits each 76b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang * new member right after it has been added. This 77b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang * allows changing the visitor info, for instance. 78b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang */ 79b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato// * @param addFields specifies whether fields should be added, or fused 80b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato// * with the present fields. 81b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang public MemberAdder(ProgramClass targetClass, 82b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang// boolean addFields, 83b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang MemberVisitor extraMemberVisitor) 84b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 85b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang this.targetClass = targetClass; 86b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang// this.addFields = addFields; 87b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang this.extraMemberVisitor = extraMemberVisitor; 88b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 89b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato constantAdder = new ConstantAdder(targetClass); 90b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato classEditor = new ClassEditor(targetClass); 91b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato constantPoolEditor = new ConstantPoolEditor(targetClass); 92b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 93b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 94b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 95b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Implementations for MemberVisitor. 96b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 97b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato public void visitProgramField(ProgramClass programClass, ProgramField programField) 98b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 99b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang //String name = programField.getName(programClass); 100b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang //String descriptor = programField.getDescriptor(programClass); 101b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato int accessFlags = programField.getAccessFlags(); 102b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 103b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang // TODO: Handle field with the same name and descriptor in the target class. 104b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang // We currently avoid this case, since renaming the identical field 105b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang // still causes confused field references. 106b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang //// Does the target class already have such a field? 107b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang //ProgramField targetField = (ProgramField)targetClass.findField(name, descriptor); 108b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang //if (targetField != null) 109b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang //{ 110b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang // // Is the field private or static? 111b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang // int targetAccessFlags = targetField.getAccessFlags(); 112b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang // if ((targetAccessFlags & 113b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang // (ClassConstants.INTERNAL_ACC_PRIVATE | 114b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang // ClassConstants.INTERNAL_ACC_STATIC)) != 0) 115b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang // { 116b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang // if (DEBUG) 117b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang // { 118b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang // System.out.println("MemberAdder: renaming field ["+targetClass+"."+targetField.getName(targetClass)+" "+targetField.getDescriptor(targetClass)+"]"); 119b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang // } 120b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang // 121b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang // // Rename the private or static field. 122b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang // targetField.u2nameIndex = 123b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang // constantPoolEditor.addUtf8Constant(newUniqueMemberName(name, targetClass.getName())); 124b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang // } 125b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang // else 126b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang // { 127b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang // // Keep the non-private and non-static field, but update its 128b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang // // contents, in order to keep any references to it valid. 129b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang // if (DEBUG) 130b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang // { 131b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang // System.out.println("MemberAdder: updating field ["+programClass+"."+programField.getName(programClass)+" "+programField.getDescriptor(programClass)+"] into ["+targetClass.getName()+"]"); 132b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang // } 133b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang // 134b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang // // Combine the access flags. 135b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang // targetField.u2accessFlags = accessFlags | targetAccessFlags; 136b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang // 137b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang // // Add and replace any attributes. 138b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang // programField.attributesAccept(programClass, 139b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang // new AttributeAdder(targetClass, 140b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang // targetField, 141b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang // true)); 142b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang // 143b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang // // Don't add a new field. 144b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang // return; 145b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang // } 146b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang //} 147b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 148b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato if (DEBUG) 149b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 150b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato System.out.println("MemberAdder: copying field ["+programClass+"."+programField.getName(programClass)+" "+programField.getDescriptor(programClass)+"] into ["+targetClass.getName()+"]"); 151b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 152b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 153b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Create a copy of the field. 154b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato ProgramField newProgramField = 155b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato new ProgramField(accessFlags, 156b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato constantAdder.addConstant(programClass, programField.u2nameIndex), 157b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato constantAdder.addConstant(programClass, programField.u2descriptorIndex), 158b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 0, 159b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato programField.u2attributesCount > 0 ? 160b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato new Attribute[programField.u2attributesCount] : 161b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato EMPTY_ATTRIBUTES, 162b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato programField.referencedClass); 163b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 164b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Link to its visitor info. 165b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato newProgramField.setVisitorInfo(programField); 166b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 167b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Copy its attributes. 168b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato programField.attributesAccept(programClass, 169b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato new AttributeAdder(targetClass, 170b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato newProgramField, 171b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato false)); 172b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 173b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Add the completed field. 174b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato classEditor.addField(newProgramField); 175b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang 176b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang // Visit the newly added field, if necessary. 177b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang if (extraMemberVisitor != null) 178b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang { 179b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang extraMemberVisitor.visitProgramField(targetClass, newProgramField); 180b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang } 181b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 182b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 183b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 184b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato public void visitProgramMethod(ProgramClass programClass, ProgramMethod programMethod) 185b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 186b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato String name = programMethod.getName(programClass); 187b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato String descriptor = programMethod.getDescriptor(programClass); 188b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato int accessFlags = programMethod.getAccessFlags(); 189b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 190b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Does the target class already have such a method? 191b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato ProgramMethod targetMethod = (ProgramMethod)targetClass.findMethod(name, descriptor); 192b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato if (targetMethod != null) 193b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 194b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // is this source method abstract? 195b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato if ((accessFlags & ClassConstants.INTERNAL_ACC_ABSTRACT) != 0) 196b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 197b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Keep the target method. 198b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato if (DEBUG) 199b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 200b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato System.out.println("MemberAdder: skipping abstract method ["+programClass.getName()+"."+programMethod.getName(programClass)+programMethod.getDescriptor(programClass)+"] into ["+targetClass.getName()+"]"); 201b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 202b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 203b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Don't add a new method. 204b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato return; 205b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 206b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 207b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Is the target method abstract? 208b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato int targetAccessFlags = targetMethod.getAccessFlags(); 209b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato if ((targetAccessFlags & ClassConstants.INTERNAL_ACC_ABSTRACT) != 0) 210b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 211b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Keep the abstract method, but update its contents, in order 212b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // to keep any references to it valid. 213b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato if (DEBUG) 214b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 215b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato System.out.println("MemberAdder: updating method ["+programClass.getName()+"."+programMethod.getName(programClass)+programMethod.getDescriptor(programClass)+"] into ["+targetClass.getName()+"]"); 216b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 217b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 218b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Replace the access flags. 219b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato targetMethod.u2accessFlags = 220b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato accessFlags & ~ClassConstants.INTERNAL_ACC_FINAL; 221b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 222b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Add and replace the attributes. 223b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato programMethod.attributesAccept(programClass, 224b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato new AttributeAdder(targetClass, 225b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato targetMethod, 226b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato true)); 227b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 228b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Don't add a new method. 229b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato return; 230b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 231b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 232b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato if (DEBUG) 233b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 234b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato System.out.println("MemberAdder: renaming method ["+targetClass.getName()+"."+targetMethod.getName(targetClass)+targetMethod.getDescriptor(targetClass)+"]"); 235b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 236b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 237b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang // TODO: Handle non-abstract method with the same name and descriptor in the target class. 238b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang // We currently avoid this case, since renaming the identical method 239b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang // still causes confused method references. 240b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang //// Rename the private (non-abstract) or static method. 241b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang //targetMethod.u2nameIndex = 242b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang // constantPoolEditor.addUtf8Constant(newUniqueMemberName(name, descriptor)); 243b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 244b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 245b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato if (DEBUG) 246b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 247b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato System.out.println("MemberAdder: copying method ["+programClass.getName()+"."+programMethod.getName(programClass)+programMethod.getDescriptor(programClass)+"] into ["+targetClass.getName()+"]"); 248b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 249b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 250b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Create a copy of the method. 251b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato ProgramMethod newProgramMethod = 252b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato new ProgramMethod(accessFlags & ~ClassConstants.INTERNAL_ACC_FINAL, 253b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato constantAdder.addConstant(programClass, programMethod.u2nameIndex), 254b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato constantAdder.addConstant(programClass, programMethod.u2descriptorIndex), 255b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 0, 256b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato programMethod.u2attributesCount > 0 ? 257b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato new Attribute[programMethod.u2attributesCount] : 258b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato EMPTY_ATTRIBUTES, 259b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato programMethod.referencedClasses != null ? 260b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato (Clazz[])programMethod.referencedClasses.clone() : 261b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato null); 262b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 263b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Link to its visitor info. 264b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato newProgramMethod.setVisitorInfo(programMethod); 265b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 266b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Copy its attributes. 267b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato programMethod.attributesAccept(programClass, 268b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato new AttributeAdder(targetClass, 269b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato newProgramMethod, 270b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato false)); 271b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 272b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Add the completed method. 273b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato classEditor.addMethod(newProgramMethod); 274b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang 275b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang // Visit the newly added method, if necessary. 276b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang if (extraMemberVisitor != null) 277b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang { 278b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang extraMemberVisitor.visitProgramMethod(targetClass, newProgramMethod); 279b9cc48a43ed984587c939d02fba5316bf5c0df6eYing Wang } 280b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 281b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 282b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 283b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Small utility methods. 284b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 285b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato /** 286b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * Returns a unique class member name, based on the given name and descriptor. 287b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato */ 288b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato private String newUniqueMemberName(String name, String descriptor) 289b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 290b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato return name.equals(ClassConstants.INTERNAL_METHOD_NAME_INIT) ? 291b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato ClassConstants.INTERNAL_METHOD_NAME_INIT : 292b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato name + ClassConstants.SPECIAL_MEMBER_SEPARATOR + Long.toHexString(Math.abs((descriptor).hashCode())); 293b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 294b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato} 295