dalvik/system/DexClassLoaderTest.java

/*
 * Copyright (C) 2011 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 dalvik.system;

import java.io.FilenameFilter;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import libcore.io.Streams;
import junit.framework.TestCase;

/**
 * Tests for the class {@link DexClassLoader}.
 */
public class DexClassLoaderTest extends TestCase {
    // Use /data not /sdcard because optimized cannot be noexec mounted
    private static final File WORKING_DIR;
    static {
      // First try to use the test runner directory for cts, fall back to
      // shell-writable directory for vogar
      File runner_dir = new File("/data/data/android.core.tests.runner");
      if (runner_dir.exists()) {
        WORKING_DIR = runner_dir;
      } else {
        WORKING_DIR = new File("/data/local/tmp");
      }
    }
    private static final File TMP_DIR = new File(WORKING_DIR, "loading-test");
    private static final String PACKAGE_PATH = "dalvik/system/";
    private static final String JAR_NAME = "loading-test.jar";
    private static final String DEX_NAME = "loading-test.dex";
    private static final String JAR2_NAME = "loading-test2.jar";
    private static final String DEX2_NAME = "loading-test2.dex";
    private static final File JAR_FILE = new File(TMP_DIR, JAR_NAME);
    private static final File DEX_FILE = new File(TMP_DIR, DEX_NAME);
    private static final File JAR2_FILE = new File(TMP_DIR, JAR2_NAME);
    private static final File DEX2_FILE = new File(TMP_DIR, DEX2_NAME);
    private static final File DEFAULT_OPTIMIZED_DIR = new File(TMP_DIR, "optimized");
    // Init tests need to use different optimized directories because the tests are executed in the
    // same runtime. This means we can't reliably count the number of generated file since they
    // might be cached by the runtime.
    private static final File INIT1_OPTIMIZED_DIR = new File(TMP_DIR, "optimized_init1");
    private static final File INIT2_OPTIMIZED_DIR = new File(TMP_DIR, "optimized_init2");

    private static enum Configuration {
        /** just one classpath element, a raw dex file */
        ONE_DEX(1, DEX_FILE),
        ONE_DEX_INIT(INIT1_OPTIMIZED_DIR, 1, DEX_FILE),

        /** just one classpath element, a jar file */
        ONE_JAR(1, JAR_FILE),
        ONE_JAR_INIT(INIT1_OPTIMIZED_DIR, 1, JAR_FILE),

        /** two classpath elements, both raw dex files */
        TWO_DEX(2, DEX_FILE, DEX2_FILE),
        TWO_DEX_INIT(INIT2_OPTIMIZED_DIR, 2, DEX_FILE, DEX2_FILE),

        /** two classpath elements, both jar files */
        TWO_JAR(2, JAR_FILE, JAR2_FILE),
        TWO_JAR_INIT(INIT2_OPTIMIZED_DIR, 2, JAR_FILE, JAR2_FILE);

        public final int expectedFiles;
        public final File optimizedDir;
        public final String path;

        Configuration(int expectedFiles, File... files) {
            this(DEFAULT_OPTIMIZED_DIR, expectedFiles, files);
        }

        Configuration(File optimizedDir, int expectedFiles, File... files) {
            assertTrue(files != null && files.length > 0);

            this.expectedFiles = expectedFiles;
            this.optimizedDir = optimizedDir;
            String path = files[0].getAbsolutePath();
            for (int i = 1; i < files.length; i++) {
                path += File.pathSeparator + files[i].getAbsolutePath();
            }
            this.path = path;
        }
    }

    protected void setUp() throws Exception {
        assertTrue(TMP_DIR.exists() || TMP_DIR.mkdirs());
        assertTrue(DEFAULT_OPTIMIZED_DIR.exists() || DEFAULT_OPTIMIZED_DIR.mkdirs());
        assertTrue(INIT1_OPTIMIZED_DIR.exists() || INIT1_OPTIMIZED_DIR.mkdirs());
        assertTrue(INIT2_OPTIMIZED_DIR.exists() || INIT2_OPTIMIZED_DIR.mkdirs());

        ClassLoader cl = DexClassLoaderTest.class.getClassLoader();
        copyResource(cl, JAR_NAME, JAR_FILE);
        copyResource(cl, DEX_NAME, DEX_FILE);
        copyResource(cl, JAR2_NAME, JAR2_FILE);
        copyResource(cl, DEX2_NAME, DEX2_FILE);
    }

