langtools/src/share/classes/com/sun/tools/jdeps/ClassFileReader.java - platform/libcore - Git at Google

 /*
  * Copyright (c) 2012, Oracle and/or its affiliates. All rights reserved.
  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
  *
  * This code is free software; you can redistribute it and/or modify it
  * under the terms of the GNU General Public License version 2 only, as
  * published by the Free Software Foundation.  Oracle designates this
  * particular file as subject to the "Classpath" exception as provided
  * by Oracle in the LICENSE file that accompanied this code.
  *
  * This code 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 General Public License
  * version 2 for more details (a copy is included in the LICENSE file that
  * accompanied this code).
  *
  * You should have received a copy of the GNU General Public License version
  * 2 along with this work; if not, write to the Free Software Foundation,
  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  *
  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  * or visit www.oracle.com if you need additional information or have any
  * questions.
  */
 package com.sun.tools.jdeps;

 import com.sun.tools.classfile.ClassFile;
 import com.sun.tools.classfile.ConstantPoolException;
 import com.sun.tools.classfile.Dependencies.ClassFileError;
 import java.io.*;
 import java.nio.file.FileVisitResult;
 import java.nio.file.Files;
 import java.nio.file.Path;
 import java.nio.file.SimpleFileVisitor;
 import java.nio.file.attribute.BasicFileAttributes;
 import java.util.*;
 import java.util.jar.JarEntry;
 import java.util.jar.JarFile;

 /**
  * ClassFileReader reads ClassFile(s) of a given path that can be
  * a .class file, a directory, or a JAR file.
  */
 public class ClassFileReader {
     /**
      * Returns a ClassFileReader instance of a given path.
      */
     public static ClassFileReader newInstance(Path path) throws IOException {
         if (!Files.exists(path)) {
             throw new FileNotFoundException(path.toString());
         }

         if (Files.isDirectory(path)) {
             return new DirectoryReader(path);
         } else if (path.getFileName().toString().endsWith(".jar")) {
             return new JarFileReader(path);
         } else {
             return new ClassFileReader(path);
         }
     }

     /**
      * Returns a ClassFileReader instance of a given JarFile.
      */
     public static ClassFileReader newInstance(Path path, JarFile jf) throws IOException {
         return new JarFileReader(path, jf);
     }

     protected final Path path;
     protected final String baseFileName;
     private ClassFileReader(Path path) {
         this.path = path;
         this.baseFileName = path.getFileName() != null
                                 ? path.getFileName().toString()
                                 : path.toString();
     }

     public String getFileName() {
         return baseFileName;
     }

     /**
      * Returns the ClassFile matching the given binary name
      * or a fully-qualified class name.
      */
     public ClassFile getClassFile(String name) throws IOException {
         if (name.indexOf('.') > 0) {
             int i = name.lastIndexOf('.');
             String pathname = name.replace('.', File.separatorChar) + ".class";
             if (baseFileName.equals(pathname) ||
                     baseFileName.equals(pathname.substring(0, i) + "$" +
                                         pathname.substring(i+1, pathname.length()))) {
                 return readClassFile(path);
             }
         } else {
             if (baseFileName.equals(name.replace('/', File.separatorChar) + ".class")) {
                 return readClassFile(path);
             }
         }
         return null;
     }

     public Iterable<ClassFile> getClassFiles() throws IOException {
         return new Iterable<ClassFile>() {
             public Iterator<ClassFile> iterator() {
                 return new FileIterator();
             }
         };
     }

     protected ClassFile readClassFile(Path p) throws IOException {
         InputStream is = null;
         try {
             is = Files.newInputStream(p);
             return ClassFile.read(is);
         } catch (ConstantPoolException e) {
             throw new ClassFileError(e);
         } finally {
             if (is != null) {
                 is.close();
             }
         }
     }

     class FileIterator implements Iterator<ClassFile> {
         int count;
         FileIterator() {
             this.count = 0;
         }
         public boolean hasNext() {
             return count == 0 && baseFileName.endsWith(".class");
         }

         public ClassFile next() {
             if (!hasNext()) {
                 throw new NoSuchElementException();
             }
             try {
                 ClassFile cf = readClassFile(path);
                 count++;
                 return cf;
             } catch (IOException e) {
                 throw new ClassFileError(e);
             }
         }

         public void remove() {
             throw new UnsupportedOperationException("Not supported yet.");
         }
     }

     public String toString() {
         return path.toString();
     }

     private static class DirectoryReader extends ClassFileReader {
         DirectoryReader(Path path) throws IOException {
             super(path);
         }

         public ClassFile getClassFile(String name) throws IOException {
             if (name.indexOf('.') > 0) {
                 int i = name.lastIndexOf('.');
                 String pathname = name.replace('.', File.separatorChar) + ".class";
                 Path p = path.resolve(pathname);
                 if (!Files.exists(p)) {
                     p = path.resolve(pathname.substring(0, i) + "$" +
                                      pathname.substring(i+1, pathname.length()));
                 }
                 if (Files.exists(p)) {
                     return readClassFile(p);
                 }
             } else {
                 Path p = path.resolve(name + ".class");
                 if (Files.exists(p)) {
                     return readClassFile(p);
                 }
             }
             return null;
         }

         public Iterable<ClassFile> getClassFiles() throws IOException {
             final Iterator<ClassFile> iter = new DirectoryIterator();
             return new Iterable<ClassFile>() {
                 public Iterator<ClassFile> iterator() {
                     return iter;
                 }
             };
         }

         private List<Path> walkTree(Path dir) throws IOException {
             final List<Path> files = new ArrayList<>();
             Files.walkFileTree(dir, new SimpleFileVisitor<Path>() {
                 public FileVisitResult visitFile(Path file, BasicFileAttributes attrs)
                         throws IOException {
                     if (file.getFileName().toString().endsWith(".class")) {
                         files.add(file);
                     }
                     return FileVisitResult.CONTINUE;
                 }
             });
             return files;
         }

         class DirectoryIterator implements Iterator<ClassFile> {
             private List<Path> entries;
             private int index = 0;
             DirectoryIterator() throws IOException {
                 entries = walkTree(path);
                 index = 0;
             }

             public boolean hasNext() {
                 return index != entries.size();
             }

             public ClassFile next() {
                 if (!hasNext()) {
                     throw new NoSuchElementException();
                 }
                 Path path = entries.get(index++);
                 try {
                     return readClassFile(path);
                 } catch (IOException e) {
                     throw new ClassFileError(e);
                 }
             }

             public void remove() {
                 throw new UnsupportedOperationException("Not supported yet.");
             }
         }
     }

     private static class JarFileReader extends ClassFileReader {
         final JarFile jarfile;
         JarFileReader(Path path) throws IOException {
             this(path, new JarFile(path.toFile()));
         }
         JarFileReader(Path path, JarFile jf) throws IOException {
             super(path);
             this.jarfile = jf;
         }

         public ClassFile getClassFile(String name) throws IOException {
             if (name.indexOf('.') > 0) {
                 int i = name.lastIndexOf('.');
                 String entryName = name.replace('.', '/') + ".class";
                 JarEntry e = jarfile.getJarEntry(entryName);
                 if (e == null) {
                     e = jarfile.getJarEntry(entryName.substring(0, i) + "$"
                             + entryName.substring(i + 1, entryName.length()));
                 }
                 if (e != null) {
                     return readClassFile(e);
                 }
             } else {
                 JarEntry e = jarfile.getJarEntry(name + ".class");
                 if (e != null) {
                     return readClassFile(e);
                 }
             }
             return null;
         }

         private ClassFile readClassFile(JarEntry e) throws IOException {
             InputStream is = null;
             try {
                 is = jarfile.getInputStream(e);
                 return ClassFile.read(is);
             } catch (ConstantPoolException ex) {
                 throw new ClassFileError(ex);
             } finally {
                 if (is != null)
                     is.close();
             }
         }

         public Iterable<ClassFile> getClassFiles() throws IOException {
             final Iterator<ClassFile> iter = new JarFileIterator();
             return new Iterable<ClassFile>() {
                 public Iterator<ClassFile> iterator() {
                     return iter;
                 }
             };
         }

         class JarFileIterator implements Iterator<ClassFile> {
             private Enumeration<JarEntry> entries;
             private JarEntry nextEntry;
             JarFileIterator() {
                 this.entries = jarfile.entries();
                 while (entries.hasMoreElements()) {
                     JarEntry e = entries.nextElement();
                     String name = e.getName();
                     if (name.endsWith(".class")) {
                         this.nextEntry = e;
                         break;
                     }
                 }
             }

             public boolean hasNext() {
                 return nextEntry != null;
             }

             public ClassFile next() {
                 if (!hasNext()) {
                     throw new NoSuchElementException();
                 }

                 ClassFile cf;
                 try {
                     cf = readClassFile(nextEntry);
                 } catch (IOException ex) {
                     throw new ClassFileError(ex);
                 }
                 JarEntry entry = nextEntry;
                 nextEntry = null;
                 while (entries.hasMoreElements()) {
                     JarEntry e = entries.nextElement();
                     String name = e.getName();
                     if (name.endsWith(".class")) {
                         nextEntry = e;
                         break;
                     }
                 }
                 return cf;
             }

             public void remove() {
                 throw new UnsupportedOperationException("Not supported yet.");
             }
         }
     }
 }
	/*
	* Copyright (c) 2012, Oracle and/or its affiliates. All rights reserved.
	* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
	*
	* This code is free software; you can redistribute it and/or modify it
	* under the terms of the GNU General Public License version 2 only, as
	* published by the Free Software Foundation. Oracle designates this
	* particular file as subject to the "Classpath" exception as provided
	* by Oracle in the LICENSE file that accompanied this code.
	*
	* This code 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 General Public License
	* version 2 for more details (a copy is included in the LICENSE file that
	* accompanied this code).
	*
	* You should have received a copy of the GNU General Public License version
	* 2 along with this work; if not, write to the Free Software Foundation,
	* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
	*
	* Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
	* or visit www.oracle.com if you need additional information or have any
	* questions.
	*/
	package com.sun.tools.jdeps;

	import com.sun.tools.classfile.ClassFile;
	import com.sun.tools.classfile.ConstantPoolException;
	import com.sun.tools.classfile.Dependencies.ClassFileError;
	import java.io.*;
	import java.nio.file.FileVisitResult;
	import java.nio.file.Files;
	import java.nio.file.Path;
	import java.nio.file.SimpleFileVisitor;
	import java.nio.file.attribute.BasicFileAttributes;
	import java.util.*;
	import java.util.jar.JarEntry;
	import java.util.jar.JarFile;

	/**
	* ClassFileReader reads ClassFile(s) of a given path that can be
	* a .class file, a directory, or a JAR file.
	*/
	public class ClassFileReader {
	/**
	* Returns a ClassFileReader instance of a given path.
	*/
	public static ClassFileReader newInstance(Path path) throws IOException {
	if (!Files.exists(path)) {
	throw new FileNotFoundException(path.toString());
	}

	if (Files.isDirectory(path)) {
	return new DirectoryReader(path);
	} else if (path.getFileName().toString().endsWith(".jar")) {
	return new JarFileReader(path);
	} else {
	return new ClassFileReader(path);
	}
	}

	/**
	* Returns a ClassFileReader instance of a given JarFile.
	*/
	public static ClassFileReader newInstance(Path path, JarFile jf) throws IOException {
	return new JarFileReader(path, jf);
	}

	protected final Path path;
	protected final String baseFileName;
	private ClassFileReader(Path path) {
	this.path = path;
	this.baseFileName = path.getFileName() != null
	? path.getFileName().toString()
	: path.toString();
	}

	public String getFileName() {
	return baseFileName;
	}

	/**
	* Returns the ClassFile matching the given binary name
	* or a fully-qualified class name.
	*/
	public ClassFile getClassFile(String name) throws IOException {
	if (name.indexOf('.') > 0) {
	int i = name.lastIndexOf('.');
	String pathname = name.replace('.', File.separatorChar) + ".class";
	if (baseFileName.equals(pathname) \|\|
	baseFileName.equals(pathname.substring(0, i) + "$" +
	pathname.substring(i+1, pathname.length()))) {
	return readClassFile(path);
	}
	} else {
	if (baseFileName.equals(name.replace('/', File.separatorChar) + ".class")) {
	return readClassFile(path);
	}
	}
	return null;
	}

	public Iterable<ClassFile> getClassFiles() throws IOException {
	return new Iterable<ClassFile>() {
	public Iterator<ClassFile> iterator() {
	return new FileIterator();
	}
	};
	}

	protected ClassFile readClassFile(Path p) throws IOException {
	InputStream is = null;
	try {
	is = Files.newInputStream(p);
	return ClassFile.read(is);
	} catch (ConstantPoolException e) {
	throw new ClassFileError(e);
	} finally {
	if (is != null) {
	is.close();
	}
	}
	}

	class FileIterator implements Iterator<ClassFile> {
	int count;
	FileIterator() {
	this.count = 0;
	}
	public boolean hasNext() {
	return count == 0 && baseFileName.endsWith(".class");
	}

	public ClassFile next() {
	if (!hasNext()) {
	throw new NoSuchElementException();
	}
	try {
	ClassFile cf = readClassFile(path);
	count++;
	return cf;
	} catch (IOException e) {
	throw new ClassFileError(e);
	}
	}

	public void remove() {
	throw new UnsupportedOperationException("Not supported yet.");
	}
	}

	public String toString() {
	return path.toString();
	}

	private static class DirectoryReader extends ClassFileReader {
	DirectoryReader(Path path) throws IOException {
	super(path);
	}

	public ClassFile getClassFile(String name) throws IOException {
	if (name.indexOf('.') > 0) {
	int i = name.lastIndexOf('.');
	String pathname = name.replace('.', File.separatorChar) + ".class";
	Path p = path.resolve(pathname);
	if (!Files.exists(p)) {
	p = path.resolve(pathname.substring(0, i) + "$" +
	pathname.substring(i+1, pathname.length()));
	}
	if (Files.exists(p)) {
	return readClassFile(p);
	}
	} else {
	Path p = path.resolve(name + ".class");
	if (Files.exists(p)) {
	return readClassFile(p);
	}
	}
	return null;
	}

	public Iterable<ClassFile> getClassFiles() throws IOException {
	final Iterator<ClassFile> iter = new DirectoryIterator();
	return new Iterable<ClassFile>() {
	public Iterator<ClassFile> iterator() {
	return iter;
	}
	};
	}

	private List<Path> walkTree(Path dir) throws IOException {
	final List<Path> files = new ArrayList<>();
	Files.walkFileTree(dir, new SimpleFileVisitor<Path>() {
	public FileVisitResult visitFile(Path file, BasicFileAttributes attrs)
	throws IOException {
	if (file.getFileName().toString().endsWith(".class")) {
	files.add(file);
	}
	return FileVisitResult.CONTINUE;
	}
	});
	return files;
	}

	class DirectoryIterator implements Iterator<ClassFile> {
	private List<Path> entries;
	private int index = 0;
	DirectoryIterator() throws IOException {
	entries = walkTree(path);
	index = 0;
	}

	public boolean hasNext() {
	return index != entries.size();
	}

	public ClassFile next() {
	if (!hasNext()) {
	throw new NoSuchElementException();
	}
	Path path = entries.get(index++);
	try {
	return readClassFile(path);
	} catch (IOException e) {
	throw new ClassFileError(e);
	}
	}

	public void remove() {
	throw new UnsupportedOperationException("Not supported yet.");
	}
	}
	}

	private static class JarFileReader extends ClassFileReader {
	final JarFile jarfile;
	JarFileReader(Path path) throws IOException {
	this(path, new JarFile(path.toFile()));
	}
	JarFileReader(Path path, JarFile jf) throws IOException {
	super(path);
	this.jarfile = jf;
	}

	public ClassFile getClassFile(String name) throws IOException {
	if (name.indexOf('.') > 0) {
	int i = name.lastIndexOf('.');
	String entryName = name.replace('.', '/') + ".class";
	JarEntry e = jarfile.getJarEntry(entryName);
	if (e == null) {
	e = jarfile.getJarEntry(entryName.substring(0, i) + "$"
	+ entryName.substring(i + 1, entryName.length()));
	}
	if (e != null) {
	return readClassFile(e);
	}
	} else {
	JarEntry e = jarfile.getJarEntry(name + ".class");
	if (e != null) {
	return readClassFile(e);
	}
	}
	return null;
	}

	private ClassFile readClassFile(JarEntry e) throws IOException {
	InputStream is = null;
	try {
	is = jarfile.getInputStream(e);
	return ClassFile.read(is);
	} catch (ConstantPoolException ex) {
	throw new ClassFileError(ex);
	} finally {
	if (is != null)
	is.close();
	}
	}

	public Iterable<ClassFile> getClassFiles() throws IOException {
	final Iterator<ClassFile> iter = new JarFileIterator();
	return new Iterable<ClassFile>() {
	public Iterator<ClassFile> iterator() {
	return iter;
	}
	};
	}

	class JarFileIterator implements Iterator<ClassFile> {
	private Enumeration<JarEntry> entries;
	private JarEntry nextEntry;
	JarFileIterator() {
	this.entries = jarfile.entries();
	while (entries.hasMoreElements()) {
	JarEntry e = entries.nextElement();
	String name = e.getName();
	if (name.endsWith(".class")) {
	this.nextEntry = e;
	break;
	}
	}
	}

	public boolean hasNext() {
	return nextEntry != null;
	}

	public ClassFile next() {
	if (!hasNext()) {
	throw new NoSuchElementException();
	}

	ClassFile cf;
	try {
	cf = readClassFile(nextEntry);
	} catch (IOException ex) {
	throw new ClassFileError(ex);
	}
	JarEntry entry = nextEntry;
	nextEntry = null;
	while (entries.hasMoreElements()) {
	JarEntry e = entries.nextElement();
	String name = e.getName();
	if (name.endsWith(".class")) {
	nextEntry = e;
	break;
	}
	}
	return cf;
	}

	public void remove() {
	throw new UnsupportedOperationException("Not supported yet.");
	}
	}
	}
	}