• Home
• biz.aQute.bnd
• biz.aQute.bndlib
• 3.3.0
• source code
• DirectoryInputStream.java

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

aQute.lib.putjar.DirectoryInputStream Maven / Gradle / Ivy

package aQute.lib.putjar;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.zip.CRC32;

import aQute.lib.io.IOConstants;
import aQute.libg.fileiterator.FileIterator;

public class DirectoryInputStream extends InputStream {
	static final int			BUFFER_SIZE	= IOConstants.PAGE_SIZE * 16;

	final File					root;
	final FileIterator			fi;
	File						element;
	int							entries		= 0;
	int							state		= START;
	long						where		= 0;

	final static int			START		= 0;
	final static int			HEADER		= 1;
	final static int			DATA		= 2;
	final static int			DIRECTORY	= 4;
	final static int			EOF			= 5;

	final static InputStream	eof			= new ByteArrayInputStream(new byte[0]);
	ByteArrayOutputStream		directory	= new ByteArrayOutputStream();
	InputStream					current		= eof;

	public DirectoryInputStream(File dir) {
		root = dir;
		fi = new FileIterator(dir);
	}

	@Override
	public int read() throws IOException {
		if (fi == null)
			return -1;

		int c = current.read();
		if (c < 0) {
			next();
			c = current.read();
		}
		if (c >= 0)
			where++;

		return c;
	}

	void next() throws IOException {
		switch (state) {
			case START :
			case DATA :
				nextHeader();
				break;

			case HEADER :
				if (element.isFile() && element.length() > 0) {
					current = new FileInputStream(element);
					state = DATA;
				} else
					nextHeader();
				break;

			case DIRECTORY :
				state = EOF;
				current = eof;
				break;

			case EOF :
				break;
		}
	}

	private void nextHeader() throws IOException {
		if (fi.hasNext()) {
			element = fi.next();
			state = HEADER;
			current = getHeader(root, element);
			entries++;
		} else {
			current = getDirectory();
			state = DIRECTORY;
		}
	}

	/**
	 * 
	 *  end of central dir signature 4 bytes (0x06054b50) number of this
	 * disk 2 bytes number of the disk with the start of the central directory 2
	 * bytes total number of entries in the central directory on this disk 2
	 * bytes total number of entries in the central directory 2 bytes size of
	 * the central directory 4 bytes offset of start of central directory with
	 * respect to the starting disk number 4 bytes .ZIP file comment length 2
	 * bytes .ZIP file comment (variable size)
	 * 
	 */
	InputStream getDirectory() throws IOException {
		long where = this.where;
		int sizeDirectory = directory.size();

		writeInt(directory, 0x504b0506); // Signature
		writeShort(directory, 0); // # of disk
		writeShort(directory, 0); // # of the disk with start of the central
		// dir
		writeShort(directory, entries); // # of entries
		writeInt(directory, sizeDirectory); // Size of central dir
		writeInt(directory, (int) where);
		writeShort(directory, 0);

		directory.close();

		byte[] data = directory.toByteArray();
		return new ByteArrayInputStream(data);
	}

	private void writeShort(OutputStream out, int v) throws IOException {
		for (int i = 0; i < 2; i++) {
			out.write((byte) (v & 0xFF));
			v = v >> 8;
		}
	}

	private void writeInt(OutputStream out, int v) throws IOException {
		for (int i = 0; i < 4; i++) {
			out.write((byte) (v & 0xFF));
			v = v >> 8;
		}
	}

	/**
	 * Local file header:
	 * 
	 * 
	 *  local file header signature 4 bytes (0x04034b50)
	 * version needed to extract 2 bytes general purpose bit flag 2 bytes
	 * compression method 2 bytes last mod file time 2 bytes last mod file date
	 * 2 bytes crc-32 4 bytes compressed size 4 bytes uncompressed size 4 bytes
	 * file name length 2 bytes extra field length 2 bytes file name (variable
	 * size) extra field (variable size) central file header signature 4 bytes
	 * (0x02014b50) version made by 2 bytes version needed to extract 2 bytes
	 * general purpose bit flag 2 bytes compression method 2 bytes last mod file
	 * time 2 bytes last mod file date 2 bytes crc-32 4 bytes compressed size 4
	 * bytes uncompressed size 4 bytes file name length 2 bytes extra field
	 * length 2 bytes file comment length 2 bytes disk number start 2 bytes
	 * internal file attributes 2 bytes external file attributes 4 bytes
	 * relative offset of local header 4 bytes file name (variable size) extra
	 * field (variable size) file comment (variable size)
	 * 
	 *