mirror of
https://github.com/openjdk/jdk.git
synced 2026-02-20 07:15:31 +00:00
276 lines
9.8 KiB
Java
276 lines
9.8 KiB
Java
/*
|
|
* Copyright (c) 2009, 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.
|
|
*
|
|
* 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.classanalyzer;
|
|
|
|
import java.io.BufferedInputStream;
|
|
import java.io.File;
|
|
import java.io.FileInputStream;
|
|
import java.io.IOException;
|
|
import java.io.InputStream;
|
|
import java.util.ArrayList;
|
|
import java.util.Enumeration;
|
|
import java.util.HashSet;
|
|
import java.util.List;
|
|
import java.util.Set;
|
|
import java.util.jar.JarEntry;
|
|
import java.util.jar.JarFile;
|
|
|
|
/**
|
|
*
|
|
* @author mchung
|
|
*/
|
|
public class ClassPath {
|
|
|
|
public class FileInfo {
|
|
|
|
File file;
|
|
JarFile jarfile;
|
|
int classCount;
|
|
long filesize;
|
|
|
|
FileInfo(File f) throws IOException {
|
|
this.file = f;
|
|
this.classCount = 0;
|
|
if (file.getName().endsWith(".jar")) {
|
|
this.filesize = file.length();
|
|
jarfile = new JarFile(f);
|
|
}
|
|
}
|
|
|
|
File getFile() {
|
|
return file;
|
|
}
|
|
|
|
JarFile getJarFile() {
|
|
return jarfile;
|
|
}
|
|
|
|
String getName() throws IOException {
|
|
return file.getCanonicalPath();
|
|
}
|
|
}
|
|
private List<FileInfo> fileList = new ArrayList<FileInfo>();
|
|
private static ClassPath instance = new ClassPath();
|
|
|
|
static List<FileInfo> getFileInfos() {
|
|
return instance.fileList;
|
|
}
|
|
|
|
static ClassPath setJDKHome(String jdkhome) throws IOException {
|
|
List<File> files = new ArrayList<File>();
|
|
File jre = new File(jdkhome, "jre");
|
|
File lib = new File(jdkhome, "lib");
|
|
if (jre.exists() && jre.isDirectory()) {
|
|
listFiles(new File(jre, "lib"), ".jar", files);
|
|
} else if (lib.exists() && lib.isDirectory()) {
|
|
// either a JRE or a jdk build image
|
|
listFiles(lib, ".jar", files);
|
|
|
|
File classes = new File(jdkhome, "classes");
|
|
if (classes.exists() && classes.isDirectory()) {
|
|
// jdk build outputdir
|
|
instance.add(classes);
|
|
}
|
|
} else {
|
|
throw new RuntimeException("\"" + jdkhome + "\" not a JDK home");
|
|
}
|
|
|
|
for (File f : files) {
|
|
instance.add(f);
|
|
}
|
|
return instance;
|
|
}
|
|
|
|
static ClassPath setClassPath(String path) throws IOException {
|
|
if (path.endsWith(".class")) {
|
|
// one class file
|
|
File f = new File(path);
|
|
if (!f.exists()) {
|
|
throw new RuntimeException("Classfile \"" + f + "\" doesn't exist");
|
|
}
|
|
|
|
instance.add(f);
|
|
} else {
|
|
List<File> jarFiles = new ArrayList<File>();
|
|
String[] locs = path.split(File.pathSeparator);
|
|
for (String p : locs) {
|
|
File f = new File(p);
|
|
if (!f.exists()) {
|
|
throw new RuntimeException("\"" + f + "\" doesn't exist");
|
|
}
|
|
|
|
if (f.isDirectory()) {
|
|
instance.add(f); // add the directory to look up .class files
|
|
listFiles(f, ".jar", jarFiles);
|
|
} else if (p.endsWith(".jar")) {
|
|
// jar files
|
|
jarFiles.add(f);
|
|
} else {
|
|
throw new RuntimeException("Invalid file \"" + f);
|
|
}
|
|
}
|
|
// add jarFiles if any
|
|
for (File f : jarFiles) {
|
|
instance.add(f);
|
|
}
|
|
}
|
|
|
|
return instance;
|
|
}
|
|
|
|
private void add(File f) throws IOException {
|
|
fileList.add(new FileInfo(f));
|
|
}
|
|
|
|
public static InputStream open(String pathname) throws IOException {
|
|
for (FileInfo fi : instance.fileList) {
|
|
if (fi.getName().endsWith(".jar")) {
|
|
String path = pathname.replace(File.separatorChar, '/');
|
|
JarEntry e = fi.jarfile.getJarEntry(path);
|
|
if (e != null) {
|
|
return fi.jarfile.getInputStream(e);
|
|
}
|
|
} else if (fi.getFile().isDirectory()) {
|
|
File f = new File(fi.getFile(), pathname);
|
|
if (f.exists()) {
|
|
return new FileInputStream(f);
|
|
}
|
|
} else if (fi.file.isFile()) {
|
|
if (fi.getName().endsWith(File.separator + pathname)) {
|
|
return new FileInputStream(fi.file);
|
|
}
|
|
}
|
|
}
|
|
return null;
|
|
}
|
|
|
|
static ClassFileParser parserForClass(String classname) throws IOException {
|
|
String pathname = classname.replace('.', File.separatorChar) + ".class";
|
|
|
|
ClassFileParser cfparser = null;
|
|
for (FileInfo fi : instance.fileList) {
|
|
if (fi.getName().endsWith(".class")) {
|
|
if (fi.getName().endsWith(File.separator + pathname)) {
|
|
cfparser = ClassFileParser.newParser(fi.getFile(), true);
|
|
break;
|
|
}
|
|
} else if (fi.getName().endsWith(".jar")) {
|
|
JarEntry e = fi.jarfile.getJarEntry(classname.replace('.', '/') + ".class");
|
|
if (e != null) {
|
|
cfparser = ClassFileParser.newParser(fi.jarfile.getInputStream(e), e.getSize(), true);
|
|
break;
|
|
}
|
|
} else if (fi.getFile().isDirectory()) {
|
|
File f = new File(fi.getFile(), pathname);
|
|
if (f.exists()) {
|
|
cfparser = ClassFileParser.newParser(f, true);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
return cfparser;
|
|
}
|
|
|
|
public static void parseAllClassFiles() throws IOException {
|
|
instance.parseFiles();
|
|
}
|
|
|
|
private void parseFiles() throws IOException {
|
|
Set<Klass> classes = new HashSet<Klass>();
|
|
|
|
int count = 0;
|
|
for (FileInfo fi : fileList) {
|
|
// filter out public generated classes (i.e. not public API)
|
|
// javax.management.remote.rmi._RMIConnectionImpl_Tie
|
|
// javax.management.remote.rmi._RMIServerImpl_Tie
|
|
if (fi.getName().endsWith(".class")) {
|
|
parseClass(fi);
|
|
} else if (fi.getName().endsWith(".jar")) {
|
|
Enumeration<JarEntry> entries = fi.jarfile.entries();
|
|
while (entries.hasMoreElements()) {
|
|
JarEntry e = entries.nextElement();
|
|
if (e.getName().endsWith(".class")) {
|
|
ClassFileParser cfparser = ClassFileParser.newParser(fi.jarfile.getInputStream(e), e.getSize(), true);
|
|
cfparser.parseDependency(false);
|
|
fi.classCount++;
|
|
} else if (!e.isDirectory() && ResourceFile.isResource(e.getName())) {
|
|
ResourceFile.addResource(e.getName(), fi.jarfile.getInputStream(e));
|
|
}
|
|
}
|
|
} else if (fi.getFile().isDirectory()) {
|
|
List<File> files = new ArrayList<File>();
|
|
listFiles(fi.getFile(), "", files);
|
|
for (File f : files) {
|
|
if (f.getName().endsWith(".class")) {
|
|
parseClass(fi, f);
|
|
} else if (!f.isDirectory() && ResourceFile.isResource(f.getCanonicalPath())) {
|
|
String pathname = f.getCanonicalPath();
|
|
String dir = fi.getName();
|
|
if (!pathname.startsWith(dir)) {
|
|
throw new RuntimeException("Incorrect pathname " + pathname);
|
|
}
|
|
String name = pathname.substring(dir.length() + 1, pathname.length());
|
|
BufferedInputStream in = new BufferedInputStream(new FileInputStream(f));
|
|
try {
|
|
ResourceFile.addResource(name, in);
|
|
} finally {
|
|
in.close();
|
|
}
|
|
}
|
|
}
|
|
} else {
|
|
// should not reach here
|
|
throw new RuntimeException("Unexpected class path: " + fi.getFile());
|
|
}
|
|
}
|
|
}
|
|
|
|
private void parseClass(FileInfo fi) throws IOException {
|
|
parseClass(fi, fi.getFile());
|
|
}
|
|
|
|
private void parseClass(FileInfo fi, File f) throws IOException {
|
|
ClassFileParser cfparser = ClassFileParser.newParser(f, true);
|
|
cfparser.parseDependency(false);
|
|
fi.classCount++;
|
|
// need to update the filesize for this directory
|
|
fi.filesize += fi.getFile().length();
|
|
|
|
}
|
|
|
|
public static void listFiles(File path, String suffix, List<File> result) {
|
|
if (path.isDirectory()) {
|
|
File[] children = path.listFiles();
|
|
for (File c : children) {
|
|
listFiles(c, suffix, result);
|
|
}
|
|
|
|
} else {
|
|
if (suffix.isEmpty() || path.getName().endsWith(suffix)) {
|
|
result.add(path);
|
|
}
|
|
}
|
|
}
|
|
}
|