• file
• revisions
• annotate
• diff
• comparison
• raw

test/tools/javac/lambdaShapes/org/openjdk/tests/separate/DirectedClassLoader.java@2eb010b6cb22 (annotated)

test/tools/javac/lambdaShapes/org/openjdk/tests/separate/DirectedClassLoader.java

Thu, 31 Aug 2017 15:17:03 +0800

author

aoqi

date

Thu, 31 Aug 2017 15:17:03 +0800

changeset 2525

2eb010b6cb22

parent 1448

7d34e91f66bb

parent 0

959103a6100f

permissions

-rw-r--r--

merge

aoqi@0	1	/*
aoqi@0	2	* Copyright (c) 2012, Oracle and/or its affiliates. All rights reserved.
aoqi@0	3	* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
aoqi@0	4	*
aoqi@0	5	* This code is free software; you can redistribute it and/or modify it
aoqi@0	6	* under the terms of the GNU General Public License version 2 only, as
aoqi@0	7	* published by the Free Software Foundation. Oracle designates this
aoqi@0	8	* particular file as subject to the "Classpath" exception as provided
aoqi@0	9	* by Oracle in the LICENSE file that accompanied this code.
aoqi@0	10	*
aoqi@0	11	* This code is distributed in the hope that it will be useful, but WITHOUT
aoqi@0	12	* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
aoqi@0	13	* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
aoqi@0	14	* version 2 for more details (a copy is included in the LICENSE file that
aoqi@0	15	* accompanied this code).
aoqi@0	16	*
aoqi@0	17	* You should have received a copy of the GNU General Public License version
aoqi@0	18	* 2 along with this work; if not, write to the Free Software Foundation,
aoqi@0	19	* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
aoqi@0	20	*
aoqi@0	21	* Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
aoqi@0	22	* or visit www.oracle.com if you need additional information or have any
aoqi@0	23	* questions.
aoqi@0	24	*/
aoqi@0	25
aoqi@0	26	package org.openjdk.tests.separate;
aoqi@0	27
aoqi@0	28	import java.util.HashMap;
aoqi@0	29	import java.io.File;
aoqi@0	30	import java.io.FileInputStream;
aoqi@0	31	import java.io.IOException;
aoqi@0	32
aoqi@0	33	class DirectedClassLoader extends ClassLoader {
aoqi@0	34
aoqi@0	35	private HashMap<String,File> loadLocations;
aoqi@0	36	private File defaultLocation;
aoqi@0	37	private ClassFilePreprocessor[] preprocessors;
aoqi@0	38
aoqi@0	39	public DirectedClassLoader(
aoqi@0	40	HashMap<String,File> locations, File fallback,
aoqi@0	41	ClassFilePreprocessor ... preprocessors) {
aoqi@0	42	loadLocations = new HashMap<>(locations);
aoqi@0	43	defaultLocation = fallback;
aoqi@0	44	this.preprocessors = preprocessors;
aoqi@0	45	}
aoqi@0	46
aoqi@0	47	public DirectedClassLoader(
aoqi@0	48	File fallback, ClassFilePreprocessor ... preprocessors) {
aoqi@0	49	loadLocations = new HashMap<>();
aoqi@0	50	defaultLocation = fallback;
aoqi@0	51	this.preprocessors = preprocessors;
aoqi@0	52	}
aoqi@0	53
aoqi@0	54	public DirectedClassLoader(ClassFilePreprocessor ... preprocessors) {
aoqi@0	55	this((File)null, preprocessors);
aoqi@0	56	}
aoqi@0	57
aoqi@0	58	public void setDefaultLocation(File dir) { this.defaultLocation = dir; }
aoqi@0	59	public void setLocationFor(String name, File dir) {
aoqi@0	60	loadLocations.put(name, dir);
aoqi@0	61	}
aoqi@0	62
aoqi@0	63	@Override
aoqi@0	64	protected Class<?> findClass(String name) {
aoqi@0	65	String path = name.replace(".", File.separator) + ".class";
aoqi@0	66
aoqi@0	67	File location = loadLocations.get(name);
aoqi@0	68	if (location == null || !(new File(location, path)).exists()) {
aoqi@0	69	File def = new File(defaultLocation, path);
aoqi@0	70	if (def.exists()) {
aoqi@0	71	return defineFrom(name, new File(location, path));
aoqi@0	72	}
aoqi@0	73	} else {
aoqi@0	74	return defineFrom(name, new File(location, path));
aoqi@0	75	}
aoqi@0	76	return null;
aoqi@0	77	}
aoqi@0	78
aoqi@0	79	private Class<?> defineFrom(String name, File file) {
aoqi@0	80	FileInputStream fis = null;
aoqi@0	81	try {
aoqi@0	82	try {
aoqi@0	83	fis = new FileInputStream(file);
aoqi@0	84	byte[] bytes = new byte[fis.available()];
aoqi@0	85	int read = fis.read(bytes);
aoqi@0	86	if (read != bytes.length) {
aoqi@0	87	return null;
aoqi@0	88	}
aoqi@0	89	if (preprocessors != null) {
aoqi@0	90	for (ClassFilePreprocessor cfp : preprocessors) {
aoqi@0	91	bytes = cfp.preprocess(name, bytes);
aoqi@0	92	}
aoqi@0	93	}
aoqi@0	94	return defineClass(name, bytes, 0, bytes.length);
aoqi@0	95	} finally {
aoqi@0	96	fis.close();
aoqi@0	97	}
aoqi@0	98	} catch (IOException e) {}
aoqi@0	99	return null;
aoqi@0	100	}
aoqi@0	101	}