    protected void tearDown() {
        cleanUpDir(DEFAULT_OPTIMIZED_DIR);
        cleanUpDir(INIT1_OPTIMIZED_DIR);
        cleanUpDir(INIT2_OPTIMIZED_DIR);
    }

    private void cleanUpDir(File dir) {
        if (!dir.isDirectory()) {
            return;
        }
        File[] files = dir.listFiles();
        for (File file : files) {
            assertTrue(file.delete());
        }
    }

    /**
     * Copy a resource in the package directory to the indicated
     * target file, but only if the target file doesn't exist.
     */
    private static void copyResource(ClassLoader loader, String resourceName,
            File destination) throws IOException {
        if (destination.exists()) {
            return;
        }

        InputStream in = loader.getResourceAsStream(PACKAGE_PATH + resourceName);
        if (in == null) {
            throw new IllegalStateException("Unable to find resource: " + PACKAGE_PATH + resourceName);
        }


        try (FileOutputStream out = new FileOutputStream(destination)) {
            Streams.copy(in, out);
        } finally {
            in.close();
        }
    }

    /**
     * Helper to construct an instance to test.
     *
     * @param config how to configure the classpath
     */
    private static DexClassLoader createInstance(Configuration config) {
        return new DexClassLoader(
            config.path, config.optimizedDir.getAbsolutePath(), null,
            ClassLoader.getSystemClassLoader());
    }

    /**
     * Helper to construct an instance to test, using the jar file as
     * the source, and call a named no-argument static method on a
     * named class.
     *
     * @param config how to configure the classpath
     */
    public static Object createInstanceAndCallStaticMethod(
            Configuration config, String className, String methodName)
            throws ClassNotFoundException, NoSuchMethodException,
            IllegalAccessException, InvocationTargetException {
        DexClassLoader dcl = createInstance(config);
        Class c = dcl.loadClass(className);
        Method m = c.getMethod(methodName, (Class[]) null);
        return m.invoke(null, (Object[]) null);
    }

    /*
     * Tests that are parametric with respect to whether to use a jar
     * file or a dex file as the source of the code
     */

    static final FilenameFilter DEX_FILE_NAME_FILTER = new FilenameFilter() {
        @Override
        public boolean accept(File file, String s) {
            return s.endsWith(".dex");
        }
    };

    /**
     * Just a trivial test of construction. This one merely makes
     * sure that a valid construction doesn't fail. It doesn't try
     * to verify anything about the constructed instance, other than
     * checking for the existence of optimized dex files.
     */
    private static void test_init(Configuration config) {
        createInstance(config);

        int expectedFiles = config.expectedFiles;
        int actualFiles = config.optimizedDir.listFiles(DEX_FILE_NAME_FILTER).length;

        assertEquals(expectedFiles, actualFiles);
    }

    /**
     * Check that a class in the jar/dex file may be used successfully. In this
     * case, a trivial static method is called.
     */
    private static void test_simpleUse(Configuration config) throws Exception {
        String result = (String)
            createInstanceAndCallStaticMethod(config, "test.Test1", "test");

        assertSame("blort", result);
    }

    /*
     * All the following tests are just pass-throughs to test code
     * that lives inside the loading-test dex/jar file.
     */

    private static void test_constructor(Configuration config)
            throws Exception {
        createInstanceAndCallStaticMethod(
            config, "test.TestMethods", "test_constructor");
    }

    private static void test_callStaticMethod(Configuration config)
            throws Exception {
        createInstanceAndCallStaticMethod(
            config, "test.TestMethods", "test_callStaticMethod");
    }

    private static void test_getStaticVariable(Configuration config)
            throws Exception {
        createInstanceAndCallStaticMethod(
            config, "test.TestMethods", "test_getStaticVariable");
    }

    private static void test_callInstanceMethod(Configuration config)
            throws Exception {
        createInstanceAndCallStaticMethod(
            config, "test.TestMethods", "test_callInstanceMethod");
    }

    private static void test_getInstanceVariable(Configuration config)
            throws Exception {
        createInstanceAndCallStaticMethod(
            config, "test.TestMethods", "test_getInstanceVariable");
    }

    private static void test_diff_constructor(Configuration config)
            throws Exception {
        createInstanceAndCallStaticMethod(
            config, "test.TestMethods", "test_diff_constructor");
    }

