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.jfrnext.impl;
  34 
  35 import java.lang.reflect.Method;
  36 import java.security.ProtectionDomain;
  37 import java.util.logging.Level;
  38 import java.util.logging.Logger;
  39 
  40 import org.objectweb.asm.ClassVisitor;
  41 import org.objectweb.asm.ClassWriter;
  42 import org.objectweb.asm.MethodVisitor;
  43 import org.objectweb.asm.Opcodes;
  44 import org.openjdk.jmc.agent.jfr.JFRTransformDescriptor;
  45 import org.openjdk.jmc.agent.util.TypeUtils;
  46 
  47 public class JFRNextClassVisitor extends ClassVisitor {
  48         private final JFRTransformDescriptor transformDescriptor;
  49         private final ClassLoader definingClassLoader;
  50         private final ProtectionDomain protectionDomain;
  51 
  52         public JFRNextClassVisitor(ClassWriter cv, JFRTransformDescriptor descriptor, ClassLoader definingLoader,
  53                         ProtectionDomain protectionDomain) {
  54                 super(Opcodes.ASM5, cv);
  55                 this.transformDescriptor = descriptor;
  56                 this.definingClassLoader = definingLoader;
  57                 this.protectionDomain = protectionDomain;
  58         }
  59 
  60         @Override
  61         public MethodVisitor visitMethod(int access, String name, String desc, String signature, String[] exceptions) {
  62                 MethodVisitor mv = super.visitMethod(access, name, desc, signature, exceptions);
  63                 if (name.equals(transformDescriptor.getMethod().getName())
  64                                 && desc.equals(transformDescriptor.getMethod().getSignature())) {
  65                         return new JFRNextMethodAdvisor(transformDescriptor, Opcodes.ASM5, mv, access, name, desc);
  66                 }
  67                 return mv;
  68         }
  69 
  70         @Override
  71         public void visitEnd() {
  72                 try {
  73                         reflectiveRegister(generateEventClass());
  74                 } catch (Exception e) {
  75                         Logger.getLogger(JFRNextClassVisitor.class.getName()).log(Level.SEVERE,
  76                                         "Failed to generate event class for " + transformDescriptor.toString(), e); //$NON-NLS-1$
  77                 }
  78                 super.visitEnd();
  79         }
  80 
  81         // NOTE: multi-release jars should let us compile against jdk9 and do a direct call here
  82         private void reflectiveRegister(Class<?> generateEventClass) throws Exception {
  83                 Class<?> jfr = Class.forName("jdk.jfr.FlightRecorder"); //$NON-NLS-1$
  84                 Method registerMethod = jfr.getDeclaredMethod("register", Class.class); //$NON-NLS-1$
  85                 registerMethod.invoke(null,  generateEventClass);               
  86         }
  87 
  88         private Class<?> generateEventClass() throws Exception {
  89                 byte[] eventClass = JFRNextEventClassGenerator.generateEventClass(transformDescriptor);
  90                 return TypeUtils.getUnsafe().defineClass(transformDescriptor.getEventClassName(), eventClass, 0,
  91                                 eventClass.length, definingClassLoader, protectionDomain);
  92         }
  93 }