1b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato/* 2b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * ProGuard -- shrinking, optimization, obfuscation, and preverification 3b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * of Java bytecode. 4b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * 5b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * 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 Onoratoimport proguard.classfile.util.*; 27b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onoratoimport proguard.classfile.visitor.*; 28b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 29b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato/** 30b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * This ConstantVisitor fixes the access modifiers of all classes and class 31b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * members that are referenced by the constants that it visits. 32b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * 33b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * @author Eric Lafortune 34b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato */ 35b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onoratopublic class AccessFixer 36b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onoratoextends SimplifiedVisitor 37b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onoratoimplements ConstantVisitor, 38b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato ClassVisitor, 39b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato MemberVisitor 40b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato{ 41b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato private MyReferencedClassFinder referencedClassFinder = new MyReferencedClassFinder(); 42b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 43b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato private Clazz referencingClass; 44b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato private Clazz referencedClass; 45b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 46b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 47b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Implementations for ConstantVisitor. 48b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 49b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato public void visitAnyConstant(Clazz clazz, Constant constant) {} 50b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 51b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 52b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato public void visitStringConstant(Clazz clazz, StringConstant stringConstant) 53b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 54b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato referencingClass = clazz; 55b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato referencedClass = stringConstant.referencedClass; 56b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 57b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Make sure the access flags of the referenced class or class member, 58b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // if any, are acceptable. 59b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato stringConstant.referencedClassAccept(this); 60b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato stringConstant.referencedMemberAccept(this); 61b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 62b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 63b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 64b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato public void visitAnyRefConstant(Clazz clazz, RefConstant refConstant) 65b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 66b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato referencingClass = clazz; 67b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 68b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Remember the specified class, since it might be different from 69b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // the referenced class that acutally contains the class member. 70b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato clazz.constantPoolEntryAccept(refConstant.u2classIndex, referencedClassFinder); 71b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 72b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Make sure the access flags of the referenced class member are 73b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // acceptable. 74b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato refConstant.referencedMemberAccept(this); 75b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 76b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 77b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 78b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato public void visitClassConstant(Clazz clazz, ClassConstant classConstant) 79b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 80b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato referencingClass = clazz; 81b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 82b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Make sure the access flags of the referenced class are acceptable. 83b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato classConstant.referencedClassAccept(this); 84b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 85b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 86b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 87b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Implementations for ClassVisitor. 88b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 89b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato public void visitLibraryClass(LibraryClass libraryClass) {} 90b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 91b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 92b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato public void visitProgramClass(ProgramClass programClass) 93b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 94b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato int currentAccessFlags = programClass.getAccessFlags(); 95b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato int currentAccessLevel = AccessUtil.accessLevel(currentAccessFlags); 96b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 97b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Compute the required access level. 98b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato Clazz referencingClass = this.referencingClass; 99b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato int requiredAccessLevel = 100b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato inSamePackage(programClass, referencingClass) ? AccessUtil.PACKAGE_VISIBLE : 101b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato AccessUtil.PUBLIC; 102b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 103b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Fix the class access flags if necessary. 104b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato if (currentAccessLevel < requiredAccessLevel) 105b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 106b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato programClass.u2accessFlags = 107b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato AccessUtil.replaceAccessFlags(currentAccessFlags, 108b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato AccessUtil.accessFlags(requiredAccessLevel)); 109b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 110b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 111b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 112b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 113b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Implementations for MemberVisitor. 114b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 115b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato public void visitLibraryMember(LibraryClass libraryClass, LibraryMember libraryMember) {} 116b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 117b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 118b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato public void visitProgramMember(ProgramClass programClass, ProgramMember programMember) 119b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 120b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato int currentAccessFlags = programMember.getAccessFlags(); 121b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato int currentAccessLevel = AccessUtil.accessLevel(currentAccessFlags); 122b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 123b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Compute the required access level. 124b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato int requiredAccessLevel = 125b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato programClass.equals(referencingClass) ? AccessUtil.PRIVATE : 126b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato inSamePackage(programClass, referencingClass) ? AccessUtil.PACKAGE_VISIBLE : 127b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato referencedClass.extends_(referencingClass) && 128b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato referencingClass.extends_(programClass) ? AccessUtil.PROTECTED : 129b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato AccessUtil.PUBLIC; 130b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 131b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Fix the class member access flags if necessary. 132b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato if (currentAccessLevel < requiredAccessLevel) 133b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 134b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato programMember.u2accessFlags = 135b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato AccessUtil.replaceAccessFlags(currentAccessFlags, 136b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato AccessUtil.accessFlags(requiredAccessLevel)); 137b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 138b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 139b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 140b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 141b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato /** 142b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * This ConstantVisitor returns the referenced class of the class constant 143b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * that it visits. 144b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato */ 145b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato private class MyReferencedClassFinder 146b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato extends SimplifiedVisitor 147b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato implements ConstantVisitor 148b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 149b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Implementations for ConstantVisitor. 150b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato public void visitClassConstant(Clazz clazz, ClassConstant classConstant) 151b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 152b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato referencedClass = classConstant.referencedClass; 153b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 154b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 155b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 156b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 157b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Small utility methods. 158b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 159b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato private boolean inSamePackage(ProgramClass class1, Clazz class2) 160b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 161b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato return ClassUtil.internalPackageName(class1.getName()).equals( 162b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato ClassUtil.internalPackageName(class2.getName())); 163b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 164b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato} 165