1 /* 2 * Copyright (c) 2015, 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. 8 * 9 * This code is distributed in the hope that it will be useful, but WITHOUT 10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 11 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 12 * version 2 for more details (a copy is included in the LICENSE file that 13 * accompanied this code). 14 * 15 * You should have received a copy of the GNU General Public License version 16 * 2 along with this work; if not, write to the Free Software Foundation, 17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 18 * 19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA 20 * or visit www.oracle.com if you need additional information or have any 21 * questions. 22 */ 23 24 import java.io.File; 25 import java.io.IOException; 26 import java.net.URL; 27 import java.nio.file.Files; 28 import java.security.AllPermission; 29 import java.security.Permissions; 30 import java.security.ProtectionDomain; 31 32 33 /** 34 * A custom ClassLoader to load the concrete LoggerFinder class 35 * with all permissions. 36 * 37 * @author danielfuchs 38 */ 39 public class CustomSystemClassLoader extends ClassLoader { 40 41 42 Class<?> finderClass = null; 43 44 public CustomSystemClassLoader() { 45 super(); 46 } 47 public CustomSystemClassLoader(ClassLoader parent) { 48 super(parent); 49 } 50 51 private Class<?> defineFinderClass(String name) 52 throws ClassNotFoundException { 53 final Object obj = getClassLoadingLock(name); 54 synchronized(obj) { 55 if (finderClass != null) return finderClass; 56 57 URL url = this.getClass().getProtectionDomain().getCodeSource().getLocation(); 58 File file = new File(url.getPath(), name+".class"); 59 if (file.canRead()) { 60 try { 61 byte[] b = Files.readAllBytes(file.toPath()); 62 Permissions perms = new Permissions(); 63 perms.add(new AllPermission()); 64 finderClass = defineClass( 65 name, b, 0, b.length, new ProtectionDomain( 66 this.getClass().getProtectionDomain().getCodeSource(), 67 perms)); 68 System.out.println("Loaded " + name); 69 return finderClass; 70 } catch (Throwable ex) { 71 ex.printStackTrace(); 72 throw new ClassNotFoundException(name, ex); 73 } 74 } else { 75 throw new ClassNotFoundException(name, 76 new IOException(file.toPath() + ": can't read")); 77 } 78 } 79 } 80 81 @Override 82 public synchronized Class<?> loadClass(String name, boolean resolve) throws ClassNotFoundException { 83 if (name.endsWith("$BaseLoggerFinder")) { 84 Class<?> c = defineFinderClass(name); 85 if (resolve) { 86 resolveClass(c); 87 } 88 return c; 89 } 90 return super.loadClass(name, resolve); 91 } 92 93 @Override 94 protected Class<?> findClass(String name) throws ClassNotFoundException { 95 if (name.endsWith("$BaseLoggerFinder")) { 96 return defineFinderClass(name); 97 } 98 return super.findClass(name); 99 } 100 101 } | 1 /* 2 * Copyright (c) 2015, 2018, 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. 8 * 9 * This code is distributed in the hope that it will be useful, but WITHOUT 10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 11 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 12 * version 2 for more details (a copy is included in the LICENSE file that 13 * accompanied this code). 14 * 15 * You should have received a copy of the GNU General Public License version 16 * 2 along with this work; if not, write to the Free Software Foundation, 17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 18 * 19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA 20 * or visit www.oracle.com if you need additional information or have any 21 * questions. 22 */ 23 24 import java.io.File; 25 import java.io.IOException; 26 import java.net.URL; 27 import java.nio.file.Files; 28 import java.security.AllPermission; 29 import java.security.Permissions; 30 import java.security.ProtectionDomain; 31 import java.util.concurrent.ConcurrentHashMap; 32 33 34 /** 35 * A custom ClassLoader to load the concrete LoggerFinder class 36 * with all permissions. 37 * 38 * @author danielfuchs 39 */ 40 public class CustomSystemClassLoader extends ClassLoader { 41 42 43 private final ConcurrentHashMap<String, Class<?>> classes = new ConcurrentHashMap<>(); 44 45 public CustomSystemClassLoader() { 46 super(); 47 } 48 public CustomSystemClassLoader(ClassLoader parent) { 49 super(parent); 50 } 51 52 private Class<?> defineFinderClass(String name) 53 throws ClassNotFoundException { 54 Class<?> finderClass = classes.get(name); 55 if (finderClass != null) return finderClass; 56 57 final Object obj = getClassLoadingLock(name); 58 synchronized(obj) { 59 finderClass = classes.get(name); 60 if (finderClass != null) return finderClass; 61 62 URL url = this.getClass().getProtectionDomain().getCodeSource().getLocation(); 63 File file = new File(url.getPath(), name+".class"); 64 if (file.canRead()) { 65 try { 66 byte[] b = Files.readAllBytes(file.toPath()); 67 Permissions perms = new Permissions(); 68 perms.add(new AllPermission()); 69 finderClass = defineClass( 70 name, b, 0, b.length, new ProtectionDomain( 71 this.getClass().getProtectionDomain().getCodeSource(), 72 perms)); 73 System.out.println("Loaded " + name); 74 classes.put(name, finderClass); 75 return finderClass; 76 } catch (Throwable ex) { 77 ex.printStackTrace(); 78 throw new ClassNotFoundException(name, ex); 79 } 80 } else { 81 throw new ClassNotFoundException(name, 82 new IOException(file.toPath() + ": can't read")); 83 } 84 } 85 } 86 87 @Override 88 public synchronized Class<?> loadClass(String name, boolean resolve) throws ClassNotFoundException { 89 if (name.endsWith("BaseLoggerFinder") || name.startsWith("BaseLoggerFinder$")) { 90 Class<?> c = defineFinderClass(name); 91 if (resolve) { 92 resolveClass(c); 93 } 94 return c; 95 } 96 return super.loadClass(name, resolve); 97 } 98 99 @Override 100 protected Class<?> findClass(String name) throws ClassNotFoundException { 101 if (name.endsWith("BaseLoggerFinder") || name.startsWith("BaseLoggerFinder$")) { 102 return defineFinderClass(name); 103 } 104 return super.findClass(name); 105 } 106 107 } |