/*
* Copyright 2015-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.weaving;
import java.util.List;
import javax.annotation.Nullable;
import org.immutables.value.Value;
import org.objectweb.asm.AnnotationVisitor;
import org.objectweb.asm.ClassVisitor;
import org.objectweb.asm.MethodVisitor;
import static com.google.common.base.Preconditions.checkNotNull;
import static org.objectweb.asm.Opcodes.ACC_BRIDGE;
import static org.objectweb.asm.Opcodes.ASM5;
class ThinClassVisitor extends ClassVisitor {
private final ImmutableThinClass.Builder thinClassBuilder = ImmutableThinClass.builder();
private @Nullable ThinClass thinClass;
ThinClassVisitor() {
super(ASM5);
}
@Override
public void visit(int version, int access, String name, @Nullable String signature,
@Nullable String superName, String/*@Nullable*/[] interfaces) {
thinClassBuilder.access(access);
thinClassBuilder.name(name);
thinClassBuilder.superName(superName);
if (interfaces != null) {
thinClassBuilder.addInterfaces(interfaces);
}
}
@Override
public @Nullable AnnotationVisitor visitAnnotation(String desc, boolean visible) {
thinClassBuilder.addAnnotations(desc);
return null;
}
@Override
public MethodVisitor visitMethod(int access, String name, String desc,
@Nullable String signature, String /*@Nullable*/[] exceptions) {
ImmutableThinMethod.Builder thinMethodBuilder = ImmutableThinMethod.builder();
thinMethodBuilder.access(access);
thinMethodBuilder.name(name);
thinMethodBuilder.desc(desc);
thinMethodBuilder.signature(signature);
if (exceptions != null) {
thinMethodBuilder.addExceptions(exceptions);
}
return new AnnotationCaptureMethodVisitor(thinMethodBuilder);
}
@Override
public void visitEnd() {
thinClass = thinClassBuilder.build();
}
ThinClass getThinClass() {
return checkNotNull(thinClass);
}
@Value.Immutable
interface ThinClass {
int access();
String name();
@Nullable
String superName();
List<String> interfaces();
List<String> annotations();
List<ThinMethod> nonBridgeMethods();
List<ThinMethod> bridgeMethods();
}
@Value.Immutable
interface ThinMethod {
int access();
String name();
String desc();
@Nullable
String signature();
List<String> exceptions();
List<String> annotations();
}
private class AnnotationCaptureMethodVisitor extends MethodVisitor {
private final ImmutableThinMethod.Builder thinMethodBuilder;
private AnnotationCaptureMethodVisitor(ImmutableThinMethod.Builder thinMethodBuilder) {
super(ASM5);
this.thinMethodBuilder = thinMethodBuilder;
}
@Override
public @Nullable AnnotationVisitor visitAnnotation(String desc, boolean visible) {
thinMethodBuilder.addAnnotations(desc);
return null;
}
@Override
public void visitEnd() {
ThinMethod thinMethod = thinMethodBuilder.build();
if ((thinMethod.access() & ACC_BRIDGE) != 0) {
thinClassBuilder.addBridgeMethods(thinMethod);
} else {
thinClassBuilder.addNonBridgeMethods(thinMethod);
}
}
}
}