Register transformation hook only when needed

This prevents us from attempting to transform classes when not needed.
Every transformation is expensive, hence we want to avoid them if we
can.

If another agent triggers transformations of mocked classes the objects
of the classes will become unmocked without knowledge of mockito breaking
many assuptions.

Having another agent registered is very rare, i.e. currently the only
other usage is profiling using Android Studio. Still, we want to support
this, hence we might want to revert this change later.

Wifi-tests runtime 590s -> 417s

Test: frameworks/opt/net/wifi/tests/wifitests/runtests.sh
      atest CtsInlineMockingTestCases CtsMockingTestCase
Bug: 72052158

Change-Id: I2556e749806bed4f80697d5cba38a8d8a2f7ce6a
3 files changed
tree: 837ff8a42257dbad91e62e454811d115469b189c
  1. dexmaker/
  2. dexmaker-mockito/
  3. dexmaker-mockito-inline/
  4. dexmaker-mockito-inline-dispatcher/
  5. dexmaker-mockito-inline-tests/
  6. dexmaker-mockito-tests/
  7. dexmaker-tests/
  8. lib/
  9. .gitignore
  10. Android.bp
  11. Android.mk
  12. AndroidManifest.xml
  13. bug-8108255.patch
  14. build.gradle
  15. CleanSpec.mk
  16. COPYING
  17. LICENSE
  18. MODULE_LICENSE_APACHE2
  19. NOTICE
  20. OWNERS
  21. README.md
  22. README.version
  23. update_source.sh
README.md

A Java-language API for doing compile time or runtime code generation targeting the Dalvik VM. Unlike cglib or ASM, this library creates Dalvik .dex files instead of Java .class files.

It has a small, close-to-the-metal API. This API mirrors the Dalvik bytecode specification giving you tight control over the bytecode emitted. Code is generated instruction-by-instruction; you bring your own abstract syntax tree if you need one. And since it uses Dalvik's dx tool as a backend, you get efficient register allocation and regular/wide instruction selection for free.

Class Proxies

Dexmaker includes a stock code generator for class proxies. If you just want to do AOP or class mocking, you don't need to mess around with bytecodes.

Mockito Mocks

Dexmaker includes class proxy support for Mockito. Add the mockito and the dexmaker .jar files to your Android test project's libs/ directory and you can use Mockito in your Android unit tests.

This requires Mockito 1.10.5 or newer.

Runtime Code Generation

This example generates a class and a method. It then loads that class into the current process and invokes its method.

public final class HelloWorldMaker {
    public static void main(String[] args) throws Exception {
        DexMaker dexMaker = new DexMaker();

        // Generate a HelloWorld class.
        TypeId<?> helloWorld = TypeId.get("LHelloWorld;");
        dexMaker.declare(helloWorld, "HelloWorld.generated", Modifier.PUBLIC, TypeId.OBJECT);
        generateHelloMethod(dexMaker, helloWorld);

        // Create the dex file and load it.
        File outputDir = new File(".");
        ClassLoader loader = dexMaker.generateAndLoad(HelloWorldMaker.class.getClassLoader(),
                outputDir, outputDir);
        Class<?> helloWorldClass = loader.loadClass("HelloWorld");

        // Execute our newly-generated code in-process.
        helloWorldClass.getMethod("hello").invoke(null);
    }

    /**
     * Generates Dalvik bytecode equivalent to the following method.
     *    public static void hello() {
     *        int a = 0xabcd;
     *        int b = 0xaaaa;
     *        int c = a - b;
     *        String s = Integer.toHexString(c);
     *        System.out.println(s);
     *        return;
     *    }
     */
    private static void generateHelloMethod(DexMaker dexMaker, TypeId<?> declaringType) {
        // Lookup some types we'll need along the way.
        TypeId<System> systemType = TypeId.get(System.class);
        TypeId<PrintStream> printStreamType = TypeId.get(PrintStream.class);

        // Identify the 'hello()' method on declaringType.
        MethodId hello = declaringType.getMethod(TypeId.VOID, "hello");

        // Declare that method on the dexMaker. Use the returned Code instance
        // as a builder that we can append instructions to.
        Code code = dexMaker.declare(hello, Modifier.STATIC | Modifier.PUBLIC);

        // Declare all the locals we'll need up front. The API requires this.
        Local<Integer> a = code.newLocal(TypeId.INT);
        Local<Integer> b = code.newLocal(TypeId.INT);
        Local<Integer> c = code.newLocal(TypeId.INT);
        Local<String> s = code.newLocal(TypeId.STRING);
        Local<PrintStream> localSystemOut = code.newLocal(printStreamType);

        // int a = 0xabcd;
        code.loadConstant(a, 0xabcd);

        // int b = 0xaaaa;
        code.loadConstant(b, 0xaaaa);

        // int c = a - b;
        code.op(BinaryOp.SUBTRACT, c, a, b);

        // String s = Integer.toHexString(c);
        MethodId<Integer, String> toHexString
                = TypeId.get(Integer.class).getMethod(TypeId.STRING, "toHexString", TypeId.INT);
        code.invokeStatic(toHexString, s, c);

        // System.out.println(s);
        FieldId<System, PrintStream> systemOutField = systemType.getField(printStreamType, "out");
        code.sget(systemOutField, localSystemOut);
        MethodId<PrintStream, Void> printlnMethod = printStreamType.getMethod(
                TypeId.VOID, "println", TypeId.STRING);
        code.invokeVirtual(printlnMethod, null, localSystemOut, s);

        // return;
        code.returnVoid();
    }
}

Use it in your app

Maven users can get dexmaker from Sonatype's central repository. The Mockito dependency is optional.

  <dependency>
    <groupId>com.google.dexmaker</groupId>
    <artifactId>dexmaker</artifactId>
    <version>1.2</version>
  </dependency>
  <dependency>
    <groupId>com.google.dexmaker</groupId>
    <artifactId>dexmaker-mockito</artifactId>
    <version>1.2</version>
  </dependency>

Download dexmaker-1.2.jar and dexmaker-mockito-1.2.jar.

Run the Unit Tests

The unit tests for dexmaker must be run on a dalvikvm. In order to do this, you can use Vogar in the following fashion:

$ java -jar vogar.jar --mode device --sourcepath /path/to/dexmaker/dexmaker/src/test/java --sourcepath /path/to/dexmaker/dexmaker/src/main/java --sourcepath /path/to/dexmaker/dx/src/main/java --device-dir /data/dexmaker /path/to/dexmaker/dexmaker/src/test/

Download vogar.jar.