src/share/classes/com/sun/tools/javac/zip/ZipFileIndex.java

Thu, 22 May 2008 15:51:41 -0700

author
jjg
date
Thu, 22 May 2008 15:51:41 -0700
changeset 36
58e352559a41
parent 1
9a66ca7c79fa
permissions
-rw-r--r--

6705945: com.sun.tools.javac.zip files do not have valid copyright
Reviewed-by: mcimadamore

jjg@36 1 /*
jjg@36 2 * Copyright 2007-2008 Sun Microsystems, Inc. All Rights Reserved.
jjg@36 3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
jjg@36 4 *
jjg@36 5 * This code is free software; you can redistribute it and/or modify it
jjg@36 6 * under the terms of the GNU General Public License version 2 only, as
jjg@36 7 * published by the Free Software Foundation. Sun designates this
jjg@36 8 * particular file as subject to the "Classpath" exception as provided
jjg@36 9 * by Sun in the LICENSE file that accompanied this code.
jjg@36 10 *
jjg@36 11 * This code is distributed in the hope that it will be useful, but WITHOUT
jjg@36 12 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
jjg@36 13 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
jjg@36 14 * version 2 for more details (a copy is included in the LICENSE file that
jjg@36 15 * accompanied this code).
jjg@36 16 *
jjg@36 17 * You should have received a copy of the GNU General Public License version
jjg@36 18 * 2 along with this work; if not, write to the Free Software Foundation,
jjg@36 19 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
jjg@36 20 *
jjg@36 21 * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
jjg@36 22 * CA 95054 USA or visit www.sun.com if you need additional information or
jjg@36 23 * have any questions.
jjg@36 24 */
jjg@36 25
duke@1 26 package com.sun.tools.javac.zip;
duke@1 27
duke@1 28 import java.io.*;
duke@1 29 import java.text.MessageFormat;
duke@1 30 import java.util.*;
duke@1 31 import java.util.List;
duke@1 32 import java.util.concurrent.locks.ReentrantLock;
duke@1 33 import java.util.zip.*;
duke@1 34
duke@1 35 /** This class implements building of index of a zip archive and access to it's context.
duke@1 36 * It also uses prebuild index if available. It supports invocations where it will
duke@1 37 * serialize an optimized zip index file to disk.
duke@1 38 *
duke@1 39 * In oreder to use secondary index file make sure the option "usezipindex" is in the Options object,
duke@1 40 * when JavacFileManager is invoked. (You can pass "-XDusezipindex" on the command line.
duke@1 41 *
duke@1 42 * Location where to look for/generate optimized zip index files can be provided using
duke@1 43 * "-XDcachezipindexdir=<directory>". If this flag is not provided, the dfault location is
duke@1 44 * the value of the "java.io.tmpdir" system property.
duke@1 45 *
duke@1 46 * If key "-XDwritezipindexfiles" is specified, there will be new optimized index file
duke@1 47 * created for each archive, used by the compiler for compilation, at location,
duke@1 48 * specified by "cachezipindexdir" option.
duke@1 49 *
duke@1 50 * If nonBatchMode option is specified (-XDnonBatchMode) the compiler will use timestamp
duke@1 51 * checking to reindex the zip files if it is needed. In batch mode the timestamps are not checked
duke@1 52 * and the compiler uses the cached indexes.
duke@1 53 */
duke@1 54 public class ZipFileIndex {
duke@1 55 private static final String MIN_CHAR = String.valueOf(Character.MIN_VALUE);
duke@1 56 private static final String MAX_CHAR = String.valueOf(Character.MAX_VALUE);
duke@1 57
duke@1 58 public final static long NOT_MODIFIED = Long.MIN_VALUE;
duke@1 59
duke@1 60 private static Map<File, ZipFileIndex> zipFileIndexCache = new HashMap<File, ZipFileIndex>();
duke@1 61 private static ReentrantLock lock = new ReentrantLock();
duke@1 62
duke@1 63 private static boolean NON_BATCH_MODE = System.getProperty("nonBatchMode") != null;// TODO: Use -XD compiler switch for this.
duke@1 64
duke@1 65 private Map<String, DirectoryEntry> directories = Collections.<String, DirectoryEntry>emptyMap();
duke@1 66 private Set<String> allDirs = Collections.<String>emptySet();
duke@1 67
duke@1 68 // ZipFileIndex data entries
duke@1 69 private File zipFile;
duke@1 70 private long zipFileLastModified = NOT_MODIFIED;
duke@1 71 private RandomAccessFile zipRandomFile;
duke@1 72 private ZipFileIndexEntry[] entries;
duke@1 73
duke@1 74 private boolean readFromIndex = false;
duke@1 75 private File zipIndexFile = null;
duke@1 76 private boolean triedToReadIndex = false;
duke@1 77 private int symbolFilePrefixLength = 0;
duke@1 78 private boolean hasPopulatedData = false;
duke@1 79 private long lastReferenceTimeStamp = NOT_MODIFIED;
duke@1 80
duke@1 81 private boolean usePreindexedCache = false;
duke@1 82 private String preindexedCacheLocation = null;
duke@1 83
duke@1 84 private boolean writeIndex = false;
duke@1 85
duke@1 86 /**
duke@1 87 * Returns a list of all ZipFileIndex entries
duke@1 88 *
duke@1 89 * @return A list of ZipFileIndex entries, or an empty list
duke@1 90 */
duke@1 91 public static List<ZipFileIndex> getZipFileIndexes() {
duke@1 92 return getZipFileIndexes(false);
duke@1 93 }
duke@1 94
duke@1 95 /**
duke@1 96 * Returns a list of all ZipFileIndex entries
duke@1 97 *
duke@1 98 * @param openedOnly If true it returns a list of only opened ZipFileIndex entries, otherwise
duke@1 99 * all ZipFileEntry(s) are included into the list.
duke@1 100 * @return A list of ZipFileIndex entries, or an empty list
duke@1 101 */
duke@1 102 public static List<ZipFileIndex> getZipFileIndexes(boolean openedOnly) {
duke@1 103 List<ZipFileIndex> zipFileIndexes = new ArrayList<ZipFileIndex>();
duke@1 104 lock.lock();
duke@1 105 try {
duke@1 106 zipFileIndexes.addAll(zipFileIndexCache.values());
duke@1 107
duke@1 108 if (openedOnly) {
duke@1 109 for(ZipFileIndex elem : zipFileIndexes) {
duke@1 110 if (!elem.isOpen()) {
duke@1 111 zipFileIndexes.remove(elem);
duke@1 112 }
duke@1 113 }
duke@1 114 }
duke@1 115 }
duke@1 116 finally {
duke@1 117 lock.unlock();
duke@1 118 }
duke@1 119 return zipFileIndexes;
duke@1 120 }
duke@1 121
duke@1 122 public boolean isOpen() {
duke@1 123 lock.lock();
duke@1 124 try {
duke@1 125 return zipRandomFile != null;
duke@1 126 }
duke@1 127 finally {
duke@1 128 lock.unlock();
duke@1 129 }
duke@1 130 }
duke@1 131
duke@1 132 public static ZipFileIndex getZipFileIndex(File zipFile, int symbolFilePrefixLen, boolean useCache, String cacheLocation, boolean writeIndex) throws IOException {
duke@1 133 ZipFileIndex zi = null;
duke@1 134 lock.lock();
duke@1 135 try {
duke@1 136 zi = getExistingZipIndex(zipFile);
duke@1 137
duke@1 138 if (zi == null || (zi != null && zipFile.lastModified() != zi.zipFileLastModified)) {
duke@1 139 zi = new ZipFileIndex(zipFile, symbolFilePrefixLen, writeIndex,
duke@1 140 useCache, cacheLocation);
duke@1 141 zipFileIndexCache.put(zipFile, zi);
duke@1 142 }
duke@1 143 }
duke@1 144 finally {
duke@1 145 lock.unlock();
duke@1 146 }
duke@1 147 return zi;
duke@1 148 }
duke@1 149
duke@1 150 public static ZipFileIndex getExistingZipIndex(File zipFile) {
duke@1 151 lock.lock();
duke@1 152 try {
duke@1 153 return zipFileIndexCache.get(zipFile);
duke@1 154 }
duke@1 155 finally {
duke@1 156 lock.unlock();
duke@1 157 }
duke@1 158 }
duke@1 159
duke@1 160 public static void clearCache() {
duke@1 161 lock.lock();
duke@1 162 try {
duke@1 163 zipFileIndexCache.clear();
duke@1 164 }
duke@1 165 finally {
duke@1 166 lock.unlock();
duke@1 167 }
duke@1 168 }
duke@1 169
duke@1 170 public static void clearCache(long timeNotUsed) {
duke@1 171 lock.lock();
duke@1 172 try {
duke@1 173 Iterator<File> cachedFileIterator = zipFileIndexCache.keySet().iterator();
duke@1 174 while (cachedFileIterator.hasNext()) {
duke@1 175 File cachedFile = cachedFileIterator.next();
duke@1 176 ZipFileIndex cachedZipIndex = zipFileIndexCache.get(cachedFile);
duke@1 177 if (cachedZipIndex != null) {
duke@1 178 long timeToTest = cachedZipIndex.lastReferenceTimeStamp + timeNotUsed;
duke@1 179 if (timeToTest < cachedZipIndex.lastReferenceTimeStamp || // Overflow...
duke@1 180 System.currentTimeMillis() > timeToTest) {
duke@1 181 zipFileIndexCache.remove(cachedFile);
duke@1 182 }
duke@1 183 }
duke@1 184 }
duke@1 185 }
duke@1 186 finally {
duke@1 187 lock.unlock();
duke@1 188 }
duke@1 189 }
duke@1 190
duke@1 191 public static void removeFromCache(File file) {
duke@1 192 lock.lock();
duke@1 193 try {
duke@1 194 zipFileIndexCache.remove(file);
duke@1 195 }
duke@1 196 finally {
duke@1 197 lock.unlock();
duke@1 198 }
duke@1 199 }
duke@1 200
duke@1 201 /** Sets already opened list of ZipFileIndexes from an outside client
duke@1 202 * of the compiler. This functionality should be used in a non-batch clients of the compiler.
duke@1 203 */
duke@1 204 public static void setOpenedIndexes(List<ZipFileIndex>indexes) throws IllegalStateException {
duke@1 205 lock.lock();
duke@1 206 try {
duke@1 207 if (zipFileIndexCache.isEmpty()) {
duke@1 208 throw new IllegalStateException("Setting opened indexes should be called only when the ZipFileCache is empty. Call JavacFileManager.flush() before calling this method.");
duke@1 209 }
duke@1 210
duke@1 211 for (ZipFileIndex zfi : indexes) {
duke@1 212 zipFileIndexCache.put(zfi.zipFile, zfi);
duke@1 213 }
duke@1 214 }
duke@1 215 finally {
duke@1 216 lock.unlock();
duke@1 217 }
duke@1 218 }
duke@1 219
duke@1 220 private ZipFileIndex(File zipFile, int symbolFilePrefixLen, boolean writeIndex,
duke@1 221 boolean useCache, String cacheLocation) throws IOException {
duke@1 222 this.zipFile = zipFile;
duke@1 223 this.symbolFilePrefixLength = symbolFilePrefixLen;
duke@1 224 this.writeIndex = writeIndex;
duke@1 225 this.usePreindexedCache = useCache;
duke@1 226 this.preindexedCacheLocation = cacheLocation;
duke@1 227
duke@1 228 if (zipFile != null) {
duke@1 229 this.zipFileLastModified = zipFile.lastModified();
duke@1 230 }
duke@1 231
duke@1 232 // Validate integrity of the zip file
duke@1 233 checkIndex();
duke@1 234 }
duke@1 235
duke@1 236 public String toString() {
duke@1 237 return "ZipFileIndex of file:(" + zipFile + ")";
duke@1 238 }
duke@1 239
duke@1 240 // Just in case...
duke@1 241 protected void finalize() {
duke@1 242 closeFile();
duke@1 243 }
duke@1 244
duke@1 245 private boolean isUpToDate() {
duke@1 246 if (zipFile != null &&
duke@1 247 ((!NON_BATCH_MODE) || zipFileLastModified == zipFile.lastModified()) &&
duke@1 248 hasPopulatedData) {
duke@1 249 return true;
duke@1 250 }
duke@1 251
duke@1 252 return false;
duke@1 253 }
duke@1 254
duke@1 255 /**
duke@1 256 * Here we need to make sure that the ZipFileIndex is valid. Check the timestamp of the file and
duke@1 257 * if its the same as the one at the time the index was build we don't need to reopen anything.
duke@1 258 */
duke@1 259 private void checkIndex() throws IOException {
duke@1 260 boolean isUpToDate = true;
duke@1 261 if (!isUpToDate()) {
duke@1 262 closeFile();
duke@1 263 isUpToDate = false;
duke@1 264 }
duke@1 265
duke@1 266 if (zipRandomFile != null || isUpToDate) {
duke@1 267 lastReferenceTimeStamp = System.currentTimeMillis();
duke@1 268 return;
duke@1 269 }
duke@1 270
duke@1 271 hasPopulatedData = true;
duke@1 272
duke@1 273 if (readIndex()) {
duke@1 274 lastReferenceTimeStamp = System.currentTimeMillis();
duke@1 275 return;
duke@1 276 }
duke@1 277
duke@1 278 directories = Collections.<String, DirectoryEntry>emptyMap();
duke@1 279 allDirs = Collections.<String>emptySet();
duke@1 280
duke@1 281 try {
duke@1 282 openFile();
duke@1 283 long totalLength = zipRandomFile.length();
duke@1 284 ZipDirectory directory = new ZipDirectory(zipRandomFile, 0L, totalLength, this);
duke@1 285 directory.buildIndex();
duke@1 286 } finally {
duke@1 287 if (zipRandomFile != null) {
duke@1 288 closeFile();
duke@1 289 }
duke@1 290 }
duke@1 291
duke@1 292 lastReferenceTimeStamp = System.currentTimeMillis();
duke@1 293 }
duke@1 294
duke@1 295 private void openFile() throws FileNotFoundException {
duke@1 296 if (zipRandomFile == null && zipFile != null) {
duke@1 297 zipRandomFile = new RandomAccessFile(zipFile, "r");
duke@1 298 }
duke@1 299 }
duke@1 300
duke@1 301 private void cleanupState() {
duke@1 302 // Make sure there is a valid but empty index if the file doesn't exist
duke@1 303 entries = ZipFileIndexEntry.EMPTY_ARRAY;
duke@1 304 directories = Collections.<String, DirectoryEntry>emptyMap();
duke@1 305 zipFileLastModified = NOT_MODIFIED;
duke@1 306 allDirs = Collections.<String>emptySet();
duke@1 307 }
duke@1 308
duke@1 309 public void close() {
duke@1 310 lock.lock();
duke@1 311 try {
duke@1 312 writeIndex();
duke@1 313 closeFile();
duke@1 314 }
duke@1 315 finally {
duke@1 316 lock.unlock();
duke@1 317 }
duke@1 318 }
duke@1 319
duke@1 320 private void closeFile() {
duke@1 321 if (zipRandomFile != null) {
duke@1 322 try {
duke@1 323 zipRandomFile.close();
duke@1 324 } catch (IOException ex) {
duke@1 325 }
duke@1 326 zipRandomFile = null;
duke@1 327 }
duke@1 328 }
duke@1 329
duke@1 330 /**
duke@1 331 * Returns the ZipFileIndexEntry for an absolute path, if there is one.
duke@1 332 */
duke@1 333 public ZipFileIndexEntry getZipIndexEntry(String path) {
duke@1 334 if (File.separatorChar != '/') {
duke@1 335 path = path.replace('/', File.separatorChar);
duke@1 336 }
duke@1 337 lock.lock();
duke@1 338 try {
duke@1 339 checkIndex();
duke@1 340 String lookFor = "";
duke@1 341 int lastSepIndex = path.lastIndexOf(File.separatorChar);
duke@1 342 boolean noSeparator = false;
duke@1 343 if (lastSepIndex == -1) {
duke@1 344 noSeparator = true;
duke@1 345 }
duke@1 346
duke@1 347 DirectoryEntry de = directories.get(noSeparator ? "" : path.substring(0, lastSepIndex));
duke@1 348
duke@1 349 lookFor = path.substring(noSeparator ? 0 : lastSepIndex + 1);
duke@1 350
duke@1 351 return de == null ? null : de.getEntry(lookFor);
duke@1 352 }
duke@1 353 catch (IOException e) {
duke@1 354 return null;
duke@1 355 }
duke@1 356 finally {
duke@1 357 lock.unlock();
duke@1 358 }
duke@1 359 }
duke@1 360
duke@1 361 /**
duke@1 362 * Returns a javac List of filenames within an absolute path in the ZipFileIndex.
duke@1 363 */
duke@1 364 public com.sun.tools.javac.util.List<String> getFiles(String path) {
duke@1 365 if (File.separatorChar != '/') {
duke@1 366 path = path.replace('/', File.separatorChar);
duke@1 367 }
duke@1 368
duke@1 369 lock.lock();
duke@1 370 try {
duke@1 371 checkIndex();
duke@1 372
duke@1 373 DirectoryEntry de = directories.get(path);
duke@1 374 com.sun.tools.javac.util.List<String> ret = de == null ? null : de.getFiles();
duke@1 375
duke@1 376 if (ret == null) {
duke@1 377 return com.sun.tools.javac.util.List.<String>nil();
duke@1 378 }
duke@1 379 return ret;
duke@1 380 }
duke@1 381 catch (IOException e) {
duke@1 382 return com.sun.tools.javac.util.List.<String>nil();
duke@1 383 }
duke@1 384 finally {
duke@1 385 lock.unlock();
duke@1 386 }
duke@1 387 }
duke@1 388
duke@1 389 public List<String> getAllDirectories(String path) {
duke@1 390
duke@1 391 if (File.separatorChar != '/') {
duke@1 392 path = path.replace('/', File.separatorChar);
duke@1 393 }
duke@1 394
duke@1 395 lock.lock();
duke@1 396 try {
duke@1 397 checkIndex();
duke@1 398 path = path.intern();
duke@1 399
duke@1 400 DirectoryEntry de = directories.get(path);
duke@1 401 com.sun.tools.javac.util.List<String> ret = de == null ? null : de.getDirectories();
duke@1 402
duke@1 403 if (ret == null) {
duke@1 404 return com.sun.tools.javac.util.List.<String>nil();
duke@1 405 }
duke@1 406
duke@1 407 return ret;
duke@1 408 }
duke@1 409 catch (IOException e) {
duke@1 410 return com.sun.tools.javac.util.List.<String>nil();
duke@1 411 }
duke@1 412 finally {
duke@1 413 lock.unlock();
duke@1 414 }
duke@1 415 }
duke@1 416
duke@1 417 public Set<String> getAllDirectories() {
duke@1 418 lock.lock();
duke@1 419 try {
duke@1 420 checkIndex();
duke@1 421 if (allDirs == Collections.EMPTY_SET) {
duke@1 422 Set<String> alldirs = new HashSet<String>();
duke@1 423 Iterator<String> dirsIter = directories.keySet().iterator();
duke@1 424 while (dirsIter.hasNext()) {
duke@1 425 alldirs.add(new String(dirsIter.next()));
duke@1 426 }
duke@1 427
duke@1 428 allDirs = alldirs;
duke@1 429 }
duke@1 430
duke@1 431 return allDirs;
duke@1 432 }
duke@1 433 catch (IOException e) {
duke@1 434 return Collections.<String>emptySet();
duke@1 435 }
duke@1 436 finally {
duke@1 437 lock.unlock();
duke@1 438 }
duke@1 439 }
duke@1 440
duke@1 441 /**
duke@1 442 * Tests if a specific path exists in the zip. This method will return true
duke@1 443 * for file entries and directories.
duke@1 444 *
duke@1 445 * @param path A path within the zip.
duke@1 446 * @return True if the path is a file or dir, false otherwise.
duke@1 447 */
duke@1 448 public boolean contains(String path) {
duke@1 449 lock.lock();
duke@1 450 try {
duke@1 451 checkIndex();
duke@1 452 return getZipIndexEntry(path) != null;
duke@1 453 }
duke@1 454 catch (IOException e) {
duke@1 455 return false;
duke@1 456 }
duke@1 457 finally {
duke@1 458 lock.unlock();
duke@1 459 }
duke@1 460 }
duke@1 461
duke@1 462 public boolean isDirectory(String path) throws IOException {
duke@1 463 lock.lock();
duke@1 464 try {
duke@1 465 // The top level in a zip file is always a directory.
duke@1 466 if (path.length() == 0) {
duke@1 467 lastReferenceTimeStamp = System.currentTimeMillis();
duke@1 468 return true;
duke@1 469 }
duke@1 470
duke@1 471 if (File.separatorChar != '/')
duke@1 472 path = path.replace('/', File.separatorChar);
duke@1 473 checkIndex();
duke@1 474 return directories.get(path) != null;
duke@1 475 }
duke@1 476 finally {
duke@1 477 lock.unlock();
duke@1 478 }
duke@1 479 }
duke@1 480
duke@1 481 public long getLastModified(String path) throws IOException {
duke@1 482 lock.lock();
duke@1 483 try {
duke@1 484 ZipFileIndexEntry entry = getZipIndexEntry(path);
duke@1 485 if (entry == null)
duke@1 486 throw new FileNotFoundException();
duke@1 487 return entry.getLastModified();
duke@1 488 }
duke@1 489 finally {
duke@1 490 lock.unlock();
duke@1 491 }
duke@1 492 }
duke@1 493
duke@1 494 public int length(String path) throws IOException {
duke@1 495 lock.lock();
duke@1 496 try {
duke@1 497 ZipFileIndexEntry entry = getZipIndexEntry(path);
duke@1 498 if (entry == null)
duke@1 499 throw new FileNotFoundException();
duke@1 500
duke@1 501 if (entry.isDir) {
duke@1 502 return 0;
duke@1 503 }
duke@1 504
duke@1 505 byte[] header = getHeader(entry);
duke@1 506 // entry is not compressed?
duke@1 507 if (get2ByteLittleEndian(header, 8) == 0) {
duke@1 508 return entry.compressedSize;
duke@1 509 } else {
duke@1 510 return entry.size;
duke@1 511 }
duke@1 512 }
duke@1 513 finally {
duke@1 514 lock.unlock();
duke@1 515 }
duke@1 516 }
duke@1 517
duke@1 518 public byte[] read(String path) throws IOException {
duke@1 519 lock.lock();
duke@1 520 try {
duke@1 521 ZipFileIndexEntry entry = getZipIndexEntry(path);
duke@1 522 if (entry == null)
duke@1 523 throw new FileNotFoundException(MessageFormat.format("Path not found in ZIP: {0}", path));
duke@1 524 return read(entry);
duke@1 525 }
duke@1 526 finally {
duke@1 527 lock.unlock();
duke@1 528 }
duke@1 529 }
duke@1 530
duke@1 531 public byte[] read(ZipFileIndexEntry entry) throws IOException {
duke@1 532 lock.lock();
duke@1 533 try {
duke@1 534 openFile();
duke@1 535 byte[] result = readBytes(entry);
duke@1 536 closeFile();
duke@1 537 return result;
duke@1 538 }
duke@1 539 finally {
duke@1 540 lock.unlock();
duke@1 541 }
duke@1 542 }
duke@1 543
duke@1 544 public int read(String path, byte[] buffer) throws IOException {
duke@1 545 lock.lock();
duke@1 546 try {
duke@1 547 ZipFileIndexEntry entry = getZipIndexEntry(path);
duke@1 548 if (entry == null)
duke@1 549 throw new FileNotFoundException();
duke@1 550 return read(entry, buffer);
duke@1 551 }
duke@1 552 finally {
duke@1 553 lock.unlock();
duke@1 554 }
duke@1 555 }
duke@1 556
duke@1 557 public int read(ZipFileIndexEntry entry, byte[] buffer)
duke@1 558 throws IOException {
duke@1 559 lock.lock();
duke@1 560 try {
duke@1 561 int result = readBytes(entry, buffer);
duke@1 562 return result;
duke@1 563 }
duke@1 564 finally {
duke@1 565 lock.unlock();
duke@1 566 }
duke@1 567 }
duke@1 568
duke@1 569 private byte[] readBytes(ZipFileIndexEntry entry) throws IOException {
duke@1 570 byte[] header = getHeader(entry);
duke@1 571 int csize = entry.compressedSize;
duke@1 572 byte[] cbuf = new byte[csize];
duke@1 573 zipRandomFile.skipBytes(get2ByteLittleEndian(header, 26) + get2ByteLittleEndian(header, 28));
duke@1 574 zipRandomFile.readFully(cbuf, 0, csize);
duke@1 575
duke@1 576 // is this compressed - offset 8 in the ZipEntry header
duke@1 577 if (get2ByteLittleEndian(header, 8) == 0)
duke@1 578 return cbuf;
duke@1 579
duke@1 580 int size = entry.size;
duke@1 581 byte[] buf = new byte[size];
duke@1 582 if (inflate(cbuf, buf) != size)
duke@1 583 throw new ZipException("corrupted zip file");
duke@1 584
duke@1 585 return buf;
duke@1 586 }
duke@1 587
duke@1 588 /**
duke@1 589 *
duke@1 590 */
duke@1 591 private int readBytes(ZipFileIndexEntry entry, byte[] buffer) throws IOException {
duke@1 592 byte[] header = getHeader(entry);
duke@1 593
duke@1 594 // entry is not compressed?
duke@1 595 if (get2ByteLittleEndian(header, 8) == 0) {
duke@1 596 zipRandomFile.skipBytes(get2ByteLittleEndian(header, 26) + get2ByteLittleEndian(header, 28));
duke@1 597 int offset = 0;
duke@1 598 int size = buffer.length;
duke@1 599 while (offset < size) {
duke@1 600 int count = zipRandomFile.read(buffer, offset, size - offset);
duke@1 601 if (count == -1)
duke@1 602 break;
duke@1 603 offset += count;
duke@1 604 }
duke@1 605 return entry.size;
duke@1 606 }
duke@1 607
duke@1 608 int csize = entry.compressedSize;
duke@1 609 byte[] cbuf = new byte[csize];
duke@1 610 zipRandomFile.skipBytes(get2ByteLittleEndian(header, 26) + get2ByteLittleEndian(header, 28));
duke@1 611 zipRandomFile.readFully(cbuf, 0, csize);
duke@1 612
duke@1 613 int count = inflate(cbuf, buffer);
duke@1 614 if (count == -1)
duke@1 615 throw new ZipException("corrupted zip file");
duke@1 616
duke@1 617 return entry.size;
duke@1 618 }
duke@1 619
duke@1 620 //----------------------------------------------------------------------------
duke@1 621 // Zip utilities
duke@1 622 //----------------------------------------------------------------------------
duke@1 623
duke@1 624 private byte[] getHeader(ZipFileIndexEntry entry) throws IOException {
duke@1 625 zipRandomFile.seek(entry.offset);
duke@1 626 byte[] header = new byte[30];
duke@1 627 zipRandomFile.readFully(header);
duke@1 628 if (get4ByteLittleEndian(header, 0) != 0x04034b50)
duke@1 629 throw new ZipException("corrupted zip file");
duke@1 630 if ((get2ByteLittleEndian(header, 6) & 1) != 0)
duke@1 631 throw new ZipException("encrypted zip file"); // offset 6 in the header of the ZipFileEntry
duke@1 632 return header;
duke@1 633 }
duke@1 634
duke@1 635 /*
duke@1 636 * Inflate using the java.util.zip.Inflater class
duke@1 637 */
duke@1 638 private static Inflater inflater;
duke@1 639 private int inflate(byte[] src, byte[] dest) {
duke@1 640
duke@1 641 // construct the inflater object or reuse an existing one
duke@1 642 if (inflater == null)
duke@1 643 inflater = new Inflater(true);
duke@1 644
duke@1 645 synchronized (inflater) {
duke@1 646 inflater.reset();
duke@1 647 inflater.setInput(src);
duke@1 648 try {
duke@1 649 return inflater.inflate(dest);
duke@1 650 } catch (DataFormatException ex) {
duke@1 651 return -1;
duke@1 652 }
duke@1 653 }
duke@1 654 }
duke@1 655
duke@1 656 /**
duke@1 657 * return the two bytes buf[pos], buf[pos+1] as an unsigned integer in little
duke@1 658 * endian format.
duke@1 659 */
duke@1 660 private static int get2ByteLittleEndian(byte[] buf, int pos) {
duke@1 661 return (buf[pos] & 0xFF) + ((buf[pos+1] & 0xFF) << 8);
duke@1 662 }
duke@1 663
duke@1 664 /**
duke@1 665 * return the 4 bytes buf[i..i+3] as an integer in little endian format.
duke@1 666 */
duke@1 667 private static int get4ByteLittleEndian(byte[] buf, int pos) {
duke@1 668 return (buf[pos] & 0xFF) + ((buf[pos + 1] & 0xFF) << 8) +
duke@1 669 ((buf[pos + 2] & 0xFF) << 16) + ((buf[pos + 3] & 0xFF) << 24);
duke@1 670 }
duke@1 671
duke@1 672 /* ----------------------------------------------------------------------------
duke@1 673 * ZipDirectory
duke@1 674 * ----------------------------------------------------------------------------*/
duke@1 675
duke@1 676 private class ZipDirectory {
duke@1 677 private String lastDir;
duke@1 678 private int lastStart;
duke@1 679 private int lastLen;
duke@1 680
duke@1 681 byte[] zipDir;
duke@1 682 RandomAccessFile zipRandomFile = null;
duke@1 683 ZipFileIndex zipFileIndex = null;
duke@1 684
duke@1 685 public ZipDirectory(RandomAccessFile zipRandomFile, long start, long end, ZipFileIndex index) throws IOException {
duke@1 686 this.zipRandomFile = zipRandomFile;
duke@1 687 this.zipFileIndex = index;
duke@1 688
duke@1 689 findCENRecord(start, end);
duke@1 690 }
duke@1 691
duke@1 692 /*
duke@1 693 * Reads zip file central directory.
duke@1 694 * For more details see readCEN in zip_util.c from the JDK sources.
duke@1 695 * This is a Java port of that function.
duke@1 696 */
duke@1 697 private void findCENRecord(long start, long end) throws IOException {
duke@1 698 long totalLength = end - start;
duke@1 699 int endbuflen = 1024;
duke@1 700 byte[] endbuf = new byte[endbuflen];
duke@1 701 long endbufend = end - start;
duke@1 702
duke@1 703 // There is a variable-length field after the dir offset record. We need to do consequential search.
duke@1 704 while (endbufend >= 22) {
duke@1 705 if (endbufend < endbuflen)
duke@1 706 endbuflen = (int)endbufend;
duke@1 707 long endbufpos = endbufend - endbuflen;
duke@1 708 zipRandomFile.seek(start + endbufpos);
duke@1 709 zipRandomFile.readFully(endbuf, 0, endbuflen);
duke@1 710 int i = endbuflen - 22;
duke@1 711 while (i >= 0 &&
duke@1 712 !(endbuf[i] == 0x50 &&
duke@1 713 endbuf[i + 1] == 0x4b &&
duke@1 714 endbuf[i + 2] == 0x05 &&
duke@1 715 endbuf[i + 3] == 0x06 &&
duke@1 716 endbufpos + i + 22 +
duke@1 717 get2ByteLittleEndian(endbuf, i + 20) == totalLength)) {
duke@1 718 i--;
duke@1 719 }
duke@1 720
duke@1 721 if (i >= 0) {
duke@1 722 zipDir = new byte[get4ByteLittleEndian(endbuf, i + 12) + 2];
duke@1 723 zipDir[0] = endbuf[i + 10];
duke@1 724 zipDir[1] = endbuf[i + 11];
duke@1 725 zipRandomFile.seek(start + get4ByteLittleEndian(endbuf, i + 16));
duke@1 726 zipRandomFile.readFully(zipDir, 2, zipDir.length - 2);
duke@1 727 return;
duke@1 728 } else {
duke@1 729 endbufend = endbufpos + 21;
duke@1 730 }
duke@1 731 }
duke@1 732 throw new ZipException("cannot read zip file");
duke@1 733 }
duke@1 734 private void buildIndex() throws IOException {
duke@1 735 int entryCount = get2ByteLittleEndian(zipDir, 0);
duke@1 736
duke@1 737 entries = new ZipFileIndexEntry[entryCount];
duke@1 738 // Add each of the files
duke@1 739 if (entryCount > 0) {
duke@1 740 directories = new HashMap<String, DirectoryEntry>();
duke@1 741 ArrayList<ZipFileIndexEntry> entryList = new ArrayList<ZipFileIndexEntry>();
duke@1 742 int pos = 2;
duke@1 743 for (int i = 0; i < entryCount; i++) {
duke@1 744 pos = readEntry(pos, entryList, directories);
duke@1 745 }
duke@1 746
duke@1 747 // Add the accumulated dirs into the same list
duke@1 748 Iterator i = directories.keySet().iterator();
duke@1 749 while (i.hasNext()) {
duke@1 750 ZipFileIndexEntry zipFileIndexEntry = new ZipFileIndexEntry( (String) i.next());
duke@1 751 zipFileIndexEntry.isDir = true;
duke@1 752 entryList.add(zipFileIndexEntry);
duke@1 753 }
duke@1 754
duke@1 755 entries = entryList.toArray(new ZipFileIndexEntry[entryList.size()]);
duke@1 756 Arrays.sort(entries);
duke@1 757 } else {
duke@1 758 cleanupState();
duke@1 759 }
duke@1 760 }
duke@1 761
duke@1 762 private int readEntry(int pos, List<ZipFileIndexEntry> entryList,
duke@1 763 Map<String, DirectoryEntry> directories) throws IOException {
duke@1 764 if (get4ByteLittleEndian(zipDir, pos) != 0x02014b50) {
duke@1 765 throw new ZipException("cannot read zip file entry");
duke@1 766 }
duke@1 767
duke@1 768 int dirStart = pos + 46;
duke@1 769 int fileStart = dirStart;
duke@1 770 int fileEnd = fileStart + get2ByteLittleEndian(zipDir, pos + 28);
duke@1 771
duke@1 772 if (zipFileIndex.symbolFilePrefixLength != 0 &&
duke@1 773 ((fileEnd - fileStart) >= symbolFilePrefixLength)) {
duke@1 774 dirStart += zipFileIndex.symbolFilePrefixLength;
duke@1 775 fileStart += zipFileIndex.symbolFilePrefixLength;
duke@1 776 }
duke@1 777
duke@1 778 // Use the OS's path separator. Keep the position of the last one.
duke@1 779 for (int index = fileStart; index < fileEnd; index++) {
duke@1 780 byte nextByte = zipDir[index];
duke@1 781 if (nextByte == (byte)'\\' || nextByte == (byte)'/') {
duke@1 782 zipDir[index] = (byte)File.separatorChar;
duke@1 783 fileStart = index + 1;
duke@1 784 }
duke@1 785 }
duke@1 786
duke@1 787 String directory = null;
duke@1 788 if (fileStart == dirStart)
duke@1 789 directory = "";
duke@1 790 else if (lastDir != null && lastLen == fileStart - dirStart - 1) {
duke@1 791 int index = lastLen - 1;
duke@1 792 while (zipDir[lastStart + index] == zipDir[dirStart + index]) {
duke@1 793 if (index == 0) {
duke@1 794 directory = lastDir;
duke@1 795 break;
duke@1 796 }
duke@1 797 index--;
duke@1 798 }
duke@1 799 }
duke@1 800
duke@1 801 // Sub directories
duke@1 802 if (directory == null) {
duke@1 803 lastStart = dirStart;
duke@1 804 lastLen = fileStart - dirStart - 1;
duke@1 805
duke@1 806 directory = new String(zipDir, dirStart, lastLen, "UTF-8").intern();
duke@1 807 lastDir = directory;
duke@1 808
duke@1 809 // Enter also all the parent directories
duke@1 810 String tempDirectory = directory;
duke@1 811
duke@1 812 while (directories.get(tempDirectory) == null) {
duke@1 813 directories.put(tempDirectory, new DirectoryEntry(tempDirectory, zipFileIndex));
duke@1 814 int separator = tempDirectory.lastIndexOf(File.separatorChar);
duke@1 815 if (separator == -1)
duke@1 816 break;
duke@1 817 tempDirectory = tempDirectory.substring(0, separator);
duke@1 818 }
duke@1 819 }
duke@1 820 else {
duke@1 821 directory = directory.intern();
duke@1 822 if (directories.get(directory) == null) {
duke@1 823 directories.put(directory, new DirectoryEntry(directory, zipFileIndex));
duke@1 824 }
duke@1 825 }
duke@1 826
duke@1 827 // For each dir create also a file
duke@1 828 if (fileStart != fileEnd) {
duke@1 829 ZipFileIndexEntry entry = new ZipFileIndexEntry(directory,
duke@1 830 new String(zipDir, fileStart, fileEnd - fileStart, "UTF-8"));
duke@1 831
duke@1 832 entry.setNativeTime(get4ByteLittleEndian(zipDir, pos + 12));
duke@1 833 entry.compressedSize = get4ByteLittleEndian(zipDir, pos + 20);
duke@1 834 entry.size = get4ByteLittleEndian(zipDir, pos + 24);
duke@1 835 entry.offset = get4ByteLittleEndian(zipDir, pos + 42);
duke@1 836 entryList.add(entry);
duke@1 837 }
duke@1 838
duke@1 839 return pos + 46 +
duke@1 840 get2ByteLittleEndian(zipDir, pos + 28) +
duke@1 841 get2ByteLittleEndian(zipDir, pos + 30) +
duke@1 842 get2ByteLittleEndian(zipDir, pos + 32);
duke@1 843 }
duke@1 844 }
duke@1 845
duke@1 846 /**
duke@1 847 * Returns the last modified timestamp of a zip file.
duke@1 848 * @return long
duke@1 849 */
duke@1 850 public long getZipFileLastModified() throws IOException {
duke@1 851 lock.lock();
duke@1 852 try {
duke@1 853 checkIndex();
duke@1 854 return zipFileLastModified;
duke@1 855 }
duke@1 856 finally {
duke@1 857 lock.unlock();
duke@1 858 }
duke@1 859 }
duke@1 860
duke@1 861 /** ------------------------------------------------------------------------
duke@1 862 * DirectoryEntry class
duke@1 863 * -------------------------------------------------------------------------*/
duke@1 864 static class DirectoryEntry {
duke@1 865 private boolean filesInited;
duke@1 866 private boolean directoriesInited;
duke@1 867 private boolean zipFileEntriesInited;
duke@1 868 private boolean entriesInited;
duke@1 869
duke@1 870 private long writtenOffsetOffset = 0;
duke@1 871
duke@1 872 private String dirName;
duke@1 873
duke@1 874 private com.sun.tools.javac.util.List<String> zipFileEntriesFiles = com.sun.tools.javac.util.List.<String>nil();
duke@1 875 private com.sun.tools.javac.util.List<String> zipFileEntriesDirectories = com.sun.tools.javac.util.List.<String>nil();
duke@1 876 private com.sun.tools.javac.util.List<ZipFileIndexEntry> zipFileEntries = com.sun.tools.javac.util.List.<ZipFileIndexEntry>nil();
duke@1 877
duke@1 878 private List<ZipFileIndexEntry> entries = new ArrayList<ZipFileIndexEntry>();
duke@1 879
duke@1 880 private ZipFileIndex zipFileIndex;
duke@1 881
duke@1 882 private int numEntries;
duke@1 883
duke@1 884 DirectoryEntry(String dirName, ZipFileIndex index) {
duke@1 885 filesInited = false;
duke@1 886 directoriesInited = false;
duke@1 887 entriesInited = false;
duke@1 888
duke@1 889 if (File.separatorChar == '/') {
duke@1 890 dirName.replace('\\', '/');
duke@1 891 }
duke@1 892 else {
duke@1 893 dirName.replace('/', '\\');
duke@1 894 }
duke@1 895
duke@1 896 this.dirName = dirName.intern();
duke@1 897 this.zipFileIndex = index;
duke@1 898 }
duke@1 899
duke@1 900 private com.sun.tools.javac.util.List<String> getFiles() {
duke@1 901 if (filesInited) {
duke@1 902 return zipFileEntriesFiles;
duke@1 903 }
duke@1 904
duke@1 905 initEntries();
duke@1 906
duke@1 907 for (ZipFileIndexEntry e : entries) {
duke@1 908 if (!e.isDir) {
duke@1 909 zipFileEntriesFiles = zipFileEntriesFiles.append(e.name);
duke@1 910 }
duke@1 911 }
duke@1 912 filesInited = true;
duke@1 913 return zipFileEntriesFiles;
duke@1 914 }
duke@1 915
duke@1 916 private com.sun.tools.javac.util.List<String> getDirectories() {
duke@1 917 if (directoriesInited) {
duke@1 918 return zipFileEntriesFiles;
duke@1 919 }
duke@1 920
duke@1 921 initEntries();
duke@1 922
duke@1 923 for (ZipFileIndexEntry e : entries) {
duke@1 924 if (e.isDir) {
duke@1 925 zipFileEntriesDirectories = zipFileEntriesDirectories.append(e.name);
duke@1 926 }
duke@1 927 }
duke@1 928
duke@1 929 directoriesInited = true;
duke@1 930
duke@1 931 return zipFileEntriesDirectories;
duke@1 932 }
duke@1 933
duke@1 934 private com.sun.tools.javac.util.List<ZipFileIndexEntry> getEntries() {
duke@1 935 if (zipFileEntriesInited) {
duke@1 936 return zipFileEntries;
duke@1 937 }
duke@1 938
duke@1 939 initEntries();
duke@1 940
duke@1 941 zipFileEntries = com.sun.tools.javac.util.List.nil();
duke@1 942 for (ZipFileIndexEntry zfie : entries) {
duke@1 943 zipFileEntries = zipFileEntries.append(zfie);
duke@1 944 }
duke@1 945
duke@1 946 zipFileEntriesInited = true;
duke@1 947
duke@1 948 return zipFileEntries;
duke@1 949 }
duke@1 950
duke@1 951 private ZipFileIndexEntry getEntry(String rootName) {
duke@1 952 initEntries();
duke@1 953 int index = Collections.binarySearch(entries, new ZipFileIndexEntry(dirName, rootName));
duke@1 954 if (index < 0) {
duke@1 955 return null;
duke@1 956 }
duke@1 957
duke@1 958 return entries.get(index);
duke@1 959 }
duke@1 960
duke@1 961 private void initEntries() {
duke@1 962 if (entriesInited) {
duke@1 963 return;
duke@1 964 }
duke@1 965
duke@1 966 if (!zipFileIndex.readFromIndex) {
duke@1 967 int from = -Arrays.binarySearch(zipFileIndex.entries,
duke@1 968 new ZipFileIndexEntry(dirName, ZipFileIndex.MIN_CHAR)) - 1;
duke@1 969 int to = -Arrays.binarySearch(zipFileIndex.entries,
duke@1 970 new ZipFileIndexEntry(dirName, MAX_CHAR)) - 1;
duke@1 971
duke@1 972 boolean emptyList = false;
duke@1 973
duke@1 974 for (int i = from; i < to; i++) {
duke@1 975 entries.add(zipFileIndex.entries[i]);
duke@1 976 }
duke@1 977 } else {
duke@1 978 File indexFile = zipFileIndex.getIndexFile();
duke@1 979 if (indexFile != null) {
duke@1 980 RandomAccessFile raf = null;
duke@1 981 try {
duke@1 982 raf = new RandomAccessFile(indexFile, "r");
duke@1 983 raf.seek(writtenOffsetOffset);
duke@1 984
duke@1 985 for (int nFiles = 0; nFiles < numEntries; nFiles++) {
duke@1 986 // Read the name bytes
duke@1 987 int zfieNameBytesLen = raf.readInt();
duke@1 988 byte [] zfieNameBytes = new byte[zfieNameBytesLen];
duke@1 989 raf.read(zfieNameBytes);
duke@1 990 String eName = new String(zfieNameBytes, "UTF-8");
duke@1 991
duke@1 992 // Read isDir
duke@1 993 boolean eIsDir = raf.readByte() == (byte)0 ? false : true;
duke@1 994
duke@1 995 // Read offset of bytes in the real Jar/Zip file
duke@1 996 int eOffset = raf.readInt();
duke@1 997
duke@1 998 // Read size of the file in the real Jar/Zip file
duke@1 999 int eSize = raf.readInt();
duke@1 1000
duke@1 1001 // Read compressed size of the file in the real Jar/Zip file
duke@1 1002 int eCsize = raf.readInt();
duke@1 1003
duke@1 1004 // Read java time stamp of the file in the real Jar/Zip file
duke@1 1005 long eJavaTimestamp = raf.readLong();
duke@1 1006
duke@1 1007 ZipFileIndexEntry rfie = new ZipFileIndexEntry(dirName, eName);
duke@1 1008 rfie.isDir = eIsDir;
duke@1 1009 rfie.offset = eOffset;
duke@1 1010 rfie.size = eSize;
duke@1 1011 rfie.compressedSize = eCsize;
duke@1 1012 rfie.javatime = eJavaTimestamp;
duke@1 1013 entries.add(rfie);
duke@1 1014 }
duke@1 1015 } catch (Throwable t) {
duke@1 1016 // Do nothing
duke@1 1017 } finally {
duke@1 1018 try {
duke@1 1019 if (raf == null) {
duke@1 1020 raf.close();
duke@1 1021 }
duke@1 1022 } catch (Throwable t) {
duke@1 1023 // Do nothing
duke@1 1024 }
duke@1 1025 }
duke@1 1026 }
duke@1 1027 }
duke@1 1028
duke@1 1029 entriesInited = true;
duke@1 1030 }
duke@1 1031
duke@1 1032 List<ZipFileIndexEntry> getEntriesAsCollection() {
duke@1 1033 initEntries();
duke@1 1034
duke@1 1035 return entries;
duke@1 1036 }
duke@1 1037 }
duke@1 1038
duke@1 1039 private boolean readIndex() {
duke@1 1040 if (triedToReadIndex || !usePreindexedCache) {
duke@1 1041 return false;
duke@1 1042 }
duke@1 1043
duke@1 1044 boolean ret = false;
duke@1 1045 lock.lock();
duke@1 1046 try {
duke@1 1047 triedToReadIndex = true;
duke@1 1048 RandomAccessFile raf = null;
duke@1 1049 try {
duke@1 1050 File indexFileName = getIndexFile();
duke@1 1051 raf = new RandomAccessFile(indexFileName, "r");
duke@1 1052
duke@1 1053 long fileStamp = raf.readLong();
duke@1 1054 if (zipFile.lastModified() != fileStamp) {
duke@1 1055 ret = false;
duke@1 1056 } else {
duke@1 1057 directories = new HashMap<String, DirectoryEntry>();
duke@1 1058 int numDirs = raf.readInt();
duke@1 1059 for (int nDirs = 0; nDirs < numDirs; nDirs++) {
duke@1 1060 int dirNameBytesLen = raf.readInt();
duke@1 1061 byte [] dirNameBytes = new byte[dirNameBytesLen];
duke@1 1062 raf.read(dirNameBytes);
duke@1 1063
duke@1 1064 String dirNameStr = new String(dirNameBytes, "UTF-8");
duke@1 1065 DirectoryEntry de = new DirectoryEntry(dirNameStr, this);
duke@1 1066 de.numEntries = raf.readInt();
duke@1 1067 de.writtenOffsetOffset = raf.readLong();
duke@1 1068 directories.put(dirNameStr, de);
duke@1 1069 }
duke@1 1070 ret = true;
duke@1 1071 zipFileLastModified = fileStamp;
duke@1 1072 }
duke@1 1073 } catch (Throwable t) {
duke@1 1074 // Do nothing
duke@1 1075 } finally {
duke@1 1076 if (raf != null) {
duke@1 1077 try {
duke@1 1078 raf.close();
duke@1 1079 } catch (Throwable tt) {
duke@1 1080 // Do nothing
duke@1 1081 }
duke@1 1082 }
duke@1 1083 }
duke@1 1084 if (ret == true) {
duke@1 1085 readFromIndex = true;
duke@1 1086 }
duke@1 1087 }
duke@1 1088 finally {
duke@1 1089 lock.unlock();
duke@1 1090 }
duke@1 1091
duke@1 1092 return ret;
duke@1 1093 }
duke@1 1094
duke@1 1095 private boolean writeIndex() {
duke@1 1096 boolean ret = false;
duke@1 1097 if (readFromIndex || !usePreindexedCache) {
duke@1 1098 return true;
duke@1 1099 }
duke@1 1100
duke@1 1101 if (!writeIndex) {
duke@1 1102 return true;
duke@1 1103 }
duke@1 1104
duke@1 1105 File indexFile = getIndexFile();
duke@1 1106 if (indexFile == null) {
duke@1 1107 return false;
duke@1 1108 }
duke@1 1109
duke@1 1110 RandomAccessFile raf = null;
duke@1 1111 long writtenSoFar = 0;
duke@1 1112 try {
duke@1 1113 raf = new RandomAccessFile(indexFile, "rw");
duke@1 1114
duke@1 1115 raf.writeLong(zipFileLastModified);
duke@1 1116 writtenSoFar += 8;
duke@1 1117
duke@1 1118
duke@1 1119 Iterator<String> iterDirName = directories.keySet().iterator();
duke@1 1120 List<DirectoryEntry> directoriesToWrite = new ArrayList<DirectoryEntry>();
duke@1 1121 Map<String, Long> offsets = new HashMap<String, Long>();
duke@1 1122 raf.writeInt(directories.keySet().size());
duke@1 1123 writtenSoFar += 4;
duke@1 1124
duke@1 1125 while(iterDirName.hasNext()) {
duke@1 1126 String dirName = iterDirName.next();
duke@1 1127 DirectoryEntry dirEntry = directories.get(dirName);
duke@1 1128
duke@1 1129 directoriesToWrite.add(dirEntry);
duke@1 1130
duke@1 1131 // Write the dir name bytes
duke@1 1132 byte [] dirNameBytes = dirName.getBytes("UTF-8");
duke@1 1133 int dirNameBytesLen = dirNameBytes.length;
duke@1 1134 raf.writeInt(dirNameBytesLen);
duke@1 1135 writtenSoFar += 4;
duke@1 1136
duke@1 1137 raf.write(dirNameBytes);
duke@1 1138 writtenSoFar += dirNameBytesLen;
duke@1 1139
duke@1 1140 // Write the number of files in the dir
duke@1 1141 List dirEntries = dirEntry.getEntriesAsCollection();
duke@1 1142 raf.writeInt(dirEntries.size());
duke@1 1143 writtenSoFar += 4;
duke@1 1144
duke@1 1145 offsets.put(dirName, new Long(writtenSoFar));
duke@1 1146
duke@1 1147 // Write the offset of the file's data in the dir
duke@1 1148 dirEntry.writtenOffsetOffset = 0L;
duke@1 1149 raf.writeLong(0L);
duke@1 1150 writtenSoFar += 8;
duke@1 1151 }
duke@1 1152
duke@1 1153 for (DirectoryEntry de : directoriesToWrite) {
duke@1 1154 // Fix up the offset in the directory table
duke@1 1155 long currFP = raf.getFilePointer();
duke@1 1156
duke@1 1157 long offsetOffset = offsets.get(de.dirName).longValue();
duke@1 1158 raf.seek(offsetOffset);
duke@1 1159 raf.writeLong(writtenSoFar);
duke@1 1160
duke@1 1161 raf.seek(currFP);
duke@1 1162
duke@1 1163 // Now write each of the files in the DirectoryEntry
duke@1 1164 List<ZipFileIndexEntry> entries = de.getEntriesAsCollection();
duke@1 1165 for (ZipFileIndexEntry zfie : entries) {
duke@1 1166 // Write the name bytes
duke@1 1167 byte [] zfieNameBytes = zfie.name.getBytes("UTF-8");
duke@1 1168 int zfieNameBytesLen = zfieNameBytes.length;
duke@1 1169 raf.writeInt(zfieNameBytesLen);
duke@1 1170 writtenSoFar += 4;
duke@1 1171 raf.write(zfieNameBytes);
duke@1 1172 writtenSoFar += zfieNameBytesLen;
duke@1 1173
duke@1 1174 // Write isDir
duke@1 1175 raf.writeByte(zfie.isDir ? (byte)1 : (byte)0);
duke@1 1176 writtenSoFar += 1;
duke@1 1177
duke@1 1178 // Write offset of bytes in the real Jar/Zip file
duke@1 1179 raf.writeInt(zfie.offset);
duke@1 1180 writtenSoFar += 4;
duke@1 1181
duke@1 1182 // Write size of the file in the real Jar/Zip file
duke@1 1183 raf.writeInt(zfie.size);
duke@1 1184 writtenSoFar += 4;
duke@1 1185
duke@1 1186 // Write compressed size of the file in the real Jar/Zip file
duke@1 1187 raf.writeInt(zfie.compressedSize);
duke@1 1188 writtenSoFar += 4;
duke@1 1189
duke@1 1190 // Write java time stamp of the file in the real Jar/Zip file
duke@1 1191 raf.writeLong(zfie.getLastModified());
duke@1 1192 writtenSoFar += 8;
duke@1 1193 }
duke@1 1194 }
duke@1 1195 } catch (Throwable t) {
duke@1 1196 // Do nothing
duke@1 1197 } finally {
duke@1 1198 try {
duke@1 1199 if (raf != null) {
duke@1 1200 raf.close();
duke@1 1201 }
duke@1 1202 } catch(IOException ioe) {
duke@1 1203 // Do nothing
duke@1 1204 }
duke@1 1205 }
duke@1 1206
duke@1 1207 return ret;
duke@1 1208 }
duke@1 1209
duke@1 1210 public boolean writeZipIndex() {
duke@1 1211 lock.lock();
duke@1 1212 try {
duke@1 1213 return writeIndex();
duke@1 1214 }
duke@1 1215 finally {
duke@1 1216 lock.unlock();
duke@1 1217 }
duke@1 1218 }
duke@1 1219
duke@1 1220 private File getIndexFile() {
duke@1 1221 if (zipIndexFile == null) {
duke@1 1222 if (zipFile == null) {
duke@1 1223 return null;
duke@1 1224 }
duke@1 1225
duke@1 1226 zipIndexFile = new File((preindexedCacheLocation == null ? "" : preindexedCacheLocation) +
duke@1 1227 zipFile.getName() + ".index");
duke@1 1228 }
duke@1 1229
duke@1 1230 return zipIndexFile;
duke@1 1231 }
duke@1 1232
duke@1 1233 public File getZipFile() {
duke@1 1234 return zipFile;
duke@1 1235 }
duke@1 1236 }

mercurial