org.glowroot.agent.weaving.ClassAnalyzer Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of glowroot-agent-it-harness Show documentation
Show all versions of glowroot-agent-it-harness Show documentation
Glowroot Agent Integration Test Harness
/*
* Copyright 2015-2018 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.weaving;
import java.lang.reflect.Modifier;
import java.security.CodeSource;
import java.util.Arrays;
import java.util.Collection;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.regex.Pattern;
import org.glowroot.agent.shaded.org.glowroot.agent.it.harness.shaded.com.google.common.base.Joiner;
import org.glowroot.agent.shaded.org.glowroot.agent.it.harness.shaded.com.google.common.collect.ImmutableList;
import org.glowroot.agent.shaded.org.glowroot.agent.it.harness.shaded.com.google.common.collect.ImmutableMap;
import org.glowroot.agent.shaded.org.glowroot.agent.it.harness.shaded.com.google.common.collect.ImmutableSet;
import org.glowroot.agent.shaded.org.glowroot.agent.it.harness.shaded.com.google.common.collect.Lists;
import org.glowroot.agent.shaded.org.glowroot.agent.it.harness.shaded.com.google.common.collect.Maps;
import org.glowroot.agent.shaded.org.glowroot.agent.it.harness.shaded.com.google.common.collect.Sets;
import org.glowroot.agent.shaded.org.checkerframework.checker.nullness.qual.MonotonicNonNull;
import org.glowroot.agent.shaded.org.checkerframework.checker.nullness.qual.Nullable;
import org.glowroot.agent.shaded.org.checkerframework.checker.nullness.qual.RequiresNonNull;
import org.immutables.value.Value;
import org.glowroot.agent.shaded.org.objectweb.asm.ClassReader;
import org.glowroot.agent.shaded.org.objectweb.asm.ClassVisitor;
import org.glowroot.agent.shaded.org.objectweb.asm.MethodVisitor;
import org.glowroot.agent.shaded.org.objectweb.asm.Type;
import org.glowroot.agent.shaded.org.glowroot.agent.shaded.org.slf4j.Logger;
import org.glowroot.agent.shaded.org.glowroot.agent.shaded.org.slf4j.LoggerFactory;
import org.glowroot.agent.config.ImmutableInstrumentationConfig;
import org.glowroot.agent.config.InstrumentationConfig;
import org.glowroot.agent.weaving.AnalyzedWorld.ParseContext;
import org.glowroot.agent.weaving.ClassLoaders.LazyDefinedClass;
import org.glowroot.agent.weaving.ThinClassVisitor.ThinClass;
import org.glowroot.agent.weaving.ThinClassVisitor.ThinMethod;
import org.glowroot.agent.shaded.org.glowroot.wire.api.model.AgentConfigOuterClass.AgentConfig.InstrumentationConfig.CaptureKind;
import static org.glowroot.agent.shaded.org.glowroot.agent.it.harness.shaded.com.google.common.base.Preconditions.checkNotNull;
import static org.glowroot.agent.shaded.org.objectweb.asm.Opcodes.ACC_BRIDGE;
import static org.glowroot.agent.shaded.org.objectweb.asm.Opcodes.ASM6;
import static org.glowroot.agent.shaded.org.objectweb.asm.Opcodes.INVOKESPECIAL;
class ClassAnalyzer {
private static final Logger logger = LoggerFactory.getLogger(ClassAnalyzer.class);
private final ThinClass thinClass;
private final String className;
private final boolean intf;
private final ImmutableAnalyzedClass.Builder analyzedClassBuilder;
private final ImmutableList adviceMatchers;
private final ImmutableList superAnalyzedClasses;
private final ImmutableList matchedShimTypes;
private final ImmutableList matchedMixinTypes;
private final boolean hasMainMethod;
private final ImmutableSet superClassNames;
private final boolean shortCircuitBeforeAnalyzeMethods;
private final byte[] classBytes;
private @MonotonicNonNull Map> methodAdvisors;
private @MonotonicNonNull List methodsThatOnlyNowFulfillAdvice;
// this is used to propagate bridge method advice to its target
private @MonotonicNonNull Map> bridgeTargetAdvisors;
ClassAnalyzer(ThinClass thinClass, List advisors, List shimTypes,
List mixinTypes, @Nullable ClassLoader loader, AnalyzedWorld analyzedWorld,
@Nullable CodeSource codeSource, byte[] classBytes,
boolean noLongerNeedToWeaveMainMethods) {
this.thinClass = thinClass;
ImmutableList interfaceNames = ClassNames.fromInternalNames(thinClass.interfaces());
className = ClassNames.fromInternalName(thinClass.name());
intf = Modifier.isInterface(thinClass.access());
String superClassName = ClassNames.fromInternalName(thinClass.superName());
analyzedClassBuilder = ImmutableAnalyzedClass.builder()
.modifiers(thinClass.access())
.name(className)
.superName(superClassName)
.addAllInterfaceNames(interfaceNames);
boolean ejbRemote = false;
boolean ejbStateless = false;
for (String annotation : thinClass.annotations()) {
if (annotation.equals("Ljavax/ejb/Remote;")) {
ejbRemote = true;
} else if (annotation.equals("Ljavax/ejb/Stateless;")) {
ejbStateless = true;
}
}
ParseContext parseContext = ImmutableParseContext.of(className, codeSource);
List interfaceAnalyzedHierarchy = Lists.newArrayList();
// it's ok if there are duplicates in the superAnalyzedClasses list (e.g. an interface
// that appears twice in a type hierarchy), it's rare, dups don't cause an issue for
// callers, and so it doesn't seem worth the (minor) performance hit to de-dup every
// time
List superAnalyzedClasses = Lists.newArrayList();
for (String interfaceName : interfaceNames) {
interfaceAnalyzedHierarchy.addAll(
analyzedWorld.getAnalyzedHierarchy(interfaceName, loader, parseContext));
}
superAnalyzedClasses.addAll(interfaceAnalyzedHierarchy);
if (intf) {
matchedShimTypes = getMatchedShimTypes(shimTypes, className,
ImmutableList.of(), ImmutableList.of());
matchedMixinTypes = getMatchedMixinTypes(mixinTypes, className,
ImmutableList.of(), ImmutableList.of());
hasMainMethod = false;
} else {
List superAnalyzedHierarchy =
analyzedWorld.getAnalyzedHierarchy(superClassName, loader, parseContext);
superAnalyzedClasses.addAll(superAnalyzedHierarchy);
matchedShimTypes = getMatchedShimTypes(shimTypes, className, superAnalyzedHierarchy,
interfaceAnalyzedHierarchy);
matchedMixinTypes = getMatchedMixinTypes(mixinTypes, className, superAnalyzedHierarchy,
interfaceAnalyzedHierarchy);
if (noLongerNeedToWeaveMainMethods) {
hasMainMethod = false;
} else {
hasMainMethod = hasMainMethod(thinClass.nonBridgeMethods())
|| className.equals("org.apache.commons.daemon.support.DaemonLoader");
}
}
analyzedClassBuilder.addAllShimTypes(matchedShimTypes);
analyzedClassBuilder.addAllMixinTypes(matchedMixinTypes);
if ((ejbRemote || ejbStateless) && !intf) {
Set ejbRemoteInterfaces =
getEjbRemoteInterfaces(thinClass, superAnalyzedClasses);
if (ejbRemoteInterfaces.isEmpty()) {
this.superAnalyzedClasses = ImmutableList.copyOf(superAnalyzedClasses);
analyzedClassBuilder.ejbRemote(ejbRemote);
} else if (loader == null) {
logger.warn("instrumenting @javax.ejb.Remote not currently supported in bootstrap"
+ " class loader: {}", className);
this.superAnalyzedClasses = ImmutableList.copyOf(superAnalyzedClasses);
analyzedClassBuilder.ejbRemote(false);
} else {
List ejbHackedSuperAnalyzedClasses =
hack(thinClass, loader, superAnalyzedClasses, ejbRemoteInterfaces);
this.superAnalyzedClasses = ImmutableList.copyOf(ejbHackedSuperAnalyzedClasses);
analyzedClassBuilder.ejbRemote(true);
}
} else {
this.superAnalyzedClasses = ImmutableList.copyOf(superAnalyzedClasses);
analyzedClassBuilder.ejbRemote(ejbRemote);
}
Set superClassNames = Sets.newHashSet();
superClassNames.add(className);
for (AnalyzedClass analyzedClass : superAnalyzedClasses) {
superClassNames.add(analyzedClass.name());
}
this.superClassNames = ImmutableSet.copyOf(superClassNames);
adviceMatchers = AdviceMatcher.getAdviceMatchers(className, thinClass.annotations(),
superClassNames, advisors);
if (intf) {
shortCircuitBeforeAnalyzeMethods = false;
} else {
shortCircuitBeforeAnalyzeMethods =
!hasSuperAdvice(this.superAnalyzedClasses) && matchedShimTypes.isEmpty()
&& matchedMixinTypes.isEmpty() && adviceMatchers.isEmpty();
}
this.classBytes = classBytes;
}
void analyzeMethods() {
methodAdvisors = Maps.newHashMap();
bridgeTargetAdvisors = Maps.newHashMap();
for (ThinMethod bridgeMethod : thinClass.bridgeMethods()) {
List advisors = analyzeMethod(bridgeMethod);
if (!advisors.isEmpty()) {
// don't add advisors to bridge method
// instead propagate bridge method advice to its target
ThinMethod targetMethod = getTargetMethod(bridgeMethod);
if (targetMethod != null) {
bridgeTargetAdvisors.put(targetMethod, advisors);
}
}
}
for (ThinMethod nonBridgeMethod : thinClass.nonBridgeMethods()) {
List advisors = analyzeMethod(nonBridgeMethod);
if (!advisors.isEmpty()) {
methodAdvisors.put(nonBridgeMethod.name() + nonBridgeMethod.desc(), advisors);
}
}
AnalyzedClass mostlyAnalyzedClass = analyzedClassBuilder.build();
methodsThatOnlyNowFulfillAdvice = getMethodsThatOnlyNowFulfillAdvice(mostlyAnalyzedClass);
analyzedClassBuilder.addAllAnalyzedMethods(methodsThatOnlyNowFulfillAdvice);
}
boolean isWeavingRequired() {
checkNotNull(methodAdvisors);
checkNotNull(methodsThatOnlyNowFulfillAdvice);
if (Modifier.isInterface(thinClass.access())) {
// FIXME only need to return true if any default methods have advice
return !methodAdvisors.isEmpty() || !matchedMixinTypes.isEmpty();
}
return !methodAdvisors.isEmpty() || !methodsThatOnlyNowFulfillAdvice.isEmpty()
|| !matchedShimTypes.isEmpty() || !matchedMixinTypes.isEmpty() || hasMainMethod;
}
ImmutableList getMatchedShimTypes() {
return matchedShimTypes;
}
ImmutableList getMatchedMixinTypes() {
return matchedMixinTypes;
}
Map> getMethodAdvisors() {
return checkNotNull(methodAdvisors);
}
AnalyzedClass getAnalyzedClass() {
return analyzedClassBuilder.build();
}
List getMethodsThatOnlyNowFulfillAdvice() {
return checkNotNull(methodsThatOnlyNowFulfillAdvice);
}
@RequiresNonNull("bridgeTargetAdvisors")
private List analyzeMethod(ThinMethod thinMethod) {
List parameterTypes = Arrays.asList(Type.getArgumentTypes(thinMethod.desc()));
if (Modifier.isFinal(thinMethod.access()) && Modifier.isPublic(thinMethod.access())) {
ImmutablePublicFinalMethod.Builder builder = ImmutablePublicFinalMethod.builder()
.name(thinMethod.name());
for (Type parameterType : parameterTypes) {
builder.addParameterTypes(parameterType.getClassName());
}
analyzedClassBuilder.addPublicFinalMethods(builder.build());
}
if (shortCircuitBeforeAnalyzeMethods) {
return ImmutableList.of();
}
Type returnType = Type.getReturnType(thinMethod.desc());
List methodAnnotations = thinMethod.annotations();
List matchingAdvisors = getMatchingAdvisors(thinMethod, methodAnnotations,
parameterTypes, returnType);
boolean intfMethod = intf && !Modifier.isStatic(thinMethod.access());
if (matchingAdvisors.isEmpty() && !intfMethod) {
return ImmutableList.of();
}
ImmutableAnalyzedMethod.Builder builder = ImmutableAnalyzedMethod.builder();
builder.name(thinMethod.name());
for (Type parameterType : parameterTypes) {
builder.addParameterTypes(parameterType.getClassName());
}
builder.returnType(returnType.getClassName())
.modifiers(thinMethod.access())
.signature(thinMethod.signature());
for (String exception : thinMethod.exceptions()) {
builder.addExceptions(ClassNames.fromInternalName(exception));
}
List subTypeRestrictedAdvisors = Lists.newArrayList();
for (Iterator i = matchingAdvisors.iterator(); i.hasNext();) {
Advice advice = i.next();
if (!isSubTypeRestrictionMatch(advice, superClassNames)) {
subTypeRestrictedAdvisors.add(advice);
i.remove();
}
}
builder.addAllAdvisors(matchingAdvisors);
builder.addAllSubTypeRestrictedAdvisors(subTypeRestrictedAdvisors);
analyzedClassBuilder.addAnalyzedMethods(builder.build());
return matchingAdvisors;
}
// returns mutable list if non-empty so items can be removed
@RequiresNonNull("bridgeTargetAdvisors")
private List getMatchingAdvisors(ThinMethod thinMethod, List methodAnnotations,
List parameterTypes, Type returnType) {
Set matchingAdvisors = Sets.newHashSet();
for (AdviceMatcher adviceMatcher : adviceMatchers) {
if (adviceMatcher.isMethodLevelMatch(thinMethod.name(), methodAnnotations,
parameterTypes, returnType, thinMethod.access())) {
matchingAdvisors.add(adviceMatcher.advice());
}
}
// look at super types
for (AnalyzedClass superAnalyzedClass : superAnalyzedClasses) {
for (AnalyzedMethod analyzedMethod : superAnalyzedClass.analyzedMethods()) {
if (analyzedMethod.isOverriddenBy(thinMethod.name(), parameterTypes)) {
matchingAdvisors.addAll(analyzedMethod.advisors());
for (Advice subTypeRestrictedAdvice : analyzedMethod
.subTypeRestrictedAdvisors()) {
if (isSubTypeRestrictionMatch(subTypeRestrictedAdvice, superClassNames)) {
matchingAdvisors.add(subTypeRestrictedAdvice);
}
}
}
}
}
List extraAdvisors = bridgeTargetAdvisors.get(thinMethod);
if (extraAdvisors != null) {
matchingAdvisors.addAll(extraAdvisors);
}
// sort since the order affects advice nesting
return sortAdvisors(matchingAdvisors);
}
private @Nullable ThinMethod getTargetMethod(ThinMethod bridgeMethod) {
List possibleTargetMethods = getPossibleTargetMethods(bridgeMethod);
if (possibleTargetMethods.isEmpty()) {
// probably a visibility bridge for public method in package-private super class
return null;
}
BridgeMethodClassVisitor bmcv = new BridgeMethodClassVisitor();
new ClassReader(classBytes).accept(bmcv, ClassReader.SKIP_FRAMES);
Map bridgeMethodMap = bmcv.getBridgeTargetMethods();
String targetMethod = bridgeMethodMap.get(bridgeMethod.name() + bridgeMethod.desc());
if (targetMethod == null) {
// probably a visibility bridge for public method in package-private super class
return null;
}
for (ThinMethod possibleTargetMethod : possibleTargetMethods) {
if (targetMethod.equals(possibleTargetMethod.name() + possibleTargetMethod.desc())) {
return possibleTargetMethod;
}
}
logger.warn("could not find match for bridge method in {}: {}", className, bridgeMethod);
return null;
}
private List getPossibleTargetMethods(ThinMethod bridgeMethod) {
List possibleTargetMethods = Lists.newArrayList();
for (ThinMethod possibleTargetMethod : thinClass.nonBridgeMethods()) {
if (!possibleTargetMethod.name().equals(bridgeMethod.name())) {
continue;
}
Type[] bridgeMethodParamTypes = Type.getArgumentTypes(bridgeMethod.desc());
Type[] possibleTargetMethodParamTypes =
Type.getArgumentTypes(possibleTargetMethod.desc());
if (possibleTargetMethodParamTypes.length != bridgeMethodParamTypes.length) {
continue;
}
possibleTargetMethods.add(possibleTargetMethod);
}
return possibleTargetMethods;
}
private List getMethodsThatOnlyNowFulfillAdvice(AnalyzedClass analyzedClass) {
if (analyzedClass.isAbstract()) {
return ImmutableList.of();
}
Map> matchingAdvisorSets =
getInheritedInterfaceMethodsWithAdvice();
if (matchingAdvisorSets.isEmpty()) {
return ImmutableList.of();
}
for (AnalyzedMethod analyzedMethod : analyzedClass.analyzedMethods()) {
matchingAdvisorSets.remove(AnalyzedMethodKey.wrap(analyzedMethod));
}
removeAdviceAlreadyWovenIntoSuperClass(matchingAdvisorSets);
removeMethodsThatWouldOverridePublicFinalMethodsFromSuperClass(matchingAdvisorSets);
List methodsThatOnlyNowFulfillAdvice = Lists.newArrayList();
for (Map.Entry> entry : matchingAdvisorSets.entrySet()) {
AnalyzedMethod inheritedMethod = checkNotNull(entry.getKey().analyzedMethod());
Set advisors = entry.getValue();
if (!advisors.isEmpty()) {
methodsThatOnlyNowFulfillAdvice.add(ImmutableAnalyzedMethod.builder()
.copyFrom(inheritedMethod)
.advisors(advisors)
.build());
}
}
return methodsThatOnlyNowFulfillAdvice;
}
private Map> getInheritedInterfaceMethodsWithAdvice() {
Map> matchingAdvisorSets = Maps.newHashMap();
for (AnalyzedClass superAnalyzedClass : superAnalyzedClasses) {
for (AnalyzedMethod superAnalyzedMethod : superAnalyzedClass.analyzedMethods()) {
AnalyzedMethodKey key = AnalyzedMethodKey.wrap(superAnalyzedMethod);
Set matchingAdvisorSet = matchingAdvisorSets.get(key);
if (matchingAdvisorSet == null) {
matchingAdvisorSet = Sets.newHashSet();
matchingAdvisorSets.put(key, matchingAdvisorSet);
}
matchingAdvisorSet.addAll(superAnalyzedMethod.advisors());
for (Advice advice : superAnalyzedMethod.subTypeRestrictedAdvisors()) {
if (isSubTypeRestrictionMatch(advice, superClassNames)) {
matchingAdvisorSet.add(advice);
}
}
}
}
return matchingAdvisorSets;
}
private void removeAdviceAlreadyWovenIntoSuperClass(
Map> matchingAdvisorSets) {
for (AnalyzedClass superAnalyzedClass : superAnalyzedClasses) {
for (AnalyzedMethod superAnalyzedMethod : superAnalyzedClass.analyzedMethods()) {
if (Modifier.isAbstract(superAnalyzedMethod.modifiers())) {
continue;
}
Set matchingAdvisorSet =
matchingAdvisorSets.get(AnalyzedMethodKey.wrap(superAnalyzedMethod));
if (matchingAdvisorSet == null) {
continue;
}
matchingAdvisorSet.removeAll(superAnalyzedMethod.advisors());
}
}
}
private void removeMethodsThatWouldOverridePublicFinalMethodsFromSuperClass(
Map> matchingAdvisorSets) {
for (AnalyzedClass superAnalyzedClass : superAnalyzedClasses) {
for (PublicFinalMethod publicFinalMethod : superAnalyzedClass.publicFinalMethods()) {
ImmutableAnalyzedMethodKey key = ImmutableAnalyzedMethodKey.builder()
.name(publicFinalMethod.name())
.addAllParameterTypes(publicFinalMethod.parameterTypes())
.build();
Set value = matchingAdvisorSets.remove(key);
// need to check if empty due to removeAdviceAlreadyWovenIntoSuperClass() above
if (value != null && !value.isEmpty()) {
logOverridePublicFinalMethodInfo(superAnalyzedClass, publicFinalMethod);
}
}
}
}
private void logOverridePublicFinalMethodInfo(AnalyzedClass superAnalyzedClass,
PublicFinalMethod publicFinalMethod) {
String format = "Not able to override and instrument final method {}.{}({}) which is"
+ " being used by a subclass {} to implement one or more instrumented interfaces";
String arg1 = superAnalyzedClass.name();
String arg2 = publicFinalMethod.name();
String arg3 = Joiner.on(", ").join(publicFinalMethod.parameterTypes());
String arg4 = className;
if (superAnalyzedClass.name().equals("org.jooq.tools.jdbc.JDBC41ResultSet")) {
// this is a known case, and is only worth logging at debug level since all that these
// methods in JDBC41ResultSet do is throw an UnsupportedOperationException
logger.debug(format, arg1, arg2, arg3, arg4);
} else {
logger.info(format, arg1, arg2, arg3, arg4);
}
}
private static ImmutableList getMatchedShimTypes(List shimTypes,
String className, List superAnalyzedHierarchy,
List interfaceAnalyzedHierarchy) {
Set matchedShimTypes = Sets.newHashSet();
for (ShimType shimType : shimTypes) {
// currently only exact matching is supported
if (shimType.targets().contains(className)) {
matchedShimTypes.add(shimType);
}
}
for (AnalyzedClass interfaceAnalyzedClass : interfaceAnalyzedHierarchy) {
matchedShimTypes.addAll(interfaceAnalyzedClass.shimTypes());
}
// remove shims that were already implemented in a super class
for (AnalyzedClass superAnalyzedClass : superAnalyzedHierarchy) {
if (!superAnalyzedClass.isInterface()) {
matchedShimTypes.removeAll(superAnalyzedClass.shimTypes());
}
}
return ImmutableList.copyOf(matchedShimTypes);
}
private static ImmutableList getMatchedMixinTypes(List mixinTypes,
String className, List superAnalyzedHierarchy,
List interfaceAnalyzedHierarchy) {
Set matchedMixinTypes = Sets.newHashSet();
for (MixinType mixinType : mixinTypes) {
// currently only exact matching is supported
if (mixinType.targets().contains(className)) {
matchedMixinTypes.add(mixinType);
}
}
for (AnalyzedClass interfaceAnalyzedClass : interfaceAnalyzedHierarchy) {
matchedMixinTypes.addAll(interfaceAnalyzedClass.mixinTypes());
}
for (AnalyzedClass superAnalyzedClass : superAnalyzedHierarchy) {
if (superAnalyzedClass.name().equals("java.lang.Thread")) {
matchedMixinTypes.addAll(superAnalyzedClass.mixinTypes());
}
}
// remove mixins that were already implemented in a super class
for (AnalyzedClass superAnalyzedClass : superAnalyzedHierarchy) {
if (!superAnalyzedClass.isInterface()
&& !superAnalyzedClass.name().equals("java.lang.Thread")) {
matchedMixinTypes.removeAll(superAnalyzedClass.mixinTypes());
}
}
return ImmutableList.copyOf(matchedMixinTypes);
}
private static boolean hasMainMethod(List methods) {
for (ThinMethod method : methods) {
if (method.name().equals("main") && method.desc().equals("([Ljava/lang/String;)V")) {
return true;
}
}
return false;
}
private static Set getEjbRemoteInterfaces(ThinClass thinClass,
List superAnalyzedClasses) {
Set ejbRemoteInterfaces = Sets.newHashSet();
for (Type ejbRemoteInterface : thinClass.ejbRemoteInterfaces()) {
ejbRemoteInterfaces.add(ejbRemoteInterface.getClassName());
}
for (AnalyzedClass superAnalyzedClass : superAnalyzedClasses) {
if (superAnalyzedClass.isInterface() && superAnalyzedClass.ejbRemote()) {
ejbRemoteInterfaces.add(superAnalyzedClass.name());
}
}
return ejbRemoteInterfaces;
}
private static List hack(ThinClass thinClass, ClassLoader loader,
List superAnalyzedClasses, Set ejbRemoteInterfaces) {
Map> superInterfaceNames = Maps.newHashMap();
for (AnalyzedClass analyzedClass : superAnalyzedClasses) {
if (analyzedClass.isInterface()) {
superInterfaceNames.put(analyzedClass.name(), analyzedClass.interfaceNames());
}
}
Map interfaceNamesToInstrument = Maps.newHashMap();
for (String ejbRemoteInterface : ejbRemoteInterfaces) {
addToInterfaceNamesToInstrument(ejbRemoteInterface, superInterfaceNames,
interfaceNamesToInstrument, ejbRemoteInterface);
}
List instrumentationConfigs = Lists.newArrayList();
for (Map.Entry entry : interfaceNamesToInstrument.entrySet()) {
String shortClassName = entry.getValue();
int index = shortClassName.lastIndexOf('.');
if (index != -1) {
shortClassName = shortClassName.substring(index + 1);
}
index = shortClassName.lastIndexOf('$');
if (index != -1) {
shortClassName = shortClassName.substring(index + 1);
}
instrumentationConfigs.add(ImmutableInstrumentationConfig.builder()
.className(entry.getKey())
.subTypeRestriction(ClassNames.fromInternalName(thinClass.name()))
.methodName("*")
.addMethodParameterTypes("..")
.captureKind(CaptureKind.TRANSACTION)
.transactionType("Background")
.transactionNameTemplate("EJB remote: " + shortClassName + "#{{methodName}}")
.traceEntryMessageTemplate(
"EJB remote: " + entry.getValue() + ".{{methodName}}()")
.timerName("ejb remote")
.build());
}
ImmutableMap newAdvisors =
AdviceGenerator.createAdvisors(instrumentationConfigs, null, false);
try {
ClassLoaders.defineClassesInClassLoader(newAdvisors.values(), loader);
} catch (Exception e) {
logger.error(e.getMessage(), e);
}
Map newAdvisorsByRemoteInterface = Maps.newHashMap();
for (Advice advice : newAdvisors.keySet()) {
newAdvisorsByRemoteInterface.put(advice.pointcut().className(), advice);
}
List ejbHackedSuperAnalyzedClasses = Lists.newArrayList();
for (AnalyzedClass superAnalyzedClass : superAnalyzedClasses) {
Advice advice = newAdvisorsByRemoteInterface.get(superAnalyzedClass.name());
if (advice == null) {
ejbHackedSuperAnalyzedClasses.add(superAnalyzedClass);
} else {
ImmutableAnalyzedClass.Builder builder = ImmutableAnalyzedClass.builder()
.copyFrom(superAnalyzedClass);
List analyzedMethods = Lists.newArrayList();
for (AnalyzedMethod analyzedMethod : superAnalyzedClass.analyzedMethods()) {
analyzedMethods.add(ImmutableAnalyzedMethod.builder()
.copyFrom(analyzedMethod)
.addSubTypeRestrictedAdvisors(advice)
.build());
}
builder.analyzedMethods(analyzedMethods);
ejbHackedSuperAnalyzedClasses.add(builder.build());
}
}
return ejbHackedSuperAnalyzedClasses;
}
private static void addToInterfaceNamesToInstrument(String interfaceName,
Map> superInterfaceNamesMap,
Map interfaceNamesToInstrument, String ejbRemoteInterface) {
interfaceNamesToInstrument.put(interfaceName, ejbRemoteInterface);
List superInterfaceNames = superInterfaceNamesMap.get(interfaceName);
if (superInterfaceNames != null) {
for (String superInterfaceName : superInterfaceNames) {
addToInterfaceNamesToInstrument(superInterfaceName, superInterfaceNamesMap,
interfaceNamesToInstrument, ejbRemoteInterface);
}
}
}
private static boolean hasSuperAdvice(List superAnalyzedClasses) {
for (AnalyzedClass superAnalyzedClass : superAnalyzedClasses) {
if (!superAnalyzedClass.analyzedMethods().isEmpty()) {
return true;
}
}
return false;
}
private static boolean isSubTypeRestrictionMatch(Advice advice, Set superClassNames) {
Pattern pointcutSubTypeRestrictionPattern = advice.pointcutSubTypeRestrictionPattern();
if (pointcutSubTypeRestrictionPattern == null) {
String subTypeRestriction = advice.pointcut().subTypeRestriction();
return subTypeRestriction.isEmpty() || superClassNames.contains(subTypeRestriction);
}
for (String superClassName : superClassNames) {
if (pointcutSubTypeRestrictionPattern.matcher(superClassName).matches()) {
return true;
}
}
return false;
}
static List sortAdvisors(Collection matchingAdvisors) {
switch (matchingAdvisors.size()) {
case 0:
return ImmutableList.of();
case 1:
return Lists.newArrayList(matchingAdvisors);
default:
return Advice.ordering.sortedCopy(matchingAdvisors);
}
}
// AnalyzedMethod equivalence defined only in terms of method name and parameter types
// so that overridden methods will be equivalent
@Value.Immutable
abstract static class AnalyzedMethodKey {
abstract String name();
abstract ImmutableList parameterTypes();
// null is only used when constructing key purely for lookup
@Value.Auxiliary
abstract @Nullable AnalyzedMethod analyzedMethod();
private static AnalyzedMethodKey wrap(AnalyzedMethod analyzedMethod) {
return ImmutableAnalyzedMethodKey.builder()
.name(analyzedMethod.name())
.addAllParameterTypes(analyzedMethod.parameterTypes())
.analyzedMethod(analyzedMethod)
.build();
}
}
private static class BridgeMethodClassVisitor extends ClassVisitor {
private final Map bridgeTargetMethods = Maps.newHashMap();
private BridgeMethodClassVisitor() {
super(ASM6);
}
public Map getBridgeTargetMethods() {
return bridgeTargetMethods;
}
@Override
public @Nullable MethodVisitor visitMethod(int access, String name, String desc,
@Nullable String signature, String /*@Nullable*/ [] exceptions) {
if ((access & ACC_BRIDGE) == 0) {
return null;
}
return new BridgeMethodVisitor(name, desc);
}
private class BridgeMethodVisitor extends MethodVisitor {
private String bridgeMethodName;
private String bridgeMethodDesc;
private int bridgeMethodParamCount;
private boolean found;
private BridgeMethodVisitor(String bridgeMethodName, String bridgeMethodDesc) {
super(ASM6);
this.bridgeMethodName = bridgeMethodName;
this.bridgeMethodDesc = bridgeMethodDesc;
bridgeMethodParamCount = Type.getArgumentTypes(bridgeMethodDesc).length;
}
@Override
public void visitMethodInsn(int opcode, String owner, String name, String desc,
boolean itf) {
if (found) {
return;
}
if (!name.equals(bridgeMethodName)) {
return;
}
if (Type.getArgumentTypes(desc).length != bridgeMethodParamCount) {
return;
}
if (opcode == INVOKESPECIAL) {
// this is a generated bridge method that just calls super, see
// WeaverTest.shouldHandleBridgeCallingSuper(), presumably the super method
// already matches advice
return;
}
bridgeTargetMethods.put(this.bridgeMethodName + this.bridgeMethodDesc, name + desc);
found = true;
}
}
}
}