| /* |
| * Copyright (C) 2016 The Android Open Source Project |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| import static art.Redefinition.addCommonTransformationResult; |
| import static art.Redefinition.enableCommonRetransformation; |
| import static art.Redefinition.setPopRetransformations; |
| |
| import java.lang.reflect.*; |
| import java.util.Base64; |
| |
| class Main { |
| public static String TEST_NAME = "934-load-transform"; |
| |
| /** |
| * base64 encoded class/dex file for |
| * class Transform { |
| * public void sayHi() { |
| * System.out.println("Goodbye"); |
| * } |
| * } |
| */ |
| private static final byte[] CLASS_BYTES = Base64.getDecoder().decode( |
| "yv66vgAAADQAHAoABgAOCQAPABAIABEKABIAEwcAFAcAFQEABjxpbml0PgEAAygpVgEABENvZGUB" + |
| "AA9MaW5lTnVtYmVyVGFibGUBAAVzYXlIaQEAClNvdXJjZUZpbGUBAA5UcmFuc2Zvcm0uamF2YQwA" + |
| "BwAIBwAWDAAXABgBAAdHb29kYnllBwAZDAAaABsBAAlUcmFuc2Zvcm0BABBqYXZhL2xhbmcvT2Jq" + |
| "ZWN0AQAQamF2YS9sYW5nL1N5c3RlbQEAA291dAEAFUxqYXZhL2lvL1ByaW50U3RyZWFtOwEAE2ph" + |
| "dmEvaW8vUHJpbnRTdHJlYW0BAAdwcmludGxuAQAVKExqYXZhL2xhbmcvU3RyaW5nOylWACAABQAG" + |
| "AAAAAAACAAAABwAIAAEACQAAAB0AAQABAAAABSq3AAGxAAAAAQAKAAAABgABAAAAEQABAAsACAAB" + |
| "AAkAAAAlAAIAAQAAAAmyAAISA7YABLEAAAABAAoAAAAKAAIAAAATAAgAFAABAAwAAAACAA0="); |
| private static final byte[] DEX_BYTES = Base64.getDecoder().decode( |
| "ZGV4CjAzNQCLXSBQ5FiS3f16krSYZFF8xYZtFVp0GRXMAgAAcAAAAHhWNBIAAAAAAAAAACwCAAAO" + |
| "AAAAcAAAAAYAAACoAAAAAgAAAMAAAAABAAAA2AAAAAQAAADgAAAAAQAAAAABAACsAQAAIAEAAGIB" + |
| "AABqAQAAcwEAAIABAACXAQAAqwEAAL8BAADTAQAA4wEAAOYBAADqAQAA/gEAAAMCAAAMAgAAAgAA" + |
| "AAMAAAAEAAAABQAAAAYAAAAIAAAACAAAAAUAAAAAAAAACQAAAAUAAABcAQAABAABAAsAAAAAAAAA" + |
| "AAAAAAAAAAANAAAAAQABAAwAAAACAAAAAAAAAAAAAAAAAAAAAgAAAAAAAAAHAAAAAAAAAB4CAAAA" + |
| "AAAAAQABAAEAAAATAgAABAAAAHAQAwAAAA4AAwABAAIAAAAYAgAACQAAAGIAAAAbAQEAAABuIAIA" + |
| "EAAOAAAAAQAAAAMABjxpbml0PgAHR29vZGJ5ZQALTFRyYW5zZm9ybTsAFUxqYXZhL2lvL1ByaW50" + |
| "U3RyZWFtOwASTGphdmEvbGFuZy9PYmplY3Q7ABJMamF2YS9sYW5nL1N0cmluZzsAEkxqYXZhL2xh" + |
| "bmcvU3lzdGVtOwAOVHJhbnNmb3JtLmphdmEAAVYAAlZMABJlbWl0dGVyOiBqYWNrLTMuMzYAA291" + |
| "dAAHcHJpbnRsbgAFc2F5SGkAEQAHDgATAAcOhQAAAAEBAICABKACAQG4Ag0AAAAAAAAAAQAAAAAA" + |
| "AAABAAAADgAAAHAAAAACAAAABgAAAKgAAAADAAAAAgAAAMAAAAAEAAAAAQAAANgAAAAFAAAABAAA" + |
| "AOAAAAAGAAAAAQAAAAABAAABIAAAAgAAACABAAABEAAAAQAAAFwBAAACIAAADgAAAGIBAAADIAAA" + |
| "AgAAABMCAAAAIAAAAQAAAB4CAAAAEAAAAQAAACwCAAA="); |
| |
| public static ClassLoader getClassLoaderFor(String location) throws Exception { |
| try { |
| Class<?> class_loader_class = Class.forName("dalvik.system.PathClassLoader"); |
| Constructor<?> ctor = class_loader_class.getConstructor(String.class, ClassLoader.class); |
| /* on Dalvik, this is a DexFile; otherwise, it's null */ |
| return (ClassLoader)ctor.newInstance(location + "/" + TEST_NAME + "-ex.jar", |
| Main.class.getClassLoader()); |
| } catch (ClassNotFoundException e) { |
| // Running on RI. Use URLClassLoader. |
| return new java.net.URLClassLoader( |
| new java.net.URL[] { new java.net.URL("file://" + location + "/classes-ex/") }); |
| } |
| } |
| |
| public static void main(String[] args) { |
| // Don't pop transformations. Make sure that even if 2 threads race to define the class both |
| // will get the same result. |
| setPopRetransformations(false); |
| addCommonTransformationResult("Transform", CLASS_BYTES, DEX_BYTES); |
| enableCommonRetransformation(true); |
| try { |
| /* this is the "alternate" DEX/Jar file */ |
| ClassLoader new_loader = getClassLoaderFor(System.getenv("DEX_LOCATION")); |
| Class<?> klass = (Class<?>)new_loader.loadClass("TestMain"); |
| if (klass == null) { |
| throw new AssertionError("loadClass failed"); |
| } |
| Method run_test = klass.getMethod("runTest"); |
| run_test.invoke(null); |
| } catch (Exception e) { |
| System.out.println(e.toString()); |
| e.printStackTrace(System.out); |
| } |
| } |
| } |