Instrument-Demo

 

package com.sean.source;

public class TransClass {
 public int getNumber() {
  return 1;
 }
}

import com.sean.source.TransClass;

/**
 *
 */

/**
 * @author Sean_Zhang
 *
 */
public class TransformMain {
 public static void main(String[] args) {
  TransClass tc = new TransClass();
  System.out.println(tc.getNumber());
 }
}

 

Instrument:

import java.lang.instrument.ClassDefinition;
import java.lang.instrument.Instrumentation;
import java.lang.instrument.UnmodifiableClassException;

import com.sean.instrument.Transformer;

/**
 * @author Sean_Zhang
 *
 */
public class AgentMain {
 public static void premain(String agentArgs, Instrumentation inst)
   throws ClassNotFoundException, UnmodifiableClassException,
   InterruptedException {
//  ClassDefinition def = new ClassDefinition(com.sean.source.TransClass.class,new Transformer().getBytesFromFile(Transformer.TRANSFORM_CLASS_NAME));
//        inst.redefineClasses(new ClassDefinition[] { def });
//        System.out.println("success");
  inst.addTransformer(new Transformer());
//  inst.retransformClasses(com.sean.source.TransClass.class);
 }
}

/**
 *
 */
package com.sean.agent;

import java.lang.instrument.Instrumentation;
import java.lang.instrument.UnmodifiableClassException;

import com.sean.instrument.Transformer;

/**
 * @author Sean_Zhang
 *
 */
public class AgentMain {
 public static void premain(String agentArgs, Instrumentation inst)
   throws ClassNotFoundException, UnmodifiableClassException,
   InterruptedException {
  inst.addTransformer(new Transformer());
//  inst.retransformClasses(com.sean.source.TransClass.class);
 }
}

/**
 *
 */
package com.sean.instrument;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.lang.instrument.ClassFileTransformer;
import java.lang.instrument.IllegalClassFormatException;
import java.security.ProtectionDomain;

/**
 * @author Sean_Zhang
 *
 */
public class Transformer implements ClassFileTransformer {
 public static final String TRANSFORM_CLASS_NAME = "com/sean/source/TransClass.class";

 public  byte[] getBytesFromFile(String fileName) {
  System.out.println("className:" + fileName);
        try {
         // precondition
         
            File file = new File(fileName);
            InputStream is = new FileInputStream(file);
            long length = file.length();
            byte[] bytes = new byte[(int) length];
            // Read in the bytes
            int offset = 0;
            int numRead = 0;
            while (offset <bytes.length && (numRead = is.read(bytes, offset, bytes.length - offset)) >= 0) {
                offset += numRead;
            }

            if (offset < bytes.length) {
                throw new IOException("Could not completely read file "
                        + fileName);
            }
            is.close();
            return bytes;
        } catch (Exception e) {
            System.out.println("error occurs in _ClassTransformer!"
                    + e.getClass().getName());
            return null;
        }
 }
 
 
 public byte[] transform(ClassLoader loader, String className,
   Class<?> classBeingRedefined, ProtectionDomain protectionDomain,
   byte[] classfileBuffer) throws IllegalClassFormatException {
  if(className.contains("com/sean/source/TransClass")) {
   return getBytesFromFile(TRANSFORM_CLASS_NAME);
  }
  return null;
 }
 
 public static void main(String[] args) {
  Transformer t = new Transformer();
  byte[] classfileBuffer = t.getBytesFromFile(TRANSFORM_CLASS_NAME);
 }
}