test/1901-get-bytecodes/src/art/Test1901.java

/*
 * Copyright (C) 2017 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.
 */

package art;

import java.lang.reflect.Constructor;
import java.lang.reflect.Method;
import java.nio.ByteBuffer;
import java.util.Arrays;
import java.util.Base64;

public class Test1901 {
  // Class & Dex file containing the following class.
  // Using this representation to prevent any changes to the compiler or the file formats from
  // changing the output of this test.
  //
  // package art;
  // public class Target {
  //   public void doNothing() {
  //     return;
  //   }
  //
  //   public static void staticNothing() {
  //     return;
  //   }
  //
  //   public void sayHi() {
  //     System.out.println("hello");
  //   }
  // }
  public static byte[] CLASS_BYTES = Base64.getDecoder().decode(
    "yv66vgAAADQAHgoABgAQCQARABIIABMKABQAFQcAFgcAFwEABjxpbml0PgEAAygpVgEABENvZGUB" +
    "AA9MaW5lTnVtYmVyVGFibGUBAAlkb05vdGhpbmcBAA1zdGF0aWNOb3RoaW5nAQAFc2F5SGkBAApT" +
    "b3VyY2VGaWxlAQALVGFyZ2V0LmphdmEMAAcACAcAGAwAGQAaAQAFaGVsbG8HABsMABwAHQEACmFy" +
    "dC9UYXJnZXQBABBqYXZhL2xhbmcvT2JqZWN0AQAQamF2YS9sYW5nL1N5c3RlbQEAA291dAEAFUxq" +
    "YXZhL2lvL1ByaW50U3RyZWFtOwEAE2phdmEvaW8vUHJpbnRTdHJlYW0BAAdwcmludGxuAQAVKExq" +
    "YXZhL2xhbmcvU3RyaW5nOylWACEABQAGAAAAAAAEAAEABwAIAAEACQAAAB0AAQABAAAABSq3AAGx" +
    "AAAAAQAKAAAABgABAAAAAgABAAsACAABAAkAAAAZAAAAAQAAAAGxAAAAAQAKAAAABgABAAAABAAJ" +
    "AAwACAABAAkAAAAZAAAAAAAAAAGxAAAAAQAKAAAABgABAAAACAABAA0ACAABAAkAAAAlAAIAAQAA" +
    "AAmyAAISA7YABLEAAAABAAoAAAAKAAIAAAAMAAgADQABAA4AAAACAA8=");
  public static byte[] DEX_BYTES = Base64.getDecoder().decode(
    "ZGV4CjAzNQAbYkxNjiZ8a+fNWF4smR2+uXbrq88/FNoYAwAAcAAAAHhWNBIAAAAAAAAAAHgCAAAP" +
    "AAAAcAAAAAYAAACsAAAAAgAAAMQAAAABAAAA3AAAAAYAAADkAAAAAQAAABQBAADkAQAANAEAAJoB" +
    "AACiAQAAsAEAAMcBAADbAQAA7wEAAAMCAAAQAgAAEwIAABcCAAAiAgAAKQIAAC4CAAA3AgAAPgIA" +
    "AAEAAAACAAAAAwAAAAQAAAAFAAAABwAAAAcAAAAFAAAAAAAAAAgAAAAFAAAAlAEAAAQAAQALAAAA" +
    "AAAAAAAAAAAAAAAACQAAAAAAAAANAAAAAAAAAA4AAAABAAEADAAAAAIAAAAAAAAAAAAAAAEAAAAC" +
    "AAAAAAAAAAYAAAAAAAAAYgIAAAAAAAABAAEAAQAAAE0CAAAEAAAAcBAFAAAADgAAAAAAAAAAAFIC" +
    "AAABAAAADgAAAAEAAQAAAAAAVwIAAAEAAAAOAAAAAwABAAIAAABcAgAACAAAAGIAAAAaAQoAbiAE" +
    "ABAADgABAAAAAwAGPGluaXQ+AAxMYXJ0L1RhcmdldDsAFUxqYXZhL2lvL1ByaW50U3RyZWFtOwAS" +
    "TGphdmEvbGFuZy9PYmplY3Q7ABJMamF2YS9sYW5nL1N0cmluZzsAEkxqYXZhL2xhbmcvU3lzdGVt" +
    "OwALVGFyZ2V0LmphdmEAAVYAAlZMAAlkb05vdGhpbmcABWhlbGxvAANvdXQAB3ByaW50bG4ABXNh" +
    "eUhpAA1zdGF0aWNOb3RoaW5nAAIABw4ACAAHDgAEAAcOAAwABw54AAAAAgIAgYAEtAIDCcwCAQHg" +
    "AgEB9AINAAAAAAAAAAEAAAAAAAAAAQAAAA8AAABwAAAAAgAAAAYAAACsAAAAAwAAAAIAAADEAAAA" +
    "BAAAAAEAAADcAAAABQAAAAYAAADkAAAABgAAAAEAAAAUAQAAASAAAAQAAAA0AQAAARAAAAEAAACU" +
    "AQAAAiAAAA8AAACaAQAAAyAAAAQAAABNAgAAACAAAAEAAABiAgAAABAAAAEAAAB4AgAA");

