Java tutorial
/* * Copyright 2015 the original author or authors. * @https://github.com/scouter-project/scouter * * 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 scouter.agent; import org.objectweb.asm.AnnotationVisitor; import org.objectweb.asm.ClassReader; import org.objectweb.asm.ClassVisitor; import org.objectweb.asm.ClassWriter; import org.objectweb.asm.Opcodes; import scouter.agent.asm.*; import scouter.agent.asm.asyncsupport.AsyncContextDispatchASM; import scouter.agent.asm.asyncsupport.CallRunnableASM; import scouter.agent.asm.asyncsupport.HystrixCommandASM; import scouter.agent.asm.asyncsupport.RequestStartAsyncASM; import scouter.agent.asm.asyncsupport.executor.ExecutorServiceASM; import scouter.agent.asm.asyncsupport.spring.SpringAsyncExecutionASM; import scouter.agent.asm.asyncsupport.spring.SpringAsyncExecutionAspectSupportDoSubmitASM; import scouter.agent.asm.kafka.KafkaProducerASM; import scouter.agent.asm.rabbit.RabbitPublisherASM; import scouter.agent.asm.redis.JedisCommandASM; import scouter.agent.asm.redis.JedisProtocolASM; import scouter.agent.asm.redis.LettuceASM; import scouter.agent.asm.redis.RedisCacheKeyASM; import scouter.agent.asm.redis.RedisKeyASM; import scouter.agent.asm.util.AsmUtil; import scouter.agent.util.AsyncRunner; import scouter.lang.conf.ConfObserver; import scouter.util.FileUtil; import scouter.util.IntSet; import java.io.IOException; import java.io.InputStream; import java.lang.instrument.ClassFileTransformer; import java.lang.instrument.IllegalClassFormatException; import java.security.ProtectionDomain; import java.util.ArrayList; import java.util.List; public class AgentTransformer implements ClassFileTransformer { public static final String JAVA_UTIL_MAP = "java/util/Map"; public static ThreadLocal<ClassLoader> hookingCtx = new ThreadLocal<ClassLoader>(); protected static List<IASM> asms = new ArrayList<IASM>(); // hook ? ? ?? ?. private static int hook_signature; static { final Configure conf = Configure.getInstance(); reload(); hook_signature = conf.getHookSignature(); ConfObserver.add("AgentTransformer", new Runnable() { public void run() { if (conf.getHookSignature() != hook_signature) { reload(); } hook_signature = conf.getHookSignature(); } }); } public static void reload() { Configure conf = Configure.getInstance(); List<IASM> temp = new ArrayList<IASM>(); temp.add(new HttpServiceASM()); temp.add(new ServiceASM()); temp.add(new RequestStartAsyncASM()); temp.add(new AsyncContextDispatchASM()); temp.add(new JDBCPreparedStatementASM()); temp.add(new JDBCResultSetASM()); temp.add(new JDBCStatementASM()); temp.add(new SqlMapASM()); temp.add(new UserTxASM()); temp.add(new JDBCGetConnectionASM()); temp.add(new JDBCConnectionOpenASM()); temp.add(new JDBCDriverASM()); temp.add(new InitialContextASM()); temp.add(new CapArgsASM()); temp.add(new CapReturnASM()); temp.add(new CapThisASM()); temp.add(new MethodASM()); temp.add(new ApicallASM()); temp.add(new ApicallInfoASM()); temp.add(new ApiCallResponseObjectASM()); temp.add(new ApicallSpringHandleResponseASM()); temp.add(new ApicallSpringHttpAccessorASM()); temp.add(new ApicallJavaHttpRequestASM()); temp.add(new SpringAsyncExecutionASM()); temp.add(new SpringAsyncExecutionAspectSupportDoSubmitASM()); temp.add(new CallRunnableASM()); temp.add(new ExecutorServiceASM()); temp.add(new JedisCommandASM()); temp.add(new RedisKeyASM()); temp.add(new RedisCacheKeyASM()); temp.add(new JedisProtocolASM()); temp.add(new LettuceASM()); temp.add(new KafkaProducerASM()); temp.add(new RabbitPublisherASM()); temp.add(new SpringReqMapASM()); temp.add(new HystrixCommandASM()); temp.add(new SocketASM()); temp.add(new JspServletASM()); temp.add(new MapImplASM()); temp.add(new UserExceptionASM()); temp.add(new UserExceptionHandlerASM()); temp.add(new AddFieldASM()); asms = temp; } // ////////////////////////////////////////////////////////////// // boot class? Hooking? ? ?. private static IntSet asynchook = new IntSet(); static { asynchook.add("sun/net/www/protocol/http/HttpURLConnection".hashCode()); asynchook.add("sun/net/www/http/HttpClient".hashCode()); asynchook.add("java/net/Socket".hashCode()); asynchook.add("java/nio/channels/SocketChannel".hashCode()); asynchook.add("sun/nio/ch/SocketChannelImpl".hashCode()); asynchook.add("javax/naming/InitialContext".hashCode()); asynchook.add("java/util/concurrent/AbstractExecutorService".hashCode()); asynchook.add("java/util/concurrent/ThreadPoolExecutor".hashCode()); } private Configure conf = Configure.getInstance(); private Logger.FileLog bciOut; public byte[] transform(final ClassLoader loader, String className, final Class classBeingRedefined, ProtectionDomain protectionDomain, byte[] classfileBuffer) throws IllegalClassFormatException { try { hookingCtx.set(loader); // if(className != null && (className.indexOf("http") >= 0 || className.indexOf("Http") >= 0)) { // System.out.println("[!!!!!!!!] loading ...http className = " + className); // } if (className == null) return null; if (classBeingRedefined == null) { if (asynchook.contains(className.hashCode())) { AsyncRunner.getInstance().add(loader, className, classfileBuffer); return null; } if (loader == null) { if (conf._hook_boot_prefix == null || conf._hook_boot_prefix.length() == 0 || false == className.startsWith(conf._hook_boot_prefix)) { return null; } } } if (className.startsWith("scouter/")) { return null; } // classfileBuffer = DirectPatch.patch(className, classfileBuffer); ObjTypeDetector.check(className); final ClassDesc classDesc = new ClassDesc(); ClassReader cr = new ClassReader(classfileBuffer); cr.accept(new ClassVisitor(Opcodes.ASM7) { public void visit(int version, int access, String name, String signature, String superName, String[] interfaces) { classDesc.set(version, access, name, signature, superName, interfaces); if (conf._hook_map_impl_enabled) { classDesc.isMapImpl = isMapImpl(superName, interfaces, loader); } super.visit(version, access, name, signature, superName, interfaces); } @Override public AnnotationVisitor visitAnnotation(String desc, boolean visible) { classDesc.anotation += desc; return super.visitAnnotation(desc, visible); } }, 0); if (AsmUtil.isInterface(classDesc.access)) { return null; } classDesc.classBeingRedefined = classBeingRedefined; ClassWriter cw = getClassWriter(classDesc); ClassVisitor cv = cw; List<IASM> workAsms = asms; for (int i = workAsms.size() - 1; i >= 0; i--) { cv = workAsms.get(i).transform(cv, className, classDesc); if (cv != cw) { cr = new ClassReader(classfileBuffer); cr.accept(cv, ClassReader.EXPAND_FRAMES); classfileBuffer = cw.toByteArray(); cv = cw = getClassWriter(classDesc); if (conf._log_asm_enabled) { if (this.bciOut == null) { this.bciOut = new Logger.FileLog("./scouter.bci"); } this.bciOut.println(className + "\t\t[" + loader + "]"); } } } return classfileBuffer; } catch (Throwable t) { Logger.println("A101", "Transformer Error : " + className, t); t.printStackTrace(); } finally { hookingCtx.set(null); } return null; } private boolean isMapImpl(String superName, String[] interfaces, ClassLoader loader) { String[] classes = new String[interfaces.length + 1]; System.arraycopy(interfaces, 0, classes, 0, interfaces.length); classes[classes.length - 1] = superName; for (int i = 0; i < classes.length; i++) { if (isMapImpl(classes[i], loader)) { return true; } } return false; } private boolean isMapImpl(String clazz, ClassLoader loader) { if ("java/lang/Object".equals(clazz)) { return false; } if (JAVA_UTIL_MAP.equals(clazz)) { return true; } if (loader == null) { loader = ClassLoader.getSystemClassLoader(); } InputStream in = loader.getResourceAsStream(clazz + ".class"); try { ClassReader classReader = new ClassReader(in); String[] interfaces = classReader.getInterfaces(); if (interfaces != null && interfaces.length > 0) { for (int i = 0; i < interfaces.length; i++) { if (JAVA_UTIL_MAP.equals(interfaces[i])) { return true; } isMapImpl(interfaces[i], loader); } } String superClassName = classReader.getSuperName(); if (superClassName == null) { return false; } if (JAVA_UTIL_MAP.equals(superClassName)) { return true; } isMapImpl(superClassName, loader); } catch (IOException e) { System.out.println("[A189]-isMapImpl-check super class : " + clazz); } return false; } private ClassWriter getClassWriter(final ClassDesc classDesc) { ClassWriter cw; switch (classDesc.version) { case Opcodes.V1_1: case Opcodes.V1_2: case Opcodes.V1_3: case Opcodes.V1_4: case Opcodes.V1_5: case Opcodes.V1_6: cw = new ScouterClassWriter(ClassWriter.COMPUTE_MAXS); break; default: cw = new ScouterClassWriter(ClassWriter.COMPUTE_FRAMES | ClassWriter.COMPUTE_MAXS); } return cw; } private void dump(String className, byte[] bytes) { String fname = "/tmp/" + className.replace('/', '_'); FileUtil.save(fname, bytes); } }