0N/A/*
2362N/A * Copyright (c) 2005, Oracle and/or its affiliates. All rights reserved.
0N/A * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
0N/A *
0N/A * This code is free software; you can redistribute it and/or modify it
0N/A * under the terms of the GNU General Public License version 2 only, as
0N/A * published by the Free Software Foundation.
0N/A *
0N/A * This code is distributed in the hope that it will be useful, but WITHOUT
0N/A * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
0N/A * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
0N/A * version 2 for more details (a copy is included in the LICENSE file that
0N/A * accompanied this code).
0N/A *
0N/A * You should have received a copy of the GNU General Public License version
0N/A * 2 along with this work; if not, write to the Free Software Foundation,
0N/A * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
0N/A *
2362N/A * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
2362N/A * or visit www.oracle.com if you need additional information or have any
2362N/A * questions.
0N/A */
0N/A
0N/A/*
0N/A *
0N/A *
0N/A * Unit test for Instrumentation appendToBootstrapClassLoaderSearch method.
0N/A * The test works as follows:
0N/A *
0N/A * 1. Load class "Application" and execute the doSomething() method
0N/A *
0N/A * 2. Append Tracer.jar to the boot class path - Tracer.jar has a single
0N/A * class org.tools.Tracer.
0N/A *
0N/A * 3. Redefine class "Application" - the redefined version has an instrumented
0N/A * version of doSomething() that invokes a method in org.tools.Tracer.
0N/A *
0N/A * 4. Re-execute doSomething() - this should provoke the loading of org.tools.Tracer
0N/A * from the jar file. If updated version of doSomething() executes then test
0N/A * passes.
0N/A */
0N/Aimport java.lang.instrument.*;
0N/Aimport java.util.jar.JarFile;
0N/Aimport java.io.*;
0N/A
0N/Apublic class DynamicTest {
0N/A
0N/A public static void main(String args[]) throws Exception {
0N/A
0N/A // Load Application
0N/A Application app = new Application();
0N/A if (app.doSomething() != 1) {
0N/A throw new RuntimeException("Test configuration error - doSomething should return 1");
0N/A }
0N/A
0N/A // Add org.tools.Tracer package to the boot class path
0N/A JarFile jf = new JarFile("Tracer.jar");
0N/A Agent.getInstrumentation().appendToBootstrapClassLoaderSearch(jf);
0N/A
0N/A // Redefine Application with the instrumented version
0N/A File f = new File("InstrumentedApplication.bytes");
0N/A int len = (int)f.length();
0N/A byte[] def = new byte[len];
0N/A
0N/A FileInputStream fis = new FileInputStream(f);
0N/A int nread = 0;
0N/A do {
0N/A int n = fis.read(def, nread, len-nread);
0N/A if (n > 0) {
0N/A nread += n;
0N/A }
0N/A } while (nread < len);
0N/A
0N/A ClassDefinition classDefs = new ClassDefinition(Application.class, def);
0N/A Agent.getInstrumentation().redefineClasses(new ClassDefinition[] { classDefs } );
0N/A
0N/A // Re-execute doSomething() - should get 3 messages printed
0N/A int res = app.doSomething();
0N/A if (res != 3) {
0N/A throw new RuntimeException("FAIL: redefined Application returned: " + res);
0N/A }
0N/A System.out.println("PASS: Test passed.");
0N/A }
0N/A}