    private static void test_diff_callStaticMethod(Configuration config)
            throws Exception {
        createInstanceAndCallStaticMethod(
            config, "test.TestMethods", "test_diff_callStaticMethod");
    }

    private static void test_diff_getStaticVariable(Configuration config)
            throws Exception {
        createInstanceAndCallStaticMethod(
            config, "test.TestMethods", "test_diff_getStaticVariable");
    }

    private static void test_diff_callInstanceMethod(Configuration config)
            throws Exception {
        createInstanceAndCallStaticMethod(
            config, "test.TestMethods", "test_diff_callInstanceMethod");
    }

    private static void test_diff_getInstanceVariable(Configuration config)
            throws Exception {
        createInstanceAndCallStaticMethod(
            config, "test.TestMethods", "test_diff_getInstanceVariable");
    }

    /*
     * These methods are all essentially just calls to the
     * parametrically-defined tests above.
     */

    // ONE_JAR

    public void test_oneJar_init() throws Exception {
        test_init(Configuration.ONE_JAR_INIT);
    }

    public void test_oneJar_simpleUse() throws Exception {
        test_simpleUse(Configuration.ONE_JAR);
    }

    public void test_oneJar_constructor() throws Exception {
        test_constructor(Configuration.ONE_JAR);
    }

    public void test_oneJar_callStaticMethod() throws Exception {
        test_callStaticMethod(Configuration.ONE_JAR);
    }

    public void test_oneJar_getStaticVariable() throws Exception {
        test_getStaticVariable(Configuration.ONE_JAR);
    }

    public void test_oneJar_callInstanceMethod() throws Exception {
        test_callInstanceMethod(Configuration.ONE_JAR);
    }

    public void test_oneJar_getInstanceVariable() throws Exception {
        test_getInstanceVariable(Configuration.ONE_JAR);
    }

    // ONE_DEX

    public void test_oneDex_init() throws Exception {
        test_init(Configuration.ONE_DEX_INIT);
    }

    public void test_oneDex_simpleUse() throws Exception {
        test_simpleUse(Configuration.ONE_DEX);
    }

    public void test_oneDex_constructor() throws Exception {
        test_constructor(Configuration.ONE_DEX);
    }

    public void test_oneDex_callStaticMethod() throws Exception {
        test_callStaticMethod(Configuration.ONE_DEX);
    }

    public void test_oneDex_getStaticVariable() throws Exception {
        test_getStaticVariable(Configuration.ONE_DEX);
    }

    public void test_oneDex_callInstanceMethod() throws Exception {
        test_callInstanceMethod(Configuration.ONE_DEX);
    }

    public void test_oneDex_getInstanceVariable() throws Exception {
        test_getInstanceVariable(Configuration.ONE_DEX);
    }

    // TWO_JAR

    public void test_twoJar_init() throws Exception {
        test_init(Configuration.TWO_JAR_INIT);
    }

    public void test_twoJar_simpleUse() throws Exception {
        test_simpleUse(Configuration.TWO_JAR);
    }

    public void test_twoJar_constructor() throws Exception {
        test_constructor(Configuration.TWO_JAR);
    }

    public void test_twoJar_callStaticMethod() throws Exception {
        test_callStaticMethod(Configuration.TWO_JAR);
    }

    public void test_twoJar_getStaticVariable() throws Exception {
        test_getStaticVariable(Configuration.TWO_JAR);
    }

    public void test_twoJar_callInstanceMethod() throws Exception {
        test_callInstanceMethod(Configuration.TWO_JAR);
    }

    public void test_twoJar_getInstanceVariable() throws Exception {
        test_getInstanceVariable(Configuration.TWO_JAR);
    }

    public static void test_twoJar_diff_constructor() throws Exception {
        test_diff_constructor(Configuration.TWO_JAR);
    }

    public static void test_twoJar_diff_callStaticMethod() throws Exception {
        test_diff_callStaticMethod(Configuration.TWO_JAR);
    }

    public static void test_twoJar_diff_getStaticVariable() throws Exception {
        test_diff_getStaticVariable(Configuration.TWO_JAR);
    }

    public static void test_twoJar_diff_callInstanceMethod()
            throws Exception {
        test_diff_callInstanceMethod(Configuration.TWO_JAR);
    }

