Home | History | Annotate | Download | only in src 
      1 /*
      2  * Copyright (C) 2016 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.*;
     18 import java.util.Base64;
     19 
     20 import art.Redefinition;
     21 
     22 class Main {
     23   public static String TEST_NAME = "935-non-retransformable";
     24 
     25   /**
     26    * base64 encoded class/dex file for
     27    * class Transform {
     28    *   public void sayHi() {
     29    *     System.out.println("Hello");
     30    *   }
     31    *   public void sayGoodbye() {
     32    *    System.out.println("Goodbye");
     33    *   }
     34    * }
     35    */
     36   private static final byte[] CLASS_BYTES = Base64.getDecoder().decode(
     37     "yv66vgAAADQAHwoABwAQCQARABIIABMKABQAFQgAFgcAFwcAGAEABjxpbml0PgEAAygpVgEABENv" +
     38     "ZGUBAA9MaW5lTnVtYmVyVGFibGUBAAVzYXlIaQEACnNheUdvb2RieWUBAApTb3VyY2VGaWxlAQAO" +
     39     "VHJhbnNmb3JtLmphdmEMAAgACQcAGQwAGgAbAQAFSGVsbG8HABwMAB0AHgEAB0dvb2RieWUBAAlU" +
     40     "cmFuc2Zvcm0BABBqYXZhL2xhbmcvT2JqZWN0AQAQamF2YS9sYW5nL1N5c3RlbQEAA291dAEAFUxq" +
     41     "YXZhL2lvL1ByaW50U3RyZWFtOwEAE2phdmEvaW8vUHJpbnRTdHJlYW0BAAdwcmludGxuAQAVKExq" +
     42     "YXZhL2xhbmcvU3RyaW5nOylWACAABgAHAAAAAAADAAAACAAJAAEACgAAAB0AAQABAAAABSq3AAGx" +
     43     "AAAAAQALAAAABgABAAAAAQABAAwACQABAAoAAAAlAAIAAQAAAAmyAAISA7YABLEAAAABAAsAAAAK" +
     44     "AAIAAAADAAgABAABAA0ACQABAAoAAAAlAAIAAQAAAAmyAAISBbYABLEAAAABAAsAAAAKAAIAAAAG" +
     45     "AAgABwABAA4AAAACAA8=");
     46   private static final byte[] DEX_BYTES = Base64.getDecoder().decode(
     47     "ZGV4CjAzNQDpaN+7jX/ZLl9Jr0HAEV7nqL1YDuakKakgAwAAcAAAAHhWNBIAAAAAAAAAAIACAAAQ" +
     48     "AAAAcAAAAAYAAACwAAAAAgAAAMgAAAABAAAA4AAAAAUAAADoAAAAAQAAABABAADwAQAAMAEAAJYB" +
     49     "AACeAQAApwEAAK4BAAC7AQAA0gEAAOYBAAD6AQAADgIAAB4CAAAhAgAAJQIAADkCAAA+AgAARwIA" +
     50     "AFMCAAADAAAABAAAAAUAAAAGAAAABwAAAAkAAAAJAAAABQAAAAAAAAAKAAAABQAAAJABAAAEAAEA" +
     51     "DAAAAAAAAAAAAAAAAAAAAA4AAAAAAAAADwAAAAEAAQANAAAAAgAAAAAAAAAAAAAAAAAAAAIAAAAA" +
     52     "AAAACAAAAAAAAABrAgAAAAAAAAEAAQABAAAAWgIAAAQAAABwEAQAAAAOAAMAAQACAAAAXwIAAAkA" +
     53     "AABiAAAAGwEBAAAAbiADABAADgAAAAMAAQACAAAAZQIAAAkAAABiAAAAGwECAAAAbiADABAADgAA" +
     54     "AAEAAAADAAY8aW5pdD4AB0dvb2RieWUABUhlbGxvAAtMVHJhbnNmb3JtOwAVTGphdmEvaW8vUHJp" +
     55     "bnRTdHJlYW07ABJMamF2YS9sYW5nL09iamVjdDsAEkxqYXZhL2xhbmcvU3RyaW5nOwASTGphdmEv" +
     56     "bGFuZy9TeXN0ZW07AA5UcmFuc2Zvcm0uamF2YQABVgACVkwAEmVtaXR0ZXI6IGphY2stNC4yMgAD" +
     57     "b3V0AAdwcmludGxuAApzYXlHb29kYnllAAVzYXlIaQABAAcOAAYABw6HAAMABw6HAAAAAQIAgIAE" +
     58     "sAIBAcgCAQHsAgAAAA0AAAAAAAAAAQAAAAAAAAABAAAAEAAAAHAAAAACAAAABgAAALAAAAADAAAA" +
     59     "AgAAAMgAAAAEAAAAAQAAAOAAAAAFAAAABQAAAOgAAAAGAAAAAQAAABABAAABIAAAAwAAADABAAAB" +
     60     "EAAAAQAAAJABAAACIAAAEAAAAJYBAAADIAAAAwAAAFoCAAAAIAAAAQAAAGsCAAAAEAAAAQAAAIAC" +
     61     "AAA=");
     62 
     63 
     64   public static ClassLoader getClassLoaderFor(String location) throws Exception {
     65     try {
     66       Class<?> class_loader_class = Class.forName("dalvik.system.PathClassLoader");
     67       Constructor<?> ctor = class_loader_class.getConstructor(String.class, ClassLoader.class);
     68       /* on Dalvik, this is a DexFile; otherwise, it's null */
     69       return (ClassLoader)ctor.newInstance(location + "/" + TEST_NAME + "-ex.jar",
     70                                            Main.class.getClassLoader());
     71     } catch (ClassNotFoundException e) {
     72       // Running on RI. Use URLClassLoader.
     73       return new java.net.URLClassLoader(
     74           new java.net.URL[] { new java.net.URL("file://" + location + "/classes-ex/") });
     75     }
     76   }
     77 
     78   public static void main(String[] args) {
     79     Redefinition.setPopRetransformations(false);
     80     Redefinition.addCommonTransformationResult("Transform", CLASS_BYTES, DEX_BYTES);
     81     Redefinition.enableCommonRetransformation(true);
     82     try {
     83       /* this is the "alternate" DEX/Jar file */
     84       ClassLoader new_loader = getClassLoaderFor(System.getenv("DEX_LOCATION"));
     85       Class<?> klass = (Class<?>)new_loader.loadClass("TestMain");
     86       if (klass == null) {
     87         throw new AssertionError("loadClass failed");
     88       }
     89       Method run_test = klass.getMethod("runTest");
     90       run_test.invoke(null);
     91 
     92       // Remove the original transformation. It has been used by now.
     93       Redefinition.popTransformationFor("Transform");
     94       // Make sure we don't get called for transformation again.
     95       Redefinition.addCommonTransformationResult("Transform", new byte[0], new byte[0]);
     96       Redefinition.doCommonClassRetransformation(new_loader.loadClass("Transform"));
     97       run_test.invoke(null);
     98     } catch (Exception e) {
     99       System.out.println(e.toString());
    100       e.printStackTrace(System.out);
    101     }
    102   }
    103 }
    104