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 }