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.obfuscate; 22b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 23b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onoratoimport proguard.classfile.*; 24b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onoratoimport proguard.classfile.util.SimplifiedVisitor; 25b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onoratoimport proguard.classfile.visitor.MemberVisitor; 26b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 27b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onoratoimport java.util.Map; 28b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 29b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato/** 30b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * This MemberVisitor collects all new (obfuscation) names of the members 31b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * that it visits. 32b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * 33b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * @see MemberObfuscator 34b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * 35b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * @author Eric Lafortune 36b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato */ 37b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onoratopublic class MemberNameCollector 38b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onoratoextends SimplifiedVisitor 39b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onoratoimplements MemberVisitor 40b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato{ 41b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato private final boolean allowAggressiveOverloading; 42b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato private final Map descriptorMap; 43b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 44b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 45b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato /** 46b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * Creates a new MemberNameCollector. 47b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * @param allowAggressiveOverloading a flag that specifies whether class 48b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * members can be overloaded aggressively. 49b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * @param descriptorMap the map of descriptors to 50b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * [new name - old name] maps. 51b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato */ 52b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato public MemberNameCollector(boolean allowAggressiveOverloading, 53b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato Map descriptorMap) 54b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 55b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato this.allowAggressiveOverloading = allowAggressiveOverloading; 56b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato this.descriptorMap = descriptorMap; 57b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 58b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 59b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 60b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Implementations for MemberVisitor. 61b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 62b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato public void visitAnyMember(Clazz clazz, Member member) 63b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 64b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Special cases: <clinit> and <init> are always kept unchanged. 65b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // We can ignore them here. 66b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato String name = member.getName(clazz); 67b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato if (name.equals(ClassConstants.INTERNAL_METHOD_NAME_CLINIT) || 68b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato name.equals(ClassConstants.INTERNAL_METHOD_NAME_INIT)) 69b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 70b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato return; 71b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 72b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 73b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Get the member's new name. 74b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato String newName = MemberObfuscator.newMemberName(member); 75b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 76b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Remember it, if it has already been set. 77b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato if (newName != null) 78b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 79b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Get the member's descriptor. 80b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato String descriptor = member.getDescriptor(clazz); 81b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 82b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Check whether we're allowed to do aggressive overloading 83b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato if (!allowAggressiveOverloading) 84b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 85b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Trim the return argument from the descriptor if not. 86b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Works for fields and methods alike. 87b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato descriptor = descriptor.substring(0, descriptor.indexOf(')')+1); 88b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 89b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 90b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Put the [descriptor - new name] in the map, 91b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // creating a new [new name - old name] map if necessary. 92b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato Map nameMap = MemberObfuscator.retrieveNameMap(descriptorMap, descriptor); 93b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 94b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Isn't there another original name for this new name, or should 95b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // this original name get priority? 96b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato String otherName = (String)nameMap.get(newName); 97b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato if (otherName == null || 98b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato MemberObfuscator.hasFixedNewMemberName(member) || 99b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato name.compareTo(otherName) < 0) 100b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 101b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Remember not to use the new name again in this name space. 102b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato nameMap.put(newName, name); 103b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 104b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 105b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 106b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato} 107