    public static void test_twoJar_diff_getInstanceVariable()
            throws Exception {
        test_diff_getInstanceVariable(Configuration.TWO_JAR);
    }

    // TWO_DEX

    public void test_twoDex_init() throws Exception {
        test_init(Configuration.TWO_DEX_INIT);
    }

    public void test_twoDex_simpleUse() throws Exception {
        test_simpleUse(Configuration.TWO_DEX);
    }

    public void test_twoDex_constructor() throws Exception {
        test_constructor(Configuration.TWO_DEX);
    }

    public void test_twoDex_callStaticMethod() throws Exception {
        test_callStaticMethod(Configuration.TWO_DEX);
    }

    public void test_twoDex_getStaticVariable() throws Exception {
        test_getStaticVariable(Configuration.TWO_DEX);
    }

    public void test_twoDex_callInstanceMethod() throws Exception {
        test_callInstanceMethod(Configuration.TWO_DEX);
    }

    public void test_twoDex_getInstanceVariable() throws Exception {
        test_getInstanceVariable(Configuration.TWO_DEX);
    }

    public static void test_twoDex_diff_constructor() throws Exception {
        test_diff_constructor(Configuration.TWO_DEX);
    }

    public static void test_twoDex_diff_callStaticMethod() throws Exception {
        test_diff_callStaticMethod(Configuration.TWO_DEX);
    }

    public static void test_twoDex_diff_getStaticVariable() throws Exception {
        test_diff_getStaticVariable(Configuration.TWO_DEX);
    }

    public static void test_twoDex_diff_callInstanceMethod()
            throws Exception {
        test_diff_callInstanceMethod(Configuration.TWO_DEX);
    }

    public static void test_twoDex_diff_getInstanceVariable()
            throws Exception {
        test_diff_getInstanceVariable(Configuration.TWO_DEX);
    }

    /*
     * Tests specifically for resource-related functionality.  Since
     * raw dex files don't contain resources, these test only work
     * with jar files. The first couple methods here are helpers,
     * and they are followed by the tests per se.
     */

    /**
     * Check that a given resource (by name) is retrievable and contains
     * the given expected contents.
     */
    private static void test_directGetResourceAsStream(Configuration config,
            String resourceName, String expectedContents)
            throws Exception {
        DexClassLoader dcl = createInstance(config);
        InputStream in = dcl.getResourceAsStream(resourceName);
        byte[] contents = Streams.readFully(in);
        String s = new String(contents, "UTF-8");

        assertEquals(expectedContents, s);
    }

    /**
     * Check that a resource in the jar file is retrievable and contains
     * the expected contents.
     */
    private static void test_directGetResourceAsStream(Configuration config)
            throws Exception {
        test_directGetResourceAsStream(
            config, "test/Resource1.txt", "Muffins are tasty!\n");
    }

    /**
     * Check that a resource in the jar file can be retrieved from
     * a class within that jar file.
     */
    private static void test_getResourceAsStream(Configuration config)
            throws Exception {
        createInstanceAndCallStaticMethod(
            config, "test.TestMethods", "test_getResourceAsStream");
    }

    public void test_oneJar_directGetResourceAsStream() throws Exception {
        test_directGetResourceAsStream(Configuration.ONE_JAR);
    }

    public void test_oneJar_getResourceAsStream() throws Exception {
        test_getResourceAsStream(Configuration.ONE_JAR);
    }

    public void test_twoJar_directGetResourceAsStream() throws Exception {
        test_directGetResourceAsStream(Configuration.TWO_JAR);
    }

    public void test_twoJar_getResourceAsStream() throws Exception {
        test_getResourceAsStream(Configuration.TWO_JAR);
    }

    /**
     * Check that a resource in the second jar file is retrievable and
     * contains the expected contents.
     */
    public void test_twoJar_diff_directGetResourceAsStream()
            throws Exception {
        test_directGetResourceAsStream(
            Configuration.TWO_JAR, "test2/Resource2.txt",
            "Who doesn't like a good biscuit?\n");
    }

    /**
     * Check that a resource in a jar file can be retrieved from
     * a class within the other jar file.
     */
    public void test_twoJar_diff_getResourceAsStream()
            throws Exception {
        createInstanceAndCallStaticMethod(
            Configuration.TWO_JAR, "test.TestMethods",
            "test_diff_getResourceAsStream");
    }
}