1*b1cdbd2cSJim Jagielski /************************************************************** 2*b1cdbd2cSJim Jagielski * 3*b1cdbd2cSJim Jagielski * Licensed to the Apache Software Foundation (ASF) under one 4*b1cdbd2cSJim Jagielski * or more contributor license agreements. See the NOTICE file 5*b1cdbd2cSJim Jagielski * distributed with this work for additional information 6*b1cdbd2cSJim Jagielski * regarding copyright ownership. The ASF licenses this file 7*b1cdbd2cSJim Jagielski * to you under the Apache License, Version 2.0 (the 8*b1cdbd2cSJim Jagielski * "License"); you may not use this file except in compliance 9*b1cdbd2cSJim Jagielski * with the License. You may obtain a copy of the License at 10*b1cdbd2cSJim Jagielski * 11*b1cdbd2cSJim Jagielski * http://www.apache.org/licenses/LICENSE-2.0 12*b1cdbd2cSJim Jagielski * 13*b1cdbd2cSJim Jagielski * Unless required by applicable law or agreed to in writing, 14*b1cdbd2cSJim Jagielski * software distributed under the License is distributed on an 15*b1cdbd2cSJim Jagielski * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY 16*b1cdbd2cSJim Jagielski * KIND, either express or implied. See the License for the 17*b1cdbd2cSJim Jagielski * specific language governing permissions and limitations 18*b1cdbd2cSJim Jagielski * under the License. 19*b1cdbd2cSJim Jagielski * 20*b1cdbd2cSJim Jagielski *************************************************************/ 21*b1cdbd2cSJim Jagielski 22*b1cdbd2cSJim Jagielski 23*b1cdbd2cSJim Jagielski 24*b1cdbd2cSJim Jagielski import java.io.IOException; 25*b1cdbd2cSJim Jagielski import java.io.InputStream; 26*b1cdbd2cSJim Jagielski import java.io.FileInputStream; 27*b1cdbd2cSJim Jagielski import java.io.RandomAccessFile; 28*b1cdbd2cSJim Jagielski import java.io.BufferedInputStream; 29*b1cdbd2cSJim Jagielski import java.io.ByteArrayOutputStream; 30*b1cdbd2cSJim Jagielski import java.util.Enumeration; 31*b1cdbd2cSJim Jagielski import java.util.zip.ZipFile; 32*b1cdbd2cSJim Jagielski import java.util.zip.ZipEntry; 33*b1cdbd2cSJim Jagielski 34*b1cdbd2cSJim Jagielski public class XmlZipExtract 35*b1cdbd2cSJim Jagielski { 36*b1cdbd2cSJim Jagielski 37*b1cdbd2cSJim Jagielski public static final String CONTENT = "Content.xml"; 38*b1cdbd2cSJim Jagielski public static final String OLDCONTENT = "content.xml"; 39*b1cdbd2cSJim Jagielski private static final int BUFFER_SIZE = 2048; 40*b1cdbd2cSJim Jagielski 41*b1cdbd2cSJim Jagielski 42*b1cdbd2cSJim Jagielski /** 43*b1cdbd2cSJim Jagielski * Full path of the Zip file to process. 44*b1cdbd2cSJim Jagielski */ 45*b1cdbd2cSJim Jagielski private String filename = null; 46*b1cdbd2cSJim Jagielski 47*b1cdbd2cSJim Jagielski 48*b1cdbd2cSJim Jagielski /** 49*b1cdbd2cSJim Jagielski * Constructor 50*b1cdbd2cSJim Jagielski * 51*b1cdbd2cSJim Jagielski * @param filename Full Path to Zip file to process 52*b1cdbd2cSJim Jagielski * 53*b1cdbd2cSJim Jagielski */ XmlZipExtract(String filename)54*b1cdbd2cSJim Jagielski public XmlZipExtract(String filename) { 55*b1cdbd2cSJim Jagielski this.filename = filename; 56*b1cdbd2cSJim Jagielski } 57*b1cdbd2cSJim Jagielski 58*b1cdbd2cSJim Jagielski /** 59*b1cdbd2cSJim Jagielski * Copies Content.xml from zip file onto the filename passed as 60*b1cdbd2cSJim Jagielski * an argument 61*b1cdbd2cSJim Jagielski * 62*b1cdbd2cSJim Jagielski * @param fname Full Path to file to which contents have to be copied 63*b1cdbd2cSJim Jagielski * 64*b1cdbd2cSJim Jagielski */ getContentXml(String fname)65*b1cdbd2cSJim Jagielski public void getContentXml(String fname) throws IOException 66*b1cdbd2cSJim Jagielski { 67*b1cdbd2cSJim Jagielski try 68*b1cdbd2cSJim Jagielski { 69*b1cdbd2cSJim Jagielski getContentXmlInt(fname, XmlZipExtract.CONTENT); 70*b1cdbd2cSJim Jagielski } 71*b1cdbd2cSJim Jagielski catch (NullPointerException e1) 72*b1cdbd2cSJim Jagielski { 73*b1cdbd2cSJim Jagielski // If the new name of the content file failed, try 74*b1cdbd2cSJim Jagielski // the older name. 75*b1cdbd2cSJim Jagielski // 76*b1cdbd2cSJim Jagielski System.out.println(filename + " Content.xml does not exist, trying content.xml..."); 77*b1cdbd2cSJim Jagielski try 78*b1cdbd2cSJim Jagielski { 79*b1cdbd2cSJim Jagielski getContentXmlInt(fname, XmlZipExtract.OLDCONTENT); 80*b1cdbd2cSJim Jagielski } 81*b1cdbd2cSJim Jagielski catch (NullPointerException e2) 82*b1cdbd2cSJim Jagielski { 83*b1cdbd2cSJim Jagielski System.out.println(filename + " content.xml does not exist, trying content.xml..."); 84*b1cdbd2cSJim Jagielski throw e2; 85*b1cdbd2cSJim Jagielski } 86*b1cdbd2cSJim Jagielski } 87*b1cdbd2cSJim Jagielski } 88*b1cdbd2cSJim Jagielski getContentXmlInt(String fname, String cname)89*b1cdbd2cSJim Jagielski public void getContentXmlInt(String fname, String cname) throws IOException 90*b1cdbd2cSJim Jagielski { 91*b1cdbd2cSJim Jagielski byte b[] = getEntry(cname); 92*b1cdbd2cSJim Jagielski 93*b1cdbd2cSJim Jagielski RandomAccessFile raf=null; 94*b1cdbd2cSJim Jagielski raf = new RandomAccessFile(fname, "rw"); 95*b1cdbd2cSJim Jagielski raf.write(b); 96*b1cdbd2cSJim Jagielski raf.close(); 97*b1cdbd2cSJim Jagielski } 98*b1cdbd2cSJim Jagielski 99*b1cdbd2cSJim Jagielski /** 100*b1cdbd2cSJim Jagielski * Get the specified entry in the zip file as a stream. 101*b1cdbd2cSJim Jagielski * 102*b1cdbd2cSJim Jagielski * @param entryName The name of the entry in the zipfile to get. 103*b1cdbd2cSJim Jagielski * This should be one of the constants defined above. 104*b1cdbd2cSJim Jagielski * 105*b1cdbd2cSJim Jagielski * @return byte[] bits for entryName 106*b1cdbd2cSJim Jagielski * 107*b1cdbd2cSJim Jagielski * @throws IOException if something goes wrong 108*b1cdbd2cSJim Jagielski */ getEntry(String entryName)109*b1cdbd2cSJim Jagielski public byte[] getEntry(String entryName) throws IOException 110*b1cdbd2cSJim Jagielski { 111*b1cdbd2cSJim Jagielski ZipFile zf = new ZipFile(filename); 112*b1cdbd2cSJim Jagielski ZipEntry ze = zf.getEntry(entryName); 113*b1cdbd2cSJim Jagielski byte[] bits = readStream(zf.getInputStream(ze)); 114*b1cdbd2cSJim Jagielski zf.close(); 115*b1cdbd2cSJim Jagielski return bits; 116*b1cdbd2cSJim Jagielski } 117*b1cdbd2cSJim Jagielski 118*b1cdbd2cSJim Jagielski 119*b1cdbd2cSJim Jagielski /** 120*b1cdbd2cSJim Jagielski * Read an InputStream into an array of bytes. 121*b1cdbd2cSJim Jagielski * 122*b1cdbd2cSJim Jagielski * @param is InputStream of data from Zip file 123*b1cdbd2cSJim Jagielski * 124*b1cdbd2cSJim Jagielski * @return an array of Bytes 125*b1cdbd2cSJim Jagielski */ readStream(InputStream is)126*b1cdbd2cSJim Jagielski private byte[] readStream(InputStream is) throws IOException 127*b1cdbd2cSJim Jagielski { 128*b1cdbd2cSJim Jagielski BufferedInputStream bis = new BufferedInputStream(is); 129*b1cdbd2cSJim Jagielski ByteArrayOutputStream baos = new ByteArrayOutputStream(); 130*b1cdbd2cSJim Jagielski byte[] buffer = new byte[BUFFER_SIZE]; 131*b1cdbd2cSJim Jagielski int eof = 0; 132*b1cdbd2cSJim Jagielski while ((eof = bis.read(buffer, 0, buffer.length)) > 0) { 133*b1cdbd2cSJim Jagielski baos.write(buffer, 0, eof); 134*b1cdbd2cSJim Jagielski } 135*b1cdbd2cSJim Jagielski 136*b1cdbd2cSJim Jagielski return baos.toByteArray(); 137*b1cdbd2cSJim Jagielski } 138*b1cdbd2cSJim Jagielski } 139*b1cdbd2cSJim Jagielski 140*b1cdbd2cSJim Jagielski 141