aoqi@0: /*
aoqi@0:  * Copyright (c) 2012, Oracle and/or its affiliates. All rights reserved.
aoqi@0:  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
aoqi@0:  *
aoqi@0:  * This code is free software; you can redistribute it and/or modify it
aoqi@0:  * under the terms of the GNU General Public License version 2 only, as
aoqi@0:  * published by the Free Software Foundation.  Oracle designates this
aoqi@0:  * particular file as subject to the "Classpath" exception as provided
aoqi@0:  * by Oracle in the LICENSE file that accompanied this code.
aoqi@0:  *
aoqi@0:  * This code is distributed in the hope that it will be useful, but WITHOUT
aoqi@0:  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
aoqi@0:  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
aoqi@0:  * version 2 for more details (a copy is included in the LICENSE file that
aoqi@0:  * accompanied this code).
aoqi@0:  *
aoqi@0:  * You should have received a copy of the GNU General Public License version
aoqi@0:  * 2 along with this work; if not, write to the Free Software Foundation,
aoqi@0:  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
aoqi@0:  *
aoqi@0:  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
aoqi@0:  * or visit www.oracle.com if you need additional information or have any
aoqi@0:  * questions.
aoqi@0:  */
aoqi@0: 
aoqi@0: package org.openjdk.tests.separate;
aoqi@0: 
aoqi@0: import java.util.HashMap;
aoqi@0: import java.io.File;
aoqi@0: import java.io.FileInputStream;
aoqi@0: import java.io.IOException;
aoqi@0: 
aoqi@0: class DirectedClassLoader extends ClassLoader {
aoqi@0: 
aoqi@0:     private HashMap<String,File> loadLocations;
aoqi@0:     private File defaultLocation;
aoqi@0:     private ClassFilePreprocessor[] preprocessors;
aoqi@0: 
aoqi@0:     public DirectedClassLoader(
aoqi@0:             HashMap<String,File> locations, File fallback,
aoqi@0:             ClassFilePreprocessor ... preprocessors) {
aoqi@0:         loadLocations = new HashMap<>(locations);
aoqi@0:         defaultLocation = fallback;
aoqi@0:         this.preprocessors = preprocessors;
aoqi@0:     }
aoqi@0: 
aoqi@0:     public DirectedClassLoader(
aoqi@0:             File fallback, ClassFilePreprocessor ... preprocessors) {
aoqi@0:         loadLocations = new HashMap<>();
aoqi@0:         defaultLocation = fallback;
aoqi@0:         this.preprocessors = preprocessors;
aoqi@0:     }
aoqi@0: 
aoqi@0:     public DirectedClassLoader(ClassFilePreprocessor ... preprocessors) {
aoqi@0:         this((File)null, preprocessors);
aoqi@0:     }
aoqi@0: 
aoqi@0:     public void setDefaultLocation(File dir) { this.defaultLocation = dir; }
aoqi@0:     public void setLocationFor(String name, File dir) {
aoqi@0:         loadLocations.put(name, dir);
aoqi@0:     }
aoqi@0: 
aoqi@0:     @Override
aoqi@0:     protected Class<?> findClass(String name) {
aoqi@0:         String path = name.replace(".", File.separator) + ".class";
aoqi@0: 
aoqi@0:         File location = loadLocations.get(name);
aoqi@0:         if (location == null || !(new File(location, path)).exists()) {
aoqi@0:             File def = new File(defaultLocation, path);
aoqi@0:             if (def.exists()) {
aoqi@0:                 return defineFrom(name, new File(location, path));
aoqi@0:             }
aoqi@0:         } else {
aoqi@0:             return defineFrom(name, new File(location, path));
aoqi@0:         }
aoqi@0:         return null;
aoqi@0:     }
aoqi@0: 
aoqi@0:     private Class<?> defineFrom(String name, File file) {
aoqi@0:         FileInputStream fis = null;
aoqi@0:         try {
aoqi@0:             try {
aoqi@0:                 fis = new FileInputStream(file);
aoqi@0:                 byte[] bytes = new byte[fis.available()];
aoqi@0:                 int read = fis.read(bytes);
aoqi@0:                 if (read != bytes.length) {
aoqi@0:                     return null;
aoqi@0:                 }
aoqi@0:                 if (preprocessors != null) {
aoqi@0:                     for (ClassFilePreprocessor cfp : preprocessors) {
aoqi@0:                         bytes = cfp.preprocess(name, bytes);
aoqi@0:                     }
aoqi@0:                  }
aoqi@0:                 return defineClass(name, bytes, 0, bytes.length);
aoqi@0:             } finally {
aoqi@0:                 fis.close();
aoqi@0:             }
aoqi@0:         } catch (IOException e) {}
aoqi@0:         return null;
aoqi@0:     }
aoqi@0: }