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.io; 22b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 23b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onoratoimport proguard.classfile.ClassConstants; 24b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 25b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onoratoimport java.io.*; 26b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 27b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 28b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato/** 29b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * This DataEntryWriter writes data entries to individual files in a given 30b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * directory. 31b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * 32b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * @author Eric Lafortune 33b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato */ 34b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onoratopublic class DirectoryWriter implements DataEntryWriter 35b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato{ 36b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato private final File baseFile; 37b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato private final boolean isFile; 38b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 39b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato private File currentFile; 40b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato private OutputStream currentOutputStream; 41b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato private Finisher currentFinisher; 42b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 43b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 44b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato /** 45b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * Creates a new DirectoryWriter. 46b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * @param baseFile the base directory to which all files will be written. 47b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato */ 48b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato public DirectoryWriter(File baseFile, 49b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato boolean isFile) 50b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 51b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato this.baseFile = baseFile; 52b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato this.isFile = isFile; 53b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 54b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 55b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 56b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Implementations for DataEntryWriter. 57b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 58b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato public boolean createDirectory(DataEntry dataEntry) throws IOException 59b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 60b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Should we close the current file? 61b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato if (!isFile && 62b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato currentFile != null) 63b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 64b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato closeEntry(); 65b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 66b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 67b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato File directory = getFile(dataEntry); 68b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato if (!directory.exists() && 69b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato !directory.mkdirs()) 70b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 71b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato throw new IOException("Can't create directory [" + directory.getPath() + "]"); 72b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 73b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 74b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato return true; 75b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 76b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 77b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 78b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato public OutputStream getOutputStream(DataEntry dataEntry) throws IOException 79b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 80b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato return getOutputStream(dataEntry, null); 81b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 82b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 83b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 84b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato public OutputStream getOutputStream(DataEntry dataEntry, 85b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato Finisher finisher) throws IOException 86b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 87b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato File file = getFile(dataEntry); 88b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 89b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Should we close the current file? 90b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato if (!isFile && 91b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato currentFile != null && 92b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato !currentFile.equals(file)) 93b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 94b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato closeEntry(); 95b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 96b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 97b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Do we need a new stream? 98b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato if (currentOutputStream == null) 99b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 100b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Make sure the parent directories exist. 101b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato File parentDirectory = file.getParentFile(); 102b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato if (parentDirectory != null && 103b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato !parentDirectory.exists() && 104b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato !parentDirectory.mkdirs()) 105b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 106b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato throw new IOException("Can't create directory [" + parentDirectory.getPath() + "]"); 107b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 108b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 109b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Open a new output stream for writing to the file. 110b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato currentOutputStream = 111b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato new BufferedOutputStream( 112b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato new FileOutputStream(file)); 113b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 114b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato currentFinisher = finisher; 115b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato currentFile = file; 116b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 117b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 118b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato return currentOutputStream; 119b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 120b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 121b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 122b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato public void close() throws IOException 123b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 124b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Close the file stream, if any. 125b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato closeEntry(); 126b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 127b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 128b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 129b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Small utility methods. 130b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 131b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato /** 132b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * Returns the file for the given data entry. 133b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato */ 134b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato private File getFile(DataEntry dataEntry) 135b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 136b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Use the specified file, or construct a new file. 137b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato return isFile ? 138b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato baseFile : 139b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato new File(baseFile, 140b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato dataEntry.getName().replace(ClassConstants.INTERNAL_PACKAGE_SEPARATOR, 141b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato File.separatorChar)); 142b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 143b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 144b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 145b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato /** 146b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato * Closes the previous file, if any. 147b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato */ 148b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato private void closeEntry() throws IOException 149b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 150b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Close the file stream, if any. 151b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato if (currentOutputStream != null) 152b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 153b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato // Let any finisher finish up first. 154b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato if (currentFinisher != null) 155b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato { 156b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato currentFinisher.finish(); 157b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato currentFinisher = null; 158b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 159b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato 160b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato currentOutputStream.close(); 161b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato currentOutputStream = null; 162b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato currentFile = null; 163b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 164b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato } 165b72c5c2e5482cf10117b2b25f642f7616b2326c3Joe Onorato} 166