159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta/* 259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * Copyright (c) 2009-2010 jMonkeyEngine 359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * All rights reserved. 459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * 559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * Redistribution and use in source and binary forms, with or without 659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * modification, are permitted provided that the following conditions are 759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * met: 859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * 959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * * Redistributions of source code must retain the above copyright 1059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * notice, this list of conditions and the following disclaimer. 1159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * 1259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * * Redistributions in binary form must reproduce the above copyright 1359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * notice, this list of conditions and the following disclaimer in the 1459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * documentation and/or other materials provided with the distribution. 1559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * 1659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * * Neither the name of 'jMonkeyEngine' nor the names of its contributors 1759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * may be used to endorse or promote products derived from this software 1859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * without specific prior written permission. 1959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * 2059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 2159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED 2259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 2359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR 2459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, 2559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, 2659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR 2759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF 2859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING 2959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS 3059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 3159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta */ 3259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 3359b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartapackage com.jme3.asset.plugins; 3459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 3559b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport com.jme3.asset.AssetInfo; 3659b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport com.jme3.asset.AssetKey; 3759b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport com.jme3.asset.AssetLocator; 3859b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport com.jme3.asset.AssetManager; 3959b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.io.IOException; 4059b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.io.InputStream; 4159b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.net.HttpURLConnection; 4259b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.net.URL; 4359b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.nio.ByteBuffer; 4459b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.nio.CharBuffer; 4559b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.nio.charset.CharacterCodingException; 4659b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.nio.charset.Charset; 4759b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.nio.charset.CharsetDecoder; 4859b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.nio.charset.CoderResult; 4959b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.util.HashMap; 5059b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.util.logging.Level; 5159b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.util.logging.Logger; 5259b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.util.zip.Inflater; 5359b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.util.zip.InflaterInputStream; 5459b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.util.zip.ZipEntry; 5559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 5659b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartapublic class HttpZipLocator implements AssetLocator { 5759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 5859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private static final Logger logger = Logger.getLogger(HttpZipLocator.class.getName()); 5959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 6059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private URL zipUrl; 6159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private String rootPath = ""; 6259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private int numEntries; 6359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private int tableOffset; 6459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private int tableLength; 6559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private HashMap<String, ZipEntry2> entries; 6659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 6759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private static final ByteBuffer byteBuf = ByteBuffer.allocate(250); 6859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private static final CharBuffer charBuf = CharBuffer.allocate(250); 6959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private static final CharsetDecoder utf8Decoder; 7059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 7159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta public static final long LOCSIG = 0x4034b50, EXTSIG = 0x8074b50, 7259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta CENSIG = 0x2014b50, ENDSIG = 0x6054b50; 7359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 7459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta public static final int LOCHDR = 30, EXTHDR = 16, CENHDR = 46, ENDHDR = 22, 7559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta LOCVER = 4, LOCFLG = 6, LOCHOW = 8, LOCTIM = 10, LOCCRC = 14, 7659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta LOCSIZ = 18, LOCLEN = 22, LOCNAM = 26, LOCEXT = 28, EXTCRC = 4, 7759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta EXTSIZ = 8, EXTLEN = 12, CENVEM = 4, CENVER = 6, CENFLG = 8, 7859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta CENHOW = 10, CENTIM = 12, CENCRC = 16, CENSIZ = 20, CENLEN = 24, 7959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta CENNAM = 28, CENEXT = 30, CENCOM = 32, CENDSK = 34, CENATT = 36, 8059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta CENATX = 38, CENOFF = 42, ENDSUB = 8, ENDTOT = 10, ENDSIZ = 12, 8159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta ENDOFF = 16, ENDCOM = 20; 8259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 8359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta static { 8459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta Charset utf8 = Charset.forName("UTF-8"); 8559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta utf8Decoder = utf8.newDecoder(); 8659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 8759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 8859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private static class ZipEntry2 { 8959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta String name; 9059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta int length; 9159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta int offset; 9259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta int compSize; 9359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta long crc; 9459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta boolean deflate; 9559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 9659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta @Override 9759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta public String toString(){ 9859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return "ZipEntry[name=" + name + 9959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta ", length=" + length + 10059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta ", compSize=" + compSize + 10159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta ", offset=" + offset + "]"; 10259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 10359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 10459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 10559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private static int get16(byte[] b, int off) { 10659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return (b[off++] & 0xff) | 10759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta ((b[off] & 0xff) << 8); 10859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 10959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 11059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private static int get32(byte[] b, int off) { 11159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return (b[off++] & 0xff) | 11259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta ((b[off++] & 0xff) << 8) | 11359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta ((b[off++] & 0xff) << 16) | 11459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta ((b[off] & 0xff) << 24); 11559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 11659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 11759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private static long getu32(byte[] b, int off) throws IOException{ 11859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return (b[off++]&0xff) | 11959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta ((b[off++]&0xff) << 8) | 12059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta ((b[off++]&0xff) << 16) | 12159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta (((long)(b[off]&0xff)) << 24); 12259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 12359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 12459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private static String getUTF8String(byte[] b, int off, int len) throws CharacterCodingException { 12559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta StringBuilder sb = new StringBuilder(); 12659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 12759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta int read = 0; 12859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta while (read < len){ 12959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // Either read n remaining bytes in b or 250 if n is higher. 13059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta int toRead = Math.min(len - read, byteBuf.capacity()); 13159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 13259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta boolean endOfInput = toRead < byteBuf.capacity(); 13359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 13459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // read 'toRead' bytes into byteBuf 13559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta byteBuf.put(b, off + read, toRead); 13659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 13759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // set limit to position and set position to 0 13859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // so data can be decoded 13959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta byteBuf.flip(); 14059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 14159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // decode data in byteBuf 14259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta CoderResult result = utf8Decoder.decode(byteBuf, charBuf, endOfInput); 14359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 14459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // if the result is not an underflow its an error 14559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // that cannot be handled. 14659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // if the error is an underflow and its the end of input 14759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // then the decoder expects more bytes but there are no more => error 14859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (!result.isUnderflow() || !endOfInput){ 14959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta result.throwException(); 15059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 15159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 15259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // flip the char buf to get the string just decoded 15359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta charBuf.flip(); 15459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 15559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // append the decoded data into the StringBuilder 15659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta sb.append(charBuf.toString()); 15759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 15859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // clear buffers for next use 15959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta byteBuf.clear(); 16059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta charBuf.clear(); 16159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 16259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta read += toRead; 16359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 16459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 16559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return sb.toString(); 16659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 16759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 16859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private InputStream readData(int offset, int length) throws IOException{ 16959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta HttpURLConnection conn = (HttpURLConnection) zipUrl.openConnection(); 17059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta conn.setDoOutput(false); 17159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta conn.setUseCaches(false); 17259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta conn.setInstanceFollowRedirects(false); 17359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta String range = "-"; 17459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (offset != Integer.MAX_VALUE){ 17559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta range = offset + range; 17659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 17759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (length != Integer.MAX_VALUE){ 17859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (offset != Integer.MAX_VALUE){ 17959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta range = range + (offset + length - 1); 18059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta }else{ 18159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta range = range + length; 18259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 18359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 18459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 18559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta conn.setRequestProperty("Range", "bytes=" + range); 18659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta conn.connect(); 18759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (conn.getResponseCode() == HttpURLConnection.HTTP_PARTIAL){ 18859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return conn.getInputStream(); 18959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta }else if (conn.getResponseCode() == HttpURLConnection.HTTP_OK){ 19059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta throw new IOException("Your server does not support HTTP feature Content-Range. Please contact your server administrator."); 19159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta }else{ 19259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta throw new IOException(conn.getResponseCode() + " " + conn.getResponseMessage()); 19359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 19459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 19559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 19659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private int readTableEntry(byte[] table, int offset) throws IOException{ 19759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (get32(table, offset) != CENSIG){ 19859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta throw new IOException("Central directory error, expected 'PK12'"); 19959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 20059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 20159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta int nameLen = get16(table, offset + CENNAM); 20259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta int extraLen = get16(table, offset + CENEXT); 20359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta int commentLen = get16(table, offset + CENCOM); 20459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta int newOffset = offset + CENHDR + nameLen + extraLen + commentLen; 20559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 20659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta int flags = get16(table, offset + CENFLG); 20759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if ((flags & 1) == 1){ 20859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // ignore this entry, it uses encryption 20959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return newOffset; 21059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 21159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 21259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta int method = get16(table, offset + CENHOW); 21359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (method != ZipEntry.DEFLATED && method != ZipEntry.STORED){ 21459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // ignore this entry, it uses unknown compression method 21559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return newOffset; 21659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 21759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 21859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta String name = getUTF8String(table, offset + CENHDR, nameLen); 21959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (name.charAt(name.length()-1) == '/'){ 22059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // ignore this entry, it is directory node 22159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // or it has no name (?) 22259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return newOffset; 22359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 22459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 22559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta ZipEntry2 entry = new ZipEntry2(); 22659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta entry.name = name; 22759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta entry.deflate = (method == ZipEntry.DEFLATED); 22859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta entry.crc = getu32(table, offset + CENCRC); 22959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta entry.length = get32(table, offset + CENLEN); 23059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta entry.compSize = get32(table, offset + CENSIZ); 23159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta entry.offset = get32(table, offset + CENOFF); 23259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 23359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // we want offset directly into file data .. 23459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // move the offset forward to skip the LOC header 23559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta entry.offset += LOCHDR + nameLen + extraLen; 23659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 23759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta entries.put(entry.name, entry); 23859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 23959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return newOffset; 24059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 24159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 24259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private void fillByteArray(byte[] array, InputStream source) throws IOException{ 24359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta int total = 0; 24459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta int length = array.length; 24559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta while (total < length) { 24659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta int read = source.read(array, total, length - total); 24759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (read < 0) 24859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta throw new IOException("Failed to read entire array"); 24959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 25059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta total += read; 25159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 25259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 25359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 25459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private void readCentralDirectory() throws IOException{ 25559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta InputStream in = readData(tableOffset, tableLength); 25659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta byte[] header = new byte[tableLength]; 25759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 25859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // Fix for "PK12 bug in town.zip": sometimes 25959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // not entire byte array will be read with InputStream.read() 26059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // (especially for big headers) 26159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta fillByteArray(header, in); 26259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 26359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta// in.read(header); 26459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta in.close(); 26559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 26659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta entries = new HashMap<String, ZipEntry2>(numEntries); 26759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta int offset = 0; 26859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta for (int i = 0; i < numEntries; i++){ 26959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta offset = readTableEntry(header, offset); 27059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 27159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 27259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 27359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private void readEndHeader() throws IOException{ 27459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 27559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta// InputStream in = readData(Integer.MAX_VALUE, ENDHDR); 27659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta// byte[] header = new byte[ENDHDR]; 27759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta// fillByteArray(header, in); 27859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta// in.close(); 27959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta// 28059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta// if (get32(header, 0) != ENDSIG){ 28159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta// throw new IOException("End header error, expected 'PK56'"); 28259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta// } 28359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 28459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // Fix for "PK56 bug in town.zip": 28559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // If there's a zip comment inside the end header, 28659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // PK56 won't appear in the -22 position relative to the end of the 28759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // file! 28859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // In that case, we have to search for it. 28959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // Increase search space to 200 bytes 29059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 29159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta InputStream in = readData(Integer.MAX_VALUE, 200); 29259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta byte[] header = new byte[200]; 29359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta fillByteArray(header, in); 29459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta in.close(); 29559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 29659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta int offset = -1; 29759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta for (int i = 200 - 22; i >= 0; i--){ 29859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (header[i] == (byte) (ENDSIG & 0xff) 29959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta && get32(header, i) == ENDSIG){ 30059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // found location 30159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta offset = i; 30259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta break; 30359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 30459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 30559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (offset == -1) 30659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta throw new IOException("Cannot find Zip End Header in file!"); 30759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 30859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta numEntries = get16(header, offset + ENDTOT); 30959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta tableLength = get32(header, offset + ENDSIZ); 31059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta tableOffset = get32(header, offset + ENDOFF); 31159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 31259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 31359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta public void load(URL url) throws IOException { 31459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (!url.getProtocol().equals("http")) 31559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta throw new UnsupportedOperationException(); 31659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 31759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta zipUrl = url; 31859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta readEndHeader(); 31959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta readCentralDirectory(); 32059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 32159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 32259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private InputStream openStream(ZipEntry2 entry) throws IOException{ 32359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta InputStream in = readData(entry.offset, entry.compSize); 32459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (entry.deflate){ 32559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return new InflaterInputStream(in, new Inflater(true)); 32659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 32759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return in; 32859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 32959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 33059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta public InputStream openStream(String name) throws IOException{ 33159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta ZipEntry2 entry = entries.get(name); 33259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (entry == null) 33359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta throw new RuntimeException("Entry not found: "+name); 33459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 33559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return openStream(entry); 33659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 33759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 33859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta public void setRootPath(String path){ 33959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (!rootPath.equals(path)){ 34059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta rootPath = path; 34159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta try { 34259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta load(new URL(path)); 34359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } catch (IOException ex) { 34459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta logger.log(Level.WARNING, "Failed to set root path "+path, ex); 34559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 34659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 34759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 34859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 34959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta public AssetInfo locate(AssetManager manager, AssetKey key){ 35059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta final ZipEntry2 entry = entries.get(key.getName()); 35159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (entry == null) 35259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return null; 35359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 35459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return new AssetInfo(manager, key){ 35559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta @Override 35659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta public InputStream openStream() { 35759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta try { 35859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return HttpZipLocator.this.openStream(entry); 35959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } catch (IOException ex) { 36059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta logger.log(Level.WARNING, "Error retrieving "+entry.name, ex); 36159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return null; 36259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 36359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 36459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta }; 36559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 36659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 36759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta} 368