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.util; 22b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 23b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onoratoimport proguard.classfile.*; 24b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onoratoimport proguard.classfile.visitor.*; 25b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 26b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onoratoimport java.util.*; 27b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 28b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato/** 299f606f95f03a75961498803e24bee6799a7c0885Ying Wang * This ClassVisitor links all corresponding non-private methods in the class 309f606f95f03a75961498803e24bee6799a7c0885Ying Wang * hierarchies of all visited classes. Visited classes are typically all class 319f606f95f03a75961498803e24bee6799a7c0885Ying Wang * files that are not being subclassed. Chains of links that have been created 329f606f95f03a75961498803e24bee6799a7c0885Ying Wang * in previous invocations are merged with new chains of links, in order to 339f606f95f03a75961498803e24bee6799a7c0885Ying Wang * create a consistent set of chains. 349f606f95f03a75961498803e24bee6799a7c0885Ying Wang * <p> 359f606f95f03a75961498803e24bee6799a7c0885Ying Wang * As a MemberVisitor, it links all corresponding class members that it visits, 369f606f95f03a75961498803e24bee6799a7c0885Ying Wang * including fields and private class members. 379f606f95f03a75961498803e24bee6799a7c0885Ying Wang * <p> 389f606f95f03a75961498803e24bee6799a7c0885Ying Wang * Class initialization methods and constructors are always ignored. 39b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * 40b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * @author Eric Lafortune 41b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato */ 42b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onoratopublic class MethodLinker 43b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onoratoextends SimplifiedVisitor 44b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onoratoimplements ClassVisitor, 45b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato MemberVisitor 46b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato{ 47b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // An object that is reset and reused every time. 48b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // The map: [class member name+' '+descriptor - class member info] 49b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato private final Map memberMap = new HashMap(); 50b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 51b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 52b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Implementations for ClassVisitor. 53b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 54b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato public void visitAnyClass(Clazz clazz) 55b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 56b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Collect all non-private members in this class hierarchy. 57b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato clazz.hierarchyAccept(true, true, true, false, 58b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato new AllMethodVisitor( 599f606f95f03a75961498803e24bee6799a7c0885Ying Wang new MemberAccessFilter(0, ClassConstants.INTERNAL_ACC_PRIVATE, 60b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato this))); 61b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 62b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Clean up for the next class hierarchy. 63b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato memberMap.clear(); 64b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 65b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 66b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 67b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Implementations for MemberVisitor. 68b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 69b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato public void visitAnyMember(Clazz clazz, Member member) 70b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 71b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Get the class member's name and descriptor. 72b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato String name = member.getName(clazz); 73b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato String descriptor = member.getDescriptor(clazz); 74b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 75b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Special cases: <clinit> and <init> are always kept unchanged. 76b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // We can ignore them here. 77b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato if (name.equals(ClassConstants.INTERNAL_METHOD_NAME_CLINIT) || 78b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato name.equals(ClassConstants.INTERNAL_METHOD_NAME_INIT)) 79b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 80b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato return; 81b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 82b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 83b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // See if we've already come across a method with the same name and 84b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // descriptor. 85b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato String key = name + ' ' + descriptor; 86b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato Member otherMember = (Member)memberMap.get(key); 87b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 88b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato if (otherMember == null) 89b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 90b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Get the last method in the chain. 91b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato Member thisLastMember = lastMember(member); 92b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 93b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Store the new class method in the map. 94b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato memberMap.put(key, thisLastMember); 95b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 96b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato else 97b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 98b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Link both members. 99b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato link(member, otherMember); 100b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 101b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 102b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 103b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 104b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Small utility methods. 105b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 106b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato /** 107b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * Links the two given class members. 108b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato */ 109b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato private static void link(Member member1, Member member2) 110b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 111b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Get the last methods in the both chains. 112b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato Member lastMember1 = lastMember(member1); 113b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato Member lastMember2 = lastMember(member2); 114b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 115b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Check if both link chains aren't already ending in the same element. 116b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato if (!lastMember1.equals(lastMember2)) 117b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 118b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Merge the two chains, with the library members last. 119b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato if (lastMember2 instanceof LibraryMember) 120b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 121b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato lastMember1.setVisitorInfo(lastMember2); 122b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 123b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato else 124b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 125b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato lastMember2.setVisitorInfo(lastMember1); 126b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 127b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 128b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 129b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 130b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 131b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato /** 132b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * Finds the last class member in the linked list of related class members. 133b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * @param member the given class member. 134b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * @return the last class member in the linked list. 135b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato */ 136b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato public static Member lastMember(Member member) 137b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 138b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato Member lastMember = member; 139b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato while (lastMember.getVisitorInfo() != null && 140b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato lastMember.getVisitorInfo() instanceof Member) 141b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 142b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato lastMember = (Member)lastMember.getVisitorInfo(); 143b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 144b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 145b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato return lastMember; 146b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 147b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 148b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 149b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato /** 150b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * Finds the last visitor accepter in the linked list of visitors. 151b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * @param visitorAccepter the given method. 152b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * @return the last method in the linked list. 153b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato */ 154b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato public static VisitorAccepter lastVisitorAccepter(VisitorAccepter visitorAccepter) 155b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 156b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato VisitorAccepter lastVisitorAccepter = visitorAccepter; 157b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato while (lastVisitorAccepter.getVisitorInfo() != null && 158b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato lastVisitorAccepter.getVisitorInfo() instanceof VisitorAccepter) 159b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 160b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato lastVisitorAccepter = (VisitorAccepter)lastVisitorAccepter.getVisitorInfo(); 161b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 162b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 163b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato return lastVisitorAccepter; 164b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 165b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato} 166