Java tutorial
/* * Copyright 2013-2016 the original author or authors. * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package org.glowroot.agent.advicegen; import java.util.List; import java.util.Map; import java.util.concurrent.atomic.AtomicInteger; import javax.annotation.Nullable; import com.google.common.collect.ImmutableMap; import com.google.common.collect.Maps; import org.checkerframework.checker.nullness.qual.RequiresNonNull; import org.objectweb.asm.AnnotationVisitor; import org.objectweb.asm.ClassWriter; import org.objectweb.asm.Label; import org.objectweb.asm.MethodVisitor; import org.objectweb.asm.Type; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import org.glowroot.agent.config.InstrumentationConfig; import org.glowroot.agent.plugin.api.ThreadContext.Priority; import org.glowroot.agent.weaving.Advice; import org.glowroot.agent.weaving.AdviceBuilder; import org.glowroot.agent.weaving.ClassLoaders.LazyDefinedClass; import org.glowroot.agent.weaving.ImmutableLazyDefinedClass; import org.glowroot.wire.api.model.AgentConfigOuterClass.AgentConfig.CaptureKind; import static com.google.common.base.Preconditions.checkNotNull; import static org.objectweb.asm.Opcodes.ACC_FINAL; import static org.objectweb.asm.Opcodes.ACC_PRIVATE; import static org.objectweb.asm.Opcodes.ACC_PUBLIC; import static org.objectweb.asm.Opcodes.ACC_STATIC; import static org.objectweb.asm.Opcodes.ACC_SUPER; import static org.objectweb.asm.Opcodes.ACONST_NULL; import static org.objectweb.asm.Opcodes.ALOAD; import static org.objectweb.asm.Opcodes.ARETURN; import static org.objectweb.asm.Opcodes.BIPUSH; import static org.objectweb.asm.Opcodes.CHECKCAST; import static org.objectweb.asm.Opcodes.GETFIELD; import static org.objectweb.asm.Opcodes.GETSTATIC; import static org.objectweb.asm.Opcodes.GOTO; import static org.objectweb.asm.Opcodes.IFEQ; import static org.objectweb.asm.Opcodes.IFNE; import static org.objectweb.asm.Opcodes.IFNONNULL; import static org.objectweb.asm.Opcodes.INSTANCEOF; import static org.objectweb.asm.Opcodes.INVOKEINTERFACE; import static org.objectweb.asm.Opcodes.INVOKESPECIAL; import static org.objectweb.asm.Opcodes.INVOKESTATIC; import static org.objectweb.asm.Opcodes.INVOKEVIRTUAL; import static org.objectweb.asm.Opcodes.IRETURN; import static org.objectweb.asm.Opcodes.PUTFIELD; import static org.objectweb.asm.Opcodes.PUTSTATIC; import static org.objectweb.asm.Opcodes.RETURN; import static org.objectweb.asm.Opcodes.V1_5; public class AdviceGenerator { private static final Logger logger = LoggerFactory.getLogger(AdviceGenerator.class); private static final AtomicInteger counter = new AtomicInteger(); private final InstrumentationConfig config; private final @Nullable String pluginId; private final String adviceInternalName; private final @Nullable String methodMetaInternalName; private final int uniqueNum; public static ImmutableMap<Advice, LazyDefinedClass> createAdvisors(List<InstrumentationConfig> configs, @Nullable String pluginId, boolean reweavable) { Map<Advice, LazyDefinedClass> advisors = Maps.newHashMap(); for (InstrumentationConfig config : configs) { if (!config.validationErrors().isEmpty()) { continue; } try { LazyDefinedClass lazyAdviceClass = new AdviceGenerator(config, pluginId).generate(); Advice advice = new AdviceBuilder(lazyAdviceClass, reweavable).build(); advisors.put(advice, lazyAdviceClass); } catch (Exception e) { logger.error("error creating advice for advice config: {}", config, e); } } return ImmutableMap.copyOf(advisors); } private AdviceGenerator(InstrumentationConfig config, @Nullable String pluginId) { this.config = config; this.pluginId = pluginId; uniqueNum = counter.incrementAndGet(); adviceInternalName = "org/glowroot/agent/advicegen/GeneratedAdvice" + uniqueNum; if (config.isTraceEntryOrGreater() || !config.transactionNameTemplate().isEmpty() || !config.transactionUserTemplate().isEmpty() || !config.transactionAttributeTemplates().isEmpty()) { // templates are used, so method meta is needed methodMetaInternalName = "org/glowroot/agent/advicegen/GeneratedMethodMeta" + uniqueNum; } else { methodMetaInternalName = null; } } private LazyDefinedClass generate() throws Exception { LazyDefinedClass methodMetaClass = null; if (methodMetaInternalName != null) { methodMetaClass = generateMethodMetaClass(config); } ClassWriter cw = new ClassWriter(ClassWriter.COMPUTE_MAXS + ClassWriter.COMPUTE_FRAMES); String[] interfaces = null; if (!config.enabledProperty().isEmpty() || !config.traceEntryEnabledProperty().isEmpty()) { interfaces = new String[] { "org/glowroot/agent/plugin/api/config/ConfigListener" }; } cw.visit(V1_5, ACC_PUBLIC + ACC_SUPER, adviceInternalName, null, "java/lang/Object", interfaces); addClassAnnotation(cw); addStaticFields(cw); addStaticInitializer(cw); if (pluginId != null && !config.enabledProperty().isEmpty()) { addIsEnabledMethod(cw); } if (config.isTraceEntryOrGreater()) { // methodMetaInternalName is non-null when entry or greater checkNotNull(methodMetaInternalName); addOnBeforeMethod(cw); addOnThrowMethod(cw); addOnReturnMethod(cw); } else if (config.captureKind() == CaptureKind.TIMER) { addOnBeforeMethodTimerOnly(cw); addOnAfterMethodTimerOnly(cw); } else { addOnBeforeMethodOther(cw); } cw.visitEnd(); ImmutableLazyDefinedClass.Builder builder = ImmutableLazyDefinedClass.builder() .type(Type.getObjectType(adviceInternalName)).bytes(cw.toByteArray()); if (methodMetaClass != null) { builder.addDependencies(methodMetaClass); } return builder.build(); } private void addClassAnnotation(ClassWriter cw) { AnnotationVisitor annotationVisitor = cw.visitAnnotation("Lorg/glowroot/agent/plugin/api/weaving/Pointcut;", true); annotationVisitor.visit("className", config.className()); annotationVisitor.visit("classAnnotation", config.classAnnotation()); annotationVisitor.visit("methodDeclaringClassName", config.methodDeclaringClassName()); annotationVisitor.visit("methodName", config.methodName()); annotationVisitor.visit("methodAnnotation", config.methodAnnotation()); AnnotationVisitor arrayAnnotationVisitor = annotationVisitor.visitArray("methodParameterTypes"); for (String methodParameterType : config.methodParameterTypes()) { arrayAnnotationVisitor.visit(null, methodParameterType); } arrayAnnotationVisitor.visitEnd(); String timerName = config.timerName(); if (config.isTimerOrGreater()) { if (timerName.isEmpty()) { annotationVisitor.visit("timerName", "<no timer name provided>"); } else { annotationVisitor.visit("timerName", timerName); } } String nestingGroup = config.nestingGroup(); if (!nestingGroup.isEmpty()) { annotationVisitor.visit("nestingGroup", nestingGroup); } else if (!config.traceEntryCaptureSelfNested()) { annotationVisitor.visit("nestingGroup", "__GeneratedAdvice" + uniqueNum); } annotationVisitor.visit("priority", config.priority()); annotationVisitor.visitEnd(); } private void addStaticFields(ClassWriter cw) { if (pluginId != null) { cw.visitField(ACC_PRIVATE + ACC_FINAL + ACC_STATIC, "configService", "Lorg/glowroot/agent/plugin/api/config/ConfigService;", null, null).visitEnd(); } if (config.isTimerOrGreater()) { cw.visitField(ACC_PRIVATE + ACC_FINAL + ACC_STATIC, "timerName", "Lorg/glowroot/agent/plugin/api/TimerName;", null, null).visitEnd(); } if (!config.enabledProperty().isEmpty()) { cw.visitField(ACC_PRIVATE + ACC_STATIC + ACC_FINAL, "enabled", "Lorg/glowroot/agent/plugin/api/config/BooleanProperty;", null, null).visitEnd(); } if (!config.traceEntryEnabledProperty().isEmpty()) { cw.visitField(ACC_PRIVATE + ACC_STATIC + ACC_FINAL, "entryEnabled", "Lorg/glowroot/agent/plugin/api/config/BooleanProperty;", null, null).visitEnd(); } } private void addStaticInitializer(ClassWriter cw) { MethodVisitor mv = cw.visitMethod(ACC_STATIC, "<clinit>", "()V", null, null); mv.visitCode(); if (pluginId != null) { mv.visitLdcInsn(pluginId); mv.visitMethodInsn(INVOKESTATIC, "org/glowroot/agent/plugin/api/Agent", "getConfigService", "(Ljava/lang/String;)Lorg/glowroot/agent/plugin/api/config/ConfigService;", false); mv.visitFieldInsn(PUTSTATIC, adviceInternalName, "configService", "Lorg/glowroot/agent/plugin/api/config/ConfigService;"); } if (config.isTimerOrGreater()) { mv.visitLdcInsn(Type.getObjectType(adviceInternalName)); mv.visitMethodInsn(INVOKESTATIC, "org/glowroot/agent/plugin/api/Agent", "getTimerName", "(Ljava/lang/Class;)Lorg/glowroot/agent/plugin/api/TimerName;", false); mv.visitFieldInsn(PUTSTATIC, adviceInternalName, "timerName", "Lorg/glowroot/agent/plugin/api/TimerName;"); } if (!config.enabledProperty().isEmpty() && pluginId != null) { mv.visitFieldInsn(GETSTATIC, adviceInternalName, "configService", "Lorg/glowroot/agent/plugin/api/config/ConfigService;"); mv.visitLdcInsn(config.enabledProperty()); mv.visitMethodInsn(INVOKEINTERFACE, "org/glowroot/agent/plugin/api/config/ConfigService", "getBooleanProperty", "(Ljava/lang/String;)Lorg/glowroot/agent/plugin/api/config/BooleanProperty;", true); mv.visitFieldInsn(PUTSTATIC, adviceInternalName, "enabled", "Lorg/glowroot/agent/plugin/api/config/BooleanProperty;"); } if (!config.traceEntryEnabledProperty().isEmpty() && pluginId != null) { mv.visitFieldInsn(GETSTATIC, adviceInternalName, "configService", "Lorg/glowroot/agent/plugin/api/config/ConfigService;"); mv.visitLdcInsn(config.traceEntryEnabledProperty()); mv.visitMethodInsn(INVOKEINTERFACE, "org/glowroot/agent/plugin/api/config/ConfigService", "getBooleanProperty", "(Ljava/lang/String;)Lorg/glowroot/agent/plugin/api/config/BooleanProperty;", true); mv.visitFieldInsn(PUTSTATIC, adviceInternalName, "entryEnabled", "Lorg/glowroot/agent/plugin/api/config/BooleanProperty;"); } mv.visitInsn(RETURN); mv.visitMaxs(0, 0); mv.visitEnd(); } private void addIsEnabledMethod(ClassWriter cw) { MethodVisitor mv = cw.visitMethod(ACC_PUBLIC + ACC_STATIC, "isEnabled", "()Z", null, null); visitAnnotation(mv, "Lorg/glowroot/agent/plugin/api/weaving/IsEnabled;"); mv.visitCode(); mv.visitFieldInsn(GETSTATIC, adviceInternalName, "enabled", "Lorg/glowroot/agent/plugin/api/config/BooleanProperty;"); mv.visitMethodInsn(INVOKEINTERFACE, "org/glowroot/agent/plugin/api/config/BooleanProperty", "value", "()Z", true); mv.visitInsn(IRETURN); mv.visitMaxs(0, 0); mv.visitEnd(); } @RequiresNonNull("methodMetaInternalName") private void addOnBeforeMethod(ClassWriter cw) { MethodVisitor mv = visitOnBeforeMethod(cw, "Lorg/glowroot/agent/plugin/api/TraceEntry;"); mv.visitCode(); if (!config.traceEntryEnabledProperty().isEmpty() && pluginId != null) { mv.visitFieldInsn(GETSTATIC, adviceInternalName, "entryEnabled", "Lorg/glowroot/agent/plugin/api/config/BooleanProperty;"); mv.visitMethodInsn(INVOKEINTERFACE, "org/glowroot/agent/plugin/api/config/BooleanProperty", "value", "()Z", true); Label label = new Label(); mv.visitJumpInsn(IFNE, label); // entryEnabled is false, collect timer only mv.visitVarInsn(ALOAD, 0); mv.visitFieldInsn(GETSTATIC, adviceInternalName, "timerName", "Lorg/glowroot/agent/plugin/api/TimerName;"); mv.visitMethodInsn(INVOKEINTERFACE, "org/glowroot/agent/plugin/api/ThreadContext", "startTimer", "(Lorg/glowroot/agent/plugin/api/TimerName;)" + "Lorg/glowroot/agent/plugin/api/Timer;", true); mv.visitInsn(ARETURN); mv.visitLabel(label); } mv.visitVarInsn(ALOAD, 0); if (config.isTransaction()) { String transactionType = config.transactionType(); if (transactionType.isEmpty()) { mv.visitLdcInsn("<no transaction type provided>"); } else { mv.visitLdcInsn(transactionType); } mv.visitVarInsn(ALOAD, 4); mv.visitMethodInsn(INVOKEVIRTUAL, methodMetaInternalName, "getTransactionNameTemplate", "()Lorg/glowroot/agent/advicegen/MessageTemplate;", false); mv.visitVarInsn(ALOAD, 1); mv.visitVarInsn(ALOAD, 2); mv.visitVarInsn(ALOAD, 3); mv.visitMethodInsn(INVOKESTATIC, "org/glowroot/agent/advicegen/GenericMessageSupplier", "create", "(Lorg/glowroot/agent/advicegen/MessageTemplate;" + "Ljava/lang/Object;Ljava/lang/String;[Ljava/lang/Object;)" + "Lorg/glowroot/agent/advicegen/GenericMessageSupplier;", false); mv.visitMethodInsn(INVOKEVIRTUAL, "org/glowroot/agent/advicegen/GenericMessageSupplier", "getMessageText", "()Ljava/lang/String;", false); } mv.visitVarInsn(ALOAD, 4); mv.visitMethodInsn(INVOKEVIRTUAL, methodMetaInternalName, "getMessageTemplate", "()Lorg/glowroot/agent/advicegen/MessageTemplate;", false); mv.visitVarInsn(ALOAD, 1); mv.visitVarInsn(ALOAD, 2); mv.visitVarInsn(ALOAD, 3); mv.visitMethodInsn(INVOKESTATIC, "org/glowroot/agent/advicegen/GenericMessageSupplier", "create", "(Lorg/glowroot/agent/advicegen/MessageTemplate;" + "Ljava/lang/Object;Ljava/lang/String;[Ljava/lang/Object;)" + "Lorg/glowroot/agent/advicegen/GenericMessageSupplier;", false); mv.visitFieldInsn(GETSTATIC, adviceInternalName, "timerName", "Lorg/glowroot/agent/plugin/api/TimerName;"); if (config.isTransaction()) { mv.visitMethodInsn(INVOKEINTERFACE, "org/glowroot/agent/plugin/api/OptionalThreadContext", "startTransaction", "(Ljava/lang/String;Ljava/lang/String;" + "Lorg/glowroot/agent/plugin/api/MessageSupplier;" + "Lorg/glowroot/agent/plugin/api/TimerName;)" + "Lorg/glowroot/agent/plugin/api/TraceEntry;", true); } else { mv.visitMethodInsn(INVOKEINTERFACE, "org/glowroot/agent/plugin/api/ThreadContext", "startTraceEntry", "(Lorg/glowroot/agent/plugin/api/MessageSupplier;" + "Lorg/glowroot/agent/plugin/api/TimerName;)" + "Lorg/glowroot/agent/plugin/api/TraceEntry;", true); } addCodeForOptionalTransactionAttributes(mv); mv.visitInsn(ARETURN); mv.visitMaxs(0, 0); mv.visitEnd(); } private void addOnBeforeMethodTimerOnly(ClassWriter cw) { MethodVisitor mv = visitOnBeforeMethod(cw, "Lorg/glowroot/agent/plugin/api/Timer;"); mv.visitCode(); mv.visitVarInsn(ALOAD, 0); mv.visitFieldInsn(GETSTATIC, adviceInternalName, "timerName", "Lorg/glowroot/agent/plugin/api/TimerName;"); mv.visitMethodInsn(INVOKEINTERFACE, "org/glowroot/agent/plugin/api/ThreadContext", "startTimer", "(Lorg/glowroot/agent/plugin/api/TimerName;)" + "Lorg/glowroot/agent/plugin/api/Timer;", true); addCodeForOptionalTransactionAttributes(mv); mv.visitInsn(ARETURN); mv.visitMaxs(0, 0); mv.visitEnd(); } private void addOnBeforeMethodOther(ClassWriter cw) { MethodVisitor mv = visitOnBeforeMethod(cw, "V"); mv.visitCode(); addCodeForOptionalTransactionAttributes(mv); mv.visitInsn(RETURN); mv.visitMaxs(0, 0); mv.visitEnd(); } private MethodVisitor visitOnBeforeMethod(ClassWriter cw, String returnInternalName) { StringBuilder desc = new StringBuilder(); desc.append("("); if (config.isTransaction()) { desc.append("Lorg/glowroot/agent/plugin/api/OptionalThreadContext;"); } else { desc.append("Lorg/glowroot/agent/plugin/api/ThreadContext;"); } if (methodMetaInternalName != null) { desc.append("Ljava/lang/Object;Ljava/lang/String;[Ljava/lang/Object;L"); desc.append(methodMetaInternalName); desc.append(";)"); } else { desc.append(")"); } desc.append(returnInternalName); MethodVisitor mv = cw.visitMethod(ACC_PUBLIC + ACC_STATIC, "onBefore", desc.toString(), null, null); visitAnnotation(mv, "Lorg/glowroot/agent/plugin/api/weaving/OnBefore;"); if (config.isTransaction()) { mv.visitParameterAnnotation(0, "Lorg/glowroot/agent/plugin/api/weaving/BindOptionalThreadContext;", true).visitEnd(); } else { mv.visitParameterAnnotation(0, "Lorg/glowroot/agent/plugin/api/weaving/BindThreadContext;", true) .visitEnd(); } if (methodMetaInternalName != null) { mv.visitParameterAnnotation(1, "Lorg/glowroot/agent/plugin/api/weaving/BindReceiver;", true).visitEnd(); mv.visitParameterAnnotation(2, "Lorg/glowroot/agent/plugin/api/weaving/BindMethodName;", true) .visitEnd(); mv.visitParameterAnnotation(3, "Lorg/glowroot/agent/plugin/api/weaving/BindParameterArray;", true) .visitEnd(); mv.visitParameterAnnotation(4, "Lorg/glowroot/agent/plugin/api/weaving/BindMethodMeta;", true) .visitEnd(); } return mv; } private void addCodeForOptionalTransactionAttributes(MethodVisitor mv) { if (!config.transactionType().isEmpty() && !config.isTransaction()) { mv.visitVarInsn(ALOAD, 0); mv.visitLdcInsn(config.transactionType()); mv.visitIntInsn(BIPUSH, Priority.USER_CONFIG); mv.visitMethodInsn(INVOKEINTERFACE, "org/glowroot/agent/plugin/api/ThreadContext", "setTransactionType", "(Ljava/lang/String;I)V", true); } if (!config.transactionNameTemplate().isEmpty() && !config.isTransaction()) { mv.visitVarInsn(ALOAD, 0); mv.visitVarInsn(ALOAD, 4); // methodMetaInternalName is non-null when transactionNameTemplate is non-empty checkNotNull(methodMetaInternalName); mv.visitMethodInsn(INVOKEVIRTUAL, methodMetaInternalName, "getTransactionNameTemplate", "()Lorg/glowroot/agent/advicegen/MessageTemplate;", false); mv.visitVarInsn(ALOAD, 1); mv.visitVarInsn(ALOAD, 2); mv.visitVarInsn(ALOAD, 3); mv.visitMethodInsn(INVOKESTATIC, "org/glowroot/agent/advicegen/GenericMessageSupplier", "create", "(Lorg/glowroot/agent/advicegen/MessageTemplate;Ljava/lang/Object;" + "Ljava/lang/String;[Ljava/lang/Object;)" + "Lorg/glowroot/agent/advicegen/GenericMessageSupplier;", false); mv.visitMethodInsn(INVOKEVIRTUAL, "org/glowroot/agent/advicegen/GenericMessageSupplier", "getMessageText", "()Ljava/lang/String;", false); mv.visitIntInsn(BIPUSH, Priority.USER_CONFIG); mv.visitMethodInsn(INVOKEINTERFACE, "org/glowroot/agent/plugin/api/ThreadContext", "setTransactionName", "(Ljava/lang/String;I)V", true); } if (!config.transactionUserTemplate().isEmpty()) { mv.visitVarInsn(ALOAD, 0); mv.visitVarInsn(ALOAD, 4); // methodMetaInternalName is non-null when transactionUserTemplate is non-empty checkNotNull(methodMetaInternalName); mv.visitMethodInsn(INVOKEVIRTUAL, methodMetaInternalName, "getTransactionUserTemplate", "()Lorg/glowroot/agent/advicegen/MessageTemplate;", false); mv.visitVarInsn(ALOAD, 1); mv.visitVarInsn(ALOAD, 2); mv.visitVarInsn(ALOAD, 3); mv.visitMethodInsn(INVOKESTATIC, "org/glowroot/agent/advicegen/GenericMessageSupplier", "create", "(Lorg/glowroot/agent/advicegen/MessageTemplate;Ljava/lang/Object;" + "Ljava/lang/String;[Ljava/lang/Object;)" + "Lorg/glowroot/agent/advicegen/GenericMessageSupplier;", false); mv.visitMethodInsn(INVOKEVIRTUAL, "org/glowroot/agent/advicegen/GenericMessageSupplier", "getMessageText", "()Ljava/lang/String;", false); mv.visitIntInsn(BIPUSH, Priority.USER_CONFIG); mv.visitMethodInsn(INVOKEINTERFACE, "org/glowroot/agent/plugin/api/ThreadContext", "setTransactionUser", "(Ljava/lang/String;I)V", true); } int i = 0; for (String attrName : config.transactionAttributeTemplates().keySet()) { mv.visitVarInsn(ALOAD, 0); mv.visitLdcInsn(attrName); mv.visitVarInsn(ALOAD, 4); // methodMetaInternalName is non-null when transactionAttributeTemplates is // non-empty checkNotNull(methodMetaInternalName); mv.visitMethodInsn(INVOKEVIRTUAL, methodMetaInternalName, "getTransactionAttributeTemplate" + i++, "()Lorg/glowroot/agent/advicegen/MessageTemplate;", false); mv.visitVarInsn(ALOAD, 1); mv.visitVarInsn(ALOAD, 2); mv.visitVarInsn(ALOAD, 3); mv.visitMethodInsn(INVOKESTATIC, "org/glowroot/agent/advicegen/GenericMessageSupplier", "create", "(Lorg/glowroot/agent/advicegen/MessageTemplate;" + "Ljava/lang/Object;Ljava/lang/String;[Ljava/lang/Object;)" + "Lorg/glowroot/agent/advicegen/GenericMessageSupplier;", false); mv.visitMethodInsn(INVOKEVIRTUAL, "org/glowroot/agent/advicegen/GenericMessageSupplier", "getMessageText", "()Ljava/lang/String;", false); mv.visitMethodInsn(INVOKEINTERFACE, "org/glowroot/agent/plugin/api/ThreadContext", "addTransactionAttribute", "(Ljava/lang/String;Ljava/lang/String;)V", true); } Integer slowThresholdMillis = config.transactionSlowThresholdMillis(); if (slowThresholdMillis != null) { mv.visitVarInsn(ALOAD, 0); mv.visitLdcInsn(slowThresholdMillis.longValue()); mv.visitFieldInsn(GETSTATIC, "java/util/concurrent/TimeUnit", "MILLISECONDS", "Ljava/util/concurrent/TimeUnit;"); mv.visitIntInsn(BIPUSH, Priority.USER_CONFIG); mv.visitMethodInsn(INVOKEINTERFACE, "org/glowroot/agent/plugin/api/ThreadContext", "setTransactionSlowThreshold", "(JLjava/util/concurrent/TimeUnit;I)V", true); } } private void addOnAfterMethodTimerOnly(ClassWriter cw) { MethodVisitor mv = cw.visitMethod(ACC_PUBLIC + ACC_STATIC, "onAfter", "(Lorg/glowroot/agent/plugin/api/Timer;)V", null, null); visitAnnotation(mv, "Lorg/glowroot/agent/plugin/api/weaving/OnAfter;"); mv.visitParameterAnnotation(0, "Lorg/glowroot/agent/plugin/api/weaving/BindTraveler;", true).visitEnd(); mv.visitCode(); mv.visitVarInsn(ALOAD, 0); mv.visitMethodInsn(INVOKEINTERFACE, "org/glowroot/agent/plugin/api/Timer", "stop", "()V", true); mv.visitInsn(RETURN); mv.visitMaxs(0, 0); mv.visitEnd(); } private void addOnReturnMethod(ClassWriter cw) { boolean entryOrTimer = !config.traceEntryEnabledProperty().isEmpty(); String travelerType = entryOrTimer ? "Ljava/lang/Object;" : "Lorg/glowroot/agent/plugin/api/TraceEntry;"; MethodVisitor mv = cw.visitMethod(ACC_PUBLIC + ACC_STATIC, "onReturn", "(Lorg/glowroot/agent/plugin/api/weaving/OptionalReturn;" + travelerType + ")V", null, null); mv.visitParameterAnnotation(0, "Lorg/glowroot/agent/plugin/api/weaving/BindOptionalReturn;", true) .visitEnd(); mv.visitParameterAnnotation(1, "Lorg/glowroot/agent/plugin/api/weaving/BindTraveler;", true).visitEnd(); int travelerParamIndex = 1; visitAnnotation(mv, "Lorg/glowroot/agent/plugin/api/weaving/OnReturn;"); mv.visitCode(); if (!config.traceEntryEnabledProperty().isEmpty()) { mv.visitVarInsn(ALOAD, travelerParamIndex); // TraceEntryImpl implements both TraceEntry and Timer so cannot check instanceof Timer // to differentiate here (but can check isntanceof TraceEntry) mv.visitTypeInsn(INSTANCEOF, "org/glowroot/agent/plugin/api/TraceEntry"); Label label = new Label(); mv.visitJumpInsn(IFNE, label); mv.visitVarInsn(ALOAD, travelerParamIndex); mv.visitTypeInsn(CHECKCAST, "org/glowroot/agent/plugin/api/Timer"); mv.visitMethodInsn(INVOKEINTERFACE, "org/glowroot/agent/plugin/api/Timer", "stop", "()V", true); mv.visitInsn(RETURN); mv.visitLabel(label); } mv.visitVarInsn(ALOAD, 1); mv.visitVarInsn(ALOAD, 0); mv.visitMethodInsn(INVOKEINTERFACE, "org/glowroot/agent/plugin/api/weaving/OptionalReturn", "isVoid", "()Z", true); Label notVoidLabel = new Label(); Label endIfLabel = new Label(); mv.visitJumpInsn(IFEQ, notVoidLabel); mv.visitLdcInsn("void"); mv.visitJumpInsn(GOTO, endIfLabel); mv.visitLabel(notVoidLabel); mv.visitVarInsn(ALOAD, 0); mv.visitMethodInsn(INVOKEINTERFACE, "org/glowroot/agent/plugin/api/weaving/OptionalReturn", "getValue", "()Ljava/lang/Object;", true); mv.visitLabel(endIfLabel); mv.visitMethodInsn(INVOKESTATIC, "org/glowroot/agent/advicegen/GenericMessageSupplier", "updateWithReturnValue", "(Lorg/glowroot/agent/plugin/api/TraceEntry;Ljava/lang/Object;)V", false); mv.visitVarInsn(ALOAD, travelerParamIndex); Integer stackTraceThresholdMillis = config.traceEntryStackThresholdMillis(); if (stackTraceThresholdMillis == null) { mv.visitMethodInsn(INVOKEINTERFACE, "org/glowroot/agent/plugin/api/TraceEntry", "end", "()V", true); } else { mv.visitLdcInsn(stackTraceThresholdMillis.longValue()); mv.visitFieldInsn(GETSTATIC, "java/util/concurrent/TimeUnit", "MILLISECONDS", "Ljava/util/concurrent/TimeUnit;"); mv.visitMethodInsn(INVOKEINTERFACE, "org/glowroot/agent/plugin/api/TraceEntry", "endWithStackTrace", "(JLjava/util/concurrent/TimeUnit;)V", true); } mv.visitInsn(RETURN); mv.visitMaxs(0, 0); mv.visitEnd(); } private void addOnThrowMethod(ClassWriter cw) { MethodVisitor mv = cw.visitMethod(ACC_PUBLIC + ACC_STATIC, "onThrow", "(Ljava/lang/Throwable;Lorg/glowroot/agent/plugin/api/TraceEntry;)V", null, null); visitAnnotation(mv, "Lorg/glowroot/agent/plugin/api/weaving/OnThrow;"); mv.visitParameterAnnotation(0, "Lorg/glowroot/agent/plugin/api/weaving/BindThrowable;", true).visitEnd(); mv.visitParameterAnnotation(1, "Lorg/glowroot/agent/plugin/api/weaving/BindTraveler;", true).visitEnd(); mv.visitCode(); if (!config.traceEntryEnabledProperty().isEmpty()) { mv.visitVarInsn(ALOAD, 1); Label l0 = new Label(); mv.visitJumpInsn(IFNONNULL, l0); mv.visitInsn(RETURN); mv.visitLabel(l0); } mv.visitVarInsn(ALOAD, 1); mv.visitVarInsn(ALOAD, 0); mv.visitMethodInsn(INVOKEINTERFACE, "org/glowroot/agent/plugin/api/TraceEntry", "endWithError", "(Ljava/lang/Throwable;)V", true); mv.visitInsn(RETURN); mv.visitMaxs(0, 0); mv.visitEnd(); } @RequiresNonNull("methodMetaInternalName") private LazyDefinedClass generateMethodMetaClass(InstrumentationConfig config) throws Exception { ClassWriter cw = new ClassWriter(ClassWriter.COMPUTE_MAXS + ClassWriter.COMPUTE_FRAMES); cw.visit(V1_5, ACC_PUBLIC + ACC_SUPER, methodMetaInternalName, null, "java/lang/Object", null); cw.visitField(ACC_PRIVATE + ACC_FINAL, "messageTemplate", "Lorg/glowroot/agent/advicegen/MessageTemplate;", null, null).visitEnd(); if (!config.transactionNameTemplate().isEmpty()) { cw.visitField(ACC_PRIVATE + ACC_FINAL, "transactionNameTemplate", "Lorg/glowroot/agent/advicegen/MessageTemplate;", null, null).visitEnd(); } if (!config.transactionUserTemplate().isEmpty()) { cw.visitField(ACC_PRIVATE + ACC_FINAL, "transactionUserTemplate", "Lorg/glowroot/agent/advicegen/MessageTemplate;", null, null).visitEnd(); } for (int i = 0; i < config.transactionAttributeTemplates().size(); i++) { cw.visitField(ACC_PRIVATE + ACC_FINAL, "transactionAttributeTemplate" + i, "Lorg/glowroot/agent/advicegen/MessageTemplate;", null, null).visitEnd(); } generateMethodMetaConstructor(cw); generateMethodMetaGetter(cw, "messageTemplate", "getMessageTemplate"); if (!config.transactionNameTemplate().isEmpty()) { generateMethodMetaGetter(cw, "transactionNameTemplate", "getTransactionNameTemplate"); } if (!config.transactionUserTemplate().isEmpty()) { generateMethodMetaGetter(cw, "transactionUserTemplate", "getTransactionUserTemplate"); } for (int i = 0; i < config.transactionAttributeTemplates().size(); i++) { generateMethodMetaGetter(cw, "transactionAttributeTemplate" + i, "getTransactionAttributeTemplate" + i); } cw.visitEnd(); return ImmutableLazyDefinedClass.builder().type(Type.getObjectType(methodMetaInternalName)) .bytes(cw.toByteArray()).build(); } @RequiresNonNull("methodMetaInternalName") private void generateMethodMetaConstructor(ClassWriter cw) { MethodVisitor mv = cw.visitMethod(ACC_PUBLIC, "<init>", "(Ljava/lang/reflect/Method;)V", null, null); mv.visitCode(); mv.visitVarInsn(ALOAD, 0); mv.visitMethodInsn(INVOKESPECIAL, "java/lang/Object", "<init>", "()V", false); mv.visitVarInsn(ALOAD, 0); if (config.isTraceEntryOrGreater()) { String messageTemplate = config.traceEntryMessageTemplate(); if (messageTemplate.isEmpty() && config.isTransaction()) { messageTemplate = config.transactionNameTemplate(); } if (messageTemplate.isEmpty()) { mv.visitLdcInsn("<no message template provided>"); } else { mv.visitLdcInsn(messageTemplate); } mv.visitVarInsn(ALOAD, 1); mv.visitMethodInsn(INVOKESTATIC, "org/glowroot/agent/advicegen/MessageTemplate", "create", "(Ljava/lang/String;Ljava/lang/reflect/Method;)" + "Lorg/glowroot/agent/advicegen/MessageTemplate;", false); } else { mv.visitInsn(ACONST_NULL); } mv.visitFieldInsn(PUTFIELD, methodMetaInternalName, "messageTemplate", "Lorg/glowroot/agent/advicegen/MessageTemplate;"); if (!config.transactionNameTemplate().isEmpty()) { mv.visitVarInsn(ALOAD, 0); mv.visitLdcInsn(config.transactionNameTemplate()); mv.visitVarInsn(ALOAD, 1); mv.visitMethodInsn(INVOKESTATIC, "org/glowroot/agent/advicegen/MessageTemplate", "create", "(Ljava/lang/String;Ljava/lang/reflect/Method;)" + "Lorg/glowroot/agent/advicegen/MessageTemplate;", false); mv.visitFieldInsn(PUTFIELD, methodMetaInternalName, "transactionNameTemplate", "Lorg/glowroot/agent/advicegen/MessageTemplate;"); } if (!config.transactionUserTemplate().isEmpty()) { mv.visitVarInsn(ALOAD, 0); mv.visitLdcInsn(config.transactionUserTemplate()); mv.visitVarInsn(ALOAD, 1); mv.visitMethodInsn(INVOKESTATIC, "org/glowroot/agent/advicegen/MessageTemplate", "create", "(Ljava/lang/String;Ljava/lang/reflect/Method;)" + "Lorg/glowroot/agent/advicegen/MessageTemplate;", false); mv.visitFieldInsn(PUTFIELD, methodMetaInternalName, "transactionUserTemplate", "Lorg/glowroot/agent/advicegen/MessageTemplate;"); } int i = 0; for (String attrTemplate : config.transactionAttributeTemplates().values()) { mv.visitVarInsn(ALOAD, 0); mv.visitLdcInsn(attrTemplate); mv.visitVarInsn(ALOAD, 1); mv.visitMethodInsn(INVOKESTATIC, "org/glowroot/agent/advicegen/MessageTemplate", "create", "(Ljava/lang/String;Ljava/lang/reflect/Method;)" + "Lorg/glowroot/agent/advicegen/MessageTemplate;", false); mv.visitFieldInsn(PUTFIELD, methodMetaInternalName, "transactionAttributeTemplate" + i++, "Lorg/glowroot/agent/advicegen/MessageTemplate;"); } mv.visitInsn(RETURN); mv.visitMaxs(0, 0); mv.visitEnd(); } @RequiresNonNull("methodMetaInternalName") private void generateMethodMetaGetter(ClassWriter cw, String fieldName, String methodName) { MethodVisitor mv = cw.visitMethod(ACC_PUBLIC, methodName, "()Lorg/glowroot/agent/advicegen/MessageTemplate;", null, null); mv.visitCode(); mv.visitVarInsn(ALOAD, 0); mv.visitFieldInsn(GETFIELD, methodMetaInternalName, fieldName, "Lorg/glowroot/agent/advicegen/MessageTemplate;"); mv.visitInsn(ARETURN); mv.visitMaxs(0, 0); mv.visitEnd(); } private static void visitAnnotation(MethodVisitor mv, String desc) { AnnotationVisitor av = mv.visitAnnotation(desc, true); checkNotNull(av); av.visitEnd(); } }