1 /* 2 * Copyright (c) 2018, Oracle and/or its affiliates. All rights reserved. 3 * 4 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. 5 * 6 * The contents of this file are subject to the terms of either the Universal Permissive License 7 * v 1.0 as shown at http://oss.oracle.com/licenses/upl 8 * 9 * or the following license: 10 * 11 * Redistribution and use in source and binary forms, with or without modification, are permitted 12 * provided that the following conditions are met: 13 * 14 * 1. Redistributions of source code must retain the above copyright notice, this list of conditions 15 * and the following disclaimer. 16 * 17 * 2. Redistributions in binary form must reproduce the above copyright notice, this list of 18 * conditions and the following disclaimer in the documentation and/or other materials provided with 19 * the distribution. 20 * 21 * 3. Neither the name of the copyright holder nor the names of its contributors may be used to 22 * endorse or promote products derived from this software without specific prior written permission. 23 * 24 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR 25 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND 26 * FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR 27 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 28 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 29 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, 30 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY 31 * WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 32 */ 33 package org.openjdk.jmc.agent.test; 34 35 import static org.junit.Assert.assertFalse; 36 import static org.junit.Assert.assertNotNull; 37 import static org.junit.Assert.assertTrue; 38 39 import java.io.IOException; 40 import java.io.PrintWriter; 41 import java.lang.instrument.IllegalClassFormatException; 42 import java.util.concurrent.atomic.AtomicInteger; 43 import java.util.logging.Level; 44 45 import javax.xml.stream.XMLStreamException; 46 47 import org.junit.Test; 48 import org.objectweb.asm.ClassReader; 49 import org.objectweb.asm.Type; 50 import org.objectweb.asm.util.CheckClassAdapter; 51 import org.objectweb.asm.util.TraceClassVisitor; 52 import org.openjdk.jmc.agent.Agent; 53 import org.openjdk.jmc.agent.TransformRegistry; 54 import org.openjdk.jmc.agent.Transformer; 55 import org.openjdk.jmc.agent.impl.DefaultTransformRegistry; 56 import org.openjdk.jmc.agent.test.util.TestToolkit; 57 58 public class TestJFRTransformer { 59 private static AtomicInteger runCount = new AtomicInteger(0); 60 61 @Test 62 public void testRunTransforms() throws XMLStreamException, IllegalClassFormatException, IOException { 63 TransformRegistry registry = DefaultTransformRegistry 64 .from(TestToolkit.getProbesXML("RunTransforms" + runCount.getAndIncrement())); //$NON-NLS-1$ 65 66 assertTrue(registry.hasPendingTransforms(Type.getInternalName(InstrumentMe.class))); 67 68 Transformer jfrTransformer = new Transformer(registry); 69 byte[] transformedClass = jfrTransformer.transform(InstrumentMe.class.getClassLoader(), 70 Type.getInternalName(InstrumentMe.class), InstrumentMe.class, null, 71 TestToolkit.getByteCode(InstrumentMe.class)); 72 73 assertNotNull(transformedClass); 74 assertFalse(registry.hasPendingTransforms(Type.getInternalName(InstrumentMe.class))); 75 76 if (Agent.getLogger().isLoggable(Level.FINE)) { 77 // If we've asked for verbose information, we write the generated class 78 // and also dump the registry contents to stdout. 79 TraceClassVisitor visitor = new TraceClassVisitor(new PrintWriter(System.out)); 80 CheckClassAdapter checkAdapter = new CheckClassAdapter(visitor); 81 ClassReader reader = new ClassReader(transformedClass); 82 reader.accept(checkAdapter, 0); 83 System.out.println(registry); 84 } 85 } 86 }