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 static art.Redefinition.doCommonClassRedefinition; 18 19 import java.util.Base64; 20 import java.util.function.Consumer; 21 import java.lang.reflect.Method; 22 23 public class Main { 24 static final boolean ALWAYS_PRINT = false; 25 26 // import java.util.function.Consumer; 27 // class Transform { 28 // public void sayHi(int recur, Consumer<String> reporter, Runnable r) { 29 // privateSayHi(recur, reporter, r); 30 // } 31 // private void privateSayHi(int recur, Consumer<String> reporter, Runnable r) { 32 // reporter.accpet("hello" + recur + " - transformed"); 33 // if (recur == 1) { 34 // r.run(); 35 // privateSayHi(recur - 1, reporter, r); 36 // } else if (recur != 0) { 37 // privateSayHi(recur - 1, reporter, r); 38 // } 39 // reporter.accept("goodbye" + recur + " - transformed"); 40 // } 41 // } 42 private static final byte[] CLASS_BYTES = Base64.getDecoder().decode( 43 "yv66vgAAADQANAoADgAbCgANABwHAB0KAAMAGwgAHgoAAwAfCgADACAIACEKAAMAIgsAIwAkCwAl" + 44 "ACYIACcHACgHACkBAAY8aW5pdD4BAAMoKVYBAARDb2RlAQAPTGluZU51bWJlclRhYmxlAQAFc2F5" + 45 "SGkBADUoSUxqYXZhL3V0aWwvZnVuY3Rpb24vQ29uc3VtZXI7TGphdmEvbGFuZy9SdW5uYWJsZTsp" + 46 "VgEACVNpZ25hdHVyZQEASShJTGphdmEvdXRpbC9mdW5jdGlvbi9Db25zdW1lcjxMamF2YS9sYW5n" + 47 "L1N0cmluZzs+O0xqYXZhL2xhbmcvUnVubmFibGU7KVYBAAxwcml2YXRlU2F5SGkBAA1TdGFja01h" + 48 "cFRhYmxlAQAKU291cmNlRmlsZQEADlRyYW5zZm9ybS5qYXZhDAAPABAMABcAFAEAF2phdmEvbGFu" + 49 "Zy9TdHJpbmdCdWlsZGVyAQAFaGVsbG8MACoAKwwAKgAsAQAOIC0gdHJhbnNmb3JtZWQMAC0ALgcA" + 50 "LwwAMAAxBwAyDAAzABABAAdnb29kYnllAQAJVHJhbnNmb3JtAQAQamF2YS9sYW5nL09iamVjdAEA" + 51 "BmFwcGVuZAEALShMamF2YS9sYW5nL1N0cmluZzspTGphdmEvbGFuZy9TdHJpbmdCdWlsZGVyOwEA" + 52 "HChJKUxqYXZhL2xhbmcvU3RyaW5nQnVpbGRlcjsBAAh0b1N0cmluZwEAFCgpTGphdmEvbGFuZy9T" + 53 "dHJpbmc7AQAbamF2YS91dGlsL2Z1bmN0aW9uL0NvbnN1bWVyAQAGYWNjZXB0AQAVKExqYXZhL2xh" + 54 "bmcvT2JqZWN0OylWAQASamF2YS9sYW5nL1J1bm5hYmxlAQADcnVuACAADQAOAAAAAAADAAAADwAQ" + 55 "AAEAEQAAAB0AAQABAAAABSq3AAGxAAAAAQASAAAABgABAAAAAgABABMAFAACABEAAAAkAAQABAAA" + 56 "AAgqGywttwACsQAAAAEAEgAAAAoAAgAAAAQABwAFABUAAAACABYAAgAXABQAAgARAAAAnwAEAAQA" + 57 "AABhLLsAA1m3AAQSBbYABhu2AAcSCLYABrYACbkACgIAGwSgABUtuQALAQAqGwRkLC23AAKnABAb" + 58 "mQAMKhsEZCwttwACLLsAA1m3AAQSDLYABhu2AAcSCLYABrYACbkACgIAsQAAAAIAEgAAACIACAAA" + 59 "AAcAHgAIACMACQApAAoANQALADkADABCAA4AYAAPABgAAAAEAAI1DAAVAAAAAgAWAAEAGQAAAAIA" + 60 "Gg=="); 61 private static final byte[] DEX_BYTES = Base64.getDecoder().decode( 62 "ZGV4CjAzNQCevtlr8B0kh/duuDYqXkGz/w9lMmtCCuRoBQAAcAAAAHhWNBIAAAAAAAAAALAEAAAg" + 63 "AAAAcAAAAAkAAADwAAAABgAAABQBAAAAAAAAAAAAAAoAAABcAQAAAQAAAKwBAACcAwAAzAEAAPYC" + 64 "AAAGAwAACgMAAA4DAAARAwAAGQMAAB0DAAAgAwAAIwMAACcDAAArAwAAOAMAAFcDAABrAwAAgQMA" + 65 "AJUDAACwAwAAzgMAAO0DAAD9AwAAAAQAAAYEAAAKBAAAEgQAABoEAAAuBAAANwQAAD4EAABMBAAA" + 66 "UQQAAFgEAABiBAAABgAAAAoAAAALAAAADAAAAA0AAAAOAAAADwAAABEAAAATAAAABwAAAAUAAAAA" + 67 "AAAACAAAAAYAAADUAgAACQAAAAYAAADcAgAAEwAAAAgAAAAAAAAAFAAAAAgAAADkAgAAFQAAAAgA" + 68 "AADwAgAAAQADAAQAAAABAAQAGwAAAAEABAAdAAAAAwADAAQAAAAEAAMAHAAAAAYAAwAEAAAABgAB" + 69 "ABcAAAAGAAIAFwAAAAYAAAAeAAAABwAFABYAAAABAAAAAAAAAAMAAAAAAAAAEgAAALQCAACeBAAA" + 70 "AAAAAAEAAACKBAAAAQABAAEAAABpBAAABAAAAHAQAwAAAA4ABgAEAAQAAABuBAAAUAAAACIABgBw" + 71 "EAUAAAAbARoAAABuIAcAEAAMAG4gBgAwAAwAGwEAAAAAbiAHABAADABuEAgAAAAMAHIgCQAEABIQ" + 72 "MwMpAHIQBAAFANgAA/9wQAEAAlQiAAYAcBAFAAAAGwEZAAAAbiAHABAADABuIAYAMAAMABsBAAAA" + 73 "AG4gBwAQAAwAbhAIAAAADAByIAkABAAOADgD4f/YAAP/cEABAAJUKNoEAAQABAAAAIEEAAAEAAAA" + 74 "cEABABAyDgAAAAAAAAAAAAIAAAAAAAAAAQAAAMwBAAACAAAAzAEAAAEAAAAAAAAAAQAAAAUAAAAD" + 75 "AAAAAAAHAAQAAAABAAAAAwAOIC0gdHJhbnNmb3JtZWQAAihJAAIpVgABPAAGPGluaXQ+AAI+OwAB" + 76 "SQABTAACTEkAAkxMAAtMVHJhbnNmb3JtOwAdTGRhbHZpay9hbm5vdGF0aW9uL1NpZ25hdHVyZTsA" + 77 "EkxqYXZhL2xhbmcvT2JqZWN0OwAUTGphdmEvbGFuZy9SdW5uYWJsZTsAEkxqYXZhL2xhbmcvU3Ry" + 78 "aW5nOwAZTGphdmEvbGFuZy9TdHJpbmdCdWlsZGVyOwAcTGphdmEvdXRpbC9mdW5jdGlvbi9Db25z" + 79 "dW1lcgAdTGphdmEvdXRpbC9mdW5jdGlvbi9Db25zdW1lcjsADlRyYW5zZm9ybS5qYXZhAAFWAARW" + 80 "SUxMAAJWTAAGYWNjZXB0AAZhcHBlbmQAEmVtaXR0ZXI6IGphY2stNC4yNAAHZ29vZGJ5ZQAFaGVs" + 81 "bG8ADHByaXZhdGVTYXlIaQADcnVuAAVzYXlIaQAIdG9TdHJpbmcABXZhbHVlAAIABw4ABwMAAAAH" + 82 "DgEeDzw8XQEeDxktAAQDAAAABw48AAICAR8cBxcBFxAXAxcOFwUXDRcCAAACAQCAgATUAwEC7AMC" + 83 "AZwFDwAAAAAAAAABAAAAAAAAAAEAAAAgAAAAcAAAAAIAAAAJAAAA8AAAAAMAAAAGAAAAFAEAAAUA" + 84 "AAAKAAAAXAEAAAYAAAABAAAArAEAAAMQAAABAAAAzAEAAAEgAAADAAAA1AEAAAYgAAABAAAAtAIA" + 85 "AAEQAAAEAAAA1AIAAAIgAAAgAAAA9gIAAAMgAAADAAAAaQQAAAQgAAABAAAAigQAAAAgAAABAAAA" + 86 "ngQAAAAQAAABAAAAsAQAAA=="); 87 88 // A class that we can use to keep track of the output of this test. 89 private static class TestWatcher implements Consumer<String> { 90 private StringBuilder sb; TestWatcher()91 public TestWatcher() { 92 sb = new StringBuilder(); 93 } 94 95 @Override accept(String s)96 public void accept(String s) { 97 if (Main.ALWAYS_PRINT) { 98 System.out.println(s); 99 } 100 sb.append(s); 101 sb.append('\n'); 102 } 103 getOutput()104 public String getOutput() { 105 return sb.toString(); 106 } 107 clear()108 public void clear() { 109 sb = new StringBuilder(); 110 } 111 } 112 main(String[] args)113 public static void main(String[] args) { 114 doTest(new Transform()); 115 } 116 117 private static boolean retry = false; 118 doTest(Transform t)119 public static void doTest(Transform t) { 120 final TestWatcher reporter = new TestWatcher(); 121 Method say_hi_method; 122 Method private_say_hi_method; 123 // Figure out if we can even JIT at all. 124 final boolean has_jit = hasJit(); 125 try { 126 say_hi_method = Transform.class.getDeclaredMethod( 127 "sayHi", int.class, Consumer.class, Runnable.class); 128 private_say_hi_method = Transform.class.getDeclaredMethod( 129 "privateSayHi", int.class, Consumer.class, Runnable.class); 130 } catch (Exception e) { 131 System.out.println("Unable to find methods!"); 132 e.printStackTrace(System.out); 133 return; 134 } 135 // Makes sure the stack is the way we want it for the test and does the redefinition. It will 136 // set the retry boolean to true if we need to go around again due to jit code being GCd. 137 Runnable do_redefinition = () -> { 138 if (has_jit && 139 (Main.isInterpretedFunction(say_hi_method, true) || 140 Main.isInterpretedFunction(private_say_hi_method, true))) { 141 // Try again. We are not running the right jitted methods/cannot redefine them now. 142 retry = true; 143 } else { 144 // Actually do the redefinition. The stack looks good. 145 retry = false; 146 reporter.accept("transforming calling function"); 147 doCommonClassRedefinition(Transform.class, CLASS_BYTES, DEX_BYTES); 148 } 149 }; 150 do { 151 // Run ensureJitCompiled here since it might get GCd 152 ensureJitCompiled(Transform.class, "sayHi"); 153 ensureJitCompiled(Transform.class, "privateSayHi"); 154 // Clear output. 155 reporter.clear(); 156 t.sayHi(2, reporter, () -> { reporter.accept("Not doing anything here"); }); 157 t.sayHi(2, reporter, do_redefinition); 158 t.sayHi(2, reporter, () -> { reporter.accept("Not doing anything here"); }); 159 } while(retry); 160 System.out.println(reporter.getOutput()); 161 } 162 hasJit()163 private static native boolean hasJit(); 164 isInterpretedFunction(Method m, boolean require_deoptimizable)165 private static native boolean isInterpretedFunction(Method m, boolean require_deoptimizable); 166 ensureJitCompiled(Class c, String name)167 private static native void ensureJitCompiled(Class c, String name); 168 } 169