nestedzipfilecodebase.java

来自「A static analysis tool to find bugs in J」· Java 代码 · 共 120 行

/*
 * FindBugs - Find Bugs in Java programs
 * Copyright (C) 2006, University of Maryland
 * 
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 * 
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 * 
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */

package edu.umd.cs.findbugs.classfile.impl;

import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.Iterator;

import edu.umd.cs.findbugs.classfile.ICodeBase;
import edu.umd.cs.findbugs.classfile.ICodeBaseEntry;
import edu.umd.cs.findbugs.classfile.ICodeBaseIterator;
import edu.umd.cs.findbugs.classfile.ICodeBaseLocator;
import edu.umd.cs.findbugs.classfile.IScannableCodeBase;
import edu.umd.cs.findbugs.classfile.ResourceNotFoundException;
import edu.umd.cs.findbugs.io.IO;

/**
 * A scannable code base class for a zip (or Jar) file nested inside
 * some other codebase.  These are handled by extracting the nested
 * zip/jar file to a temporary file, and delegating to an
 * internal ZipFileCodeBase that reads from the temporary file.
 * 
 * @author David Hovemeyer
 */
public class NestedZipFileCodeBase extends AbstractScannableCodeBase implements IScannableCodeBase {
	private ICodeBase parentCodeBase;
	private String resourceName;
	private File tempFile;
	private ZipFileCodeBase delegateCodeBase;

	/**
	 * Constructor.
	 * 
	 * @param codeBaseLocator the codebase locator for this codebase
	 */
	public NestedZipFileCodeBase(NestedZipFileCodeBaseLocator codeBaseLocator)
			throws ResourceNotFoundException, IOException {
		super(codeBaseLocator);
		this.parentCodeBase = codeBaseLocator.getParentCodeBase();
		this.resourceName = codeBaseLocator.getResourceName();

		InputStream inputStream = null;
		OutputStream outputStream = null;
		try {
			// Create a temp file
			this.tempFile = File.createTempFile("findbugs", ".zip");
			tempFile.deleteOnExit(); // just in case we crash before the codebase is closed

			// Copy nested zipfile to the temporary file
			// FIXME: potentially long blocking operation - should be interruptible
			inputStream = parentCodeBase.lookupResource(resourceName).openResource();
			outputStream = new BufferedOutputStream(new FileOutputStream(tempFile));
			IO.copy(inputStream, outputStream);
			outputStream.flush();

			// Create the delegate to read from the temporary file
			delegateCodeBase = new ZipFileCodeBase(codeBaseLocator, tempFile);
		} finally {
			if (inputStream != null) {
				IO.close(inputStream);
			}

			if (outputStream != null) {
				IO.close(outputStream);
			}
		}
	}

	/* (non-Javadoc)
	 * @see edu.umd.cs.findbugs.classfile.IScannableCodeBase#iterator()
	 */
	public ICodeBaseIterator iterator() throws InterruptedException {
		return new DelegatingCodeBaseIterator(this, delegateCodeBase);
	}

	/* (non-Javadoc)
	 * @see edu.umd.cs.findbugs.classfile.ICodeBase#lookupResource(java.lang.String)
	 */
	public ICodeBaseEntry lookupResource(String resourceName) throws ResourceNotFoundException {
		ICodeBaseEntry delegateCodeBaseEntry = delegateCodeBase.lookupResource(resourceName);
		return new DelegatingCodeBaseEntry(this, delegateCodeBaseEntry);
	}

	/* (non-Javadoc)
	 * @see edu.umd.cs.findbugs.classfile.ICodeBase#getPathName()
	 */
	public String getPathName() {
		return null;
	}

	/* (non-Javadoc)
	 * @see edu.umd.cs.findbugs.classfile.ICodeBase#close()
	 */
	public void close() {
		delegateCodeBase.close();
		tempFile.delete();
	}
}