1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000 1.2 +++ b/test/tools/javac/lambdaShapes/org/openjdk/tests/separate/DirectedClassLoader.java Tue Nov 20 09:58:55 2012 -0800 1.3 @@ -0,0 +1,101 @@ 1.4 +/* 1.5 + * Copyright (c) 2012 Oracle and/or its affiliates. All rights reserved. 1.6 + * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. 1.7 + * 1.8 + * This code is free software; you can redistribute it and/or modify it 1.9 + * under the terms of the GNU General Public License version 2 only, as 1.10 + * published by the Free Software Foundation. Oracle designates this 1.11 + * particular file as subject to the "Classpath" exception as provided 1.12 + * by Oracle in the LICENSE file that accompanied this code. 1.13 + * 1.14 + * This code is distributed in the hope that it will be useful, but WITHOUT 1.15 + * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 1.16 + * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 1.17 + * version 2 for more details (a copy is included in the LICENSE file that 1.18 + * accompanied this code). 1.19 + * 1.20 + * You should have received a copy of the GNU General Public License version 1.21 + * 2 along with this work; if not, write to the Free Software Foundation, 1.22 + * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 1.23 + * 1.24 + * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA 1.25 + * or visit www.oracle.com if you need additional information or have any 1.26 + * questions. 1.27 + */ 1.28 + 1.29 +package org.openjdk.tests.separate; 1.30 + 1.31 +import java.util.HashMap; 1.32 +import java.io.File; 1.33 +import java.io.FileInputStream; 1.34 +import java.io.IOException; 1.35 + 1.36 +class DirectedClassLoader extends ClassLoader { 1.37 + 1.38 + private HashMap<String,File> loadLocations; 1.39 + private File defaultLocation; 1.40 + private ClassFilePreprocessor[] preprocessors; 1.41 + 1.42 + public DirectedClassLoader( 1.43 + HashMap<String,File> locations, File fallback, 1.44 + ClassFilePreprocessor ... preprocessors) { 1.45 + loadLocations = new HashMap<>(locations); 1.46 + defaultLocation = fallback; 1.47 + this.preprocessors = preprocessors; 1.48 + } 1.49 + 1.50 + public DirectedClassLoader( 1.51 + File fallback, ClassFilePreprocessor ... preprocessors) { 1.52 + loadLocations = new HashMap<>(); 1.53 + defaultLocation = fallback; 1.54 + this.preprocessors = preprocessors; 1.55 + } 1.56 + 1.57 + public DirectedClassLoader(ClassFilePreprocessor ... preprocessors) { 1.58 + this((File)null, preprocessors); 1.59 + } 1.60 + 1.61 + public void setDefaultLocation(File dir) { this.defaultLocation = dir; } 1.62 + public void setLocationFor(String name, File dir) { 1.63 + loadLocations.put(name, dir); 1.64 + } 1.65 + 1.66 + @Override 1.67 + protected Class<?> findClass(String name) { 1.68 + String path = name.replace(".", File.separator) + ".class"; 1.69 + 1.70 + File location = loadLocations.get(name); 1.71 + if (location == null || !(new File(location, path)).exists()) { 1.72 + File def = new File(defaultLocation, path); 1.73 + if (def.exists()) { 1.74 + return defineFrom(name, new File(location, path)); 1.75 + } 1.76 + } else { 1.77 + return defineFrom(name, new File(location, path)); 1.78 + } 1.79 + return null; 1.80 + } 1.81 + 1.82 + private Class<?> defineFrom(String name, File file) { 1.83 + FileInputStream fis = null; 1.84 + try { 1.85 + try { 1.86 + fis = new FileInputStream(file); 1.87 + byte[] bytes = new byte[fis.available()]; 1.88 + int read = fis.read(bytes); 1.89 + if (read != bytes.length) { 1.90 + return null; 1.91 + } 1.92 + if (preprocessors != null) { 1.93 + for (ClassFilePreprocessor cfp : preprocessors) { 1.94 + bytes = cfp.preprocess(name, bytes); 1.95 + } 1.96 + } 1.97 + return defineClass(name, bytes, 0, bytes.length); 1.98 + } finally { 1.99 + fis.close(); 1.100 + } 1.101 + } catch (IOException e) {} 1.102 + return null; 1.103 + } 1.104 +}