1 /* 2 * Copyright (C) 2015 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 import java.lang.reflect.Constructor; 18 import java.lang.reflect.Field; 19 20 /** 21 * PathClassLoader test. 22 */ 23 public class Main { 24 25 private static ClassLoader createClassLoader(String dexPath, ClassLoader parent) { 26 try { 27 Class<?> myClassLoaderClass = Class.forName("MyPathClassLoader"); 28 Constructor<?> constructor = myClassLoaderClass.getConstructor(String.class, 29 ClassLoader.class); 30 return (ClassLoader)constructor.newInstance(dexPath, parent); 31 } catch (Exception e) { 32 // Ups, not available?!?! 33 throw new RuntimeException(e); 34 } 35 } 36 37 /** 38 * Main entry point. 39 */ 40 public static void main(String[] args) throws Exception { 41 // Check the class-path for the second file. We'll use that one as the source of the 42 // new classloader. 43 String cp = System.getProperty("java.class.path"); 44 if (cp.split(System.getProperty("path.separator")).length != 1) { 45 throw new IllegalStateException("Didn't find exactly one classpath element in " + cp); 46 } 47 if (!cp.endsWith("classloader2.jar")) { 48 throw new IllegalStateException("Don't understand classpath " + cp); 49 } 50 cp = cp.replace("classloader2.jar", "classloader2-ex.jar"); 51 52 ClassLoader myClassLoader = createClassLoader( 53 cp, ClassLoader.getSystemClassLoader().getParent()); 54 55 // Now load our test class. 56 Class<?> srcClass = A.class; 57 Class<?> exClass = myClassLoader.loadClass("A"); 58 59 // First check: classes should be different. 60 if (srcClass == exClass) { 61 throw new IllegalStateException("Loaded class instances are the same"); 62 } 63 64 // Secondary checks: get the static field values and make sure they aren't the same. 65 String srcValue = (String)srcClass.getDeclaredField("value").get(null); 66 if (!"Src-A".equals(srcValue)) { 67 throw new IllegalStateException("Expected Src-A, found " + srcValue); 68 } 69 String exValue = (String)exClass.getDeclaredField("value").get(null); 70 if (!"Ex-A".equals(exValue)) { 71 throw new IllegalStateException("Expected Ex-A, found " + exValue); 72 } 73 74 // Try to load a dex file with bad dex code. Use new instance to force verification. 75 try { 76 Class<?> badClass = Main.class.getClassLoader().loadClass("B"); 77 System.out.println("Loaded class B."); 78 badClass.newInstance(); 79 System.out.println("Should not be able to instantiate B with bad dex bytecode."); 80 } catch (VerifyError e) { 81 System.out.println("Caught VerifyError."); 82 } 83 84 // Make sure the same error is rethrown when reloading the bad class. 85 try { 86 Class<?> badClass = Main.class.getClassLoader().loadClass("B"); 87 System.out.println("Loaded class B."); 88 badClass.newInstance(); 89 System.out.println("Should not be able to instantiate B with bad dex bytecode."); 90 } catch (NoClassDefFoundError e) { 91 if (e.getCause() instanceof VerifyError) { 92 System.out.println("Caught wrapped VerifyError."); 93 } else { 94 e.printStackTrace(System.out); 95 } 96 } 97 98 System.out.println("Everything OK."); 99 } 100 } 101
