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.io; 22b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 23b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onoratoimport proguard.classfile.*; 24b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 25b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onoratoimport java.io.*; 26b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 27b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato/** 28b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * This DataEntryReader writes the resource data entries that it reads to a 29b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * given DataEntryWriter, updating their contents based on the renamed classes 30b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * in the given ClassPool. 31b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * 32b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * @author Eric Lafortune 33b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato */ 34b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onoratopublic class DataEntryRewriter extends DataEntryCopier 35b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato{ 36b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato private final ClassPool classPool; 37b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 38b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 39b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato /** 40b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * Creates a new DataEntryRewriter. 41b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato */ 42b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato public DataEntryRewriter(ClassPool classPool, 43b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato DataEntryWriter dataEntryWriter) 44b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 45b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato super(dataEntryWriter); 46b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 47b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato this.classPool = classPool; 48b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 49b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 50b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 51b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Implementations for DataEntryCopier. 52b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 53b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato protected void copyData(InputStream inputStream, 54b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato OutputStream outputStream) 55b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato throws IOException 56b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 57b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato Reader reader = new BufferedReader(new InputStreamReader(inputStream)); 58b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato Writer writer = new BufferedWriter(new OutputStreamWriter(outputStream)); 59b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 60b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato copyData(reader, writer); 61b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 62b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato writer.flush(); 63b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato outputStream.flush(); 64b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 65b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 66b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 67b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato /** 68b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * Copies all data that it can read from the given reader to the given 69b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * writer. 70b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato */ 71b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato protected void copyData(Reader reader, 72b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato Writer writer) 73b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato throws IOException 74b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 75b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato StringBuffer word = new StringBuffer(); 76b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 77b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato while (true) 78b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 79b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato int i = reader.read(); 80b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato if (i < 0) 81b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 82b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato break; 83b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 84b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 85b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Is the character part of a word? 86b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato char c = (char)i; 87b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato if (Character.isJavaIdentifierPart(c) || 88b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato c == '.' || 89b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato c == '-') 90b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 91b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Collect the characters in this word. 92b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato word.append(c); 93b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 94b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato else 95b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 96b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Write out the updated word, if any. 97b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato writeUpdatedWord(writer, word.toString()); 98b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato word.setLength(0); 99b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 100b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Write out the character that terminated it. 101b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato writer.write(c); 102b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 103b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 104b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 105b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Write out the final word. 106b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato writeUpdatedWord(writer, word.toString()); 107b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 108b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 109b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 110b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Small utility methods. 111b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 112b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato /** 113b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * Writes the given word to the given writer, after having adapted it, 114b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * based on the renamed class names. 115b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato */ 116b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato private void writeUpdatedWord(Writer writer, String word) 117b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato throws IOException 118b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 119b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato if (word.length() > 0) 120b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 121b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato String newWord = word; 122b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 123b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato boolean containsDots = word.indexOf('.') >= 0; 124b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 125b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Replace dots by forward slashes. 126b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato String className = containsDots ? 127b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato word.replace('.', ClassConstants.INTERNAL_PACKAGE_SEPARATOR) : 128b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato word; 129b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 130b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Find the class corrsponding to the word. 131b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato Clazz clazz = classPool.getClass(className); 132b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato if (clazz != null) 133b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 134b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Update the word if necessary. 135b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato String newClassName = clazz.getName(); 136b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato if (!className.equals(newClassName)) 137b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 138b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Replace forward slashes by dots. 139b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato newWord = containsDots ? 140b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato newClassName.replace(ClassConstants.INTERNAL_PACKAGE_SEPARATOR, '.') : 141b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato newClassName; 142b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 143b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 144b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 145b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato writer.write(newWord); 146b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 147b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 148b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato} 149