1 /*

     2  * Copyright 2005 Sun Microsystems, Inc.  All Rights Reserved.

     3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

     4  *

     5  * This code is free software; you can redistribute it and/or modify it

     6  * under the terms of the GNU General Public License version 2 only, as

     7  * published by the Free Software Foundation.

     8  *

     9  * This code is distributed in the hope that it will be useful, but WITHOUT

    10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

    11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

    12  * version 2 for more details (a copy is included in the LICENSE file that

    13  * accompanied this code).

    14  *

    15  * You should have received a copy of the GNU General Public License version

    16  * 2 along with this work; if not, write to the Free Software Foundation,

    17  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

    18  *

    19  * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,

    20  * CA 95054 USA or visit www.sun.com if you need additional information or

    21  * have any questions.

    22  */

    23 

    24 /*

    25  *

    26  *

    27  * Basic unit test for Instrumentation appendToBootstrapClassLoaderSearch and

    28  * appendToSystemClassLoaderSearch methods:

    29  *

    30  * 1. Adds a jar file to the bootstrap class loader search; Loads a class known

    31  *    to be in the jar file and checks that it was loaded by the bootstrap class

    32  *    loader.

    33  *

    34  * 2. Adds a jar file to the system class loader search; Loads a class known

    35  *    to be in the jar file and checks that it was indeed loaded by the system

    36  *    class loader.

    37  */

    38 import java.lang.instrument.Instrumentation;

    39 import java.util.jar.JarFile;

    40 import java.io.IOException;

    41 

    42 public class BasicTest {

    43 

    44     // count of failures

    45     static int failures = 0;

    46 

    47     // check that the given class is loaded by the given loader

    48     static void checkLoadedByLoader(String name, ClassLoader loader) {

    49         try {

    50             Class cl = Class.forName(name);

    51             ClassLoader actual = cl.getClassLoader();

    52             String loaderName = (actual == null) ? "boot class loader" : actual.toString();

    53             if (actual != loader) {

    54                 System.out.println("FAIL: " + name + " incorrectly loaded by: " + loaderName);

    55                 failures++;

    56             } else {

    57                 System.out.println("PASS: " + name + " loaded by: " + loaderName);

    58             }

    59         } catch (Exception x) {

    60             System.out.println("FAIL: Unable to load " + name + ":" + x);

    61             failures++;

    62         }

    63     }

    64 

    65     public static void main(String args[]) throws IOException {

    66         JarFile bootclasses = new JarFile("BootSupport.jar");

    67         JarFile agentclasses = new JarFile("AgentSupport.jar");

    68 

    69         Instrumentation ins = Agent.getInstrumentation();

    70 

    71         // Test 1: Add BootSupport.jar to boot class path and check that

    72         // BootSupport is loaded by the bootstrap class loader

    73         ins.appendToBootstrapClassLoaderSearch(bootclasses);

    74         checkLoadedByLoader("BootSupport", null);

    75 

    76         // Test 2: Add AgentSupport.jar to the system class path and check that

    77         // AgentSupport is loaded by the system class loader.

    78         try {

    79             ins.appendToSystemClassLoaderSearch(agentclasses);

    80             checkLoadedByLoader("AgentSupport", ClassLoader.getSystemClassLoader());

    81         } catch (UnsupportedOperationException x) {

    82             System.out.println("System class loader does not support adding to class path");

    83         }

    84 

    85         // throw exception if a test failed

    86         if (failures > 0) {

    87             throw new RuntimeException(failures + " test(s) failed.");

    88         }

    89     }

    90 }