1 /* 2 * Copyright (c) 2017, Oracle and/or its affiliates. All rights reserved. 3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. 4 * 5 * This code is free software; you can redistribute it and/or modify it 6 * under the terms of the GNU General Public License version 2 only, as 7 * published by the Free Software Foundation. Oracle designates this 8 * particular file as subject to the "Classpath" exception as provided 9 * by Oracle in the LICENSE file that accompanied this code. 10 * 11 * This code is distributed in the hope that it will be useful, but WITHOUT 12 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 13 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 14 * version 2 for more details (a copy is included in the LICENSE file that 15 * accompanied this code). 16 * 17 * You should have received a copy of the GNU General Public License version 18 * 2 along with this work; if not, write to the Free Software Foundation, 19 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 20 * 21 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA 22 * or visit www.oracle.com if you need additional information or have any 23 * questions. 24 */ 25 26 package jdk.nashorn.tools.jjs; 27 28 import java.io.IOException; 29 import java.io.File; 30 import java.util.Collections; 31 import java.util.EnumSet; 32 import java.util.HashSet; 33 import java.util.List; 34 import java.util.Set; 35 import java.util.stream.Collectors; 36 import java.util.stream.Stream; 37 import javax.tools.JavaCompiler; 38 import javax.tools.JavaFileManager.Location; 39 import javax.tools.JavaFileObject; 40 import javax.tools.StandardJavaFileManager; 41 import javax.tools.StandardLocation; 42 import javax.tools.ToolProvider; 43 import jdk.nashorn.internal.runtime.Context; 44 45 /** 46 * A javac package helper that uses javac to complete package names. 47 */ 48 final class JavacPackagesHelper extends PackagesHelper { 49 // JavaCompiler may be null on certain platforms (eg. JRE) 50 private static final JavaCompiler compiler; 51 static { 52 // Use javac only if security manager is not around! 53 compiler = System.getSecurityManager() == null? ToolProvider.getSystemJavaCompiler() : null; 54 } 55 56 /** 57 * Is this class available? 58 * 59 * @return true if javac is available 60 */ isAvailable()61 static boolean isAvailable() { 62 return compiler != null; 63 } 64 65 private final boolean modulePathSet; 66 private final StandardJavaFileManager fm; 67 private final Set<JavaFileObject.Kind> fileKinds; 68 69 /** 70 * Construct a new JavacPackagesHelper. 71 * 72 * @param context the current Nashorn Context 73 */ JavacPackagesHelper(final Context context)74 JavacPackagesHelper(final Context context) throws IOException { 75 super(context); 76 final String modulePath = context.getEnv()._module_path; 77 this.modulePathSet = modulePath != null && !modulePath.isEmpty(); 78 if (isAvailable()) { 79 final String classPath = context.getEnv()._classpath; 80 fm = compiler.getStandardFileManager(null, null, null); 81 fileKinds = EnumSet.of(JavaFileObject.Kind.CLASS); 82 83 if (this.modulePathSet) { 84 fm.setLocation(StandardLocation.MODULE_PATH, getFiles(modulePath)); 85 } 86 87 if (classPath != null && !classPath.isEmpty()) { 88 fm.setLocation(StandardLocation.CLASS_PATH, getFiles(classPath)); 89 } else { 90 // no classpath set. Make sure that it is empty and not any default like "." 91 fm.setLocation(StandardLocation.CLASS_PATH, Collections.<File>emptyList()); 92 } 93 } else { 94 // javac is not available - caller should have checked! 95 throw new IllegalStateException("JavacPackagesHelper is not available!"); 96 } 97 } 98 99 100 @Override close()101 void close() throws IOException { 102 if (fm != null) { 103 fm.close(); 104 } 105 } 106 107 @Override listPackage(final String pkg)108 Set<String> listPackage(final String pkg) throws IOException { 109 final Set<String> props = new HashSet<>(); 110 listPackage(StandardLocation.PLATFORM_CLASS_PATH, pkg, props); 111 if (this.modulePathSet) { 112 for (Set<Location> locs : fm.listLocationsForModules(StandardLocation.MODULE_PATH)) { 113 for (Location loc : locs) { 114 listPackage(loc, pkg, props); 115 } 116 } 117 } 118 listPackage(StandardLocation.CLASS_PATH, pkg, props); 119 return props; 120 } 121 listPackage(final Location loc, final String pkg, final Set<String> props)122 private void listPackage(final Location loc, final String pkg, final Set<String> props) 123 throws IOException { 124 for (JavaFileObject file : fm.list(loc, pkg, fileKinds, true)) { 125 final String binaryName = fm.inferBinaryName(loc, file); 126 // does not start with the given package prefix 127 if (!binaryName.startsWith(pkg + ".")) { 128 continue; 129 } 130 131 final int nextDot = binaryName.indexOf('.', pkg.length() + 1); 132 final int start = pkg.length() + 1; 133 134 if (nextDot != -1) { 135 // subpackage - eg. "regex" for "java.util" 136 final String pkgName = binaryName.substring(start, nextDot); 137 if (isPackageAccessible(binaryName.substring(0, nextDot))) { 138 props.add(binaryName.substring(start, nextDot)); 139 } 140 } else { 141 // class - filter out nested, inner, anonymous, local classes. 142 // Dynalink supported public nested classes as properties of 143 // StaticClass object anyway. We don't want to expose those 144 // "$" internal names as properties of package object. 145 146 final String clsName = binaryName.substring(start); 147 if (clsName.indexOf('$') == -1 && isClassAccessible(binaryName)) { 148 props.add(clsName); 149 } 150 } 151 } 152 } 153 154 // return list of File objects for the given class path getFiles(final String classPath)155 private static List<File> getFiles(final String classPath) { 156 return Stream.of(classPath.split(File.pathSeparator)) 157 .map(File::new) 158 .collect(Collectors.toList()); 159 } 160 } 161