  public static byte[][] DO_NOTHING_BYTECODES = new byte[][] {
    // Dex Bytecodes for doNothing
    // 0e00           |0000: return-void
    new byte[] { 14, 0 },
    // Java bytecodes
    // 0: return
    new byte[] { -79 },
  };

  public static byte[][] STATIC_NOTHING_BYTECODES = new byte[][] {
    // Dex Bytecodes for staticNothing
    // 0e00           |0000: return-void
    new byte[] { 14, 0 },
    // Java bytecodes
    // 0: return
    new byte[] { -79 },
  };

  public static byte[][] SAY_HI_NOTHING_BYTECODES = new byte[][] {
    // Dex Bytecodes for sayHi
    // 6200 0000      |0000: sget-object v0, Ljava/lang/System;.out:Ljava/io/PrintStream; // field@0000
    // 1a01 0a00      |0002: const-string v1, "hello" // string@000a
    // 6e20 0400 1000 |0004: invoke-virtual {v0, v1}, Ljava/io/PrintStream;.println:(Ljava/lang/String;)V // method@0004
    // 0e00           |0007: return-void
    new byte[] { 98, 0, 0, 0, 26, 1, 10, 0, 110, 32, 4, 0, 16, 0, 14, 0 },
    // Java bytecodes
    // 0: getstatic     #2  // Field java/lang/System.out:Ljava/io/PrintStream;
    // 3: ldc           #3  // String hello
    // 5: invokevirtual #4  // Method java/io/PrintStream.println:(Ljava/lang/String;)V
    // 8: return
    new byte[] { -78, 0, 2, 18, 3, -74, 0, 4, -79 },
  };

  public static ClassLoader getClassLoader() throws Exception {
    try {
      Class<?> class_loader_class = Class.forName("dalvik.system.InMemoryDexClassLoader");
      Constructor<?> ctor = class_loader_class.getConstructor(ByteBuffer.class, ClassLoader.class);
      // We are on art since we got the InMemoryDexClassLoader.
      return (ClassLoader)ctor.newInstance(
          ByteBuffer.wrap(DEX_BYTES), Test1901.class.getClassLoader());
    } catch (ClassNotFoundException e) {
      // Running on RI.
      return new ClassLoader(Test1901.class.getClassLoader()) {
        protected Class<?> findClass(String name) throws ClassNotFoundException {
          if (name.equals("art.Target")) {
            return defineClass(name, CLASS_BYTES, 0, CLASS_BYTES.length);
          } else {
            return super.findClass(name);
          }
        }
      };
    }
  }

  public static void CheckMethodBytes(Method m, byte[][] possible_bytecodes) {
    byte[] real_codes = getBytecodes(m);
    for (byte[] pos : possible_bytecodes) {
      if (Arrays.equals(pos, real_codes)) {
        return;
      }
    }
    System.out.println("Unexpected bytecodes for " + m);
    System.out.println("Received: " + Arrays.toString(real_codes));
    System.out.println("Expected one of:");
    for (byte[] pos : possible_bytecodes) {
      System.out.println("\t" + Arrays.toString(pos));
    }
  }

  public static void run() throws Exception {
    Class<?> target = getClassLoader().loadClass("art.Target");
    CheckMethodBytes(target.getDeclaredMethod("doNothing"), DO_NOTHING_BYTECODES);
    CheckMethodBytes(target.getDeclaredMethod("staticNothing"), STATIC_NOTHING_BYTECODES);
    CheckMethodBytes(target.getDeclaredMethod("sayHi"), SAY_HI_NOTHING_BYTECODES);
  }

  public static native byte[] getBytecodes(Method m);
}